blob: e8a4d529faac6b84d94fd232c211d4618a8df517 [file] [log] [blame]
Andrew Scull5e1ddfa2018-08-14 10:06:54 +01001//===- 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
51namespace llvm {
52
53class BasicBlock;
54class BlockAddress;
55class DataLayout;
56class DIExpression;
57class DILocalVariable;
58class DILocation;
59class Function;
60class GlobalValue;
61class MachineConstantPool;
62class MachineFrameInfo;
63class MachineFunction;
64class MachineJumpTableInfo;
65class MachineModuleInfo;
66class MachineRegisterInfo;
67class MCContext;
68class MCInstrDesc;
69class Pass;
70class PseudoSourceValueManager;
71class raw_ostream;
72class SlotIndexes;
73class TargetMachine;
74class TargetRegisterClass;
75class TargetSubtargetInfo;
Andrew Scullcdfcccc2018-10-05 20:58:37 +010076struct WasmEHFuncInfo;
Andrew Scull5e1ddfa2018-08-14 10:06:54 +010077struct WinEHFuncInfo;
78
79template <> struct ilist_alloc_traits<MachineBasicBlock> {
80 void deleteNode(MachineBasicBlock *MBB);
81};
82
83template <> struct ilist_callback_traits<MachineBasicBlock> {
Andrew Scullcdfcccc2018-10-05 20:58:37 +010084 void addNodeToList(MachineBasicBlock* N);
85 void removeNodeFromList(MachineBasicBlock* N);
Andrew Scull5e1ddfa2018-08-14 10:06:54 +010086
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.
97struct MachineFunctionInfo {
98 virtual ~MachineFunctionInfo();
99
Andrew Scullcdfcccc2018-10-05 20:58:37 +0100100 /// Factory function: default behavior is to call new using the
Andrew Scull5e1ddfa2018-08-14 10:06:54 +0100101 /// 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.
113class 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
119public:
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
200private:
201 BitVector Properties =
202 BitVector(static_cast<unsigned>(Property::LastProperty)+1);
203};
204
205struct 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.
214struct 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
226class 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
Andrew Scullcdfcccc2018-10-05 20:58:37 +0100249 // 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
Andrew Scull5e1ddfa2018-08-14 10:06:54 +0100253 // 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;
Andrew Scullcdfcccc2018-10-05 20:58:37 +0100327 bool HasEHScopes = false;
Andrew Scull5e1ddfa2018-08-14 10:06:54 +0100328 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
354public:
355 struct VariableDbgInfo {
356 const DILocalVariable *Var;
357 const DIExpression *Expr;
Andrew Scullcdfcccc2018-10-05 20:58:37 +0100358 // The Slot can be negative for fixed stack objects.
359 int Slot;
Andrew Scull5e1ddfa2018-08-14 10:06:54 +0100360 const DILocation *Loc;
361
362 VariableDbgInfo(const DILocalVariable *Var, const DIExpression *Expr,
Andrew Scullcdfcccc2018-10-05 20:58:37 +0100363 int Slot, const DILocation *Loc)
Andrew Scull5e1ddfa2018-08-14 10:06:54 +0100364 : Var(Var), Expr(Expr), Slot(Slot), Loc(Loc) {}
365 };
366 using VariableDbgInfoMapTy = SmallVector<VariableDbgInfo, 4>;
367 VariableDbgInfoMapTy VariableDbgInfos;
368
Andrew Scullcdfcccc2018-10-05 20:58:37 +0100369 MachineFunction(const Function &F, const TargetMachine &Target,
Andrew Scull5e1ddfa2018-08-14 10:06:54 +0100370 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
Andrew Scullcdfcccc2018-10-05 20:58:37 +0100440 /// 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
Andrew Scull5e1ddfa2018-08-14 10:06:54 +0100446 /// 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
Andrew Scullcdfcccc2018-10-05 20:58:37 +0100625 /// Adds the MBB to the internal numbering. Returns the unique number
Andrew Scull5e1ddfa2018-08-14 10:06:54 +0100626 /// 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
Andrew Scullcdfcccc2018-10-05 20:58:37 +0100711 /// Allocate and initialize a register mask with @p NumRegister bits.
712 uint32_t *allocateRegMask();
Andrew Scull5e1ddfa2018-08-14 10:06:54 +0100713
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
Andrew Scullcdfcccc2018-10-05 20:58:37 +0100769 bool hasEHScopes() const { return HasEHScopes; }
770 void setHasEHScopes(bool V) { HasEHScopes = V; }
771
Andrew Scull5e1ddfa2018-08-14 10:06:54 +0100772 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,
Andrew Scullcdfcccc2018-10-05 20:58:37 +0100806 const BlockAddress *RecoverBA);
Andrew Scull5e1ddfa2018-08-14 10:06:54 +0100807
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,
Andrew Scullcdfcccc2018-10-05 20:58:37 +0100873 int Slot, const DILocation *Loc) {
Andrew Scull5e1ddfa2018-08-14 10:06:54 +0100874 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.
888void 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//
901template <> 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};
918template <> 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//
943template <> struct GraphTraits<Inverse<MachineFunction*>> :
944 public GraphTraits<Inverse<MachineBasicBlock*>> {
945 static NodeRef getEntryNode(Inverse<MachineFunction *> G) {
946 return &G.Graph->front();
947 }
948};
949template <> 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