SjLjEHPrepare.cpp revision 631d11765b9f7ebdb49679c92bd41083fbab3f62
18b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach//===- SjLjEHPass.cpp - Eliminate Invoke & Unwind instructions -----------===// 28b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach// 38b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach// The LLVM Compiler Infrastructure 48b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach// 58b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach// This file is distributed under the University of Illinois Open Source 68b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach// License. See LICENSE.TXT for details. 78b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach// 88b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach//===----------------------------------------------------------------------===// 98b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach// 108b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach// This transformation is designed for use by code generators which use SjLj 118b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach// based exception handling. 128b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach// 138b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach//===----------------------------------------------------------------------===// 148b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 158b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach#define DEBUG_TYPE "sjljehprepare" 168b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach#include "llvm/Transforms/Scalar.h" 178b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach#include "llvm/Constants.h" 188b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach#include "llvm/DerivedTypes.h" 198b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach#include "llvm/Instructions.h" 208b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach#include "llvm/Intrinsics.h" 218b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach#include "llvm/LLVMContext.h" 228b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach#include "llvm/Module.h" 238b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach#include "llvm/Pass.h" 24f788854d20b12c60fd8b43c587adb3227b6b1bffBenjamin Kramer#include "llvm/CodeGen/Passes.h" 252b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling#include "llvm/Target/TargetData.h" 268b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach#include "llvm/Target/TargetLowering.h" 27f788854d20b12c60fd8b43c587adb3227b6b1bffBenjamin Kramer#include "llvm/Transforms/Utils/BasicBlockUtils.h" 28f788854d20b12c60fd8b43c587adb3227b6b1bffBenjamin Kramer#include "llvm/Transforms/Utils/Local.h" 292b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling#include "llvm/Support/CommandLine.h" 30d36b3e36811e8b50476076ee4986068481916f66Bill Wendling#include "llvm/Support/Debug.h" 312b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling#include "llvm/Support/IRBuilder.h" 32aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling#include "llvm/ADT/DenseMap.h" 33d36b3e36811e8b50476076ee4986068481916f66Bill Wendling#include "llvm/ADT/SmallVector.h" 34d36b3e36811e8b50476076ee4986068481916f66Bill Wendling#include "llvm/ADT/Statistic.h" 35f788854d20b12c60fd8b43c587adb3227b6b1bffBenjamin Kramer#include <set> 368b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbachusing namespace llvm; 378b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 382b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendlingstatic cl::opt<bool> DisableOldSjLjEH("disable-old-sjlj-eh", cl::Hidden, 392b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling cl::desc("Disable the old SjLj EH preparation pass")); 402b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 418b818d7e98309125c6058c4ea72a7dc73b031db2Jim GrosbachSTATISTIC(NumInvokes, "Number of invokes replaced"); 428b818d7e98309125c6058c4ea72a7dc73b031db2Jim GrosbachSTATISTIC(NumUnwinds, "Number of unwinds replaced"); 438b818d7e98309125c6058c4ea72a7dc73b031db2Jim GrosbachSTATISTIC(NumSpilled, "Number of registers live across unwind edges"); 448b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 458b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbachnamespace { 466726b6d75a8b679068a58cb954ba97cf9d1690baNick Lewycky class SjLjEHPass : public FunctionPass { 478b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach const TargetLowering *TLI; 48db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *FunctionContextTy; 498b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach Constant *RegisterFn; 508b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach Constant *UnregisterFn; 518b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach Constant *BuiltinSetjmpFn; 528b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach Constant *FrameAddrFn; 530798eddd07b8dc827a4e6e9028c4c3a8d9444286Jim Grosbach Constant *StackAddrFn; 540798eddd07b8dc827a4e6e9028c4c3a8d9444286Jim Grosbach Constant *StackRestoreFn; 558b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach Constant *LSDAAddrFn; 568b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach Value *PersonalityFn; 57b01bbdcc1af27bd90b552bb1b62b48916e0d4be3Duncan Sands Constant *SelectorFn; 588b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach Constant *ExceptionFn; 59ca752c9020a1b1cf151142bd9e0cbca9af12d807Jim Grosbach Constant *CallSiteFn; 60e4ad387a5a88dae20f0f7578e55170bbc8eee2a9Jim Grosbach Constant *DispatchSetupFn; 61cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling Constant *FuncCtxFn; 628b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach Value *CallSite; 63aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling DenseMap<InvokeInst*, BasicBlock*> LPadSuccMap; 648b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach public: 658b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach static char ID; // Pass identification, replacement for typeid 668b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach explicit SjLjEHPass(const TargetLowering *tli = NULL) 6790c579de5a383cee278acc3f7e7b9d0a656e6a35Owen Anderson : FunctionPass(ID), TLI(tli) { } 688b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach bool doInitialization(Module &M); 698b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach bool runOnFunction(Function &F); 708b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 71d36b3e36811e8b50476076ee4986068481916f66Bill Wendling virtual void getAnalysisUsage(AnalysisUsage &AU) const {} 728b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach const char *getPassName() const { 738b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach return "SJLJ Exception Handling preparation"; 748b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach } 758b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 768b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach private: 772b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling bool setupEntryBlockAndCallSites(Function &F); 78631d11765b9f7ebdb49679c92bd41083fbab3f62Bill Wendling Value *setupFunctionContext(Function &F, ArrayRef<LandingPadInst*> LPads); 792b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 80b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach void insertCallSiteStore(Instruction *I, int Number, Value *CallSite); 81b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach void markInvokeCallSite(InvokeInst *II, int InvokeNo, Value *CallSite, 820bb61c56bcad7a738d50f705b5966be446ef06c1Jim Grosbach SwitchInst *CatchSwitch); 83288694f430bc227d12b74f84f7e26e64d3b7f56bJim Grosbach void splitLiveRangesAcrossInvokes(SmallVector<InvokeInst*,16> &Invokes); 843669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling void splitLandingPad(InvokeInst *II); 858b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach bool insertSjLjEHSupport(Function &F); 868b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach }; 878b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach} // end anonymous namespace 888b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 898b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbachchar SjLjEHPass::ID = 0; 908b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 918b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach// Public Interface To the SjLjEHPass pass. 928b818d7e98309125c6058c4ea72a7dc73b031db2Jim GrosbachFunctionPass *llvm::createSjLjEHPass(const TargetLowering *TLI) { 938b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach return new SjLjEHPass(TLI); 948b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach} 95a235d13217ff14621a88f3ea96a8a3b980c56d02Jim Grosbach// doInitialization - Set up decalarations and types needed to process 96a235d13217ff14621a88f3ea96a8a3b980c56d02Jim Grosbach// exceptions. 978b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbachbool SjLjEHPass::doInitialization(Module &M) { 988b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // Build the function context structure. 998b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // builtin_setjmp uses a five word jbuf 1005fdd6c8793462549e3593890ec61573da06e3346Jay Foad Type *VoidPtrTy = Type::getInt8PtrTy(M.getContext()); 1015fdd6c8793462549e3593890ec61573da06e3346Jay Foad Type *Int32Ty = Type::getInt32Ty(M.getContext()); 1028b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach FunctionContextTy = 103b2318662b6d2e6d9ea9917fd280dde0ba9a938adChris Lattner StructType::get(VoidPtrTy, // __prev 1048b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach Int32Ty, // call_site 1058b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach ArrayType::get(Int32Ty, 4), // __data 1068b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach VoidPtrTy, // __personality 1078b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach VoidPtrTy, // __lsda 1088b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach ArrayType::get(VoidPtrTy, 5), // __jbuf 1098b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach NULL); 1108b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach RegisterFn = M.getOrInsertFunction("_Unwind_SjLj_Register", 1118b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach Type::getVoidTy(M.getContext()), 1128b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach PointerType::getUnqual(FunctionContextTy), 1138b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach (Type *)0); 1148b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach UnregisterFn = 1158b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach M.getOrInsertFunction("_Unwind_SjLj_Unregister", 1168b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach Type::getVoidTy(M.getContext()), 1178b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach PointerType::getUnqual(FunctionContextTy), 1188b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach (Type *)0); 1198b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach FrameAddrFn = Intrinsic::getDeclaration(&M, Intrinsic::frameaddress); 1200798eddd07b8dc827a4e6e9028c4c3a8d9444286Jim Grosbach StackAddrFn = Intrinsic::getDeclaration(&M, Intrinsic::stacksave); 1210798eddd07b8dc827a4e6e9028c4c3a8d9444286Jim Grosbach StackRestoreFn = Intrinsic::getDeclaration(&M, Intrinsic::stackrestore); 1228b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach BuiltinSetjmpFn = Intrinsic::getDeclaration(&M, Intrinsic::eh_sjlj_setjmp); 1238b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach LSDAAddrFn = Intrinsic::getDeclaration(&M, Intrinsic::eh_sjlj_lsda); 124b01bbdcc1af27bd90b552bb1b62b48916e0d4be3Duncan Sands SelectorFn = Intrinsic::getDeclaration(&M, Intrinsic::eh_selector); 1258b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach ExceptionFn = Intrinsic::getDeclaration(&M, Intrinsic::eh_exception); 126ca752c9020a1b1cf151142bd9e0cbca9af12d807Jim Grosbach CallSiteFn = Intrinsic::getDeclaration(&M, Intrinsic::eh_sjlj_callsite); 127e4ad387a5a88dae20f0f7578e55170bbc8eee2a9Jim Grosbach DispatchSetupFn 128e4ad387a5a88dae20f0f7578e55170bbc8eee2a9Jim Grosbach = Intrinsic::getDeclaration(&M, Intrinsic::eh_sjlj_dispatch_setup); 129cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling FuncCtxFn = Intrinsic::getDeclaration(&M, Intrinsic::eh_sjlj_functioncontext); 130a235d13217ff14621a88f3ea96a8a3b980c56d02Jim Grosbach PersonalityFn = 0; 1318b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 1328b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach return true; 1338b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach} 1348b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 135b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach/// insertCallSiteStore - Insert a store of the call-site value to the 136b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach/// function context 137b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbachvoid SjLjEHPass::insertCallSiteStore(Instruction *I, int Number, 138b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach Value *CallSite) { 139b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach ConstantInt *CallSiteNoC = ConstantInt::get(Type::getInt32Ty(I->getContext()), 140b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach Number); 141b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach // Insert a store of the call-site number 142b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach new StoreInst(CallSiteNoC, CallSite, true, I); // volatile 143b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach} 144b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach 1453669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling/// splitLandingPad - Split a landing pad. This takes considerable care because 1463669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling/// of PHIs and other nasties. The problem is that the jump table needs to jump 1473669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling/// to the landing pad block. However, the landing pad block can be jumped to 1483669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling/// only by an invoke instruction. So we clone the landingpad instruction into 1493669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling/// its own basic block, have the invoke jump to there. The landingpad 1503669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling/// instruction's basic block's successor is now the target for the jump table. 1513669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling/// 1523669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling/// But because of PHI nodes, we need to create another basic block for the jump 1533669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling/// table to jump to. This is definitely a hack, because the values for the PHI 1543669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling/// nodes may not be defined on the edge from the jump table. But that's okay, 1553669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling/// because the jump table is simply a construct to mimic what is happening in 1563669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling/// the CFG. So the values are mysteriously there, even though there is no value 1573669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling/// for the PHI from the jump table's edge (hence calling this a hack). 1583669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendlingvoid SjLjEHPass::splitLandingPad(InvokeInst *II) { 1593669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling SmallVector<BasicBlock*, 2> NewBBs; 1603669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling SplitLandingPadPredecessors(II->getUnwindDest(), II->getParent(), 1613669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling ".1", ".2", this, NewBBs); 1623669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling 1633669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling // Create an empty block so that the jump table has something to jump to 1643669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling // which doesn't have any PHI nodes. 1653669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling BasicBlock *LPad = NewBBs[0]; 1663669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling BasicBlock *Succ = *succ_begin(LPad); 1673669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling BasicBlock *JumpTo = BasicBlock::Create(II->getContext(), "jt.land", 1683669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling LPad->getParent(), Succ); 1693669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling LPad->getTerminator()->eraseFromParent(); 1703669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling BranchInst::Create(JumpTo, LPad); 1713669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling BranchInst::Create(Succ, JumpTo); 1723669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling LPadSuccMap[II] = JumpTo; 1733669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling 1743669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling for (BasicBlock::iterator I = Succ->begin(); isa<PHINode>(I); ++I) { 1753669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling PHINode *PN = cast<PHINode>(I); 1763669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling Value *Val = PN->removeIncomingValue(LPad, false); 1773669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling PN->addIncoming(Val, JumpTo); 1783669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling } 1793669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling} 1803669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling 1818b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach/// markInvokeCallSite - Insert code to mark the call_site for this invoke 182b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbachvoid SjLjEHPass::markInvokeCallSite(InvokeInst *II, int InvokeNo, 1830bb61c56bcad7a738d50f705b5966be446ef06c1Jim Grosbach Value *CallSite, 1840bb61c56bcad7a738d50f705b5966be446ef06c1Jim Grosbach SwitchInst *CatchSwitch) { 1858b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach ConstantInt *CallSiteNoC= ConstantInt::get(Type::getInt32Ty(II->getContext()), 186b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach InvokeNo); 1870bb61c56bcad7a738d50f705b5966be446ef06c1Jim Grosbach // The runtime comes back to the dispatcher with the call_site - 1 in 1880bb61c56bcad7a738d50f705b5966be446ef06c1Jim Grosbach // the context. Odd, but there it is. 1890bb61c56bcad7a738d50f705b5966be446ef06c1Jim Grosbach ConstantInt *SwitchValC = ConstantInt::get(Type::getInt32Ty(II->getContext()), 1903669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling InvokeNo - 1); 1918b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 1928b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // If the unwind edge has phi nodes, split the edge. 1938b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach if (isa<PHINode>(II->getUnwindDest()->begin())) { 19451fb91c04c920703fbd4a67ac3f85c0971b2c5a5Bill Wendling // FIXME: New EH - This if-condition will be always true in the new scheme. 1953669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling if (II->getUnwindDest()->isLandingPad()) 1963669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling splitLandingPad(II); 1973669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling else 19851fb91c04c920703fbd4a67ac3f85c0971b2c5a5Bill Wendling SplitCriticalEdge(II, 1, this); 1998b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 2008b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // If there are any phi nodes left, they must have a single predecessor. 2018b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach while (PHINode *PN = dyn_cast<PHINode>(II->getUnwindDest()->begin())) { 2028b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach PN->replaceAllUsesWith(PN->getIncomingValue(0)); 2038b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach PN->eraseFromParent(); 2048b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach } 2058b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach } 2068b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 207b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach // Insert the store of the call site value 208b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach insertCallSiteStore(II, InvokeNo, CallSite); 209b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach 210b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach // Record the call site value for the back end so it stays associated with 211b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach // the invoke. 212ca752c9020a1b1cf151142bd9e0cbca9af12d807Jim Grosbach CallInst::Create(CallSiteFn, CallSiteNoC, "", II); 2138b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 2140bb61c56bcad7a738d50f705b5966be446ef06c1Jim Grosbach // Add a switch case to our unwind block. 215aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling if (BasicBlock *SuccBB = LPadSuccMap[II]) { 216aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling CatchSwitch->addCase(SwitchValC, SuccBB); 217aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling } else { 218aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling CatchSwitch->addCase(SwitchValC, II->getUnwindDest()); 219aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling } 220aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling 221ca752c9020a1b1cf151142bd9e0cbca9af12d807Jim Grosbach // We still want this to look like an invoke so we emit the LSDA properly, 222ca752c9020a1b1cf151142bd9e0cbca9af12d807Jim Grosbach // so we don't transform the invoke into a call here. 2238b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach} 2248b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 2258b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach/// MarkBlocksLiveIn - Insert BB and all of its predescessors into LiveBBs until 2268b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach/// we reach blocks we've already seen. 2278b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbachstatic void MarkBlocksLiveIn(BasicBlock *BB, std::set<BasicBlock*> &LiveBBs) { 2288b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach if (!LiveBBs.insert(BB).second) return; // already been here. 2298b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 2308b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach for (pred_iterator PI = pred_begin(BB), E = pred_end(BB); PI != E; ++PI) 2318b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach MarkBlocksLiveIn(*PI, LiveBBs); 2328b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach} 2338b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 234606f3d696ff78211524fa49c3ca0c4de6db37041Jim Grosbach/// splitLiveRangesAcrossInvokes - Each value that is live across an unwind edge 235606f3d696ff78211524fa49c3ca0c4de6db37041Jim Grosbach/// we spill into a stack location, guaranteeing that there is nothing live 236606f3d696ff78211524fa49c3ca0c4de6db37041Jim Grosbach/// across the unwind edge. This process also splits all critical edges 237606f3d696ff78211524fa49c3ca0c4de6db37041Jim Grosbach/// coming out of invoke's. 2382f3257ef04a40adfeb6440d1b02f7bbfee0dd4fcJim Grosbach/// FIXME: Move this function to a common utility file (Local.cpp?) so 2392f3257ef04a40adfeb6440d1b02f7bbfee0dd4fcJim Grosbach/// both SjLj and LowerInvoke can use it. 2408b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbachvoid SjLjEHPass:: 241288694f430bc227d12b74f84f7e26e64d3b7f56bJim GrosbachsplitLiveRangesAcrossInvokes(SmallVector<InvokeInst*,16> &Invokes) { 2428b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // First step, split all critical edges from invoke instructions. 2438b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach for (unsigned i = 0, e = Invokes.size(); i != e; ++i) { 2448b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach InvokeInst *II = Invokes[i]; 2458b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach SplitCriticalEdge(II, 0, this); 246aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling 247aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling // FIXME: New EH - This if-condition will be always true in the new scheme. 2483669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling if (II->getUnwindDest()->isLandingPad()) 2493669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling splitLandingPad(II); 2503669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling else 251aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling SplitCriticalEdge(II, 1, this); 252aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling 2538b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach assert(!isa<PHINode>(II->getNormalDest()) && 2548b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach !isa<PHINode>(II->getUnwindDest()) && 255d36b3e36811e8b50476076ee4986068481916f66Bill Wendling "Critical edge splitting left single entry phi nodes?"); 2568b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach } 2578b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 2588b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach Function *F = Invokes.back()->getParent()->getParent(); 2598b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 2608b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // To avoid having to handle incoming arguments specially, we lower each arg 2618b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // to a copy instruction in the entry block. This ensures that the argument 2628b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // value itself cannot be live across the entry block. 2638b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach BasicBlock::iterator AfterAllocaInsertPt = F->begin()->begin(); 2648b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach while (isa<AllocaInst>(AfterAllocaInsertPt) && 2658b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach isa<ConstantInt>(cast<AllocaInst>(AfterAllocaInsertPt)->getArraySize())) 2668b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach ++AfterAllocaInsertPt; 2678b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach for (Function::arg_iterator AI = F->arg_begin(), E = F->arg_end(); 2688b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach AI != E; ++AI) { 269db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *Ty = AI->getType(); 270e70fc8ea6427934ee780db68d72aca07c629e0cdJim Grosbach // Aggregate types can't be cast, but are legal argument types, so we have 271dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach // to handle them differently. We use an extract/insert pair as a 272dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach // lightweight method to achieve the same goal. 273e70fc8ea6427934ee780db68d72aca07c629e0cdJim Grosbach if (isa<StructType>(Ty) || isa<ArrayType>(Ty) || isa<VectorType>(Ty)) { 274e70fc8ea6427934ee780db68d72aca07c629e0cdJim Grosbach Instruction *EI = ExtractValueInst::Create(AI, 0, "",AfterAllocaInsertPt); 275dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach Instruction *NI = InsertValueInst::Create(AI, EI, 0); 276dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach NI->insertAfter(EI); 277dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach AI->replaceAllUsesWith(NI); 278e70fc8ea6427934ee780db68d72aca07c629e0cdJim Grosbach // Set the operand of the instructions back to the AllocaInst. 279dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach EI->setOperand(0, AI); 280dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach NI->setOperand(0, AI); 281dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach } else { 282dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach // This is always a no-op cast because we're casting AI to AI->getType() 283dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach // so src and destination types are identical. BitCast is the only 284dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach // possibility. 285dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach CastInst *NC = new BitCastInst( 286dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach AI, AI->getType(), AI->getName()+".tmp", AfterAllocaInsertPt); 287dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach AI->replaceAllUsesWith(NC); 288dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach // Set the operand of the cast instruction back to the AllocaInst. 289dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach // Normally it's forbidden to replace a CastInst's operand because it 290dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach // could cause the opcode to reflect an illegal conversion. However, 291dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach // we're replacing it here with the same value it was constructed with. 292dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach // We do this because the above replaceAllUsesWith() clobbered the 293dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach // operand, but we want this one to remain. 294dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach NC->setOperand(0, AI); 295dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach } 2968b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach } 2978b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 2988b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // Finally, scan the code looking for instructions with bad live ranges. 2998b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach for (Function::iterator BB = F->begin(), E = F->end(); BB != E; ++BB) 3008b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach for (BasicBlock::iterator II = BB->begin(), E = BB->end(); II != E; ++II) { 3018b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // Ignore obvious cases we don't have to handle. In particular, most 3028b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // instructions either have no uses or only have a single use inside the 3038b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // current block. Ignore them quickly. 3048b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach Instruction *Inst = II; 3058b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach if (Inst->use_empty()) continue; 3068b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach if (Inst->hasOneUse() && 3078b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach cast<Instruction>(Inst->use_back())->getParent() == BB && 3088b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach !isa<PHINode>(Inst->use_back())) continue; 3098b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 3108b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // If this is an alloca in the entry block, it's not a real register 3118b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // value. 3128b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach if (AllocaInst *AI = dyn_cast<AllocaInst>(Inst)) 3138b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach if (isa<ConstantInt>(AI->getArraySize()) && BB == F->begin()) 3148b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach continue; 3158b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 3168b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // Avoid iterator invalidation by copying users to a temporary vector. 317606f3d696ff78211524fa49c3ca0c4de6db37041Jim Grosbach SmallVector<Instruction*,16> Users; 3188b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach for (Value::use_iterator UI = Inst->use_begin(), E = Inst->use_end(); 3198b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach UI != E; ++UI) { 3208b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach Instruction *User = cast<Instruction>(*UI); 3218b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach if (User->getParent() != BB || isa<PHINode>(User)) 3228b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach Users.push_back(User); 3238b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach } 3248b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 3258b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // Find all of the blocks that this value is live in. 3268b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach std::set<BasicBlock*> LiveBBs; 3278b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach LiveBBs.insert(Inst->getParent()); 3288b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach while (!Users.empty()) { 3298b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach Instruction *U = Users.back(); 3308b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach Users.pop_back(); 3318b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 3328b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach if (!isa<PHINode>(U)) { 3338b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach MarkBlocksLiveIn(U->getParent(), LiveBBs); 3348b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach } else { 3358b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // Uses for a PHI node occur in their predecessor block. 3368b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach PHINode *PN = cast<PHINode>(U); 3378b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach for (unsigned i = 0, e = PN->getNumIncomingValues(); i != e; ++i) 3388b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach if (PN->getIncomingValue(i) == Inst) 3398b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach MarkBlocksLiveIn(PN->getIncomingBlock(i), LiveBBs); 3408b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach } 3418b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach } 3428b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 3438b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // Now that we know all of the blocks that this thing is live in, see if 3448b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // it includes any of the unwind locations. 3458b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach bool NeedsSpill = false; 3468b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach for (unsigned i = 0, e = Invokes.size(); i != e; ++i) { 3478b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach BasicBlock *UnwindBlock = Invokes[i]->getUnwindDest(); 3488b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach if (UnwindBlock != BB && LiveBBs.count(UnwindBlock)) { 3498b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach NeedsSpill = true; 3508b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach } 3518b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach } 3528b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 3538b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // If we decided we need a spill, do it. 354956352eb4682ba7000486d8712a4435a843a48d4Jim Grosbach // FIXME: Spilling this way is overkill, as it forces all uses of 355956352eb4682ba7000486d8712a4435a843a48d4Jim Grosbach // the value to be reloaded from the stack slot, even those that aren't 356956352eb4682ba7000486d8712a4435a843a48d4Jim Grosbach // in the unwind blocks. We should be more selective. 3578b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach if (NeedsSpill) { 3588b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach ++NumSpilled; 3598b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach DemoteRegToStack(*Inst, true); 3608b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach } 3618b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach } 3628b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach} 3638b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 364aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling/// CreateLandingPadLoad - Load the exception handling values and insert them 365aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling/// into a structure. 366aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendlingstatic Instruction *CreateLandingPadLoad(Function &F, Value *ExnAddr, 367aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling Value *SelAddr, 368aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling BasicBlock::iterator InsertPt) { 369aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling Value *Exn = new LoadInst(ExnAddr, "exn", false, 370aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling InsertPt); 371aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling Type *Ty = Type::getInt8PtrTy(F.getContext()); 372aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling Exn = CastInst::Create(Instruction::IntToPtr, Exn, Ty, "", InsertPt); 373aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling Value *Sel = new LoadInst(SelAddr, "sel", false, InsertPt); 374aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling 375aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling Ty = StructType::get(Exn->getType(), Sel->getType(), NULL); 376aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling InsertValueInst *LPadVal = InsertValueInst::Create(llvm::UndefValue::get(Ty), 377aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling Exn, 0, 378aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling "lpad.val", InsertPt); 379aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling return InsertValueInst::Create(LPadVal, Sel, 1, "lpad.val", InsertPt); 380aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling} 381aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling 382aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling/// ReplaceLandingPadVal - Replace the landingpad instruction's value with a 383aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling/// load from the stored values (via CreateLandingPadLoad). This looks through 384aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling/// PHI nodes, and removes them if they are dead. 385aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendlingstatic void ReplaceLandingPadVal(Function &F, Instruction *Inst, Value *ExnAddr, 386aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling Value *SelAddr) { 387aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling if (Inst->use_empty()) return; 388aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling 389aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling while (!Inst->use_empty()) { 390aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling Instruction *I = cast<Instruction>(Inst->use_back()); 391aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling 392aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling if (PHINode *PN = dyn_cast<PHINode>(I)) { 393aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling ReplaceLandingPadVal(F, PN, ExnAddr, SelAddr); 394aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling if (PN->use_empty()) PN->eraseFromParent(); 395aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling continue; 396aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling } 397aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling 398fc8713ff74c2ed83d453057b24c578a0709fc180Bill Wendling I->replaceUsesOfWith(Inst, CreateLandingPadLoad(F, ExnAddr, SelAddr, I)); 399aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling } 400aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling} 401aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling 4028b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbachbool SjLjEHPass::insertSjLjEHSupport(Function &F) { 403606f3d696ff78211524fa49c3ca0c4de6db37041Jim Grosbach SmallVector<ReturnInst*,16> Returns; 404606f3d696ff78211524fa49c3ca0c4de6db37041Jim Grosbach SmallVector<UnwindInst*,16> Unwinds; 405606f3d696ff78211524fa49c3ca0c4de6db37041Jim Grosbach SmallVector<InvokeInst*,16> Invokes; 4068b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 4078b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // Look through the terminators of the basic blocks to find invokes, returns 408b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach // and unwinds. 409b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach for (Function::iterator BB = F.begin(), E = F.end(); BB != E; ++BB) { 4108b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach if (ReturnInst *RI = dyn_cast<ReturnInst>(BB->getTerminator())) { 4118b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // Remember all return instructions in case we insert an invoke into this 4128b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // function. 4138b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach Returns.push_back(RI); 4148b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach } else if (InvokeInst *II = dyn_cast<InvokeInst>(BB->getTerminator())) { 4158b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach Invokes.push_back(II); 4168b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach } else if (UnwindInst *UI = dyn_cast<UnwindInst>(BB->getTerminator())) { 4178b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach Unwinds.push_back(UI); 4188b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach } 419b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach } 4208d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 4218d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling NumInvokes += Invokes.size(); 4228d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling NumUnwinds += Unwinds.size(); 4238d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 4248d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // If we don't have any invokes, there's nothing to do. 4258d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling if (Invokes.empty()) return false; 4268b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 4270798eddd07b8dc827a4e6e9028c4c3a8d9444286Jim Grosbach // Find the eh.selector.*, eh.exception and alloca calls. 4280798eddd07b8dc827a4e6e9028c4c3a8d9444286Jim Grosbach // 4290798eddd07b8dc827a4e6e9028c4c3a8d9444286Jim Grosbach // Remember any allocas() that aren't in the entry block, as the 4300798eddd07b8dc827a4e6e9028c4c3a8d9444286Jim Grosbach // jmpbuf saved SP will need to be updated for them. 4310798eddd07b8dc827a4e6e9028c4c3a8d9444286Jim Grosbach // 4320798eddd07b8dc827a4e6e9028c4c3a8d9444286Jim Grosbach // We'll use the first eh.selector to determine the right personality 4330798eddd07b8dc827a4e6e9028c4c3a8d9444286Jim Grosbach // function to use. For SJLJ, we always use the same personality for the 4340798eddd07b8dc827a4e6e9028c4c3a8d9444286Jim Grosbach // whole function, not on a per-selector basis. 435a235d13217ff14621a88f3ea96a8a3b980c56d02Jim Grosbach // FIXME: That's a bit ugly. Better way? 436a235d13217ff14621a88f3ea96a8a3b980c56d02Jim Grosbach SmallVector<CallInst*,16> EH_Selectors; 437a235d13217ff14621a88f3ea96a8a3b980c56d02Jim Grosbach SmallVector<CallInst*,16> EH_Exceptions; 4380798eddd07b8dc827a4e6e9028c4c3a8d9444286Jim Grosbach SmallVector<Instruction*,16> JmpbufUpdatePoints; 4398d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 440cfcccef926bd142c885dfa3f506bff933939350cBill Wendling for (Function::iterator BB = F.begin(), E = F.end(); BB != E; ++BB) { 441cfcccef926bd142c885dfa3f506bff933939350cBill Wendling // Note: Skip the entry block since there's nothing there that interests 442cfcccef926bd142c885dfa3f506bff933939350cBill Wendling // us. eh.selector and eh.exception shouldn't ever be there, and we 443cfcccef926bd142c885dfa3f506bff933939350cBill Wendling // want to disregard any allocas that are there. 444cfcccef926bd142c885dfa3f506bff933939350cBill Wendling // 445cfcccef926bd142c885dfa3f506bff933939350cBill Wendling // FIXME: This is awkward. The new EH scheme won't need to skip the entry 446cfcccef926bd142c885dfa3f506bff933939350cBill Wendling // block. 447cfcccef926bd142c885dfa3f506bff933939350cBill Wendling if (BB == F.begin()) { 448cfcccef926bd142c885dfa3f506bff933939350cBill Wendling if (InvokeInst *II = dyn_cast<InvokeInst>(F.begin()->getTerminator())) { 449cfcccef926bd142c885dfa3f506bff933939350cBill Wendling // FIXME: This will be always non-NULL in the new EH. 450cfcccef926bd142c885dfa3f506bff933939350cBill Wendling if (LandingPadInst *LPI = II->getUnwindDest()->getLandingPadInst()) 451cfcccef926bd142c885dfa3f506bff933939350cBill Wendling if (!PersonalityFn) PersonalityFn = LPI->getPersonalityFn(); 452cfcccef926bd142c885dfa3f506bff933939350cBill Wendling } 453cfcccef926bd142c885dfa3f506bff933939350cBill Wendling 454cfcccef926bd142c885dfa3f506bff933939350cBill Wendling continue; 455cfcccef926bd142c885dfa3f506bff933939350cBill Wendling } 456cfcccef926bd142c885dfa3f506bff933939350cBill Wendling 457a235d13217ff14621a88f3ea96a8a3b980c56d02Jim Grosbach for (BasicBlock::iterator I = BB->begin(), E = BB->end(); I != E; ++I) { 458a235d13217ff14621a88f3ea96a8a3b980c56d02Jim Grosbach if (CallInst *CI = dyn_cast<CallInst>(I)) { 459b01bbdcc1af27bd90b552bb1b62b48916e0d4be3Duncan Sands if (CI->getCalledFunction() == SelectorFn) { 46045c097fe43d565905041d2a9fc77567d2deca709Gabor Greif if (!PersonalityFn) PersonalityFn = CI->getArgOperand(1); 461a235d13217ff14621a88f3ea96a8a3b980c56d02Jim Grosbach EH_Selectors.push_back(CI); 462a235d13217ff14621a88f3ea96a8a3b980c56d02Jim Grosbach } else if (CI->getCalledFunction() == ExceptionFn) { 463a235d13217ff14621a88f3ea96a8a3b980c56d02Jim Grosbach EH_Exceptions.push_back(CI); 4640798eddd07b8dc827a4e6e9028c4c3a8d9444286Jim Grosbach } else if (CI->getCalledFunction() == StackRestoreFn) { 4650798eddd07b8dc827a4e6e9028c4c3a8d9444286Jim Grosbach JmpbufUpdatePoints.push_back(CI); 466a235d13217ff14621a88f3ea96a8a3b980c56d02Jim Grosbach } 4670798eddd07b8dc827a4e6e9028c4c3a8d9444286Jim Grosbach } else if (AllocaInst *AI = dyn_cast<AllocaInst>(I)) { 4680798eddd07b8dc827a4e6e9028c4c3a8d9444286Jim Grosbach JmpbufUpdatePoints.push_back(AI); 469aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling } else if (InvokeInst *II = dyn_cast<InvokeInst>(I)) { 470aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling // FIXME: This will be always non-NULL in the new EH. 471aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling if (LandingPadInst *LPI = II->getUnwindDest()->getLandingPadInst()) 472aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling if (!PersonalityFn) PersonalityFn = LPI->getPersonalityFn(); 473a235d13217ff14621a88f3ea96a8a3b980c56d02Jim Grosbach } 474a235d13217ff14621a88f3ea96a8a3b980c56d02Jim Grosbach } 475a235d13217ff14621a88f3ea96a8a3b980c56d02Jim Grosbach } 4768d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 477a235d13217ff14621a88f3ea96a8a3b980c56d02Jim Grosbach // If we don't have any eh.selector calls, we can't determine the personality 478a235d13217ff14621a88f3ea96a8a3b980c56d02Jim Grosbach // function. Without a personality function, we can't process exceptions. 479a235d13217ff14621a88f3ea96a8a3b980c56d02Jim Grosbach if (!PersonalityFn) return false; 480a235d13217ff14621a88f3ea96a8a3b980c56d02Jim Grosbach 4818d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // We have invokes, so we need to add register/unregister calls to get this 4828d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // function onto the global unwind stack. 4838d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // 4848d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // First thing we need to do is scan the whole function for values that are 4858d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // live across unwind edges. Each value that is live across an unwind edge we 4868d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // spill into a stack location, guaranteeing that there is nothing live across 4878d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // the unwind edge. This process also splits all critical edges coming out of 4888d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // invoke's. 4898d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling splitLiveRangesAcrossInvokes(Invokes); 4908d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 491aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling 492aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling SmallVector<LandingPadInst*, 16> LandingPads; 493aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling for (Function::iterator BB = F.begin(), E = F.end(); BB != E; ++BB) { 494aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling if (InvokeInst *II = dyn_cast<InvokeInst>(BB->getTerminator())) 495aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling // FIXME: This will be always non-NULL in the new EH. 496aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling if (LandingPadInst *LPI = II->getUnwindDest()->getLandingPadInst()) 497aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling LandingPads.push_back(LPI); 498aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling } 499aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling 500aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling 5018d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling BasicBlock *EntryBB = F.begin(); 5028d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // Create an alloca for the incoming jump buffer ptr and the new jump buffer 5038d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // that needs to be restored on all exits from the function. This is an 5048d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // alloca because the value needs to be added to the global context list. 5058d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling unsigned Align = 4; // FIXME: Should be a TLI check? 5068d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling AllocaInst *FunctionContext = 5078d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling new AllocaInst(FunctionContextTy, 0, Align, 5088d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling "fcn_context", F.begin()->begin()); 5098d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 5108d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Value *Idxs[2]; 511db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *Int32Ty = Type::getInt32Ty(F.getContext()); 5128d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Value *Zero = ConstantInt::get(Int32Ty, 0); 5138d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // We need to also keep around a reference to the call_site field 5148d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Idxs[0] = Zero; 5158d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Idxs[1] = ConstantInt::get(Int32Ty, 1); 516a9203109f4ac95aa7e9624f2838e3d89623ec902Jay Foad CallSite = GetElementPtrInst::Create(FunctionContext, Idxs, "call_site", 5178d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling EntryBB->getTerminator()); 5188d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 5198d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // The exception selector comes back in context->data[1] 5208d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Idxs[1] = ConstantInt::get(Int32Ty, 2); 521a9203109f4ac95aa7e9624f2838e3d89623ec902Jay Foad Value *FCData = GetElementPtrInst::Create(FunctionContext, Idxs, "fc_data", 5228d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling EntryBB->getTerminator()); 5238d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Idxs[1] = ConstantInt::get(Int32Ty, 1); 524a9203109f4ac95aa7e9624f2838e3d89623ec902Jay Foad Value *SelectorAddr = GetElementPtrInst::Create(FCData, Idxs, 5258d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling "exc_selector_gep", 5268d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling EntryBB->getTerminator()); 5278d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // The exception value comes back in context->data[0] 5288d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Idxs[1] = Zero; 529a9203109f4ac95aa7e9624f2838e3d89623ec902Jay Foad Value *ExceptionAddr = GetElementPtrInst::Create(FCData, Idxs, 5308d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling "exception_gep", 5318d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling EntryBB->getTerminator()); 5328d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 5338d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // The result of the eh.selector call will be replaced with a a reference to 5348d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // the selector value returned in the function context. We leave the selector 5358d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // itself so the EH analysis later can use it. 5368d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling for (int i = 0, e = EH_Selectors.size(); i < e; ++i) { 5378d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling CallInst *I = EH_Selectors[i]; 5388d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Value *SelectorVal = new LoadInst(SelectorAddr, "select_val", true, I); 5398d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling I->replaceAllUsesWith(SelectorVal); 5408d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling } 5418b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 5428d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // eh.exception calls are replaced with references to the proper location in 5438d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // the context. Unlike eh.selector, the eh.exception calls are removed 5448d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // entirely. 5458d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling for (int i = 0, e = EH_Exceptions.size(); i < e; ++i) { 5468d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling CallInst *I = EH_Exceptions[i]; 5478d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // Possible for there to be duplicates, so check to make sure the 5488d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // instruction hasn't already been removed. 5498d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling if (!I->getParent()) continue; 5508d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Value *Val = new LoadInst(ExceptionAddr, "exception", true, I); 551db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *Ty = Type::getInt8PtrTy(F.getContext()); 5528d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Val = CastInst::Create(Instruction::IntToPtr, Val, Ty, "", I); 5538d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 5548d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling I->replaceAllUsesWith(Val); 5558d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling I->eraseFromParent(); 5568d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling } 5578b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 558aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling for (unsigned i = 0, e = LandingPads.size(); i != e; ++i) 559aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling ReplaceLandingPadVal(F, LandingPads[i], ExceptionAddr, SelectorAddr); 560aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling 5618d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // The entry block changes to have the eh.sjlj.setjmp, with a conditional 5628d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // branch to a dispatch block for non-zero returns. If we return normally, 5638d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // we're not handling an exception and just register the function context and 5648d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // continue. 5658d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 5668d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // Create the dispatch block. The dispatch block is basically a big switch 5678d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // statement that goes to all of the invoke landing pads. 5688d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling BasicBlock *DispatchBlock = 5698d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling BasicBlock::Create(F.getContext(), "eh.sjlj.setjmp.catch", &F); 5708d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 5718d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // Insert a load of the callsite in the dispatch block, and a switch on its 572cc88595bce813464f4dc0c8351f7311bb233a3e2Bill Wendling // value. By default, we issue a trap statement. 573cc88595bce813464f4dc0c8351f7311bb233a3e2Bill Wendling BasicBlock *TrapBlock = 574cc88595bce813464f4dc0c8351f7311bb233a3e2Bill Wendling BasicBlock::Create(F.getContext(), "trapbb", &F); 575cc88595bce813464f4dc0c8351f7311bb233a3e2Bill Wendling CallInst::Create(Intrinsic::getDeclaration(F.getParent(), Intrinsic::trap), 576cc88595bce813464f4dc0c8351f7311bb233a3e2Bill Wendling "", TrapBlock); 577cc88595bce813464f4dc0c8351f7311bb233a3e2Bill Wendling new UnreachableInst(F.getContext(), TrapBlock); 5788d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 5798d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Value *DispatchLoad = new LoadInst(CallSite, "invoke.num", true, 5808d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling DispatchBlock); 5818d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling SwitchInst *DispatchSwitch = 582cc88595bce813464f4dc0c8351f7311bb233a3e2Bill Wendling SwitchInst::Create(DispatchLoad, TrapBlock, Invokes.size(), 583e4ad387a5a88dae20f0f7578e55170bbc8eee2a9Jim Grosbach DispatchBlock); 5848d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // Split the entry block to insert the conditional branch for the setjmp. 5858d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling BasicBlock *ContBlock = EntryBB->splitBasicBlock(EntryBB->getTerminator(), 5868d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling "eh.sjlj.setjmp.cont"); 5878d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 5888d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // Populate the Function Context 5898d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // 1. LSDA address 5908d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // 2. Personality function address 5918d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // 3. jmpbuf (save SP, FP and call eh.sjlj.setjmp) 5928d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 5938d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // LSDA address 5948d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Idxs[0] = Zero; 5958d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Idxs[1] = ConstantInt::get(Int32Ty, 4); 5968d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Value *LSDAFieldPtr = 597a9203109f4ac95aa7e9624f2838e3d89623ec902Jay Foad GetElementPtrInst::Create(FunctionContext, Idxs, "lsda_gep", 5988d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling EntryBB->getTerminator()); 5998d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Value *LSDA = CallInst::Create(LSDAAddrFn, "lsda_addr", 6008d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling EntryBB->getTerminator()); 6018d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling new StoreInst(LSDA, LSDAFieldPtr, true, EntryBB->getTerminator()); 6028d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 6038d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Idxs[1] = ConstantInt::get(Int32Ty, 3); 6048d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Value *PersonalityFieldPtr = 605a9203109f4ac95aa7e9624f2838e3d89623ec902Jay Foad GetElementPtrInst::Create(FunctionContext, Idxs, "lsda_gep", 6068d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling EntryBB->getTerminator()); 6078d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling new StoreInst(PersonalityFn, PersonalityFieldPtr, true, 6088d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling EntryBB->getTerminator()); 6098d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 6108d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // Save the frame pointer. 6118d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Idxs[1] = ConstantInt::get(Int32Ty, 5); 6128d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Value *JBufPtr 613a9203109f4ac95aa7e9624f2838e3d89623ec902Jay Foad = GetElementPtrInst::Create(FunctionContext, Idxs, "jbuf_gep", 6148b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach EntryBB->getTerminator()); 6158d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Idxs[1] = ConstantInt::get(Int32Ty, 0); 6168d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Value *FramePtr = 617a9203109f4ac95aa7e9624f2838e3d89623ec902Jay Foad GetElementPtrInst::Create(JBufPtr, Idxs, "jbuf_fp_gep", 6188d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling EntryBB->getTerminator()); 6198d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 6208d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Value *Val = CallInst::Create(FrameAddrFn, 6218d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling ConstantInt::get(Int32Ty, 0), 6228d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling "fp", 6238b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach EntryBB->getTerminator()); 6248d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling new StoreInst(Val, FramePtr, true, EntryBB->getTerminator()); 6258d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 6268d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // Save the stack pointer. 6278d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Idxs[1] = ConstantInt::get(Int32Ty, 2); 6288d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Value *StackPtr = 629a9203109f4ac95aa7e9624f2838e3d89623ec902Jay Foad GetElementPtrInst::Create(JBufPtr, Idxs, "jbuf_sp_gep", 6308d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling EntryBB->getTerminator()); 6318d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 6328d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Val = CallInst::Create(StackAddrFn, "sp", EntryBB->getTerminator()); 6338d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling new StoreInst(Val, StackPtr, true, EntryBB->getTerminator()); 6348d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 6358d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // Call the setjmp instrinsic. It fills in the rest of the jmpbuf. 6368d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Value *SetjmpArg = 6378d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling CastInst::Create(Instruction::BitCast, JBufPtr, 6388d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Type::getInt8PtrTy(F.getContext()), "", 6398d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling EntryBB->getTerminator()); 6408d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Value *DispatchVal = CallInst::Create(BuiltinSetjmpFn, SetjmpArg, 6418d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling "dispatch", 6428d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling EntryBB->getTerminator()); 643f05b1dcf870346094f8aaee8e387c92d3e47e98dBill Wendling 644f05b1dcf870346094f8aaee8e387c92d3e47e98dBill Wendling // Add a call to dispatch_setup after the setjmp call. This is expanded to any 645f05b1dcf870346094f8aaee8e387c92d3e47e98dBill Wendling // target-specific setup that needs to be done. 64661512ba251097888963a8f07a35605564bcfc537Bill Wendling CallInst::Create(DispatchSetupFn, DispatchVal, "", EntryBB->getTerminator()); 647f05b1dcf870346094f8aaee8e387c92d3e47e98dBill Wendling 6488d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // check the return value of the setjmp. non-zero goes to dispatcher. 6498d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Value *IsNormal = new ICmpInst(EntryBB->getTerminator(), 6508d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling ICmpInst::ICMP_EQ, DispatchVal, Zero, 6518d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling "notunwind"); 6528d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // Nuke the uncond branch. 6538d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling EntryBB->getTerminator()->eraseFromParent(); 6548d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 6558d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // Put in a new condbranch in its place. 6568d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling BranchInst::Create(ContBlock, DispatchBlock, IsNormal, EntryBB); 6578d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 6588d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // Register the function context and make sure it's known to not throw 6598d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling CallInst *Register = 6608d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling CallInst::Create(RegisterFn, FunctionContext, "", 6618d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling ContBlock->getTerminator()); 6628d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Register->setDoesNotThrow(); 6638d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 6648d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // At this point, we are all set up, update the invoke instructions to mark 6658d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // their call_site values, and fill in the dispatch switch accordingly. 6668d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling for (unsigned i = 0, e = Invokes.size(); i != e; ++i) 6678d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling markInvokeCallSite(Invokes[i], i+1, CallSite, DispatchSwitch); 6688d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 6698d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // Mark call instructions that aren't nounwind as no-action (call_site == 6708d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // -1). Skip the entry block, as prior to then, no function context has been 6718d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // created for this function and any unexpected exceptions thrown will go 6728d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // directly to the caller's context, which is what we want anyway, so no need 6738d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // to do anything here. 6748d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling for (Function::iterator BB = F.begin(), E = F.end(); ++BB != E;) { 6758d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling for (BasicBlock::iterator I = BB->begin(), end = BB->end(); I != end; ++I) 6768d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling if (CallInst *CI = dyn_cast<CallInst>(I)) { 6778d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // Ignore calls to the EH builtins (eh.selector, eh.exception) 6788d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Constant *Callee = CI->getCalledFunction(); 6798d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling if (Callee != SelectorFn && Callee != ExceptionFn 6808d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling && !CI->doesNotThrow()) 6818d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling insertCallSiteStore(CI, -1, CallSite); 6823ae96d69ef0695460eb3d046427d22dcda10d9f0Bill Wendling } else if (ResumeInst *RI = dyn_cast<ResumeInst>(I)) { 6833ae96d69ef0695460eb3d046427d22dcda10d9f0Bill Wendling insertCallSiteStore(RI, -1, CallSite); 6848d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling } 6858d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling } 6868b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 6878d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // Replace all unwinds with a branch to the unwind handler. 6888d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // ??? Should this ever happen with sjlj exceptions? 6898d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling for (unsigned i = 0, e = Unwinds.size(); i != e; ++i) { 690cc88595bce813464f4dc0c8351f7311bb233a3e2Bill Wendling BranchInst::Create(TrapBlock, Unwinds[i]); 6918d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Unwinds[i]->eraseFromParent(); 6928d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling } 6930798eddd07b8dc827a4e6e9028c4c3a8d9444286Jim Grosbach 6948d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // Following any allocas not in the entry block, update the saved SP in the 6958d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // jmpbuf to the new value. 6968d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling for (unsigned i = 0, e = JmpbufUpdatePoints.size(); i != e; ++i) { 6978d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Instruction *AI = JmpbufUpdatePoints[i]; 6988d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Instruction *StackAddr = CallInst::Create(StackAddrFn, "sp"); 6998d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling StackAddr->insertAfter(AI); 7008d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Instruction *StoreStackAddr = new StoreInst(StackAddr, StackPtr, true); 7018d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling StoreStackAddr->insertAfter(StackAddr); 7028b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach } 7038b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 7048d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // Finally, for any returns from this function, if this function contains an 7058d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // invoke, add a call to unregister the function context. 7068d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling for (unsigned i = 0, e = Returns.size(); i != e; ++i) 7078d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling CallInst::Create(UnregisterFn, FunctionContext, "", Returns[i]); 7088d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 7098b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach return true; 7108b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach} 7118b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 7122b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling/// setupFunctionContext - Allocate the function context on the stack and fill 7132b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling/// it with all of the data that we know at this point. 714631d11765b9f7ebdb49679c92bd41083fbab3f62Bill WendlingValue *SjLjEHPass:: 715cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill WendlingsetupFunctionContext(Function &F, ArrayRef<LandingPadInst*> LPads) { 7162b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling BasicBlock *EntryBB = F.begin(); 7172b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 7182b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // Create an alloca for the incoming jump buffer ptr and the new jump buffer 7192b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // that needs to be restored on all exits from the function. This is an alloca 7202b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // because the value needs to be added to the global context list. 7212b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling unsigned Align = 7222b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling TLI->getTargetData()->getPrefTypeAlignment(FunctionContextTy); 7232b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling AllocaInst *FuncCtx = 7242b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling new AllocaInst(FunctionContextTy, 0, Align, "fn_context", EntryBB->begin()); 7252b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 7262b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // Fill in the function context structure. 7272b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Value *Idxs[2]; 7282b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Type *Int32Ty = Type::getInt32Ty(F.getContext()); 7292b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Value *Zero = ConstantInt::get(Int32Ty, 0); 7302b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Value *One = ConstantInt::get(Int32Ty, 1); 7312b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 7322b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // Keep around a reference to the call_site field. 7332b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Idxs[0] = Zero; 7342b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Idxs[1] = One; 7352b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling CallSite = GetElementPtrInst::Create(FuncCtx, Idxs, "call_site", 7362b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling EntryBB->getTerminator()); 7372b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 7382b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // Reference the __data field. 7392b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Idxs[1] = ConstantInt::get(Int32Ty, 2); 7402b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Value *FCData = GetElementPtrInst::Create(FuncCtx, Idxs, "__data", 7412b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling EntryBB->getTerminator()); 7422b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 7432b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // The exception value comes back in context->__data[0]. 7442b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Idxs[1] = Zero; 7452b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Value *ExceptionAddr = GetElementPtrInst::Create(FCData, Idxs, 7462b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling "exception_gep", 7472b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling EntryBB->getTerminator()); 7482b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 7492b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // The exception selector comes back in context->__data[1]. 7502b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Idxs[1] = One; 7512b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Value *SelectorAddr = GetElementPtrInst::Create(FCData, Idxs, 7522b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling "exn_selector_gep", 7532b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling EntryBB->getTerminator()); 7542b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 7552b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling for (unsigned I = 0, E = LPads.size(); I != E; ++I) { 7562b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling LandingPadInst *LPI = LPads[I]; 7572b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling IRBuilder<> Builder(LPI->getParent()->getFirstInsertionPt()); 7582b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 7592b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Value *ExnVal = Builder.CreateLoad(ExceptionAddr, true, "exn_val"); 7602b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling ExnVal = Builder.CreateIntToPtr(ExnVal, Type::getInt8PtrTy(F.getContext())); 7612b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Value *SelVal = Builder.CreateLoad(SelectorAddr, true, "exn_selector_val"); 7622b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 7632b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Type *LPadType = LPI->getType(); 7642b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Value *LPadVal = UndefValue::get(LPadType); 7652b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling LPadVal = Builder.CreateInsertValue(LPadVal, ExnVal, 0, "lpad.val"); 7662b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling LPadVal = Builder.CreateInsertValue(LPadVal, SelVal, 1, "lpad.val"); 7672b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 7682b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling LPI->replaceAllUsesWith(LPadVal); 7692b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling } 7702b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 7712b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // Personality function 7722b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Idxs[1] = ConstantInt::get(Int32Ty, 3); 7732b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling if (!PersonalityFn) 7742b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling PersonalityFn = LPads[0]->getPersonalityFn(); 7752b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Value *PersonalityFieldPtr = 7762b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling GetElementPtrInst::Create(FuncCtx, Idxs, "pers_fn_gep", 7772b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling EntryBB->getTerminator()); 7782b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling new StoreInst(PersonalityFn, PersonalityFieldPtr, true, 7792b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling EntryBB->getTerminator()); 7802b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 7812b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // LSDA address 7822b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Idxs[1] = ConstantInt::get(Int32Ty, 4); 7832b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Value *LSDAFieldPtr = 7842b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling GetElementPtrInst::Create(FuncCtx, Idxs, "lsda_gep", 7852b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling EntryBB->getTerminator()); 7862b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Value *LSDA = CallInst::Create(LSDAAddrFn, "lsda_addr", 7872b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling EntryBB->getTerminator()); 7882b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling new StoreInst(LSDA, LSDAFieldPtr, true, EntryBB->getTerminator()); 7892b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 790631d11765b9f7ebdb49679c92bd41083fbab3f62Bill Wendling return FuncCtx; 7912b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling} 7922b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 7932b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling/// setupEntryBlockAndCallSites - Setup the entry block by creating and filling 7942b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling/// the function context and marking the call sites with the appropriate 7952b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling/// values. These values are used by the DWARF EH emitter. 7962b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendlingbool SjLjEHPass::setupEntryBlockAndCallSites(Function &F) { 797cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling SmallVector<ReturnInst*, 16> Returns; 7982b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling SmallVector<InvokeInst*, 16> Invokes; 7992b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling SmallVector<LandingPadInst*, 16> LPads; 8002b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 8012b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // Look through the terminators of the basic blocks to find invokes. 8022b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling for (Function::iterator BB = F.begin(), E = F.end(); BB != E; ++BB) 8032b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling if (InvokeInst *II = dyn_cast<InvokeInst>(BB->getTerminator())) { 8042b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Invokes.push_back(II); 8052b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling LPads.push_back(II->getUnwindDest()->getLandingPadInst()); 806cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling } else if (ReturnInst *RI = dyn_cast<ReturnInst>(BB->getTerminator())) { 807cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling Returns.push_back(RI); 8082b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling } 8092b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 8102b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling if (Invokes.empty()) return false; 8112b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 812631d11765b9f7ebdb49679c92bd41083fbab3f62Bill Wendling Value *FuncCtx = setupFunctionContext(F, LPads); 813cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling BasicBlock *EntryBB = F.begin(); 814cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling Type *Int32Ty = Type::getInt32Ty(F.getContext()); 815631d11765b9f7ebdb49679c92bd41083fbab3f62Bill Wendling 816cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling Value *Idxs[2] = { 817631d11765b9f7ebdb49679c92bd41083fbab3f62Bill Wendling ConstantInt::get(Int32Ty, 0), 0 818cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling }; 819631d11765b9f7ebdb49679c92bd41083fbab3f62Bill Wendling 820631d11765b9f7ebdb49679c92bd41083fbab3f62Bill Wendling // Get a reference to the jump buffer. 821631d11765b9f7ebdb49679c92bd41083fbab3f62Bill Wendling Idxs[1] = ConstantInt::get(Int32Ty, 5); 822631d11765b9f7ebdb49679c92bd41083fbab3f62Bill Wendling Value *JBufPtr = 823631d11765b9f7ebdb49679c92bd41083fbab3f62Bill Wendling GetElementPtrInst::Create(FuncCtx, Idxs, "jbuf_gep", 824631d11765b9f7ebdb49679c92bd41083fbab3f62Bill Wendling EntryBB->getTerminator()); 825631d11765b9f7ebdb49679c92bd41083fbab3f62Bill Wendling 826631d11765b9f7ebdb49679c92bd41083fbab3f62Bill Wendling // Save the frame pointer. 827631d11765b9f7ebdb49679c92bd41083fbab3f62Bill Wendling Idxs[1] = ConstantInt::get(Int32Ty, 0); 828631d11765b9f7ebdb49679c92bd41083fbab3f62Bill Wendling Value *FramePtr = 829631d11765b9f7ebdb49679c92bd41083fbab3f62Bill Wendling GetElementPtrInst::Create(JBufPtr, Idxs, "jbuf_fp_gep", 830631d11765b9f7ebdb49679c92bd41083fbab3f62Bill Wendling EntryBB->getTerminator()); 831631d11765b9f7ebdb49679c92bd41083fbab3f62Bill Wendling 832631d11765b9f7ebdb49679c92bd41083fbab3f62Bill Wendling Value *Val = CallInst::Create(FrameAddrFn, 833631d11765b9f7ebdb49679c92bd41083fbab3f62Bill Wendling ConstantInt::get(Int32Ty, 0), 834631d11765b9f7ebdb49679c92bd41083fbab3f62Bill Wendling "fp", 835631d11765b9f7ebdb49679c92bd41083fbab3f62Bill Wendling EntryBB->getTerminator()); 836631d11765b9f7ebdb49679c92bd41083fbab3f62Bill Wendling new StoreInst(Val, FramePtr, true, EntryBB->getTerminator()); 837631d11765b9f7ebdb49679c92bd41083fbab3f62Bill Wendling 838631d11765b9f7ebdb49679c92bd41083fbab3f62Bill Wendling // Save the stack pointer. 839631d11765b9f7ebdb49679c92bd41083fbab3f62Bill Wendling Idxs[1] = ConstantInt::get(Int32Ty, 2); 840cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling Value *StackPtr = 841631d11765b9f7ebdb49679c92bd41083fbab3f62Bill Wendling GetElementPtrInst::Create(JBufPtr, Idxs, "jbuf_sp_gep", 842cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling EntryBB->getTerminator()); 843cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling 844631d11765b9f7ebdb49679c92bd41083fbab3f62Bill Wendling Val = CallInst::Create(StackAddrFn, "sp", EntryBB->getTerminator()); 845cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling new StoreInst(Val, StackPtr, true, EntryBB->getTerminator()); 846cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling 847cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling // Call the setjmp instrinsic. It fills in the rest of the jmpbuf. 848cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling Value *SetjmpArg = 849631d11765b9f7ebdb49679c92bd41083fbab3f62Bill Wendling CastInst::Create(Instruction::BitCast, JBufPtr, 850cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling Type::getInt8PtrTy(F.getContext()), "", 851cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling EntryBB->getTerminator()); 852cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling Value *DispatchVal = CallInst::Create(BuiltinSetjmpFn, SetjmpArg, 853cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling "dispatch", 854cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling EntryBB->getTerminator()); 855cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling 856cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling // Add a call to dispatch_setup after the setjmp call. This is expanded to any 857cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling // target-specific setup that needs to be done. 858cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling CallInst::Create(DispatchSetupFn, DispatchVal, "", EntryBB->getTerminator()); 859cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling 860cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling // Store a pointer to the function context so that the back-end will know 861cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling // where to look for it. 862cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling Value *FuncCtxArg = 863631d11765b9f7ebdb49679c92bd41083fbab3f62Bill Wendling CastInst::Create(Instruction::BitCast, FuncCtx, 864cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling Type::getInt8PtrTy(F.getContext()), "", 865cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling EntryBB->getTerminator()); 866cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling CallInst::Create(FuncCtxFn, FuncCtxArg, "", EntryBB->getTerminator()); 8672b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 8682b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // At this point, we are all set up, update the invoke instructions to mark 8692b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // their call_site values, and fill in the dispatch switch accordingly. 8702b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling for (unsigned I = 0, E = Invokes.size(); I != E; ++I) { 8712b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling insertCallSiteStore(Invokes[I], I + 1, CallSite); 8722b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 8732b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling ConstantInt *CallSiteNum = 8742b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling ConstantInt::get(Type::getInt32Ty(F.getContext()), I + 1); 8752b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 8762b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // Record the call site value for the back end so it stays associated with 8772b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // the invoke. 8782b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling CallInst::Create(CallSiteFn, CallSiteNum, "", Invokes[I]); 8792b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling } 8802b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 8812b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // Mark call instructions that aren't nounwind as no-action (call_site == 8822b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // -1). Skip the entry block, as prior to then, no function context has been 8832b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // created for this function and any unexpected exceptions thrown will go 8842b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // directly to the caller's context, which is what we want anyway, so no need 8852b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // to do anything here. 8862b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling for (Function::iterator BB = F.begin(), E = F.end(); ++BB != E;) { 8872b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling for (BasicBlock::iterator I = BB->begin(), end = BB->end(); I != end; ++I) 8882b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling if (CallInst *CI = dyn_cast<CallInst>(I)) { 8892b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling if (!CI->doesNotThrow()) 8902b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling insertCallSiteStore(CI, -1, CallSite); 8912b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling } else if (ResumeInst *RI = dyn_cast<ResumeInst>(I)) { 8922b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling insertCallSiteStore(RI, -1, CallSite); 8932b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling } 8942b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling } 8952b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 896cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling // Register the function context and make sure it's known to not throw 897631d11765b9f7ebdb49679c92bd41083fbab3f62Bill Wendling CallInst *Register = CallInst::Create(RegisterFn, FuncCtx, "", 898cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling EntryBB->getTerminator()); 899cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling Register->setDoesNotThrow(); 900cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling 901cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling // Finally, for any returns from this function, if this function contains an 902cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling // invoke, add a call to unregister the function context. 903cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling for (unsigned I = 0, E = Returns.size(); I != E; ++I) 904631d11765b9f7ebdb49679c92bd41083fbab3f62Bill Wendling CallInst::Create(UnregisterFn, FuncCtx, "", Returns[I]); 905cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling 9062b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling return true; 9072b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling} 9082b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 9098b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbachbool SjLjEHPass::runOnFunction(Function &F) { 9102b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling bool Res = false; 9112b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling if (!DisableOldSjLjEH) 9122b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Res = insertSjLjEHSupport(F); 9132b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling else 9142b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Res = setupEntryBlockAndCallSites(F); 9158b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach return Res; 9168b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach} 917