1 //===- llvm/CodeGen/AsmPrinter.h - AsmPrinter Framework ---------*- C++ -*-===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 //
9 // This file contains a class to be used as the base class for target specific
10 // asm writers.  This class primarily handles common functionality used by
11 // all asm writers.
12 //
13 //===----------------------------------------------------------------------===//
14 
15 #ifndef LLVM_CODEGEN_ASMPRINTER_H
16 #define LLVM_CODEGEN_ASMPRINTER_H
17 
18 #include "llvm/ADT/DenseMap.h"
19 #include "llvm/ADT/MapVector.h"
20 #include "llvm/ADT/SmallVector.h"
21 #include "llvm/BinaryFormat/Dwarf.h"
22 #include "llvm/CodeGen/AsmPrinterHandler.h"
23 #include "llvm/CodeGen/DwarfStringPoolEntry.h"
24 #include "llvm/CodeGen/MachineFunctionPass.h"
25 #include "llvm/CodeGen/StackMaps.h"
26 #include "llvm/DebugInfo/CodeView/CodeView.h"
27 #include "llvm/IR/InlineAsm.h"
28 #include "llvm/Support/ErrorHandling.h"
29 #include <cstdint>
30 #include <memory>
31 #include <utility>
32 #include <vector>
33 
34 namespace llvm {
35 
36 class AddrLabelMap;
37 class BasicBlock;
38 class BlockAddress;
39 class Constant;
40 class ConstantArray;
41 class DataLayout;
42 class DIE;
43 class DIEAbbrev;
44 class DwarfDebug;
45 class GCMetadataPrinter;
46 class GCStrategy;
47 class GlobalAlias;
48 class GlobalObject;
49 class GlobalValue;
50 class GlobalVariable;
51 class MachineBasicBlock;
52 class MachineConstantPoolValue;
53 class MachineDominatorTree;
54 class MachineFunction;
55 class MachineInstr;
56 class MachineJumpTableInfo;
57 class MachineLoopInfo;
58 class MachineModuleInfo;
59 class MachineOptimizationRemarkEmitter;
60 class MCAsmInfo;
61 class MCCFIInstruction;
62 class MCContext;
63 class MCExpr;
64 class MCInst;
65 class MCSection;
66 class MCStreamer;
67 class MCSubtargetInfo;
68 class MCSymbol;
69 class MCTargetOptions;
70 class MDNode;
71 class Module;
72 class PseudoProbeHandler;
73 class raw_ostream;
74 class StringRef;
75 class TargetLoweringObjectFile;
76 class TargetMachine;
77 class Twine;
78 
79 namespace remarks {
80 class RemarkStreamer;
81 }
82 
83 /// This class is intended to be used as a driving class for all asm writers.
84 class AsmPrinter : public MachineFunctionPass {
85 public:
86   /// Target machine description.
87   TargetMachine &TM;
88 
89   /// Target Asm Printer information.
90   const MCAsmInfo *MAI = nullptr;
91 
92   /// This is the context for the output file that we are streaming. This owns
93   /// all of the global MC-related objects for the generated translation unit.
94   MCContext &OutContext;
95 
96   /// This is the MCStreamer object for the file we are generating. This
97   /// contains the transient state for the current translation unit that we are
98   /// generating (such as the current section etc).
99   std::unique_ptr<MCStreamer> OutStreamer;
100 
101   /// The current machine function.
102   MachineFunction *MF = nullptr;
103 
104   /// This is a pointer to the current MachineModuleInfo.
105   MachineModuleInfo *MMI = nullptr;
106 
107   /// This is a pointer to the current MachineDominatorTree.
108   MachineDominatorTree *MDT = nullptr;
109 
110   /// This is a pointer to the current MachineLoopInfo.
111   MachineLoopInfo *MLI = nullptr;
112 
113   /// Optimization remark emitter.
114   MachineOptimizationRemarkEmitter *ORE = nullptr;
115 
116   /// The symbol for the entry in __patchable_function_entires.
117   MCSymbol *CurrentPatchableFunctionEntrySym = nullptr;
118 
119   /// The symbol for the current function. This is recalculated at the beginning
120   /// of each call to runOnMachineFunction().
121   MCSymbol *CurrentFnSym = nullptr;
122 
123   /// The symbol for the current function descriptor on AIX. This is created
124   /// at the beginning of each call to SetupMachineFunction().
125   MCSymbol *CurrentFnDescSym = nullptr;
126 
127   /// The symbol used to represent the start of the current function for the
128   /// purpose of calculating its size (e.g. using the .size directive). By
129   /// default, this is equal to CurrentFnSym.
130   MCSymbol *CurrentFnSymForSize = nullptr;
131 
132   /// Map a basic block section ID to the begin and end symbols of that section
133   ///  which determine the section's range.
134   struct MBBSectionRange {
135     MCSymbol *BeginLabel, *EndLabel;
136   };
137 
138   MapVector<unsigned, MBBSectionRange> MBBSectionRanges;
139 
140   /// Map global GOT equivalent MCSymbols to GlobalVariables and keep track of
141   /// its number of uses by other globals.
142   using GOTEquivUsePair = std::pair<const GlobalVariable *, unsigned>;
143   MapVector<const MCSymbol *, GOTEquivUsePair> GlobalGOTEquivs;
144 
145   /// struct HandlerInfo and Handlers permit users or target extended
146   /// AsmPrinter to add their own handlers.
147   struct HandlerInfo {
148     std::unique_ptr<AsmPrinterHandler> Handler;
149     StringRef TimerName;
150     StringRef TimerDescription;
151     StringRef TimerGroupName;
152     StringRef TimerGroupDescription;
153 
HandlerInfoHandlerInfo154     HandlerInfo(std::unique_ptr<AsmPrinterHandler> Handler, StringRef TimerName,
155                 StringRef TimerDescription, StringRef TimerGroupName,
156                 StringRef TimerGroupDescription)
157         : Handler(std::move(Handler)), TimerName(TimerName),
158           TimerDescription(TimerDescription), TimerGroupName(TimerGroupName),
159           TimerGroupDescription(TimerGroupDescription) {}
160   };
161 
162   // Flags representing which CFI section is required for a function/module.
163   enum class CFISection : unsigned {
164     None = 0, ///< Do not emit either .eh_frame or .debug_frame
165     EH = 1,   ///< Emit .eh_frame
166     Debug = 2 ///< Emit .debug_frame
167   };
168 
169 private:
170   MCSymbol *CurrentFnEnd = nullptr;
171 
172   /// Map a basic block section ID to the exception symbol associated with that
173   /// section. Map entries are assigned and looked up via
174   /// AsmPrinter::getMBBExceptionSym.
175   DenseMap<unsigned, MCSymbol *> MBBSectionExceptionSyms;
176 
177   // The symbol used to represent the start of the current BB section of the
178   // function. This is used to calculate the size of the BB section.
179   MCSymbol *CurrentSectionBeginSym = nullptr;
180 
181   /// This map keeps track of which symbol is being used for the specified basic
182   /// block's address of label.
183   std::unique_ptr<AddrLabelMap> AddrLabelSymbols;
184 
185   /// The garbage collection metadata printer table.
186   DenseMap<GCStrategy *, std::unique_ptr<GCMetadataPrinter>> GCMetadataPrinters;
187 
188   /// Emit comments in assembly output if this is true.
189   bool VerboseAsm;
190 
191   /// Output stream for the stack usage file (i.e., .su file).
192   std::unique_ptr<raw_fd_ostream> StackUsageStream;
193 
194   /// List of symbols to be inserted into PC sections.
195   DenseMap<const MDNode *, SmallVector<const MCSymbol *>> PCSectionsSymbols;
196 
197   static char ID;
198 
199 protected:
200   MCSymbol *CurrentFnBegin = nullptr;
201 
202   /// For dso_local functions, the current $local alias for the function.
203   MCSymbol *CurrentFnBeginLocal = nullptr;
204 
205   /// A vector of all debug/EH info emitters we should use. This vector
206   /// maintains ownership of the emitters.
207   std::vector<HandlerInfo> Handlers;
208   size_t NumUserHandlers = 0;
209 
210   StackMaps SM;
211 
212 private:
213   /// If generated on the fly this own the instance.
214   std::unique_ptr<MachineDominatorTree> OwnedMDT;
215 
216   /// If generated on the fly this own the instance.
217   std::unique_ptr<MachineLoopInfo> OwnedMLI;
218 
219   /// If the target supports dwarf debug info, this pointer is non-null.
220   DwarfDebug *DD = nullptr;
221 
222   /// A handler that supports pseudo probe emission with embedded inline
223   /// context.
224   PseudoProbeHandler *PP = nullptr;
225 
226   /// CFISection type the module needs i.e. either .eh_frame or .debug_frame.
227   CFISection ModuleCFISection = CFISection::None;
228 
229   /// True if the module contains split-stack functions. This is used to
230   /// emit .note.GNU-split-stack section as required by the linker for
231   /// special handling split-stack function calling no-split-stack function.
232   bool HasSplitStack = false;
233 
234   /// True if the module contains no-split-stack functions. This is used to emit
235   /// .note.GNU-no-split-stack section when it also contains functions without a
236   /// split stack prologue.
237   bool HasNoSplitStack = false;
238 
239 protected:
240   explicit AsmPrinter(TargetMachine &TM, std::unique_ptr<MCStreamer> Streamer);
241 
242 public:
243   ~AsmPrinter() override;
244 
getDwarfDebug()245   DwarfDebug *getDwarfDebug() { return DD; }
getDwarfDebug()246   DwarfDebug *getDwarfDebug() const { return DD; }
247 
248   uint16_t getDwarfVersion() const;
249   void setDwarfVersion(uint16_t Version);
250 
251   bool isDwarf64() const;
252 
253   /// Returns 4 for DWARF32 and 8 for DWARF64.
254   unsigned int getDwarfOffsetByteSize() const;
255 
256   /// Returns 4 for DWARF32 and 12 for DWARF64.
257   unsigned int getUnitLengthFieldByteSize() const;
258 
259   /// Returns information about the byte size of DW_FORM values.
260   dwarf::FormParams getDwarfFormParams() const;
261 
262   bool isPositionIndependent() const;
263 
264   /// Return true if assembly output should contain comments.
isVerbose()265   bool isVerbose() const { return VerboseAsm; }
266 
267   /// Return a unique ID for the current function.
268   unsigned getFunctionNumber() const;
269 
270   /// Return symbol for the function pseudo stack if the stack frame is not a
271   /// register based.
getFunctionFrameSymbol()272   virtual const MCSymbol *getFunctionFrameSymbol() const { return nullptr; }
273 
getFunctionBegin()274   MCSymbol *getFunctionBegin() const { return CurrentFnBegin; }
getFunctionEnd()275   MCSymbol *getFunctionEnd() const { return CurrentFnEnd; }
276 
277   // Return the exception symbol associated with the MBB section containing a
278   // given basic block.
279   MCSymbol *getMBBExceptionSym(const MachineBasicBlock &MBB);
280 
281   /// Return the symbol to be used for the specified basic block when its
282   /// address is taken.  This cannot be its normal LBB label because the block
283   /// may be accessed outside its containing function.
getAddrLabelSymbol(const BasicBlock * BB)284   MCSymbol *getAddrLabelSymbol(const BasicBlock *BB) {
285     return getAddrLabelSymbolToEmit(BB).front();
286   }
287 
288   /// Return the symbol to be used for the specified basic block when its
289   /// address is taken.  If other blocks were RAUW'd to this one, we may have
290   /// to emit them as well, return the whole set.
291   ArrayRef<MCSymbol *> getAddrLabelSymbolToEmit(const BasicBlock *BB);
292 
293   /// If the specified function has had any references to address-taken blocks
294   /// generated, but the block got deleted, return the symbol now so we can
295   /// emit it.  This prevents emitting a reference to a symbol that has no
296   /// definition.
297   void takeDeletedSymbolsForFunction(const Function *F,
298                                      std::vector<MCSymbol *> &Result);
299 
300   /// Return information about object file lowering.
301   const TargetLoweringObjectFile &getObjFileLowering() const;
302 
303   /// Return information about data layout.
304   const DataLayout &getDataLayout() const;
305 
306   /// Return the pointer size from the TargetMachine
307   unsigned getPointerSize() const;
308 
309   /// Return information about subtarget.
310   const MCSubtargetInfo &getSubtargetInfo() const;
311 
312   void EmitToStreamer(MCStreamer &S, const MCInst &Inst);
313 
314   /// Emits inital debug location directive.
315   void emitInitialRawDwarfLocDirective(const MachineFunction &MF);
316 
317   /// Return the current section we are emitting to.
318   const MCSection *getCurrentSection() const;
319 
320   void getNameWithPrefix(SmallVectorImpl<char> &Name,
321                          const GlobalValue *GV) const;
322 
323   MCSymbol *getSymbol(const GlobalValue *GV) const;
324 
325   /// Similar to getSymbol() but preferred for references. On ELF, this uses a
326   /// local symbol if a reference to GV is guaranteed to be resolved to the
327   /// definition in the same module.
328   MCSymbol *getSymbolPreferLocal(const GlobalValue &GV) const;
329 
doesDwarfUseRelocationsAcrossSections()330   bool doesDwarfUseRelocationsAcrossSections() const {
331     return DwarfUsesRelocationsAcrossSections;
332   }
333 
setDwarfUsesRelocationsAcrossSections(bool Enable)334   void setDwarfUsesRelocationsAcrossSections(bool Enable) {
335     DwarfUsesRelocationsAcrossSections = Enable;
336   }
337 
338   //===------------------------------------------------------------------===//
339   // XRay instrumentation implementation.
340   //===------------------------------------------------------------------===//
341 public:
342   // This describes the kind of sled we're storing in the XRay table.
343   enum class SledKind : uint8_t {
344     FUNCTION_ENTER = 0,
345     FUNCTION_EXIT = 1,
346     TAIL_CALL = 2,
347     LOG_ARGS_ENTER = 3,
348     CUSTOM_EVENT = 4,
349     TYPED_EVENT = 5,
350   };
351 
352   // The table will contain these structs that point to the sled, the function
353   // containing the sled, and what kind of sled (and whether they should always
354   // be instrumented). We also use a version identifier that the runtime can use
355   // to decide what to do with the sled, depending on the version of the sled.
356   struct XRayFunctionEntry {
357     const MCSymbol *Sled;
358     const MCSymbol *Function;
359     SledKind Kind;
360     bool AlwaysInstrument;
361     const class Function *Fn;
362     uint8_t Version;
363 
364     void emit(int, MCStreamer *) const;
365   };
366 
367   // All the sleds to be emitted.
368   SmallVector<XRayFunctionEntry, 4> Sleds;
369 
370   // Helper function to record a given XRay sled.
371   void recordSled(MCSymbol *Sled, const MachineInstr &MI, SledKind Kind,
372                   uint8_t Version = 0);
373 
374   /// Emit a table with all XRay instrumentation points.
375   void emitXRayTable();
376 
377   void emitPatchableFunctionEntries();
378 
379   //===------------------------------------------------------------------===//
380   // MachineFunctionPass Implementation.
381   //===------------------------------------------------------------------===//
382 
383   /// Record analysis usage.
384   void getAnalysisUsage(AnalysisUsage &AU) const override;
385 
386   /// Set up the AsmPrinter when we are working on a new module. If your pass
387   /// overrides this, it must make sure to explicitly call this implementation.
388   bool doInitialization(Module &M) override;
389 
390   /// Shut down the asmprinter. If you override this in your pass, you must make
391   /// sure to call it explicitly.
392   bool doFinalization(Module &M) override;
393 
394   /// Emit the specified function out to the OutStreamer.
runOnMachineFunction(MachineFunction & MF)395   bool runOnMachineFunction(MachineFunction &MF) override {
396     SetupMachineFunction(MF);
397     emitFunctionBody();
398     return false;
399   }
400 
401   //===------------------------------------------------------------------===//
402   // Coarse grained IR lowering routines.
403   //===------------------------------------------------------------------===//
404 
405   /// This should be called when a new MachineFunction is being processed from
406   /// runOnMachineFunction.
407   virtual void SetupMachineFunction(MachineFunction &MF);
408 
409   /// This method emits the body and trailer for a function.
410   void emitFunctionBody();
411 
412   void emitCFIInstruction(const MachineInstr &MI);
413 
414   void emitFrameAlloc(const MachineInstr &MI);
415 
416   void emitStackSizeSection(const MachineFunction &MF);
417 
418   void emitStackUsage(const MachineFunction &MF);
419 
420   void emitBBAddrMapSection(const MachineFunction &MF);
421 
422   void emitKCFITrapEntry(const MachineFunction &MF, const MCSymbol *Symbol);
423   virtual void emitKCFITypeId(const MachineFunction &MF);
424 
425   void emitPseudoProbe(const MachineInstr &MI);
426 
427   void emitRemarksSection(remarks::RemarkStreamer &RS);
428 
429   /// Emits a label as reference for PC sections.
430   void emitPCSectionsLabel(const MachineFunction &MF, const MDNode &MD);
431 
432   /// Emits the PC sections collected from instructions.
433   void emitPCSections(const MachineFunction &MF);
434 
435   /// Get the CFISection type for a function.
436   CFISection getFunctionCFISectionType(const Function &F) const;
437 
438   /// Get the CFISection type for a function.
439   CFISection getFunctionCFISectionType(const MachineFunction &MF) const;
440 
441   /// Get the CFISection type for the module.
getModuleCFISectionType()442   CFISection getModuleCFISectionType() const { return ModuleCFISection; }
443 
444   bool needsSEHMoves();
445 
446   /// Since emitting CFI unwind information is entangled with supporting the
447   /// exceptions, this returns true for platforms which use CFI unwind
448   /// information for other purposes (debugging, sanitizers, ...) when
449   /// `MCAsmInfo::ExceptionsType == ExceptionHandling::None`.
450   bool usesCFIWithoutEH() const;
451 
452   /// Print to the current output stream assembly representations of the
453   /// constants in the constant pool MCP. This is used to print out constants
454   /// which have been "spilled to memory" by the code generator.
455   virtual void emitConstantPool();
456 
457   /// Print assembly representations of the jump tables used by the current
458   /// function to the current output stream.
459   virtual void emitJumpTableInfo();
460 
461   /// Emit the specified global variable to the .s file.
462   virtual void emitGlobalVariable(const GlobalVariable *GV);
463 
464   /// Check to see if the specified global is a special global used by LLVM. If
465   /// so, emit it and return true, otherwise do nothing and return false.
466   bool emitSpecialLLVMGlobal(const GlobalVariable *GV);
467 
468   /// `llvm.global_ctors` and `llvm.global_dtors` are arrays of Structor
469   /// structs.
470   ///
471   /// Priority - init priority
472   /// Func - global initialization or global clean-up function
473   /// ComdatKey - associated data
474   struct Structor {
475     int Priority = 0;
476     Constant *Func = nullptr;
477     GlobalValue *ComdatKey = nullptr;
478 
479     Structor() = default;
480   };
481 
482   /// This method gathers an array of Structors and then sorts them out by
483   /// Priority.
484   /// @param List The initializer of `llvm.global_ctors` or `llvm.global_dtors`
485   /// array.
486   /// @param[out] Structors Sorted Structor structs by Priority.
487   void preprocessXXStructorList(const DataLayout &DL, const Constant *List,
488                                 SmallVector<Structor, 8> &Structors);
489 
490   /// This method emits `llvm.global_ctors` or `llvm.global_dtors` list.
491   virtual void emitXXStructorList(const DataLayout &DL, const Constant *List,
492                                   bool IsCtor);
493 
494   /// Emit an alignment directive to the specified power of two boundary. If a
495   /// global value is specified, and if that global has an explicit alignment
496   /// requested, it will override the alignment request if required for
497   /// correctness.
498   void emitAlignment(Align Alignment, const GlobalObject *GV = nullptr,
499                      unsigned MaxBytesToEmit = 0) const;
500 
501   /// Lower the specified LLVM Constant to an MCExpr.
502   virtual const MCExpr *lowerConstant(const Constant *CV);
503 
504   /// Print a general LLVM constant to the .s file.
505   /// On AIX, when an alias refers to a sub-element of a global variable, the
506   /// label of that alias needs to be emitted before the corresponding element.
507   using AliasMapTy = DenseMap<uint64_t, SmallVector<const GlobalAlias *, 1>>;
508   void emitGlobalConstant(const DataLayout &DL, const Constant *CV,
509                           AliasMapTy *AliasList = nullptr);
510 
511   /// Unnamed constant global variables solely contaning a pointer to
512   /// another globals variable act like a global variable "proxy", or GOT
513   /// equivalents, i.e., it's only used to hold the address of the latter. One
514   /// optimization is to replace accesses to these proxies by using the GOT
515   /// entry for the final global instead. Hence, we select GOT equivalent
516   /// candidates among all the module global variables, avoid emitting them
517   /// unnecessarily and finally replace references to them by pc relative
518   /// accesses to GOT entries.
519   void computeGlobalGOTEquivs(Module &M);
520 
521   /// Constant expressions using GOT equivalent globals may not be
522   /// eligible for PC relative GOT entry conversion, in such cases we need to
523   /// emit the proxies we previously omitted in EmitGlobalVariable.
524   void emitGlobalGOTEquivs();
525 
526   /// Emit the stack maps.
527   void emitStackMaps();
528 
529   //===------------------------------------------------------------------===//
530   // Overridable Hooks
531   //===------------------------------------------------------------------===//
532 
addAsmPrinterHandler(HandlerInfo Handler)533   void addAsmPrinterHandler(HandlerInfo Handler) {
534     Handlers.insert(Handlers.begin(), std::move(Handler));
535     NumUserHandlers++;
536   }
537 
538   // Targets can, or in the case of EmitInstruction, must implement these to
539   // customize output.
540 
541   /// This virtual method can be overridden by targets that want to emit
542   /// something at the start of their file.
emitStartOfAsmFile(Module &)543   virtual void emitStartOfAsmFile(Module &) {}
544 
545   /// This virtual method can be overridden by targets that want to emit
546   /// something at the end of their file.
emitEndOfAsmFile(Module &)547   virtual void emitEndOfAsmFile(Module &) {}
548 
549   /// Targets can override this to emit stuff before the first basic block in
550   /// the function.
emitFunctionBodyStart()551   virtual void emitFunctionBodyStart() {}
552 
553   /// Targets can override this to emit stuff after the last basic block in the
554   /// function.
emitFunctionBodyEnd()555   virtual void emitFunctionBodyEnd() {}
556 
557   /// Targets can override this to emit stuff at the start of a basic block.
558   /// By default, this method prints the label for the specified
559   /// MachineBasicBlock, an alignment (if present) and a comment describing it
560   /// if appropriate.
561   virtual void emitBasicBlockStart(const MachineBasicBlock &MBB);
562 
563   /// Targets can override this to emit stuff at the end of a basic block.
564   virtual void emitBasicBlockEnd(const MachineBasicBlock &MBB);
565 
566   /// Targets should implement this to emit instructions.
emitInstruction(const MachineInstr *)567   virtual void emitInstruction(const MachineInstr *) {
568     llvm_unreachable("EmitInstruction not implemented");
569   }
570 
571   /// Return the symbol for the specified constant pool entry.
572   virtual MCSymbol *GetCPISymbol(unsigned CPID) const;
573 
574   virtual void emitFunctionEntryLabel();
575 
emitFunctionDescriptor()576   virtual void emitFunctionDescriptor() {
577     llvm_unreachable("Function descriptor is target-specific.");
578   }
579 
580   virtual void emitMachineConstantPoolValue(MachineConstantPoolValue *MCPV);
581 
582   /// Targets can override this to change how global constants that are part of
583   /// a C++ static/global constructor list are emitted.
emitXXStructor(const DataLayout & DL,const Constant * CV)584   virtual void emitXXStructor(const DataLayout &DL, const Constant *CV) {
585     emitGlobalConstant(DL, CV);
586   }
587 
588   /// Return true if the basic block has exactly one predecessor and the control
589   /// transfer mechanism between the predecessor and this block is a
590   /// fall-through.
591   virtual bool
592   isBlockOnlyReachableByFallthrough(const MachineBasicBlock *MBB) const;
593 
594   /// Targets can override this to customize the output of IMPLICIT_DEF
595   /// instructions in verbose mode.
596   virtual void emitImplicitDef(const MachineInstr *MI) const;
597 
598   /// getSubtargetInfo() cannot be used where this is needed because we don't
599   /// have a MachineFunction when we're lowering a GlobalIFunc, and
600   /// getSubtargetInfo requires one. Override the implementation in targets
601   /// that support the Mach-O IFunc lowering.
getIFuncMCSubtargetInfo()602   virtual const MCSubtargetInfo *getIFuncMCSubtargetInfo() const {
603     return nullptr;
604   }
605 
emitMachOIFuncStubBody(Module & M,const GlobalIFunc & GI,MCSymbol * LazyPointer)606   virtual void emitMachOIFuncStubBody(Module &M, const GlobalIFunc &GI,
607                                       MCSymbol *LazyPointer) {
608     llvm_unreachable(
609         "Mach-O IFunc lowering is not yet supported on this target");
610   }
611 
emitMachOIFuncStubHelperBody(Module & M,const GlobalIFunc & GI,MCSymbol * LazyPointer)612   virtual void emitMachOIFuncStubHelperBody(Module &M, const GlobalIFunc &GI,
613                                             MCSymbol *LazyPointer) {
614     llvm_unreachable(
615         "Mach-O IFunc lowering is not yet supported on this target");
616   }
617 
618   /// Emit N NOP instructions.
619   void emitNops(unsigned N);
620 
621   //===------------------------------------------------------------------===//
622   // Symbol Lowering Routines.
623   //===------------------------------------------------------------------===//
624 
625   MCSymbol *createTempSymbol(const Twine &Name) const;
626 
627   /// Return the MCSymbol for a private symbol with global value name as its
628   /// base, with the specified suffix.
629   MCSymbol *getSymbolWithGlobalValueBase(const GlobalValue *GV,
630                                          StringRef Suffix) const;
631 
632   /// Return the MCSymbol for the specified ExternalSymbol.
633   MCSymbol *GetExternalSymbolSymbol(Twine Sym) const;
634 
635   /// Return the symbol for the specified jump table entry.
636   MCSymbol *GetJTISymbol(unsigned JTID, bool isLinkerPrivate = false) const;
637 
638   /// Return the symbol for the specified jump table .set
639   /// FIXME: privatize to AsmPrinter.
640   MCSymbol *GetJTSetSymbol(unsigned UID, unsigned MBBID) const;
641 
642   /// Return the MCSymbol used to satisfy BlockAddress uses of the specified
643   /// basic block.
644   MCSymbol *GetBlockAddressSymbol(const BlockAddress *BA) const;
645   MCSymbol *GetBlockAddressSymbol(const BasicBlock *BB) const;
646 
647   //===------------------------------------------------------------------===//
648   // Emission Helper Routines.
649   //===------------------------------------------------------------------===//
650 
651   /// This is just convenient handler for printing offsets.
652   void printOffset(int64_t Offset, raw_ostream &OS) const;
653 
654   /// Emit a byte directive and value.
655   void emitInt8(int Value) const;
656 
657   /// Emit a short directive and value.
658   void emitInt16(int Value) const;
659 
660   /// Emit a long directive and value.
661   void emitInt32(int Value) const;
662 
663   /// Emit a long long directive and value.
664   void emitInt64(uint64_t Value) const;
665 
666   /// Emit the specified signed leb128 value.
667   void emitSLEB128(int64_t Value, const char *Desc = nullptr) const;
668 
669   /// Emit the specified unsigned leb128 value.
670   void emitULEB128(uint64_t Value, const char *Desc = nullptr,
671                    unsigned PadTo = 0) const;
672 
673   /// Emit something like ".long Hi-Lo" where the size in bytes of the directive
674   /// is specified by Size and Hi/Lo specify the labels.  This implicitly uses
675   /// .set if it is available.
676   void emitLabelDifference(const MCSymbol *Hi, const MCSymbol *Lo,
677                            unsigned Size) const;
678 
679   /// Emit something like ".uleb128 Hi-Lo".
680   void emitLabelDifferenceAsULEB128(const MCSymbol *Hi,
681                                     const MCSymbol *Lo) const;
682 
683   /// Emit something like ".long Label+Offset" where the size in bytes of the
684   /// directive is specified by Size and Label specifies the label.  This
685   /// implicitly uses .set if it is available.
686   void emitLabelPlusOffset(const MCSymbol *Label, uint64_t Offset,
687                            unsigned Size, bool IsSectionRelative = false) const;
688 
689   /// Emit something like ".long Label" where the size in bytes of the directive
690   /// is specified by Size and Label specifies the label.
691   void emitLabelReference(const MCSymbol *Label, unsigned Size,
692                           bool IsSectionRelative = false) const {
693     emitLabelPlusOffset(Label, 0, Size, IsSectionRelative);
694   }
695 
696   //===------------------------------------------------------------------===//
697   // Dwarf Emission Helper Routines
698   //===------------------------------------------------------------------===//
699 
700   /// Emit a .byte 42 directive that corresponds to an encoding.  If verbose
701   /// assembly output is enabled, we output comments describing the encoding.
702   /// Desc is a string saying what the encoding is specifying (e.g. "LSDA").
703   void emitEncodingByte(unsigned Val, const char *Desc = nullptr) const;
704 
705   /// Return the size of the encoding in bytes.
706   unsigned GetSizeOfEncodedValue(unsigned Encoding) const;
707 
708   /// Emit reference to a ttype global with a specified encoding.
709   virtual void emitTTypeReference(const GlobalValue *GV, unsigned Encoding);
710 
711   /// Emit a reference to a symbol for use in dwarf. Different object formats
712   /// represent this in different ways. Some use a relocation others encode
713   /// the label offset in its section.
714   void emitDwarfSymbolReference(const MCSymbol *Label,
715                                 bool ForceOffset = false) const;
716 
717   /// Emit the 4- or 8-byte offset of a string from the start of its section.
718   ///
719   /// When possible, emit a DwarfStringPool section offset without any
720   /// relocations, and without using the symbol.  Otherwise, defers to \a
721   /// emitDwarfSymbolReference().
722   ///
723   /// The length of the emitted value depends on the DWARF format.
724   void emitDwarfStringOffset(DwarfStringPoolEntry S) const;
725 
726   /// Emit the 4-or 8-byte offset of a string from the start of its section.
emitDwarfStringOffset(DwarfStringPoolEntryRef S)727   void emitDwarfStringOffset(DwarfStringPoolEntryRef S) const {
728     emitDwarfStringOffset(S.getEntry());
729   }
730 
731   /// Emit something like ".long Label + Offset" or ".quad Label + Offset"
732   /// depending on the DWARF format.
733   void emitDwarfOffset(const MCSymbol *Label, uint64_t Offset) const;
734 
735   /// Emit 32- or 64-bit value depending on the DWARF format.
736   void emitDwarfLengthOrOffset(uint64_t Value) const;
737 
738   /// Emit a unit length field. The actual format, DWARF32 or DWARF64, is chosen
739   /// according to the settings.
740   void emitDwarfUnitLength(uint64_t Length, const Twine &Comment) const;
741 
742   /// Emit a unit length field. The actual format, DWARF32 or DWARF64, is chosen
743   /// according to the settings.
744   /// Return the end symbol generated inside, the caller needs to emit it.
745   MCSymbol *emitDwarfUnitLength(const Twine &Prefix,
746                                 const Twine &Comment) const;
747 
748   /// Emit reference to a call site with a specified encoding
749   void emitCallSiteOffset(const MCSymbol *Hi, const MCSymbol *Lo,
750                           unsigned Encoding) const;
751   /// Emit an integer value corresponding to the call site encoding
752   void emitCallSiteValue(uint64_t Value, unsigned Encoding) const;
753 
754   /// Get the value for DW_AT_APPLE_isa. Zero if no isa encoding specified.
getISAEncoding()755   virtual unsigned getISAEncoding() { return 0; }
756 
757   /// Emit the directive and value for debug thread local expression
758   ///
759   /// \p Value - The value to emit.
760   /// \p Size - The size of the integer (in bytes) to emit.
761   virtual void emitDebugValue(const MCExpr *Value, unsigned Size) const;
762 
763   //===------------------------------------------------------------------===//
764   // Dwarf Lowering Routines
765   //===------------------------------------------------------------------===//
766 
767   /// Emit frame instruction to describe the layout of the frame.
768   void emitCFIInstruction(const MCCFIInstruction &Inst) const;
769 
770   /// Emit Dwarf abbreviation table.
emitDwarfAbbrevs(const T & Abbrevs)771   template <typename T> void emitDwarfAbbrevs(const T &Abbrevs) const {
772     // For each abbreviation.
773     for (const auto &Abbrev : Abbrevs)
774       emitDwarfAbbrev(*Abbrev);
775 
776     // Mark end of abbreviations.
777     emitULEB128(0, "EOM(3)");
778   }
779 
780   void emitDwarfAbbrev(const DIEAbbrev &Abbrev) const;
781 
782   /// Recursively emit Dwarf DIE tree.
783   void emitDwarfDIE(const DIE &Die) const;
784 
785   //===------------------------------------------------------------------===//
786   // CodeView Helper Routines
787   //===------------------------------------------------------------------===//
788 
789   /// Gets information required to create a CodeView debug symbol for a jump
790   /// table.
791   /// Return value is <Base Address, Base Offset, Branch Address, Entry Size>
792   virtual std::tuple<const MCSymbol *, uint64_t, const MCSymbol *,
793                      codeview::JumpTableEntrySize>
794   getCodeViewJumpTableInfo(int JTI, const MachineInstr *BranchInstr,
795                            const MCSymbol *BranchLabel) const;
796 
797   //===------------------------------------------------------------------===//
798   // Inline Asm Support
799   //===------------------------------------------------------------------===//
800 
801   // These are hooks that targets can override to implement inline asm
802   // support.  These should probably be moved out of AsmPrinter someday.
803 
804   /// Print information related to the specified machine instr that is
805   /// independent of the operand, and may be independent of the instr itself.
806   /// This can be useful for portably encoding the comment character or other
807   /// bits of target-specific knowledge into the asmstrings.  The syntax used is
808   /// ${:comment}.  Targets can override this to add support for their own
809   /// strange codes.
810   virtual void PrintSpecial(const MachineInstr *MI, raw_ostream &OS,
811                             StringRef Code) const;
812 
813   /// Print the MachineOperand as a symbol. Targets with complex handling of
814   /// symbol references should override the base implementation.
815   virtual void PrintSymbolOperand(const MachineOperand &MO, raw_ostream &OS);
816 
817   /// Print the specified operand of MI, an INLINEASM instruction, using the
818   /// specified assembler variant.  Targets should override this to format as
819   /// appropriate.  This method can return true if the operand is erroneous.
820   virtual bool PrintAsmOperand(const MachineInstr *MI, unsigned OpNo,
821                                const char *ExtraCode, raw_ostream &OS);
822 
823   /// Print the specified operand of MI, an INLINEASM instruction, using the
824   /// specified assembler variant as an address. Targets should override this to
825   /// format as appropriate.  This method can return true if the operand is
826   /// erroneous.
827   virtual bool PrintAsmMemoryOperand(const MachineInstr *MI, unsigned OpNo,
828                                      const char *ExtraCode, raw_ostream &OS);
829 
830   /// Let the target do anything it needs to do before emitting inlineasm.
831   /// \p StartInfo - the subtarget info before parsing inline asm
832   virtual void emitInlineAsmStart() const;
833 
834   /// Let the target do anything it needs to do after emitting inlineasm.
835   /// This callback can be used restore the original mode in case the
836   /// inlineasm contains directives to switch modes.
837   /// \p StartInfo - the original subtarget info before inline asm
838   /// \p EndInfo   - the final subtarget info after parsing the inline asm,
839   ///                or NULL if the value is unknown.
840   virtual void emitInlineAsmEnd(const MCSubtargetInfo &StartInfo,
841                                 const MCSubtargetInfo *EndInfo) const;
842 
843   /// This emits visibility information about symbol, if this is supported by
844   /// the target.
845   void emitVisibility(MCSymbol *Sym, unsigned Visibility,
846                       bool IsDefinition = true) const;
847 
848   /// This emits linkage information about \p GVSym based on \p GV, if this is
849   /// supported by the target.
850   virtual void emitLinkage(const GlobalValue *GV, MCSymbol *GVSym) const;
851 
852   /// Return the alignment for the specified \p GV.
853   static Align getGVAlignment(const GlobalObject *GV, const DataLayout &DL,
854                               Align InAlign = Align(1));
855 
856 private:
857   /// Private state for PrintSpecial()
858   // Assign a unique ID to this machine instruction.
859   mutable const MachineInstr *LastMI = nullptr;
860   mutable unsigned LastFn = 0;
861   mutable unsigned Counter = ~0U;
862 
863   bool DwarfUsesRelocationsAcrossSections = false;
864 
865   /// This method emits the header for the current function.
866   virtual void emitFunctionHeader();
867 
868   /// This method emits a comment next to header for the current function.
869   virtual void emitFunctionHeaderComment();
870 
871   /// Emit a blob of inline asm to the output streamer.
872   void
873   emitInlineAsm(StringRef Str, const MCSubtargetInfo &STI,
874                 const MCTargetOptions &MCOptions,
875                 const MDNode *LocMDNode = nullptr,
876                 InlineAsm::AsmDialect AsmDialect = InlineAsm::AD_ATT) const;
877 
878   /// This method formats and emits the specified machine instruction that is an
879   /// inline asm.
880   void emitInlineAsm(const MachineInstr *MI) const;
881 
882   /// Add inline assembly info to the diagnostics machinery, so we can
883   /// emit file and position info. Returns SrcMgr memory buffer position.
884   unsigned addInlineAsmDiagBuffer(StringRef AsmStr,
885                                   const MDNode *LocMDNode) const;
886 
887   //===------------------------------------------------------------------===//
888   // Internal Implementation Details
889   //===------------------------------------------------------------------===//
890 
891   void emitJumpTableEntry(const MachineJumpTableInfo *MJTI,
892                           const MachineBasicBlock *MBB, unsigned uid) const;
893   void emitLLVMUsedList(const ConstantArray *InitList);
894   /// Emit llvm.ident metadata in an '.ident' directive.
895   void emitModuleIdents(Module &M);
896   /// Emit bytes for llvm.commandline metadata.
897   virtual void emitModuleCommandLines(Module &M);
898 
899   GCMetadataPrinter *getOrCreateGCPrinter(GCStrategy &S);
900   void emitGlobalAlias(Module &M, const GlobalAlias &GA);
901   void emitGlobalIFunc(Module &M, const GlobalIFunc &GI);
902 
903 private:
904   /// This method decides whether the specified basic block requires a label.
905   bool shouldEmitLabelForBasicBlock(const MachineBasicBlock &MBB) const;
906 
907 protected:
shouldEmitWeakSwiftAsyncExtendedFramePointerFlags()908   virtual bool shouldEmitWeakSwiftAsyncExtendedFramePointerFlags() const {
909     return false;
910   }
911 };
912 
913 } // end namespace llvm
914 
915 #endif // LLVM_CODEGEN_ASMPRINTER_H
916