ExternalVendorCode/AFLplusplus
1
0
Fork 0
mirror of https://github.com/AFLplusplus/AFLplusplus.git synced 2025-06-14 02:58:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

blacklist function support for llvm_mode

Browse Source
This commit is contained in:
van Hauser
2020-01-15 10:10:25 +01:00
parent d0ea8f8433
commit d1d5e7c02a
6 changed files with 62 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
llvm_mode/afl-llvm-pass.so.cc
View File

@ -75,6 +75,28 @@ class AFLCoverage : public ModulePass {
}
// ripped from aflgo
static bool isBlacklisted(const Function *F) {
static const SmallVector<std::string, 4> Blacklist = {
"asan.",
"llvm.",
"sancov.",
"__ubsan_handle_",
};
for (auto const &BlacklistFunc : Blacklist) {
if (F->getName().startswith(BlacklistFunc)) { return true; }
}
return false;
}
bool runOnModule(Module &M) override;
// StringRef getPassName() const override {
@ -156,13 +178,11 @@ bool AFLCoverage::runOnModule(Module &M) {
/* Instrument all the things! */
const char *IntrinsicPrefix = "llvm.";
int inst_blocks = 0;
for (auto &F : M) {
auto Fname = F.getName();
if (Fname.startswith(IntrinsicPrefix)) continue;
if (isBlacklisted(&F)) continue;
for (auto &BB : F) {
@ -377,6 +397,7 @@ bool AFLCoverage::runOnModule(Module &M) {
inst_blocks++;
}
}
/* Say something nice. */