chore: adds check for riscv64 and riscv32 archs (#2993)

AFLplusplus · Feb 17, 2025 · 94d6c51 · 94d6c51
1 parent 9a2a42c
commit 94d6c51
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/libafl_qemu/runtime/libafl_qemu_arch.h b/libafl_qemu/runtime/libafl_qemu_arch.h
@@ -30,12 +30,12 @@
       // Linux kernel
       #include <asm-generic/int-ll64.h>
 
-      #if defined(__x86_64__) || defined(__aarch64__)
+      #if defined(__x86_64__) || defined(__aarch64__) || (defined(__riscv) && __riscv_xlen == 64)
         typedef __u64 libafl_word;
         #define LIBAFL_CALLING_CONVENTION __attribute__(())
       #endif
 
-      #ifdef __arm__
+      #if defined(__arm__) || (defined(__riscv) && __riscv_xlen == 32)
         typedef __u32 libafl_word;
         #define LIBAFL_CALLING_CONVENTION __attribute__(())
       #endif
@@ -47,12 +47,12 @@
 
       #define noinline __attribute__((noinline))
 
-      #if defined(__x86_64__) || defined(__aarch64__)
+      #if defined(__x86_64__) || defined(__aarch64__) || (defined(__riscv) && __riscv_xlen == 64)
         typedef uint64_t libafl_word;
         #define LIBAFL_CALLING_CONVENTION __attribute__(())
       #endif
 
-      #ifdef __arm__
+      #if defined(__arm__) || (defined(__riscv) && __riscv_xlen == 32)
         typedef uint32_t libafl_word;
         #define LIBAFL_CALLING_CONVENTION __attribute__(())
       #endif
@@ -66,12 +66,12 @@
 
     #define noinline __attribute__((noinline))
 
-    #if defined(__x86_64__) || defined(__aarch64__)
+    #if defined(__x86_64__) || defined(__aarch64__) || (defined(__riscv) && __riscv_xlen == 64)
       typedef uint64_t libafl_word;
       #define LIBAFL_CALLING_CONVENTION __attribute__(())
     #endif
 
-    #ifdef __arm__
+    #if defined(__arm__) || (defined(__riscv) && __riscv_xlen == 32)
       typedef uint32_t libafl_word;
       #define LIBAFL_CALLING_CONVENTION __attribute__(())
     #endif