llvm-project/lld/wasm/InputElement.h
Derek Schuff 7f409cd82b
[Object][Wasm] Allow parsing of GC types in type and table sections (#79235)
This change allows a WasmObjectFile to be created from a wasm file even 
if it uses typed funcrefs and GC types. It does not significantly change how 
lib/Object models its various internal types (e.g. WasmSignature,
WasmElemSegment), so LLVM does not really "support" or understand such
files, but it is sufficient to parse the type, global and element sections, discarding
types that are not understood. This is useful for low-level binary tools such as
nm and objcopy, which use only limited aspects of the binary (such as function
definitions) or deal with sections as opaque blobs.

This is done by allowing `WasmValType` to have a value of `OTHERREF`
(representing any unmodeled reference type), and adding a field to
`WasmSignature` indicating it's a placeholder for an unmodeled reference 
type (since there is a 1:1 correspondence between WasmSignature objects
and types in the type section).
Then the object file parsers for the type and element sections are expanded
to parse encoded reference types and discard any unmodeled fields.
2024-01-25 09:48:38 -08:00

107 lines
2.8 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) {
assert(s.Kind == WasmSignature::Tag);
}
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