diff --git a/llvm/include/llvm/FuzzMutate/IRMutator.h b/llvm/include/llvm/FuzzMutate/IRMutator.h --- a/llvm/include/llvm/FuzzMutate/IRMutator.h +++ b/llvm/include/llvm/FuzzMutate/IRMutator.h @@ -81,6 +81,7 @@ RandomIRBuilder &IB); public: + InjectorIRStrategy() : Operations(getDefaultOps()) {} InjectorIRStrategy(std::vector &&Operations) : Operations(std::move(Operations)) {} static std::vector getDefaultOps(); diff --git a/llvm/lib/FuzzMutate/IRMutator.cpp b/llvm/lib/FuzzMutate/IRMutator.cpp --- a/llvm/lib/FuzzMutate/IRMutator.cpp +++ b/llvm/lib/FuzzMutate/IRMutator.cpp @@ -45,7 +45,10 @@ } void IRMutationStrategy::mutate(Function &F, RandomIRBuilder &IB) { - mutate(*makeSampler(IB.Rand, make_pointer_range(F)).getSelection(), IB); + auto Range = make_filter_range(make_pointer_range(F), + [](BasicBlock *BB) { return !BB->isEHPad(); }); + + mutate(*makeSampler(IB.Rand, Range).getSelection(), IB); } void IRMutationStrategy::mutate(BasicBlock &BB, RandomIRBuilder &IB) { @@ -566,7 +569,6 @@ } void ShuffleBlockStrategy::mutate(BasicBlock &BB, RandomIRBuilder &IB) { - SmallPtrSet AliveInsts; for (auto &I : make_early_inc_range(make_range( BB.getFirstInsertionPt(), BB.getTerminator()->getIterator()))) { diff --git a/llvm/lib/FuzzMutate/RandomIRBuilder.cpp b/llvm/lib/FuzzMutate/RandomIRBuilder.cpp --- a/llvm/lib/FuzzMutate/RandomIRBuilder.cpp +++ b/llvm/lib/FuzzMutate/RandomIRBuilder.cpp @@ -264,7 +264,7 @@ case Instruction::CallBr: { const CallBase *II = cast(I); const Function *Callee = II->getCalledFunction(); - return !Callee->hasParamAttribute(OperandNo, Attribute::ImmArg); + return Callee && !Callee->hasParamAttribute(OperandNo, Attribute::ImmArg); } default: break; diff --git a/llvm/unittests/FuzzMutate/StrategiesTest.cpp b/llvm/unittests/FuzzMutate/StrategiesTest.cpp --- a/llvm/unittests/FuzzMutate/StrategiesTest.cpp +++ b/llvm/unittests/FuzzMutate/StrategiesTest.cpp @@ -640,4 +640,30 @@ }"; VerifyBlockShuffle(Source); } + +TEST(AllStrategies, SkipEHPad) { + StringRef Source = "\n\ + define void @f(i32 %x) personality ptr @__CxxFrameHandler3 { \n\ + entry: \n\ + invoke void @g() to label %try.cont unwind label %catch.dispatch \n\ + catch.dispatch: \n\ + %0 = catchswitch within none [label %catch] unwind to caller \n\ + catch: \n\ + %1 = catchpad within %0 [ptr null, i32 64, ptr null] \n\ + catchret from %1 to label %try.cont \n\ + try.cont: \n\ + ret void \n\ + } \n\ + declare void @g() \n\ + declare i32 @__CxxFrameHandler3(...) \n\ + "; + + mutateAndVerifyModule(Source); + mutateAndVerifyModule(Source); + mutateAndVerifyModule(Source); + mutateAndVerifyModule(Source); + mutateAndVerifyModule(Source); + mutateAndVerifyModule(Source); + mutateAndVerifyModule(Source); +} } // namespace