about summary refs log tree commit diff
path: root/llvm_mode/afl-clang-fast.c
diff options
context:
space:
mode:
authorvan Hauser <vh@thc.org>2020-04-12 10:58:58 +0200
committervan Hauser <vh@thc.org>2020-04-12 10:58:58 +0200
commit5e09e13ece5f074765df65bdec0482509bc23637 (patch)
treebdb42d0d5d2d541ca6563a81284243c480d801a8 /llvm_mode/afl-clang-fast.c
parenteec725a345b2e1cf396fd96970333677f701e42e (diff)
downloadafl++-5e09e13ece5f074765df65bdec0482509bc23637.tar.gz
m32/m64 support for LTO and code-format for test/
Diffstat (limited to 'llvm_mode/afl-clang-fast.c')
-rw-r--r--llvm_mode/afl-clang-fast.c15
1 files changed, 9 insertions, 6 deletions
diff --git a/llvm_mode/afl-clang-fast.c b/llvm_mode/afl-clang-fast.c
index fa76a11e..2114ccf3 100644
--- a/llvm_mode/afl-clang-fast.c
+++ b/llvm_mode/afl-clang-fast.c
@@ -479,18 +479,19 @@ static void edit_params(u32 argc, char **argv, char **envp) {
 
   }
 
-  if (instrument_mode == INSTRUMENT_LTO)
-    cc_params[cc_par_cnt++] = alloc_printf("%s/afl-llvm-rt-lto.o", obj_path);
-
 #ifndef __ANDROID__
   switch (bit_mode) {
 
     case 0:
-      cc_params[cc_par_cnt++] = alloc_printf("%s/afl-llvm-rt.o", obj_path);
+      cc_params[cc_par_cnt++] =
+          alloc_printf("%s/afl-llvm-rt%s.o", obj_path,
+                       instrument_mode == INSTRUMENT_LTO ? "-lto" : "");
       break;
 
     case 32:
-      cc_params[cc_par_cnt++] = alloc_printf("%s/afl-llvm-rt-32.o", obj_path);
+      cc_params[cc_par_cnt++] =
+          alloc_printf("%s/afl-llvm-rt%s-32.o", obj_path,
+                       instrument_mode == INSTRUMENT_LTO ? "-lto" : "");
 
       if (access(cc_params[cc_par_cnt - 1], R_OK))
         FATAL("-m32 is not supported by your compiler");
@@ -498,7 +499,9 @@ static void edit_params(u32 argc, char **argv, char **envp) {
       break;
 
     case 64:
-      cc_params[cc_par_cnt++] = alloc_printf("%s/afl-llvm-rt-64.o", obj_path);
+      cc_params[cc_par_cnt++] =
+          alloc_printf("%s/afl-llvm-rt%s-64.o", obj_path,
+                       instrument_mode == INSTRUMENT_LTO ? "-lto" : "");
 
       if (access(cc_params[cc_par_cnt - 1], R_OK))
         FATAL("-m64 is not supported by your compiler");