diff options
| author | Andrea Fioraldi <andreafioraldi@gmail.com> | 2020-02-18 15:27:35 +0100 |
|---|---|---|
| committer | Andrea Fioraldi <andreafioraldi@gmail.com> | 2020-02-18 15:27:35 +0100 |
| commit | a83691d6626a4a3b4881534d5680153a5df2e951 (patch) | |
| tree | 8d8ba3246dc6ee5f5d125bf44f2a1b1bb37d03b9 /llvm_mode/cmplog-routines-pass.cc | |
| parent | 13296af49168c4b63f3d4ea1e31f278317114e5c (diff) | |
| download | afl++-a83691d6626a4a3b4881534d5680153a5df2e951.tar.gz | |
fix insertion of __cmplog_rtn_hook
Diffstat (limited to 'llvm_mode/cmplog-routines-pass.cc')
| -rw-r--r-- | llvm_mode/cmplog-routines-pass.cc | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/llvm_mode/cmplog-routines-pass.cc b/llvm_mode/cmplog-routines-pass.cc index 309ea65e..7e3acae2 100644 --- a/llvm_mode/cmplog-routines-pass.cc +++ b/llvm_mode/cmplog-routines-pass.cc @@ -280,9 +280,8 @@ bool CmpLogRoutines::hookRtns(Module &M) { Value *v1P = callInst->getArgOperand(0), *v2P = callInst->getArgOperand(1); - BasicBlock *bb = callInst->getParent(); - BasicBlock::iterator IP = bb->getFirstInsertionPt(); - IRBuilder<> IRB(&*IP); + IRBuilder<> IRB(callInst->getParent()); + IRB.SetInsertPoint(callInst); std::vector<Value*> args; args.push_back(v1P); @@ -290,7 +289,7 @@ bool CmpLogRoutines::hookRtns(Module &M) { IRB.CreateCall(cmplogHookFn, args, "tmp"); - errs() << callInst->getCalledFunction()->getName() << "\n"; + // errs() << callInst->getCalledFunction()->getName() << "\n"; } |