1 //===- InputElement.h ----------------------------------------*- 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 LLD_WASM_INPUT_ELEMENT_H
10 #define LLD_WASM_INPUT_ELEMENT_H
11
12 #include "Config.h"
13 #include "InputFiles.h"
14 #include "WriterUtils.h"
15 #include "lld/Common/LLVM.h"
16 #include "llvm/Object/Wasm.h"
17
18 namespace lld {
19 namespace wasm {
20
21 // Represents a single element (Global, Tag, Table, etc) within an input
22 // file.
23 class InputElement {
24 protected:
InputElement(StringRef name,ObjFile * f)25 InputElement(StringRef name, ObjFile *f)
26 : file(f), live(!config->gcSections), name(name) {}
27
28 public:
getName()29 StringRef getName() const { return name; }
getAssignedIndex()30 uint32_t getAssignedIndex() const { return assignedIndex.getValue(); }
hasAssignedIndex()31 bool hasAssignedIndex() const { return assignedIndex.hasValue(); }
assignIndex(uint32_t index)32 void assignIndex(uint32_t index) {
33 assert(!hasAssignedIndex());
34 assignedIndex = index;
35 }
36
37 ObjFile *file;
38 bool live = false;
39
40 protected:
41 StringRef name;
42 llvm::Optional<uint32_t> assignedIndex;
43 };
44
intConst(uint64_t value,bool is64)45 inline WasmInitExpr intConst(uint64_t value, bool is64) {
46 WasmInitExpr ie;
47 if (is64) {
48 ie.Opcode = llvm::wasm::WASM_OPCODE_I64_CONST;
49 ie.Value.Int64 = static_cast<int64_t>(value);
50 } else {
51 ie.Opcode = llvm::wasm::WASM_OPCODE_I32_CONST;
52 ie.Value.Int32 = static_cast<int32_t>(value);
53 }
54 return ie;
55 }
56
57 class InputGlobal : public InputElement {
58 public:
InputGlobal(const WasmGlobal & g,ObjFile * f)59 InputGlobal(const WasmGlobal &g, ObjFile *f)
60 : InputElement(g.SymbolName, f), type(g.Type), initExpr(g.InitExpr) {}
61
getType()62 const WasmGlobalType &getType() const { return type; }
getInitExpr()63 const WasmInitExpr &getInitExpr() const { return initExpr; }
64
setPointerValue(uint64_t value)65 void setPointerValue(uint64_t value) {
66 initExpr = intConst(value, config->is64.getValueOr(false));
67 }
68
69 private:
70 WasmGlobalType type;
71 WasmInitExpr initExpr;
72 };
73
74 class InputTag : public InputElement {
75 public:
InputTag(const WasmSignature & s,const WasmTag & t,ObjFile * f)76 InputTag(const WasmSignature &s, const WasmTag &t, ObjFile *f)
77 : InputElement(t.SymbolName, f), signature(s), type(t.Type) {}
78
getType()79 const WasmTagType &getType() const { return type; }
80
81 const WasmSignature &signature;
82
83 private:
84 WasmTagType type;
85 };
86
87 class InputTable : public InputElement {
88 public:
InputTable(const WasmTable & t,ObjFile * f)89 InputTable(const WasmTable &t, ObjFile *f)
90 : InputElement(t.SymbolName, f), type(t.Type) {}
91
getType()92 const WasmTableType &getType() const { return type; }
setLimits(const WasmLimits & limits)93 void setLimits(const WasmLimits &limits) { type.Limits = limits; }
94
95 private:
96 WasmTableType type;
97 };
98
99 } // namespace wasm
100
toString(const wasm::InputElement * d)101 inline std::string toString(const wasm::InputElement *d) {
102 return (toString(d->file) + ":(" + d->getName() + ")").str();
103 }
104
105 } // namespace lld
106
107 #endif // LLD_WASM_INPUT_ELEMENT_H
108