MCELFStreamer.h revision a3863ea2dacafc925a8272ebf9884fc64bef686c 
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                             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
82  virtual void EmitFileDirective(StringRef Filename);
83
84  virtual void EmitTCEntry(const MCSymbol &S);
85
86  virtual void EmitValueToAlignment(unsigned, int64_t, unsigned, unsigned);
87
88  virtual void FinishImpl();
89  /// @}
90
91  static bool classof(const MCStreamer *S) {
92    return S->getKind() == SK_ELFStreamer || S->getKind() == SK_ARMELFStreamer;
93  }
94
95private:
96  virtual void EmitInstToFragment(const MCInst &Inst);
97  virtual void EmitInstToData(const MCInst &Inst);
98
99  virtual void EmitBundleAlignMode(unsigned AlignPow2);
100  virtual void EmitBundleLock(bool AlignToEnd);
101  virtual void EmitBundleUnlock();
102
103  void fixSymbolsInTLSFixups(const MCExpr *expr);
104
105  struct LocalCommon {
106    MCSymbolData *SD;
107    uint64_t Size;
108    unsigned ByteAlignment;
109  };
110
111  std::vector<LocalCommon> LocalCommons;
112
113  SmallPtrSet<MCSymbol *, 16> BindingExplicitlySet;
114
115
116  void SetSection(StringRef Section, unsigned Type, unsigned Flags,
117                  SectionKind Kind);
118  void SetSectionData();
119  void SetSectionText();
120  void SetSectionBss();
121};
122
123} // end namespace llvm
124
125#endif
126