diff options
| -rw-r--r-- | tools/clang-fuzzer/ExampleClangProtoFuzzer.cpp | 18 |
1 files changed, 17 insertions, 1 deletions
diff --git a/tools/clang-fuzzer/ExampleClangProtoFuzzer.cpp b/tools/clang-fuzzer/ExampleClangProtoFuzzer.cpp index 23123277c3..ab734e85b8 100644 --- a/tools/clang-fuzzer/ExampleClangProtoFuzzer.cpp +++ b/tools/clang-fuzzer/ExampleClangProtoFuzzer.cpp @@ -20,9 +20,25 @@ #include "src/libfuzzer/libfuzzer_macro.h" +#include <cstring> + using namespace clang_fuzzer; +static std::vector<const char *> CLArgs; + +extern "C" int LLVMFuzzerInitialize(int *argc, char ***argv) { + CLArgs.push_back("-O2"); + for (int I = 1; I < *argc; I++) { + if (strcmp((*argv)[I], "-ignore_remaining_args=1") == 0) { + for (I++; I < *argc; I++) + CLArgs.push_back((*argv)[I]); + break; + } + } + return 0; +} + DEFINE_BINARY_PROTO_FUZZER(const Function& input) { auto S = FunctionToString(input); - HandleCXX(S, {"-O2"}); + HandleCXX(S, CLArgs); } |