Fix autotokens (AFLplusplus#2407)

cube0x8 · Jul 16, 2024 · 074a16f · 074a16f
1 parent 4370a84
commit 074a16f
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/libafl_cc/src/autotokens-pass.cc b/libafl_cc/src/autotokens-pass.cc
@@ -498,6 +498,11 @@ bool AutoTokensPass::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;
               optLen = ilen->getZExtValue();