about summary refs log tree commit diff
diff options
context:
space:
mode:
authorvanhauser-thc <vh@thc.org>2024-07-15 23:42:55 +0200
committervanhauser-thc <vh@thc.org>2024-07-15 23:42:55 +0200
commiteac53afe7b99bdb2729ca01791b4bac92a7ae4d9 (patch)
treef618652624b865071af65f0202b61354854f2d1f
parent55a2362348cd467b65d6aea33e93ec44c6de1a38 (diff)
downloadafl++-eac53afe7b99bdb2729ca01791b4bac92a7ae4d9.tar.gz
fix autodict
-rw-r--r--instrumentation/afl-llvm-dict2file.so.cc7
1 files changed, 7 insertions, 0 deletions
diff --git a/instrumentation/afl-llvm-dict2file.so.cc b/instrumentation/afl-llvm-dict2file.so.cc
index b93f61f0..6559bc84 100644
--- a/instrumentation/afl-llvm-dict2file.so.cc
+++ b/instrumentation/afl-llvm-dict2file.so.cc
@@ -661,6 +661,13 @@ bool AFLdict2filePass::runOnModule(Module &M) {
             Value       *op2 = callInst->getArgOperand(2);
             ConstantInt *ilen = dyn_cast<ConstantInt>(op2);
 
+            if (!ilen) {
+
+              op2 = callInst->getArgOperand(1);
+              ilen = dyn_cast<ConstantInt>(op2);
+
+            }
+
             if (ilen) {
 
               uint64_t literalLength = optLen;