SjLjEHPrepare.cpp revision cc8cf97be434b4e2b9c0388db1bfb709f09098eb
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); 78cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling std::pair<Value*, Value*> 79cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling setupFunctionContext(Function &F, ArrayRef<LandingPadInst*> LPads); 802b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 81b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach void insertCallSiteStore(Instruction *I, int Number, Value *CallSite); 82b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach void markInvokeCallSite(InvokeInst *II, int InvokeNo, Value *CallSite, 830bb61c56bcad7a738d50f705b5966be446ef06c1Jim Grosbach SwitchInst *CatchSwitch); 84288694f430bc227d12b74f84f7e26e64d3b7f56bJim Grosbach void splitLiveRangesAcrossInvokes(SmallVector<InvokeInst*,16> &Invokes); 853669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling void splitLandingPad(InvokeInst *II); 868b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach bool insertSjLjEHSupport(Function &F); 878b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach }; 888b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach} // end anonymous namespace 898b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 908b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbachchar SjLjEHPass::ID = 0; 918b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 928b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach// Public Interface To the SjLjEHPass pass. 938b818d7e98309125c6058c4ea72a7dc73b031db2Jim GrosbachFunctionPass *llvm::createSjLjEHPass(const TargetLowering *TLI) { 948b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach return new SjLjEHPass(TLI); 958b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach} 96a235d13217ff14621a88f3ea96a8a3b980c56d02Jim Grosbach// doInitialization - Set up decalarations and types needed to process 97a235d13217ff14621a88f3ea96a8a3b980c56d02Jim Grosbach// exceptions. 988b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbachbool SjLjEHPass::doInitialization(Module &M) { 998b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // Build the function context structure. 1008b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // builtin_setjmp uses a five word jbuf 1015fdd6c8793462549e3593890ec61573da06e3346Jay Foad Type *VoidPtrTy = Type::getInt8PtrTy(M.getContext()); 1025fdd6c8793462549e3593890ec61573da06e3346Jay Foad Type *Int32Ty = Type::getInt32Ty(M.getContext()); 1038b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach FunctionContextTy = 104b2318662b6d2e6d9ea9917fd280dde0ba9a938adChris Lattner StructType::get(VoidPtrTy, // __prev 1058b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach Int32Ty, // call_site 1068b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach ArrayType::get(Int32Ty, 4), // __data 1078b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach VoidPtrTy, // __personality 1088b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach VoidPtrTy, // __lsda 1098b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach ArrayType::get(VoidPtrTy, 5), // __jbuf 1108b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach NULL); 1118b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach RegisterFn = M.getOrInsertFunction("_Unwind_SjLj_Register", 1128b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach Type::getVoidTy(M.getContext()), 1138b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach PointerType::getUnqual(FunctionContextTy), 1148b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach (Type *)0); 1158b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach UnregisterFn = 1168b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach M.getOrInsertFunction("_Unwind_SjLj_Unregister", 1178b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach Type::getVoidTy(M.getContext()), 1188b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach PointerType::getUnqual(FunctionContextTy), 1198b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach (Type *)0); 1208b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach FrameAddrFn = Intrinsic::getDeclaration(&M, Intrinsic::frameaddress); 1210798eddd07b8dc827a4e6e9028c4c3a8d9444286Jim Grosbach StackAddrFn = Intrinsic::getDeclaration(&M, Intrinsic::stacksave); 1220798eddd07b8dc827a4e6e9028c4c3a8d9444286Jim Grosbach StackRestoreFn = Intrinsic::getDeclaration(&M, Intrinsic::stackrestore); 1238b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach BuiltinSetjmpFn = Intrinsic::getDeclaration(&M, Intrinsic::eh_sjlj_setjmp); 1248b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach LSDAAddrFn = Intrinsic::getDeclaration(&M, Intrinsic::eh_sjlj_lsda); 125b01bbdcc1af27bd90b552bb1b62b48916e0d4be3Duncan Sands SelectorFn = Intrinsic::getDeclaration(&M, Intrinsic::eh_selector); 1268b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach ExceptionFn = Intrinsic::getDeclaration(&M, Intrinsic::eh_exception); 127ca752c9020a1b1cf151142bd9e0cbca9af12d807Jim Grosbach CallSiteFn = Intrinsic::getDeclaration(&M, Intrinsic::eh_sjlj_callsite); 128e4ad387a5a88dae20f0f7578e55170bbc8eee2a9Jim Grosbach DispatchSetupFn 129e4ad387a5a88dae20f0f7578e55170bbc8eee2a9Jim Grosbach = Intrinsic::getDeclaration(&M, Intrinsic::eh_sjlj_dispatch_setup); 130cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling FuncCtxFn = Intrinsic::getDeclaration(&M, Intrinsic::eh_sjlj_functioncontext); 131a235d13217ff14621a88f3ea96a8a3b980c56d02Jim Grosbach PersonalityFn = 0; 1328b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 1338b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach return true; 1348b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach} 1358b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 136b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach/// insertCallSiteStore - Insert a store of the call-site value to the 137b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach/// function context 138b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbachvoid SjLjEHPass::insertCallSiteStore(Instruction *I, int Number, 139b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach Value *CallSite) { 140b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach ConstantInt *CallSiteNoC = ConstantInt::get(Type::getInt32Ty(I->getContext()), 141b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach Number); 142b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach // Insert a store of the call-site number 143b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach new StoreInst(CallSiteNoC, CallSite, true, I); // volatile 144b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach} 145b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach 1463669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling/// splitLandingPad - Split a landing pad. This takes considerable care because 1473669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling/// of PHIs and other nasties. The problem is that the jump table needs to jump 1483669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling/// to the landing pad block. However, the landing pad block can be jumped to 1493669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling/// only by an invoke instruction. So we clone the landingpad instruction into 1503669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling/// its own basic block, have the invoke jump to there. The landingpad 1513669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling/// instruction's basic block's successor is now the target for the jump table. 1523669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling/// 1533669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling/// But because of PHI nodes, we need to create another basic block for the jump 1543669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling/// table to jump to. This is definitely a hack, because the values for the PHI 1553669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling/// nodes may not be defined on the edge from the jump table. But that's okay, 1563669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling/// because the jump table is simply a construct to mimic what is happening in 1573669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling/// the CFG. So the values are mysteriously there, even though there is no value 1583669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling/// for the PHI from the jump table's edge (hence calling this a hack). 1593669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendlingvoid SjLjEHPass::splitLandingPad(InvokeInst *II) { 1603669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling SmallVector<BasicBlock*, 2> NewBBs; 1613669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling SplitLandingPadPredecessors(II->getUnwindDest(), II->getParent(), 1623669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling ".1", ".2", this, NewBBs); 1633669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling 1643669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling // Create an empty block so that the jump table has something to jump to 1653669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling // which doesn't have any PHI nodes. 1663669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling BasicBlock *LPad = NewBBs[0]; 1673669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling BasicBlock *Succ = *succ_begin(LPad); 1683669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling BasicBlock *JumpTo = BasicBlock::Create(II->getContext(), "jt.land", 1693669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling LPad->getParent(), Succ); 1703669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling LPad->getTerminator()->eraseFromParent(); 1713669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling BranchInst::Create(JumpTo, LPad); 1723669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling BranchInst::Create(Succ, JumpTo); 1733669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling LPadSuccMap[II] = JumpTo; 1743669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling 1753669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling for (BasicBlock::iterator I = Succ->begin(); isa<PHINode>(I); ++I) { 1763669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling PHINode *PN = cast<PHINode>(I); 1773669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling Value *Val = PN->removeIncomingValue(LPad, false); 1783669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling PN->addIncoming(Val, JumpTo); 1793669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling } 1803669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling} 1813669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling 1828b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach/// markInvokeCallSite - Insert code to mark the call_site for this invoke 183b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbachvoid SjLjEHPass::markInvokeCallSite(InvokeInst *II, int InvokeNo, 1840bb61c56bcad7a738d50f705b5966be446ef06c1Jim Grosbach Value *CallSite, 1850bb61c56bcad7a738d50f705b5966be446ef06c1Jim Grosbach SwitchInst *CatchSwitch) { 1868b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach ConstantInt *CallSiteNoC= ConstantInt::get(Type::getInt32Ty(II->getContext()), 187b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach InvokeNo); 1880bb61c56bcad7a738d50f705b5966be446ef06c1Jim Grosbach // The runtime comes back to the dispatcher with the call_site - 1 in 1890bb61c56bcad7a738d50f705b5966be446ef06c1Jim Grosbach // the context. Odd, but there it is. 1900bb61c56bcad7a738d50f705b5966be446ef06c1Jim Grosbach ConstantInt *SwitchValC = ConstantInt::get(Type::getInt32Ty(II->getContext()), 1913669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling InvokeNo - 1); 1928b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 1938b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // If the unwind edge has phi nodes, split the edge. 1948b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach if (isa<PHINode>(II->getUnwindDest()->begin())) { 19551fb91c04c920703fbd4a67ac3f85c0971b2c5a5Bill Wendling // FIXME: New EH - This if-condition will be always true in the new scheme. 1963669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling if (II->getUnwindDest()->isLandingPad()) 1973669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling splitLandingPad(II); 1983669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling else 19951fb91c04c920703fbd4a67ac3f85c0971b2c5a5Bill Wendling SplitCriticalEdge(II, 1, this); 2008b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 2018b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // If there are any phi nodes left, they must have a single predecessor. 2028b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach while (PHINode *PN = dyn_cast<PHINode>(II->getUnwindDest()->begin())) { 2038b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach PN->replaceAllUsesWith(PN->getIncomingValue(0)); 2048b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach PN->eraseFromParent(); 2058b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach } 2068b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach } 2078b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 208b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach // Insert the store of the call site value 209b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach insertCallSiteStore(II, InvokeNo, CallSite); 210b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach 211b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach // Record the call site value for the back end so it stays associated with 212b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach // the invoke. 213ca752c9020a1b1cf151142bd9e0cbca9af12d807Jim Grosbach CallInst::Create(CallSiteFn, CallSiteNoC, "", II); 2148b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 2150bb61c56bcad7a738d50f705b5966be446ef06c1Jim Grosbach // Add a switch case to our unwind block. 216aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling if (BasicBlock *SuccBB = LPadSuccMap[II]) { 217aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling CatchSwitch->addCase(SwitchValC, SuccBB); 218aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling } else { 219aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling CatchSwitch->addCase(SwitchValC, II->getUnwindDest()); 220aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling } 221aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling 222ca752c9020a1b1cf151142bd9e0cbca9af12d807Jim Grosbach // We still want this to look like an invoke so we emit the LSDA properly, 223ca752c9020a1b1cf151142bd9e0cbca9af12d807Jim Grosbach // so we don't transform the invoke into a call here. 2248b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach} 2258b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 2268b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach/// MarkBlocksLiveIn - Insert BB and all of its predescessors into LiveBBs until 2278b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach/// we reach blocks we've already seen. 2288b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbachstatic void MarkBlocksLiveIn(BasicBlock *BB, std::set<BasicBlock*> &LiveBBs) { 2298b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach if (!LiveBBs.insert(BB).second) return; // already been here. 2308b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 2318b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach for (pred_iterator PI = pred_begin(BB), E = pred_end(BB); PI != E; ++PI) 2328b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach MarkBlocksLiveIn(*PI, LiveBBs); 2338b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach} 2348b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 235606f3d696ff78211524fa49c3ca0c4de6db37041Jim Grosbach/// splitLiveRangesAcrossInvokes - Each value that is live across an unwind edge 236606f3d696ff78211524fa49c3ca0c4de6db37041Jim Grosbach/// we spill into a stack location, guaranteeing that there is nothing live 237606f3d696ff78211524fa49c3ca0c4de6db37041Jim Grosbach/// across the unwind edge. This process also splits all critical edges 238606f3d696ff78211524fa49c3ca0c4de6db37041Jim Grosbach/// coming out of invoke's. 2392f3257ef04a40adfeb6440d1b02f7bbfee0dd4fcJim Grosbach/// FIXME: Move this function to a common utility file (Local.cpp?) so 2402f3257ef04a40adfeb6440d1b02f7bbfee0dd4fcJim Grosbach/// both SjLj and LowerInvoke can use it. 2418b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbachvoid SjLjEHPass:: 242288694f430bc227d12b74f84f7e26e64d3b7f56bJim GrosbachsplitLiveRangesAcrossInvokes(SmallVector<InvokeInst*,16> &Invokes) { 2438b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // First step, split all critical edges from invoke instructions. 2448b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach for (unsigned i = 0, e = Invokes.size(); i != e; ++i) { 2458b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach InvokeInst *II = Invokes[i]; 2468b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach SplitCriticalEdge(II, 0, this); 247aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling 248aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling // FIXME: New EH - This if-condition will be always true in the new scheme. 2493669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling if (II->getUnwindDest()->isLandingPad()) 2503669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling splitLandingPad(II); 2513669915c6dc8d57a4525fb0c592e6c8c1df098f5Bill Wendling else 252aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling SplitCriticalEdge(II, 1, this); 253aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling 2548b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach assert(!isa<PHINode>(II->getNormalDest()) && 2558b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach !isa<PHINode>(II->getUnwindDest()) && 256d36b3e36811e8b50476076ee4986068481916f66Bill Wendling "Critical edge splitting left single entry phi nodes?"); 2578b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach } 2588b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 2598b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach Function *F = Invokes.back()->getParent()->getParent(); 2608b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 2618b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // To avoid having to handle incoming arguments specially, we lower each arg 2628b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // to a copy instruction in the entry block. This ensures that the argument 2638b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // value itself cannot be live across the entry block. 2648b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach BasicBlock::iterator AfterAllocaInsertPt = F->begin()->begin(); 2658b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach while (isa<AllocaInst>(AfterAllocaInsertPt) && 2668b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach isa<ConstantInt>(cast<AllocaInst>(AfterAllocaInsertPt)->getArraySize())) 2678b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach ++AfterAllocaInsertPt; 2688b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach for (Function::arg_iterator AI = F->arg_begin(), E = F->arg_end(); 2698b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach AI != E; ++AI) { 270db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *Ty = AI->getType(); 271e70fc8ea6427934ee780db68d72aca07c629e0cdJim Grosbach // Aggregate types can't be cast, but are legal argument types, so we have 272dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach // to handle them differently. We use an extract/insert pair as a 273dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach // lightweight method to achieve the same goal. 274e70fc8ea6427934ee780db68d72aca07c629e0cdJim Grosbach if (isa<StructType>(Ty) || isa<ArrayType>(Ty) || isa<VectorType>(Ty)) { 275e70fc8ea6427934ee780db68d72aca07c629e0cdJim Grosbach Instruction *EI = ExtractValueInst::Create(AI, 0, "",AfterAllocaInsertPt); 276dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach Instruction *NI = InsertValueInst::Create(AI, EI, 0); 277dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach NI->insertAfter(EI); 278dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach AI->replaceAllUsesWith(NI); 279e70fc8ea6427934ee780db68d72aca07c629e0cdJim Grosbach // Set the operand of the instructions back to the AllocaInst. 280dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach EI->setOperand(0, AI); 281dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach NI->setOperand(0, AI); 282dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach } else { 283dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach // This is always a no-op cast because we're casting AI to AI->getType() 284dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach // so src and destination types are identical. BitCast is the only 285dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach // possibility. 286dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach CastInst *NC = new BitCastInst( 287dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach AI, AI->getType(), AI->getName()+".tmp", AfterAllocaInsertPt); 288dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach AI->replaceAllUsesWith(NC); 289dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach // Set the operand of the cast instruction back to the AllocaInst. 290dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach // Normally it's forbidden to replace a CastInst's operand because it 291dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach // could cause the opcode to reflect an illegal conversion. However, 292dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach // we're replacing it here with the same value it was constructed with. 293dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach // We do this because the above replaceAllUsesWith() clobbered the 294dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach // operand, but we want this one to remain. 295dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach NC->setOperand(0, AI); 296dc58b25ebc4d6b79f36db09a0b718ca957a610ebJim Grosbach } 2978b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach } 2988b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 2998b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // Finally, scan the code looking for instructions with bad live ranges. 3008b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach for (Function::iterator BB = F->begin(), E = F->end(); BB != E; ++BB) 3018b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach for (BasicBlock::iterator II = BB->begin(), E = BB->end(); II != E; ++II) { 3028b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // Ignore obvious cases we don't have to handle. In particular, most 3038b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // instructions either have no uses or only have a single use inside the 3048b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // current block. Ignore them quickly. 3058b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach Instruction *Inst = II; 3068b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach if (Inst->use_empty()) continue; 3078b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach if (Inst->hasOneUse() && 3088b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach cast<Instruction>(Inst->use_back())->getParent() == BB && 3098b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach !isa<PHINode>(Inst->use_back())) continue; 3108b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 3118b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // If this is an alloca in the entry block, it's not a real register 3128b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // value. 3138b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach if (AllocaInst *AI = dyn_cast<AllocaInst>(Inst)) 3148b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach if (isa<ConstantInt>(AI->getArraySize()) && BB == F->begin()) 3158b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach continue; 3168b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 3178b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // Avoid iterator invalidation by copying users to a temporary vector. 318606f3d696ff78211524fa49c3ca0c4de6db37041Jim Grosbach SmallVector<Instruction*,16> Users; 3198b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach for (Value::use_iterator UI = Inst->use_begin(), E = Inst->use_end(); 3208b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach UI != E; ++UI) { 3218b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach Instruction *User = cast<Instruction>(*UI); 3228b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach if (User->getParent() != BB || isa<PHINode>(User)) 3238b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach Users.push_back(User); 3248b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach } 3258b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 3268b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // Find all of the blocks that this value is live in. 3278b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach std::set<BasicBlock*> LiveBBs; 3288b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach LiveBBs.insert(Inst->getParent()); 3298b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach while (!Users.empty()) { 3308b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach Instruction *U = Users.back(); 3318b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach Users.pop_back(); 3328b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 3338b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach if (!isa<PHINode>(U)) { 3348b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach MarkBlocksLiveIn(U->getParent(), LiveBBs); 3358b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach } else { 3368b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // Uses for a PHI node occur in their predecessor block. 3378b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach PHINode *PN = cast<PHINode>(U); 3388b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach for (unsigned i = 0, e = PN->getNumIncomingValues(); i != e; ++i) 3398b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach if (PN->getIncomingValue(i) == Inst) 3408b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach MarkBlocksLiveIn(PN->getIncomingBlock(i), LiveBBs); 3418b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach } 3428b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach } 3438b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 3448b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // Now that we know all of the blocks that this thing is live in, see if 3458b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // it includes any of the unwind locations. 3468b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach bool NeedsSpill = false; 3478b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach for (unsigned i = 0, e = Invokes.size(); i != e; ++i) { 3488b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach BasicBlock *UnwindBlock = Invokes[i]->getUnwindDest(); 3498b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach if (UnwindBlock != BB && LiveBBs.count(UnwindBlock)) { 3508b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach NeedsSpill = true; 3518b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach } 3528b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach } 3538b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 3548b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // If we decided we need a spill, do it. 355956352eb4682ba7000486d8712a4435a843a48d4Jim Grosbach // FIXME: Spilling this way is overkill, as it forces all uses of 356956352eb4682ba7000486d8712a4435a843a48d4Jim Grosbach // the value to be reloaded from the stack slot, even those that aren't 357956352eb4682ba7000486d8712a4435a843a48d4Jim Grosbach // in the unwind blocks. We should be more selective. 3588b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach if (NeedsSpill) { 3598b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach ++NumSpilled; 3608b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach DemoteRegToStack(*Inst, true); 3618b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach } 3628b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach } 3638b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach} 3648b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 365aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling/// CreateLandingPadLoad - Load the exception handling values and insert them 366aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling/// into a structure. 367aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendlingstatic Instruction *CreateLandingPadLoad(Function &F, Value *ExnAddr, 368aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling Value *SelAddr, 369aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling BasicBlock::iterator InsertPt) { 370aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling Value *Exn = new LoadInst(ExnAddr, "exn", false, 371aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling InsertPt); 372aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling Type *Ty = Type::getInt8PtrTy(F.getContext()); 373aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling Exn = CastInst::Create(Instruction::IntToPtr, Exn, Ty, "", InsertPt); 374aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling Value *Sel = new LoadInst(SelAddr, "sel", false, InsertPt); 375aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling 376aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling Ty = StructType::get(Exn->getType(), Sel->getType(), NULL); 377aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling InsertValueInst *LPadVal = InsertValueInst::Create(llvm::UndefValue::get(Ty), 378aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling Exn, 0, 379aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling "lpad.val", InsertPt); 380aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling return InsertValueInst::Create(LPadVal, Sel, 1, "lpad.val", InsertPt); 381aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling} 382aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling 383aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling/// ReplaceLandingPadVal - Replace the landingpad instruction's value with a 384aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling/// load from the stored values (via CreateLandingPadLoad). This looks through 385aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling/// PHI nodes, and removes them if they are dead. 386aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendlingstatic void ReplaceLandingPadVal(Function &F, Instruction *Inst, Value *ExnAddr, 387aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling Value *SelAddr) { 388aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling if (Inst->use_empty()) return; 389aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling 390aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling while (!Inst->use_empty()) { 391aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling Instruction *I = cast<Instruction>(Inst->use_back()); 392aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling 393aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling if (PHINode *PN = dyn_cast<PHINode>(I)) { 394aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling ReplaceLandingPadVal(F, PN, ExnAddr, SelAddr); 395aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling if (PN->use_empty()) PN->eraseFromParent(); 396aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling continue; 397aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling } 398aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling 399fc8713ff74c2ed83d453057b24c578a0709fc180Bill Wendling I->replaceUsesOfWith(Inst, CreateLandingPadLoad(F, ExnAddr, SelAddr, I)); 400aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling } 401aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling} 402aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling 4038b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbachbool SjLjEHPass::insertSjLjEHSupport(Function &F) { 404606f3d696ff78211524fa49c3ca0c4de6db37041Jim Grosbach SmallVector<ReturnInst*,16> Returns; 405606f3d696ff78211524fa49c3ca0c4de6db37041Jim Grosbach SmallVector<UnwindInst*,16> Unwinds; 406606f3d696ff78211524fa49c3ca0c4de6db37041Jim Grosbach SmallVector<InvokeInst*,16> Invokes; 4078b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 4088b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // Look through the terminators of the basic blocks to find invokes, returns 409b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach // and unwinds. 410b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach for (Function::iterator BB = F.begin(), E = F.end(); BB != E; ++BB) { 4118b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach if (ReturnInst *RI = dyn_cast<ReturnInst>(BB->getTerminator())) { 4128b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // Remember all return instructions in case we insert an invoke into this 4138b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach // function. 4148b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach Returns.push_back(RI); 4158b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach } else if (InvokeInst *II = dyn_cast<InvokeInst>(BB->getTerminator())) { 4168b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach Invokes.push_back(II); 4178b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach } else if (UnwindInst *UI = dyn_cast<UnwindInst>(BB->getTerminator())) { 4188b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach Unwinds.push_back(UI); 4198b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach } 420b58a59b3c1563ab5bd7225748e32933bf6d76c03Jim Grosbach } 4218d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 4228d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling NumInvokes += Invokes.size(); 4238d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling NumUnwinds += Unwinds.size(); 4248d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 4258d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // If we don't have any invokes, there's nothing to do. 4268d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling if (Invokes.empty()) return false; 4278b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 4280798eddd07b8dc827a4e6e9028c4c3a8d9444286Jim Grosbach // Find the eh.selector.*, eh.exception and alloca calls. 4290798eddd07b8dc827a4e6e9028c4c3a8d9444286Jim Grosbach // 4300798eddd07b8dc827a4e6e9028c4c3a8d9444286Jim Grosbach // Remember any allocas() that aren't in the entry block, as the 4310798eddd07b8dc827a4e6e9028c4c3a8d9444286Jim Grosbach // jmpbuf saved SP will need to be updated for them. 4320798eddd07b8dc827a4e6e9028c4c3a8d9444286Jim Grosbach // 4330798eddd07b8dc827a4e6e9028c4c3a8d9444286Jim Grosbach // We'll use the first eh.selector to determine the right personality 4340798eddd07b8dc827a4e6e9028c4c3a8d9444286Jim Grosbach // function to use. For SJLJ, we always use the same personality for the 4350798eddd07b8dc827a4e6e9028c4c3a8d9444286Jim Grosbach // whole function, not on a per-selector basis. 436a235d13217ff14621a88f3ea96a8a3b980c56d02Jim Grosbach // FIXME: That's a bit ugly. Better way? 437a235d13217ff14621a88f3ea96a8a3b980c56d02Jim Grosbach SmallVector<CallInst*,16> EH_Selectors; 438a235d13217ff14621a88f3ea96a8a3b980c56d02Jim Grosbach SmallVector<CallInst*,16> EH_Exceptions; 4390798eddd07b8dc827a4e6e9028c4c3a8d9444286Jim Grosbach SmallVector<Instruction*,16> JmpbufUpdatePoints; 4408d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 441cfcccef926bd142c885dfa3f506bff933939350cBill Wendling for (Function::iterator BB = F.begin(), E = F.end(); BB != E; ++BB) { 442cfcccef926bd142c885dfa3f506bff933939350cBill Wendling // Note: Skip the entry block since there's nothing there that interests 443cfcccef926bd142c885dfa3f506bff933939350cBill Wendling // us. eh.selector and eh.exception shouldn't ever be there, and we 444cfcccef926bd142c885dfa3f506bff933939350cBill Wendling // want to disregard any allocas that are there. 445cfcccef926bd142c885dfa3f506bff933939350cBill Wendling // 446cfcccef926bd142c885dfa3f506bff933939350cBill Wendling // FIXME: This is awkward. The new EH scheme won't need to skip the entry 447cfcccef926bd142c885dfa3f506bff933939350cBill Wendling // block. 448cfcccef926bd142c885dfa3f506bff933939350cBill Wendling if (BB == F.begin()) { 449cfcccef926bd142c885dfa3f506bff933939350cBill Wendling if (InvokeInst *II = dyn_cast<InvokeInst>(F.begin()->getTerminator())) { 450cfcccef926bd142c885dfa3f506bff933939350cBill Wendling // FIXME: This will be always non-NULL in the new EH. 451cfcccef926bd142c885dfa3f506bff933939350cBill Wendling if (LandingPadInst *LPI = II->getUnwindDest()->getLandingPadInst()) 452cfcccef926bd142c885dfa3f506bff933939350cBill Wendling if (!PersonalityFn) PersonalityFn = LPI->getPersonalityFn(); 453cfcccef926bd142c885dfa3f506bff933939350cBill Wendling } 454cfcccef926bd142c885dfa3f506bff933939350cBill Wendling 455cfcccef926bd142c885dfa3f506bff933939350cBill Wendling continue; 456cfcccef926bd142c885dfa3f506bff933939350cBill Wendling } 457cfcccef926bd142c885dfa3f506bff933939350cBill Wendling 458a235d13217ff14621a88f3ea96a8a3b980c56d02Jim Grosbach for (BasicBlock::iterator I = BB->begin(), E = BB->end(); I != E; ++I) { 459a235d13217ff14621a88f3ea96a8a3b980c56d02Jim Grosbach if (CallInst *CI = dyn_cast<CallInst>(I)) { 460b01bbdcc1af27bd90b552bb1b62b48916e0d4be3Duncan Sands if (CI->getCalledFunction() == SelectorFn) { 46145c097fe43d565905041d2a9fc77567d2deca709Gabor Greif if (!PersonalityFn) PersonalityFn = CI->getArgOperand(1); 462a235d13217ff14621a88f3ea96a8a3b980c56d02Jim Grosbach EH_Selectors.push_back(CI); 463a235d13217ff14621a88f3ea96a8a3b980c56d02Jim Grosbach } else if (CI->getCalledFunction() == ExceptionFn) { 464a235d13217ff14621a88f3ea96a8a3b980c56d02Jim Grosbach EH_Exceptions.push_back(CI); 4650798eddd07b8dc827a4e6e9028c4c3a8d9444286Jim Grosbach } else if (CI->getCalledFunction() == StackRestoreFn) { 4660798eddd07b8dc827a4e6e9028c4c3a8d9444286Jim Grosbach JmpbufUpdatePoints.push_back(CI); 467a235d13217ff14621a88f3ea96a8a3b980c56d02Jim Grosbach } 4680798eddd07b8dc827a4e6e9028c4c3a8d9444286Jim Grosbach } else if (AllocaInst *AI = dyn_cast<AllocaInst>(I)) { 4690798eddd07b8dc827a4e6e9028c4c3a8d9444286Jim Grosbach JmpbufUpdatePoints.push_back(AI); 470aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling } else if (InvokeInst *II = dyn_cast<InvokeInst>(I)) { 471aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling // FIXME: This will be always non-NULL in the new EH. 472aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling if (LandingPadInst *LPI = II->getUnwindDest()->getLandingPadInst()) 473aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling if (!PersonalityFn) PersonalityFn = LPI->getPersonalityFn(); 474a235d13217ff14621a88f3ea96a8a3b980c56d02Jim Grosbach } 475a235d13217ff14621a88f3ea96a8a3b980c56d02Jim Grosbach } 476a235d13217ff14621a88f3ea96a8a3b980c56d02Jim Grosbach } 4778d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 478a235d13217ff14621a88f3ea96a8a3b980c56d02Jim Grosbach // If we don't have any eh.selector calls, we can't determine the personality 479a235d13217ff14621a88f3ea96a8a3b980c56d02Jim Grosbach // function. Without a personality function, we can't process exceptions. 480a235d13217ff14621a88f3ea96a8a3b980c56d02Jim Grosbach if (!PersonalityFn) return false; 481a235d13217ff14621a88f3ea96a8a3b980c56d02Jim Grosbach 4828d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // We have invokes, so we need to add register/unregister calls to get this 4838d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // function onto the global unwind stack. 4848d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // 4858d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // First thing we need to do is scan the whole function for values that are 4868d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // live across unwind edges. Each value that is live across an unwind edge we 4878d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // spill into a stack location, guaranteeing that there is nothing live across 4888d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // the unwind edge. This process also splits all critical edges coming out of 4898d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // invoke's. 4908d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling splitLiveRangesAcrossInvokes(Invokes); 4918d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 492aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling 493aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling SmallVector<LandingPadInst*, 16> LandingPads; 494aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling for (Function::iterator BB = F.begin(), E = F.end(); BB != E; ++BB) { 495aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling if (InvokeInst *II = dyn_cast<InvokeInst>(BB->getTerminator())) 496aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling // FIXME: This will be always non-NULL in the new EH. 497aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling if (LandingPadInst *LPI = II->getUnwindDest()->getLandingPadInst()) 498aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling LandingPads.push_back(LPI); 499aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling } 500aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling 501aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling 5028d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling BasicBlock *EntryBB = F.begin(); 5038d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // Create an alloca for the incoming jump buffer ptr and the new jump buffer 5048d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // that needs to be restored on all exits from the function. This is an 5058d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // alloca because the value needs to be added to the global context list. 5068d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling unsigned Align = 4; // FIXME: Should be a TLI check? 5078d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling AllocaInst *FunctionContext = 5088d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling new AllocaInst(FunctionContextTy, 0, Align, 5098d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling "fcn_context", F.begin()->begin()); 5108d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 5118d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Value *Idxs[2]; 512db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *Int32Ty = Type::getInt32Ty(F.getContext()); 5138d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Value *Zero = ConstantInt::get(Int32Ty, 0); 5148d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // We need to also keep around a reference to the call_site field 5158d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Idxs[0] = Zero; 5168d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Idxs[1] = ConstantInt::get(Int32Ty, 1); 517a9203109f4ac95aa7e9624f2838e3d89623ec902Jay Foad CallSite = GetElementPtrInst::Create(FunctionContext, Idxs, "call_site", 5188d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling EntryBB->getTerminator()); 5198d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 5208d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // The exception selector comes back in context->data[1] 5218d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Idxs[1] = ConstantInt::get(Int32Ty, 2); 522a9203109f4ac95aa7e9624f2838e3d89623ec902Jay Foad Value *FCData = GetElementPtrInst::Create(FunctionContext, Idxs, "fc_data", 5238d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling EntryBB->getTerminator()); 5248d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Idxs[1] = ConstantInt::get(Int32Ty, 1); 525a9203109f4ac95aa7e9624f2838e3d89623ec902Jay Foad Value *SelectorAddr = GetElementPtrInst::Create(FCData, Idxs, 5268d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling "exc_selector_gep", 5278d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling EntryBB->getTerminator()); 5288d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // The exception value comes back in context->data[0] 5298d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Idxs[1] = Zero; 530a9203109f4ac95aa7e9624f2838e3d89623ec902Jay Foad Value *ExceptionAddr = GetElementPtrInst::Create(FCData, Idxs, 5318d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling "exception_gep", 5328d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling EntryBB->getTerminator()); 5338d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 5348d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // The result of the eh.selector call will be replaced with a a reference to 5358d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // the selector value returned in the function context. We leave the selector 5368d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // itself so the EH analysis later can use it. 5378d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling for (int i = 0, e = EH_Selectors.size(); i < e; ++i) { 5388d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling CallInst *I = EH_Selectors[i]; 5398d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Value *SelectorVal = new LoadInst(SelectorAddr, "select_val", true, I); 5408d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling I->replaceAllUsesWith(SelectorVal); 5418d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling } 5428b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 5438d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // eh.exception calls are replaced with references to the proper location in 5448d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // the context. Unlike eh.selector, the eh.exception calls are removed 5458d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // entirely. 5468d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling for (int i = 0, e = EH_Exceptions.size(); i < e; ++i) { 5478d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling CallInst *I = EH_Exceptions[i]; 5488d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // Possible for there to be duplicates, so check to make sure the 5498d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // instruction hasn't already been removed. 5508d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling if (!I->getParent()) continue; 5518d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Value *Val = new LoadInst(ExceptionAddr, "exception", true, I); 552db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *Ty = Type::getInt8PtrTy(F.getContext()); 5538d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Val = CastInst::Create(Instruction::IntToPtr, Val, Ty, "", I); 5548d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 5558d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling I->replaceAllUsesWith(Val); 5568d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling I->eraseFromParent(); 5578d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling } 5588b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 559aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling for (unsigned i = 0, e = LandingPads.size(); i != e; ++i) 560aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling ReplaceLandingPadVal(F, LandingPads[i], ExceptionAddr, SelectorAddr); 561aef508dfc1d844bb3a4f7606f54bf4c616c10effBill Wendling 5628d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // The entry block changes to have the eh.sjlj.setjmp, with a conditional 5638d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // branch to a dispatch block for non-zero returns. If we return normally, 5648d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // we're not handling an exception and just register the function context and 5658d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // continue. 5668d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 5678d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // Create the dispatch block. The dispatch block is basically a big switch 5688d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // statement that goes to all of the invoke landing pads. 5698d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling BasicBlock *DispatchBlock = 5708d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling BasicBlock::Create(F.getContext(), "eh.sjlj.setjmp.catch", &F); 5718d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 5728d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // Insert a load of the callsite in the dispatch block, and a switch on its 573cc88595bce813464f4dc0c8351f7311bb233a3e2Bill Wendling // value. By default, we issue a trap statement. 574cc88595bce813464f4dc0c8351f7311bb233a3e2Bill Wendling BasicBlock *TrapBlock = 575cc88595bce813464f4dc0c8351f7311bb233a3e2Bill Wendling BasicBlock::Create(F.getContext(), "trapbb", &F); 576cc88595bce813464f4dc0c8351f7311bb233a3e2Bill Wendling CallInst::Create(Intrinsic::getDeclaration(F.getParent(), Intrinsic::trap), 577cc88595bce813464f4dc0c8351f7311bb233a3e2Bill Wendling "", TrapBlock); 578cc88595bce813464f4dc0c8351f7311bb233a3e2Bill Wendling new UnreachableInst(F.getContext(), TrapBlock); 5798d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 5808d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Value *DispatchLoad = new LoadInst(CallSite, "invoke.num", true, 5818d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling DispatchBlock); 5828d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling SwitchInst *DispatchSwitch = 583cc88595bce813464f4dc0c8351f7311bb233a3e2Bill Wendling SwitchInst::Create(DispatchLoad, TrapBlock, Invokes.size(), 584e4ad387a5a88dae20f0f7578e55170bbc8eee2a9Jim Grosbach DispatchBlock); 5858d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // Split the entry block to insert the conditional branch for the setjmp. 5868d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling BasicBlock *ContBlock = EntryBB->splitBasicBlock(EntryBB->getTerminator(), 5878d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling "eh.sjlj.setjmp.cont"); 5888d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 5898d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // Populate the Function Context 5908d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // 1. LSDA address 5918d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // 2. Personality function address 5928d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // 3. jmpbuf (save SP, FP and call eh.sjlj.setjmp) 5938d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 5948d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // LSDA address 5958d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Idxs[0] = Zero; 5968d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Idxs[1] = ConstantInt::get(Int32Ty, 4); 5978d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Value *LSDAFieldPtr = 598a9203109f4ac95aa7e9624f2838e3d89623ec902Jay Foad GetElementPtrInst::Create(FunctionContext, Idxs, "lsda_gep", 5998d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling EntryBB->getTerminator()); 6008d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Value *LSDA = CallInst::Create(LSDAAddrFn, "lsda_addr", 6018d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling EntryBB->getTerminator()); 6028d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling new StoreInst(LSDA, LSDAFieldPtr, true, EntryBB->getTerminator()); 6038d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 6048d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Idxs[1] = ConstantInt::get(Int32Ty, 3); 6058d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Value *PersonalityFieldPtr = 606a9203109f4ac95aa7e9624f2838e3d89623ec902Jay Foad GetElementPtrInst::Create(FunctionContext, Idxs, "lsda_gep", 6078d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling EntryBB->getTerminator()); 6088d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling new StoreInst(PersonalityFn, PersonalityFieldPtr, true, 6098d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling EntryBB->getTerminator()); 6108d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 6118d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // Save the frame pointer. 6128d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Idxs[1] = ConstantInt::get(Int32Ty, 5); 6138d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Value *JBufPtr 614a9203109f4ac95aa7e9624f2838e3d89623ec902Jay Foad = GetElementPtrInst::Create(FunctionContext, Idxs, "jbuf_gep", 6158b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach EntryBB->getTerminator()); 6168d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Idxs[1] = ConstantInt::get(Int32Ty, 0); 6178d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Value *FramePtr = 618a9203109f4ac95aa7e9624f2838e3d89623ec902Jay Foad GetElementPtrInst::Create(JBufPtr, Idxs, "jbuf_fp_gep", 6198d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling EntryBB->getTerminator()); 6208d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 6218d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Value *Val = CallInst::Create(FrameAddrFn, 6228d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling ConstantInt::get(Int32Ty, 0), 6238d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling "fp", 6248b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach EntryBB->getTerminator()); 6258d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling new StoreInst(Val, FramePtr, true, EntryBB->getTerminator()); 6268d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 6278d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // Save the stack pointer. 6288d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Idxs[1] = ConstantInt::get(Int32Ty, 2); 6298d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Value *StackPtr = 630a9203109f4ac95aa7e9624f2838e3d89623ec902Jay Foad GetElementPtrInst::Create(JBufPtr, Idxs, "jbuf_sp_gep", 6318d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling EntryBB->getTerminator()); 6328d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 6338d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Val = CallInst::Create(StackAddrFn, "sp", EntryBB->getTerminator()); 6348d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling new StoreInst(Val, StackPtr, true, EntryBB->getTerminator()); 6358d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 6368d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // Call the setjmp instrinsic. It fills in the rest of the jmpbuf. 6378d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Value *SetjmpArg = 6388d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling CastInst::Create(Instruction::BitCast, JBufPtr, 6398d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Type::getInt8PtrTy(F.getContext()), "", 6408d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling EntryBB->getTerminator()); 6418d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Value *DispatchVal = CallInst::Create(BuiltinSetjmpFn, SetjmpArg, 6428d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling "dispatch", 6438d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling EntryBB->getTerminator()); 644f05b1dcf870346094f8aaee8e387c92d3e47e98dBill Wendling 645f05b1dcf870346094f8aaee8e387c92d3e47e98dBill Wendling // Add a call to dispatch_setup after the setjmp call. This is expanded to any 646f05b1dcf870346094f8aaee8e387c92d3e47e98dBill Wendling // target-specific setup that needs to be done. 64761512ba251097888963a8f07a35605564bcfc537Bill Wendling CallInst::Create(DispatchSetupFn, DispatchVal, "", EntryBB->getTerminator()); 648f05b1dcf870346094f8aaee8e387c92d3e47e98dBill Wendling 6498d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // check the return value of the setjmp. non-zero goes to dispatcher. 6508d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Value *IsNormal = new ICmpInst(EntryBB->getTerminator(), 6518d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling ICmpInst::ICMP_EQ, DispatchVal, Zero, 6528d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling "notunwind"); 6538d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // Nuke the uncond branch. 6548d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling EntryBB->getTerminator()->eraseFromParent(); 6558d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 6568d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // Put in a new condbranch in its place. 6578d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling BranchInst::Create(ContBlock, DispatchBlock, IsNormal, EntryBB); 6588d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 6598d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // Register the function context and make sure it's known to not throw 6608d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling CallInst *Register = 6618d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling CallInst::Create(RegisterFn, FunctionContext, "", 6628d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling ContBlock->getTerminator()); 6638d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Register->setDoesNotThrow(); 6648d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 6658d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // At this point, we are all set up, update the invoke instructions to mark 6668d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // their call_site values, and fill in the dispatch switch accordingly. 6678d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling for (unsigned i = 0, e = Invokes.size(); i != e; ++i) 6688d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling markInvokeCallSite(Invokes[i], i+1, CallSite, DispatchSwitch); 6698d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 6708d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // Mark call instructions that aren't nounwind as no-action (call_site == 6718d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // -1). Skip the entry block, as prior to then, no function context has been 6728d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // created for this function and any unexpected exceptions thrown will go 6738d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // directly to the caller's context, which is what we want anyway, so no need 6748d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // to do anything here. 6758d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling for (Function::iterator BB = F.begin(), E = F.end(); ++BB != E;) { 6768d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling for (BasicBlock::iterator I = BB->begin(), end = BB->end(); I != end; ++I) 6778d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling if (CallInst *CI = dyn_cast<CallInst>(I)) { 6788d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // Ignore calls to the EH builtins (eh.selector, eh.exception) 6798d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Constant *Callee = CI->getCalledFunction(); 6808d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling if (Callee != SelectorFn && Callee != ExceptionFn 6818d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling && !CI->doesNotThrow()) 6828d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling insertCallSiteStore(CI, -1, CallSite); 6833ae96d69ef0695460eb3d046427d22dcda10d9f0Bill Wendling } else if (ResumeInst *RI = dyn_cast<ResumeInst>(I)) { 6843ae96d69ef0695460eb3d046427d22dcda10d9f0Bill Wendling insertCallSiteStore(RI, -1, CallSite); 6858d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling } 6868d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling } 6878b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 6888d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // Replace all unwinds with a branch to the unwind handler. 6898d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // ??? Should this ever happen with sjlj exceptions? 6908d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling for (unsigned i = 0, e = Unwinds.size(); i != e; ++i) { 691cc88595bce813464f4dc0c8351f7311bb233a3e2Bill Wendling BranchInst::Create(TrapBlock, Unwinds[i]); 6928d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Unwinds[i]->eraseFromParent(); 6938d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling } 6940798eddd07b8dc827a4e6e9028c4c3a8d9444286Jim Grosbach 6958d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // Following any allocas not in the entry block, update the saved SP in the 6968d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // jmpbuf to the new value. 6978d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling for (unsigned i = 0, e = JmpbufUpdatePoints.size(); i != e; ++i) { 6988d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Instruction *AI = JmpbufUpdatePoints[i]; 6998d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Instruction *StackAddr = CallInst::Create(StackAddrFn, "sp"); 7008d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling StackAddr->insertAfter(AI); 7018d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling Instruction *StoreStackAddr = new StoreInst(StackAddr, StackPtr, true); 7028d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling StoreStackAddr->insertAfter(StackAddr); 7038b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach } 7048b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 7058d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // Finally, for any returns from this function, if this function contains an 7068d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling // invoke, add a call to unregister the function context. 7078d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling for (unsigned i = 0, e = Returns.size(); i != e; ++i) 7088d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling CallInst::Create(UnregisterFn, FunctionContext, "", Returns[i]); 7098d90b7190f9411e0e48d8a29705240f7679ae201Bill Wendling 7108b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach return true; 7118b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach} 7128b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach 7132b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling/// setupFunctionContext - Allocate the function context on the stack and fill 7142b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling/// it with all of the data that we know at this point. 715cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendlingstd::pair<Value*, Value*> SjLjEHPass:: 716cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill WendlingsetupFunctionContext(Function &F, ArrayRef<LandingPadInst*> LPads) { 7172b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling BasicBlock *EntryBB = F.begin(); 7182b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 7192b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // Create an alloca for the incoming jump buffer ptr and the new jump buffer 7202b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // that needs to be restored on all exits from the function. This is an alloca 7212b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // because the value needs to be added to the global context list. 7222b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling unsigned Align = 7232b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling TLI->getTargetData()->getPrefTypeAlignment(FunctionContextTy); 7242b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling AllocaInst *FuncCtx = 7252b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling new AllocaInst(FunctionContextTy, 0, Align, "fn_context", EntryBB->begin()); 7262b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 7272b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // Fill in the function context structure. 7282b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Value *Idxs[2]; 7292b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Type *Int32Ty = Type::getInt32Ty(F.getContext()); 7302b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Value *Zero = ConstantInt::get(Int32Ty, 0); 7312b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Value *One = ConstantInt::get(Int32Ty, 1); 7322b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 7332b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // Keep around a reference to the call_site field. 7342b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Idxs[0] = Zero; 7352b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Idxs[1] = One; 7362b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling CallSite = GetElementPtrInst::Create(FuncCtx, Idxs, "call_site", 7372b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling EntryBB->getTerminator()); 7382b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 7392b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // Reference the __data field. 7402b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Idxs[1] = ConstantInt::get(Int32Ty, 2); 7412b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Value *FCData = GetElementPtrInst::Create(FuncCtx, Idxs, "__data", 7422b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling EntryBB->getTerminator()); 7432b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 7442b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // The exception value comes back in context->__data[0]. 7452b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Idxs[1] = Zero; 7462b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Value *ExceptionAddr = GetElementPtrInst::Create(FCData, Idxs, 7472b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling "exception_gep", 7482b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling EntryBB->getTerminator()); 7492b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 7502b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // The exception selector comes back in context->__data[1]. 7512b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Idxs[1] = One; 7522b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Value *SelectorAddr = GetElementPtrInst::Create(FCData, Idxs, 7532b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling "exn_selector_gep", 7542b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling EntryBB->getTerminator()); 7552b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 7562b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling for (unsigned I = 0, E = LPads.size(); I != E; ++I) { 7572b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling LandingPadInst *LPI = LPads[I]; 7582b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling IRBuilder<> Builder(LPI->getParent()->getFirstInsertionPt()); 7592b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 7602b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Value *ExnVal = Builder.CreateLoad(ExceptionAddr, true, "exn_val"); 7612b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling ExnVal = Builder.CreateIntToPtr(ExnVal, Type::getInt8PtrTy(F.getContext())); 7622b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Value *SelVal = Builder.CreateLoad(SelectorAddr, true, "exn_selector_val"); 7632b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 7642b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Type *LPadType = LPI->getType(); 7652b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Value *LPadVal = UndefValue::get(LPadType); 7662b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling LPadVal = Builder.CreateInsertValue(LPadVal, ExnVal, 0, "lpad.val"); 7672b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling LPadVal = Builder.CreateInsertValue(LPadVal, SelVal, 1, "lpad.val"); 7682b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 7692b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling LPI->replaceAllUsesWith(LPadVal); 7702b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling } 7712b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 7722b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // Personality function 7732b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Idxs[1] = ConstantInt::get(Int32Ty, 3); 7742b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling if (!PersonalityFn) 7752b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling PersonalityFn = LPads[0]->getPersonalityFn(); 7762b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Value *PersonalityFieldPtr = 7772b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling GetElementPtrInst::Create(FuncCtx, Idxs, "pers_fn_gep", 7782b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling EntryBB->getTerminator()); 7792b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling new StoreInst(PersonalityFn, PersonalityFieldPtr, true, 7802b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling EntryBB->getTerminator()); 7812b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 7822b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // LSDA address 7832b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Idxs[1] = ConstantInt::get(Int32Ty, 4); 7842b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Value *LSDAFieldPtr = 7852b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling GetElementPtrInst::Create(FuncCtx, Idxs, "lsda_gep", 7862b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling EntryBB->getTerminator()); 7872b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Value *LSDA = CallInst::Create(LSDAAddrFn, "lsda_addr", 7882b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling EntryBB->getTerminator()); 7892b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling new StoreInst(LSDA, LSDAFieldPtr, true, EntryBB->getTerminator()); 7902b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 7912b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // Get a reference to the jump buffer. 7922b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Idxs[1] = ConstantInt::get(Int32Ty, 5); 7932b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Value *JBufPtr = 7942b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling GetElementPtrInst::Create(FuncCtx, Idxs, "jbuf_gep", 7952b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling EntryBB->getTerminator()); 7962b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Idxs[1] = Zero; 7972b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Value *FramePtr = 7982b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling GetElementPtrInst::Create(JBufPtr, Idxs, "jbuf_fp_gep", 7992b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling EntryBB->getTerminator()); 8002b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 8012b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // Save the frame pointer. 8022b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Value *Val = CallInst::Create(FrameAddrFn, 8032b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling ConstantInt::get(Int32Ty, 0), 8042b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling "fp", 8052b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling EntryBB->getTerminator()); 8062b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling new StoreInst(Val, FramePtr, true, EntryBB->getTerminator()); 8072b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 808cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling return std::make_pair(FuncCtx, JBufPtr); 8092b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling} 8102b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 8112b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling/// setupEntryBlockAndCallSites - Setup the entry block by creating and filling 8122b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling/// the function context and marking the call sites with the appropriate 8132b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling/// values. These values are used by the DWARF EH emitter. 8142b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendlingbool SjLjEHPass::setupEntryBlockAndCallSites(Function &F) { 815cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling SmallVector<ReturnInst*, 16> Returns; 8162b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling SmallVector<InvokeInst*, 16> Invokes; 8172b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling SmallVector<LandingPadInst*, 16> LPads; 8182b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 8192b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // Look through the terminators of the basic blocks to find invokes. 8202b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling for (Function::iterator BB = F.begin(), E = F.end(); BB != E; ++BB) 8212b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling if (InvokeInst *II = dyn_cast<InvokeInst>(BB->getTerminator())) { 8222b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Invokes.push_back(II); 8232b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling LPads.push_back(II->getUnwindDest()->getLandingPadInst()); 824cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling } else if (ReturnInst *RI = dyn_cast<ReturnInst>(BB->getTerminator())) { 825cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling Returns.push_back(RI); 8262b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling } 8272b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 8282b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling if (Invokes.empty()) return false; 8292b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 830cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling std::pair<Value*, Value*> FuncCtx = setupFunctionContext(F, LPads); 831cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling BasicBlock *EntryBB = F.begin(); 832cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling 833cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling // Save the stack pointer. 834cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling Type *Int32Ty = Type::getInt32Ty(F.getContext()); 835cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling Value *Idxs[2] = { 836cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling ConstantInt::get(Int32Ty, 0), ConstantInt::get(Int32Ty, 2) 837cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling }; 838cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling Value *StackPtr = 839cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling GetElementPtrInst::Create(FuncCtx.second, Idxs, "jbuf_sp_gep", 840cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling EntryBB->getTerminator()); 841cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling 842cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling Value *Val = CallInst::Create(StackAddrFn, "sp", EntryBB->getTerminator()); 843cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling new StoreInst(Val, StackPtr, true, EntryBB->getTerminator()); 844cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling 845cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling // Call the setjmp instrinsic. It fills in the rest of the jmpbuf. 846cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling Value *SetjmpArg = 847cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling CastInst::Create(Instruction::BitCast, FuncCtx.second, 848cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling Type::getInt8PtrTy(F.getContext()), "", 849cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling EntryBB->getTerminator()); 850cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling Value *DispatchVal = CallInst::Create(BuiltinSetjmpFn, SetjmpArg, 851cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling "dispatch", 852cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling EntryBB->getTerminator()); 853cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling 854cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling // Add a call to dispatch_setup after the setjmp call. This is expanded to any 855cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling // target-specific setup that needs to be done. 856cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling CallInst::Create(DispatchSetupFn, DispatchVal, "", EntryBB->getTerminator()); 857cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling 858cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling // Store a pointer to the function context so that the back-end will know 859cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling // where to look for it. 860cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling Value *FuncCtxArg = 861cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling CastInst::Create(Instruction::BitCast, FuncCtx.first, 862cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling Type::getInt8PtrTy(F.getContext()), "", 863cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling EntryBB->getTerminator()); 864cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling CallInst::Create(FuncCtxFn, FuncCtxArg, "", EntryBB->getTerminator()); 8652b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 8662b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // At this point, we are all set up, update the invoke instructions to mark 8672b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // their call_site values, and fill in the dispatch switch accordingly. 8682b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling for (unsigned I = 0, E = Invokes.size(); I != E; ++I) { 8692b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling insertCallSiteStore(Invokes[I], I + 1, CallSite); 8702b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 8712b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling ConstantInt *CallSiteNum = 8722b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling ConstantInt::get(Type::getInt32Ty(F.getContext()), I + 1); 8732b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 8742b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // Record the call site value for the back end so it stays associated with 8752b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // the invoke. 8762b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling CallInst::Create(CallSiteFn, CallSiteNum, "", Invokes[I]); 8772b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling } 8782b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 8792b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // Mark call instructions that aren't nounwind as no-action (call_site == 8802b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // -1). Skip the entry block, as prior to then, no function context has been 8812b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // created for this function and any unexpected exceptions thrown will go 8822b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // directly to the caller's context, which is what we want anyway, so no need 8832b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling // to do anything here. 8842b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling for (Function::iterator BB = F.begin(), E = F.end(); ++BB != E;) { 8852b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling for (BasicBlock::iterator I = BB->begin(), end = BB->end(); I != end; ++I) 8862b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling if (CallInst *CI = dyn_cast<CallInst>(I)) { 8872b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling if (!CI->doesNotThrow()) 8882b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling insertCallSiteStore(CI, -1, CallSite); 8892b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling } else if (ResumeInst *RI = dyn_cast<ResumeInst>(I)) { 8902b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling insertCallSiteStore(RI, -1, CallSite); 8912b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling } 8922b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling } 8932b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 894cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling // Register the function context and make sure it's known to not throw 895cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling CallInst *Register = CallInst::Create(RegisterFn, FuncCtx.first, "", 896cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling EntryBB->getTerminator()); 897cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling Register->setDoesNotThrow(); 898cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling 899cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling // Finally, for any returns from this function, if this function contains an 900cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling // invoke, add a call to unregister the function context. 901cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling for (unsigned I = 0, E = Returns.size(); I != E; ++I) 902cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling CallInst::Create(UnregisterFn, FuncCtx.first, "", Returns[I]); 903cc8cf97be434b4e2b9c0388db1bfb709f09098ebBill Wendling 9042b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling return true; 9052b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling} 9062b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling 9078b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbachbool SjLjEHPass::runOnFunction(Function &F) { 9082b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling bool Res = false; 9092b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling if (!DisableOldSjLjEH) 9102b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Res = insertSjLjEHSupport(F); 9112b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling else 9122b6bd7ba5819d8c73c89291c6de89a86dbbffe9cBill Wendling Res = setupEntryBlockAndCallSites(F); 9138b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach return Res; 9148b818d7e98309125c6058c4ea72a7dc73b031db2Jim Grosbach} 915