1 //===- MCWasmStreamer.h - MCStreamer Wasm Object File Interface -*- C++ -*-===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 
9 #ifndef LLVM_MC_MCWASMSTREAMER_H
10 #define LLVM_MC_MCWASMSTREAMER_H
11 
12 #include "MCAsmBackend.h"
13 #include "MCCodeEmitter.h"
14 #include "llvm/ADT/SmallPtrSet.h"
15 #include "llvm/MC/MCDirectives.h"
16 #include "llvm/MC/MCObjectStreamer.h"
17 #include "llvm/MC/MCObjectWriter.h"
18 #include "llvm/MC/SectionKind.h"
19 #include "llvm/Support/DataTypes.h"
20 
21 namespace llvm {
22 class MCAssembler;
23 class MCExpr;
24 class MCInst;
25 class raw_ostream;
26 
27 class MCWasmStreamer : public MCObjectStreamer {
28 public:
29   MCWasmStreamer(MCContext &Context, std::unique_ptr<MCAsmBackend> TAB,
30                  std::unique_ptr<MCObjectWriter> OW,
31                  std::unique_ptr<MCCodeEmitter> Emitter)
32       : MCObjectStreamer(Context, std::move(TAB), std::move(OW),
33                          std::move(Emitter)),
34         SeenIdent(false) {}
35 
36   ~MCWasmStreamer() override;
37 
38   /// state management
39   void reset() override {
40     SeenIdent = false;
41     MCObjectStreamer::reset();
42   }
43 
44   /// \name MCStreamer Interface
45   /// @{
46 
47   void ChangeSection(MCSection *Section, const MCExpr *Subsection) override;
48   void EmitAssemblerFlag(MCAssemblerFlag Flag) override;
49   void EmitThumbFunc(MCSymbol *Func) override;
50   void EmitWeakReference(MCSymbol *Alias, const MCSymbol *Symbol) override;
51   bool EmitSymbolAttribute(MCSymbol *Symbol, MCSymbolAttr Attribute) override;
52   void EmitSymbolDesc(MCSymbol *Symbol, unsigned DescValue) override;
53   void EmitCommonSymbol(MCSymbol *Symbol, uint64_t Size,
54                         unsigned ByteAlignment) override;
55 
56   void emitELFSize(MCSymbol *Symbol, const MCExpr *Value) override;
57 
58   void EmitLocalCommonSymbol(MCSymbol *Symbol, uint64_t Size,
59                              unsigned ByteAlignment) override;
60 
61   void EmitZerofill(MCSection *Section, MCSymbol *Symbol = nullptr,
62                     uint64_t Size = 0, unsigned ByteAlignment = 0,
63                     SMLoc Loc = SMLoc()) override;
64   void EmitTBSSSymbol(MCSection *Section, MCSymbol *Symbol, uint64_t Size,
65                       unsigned ByteAlignment = 0) override;
66   void EmitValueImpl(const MCExpr *Value, unsigned Size,
67                      SMLoc Loc = SMLoc()) override;
68 
69   void EmitIdent(StringRef IdentString) override;
70 
71   void EmitValueToAlignment(unsigned, int64_t, unsigned, unsigned) override;
72 
73   void FinishImpl() override;
74 
75 private:
76   void EmitInstToFragment(const MCInst &Inst, const MCSubtargetInfo &) override;
77   void EmitInstToData(const MCInst &Inst, const MCSubtargetInfo &) override;
78 
79   /// Merge the content of the fragment \p EF into the fragment \p DF.
80   void mergeFragment(MCDataFragment *, MCDataFragment *);
81 
82   bool SeenIdent;
83 };
84 
85 } // end namespace llvm
86 
87 #endif
88