Add support for subsections to the ELF assembler. Fixes PR8717.
[oota-llvm.git] / include / llvm / MC / MCELFStreamer.h
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 MCSymbol;
27 class MCSymbolData;
28 class raw_ostream;
29
30 class MCELFStreamer : public MCObjectStreamer {
31 protected:
32   MCELFStreamer(StreamerKind Kind, MCContext &Context, MCAsmBackend &TAB,
33                 raw_ostream &OS, MCCodeEmitter *Emitter)
34       : MCObjectStreamer(Kind, Context, TAB, OS, Emitter) {}
35
36 public:
37   MCELFStreamer(MCContext &Context, MCAsmBackend &TAB, raw_ostream &OS,
38                 MCCodeEmitter *Emitter)
39       : MCObjectStreamer(SK_ELFStreamer, Context, TAB, OS, Emitter) {}
40
41   MCELFStreamer(MCContext &Context, MCAsmBackend &TAB, raw_ostream &OS,
42                 MCCodeEmitter *Emitter, MCAssembler *Assembler)
43       : MCObjectStreamer(SK_ELFStreamer, Context, TAB, OS, Emitter,
44                          Assembler) {}
45
46   virtual ~MCELFStreamer();
47
48   /// @name MCStreamer Interface
49   /// @{
50
51   virtual void InitSections();
52   virtual void InitToTextSection();
53   virtual void ChangeSection(const MCSection *Section,
54                              const MCExpr *Subsection);
55   virtual void EmitLabel(MCSymbol *Symbol);
56   virtual void EmitDebugLabel(MCSymbol *Symbol);
57   virtual void EmitAssemblerFlag(MCAssemblerFlag Flag);
58   virtual void EmitThumbFunc(MCSymbol *Func);
59   virtual void EmitWeakReference(MCSymbol *Alias, const MCSymbol *Symbol);
60   virtual void EmitSymbolAttribute(MCSymbol *Symbol, MCSymbolAttr Attribute);
61   virtual void EmitSymbolDesc(MCSymbol *Symbol, unsigned DescValue);
62   virtual void EmitCommonSymbol(MCSymbol *Symbol, uint64_t Size,
63                                 unsigned ByteAlignment);
64   virtual void BeginCOFFSymbolDef(const MCSymbol *Symbol);
65   virtual void EmitCOFFSymbolStorageClass(int StorageClass);
66   virtual void EmitCOFFSymbolType(int Type);
67   virtual void EndCOFFSymbolDef();
68
69   virtual MCSymbolData &getOrCreateSymbolData(MCSymbol *Symbol);
70
71   virtual void EmitELFSize(MCSymbol *Symbol, const MCExpr *Value);
72
73   virtual void EmitLocalCommonSymbol(MCSymbol *Symbol, uint64_t Size,
74                                      unsigned ByteAlignment);
75
76   virtual void EmitZerofill(const MCSection *Section, MCSymbol *Symbol = 0,
77                             uint64_t Size = 0, unsigned ByteAlignment = 0);
78   virtual void EmitTBSSSymbol(const MCSection *Section, MCSymbol *Symbol,
79                               uint64_t Size, unsigned ByteAlignment = 0);
80   virtual void EmitValueImpl(const MCExpr *Value, unsigned Size,
81                              unsigned AddrSpace);
82
83   virtual void EmitFileDirective(StringRef Filename);
84
85   virtual void EmitTCEntry(const MCSymbol &S);
86
87   virtual void EmitValueToAlignment(unsigned, int64_t, unsigned, unsigned);
88
89   virtual void FinishImpl();
90   /// @}
91
92   static bool classof(const MCStreamer *S) {
93     return S->getKind() == SK_ELFStreamer || S->getKind() == SK_ARMELFStreamer;
94   }
95
96 private:
97   virtual void EmitInstToFragment(const MCInst &Inst);
98   virtual void EmitInstToData(const MCInst &Inst);
99
100   virtual void EmitBundleAlignMode(unsigned AlignPow2);
101   virtual void EmitBundleLock(bool AlignToEnd);
102   virtual void EmitBundleUnlock();
103
104   void fixSymbolsInTLSFixups(const MCExpr *expr);
105
106   struct LocalCommon {
107     MCSymbolData *SD;
108     uint64_t Size;
109     unsigned ByteAlignment;
110   };
111
112   std::vector<LocalCommon> LocalCommons;
113
114   SmallPtrSet<MCSymbol *, 16> BindingExplicitlySet;
115
116
117   void SetSection(StringRef Section, unsigned Type, unsigned Flags,
118                   SectionKind Kind);
119   void SetSectionData();
120   void SetSectionText();
121   void SetSectionBss();
122 };
123
124 } // end namespace llvm
125
126 #endif