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