llvm-project/lld/wasm/InputElement.h
2023-01-02 18:29:04 -08:00

105 lines
2.7 KiB
C++

//===- InputElement.h ----------------------------------------*- C++ -*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#ifndef LLD_WASM_INPUT_ELEMENT_H
#define LLD_WASM_INPUT_ELEMENT_H
#include "Config.h"
#include "InputFiles.h"
#include "WriterUtils.h"
#include "lld/Common/LLVM.h"
#include "llvm/Object/Wasm.h"
#include <optional>
namespace lld {
namespace wasm {
// Represents a single element (Global, Tag, Table, etc) within an input
// file.
class InputElement {
protected:
InputElement(StringRef name, ObjFile *f)
: file(f), live(!config->gcSections), name(name) {}
public:
StringRef getName() const { return name; }
uint32_t getAssignedIndex() const { return *assignedIndex; }
bool hasAssignedIndex() const { return assignedIndex.has_value(); }
void assignIndex(uint32_t index) {
assert(!hasAssignedIndex());
assignedIndex = index;
}
ObjFile *file;
bool live = false;
protected:
StringRef name;
std::optional<uint32_t> assignedIndex;
};
inline WasmInitExpr intConst(uint64_t value, bool is64) {
WasmInitExpr ie;
ie.Extended = false;
if (is64) {
ie.Inst.Opcode = llvm::wasm::WASM_OPCODE_I64_CONST;
ie.Inst.Value.Int64 = static_cast<int64_t>(value);
} else {
ie.Inst.Opcode = llvm::wasm::WASM_OPCODE_I32_CONST;
ie.Inst.Value.Int32 = static_cast<int32_t>(value);
}
return ie;
}
class InputGlobal : public InputElement {
public:
InputGlobal(const WasmGlobal &g, ObjFile *f)
: InputElement(g.SymbolName, f), type(g.Type), initExpr(g.InitExpr) {}
const WasmGlobalType &getType() const { return type; }
const WasmInitExpr &getInitExpr() const { return initExpr; }
void setPointerValue(uint64_t value) {
initExpr = intConst(value, config->is64.value_or(false));
}
private:
WasmGlobalType type;
WasmInitExpr initExpr;
};
class InputTag : public InputElement {
public:
InputTag(const WasmSignature &s, const WasmTag &t, ObjFile *f)
: InputElement(t.SymbolName, f), signature(s) {}
const WasmSignature &signature;
};
class InputTable : public InputElement {
public:
InputTable(const WasmTable &t, ObjFile *f)
: InputElement(t.SymbolName, f), type(t.Type) {}
const WasmTableType &getType() const { return type; }
void setLimits(const WasmLimits &limits) { type.Limits = limits; }
private:
WasmTableType type;
};
} // namespace wasm
inline std::string toString(const wasm::InputElement *d) {
return (toString(d->file) + ":(" + d->getName() + ")").str();
}
} // namespace lld
#endif // LLD_WASM_INPUT_ELEMENT_H