From 48f7f7a17b8ac848e261cfc47800e1e6d7a51bcb Mon Sep 17 00:00:00 2001 From: Rick van Schijndel Date: Wed, 15 Apr 2020 12:08:33 +0200 Subject: afl-clang-fast: fail when binary name can't be used to determine build mode (#318) --- llvm_mode/afl-clang-fast.c | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) (limited to 'llvm_mode/afl-clang-fast.c') diff --git a/llvm_mode/afl-clang-fast.c b/llvm_mode/afl-clang-fast.c index 49318f2e..f58c22dd 100644 --- a/llvm_mode/afl-clang-fast.c +++ b/llvm_mode/afl-clang-fast.c @@ -184,7 +184,7 @@ static void edit_params(u32 argc, char **argv, char **envp) { sprintf(llvm_fullpath, CLANGPP_BIN); cc_params[0] = alt_cxx && *alt_cxx ? alt_cxx : (u8 *)llvm_fullpath; - } else { + } else if (!strcmp(name, "afl-clang-fast") || !strcmp(name, "afl-clang-lto")) { u8 *alt_cc = getenv("AFL_CC"); if (USE_BINDIR) @@ -193,6 +193,9 @@ static void edit_params(u32 argc, char **argv, char **envp) { sprintf(llvm_fullpath, CLANG_BIN); cc_params[0] = alt_cc && *alt_cc ? alt_cc : (u8 *)llvm_fullpath; + } else { + fprintf(stderr, "Name of the binary: %s\n", argv[0]); + FATAL("Name of the binary is not a known name, expected afl-clang-fast(++) or afl-clang-lto(++)"); } /* There are three ways to compile with afl-clang-fast. In the traditional -- cgit 1.4.1