From 0cedc8014bed28e0f2ae041373d1b57271d0e6f8 Mon Sep 17 00:00:00 2001 From: Sebastian Ă–sterlund Date: Fri, 29 May 2020 10:51:34 +0200 Subject: Support filename pattern matching in whitelist Allow the whitelist specified by AFL_LLVM_WHITELIST contain entries with UNIX shell-style wildcard pattern matching. --- llvm_mode/afl-llvm-common.cc | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'llvm_mode/afl-llvm-common.cc') diff --git a/llvm_mode/afl-llvm-common.cc b/llvm_mode/afl-llvm-common.cc index 35eabbf0..e97423a0 100644 --- a/llvm_mode/afl-llvm-common.cc +++ b/llvm_mode/afl-llvm-common.cc @@ -7,6 +7,7 @@ #include #include #include +#include #include #include @@ -155,9 +156,8 @@ bool isInWhitelist(llvm::Function *F) { * specified in the list. */ if (instFilename.str().length() >= it->length()) { - if (instFilename.str().compare( - instFilename.str().length() - it->length(), it->length(), - *it) == 0) { + if (fnmatch((*it).c_str(), + instFilename.str().c_str(), FNM_PATHNAME) == 0) { return true; -- cgit 1.4.1