1 //===- MCSymbolWasm.h -  ----------------------------------------*- 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 #ifndef LLVM_MC_MCSYMBOLWASM_H
10 #define LLVM_MC_MCSYMBOLWASM_H
11 
12 #include "llvm/BinaryFormat/Wasm.h"
13 #include "llvm/MC/MCSymbol.h"
14 
15 namespace llvm {
16 
17 class MCSymbolWasm : public MCSymbol {
18   wasm::WasmSymbolType Type = wasm::WASM_SYMBOL_TYPE_DATA;
19   bool IsWeak = false;
20   bool IsHidden = false;
21   bool IsComdat = false;
22   Optional<std::string> ImportModule;
23   Optional<std::string> ImportName;
24   wasm::WasmSignature *Signature = nullptr;
25   Optional<wasm::WasmGlobalType> GlobalType;
26   Optional<wasm::WasmEventType> EventType;
27 
28   /// An expression describing how to calculate the size of a symbol. If a
29   /// symbol has no size this field will be NULL.
30   const MCExpr *SymbolSize = nullptr;
31 
32 public:
33   // Use a module name of "env" for now, for compatibility with existing tools.
34   // This is temporary, and may change, as the ABI is not yet stable.
MCSymbolWasm(const StringMapEntry<bool> * Name,bool isTemporary)35   MCSymbolWasm(const StringMapEntry<bool> *Name, bool isTemporary)
36       : MCSymbol(SymbolKindWasm, Name, isTemporary) {}
classof(const MCSymbol * S)37   static bool classof(const MCSymbol *S) { return S->isWasm(); }
38 
getSize()39   const MCExpr *getSize() const { return SymbolSize; }
setSize(const MCExpr * SS)40   void setSize(const MCExpr *SS) { SymbolSize = SS; }
41 
isFunction()42   bool isFunction() const { return Type == wasm::WASM_SYMBOL_TYPE_FUNCTION; }
isData()43   bool isData() const { return Type == wasm::WASM_SYMBOL_TYPE_DATA; }
isGlobal()44   bool isGlobal() const { return Type == wasm::WASM_SYMBOL_TYPE_GLOBAL; }
isSection()45   bool isSection() const { return Type == wasm::WASM_SYMBOL_TYPE_SECTION; }
isEvent()46   bool isEvent() const { return Type == wasm::WASM_SYMBOL_TYPE_EVENT; }
getType()47   wasm::WasmSymbolType getType() const { return Type; }
setType(wasm::WasmSymbolType type)48   void setType(wasm::WasmSymbolType type) { Type = type; }
49 
isWeak()50   bool isWeak() const { return IsWeak; }
setWeak(bool isWeak)51   void setWeak(bool isWeak) { IsWeak = isWeak; }
52 
isHidden()53   bool isHidden() const { return IsHidden; }
setHidden(bool isHidden)54   void setHidden(bool isHidden) { IsHidden = isHidden; }
55 
isComdat()56   bool isComdat() const { return IsComdat; }
setComdat(bool isComdat)57   void setComdat(bool isComdat) { IsComdat = isComdat; }
58 
getImportModule()59   const StringRef getImportModule() const {
60       if (ImportModule.hasValue()) {
61           return ImportModule.getValue();
62       }
63       return "env";
64   }
setImportModule(StringRef Name)65   void setImportModule(StringRef Name) { ImportModule = Name; }
66 
getImportName()67   const StringRef getImportName() const {
68       if (ImportName.hasValue()) {
69           return ImportName.getValue();
70       }
71       return getName();
72   }
setImportName(StringRef Name)73   void setImportName(StringRef Name) { ImportName = Name; }
74 
getSignature()75   const wasm::WasmSignature *getSignature() const { return Signature; }
setSignature(wasm::WasmSignature * Sig)76   void setSignature(wasm::WasmSignature *Sig) { Signature = Sig; }
77 
getGlobalType()78   const wasm::WasmGlobalType &getGlobalType() const {
79     assert(GlobalType.hasValue());
80     return GlobalType.getValue();
81   }
setGlobalType(wasm::WasmGlobalType GT)82   void setGlobalType(wasm::WasmGlobalType GT) { GlobalType = GT; }
83 
getEventType()84   const wasm::WasmEventType &getEventType() const {
85     assert(EventType.hasValue());
86     return EventType.getValue();
87   }
setEventType(wasm::WasmEventType ET)88   void setEventType(wasm::WasmEventType ET) { EventType = ET; }
89 };
90 
91 } // end namespace llvm
92 
93 #endif // LLVM_MC_MCSYMBOLWASM_H
94