3b0152bba41da8ec540ee6dbcbe7568884703c59
[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
19 #include <vector>
20
21 namespace llvm {
22 class MCAsmBackend;
23 class MCAssembler;
24 class MCCodeEmitter;
25 class MCExpr;
26 class MCInst;
27 class MCSymbol;
28 class MCSymbolData;
29 class raw_ostream;
30
31 class MCELFStreamer : public MCObjectStreamer {
32 public:
33   MCELFStreamer(MCContext &Context, MCAsmBackend &TAB,
34                 raw_ostream &OS, MCCodeEmitter *Emitter)
35     : MCObjectStreamer(Context, TAB, OS, Emitter) {}
36
37   MCELFStreamer(MCContext &Context, MCAsmBackend &TAB,
38                 raw_ostream &OS, MCCodeEmitter *Emitter,
39                 MCAssembler *Assembler)
40     : MCObjectStreamer(Context, TAB, OS, Emitter, Assembler) {}
41
42   virtual ~MCELFStreamer();
43
44   /// @name MCStreamer Interface
45   /// @{
46
47   virtual void InitSections();
48   virtual void ChangeSection(const MCSection *Section);
49   virtual void EmitLabel(MCSymbol *Symbol);
50   virtual void EmitAssemblerFlag(MCAssemblerFlag Flag);
51   virtual void EmitThumbFunc(MCSymbol *Func);
52   virtual void EmitWeakReference(MCSymbol *Alias, const MCSymbol *Symbol);
53   virtual void EmitSymbolAttribute(MCSymbol *Symbol, MCSymbolAttr Attribute);
54   virtual void EmitSymbolDesc(MCSymbol *Symbol, unsigned DescValue);
55   virtual void EmitCommonSymbol(MCSymbol *Symbol, uint64_t Size,
56                                 unsigned ByteAlignment);
57   virtual void BeginCOFFSymbolDef(const MCSymbol *Symbol);
58   virtual void EmitCOFFSymbolStorageClass(int StorageClass);
59   virtual void EmitCOFFSymbolType(int Type);
60   virtual void EndCOFFSymbolDef();
61
62   virtual void EmitELFSize(MCSymbol *Symbol, const MCExpr *Value);
63
64   virtual void EmitLocalCommonSymbol(MCSymbol *Symbol, uint64_t Size,
65                                      unsigned ByteAlignment);
66
67   virtual void EmitZerofill(const MCSection *Section, MCSymbol *Symbol = 0,
68                             uint64_t Size = 0, unsigned ByteAlignment = 0);
69   virtual void EmitTBSSSymbol(const MCSection *Section, MCSymbol *Symbol,
70                               uint64_t Size, unsigned ByteAlignment = 0);
71   virtual void EmitValueImpl(const MCExpr *Value, unsigned Size,
72                              unsigned AddrSpace);
73
74   virtual void EmitFileDirective(StringRef Filename);
75
76   virtual void EmitTCEntry(const MCSymbol &S);
77
78   virtual void FinishImpl();
79   /// @}
80
81 private:
82   virtual void EmitInstToFragment(const MCInst &Inst);
83   virtual void EmitInstToData(const MCInst &Inst);
84
85   void fixSymbolsInTLSFixups(const MCExpr *expr);
86
87   struct LocalCommon {
88     MCSymbolData *SD;
89     uint64_t Size;
90     unsigned ByteAlignment;
91   };
92
93   std::vector<LocalCommon> LocalCommons;
94
95   SmallPtrSet<MCSymbol *, 16> BindingExplicitlySet;
96
97
98   void SetSection(StringRef Section, unsigned Type, unsigned Flags,
99                   SectionKind Kind);
100   void SetSectionData();
101   void SetSectionText();
102   void SetSectionBss();
103 };
104
105 } // end namespace llvm
106
107 #endif