MCELFStreamer.h revision 5da3665cc501ed8928e63678254357214ec0b9eb
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
20namespace llvm {
21class MCAsmBackend;
22class MCAssembler;
23class MCCodeEmitter;
24class MCExpr;
25class MCInst;
26class MCSymbol;
27class MCSymbolData;
28class raw_ostream;
29
30class MCELFStreamer : public MCObjectStreamer {
31protected:
32  MCELFStreamer(StreamerKind Kind, MCContext &Context, MCAsmBackend &TAB,
33                raw_ostream &OS, MCCodeEmitter *Emitter)
34      : MCObjectStreamer(Kind, Context, TAB, OS, Emitter) {}
35
36public:
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  virtual void EmitLabel(MCSymbol *Symbol);
55  virtual void EmitDebugLabel(MCSymbol *Symbol);
56  virtual void EmitAssemblerFlag(MCAssemblerFlag Flag);
57  virtual void EmitThumbFunc(MCSymbol *Func);
58  virtual void EmitWeakReference(MCSymbol *Alias, const MCSymbol *Symbol);
59  virtual void EmitSymbolAttribute(MCSymbol *Symbol, MCSymbolAttr Attribute);
60  virtual void EmitSymbolDesc(MCSymbol *Symbol, unsigned DescValue);
61  virtual void EmitCommonSymbol(MCSymbol *Symbol, uint64_t Size,
62                                unsigned ByteAlignment);
63  virtual void BeginCOFFSymbolDef(const MCSymbol *Symbol);
64  virtual void EmitCOFFSymbolStorageClass(int StorageClass);
65  virtual void EmitCOFFSymbolType(int Type);
66  virtual void EndCOFFSymbolDef();
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                             unsigned AddrSpace);
79
80  virtual void EmitFileDirective(StringRef Filename);
81
82  virtual void EmitTCEntry(const MCSymbol &S);
83
84  virtual void EmitValueToAlignment(unsigned, int64_t, unsigned, unsigned);
85
86  virtual void FinishImpl();
87  /// @}
88
89  static bool classof(const MCStreamer *S) {
90    return S->getKind() == SK_ELFStreamer && S->getKind() == SK_ARMELFStreamer;
91  }
92
93private:
94  virtual void EmitInstToFragment(const MCInst &Inst);
95  virtual void EmitInstToData(const MCInst &Inst);
96
97  virtual void EmitBundleAlignMode(unsigned AlignPow2);
98  virtual void EmitBundleLock(bool AlignToEnd);
99  virtual void EmitBundleUnlock();
100
101  void fixSymbolsInTLSFixups(const MCExpr *expr);
102
103  struct LocalCommon {
104    MCSymbolData *SD;
105    uint64_t Size;
106    unsigned ByteAlignment;
107  };
108
109  std::vector<LocalCommon> LocalCommons;
110
111  SmallPtrSet<MCSymbol *, 16> BindingExplicitlySet;
112
113
114  void SetSection(StringRef Section, unsigned Type, unsigned Flags,
115                  SectionKind Kind);
116  void SetSectionData();
117  void SetSectionText();
118  void SetSectionBss();
119};
120
121} // end namespace llvm
122
123#endif
124