1 //===- MCELFStreamer.h - MCStreamer ELF Object File Interface ---*- 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 #ifndef LLVM_MC_MCELFSTREAMER_H 11 #define LLVM_MC_MCELFSTREAMER_H 12 13 #include "llvm/ADT/SmallPtrSet.h" 14 #include "llvm/MC/MCDirectives.h" 15 #include "llvm/MC/MCObjectStreamer.h" 16 #include "llvm/MC/SectionKind.h" 17 #include "llvm/Support/DataTypes.h" 18 #include <vector> 19 20 namespace llvm { 21 class MCAsmBackend; 22 class MCAssembler; 23 class MCCodeEmitter; 24 class MCExpr; 25 class MCInst; 26 class raw_ostream; 27 28 class MCELFStreamer : public MCObjectStreamer { 29 public: MCELFStreamer(MCContext & Context,MCAsmBackend & TAB,raw_pwrite_stream & OS,MCCodeEmitter * Emitter)30 MCELFStreamer(MCContext &Context, MCAsmBackend &TAB, raw_pwrite_stream &OS, 31 MCCodeEmitter *Emitter) 32 : MCObjectStreamer(Context, TAB, OS, Emitter), SeenIdent(false) {} 33 34 ~MCELFStreamer() override; 35 36 /// state management reset()37 void reset() override { 38 SeenIdent = false; 39 LocalCommons.clear(); 40 BundleGroups.clear(); 41 MCObjectStreamer::reset(); 42 } 43 44 /// \name MCStreamer Interface 45 /// @{ 46 47 void InitSections(bool NoExecStack) override; 48 void ChangeSection(MCSection *Section, const MCExpr *Subsection) override; 49 void EmitLabel(MCSymbol *Symbol) override; 50 void EmitAssemblerFlag(MCAssemblerFlag Flag) override; 51 void EmitThumbFunc(MCSymbol *Func) override; 52 void EmitWeakReference(MCSymbol *Alias, const MCSymbol *Symbol) override; 53 bool EmitSymbolAttribute(MCSymbol *Symbol, MCSymbolAttr Attribute) override; 54 void EmitSymbolDesc(MCSymbol *Symbol, unsigned DescValue) override; 55 void EmitCommonSymbol(MCSymbol *Symbol, uint64_t Size, 56 unsigned ByteAlignment) override; 57 void BeginCOFFSymbolDef(const MCSymbol *Symbol) override; 58 void EmitCOFFSymbolStorageClass(int StorageClass) override; 59 void EmitCOFFSymbolType(int Type) override; 60 void EndCOFFSymbolDef() override; 61 62 void emitELFSize(MCSymbolELF *Symbol, const MCExpr *Value) override; 63 64 void EmitLocalCommonSymbol(MCSymbol *Symbol, uint64_t Size, 65 unsigned ByteAlignment) override; 66 67 void EmitZerofill(MCSection *Section, MCSymbol *Symbol = nullptr, 68 uint64_t Size = 0, unsigned ByteAlignment = 0) override; 69 void EmitTBSSSymbol(MCSection *Section, MCSymbol *Symbol, uint64_t Size, 70 unsigned ByteAlignment = 0) override; 71 void EmitValueImpl(const MCExpr *Value, unsigned Size, 72 const SMLoc &Loc = SMLoc()) override; 73 74 void EmitFileDirective(StringRef Filename) override; 75 76 void EmitIdent(StringRef IdentString) override; 77 78 void EmitValueToAlignment(unsigned, int64_t, unsigned, unsigned) override; 79 80 void Flush() override; 81 82 void FinishImpl() override; 83 84 void EmitBundleAlignMode(unsigned AlignPow2) override; 85 void EmitBundleLock(bool AlignToEnd) override; 86 void EmitBundleUnlock() override; 87 88 private: 89 bool isBundleLocked() const; 90 void EmitInstToFragment(const MCInst &Inst, const MCSubtargetInfo &) override; 91 void EmitInstToData(const MCInst &Inst, const MCSubtargetInfo &) override; 92 93 void fixSymbolsInTLSFixups(const MCExpr *expr); 94 95 /// \brief Merge the content of the fragment \p EF into the fragment \p DF. 96 void mergeFragment(MCDataFragment *, MCDataFragment *); 97 98 bool SeenIdent; 99 100 struct LocalCommon { 101 const MCSymbol *Symbol; 102 uint64_t Size; 103 unsigned ByteAlignment; 104 }; 105 106 std::vector<LocalCommon> LocalCommons; 107 108 /// BundleGroups - The stack of fragments holding the bundle-locked 109 /// instructions. 110 llvm::SmallVector<MCDataFragment *, 4> BundleGroups; 111 }; 112 113 MCELFStreamer *createARMELFStreamer(MCContext &Context, MCAsmBackend &TAB, 114 raw_pwrite_stream &OS, 115 MCCodeEmitter *Emitter, bool RelaxAll, 116 bool IsThumb); 117 118 } // end namespace llvm 119 120 #endif 121