Index: include/llvm/BinaryFormat/Wasm.h =================================================================== --- include/llvm/BinaryFormat/Wasm.h +++ include/llvm/BinaryFormat/Wasm.h @@ -115,8 +115,14 @@ int64_t Addend; // A value to add to the symbol. }; +struct WasmInitFunc { + uint32_t Priority; + uint32_t FunctionIndex; +}; + struct WasmLinkingData { uint32_t DataSize; + std::vector InitFunctions; }; enum : unsigned { @@ -185,6 +191,7 @@ WASM_SYMBOL_INFO = 0x2, WASM_DATA_SIZE = 0x3, WASM_SEGMENT_INFO = 0x5, + WASM_INIT_FUNCS = 0x6, }; const unsigned WASM_SYMBOL_BINDING_MASK = 0x3; Index: include/llvm/Object/Wasm.h =================================================================== --- include/llvm/Object/Wasm.h +++ include/llvm/Object/Wasm.h @@ -206,6 +206,7 @@ bool isRelocatableObject() const override; private: + bool isValidFunctionIndex(uint32_t Index) const; const WasmSection &getWasmSection(DataRefImpl Ref) const; const wasm::WasmRelocation &getWasmRelocation(DataRefImpl Ref) const; Index: include/llvm/ObjectYAML/WasmYAML.h =================================================================== --- include/llvm/ObjectYAML/WasmYAML.h +++ include/llvm/ObjectYAML/WasmYAML.h @@ -131,6 +131,11 @@ SymbolFlags Flags; }; +struct InitFunction { + uint32_t Priority; + uint32_t FunctionIndex; +}; + struct Section { explicit Section(SectionType SecType) : Type(SecType) {} virtual ~Section(); @@ -173,6 +178,7 @@ uint32_t DataSize; std::vector SymbolInfos; std::vector SegmentInfos; + std::vector InitFunctions; }; struct TypeSection : Section { @@ -309,6 +315,7 @@ LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::WasmYAML::NameEntry) LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::WasmYAML::SegmentInfo) LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::WasmYAML::SymbolInfo) +LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::WasmYAML::InitFunction) namespace llvm { namespace yaml { @@ -401,6 +408,10 @@ static void mapping(IO &IO, WasmYAML::SymbolInfo &Info); }; +template <> struct MappingTraits { + static void mapping(IO &IO, WasmYAML::InitFunction &Init); +}; + template <> struct ScalarEnumerationTraits { static void enumeration(IO &IO, WasmYAML::ValueType &Type); }; Index: lib/Object/WasmObjectFile.cpp =================================================================== --- lib/Object/WasmObjectFile.cpp +++ lib/Object/WasmObjectFile.cpp @@ -398,6 +398,21 @@ } break; } + case wasm::WASM_INIT_FUNCS: { + uint32_t Count = readVaruint32(Ptr); + LinkingData.InitFunctions.reserve(Count); + for (uint32_t i = 0; i < Count; i++) { + wasm::WasmInitFunc Init; + Init.Priority = readVaruint32(Ptr); + Init.FunctionIndex = readVaruint32(Ptr); + if (!isValidFunctionIndex(Init.FunctionIndex)) + return make_error("Invalid function index: " + + Twine(Init.FunctionIndex), + object_error::parse_failed); + LinkingData.InitFunctions.emplace_back(Init); + } + break; + } default: Ptr += Size; break; @@ -656,9 +671,13 @@ return Error::success(); } +bool WasmObjectFile::isValidFunctionIndex(uint32_t Index) const { + return Index < FunctionTypes.size() + NumImportedFunctions; +} + Error WasmObjectFile::parseStartSection(const uint8_t *Ptr, const uint8_t *End) { StartFunction = readVaruint32(Ptr); - if (StartFunction >= FunctionTypes.size()) + if (!isValidFunctionIndex(StartFunction)) return make_error("Invalid start function", object_error::parse_failed); return Error::success(); Index: lib/ObjectYAML/WasmYAML.cpp =================================================================== --- lib/ObjectYAML/WasmYAML.cpp +++ lib/ObjectYAML/WasmYAML.cpp @@ -60,6 +60,7 @@ IO.mapRequired("DataSize", Section.DataSize); IO.mapOptional("SymbolInfo", Section.SymbolInfos); IO.mapOptional("SegmentInfo", Section.SegmentInfos); + IO.mapOptional("InitFunctions", Section.InitFunctions); } static void sectionMapping(IO &IO, WasmYAML::CustomSection &Section) { @@ -359,6 +360,12 @@ IO.mapRequired("Content", Segment.Content); } +void MappingTraits::mapping( + IO &IO, WasmYAML::InitFunction &Init) { + IO.mapRequired("Priority", Init.Priority); + IO.mapRequired("FunctionIndex", Init.FunctionIndex); +} + void MappingTraits::mapping(IO &IO, WasmYAML::SymbolInfo &Info) { IO.mapRequired("Name", Info.Name); Index: test/ObjectYAML/wasm/linking_section.yaml =================================================================== --- test/ObjectYAML/wasm/linking_section.yaml +++ test/ObjectYAML/wasm/linking_section.yaml @@ -41,6 +41,9 @@ Alignment: 2 Flags: [ ] Name: moredata + InitFunctions: + - Priority: 1 + FunctionIndex: 0 ... # CHECK: - Type: CUSTOM # CHECK-NEXT: Name: linking @@ -57,4 +60,7 @@ # CHECK-NEXT: Name: moredata # CHECK-NEXT: Alignment: 2 # CHECK-NEXT: Flags: [ ] +# CHECK-NEXT: InitFunctions: +# CHECK-NEXT: - Priority: 1 +# CHECK-NEXT: FunctionIndex: 0 # CHECK-NEXT: ... Index: tools/obj2yaml/wasm2yaml.cpp =================================================================== --- tools/obj2yaml/wasm2yaml.cpp +++ tools/obj2yaml/wasm2yaml.cpp @@ -80,11 +80,15 @@ for (const object::SymbolRef& Sym: Obj.symbols()) { const object::WasmSymbol Symbol = Obj.getWasmSymbol(Sym); if (Symbol.Flags != 0) { - WasmYAML::SymbolInfo Info = { Symbol.Name, Symbol.Flags }; - LinkingSec->SymbolInfos.push_back(Info); + WasmYAML::SymbolInfo Info{Symbol.Name, Symbol.Flags}; + LinkingSec->SymbolInfos.emplace_back(Info); } } LinkingSec->DataSize = Obj.linkingData().DataSize; + for (const wasm::WasmInitFunc &Func : Obj.linkingData().InitFunctions) { + WasmYAML::InitFunction F{Func.Priority, Func.FunctionIndex}; + LinkingSec->InitFunctions.emplace_back(F); + } CustomSec = std::move(LinkingSec); } else { CustomSec = make_unique(WasmSec.Name); Index: tools/yaml2obj/yaml2wasm.cpp =================================================================== --- tools/yaml2obj/yaml2wasm.cpp +++ tools/yaml2obj/yaml2wasm.cpp @@ -162,6 +162,17 @@ } SubSection.Done(); } + + // INIT_FUNCS subsection + if (Section.InitFunctions.size()) { + encodeULEB128(wasm::WASM_INIT_FUNCS, OS); + encodeULEB128(Section.InitFunctions.size(), SubSection.GetStream()); + for (const WasmYAML::InitFunction &Func : Section.InitFunctions) { + encodeULEB128(Func.Priority, SubSection.GetStream()); + encodeULEB128(Func.FunctionIndex, SubSection.GetStream()); + } + SubSection.Done(); + } return 0; }