[WinEH] Add some support for code generating catchpad

We can now run 32-bit programs with empty catch bodies.  The next step
is to change PEI so that we get funclet prologues and epilogues.

llvm-svn: 246235

1 files changed, 10 insertions, 0 deletions

diff --git a/llvm/lib/Target/X86/X86WinEHState.cpp b/llvm/lib/Target/X86/X86WinEHState.cpp
index d039a0d4ca6..19f574ad746 100644
--- a/llvm/lib/Target/X86/X86WinEHState.cpp
+++ b/llvm/lib/Target/X86/X86WinEHState.cpp
@@ -471,6 +471,8 @@ int WinEHStatePass::escapeRegNode(Function &F) {
 void WinEHStatePass::addCXXStateStoresToFunclet(Value *ParentRegNode,
                                                 WinEHFuncInfo &FuncInfo,
                                                 Function &F, int BaseState) {
+  Function *RestoreFrame =
+      Intrinsic::getDeclaration(TheModule, Intrinsic::x86_seh_restoreframe);
   // Iterate all the instructions and emit state number stores.
   for (BasicBlock &BB : F) {
     for (Instruction &I : BB) {
@@ -489,6 +491,14 @@ void WinEHStatePass::addCXXStateStoresToFunclet(Value *ParentRegNode,
         insertStateNumberStore(ParentRegNode, II, State);
       }
     }
+
+    // Insert calls to llvm.x86.seh.restoreframe at catchret destinations.
+    if (auto *CR = dyn_cast<CatchReturnInst>(BB.getTerminator())) {
+      //llvm::errs() << "BB: " << BB << '\n';
+      //llvm::errs() << "CR->getSuccessor(): " << *CR->getSuccessor() << '\n';
+      IRBuilder<> Builder(CR->getSuccessor()->begin());
+      Builder.CreateCall(RestoreFrame, {});
+    }
   }
 }