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