diff options
| author | van Hauser <vh@thc.org> | 2023-02-23 11:49:28 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-02-23 11:49:28 +0100 |
| commit | eb5c1ee4f3ffd6e42a3637082ab4ea089052ad8b (patch) | |
| tree | b0ebfc066252a9c78dfe90cf811b2b27f43cc215 /instrumentation/afl-llvm-dict2file.so.cc | |
| parent | a293281b9bbb4e61bfbe5e67ab108d9e5ee40192 (diff) | |
| parent | eeccb2da69d7e6f32ee74c431e7c5053e8379dff (diff) | |
| download | afl++-eb5c1ee4f3ffd6e42a3637082ab4ea089052ad8b.tar.gz | |
Merge pull request #1650 from AFLplusplus/autotoken
Autotoken
Diffstat (limited to 'instrumentation/afl-llvm-dict2file.so.cc')
| -rw-r--r-- | instrumentation/afl-llvm-dict2file.so.cc | 17 |
1 files changed, 14 insertions, 3 deletions
diff --git a/instrumentation/afl-llvm-dict2file.so.cc b/instrumentation/afl-llvm-dict2file.so.cc index bbbbe32c..97f1d47f 100644 --- a/instrumentation/afl-llvm-dict2file.so.cc +++ b/instrumentation/afl-llvm-dict2file.so.cc @@ -182,7 +182,7 @@ bool AFLdict2filePass::runOnModule(Module &M) { DenseMap<Value *, std::string *> valueMap; char *ptr; - int found = 0; + int found = 0, handle_main = 1; /* Show a banner */ setvbuf(stdout, NULL, _IONBF, 0); @@ -192,10 +192,14 @@ bool AFLdict2filePass::runOnModule(Module &M) { SAYF(cCYA "afl-llvm-dict2file" VERSION cRST " by Marc \"vanHauser\" Heuse <mh@mh-sec.de>\n"); - } else + } else { be_quiet = 1; + } + + if (getenv("AFL_LLVM_DICT2FILE_NO_MAIN")) { handle_main = 0; } + scanForDangerousFunctions(&M); ptr = getenv("AFL_LLVM_DICT2FILE"); @@ -210,7 +214,14 @@ bool AFLdict2filePass::runOnModule(Module &M) { for (auto &F : M) { - if (isIgnoreFunction(&F)) continue; + if (!handle_main && + (!F.getName().compare("main") || !F.getName().compare("_main"))) { + + continue; + + } + + if (isIgnoreFunction(&F)) { continue; } if (!isInInstrumentList(&F, MNAME) || !F.size()) { continue; } /* Some implementation notes. |