From 5f715522f1830c5555696b867fccd1ddfd1b4640 Mon Sep 17 00:00:00 2001 From: Andrew Kaylor Date: Thu, 23 Apr 2015 18:37:39 +0000 Subject: [PATCH] [WinEH] Handle stubs for outlined functions that have only unreached terminators. llvm-svn: 235618 --- llvm/lib/CodeGen/WinEHPrepare.cpp | 25 ++++++++++++++++--------- 1 file changed, 16 insertions(+), 9 deletions(-) diff --git a/llvm/lib/CodeGen/WinEHPrepare.cpp b/llvm/lib/CodeGen/WinEHPrepare.cpp index 57671f0..03f53cd 100644 --- a/llvm/lib/CodeGen/WinEHPrepare.cpp +++ b/llvm/lib/CodeGen/WinEHPrepare.cpp @@ -1037,6 +1037,7 @@ static BasicBlock *createStubLandingPad(Function *Handler, void WinEHPrepare::addStubInvokeToHandlerIfNeeded(Function *Handler, Value *PersonalityFn) { ReturnInst *Ret = nullptr; + UnreachableInst *Unreached = nullptr; for (BasicBlock &BB : *Handler) { TerminatorInst *Terminator = BB.getTerminator(); // If we find an invoke, there is nothing to be done. @@ -1044,18 +1045,24 @@ void WinEHPrepare::addStubInvokeToHandlerIfNeeded(Function *Handler, if (II) return; // If we've already recorded a return instruction, keep looking for invokes. - if (Ret) - continue; - // If we haven't recorded a return instruction yet, try this terminator. - Ret = dyn_cast(Terminator); + if (!Ret) + Ret = dyn_cast(Terminator); + // If we haven't recorded an unreachable instruction, try this terminator. + if (!Unreached) + Unreached = dyn_cast(Terminator); } // If we got this far, the handler contains no invokes. We should have seen - // at least one return. We'll insert an invoke of llvm.donothing ahead of - // that return. - assert(Ret); - BasicBlock *OldRetBB = Ret->getParent(); - BasicBlock *NewRetBB = SplitBlock(OldRetBB, Ret); + // at least one return or unreachable instruction. We'll insert an invoke of + // llvm.donothing ahead of that instruction. + assert(Ret || Unreached); + TerminatorInst *Term; + if (Ret) + Term = Ret; + else + Term = Unreached; + BasicBlock *OldRetBB = Term->getParent(); + BasicBlock *NewRetBB = SplitBlock(OldRetBB, Term); // SplitBlock adds an unconditional branch instruction at the end of the // parent block. We want to replace that with an invoke call, so we can // erase it now. -- 2.7.4