1f22ef01cSRoman Divacky //===-- MachineFunction.cpp -----------------------------------------------===// 2f22ef01cSRoman Divacky // 3f22ef01cSRoman Divacky // The LLVM Compiler Infrastructure 4f22ef01cSRoman Divacky // 5f22ef01cSRoman Divacky // This file is distributed under the University of Illinois Open Source 6f22ef01cSRoman Divacky // License. See LICENSE.TXT for details. 7f22ef01cSRoman Divacky // 8f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 9f22ef01cSRoman Divacky // 10f22ef01cSRoman Divacky // Collect native machine code information for a function. This allows 11f22ef01cSRoman Divacky // target-specific information about the generated code to be stored with each 12f22ef01cSRoman Divacky // function. 13f22ef01cSRoman Divacky // 14f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 15f22ef01cSRoman Divacky 16f22ef01cSRoman Divacky #include "llvm/CodeGen/MachineFunction.h" 17139f7f9bSDimitry Andric #include "llvm/ADT/STLExtras.h" 18139f7f9bSDimitry Andric #include "llvm/ADT/SmallString.h" 19139f7f9bSDimitry Andric #include "llvm/Analysis/ConstantFolding.h" 207d523365SDimitry Andric #include "llvm/Analysis/EHPersonalities.h" 21dff0c46cSDimitry Andric #include "llvm/CodeGen/MachineConstantPool.h" 22f22ef01cSRoman Divacky #include "llvm/CodeGen/MachineFrameInfo.h" 238f0fd8f6SDimitry Andric #include "llvm/CodeGen/MachineFunctionInitializer.h" 24139f7f9bSDimitry Andric #include "llvm/CodeGen/MachineFunctionPass.h" 25f22ef01cSRoman Divacky #include "llvm/CodeGen/MachineInstr.h" 26f22ef01cSRoman Divacky #include "llvm/CodeGen/MachineJumpTableInfo.h" 27f22ef01cSRoman Divacky #include "llvm/CodeGen/MachineModuleInfo.h" 28f22ef01cSRoman Divacky #include "llvm/CodeGen/MachineRegisterInfo.h" 29f22ef01cSRoman Divacky #include "llvm/CodeGen/Passes.h" 307d523365SDimitry Andric #include "llvm/CodeGen/PseudoSourceValue.h" 317d523365SDimitry Andric #include "llvm/CodeGen/WinEHFuncInfo.h" 32139f7f9bSDimitry Andric #include "llvm/IR/DataLayout.h" 3391bc56edSDimitry Andric #include "llvm/IR/DebugInfo.h" 34139f7f9bSDimitry Andric #include "llvm/IR/Function.h" 35875ed548SDimitry Andric #include "llvm/IR/Module.h" 363dac3a9bSDimitry Andric #include "llvm/IR/ModuleSlotTracker.h" 37f22ef01cSRoman Divacky #include "llvm/MC/MCAsmInfo.h" 38f22ef01cSRoman Divacky #include "llvm/MC/MCContext.h" 39f22ef01cSRoman Divacky #include "llvm/Support/Debug.h" 40f22ef01cSRoman Divacky #include "llvm/Support/GraphWriter.h" 41f22ef01cSRoman Divacky #include "llvm/Support/raw_ostream.h" 42139f7f9bSDimitry Andric #include "llvm/Target/TargetFrameLowering.h" 43139f7f9bSDimitry Andric #include "llvm/Target/TargetLowering.h" 44139f7f9bSDimitry Andric #include "llvm/Target/TargetMachine.h" 4539d628a0SDimitry Andric #include "llvm/Target/TargetSubtargetInfo.h" 46f22ef01cSRoman Divacky using namespace llvm; 47f22ef01cSRoman Divacky 4891bc56edSDimitry Andric #define DEBUG_TYPE "codegen" 4991bc56edSDimitry Andric 507d523365SDimitry Andric static cl::opt<unsigned> 517d523365SDimitry Andric AlignAllFunctions("align-all-functions", 527d523365SDimitry Andric cl::desc("Force the alignment of all functions."), 537d523365SDimitry Andric cl::init(0), cl::Hidden); 547d523365SDimitry Andric 558f0fd8f6SDimitry Andric void MachineFunctionInitializer::anchor() {} 568f0fd8f6SDimitry Andric 57d88c1a5aSDimitry Andric static const char *getPropertyName(MachineFunctionProperties::Property Prop) { 58d88c1a5aSDimitry Andric typedef MachineFunctionProperties::Property P; 59d88c1a5aSDimitry Andric switch(Prop) { 60d88c1a5aSDimitry Andric case P::FailedISel: return "FailedISel"; 61d88c1a5aSDimitry Andric case P::IsSSA: return "IsSSA"; 62d88c1a5aSDimitry Andric case P::Legalized: return "Legalized"; 63d88c1a5aSDimitry Andric case P::NoPHIs: return "NoPHIs"; 64d88c1a5aSDimitry Andric case P::NoVRegs: return "NoVRegs"; 65d88c1a5aSDimitry Andric case P::RegBankSelected: return "RegBankSelected"; 66d88c1a5aSDimitry Andric case P::Selected: return "Selected"; 67d88c1a5aSDimitry Andric case P::TracksLiveness: return "TracksLiveness"; 68d88c1a5aSDimitry Andric } 69d88c1a5aSDimitry Andric llvm_unreachable("Invalid machine function property"); 70d88c1a5aSDimitry Andric } 71d88c1a5aSDimitry Andric 72d88c1a5aSDimitry Andric void MachineFunctionProperties::print(raw_ostream &OS) const { 73d88c1a5aSDimitry Andric const char *Separator = ""; 74d88c1a5aSDimitry Andric for (BitVector::size_type I = 0; I < Properties.size(); ++I) { 75d88c1a5aSDimitry Andric if (!Properties[I]) 763ca95b02SDimitry Andric continue; 77d88c1a5aSDimitry Andric OS << Separator << getPropertyName(static_cast<Property>(I)); 78d88c1a5aSDimitry Andric Separator = ", "; 793ca95b02SDimitry Andric } 803ca95b02SDimitry Andric } 813ca95b02SDimitry Andric 82f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 83f22ef01cSRoman Divacky // MachineFunction implementation 84f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 85f22ef01cSRoman Divacky 868f0fd8f6SDimitry Andric // Out-of-line virtual method. 87f22ef01cSRoman Divacky MachineFunctionInfo::~MachineFunctionInfo() {} 88f22ef01cSRoman Divacky 89d88c1a5aSDimitry Andric void ilist_alloc_traits<MachineBasicBlock>::deleteNode(MachineBasicBlock *MBB) { 90f22ef01cSRoman Divacky MBB->getParent()->DeleteMachineBasicBlock(MBB); 91f22ef01cSRoman Divacky } 92f22ef01cSRoman Divacky 933ca95b02SDimitry Andric static inline unsigned getFnStackAlignment(const TargetSubtargetInfo *STI, 943ca95b02SDimitry Andric const Function *Fn) { 953ca95b02SDimitry Andric if (Fn->hasFnAttribute(Attribute::StackAlignment)) 963ca95b02SDimitry Andric return Fn->getFnStackAlignment(); 973ca95b02SDimitry Andric return STI->getFrameLowering()->getStackAlignment(); 983ca95b02SDimitry Andric } 993ca95b02SDimitry Andric 100f22ef01cSRoman Divacky MachineFunction::MachineFunction(const Function *F, const TargetMachine &TM, 10139d628a0SDimitry Andric unsigned FunctionNum, MachineModuleInfo &mmi) 102ff0cc061SDimitry Andric : Fn(F), Target(TM), STI(TM.getSubtargetImpl(*F)), Ctx(mmi.getContext()), 10339d628a0SDimitry Andric MMI(mmi) { 104d88c1a5aSDimitry Andric FunctionNumber = FunctionNum; 105d88c1a5aSDimitry Andric init(); 106d88c1a5aSDimitry Andric } 107d88c1a5aSDimitry Andric 108d88c1a5aSDimitry Andric void MachineFunction::init() { 1093ca95b02SDimitry Andric // Assume the function starts in SSA form with correct liveness. 1103ca95b02SDimitry Andric Properties.set(MachineFunctionProperties::Property::IsSSA); 1113ca95b02SDimitry Andric Properties.set(MachineFunctionProperties::Property::TracksLiveness); 11239d628a0SDimitry Andric if (STI->getRegisterInfo()) 11339d628a0SDimitry Andric RegInfo = new (Allocator) MachineRegisterInfo(this); 114f22ef01cSRoman Divacky else 11591bc56edSDimitry Andric RegInfo = nullptr; 116f785676fSDimitry Andric 11791bc56edSDimitry Andric MFInfo = nullptr; 1183ca95b02SDimitry Andric // We can realign the stack if the target supports it and the user hasn't 1193ca95b02SDimitry Andric // explicitly asked us not to. 1203ca95b02SDimitry Andric bool CanRealignSP = STI->getFrameLowering()->isStackRealignable() && 121d88c1a5aSDimitry Andric !Fn->hasFnAttribute("no-realign-stack"); 1223ca95b02SDimitry Andric FrameInfo = new (Allocator) MachineFrameInfo( 1233ca95b02SDimitry Andric getFnStackAlignment(STI, Fn), /*StackRealignable=*/CanRealignSP, 1243ca95b02SDimitry Andric /*ForceRealign=*/CanRealignSP && 125d88c1a5aSDimitry Andric Fn->hasFnAttribute(Attribute::StackAlignment)); 126f785676fSDimitry Andric 127ff0cc061SDimitry Andric if (Fn->hasFnAttribute(Attribute::StackAlignment)) 128ff0cc061SDimitry Andric FrameInfo->ensureMaxAlignment(Fn->getFnStackAlignment()); 129f785676fSDimitry Andric 130875ed548SDimitry Andric ConstantPool = new (Allocator) MachineConstantPool(getDataLayout()); 13139d628a0SDimitry Andric Alignment = STI->getTargetLowering()->getMinFunctionAlignment(); 132f785676fSDimitry Andric 133bd5abe19SDimitry Andric // FIXME: Shouldn't use pref alignment if explicit alignment is set on Fn. 1347d523365SDimitry Andric // FIXME: Use Function::optForSize(). 135ff0cc061SDimitry Andric if (!Fn->hasFnAttribute(Attribute::OptimizeForSize)) 136bd5abe19SDimitry Andric Alignment = std::max(Alignment, 13739d628a0SDimitry Andric STI->getTargetLowering()->getPrefFunctionAlignment()); 138f785676fSDimitry Andric 1397d523365SDimitry Andric if (AlignAllFunctions) 1407d523365SDimitry Andric Alignment = AlignAllFunctions; 1417d523365SDimitry Andric 14291bc56edSDimitry Andric JumpTableInfo = nullptr; 1437d523365SDimitry Andric 1447d523365SDimitry Andric if (isFuncletEHPersonality(classifyEHPersonality( 145d88c1a5aSDimitry Andric Fn->hasPersonalityFn() ? Fn->getPersonalityFn() : nullptr))) { 1467d523365SDimitry Andric WinEHInfo = new (Allocator) WinEHFuncInfo(); 1477d523365SDimitry Andric } 1487d523365SDimitry Andric 149d88c1a5aSDimitry Andric assert(Target.isCompatibleDataLayout(getDataLayout()) && 1507d523365SDimitry Andric "Can't create a MachineFunction using a Module with a " 1517d523365SDimitry Andric "Target-incompatible DataLayout attached\n"); 1527d523365SDimitry Andric 1537d523365SDimitry Andric PSVManager = llvm::make_unique<PseudoSourceValueManager>(); 154f22ef01cSRoman Divacky } 155f22ef01cSRoman Divacky 156f22ef01cSRoman Divacky MachineFunction::~MachineFunction() { 157d88c1a5aSDimitry Andric clear(); 158d88c1a5aSDimitry Andric } 159d88c1a5aSDimitry Andric 160d88c1a5aSDimitry Andric void MachineFunction::clear() { 161d88c1a5aSDimitry Andric Properties.reset(); 162139f7f9bSDimitry Andric // Don't call destructors on MachineInstr and MachineOperand. All of their 163139f7f9bSDimitry Andric // memory comes from the BumpPtrAllocator which is about to be purged. 164139f7f9bSDimitry Andric // 165139f7f9bSDimitry Andric // Do call MachineBasicBlock destructors, it contains std::vectors. 166139f7f9bSDimitry Andric for (iterator I = begin(), E = end(); I != E; I = BasicBlocks.erase(I)) 167139f7f9bSDimitry Andric I->Insts.clearAndLeakNodesUnsafely(); 168139f7f9bSDimitry Andric 169f22ef01cSRoman Divacky InstructionRecycler.clear(Allocator); 170139f7f9bSDimitry Andric OperandRecycler.clear(Allocator); 171f22ef01cSRoman Divacky BasicBlockRecycler.clear(Allocator); 1727a7e6055SDimitry Andric VariableDbgInfos.clear(); 173f22ef01cSRoman Divacky if (RegInfo) { 174f22ef01cSRoman Divacky RegInfo->~MachineRegisterInfo(); 175f22ef01cSRoman Divacky Allocator.Deallocate(RegInfo); 176f22ef01cSRoman Divacky } 177f22ef01cSRoman Divacky if (MFInfo) { 178f22ef01cSRoman Divacky MFInfo->~MachineFunctionInfo(); 179f22ef01cSRoman Divacky Allocator.Deallocate(MFInfo); 180f22ef01cSRoman Divacky } 1817ae0e2c9SDimitry Andric 1827ae0e2c9SDimitry Andric FrameInfo->~MachineFrameInfo(); 1837ae0e2c9SDimitry Andric Allocator.Deallocate(FrameInfo); 1847ae0e2c9SDimitry Andric 1857ae0e2c9SDimitry Andric ConstantPool->~MachineConstantPool(); 1867ae0e2c9SDimitry Andric Allocator.Deallocate(ConstantPool); 187f22ef01cSRoman Divacky 188f22ef01cSRoman Divacky if (JumpTableInfo) { 189f22ef01cSRoman Divacky JumpTableInfo->~MachineJumpTableInfo(); 190f22ef01cSRoman Divacky Allocator.Deallocate(JumpTableInfo); 191f22ef01cSRoman Divacky } 1927d523365SDimitry Andric 1937d523365SDimitry Andric if (WinEHInfo) { 1947d523365SDimitry Andric WinEHInfo->~WinEHFuncInfo(); 1957d523365SDimitry Andric Allocator.Deallocate(WinEHInfo); 1967d523365SDimitry Andric } 197f22ef01cSRoman Divacky } 198f22ef01cSRoman Divacky 199875ed548SDimitry Andric const DataLayout &MachineFunction::getDataLayout() const { 200875ed548SDimitry Andric return Fn->getParent()->getDataLayout(); 201875ed548SDimitry Andric } 202875ed548SDimitry Andric 2038f0fd8f6SDimitry Andric /// Get the JumpTableInfo for this function. 2048f0fd8f6SDimitry Andric /// If it does not already exist, allocate one. 205f22ef01cSRoman Divacky MachineJumpTableInfo *MachineFunction:: 206f22ef01cSRoman Divacky getOrCreateJumpTableInfo(unsigned EntryKind) { 207f22ef01cSRoman Divacky if (JumpTableInfo) return JumpTableInfo; 208f22ef01cSRoman Divacky 209f22ef01cSRoman Divacky JumpTableInfo = new (Allocator) 210f22ef01cSRoman Divacky MachineJumpTableInfo((MachineJumpTableInfo::JTEntryKind)EntryKind); 211f22ef01cSRoman Divacky return JumpTableInfo; 212f22ef01cSRoman Divacky } 213f22ef01cSRoman Divacky 21491bc56edSDimitry Andric /// Should we be emitting segmented stack stuff for the function 2158c24ff90SDimitry Andric bool MachineFunction::shouldSplitStack() const { 21691bc56edSDimitry Andric return getFunction()->hasFnAttribute("split-stack"); 21791bc56edSDimitry Andric } 21891bc56edSDimitry Andric 2198f0fd8f6SDimitry Andric /// This discards all of the MachineBasicBlock numbers and recomputes them. 2208f0fd8f6SDimitry Andric /// This guarantees that the MBB numbers are sequential, dense, and match the 2218f0fd8f6SDimitry Andric /// ordering of the blocks within the function. If a specific MachineBasicBlock 2228f0fd8f6SDimitry Andric /// is specified, only that block and those after it are renumbered. 223f22ef01cSRoman Divacky void MachineFunction::RenumberBlocks(MachineBasicBlock *MBB) { 224f22ef01cSRoman Divacky if (empty()) { MBBNumbering.clear(); return; } 225f22ef01cSRoman Divacky MachineFunction::iterator MBBI, E = end(); 22691bc56edSDimitry Andric if (MBB == nullptr) 227f22ef01cSRoman Divacky MBBI = begin(); 228f22ef01cSRoman Divacky else 2297d523365SDimitry Andric MBBI = MBB->getIterator(); 230f22ef01cSRoman Divacky 231f22ef01cSRoman Divacky // Figure out the block number this should have. 232f22ef01cSRoman Divacky unsigned BlockNo = 0; 233f22ef01cSRoman Divacky if (MBBI != begin()) 23491bc56edSDimitry Andric BlockNo = std::prev(MBBI)->getNumber() + 1; 235f22ef01cSRoman Divacky 236f22ef01cSRoman Divacky for (; MBBI != E; ++MBBI, ++BlockNo) { 237f22ef01cSRoman Divacky if (MBBI->getNumber() != (int)BlockNo) { 238f22ef01cSRoman Divacky // Remove use of the old number. 239f22ef01cSRoman Divacky if (MBBI->getNumber() != -1) { 240f22ef01cSRoman Divacky assert(MBBNumbering[MBBI->getNumber()] == &*MBBI && 241f22ef01cSRoman Divacky "MBB number mismatch!"); 24291bc56edSDimitry Andric MBBNumbering[MBBI->getNumber()] = nullptr; 243f22ef01cSRoman Divacky } 244f22ef01cSRoman Divacky 245f22ef01cSRoman Divacky // If BlockNo is already taken, set that block's number to -1. 246f22ef01cSRoman Divacky if (MBBNumbering[BlockNo]) 247f22ef01cSRoman Divacky MBBNumbering[BlockNo]->setNumber(-1); 248f22ef01cSRoman Divacky 2497d523365SDimitry Andric MBBNumbering[BlockNo] = &*MBBI; 250f22ef01cSRoman Divacky MBBI->setNumber(BlockNo); 251f22ef01cSRoman Divacky } 252f22ef01cSRoman Divacky } 253f22ef01cSRoman Divacky 254f22ef01cSRoman Divacky // Okay, all the blocks are renumbered. If we have compactified the block 255f22ef01cSRoman Divacky // numbering, shrink MBBNumbering now. 256f22ef01cSRoman Divacky assert(BlockNo <= MBBNumbering.size() && "Mismatch!"); 257f22ef01cSRoman Divacky MBBNumbering.resize(BlockNo); 258f22ef01cSRoman Divacky } 259f22ef01cSRoman Divacky 2608f0fd8f6SDimitry Andric /// Allocate a new MachineInstr. Use this instead of `new MachineInstr'. 2613ca95b02SDimitry Andric MachineInstr *MachineFunction::CreateMachineInstr(const MCInstrDesc &MCID, 2623ca95b02SDimitry Andric const DebugLoc &DL, 2633ca95b02SDimitry Andric bool NoImp) { 264f22ef01cSRoman Divacky return new (InstructionRecycler.Allocate<MachineInstr>(Allocator)) 265139f7f9bSDimitry Andric MachineInstr(*this, MCID, DL, NoImp); 266f22ef01cSRoman Divacky } 267f22ef01cSRoman Divacky 2688f0fd8f6SDimitry Andric /// Create a new MachineInstr which is a copy of the 'Orig' instruction, 2698f0fd8f6SDimitry Andric /// identical in all ways except the instruction has no parent, prev, or next. 270f22ef01cSRoman Divacky MachineInstr * 271f22ef01cSRoman Divacky MachineFunction::CloneMachineInstr(const MachineInstr *Orig) { 272f22ef01cSRoman Divacky return new (InstructionRecycler.Allocate<MachineInstr>(Allocator)) 273f22ef01cSRoman Divacky MachineInstr(*this, *Orig); 274f22ef01cSRoman Divacky } 275f22ef01cSRoman Divacky 2768f0fd8f6SDimitry Andric /// Delete the given MachineInstr. 277f22ef01cSRoman Divacky /// 278139f7f9bSDimitry Andric /// This function also serves as the MachineInstr destructor - the real 279139f7f9bSDimitry Andric /// ~MachineInstr() destructor must be empty. 280f22ef01cSRoman Divacky void 281f22ef01cSRoman Divacky MachineFunction::DeleteMachineInstr(MachineInstr *MI) { 282139f7f9bSDimitry Andric // Strip it for parts. The operand array and the MI object itself are 283139f7f9bSDimitry Andric // independently recyclable. 284139f7f9bSDimitry Andric if (MI->Operands) 285139f7f9bSDimitry Andric deallocateOperandArray(MI->CapOperands, MI->Operands); 286139f7f9bSDimitry Andric // Don't call ~MachineInstr() which must be trivial anyway because 287139f7f9bSDimitry Andric // ~MachineFunction drops whole lists of MachineInstrs wihout calling their 288139f7f9bSDimitry Andric // destructors. 289f22ef01cSRoman Divacky InstructionRecycler.Deallocate(Allocator, MI); 290f22ef01cSRoman Divacky } 291f22ef01cSRoman Divacky 2928f0fd8f6SDimitry Andric /// Allocate a new MachineBasicBlock. Use this instead of 2938f0fd8f6SDimitry Andric /// `new MachineBasicBlock'. 294f22ef01cSRoman Divacky MachineBasicBlock * 295f22ef01cSRoman Divacky MachineFunction::CreateMachineBasicBlock(const BasicBlock *bb) { 296f22ef01cSRoman Divacky return new (BasicBlockRecycler.Allocate<MachineBasicBlock>(Allocator)) 297f22ef01cSRoman Divacky MachineBasicBlock(*this, bb); 298f22ef01cSRoman Divacky } 299f22ef01cSRoman Divacky 3008f0fd8f6SDimitry Andric /// Delete the given MachineBasicBlock. 301f22ef01cSRoman Divacky void 302f22ef01cSRoman Divacky MachineFunction::DeleteMachineBasicBlock(MachineBasicBlock *MBB) { 303f22ef01cSRoman Divacky assert(MBB->getParent() == this && "MBB parent mismatch!"); 304f22ef01cSRoman Divacky MBB->~MachineBasicBlock(); 305f22ef01cSRoman Divacky BasicBlockRecycler.Deallocate(Allocator, MBB); 306f22ef01cSRoman Divacky } 307f22ef01cSRoman Divacky 3083ca95b02SDimitry Andric MachineMemOperand *MachineFunction::getMachineMemOperand( 3093ca95b02SDimitry Andric MachinePointerInfo PtrInfo, MachineMemOperand::Flags f, uint64_t s, 310d88c1a5aSDimitry Andric unsigned base_alignment, const AAMDNodes &AAInfo, const MDNode *Ranges, 311d88c1a5aSDimitry Andric SynchronizationScope SynchScope, AtomicOrdering Ordering, 312d88c1a5aSDimitry Andric AtomicOrdering FailureOrdering) { 3133ca95b02SDimitry Andric return new (Allocator) 314d88c1a5aSDimitry Andric MachineMemOperand(PtrInfo, f, s, base_alignment, AAInfo, Ranges, 315d88c1a5aSDimitry Andric SynchScope, Ordering, FailureOrdering); 316f22ef01cSRoman Divacky } 317f22ef01cSRoman Divacky 318f22ef01cSRoman Divacky MachineMemOperand * 319f22ef01cSRoman Divacky MachineFunction::getMachineMemOperand(const MachineMemOperand *MMO, 320f22ef01cSRoman Divacky int64_t Offset, uint64_t Size) { 32191bc56edSDimitry Andric if (MMO->getValue()) 322f22ef01cSRoman Divacky return new (Allocator) 3232754fe60SDimitry Andric MachineMemOperand(MachinePointerInfo(MMO->getValue(), 3242754fe60SDimitry Andric MMO->getOffset()+Offset), 325d88c1a5aSDimitry Andric MMO->getFlags(), Size, MMO->getBaseAlignment(), 326d88c1a5aSDimitry Andric AAMDNodes(), nullptr, MMO->getSynchScope(), 327d88c1a5aSDimitry Andric MMO->getOrdering(), MMO->getFailureOrdering()); 32891bc56edSDimitry Andric return new (Allocator) 32991bc56edSDimitry Andric MachineMemOperand(MachinePointerInfo(MMO->getPseudoValue(), 33091bc56edSDimitry Andric MMO->getOffset()+Offset), 331d88c1a5aSDimitry Andric MMO->getFlags(), Size, MMO->getBaseAlignment(), 332d88c1a5aSDimitry Andric AAMDNodes(), nullptr, MMO->getSynchScope(), 333d88c1a5aSDimitry Andric MMO->getOrdering(), MMO->getFailureOrdering()); 334f22ef01cSRoman Divacky } 335f22ef01cSRoman Divacky 336f22ef01cSRoman Divacky MachineInstr::mmo_iterator 337f22ef01cSRoman Divacky MachineFunction::allocateMemRefsArray(unsigned long Num) { 338f22ef01cSRoman Divacky return Allocator.Allocate<MachineMemOperand *>(Num); 339f22ef01cSRoman Divacky } 340f22ef01cSRoman Divacky 341f22ef01cSRoman Divacky std::pair<MachineInstr::mmo_iterator, MachineInstr::mmo_iterator> 342f22ef01cSRoman Divacky MachineFunction::extractLoadMemRefs(MachineInstr::mmo_iterator Begin, 343f22ef01cSRoman Divacky MachineInstr::mmo_iterator End) { 344f22ef01cSRoman Divacky // Count the number of load mem refs. 345f22ef01cSRoman Divacky unsigned Num = 0; 346f22ef01cSRoman Divacky for (MachineInstr::mmo_iterator I = Begin; I != End; ++I) 347f22ef01cSRoman Divacky if ((*I)->isLoad()) 348f22ef01cSRoman Divacky ++Num; 349f22ef01cSRoman Divacky 350f22ef01cSRoman Divacky // Allocate a new array and populate it with the load information. 351f22ef01cSRoman Divacky MachineInstr::mmo_iterator Result = allocateMemRefsArray(Num); 352f22ef01cSRoman Divacky unsigned Index = 0; 353f22ef01cSRoman Divacky for (MachineInstr::mmo_iterator I = Begin; I != End; ++I) { 354f22ef01cSRoman Divacky if ((*I)->isLoad()) { 355f22ef01cSRoman Divacky if (!(*I)->isStore()) 356f22ef01cSRoman Divacky // Reuse the MMO. 357f22ef01cSRoman Divacky Result[Index] = *I; 358f22ef01cSRoman Divacky else { 359f22ef01cSRoman Divacky // Clone the MMO and unset the store flag. 360f22ef01cSRoman Divacky MachineMemOperand *JustLoad = 3612754fe60SDimitry Andric getMachineMemOperand((*I)->getPointerInfo(), 362f22ef01cSRoman Divacky (*I)->getFlags() & ~MachineMemOperand::MOStore, 3632754fe60SDimitry Andric (*I)->getSize(), (*I)->getBaseAlignment(), 364d88c1a5aSDimitry Andric (*I)->getAAInfo(), nullptr, 365d88c1a5aSDimitry Andric (*I)->getSynchScope(), (*I)->getOrdering(), 366d88c1a5aSDimitry Andric (*I)->getFailureOrdering()); 367f22ef01cSRoman Divacky Result[Index] = JustLoad; 368f22ef01cSRoman Divacky } 369f22ef01cSRoman Divacky ++Index; 370f22ef01cSRoman Divacky } 371f22ef01cSRoman Divacky } 372f22ef01cSRoman Divacky return std::make_pair(Result, Result + Num); 373f22ef01cSRoman Divacky } 374f22ef01cSRoman Divacky 375f22ef01cSRoman Divacky std::pair<MachineInstr::mmo_iterator, MachineInstr::mmo_iterator> 376f22ef01cSRoman Divacky MachineFunction::extractStoreMemRefs(MachineInstr::mmo_iterator Begin, 377f22ef01cSRoman Divacky MachineInstr::mmo_iterator End) { 378f22ef01cSRoman Divacky // Count the number of load mem refs. 379f22ef01cSRoman Divacky unsigned Num = 0; 380f22ef01cSRoman Divacky for (MachineInstr::mmo_iterator I = Begin; I != End; ++I) 381f22ef01cSRoman Divacky if ((*I)->isStore()) 382f22ef01cSRoman Divacky ++Num; 383f22ef01cSRoman Divacky 384f22ef01cSRoman Divacky // Allocate a new array and populate it with the store information. 385f22ef01cSRoman Divacky MachineInstr::mmo_iterator Result = allocateMemRefsArray(Num); 386f22ef01cSRoman Divacky unsigned Index = 0; 387f22ef01cSRoman Divacky for (MachineInstr::mmo_iterator I = Begin; I != End; ++I) { 388f22ef01cSRoman Divacky if ((*I)->isStore()) { 389f22ef01cSRoman Divacky if (!(*I)->isLoad()) 390f22ef01cSRoman Divacky // Reuse the MMO. 391f22ef01cSRoman Divacky Result[Index] = *I; 392f22ef01cSRoman Divacky else { 393f22ef01cSRoman Divacky // Clone the MMO and unset the load flag. 394f22ef01cSRoman Divacky MachineMemOperand *JustStore = 3952754fe60SDimitry Andric getMachineMemOperand((*I)->getPointerInfo(), 396f22ef01cSRoman Divacky (*I)->getFlags() & ~MachineMemOperand::MOLoad, 3972754fe60SDimitry Andric (*I)->getSize(), (*I)->getBaseAlignment(), 398d88c1a5aSDimitry Andric (*I)->getAAInfo(), nullptr, 399d88c1a5aSDimitry Andric (*I)->getSynchScope(), (*I)->getOrdering(), 400d88c1a5aSDimitry Andric (*I)->getFailureOrdering()); 401f22ef01cSRoman Divacky Result[Index] = JustStore; 402f22ef01cSRoman Divacky } 403f22ef01cSRoman Divacky ++Index; 404f22ef01cSRoman Divacky } 405f22ef01cSRoman Divacky } 406f22ef01cSRoman Divacky return std::make_pair(Result, Result + Num); 407f22ef01cSRoman Divacky } 408f22ef01cSRoman Divacky 4097d523365SDimitry Andric const char *MachineFunction::createExternalSymbolName(StringRef Name) { 4107d523365SDimitry Andric char *Dest = Allocator.Allocate<char>(Name.size() + 1); 4117d523365SDimitry Andric std::copy(Name.begin(), Name.end(), Dest); 4127d523365SDimitry Andric Dest[Name.size()] = 0; 4137d523365SDimitry Andric return Dest; 4147d523365SDimitry Andric } 4157d523365SDimitry Andric 4163861d79fSDimitry Andric #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP) 4173ca95b02SDimitry Andric LLVM_DUMP_METHOD void MachineFunction::dump() const { 418f22ef01cSRoman Divacky print(dbgs()); 419f22ef01cSRoman Divacky } 4203861d79fSDimitry Andric #endif 4213861d79fSDimitry Andric 4223861d79fSDimitry Andric StringRef MachineFunction::getName() const { 4233861d79fSDimitry Andric assert(getFunction() && "No function!"); 4243861d79fSDimitry Andric return getFunction()->getName(); 4253861d79fSDimitry Andric } 426f22ef01cSRoman Divacky 4273ca95b02SDimitry Andric void MachineFunction::print(raw_ostream &OS, const SlotIndexes *Indexes) const { 4283861d79fSDimitry Andric OS << "# Machine code for function " << getName() << ": "; 4293ca95b02SDimitry Andric getProperties().print(OS); 430d88c1a5aSDimitry Andric OS << '\n'; 431f22ef01cSRoman Divacky 432f22ef01cSRoman Divacky // Print Frame Information 433f22ef01cSRoman Divacky FrameInfo->print(*this, OS); 434f22ef01cSRoman Divacky 435f22ef01cSRoman Divacky // Print JumpTable Information 436f22ef01cSRoman Divacky if (JumpTableInfo) 437f22ef01cSRoman Divacky JumpTableInfo->print(OS); 438f22ef01cSRoman Divacky 439f22ef01cSRoman Divacky // Print Constant Pool 440f22ef01cSRoman Divacky ConstantPool->print(OS); 441f22ef01cSRoman Divacky 44239d628a0SDimitry Andric const TargetRegisterInfo *TRI = getSubtarget().getRegisterInfo(); 443f22ef01cSRoman Divacky 444f22ef01cSRoman Divacky if (RegInfo && !RegInfo->livein_empty()) { 445f22ef01cSRoman Divacky OS << "Function Live Ins: "; 446f22ef01cSRoman Divacky for (MachineRegisterInfo::livein_iterator 447f22ef01cSRoman Divacky I = RegInfo->livein_begin(), E = RegInfo->livein_end(); I != E; ++I) { 448bd5abe19SDimitry Andric OS << PrintReg(I->first, TRI); 449f22ef01cSRoman Divacky if (I->second) 450bd5abe19SDimitry Andric OS << " in " << PrintReg(I->second, TRI); 45191bc56edSDimitry Andric if (std::next(I) != E) 452f22ef01cSRoman Divacky OS << ", "; 453f22ef01cSRoman Divacky } 454f22ef01cSRoman Divacky OS << '\n'; 455f22ef01cSRoman Divacky } 456f22ef01cSRoman Divacky 4573dac3a9bSDimitry Andric ModuleSlotTracker MST(getFunction()->getParent()); 4583dac3a9bSDimitry Andric MST.incorporateFunction(*getFunction()); 45991bc56edSDimitry Andric for (const auto &BB : *this) { 460f22ef01cSRoman Divacky OS << '\n'; 4613dac3a9bSDimitry Andric BB.print(OS, MST, Indexes); 462f22ef01cSRoman Divacky } 463f22ef01cSRoman Divacky 4643861d79fSDimitry Andric OS << "\n# End machine code for function " << getName() << ".\n\n"; 465f22ef01cSRoman Divacky } 466f22ef01cSRoman Divacky 467f22ef01cSRoman Divacky namespace llvm { 468f22ef01cSRoman Divacky template<> 469f22ef01cSRoman Divacky struct DOTGraphTraits<const MachineFunction*> : public DefaultDOTGraphTraits { 470f22ef01cSRoman Divacky 471f22ef01cSRoman Divacky DOTGraphTraits (bool isSimple=false) : DefaultDOTGraphTraits(isSimple) {} 472f22ef01cSRoman Divacky 473f22ef01cSRoman Divacky static std::string getGraphName(const MachineFunction *F) { 474ff0cc061SDimitry Andric return ("CFG for '" + F->getName() + "' function").str(); 475f22ef01cSRoman Divacky } 476f22ef01cSRoman Divacky 477f22ef01cSRoman Divacky std::string getNodeLabel(const MachineBasicBlock *Node, 478f22ef01cSRoman Divacky const MachineFunction *Graph) { 479f22ef01cSRoman Divacky std::string OutStr; 480f22ef01cSRoman Divacky { 481f22ef01cSRoman Divacky raw_string_ostream OSS(OutStr); 482f22ef01cSRoman Divacky 4832754fe60SDimitry Andric if (isSimple()) { 4842754fe60SDimitry Andric OSS << "BB#" << Node->getNumber(); 4852754fe60SDimitry Andric if (const BasicBlock *BB = Node->getBasicBlock()) 4862754fe60SDimitry Andric OSS << ": " << BB->getName(); 4872754fe60SDimitry Andric } else 488f22ef01cSRoman Divacky Node->print(OSS); 489f22ef01cSRoman Divacky } 490f22ef01cSRoman Divacky 491f22ef01cSRoman Divacky if (OutStr[0] == '\n') OutStr.erase(OutStr.begin()); 492f22ef01cSRoman Divacky 493f22ef01cSRoman Divacky // Process string output to make it nicer... 494f22ef01cSRoman Divacky for (unsigned i = 0; i != OutStr.length(); ++i) 495f22ef01cSRoman Divacky if (OutStr[i] == '\n') { // Left justify 496f22ef01cSRoman Divacky OutStr[i] = '\\'; 497f22ef01cSRoman Divacky OutStr.insert(OutStr.begin()+i+1, 'l'); 498f22ef01cSRoman Divacky } 499f22ef01cSRoman Divacky return OutStr; 500f22ef01cSRoman Divacky } 501f22ef01cSRoman Divacky }; 5023dac3a9bSDimitry Andric } 503f22ef01cSRoman Divacky 504f22ef01cSRoman Divacky void MachineFunction::viewCFG() const 505f22ef01cSRoman Divacky { 506f22ef01cSRoman Divacky #ifndef NDEBUG 5073861d79fSDimitry Andric ViewGraph(this, "mf" + getName()); 508f22ef01cSRoman Divacky #else 509ffd1746dSEd Schouten errs() << "MachineFunction::viewCFG is only available in debug builds on " 510f22ef01cSRoman Divacky << "systems with Graphviz or gv!\n"; 511f22ef01cSRoman Divacky #endif // NDEBUG 512f22ef01cSRoman Divacky } 513f22ef01cSRoman Divacky 514f22ef01cSRoman Divacky void MachineFunction::viewCFGOnly() const 515f22ef01cSRoman Divacky { 516f22ef01cSRoman Divacky #ifndef NDEBUG 5173861d79fSDimitry Andric ViewGraph(this, "mf" + getName(), true); 518f22ef01cSRoman Divacky #else 519ffd1746dSEd Schouten errs() << "MachineFunction::viewCFGOnly is only available in debug builds on " 520f22ef01cSRoman Divacky << "systems with Graphviz or gv!\n"; 521f22ef01cSRoman Divacky #endif // NDEBUG 522f22ef01cSRoman Divacky } 523f22ef01cSRoman Divacky 5248f0fd8f6SDimitry Andric /// Add the specified physical register as a live-in value and 525f22ef01cSRoman Divacky /// create a corresponding virtual register for it. 526f22ef01cSRoman Divacky unsigned MachineFunction::addLiveIn(unsigned PReg, 527dd6029ffSDimitry Andric const TargetRegisterClass *RC) { 528f22ef01cSRoman Divacky MachineRegisterInfo &MRI = getRegInfo(); 529f22ef01cSRoman Divacky unsigned VReg = MRI.getLiveInVirtReg(PReg); 530f22ef01cSRoman Divacky if (VReg) { 53191bc56edSDimitry Andric const TargetRegisterClass *VRegRC = MRI.getRegClass(VReg); 53291bc56edSDimitry Andric (void)VRegRC; 53391bc56edSDimitry Andric // A physical register can be added several times. 53491bc56edSDimitry Andric // Between two calls, the register class of the related virtual register 53591bc56edSDimitry Andric // may have been constrained to match some operation constraints. 53691bc56edSDimitry Andric // In that case, check that the current register class includes the 53791bc56edSDimitry Andric // physical register and is a sub class of the specified RC. 53891bc56edSDimitry Andric assert((VRegRC == RC || (VRegRC->contains(PReg) && 53991bc56edSDimitry Andric RC->hasSubClassEq(VRegRC))) && 54091bc56edSDimitry Andric "Register class mismatch!"); 541f22ef01cSRoman Divacky return VReg; 542f22ef01cSRoman Divacky } 543f22ef01cSRoman Divacky VReg = MRI.createVirtualRegister(RC); 544f22ef01cSRoman Divacky MRI.addLiveIn(PReg, VReg); 545f22ef01cSRoman Divacky return VReg; 546f22ef01cSRoman Divacky } 547f22ef01cSRoman Divacky 5488f0fd8f6SDimitry Andric /// Return the MCSymbol for the specified non-empty jump table. 549f22ef01cSRoman Divacky /// If isLinkerPrivate is specified, an 'l' label is returned, otherwise a 550f22ef01cSRoman Divacky /// normal 'L' label is returned. 551f22ef01cSRoman Divacky MCSymbol *MachineFunction::getJTISymbol(unsigned JTI, MCContext &Ctx, 552f22ef01cSRoman Divacky bool isLinkerPrivate) const { 553875ed548SDimitry Andric const DataLayout &DL = getDataLayout(); 554f22ef01cSRoman Divacky assert(JumpTableInfo && "No jump tables"); 555f22ef01cSRoman Divacky assert(JTI < JumpTableInfo->getJumpTables().size() && "Invalid JTI!"); 556f22ef01cSRoman Divacky 557d88c1a5aSDimitry Andric StringRef Prefix = isLinkerPrivate ? DL.getLinkerPrivateGlobalPrefix() 558875ed548SDimitry Andric : DL.getPrivateGlobalPrefix(); 559f22ef01cSRoman Divacky SmallString<60> Name; 560f22ef01cSRoman Divacky raw_svector_ostream(Name) 561f22ef01cSRoman Divacky << Prefix << "JTI" << getFunctionNumber() << '_' << JTI; 562ff0cc061SDimitry Andric return Ctx.getOrCreateSymbol(Name); 563f22ef01cSRoman Divacky } 564f22ef01cSRoman Divacky 5658f0fd8f6SDimitry Andric /// Return a function-local symbol to represent the PIC base. 5662754fe60SDimitry Andric MCSymbol *MachineFunction::getPICBaseSymbol() const { 567875ed548SDimitry Andric const DataLayout &DL = getDataLayout(); 568875ed548SDimitry Andric return Ctx.getOrCreateSymbol(Twine(DL.getPrivateGlobalPrefix()) + 5692754fe60SDimitry Andric Twine(getFunctionNumber()) + "$pb"); 5702754fe60SDimitry Andric } 571f22ef01cSRoman Divacky 572d88c1a5aSDimitry Andric /// \name Exception Handling 573d88c1a5aSDimitry Andric /// \{ 574d88c1a5aSDimitry Andric 575d88c1a5aSDimitry Andric LandingPadInfo & 576d88c1a5aSDimitry Andric MachineFunction::getOrCreateLandingPadInfo(MachineBasicBlock *LandingPad) { 577d88c1a5aSDimitry Andric unsigned N = LandingPads.size(); 578d88c1a5aSDimitry Andric for (unsigned i = 0; i < N; ++i) { 579d88c1a5aSDimitry Andric LandingPadInfo &LP = LandingPads[i]; 580d88c1a5aSDimitry Andric if (LP.LandingPadBlock == LandingPad) 581d88c1a5aSDimitry Andric return LP; 582d88c1a5aSDimitry Andric } 583d88c1a5aSDimitry Andric 584d88c1a5aSDimitry Andric LandingPads.push_back(LandingPadInfo(LandingPad)); 585d88c1a5aSDimitry Andric return LandingPads[N]; 586d88c1a5aSDimitry Andric } 587d88c1a5aSDimitry Andric 588d88c1a5aSDimitry Andric void MachineFunction::addInvoke(MachineBasicBlock *LandingPad, 589d88c1a5aSDimitry Andric MCSymbol *BeginLabel, MCSymbol *EndLabel) { 590d88c1a5aSDimitry Andric LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad); 591d88c1a5aSDimitry Andric LP.BeginLabels.push_back(BeginLabel); 592d88c1a5aSDimitry Andric LP.EndLabels.push_back(EndLabel); 593d88c1a5aSDimitry Andric } 594d88c1a5aSDimitry Andric 595d88c1a5aSDimitry Andric MCSymbol *MachineFunction::addLandingPad(MachineBasicBlock *LandingPad) { 596d88c1a5aSDimitry Andric MCSymbol *LandingPadLabel = Ctx.createTempSymbol(); 597d88c1a5aSDimitry Andric LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad); 598d88c1a5aSDimitry Andric LP.LandingPadLabel = LandingPadLabel; 599d88c1a5aSDimitry Andric return LandingPadLabel; 600d88c1a5aSDimitry Andric } 601d88c1a5aSDimitry Andric 602d88c1a5aSDimitry Andric void MachineFunction::addCatchTypeInfo(MachineBasicBlock *LandingPad, 603d88c1a5aSDimitry Andric ArrayRef<const GlobalValue *> TyInfo) { 604d88c1a5aSDimitry Andric LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad); 605d88c1a5aSDimitry Andric for (unsigned N = TyInfo.size(); N; --N) 606d88c1a5aSDimitry Andric LP.TypeIds.push_back(getTypeIDFor(TyInfo[N - 1])); 607d88c1a5aSDimitry Andric } 608d88c1a5aSDimitry Andric 609d88c1a5aSDimitry Andric void MachineFunction::addFilterTypeInfo(MachineBasicBlock *LandingPad, 610d88c1a5aSDimitry Andric ArrayRef<const GlobalValue *> TyInfo) { 611d88c1a5aSDimitry Andric LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad); 612d88c1a5aSDimitry Andric std::vector<unsigned> IdsInFilter(TyInfo.size()); 613d88c1a5aSDimitry Andric for (unsigned I = 0, E = TyInfo.size(); I != E; ++I) 614d88c1a5aSDimitry Andric IdsInFilter[I] = getTypeIDFor(TyInfo[I]); 615d88c1a5aSDimitry Andric LP.TypeIds.push_back(getFilterIDFor(IdsInFilter)); 616d88c1a5aSDimitry Andric } 617d88c1a5aSDimitry Andric 618d88c1a5aSDimitry Andric void MachineFunction::tidyLandingPads(DenseMap<MCSymbol*, uintptr_t> *LPMap) { 619d88c1a5aSDimitry Andric for (unsigned i = 0; i != LandingPads.size(); ) { 620d88c1a5aSDimitry Andric LandingPadInfo &LandingPad = LandingPads[i]; 621d88c1a5aSDimitry Andric if (LandingPad.LandingPadLabel && 622d88c1a5aSDimitry Andric !LandingPad.LandingPadLabel->isDefined() && 623d88c1a5aSDimitry Andric (!LPMap || (*LPMap)[LandingPad.LandingPadLabel] == 0)) 624d88c1a5aSDimitry Andric LandingPad.LandingPadLabel = nullptr; 625d88c1a5aSDimitry Andric 626d88c1a5aSDimitry Andric // Special case: we *should* emit LPs with null LP MBB. This indicates 627d88c1a5aSDimitry Andric // "nounwind" case. 628d88c1a5aSDimitry Andric if (!LandingPad.LandingPadLabel && LandingPad.LandingPadBlock) { 629d88c1a5aSDimitry Andric LandingPads.erase(LandingPads.begin() + i); 630d88c1a5aSDimitry Andric continue; 631d88c1a5aSDimitry Andric } 632d88c1a5aSDimitry Andric 633d88c1a5aSDimitry Andric for (unsigned j = 0, e = LandingPads[i].BeginLabels.size(); j != e; ++j) { 634d88c1a5aSDimitry Andric MCSymbol *BeginLabel = LandingPad.BeginLabels[j]; 635d88c1a5aSDimitry Andric MCSymbol *EndLabel = LandingPad.EndLabels[j]; 636d88c1a5aSDimitry Andric if ((BeginLabel->isDefined() || 637d88c1a5aSDimitry Andric (LPMap && (*LPMap)[BeginLabel] != 0)) && 638d88c1a5aSDimitry Andric (EndLabel->isDefined() || 639d88c1a5aSDimitry Andric (LPMap && (*LPMap)[EndLabel] != 0))) continue; 640d88c1a5aSDimitry Andric 641d88c1a5aSDimitry Andric LandingPad.BeginLabels.erase(LandingPad.BeginLabels.begin() + j); 642d88c1a5aSDimitry Andric LandingPad.EndLabels.erase(LandingPad.EndLabels.begin() + j); 643d88c1a5aSDimitry Andric --j; 644d88c1a5aSDimitry Andric --e; 645d88c1a5aSDimitry Andric } 646d88c1a5aSDimitry Andric 647d88c1a5aSDimitry Andric // Remove landing pads with no try-ranges. 648d88c1a5aSDimitry Andric if (LandingPads[i].BeginLabels.empty()) { 649d88c1a5aSDimitry Andric LandingPads.erase(LandingPads.begin() + i); 650d88c1a5aSDimitry Andric continue; 651d88c1a5aSDimitry Andric } 652d88c1a5aSDimitry Andric 653d88c1a5aSDimitry Andric // If there is no landing pad, ensure that the list of typeids is empty. 654d88c1a5aSDimitry Andric // If the only typeid is a cleanup, this is the same as having no typeids. 655d88c1a5aSDimitry Andric if (!LandingPad.LandingPadBlock || 656d88c1a5aSDimitry Andric (LandingPad.TypeIds.size() == 1 && !LandingPad.TypeIds[0])) 657d88c1a5aSDimitry Andric LandingPad.TypeIds.clear(); 658d88c1a5aSDimitry Andric ++i; 659d88c1a5aSDimitry Andric } 660d88c1a5aSDimitry Andric } 661d88c1a5aSDimitry Andric 662d88c1a5aSDimitry Andric void MachineFunction::addCleanup(MachineBasicBlock *LandingPad) { 663d88c1a5aSDimitry Andric LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad); 664d88c1a5aSDimitry Andric LP.TypeIds.push_back(0); 665d88c1a5aSDimitry Andric } 666d88c1a5aSDimitry Andric 667d88c1a5aSDimitry Andric void MachineFunction::addSEHCatchHandler(MachineBasicBlock *LandingPad, 668d88c1a5aSDimitry Andric const Function *Filter, 669d88c1a5aSDimitry Andric const BlockAddress *RecoverBA) { 670d88c1a5aSDimitry Andric LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad); 671d88c1a5aSDimitry Andric SEHHandler Handler; 672d88c1a5aSDimitry Andric Handler.FilterOrFinally = Filter; 673d88c1a5aSDimitry Andric Handler.RecoverBA = RecoverBA; 674d88c1a5aSDimitry Andric LP.SEHHandlers.push_back(Handler); 675d88c1a5aSDimitry Andric } 676d88c1a5aSDimitry Andric 677d88c1a5aSDimitry Andric void MachineFunction::addSEHCleanupHandler(MachineBasicBlock *LandingPad, 678d88c1a5aSDimitry Andric const Function *Cleanup) { 679d88c1a5aSDimitry Andric LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad); 680d88c1a5aSDimitry Andric SEHHandler Handler; 681d88c1a5aSDimitry Andric Handler.FilterOrFinally = Cleanup; 682d88c1a5aSDimitry Andric Handler.RecoverBA = nullptr; 683d88c1a5aSDimitry Andric LP.SEHHandlers.push_back(Handler); 684d88c1a5aSDimitry Andric } 685d88c1a5aSDimitry Andric 686d88c1a5aSDimitry Andric void MachineFunction::setCallSiteLandingPad(MCSymbol *Sym, 687d88c1a5aSDimitry Andric ArrayRef<unsigned> Sites) { 688d88c1a5aSDimitry Andric LPadToCallSiteMap[Sym].append(Sites.begin(), Sites.end()); 689d88c1a5aSDimitry Andric } 690d88c1a5aSDimitry Andric 691d88c1a5aSDimitry Andric unsigned MachineFunction::getTypeIDFor(const GlobalValue *TI) { 692d88c1a5aSDimitry Andric for (unsigned i = 0, N = TypeInfos.size(); i != N; ++i) 693d88c1a5aSDimitry Andric if (TypeInfos[i] == TI) return i + 1; 694d88c1a5aSDimitry Andric 695d88c1a5aSDimitry Andric TypeInfos.push_back(TI); 696d88c1a5aSDimitry Andric return TypeInfos.size(); 697d88c1a5aSDimitry Andric } 698d88c1a5aSDimitry Andric 699d88c1a5aSDimitry Andric int MachineFunction::getFilterIDFor(std::vector<unsigned> &TyIds) { 700d88c1a5aSDimitry Andric // If the new filter coincides with the tail of an existing filter, then 701d88c1a5aSDimitry Andric // re-use the existing filter. Folding filters more than this requires 702d88c1a5aSDimitry Andric // re-ordering filters and/or their elements - probably not worth it. 703d88c1a5aSDimitry Andric for (std::vector<unsigned>::iterator I = FilterEnds.begin(), 704d88c1a5aSDimitry Andric E = FilterEnds.end(); I != E; ++I) { 705d88c1a5aSDimitry Andric unsigned i = *I, j = TyIds.size(); 706d88c1a5aSDimitry Andric 707d88c1a5aSDimitry Andric while (i && j) 708d88c1a5aSDimitry Andric if (FilterIds[--i] != TyIds[--j]) 709d88c1a5aSDimitry Andric goto try_next; 710d88c1a5aSDimitry Andric 711d88c1a5aSDimitry Andric if (!j) 712d88c1a5aSDimitry Andric // The new filter coincides with range [i, end) of the existing filter. 713d88c1a5aSDimitry Andric return -(1 + i); 714d88c1a5aSDimitry Andric 715d88c1a5aSDimitry Andric try_next:; 716d88c1a5aSDimitry Andric } 717d88c1a5aSDimitry Andric 718d88c1a5aSDimitry Andric // Add the new filter. 719d88c1a5aSDimitry Andric int FilterID = -(1 + FilterIds.size()); 720d88c1a5aSDimitry Andric FilterIds.reserve(FilterIds.size() + TyIds.size() + 1); 721d88c1a5aSDimitry Andric FilterIds.insert(FilterIds.end(), TyIds.begin(), TyIds.end()); 722d88c1a5aSDimitry Andric FilterEnds.push_back(FilterIds.size()); 723d88c1a5aSDimitry Andric FilterIds.push_back(0); // terminator 724d88c1a5aSDimitry Andric return FilterID; 725d88c1a5aSDimitry Andric } 726d88c1a5aSDimitry Andric 727d88c1a5aSDimitry Andric void llvm::addLandingPadInfo(const LandingPadInst &I, MachineBasicBlock &MBB) { 728d88c1a5aSDimitry Andric MachineFunction &MF = *MBB.getParent(); 729d88c1a5aSDimitry Andric if (const auto *PF = dyn_cast<Function>( 730d88c1a5aSDimitry Andric I.getParent()->getParent()->getPersonalityFn()->stripPointerCasts())) 731d88c1a5aSDimitry Andric MF.getMMI().addPersonality(PF); 732d88c1a5aSDimitry Andric 733d88c1a5aSDimitry Andric if (I.isCleanup()) 734d88c1a5aSDimitry Andric MF.addCleanup(&MBB); 735d88c1a5aSDimitry Andric 736d88c1a5aSDimitry Andric // FIXME: New EH - Add the clauses in reverse order. This isn't 100% correct, 737d88c1a5aSDimitry Andric // but we need to do it this way because of how the DWARF EH emitter 738d88c1a5aSDimitry Andric // processes the clauses. 739d88c1a5aSDimitry Andric for (unsigned i = I.getNumClauses(); i != 0; --i) { 740d88c1a5aSDimitry Andric Value *Val = I.getClause(i - 1); 741d88c1a5aSDimitry Andric if (I.isCatch(i - 1)) { 742d88c1a5aSDimitry Andric MF.addCatchTypeInfo(&MBB, 743d88c1a5aSDimitry Andric dyn_cast<GlobalValue>(Val->stripPointerCasts())); 744d88c1a5aSDimitry Andric } else { 745d88c1a5aSDimitry Andric // Add filters in a list. 746d88c1a5aSDimitry Andric Constant *CVal = cast<Constant>(Val); 747d88c1a5aSDimitry Andric SmallVector<const GlobalValue *, 4> FilterList; 748d88c1a5aSDimitry Andric for (User::op_iterator II = CVal->op_begin(), IE = CVal->op_end(); 749d88c1a5aSDimitry Andric II != IE; ++II) 750d88c1a5aSDimitry Andric FilterList.push_back(cast<GlobalValue>((*II)->stripPointerCasts())); 751d88c1a5aSDimitry Andric 752d88c1a5aSDimitry Andric MF.addFilterTypeInfo(&MBB, FilterList); 753d88c1a5aSDimitry Andric } 754d88c1a5aSDimitry Andric } 755d88c1a5aSDimitry Andric } 756d88c1a5aSDimitry Andric 757d88c1a5aSDimitry Andric /// \} 758d88c1a5aSDimitry Andric 759f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 760f22ef01cSRoman Divacky // MachineFrameInfo implementation 761f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 762f22ef01cSRoman Divacky 7638f0fd8f6SDimitry Andric /// Make sure the function is at least Align bytes aligned. 764139f7f9bSDimitry Andric void MachineFrameInfo::ensureMaxAlignment(unsigned Align) { 7653ca95b02SDimitry Andric if (!StackRealignable) 76639d628a0SDimitry Andric assert(Align <= StackAlignment && 767139f7f9bSDimitry Andric "For targets without stack realignment, Align is out of limit!"); 768139f7f9bSDimitry Andric if (MaxAlignment < Align) MaxAlignment = Align; 769139f7f9bSDimitry Andric } 770139f7f9bSDimitry Andric 7718f0fd8f6SDimitry Andric /// Clamp the alignment if requested and emit a warning. 772139f7f9bSDimitry Andric static inline unsigned clampStackAlignment(bool ShouldClamp, unsigned Align, 773139f7f9bSDimitry Andric unsigned StackAlign) { 774139f7f9bSDimitry Andric if (!ShouldClamp || Align <= StackAlign) 775139f7f9bSDimitry Andric return Align; 776139f7f9bSDimitry Andric DEBUG(dbgs() << "Warning: requested alignment " << Align 777139f7f9bSDimitry Andric << " exceeds the stack alignment " << StackAlign 778139f7f9bSDimitry Andric << " when stack realignment is off" << '\n'); 779139f7f9bSDimitry Andric return StackAlign; 780139f7f9bSDimitry Andric } 781139f7f9bSDimitry Andric 7828f0fd8f6SDimitry Andric /// Create a new statically sized stack object, returning a nonnegative 7838f0fd8f6SDimitry Andric /// identifier to represent it. 784139f7f9bSDimitry Andric int MachineFrameInfo::CreateStackObject(uint64_t Size, unsigned Alignment, 78591bc56edSDimitry Andric bool isSS, const AllocaInst *Alloca) { 786139f7f9bSDimitry Andric assert(Size != 0 && "Cannot allocate zero size stack objects!"); 7873ca95b02SDimitry Andric Alignment = clampStackAlignment(!StackRealignable, Alignment, StackAlignment); 78839d628a0SDimitry Andric Objects.push_back(StackObject(Size, Alignment, 0, false, isSS, Alloca, 78939d628a0SDimitry Andric !isSS)); 790139f7f9bSDimitry Andric int Index = (int)Objects.size() - NumFixedObjects - 1; 791139f7f9bSDimitry Andric assert(Index >= 0 && "Bad frame index!"); 792139f7f9bSDimitry Andric ensureMaxAlignment(Alignment); 793139f7f9bSDimitry Andric return Index; 794139f7f9bSDimitry Andric } 795139f7f9bSDimitry Andric 7968f0fd8f6SDimitry Andric /// Create a new statically sized stack object that represents a spill slot, 7978f0fd8f6SDimitry Andric /// returning a nonnegative identifier to represent it. 798139f7f9bSDimitry Andric int MachineFrameInfo::CreateSpillStackObject(uint64_t Size, 799139f7f9bSDimitry Andric unsigned Alignment) { 8003ca95b02SDimitry Andric Alignment = clampStackAlignment(!StackRealignable, Alignment, StackAlignment); 80191bc56edSDimitry Andric CreateStackObject(Size, Alignment, true); 802139f7f9bSDimitry Andric int Index = (int)Objects.size() - NumFixedObjects - 1; 803139f7f9bSDimitry Andric ensureMaxAlignment(Alignment); 804139f7f9bSDimitry Andric return Index; 805139f7f9bSDimitry Andric } 806139f7f9bSDimitry Andric 8078f0fd8f6SDimitry Andric /// Notify the MachineFrameInfo object that a variable sized object has been 8088f0fd8f6SDimitry Andric /// created. This must be created whenever a variable sized object is created, 8098f0fd8f6SDimitry Andric /// whether or not the index returned is actually used. 810*4f00c8c6SDimitry Andric int MachineFrameInfo::CreateVariableSizedObject(unsigned Alignment, 811*4f00c8c6SDimitry Andric const AllocaInst *Alloca) { 812139f7f9bSDimitry Andric HasVarSizedObjects = true; 8133ca95b02SDimitry Andric Alignment = clampStackAlignment(!StackRealignable, Alignment, StackAlignment); 81439d628a0SDimitry Andric Objects.push_back(StackObject(0, Alignment, 0, false, false, Alloca, true)); 815139f7f9bSDimitry Andric ensureMaxAlignment(Alignment); 816139f7f9bSDimitry Andric return (int)Objects.size()-NumFixedObjects-1; 817139f7f9bSDimitry Andric } 818139f7f9bSDimitry Andric 8198f0fd8f6SDimitry Andric /// Create a new object at a fixed location on the stack. 820f22ef01cSRoman Divacky /// All fixed objects should be created before other objects are created for 821f22ef01cSRoman Divacky /// efficiency. By default, fixed objects are immutable. This returns an 822f22ef01cSRoman Divacky /// index with a negative value. 823f22ef01cSRoman Divacky int MachineFrameInfo::CreateFixedObject(uint64_t Size, int64_t SPOffset, 82439d628a0SDimitry Andric bool Immutable, bool isAliased) { 825f22ef01cSRoman Divacky assert(Size != 0 && "Cannot allocate zero size fixed stack objects!"); 826ffd1746dSEd Schouten // The alignment of the frame index can be determined from its offset from 827ffd1746dSEd Schouten // the incoming frame position. If the frame object is at offset 32 and 828ffd1746dSEd Schouten // the stack is guaranteed to be 16-byte aligned, then we know that the 8293ca95b02SDimitry Andric // object is 16-byte aligned. Note that unlike the non-fixed case, if the 8303ca95b02SDimitry Andric // stack needs realignment, we can't assume that the stack will in fact be 8313ca95b02SDimitry Andric // aligned. 8323ca95b02SDimitry Andric unsigned Align = MinAlign(SPOffset, ForcedRealign ? 1 : StackAlignment); 8333ca95b02SDimitry Andric Align = clampStackAlignment(!StackRealignable, Align, StackAlignment); 834ffd1746dSEd Schouten Objects.insert(Objects.begin(), StackObject(Size, Align, SPOffset, Immutable, 8353861d79fSDimitry Andric /*isSS*/ false, 83639d628a0SDimitry Andric /*Alloca*/ nullptr, isAliased)); 837f22ef01cSRoman Divacky return -++NumFixedObjects; 838f22ef01cSRoman Divacky } 839f22ef01cSRoman Divacky 8408f0fd8f6SDimitry Andric /// Create a spill slot at a fixed location on the stack. 8418f0fd8f6SDimitry Andric /// Returns an index with a negative value. 84291bc56edSDimitry Andric int MachineFrameInfo::CreateFixedSpillStackObject(uint64_t Size, 843d88c1a5aSDimitry Andric int64_t SPOffset, 844d88c1a5aSDimitry Andric bool Immutable) { 8453ca95b02SDimitry Andric unsigned Align = MinAlign(SPOffset, ForcedRealign ? 1 : StackAlignment); 8463ca95b02SDimitry Andric Align = clampStackAlignment(!StackRealignable, Align, StackAlignment); 847d88c1a5aSDimitry Andric Objects.insert(Objects.begin(), StackObject(Size, Align, SPOffset, Immutable, 84891bc56edSDimitry Andric /*isSS*/ true, 84939d628a0SDimitry Andric /*Alloca*/ nullptr, 85039d628a0SDimitry Andric /*isAliased*/ false)); 85191bc56edSDimitry Andric return -++NumFixedObjects; 85291bc56edSDimitry Andric } 853f22ef01cSRoman Divacky 85497bc6c73SDimitry Andric BitVector MachineFrameInfo::getPristineRegs(const MachineFunction &MF) const { 85597bc6c73SDimitry Andric const TargetRegisterInfo *TRI = MF.getSubtarget().getRegisterInfo(); 856f22ef01cSRoman Divacky BitVector BV(TRI->getNumRegs()); 857f22ef01cSRoman Divacky 858f22ef01cSRoman Divacky // Before CSI is calculated, no registers are considered pristine. They can be 859f22ef01cSRoman Divacky // freely used and PEI will make sure they are saved. 860f22ef01cSRoman Divacky if (!isCalleeSavedInfoValid()) 861f22ef01cSRoman Divacky return BV; 862f22ef01cSRoman Divacky 8637a7e6055SDimitry Andric const MachineRegisterInfo &MRI = MF.getRegInfo(); 8647a7e6055SDimitry Andric for (const MCPhysReg *CSR = MRI.getCalleeSavedRegs(); CSR && *CSR; 8657a7e6055SDimitry Andric ++CSR) 866f22ef01cSRoman Divacky BV.set(*CSR); 867f22ef01cSRoman Divacky 86897bc6c73SDimitry Andric // Saved CSRs are not pristine. 8697d523365SDimitry Andric for (auto &I : getCalleeSavedInfo()) 8707d523365SDimitry Andric for (MCSubRegIterator S(I.getReg(), TRI, true); S.isValid(); ++S) 8717d523365SDimitry Andric BV.reset(*S); 872f22ef01cSRoman Divacky 873f22ef01cSRoman Divacky return BV; 874f22ef01cSRoman Divacky } 875f22ef01cSRoman Divacky 876139f7f9bSDimitry Andric unsigned MachineFrameInfo::estimateStackSize(const MachineFunction &MF) const { 87739d628a0SDimitry Andric const TargetFrameLowering *TFI = MF.getSubtarget().getFrameLowering(); 87839d628a0SDimitry Andric const TargetRegisterInfo *RegInfo = MF.getSubtarget().getRegisterInfo(); 879139f7f9bSDimitry Andric unsigned MaxAlign = getMaxAlignment(); 880139f7f9bSDimitry Andric int Offset = 0; 881139f7f9bSDimitry Andric 882139f7f9bSDimitry Andric // This code is very, very similar to PEI::calculateFrameObjectOffsets(). 883139f7f9bSDimitry Andric // It really should be refactored to share code. Until then, changes 884139f7f9bSDimitry Andric // should keep in mind that there's tight coupling between the two. 885139f7f9bSDimitry Andric 886139f7f9bSDimitry Andric for (int i = getObjectIndexBegin(); i != 0; ++i) { 887139f7f9bSDimitry Andric int FixedOff = -getObjectOffset(i); 888139f7f9bSDimitry Andric if (FixedOff > Offset) Offset = FixedOff; 889139f7f9bSDimitry Andric } 890139f7f9bSDimitry Andric for (unsigned i = 0, e = getObjectIndexEnd(); i != e; ++i) { 891139f7f9bSDimitry Andric if (isDeadObjectIndex(i)) 892139f7f9bSDimitry Andric continue; 893139f7f9bSDimitry Andric Offset += getObjectSize(i); 894139f7f9bSDimitry Andric unsigned Align = getObjectAlignment(i); 895139f7f9bSDimitry Andric // Adjust to alignment boundary 896139f7f9bSDimitry Andric Offset = (Offset+Align-1)/Align*Align; 897139f7f9bSDimitry Andric 898139f7f9bSDimitry Andric MaxAlign = std::max(Align, MaxAlign); 899139f7f9bSDimitry Andric } 900139f7f9bSDimitry Andric 901139f7f9bSDimitry Andric if (adjustsStack() && TFI->hasReservedCallFrame(MF)) 902139f7f9bSDimitry Andric Offset += getMaxCallFrameSize(); 903139f7f9bSDimitry Andric 904139f7f9bSDimitry Andric // Round up the size to a multiple of the alignment. If the function has 905139f7f9bSDimitry Andric // any calls or alloca's, align to the target's StackAlignment value to 906139f7f9bSDimitry Andric // ensure that the callee's frame or the alloca data is suitably aligned; 907139f7f9bSDimitry Andric // otherwise, for leaf functions, align to the TransientStackAlignment 908139f7f9bSDimitry Andric // value. 909139f7f9bSDimitry Andric unsigned StackAlign; 910139f7f9bSDimitry Andric if (adjustsStack() || hasVarSizedObjects() || 911139f7f9bSDimitry Andric (RegInfo->needsStackRealignment(MF) && getObjectIndexEnd() != 0)) 912139f7f9bSDimitry Andric StackAlign = TFI->getStackAlignment(); 913139f7f9bSDimitry Andric else 914139f7f9bSDimitry Andric StackAlign = TFI->getTransientStackAlignment(); 915139f7f9bSDimitry Andric 916139f7f9bSDimitry Andric // If the frame pointer is eliminated, all frame offsets will be relative to 917139f7f9bSDimitry Andric // SP not FP. Align to MaxAlign so this works. 918139f7f9bSDimitry Andric StackAlign = std::max(StackAlign, MaxAlign); 919139f7f9bSDimitry Andric unsigned AlignMask = StackAlign - 1; 920139f7f9bSDimitry Andric Offset = (Offset + AlignMask) & ~uint64_t(AlignMask); 921139f7f9bSDimitry Andric 922139f7f9bSDimitry Andric return (unsigned)Offset; 923139f7f9bSDimitry Andric } 924f22ef01cSRoman Divacky 925f22ef01cSRoman Divacky void MachineFrameInfo::print(const MachineFunction &MF, raw_ostream &OS) const{ 926f22ef01cSRoman Divacky if (Objects.empty()) return; 927f22ef01cSRoman Divacky 92839d628a0SDimitry Andric const TargetFrameLowering *FI = MF.getSubtarget().getFrameLowering(); 929f22ef01cSRoman Divacky int ValOffset = (FI ? FI->getOffsetOfLocalArea() : 0); 930f22ef01cSRoman Divacky 931f22ef01cSRoman Divacky OS << "Frame Objects:\n"; 932f22ef01cSRoman Divacky 933f22ef01cSRoman Divacky for (unsigned i = 0, e = Objects.size(); i != e; ++i) { 934f22ef01cSRoman Divacky const StackObject &SO = Objects[i]; 935f22ef01cSRoman Divacky OS << " fi#" << (int)(i-NumFixedObjects) << ": "; 936f22ef01cSRoman Divacky if (SO.Size == ~0ULL) { 937f22ef01cSRoman Divacky OS << "dead\n"; 938f22ef01cSRoman Divacky continue; 939f22ef01cSRoman Divacky } 940f22ef01cSRoman Divacky if (SO.Size == 0) 941f22ef01cSRoman Divacky OS << "variable sized"; 942f22ef01cSRoman Divacky else 943f22ef01cSRoman Divacky OS << "size=" << SO.Size; 944f22ef01cSRoman Divacky OS << ", align=" << SO.Alignment; 945f22ef01cSRoman Divacky 946f22ef01cSRoman Divacky if (i < NumFixedObjects) 947f22ef01cSRoman Divacky OS << ", fixed"; 948f22ef01cSRoman Divacky if (i < NumFixedObjects || SO.SPOffset != -1) { 949f22ef01cSRoman Divacky int64_t Off = SO.SPOffset - ValOffset; 950f22ef01cSRoman Divacky OS << ", at location [SP"; 951f22ef01cSRoman Divacky if (Off > 0) 952f22ef01cSRoman Divacky OS << "+" << Off; 953f22ef01cSRoman Divacky else if (Off < 0) 954f22ef01cSRoman Divacky OS << Off; 955f22ef01cSRoman Divacky OS << "]"; 956f22ef01cSRoman Divacky } 957f22ef01cSRoman Divacky OS << "\n"; 958f22ef01cSRoman Divacky } 959f22ef01cSRoman Divacky } 960f22ef01cSRoman Divacky 9613861d79fSDimitry Andric #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP) 9627a7e6055SDimitry Andric LLVM_DUMP_METHOD void MachineFrameInfo::dump(const MachineFunction &MF) const { 963f22ef01cSRoman Divacky print(MF, dbgs()); 964f22ef01cSRoman Divacky } 9653861d79fSDimitry Andric #endif 966f22ef01cSRoman Divacky 967f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 968f22ef01cSRoman Divacky // MachineJumpTableInfo implementation 969f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 970f22ef01cSRoman Divacky 9718f0fd8f6SDimitry Andric /// Return the size of each entry in the jump table. 9723861d79fSDimitry Andric unsigned MachineJumpTableInfo::getEntrySize(const DataLayout &TD) const { 973f22ef01cSRoman Divacky // The size of a jump table entry is 4 bytes unless the entry is just the 974f22ef01cSRoman Divacky // address of a block, in which case it is the pointer size. 975f22ef01cSRoman Divacky switch (getEntryKind()) { 976f22ef01cSRoman Divacky case MachineJumpTableInfo::EK_BlockAddress: 977f22ef01cSRoman Divacky return TD.getPointerSize(); 978dff0c46cSDimitry Andric case MachineJumpTableInfo::EK_GPRel64BlockAddress: 979dff0c46cSDimitry Andric return 8; 980f22ef01cSRoman Divacky case MachineJumpTableInfo::EK_GPRel32BlockAddress: 981f22ef01cSRoman Divacky case MachineJumpTableInfo::EK_LabelDifference32: 982f22ef01cSRoman Divacky case MachineJumpTableInfo::EK_Custom32: 983f22ef01cSRoman Divacky return 4; 984f22ef01cSRoman Divacky case MachineJumpTableInfo::EK_Inline: 985f22ef01cSRoman Divacky return 0; 986f22ef01cSRoman Divacky } 987dff0c46cSDimitry Andric llvm_unreachable("Unknown jump table encoding!"); 988f22ef01cSRoman Divacky } 989f22ef01cSRoman Divacky 9908f0fd8f6SDimitry Andric /// Return the alignment of each entry in the jump table. 9913861d79fSDimitry Andric unsigned MachineJumpTableInfo::getEntryAlignment(const DataLayout &TD) const { 992f22ef01cSRoman Divacky // The alignment of a jump table entry is the alignment of int32 unless the 993f22ef01cSRoman Divacky // entry is just the address of a block, in which case it is the pointer 994f22ef01cSRoman Divacky // alignment. 995f22ef01cSRoman Divacky switch (getEntryKind()) { 996f22ef01cSRoman Divacky case MachineJumpTableInfo::EK_BlockAddress: 997f22ef01cSRoman Divacky return TD.getPointerABIAlignment(); 998dff0c46cSDimitry Andric case MachineJumpTableInfo::EK_GPRel64BlockAddress: 999dff0c46cSDimitry Andric return TD.getABIIntegerTypeAlignment(64); 1000f22ef01cSRoman Divacky case MachineJumpTableInfo::EK_GPRel32BlockAddress: 1001f22ef01cSRoman Divacky case MachineJumpTableInfo::EK_LabelDifference32: 1002f22ef01cSRoman Divacky case MachineJumpTableInfo::EK_Custom32: 1003f22ef01cSRoman Divacky return TD.getABIIntegerTypeAlignment(32); 1004f22ef01cSRoman Divacky case MachineJumpTableInfo::EK_Inline: 1005f22ef01cSRoman Divacky return 1; 1006f22ef01cSRoman Divacky } 1007dff0c46cSDimitry Andric llvm_unreachable("Unknown jump table encoding!"); 1008f22ef01cSRoman Divacky } 1009f22ef01cSRoman Divacky 10108f0fd8f6SDimitry Andric /// Create a new jump table entry in the jump table info. 1011f22ef01cSRoman Divacky unsigned MachineJumpTableInfo::createJumpTableIndex( 1012f22ef01cSRoman Divacky const std::vector<MachineBasicBlock*> &DestBBs) { 1013f22ef01cSRoman Divacky assert(!DestBBs.empty() && "Cannot create an empty jump table!"); 1014f22ef01cSRoman Divacky JumpTables.push_back(MachineJumpTableEntry(DestBBs)); 1015f22ef01cSRoman Divacky return JumpTables.size()-1; 1016f22ef01cSRoman Divacky } 1017f22ef01cSRoman Divacky 10188f0fd8f6SDimitry Andric /// If Old is the target of any jump tables, update the jump tables to branch 10198f0fd8f6SDimitry Andric /// to New instead. 1020f22ef01cSRoman Divacky bool MachineJumpTableInfo::ReplaceMBBInJumpTables(MachineBasicBlock *Old, 1021f22ef01cSRoman Divacky MachineBasicBlock *New) { 1022f22ef01cSRoman Divacky assert(Old != New && "Not making a change?"); 1023f22ef01cSRoman Divacky bool MadeChange = false; 1024f22ef01cSRoman Divacky for (size_t i = 0, e = JumpTables.size(); i != e; ++i) 1025f22ef01cSRoman Divacky ReplaceMBBInJumpTable(i, Old, New); 1026f22ef01cSRoman Divacky return MadeChange; 1027f22ef01cSRoman Divacky } 1028f22ef01cSRoman Divacky 10298f0fd8f6SDimitry Andric /// If Old is a target of the jump tables, update the jump table to branch to 10308f0fd8f6SDimitry Andric /// New instead. 1031f22ef01cSRoman Divacky bool MachineJumpTableInfo::ReplaceMBBInJumpTable(unsigned Idx, 1032f22ef01cSRoman Divacky MachineBasicBlock *Old, 1033f22ef01cSRoman Divacky MachineBasicBlock *New) { 1034f22ef01cSRoman Divacky assert(Old != New && "Not making a change?"); 1035f22ef01cSRoman Divacky bool MadeChange = false; 1036f22ef01cSRoman Divacky MachineJumpTableEntry &JTE = JumpTables[Idx]; 1037f22ef01cSRoman Divacky for (size_t j = 0, e = JTE.MBBs.size(); j != e; ++j) 1038f22ef01cSRoman Divacky if (JTE.MBBs[j] == Old) { 1039f22ef01cSRoman Divacky JTE.MBBs[j] = New; 1040f22ef01cSRoman Divacky MadeChange = true; 1041f22ef01cSRoman Divacky } 1042f22ef01cSRoman Divacky return MadeChange; 1043f22ef01cSRoman Divacky } 1044f22ef01cSRoman Divacky 1045f22ef01cSRoman Divacky void MachineJumpTableInfo::print(raw_ostream &OS) const { 1046f22ef01cSRoman Divacky if (JumpTables.empty()) return; 1047f22ef01cSRoman Divacky 1048f22ef01cSRoman Divacky OS << "Jump Tables:\n"; 1049f22ef01cSRoman Divacky 1050f22ef01cSRoman Divacky for (unsigned i = 0, e = JumpTables.size(); i != e; ++i) { 1051f22ef01cSRoman Divacky OS << " jt#" << i << ": "; 1052f22ef01cSRoman Divacky for (unsigned j = 0, f = JumpTables[i].MBBs.size(); j != f; ++j) 1053f22ef01cSRoman Divacky OS << " BB#" << JumpTables[i].MBBs[j]->getNumber(); 1054f22ef01cSRoman Divacky } 1055f22ef01cSRoman Divacky 1056f22ef01cSRoman Divacky OS << '\n'; 1057f22ef01cSRoman Divacky } 1058f22ef01cSRoman Divacky 10593861d79fSDimitry Andric #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP) 10603ca95b02SDimitry Andric LLVM_DUMP_METHOD void MachineJumpTableInfo::dump() const { print(dbgs()); } 10613861d79fSDimitry Andric #endif 1062f22ef01cSRoman Divacky 1063f22ef01cSRoman Divacky 1064f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 1065f22ef01cSRoman Divacky // MachineConstantPool implementation 1066f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 1067f22ef01cSRoman Divacky 1068dff0c46cSDimitry Andric void MachineConstantPoolValue::anchor() { } 1069dff0c46cSDimitry Andric 10706122f3e6SDimitry Andric Type *MachineConstantPoolEntry::getType() const { 1071f22ef01cSRoman Divacky if (isMachineConstantPoolEntry()) 1072f22ef01cSRoman Divacky return Val.MachineCPVal->getType(); 1073f22ef01cSRoman Divacky return Val.ConstVal->getType(); 1074f22ef01cSRoman Divacky } 1075f22ef01cSRoman Divacky 10767d523365SDimitry Andric bool MachineConstantPoolEntry::needsRelocation() const { 1077f22ef01cSRoman Divacky if (isMachineConstantPoolEntry()) 10787d523365SDimitry Andric return true; 10797d523365SDimitry Andric return Val.ConstVal->needsRelocation(); 1080f22ef01cSRoman Divacky } 1081f22ef01cSRoman Divacky 108291bc56edSDimitry Andric SectionKind 108391bc56edSDimitry Andric MachineConstantPoolEntry::getSectionKind(const DataLayout *DL) const { 10847d523365SDimitry Andric if (needsRelocation()) 10857d523365SDimitry Andric return SectionKind::getReadOnlyWithRel(); 108691bc56edSDimitry Andric switch (DL->getTypeAllocSize(getType())) { 108791bc56edSDimitry Andric case 4: 10887d523365SDimitry Andric return SectionKind::getMergeableConst4(); 108991bc56edSDimitry Andric case 8: 10907d523365SDimitry Andric return SectionKind::getMergeableConst8(); 109191bc56edSDimitry Andric case 16: 10927d523365SDimitry Andric return SectionKind::getMergeableConst16(); 10933ca95b02SDimitry Andric case 32: 10943ca95b02SDimitry Andric return SectionKind::getMergeableConst32(); 109591bc56edSDimitry Andric default: 10967d523365SDimitry Andric return SectionKind::getReadOnly(); 109791bc56edSDimitry Andric } 109891bc56edSDimitry Andric } 109991bc56edSDimitry Andric 1100f22ef01cSRoman Divacky MachineConstantPool::~MachineConstantPool() { 1101d88c1a5aSDimitry Andric // A constant may be a member of both Constants and MachineCPVsSharingEntries, 1102d88c1a5aSDimitry Andric // so keep track of which we've deleted to avoid double deletions. 1103d88c1a5aSDimitry Andric DenseSet<MachineConstantPoolValue*> Deleted; 1104f22ef01cSRoman Divacky for (unsigned i = 0, e = Constants.size(); i != e; ++i) 1105d88c1a5aSDimitry Andric if (Constants[i].isMachineConstantPoolEntry()) { 1106d88c1a5aSDimitry Andric Deleted.insert(Constants[i].Val.MachineCPVal); 1107f22ef01cSRoman Divacky delete Constants[i].Val.MachineCPVal; 1108d88c1a5aSDimitry Andric } 1109dd6029ffSDimitry Andric for (DenseSet<MachineConstantPoolValue*>::iterator I = 1110dd6029ffSDimitry Andric MachineCPVsSharingEntries.begin(), E = MachineCPVsSharingEntries.end(); 1111d88c1a5aSDimitry Andric I != E; ++I) { 1112d88c1a5aSDimitry Andric if (Deleted.count(*I) == 0) 1113dd6029ffSDimitry Andric delete *I; 1114f22ef01cSRoman Divacky } 1115d88c1a5aSDimitry Andric } 1116f22ef01cSRoman Divacky 11178f0fd8f6SDimitry Andric /// Test whether the given two constants can be allocated the same constant pool 11188f0fd8f6SDimitry Andric /// entry. 1119f22ef01cSRoman Divacky static bool CanShareConstantPoolEntry(const Constant *A, const Constant *B, 1120875ed548SDimitry Andric const DataLayout &DL) { 1121f22ef01cSRoman Divacky // Handle the trivial case quickly. 1122f22ef01cSRoman Divacky if (A == B) return true; 1123f22ef01cSRoman Divacky 1124f22ef01cSRoman Divacky // If they have the same type but weren't the same constant, quickly 1125f22ef01cSRoman Divacky // reject them. 1126f22ef01cSRoman Divacky if (A->getType() == B->getType()) return false; 1127f22ef01cSRoman Divacky 1128dff0c46cSDimitry Andric // We can't handle structs or arrays. 1129dff0c46cSDimitry Andric if (isa<StructType>(A->getType()) || isa<ArrayType>(A->getType()) || 1130dff0c46cSDimitry Andric isa<StructType>(B->getType()) || isa<ArrayType>(B->getType())) 1131dff0c46cSDimitry Andric return false; 1132dff0c46cSDimitry Andric 1133f22ef01cSRoman Divacky // For now, only support constants with the same size. 1134875ed548SDimitry Andric uint64_t StoreSize = DL.getTypeStoreSize(A->getType()); 1135875ed548SDimitry Andric if (StoreSize != DL.getTypeStoreSize(B->getType()) || StoreSize > 128) 1136f22ef01cSRoman Divacky return false; 1137f22ef01cSRoman Divacky 1138dff0c46cSDimitry Andric Type *IntTy = IntegerType::get(A->getContext(), StoreSize*8); 1139f22ef01cSRoman Divacky 1140dff0c46cSDimitry Andric // Try constant folding a bitcast of both instructions to an integer. If we 1141dff0c46cSDimitry Andric // get two identical ConstantInt's, then we are good to share them. We use 1142dff0c46cSDimitry Andric // the constant folding APIs to do this so that we get the benefit of 11433861d79fSDimitry Andric // DataLayout. 1144dff0c46cSDimitry Andric if (isa<PointerType>(A->getType())) 11453ca95b02SDimitry Andric A = ConstantFoldCastOperand(Instruction::PtrToInt, 11463ca95b02SDimitry Andric const_cast<Constant *>(A), IntTy, DL); 1147dff0c46cSDimitry Andric else if (A->getType() != IntTy) 11483ca95b02SDimitry Andric A = ConstantFoldCastOperand(Instruction::BitCast, const_cast<Constant *>(A), 11493ca95b02SDimitry Andric IntTy, DL); 1150dff0c46cSDimitry Andric if (isa<PointerType>(B->getType())) 11513ca95b02SDimitry Andric B = ConstantFoldCastOperand(Instruction::PtrToInt, 11523ca95b02SDimitry Andric const_cast<Constant *>(B), IntTy, DL); 1153dff0c46cSDimitry Andric else if (B->getType() != IntTy) 11543ca95b02SDimitry Andric B = ConstantFoldCastOperand(Instruction::BitCast, const_cast<Constant *>(B), 11553ca95b02SDimitry Andric IntTy, DL); 1156f22ef01cSRoman Divacky 1157dff0c46cSDimitry Andric return A == B; 1158f22ef01cSRoman Divacky } 1159f22ef01cSRoman Divacky 11608f0fd8f6SDimitry Andric /// Create a new entry in the constant pool or return an existing one. 11618f0fd8f6SDimitry Andric /// User must specify the log2 of the minimum required alignment for the object. 1162f22ef01cSRoman Divacky unsigned MachineConstantPool::getConstantPoolIndex(const Constant *C, 1163f22ef01cSRoman Divacky unsigned Alignment) { 1164f22ef01cSRoman Divacky assert(Alignment && "Alignment must be specified!"); 1165f22ef01cSRoman Divacky if (Alignment > PoolAlignment) PoolAlignment = Alignment; 1166f22ef01cSRoman Divacky 1167f22ef01cSRoman Divacky // Check to see if we already have this constant. 1168f22ef01cSRoman Divacky // 1169f22ef01cSRoman Divacky // FIXME, this could be made much more efficient for large constant pools. 1170f22ef01cSRoman Divacky for (unsigned i = 0, e = Constants.size(); i != e; ++i) 1171f22ef01cSRoman Divacky if (!Constants[i].isMachineConstantPoolEntry() && 1172875ed548SDimitry Andric CanShareConstantPoolEntry(Constants[i].Val.ConstVal, C, DL)) { 1173f22ef01cSRoman Divacky if ((unsigned)Constants[i].getAlignment() < Alignment) 1174f22ef01cSRoman Divacky Constants[i].Alignment = Alignment; 1175f22ef01cSRoman Divacky return i; 1176f22ef01cSRoman Divacky } 1177f22ef01cSRoman Divacky 1178f22ef01cSRoman Divacky Constants.push_back(MachineConstantPoolEntry(C, Alignment)); 1179f22ef01cSRoman Divacky return Constants.size()-1; 1180f22ef01cSRoman Divacky } 1181f22ef01cSRoman Divacky 1182f22ef01cSRoman Divacky unsigned MachineConstantPool::getConstantPoolIndex(MachineConstantPoolValue *V, 1183f22ef01cSRoman Divacky unsigned Alignment) { 1184f22ef01cSRoman Divacky assert(Alignment && "Alignment must be specified!"); 1185f22ef01cSRoman Divacky if (Alignment > PoolAlignment) PoolAlignment = Alignment; 1186f22ef01cSRoman Divacky 1187f22ef01cSRoman Divacky // Check to see if we already have this constant. 1188f22ef01cSRoman Divacky // 1189f22ef01cSRoman Divacky // FIXME, this could be made much more efficient for large constant pools. 1190f22ef01cSRoman Divacky int Idx = V->getExistingMachineCPValue(this, Alignment); 1191dd6029ffSDimitry Andric if (Idx != -1) { 1192dd6029ffSDimitry Andric MachineCPVsSharingEntries.insert(V); 1193f22ef01cSRoman Divacky return (unsigned)Idx; 1194dd6029ffSDimitry Andric } 1195f22ef01cSRoman Divacky 1196f22ef01cSRoman Divacky Constants.push_back(MachineConstantPoolEntry(V, Alignment)); 1197f22ef01cSRoman Divacky return Constants.size()-1; 1198f22ef01cSRoman Divacky } 1199f22ef01cSRoman Divacky 1200f22ef01cSRoman Divacky void MachineConstantPool::print(raw_ostream &OS) const { 1201f22ef01cSRoman Divacky if (Constants.empty()) return; 1202f22ef01cSRoman Divacky 1203f22ef01cSRoman Divacky OS << "Constant Pool:\n"; 1204f22ef01cSRoman Divacky for (unsigned i = 0, e = Constants.size(); i != e; ++i) { 1205f22ef01cSRoman Divacky OS << " cp#" << i << ": "; 1206f22ef01cSRoman Divacky if (Constants[i].isMachineConstantPoolEntry()) 1207f22ef01cSRoman Divacky Constants[i].Val.MachineCPVal->print(OS); 1208f22ef01cSRoman Divacky else 120991bc56edSDimitry Andric Constants[i].Val.ConstVal->printAsOperand(OS, /*PrintType=*/false); 1210f22ef01cSRoman Divacky OS << ", align=" << Constants[i].getAlignment(); 1211f22ef01cSRoman Divacky OS << "\n"; 1212f22ef01cSRoman Divacky } 1213f22ef01cSRoman Divacky } 1214f22ef01cSRoman Divacky 12153861d79fSDimitry Andric #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP) 12163ca95b02SDimitry Andric LLVM_DUMP_METHOD void MachineConstantPool::dump() const { print(dbgs()); } 12173861d79fSDimitry Andric #endif 1218