diff --git a/llvm/lib/CodeGen/SjLjEHPrepare.cpp b/llvm/lib/CodeGen/SjLjEHPrepare.cpp --- a/llvm/lib/CodeGen/SjLjEHPrepare.cpp +++ b/llvm/lib/CodeGen/SjLjEHPrepare.cpp @@ -472,8 +472,12 @@ // Finally, for any returns from this function, if this function contains an // invoke, add a call to unregister the function context. - for (ReturnInst *Return : Returns) - CallInst::Create(UnregisterFn, FuncCtx, "", Return); + for (ReturnInst *Return : Returns) { + Instruction *InsertPoint = Return; + if (CallInst *CI = Return->getParent()->getTerminatingMustTailCall()) + InsertPoint = CI; + CallInst::Create(UnregisterFn, FuncCtx, "", InsertPoint); + } return true; } diff --git a/llvm/test/CodeGen/X86/sjlj-eh-musttail.ll b/llvm/test/CodeGen/X86/sjlj-eh-musttail.ll new file mode 100644 --- /dev/null +++ b/llvm/test/CodeGen/X86/sjlj-eh-musttail.ll @@ -0,0 +1,30 @@ +; RUN: llc -mtriple i386-windows-gnu -exception-model sjlj -filetype asm -o - %s | FileCheck %s + +declare void @_Z20function_that_throwsv() +declare i32 @__gxx_personality_sj0(...) +declare i8* @__cxa_begin_catch(i8*) +declare void @__cxa_end_catch() +declare void @_callee(); + +define void @_Z8functionv() personality i8* bitcast (i32 (...)* @__gxx_personality_sj0 to i8*) { +entry: + invoke void @_Z20function_that_throwsv() + to label %try.cont unwind label %lpad + +lpad: + %0 = landingpad { i8*, i32 } + catch i8* null + %1 = extractvalue { i8*, i32 } %0, 0 + %2 = tail call i8* @__cxa_begin_catch(i8* %1) + tail call void @__cxa_end_catch() + br label %try.cont + +try.cont: + musttail call void @_callee(); + ret void +} + +; CHECK-LABEL: __Z8functionv: +; CHECK: calll __Unwind_SjLj_Unregister +; CHECK-NOT: {{.*}}: +; CHECK: jmp __callee