10b57cec5SDimitry Andric //===- InlineFunction.cpp - Code to perform function inlining -------------===// 20b57cec5SDimitry Andric // 30b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 40b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information. 50b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 60b57cec5SDimitry Andric // 70b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 80b57cec5SDimitry Andric // 90b57cec5SDimitry Andric // This file implements inlining of a function into a call site, resolving 100b57cec5SDimitry Andric // parameters and the return value as appropriate. 110b57cec5SDimitry Andric // 120b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 130b57cec5SDimitry Andric 140b57cec5SDimitry Andric #include "llvm/ADT/DenseMap.h" 150b57cec5SDimitry Andric #include "llvm/ADT/None.h" 160b57cec5SDimitry Andric #include "llvm/ADT/Optional.h" 170b57cec5SDimitry Andric #include "llvm/ADT/STLExtras.h" 180b57cec5SDimitry Andric #include "llvm/ADT/SetVector.h" 190b57cec5SDimitry Andric #include "llvm/ADT/SmallPtrSet.h" 200b57cec5SDimitry Andric #include "llvm/ADT/SmallVector.h" 210b57cec5SDimitry Andric #include "llvm/ADT/StringExtras.h" 220b57cec5SDimitry Andric #include "llvm/ADT/iterator_range.h" 230b57cec5SDimitry Andric #include "llvm/Analysis/AliasAnalysis.h" 240b57cec5SDimitry Andric #include "llvm/Analysis/AssumptionCache.h" 250b57cec5SDimitry Andric #include "llvm/Analysis/BlockFrequencyInfo.h" 260b57cec5SDimitry Andric #include "llvm/Analysis/CallGraph.h" 270b57cec5SDimitry Andric #include "llvm/Analysis/CaptureTracking.h" 280b57cec5SDimitry Andric #include "llvm/Analysis/EHPersonalities.h" 290b57cec5SDimitry Andric #include "llvm/Analysis/InstructionSimplify.h" 300b57cec5SDimitry Andric #include "llvm/Analysis/ProfileSummaryInfo.h" 310b57cec5SDimitry Andric #include "llvm/Transforms/Utils/Local.h" 320b57cec5SDimitry Andric #include "llvm/Analysis/ValueTracking.h" 330b57cec5SDimitry Andric #include "llvm/Analysis/VectorUtils.h" 340b57cec5SDimitry Andric #include "llvm/IR/Argument.h" 350b57cec5SDimitry Andric #include "llvm/IR/BasicBlock.h" 360b57cec5SDimitry Andric #include "llvm/IR/CFG.h" 370b57cec5SDimitry Andric #include "llvm/IR/Constant.h" 380b57cec5SDimitry Andric #include "llvm/IR/Constants.h" 390b57cec5SDimitry Andric #include "llvm/IR/DIBuilder.h" 400b57cec5SDimitry Andric #include "llvm/IR/DataLayout.h" 410b57cec5SDimitry Andric #include "llvm/IR/DebugInfoMetadata.h" 420b57cec5SDimitry Andric #include "llvm/IR/DebugLoc.h" 430b57cec5SDimitry Andric #include "llvm/IR/DerivedTypes.h" 440b57cec5SDimitry Andric #include "llvm/IR/Dominators.h" 450b57cec5SDimitry Andric #include "llvm/IR/Function.h" 460b57cec5SDimitry Andric #include "llvm/IR/IRBuilder.h" 470b57cec5SDimitry Andric #include "llvm/IR/InstrTypes.h" 480b57cec5SDimitry Andric #include "llvm/IR/Instruction.h" 490b57cec5SDimitry Andric #include "llvm/IR/Instructions.h" 500b57cec5SDimitry Andric #include "llvm/IR/IntrinsicInst.h" 510b57cec5SDimitry Andric #include "llvm/IR/Intrinsics.h" 520b57cec5SDimitry Andric #include "llvm/IR/LLVMContext.h" 530b57cec5SDimitry Andric #include "llvm/IR/MDBuilder.h" 540b57cec5SDimitry Andric #include "llvm/IR/Metadata.h" 550b57cec5SDimitry Andric #include "llvm/IR/Module.h" 560b57cec5SDimitry Andric #include "llvm/IR/Type.h" 570b57cec5SDimitry Andric #include "llvm/IR/User.h" 580b57cec5SDimitry Andric #include "llvm/IR/Value.h" 590b57cec5SDimitry Andric #include "llvm/Support/Casting.h" 600b57cec5SDimitry Andric #include "llvm/Support/CommandLine.h" 610b57cec5SDimitry Andric #include "llvm/Support/ErrorHandling.h" 625ffd83dbSDimitry Andric #include "llvm/Transforms/Utils/AssumeBundleBuilder.h" 630b57cec5SDimitry Andric #include "llvm/Transforms/Utils/Cloning.h" 640b57cec5SDimitry Andric #include "llvm/Transforms/Utils/ValueMapper.h" 650b57cec5SDimitry Andric #include <algorithm> 660b57cec5SDimitry Andric #include <cassert> 670b57cec5SDimitry Andric #include <cstdint> 680b57cec5SDimitry Andric #include <iterator> 690b57cec5SDimitry Andric #include <limits> 700b57cec5SDimitry Andric #include <string> 710b57cec5SDimitry Andric #include <utility> 720b57cec5SDimitry Andric #include <vector> 730b57cec5SDimitry Andric 740b57cec5SDimitry Andric using namespace llvm; 750b57cec5SDimitry Andric using ProfileCount = Function::ProfileCount; 760b57cec5SDimitry Andric 770b57cec5SDimitry Andric static cl::opt<bool> 780b57cec5SDimitry Andric EnableNoAliasConversion("enable-noalias-to-md-conversion", cl::init(true), 790b57cec5SDimitry Andric cl::Hidden, 800b57cec5SDimitry Andric cl::desc("Convert noalias attributes to metadata during inlining.")); 810b57cec5SDimitry Andric 82e8d8bef9SDimitry Andric static cl::opt<bool> 83e8d8bef9SDimitry Andric UseNoAliasIntrinsic("use-noalias-intrinsic-during-inlining", cl::Hidden, 84e8d8bef9SDimitry Andric cl::ZeroOrMore, cl::init(true), 85e8d8bef9SDimitry Andric cl::desc("Use the llvm.experimental.noalias.scope.decl " 86e8d8bef9SDimitry Andric "intrinsic during inlining.")); 87e8d8bef9SDimitry Andric 885ffd83dbSDimitry Andric // Disabled by default, because the added alignment assumptions may increase 895ffd83dbSDimitry Andric // compile-time and block optimizations. This option is not suitable for use 905ffd83dbSDimitry Andric // with frontends that emit comprehensive parameter alignment annotations. 910b57cec5SDimitry Andric static cl::opt<bool> 920b57cec5SDimitry Andric PreserveAlignmentAssumptions("preserve-alignment-assumptions-during-inlining", 935ffd83dbSDimitry Andric cl::init(false), cl::Hidden, 940b57cec5SDimitry Andric cl::desc("Convert align attributes to assumptions during inlining.")); 950b57cec5SDimitry Andric 965ffd83dbSDimitry Andric static cl::opt<bool> UpdateReturnAttributes( 975ffd83dbSDimitry Andric "update-return-attrs", cl::init(true), cl::Hidden, 985ffd83dbSDimitry Andric cl::desc("Update return attributes on calls within inlined body")); 995ffd83dbSDimitry Andric 1005ffd83dbSDimitry Andric static cl::opt<unsigned> InlinerAttributeWindow( 1015ffd83dbSDimitry Andric "max-inst-checked-for-throw-during-inlining", cl::Hidden, 1025ffd83dbSDimitry Andric cl::desc("the maximum number of instructions analyzed for may throw during " 1035ffd83dbSDimitry Andric "attribute inference in inlined body"), 1045ffd83dbSDimitry Andric cl::init(4)); 1050b57cec5SDimitry Andric 1060b57cec5SDimitry Andric namespace { 1070b57cec5SDimitry Andric 1080b57cec5SDimitry Andric /// A class for recording information about inlining a landing pad. 1090b57cec5SDimitry Andric class LandingPadInliningInfo { 1100b57cec5SDimitry Andric /// Destination of the invoke's unwind. 1110b57cec5SDimitry Andric BasicBlock *OuterResumeDest; 1120b57cec5SDimitry Andric 1130b57cec5SDimitry Andric /// Destination for the callee's resume. 1140b57cec5SDimitry Andric BasicBlock *InnerResumeDest = nullptr; 1150b57cec5SDimitry Andric 1160b57cec5SDimitry Andric /// LandingPadInst associated with the invoke. 1170b57cec5SDimitry Andric LandingPadInst *CallerLPad = nullptr; 1180b57cec5SDimitry Andric 1190b57cec5SDimitry Andric /// PHI for EH values from landingpad insts. 1200b57cec5SDimitry Andric PHINode *InnerEHValuesPHI = nullptr; 1210b57cec5SDimitry Andric 1220b57cec5SDimitry Andric SmallVector<Value*, 8> UnwindDestPHIValues; 1230b57cec5SDimitry Andric 1240b57cec5SDimitry Andric public: 1250b57cec5SDimitry Andric LandingPadInliningInfo(InvokeInst *II) 1260b57cec5SDimitry Andric : OuterResumeDest(II->getUnwindDest()) { 1270b57cec5SDimitry Andric // If there are PHI nodes in the unwind destination block, we need to keep 1280b57cec5SDimitry Andric // track of which values came into them from the invoke before removing 1290b57cec5SDimitry Andric // the edge from this block. 1300b57cec5SDimitry Andric BasicBlock *InvokeBB = II->getParent(); 1310b57cec5SDimitry Andric BasicBlock::iterator I = OuterResumeDest->begin(); 1320b57cec5SDimitry Andric for (; isa<PHINode>(I); ++I) { 1330b57cec5SDimitry Andric // Save the value to use for this edge. 1340b57cec5SDimitry Andric PHINode *PHI = cast<PHINode>(I); 1350b57cec5SDimitry Andric UnwindDestPHIValues.push_back(PHI->getIncomingValueForBlock(InvokeBB)); 1360b57cec5SDimitry Andric } 1370b57cec5SDimitry Andric 1380b57cec5SDimitry Andric CallerLPad = cast<LandingPadInst>(I); 1390b57cec5SDimitry Andric } 1400b57cec5SDimitry Andric 1410b57cec5SDimitry Andric /// The outer unwind destination is the target of 1420b57cec5SDimitry Andric /// unwind edges introduced for calls within the inlined function. 1430b57cec5SDimitry Andric BasicBlock *getOuterResumeDest() const { 1440b57cec5SDimitry Andric return OuterResumeDest; 1450b57cec5SDimitry Andric } 1460b57cec5SDimitry Andric 1470b57cec5SDimitry Andric BasicBlock *getInnerResumeDest(); 1480b57cec5SDimitry Andric 1490b57cec5SDimitry Andric LandingPadInst *getLandingPadInst() const { return CallerLPad; } 1500b57cec5SDimitry Andric 1510b57cec5SDimitry Andric /// Forward the 'resume' instruction to the caller's landing pad block. 1520b57cec5SDimitry Andric /// When the landing pad block has only one predecessor, this is 1530b57cec5SDimitry Andric /// a simple branch. When there is more than one predecessor, we need to 1540b57cec5SDimitry Andric /// split the landing pad block after the landingpad instruction and jump 1550b57cec5SDimitry Andric /// to there. 1560b57cec5SDimitry Andric void forwardResume(ResumeInst *RI, 1570b57cec5SDimitry Andric SmallPtrSetImpl<LandingPadInst*> &InlinedLPads); 1580b57cec5SDimitry Andric 1590b57cec5SDimitry Andric /// Add incoming-PHI values to the unwind destination block for the given 1600b57cec5SDimitry Andric /// basic block, using the values for the original invoke's source block. 1610b57cec5SDimitry Andric void addIncomingPHIValuesFor(BasicBlock *BB) const { 1620b57cec5SDimitry Andric addIncomingPHIValuesForInto(BB, OuterResumeDest); 1630b57cec5SDimitry Andric } 1640b57cec5SDimitry Andric 1650b57cec5SDimitry Andric void addIncomingPHIValuesForInto(BasicBlock *src, BasicBlock *dest) const { 1660b57cec5SDimitry Andric BasicBlock::iterator I = dest->begin(); 1670b57cec5SDimitry Andric for (unsigned i = 0, e = UnwindDestPHIValues.size(); i != e; ++i, ++I) { 1680b57cec5SDimitry Andric PHINode *phi = cast<PHINode>(I); 1690b57cec5SDimitry Andric phi->addIncoming(UnwindDestPHIValues[i], src); 1700b57cec5SDimitry Andric } 1710b57cec5SDimitry Andric } 1720b57cec5SDimitry Andric }; 1730b57cec5SDimitry Andric 1740b57cec5SDimitry Andric } // end anonymous namespace 1750b57cec5SDimitry Andric 1760b57cec5SDimitry Andric /// Get or create a target for the branch from ResumeInsts. 1770b57cec5SDimitry Andric BasicBlock *LandingPadInliningInfo::getInnerResumeDest() { 1780b57cec5SDimitry Andric if (InnerResumeDest) return InnerResumeDest; 1790b57cec5SDimitry Andric 1800b57cec5SDimitry Andric // Split the landing pad. 1810b57cec5SDimitry Andric BasicBlock::iterator SplitPoint = ++CallerLPad->getIterator(); 1820b57cec5SDimitry Andric InnerResumeDest = 1830b57cec5SDimitry Andric OuterResumeDest->splitBasicBlock(SplitPoint, 1840b57cec5SDimitry Andric OuterResumeDest->getName() + ".body"); 1850b57cec5SDimitry Andric 1860b57cec5SDimitry Andric // The number of incoming edges we expect to the inner landing pad. 1870b57cec5SDimitry Andric const unsigned PHICapacity = 2; 1880b57cec5SDimitry Andric 1890b57cec5SDimitry Andric // Create corresponding new PHIs for all the PHIs in the outer landing pad. 1900b57cec5SDimitry Andric Instruction *InsertPoint = &InnerResumeDest->front(); 1910b57cec5SDimitry Andric BasicBlock::iterator I = OuterResumeDest->begin(); 1920b57cec5SDimitry Andric for (unsigned i = 0, e = UnwindDestPHIValues.size(); i != e; ++i, ++I) { 1930b57cec5SDimitry Andric PHINode *OuterPHI = cast<PHINode>(I); 1940b57cec5SDimitry Andric PHINode *InnerPHI = PHINode::Create(OuterPHI->getType(), PHICapacity, 1950b57cec5SDimitry Andric OuterPHI->getName() + ".lpad-body", 1960b57cec5SDimitry Andric InsertPoint); 1970b57cec5SDimitry Andric OuterPHI->replaceAllUsesWith(InnerPHI); 1980b57cec5SDimitry Andric InnerPHI->addIncoming(OuterPHI, OuterResumeDest); 1990b57cec5SDimitry Andric } 2000b57cec5SDimitry Andric 2010b57cec5SDimitry Andric // Create a PHI for the exception values. 2020b57cec5SDimitry Andric InnerEHValuesPHI = PHINode::Create(CallerLPad->getType(), PHICapacity, 2030b57cec5SDimitry Andric "eh.lpad-body", InsertPoint); 2040b57cec5SDimitry Andric CallerLPad->replaceAllUsesWith(InnerEHValuesPHI); 2050b57cec5SDimitry Andric InnerEHValuesPHI->addIncoming(CallerLPad, OuterResumeDest); 2060b57cec5SDimitry Andric 2070b57cec5SDimitry Andric // All done. 2080b57cec5SDimitry Andric return InnerResumeDest; 2090b57cec5SDimitry Andric } 2100b57cec5SDimitry Andric 2110b57cec5SDimitry Andric /// Forward the 'resume' instruction to the caller's landing pad block. 2120b57cec5SDimitry Andric /// When the landing pad block has only one predecessor, this is a simple 2130b57cec5SDimitry Andric /// branch. When there is more than one predecessor, we need to split the 2140b57cec5SDimitry Andric /// landing pad block after the landingpad instruction and jump to there. 2150b57cec5SDimitry Andric void LandingPadInliningInfo::forwardResume( 2160b57cec5SDimitry Andric ResumeInst *RI, SmallPtrSetImpl<LandingPadInst *> &InlinedLPads) { 2170b57cec5SDimitry Andric BasicBlock *Dest = getInnerResumeDest(); 2180b57cec5SDimitry Andric BasicBlock *Src = RI->getParent(); 2190b57cec5SDimitry Andric 2200b57cec5SDimitry Andric BranchInst::Create(Dest, Src); 2210b57cec5SDimitry Andric 2220b57cec5SDimitry Andric // Update the PHIs in the destination. They were inserted in an order which 2230b57cec5SDimitry Andric // makes this work. 2240b57cec5SDimitry Andric addIncomingPHIValuesForInto(Src, Dest); 2250b57cec5SDimitry Andric 2260b57cec5SDimitry Andric InnerEHValuesPHI->addIncoming(RI->getOperand(0), Src); 2270b57cec5SDimitry Andric RI->eraseFromParent(); 2280b57cec5SDimitry Andric } 2290b57cec5SDimitry Andric 2300b57cec5SDimitry Andric /// Helper for getUnwindDestToken/getUnwindDestTokenHelper. 2310b57cec5SDimitry Andric static Value *getParentPad(Value *EHPad) { 2320b57cec5SDimitry Andric if (auto *FPI = dyn_cast<FuncletPadInst>(EHPad)) 2330b57cec5SDimitry Andric return FPI->getParentPad(); 2340b57cec5SDimitry Andric return cast<CatchSwitchInst>(EHPad)->getParentPad(); 2350b57cec5SDimitry Andric } 2360b57cec5SDimitry Andric 2370b57cec5SDimitry Andric using UnwindDestMemoTy = DenseMap<Instruction *, Value *>; 2380b57cec5SDimitry Andric 2390b57cec5SDimitry Andric /// Helper for getUnwindDestToken that does the descendant-ward part of 2400b57cec5SDimitry Andric /// the search. 2410b57cec5SDimitry Andric static Value *getUnwindDestTokenHelper(Instruction *EHPad, 2420b57cec5SDimitry Andric UnwindDestMemoTy &MemoMap) { 2430b57cec5SDimitry Andric SmallVector<Instruction *, 8> Worklist(1, EHPad); 2440b57cec5SDimitry Andric 2450b57cec5SDimitry Andric while (!Worklist.empty()) { 2460b57cec5SDimitry Andric Instruction *CurrentPad = Worklist.pop_back_val(); 2470b57cec5SDimitry Andric // We only put pads on the worklist that aren't in the MemoMap. When 2480b57cec5SDimitry Andric // we find an unwind dest for a pad we may update its ancestors, but 2490b57cec5SDimitry Andric // the queue only ever contains uncles/great-uncles/etc. of CurrentPad, 2500b57cec5SDimitry Andric // so they should never get updated while queued on the worklist. 2510b57cec5SDimitry Andric assert(!MemoMap.count(CurrentPad)); 2520b57cec5SDimitry Andric Value *UnwindDestToken = nullptr; 2530b57cec5SDimitry Andric if (auto *CatchSwitch = dyn_cast<CatchSwitchInst>(CurrentPad)) { 2540b57cec5SDimitry Andric if (CatchSwitch->hasUnwindDest()) { 2550b57cec5SDimitry Andric UnwindDestToken = CatchSwitch->getUnwindDest()->getFirstNonPHI(); 2560b57cec5SDimitry Andric } else { 2570b57cec5SDimitry Andric // Catchswitch doesn't have a 'nounwind' variant, and one might be 2580b57cec5SDimitry Andric // annotated as "unwinds to caller" when really it's nounwind (see 2590b57cec5SDimitry Andric // e.g. SimplifyCFGOpt::SimplifyUnreachable), so we can't infer the 2600b57cec5SDimitry Andric // parent's unwind dest from this. We can check its catchpads' 2610b57cec5SDimitry Andric // descendants, since they might include a cleanuppad with an 2620b57cec5SDimitry Andric // "unwinds to caller" cleanupret, which can be trusted. 2630b57cec5SDimitry Andric for (auto HI = CatchSwitch->handler_begin(), 2640b57cec5SDimitry Andric HE = CatchSwitch->handler_end(); 2650b57cec5SDimitry Andric HI != HE && !UnwindDestToken; ++HI) { 2660b57cec5SDimitry Andric BasicBlock *HandlerBlock = *HI; 2670b57cec5SDimitry Andric auto *CatchPad = cast<CatchPadInst>(HandlerBlock->getFirstNonPHI()); 2680b57cec5SDimitry Andric for (User *Child : CatchPad->users()) { 2690b57cec5SDimitry Andric // Intentionally ignore invokes here -- since the catchswitch is 2700b57cec5SDimitry Andric // marked "unwind to caller", it would be a verifier error if it 2710b57cec5SDimitry Andric // contained an invoke which unwinds out of it, so any invoke we'd 2720b57cec5SDimitry Andric // encounter must unwind to some child of the catch. 2730b57cec5SDimitry Andric if (!isa<CleanupPadInst>(Child) && !isa<CatchSwitchInst>(Child)) 2740b57cec5SDimitry Andric continue; 2750b57cec5SDimitry Andric 2760b57cec5SDimitry Andric Instruction *ChildPad = cast<Instruction>(Child); 2770b57cec5SDimitry Andric auto Memo = MemoMap.find(ChildPad); 2780b57cec5SDimitry Andric if (Memo == MemoMap.end()) { 2790b57cec5SDimitry Andric // Haven't figured out this child pad yet; queue it. 2800b57cec5SDimitry Andric Worklist.push_back(ChildPad); 2810b57cec5SDimitry Andric continue; 2820b57cec5SDimitry Andric } 2830b57cec5SDimitry Andric // We've already checked this child, but might have found that 2840b57cec5SDimitry Andric // it offers no proof either way. 2850b57cec5SDimitry Andric Value *ChildUnwindDestToken = Memo->second; 2860b57cec5SDimitry Andric if (!ChildUnwindDestToken) 2870b57cec5SDimitry Andric continue; 2880b57cec5SDimitry Andric // We already know the child's unwind dest, which can either 2890b57cec5SDimitry Andric // be ConstantTokenNone to indicate unwind to caller, or can 2900b57cec5SDimitry Andric // be another child of the catchpad. Only the former indicates 2910b57cec5SDimitry Andric // the unwind dest of the catchswitch. 2920b57cec5SDimitry Andric if (isa<ConstantTokenNone>(ChildUnwindDestToken)) { 2930b57cec5SDimitry Andric UnwindDestToken = ChildUnwindDestToken; 2940b57cec5SDimitry Andric break; 2950b57cec5SDimitry Andric } 2960b57cec5SDimitry Andric assert(getParentPad(ChildUnwindDestToken) == CatchPad); 2970b57cec5SDimitry Andric } 2980b57cec5SDimitry Andric } 2990b57cec5SDimitry Andric } 3000b57cec5SDimitry Andric } else { 3010b57cec5SDimitry Andric auto *CleanupPad = cast<CleanupPadInst>(CurrentPad); 3020b57cec5SDimitry Andric for (User *U : CleanupPad->users()) { 3030b57cec5SDimitry Andric if (auto *CleanupRet = dyn_cast<CleanupReturnInst>(U)) { 3040b57cec5SDimitry Andric if (BasicBlock *RetUnwindDest = CleanupRet->getUnwindDest()) 3050b57cec5SDimitry Andric UnwindDestToken = RetUnwindDest->getFirstNonPHI(); 3060b57cec5SDimitry Andric else 3070b57cec5SDimitry Andric UnwindDestToken = ConstantTokenNone::get(CleanupPad->getContext()); 3080b57cec5SDimitry Andric break; 3090b57cec5SDimitry Andric } 3100b57cec5SDimitry Andric Value *ChildUnwindDestToken; 3110b57cec5SDimitry Andric if (auto *Invoke = dyn_cast<InvokeInst>(U)) { 3120b57cec5SDimitry Andric ChildUnwindDestToken = Invoke->getUnwindDest()->getFirstNonPHI(); 3130b57cec5SDimitry Andric } else if (isa<CleanupPadInst>(U) || isa<CatchSwitchInst>(U)) { 3140b57cec5SDimitry Andric Instruction *ChildPad = cast<Instruction>(U); 3150b57cec5SDimitry Andric auto Memo = MemoMap.find(ChildPad); 3160b57cec5SDimitry Andric if (Memo == MemoMap.end()) { 3170b57cec5SDimitry Andric // Haven't resolved this child yet; queue it and keep searching. 3180b57cec5SDimitry Andric Worklist.push_back(ChildPad); 3190b57cec5SDimitry Andric continue; 3200b57cec5SDimitry Andric } 3210b57cec5SDimitry Andric // We've checked this child, but still need to ignore it if it 3220b57cec5SDimitry Andric // had no proof either way. 3230b57cec5SDimitry Andric ChildUnwindDestToken = Memo->second; 3240b57cec5SDimitry Andric if (!ChildUnwindDestToken) 3250b57cec5SDimitry Andric continue; 3260b57cec5SDimitry Andric } else { 3270b57cec5SDimitry Andric // Not a relevant user of the cleanuppad 3280b57cec5SDimitry Andric continue; 3290b57cec5SDimitry Andric } 3300b57cec5SDimitry Andric // In a well-formed program, the child/invoke must either unwind to 3310b57cec5SDimitry Andric // an(other) child of the cleanup, or exit the cleanup. In the 3320b57cec5SDimitry Andric // first case, continue searching. 3330b57cec5SDimitry Andric if (isa<Instruction>(ChildUnwindDestToken) && 3340b57cec5SDimitry Andric getParentPad(ChildUnwindDestToken) == CleanupPad) 3350b57cec5SDimitry Andric continue; 3360b57cec5SDimitry Andric UnwindDestToken = ChildUnwindDestToken; 3370b57cec5SDimitry Andric break; 3380b57cec5SDimitry Andric } 3390b57cec5SDimitry Andric } 3400b57cec5SDimitry Andric // If we haven't found an unwind dest for CurrentPad, we may have queued its 3410b57cec5SDimitry Andric // children, so move on to the next in the worklist. 3420b57cec5SDimitry Andric if (!UnwindDestToken) 3430b57cec5SDimitry Andric continue; 3440b57cec5SDimitry Andric 3450b57cec5SDimitry Andric // Now we know that CurrentPad unwinds to UnwindDestToken. It also exits 3460b57cec5SDimitry Andric // any ancestors of CurrentPad up to but not including UnwindDestToken's 3470b57cec5SDimitry Andric // parent pad. Record this in the memo map, and check to see if the 3480b57cec5SDimitry Andric // original EHPad being queried is one of the ones exited. 3490b57cec5SDimitry Andric Value *UnwindParent; 3500b57cec5SDimitry Andric if (auto *UnwindPad = dyn_cast<Instruction>(UnwindDestToken)) 3510b57cec5SDimitry Andric UnwindParent = getParentPad(UnwindPad); 3520b57cec5SDimitry Andric else 3530b57cec5SDimitry Andric UnwindParent = nullptr; 3540b57cec5SDimitry Andric bool ExitedOriginalPad = false; 3550b57cec5SDimitry Andric for (Instruction *ExitedPad = CurrentPad; 3560b57cec5SDimitry Andric ExitedPad && ExitedPad != UnwindParent; 3570b57cec5SDimitry Andric ExitedPad = dyn_cast<Instruction>(getParentPad(ExitedPad))) { 3580b57cec5SDimitry Andric // Skip over catchpads since they just follow their catchswitches. 3590b57cec5SDimitry Andric if (isa<CatchPadInst>(ExitedPad)) 3600b57cec5SDimitry Andric continue; 3610b57cec5SDimitry Andric MemoMap[ExitedPad] = UnwindDestToken; 3620b57cec5SDimitry Andric ExitedOriginalPad |= (ExitedPad == EHPad); 3630b57cec5SDimitry Andric } 3640b57cec5SDimitry Andric 3650b57cec5SDimitry Andric if (ExitedOriginalPad) 3660b57cec5SDimitry Andric return UnwindDestToken; 3670b57cec5SDimitry Andric 3680b57cec5SDimitry Andric // Continue the search. 3690b57cec5SDimitry Andric } 3700b57cec5SDimitry Andric 3710b57cec5SDimitry Andric // No definitive information is contained within this funclet. 3720b57cec5SDimitry Andric return nullptr; 3730b57cec5SDimitry Andric } 3740b57cec5SDimitry Andric 3750b57cec5SDimitry Andric /// Given an EH pad, find where it unwinds. If it unwinds to an EH pad, 3760b57cec5SDimitry Andric /// return that pad instruction. If it unwinds to caller, return 3770b57cec5SDimitry Andric /// ConstantTokenNone. If it does not have a definitive unwind destination, 3780b57cec5SDimitry Andric /// return nullptr. 3790b57cec5SDimitry Andric /// 3800b57cec5SDimitry Andric /// This routine gets invoked for calls in funclets in inlinees when inlining 3810b57cec5SDimitry Andric /// an invoke. Since many funclets don't have calls inside them, it's queried 3820b57cec5SDimitry Andric /// on-demand rather than building a map of pads to unwind dests up front. 3830b57cec5SDimitry Andric /// Determining a funclet's unwind dest may require recursively searching its 3840b57cec5SDimitry Andric /// descendants, and also ancestors and cousins if the descendants don't provide 3850b57cec5SDimitry Andric /// an answer. Since most funclets will have their unwind dest immediately 3860b57cec5SDimitry Andric /// available as the unwind dest of a catchswitch or cleanupret, this routine 3870b57cec5SDimitry Andric /// searches top-down from the given pad and then up. To avoid worst-case 3880b57cec5SDimitry Andric /// quadratic run-time given that approach, it uses a memo map to avoid 3890b57cec5SDimitry Andric /// re-processing funclet trees. The callers that rewrite the IR as they go 3900b57cec5SDimitry Andric /// take advantage of this, for correctness, by checking/forcing rewritten 3910b57cec5SDimitry Andric /// pads' entries to match the original callee view. 3920b57cec5SDimitry Andric static Value *getUnwindDestToken(Instruction *EHPad, 3930b57cec5SDimitry Andric UnwindDestMemoTy &MemoMap) { 3940b57cec5SDimitry Andric // Catchpads unwind to the same place as their catchswitch; 3950b57cec5SDimitry Andric // redirct any queries on catchpads so the code below can 3960b57cec5SDimitry Andric // deal with just catchswitches and cleanuppads. 3970b57cec5SDimitry Andric if (auto *CPI = dyn_cast<CatchPadInst>(EHPad)) 3980b57cec5SDimitry Andric EHPad = CPI->getCatchSwitch(); 3990b57cec5SDimitry Andric 4000b57cec5SDimitry Andric // Check if we've already determined the unwind dest for this pad. 4010b57cec5SDimitry Andric auto Memo = MemoMap.find(EHPad); 4020b57cec5SDimitry Andric if (Memo != MemoMap.end()) 4030b57cec5SDimitry Andric return Memo->second; 4040b57cec5SDimitry Andric 4050b57cec5SDimitry Andric // Search EHPad and, if necessary, its descendants. 4060b57cec5SDimitry Andric Value *UnwindDestToken = getUnwindDestTokenHelper(EHPad, MemoMap); 4070b57cec5SDimitry Andric assert((UnwindDestToken == nullptr) != (MemoMap.count(EHPad) != 0)); 4080b57cec5SDimitry Andric if (UnwindDestToken) 4090b57cec5SDimitry Andric return UnwindDestToken; 4100b57cec5SDimitry Andric 4110b57cec5SDimitry Andric // No information is available for this EHPad from itself or any of its 4120b57cec5SDimitry Andric // descendants. An unwind all the way out to a pad in the caller would 4130b57cec5SDimitry Andric // need also to agree with the unwind dest of the parent funclet, so 4140b57cec5SDimitry Andric // search up the chain to try to find a funclet with information. Put 4150b57cec5SDimitry Andric // null entries in the memo map to avoid re-processing as we go up. 4160b57cec5SDimitry Andric MemoMap[EHPad] = nullptr; 4170b57cec5SDimitry Andric #ifndef NDEBUG 4180b57cec5SDimitry Andric SmallPtrSet<Instruction *, 4> TempMemos; 4190b57cec5SDimitry Andric TempMemos.insert(EHPad); 4200b57cec5SDimitry Andric #endif 4210b57cec5SDimitry Andric Instruction *LastUselessPad = EHPad; 4220b57cec5SDimitry Andric Value *AncestorToken; 4230b57cec5SDimitry Andric for (AncestorToken = getParentPad(EHPad); 4240b57cec5SDimitry Andric auto *AncestorPad = dyn_cast<Instruction>(AncestorToken); 4250b57cec5SDimitry Andric AncestorToken = getParentPad(AncestorToken)) { 4260b57cec5SDimitry Andric // Skip over catchpads since they just follow their catchswitches. 4270b57cec5SDimitry Andric if (isa<CatchPadInst>(AncestorPad)) 4280b57cec5SDimitry Andric continue; 4290b57cec5SDimitry Andric // If the MemoMap had an entry mapping AncestorPad to nullptr, since we 4300b57cec5SDimitry Andric // haven't yet called getUnwindDestTokenHelper for AncestorPad in this 4310b57cec5SDimitry Andric // call to getUnwindDestToken, that would mean that AncestorPad had no 4320b57cec5SDimitry Andric // information in itself, its descendants, or its ancestors. If that 4330b57cec5SDimitry Andric // were the case, then we should also have recorded the lack of information 4340b57cec5SDimitry Andric // for the descendant that we're coming from. So assert that we don't 4350b57cec5SDimitry Andric // find a null entry in the MemoMap for AncestorPad. 4360b57cec5SDimitry Andric assert(!MemoMap.count(AncestorPad) || MemoMap[AncestorPad]); 4370b57cec5SDimitry Andric auto AncestorMemo = MemoMap.find(AncestorPad); 4380b57cec5SDimitry Andric if (AncestorMemo == MemoMap.end()) { 4390b57cec5SDimitry Andric UnwindDestToken = getUnwindDestTokenHelper(AncestorPad, MemoMap); 4400b57cec5SDimitry Andric } else { 4410b57cec5SDimitry Andric UnwindDestToken = AncestorMemo->second; 4420b57cec5SDimitry Andric } 4430b57cec5SDimitry Andric if (UnwindDestToken) 4440b57cec5SDimitry Andric break; 4450b57cec5SDimitry Andric LastUselessPad = AncestorPad; 4460b57cec5SDimitry Andric MemoMap[LastUselessPad] = nullptr; 4470b57cec5SDimitry Andric #ifndef NDEBUG 4480b57cec5SDimitry Andric TempMemos.insert(LastUselessPad); 4490b57cec5SDimitry Andric #endif 4500b57cec5SDimitry Andric } 4510b57cec5SDimitry Andric 4520b57cec5SDimitry Andric // We know that getUnwindDestTokenHelper was called on LastUselessPad and 4530b57cec5SDimitry Andric // returned nullptr (and likewise for EHPad and any of its ancestors up to 4540b57cec5SDimitry Andric // LastUselessPad), so LastUselessPad has no information from below. Since 4550b57cec5SDimitry Andric // getUnwindDestTokenHelper must investigate all downward paths through 4560b57cec5SDimitry Andric // no-information nodes to prove that a node has no information like this, 4570b57cec5SDimitry Andric // and since any time it finds information it records it in the MemoMap for 4580b57cec5SDimitry Andric // not just the immediately-containing funclet but also any ancestors also 4590b57cec5SDimitry Andric // exited, it must be the case that, walking downward from LastUselessPad, 4600b57cec5SDimitry Andric // visiting just those nodes which have not been mapped to an unwind dest 4610b57cec5SDimitry Andric // by getUnwindDestTokenHelper (the nullptr TempMemos notwithstanding, since 4620b57cec5SDimitry Andric // they are just used to keep getUnwindDestTokenHelper from repeating work), 4630b57cec5SDimitry Andric // any node visited must have been exhaustively searched with no information 4640b57cec5SDimitry Andric // for it found. 4650b57cec5SDimitry Andric SmallVector<Instruction *, 8> Worklist(1, LastUselessPad); 4660b57cec5SDimitry Andric while (!Worklist.empty()) { 4670b57cec5SDimitry Andric Instruction *UselessPad = Worklist.pop_back_val(); 4680b57cec5SDimitry Andric auto Memo = MemoMap.find(UselessPad); 4690b57cec5SDimitry Andric if (Memo != MemoMap.end() && Memo->second) { 4700b57cec5SDimitry Andric // Here the name 'UselessPad' is a bit of a misnomer, because we've found 4710b57cec5SDimitry Andric // that it is a funclet that does have information about unwinding to 4720b57cec5SDimitry Andric // a particular destination; its parent was a useless pad. 4730b57cec5SDimitry Andric // Since its parent has no information, the unwind edge must not escape 4740b57cec5SDimitry Andric // the parent, and must target a sibling of this pad. This local unwind 4750b57cec5SDimitry Andric // gives us no information about EHPad. Leave it and the subtree rooted 4760b57cec5SDimitry Andric // at it alone. 4770b57cec5SDimitry Andric assert(getParentPad(Memo->second) == getParentPad(UselessPad)); 4780b57cec5SDimitry Andric continue; 4790b57cec5SDimitry Andric } 4800b57cec5SDimitry Andric // We know we don't have information for UselesPad. If it has an entry in 4810b57cec5SDimitry Andric // the MemoMap (mapping it to nullptr), it must be one of the TempMemos 4820b57cec5SDimitry Andric // added on this invocation of getUnwindDestToken; if a previous invocation 4830b57cec5SDimitry Andric // recorded nullptr, it would have had to prove that the ancestors of 4840b57cec5SDimitry Andric // UselessPad, which include LastUselessPad, had no information, and that 4850b57cec5SDimitry Andric // in turn would have required proving that the descendants of 4860b57cec5SDimitry Andric // LastUselesPad, which include EHPad, have no information about 4870b57cec5SDimitry Andric // LastUselessPad, which would imply that EHPad was mapped to nullptr in 4880b57cec5SDimitry Andric // the MemoMap on that invocation, which isn't the case if we got here. 4890b57cec5SDimitry Andric assert(!MemoMap.count(UselessPad) || TempMemos.count(UselessPad)); 4900b57cec5SDimitry Andric // Assert as we enumerate users that 'UselessPad' doesn't have any unwind 4910b57cec5SDimitry Andric // information that we'd be contradicting by making a map entry for it 4920b57cec5SDimitry Andric // (which is something that getUnwindDestTokenHelper must have proved for 4930b57cec5SDimitry Andric // us to get here). Just assert on is direct users here; the checks in 4940b57cec5SDimitry Andric // this downward walk at its descendants will verify that they don't have 4950b57cec5SDimitry Andric // any unwind edges that exit 'UselessPad' either (i.e. they either have no 4960b57cec5SDimitry Andric // unwind edges or unwind to a sibling). 4970b57cec5SDimitry Andric MemoMap[UselessPad] = UnwindDestToken; 4980b57cec5SDimitry Andric if (auto *CatchSwitch = dyn_cast<CatchSwitchInst>(UselessPad)) { 4990b57cec5SDimitry Andric assert(CatchSwitch->getUnwindDest() == nullptr && "Expected useless pad"); 5000b57cec5SDimitry Andric for (BasicBlock *HandlerBlock : CatchSwitch->handlers()) { 5010b57cec5SDimitry Andric auto *CatchPad = HandlerBlock->getFirstNonPHI(); 5020b57cec5SDimitry Andric for (User *U : CatchPad->users()) { 5030b57cec5SDimitry Andric assert( 5040b57cec5SDimitry Andric (!isa<InvokeInst>(U) || 5050b57cec5SDimitry Andric (getParentPad( 5060b57cec5SDimitry Andric cast<InvokeInst>(U)->getUnwindDest()->getFirstNonPHI()) == 5070b57cec5SDimitry Andric CatchPad)) && 5080b57cec5SDimitry Andric "Expected useless pad"); 5090b57cec5SDimitry Andric if (isa<CatchSwitchInst>(U) || isa<CleanupPadInst>(U)) 5100b57cec5SDimitry Andric Worklist.push_back(cast<Instruction>(U)); 5110b57cec5SDimitry Andric } 5120b57cec5SDimitry Andric } 5130b57cec5SDimitry Andric } else { 5140b57cec5SDimitry Andric assert(isa<CleanupPadInst>(UselessPad)); 5150b57cec5SDimitry Andric for (User *U : UselessPad->users()) { 5160b57cec5SDimitry Andric assert(!isa<CleanupReturnInst>(U) && "Expected useless pad"); 5170b57cec5SDimitry Andric assert((!isa<InvokeInst>(U) || 5180b57cec5SDimitry Andric (getParentPad( 5190b57cec5SDimitry Andric cast<InvokeInst>(U)->getUnwindDest()->getFirstNonPHI()) == 5200b57cec5SDimitry Andric UselessPad)) && 5210b57cec5SDimitry Andric "Expected useless pad"); 5220b57cec5SDimitry Andric if (isa<CatchSwitchInst>(U) || isa<CleanupPadInst>(U)) 5230b57cec5SDimitry Andric Worklist.push_back(cast<Instruction>(U)); 5240b57cec5SDimitry Andric } 5250b57cec5SDimitry Andric } 5260b57cec5SDimitry Andric } 5270b57cec5SDimitry Andric 5280b57cec5SDimitry Andric return UnwindDestToken; 5290b57cec5SDimitry Andric } 5300b57cec5SDimitry Andric 5310b57cec5SDimitry Andric /// When we inline a basic block into an invoke, 5320b57cec5SDimitry Andric /// we have to turn all of the calls that can throw into invokes. 5330b57cec5SDimitry Andric /// This function analyze BB to see if there are any calls, and if so, 5340b57cec5SDimitry Andric /// it rewrites them to be invokes that jump to InvokeDest and fills in the PHI 5350b57cec5SDimitry Andric /// nodes in that block with the values specified in InvokeDestPHIValues. 5360b57cec5SDimitry Andric static BasicBlock *HandleCallsInBlockInlinedThroughInvoke( 5370b57cec5SDimitry Andric BasicBlock *BB, BasicBlock *UnwindEdge, 5380b57cec5SDimitry Andric UnwindDestMemoTy *FuncletUnwindMap = nullptr) { 5390b57cec5SDimitry Andric for (BasicBlock::iterator BBI = BB->begin(), E = BB->end(); BBI != E; ) { 5400b57cec5SDimitry Andric Instruction *I = &*BBI++; 5410b57cec5SDimitry Andric 5420b57cec5SDimitry Andric // We only need to check for function calls: inlined invoke 5430b57cec5SDimitry Andric // instructions require no special handling. 5440b57cec5SDimitry Andric CallInst *CI = dyn_cast<CallInst>(I); 5450b57cec5SDimitry Andric 5465ffd83dbSDimitry Andric if (!CI || CI->doesNotThrow() || CI->isInlineAsm()) 5470b57cec5SDimitry Andric continue; 5480b57cec5SDimitry Andric 5490b57cec5SDimitry Andric // We do not need to (and in fact, cannot) convert possibly throwing calls 5500b57cec5SDimitry Andric // to @llvm.experimental_deoptimize (resp. @llvm.experimental.guard) into 5510b57cec5SDimitry Andric // invokes. The caller's "segment" of the deoptimization continuation 5520b57cec5SDimitry Andric // attached to the newly inlined @llvm.experimental_deoptimize 5530b57cec5SDimitry Andric // (resp. @llvm.experimental.guard) call should contain the exception 5540b57cec5SDimitry Andric // handling logic, if any. 5550b57cec5SDimitry Andric if (auto *F = CI->getCalledFunction()) 5560b57cec5SDimitry Andric if (F->getIntrinsicID() == Intrinsic::experimental_deoptimize || 5570b57cec5SDimitry Andric F->getIntrinsicID() == Intrinsic::experimental_guard) 5580b57cec5SDimitry Andric continue; 5590b57cec5SDimitry Andric 5600b57cec5SDimitry Andric if (auto FuncletBundle = CI->getOperandBundle(LLVMContext::OB_funclet)) { 5610b57cec5SDimitry Andric // This call is nested inside a funclet. If that funclet has an unwind 5620b57cec5SDimitry Andric // destination within the inlinee, then unwinding out of this call would 5630b57cec5SDimitry Andric // be UB. Rewriting this call to an invoke which targets the inlined 5640b57cec5SDimitry Andric // invoke's unwind dest would give the call's parent funclet multiple 5650b57cec5SDimitry Andric // unwind destinations, which is something that subsequent EH table 5660b57cec5SDimitry Andric // generation can't handle and that the veirifer rejects. So when we 5670b57cec5SDimitry Andric // see such a call, leave it as a call. 5680b57cec5SDimitry Andric auto *FuncletPad = cast<Instruction>(FuncletBundle->Inputs[0]); 5690b57cec5SDimitry Andric Value *UnwindDestToken = 5700b57cec5SDimitry Andric getUnwindDestToken(FuncletPad, *FuncletUnwindMap); 5710b57cec5SDimitry Andric if (UnwindDestToken && !isa<ConstantTokenNone>(UnwindDestToken)) 5720b57cec5SDimitry Andric continue; 5730b57cec5SDimitry Andric #ifndef NDEBUG 5740b57cec5SDimitry Andric Instruction *MemoKey; 5750b57cec5SDimitry Andric if (auto *CatchPad = dyn_cast<CatchPadInst>(FuncletPad)) 5760b57cec5SDimitry Andric MemoKey = CatchPad->getCatchSwitch(); 5770b57cec5SDimitry Andric else 5780b57cec5SDimitry Andric MemoKey = FuncletPad; 5790b57cec5SDimitry Andric assert(FuncletUnwindMap->count(MemoKey) && 5800b57cec5SDimitry Andric (*FuncletUnwindMap)[MemoKey] == UnwindDestToken && 5810b57cec5SDimitry Andric "must get memoized to avoid confusing later searches"); 5820b57cec5SDimitry Andric #endif // NDEBUG 5830b57cec5SDimitry Andric } 5840b57cec5SDimitry Andric 5850b57cec5SDimitry Andric changeToInvokeAndSplitBasicBlock(CI, UnwindEdge); 5860b57cec5SDimitry Andric return BB; 5870b57cec5SDimitry Andric } 5880b57cec5SDimitry Andric return nullptr; 5890b57cec5SDimitry Andric } 5900b57cec5SDimitry Andric 5910b57cec5SDimitry Andric /// If we inlined an invoke site, we need to convert calls 5920b57cec5SDimitry Andric /// in the body of the inlined function into invokes. 5930b57cec5SDimitry Andric /// 5940b57cec5SDimitry Andric /// II is the invoke instruction being inlined. FirstNewBlock is the first 5950b57cec5SDimitry Andric /// block of the inlined code (the last block is the end of the function), 5960b57cec5SDimitry Andric /// and InlineCodeInfo is information about the code that got inlined. 5970b57cec5SDimitry Andric static void HandleInlinedLandingPad(InvokeInst *II, BasicBlock *FirstNewBlock, 5980b57cec5SDimitry Andric ClonedCodeInfo &InlinedCodeInfo) { 5990b57cec5SDimitry Andric BasicBlock *InvokeDest = II->getUnwindDest(); 6000b57cec5SDimitry Andric 6010b57cec5SDimitry Andric Function *Caller = FirstNewBlock->getParent(); 6020b57cec5SDimitry Andric 6030b57cec5SDimitry Andric // The inlined code is currently at the end of the function, scan from the 6040b57cec5SDimitry Andric // start of the inlined code to its end, checking for stuff we need to 6050b57cec5SDimitry Andric // rewrite. 6060b57cec5SDimitry Andric LandingPadInliningInfo Invoke(II); 6070b57cec5SDimitry Andric 6080b57cec5SDimitry Andric // Get all of the inlined landing pad instructions. 6090b57cec5SDimitry Andric SmallPtrSet<LandingPadInst*, 16> InlinedLPads; 6100b57cec5SDimitry Andric for (Function::iterator I = FirstNewBlock->getIterator(), E = Caller->end(); 6110b57cec5SDimitry Andric I != E; ++I) 6120b57cec5SDimitry Andric if (InvokeInst *II = dyn_cast<InvokeInst>(I->getTerminator())) 6130b57cec5SDimitry Andric InlinedLPads.insert(II->getLandingPadInst()); 6140b57cec5SDimitry Andric 6150b57cec5SDimitry Andric // Append the clauses from the outer landing pad instruction into the inlined 6160b57cec5SDimitry Andric // landing pad instructions. 6170b57cec5SDimitry Andric LandingPadInst *OuterLPad = Invoke.getLandingPadInst(); 6180b57cec5SDimitry Andric for (LandingPadInst *InlinedLPad : InlinedLPads) { 6190b57cec5SDimitry Andric unsigned OuterNum = OuterLPad->getNumClauses(); 6200b57cec5SDimitry Andric InlinedLPad->reserveClauses(OuterNum); 6210b57cec5SDimitry Andric for (unsigned OuterIdx = 0; OuterIdx != OuterNum; ++OuterIdx) 6220b57cec5SDimitry Andric InlinedLPad->addClause(OuterLPad->getClause(OuterIdx)); 6230b57cec5SDimitry Andric if (OuterLPad->isCleanup()) 6240b57cec5SDimitry Andric InlinedLPad->setCleanup(true); 6250b57cec5SDimitry Andric } 6260b57cec5SDimitry Andric 6270b57cec5SDimitry Andric for (Function::iterator BB = FirstNewBlock->getIterator(), E = Caller->end(); 6280b57cec5SDimitry Andric BB != E; ++BB) { 6290b57cec5SDimitry Andric if (InlinedCodeInfo.ContainsCalls) 6300b57cec5SDimitry Andric if (BasicBlock *NewBB = HandleCallsInBlockInlinedThroughInvoke( 6310b57cec5SDimitry Andric &*BB, Invoke.getOuterResumeDest())) 6320b57cec5SDimitry Andric // Update any PHI nodes in the exceptional block to indicate that there 6330b57cec5SDimitry Andric // is now a new entry in them. 6340b57cec5SDimitry Andric Invoke.addIncomingPHIValuesFor(NewBB); 6350b57cec5SDimitry Andric 6360b57cec5SDimitry Andric // Forward any resumes that are remaining here. 6370b57cec5SDimitry Andric if (ResumeInst *RI = dyn_cast<ResumeInst>(BB->getTerminator())) 6380b57cec5SDimitry Andric Invoke.forwardResume(RI, InlinedLPads); 6390b57cec5SDimitry Andric } 6400b57cec5SDimitry Andric 6410b57cec5SDimitry Andric // Now that everything is happy, we have one final detail. The PHI nodes in 6420b57cec5SDimitry Andric // the exception destination block still have entries due to the original 6430b57cec5SDimitry Andric // invoke instruction. Eliminate these entries (which might even delete the 6440b57cec5SDimitry Andric // PHI node) now. 6450b57cec5SDimitry Andric InvokeDest->removePredecessor(II->getParent()); 6460b57cec5SDimitry Andric } 6470b57cec5SDimitry Andric 6480b57cec5SDimitry Andric /// If we inlined an invoke site, we need to convert calls 6490b57cec5SDimitry Andric /// in the body of the inlined function into invokes. 6500b57cec5SDimitry Andric /// 6510b57cec5SDimitry Andric /// II is the invoke instruction being inlined. FirstNewBlock is the first 6520b57cec5SDimitry Andric /// block of the inlined code (the last block is the end of the function), 6530b57cec5SDimitry Andric /// and InlineCodeInfo is information about the code that got inlined. 6540b57cec5SDimitry Andric static void HandleInlinedEHPad(InvokeInst *II, BasicBlock *FirstNewBlock, 6550b57cec5SDimitry Andric ClonedCodeInfo &InlinedCodeInfo) { 6560b57cec5SDimitry Andric BasicBlock *UnwindDest = II->getUnwindDest(); 6570b57cec5SDimitry Andric Function *Caller = FirstNewBlock->getParent(); 6580b57cec5SDimitry Andric 6590b57cec5SDimitry Andric assert(UnwindDest->getFirstNonPHI()->isEHPad() && "unexpected BasicBlock!"); 6600b57cec5SDimitry Andric 6610b57cec5SDimitry Andric // If there are PHI nodes in the unwind destination block, we need to keep 6620b57cec5SDimitry Andric // track of which values came into them from the invoke before removing the 6630b57cec5SDimitry Andric // edge from this block. 6640b57cec5SDimitry Andric SmallVector<Value *, 8> UnwindDestPHIValues; 6650b57cec5SDimitry Andric BasicBlock *InvokeBB = II->getParent(); 6660b57cec5SDimitry Andric for (Instruction &I : *UnwindDest) { 6670b57cec5SDimitry Andric // Save the value to use for this edge. 6680b57cec5SDimitry Andric PHINode *PHI = dyn_cast<PHINode>(&I); 6690b57cec5SDimitry Andric if (!PHI) 6700b57cec5SDimitry Andric break; 6710b57cec5SDimitry Andric UnwindDestPHIValues.push_back(PHI->getIncomingValueForBlock(InvokeBB)); 6720b57cec5SDimitry Andric } 6730b57cec5SDimitry Andric 6740b57cec5SDimitry Andric // Add incoming-PHI values to the unwind destination block for the given basic 6750b57cec5SDimitry Andric // block, using the values for the original invoke's source block. 6760b57cec5SDimitry Andric auto UpdatePHINodes = [&](BasicBlock *Src) { 6770b57cec5SDimitry Andric BasicBlock::iterator I = UnwindDest->begin(); 6780b57cec5SDimitry Andric for (Value *V : UnwindDestPHIValues) { 6790b57cec5SDimitry Andric PHINode *PHI = cast<PHINode>(I); 6800b57cec5SDimitry Andric PHI->addIncoming(V, Src); 6810b57cec5SDimitry Andric ++I; 6820b57cec5SDimitry Andric } 6830b57cec5SDimitry Andric }; 6840b57cec5SDimitry Andric 6850b57cec5SDimitry Andric // This connects all the instructions which 'unwind to caller' to the invoke 6860b57cec5SDimitry Andric // destination. 6870b57cec5SDimitry Andric UnwindDestMemoTy FuncletUnwindMap; 6880b57cec5SDimitry Andric for (Function::iterator BB = FirstNewBlock->getIterator(), E = Caller->end(); 6890b57cec5SDimitry Andric BB != E; ++BB) { 6900b57cec5SDimitry Andric if (auto *CRI = dyn_cast<CleanupReturnInst>(BB->getTerminator())) { 6910b57cec5SDimitry Andric if (CRI->unwindsToCaller()) { 6920b57cec5SDimitry Andric auto *CleanupPad = CRI->getCleanupPad(); 6930b57cec5SDimitry Andric CleanupReturnInst::Create(CleanupPad, UnwindDest, CRI); 6940b57cec5SDimitry Andric CRI->eraseFromParent(); 6950b57cec5SDimitry Andric UpdatePHINodes(&*BB); 6960b57cec5SDimitry Andric // Finding a cleanupret with an unwind destination would confuse 6970b57cec5SDimitry Andric // subsequent calls to getUnwindDestToken, so map the cleanuppad 6980b57cec5SDimitry Andric // to short-circuit any such calls and recognize this as an "unwind 6990b57cec5SDimitry Andric // to caller" cleanup. 7000b57cec5SDimitry Andric assert(!FuncletUnwindMap.count(CleanupPad) || 7010b57cec5SDimitry Andric isa<ConstantTokenNone>(FuncletUnwindMap[CleanupPad])); 7020b57cec5SDimitry Andric FuncletUnwindMap[CleanupPad] = 7030b57cec5SDimitry Andric ConstantTokenNone::get(Caller->getContext()); 7040b57cec5SDimitry Andric } 7050b57cec5SDimitry Andric } 7060b57cec5SDimitry Andric 7070b57cec5SDimitry Andric Instruction *I = BB->getFirstNonPHI(); 7080b57cec5SDimitry Andric if (!I->isEHPad()) 7090b57cec5SDimitry Andric continue; 7100b57cec5SDimitry Andric 7110b57cec5SDimitry Andric Instruction *Replacement = nullptr; 7120b57cec5SDimitry Andric if (auto *CatchSwitch = dyn_cast<CatchSwitchInst>(I)) { 7130b57cec5SDimitry Andric if (CatchSwitch->unwindsToCaller()) { 7140b57cec5SDimitry Andric Value *UnwindDestToken; 7150b57cec5SDimitry Andric if (auto *ParentPad = 7160b57cec5SDimitry Andric dyn_cast<Instruction>(CatchSwitch->getParentPad())) { 7170b57cec5SDimitry Andric // This catchswitch is nested inside another funclet. If that 7180b57cec5SDimitry Andric // funclet has an unwind destination within the inlinee, then 7190b57cec5SDimitry Andric // unwinding out of this catchswitch would be UB. Rewriting this 7200b57cec5SDimitry Andric // catchswitch to unwind to the inlined invoke's unwind dest would 7210b57cec5SDimitry Andric // give the parent funclet multiple unwind destinations, which is 7220b57cec5SDimitry Andric // something that subsequent EH table generation can't handle and 7230b57cec5SDimitry Andric // that the veirifer rejects. So when we see such a call, leave it 7240b57cec5SDimitry Andric // as "unwind to caller". 7250b57cec5SDimitry Andric UnwindDestToken = getUnwindDestToken(ParentPad, FuncletUnwindMap); 7260b57cec5SDimitry Andric if (UnwindDestToken && !isa<ConstantTokenNone>(UnwindDestToken)) 7270b57cec5SDimitry Andric continue; 7280b57cec5SDimitry Andric } else { 7290b57cec5SDimitry Andric // This catchswitch has no parent to inherit constraints from, and 7300b57cec5SDimitry Andric // none of its descendants can have an unwind edge that exits it and 7310b57cec5SDimitry Andric // targets another funclet in the inlinee. It may or may not have a 7320b57cec5SDimitry Andric // descendant that definitively has an unwind to caller. In either 7330b57cec5SDimitry Andric // case, we'll have to assume that any unwinds out of it may need to 7340b57cec5SDimitry Andric // be routed to the caller, so treat it as though it has a definitive 7350b57cec5SDimitry Andric // unwind to caller. 7360b57cec5SDimitry Andric UnwindDestToken = ConstantTokenNone::get(Caller->getContext()); 7370b57cec5SDimitry Andric } 7380b57cec5SDimitry Andric auto *NewCatchSwitch = CatchSwitchInst::Create( 7390b57cec5SDimitry Andric CatchSwitch->getParentPad(), UnwindDest, 7400b57cec5SDimitry Andric CatchSwitch->getNumHandlers(), CatchSwitch->getName(), 7410b57cec5SDimitry Andric CatchSwitch); 7420b57cec5SDimitry Andric for (BasicBlock *PadBB : CatchSwitch->handlers()) 7430b57cec5SDimitry Andric NewCatchSwitch->addHandler(PadBB); 7440b57cec5SDimitry Andric // Propagate info for the old catchswitch over to the new one in 7450b57cec5SDimitry Andric // the unwind map. This also serves to short-circuit any subsequent 7460b57cec5SDimitry Andric // checks for the unwind dest of this catchswitch, which would get 7470b57cec5SDimitry Andric // confused if they found the outer handler in the callee. 7480b57cec5SDimitry Andric FuncletUnwindMap[NewCatchSwitch] = UnwindDestToken; 7490b57cec5SDimitry Andric Replacement = NewCatchSwitch; 7500b57cec5SDimitry Andric } 7510b57cec5SDimitry Andric } else if (!isa<FuncletPadInst>(I)) { 7520b57cec5SDimitry Andric llvm_unreachable("unexpected EHPad!"); 7530b57cec5SDimitry Andric } 7540b57cec5SDimitry Andric 7550b57cec5SDimitry Andric if (Replacement) { 7560b57cec5SDimitry Andric Replacement->takeName(I); 7570b57cec5SDimitry Andric I->replaceAllUsesWith(Replacement); 7580b57cec5SDimitry Andric I->eraseFromParent(); 7590b57cec5SDimitry Andric UpdatePHINodes(&*BB); 7600b57cec5SDimitry Andric } 7610b57cec5SDimitry Andric } 7620b57cec5SDimitry Andric 7630b57cec5SDimitry Andric if (InlinedCodeInfo.ContainsCalls) 7640b57cec5SDimitry Andric for (Function::iterator BB = FirstNewBlock->getIterator(), 7650b57cec5SDimitry Andric E = Caller->end(); 7660b57cec5SDimitry Andric BB != E; ++BB) 7670b57cec5SDimitry Andric if (BasicBlock *NewBB = HandleCallsInBlockInlinedThroughInvoke( 7680b57cec5SDimitry Andric &*BB, UnwindDest, &FuncletUnwindMap)) 7690b57cec5SDimitry Andric // Update any PHI nodes in the exceptional block to indicate that there 7700b57cec5SDimitry Andric // is now a new entry in them. 7710b57cec5SDimitry Andric UpdatePHINodes(NewBB); 7720b57cec5SDimitry Andric 7730b57cec5SDimitry Andric // Now that everything is happy, we have one final detail. The PHI nodes in 7740b57cec5SDimitry Andric // the exception destination block still have entries due to the original 7750b57cec5SDimitry Andric // invoke instruction. Eliminate these entries (which might even delete the 7760b57cec5SDimitry Andric // PHI node) now. 7770b57cec5SDimitry Andric UnwindDest->removePredecessor(InvokeBB); 7780b57cec5SDimitry Andric } 7790b57cec5SDimitry Andric 780e8d8bef9SDimitry Andric /// When inlining a call site that has !llvm.mem.parallel_loop_access, 781e8d8bef9SDimitry Andric /// !llvm.access.group, !alias.scope or !noalias metadata, that metadata should 782e8d8bef9SDimitry Andric /// be propagated to all memory-accessing cloned instructions. 783e8d8bef9SDimitry Andric static void PropagateCallSiteMetadata(CallBase &CB, ValueToValueMapTy &VMap) { 784e8d8bef9SDimitry Andric MDNode *MemParallelLoopAccess = 785e8d8bef9SDimitry Andric CB.getMetadata(LLVMContext::MD_mem_parallel_loop_access); 786e8d8bef9SDimitry Andric MDNode *AccessGroup = CB.getMetadata(LLVMContext::MD_access_group); 787e8d8bef9SDimitry Andric MDNode *AliasScope = CB.getMetadata(LLVMContext::MD_alias_scope); 788e8d8bef9SDimitry Andric MDNode *NoAlias = CB.getMetadata(LLVMContext::MD_noalias); 789e8d8bef9SDimitry Andric if (!MemParallelLoopAccess && !AccessGroup && !AliasScope && !NoAlias) 7900b57cec5SDimitry Andric return; 7910b57cec5SDimitry Andric 7920b57cec5SDimitry Andric for (ValueToValueMapTy::iterator VMI = VMap.begin(), VMIE = VMap.end(); 7930b57cec5SDimitry Andric VMI != VMIE; ++VMI) { 794e8d8bef9SDimitry Andric // Check that key is an instruction, to skip the Argument mapping, which 795e8d8bef9SDimitry Andric // points to an instruction in the original function, not the inlined one. 796e8d8bef9SDimitry Andric if (!VMI->second || !isa<Instruction>(VMI->first)) 7970b57cec5SDimitry Andric continue; 7980b57cec5SDimitry Andric 7990b57cec5SDimitry Andric Instruction *NI = dyn_cast<Instruction>(VMI->second); 8000b57cec5SDimitry Andric if (!NI) 8010b57cec5SDimitry Andric continue; 8020b57cec5SDimitry Andric 803e8d8bef9SDimitry Andric // This metadata is only relevant for instructions that access memory. 804e8d8bef9SDimitry Andric if (!NI->mayReadOrWriteMemory()) 805e8d8bef9SDimitry Andric continue; 806e8d8bef9SDimitry Andric 807e8d8bef9SDimitry Andric if (MemParallelLoopAccess) { 808e8d8bef9SDimitry Andric // TODO: This probably should not overwrite MemParalleLoopAccess. 809e8d8bef9SDimitry Andric MemParallelLoopAccess = MDNode::concatenate( 810e8d8bef9SDimitry Andric NI->getMetadata(LLVMContext::MD_mem_parallel_loop_access), 811e8d8bef9SDimitry Andric MemParallelLoopAccess); 812e8d8bef9SDimitry Andric NI->setMetadata(LLVMContext::MD_mem_parallel_loop_access, 813e8d8bef9SDimitry Andric MemParallelLoopAccess); 814e8d8bef9SDimitry Andric } 815e8d8bef9SDimitry Andric 816e8d8bef9SDimitry Andric if (AccessGroup) 817e8d8bef9SDimitry Andric NI->setMetadata(LLVMContext::MD_access_group, uniteAccessGroups( 818e8d8bef9SDimitry Andric NI->getMetadata(LLVMContext::MD_access_group), AccessGroup)); 819e8d8bef9SDimitry Andric 820e8d8bef9SDimitry Andric if (AliasScope) 821e8d8bef9SDimitry Andric NI->setMetadata(LLVMContext::MD_alias_scope, MDNode::concatenate( 822e8d8bef9SDimitry Andric NI->getMetadata(LLVMContext::MD_alias_scope), AliasScope)); 823e8d8bef9SDimitry Andric 824e8d8bef9SDimitry Andric if (NoAlias) 825e8d8bef9SDimitry Andric NI->setMetadata(LLVMContext::MD_noalias, MDNode::concatenate( 826e8d8bef9SDimitry Andric NI->getMetadata(LLVMContext::MD_noalias), NoAlias)); 8270b57cec5SDimitry Andric } 8280b57cec5SDimitry Andric } 8290b57cec5SDimitry Andric 830e8d8bef9SDimitry Andric /// Utility for cloning !noalias and !alias.scope metadata. When a code region 831e8d8bef9SDimitry Andric /// using scoped alias metadata is inlined, the aliasing relationships may not 832e8d8bef9SDimitry Andric /// hold between the two version. It is necessary to create a deep clone of the 833e8d8bef9SDimitry Andric /// metadata, putting the two versions in separate scope domains. 834e8d8bef9SDimitry Andric class ScopedAliasMetadataDeepCloner { 835e8d8bef9SDimitry Andric using MetadataMap = DenseMap<const MDNode *, TrackingMDNodeRef>; 8360b57cec5SDimitry Andric SetVector<const MDNode *> MD; 837e8d8bef9SDimitry Andric MetadataMap MDMap; 838e8d8bef9SDimitry Andric void addRecursiveMetadataUses(); 8390b57cec5SDimitry Andric 840e8d8bef9SDimitry Andric public: 841e8d8bef9SDimitry Andric ScopedAliasMetadataDeepCloner(const Function *F); 8420b57cec5SDimitry Andric 843e8d8bef9SDimitry Andric /// Create a new clone of the scoped alias metadata, which will be used by 844e8d8bef9SDimitry Andric /// subsequent remap() calls. 845e8d8bef9SDimitry Andric void clone(); 846e8d8bef9SDimitry Andric 847e8d8bef9SDimitry Andric /// Remap instructions in the given VMap from the original to the cloned 848e8d8bef9SDimitry Andric /// metadata. 849e8d8bef9SDimitry Andric void remap(ValueToValueMapTy &VMap); 850e8d8bef9SDimitry Andric }; 851e8d8bef9SDimitry Andric 852e8d8bef9SDimitry Andric ScopedAliasMetadataDeepCloner::ScopedAliasMetadataDeepCloner( 853e8d8bef9SDimitry Andric const Function *F) { 854e8d8bef9SDimitry Andric for (const BasicBlock &BB : *F) { 855e8d8bef9SDimitry Andric for (const Instruction &I : BB) { 856e8d8bef9SDimitry Andric if (const MDNode *M = I.getMetadata(LLVMContext::MD_alias_scope)) 8570b57cec5SDimitry Andric MD.insert(M); 858e8d8bef9SDimitry Andric if (const MDNode *M = I.getMetadata(LLVMContext::MD_noalias)) 8590b57cec5SDimitry Andric MD.insert(M); 860e8d8bef9SDimitry Andric 861e8d8bef9SDimitry Andric // We also need to clone the metadata in noalias intrinsics. 862e8d8bef9SDimitry Andric if (const auto *Decl = dyn_cast<NoAliasScopeDeclInst>(&I)) 863e8d8bef9SDimitry Andric MD.insert(Decl->getScopeList()); 864e8d8bef9SDimitry Andric } 865e8d8bef9SDimitry Andric } 866e8d8bef9SDimitry Andric addRecursiveMetadataUses(); 8670b57cec5SDimitry Andric } 8680b57cec5SDimitry Andric 869e8d8bef9SDimitry Andric void ScopedAliasMetadataDeepCloner::addRecursiveMetadataUses() { 8700b57cec5SDimitry Andric SmallVector<const Metadata *, 16> Queue(MD.begin(), MD.end()); 8710b57cec5SDimitry Andric while (!Queue.empty()) { 8720b57cec5SDimitry Andric const MDNode *M = cast<MDNode>(Queue.pop_back_val()); 873e8d8bef9SDimitry Andric for (const Metadata *Op : M->operands()) 874e8d8bef9SDimitry Andric if (const MDNode *OpMD = dyn_cast<MDNode>(Op)) 875e8d8bef9SDimitry Andric if (MD.insert(OpMD)) 876e8d8bef9SDimitry Andric Queue.push_back(OpMD); 877e8d8bef9SDimitry Andric } 8780b57cec5SDimitry Andric } 8790b57cec5SDimitry Andric 880e8d8bef9SDimitry Andric void ScopedAliasMetadataDeepCloner::clone() { 881e8d8bef9SDimitry Andric assert(MDMap.empty() && "clone() already called ?"); 882e8d8bef9SDimitry Andric 8830b57cec5SDimitry Andric SmallVector<TempMDTuple, 16> DummyNodes; 8840b57cec5SDimitry Andric for (const MDNode *I : MD) { 885e8d8bef9SDimitry Andric DummyNodes.push_back(MDTuple::getTemporary(I->getContext(), None)); 8860b57cec5SDimitry Andric MDMap[I].reset(DummyNodes.back().get()); 8870b57cec5SDimitry Andric } 8880b57cec5SDimitry Andric 8890b57cec5SDimitry Andric // Create new metadata nodes to replace the dummy nodes, replacing old 8900b57cec5SDimitry Andric // metadata references with either a dummy node or an already-created new 8910b57cec5SDimitry Andric // node. 8920b57cec5SDimitry Andric SmallVector<Metadata *, 4> NewOps; 893e8d8bef9SDimitry Andric for (const MDNode *I : MD) { 894e8d8bef9SDimitry Andric for (const Metadata *Op : I->operands()) { 895e8d8bef9SDimitry Andric if (const MDNode *M = dyn_cast<MDNode>(Op)) 8960b57cec5SDimitry Andric NewOps.push_back(MDMap[M]); 8970b57cec5SDimitry Andric else 898e8d8bef9SDimitry Andric NewOps.push_back(const_cast<Metadata *>(Op)); 8990b57cec5SDimitry Andric } 9000b57cec5SDimitry Andric 901e8d8bef9SDimitry Andric MDNode *NewM = MDNode::get(I->getContext(), NewOps); 9020b57cec5SDimitry Andric MDTuple *TempM = cast<MDTuple>(MDMap[I]); 9030b57cec5SDimitry Andric assert(TempM->isTemporary() && "Expected temporary node"); 9040b57cec5SDimitry Andric 9050b57cec5SDimitry Andric TempM->replaceAllUsesWith(NewM); 906e8d8bef9SDimitry Andric NewOps.clear(); 907e8d8bef9SDimitry Andric } 9080b57cec5SDimitry Andric } 9090b57cec5SDimitry Andric 910e8d8bef9SDimitry Andric void ScopedAliasMetadataDeepCloner::remap(ValueToValueMapTy &VMap) { 911e8d8bef9SDimitry Andric if (MDMap.empty()) 912e8d8bef9SDimitry Andric return; // Nothing to do. 913e8d8bef9SDimitry Andric 914e8d8bef9SDimitry Andric for (auto Entry : VMap) { 915e8d8bef9SDimitry Andric // Check that key is an instruction, to skip the Argument mapping, which 916e8d8bef9SDimitry Andric // points to an instruction in the original function, not the inlined one. 917e8d8bef9SDimitry Andric if (!Entry->second || !isa<Instruction>(Entry->first)) 9180b57cec5SDimitry Andric continue; 9190b57cec5SDimitry Andric 920e8d8bef9SDimitry Andric Instruction *I = dyn_cast<Instruction>(Entry->second); 921e8d8bef9SDimitry Andric if (!I) 9220b57cec5SDimitry Andric continue; 9230b57cec5SDimitry Andric 924*d409305fSDimitry Andric // Only update scopes when we find them in the map. If they are not, it is 925*d409305fSDimitry Andric // because we already handled that instruction before. This is faster than 926*d409305fSDimitry Andric // tracking which instructions we already updated. 927e8d8bef9SDimitry Andric if (MDNode *M = I->getMetadata(LLVMContext::MD_alias_scope)) 928*d409305fSDimitry Andric if (MDNode *MNew = MDMap.lookup(M)) 929*d409305fSDimitry Andric I->setMetadata(LLVMContext::MD_alias_scope, MNew); 9300b57cec5SDimitry Andric 931e8d8bef9SDimitry Andric if (MDNode *M = I->getMetadata(LLVMContext::MD_noalias)) 932*d409305fSDimitry Andric if (MDNode *MNew = MDMap.lookup(M)) 933*d409305fSDimitry Andric I->setMetadata(LLVMContext::MD_noalias, MNew); 934e8d8bef9SDimitry Andric 935e8d8bef9SDimitry Andric if (auto *Decl = dyn_cast<NoAliasScopeDeclInst>(I)) 936*d409305fSDimitry Andric if (MDNode *MNew = MDMap.lookup(Decl->getScopeList())) 937*d409305fSDimitry Andric Decl->setScopeList(MNew); 9380b57cec5SDimitry Andric } 9390b57cec5SDimitry Andric } 9400b57cec5SDimitry Andric 9410b57cec5SDimitry Andric /// If the inlined function has noalias arguments, 9420b57cec5SDimitry Andric /// then add new alias scopes for each noalias argument, tag the mapped noalias 9430b57cec5SDimitry Andric /// parameters with noalias metadata specifying the new scope, and tag all 9440b57cec5SDimitry Andric /// non-derived loads, stores and memory intrinsics with the new alias scopes. 9455ffd83dbSDimitry Andric static void AddAliasScopeMetadata(CallBase &CB, ValueToValueMapTy &VMap, 9460b57cec5SDimitry Andric const DataLayout &DL, AAResults *CalleeAAR) { 9470b57cec5SDimitry Andric if (!EnableNoAliasConversion) 9480b57cec5SDimitry Andric return; 9490b57cec5SDimitry Andric 9505ffd83dbSDimitry Andric const Function *CalledFunc = CB.getCalledFunction(); 9510b57cec5SDimitry Andric SmallVector<const Argument *, 4> NoAliasArgs; 9520b57cec5SDimitry Andric 9530b57cec5SDimitry Andric for (const Argument &Arg : CalledFunc->args()) 9545ffd83dbSDimitry Andric if (CB.paramHasAttr(Arg.getArgNo(), Attribute::NoAlias) && !Arg.use_empty()) 9550b57cec5SDimitry Andric NoAliasArgs.push_back(&Arg); 9560b57cec5SDimitry Andric 9570b57cec5SDimitry Andric if (NoAliasArgs.empty()) 9580b57cec5SDimitry Andric return; 9590b57cec5SDimitry Andric 9600b57cec5SDimitry Andric // To do a good job, if a noalias variable is captured, we need to know if 9610b57cec5SDimitry Andric // the capture point dominates the particular use we're considering. 9620b57cec5SDimitry Andric DominatorTree DT; 9630b57cec5SDimitry Andric DT.recalculate(const_cast<Function&>(*CalledFunc)); 9640b57cec5SDimitry Andric 9650b57cec5SDimitry Andric // noalias indicates that pointer values based on the argument do not alias 9660b57cec5SDimitry Andric // pointer values which are not based on it. So we add a new "scope" for each 9670b57cec5SDimitry Andric // noalias function argument. Accesses using pointers based on that argument 9680b57cec5SDimitry Andric // become part of that alias scope, accesses using pointers not based on that 9690b57cec5SDimitry Andric // argument are tagged as noalias with that scope. 9700b57cec5SDimitry Andric 9710b57cec5SDimitry Andric DenseMap<const Argument *, MDNode *> NewScopes; 9720b57cec5SDimitry Andric MDBuilder MDB(CalledFunc->getContext()); 9730b57cec5SDimitry Andric 9740b57cec5SDimitry Andric // Create a new scope domain for this function. 9750b57cec5SDimitry Andric MDNode *NewDomain = 9760b57cec5SDimitry Andric MDB.createAnonymousAliasScopeDomain(CalledFunc->getName()); 9770b57cec5SDimitry Andric for (unsigned i = 0, e = NoAliasArgs.size(); i != e; ++i) { 9780b57cec5SDimitry Andric const Argument *A = NoAliasArgs[i]; 9790b57cec5SDimitry Andric 9805ffd83dbSDimitry Andric std::string Name = std::string(CalledFunc->getName()); 9810b57cec5SDimitry Andric if (A->hasName()) { 9820b57cec5SDimitry Andric Name += ": %"; 9830b57cec5SDimitry Andric Name += A->getName(); 9840b57cec5SDimitry Andric } else { 9850b57cec5SDimitry Andric Name += ": argument "; 9860b57cec5SDimitry Andric Name += utostr(i); 9870b57cec5SDimitry Andric } 9880b57cec5SDimitry Andric 9890b57cec5SDimitry Andric // Note: We always create a new anonymous root here. This is true regardless 9900b57cec5SDimitry Andric // of the linkage of the callee because the aliasing "scope" is not just a 9910b57cec5SDimitry Andric // property of the callee, but also all control dependencies in the caller. 9920b57cec5SDimitry Andric MDNode *NewScope = MDB.createAnonymousAliasScope(NewDomain, Name); 9930b57cec5SDimitry Andric NewScopes.insert(std::make_pair(A, NewScope)); 994e8d8bef9SDimitry Andric 995e8d8bef9SDimitry Andric if (UseNoAliasIntrinsic) { 996e8d8bef9SDimitry Andric // Introduce a llvm.experimental.noalias.scope.decl for the noalias 997e8d8bef9SDimitry Andric // argument. 998e8d8bef9SDimitry Andric MDNode *AScopeList = MDNode::get(CalledFunc->getContext(), NewScope); 999e8d8bef9SDimitry Andric auto *NoAliasDecl = 1000e8d8bef9SDimitry Andric IRBuilder<>(&CB).CreateNoAliasScopeDeclaration(AScopeList); 1001e8d8bef9SDimitry Andric // Ignore the result for now. The result will be used when the 1002e8d8bef9SDimitry Andric // llvm.noalias intrinsic is introduced. 1003e8d8bef9SDimitry Andric (void)NoAliasDecl; 1004e8d8bef9SDimitry Andric } 10050b57cec5SDimitry Andric } 10060b57cec5SDimitry Andric 10070b57cec5SDimitry Andric // Iterate over all new instructions in the map; for all memory-access 10080b57cec5SDimitry Andric // instructions, add the alias scope metadata. 10090b57cec5SDimitry Andric for (ValueToValueMapTy::iterator VMI = VMap.begin(), VMIE = VMap.end(); 10100b57cec5SDimitry Andric VMI != VMIE; ++VMI) { 10110b57cec5SDimitry Andric if (const Instruction *I = dyn_cast<Instruction>(VMI->first)) { 10120b57cec5SDimitry Andric if (!VMI->second) 10130b57cec5SDimitry Andric continue; 10140b57cec5SDimitry Andric 10150b57cec5SDimitry Andric Instruction *NI = dyn_cast<Instruction>(VMI->second); 10160b57cec5SDimitry Andric if (!NI) 10170b57cec5SDimitry Andric continue; 10180b57cec5SDimitry Andric 10190b57cec5SDimitry Andric bool IsArgMemOnlyCall = false, IsFuncCall = false; 10200b57cec5SDimitry Andric SmallVector<const Value *, 2> PtrArgs; 10210b57cec5SDimitry Andric 10220b57cec5SDimitry Andric if (const LoadInst *LI = dyn_cast<LoadInst>(I)) 10230b57cec5SDimitry Andric PtrArgs.push_back(LI->getPointerOperand()); 10240b57cec5SDimitry Andric else if (const StoreInst *SI = dyn_cast<StoreInst>(I)) 10250b57cec5SDimitry Andric PtrArgs.push_back(SI->getPointerOperand()); 10260b57cec5SDimitry Andric else if (const VAArgInst *VAAI = dyn_cast<VAArgInst>(I)) 10270b57cec5SDimitry Andric PtrArgs.push_back(VAAI->getPointerOperand()); 10280b57cec5SDimitry Andric else if (const AtomicCmpXchgInst *CXI = dyn_cast<AtomicCmpXchgInst>(I)) 10290b57cec5SDimitry Andric PtrArgs.push_back(CXI->getPointerOperand()); 10300b57cec5SDimitry Andric else if (const AtomicRMWInst *RMWI = dyn_cast<AtomicRMWInst>(I)) 10310b57cec5SDimitry Andric PtrArgs.push_back(RMWI->getPointerOperand()); 10320b57cec5SDimitry Andric else if (const auto *Call = dyn_cast<CallBase>(I)) { 10330b57cec5SDimitry Andric // If we know that the call does not access memory, then we'll still 10340b57cec5SDimitry Andric // know that about the inlined clone of this call site, and we don't 10350b57cec5SDimitry Andric // need to add metadata. 10360b57cec5SDimitry Andric if (Call->doesNotAccessMemory()) 10370b57cec5SDimitry Andric continue; 10380b57cec5SDimitry Andric 10390b57cec5SDimitry Andric IsFuncCall = true; 10400b57cec5SDimitry Andric if (CalleeAAR) { 10410b57cec5SDimitry Andric FunctionModRefBehavior MRB = CalleeAAR->getModRefBehavior(Call); 10425ffd83dbSDimitry Andric if (AAResults::onlyAccessesArgPointees(MRB)) 10430b57cec5SDimitry Andric IsArgMemOnlyCall = true; 10440b57cec5SDimitry Andric } 10450b57cec5SDimitry Andric 10460b57cec5SDimitry Andric for (Value *Arg : Call->args()) { 10470b57cec5SDimitry Andric // We need to check the underlying objects of all arguments, not just 10480b57cec5SDimitry Andric // the pointer arguments, because we might be passing pointers as 10490b57cec5SDimitry Andric // integers, etc. 10500b57cec5SDimitry Andric // However, if we know that the call only accesses pointer arguments, 10510b57cec5SDimitry Andric // then we only need to check the pointer arguments. 10520b57cec5SDimitry Andric if (IsArgMemOnlyCall && !Arg->getType()->isPointerTy()) 10530b57cec5SDimitry Andric continue; 10540b57cec5SDimitry Andric 10550b57cec5SDimitry Andric PtrArgs.push_back(Arg); 10560b57cec5SDimitry Andric } 10570b57cec5SDimitry Andric } 10580b57cec5SDimitry Andric 10590b57cec5SDimitry Andric // If we found no pointers, then this instruction is not suitable for 10600b57cec5SDimitry Andric // pairing with an instruction to receive aliasing metadata. 10610b57cec5SDimitry Andric // However, if this is a call, this we might just alias with none of the 10620b57cec5SDimitry Andric // noalias arguments. 10630b57cec5SDimitry Andric if (PtrArgs.empty() && !IsFuncCall) 10640b57cec5SDimitry Andric continue; 10650b57cec5SDimitry Andric 10660b57cec5SDimitry Andric // It is possible that there is only one underlying object, but you 10670b57cec5SDimitry Andric // need to go through several PHIs to see it, and thus could be 10680b57cec5SDimitry Andric // repeated in the Objects list. 10690b57cec5SDimitry Andric SmallPtrSet<const Value *, 4> ObjSet; 10700b57cec5SDimitry Andric SmallVector<Metadata *, 4> Scopes, NoAliases; 10710b57cec5SDimitry Andric 10720b57cec5SDimitry Andric SmallSetVector<const Argument *, 4> NAPtrArgs; 10730b57cec5SDimitry Andric for (const Value *V : PtrArgs) { 10740b57cec5SDimitry Andric SmallVector<const Value *, 4> Objects; 1075e8d8bef9SDimitry Andric getUnderlyingObjects(V, Objects, /* LI = */ nullptr); 10760b57cec5SDimitry Andric 10770b57cec5SDimitry Andric for (const Value *O : Objects) 10780b57cec5SDimitry Andric ObjSet.insert(O); 10790b57cec5SDimitry Andric } 10800b57cec5SDimitry Andric 10810b57cec5SDimitry Andric // Figure out if we're derived from anything that is not a noalias 10820b57cec5SDimitry Andric // argument. 10830b57cec5SDimitry Andric bool CanDeriveViaCapture = false, UsesAliasingPtr = false; 10840b57cec5SDimitry Andric for (const Value *V : ObjSet) { 10850b57cec5SDimitry Andric // Is this value a constant that cannot be derived from any pointer 10860b57cec5SDimitry Andric // value (we need to exclude constant expressions, for example, that 10870b57cec5SDimitry Andric // are formed from arithmetic on global symbols). 10880b57cec5SDimitry Andric bool IsNonPtrConst = isa<ConstantInt>(V) || isa<ConstantFP>(V) || 10890b57cec5SDimitry Andric isa<ConstantPointerNull>(V) || 10900b57cec5SDimitry Andric isa<ConstantDataVector>(V) || isa<UndefValue>(V); 10910b57cec5SDimitry Andric if (IsNonPtrConst) 10920b57cec5SDimitry Andric continue; 10930b57cec5SDimitry Andric 10940b57cec5SDimitry Andric // If this is anything other than a noalias argument, then we cannot 10950b57cec5SDimitry Andric // completely describe the aliasing properties using alias.scope 10960b57cec5SDimitry Andric // metadata (and, thus, won't add any). 10970b57cec5SDimitry Andric if (const Argument *A = dyn_cast<Argument>(V)) { 10985ffd83dbSDimitry Andric if (!CB.paramHasAttr(A->getArgNo(), Attribute::NoAlias)) 10990b57cec5SDimitry Andric UsesAliasingPtr = true; 11000b57cec5SDimitry Andric } else { 11010b57cec5SDimitry Andric UsesAliasingPtr = true; 11020b57cec5SDimitry Andric } 11030b57cec5SDimitry Andric 11040b57cec5SDimitry Andric // If this is not some identified function-local object (which cannot 11050b57cec5SDimitry Andric // directly alias a noalias argument), or some other argument (which, 11060b57cec5SDimitry Andric // by definition, also cannot alias a noalias argument), then we could 11070b57cec5SDimitry Andric // alias a noalias argument that has been captured). 11080b57cec5SDimitry Andric if (!isa<Argument>(V) && 11090b57cec5SDimitry Andric !isIdentifiedFunctionLocal(const_cast<Value*>(V))) 11100b57cec5SDimitry Andric CanDeriveViaCapture = true; 11110b57cec5SDimitry Andric } 11120b57cec5SDimitry Andric 11130b57cec5SDimitry Andric // A function call can always get captured noalias pointers (via other 11140b57cec5SDimitry Andric // parameters, globals, etc.). 11150b57cec5SDimitry Andric if (IsFuncCall && !IsArgMemOnlyCall) 11160b57cec5SDimitry Andric CanDeriveViaCapture = true; 11170b57cec5SDimitry Andric 11180b57cec5SDimitry Andric // First, we want to figure out all of the sets with which we definitely 11190b57cec5SDimitry Andric // don't alias. Iterate over all noalias set, and add those for which: 11200b57cec5SDimitry Andric // 1. The noalias argument is not in the set of objects from which we 11210b57cec5SDimitry Andric // definitely derive. 11220b57cec5SDimitry Andric // 2. The noalias argument has not yet been captured. 11230b57cec5SDimitry Andric // An arbitrary function that might load pointers could see captured 11240b57cec5SDimitry Andric // noalias arguments via other noalias arguments or globals, and so we 11250b57cec5SDimitry Andric // must always check for prior capture. 11260b57cec5SDimitry Andric for (const Argument *A : NoAliasArgs) { 11270b57cec5SDimitry Andric if (!ObjSet.count(A) && (!CanDeriveViaCapture || 11280b57cec5SDimitry Andric // It might be tempting to skip the 11290b57cec5SDimitry Andric // PointerMayBeCapturedBefore check if 11300b57cec5SDimitry Andric // A->hasNoCaptureAttr() is true, but this is 11310b57cec5SDimitry Andric // incorrect because nocapture only guarantees 11320b57cec5SDimitry Andric // that no copies outlive the function, not 11330b57cec5SDimitry Andric // that the value cannot be locally captured. 11340b57cec5SDimitry Andric !PointerMayBeCapturedBefore(A, 11350b57cec5SDimitry Andric /* ReturnCaptures */ false, 11360b57cec5SDimitry Andric /* StoreCaptures */ false, I, &DT))) 11370b57cec5SDimitry Andric NoAliases.push_back(NewScopes[A]); 11380b57cec5SDimitry Andric } 11390b57cec5SDimitry Andric 11400b57cec5SDimitry Andric if (!NoAliases.empty()) 11410b57cec5SDimitry Andric NI->setMetadata(LLVMContext::MD_noalias, 11420b57cec5SDimitry Andric MDNode::concatenate( 11430b57cec5SDimitry Andric NI->getMetadata(LLVMContext::MD_noalias), 11440b57cec5SDimitry Andric MDNode::get(CalledFunc->getContext(), NoAliases))); 11450b57cec5SDimitry Andric 11460b57cec5SDimitry Andric // Next, we want to figure out all of the sets to which we might belong. 11470b57cec5SDimitry Andric // We might belong to a set if the noalias argument is in the set of 11480b57cec5SDimitry Andric // underlying objects. If there is some non-noalias argument in our list 11490b57cec5SDimitry Andric // of underlying objects, then we cannot add a scope because the fact 11500b57cec5SDimitry Andric // that some access does not alias with any set of our noalias arguments 11510b57cec5SDimitry Andric // cannot itself guarantee that it does not alias with this access 11520b57cec5SDimitry Andric // (because there is some pointer of unknown origin involved and the 11530b57cec5SDimitry Andric // other access might also depend on this pointer). We also cannot add 11540b57cec5SDimitry Andric // scopes to arbitrary functions unless we know they don't access any 11550b57cec5SDimitry Andric // non-parameter pointer-values. 11560b57cec5SDimitry Andric bool CanAddScopes = !UsesAliasingPtr; 11570b57cec5SDimitry Andric if (CanAddScopes && IsFuncCall) 11580b57cec5SDimitry Andric CanAddScopes = IsArgMemOnlyCall; 11590b57cec5SDimitry Andric 11600b57cec5SDimitry Andric if (CanAddScopes) 11610b57cec5SDimitry Andric for (const Argument *A : NoAliasArgs) { 11620b57cec5SDimitry Andric if (ObjSet.count(A)) 11630b57cec5SDimitry Andric Scopes.push_back(NewScopes[A]); 11640b57cec5SDimitry Andric } 11650b57cec5SDimitry Andric 11660b57cec5SDimitry Andric if (!Scopes.empty()) 11670b57cec5SDimitry Andric NI->setMetadata( 11680b57cec5SDimitry Andric LLVMContext::MD_alias_scope, 11690b57cec5SDimitry Andric MDNode::concatenate(NI->getMetadata(LLVMContext::MD_alias_scope), 11700b57cec5SDimitry Andric MDNode::get(CalledFunc->getContext(), Scopes))); 11710b57cec5SDimitry Andric } 11720b57cec5SDimitry Andric } 11730b57cec5SDimitry Andric } 11740b57cec5SDimitry Andric 11755ffd83dbSDimitry Andric static bool MayContainThrowingOrExitingCall(Instruction *Begin, 11765ffd83dbSDimitry Andric Instruction *End) { 11775ffd83dbSDimitry Andric 11785ffd83dbSDimitry Andric assert(Begin->getParent() == End->getParent() && 11795ffd83dbSDimitry Andric "Expected to be in same basic block!"); 11805ffd83dbSDimitry Andric unsigned NumInstChecked = 0; 11815ffd83dbSDimitry Andric // Check that all instructions in the range [Begin, End) are guaranteed to 11825ffd83dbSDimitry Andric // transfer execution to successor. 11835ffd83dbSDimitry Andric for (auto &I : make_range(Begin->getIterator(), End->getIterator())) 11845ffd83dbSDimitry Andric if (NumInstChecked++ > InlinerAttributeWindow || 11855ffd83dbSDimitry Andric !isGuaranteedToTransferExecutionToSuccessor(&I)) 11865ffd83dbSDimitry Andric return true; 11875ffd83dbSDimitry Andric return false; 11885ffd83dbSDimitry Andric } 11895ffd83dbSDimitry Andric 11905ffd83dbSDimitry Andric static AttrBuilder IdentifyValidAttributes(CallBase &CB) { 11915ffd83dbSDimitry Andric 11925ffd83dbSDimitry Andric AttrBuilder AB(CB.getAttributes(), AttributeList::ReturnIndex); 11935ffd83dbSDimitry Andric if (AB.empty()) 11945ffd83dbSDimitry Andric return AB; 11955ffd83dbSDimitry Andric AttrBuilder Valid; 11965ffd83dbSDimitry Andric // Only allow these white listed attributes to be propagated back to the 11975ffd83dbSDimitry Andric // callee. This is because other attributes may only be valid on the call 11985ffd83dbSDimitry Andric // itself, i.e. attributes such as signext and zeroext. 11995ffd83dbSDimitry Andric if (auto DerefBytes = AB.getDereferenceableBytes()) 12005ffd83dbSDimitry Andric Valid.addDereferenceableAttr(DerefBytes); 12015ffd83dbSDimitry Andric if (auto DerefOrNullBytes = AB.getDereferenceableOrNullBytes()) 12025ffd83dbSDimitry Andric Valid.addDereferenceableOrNullAttr(DerefOrNullBytes); 12035ffd83dbSDimitry Andric if (AB.contains(Attribute::NoAlias)) 12045ffd83dbSDimitry Andric Valid.addAttribute(Attribute::NoAlias); 12055ffd83dbSDimitry Andric if (AB.contains(Attribute::NonNull)) 12065ffd83dbSDimitry Andric Valid.addAttribute(Attribute::NonNull); 12075ffd83dbSDimitry Andric return Valid; 12085ffd83dbSDimitry Andric } 12095ffd83dbSDimitry Andric 12105ffd83dbSDimitry Andric static void AddReturnAttributes(CallBase &CB, ValueToValueMapTy &VMap) { 12115ffd83dbSDimitry Andric if (!UpdateReturnAttributes) 12125ffd83dbSDimitry Andric return; 12135ffd83dbSDimitry Andric 12145ffd83dbSDimitry Andric AttrBuilder Valid = IdentifyValidAttributes(CB); 12155ffd83dbSDimitry Andric if (Valid.empty()) 12165ffd83dbSDimitry Andric return; 12175ffd83dbSDimitry Andric auto *CalledFunction = CB.getCalledFunction(); 12185ffd83dbSDimitry Andric auto &Context = CalledFunction->getContext(); 12195ffd83dbSDimitry Andric 12205ffd83dbSDimitry Andric for (auto &BB : *CalledFunction) { 12215ffd83dbSDimitry Andric auto *RI = dyn_cast<ReturnInst>(BB.getTerminator()); 12225ffd83dbSDimitry Andric if (!RI || !isa<CallBase>(RI->getOperand(0))) 12235ffd83dbSDimitry Andric continue; 12245ffd83dbSDimitry Andric auto *RetVal = cast<CallBase>(RI->getOperand(0)); 12255ffd83dbSDimitry Andric // Sanity check that the cloned RetVal exists and is a call, otherwise we 12265ffd83dbSDimitry Andric // cannot add the attributes on the cloned RetVal. 12275ffd83dbSDimitry Andric // Simplification during inlining could have transformed the cloned 12285ffd83dbSDimitry Andric // instruction. 12295ffd83dbSDimitry Andric auto *NewRetVal = dyn_cast_or_null<CallBase>(VMap.lookup(RetVal)); 12305ffd83dbSDimitry Andric if (!NewRetVal) 12315ffd83dbSDimitry Andric continue; 12325ffd83dbSDimitry Andric // Backward propagation of attributes to the returned value may be incorrect 12335ffd83dbSDimitry Andric // if it is control flow dependent. 12345ffd83dbSDimitry Andric // Consider: 12355ffd83dbSDimitry Andric // @callee { 12365ffd83dbSDimitry Andric // %rv = call @foo() 12375ffd83dbSDimitry Andric // %rv2 = call @bar() 12385ffd83dbSDimitry Andric // if (%rv2 != null) 12395ffd83dbSDimitry Andric // return %rv2 12405ffd83dbSDimitry Andric // if (%rv == null) 12415ffd83dbSDimitry Andric // exit() 12425ffd83dbSDimitry Andric // return %rv 12435ffd83dbSDimitry Andric // } 12445ffd83dbSDimitry Andric // caller() { 12455ffd83dbSDimitry Andric // %val = call nonnull @callee() 12465ffd83dbSDimitry Andric // } 12475ffd83dbSDimitry Andric // Here we cannot add the nonnull attribute on either foo or bar. So, we 12485ffd83dbSDimitry Andric // limit the check to both RetVal and RI are in the same basic block and 12495ffd83dbSDimitry Andric // there are no throwing/exiting instructions between these instructions. 12505ffd83dbSDimitry Andric if (RI->getParent() != RetVal->getParent() || 12515ffd83dbSDimitry Andric MayContainThrowingOrExitingCall(RetVal, RI)) 12525ffd83dbSDimitry Andric continue; 12535ffd83dbSDimitry Andric // Add to the existing attributes of NewRetVal, i.e. the cloned call 12545ffd83dbSDimitry Andric // instruction. 12555ffd83dbSDimitry Andric // NB! When we have the same attribute already existing on NewRetVal, but 12565ffd83dbSDimitry Andric // with a differing value, the AttributeList's merge API honours the already 12575ffd83dbSDimitry Andric // existing attribute value (i.e. attributes such as dereferenceable, 12585ffd83dbSDimitry Andric // dereferenceable_or_null etc). See AttrBuilder::merge for more details. 12595ffd83dbSDimitry Andric AttributeList AL = NewRetVal->getAttributes(); 12605ffd83dbSDimitry Andric AttributeList NewAL = 12615ffd83dbSDimitry Andric AL.addAttributes(Context, AttributeList::ReturnIndex, Valid); 12625ffd83dbSDimitry Andric NewRetVal->setAttributes(NewAL); 12635ffd83dbSDimitry Andric } 12645ffd83dbSDimitry Andric } 12655ffd83dbSDimitry Andric 12660b57cec5SDimitry Andric /// If the inlined function has non-byval align arguments, then 12670b57cec5SDimitry Andric /// add @llvm.assume-based alignment assumptions to preserve this information. 12685ffd83dbSDimitry Andric static void AddAlignmentAssumptions(CallBase &CB, InlineFunctionInfo &IFI) { 12690b57cec5SDimitry Andric if (!PreserveAlignmentAssumptions || !IFI.GetAssumptionCache) 12700b57cec5SDimitry Andric return; 12710b57cec5SDimitry Andric 12725ffd83dbSDimitry Andric AssumptionCache *AC = &IFI.GetAssumptionCache(*CB.getCaller()); 12735ffd83dbSDimitry Andric auto &DL = CB.getCaller()->getParent()->getDataLayout(); 12740b57cec5SDimitry Andric 12750b57cec5SDimitry Andric // To avoid inserting redundant assumptions, we should check for assumptions 12760b57cec5SDimitry Andric // already in the caller. To do this, we might need a DT of the caller. 12770b57cec5SDimitry Andric DominatorTree DT; 12780b57cec5SDimitry Andric bool DTCalculated = false; 12790b57cec5SDimitry Andric 12805ffd83dbSDimitry Andric Function *CalledFunc = CB.getCalledFunction(); 12810b57cec5SDimitry Andric for (Argument &Arg : CalledFunc->args()) { 12820b57cec5SDimitry Andric unsigned Align = Arg.getType()->isPointerTy() ? Arg.getParamAlignment() : 0; 1283e8d8bef9SDimitry Andric if (Align && !Arg.hasPassPointeeByValueCopyAttr() && !Arg.hasNUses(0)) { 12840b57cec5SDimitry Andric if (!DTCalculated) { 12855ffd83dbSDimitry Andric DT.recalculate(*CB.getCaller()); 12860b57cec5SDimitry Andric DTCalculated = true; 12870b57cec5SDimitry Andric } 12880b57cec5SDimitry Andric 12890b57cec5SDimitry Andric // If we can already prove the asserted alignment in the context of the 12900b57cec5SDimitry Andric // caller, then don't bother inserting the assumption. 12915ffd83dbSDimitry Andric Value *ArgVal = CB.getArgOperand(Arg.getArgNo()); 12925ffd83dbSDimitry Andric if (getKnownAlignment(ArgVal, DL, &CB, AC, &DT) >= Align) 12930b57cec5SDimitry Andric continue; 12940b57cec5SDimitry Andric 12955ffd83dbSDimitry Andric CallInst *NewAsmp = 12965ffd83dbSDimitry Andric IRBuilder<>(&CB).CreateAlignmentAssumption(DL, ArgVal, Align); 12970b57cec5SDimitry Andric AC->registerAssumption(NewAsmp); 12980b57cec5SDimitry Andric } 12990b57cec5SDimitry Andric } 13000b57cec5SDimitry Andric } 13010b57cec5SDimitry Andric 13020b57cec5SDimitry Andric /// Once we have cloned code over from a callee into the caller, 13030b57cec5SDimitry Andric /// update the specified callgraph to reflect the changes we made. 13040b57cec5SDimitry Andric /// Note that it's possible that not all code was copied over, so only 13050b57cec5SDimitry Andric /// some edges of the callgraph may remain. 13065ffd83dbSDimitry Andric static void UpdateCallGraphAfterInlining(CallBase &CB, 13070b57cec5SDimitry Andric Function::iterator FirstNewBlock, 13080b57cec5SDimitry Andric ValueToValueMapTy &VMap, 13090b57cec5SDimitry Andric InlineFunctionInfo &IFI) { 13100b57cec5SDimitry Andric CallGraph &CG = *IFI.CG; 13115ffd83dbSDimitry Andric const Function *Caller = CB.getCaller(); 13125ffd83dbSDimitry Andric const Function *Callee = CB.getCalledFunction(); 13130b57cec5SDimitry Andric CallGraphNode *CalleeNode = CG[Callee]; 13140b57cec5SDimitry Andric CallGraphNode *CallerNode = CG[Caller]; 13150b57cec5SDimitry Andric 13160b57cec5SDimitry Andric // Since we inlined some uninlined call sites in the callee into the caller, 13170b57cec5SDimitry Andric // add edges from the caller to all of the callees of the callee. 13180b57cec5SDimitry Andric CallGraphNode::iterator I = CalleeNode->begin(), E = CalleeNode->end(); 13190b57cec5SDimitry Andric 13200b57cec5SDimitry Andric // Consider the case where CalleeNode == CallerNode. 13210b57cec5SDimitry Andric CallGraphNode::CalledFunctionsVector CallCache; 13220b57cec5SDimitry Andric if (CalleeNode == CallerNode) { 13230b57cec5SDimitry Andric CallCache.assign(I, E); 13240b57cec5SDimitry Andric I = CallCache.begin(); 13250b57cec5SDimitry Andric E = CallCache.end(); 13260b57cec5SDimitry Andric } 13270b57cec5SDimitry Andric 13280b57cec5SDimitry Andric for (; I != E; ++I) { 13295ffd83dbSDimitry Andric // Skip 'refererence' call records. 13305ffd83dbSDimitry Andric if (!I->first) 13315ffd83dbSDimitry Andric continue; 13325ffd83dbSDimitry Andric 13335ffd83dbSDimitry Andric const Value *OrigCall = *I->first; 13340b57cec5SDimitry Andric 13350b57cec5SDimitry Andric ValueToValueMapTy::iterator VMI = VMap.find(OrigCall); 13360b57cec5SDimitry Andric // Only copy the edge if the call was inlined! 13370b57cec5SDimitry Andric if (VMI == VMap.end() || VMI->second == nullptr) 13380b57cec5SDimitry Andric continue; 13390b57cec5SDimitry Andric 13400b57cec5SDimitry Andric // If the call was inlined, but then constant folded, there is no edge to 13410b57cec5SDimitry Andric // add. Check for this case. 13420b57cec5SDimitry Andric auto *NewCall = dyn_cast<CallBase>(VMI->second); 13430b57cec5SDimitry Andric if (!NewCall) 13440b57cec5SDimitry Andric continue; 13450b57cec5SDimitry Andric 13460b57cec5SDimitry Andric // We do not treat intrinsic calls like real function calls because we 13470b57cec5SDimitry Andric // expect them to become inline code; do not add an edge for an intrinsic. 13480b57cec5SDimitry Andric if (NewCall->getCalledFunction() && 13490b57cec5SDimitry Andric NewCall->getCalledFunction()->isIntrinsic()) 13500b57cec5SDimitry Andric continue; 13510b57cec5SDimitry Andric 13520b57cec5SDimitry Andric // Remember that this call site got inlined for the client of 13530b57cec5SDimitry Andric // InlineFunction. 13540b57cec5SDimitry Andric IFI.InlinedCalls.push_back(NewCall); 13550b57cec5SDimitry Andric 13560b57cec5SDimitry Andric // It's possible that inlining the callsite will cause it to go from an 13570b57cec5SDimitry Andric // indirect to a direct call by resolving a function pointer. If this 13580b57cec5SDimitry Andric // happens, set the callee of the new call site to a more precise 13590b57cec5SDimitry Andric // destination. This can also happen if the call graph node of the caller 13600b57cec5SDimitry Andric // was just unnecessarily imprecise. 13610b57cec5SDimitry Andric if (!I->second->getFunction()) 13620b57cec5SDimitry Andric if (Function *F = NewCall->getCalledFunction()) { 13630b57cec5SDimitry Andric // Indirect call site resolved to direct call. 13640b57cec5SDimitry Andric CallerNode->addCalledFunction(NewCall, CG[F]); 13650b57cec5SDimitry Andric 13660b57cec5SDimitry Andric continue; 13670b57cec5SDimitry Andric } 13680b57cec5SDimitry Andric 13690b57cec5SDimitry Andric CallerNode->addCalledFunction(NewCall, I->second); 13700b57cec5SDimitry Andric } 13710b57cec5SDimitry Andric 13720b57cec5SDimitry Andric // Update the call graph by deleting the edge from Callee to Caller. We must 13730b57cec5SDimitry Andric // do this after the loop above in case Caller and Callee are the same. 13745ffd83dbSDimitry Andric CallerNode->removeCallEdgeFor(*cast<CallBase>(&CB)); 13750b57cec5SDimitry Andric } 13760b57cec5SDimitry Andric 13770b57cec5SDimitry Andric static void HandleByValArgumentInit(Value *Dst, Value *Src, Module *M, 13780b57cec5SDimitry Andric BasicBlock *InsertBlock, 13790b57cec5SDimitry Andric InlineFunctionInfo &IFI) { 13800b57cec5SDimitry Andric Type *AggTy = cast<PointerType>(Src->getType())->getElementType(); 13810b57cec5SDimitry Andric IRBuilder<> Builder(InsertBlock, InsertBlock->begin()); 13820b57cec5SDimitry Andric 13830b57cec5SDimitry Andric Value *Size = Builder.getInt64(M->getDataLayout().getTypeStoreSize(AggTy)); 13840b57cec5SDimitry Andric 13850b57cec5SDimitry Andric // Always generate a memcpy of alignment 1 here because we don't know 13860b57cec5SDimitry Andric // the alignment of the src pointer. Other optimizations can infer 13870b57cec5SDimitry Andric // better alignment. 13885ffd83dbSDimitry Andric Builder.CreateMemCpy(Dst, /*DstAlign*/ Align(1), Src, 13895ffd83dbSDimitry Andric /*SrcAlign*/ Align(1), Size); 13900b57cec5SDimitry Andric } 13910b57cec5SDimitry Andric 13920b57cec5SDimitry Andric /// When inlining a call site that has a byval argument, 13930b57cec5SDimitry Andric /// we have to make the implicit memcpy explicit by adding it. 13940b57cec5SDimitry Andric static Value *HandleByValArgument(Value *Arg, Instruction *TheCall, 13950b57cec5SDimitry Andric const Function *CalledFunc, 13960b57cec5SDimitry Andric InlineFunctionInfo &IFI, 13970b57cec5SDimitry Andric unsigned ByValAlignment) { 13980b57cec5SDimitry Andric PointerType *ArgTy = cast<PointerType>(Arg->getType()); 13990b57cec5SDimitry Andric Type *AggTy = ArgTy->getElementType(); 14000b57cec5SDimitry Andric 14010b57cec5SDimitry Andric Function *Caller = TheCall->getFunction(); 14020b57cec5SDimitry Andric const DataLayout &DL = Caller->getParent()->getDataLayout(); 14030b57cec5SDimitry Andric 14040b57cec5SDimitry Andric // If the called function is readonly, then it could not mutate the caller's 14050b57cec5SDimitry Andric // copy of the byval'd memory. In this case, it is safe to elide the copy and 14060b57cec5SDimitry Andric // temporary. 14070b57cec5SDimitry Andric if (CalledFunc->onlyReadsMemory()) { 14080b57cec5SDimitry Andric // If the byval argument has a specified alignment that is greater than the 14090b57cec5SDimitry Andric // passed in pointer, then we either have to round up the input pointer or 14100b57cec5SDimitry Andric // give up on this transformation. 14110b57cec5SDimitry Andric if (ByValAlignment <= 1) // 0 = unspecified, 1 = no particular alignment. 14120b57cec5SDimitry Andric return Arg; 14130b57cec5SDimitry Andric 14140b57cec5SDimitry Andric AssumptionCache *AC = 14155ffd83dbSDimitry Andric IFI.GetAssumptionCache ? &IFI.GetAssumptionCache(*Caller) : nullptr; 14160b57cec5SDimitry Andric 14170b57cec5SDimitry Andric // If the pointer is already known to be sufficiently aligned, or if we can 14180b57cec5SDimitry Andric // round it up to a larger alignment, then we don't need a temporary. 14195ffd83dbSDimitry Andric if (getOrEnforceKnownAlignment(Arg, Align(ByValAlignment), DL, TheCall, 14205ffd83dbSDimitry Andric AC) >= ByValAlignment) 14210b57cec5SDimitry Andric return Arg; 14220b57cec5SDimitry Andric 14230b57cec5SDimitry Andric // Otherwise, we have to make a memcpy to get a safe alignment. This is bad 14240b57cec5SDimitry Andric // for code quality, but rarely happens and is required for correctness. 14250b57cec5SDimitry Andric } 14260b57cec5SDimitry Andric 14270b57cec5SDimitry Andric // Create the alloca. If we have DataLayout, use nice alignment. 1428480093f4SDimitry Andric Align Alignment(DL.getPrefTypeAlignment(AggTy)); 14290b57cec5SDimitry Andric 14300b57cec5SDimitry Andric // If the byval had an alignment specified, we *must* use at least that 14310b57cec5SDimitry Andric // alignment, as it is required by the byval argument (and uses of the 14320b57cec5SDimitry Andric // pointer inside the callee). 1433480093f4SDimitry Andric Alignment = max(Alignment, MaybeAlign(ByValAlignment)); 14340b57cec5SDimitry Andric 1435480093f4SDimitry Andric Value *NewAlloca = 1436480093f4SDimitry Andric new AllocaInst(AggTy, DL.getAllocaAddrSpace(), nullptr, Alignment, 1437480093f4SDimitry Andric Arg->getName(), &*Caller->begin()->begin()); 14380b57cec5SDimitry Andric IFI.StaticAllocas.push_back(cast<AllocaInst>(NewAlloca)); 14390b57cec5SDimitry Andric 14400b57cec5SDimitry Andric // Uses of the argument in the function should use our new alloca 14410b57cec5SDimitry Andric // instead. 14420b57cec5SDimitry Andric return NewAlloca; 14430b57cec5SDimitry Andric } 14440b57cec5SDimitry Andric 14450b57cec5SDimitry Andric // Check whether this Value is used by a lifetime intrinsic. 14460b57cec5SDimitry Andric static bool isUsedByLifetimeMarker(Value *V) { 14470b57cec5SDimitry Andric for (User *U : V->users()) 14480b57cec5SDimitry Andric if (IntrinsicInst *II = dyn_cast<IntrinsicInst>(U)) 14490b57cec5SDimitry Andric if (II->isLifetimeStartOrEnd()) 14500b57cec5SDimitry Andric return true; 14510b57cec5SDimitry Andric return false; 14520b57cec5SDimitry Andric } 14530b57cec5SDimitry Andric 14540b57cec5SDimitry Andric // Check whether the given alloca already has 14550b57cec5SDimitry Andric // lifetime.start or lifetime.end intrinsics. 14560b57cec5SDimitry Andric static bool hasLifetimeMarkers(AllocaInst *AI) { 14570b57cec5SDimitry Andric Type *Ty = AI->getType(); 14580b57cec5SDimitry Andric Type *Int8PtrTy = Type::getInt8PtrTy(Ty->getContext(), 14590b57cec5SDimitry Andric Ty->getPointerAddressSpace()); 14600b57cec5SDimitry Andric if (Ty == Int8PtrTy) 14610b57cec5SDimitry Andric return isUsedByLifetimeMarker(AI); 14620b57cec5SDimitry Andric 14630b57cec5SDimitry Andric // Do a scan to find all the casts to i8*. 14640b57cec5SDimitry Andric for (User *U : AI->users()) { 14650b57cec5SDimitry Andric if (U->getType() != Int8PtrTy) continue; 14660b57cec5SDimitry Andric if (U->stripPointerCasts() != AI) continue; 14670b57cec5SDimitry Andric if (isUsedByLifetimeMarker(U)) 14680b57cec5SDimitry Andric return true; 14690b57cec5SDimitry Andric } 14700b57cec5SDimitry Andric return false; 14710b57cec5SDimitry Andric } 14720b57cec5SDimitry Andric 14730b57cec5SDimitry Andric /// Return the result of AI->isStaticAlloca() if AI were moved to the entry 14740b57cec5SDimitry Andric /// block. Allocas used in inalloca calls and allocas of dynamic array size 14750b57cec5SDimitry Andric /// cannot be static. 14760b57cec5SDimitry Andric static bool allocaWouldBeStaticInEntry(const AllocaInst *AI ) { 14770b57cec5SDimitry Andric return isa<Constant>(AI->getArraySize()) && !AI->isUsedWithInAlloca(); 14780b57cec5SDimitry Andric } 14790b57cec5SDimitry Andric 14800b57cec5SDimitry Andric /// Returns a DebugLoc for a new DILocation which is a clone of \p OrigDL 14810b57cec5SDimitry Andric /// inlined at \p InlinedAt. \p IANodes is an inlined-at cache. 14820b57cec5SDimitry Andric static DebugLoc inlineDebugLoc(DebugLoc OrigDL, DILocation *InlinedAt, 14830b57cec5SDimitry Andric LLVMContext &Ctx, 14840b57cec5SDimitry Andric DenseMap<const MDNode *, MDNode *> &IANodes) { 14850b57cec5SDimitry Andric auto IA = DebugLoc::appendInlinedAt(OrigDL, InlinedAt, Ctx, IANodes); 1486e8d8bef9SDimitry Andric return DILocation::get(Ctx, OrigDL.getLine(), OrigDL.getCol(), 1487e8d8bef9SDimitry Andric OrigDL.getScope(), IA); 14880b57cec5SDimitry Andric } 14890b57cec5SDimitry Andric 14900b57cec5SDimitry Andric /// Update inlined instructions' line numbers to 14910b57cec5SDimitry Andric /// to encode location where these instructions are inlined. 14920b57cec5SDimitry Andric static void fixupLineNumbers(Function *Fn, Function::iterator FI, 14930b57cec5SDimitry Andric Instruction *TheCall, bool CalleeHasDebugInfo) { 14940b57cec5SDimitry Andric const DebugLoc &TheCallDL = TheCall->getDebugLoc(); 14950b57cec5SDimitry Andric if (!TheCallDL) 14960b57cec5SDimitry Andric return; 14970b57cec5SDimitry Andric 14980b57cec5SDimitry Andric auto &Ctx = Fn->getContext(); 14990b57cec5SDimitry Andric DILocation *InlinedAtNode = TheCallDL; 15000b57cec5SDimitry Andric 15010b57cec5SDimitry Andric // Create a unique call site, not to be confused with any other call from the 15020b57cec5SDimitry Andric // same location. 15030b57cec5SDimitry Andric InlinedAtNode = DILocation::getDistinct( 15040b57cec5SDimitry Andric Ctx, InlinedAtNode->getLine(), InlinedAtNode->getColumn(), 15050b57cec5SDimitry Andric InlinedAtNode->getScope(), InlinedAtNode->getInlinedAt()); 15060b57cec5SDimitry Andric 15070b57cec5SDimitry Andric // Cache the inlined-at nodes as they're built so they are reused, without 15080b57cec5SDimitry Andric // this every instruction's inlined-at chain would become distinct from each 15090b57cec5SDimitry Andric // other. 15100b57cec5SDimitry Andric DenseMap<const MDNode *, MDNode *> IANodes; 15110b57cec5SDimitry Andric 1512480093f4SDimitry Andric // Check if we are not generating inline line tables and want to use 1513480093f4SDimitry Andric // the call site location instead. 1514480093f4SDimitry Andric bool NoInlineLineTables = Fn->hasFnAttribute("no-inline-line-tables"); 1515480093f4SDimitry Andric 15160b57cec5SDimitry Andric for (; FI != Fn->end(); ++FI) { 15170b57cec5SDimitry Andric for (BasicBlock::iterator BI = FI->begin(), BE = FI->end(); 15180b57cec5SDimitry Andric BI != BE; ++BI) { 15190b57cec5SDimitry Andric // Loop metadata needs to be updated so that the start and end locs 15200b57cec5SDimitry Andric // reference inlined-at locations. 15215ffd83dbSDimitry Andric auto updateLoopInfoLoc = [&Ctx, &InlinedAtNode, &IANodes]( 15225ffd83dbSDimitry Andric const DILocation &Loc) -> DILocation * { 15235ffd83dbSDimitry Andric return inlineDebugLoc(&Loc, InlinedAtNode, Ctx, IANodes).get(); 15245ffd83dbSDimitry Andric }; 15255ffd83dbSDimitry Andric updateLoopMetadataDebugLocations(*BI, updateLoopInfoLoc); 15260b57cec5SDimitry Andric 1527480093f4SDimitry Andric if (!NoInlineLineTables) 15280b57cec5SDimitry Andric if (DebugLoc DL = BI->getDebugLoc()) { 15290b57cec5SDimitry Andric DebugLoc IDL = 15300b57cec5SDimitry Andric inlineDebugLoc(DL, InlinedAtNode, BI->getContext(), IANodes); 15310b57cec5SDimitry Andric BI->setDebugLoc(IDL); 15320b57cec5SDimitry Andric continue; 15330b57cec5SDimitry Andric } 15340b57cec5SDimitry Andric 1535480093f4SDimitry Andric if (CalleeHasDebugInfo && !NoInlineLineTables) 15360b57cec5SDimitry Andric continue; 15370b57cec5SDimitry Andric 1538480093f4SDimitry Andric // If the inlined instruction has no line number, or if inline info 1539480093f4SDimitry Andric // is not being generated, make it look as if it originates from the call 1540480093f4SDimitry Andric // location. This is important for ((__always_inline, __nodebug__)) 1541480093f4SDimitry Andric // functions which must use caller location for all instructions in their 1542480093f4SDimitry Andric // function body. 15430b57cec5SDimitry Andric 15440b57cec5SDimitry Andric // Don't update static allocas, as they may get moved later. 15450b57cec5SDimitry Andric if (auto *AI = dyn_cast<AllocaInst>(BI)) 15460b57cec5SDimitry Andric if (allocaWouldBeStaticInEntry(AI)) 15470b57cec5SDimitry Andric continue; 15480b57cec5SDimitry Andric 15490b57cec5SDimitry Andric BI->setDebugLoc(TheCallDL); 15500b57cec5SDimitry Andric } 1551480093f4SDimitry Andric 1552480093f4SDimitry Andric // Remove debug info intrinsics if we're not keeping inline info. 1553480093f4SDimitry Andric if (NoInlineLineTables) { 1554480093f4SDimitry Andric BasicBlock::iterator BI = FI->begin(); 1555480093f4SDimitry Andric while (BI != FI->end()) { 1556480093f4SDimitry Andric if (isa<DbgInfoIntrinsic>(BI)) { 1557480093f4SDimitry Andric BI = BI->eraseFromParent(); 1558480093f4SDimitry Andric continue; 1559480093f4SDimitry Andric } 1560480093f4SDimitry Andric ++BI; 1561480093f4SDimitry Andric } 1562480093f4SDimitry Andric } 1563480093f4SDimitry Andric 15640b57cec5SDimitry Andric } 15650b57cec5SDimitry Andric } 15660b57cec5SDimitry Andric 15670b57cec5SDimitry Andric /// Update the block frequencies of the caller after a callee has been inlined. 15680b57cec5SDimitry Andric /// 15690b57cec5SDimitry Andric /// Each block cloned into the caller has its block frequency scaled by the 15700b57cec5SDimitry Andric /// ratio of CallSiteFreq/CalleeEntryFreq. This ensures that the cloned copy of 15710b57cec5SDimitry Andric /// callee's entry block gets the same frequency as the callsite block and the 15720b57cec5SDimitry Andric /// relative frequencies of all cloned blocks remain the same after cloning. 15730b57cec5SDimitry Andric static void updateCallerBFI(BasicBlock *CallSiteBlock, 15740b57cec5SDimitry Andric const ValueToValueMapTy &VMap, 15750b57cec5SDimitry Andric BlockFrequencyInfo *CallerBFI, 15760b57cec5SDimitry Andric BlockFrequencyInfo *CalleeBFI, 15770b57cec5SDimitry Andric const BasicBlock &CalleeEntryBlock) { 15780b57cec5SDimitry Andric SmallPtrSet<BasicBlock *, 16> ClonedBBs; 1579480093f4SDimitry Andric for (auto Entry : VMap) { 15800b57cec5SDimitry Andric if (!isa<BasicBlock>(Entry.first) || !Entry.second) 15810b57cec5SDimitry Andric continue; 15820b57cec5SDimitry Andric auto *OrigBB = cast<BasicBlock>(Entry.first); 15830b57cec5SDimitry Andric auto *ClonedBB = cast<BasicBlock>(Entry.second); 15840b57cec5SDimitry Andric uint64_t Freq = CalleeBFI->getBlockFreq(OrigBB).getFrequency(); 15850b57cec5SDimitry Andric if (!ClonedBBs.insert(ClonedBB).second) { 15860b57cec5SDimitry Andric // Multiple blocks in the callee might get mapped to one cloned block in 15870b57cec5SDimitry Andric // the caller since we prune the callee as we clone it. When that happens, 15880b57cec5SDimitry Andric // we want to use the maximum among the original blocks' frequencies. 15890b57cec5SDimitry Andric uint64_t NewFreq = CallerBFI->getBlockFreq(ClonedBB).getFrequency(); 15900b57cec5SDimitry Andric if (NewFreq > Freq) 15910b57cec5SDimitry Andric Freq = NewFreq; 15920b57cec5SDimitry Andric } 15930b57cec5SDimitry Andric CallerBFI->setBlockFreq(ClonedBB, Freq); 15940b57cec5SDimitry Andric } 15950b57cec5SDimitry Andric BasicBlock *EntryClone = cast<BasicBlock>(VMap.lookup(&CalleeEntryBlock)); 15960b57cec5SDimitry Andric CallerBFI->setBlockFreqAndScale( 15970b57cec5SDimitry Andric EntryClone, CallerBFI->getBlockFreq(CallSiteBlock).getFrequency(), 15980b57cec5SDimitry Andric ClonedBBs); 15990b57cec5SDimitry Andric } 16000b57cec5SDimitry Andric 16010b57cec5SDimitry Andric /// Update the branch metadata for cloned call instructions. 16020b57cec5SDimitry Andric static void updateCallProfile(Function *Callee, const ValueToValueMapTy &VMap, 16030b57cec5SDimitry Andric const ProfileCount &CalleeEntryCount, 16045ffd83dbSDimitry Andric const CallBase &TheCall, ProfileSummaryInfo *PSI, 16050b57cec5SDimitry Andric BlockFrequencyInfo *CallerBFI) { 16060b57cec5SDimitry Andric if (!CalleeEntryCount.hasValue() || CalleeEntryCount.isSynthetic() || 16070b57cec5SDimitry Andric CalleeEntryCount.getCount() < 1) 16080b57cec5SDimitry Andric return; 16090b57cec5SDimitry Andric auto CallSiteCount = PSI ? PSI->getProfileCount(TheCall, CallerBFI) : None; 16100b57cec5SDimitry Andric int64_t CallCount = 1611e8d8bef9SDimitry Andric std::min(CallSiteCount.getValueOr(0), CalleeEntryCount.getCount()); 16120b57cec5SDimitry Andric updateProfileCallee(Callee, -CallCount, &VMap); 16130b57cec5SDimitry Andric } 16140b57cec5SDimitry Andric 16150b57cec5SDimitry Andric void llvm::updateProfileCallee( 16160b57cec5SDimitry Andric Function *Callee, int64_t entryDelta, 16170b57cec5SDimitry Andric const ValueMap<const Value *, WeakTrackingVH> *VMap) { 16180b57cec5SDimitry Andric auto CalleeCount = Callee->getEntryCount(); 16190b57cec5SDimitry Andric if (!CalleeCount.hasValue()) 16200b57cec5SDimitry Andric return; 16210b57cec5SDimitry Andric 16220b57cec5SDimitry Andric uint64_t priorEntryCount = CalleeCount.getCount(); 16230b57cec5SDimitry Andric uint64_t newEntryCount; 16240b57cec5SDimitry Andric 16250b57cec5SDimitry Andric // Since CallSiteCount is an estimate, it could exceed the original callee 16260b57cec5SDimitry Andric // count and has to be set to 0 so guard against underflow. 16270b57cec5SDimitry Andric if (entryDelta < 0 && static_cast<uint64_t>(-entryDelta) > priorEntryCount) 16280b57cec5SDimitry Andric newEntryCount = 0; 16290b57cec5SDimitry Andric else 16300b57cec5SDimitry Andric newEntryCount = priorEntryCount + entryDelta; 16310b57cec5SDimitry Andric 16320b57cec5SDimitry Andric // During inlining ? 16330b57cec5SDimitry Andric if (VMap) { 16340b57cec5SDimitry Andric uint64_t cloneEntryCount = priorEntryCount - newEntryCount; 1635480093f4SDimitry Andric for (auto Entry : *VMap) 16360b57cec5SDimitry Andric if (isa<CallInst>(Entry.first)) 16370b57cec5SDimitry Andric if (auto *CI = dyn_cast_or_null<CallInst>(Entry.second)) 16380b57cec5SDimitry Andric CI->updateProfWeight(cloneEntryCount, priorEntryCount); 16390b57cec5SDimitry Andric } 1640480093f4SDimitry Andric 1641480093f4SDimitry Andric if (entryDelta) { 1642480093f4SDimitry Andric Callee->setEntryCount(newEntryCount); 1643480093f4SDimitry Andric 16440b57cec5SDimitry Andric for (BasicBlock &BB : *Callee) 16450b57cec5SDimitry Andric // No need to update the callsite if it is pruned during inlining. 16460b57cec5SDimitry Andric if (!VMap || VMap->count(&BB)) 16470b57cec5SDimitry Andric for (Instruction &I : BB) 16480b57cec5SDimitry Andric if (CallInst *CI = dyn_cast<CallInst>(&I)) 16490b57cec5SDimitry Andric CI->updateProfWeight(newEntryCount, priorEntryCount); 16500b57cec5SDimitry Andric } 1651480093f4SDimitry Andric } 16520b57cec5SDimitry Andric 16530b57cec5SDimitry Andric /// This function inlines the called function into the basic block of the 16540b57cec5SDimitry Andric /// caller. This returns false if it is not possible to inline this call. 16550b57cec5SDimitry Andric /// The program is still in a well defined state if this occurs though. 16560b57cec5SDimitry Andric /// 16570b57cec5SDimitry Andric /// Note that this only does one level of inlining. For example, if the 16580b57cec5SDimitry Andric /// instruction 'call B' is inlined, and 'B' calls 'C', then the call to 'C' now 16590b57cec5SDimitry Andric /// exists in the instruction stream. Similarly this will inline a recursive 16600b57cec5SDimitry Andric /// function by one level. 16615ffd83dbSDimitry Andric llvm::InlineResult llvm::InlineFunction(CallBase &CB, InlineFunctionInfo &IFI, 16620b57cec5SDimitry Andric AAResults *CalleeAAR, 16630b57cec5SDimitry Andric bool InsertLifetime, 16640b57cec5SDimitry Andric Function *ForwardVarArgsTo) { 16655ffd83dbSDimitry Andric assert(CB.getParent() && CB.getFunction() && "Instruction not in function!"); 16660b57cec5SDimitry Andric 16670b57cec5SDimitry Andric // FIXME: we don't inline callbr yet. 16685ffd83dbSDimitry Andric if (isa<CallBrInst>(CB)) 16695ffd83dbSDimitry Andric return InlineResult::failure("We don't inline callbr yet."); 16700b57cec5SDimitry Andric 16710b57cec5SDimitry Andric // If IFI has any state in it, zap it before we fill it in. 16720b57cec5SDimitry Andric IFI.reset(); 16730b57cec5SDimitry Andric 16745ffd83dbSDimitry Andric Function *CalledFunc = CB.getCalledFunction(); 16750b57cec5SDimitry Andric if (!CalledFunc || // Can't inline external function or indirect 16760b57cec5SDimitry Andric CalledFunc->isDeclaration()) // call! 16775ffd83dbSDimitry Andric return InlineResult::failure("external or indirect"); 16780b57cec5SDimitry Andric 16790b57cec5SDimitry Andric // The inliner does not know how to inline through calls with operand bundles 16800b57cec5SDimitry Andric // in general ... 16815ffd83dbSDimitry Andric if (CB.hasOperandBundles()) { 16825ffd83dbSDimitry Andric for (int i = 0, e = CB.getNumOperandBundles(); i != e; ++i) { 16835ffd83dbSDimitry Andric uint32_t Tag = CB.getOperandBundleAt(i).getTagID(); 16840b57cec5SDimitry Andric // ... but it knows how to inline through "deopt" operand bundles ... 16850b57cec5SDimitry Andric if (Tag == LLVMContext::OB_deopt) 16860b57cec5SDimitry Andric continue; 16870b57cec5SDimitry Andric // ... and "funclet" operand bundles. 16880b57cec5SDimitry Andric if (Tag == LLVMContext::OB_funclet) 16890b57cec5SDimitry Andric continue; 16900b57cec5SDimitry Andric 16915ffd83dbSDimitry Andric return InlineResult::failure("unsupported operand bundle"); 16920b57cec5SDimitry Andric } 16930b57cec5SDimitry Andric } 16940b57cec5SDimitry Andric 16950b57cec5SDimitry Andric // If the call to the callee cannot throw, set the 'nounwind' flag on any 16960b57cec5SDimitry Andric // calls that we inline. 16975ffd83dbSDimitry Andric bool MarkNoUnwind = CB.doesNotThrow(); 16980b57cec5SDimitry Andric 16995ffd83dbSDimitry Andric BasicBlock *OrigBB = CB.getParent(); 17000b57cec5SDimitry Andric Function *Caller = OrigBB->getParent(); 17010b57cec5SDimitry Andric 17020b57cec5SDimitry Andric // GC poses two hazards to inlining, which only occur when the callee has GC: 17030b57cec5SDimitry Andric // 1. If the caller has no GC, then the callee's GC must be propagated to the 17040b57cec5SDimitry Andric // caller. 17050b57cec5SDimitry Andric // 2. If the caller has a differing GC, it is invalid to inline. 17060b57cec5SDimitry Andric if (CalledFunc->hasGC()) { 17070b57cec5SDimitry Andric if (!Caller->hasGC()) 17080b57cec5SDimitry Andric Caller->setGC(CalledFunc->getGC()); 17090b57cec5SDimitry Andric else if (CalledFunc->getGC() != Caller->getGC()) 17105ffd83dbSDimitry Andric return InlineResult::failure("incompatible GC"); 17110b57cec5SDimitry Andric } 17120b57cec5SDimitry Andric 17130b57cec5SDimitry Andric // Get the personality function from the callee if it contains a landing pad. 17140b57cec5SDimitry Andric Constant *CalledPersonality = 17150b57cec5SDimitry Andric CalledFunc->hasPersonalityFn() 17160b57cec5SDimitry Andric ? CalledFunc->getPersonalityFn()->stripPointerCasts() 17170b57cec5SDimitry Andric : nullptr; 17180b57cec5SDimitry Andric 17190b57cec5SDimitry Andric // Find the personality function used by the landing pads of the caller. If it 17200b57cec5SDimitry Andric // exists, then check to see that it matches the personality function used in 17210b57cec5SDimitry Andric // the callee. 17220b57cec5SDimitry Andric Constant *CallerPersonality = 17230b57cec5SDimitry Andric Caller->hasPersonalityFn() 17240b57cec5SDimitry Andric ? Caller->getPersonalityFn()->stripPointerCasts() 17250b57cec5SDimitry Andric : nullptr; 17260b57cec5SDimitry Andric if (CalledPersonality) { 17270b57cec5SDimitry Andric if (!CallerPersonality) 17280b57cec5SDimitry Andric Caller->setPersonalityFn(CalledPersonality); 17290b57cec5SDimitry Andric // If the personality functions match, then we can perform the 17300b57cec5SDimitry Andric // inlining. Otherwise, we can't inline. 17310b57cec5SDimitry Andric // TODO: This isn't 100% true. Some personality functions are proper 17320b57cec5SDimitry Andric // supersets of others and can be used in place of the other. 17330b57cec5SDimitry Andric else if (CalledPersonality != CallerPersonality) 17345ffd83dbSDimitry Andric return InlineResult::failure("incompatible personality"); 17350b57cec5SDimitry Andric } 17360b57cec5SDimitry Andric 17370b57cec5SDimitry Andric // We need to figure out which funclet the callsite was in so that we may 17380b57cec5SDimitry Andric // properly nest the callee. 17390b57cec5SDimitry Andric Instruction *CallSiteEHPad = nullptr; 17400b57cec5SDimitry Andric if (CallerPersonality) { 17410b57cec5SDimitry Andric EHPersonality Personality = classifyEHPersonality(CallerPersonality); 17420b57cec5SDimitry Andric if (isScopedEHPersonality(Personality)) { 17430b57cec5SDimitry Andric Optional<OperandBundleUse> ParentFunclet = 17445ffd83dbSDimitry Andric CB.getOperandBundle(LLVMContext::OB_funclet); 17450b57cec5SDimitry Andric if (ParentFunclet) 17460b57cec5SDimitry Andric CallSiteEHPad = cast<FuncletPadInst>(ParentFunclet->Inputs.front()); 17470b57cec5SDimitry Andric 17480b57cec5SDimitry Andric // OK, the inlining site is legal. What about the target function? 17490b57cec5SDimitry Andric 17500b57cec5SDimitry Andric if (CallSiteEHPad) { 17510b57cec5SDimitry Andric if (Personality == EHPersonality::MSVC_CXX) { 17520b57cec5SDimitry Andric // The MSVC personality cannot tolerate catches getting inlined into 17530b57cec5SDimitry Andric // cleanup funclets. 17540b57cec5SDimitry Andric if (isa<CleanupPadInst>(CallSiteEHPad)) { 17550b57cec5SDimitry Andric // Ok, the call site is within a cleanuppad. Let's check the callee 17560b57cec5SDimitry Andric // for catchpads. 17570b57cec5SDimitry Andric for (const BasicBlock &CalledBB : *CalledFunc) { 17580b57cec5SDimitry Andric if (isa<CatchSwitchInst>(CalledBB.getFirstNonPHI())) 17595ffd83dbSDimitry Andric return InlineResult::failure("catch in cleanup funclet"); 17600b57cec5SDimitry Andric } 17610b57cec5SDimitry Andric } 17620b57cec5SDimitry Andric } else if (isAsynchronousEHPersonality(Personality)) { 17630b57cec5SDimitry Andric // SEH is even less tolerant, there may not be any sort of exceptional 17640b57cec5SDimitry Andric // funclet in the callee. 17650b57cec5SDimitry Andric for (const BasicBlock &CalledBB : *CalledFunc) { 17660b57cec5SDimitry Andric if (CalledBB.isEHPad()) 17675ffd83dbSDimitry Andric return InlineResult::failure("SEH in cleanup funclet"); 17680b57cec5SDimitry Andric } 17690b57cec5SDimitry Andric } 17700b57cec5SDimitry Andric } 17710b57cec5SDimitry Andric } 17720b57cec5SDimitry Andric } 17730b57cec5SDimitry Andric 17740b57cec5SDimitry Andric // Determine if we are dealing with a call in an EHPad which does not unwind 17750b57cec5SDimitry Andric // to caller. 17760b57cec5SDimitry Andric bool EHPadForCallUnwindsLocally = false; 17775ffd83dbSDimitry Andric if (CallSiteEHPad && isa<CallInst>(CB)) { 17780b57cec5SDimitry Andric UnwindDestMemoTy FuncletUnwindMap; 17790b57cec5SDimitry Andric Value *CallSiteUnwindDestToken = 17800b57cec5SDimitry Andric getUnwindDestToken(CallSiteEHPad, FuncletUnwindMap); 17810b57cec5SDimitry Andric 17820b57cec5SDimitry Andric EHPadForCallUnwindsLocally = 17830b57cec5SDimitry Andric CallSiteUnwindDestToken && 17840b57cec5SDimitry Andric !isa<ConstantTokenNone>(CallSiteUnwindDestToken); 17850b57cec5SDimitry Andric } 17860b57cec5SDimitry Andric 17870b57cec5SDimitry Andric // Get an iterator to the last basic block in the function, which will have 17880b57cec5SDimitry Andric // the new function inlined after it. 17890b57cec5SDimitry Andric Function::iterator LastBlock = --Caller->end(); 17900b57cec5SDimitry Andric 17910b57cec5SDimitry Andric // Make sure to capture all of the return instructions from the cloned 17920b57cec5SDimitry Andric // function. 17930b57cec5SDimitry Andric SmallVector<ReturnInst*, 8> Returns; 17940b57cec5SDimitry Andric ClonedCodeInfo InlinedFunctionInfo; 17950b57cec5SDimitry Andric Function::iterator FirstNewBlock; 17960b57cec5SDimitry Andric 17970b57cec5SDimitry Andric { // Scope to destroy VMap after cloning. 17980b57cec5SDimitry Andric ValueToValueMapTy VMap; 17990b57cec5SDimitry Andric // Keep a list of pair (dst, src) to emit byval initializations. 18000b57cec5SDimitry Andric SmallVector<std::pair<Value*, Value*>, 4> ByValInit; 18010b57cec5SDimitry Andric 1802e8d8bef9SDimitry Andric // When inlining a function that contains noalias scope metadata, 1803e8d8bef9SDimitry Andric // this metadata needs to be cloned so that the inlined blocks 1804e8d8bef9SDimitry Andric // have different "unique scopes" at every call site. 1805e8d8bef9SDimitry Andric // Track the metadata that must be cloned. Do this before other changes to 1806e8d8bef9SDimitry Andric // the function, so that we do not get in trouble when inlining caller == 1807e8d8bef9SDimitry Andric // callee. 1808e8d8bef9SDimitry Andric ScopedAliasMetadataDeepCloner SAMetadataCloner(CB.getCalledFunction()); 1809e8d8bef9SDimitry Andric 18100b57cec5SDimitry Andric auto &DL = Caller->getParent()->getDataLayout(); 18110b57cec5SDimitry Andric 18120b57cec5SDimitry Andric // Calculate the vector of arguments to pass into the function cloner, which 18130b57cec5SDimitry Andric // matches up the formal to the actual argument values. 18145ffd83dbSDimitry Andric auto AI = CB.arg_begin(); 18150b57cec5SDimitry Andric unsigned ArgNo = 0; 18160b57cec5SDimitry Andric for (Function::arg_iterator I = CalledFunc->arg_begin(), 18170b57cec5SDimitry Andric E = CalledFunc->arg_end(); I != E; ++I, ++AI, ++ArgNo) { 18180b57cec5SDimitry Andric Value *ActualArg = *AI; 18190b57cec5SDimitry Andric 18200b57cec5SDimitry Andric // When byval arguments actually inlined, we need to make the copy implied 18210b57cec5SDimitry Andric // by them explicit. However, we don't do this if the callee is readonly 18220b57cec5SDimitry Andric // or readnone, because the copy would be unneeded: the callee doesn't 18230b57cec5SDimitry Andric // modify the struct. 18245ffd83dbSDimitry Andric if (CB.isByValArgument(ArgNo)) { 18255ffd83dbSDimitry Andric ActualArg = HandleByValArgument(ActualArg, &CB, CalledFunc, IFI, 18260b57cec5SDimitry Andric CalledFunc->getParamAlignment(ArgNo)); 18270b57cec5SDimitry Andric if (ActualArg != *AI) 18280b57cec5SDimitry Andric ByValInit.push_back(std::make_pair(ActualArg, (Value*) *AI)); 18290b57cec5SDimitry Andric } 18300b57cec5SDimitry Andric 18310b57cec5SDimitry Andric VMap[&*I] = ActualArg; 18320b57cec5SDimitry Andric } 18330b57cec5SDimitry Andric 18345ffd83dbSDimitry Andric // TODO: Remove this when users have been updated to the assume bundles. 18350b57cec5SDimitry Andric // Add alignment assumptions if necessary. We do this before the inlined 18360b57cec5SDimitry Andric // instructions are actually cloned into the caller so that we can easily 18370b57cec5SDimitry Andric // check what will be known at the start of the inlined code. 18385ffd83dbSDimitry Andric AddAlignmentAssumptions(CB, IFI); 18395ffd83dbSDimitry Andric 18405ffd83dbSDimitry Andric AssumptionCache *AC = 18415ffd83dbSDimitry Andric IFI.GetAssumptionCache ? &IFI.GetAssumptionCache(*Caller) : nullptr; 18425ffd83dbSDimitry Andric 18435ffd83dbSDimitry Andric /// Preserve all attributes on of the call and its parameters. 18445ffd83dbSDimitry Andric salvageKnowledge(&CB, AC); 18450b57cec5SDimitry Andric 18460b57cec5SDimitry Andric // We want the inliner to prune the code as it copies. We would LOVE to 18470b57cec5SDimitry Andric // have no dead or constant instructions leftover after inlining occurs 18480b57cec5SDimitry Andric // (which can happen, e.g., because an argument was constant), but we'll be 18490b57cec5SDimitry Andric // happy with whatever the cloner can do. 18500b57cec5SDimitry Andric CloneAndPruneFunctionInto(Caller, CalledFunc, VMap, 18510b57cec5SDimitry Andric /*ModuleLevelChanges=*/false, Returns, ".i", 18525ffd83dbSDimitry Andric &InlinedFunctionInfo, &CB); 18530b57cec5SDimitry Andric // Remember the first block that is newly cloned over. 18540b57cec5SDimitry Andric FirstNewBlock = LastBlock; ++FirstNewBlock; 18550b57cec5SDimitry Andric 18560b57cec5SDimitry Andric if (IFI.CallerBFI != nullptr && IFI.CalleeBFI != nullptr) 18570b57cec5SDimitry Andric // Update the BFI of blocks cloned into the caller. 18580b57cec5SDimitry Andric updateCallerBFI(OrigBB, VMap, IFI.CallerBFI, IFI.CalleeBFI, 18590b57cec5SDimitry Andric CalledFunc->front()); 18600b57cec5SDimitry Andric 18615ffd83dbSDimitry Andric updateCallProfile(CalledFunc, VMap, CalledFunc->getEntryCount(), CB, 18620b57cec5SDimitry Andric IFI.PSI, IFI.CallerBFI); 18630b57cec5SDimitry Andric 18640b57cec5SDimitry Andric // Inject byval arguments initialization. 18650b57cec5SDimitry Andric for (std::pair<Value*, Value*> &Init : ByValInit) 18660b57cec5SDimitry Andric HandleByValArgumentInit(Init.first, Init.second, Caller->getParent(), 18670b57cec5SDimitry Andric &*FirstNewBlock, IFI); 18680b57cec5SDimitry Andric 18690b57cec5SDimitry Andric Optional<OperandBundleUse> ParentDeopt = 18705ffd83dbSDimitry Andric CB.getOperandBundle(LLVMContext::OB_deopt); 18710b57cec5SDimitry Andric if (ParentDeopt) { 18720b57cec5SDimitry Andric SmallVector<OperandBundleDef, 2> OpDefs; 18730b57cec5SDimitry Andric 18740b57cec5SDimitry Andric for (auto &VH : InlinedFunctionInfo.OperandBundleCallSites) { 18755ffd83dbSDimitry Andric CallBase *ICS = dyn_cast_or_null<CallBase>(VH); 18765ffd83dbSDimitry Andric if (!ICS) 18775ffd83dbSDimitry Andric continue; // instruction was DCE'd or RAUW'ed to undef 18780b57cec5SDimitry Andric 18790b57cec5SDimitry Andric OpDefs.clear(); 18800b57cec5SDimitry Andric 18815ffd83dbSDimitry Andric OpDefs.reserve(ICS->getNumOperandBundles()); 18820b57cec5SDimitry Andric 18835ffd83dbSDimitry Andric for (unsigned COBi = 0, COBe = ICS->getNumOperandBundles(); COBi < COBe; 18845ffd83dbSDimitry Andric ++COBi) { 18855ffd83dbSDimitry Andric auto ChildOB = ICS->getOperandBundleAt(COBi); 18860b57cec5SDimitry Andric if (ChildOB.getTagID() != LLVMContext::OB_deopt) { 18870b57cec5SDimitry Andric // If the inlined call has other operand bundles, let them be 18880b57cec5SDimitry Andric OpDefs.emplace_back(ChildOB); 18890b57cec5SDimitry Andric continue; 18900b57cec5SDimitry Andric } 18910b57cec5SDimitry Andric 18920b57cec5SDimitry Andric // It may be useful to separate this logic (of handling operand 18930b57cec5SDimitry Andric // bundles) out to a separate "policy" component if this gets crowded. 18940b57cec5SDimitry Andric // Prepend the parent's deoptimization continuation to the newly 18950b57cec5SDimitry Andric // inlined call's deoptimization continuation. 18960b57cec5SDimitry Andric std::vector<Value *> MergedDeoptArgs; 18970b57cec5SDimitry Andric MergedDeoptArgs.reserve(ParentDeopt->Inputs.size() + 18980b57cec5SDimitry Andric ChildOB.Inputs.size()); 18990b57cec5SDimitry Andric 1900e8d8bef9SDimitry Andric llvm::append_range(MergedDeoptArgs, ParentDeopt->Inputs); 1901e8d8bef9SDimitry Andric llvm::append_range(MergedDeoptArgs, ChildOB.Inputs); 19020b57cec5SDimitry Andric 19030b57cec5SDimitry Andric OpDefs.emplace_back("deopt", std::move(MergedDeoptArgs)); 19040b57cec5SDimitry Andric } 19050b57cec5SDimitry Andric 19065ffd83dbSDimitry Andric Instruction *NewI = CallBase::Create(ICS, OpDefs, ICS); 19070b57cec5SDimitry Andric 19080b57cec5SDimitry Andric // Note: the RAUW does the appropriate fixup in VMap, so we need to do 19090b57cec5SDimitry Andric // this even if the call returns void. 19105ffd83dbSDimitry Andric ICS->replaceAllUsesWith(NewI); 19110b57cec5SDimitry Andric 19120b57cec5SDimitry Andric VH = nullptr; 19135ffd83dbSDimitry Andric ICS->eraseFromParent(); 19140b57cec5SDimitry Andric } 19150b57cec5SDimitry Andric } 19160b57cec5SDimitry Andric 19170b57cec5SDimitry Andric // Update the callgraph if requested. 19180b57cec5SDimitry Andric if (IFI.CG) 19195ffd83dbSDimitry Andric UpdateCallGraphAfterInlining(CB, FirstNewBlock, VMap, IFI); 19200b57cec5SDimitry Andric 19210b57cec5SDimitry Andric // For 'nodebug' functions, the associated DISubprogram is always null. 19220b57cec5SDimitry Andric // Conservatively avoid propagating the callsite debug location to 19230b57cec5SDimitry Andric // instructions inlined from a function whose DISubprogram is not null. 19245ffd83dbSDimitry Andric fixupLineNumbers(Caller, FirstNewBlock, &CB, 19250b57cec5SDimitry Andric CalledFunc->getSubprogram() != nullptr); 19260b57cec5SDimitry Andric 1927e8d8bef9SDimitry Andric // Now clone the inlined noalias scope metadata. 1928e8d8bef9SDimitry Andric SAMetadataCloner.clone(); 1929e8d8bef9SDimitry Andric SAMetadataCloner.remap(VMap); 19300b57cec5SDimitry Andric 19310b57cec5SDimitry Andric // Add noalias metadata if necessary. 19325ffd83dbSDimitry Andric AddAliasScopeMetadata(CB, VMap, DL, CalleeAAR); 19335ffd83dbSDimitry Andric 19345ffd83dbSDimitry Andric // Clone return attributes on the callsite into the calls within the inlined 19355ffd83dbSDimitry Andric // function which feed into its return value. 19365ffd83dbSDimitry Andric AddReturnAttributes(CB, VMap); 19370b57cec5SDimitry Andric 1938e8d8bef9SDimitry Andric // Propagate metadata on the callsite if necessary. 1939e8d8bef9SDimitry Andric PropagateCallSiteMetadata(CB, VMap); 19400b57cec5SDimitry Andric 19410b57cec5SDimitry Andric // Register any cloned assumptions. 19420b57cec5SDimitry Andric if (IFI.GetAssumptionCache) 19430b57cec5SDimitry Andric for (BasicBlock &NewBlock : 19440b57cec5SDimitry Andric make_range(FirstNewBlock->getIterator(), Caller->end())) 19455ffd83dbSDimitry Andric for (Instruction &I : NewBlock) 19460b57cec5SDimitry Andric if (auto *II = dyn_cast<IntrinsicInst>(&I)) 19470b57cec5SDimitry Andric if (II->getIntrinsicID() == Intrinsic::assume) 19485ffd83dbSDimitry Andric IFI.GetAssumptionCache(*Caller).registerAssumption(II); 19490b57cec5SDimitry Andric } 19500b57cec5SDimitry Andric 19510b57cec5SDimitry Andric // If there are any alloca instructions in the block that used to be the entry 19520b57cec5SDimitry Andric // block for the callee, move them to the entry block of the caller. First 19530b57cec5SDimitry Andric // calculate which instruction they should be inserted before. We insert the 19540b57cec5SDimitry Andric // instructions at the end of the current alloca list. 19550b57cec5SDimitry Andric { 19560b57cec5SDimitry Andric BasicBlock::iterator InsertPoint = Caller->begin()->begin(); 19570b57cec5SDimitry Andric for (BasicBlock::iterator I = FirstNewBlock->begin(), 19580b57cec5SDimitry Andric E = FirstNewBlock->end(); I != E; ) { 19590b57cec5SDimitry Andric AllocaInst *AI = dyn_cast<AllocaInst>(I++); 19600b57cec5SDimitry Andric if (!AI) continue; 19610b57cec5SDimitry Andric 19620b57cec5SDimitry Andric // If the alloca is now dead, remove it. This often occurs due to code 19630b57cec5SDimitry Andric // specialization. 19640b57cec5SDimitry Andric if (AI->use_empty()) { 19650b57cec5SDimitry Andric AI->eraseFromParent(); 19660b57cec5SDimitry Andric continue; 19670b57cec5SDimitry Andric } 19680b57cec5SDimitry Andric 19690b57cec5SDimitry Andric if (!allocaWouldBeStaticInEntry(AI)) 19700b57cec5SDimitry Andric continue; 19710b57cec5SDimitry Andric 19720b57cec5SDimitry Andric // Keep track of the static allocas that we inline into the caller. 19730b57cec5SDimitry Andric IFI.StaticAllocas.push_back(AI); 19740b57cec5SDimitry Andric 19750b57cec5SDimitry Andric // Scan for the block of allocas that we can move over, and move them 19760b57cec5SDimitry Andric // all at once. 19770b57cec5SDimitry Andric while (isa<AllocaInst>(I) && 1978480093f4SDimitry Andric !cast<AllocaInst>(I)->use_empty() && 19790b57cec5SDimitry Andric allocaWouldBeStaticInEntry(cast<AllocaInst>(I))) { 19800b57cec5SDimitry Andric IFI.StaticAllocas.push_back(cast<AllocaInst>(I)); 19810b57cec5SDimitry Andric ++I; 19820b57cec5SDimitry Andric } 19830b57cec5SDimitry Andric 19840b57cec5SDimitry Andric // Transfer all of the allocas over in a block. Using splice means 19850b57cec5SDimitry Andric // that the instructions aren't removed from the symbol table, then 19860b57cec5SDimitry Andric // reinserted. 19870b57cec5SDimitry Andric Caller->getEntryBlock().getInstList().splice( 19880b57cec5SDimitry Andric InsertPoint, FirstNewBlock->getInstList(), AI->getIterator(), I); 19890b57cec5SDimitry Andric } 19900b57cec5SDimitry Andric } 19910b57cec5SDimitry Andric 19920b57cec5SDimitry Andric SmallVector<Value*,4> VarArgsToForward; 19930b57cec5SDimitry Andric SmallVector<AttributeSet, 4> VarArgsAttrs; 19940b57cec5SDimitry Andric for (unsigned i = CalledFunc->getFunctionType()->getNumParams(); 19955ffd83dbSDimitry Andric i < CB.getNumArgOperands(); i++) { 19965ffd83dbSDimitry Andric VarArgsToForward.push_back(CB.getArgOperand(i)); 19975ffd83dbSDimitry Andric VarArgsAttrs.push_back(CB.getAttributes().getParamAttributes(i)); 19980b57cec5SDimitry Andric } 19990b57cec5SDimitry Andric 20000b57cec5SDimitry Andric bool InlinedMustTailCalls = false, InlinedDeoptimizeCalls = false; 20010b57cec5SDimitry Andric if (InlinedFunctionInfo.ContainsCalls) { 20020b57cec5SDimitry Andric CallInst::TailCallKind CallSiteTailKind = CallInst::TCK_None; 20035ffd83dbSDimitry Andric if (CallInst *CI = dyn_cast<CallInst>(&CB)) 20040b57cec5SDimitry Andric CallSiteTailKind = CI->getTailCallKind(); 20050b57cec5SDimitry Andric 20060b57cec5SDimitry Andric // For inlining purposes, the "notail" marker is the same as no marker. 20070b57cec5SDimitry Andric if (CallSiteTailKind == CallInst::TCK_NoTail) 20080b57cec5SDimitry Andric CallSiteTailKind = CallInst::TCK_None; 20090b57cec5SDimitry Andric 20100b57cec5SDimitry Andric for (Function::iterator BB = FirstNewBlock, E = Caller->end(); BB != E; 20110b57cec5SDimitry Andric ++BB) { 20120b57cec5SDimitry Andric for (auto II = BB->begin(); II != BB->end();) { 20130b57cec5SDimitry Andric Instruction &I = *II++; 20140b57cec5SDimitry Andric CallInst *CI = dyn_cast<CallInst>(&I); 20150b57cec5SDimitry Andric if (!CI) 20160b57cec5SDimitry Andric continue; 20170b57cec5SDimitry Andric 20180b57cec5SDimitry Andric // Forward varargs from inlined call site to calls to the 20190b57cec5SDimitry Andric // ForwardVarArgsTo function, if requested, and to musttail calls. 20200b57cec5SDimitry Andric if (!VarArgsToForward.empty() && 20210b57cec5SDimitry Andric ((ForwardVarArgsTo && 20220b57cec5SDimitry Andric CI->getCalledFunction() == ForwardVarArgsTo) || 20230b57cec5SDimitry Andric CI->isMustTailCall())) { 20240b57cec5SDimitry Andric // Collect attributes for non-vararg parameters. 20250b57cec5SDimitry Andric AttributeList Attrs = CI->getAttributes(); 20260b57cec5SDimitry Andric SmallVector<AttributeSet, 8> ArgAttrs; 20270b57cec5SDimitry Andric if (!Attrs.isEmpty() || !VarArgsAttrs.empty()) { 20280b57cec5SDimitry Andric for (unsigned ArgNo = 0; 20290b57cec5SDimitry Andric ArgNo < CI->getFunctionType()->getNumParams(); ++ArgNo) 20300b57cec5SDimitry Andric ArgAttrs.push_back(Attrs.getParamAttributes(ArgNo)); 20310b57cec5SDimitry Andric } 20320b57cec5SDimitry Andric 20330b57cec5SDimitry Andric // Add VarArg attributes. 20340b57cec5SDimitry Andric ArgAttrs.append(VarArgsAttrs.begin(), VarArgsAttrs.end()); 20350b57cec5SDimitry Andric Attrs = AttributeList::get(CI->getContext(), Attrs.getFnAttributes(), 20360b57cec5SDimitry Andric Attrs.getRetAttributes(), ArgAttrs); 20370b57cec5SDimitry Andric // Add VarArgs to existing parameters. 20380b57cec5SDimitry Andric SmallVector<Value *, 6> Params(CI->arg_operands()); 20390b57cec5SDimitry Andric Params.append(VarArgsToForward.begin(), VarArgsToForward.end()); 20400b57cec5SDimitry Andric CallInst *NewCI = CallInst::Create( 20410b57cec5SDimitry Andric CI->getFunctionType(), CI->getCalledOperand(), Params, "", CI); 20420b57cec5SDimitry Andric NewCI->setDebugLoc(CI->getDebugLoc()); 20430b57cec5SDimitry Andric NewCI->setAttributes(Attrs); 20440b57cec5SDimitry Andric NewCI->setCallingConv(CI->getCallingConv()); 20450b57cec5SDimitry Andric CI->replaceAllUsesWith(NewCI); 20460b57cec5SDimitry Andric CI->eraseFromParent(); 20470b57cec5SDimitry Andric CI = NewCI; 20480b57cec5SDimitry Andric } 20490b57cec5SDimitry Andric 20500b57cec5SDimitry Andric if (Function *F = CI->getCalledFunction()) 20510b57cec5SDimitry Andric InlinedDeoptimizeCalls |= 20520b57cec5SDimitry Andric F->getIntrinsicID() == Intrinsic::experimental_deoptimize; 20530b57cec5SDimitry Andric 20540b57cec5SDimitry Andric // We need to reduce the strength of any inlined tail calls. For 20550b57cec5SDimitry Andric // musttail, we have to avoid introducing potential unbounded stack 20560b57cec5SDimitry Andric // growth. For example, if functions 'f' and 'g' are mutually recursive 20570b57cec5SDimitry Andric // with musttail, we can inline 'g' into 'f' so long as we preserve 20580b57cec5SDimitry Andric // musttail on the cloned call to 'f'. If either the inlined call site 20590b57cec5SDimitry Andric // or the cloned call site is *not* musttail, the program already has 20600b57cec5SDimitry Andric // one frame of stack growth, so it's safe to remove musttail. Here is 20610b57cec5SDimitry Andric // a table of example transformations: 20620b57cec5SDimitry Andric // 20630b57cec5SDimitry Andric // f -> musttail g -> musttail f ==> f -> musttail f 20640b57cec5SDimitry Andric // f -> musttail g -> tail f ==> f -> tail f 20650b57cec5SDimitry Andric // f -> g -> musttail f ==> f -> f 20660b57cec5SDimitry Andric // f -> g -> tail f ==> f -> f 20670b57cec5SDimitry Andric // 20680b57cec5SDimitry Andric // Inlined notail calls should remain notail calls. 20690b57cec5SDimitry Andric CallInst::TailCallKind ChildTCK = CI->getTailCallKind(); 20700b57cec5SDimitry Andric if (ChildTCK != CallInst::TCK_NoTail) 20710b57cec5SDimitry Andric ChildTCK = std::min(CallSiteTailKind, ChildTCK); 20720b57cec5SDimitry Andric CI->setTailCallKind(ChildTCK); 20730b57cec5SDimitry Andric InlinedMustTailCalls |= CI->isMustTailCall(); 20740b57cec5SDimitry Andric 20750b57cec5SDimitry Andric // Calls inlined through a 'nounwind' call site should be marked 20760b57cec5SDimitry Andric // 'nounwind'. 20770b57cec5SDimitry Andric if (MarkNoUnwind) 20780b57cec5SDimitry Andric CI->setDoesNotThrow(); 20790b57cec5SDimitry Andric } 20800b57cec5SDimitry Andric } 20810b57cec5SDimitry Andric } 20820b57cec5SDimitry Andric 20830b57cec5SDimitry Andric // Leave lifetime markers for the static alloca's, scoping them to the 20840b57cec5SDimitry Andric // function we just inlined. 20850b57cec5SDimitry Andric if (InsertLifetime && !IFI.StaticAllocas.empty()) { 20860b57cec5SDimitry Andric IRBuilder<> builder(&FirstNewBlock->front()); 20870b57cec5SDimitry Andric for (unsigned ai = 0, ae = IFI.StaticAllocas.size(); ai != ae; ++ai) { 20880b57cec5SDimitry Andric AllocaInst *AI = IFI.StaticAllocas[ai]; 20890b57cec5SDimitry Andric // Don't mark swifterror allocas. They can't have bitcast uses. 20900b57cec5SDimitry Andric if (AI->isSwiftError()) 20910b57cec5SDimitry Andric continue; 20920b57cec5SDimitry Andric 20930b57cec5SDimitry Andric // If the alloca is already scoped to something smaller than the whole 20940b57cec5SDimitry Andric // function then there's no need to add redundant, less accurate markers. 20950b57cec5SDimitry Andric if (hasLifetimeMarkers(AI)) 20960b57cec5SDimitry Andric continue; 20970b57cec5SDimitry Andric 20980b57cec5SDimitry Andric // Try to determine the size of the allocation. 20990b57cec5SDimitry Andric ConstantInt *AllocaSize = nullptr; 21000b57cec5SDimitry Andric if (ConstantInt *AIArraySize = 21010b57cec5SDimitry Andric dyn_cast<ConstantInt>(AI->getArraySize())) { 21020b57cec5SDimitry Andric auto &DL = Caller->getParent()->getDataLayout(); 21030b57cec5SDimitry Andric Type *AllocaType = AI->getAllocatedType(); 2104e8d8bef9SDimitry Andric TypeSize AllocaTypeSize = DL.getTypeAllocSize(AllocaType); 21050b57cec5SDimitry Andric uint64_t AllocaArraySize = AIArraySize->getLimitedValue(); 21060b57cec5SDimitry Andric 21070b57cec5SDimitry Andric // Don't add markers for zero-sized allocas. 21080b57cec5SDimitry Andric if (AllocaArraySize == 0) 21090b57cec5SDimitry Andric continue; 21100b57cec5SDimitry Andric 21110b57cec5SDimitry Andric // Check that array size doesn't saturate uint64_t and doesn't 21120b57cec5SDimitry Andric // overflow when it's multiplied by type size. 2113e8d8bef9SDimitry Andric if (!AllocaTypeSize.isScalable() && 2114e8d8bef9SDimitry Andric AllocaArraySize != std::numeric_limits<uint64_t>::max() && 21150b57cec5SDimitry Andric std::numeric_limits<uint64_t>::max() / AllocaArraySize >= 2116e8d8bef9SDimitry Andric AllocaTypeSize.getFixedSize()) { 21170b57cec5SDimitry Andric AllocaSize = ConstantInt::get(Type::getInt64Ty(AI->getContext()), 21180b57cec5SDimitry Andric AllocaArraySize * AllocaTypeSize); 21190b57cec5SDimitry Andric } 21200b57cec5SDimitry Andric } 21210b57cec5SDimitry Andric 21220b57cec5SDimitry Andric builder.CreateLifetimeStart(AI, AllocaSize); 21230b57cec5SDimitry Andric for (ReturnInst *RI : Returns) { 21240b57cec5SDimitry Andric // Don't insert llvm.lifetime.end calls between a musttail or deoptimize 21250b57cec5SDimitry Andric // call and a return. The return kills all local allocas. 21260b57cec5SDimitry Andric if (InlinedMustTailCalls && 21270b57cec5SDimitry Andric RI->getParent()->getTerminatingMustTailCall()) 21280b57cec5SDimitry Andric continue; 21290b57cec5SDimitry Andric if (InlinedDeoptimizeCalls && 21300b57cec5SDimitry Andric RI->getParent()->getTerminatingDeoptimizeCall()) 21310b57cec5SDimitry Andric continue; 21320b57cec5SDimitry Andric IRBuilder<>(RI).CreateLifetimeEnd(AI, AllocaSize); 21330b57cec5SDimitry Andric } 21340b57cec5SDimitry Andric } 21350b57cec5SDimitry Andric } 21360b57cec5SDimitry Andric 21370b57cec5SDimitry Andric // If the inlined code contained dynamic alloca instructions, wrap the inlined 21380b57cec5SDimitry Andric // code with llvm.stacksave/llvm.stackrestore intrinsics. 21390b57cec5SDimitry Andric if (InlinedFunctionInfo.ContainsDynamicAllocas) { 21400b57cec5SDimitry Andric Module *M = Caller->getParent(); 21410b57cec5SDimitry Andric // Get the two intrinsics we care about. 21420b57cec5SDimitry Andric Function *StackSave = Intrinsic::getDeclaration(M, Intrinsic::stacksave); 21430b57cec5SDimitry Andric Function *StackRestore=Intrinsic::getDeclaration(M,Intrinsic::stackrestore); 21440b57cec5SDimitry Andric 21450b57cec5SDimitry Andric // Insert the llvm.stacksave. 21460b57cec5SDimitry Andric CallInst *SavedPtr = IRBuilder<>(&*FirstNewBlock, FirstNewBlock->begin()) 21470b57cec5SDimitry Andric .CreateCall(StackSave, {}, "savedstack"); 21480b57cec5SDimitry Andric 21490b57cec5SDimitry Andric // Insert a call to llvm.stackrestore before any return instructions in the 21500b57cec5SDimitry Andric // inlined function. 21510b57cec5SDimitry Andric for (ReturnInst *RI : Returns) { 21520b57cec5SDimitry Andric // Don't insert llvm.stackrestore calls between a musttail or deoptimize 21530b57cec5SDimitry Andric // call and a return. The return will restore the stack pointer. 21540b57cec5SDimitry Andric if (InlinedMustTailCalls && RI->getParent()->getTerminatingMustTailCall()) 21550b57cec5SDimitry Andric continue; 21560b57cec5SDimitry Andric if (InlinedDeoptimizeCalls && RI->getParent()->getTerminatingDeoptimizeCall()) 21570b57cec5SDimitry Andric continue; 21580b57cec5SDimitry Andric IRBuilder<>(RI).CreateCall(StackRestore, SavedPtr); 21590b57cec5SDimitry Andric } 21600b57cec5SDimitry Andric } 21610b57cec5SDimitry Andric 21620b57cec5SDimitry Andric // If we are inlining for an invoke instruction, we must make sure to rewrite 21630b57cec5SDimitry Andric // any call instructions into invoke instructions. This is sensitive to which 21640b57cec5SDimitry Andric // funclet pads were top-level in the inlinee, so must be done before 21650b57cec5SDimitry Andric // rewriting the "parent pad" links. 21665ffd83dbSDimitry Andric if (auto *II = dyn_cast<InvokeInst>(&CB)) { 21670b57cec5SDimitry Andric BasicBlock *UnwindDest = II->getUnwindDest(); 21680b57cec5SDimitry Andric Instruction *FirstNonPHI = UnwindDest->getFirstNonPHI(); 21690b57cec5SDimitry Andric if (isa<LandingPadInst>(FirstNonPHI)) { 21700b57cec5SDimitry Andric HandleInlinedLandingPad(II, &*FirstNewBlock, InlinedFunctionInfo); 21710b57cec5SDimitry Andric } else { 21720b57cec5SDimitry Andric HandleInlinedEHPad(II, &*FirstNewBlock, InlinedFunctionInfo); 21730b57cec5SDimitry Andric } 21740b57cec5SDimitry Andric } 21750b57cec5SDimitry Andric 21760b57cec5SDimitry Andric // Update the lexical scopes of the new funclets and callsites. 21770b57cec5SDimitry Andric // Anything that had 'none' as its parent is now nested inside the callsite's 21780b57cec5SDimitry Andric // EHPad. 21790b57cec5SDimitry Andric 21800b57cec5SDimitry Andric if (CallSiteEHPad) { 21810b57cec5SDimitry Andric for (Function::iterator BB = FirstNewBlock->getIterator(), 21820b57cec5SDimitry Andric E = Caller->end(); 21830b57cec5SDimitry Andric BB != E; ++BB) { 21840b57cec5SDimitry Andric // Add bundle operands to any top-level call sites. 21850b57cec5SDimitry Andric SmallVector<OperandBundleDef, 1> OpBundles; 21860b57cec5SDimitry Andric for (BasicBlock::iterator BBI = BB->begin(), E = BB->end(); BBI != E;) { 21875ffd83dbSDimitry Andric CallBase *I = dyn_cast<CallBase>(&*BBI++); 21885ffd83dbSDimitry Andric if (!I) 21890b57cec5SDimitry Andric continue; 21900b57cec5SDimitry Andric 21910b57cec5SDimitry Andric // Skip call sites which are nounwind intrinsics. 21920b57cec5SDimitry Andric auto *CalledFn = 21935ffd83dbSDimitry Andric dyn_cast<Function>(I->getCalledOperand()->stripPointerCasts()); 21945ffd83dbSDimitry Andric if (CalledFn && CalledFn->isIntrinsic() && I->doesNotThrow()) 21950b57cec5SDimitry Andric continue; 21960b57cec5SDimitry Andric 21970b57cec5SDimitry Andric // Skip call sites which already have a "funclet" bundle. 21985ffd83dbSDimitry Andric if (I->getOperandBundle(LLVMContext::OB_funclet)) 21990b57cec5SDimitry Andric continue; 22000b57cec5SDimitry Andric 22015ffd83dbSDimitry Andric I->getOperandBundlesAsDefs(OpBundles); 22020b57cec5SDimitry Andric OpBundles.emplace_back("funclet", CallSiteEHPad); 22030b57cec5SDimitry Andric 22045ffd83dbSDimitry Andric Instruction *NewInst = CallBase::Create(I, OpBundles, I); 22050b57cec5SDimitry Andric NewInst->takeName(I); 22060b57cec5SDimitry Andric I->replaceAllUsesWith(NewInst); 22070b57cec5SDimitry Andric I->eraseFromParent(); 22080b57cec5SDimitry Andric 22090b57cec5SDimitry Andric OpBundles.clear(); 22100b57cec5SDimitry Andric } 22110b57cec5SDimitry Andric 22120b57cec5SDimitry Andric // It is problematic if the inlinee has a cleanupret which unwinds to 22130b57cec5SDimitry Andric // caller and we inline it into a call site which doesn't unwind but into 22140b57cec5SDimitry Andric // an EH pad that does. Such an edge must be dynamically unreachable. 22150b57cec5SDimitry Andric // As such, we replace the cleanupret with unreachable. 22160b57cec5SDimitry Andric if (auto *CleanupRet = dyn_cast<CleanupReturnInst>(BB->getTerminator())) 22170b57cec5SDimitry Andric if (CleanupRet->unwindsToCaller() && EHPadForCallUnwindsLocally) 22180b57cec5SDimitry Andric changeToUnreachable(CleanupRet, /*UseLLVMTrap=*/false); 22190b57cec5SDimitry Andric 22200b57cec5SDimitry Andric Instruction *I = BB->getFirstNonPHI(); 22210b57cec5SDimitry Andric if (!I->isEHPad()) 22220b57cec5SDimitry Andric continue; 22230b57cec5SDimitry Andric 22240b57cec5SDimitry Andric if (auto *CatchSwitch = dyn_cast<CatchSwitchInst>(I)) { 22250b57cec5SDimitry Andric if (isa<ConstantTokenNone>(CatchSwitch->getParentPad())) 22260b57cec5SDimitry Andric CatchSwitch->setParentPad(CallSiteEHPad); 22270b57cec5SDimitry Andric } else { 22280b57cec5SDimitry Andric auto *FPI = cast<FuncletPadInst>(I); 22290b57cec5SDimitry Andric if (isa<ConstantTokenNone>(FPI->getParentPad())) 22300b57cec5SDimitry Andric FPI->setParentPad(CallSiteEHPad); 22310b57cec5SDimitry Andric } 22320b57cec5SDimitry Andric } 22330b57cec5SDimitry Andric } 22340b57cec5SDimitry Andric 22350b57cec5SDimitry Andric if (InlinedDeoptimizeCalls) { 22360b57cec5SDimitry Andric // We need to at least remove the deoptimizing returns from the Return set, 22370b57cec5SDimitry Andric // so that the control flow from those returns does not get merged into the 22380b57cec5SDimitry Andric // caller (but terminate it instead). If the caller's return type does not 22390b57cec5SDimitry Andric // match the callee's return type, we also need to change the return type of 22400b57cec5SDimitry Andric // the intrinsic. 22415ffd83dbSDimitry Andric if (Caller->getReturnType() == CB.getType()) { 2242e8d8bef9SDimitry Andric llvm::erase_if(Returns, [](ReturnInst *RI) { 22430b57cec5SDimitry Andric return RI->getParent()->getTerminatingDeoptimizeCall() != nullptr; 22440b57cec5SDimitry Andric }); 22450b57cec5SDimitry Andric } else { 22460b57cec5SDimitry Andric SmallVector<ReturnInst *, 8> NormalReturns; 22470b57cec5SDimitry Andric Function *NewDeoptIntrinsic = Intrinsic::getDeclaration( 22480b57cec5SDimitry Andric Caller->getParent(), Intrinsic::experimental_deoptimize, 22490b57cec5SDimitry Andric {Caller->getReturnType()}); 22500b57cec5SDimitry Andric 22510b57cec5SDimitry Andric for (ReturnInst *RI : Returns) { 22520b57cec5SDimitry Andric CallInst *DeoptCall = RI->getParent()->getTerminatingDeoptimizeCall(); 22530b57cec5SDimitry Andric if (!DeoptCall) { 22540b57cec5SDimitry Andric NormalReturns.push_back(RI); 22550b57cec5SDimitry Andric continue; 22560b57cec5SDimitry Andric } 22570b57cec5SDimitry Andric 22580b57cec5SDimitry Andric // The calling convention on the deoptimize call itself may be bogus, 22590b57cec5SDimitry Andric // since the code we're inlining may have undefined behavior (and may 22600b57cec5SDimitry Andric // never actually execute at runtime); but all 22610b57cec5SDimitry Andric // @llvm.experimental.deoptimize declarations have to have the same 22620b57cec5SDimitry Andric // calling convention in a well-formed module. 22630b57cec5SDimitry Andric auto CallingConv = DeoptCall->getCalledFunction()->getCallingConv(); 22640b57cec5SDimitry Andric NewDeoptIntrinsic->setCallingConv(CallingConv); 22650b57cec5SDimitry Andric auto *CurBB = RI->getParent(); 22660b57cec5SDimitry Andric RI->eraseFromParent(); 22670b57cec5SDimitry Andric 2268e8d8bef9SDimitry Andric SmallVector<Value *, 4> CallArgs(DeoptCall->args()); 22690b57cec5SDimitry Andric 22700b57cec5SDimitry Andric SmallVector<OperandBundleDef, 1> OpBundles; 22710b57cec5SDimitry Andric DeoptCall->getOperandBundlesAsDefs(OpBundles); 22720b57cec5SDimitry Andric DeoptCall->eraseFromParent(); 22730b57cec5SDimitry Andric assert(!OpBundles.empty() && 22740b57cec5SDimitry Andric "Expected at least the deopt operand bundle"); 22750b57cec5SDimitry Andric 22760b57cec5SDimitry Andric IRBuilder<> Builder(CurBB); 22770b57cec5SDimitry Andric CallInst *NewDeoptCall = 22780b57cec5SDimitry Andric Builder.CreateCall(NewDeoptIntrinsic, CallArgs, OpBundles); 22790b57cec5SDimitry Andric NewDeoptCall->setCallingConv(CallingConv); 22800b57cec5SDimitry Andric if (NewDeoptCall->getType()->isVoidTy()) 22810b57cec5SDimitry Andric Builder.CreateRetVoid(); 22820b57cec5SDimitry Andric else 22830b57cec5SDimitry Andric Builder.CreateRet(NewDeoptCall); 22840b57cec5SDimitry Andric } 22850b57cec5SDimitry Andric 22860b57cec5SDimitry Andric // Leave behind the normal returns so we can merge control flow. 22870b57cec5SDimitry Andric std::swap(Returns, NormalReturns); 22880b57cec5SDimitry Andric } 22890b57cec5SDimitry Andric } 22900b57cec5SDimitry Andric 22910b57cec5SDimitry Andric // Handle any inlined musttail call sites. In order for a new call site to be 22920b57cec5SDimitry Andric // musttail, the source of the clone and the inlined call site must have been 22930b57cec5SDimitry Andric // musttail. Therefore it's safe to return without merging control into the 22940b57cec5SDimitry Andric // phi below. 22950b57cec5SDimitry Andric if (InlinedMustTailCalls) { 22960b57cec5SDimitry Andric // Check if we need to bitcast the result of any musttail calls. 22970b57cec5SDimitry Andric Type *NewRetTy = Caller->getReturnType(); 22985ffd83dbSDimitry Andric bool NeedBitCast = !CB.use_empty() && CB.getType() != NewRetTy; 22990b57cec5SDimitry Andric 23000b57cec5SDimitry Andric // Handle the returns preceded by musttail calls separately. 23010b57cec5SDimitry Andric SmallVector<ReturnInst *, 8> NormalReturns; 23020b57cec5SDimitry Andric for (ReturnInst *RI : Returns) { 23030b57cec5SDimitry Andric CallInst *ReturnedMustTail = 23040b57cec5SDimitry Andric RI->getParent()->getTerminatingMustTailCall(); 23050b57cec5SDimitry Andric if (!ReturnedMustTail) { 23060b57cec5SDimitry Andric NormalReturns.push_back(RI); 23070b57cec5SDimitry Andric continue; 23080b57cec5SDimitry Andric } 23090b57cec5SDimitry Andric if (!NeedBitCast) 23100b57cec5SDimitry Andric continue; 23110b57cec5SDimitry Andric 23120b57cec5SDimitry Andric // Delete the old return and any preceding bitcast. 23130b57cec5SDimitry Andric BasicBlock *CurBB = RI->getParent(); 23140b57cec5SDimitry Andric auto *OldCast = dyn_cast_or_null<BitCastInst>(RI->getReturnValue()); 23150b57cec5SDimitry Andric RI->eraseFromParent(); 23160b57cec5SDimitry Andric if (OldCast) 23170b57cec5SDimitry Andric OldCast->eraseFromParent(); 23180b57cec5SDimitry Andric 23190b57cec5SDimitry Andric // Insert a new bitcast and return with the right type. 23200b57cec5SDimitry Andric IRBuilder<> Builder(CurBB); 23210b57cec5SDimitry Andric Builder.CreateRet(Builder.CreateBitCast(ReturnedMustTail, NewRetTy)); 23220b57cec5SDimitry Andric } 23230b57cec5SDimitry Andric 23240b57cec5SDimitry Andric // Leave behind the normal returns so we can merge control flow. 23250b57cec5SDimitry Andric std::swap(Returns, NormalReturns); 23260b57cec5SDimitry Andric } 23270b57cec5SDimitry Andric 23280b57cec5SDimitry Andric // Now that all of the transforms on the inlined code have taken place but 23290b57cec5SDimitry Andric // before we splice the inlined code into the CFG and lose track of which 23300b57cec5SDimitry Andric // blocks were actually inlined, collect the call sites. We only do this if 23310b57cec5SDimitry Andric // call graph updates weren't requested, as those provide value handle based 23320b57cec5SDimitry Andric // tracking of inlined call sites instead. 23330b57cec5SDimitry Andric if (InlinedFunctionInfo.ContainsCalls && !IFI.CG) { 23340b57cec5SDimitry Andric // Otherwise just collect the raw call sites that were inlined. 23350b57cec5SDimitry Andric for (BasicBlock &NewBB : 23360b57cec5SDimitry Andric make_range(FirstNewBlock->getIterator(), Caller->end())) 23370b57cec5SDimitry Andric for (Instruction &I : NewBB) 23385ffd83dbSDimitry Andric if (auto *CB = dyn_cast<CallBase>(&I)) 23395ffd83dbSDimitry Andric IFI.InlinedCallSites.push_back(CB); 23400b57cec5SDimitry Andric } 23410b57cec5SDimitry Andric 23420b57cec5SDimitry Andric // If we cloned in _exactly one_ basic block, and if that block ends in a 23430b57cec5SDimitry Andric // return instruction, we splice the body of the inlined callee directly into 23440b57cec5SDimitry Andric // the calling basic block. 23450b57cec5SDimitry Andric if (Returns.size() == 1 && std::distance(FirstNewBlock, Caller->end()) == 1) { 23460b57cec5SDimitry Andric // Move all of the instructions right before the call. 23475ffd83dbSDimitry Andric OrigBB->getInstList().splice(CB.getIterator(), FirstNewBlock->getInstList(), 23480b57cec5SDimitry Andric FirstNewBlock->begin(), FirstNewBlock->end()); 23490b57cec5SDimitry Andric // Remove the cloned basic block. 23500b57cec5SDimitry Andric Caller->getBasicBlockList().pop_back(); 23510b57cec5SDimitry Andric 23520b57cec5SDimitry Andric // If the call site was an invoke instruction, add a branch to the normal 23530b57cec5SDimitry Andric // destination. 23545ffd83dbSDimitry Andric if (InvokeInst *II = dyn_cast<InvokeInst>(&CB)) { 23555ffd83dbSDimitry Andric BranchInst *NewBr = BranchInst::Create(II->getNormalDest(), &CB); 23560b57cec5SDimitry Andric NewBr->setDebugLoc(Returns[0]->getDebugLoc()); 23570b57cec5SDimitry Andric } 23580b57cec5SDimitry Andric 23590b57cec5SDimitry Andric // If the return instruction returned a value, replace uses of the call with 23600b57cec5SDimitry Andric // uses of the returned value. 23615ffd83dbSDimitry Andric if (!CB.use_empty()) { 23620b57cec5SDimitry Andric ReturnInst *R = Returns[0]; 23635ffd83dbSDimitry Andric if (&CB == R->getReturnValue()) 23645ffd83dbSDimitry Andric CB.replaceAllUsesWith(UndefValue::get(CB.getType())); 23650b57cec5SDimitry Andric else 23665ffd83dbSDimitry Andric CB.replaceAllUsesWith(R->getReturnValue()); 23670b57cec5SDimitry Andric } 23680b57cec5SDimitry Andric // Since we are now done with the Call/Invoke, we can delete it. 23695ffd83dbSDimitry Andric CB.eraseFromParent(); 23700b57cec5SDimitry Andric 23710b57cec5SDimitry Andric // Since we are now done with the return instruction, delete it also. 23720b57cec5SDimitry Andric Returns[0]->eraseFromParent(); 23730b57cec5SDimitry Andric 23740b57cec5SDimitry Andric // We are now done with the inlining. 23755ffd83dbSDimitry Andric return InlineResult::success(); 23760b57cec5SDimitry Andric } 23770b57cec5SDimitry Andric 23780b57cec5SDimitry Andric // Otherwise, we have the normal case, of more than one block to inline or 23790b57cec5SDimitry Andric // multiple return sites. 23800b57cec5SDimitry Andric 23810b57cec5SDimitry Andric // We want to clone the entire callee function into the hole between the 23820b57cec5SDimitry Andric // "starter" and "ender" blocks. How we accomplish this depends on whether 23830b57cec5SDimitry Andric // this is an invoke instruction or a call instruction. 23840b57cec5SDimitry Andric BasicBlock *AfterCallBB; 23850b57cec5SDimitry Andric BranchInst *CreatedBranchToNormalDest = nullptr; 23865ffd83dbSDimitry Andric if (InvokeInst *II = dyn_cast<InvokeInst>(&CB)) { 23870b57cec5SDimitry Andric 23880b57cec5SDimitry Andric // Add an unconditional branch to make this look like the CallInst case... 23895ffd83dbSDimitry Andric CreatedBranchToNormalDest = BranchInst::Create(II->getNormalDest(), &CB); 23900b57cec5SDimitry Andric 23910b57cec5SDimitry Andric // Split the basic block. This guarantees that no PHI nodes will have to be 23920b57cec5SDimitry Andric // updated due to new incoming edges, and make the invoke case more 23930b57cec5SDimitry Andric // symmetric to the call case. 23940b57cec5SDimitry Andric AfterCallBB = 23950b57cec5SDimitry Andric OrigBB->splitBasicBlock(CreatedBranchToNormalDest->getIterator(), 23960b57cec5SDimitry Andric CalledFunc->getName() + ".exit"); 23970b57cec5SDimitry Andric 23980b57cec5SDimitry Andric } else { // It's a call 23990b57cec5SDimitry Andric // If this is a call instruction, we need to split the basic block that 24000b57cec5SDimitry Andric // the call lives in. 24010b57cec5SDimitry Andric // 24025ffd83dbSDimitry Andric AfterCallBB = OrigBB->splitBasicBlock(CB.getIterator(), 24030b57cec5SDimitry Andric CalledFunc->getName() + ".exit"); 24040b57cec5SDimitry Andric } 24050b57cec5SDimitry Andric 24060b57cec5SDimitry Andric if (IFI.CallerBFI) { 24070b57cec5SDimitry Andric // Copy original BB's block frequency to AfterCallBB 24080b57cec5SDimitry Andric IFI.CallerBFI->setBlockFreq( 24090b57cec5SDimitry Andric AfterCallBB, IFI.CallerBFI->getBlockFreq(OrigBB).getFrequency()); 24100b57cec5SDimitry Andric } 24110b57cec5SDimitry Andric 24120b57cec5SDimitry Andric // Change the branch that used to go to AfterCallBB to branch to the first 24130b57cec5SDimitry Andric // basic block of the inlined function. 24140b57cec5SDimitry Andric // 24150b57cec5SDimitry Andric Instruction *Br = OrigBB->getTerminator(); 24160b57cec5SDimitry Andric assert(Br && Br->getOpcode() == Instruction::Br && 24170b57cec5SDimitry Andric "splitBasicBlock broken!"); 24180b57cec5SDimitry Andric Br->setOperand(0, &*FirstNewBlock); 24190b57cec5SDimitry Andric 24200b57cec5SDimitry Andric // Now that the function is correct, make it a little bit nicer. In 24210b57cec5SDimitry Andric // particular, move the basic blocks inserted from the end of the function 24220b57cec5SDimitry Andric // into the space made by splitting the source basic block. 24230b57cec5SDimitry Andric Caller->getBasicBlockList().splice(AfterCallBB->getIterator(), 24240b57cec5SDimitry Andric Caller->getBasicBlockList(), FirstNewBlock, 24250b57cec5SDimitry Andric Caller->end()); 24260b57cec5SDimitry Andric 24270b57cec5SDimitry Andric // Handle all of the return instructions that we just cloned in, and eliminate 24280b57cec5SDimitry Andric // any users of the original call/invoke instruction. 24290b57cec5SDimitry Andric Type *RTy = CalledFunc->getReturnType(); 24300b57cec5SDimitry Andric 24310b57cec5SDimitry Andric PHINode *PHI = nullptr; 24320b57cec5SDimitry Andric if (Returns.size() > 1) { 24330b57cec5SDimitry Andric // The PHI node should go at the front of the new basic block to merge all 24340b57cec5SDimitry Andric // possible incoming values. 24355ffd83dbSDimitry Andric if (!CB.use_empty()) { 24365ffd83dbSDimitry Andric PHI = PHINode::Create(RTy, Returns.size(), CB.getName(), 24370b57cec5SDimitry Andric &AfterCallBB->front()); 24380b57cec5SDimitry Andric // Anything that used the result of the function call should now use the 24390b57cec5SDimitry Andric // PHI node as their operand. 24405ffd83dbSDimitry Andric CB.replaceAllUsesWith(PHI); 24410b57cec5SDimitry Andric } 24420b57cec5SDimitry Andric 24430b57cec5SDimitry Andric // Loop over all of the return instructions adding entries to the PHI node 24440b57cec5SDimitry Andric // as appropriate. 24450b57cec5SDimitry Andric if (PHI) { 24460b57cec5SDimitry Andric for (unsigned i = 0, e = Returns.size(); i != e; ++i) { 24470b57cec5SDimitry Andric ReturnInst *RI = Returns[i]; 24480b57cec5SDimitry Andric assert(RI->getReturnValue()->getType() == PHI->getType() && 24490b57cec5SDimitry Andric "Ret value not consistent in function!"); 24500b57cec5SDimitry Andric PHI->addIncoming(RI->getReturnValue(), RI->getParent()); 24510b57cec5SDimitry Andric } 24520b57cec5SDimitry Andric } 24530b57cec5SDimitry Andric 24540b57cec5SDimitry Andric // Add a branch to the merge points and remove return instructions. 24550b57cec5SDimitry Andric DebugLoc Loc; 24560b57cec5SDimitry Andric for (unsigned i = 0, e = Returns.size(); i != e; ++i) { 24570b57cec5SDimitry Andric ReturnInst *RI = Returns[i]; 24580b57cec5SDimitry Andric BranchInst* BI = BranchInst::Create(AfterCallBB, RI); 24590b57cec5SDimitry Andric Loc = RI->getDebugLoc(); 24600b57cec5SDimitry Andric BI->setDebugLoc(Loc); 24610b57cec5SDimitry Andric RI->eraseFromParent(); 24620b57cec5SDimitry Andric } 24630b57cec5SDimitry Andric // We need to set the debug location to *somewhere* inside the 24640b57cec5SDimitry Andric // inlined function. The line number may be nonsensical, but the 24650b57cec5SDimitry Andric // instruction will at least be associated with the right 24660b57cec5SDimitry Andric // function. 24670b57cec5SDimitry Andric if (CreatedBranchToNormalDest) 24680b57cec5SDimitry Andric CreatedBranchToNormalDest->setDebugLoc(Loc); 24690b57cec5SDimitry Andric } else if (!Returns.empty()) { 24700b57cec5SDimitry Andric // Otherwise, if there is exactly one return value, just replace anything 24710b57cec5SDimitry Andric // using the return value of the call with the computed value. 24725ffd83dbSDimitry Andric if (!CB.use_empty()) { 24735ffd83dbSDimitry Andric if (&CB == Returns[0]->getReturnValue()) 24745ffd83dbSDimitry Andric CB.replaceAllUsesWith(UndefValue::get(CB.getType())); 24750b57cec5SDimitry Andric else 24765ffd83dbSDimitry Andric CB.replaceAllUsesWith(Returns[0]->getReturnValue()); 24770b57cec5SDimitry Andric } 24780b57cec5SDimitry Andric 24790b57cec5SDimitry Andric // Update PHI nodes that use the ReturnBB to use the AfterCallBB. 24800b57cec5SDimitry Andric BasicBlock *ReturnBB = Returns[0]->getParent(); 24810b57cec5SDimitry Andric ReturnBB->replaceAllUsesWith(AfterCallBB); 24820b57cec5SDimitry Andric 24830b57cec5SDimitry Andric // Splice the code from the return block into the block that it will return 24840b57cec5SDimitry Andric // to, which contains the code that was after the call. 24850b57cec5SDimitry Andric AfterCallBB->getInstList().splice(AfterCallBB->begin(), 24860b57cec5SDimitry Andric ReturnBB->getInstList()); 24870b57cec5SDimitry Andric 24880b57cec5SDimitry Andric if (CreatedBranchToNormalDest) 24890b57cec5SDimitry Andric CreatedBranchToNormalDest->setDebugLoc(Returns[0]->getDebugLoc()); 24900b57cec5SDimitry Andric 24910b57cec5SDimitry Andric // Delete the return instruction now and empty ReturnBB now. 24920b57cec5SDimitry Andric Returns[0]->eraseFromParent(); 24930b57cec5SDimitry Andric ReturnBB->eraseFromParent(); 24945ffd83dbSDimitry Andric } else if (!CB.use_empty()) { 24950b57cec5SDimitry Andric // No returns, but something is using the return value of the call. Just 24960b57cec5SDimitry Andric // nuke the result. 24975ffd83dbSDimitry Andric CB.replaceAllUsesWith(UndefValue::get(CB.getType())); 24980b57cec5SDimitry Andric } 24990b57cec5SDimitry Andric 25000b57cec5SDimitry Andric // Since we are now done with the Call/Invoke, we can delete it. 25015ffd83dbSDimitry Andric CB.eraseFromParent(); 25020b57cec5SDimitry Andric 25030b57cec5SDimitry Andric // If we inlined any musttail calls and the original return is now 25040b57cec5SDimitry Andric // unreachable, delete it. It can only contain a bitcast and ret. 2505e8d8bef9SDimitry Andric if (InlinedMustTailCalls && pred_empty(AfterCallBB)) 25060b57cec5SDimitry Andric AfterCallBB->eraseFromParent(); 25070b57cec5SDimitry Andric 25080b57cec5SDimitry Andric // We should always be able to fold the entry block of the function into the 25090b57cec5SDimitry Andric // single predecessor of the block... 25100b57cec5SDimitry Andric assert(cast<BranchInst>(Br)->isUnconditional() && "splitBasicBlock broken!"); 25110b57cec5SDimitry Andric BasicBlock *CalleeEntry = cast<BranchInst>(Br)->getSuccessor(0); 25120b57cec5SDimitry Andric 25130b57cec5SDimitry Andric // Splice the code entry block into calling block, right before the 25140b57cec5SDimitry Andric // unconditional branch. 25150b57cec5SDimitry Andric CalleeEntry->replaceAllUsesWith(OrigBB); // Update PHI nodes 25160b57cec5SDimitry Andric OrigBB->getInstList().splice(Br->getIterator(), CalleeEntry->getInstList()); 25170b57cec5SDimitry Andric 25180b57cec5SDimitry Andric // Remove the unconditional branch. 25190b57cec5SDimitry Andric OrigBB->getInstList().erase(Br); 25200b57cec5SDimitry Andric 25210b57cec5SDimitry Andric // Now we can remove the CalleeEntry block, which is now empty. 25220b57cec5SDimitry Andric Caller->getBasicBlockList().erase(CalleeEntry); 25230b57cec5SDimitry Andric 25240b57cec5SDimitry Andric // If we inserted a phi node, check to see if it has a single value (e.g. all 25250b57cec5SDimitry Andric // the entries are the same or undef). If so, remove the PHI so it doesn't 25260b57cec5SDimitry Andric // block other optimizations. 25270b57cec5SDimitry Andric if (PHI) { 25280b57cec5SDimitry Andric AssumptionCache *AC = 25295ffd83dbSDimitry Andric IFI.GetAssumptionCache ? &IFI.GetAssumptionCache(*Caller) : nullptr; 25300b57cec5SDimitry Andric auto &DL = Caller->getParent()->getDataLayout(); 25310b57cec5SDimitry Andric if (Value *V = SimplifyInstruction(PHI, {DL, nullptr, nullptr, AC})) { 25320b57cec5SDimitry Andric PHI->replaceAllUsesWith(V); 25330b57cec5SDimitry Andric PHI->eraseFromParent(); 25340b57cec5SDimitry Andric } 25350b57cec5SDimitry Andric } 25360b57cec5SDimitry Andric 25375ffd83dbSDimitry Andric return InlineResult::success(); 25380b57cec5SDimitry Andric } 2539