1 //==-- WebAssemblyTargetStreamer.h - WebAssembly Target Streamer -*- 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 /// \file
11 /// \brief This file declares WebAssembly-specific target streamer classes.
12 /// These are for implementing support for target-specific assembly directives.
13 ///
14 //===----------------------------------------------------------------------===//
15 
16 #ifndef LLVM_LIB_TARGET_WEBASSEMBLY_MCTARGETDESC_WEBASSEMBLYTARGETSTREAMER_H
17 #define LLVM_LIB_TARGET_WEBASSEMBLY_MCTARGETDESC_WEBASSEMBLYTARGETSTREAMER_H
18 
19 #include "llvm/BinaryFormat/Wasm.h"
20 #include "llvm/CodeGen/MachineValueType.h"
21 #include "llvm/MC/MCStreamer.h"
22 
23 namespace llvm {
24 
25 class MCELFStreamer;
26 class MCWasmStreamer;
27 
28 /// WebAssembly-specific streamer interface, to implement support
29 /// WebAssembly-specific assembly directives.
30 class WebAssemblyTargetStreamer : public MCTargetStreamer {
31 public:
32   explicit WebAssemblyTargetStreamer(MCStreamer &S);
33 
34   /// .param
35   virtual void emitParam(MCSymbol *Symbol, ArrayRef<MVT> Types) = 0;
36   /// .result
37   virtual void emitResult(MCSymbol *Symbol, ArrayRef<MVT> Types) = 0;
38   /// .local
39   virtual void emitLocal(ArrayRef<MVT> Types) = 0;
40   /// .globalvar
41   virtual void emitGlobal(ArrayRef<wasm::Global> Globals) = 0;
42   /// .endfunc
43   virtual void emitEndFunc() = 0;
44   /// .functype
45   virtual void emitIndirectFunctionType(MCSymbol *Symbol,
46                                         SmallVectorImpl<MVT> &Params,
47                                         SmallVectorImpl<MVT> &Results) = 0;
48   /// .indidx
49   virtual void emitIndIdx(const MCExpr *Value) = 0;
50   /// .import_global
51   virtual void emitGlobalImport(StringRef name) = 0;
52 
53 protected:
54   void emitValueType(wasm::ValType Type);
55 };
56 
57 /// This part is for ascii assembly output
58 class WebAssemblyTargetAsmStreamer final : public WebAssemblyTargetStreamer {
59   formatted_raw_ostream &OS;
60 
61 public:
62   WebAssemblyTargetAsmStreamer(MCStreamer &S, formatted_raw_ostream &OS);
63 
64   void emitParam(MCSymbol *Symbol, ArrayRef<MVT> Types) override;
65   void emitResult(MCSymbol *Symbol, ArrayRef<MVT> Types) override;
66   void emitLocal(ArrayRef<MVT> Types) override;
67   void emitGlobal(ArrayRef<wasm::Global> Globals) override;
68   void emitEndFunc() override;
69   void emitIndirectFunctionType(MCSymbol *Symbol,
70                                 SmallVectorImpl<MVT> &Params,
71                                 SmallVectorImpl<MVT> &Results) override;
72   void emitIndIdx(const MCExpr *Value) override;
73   void emitGlobalImport(StringRef name) override;
74 };
75 
76 /// This part is for ELF object output
77 class WebAssemblyTargetELFStreamer final : public WebAssemblyTargetStreamer {
78 public:
79   explicit WebAssemblyTargetELFStreamer(MCStreamer &S);
80 
81   void emitParam(MCSymbol *Symbol, ArrayRef<MVT> Types) override;
82   void emitResult(MCSymbol *Symbol, ArrayRef<MVT> Types) override;
83   void emitLocal(ArrayRef<MVT> Types) override;
84   void emitGlobal(ArrayRef<wasm::Global> Globals) override;
85   void emitEndFunc() override;
86   void emitIndirectFunctionType(MCSymbol *Symbol,
87                                 SmallVectorImpl<MVT> &Params,
88                                 SmallVectorImpl<MVT> &Results) override;
89   void emitIndIdx(const MCExpr *Value) override;
90   void emitGlobalImport(StringRef name) override;
91 };
92 
93 /// This part is for Wasm object output
94 class WebAssemblyTargetWasmStreamer final : public WebAssemblyTargetStreamer {
95 public:
96   explicit WebAssemblyTargetWasmStreamer(MCStreamer &S);
97 
98   void emitParam(MCSymbol *Symbol, ArrayRef<MVT> Types) override;
99   void emitResult(MCSymbol *Symbol, ArrayRef<MVT> Types) override;
100   void emitLocal(ArrayRef<MVT> Types) override;
101   void emitGlobal(ArrayRef<wasm::Global> Globals) override;
102   void emitEndFunc() override;
103   void emitIndirectFunctionType(MCSymbol *Symbol,
104                                 SmallVectorImpl<MVT> &Params,
105                                 SmallVectorImpl<MVT> &Results) override;
106   void emitIndIdx(const MCExpr *Value) override;
107   void emitGlobalImport(StringRef name) override;
108 };
109 
110 } // end namespace llvm
111 
112 #endif
113