[llvm] r252670 - [WinEH] Insert the MBB for EH_RESTORE after the catchret

[Reid Kleckner via llvm-commits]

Author: rnk
Date: Tue Nov 10 17:22:20 2015
New Revision: 252670

URL: http://llvm.org/viewvc/llvm-project?rev=252670&view=rev
Log:
[WinEH] Insert the MBB for EH_RESTORE after the catchret

Inserting it before the target block could be bad, we might already have
a fallthrough edge to it.

Added:
    llvm/trunk/test/CodeGen/X86/catchret-fallthrough.ll
Modified:
    llvm/trunk/lib/Target/X86/X86ISelLowering.cpp

Modified: llvm/trunk/lib/Target/X86/X86ISelLowering.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/Target/X86/X86ISelLowering.cpp?rev=252670&r1=252669&r2=252670&view=diff
==============================================================================
--- llvm/trunk/lib/Target/X86/X86ISelLowering.cpp (original)
+++ llvm/trunk/lib/Target/X86/X86ISelLowering.cpp Tue Nov 10 17:22:20 2015
@@ -21416,7 +21416,7 @@ X86TargetLowering::EmitLoweredCatchRet(M
   MachineBasicBlock *RestoreMBB =
       MF->CreateMachineBasicBlock(BB->getBasicBlock());
   assert(BB->succ_size() == 1);
-  MF->insert(TargetMBB->getIterator(), RestoreMBB);
+  MF->insert(std::next(BB->getIterator()), RestoreMBB);
   RestoreMBB->transferSuccessorsAndUpdatePHIs(BB);
   BB->addSuccessor(RestoreMBB);
   MI->getOperand(0).setMBB(RestoreMBB);

Added: llvm/trunk/test/CodeGen/X86/catchret-fallthrough.ll
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/test/CodeGen/X86/catchret-fallthrough.ll?rev=252670&view=auto
==============================================================================
--- llvm/trunk/test/CodeGen/X86/catchret-fallthrough.ll (added)
+++ llvm/trunk/test/CodeGen/X86/catchret-fallthrough.ll Tue Nov 10 17:22:20 2015
@@ -0,0 +1,46 @@
+; RUN: llc -verify-machineinstrs < %s | FileCheck %s
+
+; We used to have an issue where we inserted an MBB between invoke.cont.3 and
+; its fallthrough target of ret void.
+
+target datalayout = "e-m:x-p:32:32-i64:64-f80:32-n8:16:32-a:0:32-S32"
+target triple = "i386-pc-windows-msvc18.0.0"
+
+ at some_global = global i32 0
+
+declare i32 @__CxxFrameHandler3(...)
+
+declare void @g()
+
+define void @f() personality i32 (...)* @__CxxFrameHandler3 {
+entry:
+  invoke void @g()
+          to label %invoke.cont.3 unwind label %catch.dispatch
+
+catch.dispatch:                                   ; preds = %entry
+  %0 = catchpad [i8* null, i32 64, i8* null]
+          to label %catch unwind label %catchendblock
+
+catch:                                            ; preds = %catch.dispatch
+  catchret %0 to label %nrvo.skipdtor
+
+catchendblock:                                    ; preds = %catch, %catch.dispatch
+  catchendpad unwind to caller
+
+invoke.cont.3:                                    ; preds = %entry
+  store i32 123, i32* @some_global
+  br label %nrvo.skipdtor
+
+nrvo.skipdtor:                                    ; preds = %invoke.cont.3, %invoke.cont.4
+  ret void
+}
+
+; CHECK-LABEL: _f: # @f
+; CHECK: calll _g
+; CHECK: movl $123, _some_global
+; CHECK-NOT: jmp
+; CHECK-NOT: movl {{.*}}, %esp
+; CHECK: retl
+; CHECK: movl {{.*}}, %esp
+; CHECK: addl $12, %ebp
+; CHECK: jmp LBB0_{{.*}}