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); 172f22ef01cSRoman Divacky if (RegInfo) { 173f22ef01cSRoman Divacky RegInfo->~MachineRegisterInfo(); 174f22ef01cSRoman Divacky Allocator.Deallocate(RegInfo); 175f22ef01cSRoman Divacky } 176f22ef01cSRoman Divacky if (MFInfo) { 177f22ef01cSRoman Divacky MFInfo->~MachineFunctionInfo(); 178f22ef01cSRoman Divacky Allocator.Deallocate(MFInfo); 179f22ef01cSRoman Divacky } 1807ae0e2c9SDimitry Andric 1817ae0e2c9SDimitry Andric FrameInfo->~MachineFrameInfo(); 1827ae0e2c9SDimitry Andric Allocator.Deallocate(FrameInfo); 1837ae0e2c9SDimitry Andric 1847ae0e2c9SDimitry Andric ConstantPool->~MachineConstantPool(); 1857ae0e2c9SDimitry Andric Allocator.Deallocate(ConstantPool); 186f22ef01cSRoman Divacky 187f22ef01cSRoman Divacky if (JumpTableInfo) { 188f22ef01cSRoman Divacky JumpTableInfo->~MachineJumpTableInfo(); 189f22ef01cSRoman Divacky Allocator.Deallocate(JumpTableInfo); 190f22ef01cSRoman Divacky } 1917d523365SDimitry Andric 1927d523365SDimitry Andric if (WinEHInfo) { 1937d523365SDimitry Andric WinEHInfo->~WinEHFuncInfo(); 1947d523365SDimitry Andric Allocator.Deallocate(WinEHInfo); 1957d523365SDimitry Andric } 196f22ef01cSRoman Divacky } 197f22ef01cSRoman Divacky 198875ed548SDimitry Andric const DataLayout &MachineFunction::getDataLayout() const { 199875ed548SDimitry Andric return Fn->getParent()->getDataLayout(); 200875ed548SDimitry Andric } 201875ed548SDimitry Andric 2028f0fd8f6SDimitry Andric /// Get the JumpTableInfo for this function. 2038f0fd8f6SDimitry Andric /// If it does not already exist, allocate one. 204f22ef01cSRoman Divacky MachineJumpTableInfo *MachineFunction:: 205f22ef01cSRoman Divacky getOrCreateJumpTableInfo(unsigned EntryKind) { 206f22ef01cSRoman Divacky if (JumpTableInfo) return JumpTableInfo; 207f22ef01cSRoman Divacky 208f22ef01cSRoman Divacky JumpTableInfo = new (Allocator) 209f22ef01cSRoman Divacky MachineJumpTableInfo((MachineJumpTableInfo::JTEntryKind)EntryKind); 210f22ef01cSRoman Divacky return JumpTableInfo; 211f22ef01cSRoman Divacky } 212f22ef01cSRoman Divacky 21391bc56edSDimitry Andric /// Should we be emitting segmented stack stuff for the function 2148c24ff90SDimitry Andric bool MachineFunction::shouldSplitStack() const { 21591bc56edSDimitry Andric return getFunction()->hasFnAttribute("split-stack"); 21691bc56edSDimitry Andric } 21791bc56edSDimitry Andric 2188f0fd8f6SDimitry Andric /// This discards all of the MachineBasicBlock numbers and recomputes them. 2198f0fd8f6SDimitry Andric /// This guarantees that the MBB numbers are sequential, dense, and match the 2208f0fd8f6SDimitry Andric /// ordering of the blocks within the function. If a specific MachineBasicBlock 2218f0fd8f6SDimitry Andric /// is specified, only that block and those after it are renumbered. 222f22ef01cSRoman Divacky void MachineFunction::RenumberBlocks(MachineBasicBlock *MBB) { 223f22ef01cSRoman Divacky if (empty()) { MBBNumbering.clear(); return; } 224f22ef01cSRoman Divacky MachineFunction::iterator MBBI, E = end(); 22591bc56edSDimitry Andric if (MBB == nullptr) 226f22ef01cSRoman Divacky MBBI = begin(); 227f22ef01cSRoman Divacky else 2287d523365SDimitry Andric MBBI = MBB->getIterator(); 229f22ef01cSRoman Divacky 230f22ef01cSRoman Divacky // Figure out the block number this should have. 231f22ef01cSRoman Divacky unsigned BlockNo = 0; 232f22ef01cSRoman Divacky if (MBBI != begin()) 23391bc56edSDimitry Andric BlockNo = std::prev(MBBI)->getNumber() + 1; 234f22ef01cSRoman Divacky 235f22ef01cSRoman Divacky for (; MBBI != E; ++MBBI, ++BlockNo) { 236f22ef01cSRoman Divacky if (MBBI->getNumber() != (int)BlockNo) { 237f22ef01cSRoman Divacky // Remove use of the old number. 238f22ef01cSRoman Divacky if (MBBI->getNumber() != -1) { 239f22ef01cSRoman Divacky assert(MBBNumbering[MBBI->getNumber()] == &*MBBI && 240f22ef01cSRoman Divacky "MBB number mismatch!"); 24191bc56edSDimitry Andric MBBNumbering[MBBI->getNumber()] = nullptr; 242f22ef01cSRoman Divacky } 243f22ef01cSRoman Divacky 244f22ef01cSRoman Divacky // If BlockNo is already taken, set that block's number to -1. 245f22ef01cSRoman Divacky if (MBBNumbering[BlockNo]) 246f22ef01cSRoman Divacky MBBNumbering[BlockNo]->setNumber(-1); 247f22ef01cSRoman Divacky 2487d523365SDimitry Andric MBBNumbering[BlockNo] = &*MBBI; 249f22ef01cSRoman Divacky MBBI->setNumber(BlockNo); 250f22ef01cSRoman Divacky } 251f22ef01cSRoman Divacky } 252f22ef01cSRoman Divacky 253f22ef01cSRoman Divacky // Okay, all the blocks are renumbered. If we have compactified the block 254f22ef01cSRoman Divacky // numbering, shrink MBBNumbering now. 255f22ef01cSRoman Divacky assert(BlockNo <= MBBNumbering.size() && "Mismatch!"); 256f22ef01cSRoman Divacky MBBNumbering.resize(BlockNo); 257f22ef01cSRoman Divacky } 258f22ef01cSRoman Divacky 2598f0fd8f6SDimitry Andric /// Allocate a new MachineInstr. Use this instead of `new MachineInstr'. 2603ca95b02SDimitry Andric MachineInstr *MachineFunction::CreateMachineInstr(const MCInstrDesc &MCID, 2613ca95b02SDimitry Andric const DebugLoc &DL, 2623ca95b02SDimitry Andric bool NoImp) { 263f22ef01cSRoman Divacky return new (InstructionRecycler.Allocate<MachineInstr>(Allocator)) 264139f7f9bSDimitry Andric MachineInstr(*this, MCID, DL, NoImp); 265f22ef01cSRoman Divacky } 266f22ef01cSRoman Divacky 2678f0fd8f6SDimitry Andric /// Create a new MachineInstr which is a copy of the 'Orig' instruction, 2688f0fd8f6SDimitry Andric /// identical in all ways except the instruction has no parent, prev, or next. 269f22ef01cSRoman Divacky MachineInstr * 270f22ef01cSRoman Divacky MachineFunction::CloneMachineInstr(const MachineInstr *Orig) { 271f22ef01cSRoman Divacky return new (InstructionRecycler.Allocate<MachineInstr>(Allocator)) 272f22ef01cSRoman Divacky MachineInstr(*this, *Orig); 273f22ef01cSRoman Divacky } 274f22ef01cSRoman Divacky 2758f0fd8f6SDimitry Andric /// Delete the given MachineInstr. 276f22ef01cSRoman Divacky /// 277139f7f9bSDimitry Andric /// This function also serves as the MachineInstr destructor - the real 278139f7f9bSDimitry Andric /// ~MachineInstr() destructor must be empty. 279f22ef01cSRoman Divacky void 280f22ef01cSRoman Divacky MachineFunction::DeleteMachineInstr(MachineInstr *MI) { 281139f7f9bSDimitry Andric // Strip it for parts. The operand array and the MI object itself are 282139f7f9bSDimitry Andric // independently recyclable. 283139f7f9bSDimitry Andric if (MI->Operands) 284139f7f9bSDimitry Andric deallocateOperandArray(MI->CapOperands, MI->Operands); 285139f7f9bSDimitry Andric // Don't call ~MachineInstr() which must be trivial anyway because 286139f7f9bSDimitry Andric // ~MachineFunction drops whole lists of MachineInstrs wihout calling their 287139f7f9bSDimitry Andric // destructors. 288f22ef01cSRoman Divacky InstructionRecycler.Deallocate(Allocator, MI); 289f22ef01cSRoman Divacky } 290f22ef01cSRoman Divacky 2918f0fd8f6SDimitry Andric /// Allocate a new MachineBasicBlock. Use this instead of 2928f0fd8f6SDimitry Andric /// `new MachineBasicBlock'. 293f22ef01cSRoman Divacky MachineBasicBlock * 294f22ef01cSRoman Divacky MachineFunction::CreateMachineBasicBlock(const BasicBlock *bb) { 295f22ef01cSRoman Divacky return new (BasicBlockRecycler.Allocate<MachineBasicBlock>(Allocator)) 296f22ef01cSRoman Divacky MachineBasicBlock(*this, bb); 297f22ef01cSRoman Divacky } 298f22ef01cSRoman Divacky 2998f0fd8f6SDimitry Andric /// Delete the given MachineBasicBlock. 300f22ef01cSRoman Divacky void 301f22ef01cSRoman Divacky MachineFunction::DeleteMachineBasicBlock(MachineBasicBlock *MBB) { 302f22ef01cSRoman Divacky assert(MBB->getParent() == this && "MBB parent mismatch!"); 303f22ef01cSRoman Divacky MBB->~MachineBasicBlock(); 304f22ef01cSRoman Divacky BasicBlockRecycler.Deallocate(Allocator, MBB); 305f22ef01cSRoman Divacky } 306f22ef01cSRoman Divacky 3073ca95b02SDimitry Andric MachineMemOperand *MachineFunction::getMachineMemOperand( 3083ca95b02SDimitry Andric MachinePointerInfo PtrInfo, MachineMemOperand::Flags f, uint64_t s, 309d88c1a5aSDimitry Andric unsigned base_alignment, const AAMDNodes &AAInfo, const MDNode *Ranges, 310d88c1a5aSDimitry Andric SynchronizationScope SynchScope, AtomicOrdering Ordering, 311d88c1a5aSDimitry Andric AtomicOrdering FailureOrdering) { 3123ca95b02SDimitry Andric return new (Allocator) 313d88c1a5aSDimitry Andric MachineMemOperand(PtrInfo, f, s, base_alignment, AAInfo, Ranges, 314d88c1a5aSDimitry Andric SynchScope, Ordering, FailureOrdering); 315f22ef01cSRoman Divacky } 316f22ef01cSRoman Divacky 317f22ef01cSRoman Divacky MachineMemOperand * 318f22ef01cSRoman Divacky MachineFunction::getMachineMemOperand(const MachineMemOperand *MMO, 319f22ef01cSRoman Divacky int64_t Offset, uint64_t Size) { 32091bc56edSDimitry Andric if (MMO->getValue()) 321f22ef01cSRoman Divacky return new (Allocator) 3222754fe60SDimitry Andric MachineMemOperand(MachinePointerInfo(MMO->getValue(), 3232754fe60SDimitry Andric MMO->getOffset()+Offset), 324d88c1a5aSDimitry Andric MMO->getFlags(), Size, MMO->getBaseAlignment(), 325d88c1a5aSDimitry Andric AAMDNodes(), nullptr, MMO->getSynchScope(), 326d88c1a5aSDimitry Andric MMO->getOrdering(), MMO->getFailureOrdering()); 32791bc56edSDimitry Andric return new (Allocator) 32891bc56edSDimitry Andric MachineMemOperand(MachinePointerInfo(MMO->getPseudoValue(), 32991bc56edSDimitry Andric MMO->getOffset()+Offset), 330d88c1a5aSDimitry Andric MMO->getFlags(), Size, MMO->getBaseAlignment(), 331d88c1a5aSDimitry Andric AAMDNodes(), nullptr, MMO->getSynchScope(), 332d88c1a5aSDimitry Andric MMO->getOrdering(), MMO->getFailureOrdering()); 333f22ef01cSRoman Divacky } 334f22ef01cSRoman Divacky 335f22ef01cSRoman Divacky MachineInstr::mmo_iterator 336f22ef01cSRoman Divacky MachineFunction::allocateMemRefsArray(unsigned long Num) { 337f22ef01cSRoman Divacky return Allocator.Allocate<MachineMemOperand *>(Num); 338f22ef01cSRoman Divacky } 339f22ef01cSRoman Divacky 340f22ef01cSRoman Divacky std::pair<MachineInstr::mmo_iterator, MachineInstr::mmo_iterator> 341f22ef01cSRoman Divacky MachineFunction::extractLoadMemRefs(MachineInstr::mmo_iterator Begin, 342f22ef01cSRoman Divacky MachineInstr::mmo_iterator End) { 343f22ef01cSRoman Divacky // Count the number of load mem refs. 344f22ef01cSRoman Divacky unsigned Num = 0; 345f22ef01cSRoman Divacky for (MachineInstr::mmo_iterator I = Begin; I != End; ++I) 346f22ef01cSRoman Divacky if ((*I)->isLoad()) 347f22ef01cSRoman Divacky ++Num; 348f22ef01cSRoman Divacky 349f22ef01cSRoman Divacky // Allocate a new array and populate it with the load information. 350f22ef01cSRoman Divacky MachineInstr::mmo_iterator Result = allocateMemRefsArray(Num); 351f22ef01cSRoman Divacky unsigned Index = 0; 352f22ef01cSRoman Divacky for (MachineInstr::mmo_iterator I = Begin; I != End; ++I) { 353f22ef01cSRoman Divacky if ((*I)->isLoad()) { 354f22ef01cSRoman Divacky if (!(*I)->isStore()) 355f22ef01cSRoman Divacky // Reuse the MMO. 356f22ef01cSRoman Divacky Result[Index] = *I; 357f22ef01cSRoman Divacky else { 358f22ef01cSRoman Divacky // Clone the MMO and unset the store flag. 359f22ef01cSRoman Divacky MachineMemOperand *JustLoad = 3602754fe60SDimitry Andric getMachineMemOperand((*I)->getPointerInfo(), 361f22ef01cSRoman Divacky (*I)->getFlags() & ~MachineMemOperand::MOStore, 3622754fe60SDimitry Andric (*I)->getSize(), (*I)->getBaseAlignment(), 363d88c1a5aSDimitry Andric (*I)->getAAInfo(), nullptr, 364d88c1a5aSDimitry Andric (*I)->getSynchScope(), (*I)->getOrdering(), 365d88c1a5aSDimitry Andric (*I)->getFailureOrdering()); 366f22ef01cSRoman Divacky Result[Index] = JustLoad; 367f22ef01cSRoman Divacky } 368f22ef01cSRoman Divacky ++Index; 369f22ef01cSRoman Divacky } 370f22ef01cSRoman Divacky } 371f22ef01cSRoman Divacky return std::make_pair(Result, Result + Num); 372f22ef01cSRoman Divacky } 373f22ef01cSRoman Divacky 374f22ef01cSRoman Divacky std::pair<MachineInstr::mmo_iterator, MachineInstr::mmo_iterator> 375f22ef01cSRoman Divacky MachineFunction::extractStoreMemRefs(MachineInstr::mmo_iterator Begin, 376f22ef01cSRoman Divacky MachineInstr::mmo_iterator End) { 377f22ef01cSRoman Divacky // Count the number of load mem refs. 378f22ef01cSRoman Divacky unsigned Num = 0; 379f22ef01cSRoman Divacky for (MachineInstr::mmo_iterator I = Begin; I != End; ++I) 380f22ef01cSRoman Divacky if ((*I)->isStore()) 381f22ef01cSRoman Divacky ++Num; 382f22ef01cSRoman Divacky 383f22ef01cSRoman Divacky // Allocate a new array and populate it with the store information. 384f22ef01cSRoman Divacky MachineInstr::mmo_iterator Result = allocateMemRefsArray(Num); 385f22ef01cSRoman Divacky unsigned Index = 0; 386f22ef01cSRoman Divacky for (MachineInstr::mmo_iterator I = Begin; I != End; ++I) { 387f22ef01cSRoman Divacky if ((*I)->isStore()) { 388f22ef01cSRoman Divacky if (!(*I)->isLoad()) 389f22ef01cSRoman Divacky // Reuse the MMO. 390f22ef01cSRoman Divacky Result[Index] = *I; 391f22ef01cSRoman Divacky else { 392f22ef01cSRoman Divacky // Clone the MMO and unset the load flag. 393f22ef01cSRoman Divacky MachineMemOperand *JustStore = 3942754fe60SDimitry Andric getMachineMemOperand((*I)->getPointerInfo(), 395f22ef01cSRoman Divacky (*I)->getFlags() & ~MachineMemOperand::MOLoad, 3962754fe60SDimitry Andric (*I)->getSize(), (*I)->getBaseAlignment(), 397d88c1a5aSDimitry Andric (*I)->getAAInfo(), nullptr, 398d88c1a5aSDimitry Andric (*I)->getSynchScope(), (*I)->getOrdering(), 399d88c1a5aSDimitry Andric (*I)->getFailureOrdering()); 400f22ef01cSRoman Divacky Result[Index] = JustStore; 401f22ef01cSRoman Divacky } 402f22ef01cSRoman Divacky ++Index; 403f22ef01cSRoman Divacky } 404f22ef01cSRoman Divacky } 405f22ef01cSRoman Divacky return std::make_pair(Result, Result + Num); 406f22ef01cSRoman Divacky } 407f22ef01cSRoman Divacky 4087d523365SDimitry Andric const char *MachineFunction::createExternalSymbolName(StringRef Name) { 4097d523365SDimitry Andric char *Dest = Allocator.Allocate<char>(Name.size() + 1); 4107d523365SDimitry Andric std::copy(Name.begin(), Name.end(), Dest); 4117d523365SDimitry Andric Dest[Name.size()] = 0; 4127d523365SDimitry Andric return Dest; 4137d523365SDimitry Andric } 4147d523365SDimitry Andric 4153861d79fSDimitry Andric #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP) 4163ca95b02SDimitry Andric LLVM_DUMP_METHOD void MachineFunction::dump() const { 417f22ef01cSRoman Divacky print(dbgs()); 418f22ef01cSRoman Divacky } 4193861d79fSDimitry Andric #endif 4203861d79fSDimitry Andric 4213861d79fSDimitry Andric StringRef MachineFunction::getName() const { 4223861d79fSDimitry Andric assert(getFunction() && "No function!"); 4233861d79fSDimitry Andric return getFunction()->getName(); 4243861d79fSDimitry Andric } 425f22ef01cSRoman Divacky 4263ca95b02SDimitry Andric void MachineFunction::print(raw_ostream &OS, const SlotIndexes *Indexes) const { 4273861d79fSDimitry Andric OS << "# Machine code for function " << getName() << ": "; 4283ca95b02SDimitry Andric getProperties().print(OS); 429d88c1a5aSDimitry Andric OS << '\n'; 430f22ef01cSRoman Divacky 431f22ef01cSRoman Divacky // Print Frame Information 432f22ef01cSRoman Divacky FrameInfo->print(*this, OS); 433f22ef01cSRoman Divacky 434f22ef01cSRoman Divacky // Print JumpTable Information 435f22ef01cSRoman Divacky if (JumpTableInfo) 436f22ef01cSRoman Divacky JumpTableInfo->print(OS); 437f22ef01cSRoman Divacky 438f22ef01cSRoman Divacky // Print Constant Pool 439f22ef01cSRoman Divacky ConstantPool->print(OS); 440f22ef01cSRoman Divacky 44139d628a0SDimitry Andric const TargetRegisterInfo *TRI = getSubtarget().getRegisterInfo(); 442f22ef01cSRoman Divacky 443f22ef01cSRoman Divacky if (RegInfo && !RegInfo->livein_empty()) { 444f22ef01cSRoman Divacky OS << "Function Live Ins: "; 445f22ef01cSRoman Divacky for (MachineRegisterInfo::livein_iterator 446f22ef01cSRoman Divacky I = RegInfo->livein_begin(), E = RegInfo->livein_end(); I != E; ++I) { 447bd5abe19SDimitry Andric OS << PrintReg(I->first, TRI); 448f22ef01cSRoman Divacky if (I->second) 449bd5abe19SDimitry Andric OS << " in " << PrintReg(I->second, TRI); 45091bc56edSDimitry Andric if (std::next(I) != E) 451f22ef01cSRoman Divacky OS << ", "; 452f22ef01cSRoman Divacky } 453f22ef01cSRoman Divacky OS << '\n'; 454f22ef01cSRoman Divacky } 455f22ef01cSRoman Divacky 4563dac3a9bSDimitry Andric ModuleSlotTracker MST(getFunction()->getParent()); 4573dac3a9bSDimitry Andric MST.incorporateFunction(*getFunction()); 45891bc56edSDimitry Andric for (const auto &BB : *this) { 459f22ef01cSRoman Divacky OS << '\n'; 4603dac3a9bSDimitry Andric BB.print(OS, MST, Indexes); 461f22ef01cSRoman Divacky } 462f22ef01cSRoman Divacky 4633861d79fSDimitry Andric OS << "\n# End machine code for function " << getName() << ".\n\n"; 464f22ef01cSRoman Divacky } 465f22ef01cSRoman Divacky 466f22ef01cSRoman Divacky namespace llvm { 467f22ef01cSRoman Divacky template<> 468f22ef01cSRoman Divacky struct DOTGraphTraits<const MachineFunction*> : public DefaultDOTGraphTraits { 469f22ef01cSRoman Divacky 470f22ef01cSRoman Divacky DOTGraphTraits (bool isSimple=false) : DefaultDOTGraphTraits(isSimple) {} 471f22ef01cSRoman Divacky 472f22ef01cSRoman Divacky static std::string getGraphName(const MachineFunction *F) { 473ff0cc061SDimitry Andric return ("CFG for '" + F->getName() + "' function").str(); 474f22ef01cSRoman Divacky } 475f22ef01cSRoman Divacky 476f22ef01cSRoman Divacky std::string getNodeLabel(const MachineBasicBlock *Node, 477f22ef01cSRoman Divacky const MachineFunction *Graph) { 478f22ef01cSRoman Divacky std::string OutStr; 479f22ef01cSRoman Divacky { 480f22ef01cSRoman Divacky raw_string_ostream OSS(OutStr); 481f22ef01cSRoman Divacky 4822754fe60SDimitry Andric if (isSimple()) { 4832754fe60SDimitry Andric OSS << "BB#" << Node->getNumber(); 4842754fe60SDimitry Andric if (const BasicBlock *BB = Node->getBasicBlock()) 4852754fe60SDimitry Andric OSS << ": " << BB->getName(); 4862754fe60SDimitry Andric } else 487f22ef01cSRoman Divacky Node->print(OSS); 488f22ef01cSRoman Divacky } 489f22ef01cSRoman Divacky 490f22ef01cSRoman Divacky if (OutStr[0] == '\n') OutStr.erase(OutStr.begin()); 491f22ef01cSRoman Divacky 492f22ef01cSRoman Divacky // Process string output to make it nicer... 493f22ef01cSRoman Divacky for (unsigned i = 0; i != OutStr.length(); ++i) 494f22ef01cSRoman Divacky if (OutStr[i] == '\n') { // Left justify 495f22ef01cSRoman Divacky OutStr[i] = '\\'; 496f22ef01cSRoman Divacky OutStr.insert(OutStr.begin()+i+1, 'l'); 497f22ef01cSRoman Divacky } 498f22ef01cSRoman Divacky return OutStr; 499f22ef01cSRoman Divacky } 500f22ef01cSRoman Divacky }; 5013dac3a9bSDimitry Andric } 502f22ef01cSRoman Divacky 503f22ef01cSRoman Divacky void MachineFunction::viewCFG() const 504f22ef01cSRoman Divacky { 505f22ef01cSRoman Divacky #ifndef NDEBUG 5063861d79fSDimitry Andric ViewGraph(this, "mf" + getName()); 507f22ef01cSRoman Divacky #else 508ffd1746dSEd Schouten errs() << "MachineFunction::viewCFG is only available in debug builds on " 509f22ef01cSRoman Divacky << "systems with Graphviz or gv!\n"; 510f22ef01cSRoman Divacky #endif // NDEBUG 511f22ef01cSRoman Divacky } 512f22ef01cSRoman Divacky 513f22ef01cSRoman Divacky void MachineFunction::viewCFGOnly() const 514f22ef01cSRoman Divacky { 515f22ef01cSRoman Divacky #ifndef NDEBUG 5163861d79fSDimitry Andric ViewGraph(this, "mf" + getName(), true); 517f22ef01cSRoman Divacky #else 518ffd1746dSEd Schouten errs() << "MachineFunction::viewCFGOnly is only available in debug builds on " 519f22ef01cSRoman Divacky << "systems with Graphviz or gv!\n"; 520f22ef01cSRoman Divacky #endif // NDEBUG 521f22ef01cSRoman Divacky } 522f22ef01cSRoman Divacky 5238f0fd8f6SDimitry Andric /// Add the specified physical register as a live-in value and 524f22ef01cSRoman Divacky /// create a corresponding virtual register for it. 525f22ef01cSRoman Divacky unsigned MachineFunction::addLiveIn(unsigned PReg, 526dd6029ffSDimitry Andric const TargetRegisterClass *RC) { 527f22ef01cSRoman Divacky MachineRegisterInfo &MRI = getRegInfo(); 528f22ef01cSRoman Divacky unsigned VReg = MRI.getLiveInVirtReg(PReg); 529f22ef01cSRoman Divacky if (VReg) { 53091bc56edSDimitry Andric const TargetRegisterClass *VRegRC = MRI.getRegClass(VReg); 53191bc56edSDimitry Andric (void)VRegRC; 53291bc56edSDimitry Andric // A physical register can be added several times. 53391bc56edSDimitry Andric // Between two calls, the register class of the related virtual register 53491bc56edSDimitry Andric // may have been constrained to match some operation constraints. 53591bc56edSDimitry Andric // In that case, check that the current register class includes the 53691bc56edSDimitry Andric // physical register and is a sub class of the specified RC. 53791bc56edSDimitry Andric assert((VRegRC == RC || (VRegRC->contains(PReg) && 53891bc56edSDimitry Andric RC->hasSubClassEq(VRegRC))) && 53991bc56edSDimitry Andric "Register class mismatch!"); 540f22ef01cSRoman Divacky return VReg; 541f22ef01cSRoman Divacky } 542f22ef01cSRoman Divacky VReg = MRI.createVirtualRegister(RC); 543f22ef01cSRoman Divacky MRI.addLiveIn(PReg, VReg); 544f22ef01cSRoman Divacky return VReg; 545f22ef01cSRoman Divacky } 546f22ef01cSRoman Divacky 5478f0fd8f6SDimitry Andric /// Return the MCSymbol for the specified non-empty jump table. 548f22ef01cSRoman Divacky /// If isLinkerPrivate is specified, an 'l' label is returned, otherwise a 549f22ef01cSRoman Divacky /// normal 'L' label is returned. 550f22ef01cSRoman Divacky MCSymbol *MachineFunction::getJTISymbol(unsigned JTI, MCContext &Ctx, 551f22ef01cSRoman Divacky bool isLinkerPrivate) const { 552875ed548SDimitry Andric const DataLayout &DL = getDataLayout(); 553f22ef01cSRoman Divacky assert(JumpTableInfo && "No jump tables"); 554f22ef01cSRoman Divacky assert(JTI < JumpTableInfo->getJumpTables().size() && "Invalid JTI!"); 555f22ef01cSRoman Divacky 556d88c1a5aSDimitry Andric StringRef Prefix = isLinkerPrivate ? DL.getLinkerPrivateGlobalPrefix() 557875ed548SDimitry Andric : DL.getPrivateGlobalPrefix(); 558f22ef01cSRoman Divacky SmallString<60> Name; 559f22ef01cSRoman Divacky raw_svector_ostream(Name) 560f22ef01cSRoman Divacky << Prefix << "JTI" << getFunctionNumber() << '_' << JTI; 561ff0cc061SDimitry Andric return Ctx.getOrCreateSymbol(Name); 562f22ef01cSRoman Divacky } 563f22ef01cSRoman Divacky 5648f0fd8f6SDimitry Andric /// Return a function-local symbol to represent the PIC base. 5652754fe60SDimitry Andric MCSymbol *MachineFunction::getPICBaseSymbol() const { 566875ed548SDimitry Andric const DataLayout &DL = getDataLayout(); 567875ed548SDimitry Andric return Ctx.getOrCreateSymbol(Twine(DL.getPrivateGlobalPrefix()) + 5682754fe60SDimitry Andric Twine(getFunctionNumber()) + "$pb"); 5692754fe60SDimitry Andric } 570f22ef01cSRoman Divacky 571d88c1a5aSDimitry Andric /// \name Exception Handling 572d88c1a5aSDimitry Andric /// \{ 573d88c1a5aSDimitry Andric 574d88c1a5aSDimitry Andric LandingPadInfo & 575d88c1a5aSDimitry Andric MachineFunction::getOrCreateLandingPadInfo(MachineBasicBlock *LandingPad) { 576d88c1a5aSDimitry Andric unsigned N = LandingPads.size(); 577d88c1a5aSDimitry Andric for (unsigned i = 0; i < N; ++i) { 578d88c1a5aSDimitry Andric LandingPadInfo &LP = LandingPads[i]; 579d88c1a5aSDimitry Andric if (LP.LandingPadBlock == LandingPad) 580d88c1a5aSDimitry Andric return LP; 581d88c1a5aSDimitry Andric } 582d88c1a5aSDimitry Andric 583d88c1a5aSDimitry Andric LandingPads.push_back(LandingPadInfo(LandingPad)); 584d88c1a5aSDimitry Andric return LandingPads[N]; 585d88c1a5aSDimitry Andric } 586d88c1a5aSDimitry Andric 587d88c1a5aSDimitry Andric void MachineFunction::addInvoke(MachineBasicBlock *LandingPad, 588d88c1a5aSDimitry Andric MCSymbol *BeginLabel, MCSymbol *EndLabel) { 589d88c1a5aSDimitry Andric LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad); 590d88c1a5aSDimitry Andric LP.BeginLabels.push_back(BeginLabel); 591d88c1a5aSDimitry Andric LP.EndLabels.push_back(EndLabel); 592d88c1a5aSDimitry Andric } 593d88c1a5aSDimitry Andric 594d88c1a5aSDimitry Andric MCSymbol *MachineFunction::addLandingPad(MachineBasicBlock *LandingPad) { 595d88c1a5aSDimitry Andric MCSymbol *LandingPadLabel = Ctx.createTempSymbol(); 596d88c1a5aSDimitry Andric LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad); 597d88c1a5aSDimitry Andric LP.LandingPadLabel = LandingPadLabel; 598d88c1a5aSDimitry Andric return LandingPadLabel; 599d88c1a5aSDimitry Andric } 600d88c1a5aSDimitry Andric 601d88c1a5aSDimitry Andric void MachineFunction::addCatchTypeInfo(MachineBasicBlock *LandingPad, 602d88c1a5aSDimitry Andric ArrayRef<const GlobalValue *> TyInfo) { 603d88c1a5aSDimitry Andric LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad); 604d88c1a5aSDimitry Andric for (unsigned N = TyInfo.size(); N; --N) 605d88c1a5aSDimitry Andric LP.TypeIds.push_back(getTypeIDFor(TyInfo[N - 1])); 606d88c1a5aSDimitry Andric } 607d88c1a5aSDimitry Andric 608d88c1a5aSDimitry Andric void MachineFunction::addFilterTypeInfo(MachineBasicBlock *LandingPad, 609d88c1a5aSDimitry Andric ArrayRef<const GlobalValue *> TyInfo) { 610d88c1a5aSDimitry Andric LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad); 611d88c1a5aSDimitry Andric std::vector<unsigned> IdsInFilter(TyInfo.size()); 612d88c1a5aSDimitry Andric for (unsigned I = 0, E = TyInfo.size(); I != E; ++I) 613d88c1a5aSDimitry Andric IdsInFilter[I] = getTypeIDFor(TyInfo[I]); 614d88c1a5aSDimitry Andric LP.TypeIds.push_back(getFilterIDFor(IdsInFilter)); 615d88c1a5aSDimitry Andric } 616d88c1a5aSDimitry Andric 617d88c1a5aSDimitry Andric void MachineFunction::tidyLandingPads(DenseMap<MCSymbol*, uintptr_t> *LPMap) { 618d88c1a5aSDimitry Andric for (unsigned i = 0; i != LandingPads.size(); ) { 619d88c1a5aSDimitry Andric LandingPadInfo &LandingPad = LandingPads[i]; 620d88c1a5aSDimitry Andric if (LandingPad.LandingPadLabel && 621d88c1a5aSDimitry Andric !LandingPad.LandingPadLabel->isDefined() && 622d88c1a5aSDimitry Andric (!LPMap || (*LPMap)[LandingPad.LandingPadLabel] == 0)) 623d88c1a5aSDimitry Andric LandingPad.LandingPadLabel = nullptr; 624d88c1a5aSDimitry Andric 625d88c1a5aSDimitry Andric // Special case: we *should* emit LPs with null LP MBB. This indicates 626d88c1a5aSDimitry Andric // "nounwind" case. 627d88c1a5aSDimitry Andric if (!LandingPad.LandingPadLabel && LandingPad.LandingPadBlock) { 628d88c1a5aSDimitry Andric LandingPads.erase(LandingPads.begin() + i); 629d88c1a5aSDimitry Andric continue; 630d88c1a5aSDimitry Andric } 631d88c1a5aSDimitry Andric 632d88c1a5aSDimitry Andric for (unsigned j = 0, e = LandingPads[i].BeginLabels.size(); j != e; ++j) { 633d88c1a5aSDimitry Andric MCSymbol *BeginLabel = LandingPad.BeginLabels[j]; 634d88c1a5aSDimitry Andric MCSymbol *EndLabel = LandingPad.EndLabels[j]; 635d88c1a5aSDimitry Andric if ((BeginLabel->isDefined() || 636d88c1a5aSDimitry Andric (LPMap && (*LPMap)[BeginLabel] != 0)) && 637d88c1a5aSDimitry Andric (EndLabel->isDefined() || 638d88c1a5aSDimitry Andric (LPMap && (*LPMap)[EndLabel] != 0))) continue; 639d88c1a5aSDimitry Andric 640d88c1a5aSDimitry Andric LandingPad.BeginLabels.erase(LandingPad.BeginLabels.begin() + j); 641d88c1a5aSDimitry Andric LandingPad.EndLabels.erase(LandingPad.EndLabels.begin() + j); 642d88c1a5aSDimitry Andric --j; 643d88c1a5aSDimitry Andric --e; 644d88c1a5aSDimitry Andric } 645d88c1a5aSDimitry Andric 646d88c1a5aSDimitry Andric // Remove landing pads with no try-ranges. 647d88c1a5aSDimitry Andric if (LandingPads[i].BeginLabels.empty()) { 648d88c1a5aSDimitry Andric LandingPads.erase(LandingPads.begin() + i); 649d88c1a5aSDimitry Andric continue; 650d88c1a5aSDimitry Andric } 651d88c1a5aSDimitry Andric 652d88c1a5aSDimitry Andric // If there is no landing pad, ensure that the list of typeids is empty. 653d88c1a5aSDimitry Andric // If the only typeid is a cleanup, this is the same as having no typeids. 654d88c1a5aSDimitry Andric if (!LandingPad.LandingPadBlock || 655d88c1a5aSDimitry Andric (LandingPad.TypeIds.size() == 1 && !LandingPad.TypeIds[0])) 656d88c1a5aSDimitry Andric LandingPad.TypeIds.clear(); 657d88c1a5aSDimitry Andric ++i; 658d88c1a5aSDimitry Andric } 659d88c1a5aSDimitry Andric } 660d88c1a5aSDimitry Andric 661d88c1a5aSDimitry Andric void MachineFunction::addCleanup(MachineBasicBlock *LandingPad) { 662d88c1a5aSDimitry Andric LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad); 663d88c1a5aSDimitry Andric LP.TypeIds.push_back(0); 664d88c1a5aSDimitry Andric } 665d88c1a5aSDimitry Andric 666d88c1a5aSDimitry Andric void MachineFunction::addSEHCatchHandler(MachineBasicBlock *LandingPad, 667d88c1a5aSDimitry Andric const Function *Filter, 668d88c1a5aSDimitry Andric const BlockAddress *RecoverBA) { 669d88c1a5aSDimitry Andric LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad); 670d88c1a5aSDimitry Andric SEHHandler Handler; 671d88c1a5aSDimitry Andric Handler.FilterOrFinally = Filter; 672d88c1a5aSDimitry Andric Handler.RecoverBA = RecoverBA; 673d88c1a5aSDimitry Andric LP.SEHHandlers.push_back(Handler); 674d88c1a5aSDimitry Andric } 675d88c1a5aSDimitry Andric 676d88c1a5aSDimitry Andric void MachineFunction::addSEHCleanupHandler(MachineBasicBlock *LandingPad, 677d88c1a5aSDimitry Andric const Function *Cleanup) { 678d88c1a5aSDimitry Andric LandingPadInfo &LP = getOrCreateLandingPadInfo(LandingPad); 679d88c1a5aSDimitry Andric SEHHandler Handler; 680d88c1a5aSDimitry Andric Handler.FilterOrFinally = Cleanup; 681d88c1a5aSDimitry Andric Handler.RecoverBA = nullptr; 682d88c1a5aSDimitry Andric LP.SEHHandlers.push_back(Handler); 683d88c1a5aSDimitry Andric } 684d88c1a5aSDimitry Andric 685d88c1a5aSDimitry Andric void MachineFunction::setCallSiteLandingPad(MCSymbol *Sym, 686d88c1a5aSDimitry Andric ArrayRef<unsigned> Sites) { 687d88c1a5aSDimitry Andric LPadToCallSiteMap[Sym].append(Sites.begin(), Sites.end()); 688d88c1a5aSDimitry Andric } 689d88c1a5aSDimitry Andric 690d88c1a5aSDimitry Andric unsigned MachineFunction::getTypeIDFor(const GlobalValue *TI) { 691d88c1a5aSDimitry Andric for (unsigned i = 0, N = TypeInfos.size(); i != N; ++i) 692d88c1a5aSDimitry Andric if (TypeInfos[i] == TI) return i + 1; 693d88c1a5aSDimitry Andric 694d88c1a5aSDimitry Andric TypeInfos.push_back(TI); 695d88c1a5aSDimitry Andric return TypeInfos.size(); 696d88c1a5aSDimitry Andric } 697d88c1a5aSDimitry Andric 698d88c1a5aSDimitry Andric int MachineFunction::getFilterIDFor(std::vector<unsigned> &TyIds) { 699d88c1a5aSDimitry Andric // If the new filter coincides with the tail of an existing filter, then 700d88c1a5aSDimitry Andric // re-use the existing filter. Folding filters more than this requires 701d88c1a5aSDimitry Andric // re-ordering filters and/or their elements - probably not worth it. 702d88c1a5aSDimitry Andric for (std::vector<unsigned>::iterator I = FilterEnds.begin(), 703d88c1a5aSDimitry Andric E = FilterEnds.end(); I != E; ++I) { 704d88c1a5aSDimitry Andric unsigned i = *I, j = TyIds.size(); 705d88c1a5aSDimitry Andric 706d88c1a5aSDimitry Andric while (i && j) 707d88c1a5aSDimitry Andric if (FilterIds[--i] != TyIds[--j]) 708d88c1a5aSDimitry Andric goto try_next; 709d88c1a5aSDimitry Andric 710d88c1a5aSDimitry Andric if (!j) 711d88c1a5aSDimitry Andric // The new filter coincides with range [i, end) of the existing filter. 712d88c1a5aSDimitry Andric return -(1 + i); 713d88c1a5aSDimitry Andric 714d88c1a5aSDimitry Andric try_next:; 715d88c1a5aSDimitry Andric } 716d88c1a5aSDimitry Andric 717d88c1a5aSDimitry Andric // Add the new filter. 718d88c1a5aSDimitry Andric int FilterID = -(1 + FilterIds.size()); 719d88c1a5aSDimitry Andric FilterIds.reserve(FilterIds.size() + TyIds.size() + 1); 720d88c1a5aSDimitry Andric FilterIds.insert(FilterIds.end(), TyIds.begin(), TyIds.end()); 721d88c1a5aSDimitry Andric FilterEnds.push_back(FilterIds.size()); 722d88c1a5aSDimitry Andric FilterIds.push_back(0); // terminator 723d88c1a5aSDimitry Andric return FilterID; 724d88c1a5aSDimitry Andric } 725d88c1a5aSDimitry Andric 726d88c1a5aSDimitry Andric void llvm::addLandingPadInfo(const LandingPadInst &I, MachineBasicBlock &MBB) { 727d88c1a5aSDimitry Andric MachineFunction &MF = *MBB.getParent(); 728d88c1a5aSDimitry Andric if (const auto *PF = dyn_cast<Function>( 729d88c1a5aSDimitry Andric I.getParent()->getParent()->getPersonalityFn()->stripPointerCasts())) 730d88c1a5aSDimitry Andric MF.getMMI().addPersonality(PF); 731d88c1a5aSDimitry Andric 732d88c1a5aSDimitry Andric if (I.isCleanup()) 733d88c1a5aSDimitry Andric MF.addCleanup(&MBB); 734d88c1a5aSDimitry Andric 735d88c1a5aSDimitry Andric // FIXME: New EH - Add the clauses in reverse order. This isn't 100% correct, 736d88c1a5aSDimitry Andric // but we need to do it this way because of how the DWARF EH emitter 737d88c1a5aSDimitry Andric // processes the clauses. 738d88c1a5aSDimitry Andric for (unsigned i = I.getNumClauses(); i != 0; --i) { 739d88c1a5aSDimitry Andric Value *Val = I.getClause(i - 1); 740d88c1a5aSDimitry Andric if (I.isCatch(i - 1)) { 741d88c1a5aSDimitry Andric MF.addCatchTypeInfo(&MBB, 742d88c1a5aSDimitry Andric dyn_cast<GlobalValue>(Val->stripPointerCasts())); 743d88c1a5aSDimitry Andric } else { 744d88c1a5aSDimitry Andric // Add filters in a list. 745d88c1a5aSDimitry Andric Constant *CVal = cast<Constant>(Val); 746d88c1a5aSDimitry Andric SmallVector<const GlobalValue *, 4> FilterList; 747d88c1a5aSDimitry Andric for (User::op_iterator II = CVal->op_begin(), IE = CVal->op_end(); 748d88c1a5aSDimitry Andric II != IE; ++II) 749d88c1a5aSDimitry Andric FilterList.push_back(cast<GlobalValue>((*II)->stripPointerCasts())); 750d88c1a5aSDimitry Andric 751d88c1a5aSDimitry Andric MF.addFilterTypeInfo(&MBB, FilterList); 752d88c1a5aSDimitry Andric } 753d88c1a5aSDimitry Andric } 754d88c1a5aSDimitry Andric } 755d88c1a5aSDimitry Andric 756d88c1a5aSDimitry Andric /// \} 757d88c1a5aSDimitry Andric 758f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 759f22ef01cSRoman Divacky // MachineFrameInfo implementation 760f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 761f22ef01cSRoman Divacky 7628f0fd8f6SDimitry Andric /// Make sure the function is at least Align bytes aligned. 763139f7f9bSDimitry Andric void MachineFrameInfo::ensureMaxAlignment(unsigned Align) { 7643ca95b02SDimitry Andric if (!StackRealignable) 76539d628a0SDimitry Andric assert(Align <= StackAlignment && 766139f7f9bSDimitry Andric "For targets without stack realignment, Align is out of limit!"); 767139f7f9bSDimitry Andric if (MaxAlignment < Align) MaxAlignment = Align; 768139f7f9bSDimitry Andric } 769139f7f9bSDimitry Andric 7708f0fd8f6SDimitry Andric /// Clamp the alignment if requested and emit a warning. 771139f7f9bSDimitry Andric static inline unsigned clampStackAlignment(bool ShouldClamp, unsigned Align, 772139f7f9bSDimitry Andric unsigned StackAlign) { 773139f7f9bSDimitry Andric if (!ShouldClamp || Align <= StackAlign) 774139f7f9bSDimitry Andric return Align; 775139f7f9bSDimitry Andric DEBUG(dbgs() << "Warning: requested alignment " << Align 776139f7f9bSDimitry Andric << " exceeds the stack alignment " << StackAlign 777139f7f9bSDimitry Andric << " when stack realignment is off" << '\n'); 778139f7f9bSDimitry Andric return StackAlign; 779139f7f9bSDimitry Andric } 780139f7f9bSDimitry Andric 7818f0fd8f6SDimitry Andric /// Create a new statically sized stack object, returning a nonnegative 7828f0fd8f6SDimitry Andric /// identifier to represent it. 783139f7f9bSDimitry Andric int MachineFrameInfo::CreateStackObject(uint64_t Size, unsigned Alignment, 78491bc56edSDimitry Andric bool isSS, const AllocaInst *Alloca) { 785139f7f9bSDimitry Andric assert(Size != 0 && "Cannot allocate zero size stack objects!"); 7863ca95b02SDimitry Andric Alignment = clampStackAlignment(!StackRealignable, Alignment, StackAlignment); 78739d628a0SDimitry Andric Objects.push_back(StackObject(Size, Alignment, 0, false, isSS, Alloca, 78839d628a0SDimitry Andric !isSS)); 789139f7f9bSDimitry Andric int Index = (int)Objects.size() - NumFixedObjects - 1; 790139f7f9bSDimitry Andric assert(Index >= 0 && "Bad frame index!"); 791139f7f9bSDimitry Andric ensureMaxAlignment(Alignment); 792139f7f9bSDimitry Andric return Index; 793139f7f9bSDimitry Andric } 794139f7f9bSDimitry Andric 7958f0fd8f6SDimitry Andric /// Create a new statically sized stack object that represents a spill slot, 7968f0fd8f6SDimitry Andric /// returning a nonnegative identifier to represent it. 797139f7f9bSDimitry Andric int MachineFrameInfo::CreateSpillStackObject(uint64_t Size, 798139f7f9bSDimitry Andric unsigned Alignment) { 7993ca95b02SDimitry Andric Alignment = clampStackAlignment(!StackRealignable, Alignment, StackAlignment); 80091bc56edSDimitry Andric CreateStackObject(Size, Alignment, true); 801139f7f9bSDimitry Andric int Index = (int)Objects.size() - NumFixedObjects - 1; 802139f7f9bSDimitry Andric ensureMaxAlignment(Alignment); 803139f7f9bSDimitry Andric return Index; 804139f7f9bSDimitry Andric } 805139f7f9bSDimitry Andric 8068f0fd8f6SDimitry Andric /// Notify the MachineFrameInfo object that a variable sized object has been 8078f0fd8f6SDimitry Andric /// created. This must be created whenever a variable sized object is created, 8088f0fd8f6SDimitry Andric /// whether or not the index returned is actually used. 809*4f00c8c6SDimitry Andric int MachineFrameInfo::CreateVariableSizedObject(unsigned Alignment, 810*4f00c8c6SDimitry Andric const AllocaInst *Alloca) { 811139f7f9bSDimitry Andric HasVarSizedObjects = true; 8123ca95b02SDimitry Andric Alignment = clampStackAlignment(!StackRealignable, Alignment, StackAlignment); 81339d628a0SDimitry Andric Objects.push_back(StackObject(0, Alignment, 0, false, false, Alloca, true)); 814139f7f9bSDimitry Andric ensureMaxAlignment(Alignment); 815139f7f9bSDimitry Andric return (int)Objects.size()-NumFixedObjects-1; 816139f7f9bSDimitry Andric } 817139f7f9bSDimitry Andric 8188f0fd8f6SDimitry Andric /// Create a new object at a fixed location on the stack. 819f22ef01cSRoman Divacky /// All fixed objects should be created before other objects are created for 820f22ef01cSRoman Divacky /// efficiency. By default, fixed objects are immutable. This returns an 821f22ef01cSRoman Divacky /// index with a negative value. 822f22ef01cSRoman Divacky int MachineFrameInfo::CreateFixedObject(uint64_t Size, int64_t SPOffset, 82339d628a0SDimitry Andric bool Immutable, bool isAliased) { 824f22ef01cSRoman Divacky assert(Size != 0 && "Cannot allocate zero size fixed stack objects!"); 825ffd1746dSEd Schouten // The alignment of the frame index can be determined from its offset from 826ffd1746dSEd Schouten // the incoming frame position. If the frame object is at offset 32 and 827ffd1746dSEd Schouten // the stack is guaranteed to be 16-byte aligned, then we know that the 8283ca95b02SDimitry Andric // object is 16-byte aligned. Note that unlike the non-fixed case, if the 8293ca95b02SDimitry Andric // stack needs realignment, we can't assume that the stack will in fact be 8303ca95b02SDimitry Andric // aligned. 8313ca95b02SDimitry Andric unsigned Align = MinAlign(SPOffset, ForcedRealign ? 1 : StackAlignment); 8323ca95b02SDimitry Andric Align = clampStackAlignment(!StackRealignable, Align, StackAlignment); 833ffd1746dSEd Schouten Objects.insert(Objects.begin(), StackObject(Size, Align, SPOffset, Immutable, 8343861d79fSDimitry Andric /*isSS*/ false, 83539d628a0SDimitry Andric /*Alloca*/ nullptr, isAliased)); 836f22ef01cSRoman Divacky return -++NumFixedObjects; 837f22ef01cSRoman Divacky } 838f22ef01cSRoman Divacky 8398f0fd8f6SDimitry Andric /// Create a spill slot at a fixed location on the stack. 8408f0fd8f6SDimitry Andric /// Returns an index with a negative value. 84191bc56edSDimitry Andric int MachineFrameInfo::CreateFixedSpillStackObject(uint64_t Size, 842d88c1a5aSDimitry Andric int64_t SPOffset, 843d88c1a5aSDimitry Andric bool Immutable) { 8443ca95b02SDimitry Andric unsigned Align = MinAlign(SPOffset, ForcedRealign ? 1 : StackAlignment); 8453ca95b02SDimitry Andric Align = clampStackAlignment(!StackRealignable, Align, StackAlignment); 846d88c1a5aSDimitry Andric Objects.insert(Objects.begin(), StackObject(Size, Align, SPOffset, Immutable, 84791bc56edSDimitry Andric /*isSS*/ true, 84839d628a0SDimitry Andric /*Alloca*/ nullptr, 84939d628a0SDimitry Andric /*isAliased*/ false)); 85091bc56edSDimitry Andric return -++NumFixedObjects; 85191bc56edSDimitry Andric } 852f22ef01cSRoman Divacky 85397bc6c73SDimitry Andric BitVector MachineFrameInfo::getPristineRegs(const MachineFunction &MF) const { 85497bc6c73SDimitry Andric const TargetRegisterInfo *TRI = MF.getSubtarget().getRegisterInfo(); 855f22ef01cSRoman Divacky BitVector BV(TRI->getNumRegs()); 856f22ef01cSRoman Divacky 857f22ef01cSRoman Divacky // Before CSI is calculated, no registers are considered pristine. They can be 858f22ef01cSRoman Divacky // freely used and PEI will make sure they are saved. 859f22ef01cSRoman Divacky if (!isCalleeSavedInfoValid()) 860f22ef01cSRoman Divacky return BV; 861f22ef01cSRoman Divacky 86297bc6c73SDimitry Andric for (const MCPhysReg *CSR = TRI->getCalleeSavedRegs(&MF); CSR && *CSR; ++CSR) 863f22ef01cSRoman Divacky BV.set(*CSR); 864f22ef01cSRoman Divacky 86597bc6c73SDimitry Andric // Saved CSRs are not pristine. 8667d523365SDimitry Andric for (auto &I : getCalleeSavedInfo()) 8677d523365SDimitry Andric for (MCSubRegIterator S(I.getReg(), TRI, true); S.isValid(); ++S) 8687d523365SDimitry Andric BV.reset(*S); 869f22ef01cSRoman Divacky 870f22ef01cSRoman Divacky return BV; 871f22ef01cSRoman Divacky } 872f22ef01cSRoman Divacky 873139f7f9bSDimitry Andric unsigned MachineFrameInfo::estimateStackSize(const MachineFunction &MF) const { 87439d628a0SDimitry Andric const TargetFrameLowering *TFI = MF.getSubtarget().getFrameLowering(); 87539d628a0SDimitry Andric const TargetRegisterInfo *RegInfo = MF.getSubtarget().getRegisterInfo(); 876139f7f9bSDimitry Andric unsigned MaxAlign = getMaxAlignment(); 877139f7f9bSDimitry Andric int Offset = 0; 878139f7f9bSDimitry Andric 879139f7f9bSDimitry Andric // This code is very, very similar to PEI::calculateFrameObjectOffsets(). 880139f7f9bSDimitry Andric // It really should be refactored to share code. Until then, changes 881139f7f9bSDimitry Andric // should keep in mind that there's tight coupling between the two. 882139f7f9bSDimitry Andric 883139f7f9bSDimitry Andric for (int i = getObjectIndexBegin(); i != 0; ++i) { 884139f7f9bSDimitry Andric int FixedOff = -getObjectOffset(i); 885139f7f9bSDimitry Andric if (FixedOff > Offset) Offset = FixedOff; 886139f7f9bSDimitry Andric } 887139f7f9bSDimitry Andric for (unsigned i = 0, e = getObjectIndexEnd(); i != e; ++i) { 888139f7f9bSDimitry Andric if (isDeadObjectIndex(i)) 889139f7f9bSDimitry Andric continue; 890139f7f9bSDimitry Andric Offset += getObjectSize(i); 891139f7f9bSDimitry Andric unsigned Align = getObjectAlignment(i); 892139f7f9bSDimitry Andric // Adjust to alignment boundary 893139f7f9bSDimitry Andric Offset = (Offset+Align-1)/Align*Align; 894139f7f9bSDimitry Andric 895139f7f9bSDimitry Andric MaxAlign = std::max(Align, MaxAlign); 896139f7f9bSDimitry Andric } 897139f7f9bSDimitry Andric 898139f7f9bSDimitry Andric if (adjustsStack() && TFI->hasReservedCallFrame(MF)) 899139f7f9bSDimitry Andric Offset += getMaxCallFrameSize(); 900139f7f9bSDimitry Andric 901139f7f9bSDimitry Andric // Round up the size to a multiple of the alignment. If the function has 902139f7f9bSDimitry Andric // any calls or alloca's, align to the target's StackAlignment value to 903139f7f9bSDimitry Andric // ensure that the callee's frame or the alloca data is suitably aligned; 904139f7f9bSDimitry Andric // otherwise, for leaf functions, align to the TransientStackAlignment 905139f7f9bSDimitry Andric // value. 906139f7f9bSDimitry Andric unsigned StackAlign; 907139f7f9bSDimitry Andric if (adjustsStack() || hasVarSizedObjects() || 908139f7f9bSDimitry Andric (RegInfo->needsStackRealignment(MF) && getObjectIndexEnd() != 0)) 909139f7f9bSDimitry Andric StackAlign = TFI->getStackAlignment(); 910139f7f9bSDimitry Andric else 911139f7f9bSDimitry Andric StackAlign = TFI->getTransientStackAlignment(); 912139f7f9bSDimitry Andric 913139f7f9bSDimitry Andric // If the frame pointer is eliminated, all frame offsets will be relative to 914139f7f9bSDimitry Andric // SP not FP. Align to MaxAlign so this works. 915139f7f9bSDimitry Andric StackAlign = std::max(StackAlign, MaxAlign); 916139f7f9bSDimitry Andric unsigned AlignMask = StackAlign - 1; 917139f7f9bSDimitry Andric Offset = (Offset + AlignMask) & ~uint64_t(AlignMask); 918139f7f9bSDimitry Andric 919139f7f9bSDimitry Andric return (unsigned)Offset; 920139f7f9bSDimitry Andric } 921f22ef01cSRoman Divacky 922f22ef01cSRoman Divacky void MachineFrameInfo::print(const MachineFunction &MF, raw_ostream &OS) const{ 923f22ef01cSRoman Divacky if (Objects.empty()) return; 924f22ef01cSRoman Divacky 92539d628a0SDimitry Andric const TargetFrameLowering *FI = MF.getSubtarget().getFrameLowering(); 926f22ef01cSRoman Divacky int ValOffset = (FI ? FI->getOffsetOfLocalArea() : 0); 927f22ef01cSRoman Divacky 928f22ef01cSRoman Divacky OS << "Frame Objects:\n"; 929f22ef01cSRoman Divacky 930f22ef01cSRoman Divacky for (unsigned i = 0, e = Objects.size(); i != e; ++i) { 931f22ef01cSRoman Divacky const StackObject &SO = Objects[i]; 932f22ef01cSRoman Divacky OS << " fi#" << (int)(i-NumFixedObjects) << ": "; 933f22ef01cSRoman Divacky if (SO.Size == ~0ULL) { 934f22ef01cSRoman Divacky OS << "dead\n"; 935f22ef01cSRoman Divacky continue; 936f22ef01cSRoman Divacky } 937f22ef01cSRoman Divacky if (SO.Size == 0) 938f22ef01cSRoman Divacky OS << "variable sized"; 939f22ef01cSRoman Divacky else 940f22ef01cSRoman Divacky OS << "size=" << SO.Size; 941f22ef01cSRoman Divacky OS << ", align=" << SO.Alignment; 942f22ef01cSRoman Divacky 943f22ef01cSRoman Divacky if (i < NumFixedObjects) 944f22ef01cSRoman Divacky OS << ", fixed"; 945f22ef01cSRoman Divacky if (i < NumFixedObjects || SO.SPOffset != -1) { 946f22ef01cSRoman Divacky int64_t Off = SO.SPOffset - ValOffset; 947f22ef01cSRoman Divacky OS << ", at location [SP"; 948f22ef01cSRoman Divacky if (Off > 0) 949f22ef01cSRoman Divacky OS << "+" << Off; 950f22ef01cSRoman Divacky else if (Off < 0) 951f22ef01cSRoman Divacky OS << Off; 952f22ef01cSRoman Divacky OS << "]"; 953f22ef01cSRoman Divacky } 954f22ef01cSRoman Divacky OS << "\n"; 955f22ef01cSRoman Divacky } 956f22ef01cSRoman Divacky } 957f22ef01cSRoman Divacky 9583861d79fSDimitry Andric #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP) 959f22ef01cSRoman Divacky void MachineFrameInfo::dump(const MachineFunction &MF) const { 960f22ef01cSRoman Divacky print(MF, dbgs()); 961f22ef01cSRoman Divacky } 9623861d79fSDimitry Andric #endif 963f22ef01cSRoman Divacky 964f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 965f22ef01cSRoman Divacky // MachineJumpTableInfo implementation 966f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 967f22ef01cSRoman Divacky 9688f0fd8f6SDimitry Andric /// Return the size of each entry in the jump table. 9693861d79fSDimitry Andric unsigned MachineJumpTableInfo::getEntrySize(const DataLayout &TD) const { 970f22ef01cSRoman Divacky // The size of a jump table entry is 4 bytes unless the entry is just the 971f22ef01cSRoman Divacky // address of a block, in which case it is the pointer size. 972f22ef01cSRoman Divacky switch (getEntryKind()) { 973f22ef01cSRoman Divacky case MachineJumpTableInfo::EK_BlockAddress: 974f22ef01cSRoman Divacky return TD.getPointerSize(); 975dff0c46cSDimitry Andric case MachineJumpTableInfo::EK_GPRel64BlockAddress: 976dff0c46cSDimitry Andric return 8; 977f22ef01cSRoman Divacky case MachineJumpTableInfo::EK_GPRel32BlockAddress: 978f22ef01cSRoman Divacky case MachineJumpTableInfo::EK_LabelDifference32: 979f22ef01cSRoman Divacky case MachineJumpTableInfo::EK_Custom32: 980f22ef01cSRoman Divacky return 4; 981f22ef01cSRoman Divacky case MachineJumpTableInfo::EK_Inline: 982f22ef01cSRoman Divacky return 0; 983f22ef01cSRoman Divacky } 984dff0c46cSDimitry Andric llvm_unreachable("Unknown jump table encoding!"); 985f22ef01cSRoman Divacky } 986f22ef01cSRoman Divacky 9878f0fd8f6SDimitry Andric /// Return the alignment of each entry in the jump table. 9883861d79fSDimitry Andric unsigned MachineJumpTableInfo::getEntryAlignment(const DataLayout &TD) const { 989f22ef01cSRoman Divacky // The alignment of a jump table entry is the alignment of int32 unless the 990f22ef01cSRoman Divacky // entry is just the address of a block, in which case it is the pointer 991f22ef01cSRoman Divacky // alignment. 992f22ef01cSRoman Divacky switch (getEntryKind()) { 993f22ef01cSRoman Divacky case MachineJumpTableInfo::EK_BlockAddress: 994f22ef01cSRoman Divacky return TD.getPointerABIAlignment(); 995dff0c46cSDimitry Andric case MachineJumpTableInfo::EK_GPRel64BlockAddress: 996dff0c46cSDimitry Andric return TD.getABIIntegerTypeAlignment(64); 997f22ef01cSRoman Divacky case MachineJumpTableInfo::EK_GPRel32BlockAddress: 998f22ef01cSRoman Divacky case MachineJumpTableInfo::EK_LabelDifference32: 999f22ef01cSRoman Divacky case MachineJumpTableInfo::EK_Custom32: 1000f22ef01cSRoman Divacky return TD.getABIIntegerTypeAlignment(32); 1001f22ef01cSRoman Divacky case MachineJumpTableInfo::EK_Inline: 1002f22ef01cSRoman Divacky return 1; 1003f22ef01cSRoman Divacky } 1004dff0c46cSDimitry Andric llvm_unreachable("Unknown jump table encoding!"); 1005f22ef01cSRoman Divacky } 1006f22ef01cSRoman Divacky 10078f0fd8f6SDimitry Andric /// Create a new jump table entry in the jump table info. 1008f22ef01cSRoman Divacky unsigned MachineJumpTableInfo::createJumpTableIndex( 1009f22ef01cSRoman Divacky const std::vector<MachineBasicBlock*> &DestBBs) { 1010f22ef01cSRoman Divacky assert(!DestBBs.empty() && "Cannot create an empty jump table!"); 1011f22ef01cSRoman Divacky JumpTables.push_back(MachineJumpTableEntry(DestBBs)); 1012f22ef01cSRoman Divacky return JumpTables.size()-1; 1013f22ef01cSRoman Divacky } 1014f22ef01cSRoman Divacky 10158f0fd8f6SDimitry Andric /// If Old is the target of any jump tables, update the jump tables to branch 10168f0fd8f6SDimitry Andric /// to New instead. 1017f22ef01cSRoman Divacky bool MachineJumpTableInfo::ReplaceMBBInJumpTables(MachineBasicBlock *Old, 1018f22ef01cSRoman Divacky MachineBasicBlock *New) { 1019f22ef01cSRoman Divacky assert(Old != New && "Not making a change?"); 1020f22ef01cSRoman Divacky bool MadeChange = false; 1021f22ef01cSRoman Divacky for (size_t i = 0, e = JumpTables.size(); i != e; ++i) 1022f22ef01cSRoman Divacky ReplaceMBBInJumpTable(i, Old, New); 1023f22ef01cSRoman Divacky return MadeChange; 1024f22ef01cSRoman Divacky } 1025f22ef01cSRoman Divacky 10268f0fd8f6SDimitry Andric /// If Old is a target of the jump tables, update the jump table to branch to 10278f0fd8f6SDimitry Andric /// New instead. 1028f22ef01cSRoman Divacky bool MachineJumpTableInfo::ReplaceMBBInJumpTable(unsigned Idx, 1029f22ef01cSRoman Divacky MachineBasicBlock *Old, 1030f22ef01cSRoman Divacky MachineBasicBlock *New) { 1031f22ef01cSRoman Divacky assert(Old != New && "Not making a change?"); 1032f22ef01cSRoman Divacky bool MadeChange = false; 1033f22ef01cSRoman Divacky MachineJumpTableEntry &JTE = JumpTables[Idx]; 1034f22ef01cSRoman Divacky for (size_t j = 0, e = JTE.MBBs.size(); j != e; ++j) 1035f22ef01cSRoman Divacky if (JTE.MBBs[j] == Old) { 1036f22ef01cSRoman Divacky JTE.MBBs[j] = New; 1037f22ef01cSRoman Divacky MadeChange = true; 1038f22ef01cSRoman Divacky } 1039f22ef01cSRoman Divacky return MadeChange; 1040f22ef01cSRoman Divacky } 1041f22ef01cSRoman Divacky 1042f22ef01cSRoman Divacky void MachineJumpTableInfo::print(raw_ostream &OS) const { 1043f22ef01cSRoman Divacky if (JumpTables.empty()) return; 1044f22ef01cSRoman Divacky 1045f22ef01cSRoman Divacky OS << "Jump Tables:\n"; 1046f22ef01cSRoman Divacky 1047f22ef01cSRoman Divacky for (unsigned i = 0, e = JumpTables.size(); i != e; ++i) { 1048f22ef01cSRoman Divacky OS << " jt#" << i << ": "; 1049f22ef01cSRoman Divacky for (unsigned j = 0, f = JumpTables[i].MBBs.size(); j != f; ++j) 1050f22ef01cSRoman Divacky OS << " BB#" << JumpTables[i].MBBs[j]->getNumber(); 1051f22ef01cSRoman Divacky } 1052f22ef01cSRoman Divacky 1053f22ef01cSRoman Divacky OS << '\n'; 1054f22ef01cSRoman Divacky } 1055f22ef01cSRoman Divacky 10563861d79fSDimitry Andric #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP) 10573ca95b02SDimitry Andric LLVM_DUMP_METHOD void MachineJumpTableInfo::dump() const { print(dbgs()); } 10583861d79fSDimitry Andric #endif 1059f22ef01cSRoman Divacky 1060f22ef01cSRoman Divacky 1061f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 1062f22ef01cSRoman Divacky // MachineConstantPool implementation 1063f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 1064f22ef01cSRoman Divacky 1065dff0c46cSDimitry Andric void MachineConstantPoolValue::anchor() { } 1066dff0c46cSDimitry Andric 10676122f3e6SDimitry Andric Type *MachineConstantPoolEntry::getType() const { 1068f22ef01cSRoman Divacky if (isMachineConstantPoolEntry()) 1069f22ef01cSRoman Divacky return Val.MachineCPVal->getType(); 1070f22ef01cSRoman Divacky return Val.ConstVal->getType(); 1071f22ef01cSRoman Divacky } 1072f22ef01cSRoman Divacky 10737d523365SDimitry Andric bool MachineConstantPoolEntry::needsRelocation() const { 1074f22ef01cSRoman Divacky if (isMachineConstantPoolEntry()) 10757d523365SDimitry Andric return true; 10767d523365SDimitry Andric return Val.ConstVal->needsRelocation(); 1077f22ef01cSRoman Divacky } 1078f22ef01cSRoman Divacky 107991bc56edSDimitry Andric SectionKind 108091bc56edSDimitry Andric MachineConstantPoolEntry::getSectionKind(const DataLayout *DL) const { 10817d523365SDimitry Andric if (needsRelocation()) 10827d523365SDimitry Andric return SectionKind::getReadOnlyWithRel(); 108391bc56edSDimitry Andric switch (DL->getTypeAllocSize(getType())) { 108491bc56edSDimitry Andric case 4: 10857d523365SDimitry Andric return SectionKind::getMergeableConst4(); 108691bc56edSDimitry Andric case 8: 10877d523365SDimitry Andric return SectionKind::getMergeableConst8(); 108891bc56edSDimitry Andric case 16: 10897d523365SDimitry Andric return SectionKind::getMergeableConst16(); 10903ca95b02SDimitry Andric case 32: 10913ca95b02SDimitry Andric return SectionKind::getMergeableConst32(); 109291bc56edSDimitry Andric default: 10937d523365SDimitry Andric return SectionKind::getReadOnly(); 109491bc56edSDimitry Andric } 109591bc56edSDimitry Andric } 109691bc56edSDimitry Andric 1097f22ef01cSRoman Divacky MachineConstantPool::~MachineConstantPool() { 1098d88c1a5aSDimitry Andric // A constant may be a member of both Constants and MachineCPVsSharingEntries, 1099d88c1a5aSDimitry Andric // so keep track of which we've deleted to avoid double deletions. 1100d88c1a5aSDimitry Andric DenseSet<MachineConstantPoolValue*> Deleted; 1101f22ef01cSRoman Divacky for (unsigned i = 0, e = Constants.size(); i != e; ++i) 1102d88c1a5aSDimitry Andric if (Constants[i].isMachineConstantPoolEntry()) { 1103d88c1a5aSDimitry Andric Deleted.insert(Constants[i].Val.MachineCPVal); 1104f22ef01cSRoman Divacky delete Constants[i].Val.MachineCPVal; 1105d88c1a5aSDimitry Andric } 1106dd6029ffSDimitry Andric for (DenseSet<MachineConstantPoolValue*>::iterator I = 1107dd6029ffSDimitry Andric MachineCPVsSharingEntries.begin(), E = MachineCPVsSharingEntries.end(); 1108d88c1a5aSDimitry Andric I != E; ++I) { 1109d88c1a5aSDimitry Andric if (Deleted.count(*I) == 0) 1110dd6029ffSDimitry Andric delete *I; 1111f22ef01cSRoman Divacky } 1112d88c1a5aSDimitry Andric } 1113f22ef01cSRoman Divacky 11148f0fd8f6SDimitry Andric /// Test whether the given two constants can be allocated the same constant pool 11158f0fd8f6SDimitry Andric /// entry. 1116f22ef01cSRoman Divacky static bool CanShareConstantPoolEntry(const Constant *A, const Constant *B, 1117875ed548SDimitry Andric const DataLayout &DL) { 1118f22ef01cSRoman Divacky // Handle the trivial case quickly. 1119f22ef01cSRoman Divacky if (A == B) return true; 1120f22ef01cSRoman Divacky 1121f22ef01cSRoman Divacky // If they have the same type but weren't the same constant, quickly 1122f22ef01cSRoman Divacky // reject them. 1123f22ef01cSRoman Divacky if (A->getType() == B->getType()) return false; 1124f22ef01cSRoman Divacky 1125dff0c46cSDimitry Andric // We can't handle structs or arrays. 1126dff0c46cSDimitry Andric if (isa<StructType>(A->getType()) || isa<ArrayType>(A->getType()) || 1127dff0c46cSDimitry Andric isa<StructType>(B->getType()) || isa<ArrayType>(B->getType())) 1128dff0c46cSDimitry Andric return false; 1129dff0c46cSDimitry Andric 1130f22ef01cSRoman Divacky // For now, only support constants with the same size. 1131875ed548SDimitry Andric uint64_t StoreSize = DL.getTypeStoreSize(A->getType()); 1132875ed548SDimitry Andric if (StoreSize != DL.getTypeStoreSize(B->getType()) || StoreSize > 128) 1133f22ef01cSRoman Divacky return false; 1134f22ef01cSRoman Divacky 1135dff0c46cSDimitry Andric Type *IntTy = IntegerType::get(A->getContext(), StoreSize*8); 1136f22ef01cSRoman Divacky 1137dff0c46cSDimitry Andric // Try constant folding a bitcast of both instructions to an integer. If we 1138dff0c46cSDimitry Andric // get two identical ConstantInt's, then we are good to share them. We use 1139dff0c46cSDimitry Andric // the constant folding APIs to do this so that we get the benefit of 11403861d79fSDimitry Andric // DataLayout. 1141dff0c46cSDimitry Andric if (isa<PointerType>(A->getType())) 11423ca95b02SDimitry Andric A = ConstantFoldCastOperand(Instruction::PtrToInt, 11433ca95b02SDimitry Andric const_cast<Constant *>(A), IntTy, DL); 1144dff0c46cSDimitry Andric else if (A->getType() != IntTy) 11453ca95b02SDimitry Andric A = ConstantFoldCastOperand(Instruction::BitCast, const_cast<Constant *>(A), 11463ca95b02SDimitry Andric IntTy, DL); 1147dff0c46cSDimitry Andric if (isa<PointerType>(B->getType())) 11483ca95b02SDimitry Andric B = ConstantFoldCastOperand(Instruction::PtrToInt, 11493ca95b02SDimitry Andric const_cast<Constant *>(B), IntTy, DL); 1150dff0c46cSDimitry Andric else if (B->getType() != IntTy) 11513ca95b02SDimitry Andric B = ConstantFoldCastOperand(Instruction::BitCast, const_cast<Constant *>(B), 11523ca95b02SDimitry Andric IntTy, DL); 1153f22ef01cSRoman Divacky 1154dff0c46cSDimitry Andric return A == B; 1155f22ef01cSRoman Divacky } 1156f22ef01cSRoman Divacky 11578f0fd8f6SDimitry Andric /// Create a new entry in the constant pool or return an existing one. 11588f0fd8f6SDimitry Andric /// User must specify the log2 of the minimum required alignment for the object. 1159f22ef01cSRoman Divacky unsigned MachineConstantPool::getConstantPoolIndex(const Constant *C, 1160f22ef01cSRoman Divacky unsigned Alignment) { 1161f22ef01cSRoman Divacky assert(Alignment && "Alignment must be specified!"); 1162f22ef01cSRoman Divacky if (Alignment > PoolAlignment) PoolAlignment = Alignment; 1163f22ef01cSRoman Divacky 1164f22ef01cSRoman Divacky // Check to see if we already have this constant. 1165f22ef01cSRoman Divacky // 1166f22ef01cSRoman Divacky // FIXME, this could be made much more efficient for large constant pools. 1167f22ef01cSRoman Divacky for (unsigned i = 0, e = Constants.size(); i != e; ++i) 1168f22ef01cSRoman Divacky if (!Constants[i].isMachineConstantPoolEntry() && 1169875ed548SDimitry Andric CanShareConstantPoolEntry(Constants[i].Val.ConstVal, C, DL)) { 1170f22ef01cSRoman Divacky if ((unsigned)Constants[i].getAlignment() < Alignment) 1171f22ef01cSRoman Divacky Constants[i].Alignment = Alignment; 1172f22ef01cSRoman Divacky return i; 1173f22ef01cSRoman Divacky } 1174f22ef01cSRoman Divacky 1175f22ef01cSRoman Divacky Constants.push_back(MachineConstantPoolEntry(C, Alignment)); 1176f22ef01cSRoman Divacky return Constants.size()-1; 1177f22ef01cSRoman Divacky } 1178f22ef01cSRoman Divacky 1179f22ef01cSRoman Divacky unsigned MachineConstantPool::getConstantPoolIndex(MachineConstantPoolValue *V, 1180f22ef01cSRoman Divacky unsigned Alignment) { 1181f22ef01cSRoman Divacky assert(Alignment && "Alignment must be specified!"); 1182f22ef01cSRoman Divacky if (Alignment > PoolAlignment) PoolAlignment = Alignment; 1183f22ef01cSRoman Divacky 1184f22ef01cSRoman Divacky // Check to see if we already have this constant. 1185f22ef01cSRoman Divacky // 1186f22ef01cSRoman Divacky // FIXME, this could be made much more efficient for large constant pools. 1187f22ef01cSRoman Divacky int Idx = V->getExistingMachineCPValue(this, Alignment); 1188dd6029ffSDimitry Andric if (Idx != -1) { 1189dd6029ffSDimitry Andric MachineCPVsSharingEntries.insert(V); 1190f22ef01cSRoman Divacky return (unsigned)Idx; 1191dd6029ffSDimitry Andric } 1192f22ef01cSRoman Divacky 1193f22ef01cSRoman Divacky Constants.push_back(MachineConstantPoolEntry(V, Alignment)); 1194f22ef01cSRoman Divacky return Constants.size()-1; 1195f22ef01cSRoman Divacky } 1196f22ef01cSRoman Divacky 1197f22ef01cSRoman Divacky void MachineConstantPool::print(raw_ostream &OS) const { 1198f22ef01cSRoman Divacky if (Constants.empty()) return; 1199f22ef01cSRoman Divacky 1200f22ef01cSRoman Divacky OS << "Constant Pool:\n"; 1201f22ef01cSRoman Divacky for (unsigned i = 0, e = Constants.size(); i != e; ++i) { 1202f22ef01cSRoman Divacky OS << " cp#" << i << ": "; 1203f22ef01cSRoman Divacky if (Constants[i].isMachineConstantPoolEntry()) 1204f22ef01cSRoman Divacky Constants[i].Val.MachineCPVal->print(OS); 1205f22ef01cSRoman Divacky else 120691bc56edSDimitry Andric Constants[i].Val.ConstVal->printAsOperand(OS, /*PrintType=*/false); 1207f22ef01cSRoman Divacky OS << ", align=" << Constants[i].getAlignment(); 1208f22ef01cSRoman Divacky OS << "\n"; 1209f22ef01cSRoman Divacky } 1210f22ef01cSRoman Divacky } 1211f22ef01cSRoman Divacky 12123861d79fSDimitry Andric #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP) 12133ca95b02SDimitry Andric LLVM_DUMP_METHOD void MachineConstantPool::dump() const { print(dbgs()); } 12143861d79fSDimitry Andric #endif 1215