1 //===- llvm/CodeGen/MachineFunction.h ---------------------------*- C++ -*-===// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is distributed under the University of Illinois Open Source 6 // License. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 // 10 // Collect native machine code for a function. This class contains a list of 11 // MachineBasicBlock instances that make up the current compiled function. 12 // 13 // This class also contains pointers to various classes which hold 14 // target-specific information about the generated code. 15 // 16 //===----------------------------------------------------------------------===// 17 18 #ifndef LLVM_CODEGEN_MACHINEFUNCTION_H 19 #define LLVM_CODEGEN_MACHINEFUNCTION_H 20 21 #include "llvm/ADT/ArrayRef.h" 22 #include "llvm/ADT/BitVector.h" 23 #include "llvm/ADT/DenseMap.h" 24 #include "llvm/ADT/GraphTraits.h" 25 #include "llvm/ADT/Optional.h" 26 #include "llvm/ADT/SmallVector.h" 27 #include "llvm/ADT/StringRef.h" 28 #include "llvm/ADT/ilist.h" 29 #include "llvm/ADT/iterator.h" 30 #include "llvm/Analysis/EHPersonalities.h" 31 #include "llvm/CodeGen/MachineBasicBlock.h" 32 #include "llvm/CodeGen/MachineInstr.h" 33 #include "llvm/CodeGen/MachineMemOperand.h" 34 #include "llvm/IR/DebugLoc.h" 35 #include "llvm/IR/Instructions.h" 36 #include "llvm/IR/Metadata.h" 37 #include "llvm/MC/MCDwarf.h" 38 #include "llvm/MC/MCSymbol.h" 39 #include "llvm/Support/Allocator.h" 40 #include "llvm/Support/ArrayRecycler.h" 41 #include "llvm/Support/AtomicOrdering.h" 42 #include "llvm/Support/Compiler.h" 43 #include "llvm/Support/ErrorHandling.h" 44 #include "llvm/Support/Recycler.h" 45 #include <cassert> 46 #include <cstdint> 47 #include <memory> 48 #include <utility> 49 #include <vector> 50 51 namespace llvm { 52 53 class BasicBlock; 54 class BlockAddress; 55 class DataLayout; 56 class DIExpression; 57 class DILocalVariable; 58 class DILocation; 59 class Function; 60 class GlobalValue; 61 class MachineConstantPool; 62 class MachineFrameInfo; 63 class MachineFunction; 64 class MachineJumpTableInfo; 65 class MachineModuleInfo; 66 class MachineRegisterInfo; 67 class MCContext; 68 class MCInstrDesc; 69 class Pass; 70 class PseudoSourceValueManager; 71 class raw_ostream; 72 class SlotIndexes; 73 class TargetMachine; 74 class TargetRegisterClass; 75 class TargetSubtargetInfo; 76 struct WasmEHFuncInfo; 77 struct WinEHFuncInfo; 78 79 template <> struct ilist_alloc_traits<MachineBasicBlock> { 80 void deleteNode(MachineBasicBlock *MBB); 81 }; 82 83 template <> struct ilist_callback_traits<MachineBasicBlock> { 84 void addNodeToList(MachineBasicBlock* N); 85 void removeNodeFromList(MachineBasicBlock* N); 86 87 template <class Iterator> 88 void transferNodesFromList(ilist_callback_traits &OldList, Iterator, Iterator) { 89 llvm_unreachable("Never transfer between lists"); 90 } 91 }; 92 93 /// MachineFunctionInfo - This class can be derived from and used by targets to 94 /// hold private target-specific information for each MachineFunction. Objects 95 /// of type are accessed/created with MF::getInfo and destroyed when the 96 /// MachineFunction is destroyed. 97 struct MachineFunctionInfo { 98 virtual ~MachineFunctionInfo(); 99 100 /// Factory function: default behavior is to call new using the 101 /// supplied allocator. 102 /// 103 /// This function can be overridden in a derive class. 104 template<typename Ty> 105 static Ty *create(BumpPtrAllocator &Allocator, MachineFunction &MF) { 106 return new (Allocator.Allocate<Ty>()) Ty(MF); 107 } 108 }; 109 110 /// Properties which a MachineFunction may have at a given point in time. 111 /// Each of these has checking code in the MachineVerifier, and passes can 112 /// require that a property be set. 113 class MachineFunctionProperties { 114 // Possible TODO: Allow targets to extend this (perhaps by allowing the 115 // constructor to specify the size of the bit vector) 116 // Possible TODO: Allow requiring the negative (e.g. VRegsAllocated could be 117 // stated as the negative of "has vregs" 118 119 public: 120 // The properties are stated in "positive" form; i.e. a pass could require 121 // that the property hold, but not that it does not hold. 122 123 // Property descriptions: 124 // IsSSA: True when the machine function is in SSA form and virtual registers 125 // have a single def. 126 // NoPHIs: The machine function does not contain any PHI instruction. 127 // TracksLiveness: True when tracking register liveness accurately. 128 // While this property is set, register liveness information in basic block 129 // live-in lists and machine instruction operands (e.g. kill flags, implicit 130 // defs) is accurate. This means it can be used to change the code in ways 131 // that affect the values in registers, for example by the register 132 // scavenger. 133 // When this property is clear, liveness is no longer reliable. 134 // NoVRegs: The machine function does not use any virtual registers. 135 // Legalized: In GlobalISel: the MachineLegalizer ran and all pre-isel generic 136 // instructions have been legalized; i.e., all instructions are now one of: 137 // - generic and always legal (e.g., COPY) 138 // - target-specific 139 // - legal pre-isel generic instructions. 140 // RegBankSelected: In GlobalISel: the RegBankSelect pass ran and all generic 141 // virtual registers have been assigned to a register bank. 142 // Selected: In GlobalISel: the InstructionSelect pass ran and all pre-isel 143 // generic instructions have been eliminated; i.e., all instructions are now 144 // target-specific or non-pre-isel generic instructions (e.g., COPY). 145 // Since only pre-isel generic instructions can have generic virtual register 146 // operands, this also means that all generic virtual registers have been 147 // constrained to virtual registers (assigned to register classes) and that 148 // all sizes attached to them have been eliminated. 149 enum class Property : unsigned { 150 IsSSA, 151 NoPHIs, 152 TracksLiveness, 153 NoVRegs, 154 FailedISel, 155 Legalized, 156 RegBankSelected, 157 Selected, 158 LastProperty = Selected, 159 }; 160 161 bool hasProperty(Property P) const { 162 return Properties[static_cast<unsigned>(P)]; 163 } 164 165 MachineFunctionProperties &set(Property P) { 166 Properties.set(static_cast<unsigned>(P)); 167 return *this; 168 } 169 170 MachineFunctionProperties &reset(Property P) { 171 Properties.reset(static_cast<unsigned>(P)); 172 return *this; 173 } 174 175 /// Reset all the properties. 176 MachineFunctionProperties &reset() { 177 Properties.reset(); 178 return *this; 179 } 180 181 MachineFunctionProperties &set(const MachineFunctionProperties &MFP) { 182 Properties |= MFP.Properties; 183 return *this; 184 } 185 186 MachineFunctionProperties &reset(const MachineFunctionProperties &MFP) { 187 Properties.reset(MFP.Properties); 188 return *this; 189 } 190 191 // Returns true if all properties set in V (i.e. required by a pass) are set 192 // in this. 193 bool verifyRequiredProperties(const MachineFunctionProperties &V) const { 194 return !V.Properties.test(Properties); 195 } 196 197 /// Print the MachineFunctionProperties in human-readable form. 198 void print(raw_ostream &OS) const; 199 200 private: 201 BitVector Properties = 202 BitVector(static_cast<unsigned>(Property::LastProperty)+1); 203 }; 204 205 struct SEHHandler { 206 /// Filter or finally function. Null indicates a catch-all. 207 const Function *FilterOrFinally; 208 209 /// Address of block to recover at. Null for a finally handler. 210 const BlockAddress *RecoverBA; 211 }; 212 213 /// This structure is used to retain landing pad info for the current function. 214 struct LandingPadInfo { 215 MachineBasicBlock *LandingPadBlock; // Landing pad block. 216 SmallVector<MCSymbol *, 1> BeginLabels; // Labels prior to invoke. 217 SmallVector<MCSymbol *, 1> EndLabels; // Labels after invoke. 218 SmallVector<SEHHandler, 1> SEHHandlers; // SEH handlers active at this lpad. 219 MCSymbol *LandingPadLabel = nullptr; // Label at beginning of landing pad. 220 std::vector<int> TypeIds; // List of type ids (filters negative). 221 222 explicit LandingPadInfo(MachineBasicBlock *MBB) 223 : LandingPadBlock(MBB) {} 224 }; 225 226 class MachineFunction { 227 const Function &F; 228 const TargetMachine &Target; 229 const TargetSubtargetInfo *STI; 230 MCContext &Ctx; 231 MachineModuleInfo &MMI; 232 233 // RegInfo - Information about each register in use in the function. 234 MachineRegisterInfo *RegInfo; 235 236 // Used to keep track of target-specific per-machine function information for 237 // the target implementation. 238 MachineFunctionInfo *MFInfo; 239 240 // Keep track of objects allocated on the stack. 241 MachineFrameInfo *FrameInfo; 242 243 // Keep track of constants which are spilled to memory 244 MachineConstantPool *ConstantPool; 245 246 // Keep track of jump tables for switch instructions 247 MachineJumpTableInfo *JumpTableInfo; 248 249 // Keeps track of Wasm exception handling related data. This will be null for 250 // functions that aren't using a wasm EH personality. 251 WasmEHFuncInfo *WasmEHInfo = nullptr; 252 253 // Keeps track of Windows exception handling related data. This will be null 254 // for functions that aren't using a funclet-based EH personality. 255 WinEHFuncInfo *WinEHInfo = nullptr; 256 257 // Function-level unique numbering for MachineBasicBlocks. When a 258 // MachineBasicBlock is inserted into a MachineFunction is it automatically 259 // numbered and this vector keeps track of the mapping from ID's to MBB's. 260 std::vector<MachineBasicBlock*> MBBNumbering; 261 262 // Pool-allocate MachineFunction-lifetime and IR objects. 263 BumpPtrAllocator Allocator; 264 265 // Allocation management for instructions in function. 266 Recycler<MachineInstr> InstructionRecycler; 267 268 // Allocation management for operand arrays on instructions. 269 ArrayRecycler<MachineOperand> OperandRecycler; 270 271 // Allocation management for basic blocks in function. 272 Recycler<MachineBasicBlock> BasicBlockRecycler; 273 274 // List of machine basic blocks in function 275 using BasicBlockListType = ilist<MachineBasicBlock>; 276 BasicBlockListType BasicBlocks; 277 278 /// FunctionNumber - This provides a unique ID for each function emitted in 279 /// this translation unit. 280 /// 281 unsigned FunctionNumber; 282 283 /// Alignment - The alignment of the function. 284 unsigned Alignment; 285 286 /// ExposesReturnsTwice - True if the function calls setjmp or related 287 /// functions with attribute "returns twice", but doesn't have 288 /// the attribute itself. 289 /// This is used to limit optimizations which cannot reason 290 /// about the control flow of such functions. 291 bool ExposesReturnsTwice = false; 292 293 /// True if the function includes any inline assembly. 294 bool HasInlineAsm = false; 295 296 /// True if any WinCFI instruction have been emitted in this function. 297 Optional<bool> HasWinCFI; 298 299 /// Current high-level properties of the IR of the function (e.g. is in SSA 300 /// form or whether registers have been allocated) 301 MachineFunctionProperties Properties; 302 303 // Allocation management for pseudo source values. 304 std::unique_ptr<PseudoSourceValueManager> PSVManager; 305 306 /// List of moves done by a function's prolog. Used to construct frame maps 307 /// by debug and exception handling consumers. 308 std::vector<MCCFIInstruction> FrameInstructions; 309 310 /// \name Exception Handling 311 /// \{ 312 313 /// List of LandingPadInfo describing the landing pad information. 314 std::vector<LandingPadInfo> LandingPads; 315 316 /// Map a landing pad's EH symbol to the call site indexes. 317 DenseMap<MCSymbol*, SmallVector<unsigned, 4>> LPadToCallSiteMap; 318 319 /// Map of invoke call site index values to associated begin EH_LABEL. 320 DenseMap<MCSymbol*, unsigned> CallSiteMap; 321 322 /// CodeView label annotations. 323 std::vector<std::pair<MCSymbol *, MDNode *>> CodeViewAnnotations; 324 325 bool CallsEHReturn = false; 326 bool CallsUnwindInit = false; 327 bool HasEHScopes = false; 328 bool HasEHFunclets = false; 329 330 /// List of C++ TypeInfo used. 331 std::vector<const GlobalValue *> TypeInfos; 332 333 /// List of typeids encoding filters used. 334 std::vector<unsigned> FilterIds; 335 336 /// List of the indices in FilterIds corresponding to filter terminators. 337 std::vector<unsigned> FilterEnds; 338 339 EHPersonality PersonalityTypeCache = EHPersonality::Unknown; 340 341 /// \} 342 343 /// Clear all the members of this MachineFunction, but the ones used 344 /// to initialize again the MachineFunction. 345 /// More specifically, this deallocates all the dynamically allocated 346 /// objects and get rid of all the XXXInfo data structure, but keep 347 /// unchanged the references to Fn, Target, MMI, and FunctionNumber. 348 void clear(); 349 /// Allocate and initialize the different members. 350 /// In particular, the XXXInfo data structure. 351 /// \pre Fn, Target, MMI, and FunctionNumber are properly set. 352 void init(); 353 354 public: 355 struct VariableDbgInfo { 356 const DILocalVariable *Var; 357 const DIExpression *Expr; 358 // The Slot can be negative for fixed stack objects. 359 int Slot; 360 const DILocation *Loc; 361 362 VariableDbgInfo(const DILocalVariable *Var, const DIExpression *Expr, 363 int Slot, const DILocation *Loc) 364 : Var(Var), Expr(Expr), Slot(Slot), Loc(Loc) {} 365 }; 366 using VariableDbgInfoMapTy = SmallVector<VariableDbgInfo, 4>; 367 VariableDbgInfoMapTy VariableDbgInfos; 368 369 MachineFunction(const Function &F, const TargetMachine &Target, 370 const TargetSubtargetInfo &STI, unsigned FunctionNum, 371 MachineModuleInfo &MMI); 372 MachineFunction(const MachineFunction &) = delete; 373 MachineFunction &operator=(const MachineFunction &) = delete; 374 ~MachineFunction(); 375 376 /// Reset the instance as if it was just created. 377 void reset() { 378 clear(); 379 init(); 380 } 381 382 MachineModuleInfo &getMMI() const { return MMI; } 383 MCContext &getContext() const { return Ctx; } 384 385 PseudoSourceValueManager &getPSVManager() const { return *PSVManager; } 386 387 /// Return the DataLayout attached to the Module associated to this MF. 388 const DataLayout &getDataLayout() const; 389 390 /// Return the LLVM function that this machine code represents 391 const Function &getFunction() const { return F; } 392 393 /// getName - Return the name of the corresponding LLVM function. 394 StringRef getName() const; 395 396 /// getFunctionNumber - Return a unique ID for the current function. 397 unsigned getFunctionNumber() const { return FunctionNumber; } 398 399 /// getTarget - Return the target machine this machine code is compiled with 400 const TargetMachine &getTarget() const { return Target; } 401 402 /// getSubtarget - Return the subtarget for which this machine code is being 403 /// compiled. 404 const TargetSubtargetInfo &getSubtarget() const { return *STI; } 405 void setSubtarget(const TargetSubtargetInfo *ST) { STI = ST; } 406 407 /// getSubtarget - This method returns a pointer to the specified type of 408 /// TargetSubtargetInfo. In debug builds, it verifies that the object being 409 /// returned is of the correct type. 410 template<typename STC> const STC &getSubtarget() const { 411 return *static_cast<const STC *>(STI); 412 } 413 414 /// getRegInfo - Return information about the registers currently in use. 415 MachineRegisterInfo &getRegInfo() { return *RegInfo; } 416 const MachineRegisterInfo &getRegInfo() const { return *RegInfo; } 417 418 /// getFrameInfo - Return the frame info object for the current function. 419 /// This object contains information about objects allocated on the stack 420 /// frame of the current function in an abstract way. 421 MachineFrameInfo &getFrameInfo() { return *FrameInfo; } 422 const MachineFrameInfo &getFrameInfo() const { return *FrameInfo; } 423 424 /// getJumpTableInfo - Return the jump table info object for the current 425 /// function. This object contains information about jump tables in the 426 /// current function. If the current function has no jump tables, this will 427 /// return null. 428 const MachineJumpTableInfo *getJumpTableInfo() const { return JumpTableInfo; } 429 MachineJumpTableInfo *getJumpTableInfo() { return JumpTableInfo; } 430 431 /// getOrCreateJumpTableInfo - Get the JumpTableInfo for this function, if it 432 /// does already exist, allocate one. 433 MachineJumpTableInfo *getOrCreateJumpTableInfo(unsigned JTEntryKind); 434 435 /// getConstantPool - Return the constant pool object for the current 436 /// function. 437 MachineConstantPool *getConstantPool() { return ConstantPool; } 438 const MachineConstantPool *getConstantPool() const { return ConstantPool; } 439 440 /// getWasmEHFuncInfo - Return information about how the current function uses 441 /// Wasm exception handling. Returns null for functions that don't use wasm 442 /// exception handling. 443 const WasmEHFuncInfo *getWasmEHFuncInfo() const { return WasmEHInfo; } 444 WasmEHFuncInfo *getWasmEHFuncInfo() { return WasmEHInfo; } 445 446 /// getWinEHFuncInfo - Return information about how the current function uses 447 /// Windows exception handling. Returns null for functions that don't use 448 /// funclets for exception handling. 449 const WinEHFuncInfo *getWinEHFuncInfo() const { return WinEHInfo; } 450 WinEHFuncInfo *getWinEHFuncInfo() { return WinEHInfo; } 451 452 /// getAlignment - Return the alignment (log2, not bytes) of the function. 453 unsigned getAlignment() const { return Alignment; } 454 455 /// setAlignment - Set the alignment (log2, not bytes) of the function. 456 void setAlignment(unsigned A) { Alignment = A; } 457 458 /// ensureAlignment - Make sure the function is at least 1 << A bytes aligned. 459 void ensureAlignment(unsigned A) { 460 if (Alignment < A) Alignment = A; 461 } 462 463 /// exposesReturnsTwice - Returns true if the function calls setjmp or 464 /// any other similar functions with attribute "returns twice" without 465 /// having the attribute itself. 466 bool exposesReturnsTwice() const { 467 return ExposesReturnsTwice; 468 } 469 470 /// setCallsSetJmp - Set a flag that indicates if there's a call to 471 /// a "returns twice" function. 472 void setExposesReturnsTwice(bool B) { 473 ExposesReturnsTwice = B; 474 } 475 476 /// Returns true if the function contains any inline assembly. 477 bool hasInlineAsm() const { 478 return HasInlineAsm; 479 } 480 481 /// Set a flag that indicates that the function contains inline assembly. 482 void setHasInlineAsm(bool B) { 483 HasInlineAsm = B; 484 } 485 486 bool hasWinCFI() const { 487 assert(HasWinCFI.hasValue() && "HasWinCFI not set yet!"); 488 return *HasWinCFI; 489 } 490 void setHasWinCFI(bool v) { HasWinCFI = v; } 491 492 /// Get the function properties 493 const MachineFunctionProperties &getProperties() const { return Properties; } 494 MachineFunctionProperties &getProperties() { return Properties; } 495 496 /// getInfo - Keep track of various per-function pieces of information for 497 /// backends that would like to do so. 498 /// 499 template<typename Ty> 500 Ty *getInfo() { 501 if (!MFInfo) 502 MFInfo = Ty::template create<Ty>(Allocator, *this); 503 return static_cast<Ty*>(MFInfo); 504 } 505 506 template<typename Ty> 507 const Ty *getInfo() const { 508 return const_cast<MachineFunction*>(this)->getInfo<Ty>(); 509 } 510 511 /// getBlockNumbered - MachineBasicBlocks are automatically numbered when they 512 /// are inserted into the machine function. The block number for a machine 513 /// basic block can be found by using the MBB::getNumber method, this method 514 /// provides the inverse mapping. 515 MachineBasicBlock *getBlockNumbered(unsigned N) const { 516 assert(N < MBBNumbering.size() && "Illegal block number"); 517 assert(MBBNumbering[N] && "Block was removed from the machine function!"); 518 return MBBNumbering[N]; 519 } 520 521 /// Should we be emitting segmented stack stuff for the function 522 bool shouldSplitStack() const; 523 524 /// getNumBlockIDs - Return the number of MBB ID's allocated. 525 unsigned getNumBlockIDs() const { return (unsigned)MBBNumbering.size(); } 526 527 /// RenumberBlocks - This discards all of the MachineBasicBlock numbers and 528 /// recomputes them. This guarantees that the MBB numbers are sequential, 529 /// dense, and match the ordering of the blocks within the function. If a 530 /// specific MachineBasicBlock is specified, only that block and those after 531 /// it are renumbered. 532 void RenumberBlocks(MachineBasicBlock *MBBFrom = nullptr); 533 534 /// print - Print out the MachineFunction in a format suitable for debugging 535 /// to the specified stream. 536 void print(raw_ostream &OS, const SlotIndexes* = nullptr) const; 537 538 /// viewCFG - This function is meant for use from the debugger. You can just 539 /// say 'call F->viewCFG()' and a ghostview window should pop up from the 540 /// program, displaying the CFG of the current function with the code for each 541 /// basic block inside. This depends on there being a 'dot' and 'gv' program 542 /// in your path. 543 void viewCFG() const; 544 545 /// viewCFGOnly - This function is meant for use from the debugger. It works 546 /// just like viewCFG, but it does not include the contents of basic blocks 547 /// into the nodes, just the label. If you are only interested in the CFG 548 /// this can make the graph smaller. 549 /// 550 void viewCFGOnly() const; 551 552 /// dump - Print the current MachineFunction to cerr, useful for debugger use. 553 void dump() const; 554 555 /// Run the current MachineFunction through the machine code verifier, useful 556 /// for debugger use. 557 /// \returns true if no problems were found. 558 bool verify(Pass *p = nullptr, const char *Banner = nullptr, 559 bool AbortOnError = true) const; 560 561 // Provide accessors for the MachineBasicBlock list... 562 using iterator = BasicBlockListType::iterator; 563 using const_iterator = BasicBlockListType::const_iterator; 564 using const_reverse_iterator = BasicBlockListType::const_reverse_iterator; 565 using reverse_iterator = BasicBlockListType::reverse_iterator; 566 567 /// Support for MachineBasicBlock::getNextNode(). 568 static BasicBlockListType MachineFunction::* 569 getSublistAccess(MachineBasicBlock *) { 570 return &MachineFunction::BasicBlocks; 571 } 572 573 /// addLiveIn - Add the specified physical register as a live-in value and 574 /// create a corresponding virtual register for it. 575 unsigned addLiveIn(unsigned PReg, const TargetRegisterClass *RC); 576 577 //===--------------------------------------------------------------------===// 578 // BasicBlock accessor functions. 579 // 580 iterator begin() { return BasicBlocks.begin(); } 581 const_iterator begin() const { return BasicBlocks.begin(); } 582 iterator end () { return BasicBlocks.end(); } 583 const_iterator end () const { return BasicBlocks.end(); } 584 585 reverse_iterator rbegin() { return BasicBlocks.rbegin(); } 586 const_reverse_iterator rbegin() const { return BasicBlocks.rbegin(); } 587 reverse_iterator rend () { return BasicBlocks.rend(); } 588 const_reverse_iterator rend () const { return BasicBlocks.rend(); } 589 590 unsigned size() const { return (unsigned)BasicBlocks.size();} 591 bool empty() const { return BasicBlocks.empty(); } 592 const MachineBasicBlock &front() const { return BasicBlocks.front(); } 593 MachineBasicBlock &front() { return BasicBlocks.front(); } 594 const MachineBasicBlock & back() const { return BasicBlocks.back(); } 595 MachineBasicBlock & back() { return BasicBlocks.back(); } 596 597 void push_back (MachineBasicBlock *MBB) { BasicBlocks.push_back (MBB); } 598 void push_front(MachineBasicBlock *MBB) { BasicBlocks.push_front(MBB); } 599 void insert(iterator MBBI, MachineBasicBlock *MBB) { 600 BasicBlocks.insert(MBBI, MBB); 601 } 602 void splice(iterator InsertPt, iterator MBBI) { 603 BasicBlocks.splice(InsertPt, BasicBlocks, MBBI); 604 } 605 void splice(iterator InsertPt, MachineBasicBlock *MBB) { 606 BasicBlocks.splice(InsertPt, BasicBlocks, MBB); 607 } 608 void splice(iterator InsertPt, iterator MBBI, iterator MBBE) { 609 BasicBlocks.splice(InsertPt, BasicBlocks, MBBI, MBBE); 610 } 611 612 void remove(iterator MBBI) { BasicBlocks.remove(MBBI); } 613 void remove(MachineBasicBlock *MBBI) { BasicBlocks.remove(MBBI); } 614 void erase(iterator MBBI) { BasicBlocks.erase(MBBI); } 615 void erase(MachineBasicBlock *MBBI) { BasicBlocks.erase(MBBI); } 616 617 template <typename Comp> 618 void sort(Comp comp) { 619 BasicBlocks.sort(comp); 620 } 621 622 //===--------------------------------------------------------------------===// 623 // Internal functions used to automatically number MachineBasicBlocks 624 625 /// Adds the MBB to the internal numbering. Returns the unique number 626 /// assigned to the MBB. 627 unsigned addToMBBNumbering(MachineBasicBlock *MBB) { 628 MBBNumbering.push_back(MBB); 629 return (unsigned)MBBNumbering.size()-1; 630 } 631 632 /// removeFromMBBNumbering - Remove the specific machine basic block from our 633 /// tracker, this is only really to be used by the MachineBasicBlock 634 /// implementation. 635 void removeFromMBBNumbering(unsigned N) { 636 assert(N < MBBNumbering.size() && "Illegal basic block #"); 637 MBBNumbering[N] = nullptr; 638 } 639 640 /// CreateMachineInstr - Allocate a new MachineInstr. Use this instead 641 /// of `new MachineInstr'. 642 MachineInstr *CreateMachineInstr(const MCInstrDesc &MCID, const DebugLoc &DL, 643 bool NoImp = false); 644 645 /// Create a new MachineInstr which is a copy of \p Orig, identical in all 646 /// ways except the instruction has no parent, prev, or next. Bundling flags 647 /// are reset. 648 /// 649 /// Note: Clones a single instruction, not whole instruction bundles. 650 /// Does not perform target specific adjustments; consider using 651 /// TargetInstrInfo::duplicate() instead. 652 MachineInstr *CloneMachineInstr(const MachineInstr *Orig); 653 654 /// Clones instruction or the whole instruction bundle \p Orig and insert 655 /// into \p MBB before \p InsertBefore. 656 /// 657 /// Note: Does not perform target specific adjustments; consider using 658 /// TargetInstrInfo::duplicate() intead. 659 MachineInstr &CloneMachineInstrBundle(MachineBasicBlock &MBB, 660 MachineBasicBlock::iterator InsertBefore, const MachineInstr &Orig); 661 662 /// DeleteMachineInstr - Delete the given MachineInstr. 663 void DeleteMachineInstr(MachineInstr *MI); 664 665 /// CreateMachineBasicBlock - Allocate a new MachineBasicBlock. Use this 666 /// instead of `new MachineBasicBlock'. 667 MachineBasicBlock *CreateMachineBasicBlock(const BasicBlock *bb = nullptr); 668 669 /// DeleteMachineBasicBlock - Delete the given MachineBasicBlock. 670 void DeleteMachineBasicBlock(MachineBasicBlock *MBB); 671 672 /// getMachineMemOperand - Allocate a new MachineMemOperand. 673 /// MachineMemOperands are owned by the MachineFunction and need not be 674 /// explicitly deallocated. 675 MachineMemOperand *getMachineMemOperand( 676 MachinePointerInfo PtrInfo, MachineMemOperand::Flags f, uint64_t s, 677 unsigned base_alignment, const AAMDNodes &AAInfo = AAMDNodes(), 678 const MDNode *Ranges = nullptr, 679 SyncScope::ID SSID = SyncScope::System, 680 AtomicOrdering Ordering = AtomicOrdering::NotAtomic, 681 AtomicOrdering FailureOrdering = AtomicOrdering::NotAtomic); 682 683 /// getMachineMemOperand - Allocate a new MachineMemOperand by copying 684 /// an existing one, adjusting by an offset and using the given size. 685 /// MachineMemOperands are owned by the MachineFunction and need not be 686 /// explicitly deallocated. 687 MachineMemOperand *getMachineMemOperand(const MachineMemOperand *MMO, 688 int64_t Offset, uint64_t Size); 689 690 /// Allocate a new MachineMemOperand by copying an existing one, 691 /// replacing only AliasAnalysis information. MachineMemOperands are owned 692 /// by the MachineFunction and need not be explicitly deallocated. 693 MachineMemOperand *getMachineMemOperand(const MachineMemOperand *MMO, 694 const AAMDNodes &AAInfo); 695 696 using OperandCapacity = ArrayRecycler<MachineOperand>::Capacity; 697 698 /// Allocate an array of MachineOperands. This is only intended for use by 699 /// internal MachineInstr functions. 700 MachineOperand *allocateOperandArray(OperandCapacity Cap) { 701 return OperandRecycler.allocate(Cap, Allocator); 702 } 703 704 /// Dellocate an array of MachineOperands and recycle the memory. This is 705 /// only intended for use by internal MachineInstr functions. 706 /// Cap must be the same capacity that was used to allocate the array. 707 void deallocateOperandArray(OperandCapacity Cap, MachineOperand *Array) { 708 OperandRecycler.deallocate(Cap, Array); 709 } 710 711 /// Allocate and initialize a register mask with @p NumRegister bits. 712 uint32_t *allocateRegMask(); 713 714 /// allocateMemRefsArray - Allocate an array to hold MachineMemOperand 715 /// pointers. This array is owned by the MachineFunction. 716 MachineInstr::mmo_iterator allocateMemRefsArray(unsigned long Num); 717 718 /// extractLoadMemRefs - Allocate an array and populate it with just the 719 /// load information from the given MachineMemOperand sequence. 720 std::pair<MachineInstr::mmo_iterator, 721 MachineInstr::mmo_iterator> 722 extractLoadMemRefs(MachineInstr::mmo_iterator Begin, 723 MachineInstr::mmo_iterator End); 724 725 /// extractStoreMemRefs - Allocate an array and populate it with just the 726 /// store information from the given MachineMemOperand sequence. 727 std::pair<MachineInstr::mmo_iterator, 728 MachineInstr::mmo_iterator> 729 extractStoreMemRefs(MachineInstr::mmo_iterator Begin, 730 MachineInstr::mmo_iterator End); 731 732 /// Allocate a string and populate it with the given external symbol name. 733 const char *createExternalSymbolName(StringRef Name); 734 735 //===--------------------------------------------------------------------===// 736 // Label Manipulation. 737 738 /// getJTISymbol - Return the MCSymbol for the specified non-empty jump table. 739 /// If isLinkerPrivate is specified, an 'l' label is returned, otherwise a 740 /// normal 'L' label is returned. 741 MCSymbol *getJTISymbol(unsigned JTI, MCContext &Ctx, 742 bool isLinkerPrivate = false) const; 743 744 /// getPICBaseSymbol - Return a function-local symbol to represent the PIC 745 /// base. 746 MCSymbol *getPICBaseSymbol() const; 747 748 /// Returns a reference to a list of cfi instructions in the function's 749 /// prologue. Used to construct frame maps for debug and exception handling 750 /// comsumers. 751 const std::vector<MCCFIInstruction> &getFrameInstructions() const { 752 return FrameInstructions; 753 } 754 755 LLVM_NODISCARD unsigned addFrameInst(const MCCFIInstruction &Inst) { 756 FrameInstructions.push_back(Inst); 757 return FrameInstructions.size() - 1; 758 } 759 760 /// \name Exception Handling 761 /// \{ 762 763 bool callsEHReturn() const { return CallsEHReturn; } 764 void setCallsEHReturn(bool b) { CallsEHReturn = b; } 765 766 bool callsUnwindInit() const { return CallsUnwindInit; } 767 void setCallsUnwindInit(bool b) { CallsUnwindInit = b; } 768 769 bool hasEHScopes() const { return HasEHScopes; } 770 void setHasEHScopes(bool V) { HasEHScopes = V; } 771 772 bool hasEHFunclets() const { return HasEHFunclets; } 773 void setHasEHFunclets(bool V) { HasEHFunclets = V; } 774 775 /// Find or create an LandingPadInfo for the specified MachineBasicBlock. 776 LandingPadInfo &getOrCreateLandingPadInfo(MachineBasicBlock *LandingPad); 777 778 /// Remap landing pad labels and remove any deleted landing pads. 779 void tidyLandingPads(DenseMap<MCSymbol*, uintptr_t> *LPMap = nullptr); 780 781 /// Return a reference to the landing pad info for the current function. 782 const std::vector<LandingPadInfo> &getLandingPads() const { 783 return LandingPads; 784 } 785 786 /// Provide the begin and end labels of an invoke style call and associate it 787 /// with a try landing pad block. 788 void addInvoke(MachineBasicBlock *LandingPad, 789 MCSymbol *BeginLabel, MCSymbol *EndLabel); 790 791 /// Add a new panding pad. Returns the label ID for the landing pad entry. 792 MCSymbol *addLandingPad(MachineBasicBlock *LandingPad); 793 794 /// Provide the catch typeinfo for a landing pad. 795 void addCatchTypeInfo(MachineBasicBlock *LandingPad, 796 ArrayRef<const GlobalValue *> TyInfo); 797 798 /// Provide the filter typeinfo for a landing pad. 799 void addFilterTypeInfo(MachineBasicBlock *LandingPad, 800 ArrayRef<const GlobalValue *> TyInfo); 801 802 /// Add a cleanup action for a landing pad. 803 void addCleanup(MachineBasicBlock *LandingPad); 804 805 void addSEHCatchHandler(MachineBasicBlock *LandingPad, const Function *Filter, 806 const BlockAddress *RecoverBA); 807 808 void addSEHCleanupHandler(MachineBasicBlock *LandingPad, 809 const Function *Cleanup); 810 811 /// Return the type id for the specified typeinfo. This is function wide. 812 unsigned getTypeIDFor(const GlobalValue *TI); 813 814 /// Return the id of the filter encoded by TyIds. This is function wide. 815 int getFilterIDFor(std::vector<unsigned> &TyIds); 816 817 /// Map the landing pad's EH symbol to the call site indexes. 818 void setCallSiteLandingPad(MCSymbol *Sym, ArrayRef<unsigned> Sites); 819 820 /// Get the call site indexes for a landing pad EH symbol. 821 SmallVectorImpl<unsigned> &getCallSiteLandingPad(MCSymbol *Sym) { 822 assert(hasCallSiteLandingPad(Sym) && 823 "missing call site number for landing pad!"); 824 return LPadToCallSiteMap[Sym]; 825 } 826 827 /// Return true if the landing pad Eh symbol has an associated call site. 828 bool hasCallSiteLandingPad(MCSymbol *Sym) { 829 return !LPadToCallSiteMap[Sym].empty(); 830 } 831 832 /// Map the begin label for a call site. 833 void setCallSiteBeginLabel(MCSymbol *BeginLabel, unsigned Site) { 834 CallSiteMap[BeginLabel] = Site; 835 } 836 837 /// Get the call site number for a begin label. 838 unsigned getCallSiteBeginLabel(MCSymbol *BeginLabel) const { 839 assert(hasCallSiteBeginLabel(BeginLabel) && 840 "Missing call site number for EH_LABEL!"); 841 return CallSiteMap.lookup(BeginLabel); 842 } 843 844 /// Return true if the begin label has a call site number associated with it. 845 bool hasCallSiteBeginLabel(MCSymbol *BeginLabel) const { 846 return CallSiteMap.count(BeginLabel); 847 } 848 849 /// Record annotations associated with a particular label. 850 void addCodeViewAnnotation(MCSymbol *Label, MDNode *MD) { 851 CodeViewAnnotations.push_back({Label, MD}); 852 } 853 854 ArrayRef<std::pair<MCSymbol *, MDNode *>> getCodeViewAnnotations() const { 855 return CodeViewAnnotations; 856 } 857 858 /// Return a reference to the C++ typeinfo for the current function. 859 const std::vector<const GlobalValue *> &getTypeInfos() const { 860 return TypeInfos; 861 } 862 863 /// Return a reference to the typeids encoding filters used in the current 864 /// function. 865 const std::vector<unsigned> &getFilterIds() const { 866 return FilterIds; 867 } 868 869 /// \} 870 871 /// Collect information used to emit debugging information of a variable. 872 void setVariableDbgInfo(const DILocalVariable *Var, const DIExpression *Expr, 873 int Slot, const DILocation *Loc) { 874 VariableDbgInfos.emplace_back(Var, Expr, Slot, Loc); 875 } 876 877 VariableDbgInfoMapTy &getVariableDbgInfo() { return VariableDbgInfos; } 878 const VariableDbgInfoMapTy &getVariableDbgInfo() const { 879 return VariableDbgInfos; 880 } 881 }; 882 883 /// \name Exception Handling 884 /// \{ 885 886 /// Extract the exception handling information from the landingpad instruction 887 /// and add them to the specified machine module info. 888 void addLandingPadInfo(const LandingPadInst &I, MachineBasicBlock &MBB); 889 890 /// \} 891 892 //===--------------------------------------------------------------------===// 893 // GraphTraits specializations for function basic block graphs (CFGs) 894 //===--------------------------------------------------------------------===// 895 896 // Provide specializations of GraphTraits to be able to treat a 897 // machine function as a graph of machine basic blocks... these are 898 // the same as the machine basic block iterators, except that the root 899 // node is implicitly the first node of the function. 900 // 901 template <> struct GraphTraits<MachineFunction*> : 902 public GraphTraits<MachineBasicBlock*> { 903 static NodeRef getEntryNode(MachineFunction *F) { return &F->front(); } 904 905 // nodes_iterator/begin/end - Allow iteration over all nodes in the graph 906 using nodes_iterator = pointer_iterator<MachineFunction::iterator>; 907 908 static nodes_iterator nodes_begin(MachineFunction *F) { 909 return nodes_iterator(F->begin()); 910 } 911 912 static nodes_iterator nodes_end(MachineFunction *F) { 913 return nodes_iterator(F->end()); 914 } 915 916 static unsigned size (MachineFunction *F) { return F->size(); } 917 }; 918 template <> struct GraphTraits<const MachineFunction*> : 919 public GraphTraits<const MachineBasicBlock*> { 920 static NodeRef getEntryNode(const MachineFunction *F) { return &F->front(); } 921 922 // nodes_iterator/begin/end - Allow iteration over all nodes in the graph 923 using nodes_iterator = pointer_iterator<MachineFunction::const_iterator>; 924 925 static nodes_iterator nodes_begin(const MachineFunction *F) { 926 return nodes_iterator(F->begin()); 927 } 928 929 static nodes_iterator nodes_end (const MachineFunction *F) { 930 return nodes_iterator(F->end()); 931 } 932 933 static unsigned size (const MachineFunction *F) { 934 return F->size(); 935 } 936 }; 937 938 // Provide specializations of GraphTraits to be able to treat a function as a 939 // graph of basic blocks... and to walk it in inverse order. Inverse order for 940 // a function is considered to be when traversing the predecessor edges of a BB 941 // instead of the successor edges. 942 // 943 template <> struct GraphTraits<Inverse<MachineFunction*>> : 944 public GraphTraits<Inverse<MachineBasicBlock*>> { 945 static NodeRef getEntryNode(Inverse<MachineFunction *> G) { 946 return &G.Graph->front(); 947 } 948 }; 949 template <> struct GraphTraits<Inverse<const MachineFunction*>> : 950 public GraphTraits<Inverse<const MachineBasicBlock*>> { 951 static NodeRef getEntryNode(Inverse<const MachineFunction *> G) { 952 return &G.Graph->front(); 953 } 954 }; 955 956 } // end namespace llvm 957 958 #endif // LLVM_CODEGEN_MACHINEFUNCTION_H 959