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"
30*5f7ddb14SDimitry Andric #include "llvm/Analysis/ObjCARCAnalysisUtils.h"
31*5f7ddb14SDimitry Andric #include "llvm/Analysis/ObjCARCUtil.h"
320b57cec5SDimitry Andric #include "llvm/Analysis/ProfileSummaryInfo.h"
330b57cec5SDimitry Andric #include "llvm/Analysis/ValueTracking.h"
340b57cec5SDimitry Andric #include "llvm/Analysis/VectorUtils.h"
350b57cec5SDimitry Andric #include "llvm/IR/Argument.h"
360b57cec5SDimitry Andric #include "llvm/IR/BasicBlock.h"
370b57cec5SDimitry Andric #include "llvm/IR/CFG.h"
380b57cec5SDimitry Andric #include "llvm/IR/Constant.h"
390b57cec5SDimitry Andric #include "llvm/IR/Constants.h"
400b57cec5SDimitry Andric #include "llvm/IR/DIBuilder.h"
410b57cec5SDimitry Andric #include "llvm/IR/DataLayout.h"
420b57cec5SDimitry Andric #include "llvm/IR/DebugInfoMetadata.h"
430b57cec5SDimitry Andric #include "llvm/IR/DebugLoc.h"
440b57cec5SDimitry Andric #include "llvm/IR/DerivedTypes.h"
450b57cec5SDimitry Andric #include "llvm/IR/Dominators.h"
460b57cec5SDimitry Andric #include "llvm/IR/Function.h"
470b57cec5SDimitry Andric #include "llvm/IR/IRBuilder.h"
48*5f7ddb14SDimitry Andric #include "llvm/IR/InlineAsm.h"
490b57cec5SDimitry Andric #include "llvm/IR/InstrTypes.h"
500b57cec5SDimitry Andric #include "llvm/IR/Instruction.h"
510b57cec5SDimitry Andric #include "llvm/IR/Instructions.h"
520b57cec5SDimitry Andric #include "llvm/IR/IntrinsicInst.h"
530b57cec5SDimitry Andric #include "llvm/IR/Intrinsics.h"
540b57cec5SDimitry Andric #include "llvm/IR/LLVMContext.h"
550b57cec5SDimitry Andric #include "llvm/IR/MDBuilder.h"
560b57cec5SDimitry Andric #include "llvm/IR/Metadata.h"
570b57cec5SDimitry Andric #include "llvm/IR/Module.h"
580b57cec5SDimitry Andric #include "llvm/IR/Type.h"
590b57cec5SDimitry Andric #include "llvm/IR/User.h"
600b57cec5SDimitry Andric #include "llvm/IR/Value.h"
610b57cec5SDimitry Andric #include "llvm/Support/Casting.h"
620b57cec5SDimitry Andric #include "llvm/Support/CommandLine.h"
630b57cec5SDimitry Andric #include "llvm/Support/ErrorHandling.h"
645ffd83dbSDimitry Andric #include "llvm/Transforms/Utils/AssumeBundleBuilder.h"
650b57cec5SDimitry Andric #include "llvm/Transforms/Utils/Cloning.h"
66*5f7ddb14SDimitry Andric #include "llvm/Transforms/Utils/Local.h"
670b57cec5SDimitry Andric #include "llvm/Transforms/Utils/ValueMapper.h"
680b57cec5SDimitry Andric #include <algorithm>
690b57cec5SDimitry Andric #include <cassert>
700b57cec5SDimitry Andric #include <cstdint>
710b57cec5SDimitry Andric #include <iterator>
720b57cec5SDimitry Andric #include <limits>
730b57cec5SDimitry Andric #include <string>
740b57cec5SDimitry Andric #include <utility>
750b57cec5SDimitry Andric #include <vector>
760b57cec5SDimitry Andric 
770b57cec5SDimitry Andric using namespace llvm;
780b57cec5SDimitry Andric using ProfileCount = Function::ProfileCount;
790b57cec5SDimitry Andric 
800b57cec5SDimitry Andric static cl::opt<bool>
810b57cec5SDimitry Andric EnableNoAliasConversion("enable-noalias-to-md-conversion", cl::init(true),
820b57cec5SDimitry Andric   cl::Hidden,
830b57cec5SDimitry Andric   cl::desc("Convert noalias attributes to metadata during inlining."));
840b57cec5SDimitry Andric 
85af732203SDimitry Andric static cl::opt<bool>
86af732203SDimitry Andric     UseNoAliasIntrinsic("use-noalias-intrinsic-during-inlining", cl::Hidden,
87af732203SDimitry Andric                         cl::ZeroOrMore, cl::init(true),
88af732203SDimitry Andric                         cl::desc("Use the llvm.experimental.noalias.scope.decl "
89af732203SDimitry Andric                                  "intrinsic during inlining."));
90af732203SDimitry Andric 
915ffd83dbSDimitry Andric // Disabled by default, because the added alignment assumptions may increase
925ffd83dbSDimitry Andric // compile-time and block optimizations. This option is not suitable for use
935ffd83dbSDimitry Andric // with frontends that emit comprehensive parameter alignment annotations.
940b57cec5SDimitry Andric static cl::opt<bool>
950b57cec5SDimitry Andric PreserveAlignmentAssumptions("preserve-alignment-assumptions-during-inlining",
965ffd83dbSDimitry Andric   cl::init(false), cl::Hidden,
970b57cec5SDimitry Andric   cl::desc("Convert align attributes to assumptions during inlining."));
980b57cec5SDimitry Andric 
995ffd83dbSDimitry Andric static cl::opt<bool> UpdateReturnAttributes(
1005ffd83dbSDimitry Andric         "update-return-attrs", cl::init(true), cl::Hidden,
1015ffd83dbSDimitry Andric             cl::desc("Update return attributes on calls within inlined body"));
1025ffd83dbSDimitry Andric 
1035ffd83dbSDimitry Andric static cl::opt<unsigned> InlinerAttributeWindow(
1045ffd83dbSDimitry Andric     "max-inst-checked-for-throw-during-inlining", cl::Hidden,
1055ffd83dbSDimitry Andric     cl::desc("the maximum number of instructions analyzed for may throw during "
1065ffd83dbSDimitry Andric              "attribute inference in inlined body"),
1075ffd83dbSDimitry Andric     cl::init(4));
1080b57cec5SDimitry Andric 
1090b57cec5SDimitry Andric namespace {
1100b57cec5SDimitry Andric 
1110b57cec5SDimitry Andric   /// A class for recording information about inlining a landing pad.
1120b57cec5SDimitry Andric   class LandingPadInliningInfo {
1130b57cec5SDimitry Andric     /// Destination of the invoke's unwind.
1140b57cec5SDimitry Andric     BasicBlock *OuterResumeDest;
1150b57cec5SDimitry Andric 
1160b57cec5SDimitry Andric     /// Destination for the callee's resume.
1170b57cec5SDimitry Andric     BasicBlock *InnerResumeDest = nullptr;
1180b57cec5SDimitry Andric 
1190b57cec5SDimitry Andric     /// LandingPadInst associated with the invoke.
1200b57cec5SDimitry Andric     LandingPadInst *CallerLPad = nullptr;
1210b57cec5SDimitry Andric 
1220b57cec5SDimitry Andric     /// PHI for EH values from landingpad insts.
1230b57cec5SDimitry Andric     PHINode *InnerEHValuesPHI = nullptr;
1240b57cec5SDimitry Andric 
1250b57cec5SDimitry Andric     SmallVector<Value*, 8> UnwindDestPHIValues;
1260b57cec5SDimitry Andric 
1270b57cec5SDimitry Andric   public:
LandingPadInliningInfo(InvokeInst * II)1280b57cec5SDimitry Andric     LandingPadInliningInfo(InvokeInst *II)
1290b57cec5SDimitry Andric         : OuterResumeDest(II->getUnwindDest()) {
1300b57cec5SDimitry Andric       // If there are PHI nodes in the unwind destination block, we need to keep
1310b57cec5SDimitry Andric       // track of which values came into them from the invoke before removing
1320b57cec5SDimitry Andric       // the edge from this block.
1330b57cec5SDimitry Andric       BasicBlock *InvokeBB = II->getParent();
1340b57cec5SDimitry Andric       BasicBlock::iterator I = OuterResumeDest->begin();
1350b57cec5SDimitry Andric       for (; isa<PHINode>(I); ++I) {
1360b57cec5SDimitry Andric         // Save the value to use for this edge.
1370b57cec5SDimitry Andric         PHINode *PHI = cast<PHINode>(I);
1380b57cec5SDimitry Andric         UnwindDestPHIValues.push_back(PHI->getIncomingValueForBlock(InvokeBB));
1390b57cec5SDimitry Andric       }
1400b57cec5SDimitry Andric 
1410b57cec5SDimitry Andric       CallerLPad = cast<LandingPadInst>(I);
1420b57cec5SDimitry Andric     }
1430b57cec5SDimitry Andric 
1440b57cec5SDimitry Andric     /// The outer unwind destination is the target of
1450b57cec5SDimitry Andric     /// unwind edges introduced for calls within the inlined function.
getOuterResumeDest() const1460b57cec5SDimitry Andric     BasicBlock *getOuterResumeDest() const {
1470b57cec5SDimitry Andric       return OuterResumeDest;
1480b57cec5SDimitry Andric     }
1490b57cec5SDimitry Andric 
1500b57cec5SDimitry Andric     BasicBlock *getInnerResumeDest();
1510b57cec5SDimitry Andric 
getLandingPadInst() const1520b57cec5SDimitry Andric     LandingPadInst *getLandingPadInst() const { return CallerLPad; }
1530b57cec5SDimitry Andric 
1540b57cec5SDimitry Andric     /// Forward the 'resume' instruction to the caller's landing pad block.
1550b57cec5SDimitry Andric     /// When the landing pad block has only one predecessor, this is
1560b57cec5SDimitry Andric     /// a simple branch. When there is more than one predecessor, we need to
1570b57cec5SDimitry Andric     /// split the landing pad block after the landingpad instruction and jump
1580b57cec5SDimitry Andric     /// to there.
1590b57cec5SDimitry Andric     void forwardResume(ResumeInst *RI,
1600b57cec5SDimitry Andric                        SmallPtrSetImpl<LandingPadInst*> &InlinedLPads);
1610b57cec5SDimitry Andric 
1620b57cec5SDimitry Andric     /// Add incoming-PHI values to the unwind destination block for the given
1630b57cec5SDimitry Andric     /// basic block, using the values for the original invoke's source block.
addIncomingPHIValuesFor(BasicBlock * BB) const1640b57cec5SDimitry Andric     void addIncomingPHIValuesFor(BasicBlock *BB) const {
1650b57cec5SDimitry Andric       addIncomingPHIValuesForInto(BB, OuterResumeDest);
1660b57cec5SDimitry Andric     }
1670b57cec5SDimitry Andric 
addIncomingPHIValuesForInto(BasicBlock * src,BasicBlock * dest) const1680b57cec5SDimitry Andric     void addIncomingPHIValuesForInto(BasicBlock *src, BasicBlock *dest) const {
1690b57cec5SDimitry Andric       BasicBlock::iterator I = dest->begin();
1700b57cec5SDimitry Andric       for (unsigned i = 0, e = UnwindDestPHIValues.size(); i != e; ++i, ++I) {
1710b57cec5SDimitry Andric         PHINode *phi = cast<PHINode>(I);
1720b57cec5SDimitry Andric         phi->addIncoming(UnwindDestPHIValues[i], src);
1730b57cec5SDimitry Andric       }
1740b57cec5SDimitry Andric     }
1750b57cec5SDimitry Andric   };
1760b57cec5SDimitry Andric 
1770b57cec5SDimitry Andric } // end anonymous namespace
1780b57cec5SDimitry Andric 
1790b57cec5SDimitry Andric /// Get or create a target for the branch from ResumeInsts.
getInnerResumeDest()1800b57cec5SDimitry Andric BasicBlock *LandingPadInliningInfo::getInnerResumeDest() {
1810b57cec5SDimitry Andric   if (InnerResumeDest) return InnerResumeDest;
1820b57cec5SDimitry Andric 
1830b57cec5SDimitry Andric   // Split the landing pad.
1840b57cec5SDimitry Andric   BasicBlock::iterator SplitPoint = ++CallerLPad->getIterator();
1850b57cec5SDimitry Andric   InnerResumeDest =
1860b57cec5SDimitry Andric     OuterResumeDest->splitBasicBlock(SplitPoint,
1870b57cec5SDimitry Andric                                      OuterResumeDest->getName() + ".body");
1880b57cec5SDimitry Andric 
1890b57cec5SDimitry Andric   // The number of incoming edges we expect to the inner landing pad.
1900b57cec5SDimitry Andric   const unsigned PHICapacity = 2;
1910b57cec5SDimitry Andric 
1920b57cec5SDimitry Andric   // Create corresponding new PHIs for all the PHIs in the outer landing pad.
1930b57cec5SDimitry Andric   Instruction *InsertPoint = &InnerResumeDest->front();
1940b57cec5SDimitry Andric   BasicBlock::iterator I = OuterResumeDest->begin();
1950b57cec5SDimitry Andric   for (unsigned i = 0, e = UnwindDestPHIValues.size(); i != e; ++i, ++I) {
1960b57cec5SDimitry Andric     PHINode *OuterPHI = cast<PHINode>(I);
1970b57cec5SDimitry Andric     PHINode *InnerPHI = PHINode::Create(OuterPHI->getType(), PHICapacity,
1980b57cec5SDimitry Andric                                         OuterPHI->getName() + ".lpad-body",
1990b57cec5SDimitry Andric                                         InsertPoint);
2000b57cec5SDimitry Andric     OuterPHI->replaceAllUsesWith(InnerPHI);
2010b57cec5SDimitry Andric     InnerPHI->addIncoming(OuterPHI, OuterResumeDest);
2020b57cec5SDimitry Andric   }
2030b57cec5SDimitry Andric 
2040b57cec5SDimitry Andric   // Create a PHI for the exception values.
2050b57cec5SDimitry Andric   InnerEHValuesPHI = PHINode::Create(CallerLPad->getType(), PHICapacity,
2060b57cec5SDimitry Andric                                      "eh.lpad-body", InsertPoint);
2070b57cec5SDimitry Andric   CallerLPad->replaceAllUsesWith(InnerEHValuesPHI);
2080b57cec5SDimitry Andric   InnerEHValuesPHI->addIncoming(CallerLPad, OuterResumeDest);
2090b57cec5SDimitry Andric 
2100b57cec5SDimitry Andric   // All done.
2110b57cec5SDimitry Andric   return InnerResumeDest;
2120b57cec5SDimitry Andric }
2130b57cec5SDimitry Andric 
2140b57cec5SDimitry Andric /// Forward the 'resume' instruction to the caller's landing pad block.
2150b57cec5SDimitry Andric /// When the landing pad block has only one predecessor, this is a simple
2160b57cec5SDimitry Andric /// branch. When there is more than one predecessor, we need to split the
2170b57cec5SDimitry Andric /// landing pad block after the landingpad instruction and jump to there.
forwardResume(ResumeInst * RI,SmallPtrSetImpl<LandingPadInst * > & InlinedLPads)2180b57cec5SDimitry Andric void LandingPadInliningInfo::forwardResume(
2190b57cec5SDimitry Andric     ResumeInst *RI, SmallPtrSetImpl<LandingPadInst *> &InlinedLPads) {
2200b57cec5SDimitry Andric   BasicBlock *Dest = getInnerResumeDest();
2210b57cec5SDimitry Andric   BasicBlock *Src = RI->getParent();
2220b57cec5SDimitry Andric 
2230b57cec5SDimitry Andric   BranchInst::Create(Dest, Src);
2240b57cec5SDimitry Andric 
2250b57cec5SDimitry Andric   // Update the PHIs in the destination. They were inserted in an order which
2260b57cec5SDimitry Andric   // makes this work.
2270b57cec5SDimitry Andric   addIncomingPHIValuesForInto(Src, Dest);
2280b57cec5SDimitry Andric 
2290b57cec5SDimitry Andric   InnerEHValuesPHI->addIncoming(RI->getOperand(0), Src);
2300b57cec5SDimitry Andric   RI->eraseFromParent();
2310b57cec5SDimitry Andric }
2320b57cec5SDimitry Andric 
2330b57cec5SDimitry Andric /// Helper for getUnwindDestToken/getUnwindDestTokenHelper.
getParentPad(Value * EHPad)2340b57cec5SDimitry Andric static Value *getParentPad(Value *EHPad) {
2350b57cec5SDimitry Andric   if (auto *FPI = dyn_cast<FuncletPadInst>(EHPad))
2360b57cec5SDimitry Andric     return FPI->getParentPad();
2370b57cec5SDimitry Andric   return cast<CatchSwitchInst>(EHPad)->getParentPad();
2380b57cec5SDimitry Andric }
2390b57cec5SDimitry Andric 
2400b57cec5SDimitry Andric using UnwindDestMemoTy = DenseMap<Instruction *, Value *>;
2410b57cec5SDimitry Andric 
2420b57cec5SDimitry Andric /// Helper for getUnwindDestToken that does the descendant-ward part of
2430b57cec5SDimitry Andric /// the search.
getUnwindDestTokenHelper(Instruction * EHPad,UnwindDestMemoTy & MemoMap)2440b57cec5SDimitry Andric static Value *getUnwindDestTokenHelper(Instruction *EHPad,
2450b57cec5SDimitry Andric                                        UnwindDestMemoTy &MemoMap) {
2460b57cec5SDimitry Andric   SmallVector<Instruction *, 8> Worklist(1, EHPad);
2470b57cec5SDimitry Andric 
2480b57cec5SDimitry Andric   while (!Worklist.empty()) {
2490b57cec5SDimitry Andric     Instruction *CurrentPad = Worklist.pop_back_val();
2500b57cec5SDimitry Andric     // We only put pads on the worklist that aren't in the MemoMap.  When
2510b57cec5SDimitry Andric     // we find an unwind dest for a pad we may update its ancestors, but
2520b57cec5SDimitry Andric     // the queue only ever contains uncles/great-uncles/etc. of CurrentPad,
2530b57cec5SDimitry Andric     // so they should never get updated while queued on the worklist.
2540b57cec5SDimitry Andric     assert(!MemoMap.count(CurrentPad));
2550b57cec5SDimitry Andric     Value *UnwindDestToken = nullptr;
2560b57cec5SDimitry Andric     if (auto *CatchSwitch = dyn_cast<CatchSwitchInst>(CurrentPad)) {
2570b57cec5SDimitry Andric       if (CatchSwitch->hasUnwindDest()) {
2580b57cec5SDimitry Andric         UnwindDestToken = CatchSwitch->getUnwindDest()->getFirstNonPHI();
2590b57cec5SDimitry Andric       } else {
2600b57cec5SDimitry Andric         // Catchswitch doesn't have a 'nounwind' variant, and one might be
2610b57cec5SDimitry Andric         // annotated as "unwinds to caller" when really it's nounwind (see
2620b57cec5SDimitry Andric         // e.g. SimplifyCFGOpt::SimplifyUnreachable), so we can't infer the
2630b57cec5SDimitry Andric         // parent's unwind dest from this.  We can check its catchpads'
2640b57cec5SDimitry Andric         // descendants, since they might include a cleanuppad with an
2650b57cec5SDimitry Andric         // "unwinds to caller" cleanupret, which can be trusted.
2660b57cec5SDimitry Andric         for (auto HI = CatchSwitch->handler_begin(),
2670b57cec5SDimitry Andric                   HE = CatchSwitch->handler_end();
2680b57cec5SDimitry Andric              HI != HE && !UnwindDestToken; ++HI) {
2690b57cec5SDimitry Andric           BasicBlock *HandlerBlock = *HI;
2700b57cec5SDimitry Andric           auto *CatchPad = cast<CatchPadInst>(HandlerBlock->getFirstNonPHI());
2710b57cec5SDimitry Andric           for (User *Child : CatchPad->users()) {
2720b57cec5SDimitry Andric             // Intentionally ignore invokes here -- since the catchswitch is
2730b57cec5SDimitry Andric             // marked "unwind to caller", it would be a verifier error if it
2740b57cec5SDimitry Andric             // contained an invoke which unwinds out of it, so any invoke we'd
2750b57cec5SDimitry Andric             // encounter must unwind to some child of the catch.
2760b57cec5SDimitry Andric             if (!isa<CleanupPadInst>(Child) && !isa<CatchSwitchInst>(Child))
2770b57cec5SDimitry Andric               continue;
2780b57cec5SDimitry Andric 
2790b57cec5SDimitry Andric             Instruction *ChildPad = cast<Instruction>(Child);
2800b57cec5SDimitry Andric             auto Memo = MemoMap.find(ChildPad);
2810b57cec5SDimitry Andric             if (Memo == MemoMap.end()) {
2820b57cec5SDimitry Andric               // Haven't figured out this child pad yet; queue it.
2830b57cec5SDimitry Andric               Worklist.push_back(ChildPad);
2840b57cec5SDimitry Andric               continue;
2850b57cec5SDimitry Andric             }
2860b57cec5SDimitry Andric             // We've already checked this child, but might have found that
2870b57cec5SDimitry Andric             // it offers no proof either way.
2880b57cec5SDimitry Andric             Value *ChildUnwindDestToken = Memo->second;
2890b57cec5SDimitry Andric             if (!ChildUnwindDestToken)
2900b57cec5SDimitry Andric               continue;
2910b57cec5SDimitry Andric             // We already know the child's unwind dest, which can either
2920b57cec5SDimitry Andric             // be ConstantTokenNone to indicate unwind to caller, or can
2930b57cec5SDimitry Andric             // be another child of the catchpad.  Only the former indicates
2940b57cec5SDimitry Andric             // the unwind dest of the catchswitch.
2950b57cec5SDimitry Andric             if (isa<ConstantTokenNone>(ChildUnwindDestToken)) {
2960b57cec5SDimitry Andric               UnwindDestToken = ChildUnwindDestToken;
2970b57cec5SDimitry Andric               break;
2980b57cec5SDimitry Andric             }
2990b57cec5SDimitry Andric             assert(getParentPad(ChildUnwindDestToken) == CatchPad);
3000b57cec5SDimitry Andric           }
3010b57cec5SDimitry Andric         }
3020b57cec5SDimitry Andric       }
3030b57cec5SDimitry Andric     } else {
3040b57cec5SDimitry Andric       auto *CleanupPad = cast<CleanupPadInst>(CurrentPad);
3050b57cec5SDimitry Andric       for (User *U : CleanupPad->users()) {
3060b57cec5SDimitry Andric         if (auto *CleanupRet = dyn_cast<CleanupReturnInst>(U)) {
3070b57cec5SDimitry Andric           if (BasicBlock *RetUnwindDest = CleanupRet->getUnwindDest())
3080b57cec5SDimitry Andric             UnwindDestToken = RetUnwindDest->getFirstNonPHI();
3090b57cec5SDimitry Andric           else
3100b57cec5SDimitry Andric             UnwindDestToken = ConstantTokenNone::get(CleanupPad->getContext());
3110b57cec5SDimitry Andric           break;
3120b57cec5SDimitry Andric         }
3130b57cec5SDimitry Andric         Value *ChildUnwindDestToken;
3140b57cec5SDimitry Andric         if (auto *Invoke = dyn_cast<InvokeInst>(U)) {
3150b57cec5SDimitry Andric           ChildUnwindDestToken = Invoke->getUnwindDest()->getFirstNonPHI();
3160b57cec5SDimitry Andric         } else if (isa<CleanupPadInst>(U) || isa<CatchSwitchInst>(U)) {
3170b57cec5SDimitry Andric           Instruction *ChildPad = cast<Instruction>(U);
3180b57cec5SDimitry Andric           auto Memo = MemoMap.find(ChildPad);
3190b57cec5SDimitry Andric           if (Memo == MemoMap.end()) {
3200b57cec5SDimitry Andric             // Haven't resolved this child yet; queue it and keep searching.
3210b57cec5SDimitry Andric             Worklist.push_back(ChildPad);
3220b57cec5SDimitry Andric             continue;
3230b57cec5SDimitry Andric           }
3240b57cec5SDimitry Andric           // We've checked this child, but still need to ignore it if it
3250b57cec5SDimitry Andric           // had no proof either way.
3260b57cec5SDimitry Andric           ChildUnwindDestToken = Memo->second;
3270b57cec5SDimitry Andric           if (!ChildUnwindDestToken)
3280b57cec5SDimitry Andric             continue;
3290b57cec5SDimitry Andric         } else {
3300b57cec5SDimitry Andric           // Not a relevant user of the cleanuppad
3310b57cec5SDimitry Andric           continue;
3320b57cec5SDimitry Andric         }
3330b57cec5SDimitry Andric         // In a well-formed program, the child/invoke must either unwind to
3340b57cec5SDimitry Andric         // an(other) child of the cleanup, or exit the cleanup.  In the
3350b57cec5SDimitry Andric         // first case, continue searching.
3360b57cec5SDimitry Andric         if (isa<Instruction>(ChildUnwindDestToken) &&
3370b57cec5SDimitry Andric             getParentPad(ChildUnwindDestToken) == CleanupPad)
3380b57cec5SDimitry Andric           continue;
3390b57cec5SDimitry Andric         UnwindDestToken = ChildUnwindDestToken;
3400b57cec5SDimitry Andric         break;
3410b57cec5SDimitry Andric       }
3420b57cec5SDimitry Andric     }
3430b57cec5SDimitry Andric     // If we haven't found an unwind dest for CurrentPad, we may have queued its
3440b57cec5SDimitry Andric     // children, so move on to the next in the worklist.
3450b57cec5SDimitry Andric     if (!UnwindDestToken)
3460b57cec5SDimitry Andric       continue;
3470b57cec5SDimitry Andric 
3480b57cec5SDimitry Andric     // Now we know that CurrentPad unwinds to UnwindDestToken.  It also exits
3490b57cec5SDimitry Andric     // any ancestors of CurrentPad up to but not including UnwindDestToken's
3500b57cec5SDimitry Andric     // parent pad.  Record this in the memo map, and check to see if the
3510b57cec5SDimitry Andric     // original EHPad being queried is one of the ones exited.
3520b57cec5SDimitry Andric     Value *UnwindParent;
3530b57cec5SDimitry Andric     if (auto *UnwindPad = dyn_cast<Instruction>(UnwindDestToken))
3540b57cec5SDimitry Andric       UnwindParent = getParentPad(UnwindPad);
3550b57cec5SDimitry Andric     else
3560b57cec5SDimitry Andric       UnwindParent = nullptr;
3570b57cec5SDimitry Andric     bool ExitedOriginalPad = false;
3580b57cec5SDimitry Andric     for (Instruction *ExitedPad = CurrentPad;
3590b57cec5SDimitry Andric          ExitedPad && ExitedPad != UnwindParent;
3600b57cec5SDimitry Andric          ExitedPad = dyn_cast<Instruction>(getParentPad(ExitedPad))) {
3610b57cec5SDimitry Andric       // Skip over catchpads since they just follow their catchswitches.
3620b57cec5SDimitry Andric       if (isa<CatchPadInst>(ExitedPad))
3630b57cec5SDimitry Andric         continue;
3640b57cec5SDimitry Andric       MemoMap[ExitedPad] = UnwindDestToken;
3650b57cec5SDimitry Andric       ExitedOriginalPad |= (ExitedPad == EHPad);
3660b57cec5SDimitry Andric     }
3670b57cec5SDimitry Andric 
3680b57cec5SDimitry Andric     if (ExitedOriginalPad)
3690b57cec5SDimitry Andric       return UnwindDestToken;
3700b57cec5SDimitry Andric 
3710b57cec5SDimitry Andric     // Continue the search.
3720b57cec5SDimitry Andric   }
3730b57cec5SDimitry Andric 
3740b57cec5SDimitry Andric   // No definitive information is contained within this funclet.
3750b57cec5SDimitry Andric   return nullptr;
3760b57cec5SDimitry Andric }
3770b57cec5SDimitry Andric 
3780b57cec5SDimitry Andric /// Given an EH pad, find where it unwinds.  If it unwinds to an EH pad,
3790b57cec5SDimitry Andric /// return that pad instruction.  If it unwinds to caller, return
3800b57cec5SDimitry Andric /// ConstantTokenNone.  If it does not have a definitive unwind destination,
3810b57cec5SDimitry Andric /// return nullptr.
3820b57cec5SDimitry Andric ///
3830b57cec5SDimitry Andric /// This routine gets invoked for calls in funclets in inlinees when inlining
3840b57cec5SDimitry Andric /// an invoke.  Since many funclets don't have calls inside them, it's queried
3850b57cec5SDimitry Andric /// on-demand rather than building a map of pads to unwind dests up front.
3860b57cec5SDimitry Andric /// Determining a funclet's unwind dest may require recursively searching its
3870b57cec5SDimitry Andric /// descendants, and also ancestors and cousins if the descendants don't provide
3880b57cec5SDimitry Andric /// an answer.  Since most funclets will have their unwind dest immediately
3890b57cec5SDimitry Andric /// available as the unwind dest of a catchswitch or cleanupret, this routine
3900b57cec5SDimitry Andric /// searches top-down from the given pad and then up. To avoid worst-case
3910b57cec5SDimitry Andric /// quadratic run-time given that approach, it uses a memo map to avoid
3920b57cec5SDimitry Andric /// re-processing funclet trees.  The callers that rewrite the IR as they go
3930b57cec5SDimitry Andric /// take advantage of this, for correctness, by checking/forcing rewritten
3940b57cec5SDimitry Andric /// pads' entries to match the original callee view.
getUnwindDestToken(Instruction * EHPad,UnwindDestMemoTy & MemoMap)3950b57cec5SDimitry Andric static Value *getUnwindDestToken(Instruction *EHPad,
3960b57cec5SDimitry Andric                                  UnwindDestMemoTy &MemoMap) {
3970b57cec5SDimitry Andric   // Catchpads unwind to the same place as their catchswitch;
3980b57cec5SDimitry Andric   // redirct any queries on catchpads so the code below can
3990b57cec5SDimitry Andric   // deal with just catchswitches and cleanuppads.
4000b57cec5SDimitry Andric   if (auto *CPI = dyn_cast<CatchPadInst>(EHPad))
4010b57cec5SDimitry Andric     EHPad = CPI->getCatchSwitch();
4020b57cec5SDimitry Andric 
4030b57cec5SDimitry Andric   // Check if we've already determined the unwind dest for this pad.
4040b57cec5SDimitry Andric   auto Memo = MemoMap.find(EHPad);
4050b57cec5SDimitry Andric   if (Memo != MemoMap.end())
4060b57cec5SDimitry Andric     return Memo->second;
4070b57cec5SDimitry Andric 
4080b57cec5SDimitry Andric   // Search EHPad and, if necessary, its descendants.
4090b57cec5SDimitry Andric   Value *UnwindDestToken = getUnwindDestTokenHelper(EHPad, MemoMap);
4100b57cec5SDimitry Andric   assert((UnwindDestToken == nullptr) != (MemoMap.count(EHPad) != 0));
4110b57cec5SDimitry Andric   if (UnwindDestToken)
4120b57cec5SDimitry Andric     return UnwindDestToken;
4130b57cec5SDimitry Andric 
4140b57cec5SDimitry Andric   // No information is available for this EHPad from itself or any of its
4150b57cec5SDimitry Andric   // descendants.  An unwind all the way out to a pad in the caller would
4160b57cec5SDimitry Andric   // need also to agree with the unwind dest of the parent funclet, so
4170b57cec5SDimitry Andric   // search up the chain to try to find a funclet with information.  Put
4180b57cec5SDimitry Andric   // null entries in the memo map to avoid re-processing as we go up.
4190b57cec5SDimitry Andric   MemoMap[EHPad] = nullptr;
4200b57cec5SDimitry Andric #ifndef NDEBUG
4210b57cec5SDimitry Andric   SmallPtrSet<Instruction *, 4> TempMemos;
4220b57cec5SDimitry Andric   TempMemos.insert(EHPad);
4230b57cec5SDimitry Andric #endif
4240b57cec5SDimitry Andric   Instruction *LastUselessPad = EHPad;
4250b57cec5SDimitry Andric   Value *AncestorToken;
4260b57cec5SDimitry Andric   for (AncestorToken = getParentPad(EHPad);
4270b57cec5SDimitry Andric        auto *AncestorPad = dyn_cast<Instruction>(AncestorToken);
4280b57cec5SDimitry Andric        AncestorToken = getParentPad(AncestorToken)) {
4290b57cec5SDimitry Andric     // Skip over catchpads since they just follow their catchswitches.
4300b57cec5SDimitry Andric     if (isa<CatchPadInst>(AncestorPad))
4310b57cec5SDimitry Andric       continue;
4320b57cec5SDimitry Andric     // If the MemoMap had an entry mapping AncestorPad to nullptr, since we
4330b57cec5SDimitry Andric     // haven't yet called getUnwindDestTokenHelper for AncestorPad in this
4340b57cec5SDimitry Andric     // call to getUnwindDestToken, that would mean that AncestorPad had no
4350b57cec5SDimitry Andric     // information in itself, its descendants, or its ancestors.  If that
4360b57cec5SDimitry Andric     // were the case, then we should also have recorded the lack of information
4370b57cec5SDimitry Andric     // for the descendant that we're coming from.  So assert that we don't
4380b57cec5SDimitry Andric     // find a null entry in the MemoMap for AncestorPad.
4390b57cec5SDimitry Andric     assert(!MemoMap.count(AncestorPad) || MemoMap[AncestorPad]);
4400b57cec5SDimitry Andric     auto AncestorMemo = MemoMap.find(AncestorPad);
4410b57cec5SDimitry Andric     if (AncestorMemo == MemoMap.end()) {
4420b57cec5SDimitry Andric       UnwindDestToken = getUnwindDestTokenHelper(AncestorPad, MemoMap);
4430b57cec5SDimitry Andric     } else {
4440b57cec5SDimitry Andric       UnwindDestToken = AncestorMemo->second;
4450b57cec5SDimitry Andric     }
4460b57cec5SDimitry Andric     if (UnwindDestToken)
4470b57cec5SDimitry Andric       break;
4480b57cec5SDimitry Andric     LastUselessPad = AncestorPad;
4490b57cec5SDimitry Andric     MemoMap[LastUselessPad] = nullptr;
4500b57cec5SDimitry Andric #ifndef NDEBUG
4510b57cec5SDimitry Andric     TempMemos.insert(LastUselessPad);
4520b57cec5SDimitry Andric #endif
4530b57cec5SDimitry Andric   }
4540b57cec5SDimitry Andric 
4550b57cec5SDimitry Andric   // We know that getUnwindDestTokenHelper was called on LastUselessPad and
4560b57cec5SDimitry Andric   // returned nullptr (and likewise for EHPad and any of its ancestors up to
4570b57cec5SDimitry Andric   // LastUselessPad), so LastUselessPad has no information from below.  Since
4580b57cec5SDimitry Andric   // getUnwindDestTokenHelper must investigate all downward paths through
4590b57cec5SDimitry Andric   // no-information nodes to prove that a node has no information like this,
4600b57cec5SDimitry Andric   // and since any time it finds information it records it in the MemoMap for
4610b57cec5SDimitry Andric   // not just the immediately-containing funclet but also any ancestors also
4620b57cec5SDimitry Andric   // exited, it must be the case that, walking downward from LastUselessPad,
4630b57cec5SDimitry Andric   // visiting just those nodes which have not been mapped to an unwind dest
4640b57cec5SDimitry Andric   // by getUnwindDestTokenHelper (the nullptr TempMemos notwithstanding, since
4650b57cec5SDimitry Andric   // they are just used to keep getUnwindDestTokenHelper from repeating work),
4660b57cec5SDimitry Andric   // any node visited must have been exhaustively searched with no information
4670b57cec5SDimitry Andric   // for it found.
4680b57cec5SDimitry Andric   SmallVector<Instruction *, 8> Worklist(1, LastUselessPad);
4690b57cec5SDimitry Andric   while (!Worklist.empty()) {
4700b57cec5SDimitry Andric     Instruction *UselessPad = Worklist.pop_back_val();
4710b57cec5SDimitry Andric     auto Memo = MemoMap.find(UselessPad);
4720b57cec5SDimitry Andric     if (Memo != MemoMap.end() && Memo->second) {
4730b57cec5SDimitry Andric       // Here the name 'UselessPad' is a bit of a misnomer, because we've found
4740b57cec5SDimitry Andric       // that it is a funclet that does have information about unwinding to
4750b57cec5SDimitry Andric       // a particular destination; its parent was a useless pad.
4760b57cec5SDimitry Andric       // Since its parent has no information, the unwind edge must not escape
4770b57cec5SDimitry Andric       // the parent, and must target a sibling of this pad.  This local unwind
4780b57cec5SDimitry Andric       // gives us no information about EHPad.  Leave it and the subtree rooted
4790b57cec5SDimitry Andric       // at it alone.
4800b57cec5SDimitry Andric       assert(getParentPad(Memo->second) == getParentPad(UselessPad));
4810b57cec5SDimitry Andric       continue;
4820b57cec5SDimitry Andric     }
4830b57cec5SDimitry Andric     // We know we don't have information for UselesPad.  If it has an entry in
4840b57cec5SDimitry Andric     // the MemoMap (mapping it to nullptr), it must be one of the TempMemos
4850b57cec5SDimitry Andric     // added on this invocation of getUnwindDestToken; if a previous invocation
4860b57cec5SDimitry Andric     // recorded nullptr, it would have had to prove that the ancestors of
4870b57cec5SDimitry Andric     // UselessPad, which include LastUselessPad, had no information, and that
4880b57cec5SDimitry Andric     // in turn would have required proving that the descendants of
4890b57cec5SDimitry Andric     // LastUselesPad, which include EHPad, have no information about
4900b57cec5SDimitry Andric     // LastUselessPad, which would imply that EHPad was mapped to nullptr in
4910b57cec5SDimitry Andric     // the MemoMap on that invocation, which isn't the case if we got here.
4920b57cec5SDimitry Andric     assert(!MemoMap.count(UselessPad) || TempMemos.count(UselessPad));
4930b57cec5SDimitry Andric     // Assert as we enumerate users that 'UselessPad' doesn't have any unwind
4940b57cec5SDimitry Andric     // information that we'd be contradicting by making a map entry for it
4950b57cec5SDimitry Andric     // (which is something that getUnwindDestTokenHelper must have proved for
4960b57cec5SDimitry Andric     // us to get here).  Just assert on is direct users here; the checks in
4970b57cec5SDimitry Andric     // this downward walk at its descendants will verify that they don't have
4980b57cec5SDimitry Andric     // any unwind edges that exit 'UselessPad' either (i.e. they either have no
4990b57cec5SDimitry Andric     // unwind edges or unwind to a sibling).
5000b57cec5SDimitry Andric     MemoMap[UselessPad] = UnwindDestToken;
5010b57cec5SDimitry Andric     if (auto *CatchSwitch = dyn_cast<CatchSwitchInst>(UselessPad)) {
5020b57cec5SDimitry Andric       assert(CatchSwitch->getUnwindDest() == nullptr && "Expected useless pad");
5030b57cec5SDimitry Andric       for (BasicBlock *HandlerBlock : CatchSwitch->handlers()) {
5040b57cec5SDimitry Andric         auto *CatchPad = HandlerBlock->getFirstNonPHI();
5050b57cec5SDimitry Andric         for (User *U : CatchPad->users()) {
5060b57cec5SDimitry Andric           assert(
5070b57cec5SDimitry Andric               (!isa<InvokeInst>(U) ||
5080b57cec5SDimitry Andric                (getParentPad(
5090b57cec5SDimitry Andric                     cast<InvokeInst>(U)->getUnwindDest()->getFirstNonPHI()) ==
5100b57cec5SDimitry Andric                 CatchPad)) &&
5110b57cec5SDimitry Andric               "Expected useless pad");
5120b57cec5SDimitry Andric           if (isa<CatchSwitchInst>(U) || isa<CleanupPadInst>(U))
5130b57cec5SDimitry Andric             Worklist.push_back(cast<Instruction>(U));
5140b57cec5SDimitry Andric         }
5150b57cec5SDimitry Andric       }
5160b57cec5SDimitry Andric     } else {
5170b57cec5SDimitry Andric       assert(isa<CleanupPadInst>(UselessPad));
5180b57cec5SDimitry Andric       for (User *U : UselessPad->users()) {
5190b57cec5SDimitry Andric         assert(!isa<CleanupReturnInst>(U) && "Expected useless pad");
5200b57cec5SDimitry Andric         assert((!isa<InvokeInst>(U) ||
5210b57cec5SDimitry Andric                 (getParentPad(
5220b57cec5SDimitry Andric                      cast<InvokeInst>(U)->getUnwindDest()->getFirstNonPHI()) ==
5230b57cec5SDimitry Andric                  UselessPad)) &&
5240b57cec5SDimitry Andric                "Expected useless pad");
5250b57cec5SDimitry Andric         if (isa<CatchSwitchInst>(U) || isa<CleanupPadInst>(U))
5260b57cec5SDimitry Andric           Worklist.push_back(cast<Instruction>(U));
5270b57cec5SDimitry Andric       }
5280b57cec5SDimitry Andric     }
5290b57cec5SDimitry Andric   }
5300b57cec5SDimitry Andric 
5310b57cec5SDimitry Andric   return UnwindDestToken;
5320b57cec5SDimitry Andric }
5330b57cec5SDimitry Andric 
5340b57cec5SDimitry Andric /// When we inline a basic block into an invoke,
5350b57cec5SDimitry Andric /// we have to turn all of the calls that can throw into invokes.
5360b57cec5SDimitry Andric /// This function analyze BB to see if there are any calls, and if so,
5370b57cec5SDimitry Andric /// it rewrites them to be invokes that jump to InvokeDest and fills in the PHI
5380b57cec5SDimitry Andric /// nodes in that block with the values specified in InvokeDestPHIValues.
HandleCallsInBlockInlinedThroughInvoke(BasicBlock * BB,BasicBlock * UnwindEdge,UnwindDestMemoTy * FuncletUnwindMap=nullptr)5390b57cec5SDimitry Andric static BasicBlock *HandleCallsInBlockInlinedThroughInvoke(
5400b57cec5SDimitry Andric     BasicBlock *BB, BasicBlock *UnwindEdge,
5410b57cec5SDimitry Andric     UnwindDestMemoTy *FuncletUnwindMap = nullptr) {
5420b57cec5SDimitry Andric   for (BasicBlock::iterator BBI = BB->begin(), E = BB->end(); BBI != E; ) {
5430b57cec5SDimitry Andric     Instruction *I = &*BBI++;
5440b57cec5SDimitry Andric 
5450b57cec5SDimitry Andric     // We only need to check for function calls: inlined invoke
5460b57cec5SDimitry Andric     // instructions require no special handling.
5470b57cec5SDimitry Andric     CallInst *CI = dyn_cast<CallInst>(I);
5480b57cec5SDimitry Andric 
549*5f7ddb14SDimitry Andric     if (!CI || CI->doesNotThrow())
5500b57cec5SDimitry Andric       continue;
5510b57cec5SDimitry Andric 
552*5f7ddb14SDimitry Andric     if (CI->isInlineAsm()) {
553*5f7ddb14SDimitry Andric       InlineAsm *IA = cast<InlineAsm>(CI->getCalledOperand());
554*5f7ddb14SDimitry Andric       if (!IA->canThrow()) {
555*5f7ddb14SDimitry Andric         continue;
556*5f7ddb14SDimitry Andric       }
557*5f7ddb14SDimitry Andric     }
558*5f7ddb14SDimitry Andric 
5590b57cec5SDimitry Andric     // We do not need to (and in fact, cannot) convert possibly throwing calls
5600b57cec5SDimitry Andric     // to @llvm.experimental_deoptimize (resp. @llvm.experimental.guard) into
5610b57cec5SDimitry Andric     // invokes.  The caller's "segment" of the deoptimization continuation
5620b57cec5SDimitry Andric     // attached to the newly inlined @llvm.experimental_deoptimize
5630b57cec5SDimitry Andric     // (resp. @llvm.experimental.guard) call should contain the exception
5640b57cec5SDimitry Andric     // handling logic, if any.
5650b57cec5SDimitry Andric     if (auto *F = CI->getCalledFunction())
5660b57cec5SDimitry Andric       if (F->getIntrinsicID() == Intrinsic::experimental_deoptimize ||
5670b57cec5SDimitry Andric           F->getIntrinsicID() == Intrinsic::experimental_guard)
5680b57cec5SDimitry Andric         continue;
5690b57cec5SDimitry Andric 
5700b57cec5SDimitry Andric     if (auto FuncletBundle = CI->getOperandBundle(LLVMContext::OB_funclet)) {
5710b57cec5SDimitry Andric       // This call is nested inside a funclet.  If that funclet has an unwind
5720b57cec5SDimitry Andric       // destination within the inlinee, then unwinding out of this call would
5730b57cec5SDimitry Andric       // be UB.  Rewriting this call to an invoke which targets the inlined
5740b57cec5SDimitry Andric       // invoke's unwind dest would give the call's parent funclet multiple
5750b57cec5SDimitry Andric       // unwind destinations, which is something that subsequent EH table
5760b57cec5SDimitry Andric       // generation can't handle and that the veirifer rejects.  So when we
5770b57cec5SDimitry Andric       // see such a call, leave it as a call.
5780b57cec5SDimitry Andric       auto *FuncletPad = cast<Instruction>(FuncletBundle->Inputs[0]);
5790b57cec5SDimitry Andric       Value *UnwindDestToken =
5800b57cec5SDimitry Andric           getUnwindDestToken(FuncletPad, *FuncletUnwindMap);
5810b57cec5SDimitry Andric       if (UnwindDestToken && !isa<ConstantTokenNone>(UnwindDestToken))
5820b57cec5SDimitry Andric         continue;
5830b57cec5SDimitry Andric #ifndef NDEBUG
5840b57cec5SDimitry Andric       Instruction *MemoKey;
5850b57cec5SDimitry Andric       if (auto *CatchPad = dyn_cast<CatchPadInst>(FuncletPad))
5860b57cec5SDimitry Andric         MemoKey = CatchPad->getCatchSwitch();
5870b57cec5SDimitry Andric       else
5880b57cec5SDimitry Andric         MemoKey = FuncletPad;
5890b57cec5SDimitry Andric       assert(FuncletUnwindMap->count(MemoKey) &&
5900b57cec5SDimitry Andric              (*FuncletUnwindMap)[MemoKey] == UnwindDestToken &&
5910b57cec5SDimitry Andric              "must get memoized to avoid confusing later searches");
5920b57cec5SDimitry Andric #endif // NDEBUG
5930b57cec5SDimitry Andric     }
5940b57cec5SDimitry Andric 
5950b57cec5SDimitry Andric     changeToInvokeAndSplitBasicBlock(CI, UnwindEdge);
5960b57cec5SDimitry Andric     return BB;
5970b57cec5SDimitry Andric   }
5980b57cec5SDimitry Andric   return nullptr;
5990b57cec5SDimitry Andric }
6000b57cec5SDimitry Andric 
6010b57cec5SDimitry Andric /// If we inlined an invoke site, we need to convert calls
6020b57cec5SDimitry Andric /// in the body of the inlined function into invokes.
6030b57cec5SDimitry Andric ///
6040b57cec5SDimitry Andric /// II is the invoke instruction being inlined.  FirstNewBlock is the first
6050b57cec5SDimitry Andric /// block of the inlined code (the last block is the end of the function),
6060b57cec5SDimitry Andric /// and InlineCodeInfo is information about the code that got inlined.
HandleInlinedLandingPad(InvokeInst * II,BasicBlock * FirstNewBlock,ClonedCodeInfo & InlinedCodeInfo)6070b57cec5SDimitry Andric static void HandleInlinedLandingPad(InvokeInst *II, BasicBlock *FirstNewBlock,
6080b57cec5SDimitry Andric                                     ClonedCodeInfo &InlinedCodeInfo) {
6090b57cec5SDimitry Andric   BasicBlock *InvokeDest = II->getUnwindDest();
6100b57cec5SDimitry Andric 
6110b57cec5SDimitry Andric   Function *Caller = FirstNewBlock->getParent();
6120b57cec5SDimitry Andric 
6130b57cec5SDimitry Andric   // The inlined code is currently at the end of the function, scan from the
6140b57cec5SDimitry Andric   // start of the inlined code to its end, checking for stuff we need to
6150b57cec5SDimitry Andric   // rewrite.
6160b57cec5SDimitry Andric   LandingPadInliningInfo Invoke(II);
6170b57cec5SDimitry Andric 
6180b57cec5SDimitry Andric   // Get all of the inlined landing pad instructions.
6190b57cec5SDimitry Andric   SmallPtrSet<LandingPadInst*, 16> InlinedLPads;
6200b57cec5SDimitry Andric   for (Function::iterator I = FirstNewBlock->getIterator(), E = Caller->end();
6210b57cec5SDimitry Andric        I != E; ++I)
6220b57cec5SDimitry Andric     if (InvokeInst *II = dyn_cast<InvokeInst>(I->getTerminator()))
6230b57cec5SDimitry Andric       InlinedLPads.insert(II->getLandingPadInst());
6240b57cec5SDimitry Andric 
6250b57cec5SDimitry Andric   // Append the clauses from the outer landing pad instruction into the inlined
6260b57cec5SDimitry Andric   // landing pad instructions.
6270b57cec5SDimitry Andric   LandingPadInst *OuterLPad = Invoke.getLandingPadInst();
6280b57cec5SDimitry Andric   for (LandingPadInst *InlinedLPad : InlinedLPads) {
6290b57cec5SDimitry Andric     unsigned OuterNum = OuterLPad->getNumClauses();
6300b57cec5SDimitry Andric     InlinedLPad->reserveClauses(OuterNum);
6310b57cec5SDimitry Andric     for (unsigned OuterIdx = 0; OuterIdx != OuterNum; ++OuterIdx)
6320b57cec5SDimitry Andric       InlinedLPad->addClause(OuterLPad->getClause(OuterIdx));
6330b57cec5SDimitry Andric     if (OuterLPad->isCleanup())
6340b57cec5SDimitry Andric       InlinedLPad->setCleanup(true);
6350b57cec5SDimitry Andric   }
6360b57cec5SDimitry Andric 
6370b57cec5SDimitry Andric   for (Function::iterator BB = FirstNewBlock->getIterator(), E = Caller->end();
6380b57cec5SDimitry Andric        BB != E; ++BB) {
6390b57cec5SDimitry Andric     if (InlinedCodeInfo.ContainsCalls)
6400b57cec5SDimitry Andric       if (BasicBlock *NewBB = HandleCallsInBlockInlinedThroughInvoke(
6410b57cec5SDimitry Andric               &*BB, Invoke.getOuterResumeDest()))
6420b57cec5SDimitry Andric         // Update any PHI nodes in the exceptional block to indicate that there
6430b57cec5SDimitry Andric         // is now a new entry in them.
6440b57cec5SDimitry Andric         Invoke.addIncomingPHIValuesFor(NewBB);
6450b57cec5SDimitry Andric 
6460b57cec5SDimitry Andric     // Forward any resumes that are remaining here.
6470b57cec5SDimitry Andric     if (ResumeInst *RI = dyn_cast<ResumeInst>(BB->getTerminator()))
6480b57cec5SDimitry Andric       Invoke.forwardResume(RI, InlinedLPads);
6490b57cec5SDimitry Andric   }
6500b57cec5SDimitry Andric 
6510b57cec5SDimitry Andric   // Now that everything is happy, we have one final detail.  The PHI nodes in
6520b57cec5SDimitry Andric   // the exception destination block still have entries due to the original
6530b57cec5SDimitry Andric   // invoke instruction. Eliminate these entries (which might even delete the
6540b57cec5SDimitry Andric   // PHI node) now.
6550b57cec5SDimitry Andric   InvokeDest->removePredecessor(II->getParent());
6560b57cec5SDimitry Andric }
6570b57cec5SDimitry Andric 
6580b57cec5SDimitry Andric /// If we inlined an invoke site, we need to convert calls
6590b57cec5SDimitry Andric /// in the body of the inlined function into invokes.
6600b57cec5SDimitry Andric ///
6610b57cec5SDimitry Andric /// II is the invoke instruction being inlined.  FirstNewBlock is the first
6620b57cec5SDimitry Andric /// block of the inlined code (the last block is the end of the function),
6630b57cec5SDimitry Andric /// and InlineCodeInfo is information about the code that got inlined.
HandleInlinedEHPad(InvokeInst * II,BasicBlock * FirstNewBlock,ClonedCodeInfo & InlinedCodeInfo)6640b57cec5SDimitry Andric static void HandleInlinedEHPad(InvokeInst *II, BasicBlock *FirstNewBlock,
6650b57cec5SDimitry Andric                                ClonedCodeInfo &InlinedCodeInfo) {
6660b57cec5SDimitry Andric   BasicBlock *UnwindDest = II->getUnwindDest();
6670b57cec5SDimitry Andric   Function *Caller = FirstNewBlock->getParent();
6680b57cec5SDimitry Andric 
6690b57cec5SDimitry Andric   assert(UnwindDest->getFirstNonPHI()->isEHPad() && "unexpected BasicBlock!");
6700b57cec5SDimitry Andric 
6710b57cec5SDimitry Andric   // If there are PHI nodes in the unwind destination block, we need to keep
6720b57cec5SDimitry Andric   // track of which values came into them from the invoke before removing the
6730b57cec5SDimitry Andric   // edge from this block.
6740b57cec5SDimitry Andric   SmallVector<Value *, 8> UnwindDestPHIValues;
6750b57cec5SDimitry Andric   BasicBlock *InvokeBB = II->getParent();
6760b57cec5SDimitry Andric   for (Instruction &I : *UnwindDest) {
6770b57cec5SDimitry Andric     // Save the value to use for this edge.
6780b57cec5SDimitry Andric     PHINode *PHI = dyn_cast<PHINode>(&I);
6790b57cec5SDimitry Andric     if (!PHI)
6800b57cec5SDimitry Andric       break;
6810b57cec5SDimitry Andric     UnwindDestPHIValues.push_back(PHI->getIncomingValueForBlock(InvokeBB));
6820b57cec5SDimitry Andric   }
6830b57cec5SDimitry Andric 
6840b57cec5SDimitry Andric   // Add incoming-PHI values to the unwind destination block for the given basic
6850b57cec5SDimitry Andric   // block, using the values for the original invoke's source block.
6860b57cec5SDimitry Andric   auto UpdatePHINodes = [&](BasicBlock *Src) {
6870b57cec5SDimitry Andric     BasicBlock::iterator I = UnwindDest->begin();
6880b57cec5SDimitry Andric     for (Value *V : UnwindDestPHIValues) {
6890b57cec5SDimitry Andric       PHINode *PHI = cast<PHINode>(I);
6900b57cec5SDimitry Andric       PHI->addIncoming(V, Src);
6910b57cec5SDimitry Andric       ++I;
6920b57cec5SDimitry Andric     }
6930b57cec5SDimitry Andric   };
6940b57cec5SDimitry Andric 
6950b57cec5SDimitry Andric   // This connects all the instructions which 'unwind to caller' to the invoke
6960b57cec5SDimitry Andric   // destination.
6970b57cec5SDimitry Andric   UnwindDestMemoTy FuncletUnwindMap;
6980b57cec5SDimitry Andric   for (Function::iterator BB = FirstNewBlock->getIterator(), E = Caller->end();
6990b57cec5SDimitry Andric        BB != E; ++BB) {
7000b57cec5SDimitry Andric     if (auto *CRI = dyn_cast<CleanupReturnInst>(BB->getTerminator())) {
7010b57cec5SDimitry Andric       if (CRI->unwindsToCaller()) {
7020b57cec5SDimitry Andric         auto *CleanupPad = CRI->getCleanupPad();
7030b57cec5SDimitry Andric         CleanupReturnInst::Create(CleanupPad, UnwindDest, CRI);
7040b57cec5SDimitry Andric         CRI->eraseFromParent();
7050b57cec5SDimitry Andric         UpdatePHINodes(&*BB);
7060b57cec5SDimitry Andric         // Finding a cleanupret with an unwind destination would confuse
7070b57cec5SDimitry Andric         // subsequent calls to getUnwindDestToken, so map the cleanuppad
7080b57cec5SDimitry Andric         // to short-circuit any such calls and recognize this as an "unwind
7090b57cec5SDimitry Andric         // to caller" cleanup.
7100b57cec5SDimitry Andric         assert(!FuncletUnwindMap.count(CleanupPad) ||
7110b57cec5SDimitry Andric                isa<ConstantTokenNone>(FuncletUnwindMap[CleanupPad]));
7120b57cec5SDimitry Andric         FuncletUnwindMap[CleanupPad] =
7130b57cec5SDimitry Andric             ConstantTokenNone::get(Caller->getContext());
7140b57cec5SDimitry Andric       }
7150b57cec5SDimitry Andric     }
7160b57cec5SDimitry Andric 
7170b57cec5SDimitry Andric     Instruction *I = BB->getFirstNonPHI();
7180b57cec5SDimitry Andric     if (!I->isEHPad())
7190b57cec5SDimitry Andric       continue;
7200b57cec5SDimitry Andric 
7210b57cec5SDimitry Andric     Instruction *Replacement = nullptr;
7220b57cec5SDimitry Andric     if (auto *CatchSwitch = dyn_cast<CatchSwitchInst>(I)) {
7230b57cec5SDimitry Andric       if (CatchSwitch->unwindsToCaller()) {
7240b57cec5SDimitry Andric         Value *UnwindDestToken;
7250b57cec5SDimitry Andric         if (auto *ParentPad =
7260b57cec5SDimitry Andric                 dyn_cast<Instruction>(CatchSwitch->getParentPad())) {
7270b57cec5SDimitry Andric           // This catchswitch is nested inside another funclet.  If that
7280b57cec5SDimitry Andric           // funclet has an unwind destination within the inlinee, then
7290b57cec5SDimitry Andric           // unwinding out of this catchswitch would be UB.  Rewriting this
7300b57cec5SDimitry Andric           // catchswitch to unwind to the inlined invoke's unwind dest would
7310b57cec5SDimitry Andric           // give the parent funclet multiple unwind destinations, which is
7320b57cec5SDimitry Andric           // something that subsequent EH table generation can't handle and
7330b57cec5SDimitry Andric           // that the veirifer rejects.  So when we see such a call, leave it
7340b57cec5SDimitry Andric           // as "unwind to caller".
7350b57cec5SDimitry Andric           UnwindDestToken = getUnwindDestToken(ParentPad, FuncletUnwindMap);
7360b57cec5SDimitry Andric           if (UnwindDestToken && !isa<ConstantTokenNone>(UnwindDestToken))
7370b57cec5SDimitry Andric             continue;
7380b57cec5SDimitry Andric         } else {
7390b57cec5SDimitry Andric           // This catchswitch has no parent to inherit constraints from, and
7400b57cec5SDimitry Andric           // none of its descendants can have an unwind edge that exits it and
7410b57cec5SDimitry Andric           // targets another funclet in the inlinee.  It may or may not have a
7420b57cec5SDimitry Andric           // descendant that definitively has an unwind to caller.  In either
7430b57cec5SDimitry Andric           // case, we'll have to assume that any unwinds out of it may need to
7440b57cec5SDimitry Andric           // be routed to the caller, so treat it as though it has a definitive
7450b57cec5SDimitry Andric           // unwind to caller.
7460b57cec5SDimitry Andric           UnwindDestToken = ConstantTokenNone::get(Caller->getContext());
7470b57cec5SDimitry Andric         }
7480b57cec5SDimitry Andric         auto *NewCatchSwitch = CatchSwitchInst::Create(
7490b57cec5SDimitry Andric             CatchSwitch->getParentPad(), UnwindDest,
7500b57cec5SDimitry Andric             CatchSwitch->getNumHandlers(), CatchSwitch->getName(),
7510b57cec5SDimitry Andric             CatchSwitch);
7520b57cec5SDimitry Andric         for (BasicBlock *PadBB : CatchSwitch->handlers())
7530b57cec5SDimitry Andric           NewCatchSwitch->addHandler(PadBB);
7540b57cec5SDimitry Andric         // Propagate info for the old catchswitch over to the new one in
7550b57cec5SDimitry Andric         // the unwind map.  This also serves to short-circuit any subsequent
7560b57cec5SDimitry Andric         // checks for the unwind dest of this catchswitch, which would get
7570b57cec5SDimitry Andric         // confused if they found the outer handler in the callee.
7580b57cec5SDimitry Andric         FuncletUnwindMap[NewCatchSwitch] = UnwindDestToken;
7590b57cec5SDimitry Andric         Replacement = NewCatchSwitch;
7600b57cec5SDimitry Andric       }
7610b57cec5SDimitry Andric     } else if (!isa<FuncletPadInst>(I)) {
7620b57cec5SDimitry Andric       llvm_unreachable("unexpected EHPad!");
7630b57cec5SDimitry Andric     }
7640b57cec5SDimitry Andric 
7650b57cec5SDimitry Andric     if (Replacement) {
7660b57cec5SDimitry Andric       Replacement->takeName(I);
7670b57cec5SDimitry Andric       I->replaceAllUsesWith(Replacement);
7680b57cec5SDimitry Andric       I->eraseFromParent();
7690b57cec5SDimitry Andric       UpdatePHINodes(&*BB);
7700b57cec5SDimitry Andric     }
7710b57cec5SDimitry Andric   }
7720b57cec5SDimitry Andric 
7730b57cec5SDimitry Andric   if (InlinedCodeInfo.ContainsCalls)
7740b57cec5SDimitry Andric     for (Function::iterator BB = FirstNewBlock->getIterator(),
7750b57cec5SDimitry Andric                             E = Caller->end();
7760b57cec5SDimitry Andric          BB != E; ++BB)
7770b57cec5SDimitry Andric       if (BasicBlock *NewBB = HandleCallsInBlockInlinedThroughInvoke(
7780b57cec5SDimitry Andric               &*BB, UnwindDest, &FuncletUnwindMap))
7790b57cec5SDimitry Andric         // Update any PHI nodes in the exceptional block to indicate that there
7800b57cec5SDimitry Andric         // is now a new entry in them.
7810b57cec5SDimitry Andric         UpdatePHINodes(NewBB);
7820b57cec5SDimitry Andric 
7830b57cec5SDimitry Andric   // Now that everything is happy, we have one final detail.  The PHI nodes in
7840b57cec5SDimitry Andric   // the exception destination block still have entries due to the original
7850b57cec5SDimitry Andric   // invoke instruction. Eliminate these entries (which might even delete the
7860b57cec5SDimitry Andric   // PHI node) now.
7870b57cec5SDimitry Andric   UnwindDest->removePredecessor(InvokeBB);
7880b57cec5SDimitry Andric }
7890b57cec5SDimitry Andric 
790af732203SDimitry Andric /// When inlining a call site that has !llvm.mem.parallel_loop_access,
791af732203SDimitry Andric /// !llvm.access.group, !alias.scope or !noalias metadata, that metadata should
792af732203SDimitry Andric /// be propagated to all memory-accessing cloned instructions.
PropagateCallSiteMetadata(CallBase & CB,Function::iterator FStart,Function::iterator FEnd)793af732203SDimitry Andric static void PropagateCallSiteMetadata(CallBase &CB, Function::iterator FStart,
794af732203SDimitry Andric                                       Function::iterator FEnd) {
795af732203SDimitry Andric   MDNode *MemParallelLoopAccess =
796af732203SDimitry Andric       CB.getMetadata(LLVMContext::MD_mem_parallel_loop_access);
797af732203SDimitry Andric   MDNode *AccessGroup = CB.getMetadata(LLVMContext::MD_access_group);
798af732203SDimitry Andric   MDNode *AliasScope = CB.getMetadata(LLVMContext::MD_alias_scope);
799af732203SDimitry Andric   MDNode *NoAlias = CB.getMetadata(LLVMContext::MD_noalias);
800af732203SDimitry Andric   if (!MemParallelLoopAccess && !AccessGroup && !AliasScope && !NoAlias)
8010b57cec5SDimitry Andric     return;
8020b57cec5SDimitry Andric 
803af732203SDimitry Andric   for (BasicBlock &BB : make_range(FStart, FEnd)) {
804af732203SDimitry Andric     for (Instruction &I : BB) {
805af732203SDimitry Andric       // This metadata is only relevant for instructions that access memory.
806af732203SDimitry Andric       if (!I.mayReadOrWriteMemory())
8070b57cec5SDimitry Andric         continue;
8080b57cec5SDimitry Andric 
809af732203SDimitry Andric       if (MemParallelLoopAccess) {
810af732203SDimitry Andric         // TODO: This probably should not overwrite MemParalleLoopAccess.
811af732203SDimitry Andric         MemParallelLoopAccess = MDNode::concatenate(
812af732203SDimitry Andric             I.getMetadata(LLVMContext::MD_mem_parallel_loop_access),
813af732203SDimitry Andric             MemParallelLoopAccess);
814af732203SDimitry Andric         I.setMetadata(LLVMContext::MD_mem_parallel_loop_access,
815af732203SDimitry Andric                       MemParallelLoopAccess);
8160b57cec5SDimitry Andric       }
8170b57cec5SDimitry Andric 
818af732203SDimitry Andric       if (AccessGroup)
819af732203SDimitry Andric         I.setMetadata(LLVMContext::MD_access_group, uniteAccessGroups(
820af732203SDimitry Andric             I.getMetadata(LLVMContext::MD_access_group), AccessGroup));
821af732203SDimitry Andric 
822af732203SDimitry Andric       if (AliasScope)
823af732203SDimitry Andric         I.setMetadata(LLVMContext::MD_alias_scope, MDNode::concatenate(
824af732203SDimitry Andric             I.getMetadata(LLVMContext::MD_alias_scope), AliasScope));
825af732203SDimitry Andric 
826af732203SDimitry Andric       if (NoAlias)
827af732203SDimitry Andric         I.setMetadata(LLVMContext::MD_noalias, MDNode::concatenate(
828af732203SDimitry Andric             I.getMetadata(LLVMContext::MD_noalias), NoAlias));
8290b57cec5SDimitry Andric     }
8300b57cec5SDimitry Andric   }
8310b57cec5SDimitry Andric }
8320b57cec5SDimitry Andric 
833af732203SDimitry Andric /// Utility for cloning !noalias and !alias.scope metadata. When a code region
834af732203SDimitry Andric /// using scoped alias metadata is inlined, the aliasing relationships may not
835af732203SDimitry Andric /// hold between the two version. It is necessary to create a deep clone of the
836af732203SDimitry Andric /// metadata, putting the two versions in separate scope domains.
837af732203SDimitry Andric class ScopedAliasMetadataDeepCloner {
838af732203SDimitry Andric   using MetadataMap = DenseMap<const MDNode *, TrackingMDNodeRef>;
8390b57cec5SDimitry Andric   SetVector<const MDNode *> MD;
840af732203SDimitry Andric   MetadataMap MDMap;
841af732203SDimitry Andric   void addRecursiveMetadataUses();
8420b57cec5SDimitry Andric 
843af732203SDimitry Andric public:
844af732203SDimitry Andric   ScopedAliasMetadataDeepCloner(const Function *F);
8450b57cec5SDimitry Andric 
846af732203SDimitry Andric   /// Create a new clone of the scoped alias metadata, which will be used by
847af732203SDimitry Andric   /// subsequent remap() calls.
848af732203SDimitry Andric   void clone();
849af732203SDimitry Andric 
850af732203SDimitry Andric   /// Remap instructions in the given range from the original to the cloned
851af732203SDimitry Andric   /// metadata.
852af732203SDimitry Andric   void remap(Function::iterator FStart, Function::iterator FEnd);
853af732203SDimitry Andric };
854af732203SDimitry Andric 
ScopedAliasMetadataDeepCloner(const Function * F)855af732203SDimitry Andric ScopedAliasMetadataDeepCloner::ScopedAliasMetadataDeepCloner(
856af732203SDimitry Andric     const Function *F) {
857af732203SDimitry Andric   for (const BasicBlock &BB : *F) {
858af732203SDimitry Andric     for (const Instruction &I : BB) {
859af732203SDimitry Andric       if (const MDNode *M = I.getMetadata(LLVMContext::MD_alias_scope))
8600b57cec5SDimitry Andric         MD.insert(M);
861af732203SDimitry Andric       if (const MDNode *M = I.getMetadata(LLVMContext::MD_noalias))
8620b57cec5SDimitry Andric         MD.insert(M);
863af732203SDimitry Andric 
864af732203SDimitry Andric       // We also need to clone the metadata in noalias intrinsics.
865af732203SDimitry Andric       if (const auto *Decl = dyn_cast<NoAliasScopeDeclInst>(&I))
866af732203SDimitry Andric         MD.insert(Decl->getScopeList());
867af732203SDimitry Andric     }
868af732203SDimitry Andric   }
869af732203SDimitry Andric   addRecursiveMetadataUses();
8700b57cec5SDimitry Andric }
8710b57cec5SDimitry Andric 
addRecursiveMetadataUses()872af732203SDimitry Andric void ScopedAliasMetadataDeepCloner::addRecursiveMetadataUses() {
8730b57cec5SDimitry Andric   SmallVector<const Metadata *, 16> Queue(MD.begin(), MD.end());
8740b57cec5SDimitry Andric   while (!Queue.empty()) {
8750b57cec5SDimitry Andric     const MDNode *M = cast<MDNode>(Queue.pop_back_val());
876af732203SDimitry Andric     for (const Metadata *Op : M->operands())
877af732203SDimitry Andric       if (const MDNode *OpMD = dyn_cast<MDNode>(Op))
878af732203SDimitry Andric         if (MD.insert(OpMD))
879af732203SDimitry Andric           Queue.push_back(OpMD);
880af732203SDimitry Andric   }
8810b57cec5SDimitry Andric }
8820b57cec5SDimitry Andric 
clone()883af732203SDimitry Andric void ScopedAliasMetadataDeepCloner::clone() {
884af732203SDimitry Andric   assert(MDMap.empty() && "clone() already called ?");
885af732203SDimitry Andric 
8860b57cec5SDimitry Andric   SmallVector<TempMDTuple, 16> DummyNodes;
8870b57cec5SDimitry Andric   for (const MDNode *I : MD) {
888af732203SDimitry Andric     DummyNodes.push_back(MDTuple::getTemporary(I->getContext(), None));
8890b57cec5SDimitry Andric     MDMap[I].reset(DummyNodes.back().get());
8900b57cec5SDimitry Andric   }
8910b57cec5SDimitry Andric 
8920b57cec5SDimitry Andric   // Create new metadata nodes to replace the dummy nodes, replacing old
8930b57cec5SDimitry Andric   // metadata references with either a dummy node or an already-created new
8940b57cec5SDimitry Andric   // node.
8950b57cec5SDimitry Andric   SmallVector<Metadata *, 4> NewOps;
896af732203SDimitry Andric   for (const MDNode *I : MD) {
897af732203SDimitry Andric     for (const Metadata *Op : I->operands()) {
898af732203SDimitry Andric       if (const MDNode *M = dyn_cast<MDNode>(Op))
8990b57cec5SDimitry Andric         NewOps.push_back(MDMap[M]);
9000b57cec5SDimitry Andric       else
901af732203SDimitry Andric         NewOps.push_back(const_cast<Metadata *>(Op));
9020b57cec5SDimitry Andric     }
9030b57cec5SDimitry Andric 
904af732203SDimitry Andric     MDNode *NewM = MDNode::get(I->getContext(), NewOps);
9050b57cec5SDimitry Andric     MDTuple *TempM = cast<MDTuple>(MDMap[I]);
9060b57cec5SDimitry Andric     assert(TempM->isTemporary() && "Expected temporary node");
9070b57cec5SDimitry Andric 
9080b57cec5SDimitry Andric     TempM->replaceAllUsesWith(NewM);
909af732203SDimitry Andric     NewOps.clear();
910af732203SDimitry Andric   }
9110b57cec5SDimitry Andric }
9120b57cec5SDimitry Andric 
remap(Function::iterator FStart,Function::iterator FEnd)913af732203SDimitry Andric void ScopedAliasMetadataDeepCloner::remap(Function::iterator FStart,
914af732203SDimitry Andric                                           Function::iterator FEnd) {
915af732203SDimitry Andric   if (MDMap.empty())
916af732203SDimitry Andric     return; // Nothing to do.
9170b57cec5SDimitry Andric 
918af732203SDimitry Andric   for (BasicBlock &BB : make_range(FStart, FEnd)) {
919af732203SDimitry Andric     for (Instruction &I : BB) {
920af732203SDimitry Andric       // TODO: The null checks for the MDMap.lookup() results should no longer
921af732203SDimitry Andric       // be necessary.
922af732203SDimitry Andric       if (MDNode *M = I.getMetadata(LLVMContext::MD_alias_scope))
923af732203SDimitry Andric         if (MDNode *MNew = MDMap.lookup(M))
924af732203SDimitry Andric           I.setMetadata(LLVMContext::MD_alias_scope, MNew);
9250b57cec5SDimitry Andric 
926af732203SDimitry Andric       if (MDNode *M = I.getMetadata(LLVMContext::MD_noalias))
927af732203SDimitry Andric         if (MDNode *MNew = MDMap.lookup(M))
928af732203SDimitry Andric           I.setMetadata(LLVMContext::MD_noalias, MNew);
9290b57cec5SDimitry Andric 
930af732203SDimitry Andric       if (auto *Decl = dyn_cast<NoAliasScopeDeclInst>(&I))
931af732203SDimitry Andric         if (MDNode *MNew = MDMap.lookup(Decl->getScopeList()))
932af732203SDimitry Andric           Decl->setScopeList(MNew);
9330b57cec5SDimitry Andric     }
9340b57cec5SDimitry Andric   }
9350b57cec5SDimitry Andric }
9360b57cec5SDimitry Andric 
9370b57cec5SDimitry Andric /// If the inlined function has noalias arguments,
9380b57cec5SDimitry Andric /// then add new alias scopes for each noalias argument, tag the mapped noalias
9390b57cec5SDimitry Andric /// parameters with noalias metadata specifying the new scope, and tag all
9400b57cec5SDimitry Andric /// non-derived loads, stores and memory intrinsics with the new alias scopes.
AddAliasScopeMetadata(CallBase & CB,ValueToValueMapTy & VMap,const DataLayout & DL,AAResults * CalleeAAR,ClonedCodeInfo & InlinedFunctionInfo)9415ffd83dbSDimitry Andric static void AddAliasScopeMetadata(CallBase &CB, ValueToValueMapTy &VMap,
942*5f7ddb14SDimitry Andric                                   const DataLayout &DL, AAResults *CalleeAAR,
943*5f7ddb14SDimitry Andric                                   ClonedCodeInfo &InlinedFunctionInfo) {
9440b57cec5SDimitry Andric   if (!EnableNoAliasConversion)
9450b57cec5SDimitry Andric     return;
9460b57cec5SDimitry Andric 
9475ffd83dbSDimitry Andric   const Function *CalledFunc = CB.getCalledFunction();
9480b57cec5SDimitry Andric   SmallVector<const Argument *, 4> NoAliasArgs;
9490b57cec5SDimitry Andric 
9500b57cec5SDimitry Andric   for (const Argument &Arg : CalledFunc->args())
9515ffd83dbSDimitry Andric     if (CB.paramHasAttr(Arg.getArgNo(), Attribute::NoAlias) && !Arg.use_empty())
9520b57cec5SDimitry Andric       NoAliasArgs.push_back(&Arg);
9530b57cec5SDimitry Andric 
9540b57cec5SDimitry Andric   if (NoAliasArgs.empty())
9550b57cec5SDimitry Andric     return;
9560b57cec5SDimitry Andric 
9570b57cec5SDimitry Andric   // To do a good job, if a noalias variable is captured, we need to know if
9580b57cec5SDimitry Andric   // the capture point dominates the particular use we're considering.
9590b57cec5SDimitry Andric   DominatorTree DT;
9600b57cec5SDimitry Andric   DT.recalculate(const_cast<Function&>(*CalledFunc));
9610b57cec5SDimitry Andric 
9620b57cec5SDimitry Andric   // noalias indicates that pointer values based on the argument do not alias
9630b57cec5SDimitry Andric   // pointer values which are not based on it. So we add a new "scope" for each
9640b57cec5SDimitry Andric   // noalias function argument. Accesses using pointers based on that argument
9650b57cec5SDimitry Andric   // become part of that alias scope, accesses using pointers not based on that
9660b57cec5SDimitry Andric   // argument are tagged as noalias with that scope.
9670b57cec5SDimitry Andric 
9680b57cec5SDimitry Andric   DenseMap<const Argument *, MDNode *> NewScopes;
9690b57cec5SDimitry Andric   MDBuilder MDB(CalledFunc->getContext());
9700b57cec5SDimitry Andric 
9710b57cec5SDimitry Andric   // Create a new scope domain for this function.
9720b57cec5SDimitry Andric   MDNode *NewDomain =
9730b57cec5SDimitry Andric     MDB.createAnonymousAliasScopeDomain(CalledFunc->getName());
9740b57cec5SDimitry Andric   for (unsigned i = 0, e = NoAliasArgs.size(); i != e; ++i) {
9750b57cec5SDimitry Andric     const Argument *A = NoAliasArgs[i];
9760b57cec5SDimitry Andric 
9775ffd83dbSDimitry Andric     std::string Name = std::string(CalledFunc->getName());
9780b57cec5SDimitry Andric     if (A->hasName()) {
9790b57cec5SDimitry Andric       Name += ": %";
9800b57cec5SDimitry Andric       Name += A->getName();
9810b57cec5SDimitry Andric     } else {
9820b57cec5SDimitry Andric       Name += ": argument ";
9830b57cec5SDimitry Andric       Name += utostr(i);
9840b57cec5SDimitry Andric     }
9850b57cec5SDimitry Andric 
9860b57cec5SDimitry Andric     // Note: We always create a new anonymous root here. This is true regardless
9870b57cec5SDimitry Andric     // of the linkage of the callee because the aliasing "scope" is not just a
9880b57cec5SDimitry Andric     // property of the callee, but also all control dependencies in the caller.
9890b57cec5SDimitry Andric     MDNode *NewScope = MDB.createAnonymousAliasScope(NewDomain, Name);
9900b57cec5SDimitry Andric     NewScopes.insert(std::make_pair(A, NewScope));
991af732203SDimitry Andric 
992af732203SDimitry Andric     if (UseNoAliasIntrinsic) {
993af732203SDimitry Andric       // Introduce a llvm.experimental.noalias.scope.decl for the noalias
994af732203SDimitry Andric       // argument.
995af732203SDimitry Andric       MDNode *AScopeList = MDNode::get(CalledFunc->getContext(), NewScope);
996af732203SDimitry Andric       auto *NoAliasDecl =
997af732203SDimitry Andric           IRBuilder<>(&CB).CreateNoAliasScopeDeclaration(AScopeList);
998af732203SDimitry Andric       // Ignore the result for now. The result will be used when the
999af732203SDimitry Andric       // llvm.noalias intrinsic is introduced.
1000af732203SDimitry Andric       (void)NoAliasDecl;
1001af732203SDimitry Andric     }
10020b57cec5SDimitry Andric   }
10030b57cec5SDimitry Andric 
10040b57cec5SDimitry Andric   // Iterate over all new instructions in the map; for all memory-access
10050b57cec5SDimitry Andric   // instructions, add the alias scope metadata.
10060b57cec5SDimitry Andric   for (ValueToValueMapTy::iterator VMI = VMap.begin(), VMIE = VMap.end();
10070b57cec5SDimitry Andric        VMI != VMIE; ++VMI) {
10080b57cec5SDimitry Andric     if (const Instruction *I = dyn_cast<Instruction>(VMI->first)) {
10090b57cec5SDimitry Andric       if (!VMI->second)
10100b57cec5SDimitry Andric         continue;
10110b57cec5SDimitry Andric 
10120b57cec5SDimitry Andric       Instruction *NI = dyn_cast<Instruction>(VMI->second);
1013*5f7ddb14SDimitry Andric       if (!NI || InlinedFunctionInfo.isSimplified(I, NI))
10140b57cec5SDimitry Andric         continue;
10150b57cec5SDimitry Andric 
10160b57cec5SDimitry Andric       bool IsArgMemOnlyCall = false, IsFuncCall = false;
10170b57cec5SDimitry Andric       SmallVector<const Value *, 2> PtrArgs;
10180b57cec5SDimitry Andric 
10190b57cec5SDimitry Andric       if (const LoadInst *LI = dyn_cast<LoadInst>(I))
10200b57cec5SDimitry Andric         PtrArgs.push_back(LI->getPointerOperand());
10210b57cec5SDimitry Andric       else if (const StoreInst *SI = dyn_cast<StoreInst>(I))
10220b57cec5SDimitry Andric         PtrArgs.push_back(SI->getPointerOperand());
10230b57cec5SDimitry Andric       else if (const VAArgInst *VAAI = dyn_cast<VAArgInst>(I))
10240b57cec5SDimitry Andric         PtrArgs.push_back(VAAI->getPointerOperand());
10250b57cec5SDimitry Andric       else if (const AtomicCmpXchgInst *CXI = dyn_cast<AtomicCmpXchgInst>(I))
10260b57cec5SDimitry Andric         PtrArgs.push_back(CXI->getPointerOperand());
10270b57cec5SDimitry Andric       else if (const AtomicRMWInst *RMWI = dyn_cast<AtomicRMWInst>(I))
10280b57cec5SDimitry Andric         PtrArgs.push_back(RMWI->getPointerOperand());
10290b57cec5SDimitry Andric       else if (const auto *Call = dyn_cast<CallBase>(I)) {
10300b57cec5SDimitry Andric         // If we know that the call does not access memory, then we'll still
10310b57cec5SDimitry Andric         // know that about the inlined clone of this call site, and we don't
10320b57cec5SDimitry Andric         // need to add metadata.
10330b57cec5SDimitry Andric         if (Call->doesNotAccessMemory())
10340b57cec5SDimitry Andric           continue;
10350b57cec5SDimitry Andric 
10360b57cec5SDimitry Andric         IsFuncCall = true;
10370b57cec5SDimitry Andric         if (CalleeAAR) {
10380b57cec5SDimitry Andric           FunctionModRefBehavior MRB = CalleeAAR->getModRefBehavior(Call);
1039*5f7ddb14SDimitry Andric 
1040*5f7ddb14SDimitry Andric           // We'll retain this knowledge without additional metadata.
1041*5f7ddb14SDimitry Andric           if (AAResults::onlyAccessesInaccessibleMem(MRB))
1042*5f7ddb14SDimitry Andric             continue;
1043*5f7ddb14SDimitry Andric 
10445ffd83dbSDimitry Andric           if (AAResults::onlyAccessesArgPointees(MRB))
10450b57cec5SDimitry Andric             IsArgMemOnlyCall = true;
10460b57cec5SDimitry Andric         }
10470b57cec5SDimitry Andric 
10480b57cec5SDimitry Andric         for (Value *Arg : Call->args()) {
10490b57cec5SDimitry Andric           // We need to check the underlying objects of all arguments, not just
10500b57cec5SDimitry Andric           // the pointer arguments, because we might be passing pointers as
10510b57cec5SDimitry Andric           // integers, etc.
10520b57cec5SDimitry Andric           // However, if we know that the call only accesses pointer arguments,
10530b57cec5SDimitry Andric           // then we only need to check the pointer arguments.
10540b57cec5SDimitry Andric           if (IsArgMemOnlyCall && !Arg->getType()->isPointerTy())
10550b57cec5SDimitry Andric             continue;
10560b57cec5SDimitry Andric 
10570b57cec5SDimitry Andric           PtrArgs.push_back(Arg);
10580b57cec5SDimitry Andric         }
10590b57cec5SDimitry Andric       }
10600b57cec5SDimitry Andric 
10610b57cec5SDimitry Andric       // If we found no pointers, then this instruction is not suitable for
10620b57cec5SDimitry Andric       // pairing with an instruction to receive aliasing metadata.
10630b57cec5SDimitry Andric       // However, if this is a call, this we might just alias with none of the
10640b57cec5SDimitry Andric       // noalias arguments.
10650b57cec5SDimitry Andric       if (PtrArgs.empty() && !IsFuncCall)
10660b57cec5SDimitry Andric         continue;
10670b57cec5SDimitry Andric 
10680b57cec5SDimitry Andric       // It is possible that there is only one underlying object, but you
10690b57cec5SDimitry Andric       // need to go through several PHIs to see it, and thus could be
10700b57cec5SDimitry Andric       // repeated in the Objects list.
10710b57cec5SDimitry Andric       SmallPtrSet<const Value *, 4> ObjSet;
10720b57cec5SDimitry Andric       SmallVector<Metadata *, 4> Scopes, NoAliases;
10730b57cec5SDimitry Andric 
10740b57cec5SDimitry Andric       SmallSetVector<const Argument *, 4> NAPtrArgs;
10750b57cec5SDimitry Andric       for (const Value *V : PtrArgs) {
10760b57cec5SDimitry Andric         SmallVector<const Value *, 4> Objects;
1077af732203SDimitry Andric         getUnderlyingObjects(V, Objects, /* LI = */ nullptr);
10780b57cec5SDimitry Andric 
10790b57cec5SDimitry Andric         for (const Value *O : Objects)
10800b57cec5SDimitry Andric           ObjSet.insert(O);
10810b57cec5SDimitry Andric       }
10820b57cec5SDimitry Andric 
10830b57cec5SDimitry Andric       // Figure out if we're derived from anything that is not a noalias
10840b57cec5SDimitry Andric       // argument.
10850b57cec5SDimitry Andric       bool CanDeriveViaCapture = false, UsesAliasingPtr = false;
10860b57cec5SDimitry Andric       for (const Value *V : ObjSet) {
10870b57cec5SDimitry Andric         // Is this value a constant that cannot be derived from any pointer
10880b57cec5SDimitry Andric         // value (we need to exclude constant expressions, for example, that
10890b57cec5SDimitry Andric         // are formed from arithmetic on global symbols).
10900b57cec5SDimitry Andric         bool IsNonPtrConst = isa<ConstantInt>(V) || isa<ConstantFP>(V) ||
10910b57cec5SDimitry Andric                              isa<ConstantPointerNull>(V) ||
10920b57cec5SDimitry Andric                              isa<ConstantDataVector>(V) || isa<UndefValue>(V);
10930b57cec5SDimitry Andric         if (IsNonPtrConst)
10940b57cec5SDimitry Andric           continue;
10950b57cec5SDimitry Andric 
10960b57cec5SDimitry Andric         // If this is anything other than a noalias argument, then we cannot
10970b57cec5SDimitry Andric         // completely describe the aliasing properties using alias.scope
10980b57cec5SDimitry Andric         // metadata (and, thus, won't add any).
10990b57cec5SDimitry Andric         if (const Argument *A = dyn_cast<Argument>(V)) {
11005ffd83dbSDimitry Andric           if (!CB.paramHasAttr(A->getArgNo(), Attribute::NoAlias))
11010b57cec5SDimitry Andric             UsesAliasingPtr = true;
11020b57cec5SDimitry Andric         } else {
11030b57cec5SDimitry Andric           UsesAliasingPtr = true;
11040b57cec5SDimitry Andric         }
11050b57cec5SDimitry Andric 
11060b57cec5SDimitry Andric         // If this is not some identified function-local object (which cannot
11070b57cec5SDimitry Andric         // directly alias a noalias argument), or some other argument (which,
11080b57cec5SDimitry Andric         // by definition, also cannot alias a noalias argument), then we could
11090b57cec5SDimitry Andric         // alias a noalias argument that has been captured).
11100b57cec5SDimitry Andric         if (!isa<Argument>(V) &&
11110b57cec5SDimitry Andric             !isIdentifiedFunctionLocal(const_cast<Value*>(V)))
11120b57cec5SDimitry Andric           CanDeriveViaCapture = true;
11130b57cec5SDimitry Andric       }
11140b57cec5SDimitry Andric 
11150b57cec5SDimitry Andric       // A function call can always get captured noalias pointers (via other
11160b57cec5SDimitry Andric       // parameters, globals, etc.).
11170b57cec5SDimitry Andric       if (IsFuncCall && !IsArgMemOnlyCall)
11180b57cec5SDimitry Andric         CanDeriveViaCapture = true;
11190b57cec5SDimitry Andric 
11200b57cec5SDimitry Andric       // First, we want to figure out all of the sets with which we definitely
11210b57cec5SDimitry Andric       // don't alias. Iterate over all noalias set, and add those for which:
11220b57cec5SDimitry Andric       //   1. The noalias argument is not in the set of objects from which we
11230b57cec5SDimitry Andric       //      definitely derive.
11240b57cec5SDimitry Andric       //   2. The noalias argument has not yet been captured.
11250b57cec5SDimitry Andric       // An arbitrary function that might load pointers could see captured
11260b57cec5SDimitry Andric       // noalias arguments via other noalias arguments or globals, and so we
11270b57cec5SDimitry Andric       // must always check for prior capture.
11280b57cec5SDimitry Andric       for (const Argument *A : NoAliasArgs) {
11290b57cec5SDimitry Andric         if (!ObjSet.count(A) && (!CanDeriveViaCapture ||
11300b57cec5SDimitry Andric                                  // It might be tempting to skip the
11310b57cec5SDimitry Andric                                  // PointerMayBeCapturedBefore check if
11320b57cec5SDimitry Andric                                  // A->hasNoCaptureAttr() is true, but this is
11330b57cec5SDimitry Andric                                  // incorrect because nocapture only guarantees
11340b57cec5SDimitry Andric                                  // that no copies outlive the function, not
11350b57cec5SDimitry Andric                                  // that the value cannot be locally captured.
11360b57cec5SDimitry Andric                                  !PointerMayBeCapturedBefore(A,
11370b57cec5SDimitry Andric                                    /* ReturnCaptures */ false,
11380b57cec5SDimitry Andric                                    /* StoreCaptures */ false, I, &DT)))
11390b57cec5SDimitry Andric           NoAliases.push_back(NewScopes[A]);
11400b57cec5SDimitry Andric       }
11410b57cec5SDimitry Andric 
11420b57cec5SDimitry Andric       if (!NoAliases.empty())
11430b57cec5SDimitry Andric         NI->setMetadata(LLVMContext::MD_noalias,
11440b57cec5SDimitry Andric                         MDNode::concatenate(
11450b57cec5SDimitry Andric                             NI->getMetadata(LLVMContext::MD_noalias),
11460b57cec5SDimitry Andric                             MDNode::get(CalledFunc->getContext(), NoAliases)));
11470b57cec5SDimitry Andric 
11480b57cec5SDimitry Andric       // Next, we want to figure out all of the sets to which we might belong.
11490b57cec5SDimitry Andric       // We might belong to a set if the noalias argument is in the set of
11500b57cec5SDimitry Andric       // underlying objects. If there is some non-noalias argument in our list
11510b57cec5SDimitry Andric       // of underlying objects, then we cannot add a scope because the fact
11520b57cec5SDimitry Andric       // that some access does not alias with any set of our noalias arguments
11530b57cec5SDimitry Andric       // cannot itself guarantee that it does not alias with this access
11540b57cec5SDimitry Andric       // (because there is some pointer of unknown origin involved and the
11550b57cec5SDimitry Andric       // other access might also depend on this pointer). We also cannot add
11560b57cec5SDimitry Andric       // scopes to arbitrary functions unless we know they don't access any
11570b57cec5SDimitry Andric       // non-parameter pointer-values.
11580b57cec5SDimitry Andric       bool CanAddScopes = !UsesAliasingPtr;
11590b57cec5SDimitry Andric       if (CanAddScopes && IsFuncCall)
11600b57cec5SDimitry Andric         CanAddScopes = IsArgMemOnlyCall;
11610b57cec5SDimitry Andric 
11620b57cec5SDimitry Andric       if (CanAddScopes)
11630b57cec5SDimitry Andric         for (const Argument *A : NoAliasArgs) {
11640b57cec5SDimitry Andric           if (ObjSet.count(A))
11650b57cec5SDimitry Andric             Scopes.push_back(NewScopes[A]);
11660b57cec5SDimitry Andric         }
11670b57cec5SDimitry Andric 
11680b57cec5SDimitry Andric       if (!Scopes.empty())
11690b57cec5SDimitry Andric         NI->setMetadata(
11700b57cec5SDimitry Andric             LLVMContext::MD_alias_scope,
11710b57cec5SDimitry Andric             MDNode::concatenate(NI->getMetadata(LLVMContext::MD_alias_scope),
11720b57cec5SDimitry Andric                                 MDNode::get(CalledFunc->getContext(), Scopes)));
11730b57cec5SDimitry Andric     }
11740b57cec5SDimitry Andric   }
11750b57cec5SDimitry Andric }
11760b57cec5SDimitry Andric 
MayContainThrowingOrExitingCall(Instruction * Begin,Instruction * End)11775ffd83dbSDimitry Andric static bool MayContainThrowingOrExitingCall(Instruction *Begin,
11785ffd83dbSDimitry Andric                                             Instruction *End) {
11795ffd83dbSDimitry Andric 
11805ffd83dbSDimitry Andric   assert(Begin->getParent() == End->getParent() &&
11815ffd83dbSDimitry Andric          "Expected to be in same basic block!");
11825ffd83dbSDimitry Andric   unsigned NumInstChecked = 0;
11835ffd83dbSDimitry Andric   // Check that all instructions in the range [Begin, End) are guaranteed to
11845ffd83dbSDimitry Andric   // transfer execution to successor.
11855ffd83dbSDimitry Andric   for (auto &I : make_range(Begin->getIterator(), End->getIterator()))
11865ffd83dbSDimitry Andric     if (NumInstChecked++ > InlinerAttributeWindow ||
11875ffd83dbSDimitry Andric         !isGuaranteedToTransferExecutionToSuccessor(&I))
11885ffd83dbSDimitry Andric       return true;
11895ffd83dbSDimitry Andric   return false;
11905ffd83dbSDimitry Andric }
11915ffd83dbSDimitry Andric 
IdentifyValidAttributes(CallBase & CB)11925ffd83dbSDimitry Andric static AttrBuilder IdentifyValidAttributes(CallBase &CB) {
11935ffd83dbSDimitry Andric 
11945ffd83dbSDimitry Andric   AttrBuilder AB(CB.getAttributes(), AttributeList::ReturnIndex);
11955ffd83dbSDimitry Andric   if (AB.empty())
11965ffd83dbSDimitry Andric     return AB;
11975ffd83dbSDimitry Andric   AttrBuilder Valid;
11985ffd83dbSDimitry Andric   // Only allow these white listed attributes to be propagated back to the
11995ffd83dbSDimitry Andric   // callee. This is because other attributes may only be valid on the call
12005ffd83dbSDimitry Andric   // itself, i.e. attributes such as signext and zeroext.
12015ffd83dbSDimitry Andric   if (auto DerefBytes = AB.getDereferenceableBytes())
12025ffd83dbSDimitry Andric     Valid.addDereferenceableAttr(DerefBytes);
12035ffd83dbSDimitry Andric   if (auto DerefOrNullBytes = AB.getDereferenceableOrNullBytes())
12045ffd83dbSDimitry Andric     Valid.addDereferenceableOrNullAttr(DerefOrNullBytes);
12055ffd83dbSDimitry Andric   if (AB.contains(Attribute::NoAlias))
12065ffd83dbSDimitry Andric     Valid.addAttribute(Attribute::NoAlias);
12075ffd83dbSDimitry Andric   if (AB.contains(Attribute::NonNull))
12085ffd83dbSDimitry Andric     Valid.addAttribute(Attribute::NonNull);
12095ffd83dbSDimitry Andric   return Valid;
12105ffd83dbSDimitry Andric }
12115ffd83dbSDimitry Andric 
AddReturnAttributes(CallBase & CB,ValueToValueMapTy & VMap)12125ffd83dbSDimitry Andric static void AddReturnAttributes(CallBase &CB, ValueToValueMapTy &VMap) {
12135ffd83dbSDimitry Andric   if (!UpdateReturnAttributes)
12145ffd83dbSDimitry Andric     return;
12155ffd83dbSDimitry Andric 
12165ffd83dbSDimitry Andric   AttrBuilder Valid = IdentifyValidAttributes(CB);
12175ffd83dbSDimitry Andric   if (Valid.empty())
12185ffd83dbSDimitry Andric     return;
12195ffd83dbSDimitry Andric   auto *CalledFunction = CB.getCalledFunction();
12205ffd83dbSDimitry Andric   auto &Context = CalledFunction->getContext();
12215ffd83dbSDimitry Andric 
12225ffd83dbSDimitry Andric   for (auto &BB : *CalledFunction) {
12235ffd83dbSDimitry Andric     auto *RI = dyn_cast<ReturnInst>(BB.getTerminator());
12245ffd83dbSDimitry Andric     if (!RI || !isa<CallBase>(RI->getOperand(0)))
12255ffd83dbSDimitry Andric       continue;
12265ffd83dbSDimitry Andric     auto *RetVal = cast<CallBase>(RI->getOperand(0));
12275ffd83dbSDimitry Andric     // Sanity check that the cloned RetVal exists and is a call, otherwise we
12285ffd83dbSDimitry Andric     // cannot add the attributes on the cloned RetVal.
12295ffd83dbSDimitry Andric     // Simplification during inlining could have transformed the cloned
12305ffd83dbSDimitry Andric     // instruction.
12315ffd83dbSDimitry Andric     auto *NewRetVal = dyn_cast_or_null<CallBase>(VMap.lookup(RetVal));
12325ffd83dbSDimitry Andric     if (!NewRetVal)
12335ffd83dbSDimitry Andric       continue;
12345ffd83dbSDimitry Andric     // Backward propagation of attributes to the returned value may be incorrect
12355ffd83dbSDimitry Andric     // if it is control flow dependent.
12365ffd83dbSDimitry Andric     // Consider:
12375ffd83dbSDimitry Andric     // @callee {
12385ffd83dbSDimitry Andric     //  %rv = call @foo()
12395ffd83dbSDimitry Andric     //  %rv2 = call @bar()
12405ffd83dbSDimitry Andric     //  if (%rv2 != null)
12415ffd83dbSDimitry Andric     //    return %rv2
12425ffd83dbSDimitry Andric     //  if (%rv == null)
12435ffd83dbSDimitry Andric     //    exit()
12445ffd83dbSDimitry Andric     //  return %rv
12455ffd83dbSDimitry Andric     // }
12465ffd83dbSDimitry Andric     // caller() {
12475ffd83dbSDimitry Andric     //   %val = call nonnull @callee()
12485ffd83dbSDimitry Andric     // }
12495ffd83dbSDimitry Andric     // Here we cannot add the nonnull attribute on either foo or bar. So, we
12505ffd83dbSDimitry Andric     // limit the check to both RetVal and RI are in the same basic block and
12515ffd83dbSDimitry Andric     // there are no throwing/exiting instructions between these instructions.
12525ffd83dbSDimitry Andric     if (RI->getParent() != RetVal->getParent() ||
12535ffd83dbSDimitry Andric         MayContainThrowingOrExitingCall(RetVal, RI))
12545ffd83dbSDimitry Andric       continue;
12555ffd83dbSDimitry Andric     // Add to the existing attributes of NewRetVal, i.e. the cloned call
12565ffd83dbSDimitry Andric     // instruction.
12575ffd83dbSDimitry Andric     // NB! When we have the same attribute already existing on NewRetVal, but
12585ffd83dbSDimitry Andric     // with a differing value, the AttributeList's merge API honours the already
12595ffd83dbSDimitry Andric     // existing attribute value (i.e. attributes such as dereferenceable,
12605ffd83dbSDimitry Andric     // dereferenceable_or_null etc). See AttrBuilder::merge for more details.
12615ffd83dbSDimitry Andric     AttributeList AL = NewRetVal->getAttributes();
12625ffd83dbSDimitry Andric     AttributeList NewAL =
12635ffd83dbSDimitry Andric         AL.addAttributes(Context, AttributeList::ReturnIndex, Valid);
12645ffd83dbSDimitry Andric     NewRetVal->setAttributes(NewAL);
12655ffd83dbSDimitry Andric   }
12665ffd83dbSDimitry Andric }
12675ffd83dbSDimitry Andric 
12680b57cec5SDimitry Andric /// If the inlined function has non-byval align arguments, then
12690b57cec5SDimitry Andric /// add @llvm.assume-based alignment assumptions to preserve this information.
AddAlignmentAssumptions(CallBase & CB,InlineFunctionInfo & IFI)12705ffd83dbSDimitry Andric static void AddAlignmentAssumptions(CallBase &CB, InlineFunctionInfo &IFI) {
12710b57cec5SDimitry Andric   if (!PreserveAlignmentAssumptions || !IFI.GetAssumptionCache)
12720b57cec5SDimitry Andric     return;
12730b57cec5SDimitry Andric 
12745ffd83dbSDimitry Andric   AssumptionCache *AC = &IFI.GetAssumptionCache(*CB.getCaller());
12755ffd83dbSDimitry Andric   auto &DL = CB.getCaller()->getParent()->getDataLayout();
12760b57cec5SDimitry Andric 
12770b57cec5SDimitry Andric   // To avoid inserting redundant assumptions, we should check for assumptions
12780b57cec5SDimitry Andric   // already in the caller. To do this, we might need a DT of the caller.
12790b57cec5SDimitry Andric   DominatorTree DT;
12800b57cec5SDimitry Andric   bool DTCalculated = false;
12810b57cec5SDimitry Andric 
12825ffd83dbSDimitry Andric   Function *CalledFunc = CB.getCalledFunction();
12830b57cec5SDimitry Andric   for (Argument &Arg : CalledFunc->args()) {
12840b57cec5SDimitry Andric     unsigned Align = Arg.getType()->isPointerTy() ? Arg.getParamAlignment() : 0;
1285af732203SDimitry Andric     if (Align && !Arg.hasPassPointeeByValueCopyAttr() && !Arg.hasNUses(0)) {
12860b57cec5SDimitry Andric       if (!DTCalculated) {
12875ffd83dbSDimitry Andric         DT.recalculate(*CB.getCaller());
12880b57cec5SDimitry Andric         DTCalculated = true;
12890b57cec5SDimitry Andric       }
12900b57cec5SDimitry Andric 
12910b57cec5SDimitry Andric       // If we can already prove the asserted alignment in the context of the
12920b57cec5SDimitry Andric       // caller, then don't bother inserting the assumption.
12935ffd83dbSDimitry Andric       Value *ArgVal = CB.getArgOperand(Arg.getArgNo());
12945ffd83dbSDimitry Andric       if (getKnownAlignment(ArgVal, DL, &CB, AC, &DT) >= Align)
12950b57cec5SDimitry Andric         continue;
12960b57cec5SDimitry Andric 
12975ffd83dbSDimitry Andric       CallInst *NewAsmp =
12985ffd83dbSDimitry Andric           IRBuilder<>(&CB).CreateAlignmentAssumption(DL, ArgVal, Align);
1299*5f7ddb14SDimitry Andric       AC->registerAssumption(cast<AssumeInst>(NewAsmp));
13000b57cec5SDimitry Andric     }
13010b57cec5SDimitry Andric   }
13020b57cec5SDimitry Andric }
13030b57cec5SDimitry Andric 
13040b57cec5SDimitry Andric /// Once we have cloned code over from a callee into the caller,
13050b57cec5SDimitry Andric /// update the specified callgraph to reflect the changes we made.
13060b57cec5SDimitry Andric /// Note that it's possible that not all code was copied over, so only
13070b57cec5SDimitry Andric /// some edges of the callgraph may remain.
UpdateCallGraphAfterInlining(CallBase & CB,Function::iterator FirstNewBlock,ValueToValueMapTy & VMap,InlineFunctionInfo & IFI)13085ffd83dbSDimitry Andric static void UpdateCallGraphAfterInlining(CallBase &CB,
13090b57cec5SDimitry Andric                                          Function::iterator FirstNewBlock,
13100b57cec5SDimitry Andric                                          ValueToValueMapTy &VMap,
13110b57cec5SDimitry Andric                                          InlineFunctionInfo &IFI) {
13120b57cec5SDimitry Andric   CallGraph &CG = *IFI.CG;
13135ffd83dbSDimitry Andric   const Function *Caller = CB.getCaller();
13145ffd83dbSDimitry Andric   const Function *Callee = CB.getCalledFunction();
13150b57cec5SDimitry Andric   CallGraphNode *CalleeNode = CG[Callee];
13160b57cec5SDimitry Andric   CallGraphNode *CallerNode = CG[Caller];
13170b57cec5SDimitry Andric 
13180b57cec5SDimitry Andric   // Since we inlined some uninlined call sites in the callee into the caller,
13190b57cec5SDimitry Andric   // add edges from the caller to all of the callees of the callee.
13200b57cec5SDimitry Andric   CallGraphNode::iterator I = CalleeNode->begin(), E = CalleeNode->end();
13210b57cec5SDimitry Andric 
13220b57cec5SDimitry Andric   // Consider the case where CalleeNode == CallerNode.
13230b57cec5SDimitry Andric   CallGraphNode::CalledFunctionsVector CallCache;
13240b57cec5SDimitry Andric   if (CalleeNode == CallerNode) {
13250b57cec5SDimitry Andric     CallCache.assign(I, E);
13260b57cec5SDimitry Andric     I = CallCache.begin();
13270b57cec5SDimitry Andric     E = CallCache.end();
13280b57cec5SDimitry Andric   }
13290b57cec5SDimitry Andric 
13300b57cec5SDimitry Andric   for (; I != E; ++I) {
13315ffd83dbSDimitry Andric     // Skip 'refererence' call records.
13325ffd83dbSDimitry Andric     if (!I->first)
13335ffd83dbSDimitry Andric       continue;
13345ffd83dbSDimitry Andric 
13355ffd83dbSDimitry Andric     const Value *OrigCall = *I->first;
13360b57cec5SDimitry Andric 
13370b57cec5SDimitry Andric     ValueToValueMapTy::iterator VMI = VMap.find(OrigCall);
13380b57cec5SDimitry Andric     // Only copy the edge if the call was inlined!
13390b57cec5SDimitry Andric     if (VMI == VMap.end() || VMI->second == nullptr)
13400b57cec5SDimitry Andric       continue;
13410b57cec5SDimitry Andric 
13420b57cec5SDimitry Andric     // If the call was inlined, but then constant folded, there is no edge to
13430b57cec5SDimitry Andric     // add.  Check for this case.
13440b57cec5SDimitry Andric     auto *NewCall = dyn_cast<CallBase>(VMI->second);
13450b57cec5SDimitry Andric     if (!NewCall)
13460b57cec5SDimitry Andric       continue;
13470b57cec5SDimitry Andric 
13480b57cec5SDimitry Andric     // We do not treat intrinsic calls like real function calls because we
13490b57cec5SDimitry Andric     // expect them to become inline code; do not add an edge for an intrinsic.
13500b57cec5SDimitry Andric     if (NewCall->getCalledFunction() &&
13510b57cec5SDimitry Andric         NewCall->getCalledFunction()->isIntrinsic())
13520b57cec5SDimitry Andric       continue;
13530b57cec5SDimitry Andric 
13540b57cec5SDimitry Andric     // Remember that this call site got inlined for the client of
13550b57cec5SDimitry Andric     // InlineFunction.
13560b57cec5SDimitry Andric     IFI.InlinedCalls.push_back(NewCall);
13570b57cec5SDimitry Andric 
13580b57cec5SDimitry Andric     // It's possible that inlining the callsite will cause it to go from an
13590b57cec5SDimitry Andric     // indirect to a direct call by resolving a function pointer.  If this
13600b57cec5SDimitry Andric     // happens, set the callee of the new call site to a more precise
13610b57cec5SDimitry Andric     // destination.  This can also happen if the call graph node of the caller
13620b57cec5SDimitry Andric     // was just unnecessarily imprecise.
13630b57cec5SDimitry Andric     if (!I->second->getFunction())
13640b57cec5SDimitry Andric       if (Function *F = NewCall->getCalledFunction()) {
13650b57cec5SDimitry Andric         // Indirect call site resolved to direct call.
13660b57cec5SDimitry Andric         CallerNode->addCalledFunction(NewCall, CG[F]);
13670b57cec5SDimitry Andric 
13680b57cec5SDimitry Andric         continue;
13690b57cec5SDimitry Andric       }
13700b57cec5SDimitry Andric 
13710b57cec5SDimitry Andric     CallerNode->addCalledFunction(NewCall, I->second);
13720b57cec5SDimitry Andric   }
13730b57cec5SDimitry Andric 
13740b57cec5SDimitry Andric   // Update the call graph by deleting the edge from Callee to Caller.  We must
13750b57cec5SDimitry Andric   // do this after the loop above in case Caller and Callee are the same.
13765ffd83dbSDimitry Andric   CallerNode->removeCallEdgeFor(*cast<CallBase>(&CB));
13770b57cec5SDimitry Andric }
13780b57cec5SDimitry Andric 
HandleByValArgumentInit(Value * Dst,Value * Src,Module * M,BasicBlock * InsertBlock,InlineFunctionInfo & IFI)13790b57cec5SDimitry Andric static void HandleByValArgumentInit(Value *Dst, Value *Src, Module *M,
13800b57cec5SDimitry Andric                                     BasicBlock *InsertBlock,
13810b57cec5SDimitry Andric                                     InlineFunctionInfo &IFI) {
13820b57cec5SDimitry Andric   Type *AggTy = cast<PointerType>(Src->getType())->getElementType();
13830b57cec5SDimitry Andric   IRBuilder<> Builder(InsertBlock, InsertBlock->begin());
13840b57cec5SDimitry Andric 
13850b57cec5SDimitry Andric   Value *Size = Builder.getInt64(M->getDataLayout().getTypeStoreSize(AggTy));
13860b57cec5SDimitry Andric 
13870b57cec5SDimitry Andric   // Always generate a memcpy of alignment 1 here because we don't know
13880b57cec5SDimitry Andric   // the alignment of the src pointer.  Other optimizations can infer
13890b57cec5SDimitry Andric   // better alignment.
13905ffd83dbSDimitry Andric   Builder.CreateMemCpy(Dst, /*DstAlign*/ Align(1), Src,
13915ffd83dbSDimitry Andric                        /*SrcAlign*/ Align(1), Size);
13920b57cec5SDimitry Andric }
13930b57cec5SDimitry Andric 
13940b57cec5SDimitry Andric /// When inlining a call site that has a byval argument,
13950b57cec5SDimitry Andric /// we have to make the implicit memcpy explicit by adding it.
HandleByValArgument(Value * Arg,Instruction * TheCall,const Function * CalledFunc,InlineFunctionInfo & IFI,unsigned ByValAlignment)13960b57cec5SDimitry Andric static Value *HandleByValArgument(Value *Arg, Instruction *TheCall,
13970b57cec5SDimitry Andric                                   const Function *CalledFunc,
13980b57cec5SDimitry Andric                                   InlineFunctionInfo &IFI,
13990b57cec5SDimitry Andric                                   unsigned ByValAlignment) {
14000b57cec5SDimitry Andric   PointerType *ArgTy = cast<PointerType>(Arg->getType());
14010b57cec5SDimitry Andric   Type *AggTy = ArgTy->getElementType();
14020b57cec5SDimitry Andric 
14030b57cec5SDimitry Andric   Function *Caller = TheCall->getFunction();
14040b57cec5SDimitry Andric   const DataLayout &DL = Caller->getParent()->getDataLayout();
14050b57cec5SDimitry Andric 
14060b57cec5SDimitry Andric   // If the called function is readonly, then it could not mutate the caller's
14070b57cec5SDimitry Andric   // copy of the byval'd memory.  In this case, it is safe to elide the copy and
14080b57cec5SDimitry Andric   // temporary.
14090b57cec5SDimitry Andric   if (CalledFunc->onlyReadsMemory()) {
14100b57cec5SDimitry Andric     // If the byval argument has a specified alignment that is greater than the
14110b57cec5SDimitry Andric     // passed in pointer, then we either have to round up the input pointer or
14120b57cec5SDimitry Andric     // give up on this transformation.
14130b57cec5SDimitry Andric     if (ByValAlignment <= 1)  // 0 = unspecified, 1 = no particular alignment.
14140b57cec5SDimitry Andric       return Arg;
14150b57cec5SDimitry Andric 
14160b57cec5SDimitry Andric     AssumptionCache *AC =
14175ffd83dbSDimitry Andric         IFI.GetAssumptionCache ? &IFI.GetAssumptionCache(*Caller) : nullptr;
14180b57cec5SDimitry Andric 
14190b57cec5SDimitry Andric     // If the pointer is already known to be sufficiently aligned, or if we can
14200b57cec5SDimitry Andric     // round it up to a larger alignment, then we don't need a temporary.
14215ffd83dbSDimitry Andric     if (getOrEnforceKnownAlignment(Arg, Align(ByValAlignment), DL, TheCall,
14225ffd83dbSDimitry Andric                                    AC) >= ByValAlignment)
14230b57cec5SDimitry Andric       return Arg;
14240b57cec5SDimitry Andric 
14250b57cec5SDimitry Andric     // Otherwise, we have to make a memcpy to get a safe alignment.  This is bad
14260b57cec5SDimitry Andric     // for code quality, but rarely happens and is required for correctness.
14270b57cec5SDimitry Andric   }
14280b57cec5SDimitry Andric 
14290b57cec5SDimitry Andric   // Create the alloca.  If we have DataLayout, use nice alignment.
1430480093f4SDimitry Andric   Align Alignment(DL.getPrefTypeAlignment(AggTy));
14310b57cec5SDimitry Andric 
14320b57cec5SDimitry Andric   // If the byval had an alignment specified, we *must* use at least that
14330b57cec5SDimitry Andric   // alignment, as it is required by the byval argument (and uses of the
14340b57cec5SDimitry Andric   // pointer inside the callee).
1435480093f4SDimitry Andric   Alignment = max(Alignment, MaybeAlign(ByValAlignment));
14360b57cec5SDimitry Andric 
1437480093f4SDimitry Andric   Value *NewAlloca =
1438480093f4SDimitry Andric       new AllocaInst(AggTy, DL.getAllocaAddrSpace(), nullptr, Alignment,
1439480093f4SDimitry Andric                      Arg->getName(), &*Caller->begin()->begin());
14400b57cec5SDimitry Andric   IFI.StaticAllocas.push_back(cast<AllocaInst>(NewAlloca));
14410b57cec5SDimitry Andric 
14420b57cec5SDimitry Andric   // Uses of the argument in the function should use our new alloca
14430b57cec5SDimitry Andric   // instead.
14440b57cec5SDimitry Andric   return NewAlloca;
14450b57cec5SDimitry Andric }
14460b57cec5SDimitry Andric 
14470b57cec5SDimitry Andric // Check whether this Value is used by a lifetime intrinsic.
isUsedByLifetimeMarker(Value * V)14480b57cec5SDimitry Andric static bool isUsedByLifetimeMarker(Value *V) {
14490b57cec5SDimitry Andric   for (User *U : V->users())
14500b57cec5SDimitry Andric     if (IntrinsicInst *II = dyn_cast<IntrinsicInst>(U))
14510b57cec5SDimitry Andric       if (II->isLifetimeStartOrEnd())
14520b57cec5SDimitry Andric         return true;
14530b57cec5SDimitry Andric   return false;
14540b57cec5SDimitry Andric }
14550b57cec5SDimitry Andric 
14560b57cec5SDimitry Andric // Check whether the given alloca already has
14570b57cec5SDimitry Andric // lifetime.start or lifetime.end intrinsics.
hasLifetimeMarkers(AllocaInst * AI)14580b57cec5SDimitry Andric static bool hasLifetimeMarkers(AllocaInst *AI) {
14590b57cec5SDimitry Andric   Type *Ty = AI->getType();
14600b57cec5SDimitry Andric   Type *Int8PtrTy = Type::getInt8PtrTy(Ty->getContext(),
14610b57cec5SDimitry Andric                                        Ty->getPointerAddressSpace());
14620b57cec5SDimitry Andric   if (Ty == Int8PtrTy)
14630b57cec5SDimitry Andric     return isUsedByLifetimeMarker(AI);
14640b57cec5SDimitry Andric 
14650b57cec5SDimitry Andric   // Do a scan to find all the casts to i8*.
14660b57cec5SDimitry Andric   for (User *U : AI->users()) {
14670b57cec5SDimitry Andric     if (U->getType() != Int8PtrTy) continue;
14680b57cec5SDimitry Andric     if (U->stripPointerCasts() != AI) continue;
14690b57cec5SDimitry Andric     if (isUsedByLifetimeMarker(U))
14700b57cec5SDimitry Andric       return true;
14710b57cec5SDimitry Andric   }
14720b57cec5SDimitry Andric   return false;
14730b57cec5SDimitry Andric }
14740b57cec5SDimitry Andric 
14750b57cec5SDimitry Andric /// Return the result of AI->isStaticAlloca() if AI were moved to the entry
14760b57cec5SDimitry Andric /// block. Allocas used in inalloca calls and allocas of dynamic array size
14770b57cec5SDimitry Andric /// cannot be static.
allocaWouldBeStaticInEntry(const AllocaInst * AI)14780b57cec5SDimitry Andric static bool allocaWouldBeStaticInEntry(const AllocaInst *AI ) {
14790b57cec5SDimitry Andric   return isa<Constant>(AI->getArraySize()) && !AI->isUsedWithInAlloca();
14800b57cec5SDimitry Andric }
14810b57cec5SDimitry Andric 
14820b57cec5SDimitry Andric /// Returns a DebugLoc for a new DILocation which is a clone of \p OrigDL
14830b57cec5SDimitry Andric /// inlined at \p InlinedAt. \p IANodes is an inlined-at cache.
inlineDebugLoc(DebugLoc OrigDL,DILocation * InlinedAt,LLVMContext & Ctx,DenseMap<const MDNode *,MDNode * > & IANodes)14840b57cec5SDimitry Andric static DebugLoc inlineDebugLoc(DebugLoc OrigDL, DILocation *InlinedAt,
14850b57cec5SDimitry Andric                                LLVMContext &Ctx,
14860b57cec5SDimitry Andric                                DenseMap<const MDNode *, MDNode *> &IANodes) {
14870b57cec5SDimitry Andric   auto IA = DebugLoc::appendInlinedAt(OrigDL, InlinedAt, Ctx, IANodes);
1488af732203SDimitry Andric   return DILocation::get(Ctx, OrigDL.getLine(), OrigDL.getCol(),
1489af732203SDimitry Andric                          OrigDL.getScope(), IA);
14900b57cec5SDimitry Andric }
14910b57cec5SDimitry Andric 
14920b57cec5SDimitry Andric /// Update inlined instructions' line numbers to
14930b57cec5SDimitry Andric /// to encode location where these instructions are inlined.
fixupLineNumbers(Function * Fn,Function::iterator FI,Instruction * TheCall,bool CalleeHasDebugInfo)14940b57cec5SDimitry Andric static void fixupLineNumbers(Function *Fn, Function::iterator FI,
14950b57cec5SDimitry Andric                              Instruction *TheCall, bool CalleeHasDebugInfo) {
14960b57cec5SDimitry Andric   const DebugLoc &TheCallDL = TheCall->getDebugLoc();
14970b57cec5SDimitry Andric   if (!TheCallDL)
14980b57cec5SDimitry Andric     return;
14990b57cec5SDimitry Andric 
15000b57cec5SDimitry Andric   auto &Ctx = Fn->getContext();
15010b57cec5SDimitry Andric   DILocation *InlinedAtNode = TheCallDL;
15020b57cec5SDimitry Andric 
15030b57cec5SDimitry Andric   // Create a unique call site, not to be confused with any other call from the
15040b57cec5SDimitry Andric   // same location.
15050b57cec5SDimitry Andric   InlinedAtNode = DILocation::getDistinct(
15060b57cec5SDimitry Andric       Ctx, InlinedAtNode->getLine(), InlinedAtNode->getColumn(),
15070b57cec5SDimitry Andric       InlinedAtNode->getScope(), InlinedAtNode->getInlinedAt());
15080b57cec5SDimitry Andric 
15090b57cec5SDimitry Andric   // Cache the inlined-at nodes as they're built so they are reused, without
15100b57cec5SDimitry Andric   // this every instruction's inlined-at chain would become distinct from each
15110b57cec5SDimitry Andric   // other.
15120b57cec5SDimitry Andric   DenseMap<const MDNode *, MDNode *> IANodes;
15130b57cec5SDimitry Andric 
1514480093f4SDimitry Andric   // Check if we are not generating inline line tables and want to use
1515480093f4SDimitry Andric   // the call site location instead.
1516480093f4SDimitry Andric   bool NoInlineLineTables = Fn->hasFnAttribute("no-inline-line-tables");
1517480093f4SDimitry Andric 
15180b57cec5SDimitry Andric   for (; FI != Fn->end(); ++FI) {
15190b57cec5SDimitry Andric     for (BasicBlock::iterator BI = FI->begin(), BE = FI->end();
15200b57cec5SDimitry Andric          BI != BE; ++BI) {
15210b57cec5SDimitry Andric       // Loop metadata needs to be updated so that the start and end locs
15220b57cec5SDimitry Andric       // reference inlined-at locations.
1523*5f7ddb14SDimitry Andric       auto updateLoopInfoLoc = [&Ctx, &InlinedAtNode,
1524*5f7ddb14SDimitry Andric                                 &IANodes](Metadata *MD) -> Metadata * {
1525*5f7ddb14SDimitry Andric         if (auto *Loc = dyn_cast_or_null<DILocation>(MD))
1526*5f7ddb14SDimitry Andric           return inlineDebugLoc(Loc, InlinedAtNode, Ctx, IANodes).get();
1527*5f7ddb14SDimitry Andric         return MD;
15285ffd83dbSDimitry Andric       };
15295ffd83dbSDimitry Andric       updateLoopMetadataDebugLocations(*BI, updateLoopInfoLoc);
15300b57cec5SDimitry Andric 
1531480093f4SDimitry Andric       if (!NoInlineLineTables)
15320b57cec5SDimitry Andric         if (DebugLoc DL = BI->getDebugLoc()) {
15330b57cec5SDimitry Andric           DebugLoc IDL =
15340b57cec5SDimitry Andric               inlineDebugLoc(DL, InlinedAtNode, BI->getContext(), IANodes);
15350b57cec5SDimitry Andric           BI->setDebugLoc(IDL);
15360b57cec5SDimitry Andric           continue;
15370b57cec5SDimitry Andric         }
15380b57cec5SDimitry Andric 
1539480093f4SDimitry Andric       if (CalleeHasDebugInfo && !NoInlineLineTables)
15400b57cec5SDimitry Andric         continue;
15410b57cec5SDimitry Andric 
1542480093f4SDimitry Andric       // If the inlined instruction has no line number, or if inline info
1543480093f4SDimitry Andric       // is not being generated, make it look as if it originates from the call
1544480093f4SDimitry Andric       // location. This is important for ((__always_inline, __nodebug__))
1545480093f4SDimitry Andric       // functions which must use caller location for all instructions in their
1546480093f4SDimitry Andric       // function body.
15470b57cec5SDimitry Andric 
15480b57cec5SDimitry Andric       // Don't update static allocas, as they may get moved later.
15490b57cec5SDimitry Andric       if (auto *AI = dyn_cast<AllocaInst>(BI))
15500b57cec5SDimitry Andric         if (allocaWouldBeStaticInEntry(AI))
15510b57cec5SDimitry Andric           continue;
15520b57cec5SDimitry Andric 
15530b57cec5SDimitry Andric       BI->setDebugLoc(TheCallDL);
15540b57cec5SDimitry Andric     }
1555480093f4SDimitry Andric 
1556480093f4SDimitry Andric     // Remove debug info intrinsics if we're not keeping inline info.
1557480093f4SDimitry Andric     if (NoInlineLineTables) {
1558480093f4SDimitry Andric       BasicBlock::iterator BI = FI->begin();
1559480093f4SDimitry Andric       while (BI != FI->end()) {
1560480093f4SDimitry Andric         if (isa<DbgInfoIntrinsic>(BI)) {
1561480093f4SDimitry Andric           BI = BI->eraseFromParent();
1562480093f4SDimitry Andric           continue;
1563480093f4SDimitry Andric         }
1564480093f4SDimitry Andric         ++BI;
1565480093f4SDimitry Andric       }
1566480093f4SDimitry Andric     }
1567480093f4SDimitry Andric 
15680b57cec5SDimitry Andric   }
15690b57cec5SDimitry Andric }
15700b57cec5SDimitry Andric 
15710b57cec5SDimitry Andric /// Update the block frequencies of the caller after a callee has been inlined.
15720b57cec5SDimitry Andric ///
15730b57cec5SDimitry Andric /// Each block cloned into the caller has its block frequency scaled by the
15740b57cec5SDimitry Andric /// ratio of CallSiteFreq/CalleeEntryFreq. This ensures that the cloned copy of
15750b57cec5SDimitry Andric /// callee's entry block gets the same frequency as the callsite block and the
15760b57cec5SDimitry Andric /// relative frequencies of all cloned blocks remain the same after cloning.
updateCallerBFI(BasicBlock * CallSiteBlock,const ValueToValueMapTy & VMap,BlockFrequencyInfo * CallerBFI,BlockFrequencyInfo * CalleeBFI,const BasicBlock & CalleeEntryBlock)15770b57cec5SDimitry Andric static void updateCallerBFI(BasicBlock *CallSiteBlock,
15780b57cec5SDimitry Andric                             const ValueToValueMapTy &VMap,
15790b57cec5SDimitry Andric                             BlockFrequencyInfo *CallerBFI,
15800b57cec5SDimitry Andric                             BlockFrequencyInfo *CalleeBFI,
15810b57cec5SDimitry Andric                             const BasicBlock &CalleeEntryBlock) {
15820b57cec5SDimitry Andric   SmallPtrSet<BasicBlock *, 16> ClonedBBs;
1583480093f4SDimitry Andric   for (auto Entry : VMap) {
15840b57cec5SDimitry Andric     if (!isa<BasicBlock>(Entry.first) || !Entry.second)
15850b57cec5SDimitry Andric       continue;
15860b57cec5SDimitry Andric     auto *OrigBB = cast<BasicBlock>(Entry.first);
15870b57cec5SDimitry Andric     auto *ClonedBB = cast<BasicBlock>(Entry.second);
15880b57cec5SDimitry Andric     uint64_t Freq = CalleeBFI->getBlockFreq(OrigBB).getFrequency();
15890b57cec5SDimitry Andric     if (!ClonedBBs.insert(ClonedBB).second) {
15900b57cec5SDimitry Andric       // Multiple blocks in the callee might get mapped to one cloned block in
15910b57cec5SDimitry Andric       // the caller since we prune the callee as we clone it. When that happens,
15920b57cec5SDimitry Andric       // we want to use the maximum among the original blocks' frequencies.
15930b57cec5SDimitry Andric       uint64_t NewFreq = CallerBFI->getBlockFreq(ClonedBB).getFrequency();
15940b57cec5SDimitry Andric       if (NewFreq > Freq)
15950b57cec5SDimitry Andric         Freq = NewFreq;
15960b57cec5SDimitry Andric     }
15970b57cec5SDimitry Andric     CallerBFI->setBlockFreq(ClonedBB, Freq);
15980b57cec5SDimitry Andric   }
15990b57cec5SDimitry Andric   BasicBlock *EntryClone = cast<BasicBlock>(VMap.lookup(&CalleeEntryBlock));
16000b57cec5SDimitry Andric   CallerBFI->setBlockFreqAndScale(
16010b57cec5SDimitry Andric       EntryClone, CallerBFI->getBlockFreq(CallSiteBlock).getFrequency(),
16020b57cec5SDimitry Andric       ClonedBBs);
16030b57cec5SDimitry Andric }
16040b57cec5SDimitry Andric 
16050b57cec5SDimitry Andric /// Update the branch metadata for cloned call instructions.
updateCallProfile(Function * Callee,const ValueToValueMapTy & VMap,const ProfileCount & CalleeEntryCount,const CallBase & TheCall,ProfileSummaryInfo * PSI,BlockFrequencyInfo * CallerBFI)16060b57cec5SDimitry Andric static void updateCallProfile(Function *Callee, const ValueToValueMapTy &VMap,
16070b57cec5SDimitry Andric                               const ProfileCount &CalleeEntryCount,
16085ffd83dbSDimitry Andric                               const CallBase &TheCall, ProfileSummaryInfo *PSI,
16090b57cec5SDimitry Andric                               BlockFrequencyInfo *CallerBFI) {
16100b57cec5SDimitry Andric   if (!CalleeEntryCount.hasValue() || CalleeEntryCount.isSynthetic() ||
16110b57cec5SDimitry Andric       CalleeEntryCount.getCount() < 1)
16120b57cec5SDimitry Andric     return;
16130b57cec5SDimitry Andric   auto CallSiteCount = PSI ? PSI->getProfileCount(TheCall, CallerBFI) : None;
16140b57cec5SDimitry Andric   int64_t CallCount =
1615af732203SDimitry Andric       std::min(CallSiteCount.getValueOr(0), CalleeEntryCount.getCount());
16160b57cec5SDimitry Andric   updateProfileCallee(Callee, -CallCount, &VMap);
16170b57cec5SDimitry Andric }
16180b57cec5SDimitry Andric 
updateProfileCallee(Function * Callee,int64_t entryDelta,const ValueMap<const Value *,WeakTrackingVH> * VMap)16190b57cec5SDimitry Andric void llvm::updateProfileCallee(
16200b57cec5SDimitry Andric     Function *Callee, int64_t entryDelta,
16210b57cec5SDimitry Andric     const ValueMap<const Value *, WeakTrackingVH> *VMap) {
16220b57cec5SDimitry Andric   auto CalleeCount = Callee->getEntryCount();
16230b57cec5SDimitry Andric   if (!CalleeCount.hasValue())
16240b57cec5SDimitry Andric     return;
16250b57cec5SDimitry Andric 
16260b57cec5SDimitry Andric   uint64_t priorEntryCount = CalleeCount.getCount();
16270b57cec5SDimitry Andric   uint64_t newEntryCount;
16280b57cec5SDimitry Andric 
16290b57cec5SDimitry Andric   // Since CallSiteCount is an estimate, it could exceed the original callee
16300b57cec5SDimitry Andric   // count and has to be set to 0 so guard against underflow.
16310b57cec5SDimitry Andric   if (entryDelta < 0 && static_cast<uint64_t>(-entryDelta) > priorEntryCount)
16320b57cec5SDimitry Andric     newEntryCount = 0;
16330b57cec5SDimitry Andric   else
16340b57cec5SDimitry Andric     newEntryCount = priorEntryCount + entryDelta;
16350b57cec5SDimitry Andric 
16360b57cec5SDimitry Andric   // During inlining ?
16370b57cec5SDimitry Andric   if (VMap) {
16380b57cec5SDimitry Andric     uint64_t cloneEntryCount = priorEntryCount - newEntryCount;
1639480093f4SDimitry Andric     for (auto Entry : *VMap)
16400b57cec5SDimitry Andric       if (isa<CallInst>(Entry.first))
16410b57cec5SDimitry Andric         if (auto *CI = dyn_cast_or_null<CallInst>(Entry.second))
16420b57cec5SDimitry Andric           CI->updateProfWeight(cloneEntryCount, priorEntryCount);
16430b57cec5SDimitry Andric   }
1644480093f4SDimitry Andric 
1645480093f4SDimitry Andric   if (entryDelta) {
1646480093f4SDimitry Andric     Callee->setEntryCount(newEntryCount);
1647480093f4SDimitry Andric 
16480b57cec5SDimitry Andric     for (BasicBlock &BB : *Callee)
16490b57cec5SDimitry Andric       // No need to update the callsite if it is pruned during inlining.
16500b57cec5SDimitry Andric       if (!VMap || VMap->count(&BB))
16510b57cec5SDimitry Andric         for (Instruction &I : BB)
16520b57cec5SDimitry Andric           if (CallInst *CI = dyn_cast<CallInst>(&I))
16530b57cec5SDimitry Andric             CI->updateProfWeight(newEntryCount, priorEntryCount);
16540b57cec5SDimitry Andric   }
1655480093f4SDimitry Andric }
16560b57cec5SDimitry Andric 
1657*5f7ddb14SDimitry Andric /// An operand bundle "clang.arc.attachedcall" on a call indicates the call
1658*5f7ddb14SDimitry Andric /// result is implicitly consumed by a call to retainRV or claimRV immediately
1659*5f7ddb14SDimitry Andric /// after the call. This function inlines the retainRV/claimRV calls.
1660*5f7ddb14SDimitry Andric ///
1661*5f7ddb14SDimitry Andric /// There are three cases to consider:
1662*5f7ddb14SDimitry Andric ///
1663*5f7ddb14SDimitry Andric /// 1. If there is a call to autoreleaseRV that takes a pointer to the returned
1664*5f7ddb14SDimitry Andric ///    object in the callee return block, the autoreleaseRV call and the
1665*5f7ddb14SDimitry Andric ///    retainRV/claimRV call in the caller cancel out. If the call in the caller
1666*5f7ddb14SDimitry Andric ///    is a claimRV call, a call to objc_release is emitted.
1667*5f7ddb14SDimitry Andric ///
1668*5f7ddb14SDimitry Andric /// 2. If there is a call in the callee return block that doesn't have operand
1669*5f7ddb14SDimitry Andric ///    bundle "clang.arc.attachedcall", the operand bundle on the original call
1670*5f7ddb14SDimitry Andric ///    is transferred to the call in the callee.
1671*5f7ddb14SDimitry Andric ///
1672*5f7ddb14SDimitry Andric /// 3. Otherwise, a call to objc_retain is inserted if the call in the caller is
1673*5f7ddb14SDimitry Andric ///    a retainRV call.
1674*5f7ddb14SDimitry Andric static void
inlineRetainOrClaimRVCalls(CallBase & CB,const SmallVectorImpl<ReturnInst * > & Returns)1675*5f7ddb14SDimitry Andric inlineRetainOrClaimRVCalls(CallBase &CB,
1676*5f7ddb14SDimitry Andric                            const SmallVectorImpl<ReturnInst *> &Returns) {
1677*5f7ddb14SDimitry Andric   Module *Mod = CB.getModule();
1678*5f7ddb14SDimitry Andric   bool IsRetainRV = objcarc::hasAttachedCallOpBundle(&CB, true),
1679*5f7ddb14SDimitry Andric        IsClaimRV = !IsRetainRV;
1680*5f7ddb14SDimitry Andric 
1681*5f7ddb14SDimitry Andric   for (auto *RI : Returns) {
1682*5f7ddb14SDimitry Andric     Value *RetOpnd = objcarc::GetRCIdentityRoot(RI->getOperand(0));
1683*5f7ddb14SDimitry Andric     BasicBlock::reverse_iterator I = ++(RI->getIterator().getReverse());
1684*5f7ddb14SDimitry Andric     BasicBlock::reverse_iterator EI = RI->getParent()->rend();
1685*5f7ddb14SDimitry Andric     bool InsertRetainCall = IsRetainRV;
1686*5f7ddb14SDimitry Andric     IRBuilder<> Builder(RI->getContext());
1687*5f7ddb14SDimitry Andric 
1688*5f7ddb14SDimitry Andric     // Walk backwards through the basic block looking for either a matching
1689*5f7ddb14SDimitry Andric     // autoreleaseRV call or an unannotated call.
1690*5f7ddb14SDimitry Andric     for (; I != EI;) {
1691*5f7ddb14SDimitry Andric       auto CurI = I++;
1692*5f7ddb14SDimitry Andric 
1693*5f7ddb14SDimitry Andric       // Ignore casts.
1694*5f7ddb14SDimitry Andric       if (isa<CastInst>(*CurI))
1695*5f7ddb14SDimitry Andric         continue;
1696*5f7ddb14SDimitry Andric 
1697*5f7ddb14SDimitry Andric       if (auto *II = dyn_cast<IntrinsicInst>(&*CurI)) {
1698*5f7ddb14SDimitry Andric         if (II->getIntrinsicID() == Intrinsic::objc_autoreleaseReturnValue &&
1699*5f7ddb14SDimitry Andric             II->hasNUses(0) &&
1700*5f7ddb14SDimitry Andric             objcarc::GetRCIdentityRoot(II->getOperand(0)) == RetOpnd) {
1701*5f7ddb14SDimitry Andric           // If we've found a matching authoreleaseRV call:
1702*5f7ddb14SDimitry Andric           // - If claimRV is attached to the call, insert a call to objc_release
1703*5f7ddb14SDimitry Andric           //   and erase the autoreleaseRV call.
1704*5f7ddb14SDimitry Andric           // - If retainRV is attached to the call, just erase the autoreleaseRV
1705*5f7ddb14SDimitry Andric           //   call.
1706*5f7ddb14SDimitry Andric           if (IsClaimRV) {
1707*5f7ddb14SDimitry Andric             Builder.SetInsertPoint(II);
1708*5f7ddb14SDimitry Andric             Function *IFn =
1709*5f7ddb14SDimitry Andric                 Intrinsic::getDeclaration(Mod, Intrinsic::objc_release);
1710*5f7ddb14SDimitry Andric             Value *BC =
1711*5f7ddb14SDimitry Andric                 Builder.CreateBitCast(RetOpnd, IFn->getArg(0)->getType());
1712*5f7ddb14SDimitry Andric             Builder.CreateCall(IFn, BC, "");
1713*5f7ddb14SDimitry Andric           }
1714*5f7ddb14SDimitry Andric           II->eraseFromParent();
1715*5f7ddb14SDimitry Andric           InsertRetainCall = false;
1716*5f7ddb14SDimitry Andric         }
1717*5f7ddb14SDimitry Andric       } else if (auto *CI = dyn_cast<CallInst>(&*CurI)) {
1718*5f7ddb14SDimitry Andric         if (objcarc::GetRCIdentityRoot(CI) == RetOpnd &&
1719*5f7ddb14SDimitry Andric             !objcarc::hasAttachedCallOpBundle(CI)) {
1720*5f7ddb14SDimitry Andric           // If we've found an unannotated call that defines RetOpnd, add a
1721*5f7ddb14SDimitry Andric           // "clang.arc.attachedcall" operand bundle.
1722*5f7ddb14SDimitry Andric           Value *BundleArgs[] = {ConstantInt::get(
1723*5f7ddb14SDimitry Andric               Builder.getInt64Ty(),
1724*5f7ddb14SDimitry Andric               objcarc::getAttachedCallOperandBundleEnum(IsRetainRV))};
1725*5f7ddb14SDimitry Andric           OperandBundleDef OB("clang.arc.attachedcall", BundleArgs);
1726*5f7ddb14SDimitry Andric           auto *NewCall = CallBase::addOperandBundle(
1727*5f7ddb14SDimitry Andric               CI, LLVMContext::OB_clang_arc_attachedcall, OB, CI);
1728*5f7ddb14SDimitry Andric           NewCall->copyMetadata(*CI);
1729*5f7ddb14SDimitry Andric           CI->replaceAllUsesWith(NewCall);
1730*5f7ddb14SDimitry Andric           CI->eraseFromParent();
1731*5f7ddb14SDimitry Andric           InsertRetainCall = false;
1732*5f7ddb14SDimitry Andric         }
1733*5f7ddb14SDimitry Andric       }
1734*5f7ddb14SDimitry Andric 
1735*5f7ddb14SDimitry Andric       break;
1736*5f7ddb14SDimitry Andric     }
1737*5f7ddb14SDimitry Andric 
1738*5f7ddb14SDimitry Andric     if (InsertRetainCall) {
1739*5f7ddb14SDimitry Andric       // The retainRV is attached to the call and we've failed to find a
1740*5f7ddb14SDimitry Andric       // matching autoreleaseRV or an annotated call in the callee. Emit a call
1741*5f7ddb14SDimitry Andric       // to objc_retain.
1742*5f7ddb14SDimitry Andric       Builder.SetInsertPoint(RI);
1743*5f7ddb14SDimitry Andric       Function *IFn = Intrinsic::getDeclaration(Mod, Intrinsic::objc_retain);
1744*5f7ddb14SDimitry Andric       Value *BC = Builder.CreateBitCast(RetOpnd, IFn->getArg(0)->getType());
1745*5f7ddb14SDimitry Andric       Builder.CreateCall(IFn, BC, "");
1746*5f7ddb14SDimitry Andric     }
1747*5f7ddb14SDimitry Andric   }
1748*5f7ddb14SDimitry Andric }
1749*5f7ddb14SDimitry Andric 
17500b57cec5SDimitry Andric /// This function inlines the called function into the basic block of the
17510b57cec5SDimitry Andric /// caller. This returns false if it is not possible to inline this call.
17520b57cec5SDimitry Andric /// The program is still in a well defined state if this occurs though.
17530b57cec5SDimitry Andric ///
17540b57cec5SDimitry Andric /// Note that this only does one level of inlining.  For example, if the
17550b57cec5SDimitry Andric /// instruction 'call B' is inlined, and 'B' calls 'C', then the call to 'C' now
17560b57cec5SDimitry Andric /// exists in the instruction stream.  Similarly this will inline a recursive
17570b57cec5SDimitry Andric /// function by one level.
InlineFunction(CallBase & CB,InlineFunctionInfo & IFI,AAResults * CalleeAAR,bool InsertLifetime,Function * ForwardVarArgsTo)17585ffd83dbSDimitry Andric llvm::InlineResult llvm::InlineFunction(CallBase &CB, InlineFunctionInfo &IFI,
17590b57cec5SDimitry Andric                                         AAResults *CalleeAAR,
17600b57cec5SDimitry Andric                                         bool InsertLifetime,
17610b57cec5SDimitry Andric                                         Function *ForwardVarArgsTo) {
17625ffd83dbSDimitry Andric   assert(CB.getParent() && CB.getFunction() && "Instruction not in function!");
17630b57cec5SDimitry Andric 
17640b57cec5SDimitry Andric   // FIXME: we don't inline callbr yet.
17655ffd83dbSDimitry Andric   if (isa<CallBrInst>(CB))
17665ffd83dbSDimitry Andric     return InlineResult::failure("We don't inline callbr yet.");
17670b57cec5SDimitry Andric 
17680b57cec5SDimitry Andric   // If IFI has any state in it, zap it before we fill it in.
17690b57cec5SDimitry Andric   IFI.reset();
17700b57cec5SDimitry Andric 
17715ffd83dbSDimitry Andric   Function *CalledFunc = CB.getCalledFunction();
17720b57cec5SDimitry Andric   if (!CalledFunc ||               // Can't inline external function or indirect
17730b57cec5SDimitry Andric       CalledFunc->isDeclaration()) // call!
17745ffd83dbSDimitry Andric     return InlineResult::failure("external or indirect");
17750b57cec5SDimitry Andric 
17760b57cec5SDimitry Andric   // The inliner does not know how to inline through calls with operand bundles
17770b57cec5SDimitry Andric   // in general ...
17785ffd83dbSDimitry Andric   if (CB.hasOperandBundles()) {
17795ffd83dbSDimitry Andric     for (int i = 0, e = CB.getNumOperandBundles(); i != e; ++i) {
17805ffd83dbSDimitry Andric       uint32_t Tag = CB.getOperandBundleAt(i).getTagID();
17810b57cec5SDimitry Andric       // ... but it knows how to inline through "deopt" operand bundles ...
17820b57cec5SDimitry Andric       if (Tag == LLVMContext::OB_deopt)
17830b57cec5SDimitry Andric         continue;
17840b57cec5SDimitry Andric       // ... and "funclet" operand bundles.
17850b57cec5SDimitry Andric       if (Tag == LLVMContext::OB_funclet)
17860b57cec5SDimitry Andric         continue;
1787*5f7ddb14SDimitry Andric       if (Tag == LLVMContext::OB_clang_arc_attachedcall)
1788*5f7ddb14SDimitry Andric         continue;
17890b57cec5SDimitry Andric 
17905ffd83dbSDimitry Andric       return InlineResult::failure("unsupported operand bundle");
17910b57cec5SDimitry Andric     }
17920b57cec5SDimitry Andric   }
17930b57cec5SDimitry Andric 
17940b57cec5SDimitry Andric   // If the call to the callee cannot throw, set the 'nounwind' flag on any
17950b57cec5SDimitry Andric   // calls that we inline.
17965ffd83dbSDimitry Andric   bool MarkNoUnwind = CB.doesNotThrow();
17970b57cec5SDimitry Andric 
17985ffd83dbSDimitry Andric   BasicBlock *OrigBB = CB.getParent();
17990b57cec5SDimitry Andric   Function *Caller = OrigBB->getParent();
18000b57cec5SDimitry Andric 
18010b57cec5SDimitry Andric   // GC poses two hazards to inlining, which only occur when the callee has GC:
18020b57cec5SDimitry Andric   //  1. If the caller has no GC, then the callee's GC must be propagated to the
18030b57cec5SDimitry Andric   //     caller.
18040b57cec5SDimitry Andric   //  2. If the caller has a differing GC, it is invalid to inline.
18050b57cec5SDimitry Andric   if (CalledFunc->hasGC()) {
18060b57cec5SDimitry Andric     if (!Caller->hasGC())
18070b57cec5SDimitry Andric       Caller->setGC(CalledFunc->getGC());
18080b57cec5SDimitry Andric     else if (CalledFunc->getGC() != Caller->getGC())
18095ffd83dbSDimitry Andric       return InlineResult::failure("incompatible GC");
18100b57cec5SDimitry Andric   }
18110b57cec5SDimitry Andric 
18120b57cec5SDimitry Andric   // Get the personality function from the callee if it contains a landing pad.
18130b57cec5SDimitry Andric   Constant *CalledPersonality =
18140b57cec5SDimitry Andric       CalledFunc->hasPersonalityFn()
18150b57cec5SDimitry Andric           ? CalledFunc->getPersonalityFn()->stripPointerCasts()
18160b57cec5SDimitry Andric           : nullptr;
18170b57cec5SDimitry Andric 
18180b57cec5SDimitry Andric   // Find the personality function used by the landing pads of the caller. If it
18190b57cec5SDimitry Andric   // exists, then check to see that it matches the personality function used in
18200b57cec5SDimitry Andric   // the callee.
18210b57cec5SDimitry Andric   Constant *CallerPersonality =
18220b57cec5SDimitry Andric       Caller->hasPersonalityFn()
18230b57cec5SDimitry Andric           ? Caller->getPersonalityFn()->stripPointerCasts()
18240b57cec5SDimitry Andric           : nullptr;
18250b57cec5SDimitry Andric   if (CalledPersonality) {
18260b57cec5SDimitry Andric     if (!CallerPersonality)
18270b57cec5SDimitry Andric       Caller->setPersonalityFn(CalledPersonality);
18280b57cec5SDimitry Andric     // If the personality functions match, then we can perform the
18290b57cec5SDimitry Andric     // inlining. Otherwise, we can't inline.
18300b57cec5SDimitry Andric     // TODO: This isn't 100% true. Some personality functions are proper
18310b57cec5SDimitry Andric     //       supersets of others and can be used in place of the other.
18320b57cec5SDimitry Andric     else if (CalledPersonality != CallerPersonality)
18335ffd83dbSDimitry Andric       return InlineResult::failure("incompatible personality");
18340b57cec5SDimitry Andric   }
18350b57cec5SDimitry Andric 
18360b57cec5SDimitry Andric   // We need to figure out which funclet the callsite was in so that we may
18370b57cec5SDimitry Andric   // properly nest the callee.
18380b57cec5SDimitry Andric   Instruction *CallSiteEHPad = nullptr;
18390b57cec5SDimitry Andric   if (CallerPersonality) {
18400b57cec5SDimitry Andric     EHPersonality Personality = classifyEHPersonality(CallerPersonality);
18410b57cec5SDimitry Andric     if (isScopedEHPersonality(Personality)) {
18420b57cec5SDimitry Andric       Optional<OperandBundleUse> ParentFunclet =
18435ffd83dbSDimitry Andric           CB.getOperandBundle(LLVMContext::OB_funclet);
18440b57cec5SDimitry Andric       if (ParentFunclet)
18450b57cec5SDimitry Andric         CallSiteEHPad = cast<FuncletPadInst>(ParentFunclet->Inputs.front());
18460b57cec5SDimitry Andric 
18470b57cec5SDimitry Andric       // OK, the inlining site is legal.  What about the target function?
18480b57cec5SDimitry Andric 
18490b57cec5SDimitry Andric       if (CallSiteEHPad) {
18500b57cec5SDimitry Andric         if (Personality == EHPersonality::MSVC_CXX) {
18510b57cec5SDimitry Andric           // The MSVC personality cannot tolerate catches getting inlined into
18520b57cec5SDimitry Andric           // cleanup funclets.
18530b57cec5SDimitry Andric           if (isa<CleanupPadInst>(CallSiteEHPad)) {
18540b57cec5SDimitry Andric             // Ok, the call site is within a cleanuppad.  Let's check the callee
18550b57cec5SDimitry Andric             // for catchpads.
18560b57cec5SDimitry Andric             for (const BasicBlock &CalledBB : *CalledFunc) {
18570b57cec5SDimitry Andric               if (isa<CatchSwitchInst>(CalledBB.getFirstNonPHI()))
18585ffd83dbSDimitry Andric                 return InlineResult::failure("catch in cleanup funclet");
18590b57cec5SDimitry Andric             }
18600b57cec5SDimitry Andric           }
18610b57cec5SDimitry Andric         } else if (isAsynchronousEHPersonality(Personality)) {
18620b57cec5SDimitry Andric           // SEH is even less tolerant, there may not be any sort of exceptional
18630b57cec5SDimitry Andric           // funclet in the callee.
18640b57cec5SDimitry Andric           for (const BasicBlock &CalledBB : *CalledFunc) {
18650b57cec5SDimitry Andric             if (CalledBB.isEHPad())
18665ffd83dbSDimitry Andric               return InlineResult::failure("SEH in cleanup funclet");
18670b57cec5SDimitry Andric           }
18680b57cec5SDimitry Andric         }
18690b57cec5SDimitry Andric       }
18700b57cec5SDimitry Andric     }
18710b57cec5SDimitry Andric   }
18720b57cec5SDimitry Andric 
18730b57cec5SDimitry Andric   // Determine if we are dealing with a call in an EHPad which does not unwind
18740b57cec5SDimitry Andric   // to caller.
18750b57cec5SDimitry Andric   bool EHPadForCallUnwindsLocally = false;
18765ffd83dbSDimitry Andric   if (CallSiteEHPad && isa<CallInst>(CB)) {
18770b57cec5SDimitry Andric     UnwindDestMemoTy FuncletUnwindMap;
18780b57cec5SDimitry Andric     Value *CallSiteUnwindDestToken =
18790b57cec5SDimitry Andric         getUnwindDestToken(CallSiteEHPad, FuncletUnwindMap);
18800b57cec5SDimitry Andric 
18810b57cec5SDimitry Andric     EHPadForCallUnwindsLocally =
18820b57cec5SDimitry Andric         CallSiteUnwindDestToken &&
18830b57cec5SDimitry Andric         !isa<ConstantTokenNone>(CallSiteUnwindDestToken);
18840b57cec5SDimitry Andric   }
18850b57cec5SDimitry Andric 
18860b57cec5SDimitry Andric   // Get an iterator to the last basic block in the function, which will have
18870b57cec5SDimitry Andric   // the new function inlined after it.
18880b57cec5SDimitry Andric   Function::iterator LastBlock = --Caller->end();
18890b57cec5SDimitry Andric 
18900b57cec5SDimitry Andric   // Make sure to capture all of the return instructions from the cloned
18910b57cec5SDimitry Andric   // function.
18920b57cec5SDimitry Andric   SmallVector<ReturnInst*, 8> Returns;
18930b57cec5SDimitry Andric   ClonedCodeInfo InlinedFunctionInfo;
18940b57cec5SDimitry Andric   Function::iterator FirstNewBlock;
18950b57cec5SDimitry Andric 
18960b57cec5SDimitry Andric   { // Scope to destroy VMap after cloning.
18970b57cec5SDimitry Andric     ValueToValueMapTy VMap;
18980b57cec5SDimitry Andric     // Keep a list of pair (dst, src) to emit byval initializations.
18990b57cec5SDimitry Andric     SmallVector<std::pair<Value*, Value*>, 4> ByValInit;
19000b57cec5SDimitry Andric 
1901af732203SDimitry Andric     // When inlining a function that contains noalias scope metadata,
1902af732203SDimitry Andric     // this metadata needs to be cloned so that the inlined blocks
1903af732203SDimitry Andric     // have different "unique scopes" at every call site.
1904af732203SDimitry Andric     // Track the metadata that must be cloned. Do this before other changes to
1905af732203SDimitry Andric     // the function, so that we do not get in trouble when inlining caller ==
1906af732203SDimitry Andric     // callee.
1907af732203SDimitry Andric     ScopedAliasMetadataDeepCloner SAMetadataCloner(CB.getCalledFunction());
1908af732203SDimitry Andric 
19090b57cec5SDimitry Andric     auto &DL = Caller->getParent()->getDataLayout();
19100b57cec5SDimitry Andric 
19110b57cec5SDimitry Andric     // Calculate the vector of arguments to pass into the function cloner, which
19120b57cec5SDimitry Andric     // matches up the formal to the actual argument values.
19135ffd83dbSDimitry Andric     auto AI = CB.arg_begin();
19140b57cec5SDimitry Andric     unsigned ArgNo = 0;
19150b57cec5SDimitry Andric     for (Function::arg_iterator I = CalledFunc->arg_begin(),
19160b57cec5SDimitry Andric          E = CalledFunc->arg_end(); I != E; ++I, ++AI, ++ArgNo) {
19170b57cec5SDimitry Andric       Value *ActualArg = *AI;
19180b57cec5SDimitry Andric 
19190b57cec5SDimitry Andric       // When byval arguments actually inlined, we need to make the copy implied
19200b57cec5SDimitry Andric       // by them explicit.  However, we don't do this if the callee is readonly
19210b57cec5SDimitry Andric       // or readnone, because the copy would be unneeded: the callee doesn't
19220b57cec5SDimitry Andric       // modify the struct.
19235ffd83dbSDimitry Andric       if (CB.isByValArgument(ArgNo)) {
19245ffd83dbSDimitry Andric         ActualArg = HandleByValArgument(ActualArg, &CB, CalledFunc, IFI,
19250b57cec5SDimitry Andric                                         CalledFunc->getParamAlignment(ArgNo));
19260b57cec5SDimitry Andric         if (ActualArg != *AI)
19270b57cec5SDimitry Andric           ByValInit.push_back(std::make_pair(ActualArg, (Value*) *AI));
19280b57cec5SDimitry Andric       }
19290b57cec5SDimitry Andric 
19300b57cec5SDimitry Andric       VMap[&*I] = ActualArg;
19310b57cec5SDimitry Andric     }
19320b57cec5SDimitry Andric 
19335ffd83dbSDimitry Andric     // TODO: Remove this when users have been updated to the assume bundles.
19340b57cec5SDimitry Andric     // Add alignment assumptions if necessary. We do this before the inlined
19350b57cec5SDimitry Andric     // instructions are actually cloned into the caller so that we can easily
19360b57cec5SDimitry Andric     // check what will be known at the start of the inlined code.
19375ffd83dbSDimitry Andric     AddAlignmentAssumptions(CB, IFI);
19385ffd83dbSDimitry Andric 
19395ffd83dbSDimitry Andric     AssumptionCache *AC =
19405ffd83dbSDimitry Andric         IFI.GetAssumptionCache ? &IFI.GetAssumptionCache(*Caller) : nullptr;
19415ffd83dbSDimitry Andric 
19425ffd83dbSDimitry Andric     /// Preserve all attributes on of the call and its parameters.
19435ffd83dbSDimitry Andric     salvageKnowledge(&CB, AC);
19440b57cec5SDimitry Andric 
19450b57cec5SDimitry Andric     // We want the inliner to prune the code as it copies.  We would LOVE to
19460b57cec5SDimitry Andric     // have no dead or constant instructions leftover after inlining occurs
19470b57cec5SDimitry Andric     // (which can happen, e.g., because an argument was constant), but we'll be
19480b57cec5SDimitry Andric     // happy with whatever the cloner can do.
19490b57cec5SDimitry Andric     CloneAndPruneFunctionInto(Caller, CalledFunc, VMap,
19500b57cec5SDimitry Andric                               /*ModuleLevelChanges=*/false, Returns, ".i",
1951*5f7ddb14SDimitry Andric                               &InlinedFunctionInfo);
19520b57cec5SDimitry Andric     // Remember the first block that is newly cloned over.
19530b57cec5SDimitry Andric     FirstNewBlock = LastBlock; ++FirstNewBlock;
19540b57cec5SDimitry Andric 
1955*5f7ddb14SDimitry Andric     // Insert retainRV/clainRV runtime calls.
1956*5f7ddb14SDimitry Andric     if (objcarc::hasAttachedCallOpBundle(&CB))
1957*5f7ddb14SDimitry Andric       inlineRetainOrClaimRVCalls(CB, Returns);
1958*5f7ddb14SDimitry Andric 
1959*5f7ddb14SDimitry Andric     // Updated caller/callee profiles only when requested. For sample loader
1960*5f7ddb14SDimitry Andric     // inlining, the context-sensitive inlinee profile doesn't need to be
1961*5f7ddb14SDimitry Andric     // subtracted from callee profile, and the inlined clone also doesn't need
1962*5f7ddb14SDimitry Andric     // to be scaled based on call site count.
1963*5f7ddb14SDimitry Andric     if (IFI.UpdateProfile) {
19640b57cec5SDimitry Andric       if (IFI.CallerBFI != nullptr && IFI.CalleeBFI != nullptr)
19650b57cec5SDimitry Andric         // Update the BFI of blocks cloned into the caller.
19660b57cec5SDimitry Andric         updateCallerBFI(OrigBB, VMap, IFI.CallerBFI, IFI.CalleeBFI,
19670b57cec5SDimitry Andric                         CalledFunc->front());
19680b57cec5SDimitry Andric 
19695ffd83dbSDimitry Andric       updateCallProfile(CalledFunc, VMap, CalledFunc->getEntryCount(), CB,
19700b57cec5SDimitry Andric                         IFI.PSI, IFI.CallerBFI);
1971*5f7ddb14SDimitry Andric     }
19720b57cec5SDimitry Andric 
19730b57cec5SDimitry Andric     // Inject byval arguments initialization.
19740b57cec5SDimitry Andric     for (std::pair<Value*, Value*> &Init : ByValInit)
19750b57cec5SDimitry Andric       HandleByValArgumentInit(Init.first, Init.second, Caller->getParent(),
19760b57cec5SDimitry Andric                               &*FirstNewBlock, IFI);
19770b57cec5SDimitry Andric 
19780b57cec5SDimitry Andric     Optional<OperandBundleUse> ParentDeopt =
19795ffd83dbSDimitry Andric         CB.getOperandBundle(LLVMContext::OB_deopt);
19800b57cec5SDimitry Andric     if (ParentDeopt) {
19810b57cec5SDimitry Andric       SmallVector<OperandBundleDef, 2> OpDefs;
19820b57cec5SDimitry Andric 
19830b57cec5SDimitry Andric       for (auto &VH : InlinedFunctionInfo.OperandBundleCallSites) {
19845ffd83dbSDimitry Andric         CallBase *ICS = dyn_cast_or_null<CallBase>(VH);
19855ffd83dbSDimitry Andric         if (!ICS)
19865ffd83dbSDimitry Andric           continue; // instruction was DCE'd or RAUW'ed to undef
19870b57cec5SDimitry Andric 
19880b57cec5SDimitry Andric         OpDefs.clear();
19890b57cec5SDimitry Andric 
19905ffd83dbSDimitry Andric         OpDefs.reserve(ICS->getNumOperandBundles());
19910b57cec5SDimitry Andric 
19925ffd83dbSDimitry Andric         for (unsigned COBi = 0, COBe = ICS->getNumOperandBundles(); COBi < COBe;
19935ffd83dbSDimitry Andric              ++COBi) {
19945ffd83dbSDimitry Andric           auto ChildOB = ICS->getOperandBundleAt(COBi);
19950b57cec5SDimitry Andric           if (ChildOB.getTagID() != LLVMContext::OB_deopt) {
19960b57cec5SDimitry Andric             // If the inlined call has other operand bundles, let them be
19970b57cec5SDimitry Andric             OpDefs.emplace_back(ChildOB);
19980b57cec5SDimitry Andric             continue;
19990b57cec5SDimitry Andric           }
20000b57cec5SDimitry Andric 
20010b57cec5SDimitry Andric           // It may be useful to separate this logic (of handling operand
20020b57cec5SDimitry Andric           // bundles) out to a separate "policy" component if this gets crowded.
20030b57cec5SDimitry Andric           // Prepend the parent's deoptimization continuation to the newly
20040b57cec5SDimitry Andric           // inlined call's deoptimization continuation.
20050b57cec5SDimitry Andric           std::vector<Value *> MergedDeoptArgs;
20060b57cec5SDimitry Andric           MergedDeoptArgs.reserve(ParentDeopt->Inputs.size() +
20070b57cec5SDimitry Andric                                   ChildOB.Inputs.size());
20080b57cec5SDimitry Andric 
2009af732203SDimitry Andric           llvm::append_range(MergedDeoptArgs, ParentDeopt->Inputs);
2010af732203SDimitry Andric           llvm::append_range(MergedDeoptArgs, ChildOB.Inputs);
20110b57cec5SDimitry Andric 
20120b57cec5SDimitry Andric           OpDefs.emplace_back("deopt", std::move(MergedDeoptArgs));
20130b57cec5SDimitry Andric         }
20140b57cec5SDimitry Andric 
20155ffd83dbSDimitry Andric         Instruction *NewI = CallBase::Create(ICS, OpDefs, ICS);
20160b57cec5SDimitry Andric 
20170b57cec5SDimitry Andric         // Note: the RAUW does the appropriate fixup in VMap, so we need to do
20180b57cec5SDimitry Andric         // this even if the call returns void.
20195ffd83dbSDimitry Andric         ICS->replaceAllUsesWith(NewI);
20200b57cec5SDimitry Andric 
20210b57cec5SDimitry Andric         VH = nullptr;
20225ffd83dbSDimitry Andric         ICS->eraseFromParent();
20230b57cec5SDimitry Andric       }
20240b57cec5SDimitry Andric     }
20250b57cec5SDimitry Andric 
20260b57cec5SDimitry Andric     // Update the callgraph if requested.
20270b57cec5SDimitry Andric     if (IFI.CG)
20285ffd83dbSDimitry Andric       UpdateCallGraphAfterInlining(CB, FirstNewBlock, VMap, IFI);
20290b57cec5SDimitry Andric 
20300b57cec5SDimitry Andric     // For 'nodebug' functions, the associated DISubprogram is always null.
20310b57cec5SDimitry Andric     // Conservatively avoid propagating the callsite debug location to
20320b57cec5SDimitry Andric     // instructions inlined from a function whose DISubprogram is not null.
20335ffd83dbSDimitry Andric     fixupLineNumbers(Caller, FirstNewBlock, &CB,
20340b57cec5SDimitry Andric                      CalledFunc->getSubprogram() != nullptr);
20350b57cec5SDimitry Andric 
2036af732203SDimitry Andric     // Now clone the inlined noalias scope metadata.
2037af732203SDimitry Andric     SAMetadataCloner.clone();
2038af732203SDimitry Andric     SAMetadataCloner.remap(FirstNewBlock, Caller->end());
20390b57cec5SDimitry Andric 
20400b57cec5SDimitry Andric     // Add noalias metadata if necessary.
2041*5f7ddb14SDimitry Andric     AddAliasScopeMetadata(CB, VMap, DL, CalleeAAR, InlinedFunctionInfo);
20425ffd83dbSDimitry Andric 
20435ffd83dbSDimitry Andric     // Clone return attributes on the callsite into the calls within the inlined
20445ffd83dbSDimitry Andric     // function which feed into its return value.
20455ffd83dbSDimitry Andric     AddReturnAttributes(CB, VMap);
20460b57cec5SDimitry Andric 
2047af732203SDimitry Andric     // Propagate metadata on the callsite if necessary.
2048af732203SDimitry Andric     PropagateCallSiteMetadata(CB, FirstNewBlock, Caller->end());
20490b57cec5SDimitry Andric 
20500b57cec5SDimitry Andric     // Register any cloned assumptions.
20510b57cec5SDimitry Andric     if (IFI.GetAssumptionCache)
20520b57cec5SDimitry Andric       for (BasicBlock &NewBlock :
20530b57cec5SDimitry Andric            make_range(FirstNewBlock->getIterator(), Caller->end()))
20545ffd83dbSDimitry Andric         for (Instruction &I : NewBlock)
2055*5f7ddb14SDimitry Andric           if (auto *II = dyn_cast<AssumeInst>(&I))
20565ffd83dbSDimitry Andric             IFI.GetAssumptionCache(*Caller).registerAssumption(II);
20570b57cec5SDimitry Andric   }
20580b57cec5SDimitry Andric 
20590b57cec5SDimitry Andric   // If there are any alloca instructions in the block that used to be the entry
20600b57cec5SDimitry Andric   // block for the callee, move them to the entry block of the caller.  First
20610b57cec5SDimitry Andric   // calculate which instruction they should be inserted before.  We insert the
20620b57cec5SDimitry Andric   // instructions at the end of the current alloca list.
20630b57cec5SDimitry Andric   {
20640b57cec5SDimitry Andric     BasicBlock::iterator InsertPoint = Caller->begin()->begin();
20650b57cec5SDimitry Andric     for (BasicBlock::iterator I = FirstNewBlock->begin(),
20660b57cec5SDimitry Andric          E = FirstNewBlock->end(); I != E; ) {
20670b57cec5SDimitry Andric       AllocaInst *AI = dyn_cast<AllocaInst>(I++);
20680b57cec5SDimitry Andric       if (!AI) continue;
20690b57cec5SDimitry Andric 
20700b57cec5SDimitry Andric       // If the alloca is now dead, remove it.  This often occurs due to code
20710b57cec5SDimitry Andric       // specialization.
20720b57cec5SDimitry Andric       if (AI->use_empty()) {
20730b57cec5SDimitry Andric         AI->eraseFromParent();
20740b57cec5SDimitry Andric         continue;
20750b57cec5SDimitry Andric       }
20760b57cec5SDimitry Andric 
20770b57cec5SDimitry Andric       if (!allocaWouldBeStaticInEntry(AI))
20780b57cec5SDimitry Andric         continue;
20790b57cec5SDimitry Andric 
20800b57cec5SDimitry Andric       // Keep track of the static allocas that we inline into the caller.
20810b57cec5SDimitry Andric       IFI.StaticAllocas.push_back(AI);
20820b57cec5SDimitry Andric 
20830b57cec5SDimitry Andric       // Scan for the block of allocas that we can move over, and move them
20840b57cec5SDimitry Andric       // all at once.
20850b57cec5SDimitry Andric       while (isa<AllocaInst>(I) &&
2086480093f4SDimitry Andric              !cast<AllocaInst>(I)->use_empty() &&
20870b57cec5SDimitry Andric              allocaWouldBeStaticInEntry(cast<AllocaInst>(I))) {
20880b57cec5SDimitry Andric         IFI.StaticAllocas.push_back(cast<AllocaInst>(I));
20890b57cec5SDimitry Andric         ++I;
20900b57cec5SDimitry Andric       }
20910b57cec5SDimitry Andric 
20920b57cec5SDimitry Andric       // Transfer all of the allocas over in a block.  Using splice means
20930b57cec5SDimitry Andric       // that the instructions aren't removed from the symbol table, then
20940b57cec5SDimitry Andric       // reinserted.
20950b57cec5SDimitry Andric       Caller->getEntryBlock().getInstList().splice(
20960b57cec5SDimitry Andric           InsertPoint, FirstNewBlock->getInstList(), AI->getIterator(), I);
20970b57cec5SDimitry Andric     }
20980b57cec5SDimitry Andric   }
20990b57cec5SDimitry Andric 
21000b57cec5SDimitry Andric   SmallVector<Value*,4> VarArgsToForward;
21010b57cec5SDimitry Andric   SmallVector<AttributeSet, 4> VarArgsAttrs;
21020b57cec5SDimitry Andric   for (unsigned i = CalledFunc->getFunctionType()->getNumParams();
21035ffd83dbSDimitry Andric        i < CB.getNumArgOperands(); i++) {
21045ffd83dbSDimitry Andric     VarArgsToForward.push_back(CB.getArgOperand(i));
21055ffd83dbSDimitry Andric     VarArgsAttrs.push_back(CB.getAttributes().getParamAttributes(i));
21060b57cec5SDimitry Andric   }
21070b57cec5SDimitry Andric 
21080b57cec5SDimitry Andric   bool InlinedMustTailCalls = false, InlinedDeoptimizeCalls = false;
21090b57cec5SDimitry Andric   if (InlinedFunctionInfo.ContainsCalls) {
21100b57cec5SDimitry Andric     CallInst::TailCallKind CallSiteTailKind = CallInst::TCK_None;
21115ffd83dbSDimitry Andric     if (CallInst *CI = dyn_cast<CallInst>(&CB))
21120b57cec5SDimitry Andric       CallSiteTailKind = CI->getTailCallKind();
21130b57cec5SDimitry Andric 
21140b57cec5SDimitry Andric     // For inlining purposes, the "notail" marker is the same as no marker.
21150b57cec5SDimitry Andric     if (CallSiteTailKind == CallInst::TCK_NoTail)
21160b57cec5SDimitry Andric       CallSiteTailKind = CallInst::TCK_None;
21170b57cec5SDimitry Andric 
21180b57cec5SDimitry Andric     for (Function::iterator BB = FirstNewBlock, E = Caller->end(); BB != E;
21190b57cec5SDimitry Andric          ++BB) {
21200b57cec5SDimitry Andric       for (auto II = BB->begin(); II != BB->end();) {
21210b57cec5SDimitry Andric         Instruction &I = *II++;
21220b57cec5SDimitry Andric         CallInst *CI = dyn_cast<CallInst>(&I);
21230b57cec5SDimitry Andric         if (!CI)
21240b57cec5SDimitry Andric           continue;
21250b57cec5SDimitry Andric 
21260b57cec5SDimitry Andric         // Forward varargs from inlined call site to calls to the
21270b57cec5SDimitry Andric         // ForwardVarArgsTo function, if requested, and to musttail calls.
21280b57cec5SDimitry Andric         if (!VarArgsToForward.empty() &&
21290b57cec5SDimitry Andric             ((ForwardVarArgsTo &&
21300b57cec5SDimitry Andric               CI->getCalledFunction() == ForwardVarArgsTo) ||
21310b57cec5SDimitry Andric              CI->isMustTailCall())) {
21320b57cec5SDimitry Andric           // Collect attributes for non-vararg parameters.
21330b57cec5SDimitry Andric           AttributeList Attrs = CI->getAttributes();
21340b57cec5SDimitry Andric           SmallVector<AttributeSet, 8> ArgAttrs;
21350b57cec5SDimitry Andric           if (!Attrs.isEmpty() || !VarArgsAttrs.empty()) {
21360b57cec5SDimitry Andric             for (unsigned ArgNo = 0;
21370b57cec5SDimitry Andric                  ArgNo < CI->getFunctionType()->getNumParams(); ++ArgNo)
21380b57cec5SDimitry Andric               ArgAttrs.push_back(Attrs.getParamAttributes(ArgNo));
21390b57cec5SDimitry Andric           }
21400b57cec5SDimitry Andric 
21410b57cec5SDimitry Andric           // Add VarArg attributes.
21420b57cec5SDimitry Andric           ArgAttrs.append(VarArgsAttrs.begin(), VarArgsAttrs.end());
21430b57cec5SDimitry Andric           Attrs = AttributeList::get(CI->getContext(), Attrs.getFnAttributes(),
21440b57cec5SDimitry Andric                                      Attrs.getRetAttributes(), ArgAttrs);
21450b57cec5SDimitry Andric           // Add VarArgs to existing parameters.
21460b57cec5SDimitry Andric           SmallVector<Value *, 6> Params(CI->arg_operands());
21470b57cec5SDimitry Andric           Params.append(VarArgsToForward.begin(), VarArgsToForward.end());
21480b57cec5SDimitry Andric           CallInst *NewCI = CallInst::Create(
21490b57cec5SDimitry Andric               CI->getFunctionType(), CI->getCalledOperand(), Params, "", CI);
21500b57cec5SDimitry Andric           NewCI->setDebugLoc(CI->getDebugLoc());
21510b57cec5SDimitry Andric           NewCI->setAttributes(Attrs);
21520b57cec5SDimitry Andric           NewCI->setCallingConv(CI->getCallingConv());
21530b57cec5SDimitry Andric           CI->replaceAllUsesWith(NewCI);
21540b57cec5SDimitry Andric           CI->eraseFromParent();
21550b57cec5SDimitry Andric           CI = NewCI;
21560b57cec5SDimitry Andric         }
21570b57cec5SDimitry Andric 
21580b57cec5SDimitry Andric         if (Function *F = CI->getCalledFunction())
21590b57cec5SDimitry Andric           InlinedDeoptimizeCalls |=
21600b57cec5SDimitry Andric               F->getIntrinsicID() == Intrinsic::experimental_deoptimize;
21610b57cec5SDimitry Andric 
21620b57cec5SDimitry Andric         // We need to reduce the strength of any inlined tail calls.  For
21630b57cec5SDimitry Andric         // musttail, we have to avoid introducing potential unbounded stack
21640b57cec5SDimitry Andric         // growth.  For example, if functions 'f' and 'g' are mutually recursive
21650b57cec5SDimitry Andric         // with musttail, we can inline 'g' into 'f' so long as we preserve
21660b57cec5SDimitry Andric         // musttail on the cloned call to 'f'.  If either the inlined call site
21670b57cec5SDimitry Andric         // or the cloned call site is *not* musttail, the program already has
21680b57cec5SDimitry Andric         // one frame of stack growth, so it's safe to remove musttail.  Here is
21690b57cec5SDimitry Andric         // a table of example transformations:
21700b57cec5SDimitry Andric         //
21710b57cec5SDimitry Andric         //    f -> musttail g -> musttail f  ==>  f -> musttail f
21720b57cec5SDimitry Andric         //    f -> musttail g ->     tail f  ==>  f ->     tail f
21730b57cec5SDimitry Andric         //    f ->          g -> musttail f  ==>  f ->          f
21740b57cec5SDimitry Andric         //    f ->          g ->     tail f  ==>  f ->          f
21750b57cec5SDimitry Andric         //
21760b57cec5SDimitry Andric         // Inlined notail calls should remain notail calls.
21770b57cec5SDimitry Andric         CallInst::TailCallKind ChildTCK = CI->getTailCallKind();
21780b57cec5SDimitry Andric         if (ChildTCK != CallInst::TCK_NoTail)
21790b57cec5SDimitry Andric           ChildTCK = std::min(CallSiteTailKind, ChildTCK);
21800b57cec5SDimitry Andric         CI->setTailCallKind(ChildTCK);
21810b57cec5SDimitry Andric         InlinedMustTailCalls |= CI->isMustTailCall();
21820b57cec5SDimitry Andric 
21830b57cec5SDimitry Andric         // Calls inlined through a 'nounwind' call site should be marked
21840b57cec5SDimitry Andric         // 'nounwind'.
21850b57cec5SDimitry Andric         if (MarkNoUnwind)
21860b57cec5SDimitry Andric           CI->setDoesNotThrow();
21870b57cec5SDimitry Andric       }
21880b57cec5SDimitry Andric     }
21890b57cec5SDimitry Andric   }
21900b57cec5SDimitry Andric 
21910b57cec5SDimitry Andric   // Leave lifetime markers for the static alloca's, scoping them to the
21920b57cec5SDimitry Andric   // function we just inlined.
2193*5f7ddb14SDimitry Andric   // We need to insert lifetime intrinsics even at O0 to avoid invalid
2194*5f7ddb14SDimitry Andric   // access caused by multithreaded coroutines. The check
2195*5f7ddb14SDimitry Andric   // `Caller->isPresplitCoroutine()` would affect AlwaysInliner at O0 only.
2196*5f7ddb14SDimitry Andric   if ((InsertLifetime || Caller->isPresplitCoroutine()) &&
2197*5f7ddb14SDimitry Andric       !IFI.StaticAllocas.empty()) {
21980b57cec5SDimitry Andric     IRBuilder<> builder(&FirstNewBlock->front());
21990b57cec5SDimitry Andric     for (unsigned ai = 0, ae = IFI.StaticAllocas.size(); ai != ae; ++ai) {
22000b57cec5SDimitry Andric       AllocaInst *AI = IFI.StaticAllocas[ai];
22010b57cec5SDimitry Andric       // Don't mark swifterror allocas. They can't have bitcast uses.
22020b57cec5SDimitry Andric       if (AI->isSwiftError())
22030b57cec5SDimitry Andric         continue;
22040b57cec5SDimitry Andric 
22050b57cec5SDimitry Andric       // If the alloca is already scoped to something smaller than the whole
22060b57cec5SDimitry Andric       // function then there's no need to add redundant, less accurate markers.
22070b57cec5SDimitry Andric       if (hasLifetimeMarkers(AI))
22080b57cec5SDimitry Andric         continue;
22090b57cec5SDimitry Andric 
22100b57cec5SDimitry Andric       // Try to determine the size of the allocation.
22110b57cec5SDimitry Andric       ConstantInt *AllocaSize = nullptr;
22120b57cec5SDimitry Andric       if (ConstantInt *AIArraySize =
22130b57cec5SDimitry Andric           dyn_cast<ConstantInt>(AI->getArraySize())) {
22140b57cec5SDimitry Andric         auto &DL = Caller->getParent()->getDataLayout();
22150b57cec5SDimitry Andric         Type *AllocaType = AI->getAllocatedType();
2216af732203SDimitry Andric         TypeSize AllocaTypeSize = DL.getTypeAllocSize(AllocaType);
22170b57cec5SDimitry Andric         uint64_t AllocaArraySize = AIArraySize->getLimitedValue();
22180b57cec5SDimitry Andric 
22190b57cec5SDimitry Andric         // Don't add markers for zero-sized allocas.
22200b57cec5SDimitry Andric         if (AllocaArraySize == 0)
22210b57cec5SDimitry Andric           continue;
22220b57cec5SDimitry Andric 
22230b57cec5SDimitry Andric         // Check that array size doesn't saturate uint64_t and doesn't
22240b57cec5SDimitry Andric         // overflow when it's multiplied by type size.
2225af732203SDimitry Andric         if (!AllocaTypeSize.isScalable() &&
2226af732203SDimitry Andric             AllocaArraySize != std::numeric_limits<uint64_t>::max() &&
22270b57cec5SDimitry Andric             std::numeric_limits<uint64_t>::max() / AllocaArraySize >=
2228af732203SDimitry Andric                 AllocaTypeSize.getFixedSize()) {
22290b57cec5SDimitry Andric           AllocaSize = ConstantInt::get(Type::getInt64Ty(AI->getContext()),
22300b57cec5SDimitry Andric                                         AllocaArraySize * AllocaTypeSize);
22310b57cec5SDimitry Andric         }
22320b57cec5SDimitry Andric       }
22330b57cec5SDimitry Andric 
22340b57cec5SDimitry Andric       builder.CreateLifetimeStart(AI, AllocaSize);
22350b57cec5SDimitry Andric       for (ReturnInst *RI : Returns) {
22360b57cec5SDimitry Andric         // Don't insert llvm.lifetime.end calls between a musttail or deoptimize
22370b57cec5SDimitry Andric         // call and a return.  The return kills all local allocas.
22380b57cec5SDimitry Andric         if (InlinedMustTailCalls &&
22390b57cec5SDimitry Andric             RI->getParent()->getTerminatingMustTailCall())
22400b57cec5SDimitry Andric           continue;
22410b57cec5SDimitry Andric         if (InlinedDeoptimizeCalls &&
22420b57cec5SDimitry Andric             RI->getParent()->getTerminatingDeoptimizeCall())
22430b57cec5SDimitry Andric           continue;
22440b57cec5SDimitry Andric         IRBuilder<>(RI).CreateLifetimeEnd(AI, AllocaSize);
22450b57cec5SDimitry Andric       }
22460b57cec5SDimitry Andric     }
22470b57cec5SDimitry Andric   }
22480b57cec5SDimitry Andric 
22490b57cec5SDimitry Andric   // If the inlined code contained dynamic alloca instructions, wrap the inlined
22500b57cec5SDimitry Andric   // code with llvm.stacksave/llvm.stackrestore intrinsics.
22510b57cec5SDimitry Andric   if (InlinedFunctionInfo.ContainsDynamicAllocas) {
22520b57cec5SDimitry Andric     Module *M = Caller->getParent();
22530b57cec5SDimitry Andric     // Get the two intrinsics we care about.
22540b57cec5SDimitry Andric     Function *StackSave = Intrinsic::getDeclaration(M, Intrinsic::stacksave);
22550b57cec5SDimitry Andric     Function *StackRestore=Intrinsic::getDeclaration(M,Intrinsic::stackrestore);
22560b57cec5SDimitry Andric 
22570b57cec5SDimitry Andric     // Insert the llvm.stacksave.
22580b57cec5SDimitry Andric     CallInst *SavedPtr = IRBuilder<>(&*FirstNewBlock, FirstNewBlock->begin())
22590b57cec5SDimitry Andric                              .CreateCall(StackSave, {}, "savedstack");
22600b57cec5SDimitry Andric 
22610b57cec5SDimitry Andric     // Insert a call to llvm.stackrestore before any return instructions in the
22620b57cec5SDimitry Andric     // inlined function.
22630b57cec5SDimitry Andric     for (ReturnInst *RI : Returns) {
22640b57cec5SDimitry Andric       // Don't insert llvm.stackrestore calls between a musttail or deoptimize
22650b57cec5SDimitry Andric       // call and a return.  The return will restore the stack pointer.
22660b57cec5SDimitry Andric       if (InlinedMustTailCalls && RI->getParent()->getTerminatingMustTailCall())
22670b57cec5SDimitry Andric         continue;
22680b57cec5SDimitry Andric       if (InlinedDeoptimizeCalls && RI->getParent()->getTerminatingDeoptimizeCall())
22690b57cec5SDimitry Andric         continue;
22700b57cec5SDimitry Andric       IRBuilder<>(RI).CreateCall(StackRestore, SavedPtr);
22710b57cec5SDimitry Andric     }
22720b57cec5SDimitry Andric   }
22730b57cec5SDimitry Andric 
22740b57cec5SDimitry Andric   // If we are inlining for an invoke instruction, we must make sure to rewrite
22750b57cec5SDimitry Andric   // any call instructions into invoke instructions.  This is sensitive to which
22760b57cec5SDimitry Andric   // funclet pads were top-level in the inlinee, so must be done before
22770b57cec5SDimitry Andric   // rewriting the "parent pad" links.
22785ffd83dbSDimitry Andric   if (auto *II = dyn_cast<InvokeInst>(&CB)) {
22790b57cec5SDimitry Andric     BasicBlock *UnwindDest = II->getUnwindDest();
22800b57cec5SDimitry Andric     Instruction *FirstNonPHI = UnwindDest->getFirstNonPHI();
22810b57cec5SDimitry Andric     if (isa<LandingPadInst>(FirstNonPHI)) {
22820b57cec5SDimitry Andric       HandleInlinedLandingPad(II, &*FirstNewBlock, InlinedFunctionInfo);
22830b57cec5SDimitry Andric     } else {
22840b57cec5SDimitry Andric       HandleInlinedEHPad(II, &*FirstNewBlock, InlinedFunctionInfo);
22850b57cec5SDimitry Andric     }
22860b57cec5SDimitry Andric   }
22870b57cec5SDimitry Andric 
22880b57cec5SDimitry Andric   // Update the lexical scopes of the new funclets and callsites.
22890b57cec5SDimitry Andric   // Anything that had 'none' as its parent is now nested inside the callsite's
22900b57cec5SDimitry Andric   // EHPad.
22910b57cec5SDimitry Andric 
22920b57cec5SDimitry Andric   if (CallSiteEHPad) {
22930b57cec5SDimitry Andric     for (Function::iterator BB = FirstNewBlock->getIterator(),
22940b57cec5SDimitry Andric                             E = Caller->end();
22950b57cec5SDimitry Andric          BB != E; ++BB) {
22960b57cec5SDimitry Andric       // Add bundle operands to any top-level call sites.
22970b57cec5SDimitry Andric       SmallVector<OperandBundleDef, 1> OpBundles;
22980b57cec5SDimitry Andric       for (BasicBlock::iterator BBI = BB->begin(), E = BB->end(); BBI != E;) {
22995ffd83dbSDimitry Andric         CallBase *I = dyn_cast<CallBase>(&*BBI++);
23005ffd83dbSDimitry Andric         if (!I)
23010b57cec5SDimitry Andric           continue;
23020b57cec5SDimitry Andric 
23030b57cec5SDimitry Andric         // Skip call sites which are nounwind intrinsics.
23040b57cec5SDimitry Andric         auto *CalledFn =
23055ffd83dbSDimitry Andric             dyn_cast<Function>(I->getCalledOperand()->stripPointerCasts());
23065ffd83dbSDimitry Andric         if (CalledFn && CalledFn->isIntrinsic() && I->doesNotThrow())
23070b57cec5SDimitry Andric           continue;
23080b57cec5SDimitry Andric 
23090b57cec5SDimitry Andric         // Skip call sites which already have a "funclet" bundle.
23105ffd83dbSDimitry Andric         if (I->getOperandBundle(LLVMContext::OB_funclet))
23110b57cec5SDimitry Andric           continue;
23120b57cec5SDimitry Andric 
23135ffd83dbSDimitry Andric         I->getOperandBundlesAsDefs(OpBundles);
23140b57cec5SDimitry Andric         OpBundles.emplace_back("funclet", CallSiteEHPad);
23150b57cec5SDimitry Andric 
23165ffd83dbSDimitry Andric         Instruction *NewInst = CallBase::Create(I, OpBundles, I);
23170b57cec5SDimitry Andric         NewInst->takeName(I);
23180b57cec5SDimitry Andric         I->replaceAllUsesWith(NewInst);
23190b57cec5SDimitry Andric         I->eraseFromParent();
23200b57cec5SDimitry Andric 
23210b57cec5SDimitry Andric         OpBundles.clear();
23220b57cec5SDimitry Andric       }
23230b57cec5SDimitry Andric 
23240b57cec5SDimitry Andric       // It is problematic if the inlinee has a cleanupret which unwinds to
23250b57cec5SDimitry Andric       // caller and we inline it into a call site which doesn't unwind but into
23260b57cec5SDimitry Andric       // an EH pad that does.  Such an edge must be dynamically unreachable.
23270b57cec5SDimitry Andric       // As such, we replace the cleanupret with unreachable.
23280b57cec5SDimitry Andric       if (auto *CleanupRet = dyn_cast<CleanupReturnInst>(BB->getTerminator()))
23290b57cec5SDimitry Andric         if (CleanupRet->unwindsToCaller() && EHPadForCallUnwindsLocally)
2330*5f7ddb14SDimitry Andric           changeToUnreachable(CleanupRet);
23310b57cec5SDimitry Andric 
23320b57cec5SDimitry Andric       Instruction *I = BB->getFirstNonPHI();
23330b57cec5SDimitry Andric       if (!I->isEHPad())
23340b57cec5SDimitry Andric         continue;
23350b57cec5SDimitry Andric 
23360b57cec5SDimitry Andric       if (auto *CatchSwitch = dyn_cast<CatchSwitchInst>(I)) {
23370b57cec5SDimitry Andric         if (isa<ConstantTokenNone>(CatchSwitch->getParentPad()))
23380b57cec5SDimitry Andric           CatchSwitch->setParentPad(CallSiteEHPad);
23390b57cec5SDimitry Andric       } else {
23400b57cec5SDimitry Andric         auto *FPI = cast<FuncletPadInst>(I);
23410b57cec5SDimitry Andric         if (isa<ConstantTokenNone>(FPI->getParentPad()))
23420b57cec5SDimitry Andric           FPI->setParentPad(CallSiteEHPad);
23430b57cec5SDimitry Andric       }
23440b57cec5SDimitry Andric     }
23450b57cec5SDimitry Andric   }
23460b57cec5SDimitry Andric 
23470b57cec5SDimitry Andric   if (InlinedDeoptimizeCalls) {
23480b57cec5SDimitry Andric     // We need to at least remove the deoptimizing returns from the Return set,
23490b57cec5SDimitry Andric     // so that the control flow from those returns does not get merged into the
23500b57cec5SDimitry Andric     // caller (but terminate it instead).  If the caller's return type does not
23510b57cec5SDimitry Andric     // match the callee's return type, we also need to change the return type of
23520b57cec5SDimitry Andric     // the intrinsic.
23535ffd83dbSDimitry Andric     if (Caller->getReturnType() == CB.getType()) {
2354af732203SDimitry Andric       llvm::erase_if(Returns, [](ReturnInst *RI) {
23550b57cec5SDimitry Andric         return RI->getParent()->getTerminatingDeoptimizeCall() != nullptr;
23560b57cec5SDimitry Andric       });
23570b57cec5SDimitry Andric     } else {
23580b57cec5SDimitry Andric       SmallVector<ReturnInst *, 8> NormalReturns;
23590b57cec5SDimitry Andric       Function *NewDeoptIntrinsic = Intrinsic::getDeclaration(
23600b57cec5SDimitry Andric           Caller->getParent(), Intrinsic::experimental_deoptimize,
23610b57cec5SDimitry Andric           {Caller->getReturnType()});
23620b57cec5SDimitry Andric 
23630b57cec5SDimitry Andric       for (ReturnInst *RI : Returns) {
23640b57cec5SDimitry Andric         CallInst *DeoptCall = RI->getParent()->getTerminatingDeoptimizeCall();
23650b57cec5SDimitry Andric         if (!DeoptCall) {
23660b57cec5SDimitry Andric           NormalReturns.push_back(RI);
23670b57cec5SDimitry Andric           continue;
23680b57cec5SDimitry Andric         }
23690b57cec5SDimitry Andric 
23700b57cec5SDimitry Andric         // The calling convention on the deoptimize call itself may be bogus,
23710b57cec5SDimitry Andric         // since the code we're inlining may have undefined behavior (and may
23720b57cec5SDimitry Andric         // never actually execute at runtime); but all
23730b57cec5SDimitry Andric         // @llvm.experimental.deoptimize declarations have to have the same
23740b57cec5SDimitry Andric         // calling convention in a well-formed module.
23750b57cec5SDimitry Andric         auto CallingConv = DeoptCall->getCalledFunction()->getCallingConv();
23760b57cec5SDimitry Andric         NewDeoptIntrinsic->setCallingConv(CallingConv);
23770b57cec5SDimitry Andric         auto *CurBB = RI->getParent();
23780b57cec5SDimitry Andric         RI->eraseFromParent();
23790b57cec5SDimitry Andric 
2380af732203SDimitry Andric         SmallVector<Value *, 4> CallArgs(DeoptCall->args());
23810b57cec5SDimitry Andric 
23820b57cec5SDimitry Andric         SmallVector<OperandBundleDef, 1> OpBundles;
23830b57cec5SDimitry Andric         DeoptCall->getOperandBundlesAsDefs(OpBundles);
2384*5f7ddb14SDimitry Andric         auto DeoptAttributes = DeoptCall->getAttributes();
23850b57cec5SDimitry Andric         DeoptCall->eraseFromParent();
23860b57cec5SDimitry Andric         assert(!OpBundles.empty() &&
23870b57cec5SDimitry Andric                "Expected at least the deopt operand bundle");
23880b57cec5SDimitry Andric 
23890b57cec5SDimitry Andric         IRBuilder<> Builder(CurBB);
23900b57cec5SDimitry Andric         CallInst *NewDeoptCall =
23910b57cec5SDimitry Andric             Builder.CreateCall(NewDeoptIntrinsic, CallArgs, OpBundles);
23920b57cec5SDimitry Andric         NewDeoptCall->setCallingConv(CallingConv);
2393*5f7ddb14SDimitry Andric         NewDeoptCall->setAttributes(DeoptAttributes);
23940b57cec5SDimitry Andric         if (NewDeoptCall->getType()->isVoidTy())
23950b57cec5SDimitry Andric           Builder.CreateRetVoid();
23960b57cec5SDimitry Andric         else
23970b57cec5SDimitry Andric           Builder.CreateRet(NewDeoptCall);
23980b57cec5SDimitry Andric       }
23990b57cec5SDimitry Andric 
24000b57cec5SDimitry Andric       // Leave behind the normal returns so we can merge control flow.
24010b57cec5SDimitry Andric       std::swap(Returns, NormalReturns);
24020b57cec5SDimitry Andric     }
24030b57cec5SDimitry Andric   }
24040b57cec5SDimitry Andric 
24050b57cec5SDimitry Andric   // Handle any inlined musttail call sites.  In order for a new call site to be
24060b57cec5SDimitry Andric   // musttail, the source of the clone and the inlined call site must have been
24070b57cec5SDimitry Andric   // musttail.  Therefore it's safe to return without merging control into the
24080b57cec5SDimitry Andric   // phi below.
24090b57cec5SDimitry Andric   if (InlinedMustTailCalls) {
24100b57cec5SDimitry Andric     // Check if we need to bitcast the result of any musttail calls.
24110b57cec5SDimitry Andric     Type *NewRetTy = Caller->getReturnType();
24125ffd83dbSDimitry Andric     bool NeedBitCast = !CB.use_empty() && CB.getType() != NewRetTy;
24130b57cec5SDimitry Andric 
24140b57cec5SDimitry Andric     // Handle the returns preceded by musttail calls separately.
24150b57cec5SDimitry Andric     SmallVector<ReturnInst *, 8> NormalReturns;
24160b57cec5SDimitry Andric     for (ReturnInst *RI : Returns) {
24170b57cec5SDimitry Andric       CallInst *ReturnedMustTail =
24180b57cec5SDimitry Andric           RI->getParent()->getTerminatingMustTailCall();
24190b57cec5SDimitry Andric       if (!ReturnedMustTail) {
24200b57cec5SDimitry Andric         NormalReturns.push_back(RI);
24210b57cec5SDimitry Andric         continue;
24220b57cec5SDimitry Andric       }
24230b57cec5SDimitry Andric       if (!NeedBitCast)
24240b57cec5SDimitry Andric         continue;
24250b57cec5SDimitry Andric 
24260b57cec5SDimitry Andric       // Delete the old return and any preceding bitcast.
24270b57cec5SDimitry Andric       BasicBlock *CurBB = RI->getParent();
24280b57cec5SDimitry Andric       auto *OldCast = dyn_cast_or_null<BitCastInst>(RI->getReturnValue());
24290b57cec5SDimitry Andric       RI->eraseFromParent();
24300b57cec5SDimitry Andric       if (OldCast)
24310b57cec5SDimitry Andric         OldCast->eraseFromParent();
24320b57cec5SDimitry Andric 
24330b57cec5SDimitry Andric       // Insert a new bitcast and return with the right type.
24340b57cec5SDimitry Andric       IRBuilder<> Builder(CurBB);
24350b57cec5SDimitry Andric       Builder.CreateRet(Builder.CreateBitCast(ReturnedMustTail, NewRetTy));
24360b57cec5SDimitry Andric     }
24370b57cec5SDimitry Andric 
24380b57cec5SDimitry Andric     // Leave behind the normal returns so we can merge control flow.
24390b57cec5SDimitry Andric     std::swap(Returns, NormalReturns);
24400b57cec5SDimitry Andric   }
24410b57cec5SDimitry Andric 
24420b57cec5SDimitry Andric   // Now that all of the transforms on the inlined code have taken place but
24430b57cec5SDimitry Andric   // before we splice the inlined code into the CFG and lose track of which
24440b57cec5SDimitry Andric   // blocks were actually inlined, collect the call sites. We only do this if
24450b57cec5SDimitry Andric   // call graph updates weren't requested, as those provide value handle based
2446*5f7ddb14SDimitry Andric   // tracking of inlined call sites instead. Calls to intrinsics are not
2447*5f7ddb14SDimitry Andric   // collected because they are not inlineable.
24480b57cec5SDimitry Andric   if (InlinedFunctionInfo.ContainsCalls && !IFI.CG) {
24490b57cec5SDimitry Andric     // Otherwise just collect the raw call sites that were inlined.
24500b57cec5SDimitry Andric     for (BasicBlock &NewBB :
24510b57cec5SDimitry Andric          make_range(FirstNewBlock->getIterator(), Caller->end()))
24520b57cec5SDimitry Andric       for (Instruction &I : NewBB)
24535ffd83dbSDimitry Andric         if (auto *CB = dyn_cast<CallBase>(&I))
2454*5f7ddb14SDimitry Andric           if (!(CB->getCalledFunction() &&
2455*5f7ddb14SDimitry Andric                 CB->getCalledFunction()->isIntrinsic()))
24565ffd83dbSDimitry Andric             IFI.InlinedCallSites.push_back(CB);
24570b57cec5SDimitry Andric   }
24580b57cec5SDimitry Andric 
24590b57cec5SDimitry Andric   // If we cloned in _exactly one_ basic block, and if that block ends in a
24600b57cec5SDimitry Andric   // return instruction, we splice the body of the inlined callee directly into
24610b57cec5SDimitry Andric   // the calling basic block.
24620b57cec5SDimitry Andric   if (Returns.size() == 1 && std::distance(FirstNewBlock, Caller->end()) == 1) {
24630b57cec5SDimitry Andric     // Move all of the instructions right before the call.
24645ffd83dbSDimitry Andric     OrigBB->getInstList().splice(CB.getIterator(), FirstNewBlock->getInstList(),
24650b57cec5SDimitry Andric                                  FirstNewBlock->begin(), FirstNewBlock->end());
24660b57cec5SDimitry Andric     // Remove the cloned basic block.
24670b57cec5SDimitry Andric     Caller->getBasicBlockList().pop_back();
24680b57cec5SDimitry Andric 
24690b57cec5SDimitry Andric     // If the call site was an invoke instruction, add a branch to the normal
24700b57cec5SDimitry Andric     // destination.
24715ffd83dbSDimitry Andric     if (InvokeInst *II = dyn_cast<InvokeInst>(&CB)) {
24725ffd83dbSDimitry Andric       BranchInst *NewBr = BranchInst::Create(II->getNormalDest(), &CB);
24730b57cec5SDimitry Andric       NewBr->setDebugLoc(Returns[0]->getDebugLoc());
24740b57cec5SDimitry Andric     }
24750b57cec5SDimitry Andric 
24760b57cec5SDimitry Andric     // If the return instruction returned a value, replace uses of the call with
24770b57cec5SDimitry Andric     // uses of the returned value.
24785ffd83dbSDimitry Andric     if (!CB.use_empty()) {
24790b57cec5SDimitry Andric       ReturnInst *R = Returns[0];
24805ffd83dbSDimitry Andric       if (&CB == R->getReturnValue())
24815ffd83dbSDimitry Andric         CB.replaceAllUsesWith(UndefValue::get(CB.getType()));
24820b57cec5SDimitry Andric       else
24835ffd83dbSDimitry Andric         CB.replaceAllUsesWith(R->getReturnValue());
24840b57cec5SDimitry Andric     }
24850b57cec5SDimitry Andric     // Since we are now done with the Call/Invoke, we can delete it.
24865ffd83dbSDimitry Andric     CB.eraseFromParent();
24870b57cec5SDimitry Andric 
24880b57cec5SDimitry Andric     // Since we are now done with the return instruction, delete it also.
24890b57cec5SDimitry Andric     Returns[0]->eraseFromParent();
24900b57cec5SDimitry Andric 
24910b57cec5SDimitry Andric     // We are now done with the inlining.
24925ffd83dbSDimitry Andric     return InlineResult::success();
24930b57cec5SDimitry Andric   }
24940b57cec5SDimitry Andric 
24950b57cec5SDimitry Andric   // Otherwise, we have the normal case, of more than one block to inline or
24960b57cec5SDimitry Andric   // multiple return sites.
24970b57cec5SDimitry Andric 
24980b57cec5SDimitry Andric   // We want to clone the entire callee function into the hole between the
24990b57cec5SDimitry Andric   // "starter" and "ender" blocks.  How we accomplish this depends on whether
25000b57cec5SDimitry Andric   // this is an invoke instruction or a call instruction.
25010b57cec5SDimitry Andric   BasicBlock *AfterCallBB;
25020b57cec5SDimitry Andric   BranchInst *CreatedBranchToNormalDest = nullptr;
25035ffd83dbSDimitry Andric   if (InvokeInst *II = dyn_cast<InvokeInst>(&CB)) {
25040b57cec5SDimitry Andric 
25050b57cec5SDimitry Andric     // Add an unconditional branch to make this look like the CallInst case...
25065ffd83dbSDimitry Andric     CreatedBranchToNormalDest = BranchInst::Create(II->getNormalDest(), &CB);
25070b57cec5SDimitry Andric 
25080b57cec5SDimitry Andric     // Split the basic block.  This guarantees that no PHI nodes will have to be
25090b57cec5SDimitry Andric     // updated due to new incoming edges, and make the invoke case more
25100b57cec5SDimitry Andric     // symmetric to the call case.
25110b57cec5SDimitry Andric     AfterCallBB =
25120b57cec5SDimitry Andric         OrigBB->splitBasicBlock(CreatedBranchToNormalDest->getIterator(),
25130b57cec5SDimitry Andric                                 CalledFunc->getName() + ".exit");
25140b57cec5SDimitry Andric 
25150b57cec5SDimitry Andric   } else { // It's a call
25160b57cec5SDimitry Andric     // If this is a call instruction, we need to split the basic block that
25170b57cec5SDimitry Andric     // the call lives in.
25180b57cec5SDimitry Andric     //
25195ffd83dbSDimitry Andric     AfterCallBB = OrigBB->splitBasicBlock(CB.getIterator(),
25200b57cec5SDimitry Andric                                           CalledFunc->getName() + ".exit");
25210b57cec5SDimitry Andric   }
25220b57cec5SDimitry Andric 
25230b57cec5SDimitry Andric   if (IFI.CallerBFI) {
25240b57cec5SDimitry Andric     // Copy original BB's block frequency to AfterCallBB
25250b57cec5SDimitry Andric     IFI.CallerBFI->setBlockFreq(
25260b57cec5SDimitry Andric         AfterCallBB, IFI.CallerBFI->getBlockFreq(OrigBB).getFrequency());
25270b57cec5SDimitry Andric   }
25280b57cec5SDimitry Andric 
25290b57cec5SDimitry Andric   // Change the branch that used to go to AfterCallBB to branch to the first
25300b57cec5SDimitry Andric   // basic block of the inlined function.
25310b57cec5SDimitry Andric   //
25320b57cec5SDimitry Andric   Instruction *Br = OrigBB->getTerminator();
25330b57cec5SDimitry Andric   assert(Br && Br->getOpcode() == Instruction::Br &&
25340b57cec5SDimitry Andric          "splitBasicBlock broken!");
25350b57cec5SDimitry Andric   Br->setOperand(0, &*FirstNewBlock);
25360b57cec5SDimitry Andric 
25370b57cec5SDimitry Andric   // Now that the function is correct, make it a little bit nicer.  In
25380b57cec5SDimitry Andric   // particular, move the basic blocks inserted from the end of the function
25390b57cec5SDimitry Andric   // into the space made by splitting the source basic block.
25400b57cec5SDimitry Andric   Caller->getBasicBlockList().splice(AfterCallBB->getIterator(),
25410b57cec5SDimitry Andric                                      Caller->getBasicBlockList(), FirstNewBlock,
25420b57cec5SDimitry Andric                                      Caller->end());
25430b57cec5SDimitry Andric 
25440b57cec5SDimitry Andric   // Handle all of the return instructions that we just cloned in, and eliminate
25450b57cec5SDimitry Andric   // any users of the original call/invoke instruction.
25460b57cec5SDimitry Andric   Type *RTy = CalledFunc->getReturnType();
25470b57cec5SDimitry Andric 
25480b57cec5SDimitry Andric   PHINode *PHI = nullptr;
25490b57cec5SDimitry Andric   if (Returns.size() > 1) {
25500b57cec5SDimitry Andric     // The PHI node should go at the front of the new basic block to merge all
25510b57cec5SDimitry Andric     // possible incoming values.
25525ffd83dbSDimitry Andric     if (!CB.use_empty()) {
25535ffd83dbSDimitry Andric       PHI = PHINode::Create(RTy, Returns.size(), CB.getName(),
25540b57cec5SDimitry Andric                             &AfterCallBB->front());
25550b57cec5SDimitry Andric       // Anything that used the result of the function call should now use the
25560b57cec5SDimitry Andric       // PHI node as their operand.
25575ffd83dbSDimitry Andric       CB.replaceAllUsesWith(PHI);
25580b57cec5SDimitry Andric     }
25590b57cec5SDimitry Andric 
25600b57cec5SDimitry Andric     // Loop over all of the return instructions adding entries to the PHI node
25610b57cec5SDimitry Andric     // as appropriate.
25620b57cec5SDimitry Andric     if (PHI) {
25630b57cec5SDimitry Andric       for (unsigned i = 0, e = Returns.size(); i != e; ++i) {
25640b57cec5SDimitry Andric         ReturnInst *RI = Returns[i];
25650b57cec5SDimitry Andric         assert(RI->getReturnValue()->getType() == PHI->getType() &&
25660b57cec5SDimitry Andric                "Ret value not consistent in function!");
25670b57cec5SDimitry Andric         PHI->addIncoming(RI->getReturnValue(), RI->getParent());
25680b57cec5SDimitry Andric       }
25690b57cec5SDimitry Andric     }
25700b57cec5SDimitry Andric 
25710b57cec5SDimitry Andric     // Add a branch to the merge points and remove return instructions.
25720b57cec5SDimitry Andric     DebugLoc Loc;
25730b57cec5SDimitry Andric     for (unsigned i = 0, e = Returns.size(); i != e; ++i) {
25740b57cec5SDimitry Andric       ReturnInst *RI = Returns[i];
25750b57cec5SDimitry Andric       BranchInst* BI = BranchInst::Create(AfterCallBB, RI);
25760b57cec5SDimitry Andric       Loc = RI->getDebugLoc();
25770b57cec5SDimitry Andric       BI->setDebugLoc(Loc);
25780b57cec5SDimitry Andric       RI->eraseFromParent();
25790b57cec5SDimitry Andric     }
25800b57cec5SDimitry Andric     // We need to set the debug location to *somewhere* inside the
25810b57cec5SDimitry Andric     // inlined function. The line number may be nonsensical, but the
25820b57cec5SDimitry Andric     // instruction will at least be associated with the right
25830b57cec5SDimitry Andric     // function.
25840b57cec5SDimitry Andric     if (CreatedBranchToNormalDest)
25850b57cec5SDimitry Andric       CreatedBranchToNormalDest->setDebugLoc(Loc);
25860b57cec5SDimitry Andric   } else if (!Returns.empty()) {
25870b57cec5SDimitry Andric     // Otherwise, if there is exactly one return value, just replace anything
25880b57cec5SDimitry Andric     // using the return value of the call with the computed value.
25895ffd83dbSDimitry Andric     if (!CB.use_empty()) {
25905ffd83dbSDimitry Andric       if (&CB == Returns[0]->getReturnValue())
25915ffd83dbSDimitry Andric         CB.replaceAllUsesWith(UndefValue::get(CB.getType()));
25920b57cec5SDimitry Andric       else
25935ffd83dbSDimitry Andric         CB.replaceAllUsesWith(Returns[0]->getReturnValue());
25940b57cec5SDimitry Andric     }
25950b57cec5SDimitry Andric 
25960b57cec5SDimitry Andric     // Update PHI nodes that use the ReturnBB to use the AfterCallBB.
25970b57cec5SDimitry Andric     BasicBlock *ReturnBB = Returns[0]->getParent();
25980b57cec5SDimitry Andric     ReturnBB->replaceAllUsesWith(AfterCallBB);
25990b57cec5SDimitry Andric 
26000b57cec5SDimitry Andric     // Splice the code from the return block into the block that it will return
26010b57cec5SDimitry Andric     // to, which contains the code that was after the call.
26020b57cec5SDimitry Andric     AfterCallBB->getInstList().splice(AfterCallBB->begin(),
26030b57cec5SDimitry Andric                                       ReturnBB->getInstList());
26040b57cec5SDimitry Andric 
26050b57cec5SDimitry Andric     if (CreatedBranchToNormalDest)
26060b57cec5SDimitry Andric       CreatedBranchToNormalDest->setDebugLoc(Returns[0]->getDebugLoc());
26070b57cec5SDimitry Andric 
26080b57cec5SDimitry Andric     // Delete the return instruction now and empty ReturnBB now.
26090b57cec5SDimitry Andric     Returns[0]->eraseFromParent();
26100b57cec5SDimitry Andric     ReturnBB->eraseFromParent();
26115ffd83dbSDimitry Andric   } else if (!CB.use_empty()) {
26120b57cec5SDimitry Andric     // No returns, but something is using the return value of the call.  Just
26130b57cec5SDimitry Andric     // nuke the result.
26145ffd83dbSDimitry Andric     CB.replaceAllUsesWith(UndefValue::get(CB.getType()));
26150b57cec5SDimitry Andric   }
26160b57cec5SDimitry Andric 
26170b57cec5SDimitry Andric   // Since we are now done with the Call/Invoke, we can delete it.
26185ffd83dbSDimitry Andric   CB.eraseFromParent();
26190b57cec5SDimitry Andric 
26200b57cec5SDimitry Andric   // If we inlined any musttail calls and the original return is now
26210b57cec5SDimitry Andric   // unreachable, delete it.  It can only contain a bitcast and ret.
2622af732203SDimitry Andric   if (InlinedMustTailCalls && pred_empty(AfterCallBB))
26230b57cec5SDimitry Andric     AfterCallBB->eraseFromParent();
26240b57cec5SDimitry Andric 
26250b57cec5SDimitry Andric   // We should always be able to fold the entry block of the function into the
26260b57cec5SDimitry Andric   // single predecessor of the block...
26270b57cec5SDimitry Andric   assert(cast<BranchInst>(Br)->isUnconditional() && "splitBasicBlock broken!");
26280b57cec5SDimitry Andric   BasicBlock *CalleeEntry = cast<BranchInst>(Br)->getSuccessor(0);
26290b57cec5SDimitry Andric 
26300b57cec5SDimitry Andric   // Splice the code entry block into calling block, right before the
26310b57cec5SDimitry Andric   // unconditional branch.
26320b57cec5SDimitry Andric   CalleeEntry->replaceAllUsesWith(OrigBB);  // Update PHI nodes
26330b57cec5SDimitry Andric   OrigBB->getInstList().splice(Br->getIterator(), CalleeEntry->getInstList());
26340b57cec5SDimitry Andric 
26350b57cec5SDimitry Andric   // Remove the unconditional branch.
26360b57cec5SDimitry Andric   OrigBB->getInstList().erase(Br);
26370b57cec5SDimitry Andric 
26380b57cec5SDimitry Andric   // Now we can remove the CalleeEntry block, which is now empty.
26390b57cec5SDimitry Andric   Caller->getBasicBlockList().erase(CalleeEntry);
26400b57cec5SDimitry Andric 
26410b57cec5SDimitry Andric   // If we inserted a phi node, check to see if it has a single value (e.g. all
26420b57cec5SDimitry Andric   // the entries are the same or undef).  If so, remove the PHI so it doesn't
26430b57cec5SDimitry Andric   // block other optimizations.
26440b57cec5SDimitry Andric   if (PHI) {
26450b57cec5SDimitry Andric     AssumptionCache *AC =
26465ffd83dbSDimitry Andric         IFI.GetAssumptionCache ? &IFI.GetAssumptionCache(*Caller) : nullptr;
26470b57cec5SDimitry Andric     auto &DL = Caller->getParent()->getDataLayout();
26480b57cec5SDimitry Andric     if (Value *V = SimplifyInstruction(PHI, {DL, nullptr, nullptr, AC})) {
26490b57cec5SDimitry Andric       PHI->replaceAllUsesWith(V);
26500b57cec5SDimitry Andric       PHI->eraseFromParent();
26510b57cec5SDimitry Andric     }
26520b57cec5SDimitry Andric   }
26530b57cec5SDimitry Andric 
26545ffd83dbSDimitry Andric   return InlineResult::success();
26550b57cec5SDimitry Andric }
2656