144d95122SEugene Zelenko //===- WasmObjectFile.cpp - Wasm object file implementation ---------------===//
22c6f75ddSDerek Schuff //
32946cd70SChandler Carruth // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
42946cd70SChandler Carruth // See https://llvm.org/LICENSE.txt for license information.
52946cd70SChandler Carruth // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
62c6f75ddSDerek Schuff //
72c6f75ddSDerek Schuff //===----------------------------------------------------------------------===//
82c6f75ddSDerek Schuff 
944d95122SEugene Zelenko #include "llvm/ADT/ArrayRef.h"
104710ed7aSSam Clegg #include "llvm/ADT/DenseSet.h"
11cbda16ebSThomas Lively #include "llvm/ADT/SmallSet.h"
1244d95122SEugene Zelenko #include "llvm/ADT/StringRef.h"
13ea7caceeSSam Clegg #include "llvm/ADT/StringSet.h"
140ea52537SSimon Pilgrim #include "llvm/ADT/StringSwitch.h"
1544d95122SEugene Zelenko #include "llvm/ADT/Triple.h"
16264b5d9eSZachary Turner #include "llvm/BinaryFormat/Wasm.h"
179f5094dfSEugene Zelenko #include "llvm/MC/SubtargetFeature.h"
1844d95122SEugene Zelenko #include "llvm/Object/Binary.h"
1944d95122SEugene Zelenko #include "llvm/Object/Error.h"
2044d95122SEugene Zelenko #include "llvm/Object/ObjectFile.h"
2144d95122SEugene Zelenko #include "llvm/Object/SymbolicFile.h"
222c6f75ddSDerek Schuff #include "llvm/Object/Wasm.h"
232c6f75ddSDerek Schuff #include "llvm/Support/Endian.h"
2444d95122SEugene Zelenko #include "llvm/Support/Error.h"
2544d95122SEugene Zelenko #include "llvm/Support/ErrorHandling.h"
262c6f75ddSDerek Schuff #include "llvm/Support/LEB128.h"
27feef720bSHeejin Ahn #include "llvm/Support/ScopedPrinter.h"
2844d95122SEugene Zelenko #include <algorithm>
299f5094dfSEugene Zelenko #include <cassert>
3044d95122SEugene Zelenko #include <cstdint>
319f5094dfSEugene Zelenko #include <cstring>
322c6f75ddSDerek Schuff 
33b7787fd0SSam Clegg #define DEBUG_TYPE "wasm-object"
34b7787fd0SSam Clegg 
3544d95122SEugene Zelenko using namespace llvm;
3644d95122SEugene Zelenko using namespace object;
372c6f75ddSDerek Schuff 
print(raw_ostream & Out) const38659932b0SJF Bastien void WasmSymbol::print(raw_ostream &Out) const {
39659932b0SJF Bastien   Out << "Name=" << Info.Name
4073332d73SSam Clegg       << ", Kind=" << toString(wasm::WasmSymbolType(Info.Kind)) << ", Flags=0x"
4173332d73SSam Clegg       << Twine::utohexstr(Info.Flags);
42659932b0SJF Bastien   if (!isTypeData()) {
43659932b0SJF Bastien     Out << ", ElemIndex=" << Info.ElementIndex;
44659932b0SJF Bastien   } else if (isDefined()) {
45659932b0SJF Bastien     Out << ", Segment=" << Info.DataRef.Segment;
46659932b0SJF Bastien     Out << ", Offset=" << Info.DataRef.Offset;
47659932b0SJF Bastien     Out << ", Size=" << Info.DataRef.Size;
48659932b0SJF Bastien   }
49659932b0SJF Bastien }
50659932b0SJF Bastien 
51ddc84bf7SJF Bastien #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
dump() const52ddc84bf7SJF Bastien LLVM_DUMP_METHOD void WasmSymbol::dump() const { print(dbgs()); }
53ddc84bf7SJF Bastien #endif
54ddc84bf7SJF Bastien 
552c6f75ddSDerek Schuff Expected<std::unique_ptr<WasmObjectFile>>
createWasmObjectFile(MemoryBufferRef Buffer)562c6f75ddSDerek Schuff ObjectFile::createWasmObjectFile(MemoryBufferRef Buffer) {
572c6f75ddSDerek Schuff   Error Err = Error::success();
580eaee545SJonas Devlieghere   auto ObjectFile = std::make_unique<WasmObjectFile>(Buffer, Err);
592c6f75ddSDerek Schuff   if (Err)
60c55cf4afSBill Wendling     return std::move(Err);
612c6f75ddSDerek Schuff 
62c55cf4afSBill Wendling   return std::move(ObjectFile);
632c6f75ddSDerek Schuff }
642c6f75ddSDerek Schuff 
65d3d84fddSDerek Schuff #define VARINT7_MAX ((1 << 7) - 1)
66d3d84fddSDerek Schuff #define VARINT7_MIN (-(1 << 7))
67d3d84fddSDerek Schuff #define VARUINT7_MAX (1 << 7)
68d3d84fddSDerek Schuff #define VARUINT1_MAX (1)
69d3d84fddSDerek Schuff 
readUint8(WasmObjectFile::ReadContext & Ctx)70b7c62394SSam Clegg static uint8_t readUint8(WasmObjectFile::ReadContext &Ctx) {
71b7c62394SSam Clegg   if (Ctx.Ptr == Ctx.End)
72b7c62394SSam Clegg     report_fatal_error("EOF while reading uint8");
73b7c62394SSam Clegg   return *Ctx.Ptr++;
74b7c62394SSam Clegg }
75d3d84fddSDerek Schuff 
readUint32(WasmObjectFile::ReadContext & Ctx)76b7c62394SSam Clegg static uint32_t readUint32(WasmObjectFile::ReadContext &Ctx) {
77b7c62394SSam Clegg   if (Ctx.Ptr + 4 > Ctx.End)
78b7c62394SSam Clegg     report_fatal_error("EOF while reading uint32");
79b7c62394SSam Clegg   uint32_t Result = support::endian::read32le(Ctx.Ptr);
80b7c62394SSam Clegg   Ctx.Ptr += 4;
812c6f75ddSDerek Schuff   return Result;
822c6f75ddSDerek Schuff }
832c6f75ddSDerek Schuff 
readFloat32(WasmObjectFile::ReadContext & Ctx)84b7c62394SSam Clegg static int32_t readFloat32(WasmObjectFile::ReadContext &Ctx) {
85e3d6b978SJonas Devlieghere   if (Ctx.Ptr + 4 > Ctx.End)
86e3d6b978SJonas Devlieghere     report_fatal_error("EOF while reading float64");
87d3d84fddSDerek Schuff   int32_t Result = 0;
88b7c62394SSam Clegg   memcpy(&Result, Ctx.Ptr, sizeof(Result));
89b7c62394SSam Clegg   Ctx.Ptr += sizeof(Result);
90d3d84fddSDerek Schuff   return Result;
91d3d84fddSDerek Schuff }
92d3d84fddSDerek Schuff 
readFloat64(WasmObjectFile::ReadContext & Ctx)93b7c62394SSam Clegg static int64_t readFloat64(WasmObjectFile::ReadContext &Ctx) {
94e3d6b978SJonas Devlieghere   if (Ctx.Ptr + 8 > Ctx.End)
95e3d6b978SJonas Devlieghere     report_fatal_error("EOF while reading float64");
96d3d84fddSDerek Schuff   int64_t Result = 0;
97b7c62394SSam Clegg   memcpy(&Result, Ctx.Ptr, sizeof(Result));
98b7c62394SSam Clegg   Ctx.Ptr += sizeof(Result);
99d3d84fddSDerek Schuff   return Result;
100d3d84fddSDerek Schuff }
101d3d84fddSDerek Schuff 
readULEB128(WasmObjectFile::ReadContext & Ctx)102b7c62394SSam Clegg static uint64_t readULEB128(WasmObjectFile::ReadContext &Ctx) {
1032c6f75ddSDerek Schuff   unsigned Count;
104b7c62394SSam Clegg   const char *Error = nullptr;
105b7c62394SSam Clegg   uint64_t Result = decodeULEB128(Ctx.Ptr, &Count, Ctx.End, &Error);
106b7c62394SSam Clegg   if (Error)
107b7c62394SSam Clegg     report_fatal_error(Error);
108b7c62394SSam Clegg   Ctx.Ptr += Count;
1092c6f75ddSDerek Schuff   return Result;
1102c6f75ddSDerek Schuff }
1112c6f75ddSDerek Schuff 
readString(WasmObjectFile::ReadContext & Ctx)112b7c62394SSam Clegg static StringRef readString(WasmObjectFile::ReadContext &Ctx) {
113b7c62394SSam Clegg   uint32_t StringLen = readULEB128(Ctx);
114b7c62394SSam Clegg   if (Ctx.Ptr + StringLen > Ctx.End)
115b7c62394SSam Clegg     report_fatal_error("EOF while reading string");
116b7c62394SSam Clegg   StringRef Return =
117b7c62394SSam Clegg       StringRef(reinterpret_cast<const char *>(Ctx.Ptr), StringLen);
118b7c62394SSam Clegg   Ctx.Ptr += StringLen;
1192c6f75ddSDerek Schuff   return Return;
1202c6f75ddSDerek Schuff }
1212c6f75ddSDerek Schuff 
readLEB128(WasmObjectFile::ReadContext & Ctx)122b7c62394SSam Clegg static int64_t readLEB128(WasmObjectFile::ReadContext &Ctx) {
123d3d84fddSDerek Schuff   unsigned Count;
124b7c62394SSam Clegg   const char *Error = nullptr;
125b7c62394SSam Clegg   uint64_t Result = decodeSLEB128(Ctx.Ptr, &Count, Ctx.End, &Error);
126b7c62394SSam Clegg   if (Error)
127b7c62394SSam Clegg     report_fatal_error(Error);
128b7c62394SSam Clegg   Ctx.Ptr += Count;
129d3d84fddSDerek Schuff   return Result;
130d3d84fddSDerek Schuff }
131d3d84fddSDerek Schuff 
readVaruint1(WasmObjectFile::ReadContext & Ctx)132b7c62394SSam Clegg static uint8_t readVaruint1(WasmObjectFile::ReadContext &Ctx) {
13318c56a07SHeejin Ahn   int64_t Result = readLEB128(Ctx);
13418c56a07SHeejin Ahn   if (Result > VARUINT1_MAX || Result < 0)
1354bbc6b55SSam Clegg     report_fatal_error("LEB is outside Varuint1 range");
13618c56a07SHeejin Ahn   return Result;
137d3d84fddSDerek Schuff }
138d3d84fddSDerek Schuff 
readVarint32(WasmObjectFile::ReadContext & Ctx)139b7c62394SSam Clegg static int32_t readVarint32(WasmObjectFile::ReadContext &Ctx) {
14018c56a07SHeejin Ahn   int64_t Result = readLEB128(Ctx);
14118c56a07SHeejin Ahn   if (Result > INT32_MAX || Result < INT32_MIN)
1424bbc6b55SSam Clegg     report_fatal_error("LEB is outside Varint32 range");
14318c56a07SHeejin Ahn   return Result;
144d3d84fddSDerek Schuff }
145d3d84fddSDerek Schuff 
readVaruint32(WasmObjectFile::ReadContext & Ctx)146b7c62394SSam Clegg static uint32_t readVaruint32(WasmObjectFile::ReadContext &Ctx) {
14718c56a07SHeejin Ahn   uint64_t Result = readULEB128(Ctx);
14818c56a07SHeejin Ahn   if (Result > UINT32_MAX)
1494bbc6b55SSam Clegg     report_fatal_error("LEB is outside Varuint32 range");
15018c56a07SHeejin Ahn   return Result;
151d3d84fddSDerek Schuff }
152d3d84fddSDerek Schuff 
readVarint64(WasmObjectFile::ReadContext & Ctx)153b7c62394SSam Clegg static int64_t readVarint64(WasmObjectFile::ReadContext &Ctx) {
154b7c62394SSam Clegg   return readLEB128(Ctx);
155d3d84fddSDerek Schuff }
156d3d84fddSDerek Schuff 
readVaruint64(WasmObjectFile::ReadContext & Ctx)1573b29376eSWouter van Oortmerssen static uint64_t readVaruint64(WasmObjectFile::ReadContext &Ctx) {
1583b29376eSWouter van Oortmerssen   return readULEB128(Ctx);
1593b29376eSWouter van Oortmerssen }
1603b29376eSWouter van Oortmerssen 
readOpcode(WasmObjectFile::ReadContext & Ctx)161b7c62394SSam Clegg static uint8_t readOpcode(WasmObjectFile::ReadContext &Ctx) {
162b7c62394SSam Clegg   return readUint8(Ctx);
163d3d84fddSDerek Schuff }
164d3d84fddSDerek Schuff 
readInitExpr(wasm::WasmInitExpr & Expr,WasmObjectFile::ReadContext & Ctx)165b7c62394SSam Clegg static Error readInitExpr(wasm::WasmInitExpr &Expr,
166b7c62394SSam Clegg                           WasmObjectFile::ReadContext &Ctx) {
1679504ab32SSam Clegg   auto Start = Ctx.Ptr;
168d3d84fddSDerek Schuff 
1699504ab32SSam Clegg   Expr.Extended = false;
1709504ab32SSam Clegg   Expr.Inst.Opcode = readOpcode(Ctx);
1719504ab32SSam Clegg   switch (Expr.Inst.Opcode) {
172d3d84fddSDerek Schuff   case wasm::WASM_OPCODE_I32_CONST:
1739504ab32SSam Clegg     Expr.Inst.Value.Int32 = readVarint32(Ctx);
174d3d84fddSDerek Schuff     break;
175d3d84fddSDerek Schuff   case wasm::WASM_OPCODE_I64_CONST:
1769504ab32SSam Clegg     Expr.Inst.Value.Int64 = readVarint64(Ctx);
177d3d84fddSDerek Schuff     break;
178d3d84fddSDerek Schuff   case wasm::WASM_OPCODE_F32_CONST:
1799504ab32SSam Clegg     Expr.Inst.Value.Float32 = readFloat32(Ctx);
180d3d84fddSDerek Schuff     break;
181d3d84fddSDerek Schuff   case wasm::WASM_OPCODE_F64_CONST:
1829504ab32SSam Clegg     Expr.Inst.Value.Float64 = readFloat64(Ctx);
183d3d84fddSDerek Schuff     break;
1846a87ddacSThomas Lively   case wasm::WASM_OPCODE_GLOBAL_GET:
1859504ab32SSam Clegg     Expr.Inst.Value.Global = readULEB128(Ctx);
186d3d84fddSDerek Schuff     break;
18779aad89dSSam Clegg   case wasm::WASM_OPCODE_REF_NULL: {
18879aad89dSSam Clegg     wasm::ValType Ty = static_cast<wasm::ValType>(readULEB128(Ctx));
18979aad89dSSam Clegg     if (Ty != wasm::ValType::EXTERNREF) {
19084c6f325SSam Clegg       return make_error<GenericBinaryError>("invalid type for ref.null",
19179aad89dSSam Clegg                                             object_error::parse_failed);
19279aad89dSSam Clegg     }
19379aad89dSSam Clegg     break;
19479aad89dSSam Clegg   }
195d3d84fddSDerek Schuff   default:
1969504ab32SSam Clegg     Expr.Extended = true;
197d3d84fddSDerek Schuff   }
198d3d84fddSDerek Schuff 
1999504ab32SSam Clegg   if (!Expr.Extended) {
200b7c62394SSam Clegg     uint8_t EndOpcode = readOpcode(Ctx);
2019504ab32SSam Clegg     if (EndOpcode != wasm::WASM_OPCODE_END)
2029504ab32SSam Clegg       Expr.Extended = true;
2039504ab32SSam Clegg   }
2049504ab32SSam Clegg 
2059504ab32SSam Clegg   if (Expr.Extended) {
2069504ab32SSam Clegg     Ctx.Ptr = Start;
207*7094ab4eSKazu Hirata     while (true) {
2089504ab32SSam Clegg       uint8_t Opcode = readOpcode(Ctx);
2099504ab32SSam Clegg       switch (Opcode) {
2109504ab32SSam Clegg       case wasm::WASM_OPCODE_I32_CONST:
2119504ab32SSam Clegg       case wasm::WASM_OPCODE_GLOBAL_GET:
2129504ab32SSam Clegg       case wasm::WASM_OPCODE_REF_NULL:
2139504ab32SSam Clegg       case wasm::WASM_OPCODE_I64_CONST:
2149504ab32SSam Clegg       case wasm::WASM_OPCODE_F32_CONST:
2159504ab32SSam Clegg       case wasm::WASM_OPCODE_F64_CONST:
2169504ab32SSam Clegg         readULEB128(Ctx);
2179504ab32SSam Clegg         break;
2189504ab32SSam Clegg       case wasm::WASM_OPCODE_I32_ADD:
2199504ab32SSam Clegg       case wasm::WASM_OPCODE_I32_SUB:
2209504ab32SSam Clegg       case wasm::WASM_OPCODE_I32_MUL:
2219504ab32SSam Clegg       case wasm::WASM_OPCODE_I64_ADD:
2229504ab32SSam Clegg       case wasm::WASM_OPCODE_I64_SUB:
2239504ab32SSam Clegg       case wasm::WASM_OPCODE_I64_MUL:
2249504ab32SSam Clegg         break;
2259504ab32SSam Clegg       case wasm::WASM_OPCODE_END:
2269504ab32SSam Clegg         Expr.Body = ArrayRef<uint8_t>(Start, Ctx.Ptr - Start);
2279504ab32SSam Clegg         return Error::success();
2289504ab32SSam Clegg       default:
2299504ab32SSam Clegg         return make_error<GenericBinaryError>(
2309504ab32SSam Clegg             Twine("invalid opcode in init_expr: ") + Twine(unsigned(Opcode)),
231d3d84fddSDerek Schuff             object_error::parse_failed);
232d3d84fddSDerek Schuff       }
2339504ab32SSam Clegg     }
2349504ab32SSam Clegg   }
2359504ab32SSam Clegg 
236d3d84fddSDerek Schuff   return Error::success();
237d3d84fddSDerek Schuff }
238d3d84fddSDerek Schuff 
readLimits(WasmObjectFile::ReadContext & Ctx)239b7c62394SSam Clegg static wasm::WasmLimits readLimits(WasmObjectFile::ReadContext &Ctx) {
240d3d84fddSDerek Schuff   wasm::WasmLimits Result;
24168818062SDerek Schuff   Result.Flags = readVaruint32(Ctx);
2429ac5620cSAndy Wingo   Result.Minimum = readVaruint64(Ctx);
243d3d84fddSDerek Schuff   if (Result.Flags & wasm::WASM_LIMITS_FLAG_HAS_MAX)
2444d135b04SWouter van Oortmerssen     Result.Maximum = readVaruint64(Ctx);
245d3d84fddSDerek Schuff   return Result;
246d3d84fddSDerek Schuff }
247d3d84fddSDerek Schuff 
readTableType(WasmObjectFile::ReadContext & Ctx)2481933c9d4SAndy Wingo static wasm::WasmTableType readTableType(WasmObjectFile::ReadContext &Ctx) {
2491933c9d4SAndy Wingo   wasm::WasmTableType TableType;
2501933c9d4SAndy Wingo   TableType.ElemType = readUint8(Ctx);
2511933c9d4SAndy Wingo   TableType.Limits = readLimits(Ctx);
2521933c9d4SAndy Wingo   return TableType;
2532ffff5afSSam Clegg }
2542ffff5afSSam Clegg 
readSection(WasmSection & Section,WasmObjectFile::ReadContext & Ctx,WasmSectionOrderChecker & Checker)255feef720bSHeejin Ahn static Error readSection(WasmSection &Section, WasmObjectFile::ReadContext &Ctx,
256feef720bSHeejin Ahn                          WasmSectionOrderChecker &Checker) {
257b7c62394SSam Clegg   Section.Offset = Ctx.Ptr - Ctx.Start;
258b7c62394SSam Clegg   Section.Type = readUint8(Ctx);
259f4f37509SSam Clegg   LLVM_DEBUG(dbgs() << "readSection type=" << Section.Type << "\n");
260b7c62394SSam Clegg   uint32_t Size = readVaruint32(Ctx);
2612c6f75ddSDerek Schuff   if (Size == 0)
26284c6f325SSam Clegg     return make_error<StringError>("zero length section",
2632c6f75ddSDerek Schuff                                    object_error::parse_failed);
264b7c62394SSam Clegg   if (Ctx.Ptr + Size > Ctx.End)
26584c6f325SSam Clegg     return make_error<StringError>("section too large",
26635b50a83SVedant Kumar                                    object_error::parse_failed);
2679745afa6SSam Clegg   if (Section.Type == wasm::WASM_SEC_CUSTOM) {
2688511777dSJonas Devlieghere     WasmObjectFile::ReadContext SectionCtx;
2698511777dSJonas Devlieghere     SectionCtx.Start = Ctx.Ptr;
2708511777dSJonas Devlieghere     SectionCtx.Ptr = Ctx.Ptr;
2718511777dSJonas Devlieghere     SectionCtx.End = Ctx.Ptr + Size;
2728511777dSJonas Devlieghere 
2738511777dSJonas Devlieghere     Section.Name = readString(SectionCtx);
2748511777dSJonas Devlieghere 
2758511777dSJonas Devlieghere     uint32_t SectionNameSize = SectionCtx.Ptr - SectionCtx.Start;
2768511777dSJonas Devlieghere     Ctx.Ptr += SectionNameSize;
2778511777dSJonas Devlieghere     Size -= SectionNameSize;
2789745afa6SSam Clegg   }
279feef720bSHeejin Ahn 
280feef720bSHeejin Ahn   if (!Checker.isValidSectionOrder(Section.Type, Section.Name)) {
28184c6f325SSam Clegg     return make_error<StringError>("out of order section type: " +
282feef720bSHeejin Ahn                                        llvm::to_string(Section.Type),
283feef720bSHeejin Ahn                                    object_error::parse_failed);
284feef720bSHeejin Ahn   }
285feef720bSHeejin Ahn 
286b7c62394SSam Clegg   Section.Content = ArrayRef<uint8_t>(Ctx.Ptr, Size);
287b7c62394SSam Clegg   Ctx.Ptr += Size;
2882c6f75ddSDerek Schuff   return Error::success();
2892c6f75ddSDerek Schuff }
2902c6f75ddSDerek Schuff 
WasmObjectFile(MemoryBufferRef Buffer,Error & Err)2912c6f75ddSDerek Schuff WasmObjectFile::WasmObjectFile(MemoryBufferRef Buffer, Error &Err)
2929f5094dfSEugene Zelenko     : ObjectFile(Binary::ID_Wasm, Buffer) {
2932c6f75ddSDerek Schuff   ErrorAsOutParameter ErrAsOutParam(&Err);
2942c6f75ddSDerek Schuff   Header.Magic = getData().substr(0, 4);
2952c6f75ddSDerek Schuff   if (Header.Magic != StringRef("\0asm", 4)) {
29684c6f325SSam Clegg     Err = make_error<StringError>("invalid magic number",
29784c6f325SSam Clegg                                   object_error::parse_failed);
2982c6f75ddSDerek Schuff     return;
2992c6f75ddSDerek Schuff   }
300a845167dSJonas Devlieghere 
301b7c62394SSam Clegg   ReadContext Ctx;
3026a0746a9SFangrui Song   Ctx.Start = getData().bytes_begin();
303b7c62394SSam Clegg   Ctx.Ptr = Ctx.Start + 4;
304b7c62394SSam Clegg   Ctx.End = Ctx.Start + getData().size();
305a845167dSJonas Devlieghere 
306b7c62394SSam Clegg   if (Ctx.Ptr + 4 > Ctx.End) {
30784c6f325SSam Clegg     Err = make_error<StringError>("missing version number",
308a845167dSJonas Devlieghere                                   object_error::parse_failed);
309a845167dSJonas Devlieghere     return;
310a845167dSJonas Devlieghere   }
311a845167dSJonas Devlieghere 
312b7c62394SSam Clegg   Header.Version = readUint32(Ctx);
3132c6f75ddSDerek Schuff   if (Header.Version != wasm::WasmVersion) {
31484c6f325SSam Clegg     Err = make_error<StringError>("invalid version number: " +
31584c6f325SSam Clegg                                       Twine(Header.Version),
3162c6f75ddSDerek Schuff                                   object_error::parse_failed);
3172c6f75ddSDerek Schuff     return;
3182c6f75ddSDerek Schuff   }
3192c6f75ddSDerek Schuff 
320feef720bSHeejin Ahn   WasmSectionOrderChecker Checker;
321b7c62394SSam Clegg   while (Ctx.Ptr < Ctx.End) {
322e4b2f305SSam Clegg     WasmSection Sec;
323feef720bSHeejin Ahn     if ((Err = readSection(Sec, Ctx, Checker)))
3242c6f75ddSDerek Schuff       return;
325d3d84fddSDerek Schuff     if ((Err = parseSection(Sec)))
3262c6f75ddSDerek Schuff       return;
327d3d84fddSDerek Schuff 
3282c6f75ddSDerek Schuff     Sections.push_back(Sec);
3292c6f75ddSDerek Schuff   }
3302c6f75ddSDerek Schuff }
3312c6f75ddSDerek Schuff 
parseSection(WasmSection & Sec)332d3d84fddSDerek Schuff Error WasmObjectFile::parseSection(WasmSection &Sec) {
333b7c62394SSam Clegg   ReadContext Ctx;
334b7c62394SSam Clegg   Ctx.Start = Sec.Content.data();
335b7c62394SSam Clegg   Ctx.End = Ctx.Start + Sec.Content.size();
336b7c62394SSam Clegg   Ctx.Ptr = Ctx.Start;
337d3d84fddSDerek Schuff   switch (Sec.Type) {
338d3d84fddSDerek Schuff   case wasm::WASM_SEC_CUSTOM:
339b7c62394SSam Clegg     return parseCustomSection(Sec, Ctx);
340d3d84fddSDerek Schuff   case wasm::WASM_SEC_TYPE:
341b7c62394SSam Clegg     return parseTypeSection(Ctx);
342d3d84fddSDerek Schuff   case wasm::WASM_SEC_IMPORT:
343b7c62394SSam Clegg     return parseImportSection(Ctx);
344d3d84fddSDerek Schuff   case wasm::WASM_SEC_FUNCTION:
345b7c62394SSam Clegg     return parseFunctionSection(Ctx);
346d3d84fddSDerek Schuff   case wasm::WASM_SEC_TABLE:
347b7c62394SSam Clegg     return parseTableSection(Ctx);
348d3d84fddSDerek Schuff   case wasm::WASM_SEC_MEMORY:
349b7c62394SSam Clegg     return parseMemorySection(Ctx);
3501d891d44SHeejin Ahn   case wasm::WASM_SEC_TAG:
3511d891d44SHeejin Ahn     return parseTagSection(Ctx);
352f93426c5SHeejin Ahn   case wasm::WASM_SEC_GLOBAL:
353f93426c5SHeejin Ahn     return parseGlobalSection(Ctx);
354d3d84fddSDerek Schuff   case wasm::WASM_SEC_EXPORT:
355b7c62394SSam Clegg     return parseExportSection(Ctx);
356d3d84fddSDerek Schuff   case wasm::WASM_SEC_START:
357b7c62394SSam Clegg     return parseStartSection(Ctx);
358d3d84fddSDerek Schuff   case wasm::WASM_SEC_ELEM:
359b7c62394SSam Clegg     return parseElemSection(Ctx);
360d3d84fddSDerek Schuff   case wasm::WASM_SEC_CODE:
361b7c62394SSam Clegg     return parseCodeSection(Ctx);
362d3d84fddSDerek Schuff   case wasm::WASM_SEC_DATA:
363b7c62394SSam Clegg     return parseDataSection(Ctx);
364fef8de66SThomas Lively   case wasm::WASM_SEC_DATACOUNT:
365fef8de66SThomas Lively     return parseDataCountSection(Ctx);
366d3d84fddSDerek Schuff   default:
367fe8aabf9SSam Clegg     return make_error<GenericBinaryError>(
36884c6f325SSam Clegg         "invalid section type: " + Twine(Sec.Type), object_error::parse_failed);
369d3d84fddSDerek Schuff   }
370d3d84fddSDerek Schuff }
371d3d84fddSDerek Schuff 
parseDylinkSection(ReadContext & Ctx)372e4afbc68SSam Clegg Error WasmObjectFile::parseDylinkSection(ReadContext &Ctx) {
373b78c85a4SSam Clegg   // Legacy "dylink" section support.
374b78c85a4SSam Clegg   // See parseDylink0Section for the current "dylink.0" section parsing.
375a688a42cSSam Clegg   HasDylinkSection = true;
376e4afbc68SSam Clegg   DylinkInfo.MemorySize = readVaruint32(Ctx);
377e4afbc68SSam Clegg   DylinkInfo.MemoryAlignment = readVaruint32(Ctx);
378e4afbc68SSam Clegg   DylinkInfo.TableSize = readVaruint32(Ctx);
379e4afbc68SSam Clegg   DylinkInfo.TableAlignment = readVaruint32(Ctx);
38003801256SSam Clegg   uint32_t Count = readVaruint32(Ctx);
38103801256SSam Clegg   while (Count--) {
38203801256SSam Clegg     DylinkInfo.Needed.push_back(readString(Ctx));
38303801256SSam Clegg   }
384b78c85a4SSam Clegg 
385e4afbc68SSam Clegg   if (Ctx.Ptr != Ctx.End)
386e4afbc68SSam Clegg     return make_error<GenericBinaryError>("dylink section ended prematurely",
387e4afbc68SSam Clegg                                           object_error::parse_failed);
388e4afbc68SSam Clegg   return Error::success();
389e4afbc68SSam Clegg }
390e4afbc68SSam Clegg 
parseDylink0Section(ReadContext & Ctx)391b78c85a4SSam Clegg Error WasmObjectFile::parseDylink0Section(ReadContext &Ctx) {
392b78c85a4SSam Clegg   // See
393c71fbdd8SQuinn Pham   // https://github.com/WebAssembly/tool-conventions/blob/main/DynamicLinking.md
394b78c85a4SSam Clegg   HasDylinkSection = true;
395b78c85a4SSam Clegg 
396b78c85a4SSam Clegg   const uint8_t *OrigEnd = Ctx.End;
397b78c85a4SSam Clegg   while (Ctx.Ptr < OrigEnd) {
398b78c85a4SSam Clegg     Ctx.End = OrigEnd;
399b78c85a4SSam Clegg     uint8_t Type = readUint8(Ctx);
400b78c85a4SSam Clegg     uint32_t Size = readVaruint32(Ctx);
401b78c85a4SSam Clegg     LLVM_DEBUG(dbgs() << "readSubsection type=" << int(Type) << " size=" << Size
402b78c85a4SSam Clegg                       << "\n");
403b78c85a4SSam Clegg     Ctx.End = Ctx.Ptr + Size;
404b78c85a4SSam Clegg     uint32_t Count;
405b78c85a4SSam Clegg     switch (Type) {
406b78c85a4SSam Clegg     case wasm::WASM_DYLINK_MEM_INFO:
407b78c85a4SSam Clegg       DylinkInfo.MemorySize = readVaruint32(Ctx);
408b78c85a4SSam Clegg       DylinkInfo.MemoryAlignment = readVaruint32(Ctx);
409b78c85a4SSam Clegg       DylinkInfo.TableSize = readVaruint32(Ctx);
410b78c85a4SSam Clegg       DylinkInfo.TableAlignment = readVaruint32(Ctx);
411b78c85a4SSam Clegg       break;
412b78c85a4SSam Clegg     case wasm::WASM_DYLINK_NEEDED:
413b78c85a4SSam Clegg       Count = readVaruint32(Ctx);
414b78c85a4SSam Clegg       while (Count--) {
415b78c85a4SSam Clegg         DylinkInfo.Needed.push_back(readString(Ctx));
416b78c85a4SSam Clegg       }
417b78c85a4SSam Clegg       break;
418ef8c9135SSam Clegg     case wasm::WASM_DYLINK_EXPORT_INFO: {
419ef8c9135SSam Clegg       uint32_t Count = readVaruint32(Ctx);
420ef8c9135SSam Clegg       while (Count--) {
421ef8c9135SSam Clegg         DylinkInfo.ExportInfo.push_back({readString(Ctx), readVaruint32(Ctx)});
422ef8c9135SSam Clegg       }
423ef8c9135SSam Clegg       break;
424ef8c9135SSam Clegg     }
425659a0839SSam Clegg     case wasm::WASM_DYLINK_IMPORT_INFO: {
426659a0839SSam Clegg       uint32_t Count = readVaruint32(Ctx);
427659a0839SSam Clegg       while (Count--) {
428659a0839SSam Clegg         DylinkInfo.ImportInfo.push_back(
429659a0839SSam Clegg             {readString(Ctx), readString(Ctx), readVaruint32(Ctx)});
430659a0839SSam Clegg       }
431659a0839SSam Clegg       break;
432659a0839SSam Clegg     }
433b78c85a4SSam Clegg     default:
434210cbcf4SSam Clegg       LLVM_DEBUG(dbgs() << "unknown dylink.0 sub-section: " << Type << "\n");
435b78c85a4SSam Clegg       Ctx.Ptr += Size;
436b78c85a4SSam Clegg       break;
437b78c85a4SSam Clegg     }
438b78c85a4SSam Clegg     if (Ctx.Ptr != Ctx.End) {
439b78c85a4SSam Clegg       return make_error<GenericBinaryError>(
440b78c85a4SSam Clegg           "dylink.0 sub-section ended prematurely", object_error::parse_failed);
441b78c85a4SSam Clegg     }
442b78c85a4SSam Clegg   }
443b78c85a4SSam Clegg 
444b78c85a4SSam Clegg   if (Ctx.Ptr != Ctx.End)
445b78c85a4SSam Clegg     return make_error<GenericBinaryError>("dylink.0 section ended prematurely",
446b78c85a4SSam Clegg                                           object_error::parse_failed);
447b78c85a4SSam Clegg   return Error::success();
448b78c85a4SSam Clegg }
449b78c85a4SSam Clegg 
parseNameSection(ReadContext & Ctx)450b7c62394SSam Clegg Error WasmObjectFile::parseNameSection(ReadContext &Ctx) {
4511827005cSSam Clegg   llvm::DenseSet<uint64_t> SeenFunctions;
4521827005cSSam Clegg   llvm::DenseSet<uint64_t> SeenGlobals;
4539a72d3e3SSam Clegg   llvm::DenseSet<uint64_t> SeenSegments;
4544710ed7aSSam Clegg 
455b7c62394SSam Clegg   while (Ctx.Ptr < Ctx.End) {
456b7c62394SSam Clegg     uint8_t Type = readUint8(Ctx);
457b7c62394SSam Clegg     uint32_t Size = readVaruint32(Ctx);
458b7c62394SSam Clegg     const uint8_t *SubSectionEnd = Ctx.Ptr + Size;
459d3d84fddSDerek Schuff     switch (Type) {
4601827005cSSam Clegg     case wasm::WASM_NAMES_FUNCTION:
4619a72d3e3SSam Clegg     case wasm::WASM_NAMES_GLOBAL:
4629a72d3e3SSam Clegg     case wasm::WASM_NAMES_DATA_SEGMENT: {
463b7c62394SSam Clegg       uint32_t Count = readVaruint32(Ctx);
464d3d84fddSDerek Schuff       while (Count--) {
465b7c62394SSam Clegg         uint32_t Index = readVaruint32(Ctx);
466b7c62394SSam Clegg         StringRef Name = readString(Ctx);
4679a72d3e3SSam Clegg         wasm::NameType nameType = wasm::NameType::FUNCTION;
4681827005cSSam Clegg         if (Type == wasm::WASM_NAMES_FUNCTION) {
4691827005cSSam Clegg           if (!SeenFunctions.insert(Index).second)
4701827005cSSam Clegg             return make_error<GenericBinaryError>(
47184c6f325SSam Clegg                 "function named more than once", object_error::parse_failed);
4729f3fe42eSSam Clegg           if (!isValidFunctionIndex(Index) || Name.empty())
4735a082d9cSDerek Schuff             return make_error<GenericBinaryError>("invalid function name entry",
4749f3fe42eSSam Clegg                                                   object_error::parse_failed);
4751827005cSSam Clegg 
476ef90ff36SNicholas Wilson           if (isDefinedFunctionIndex(Index))
477ef90ff36SNicholas Wilson             getDefinedFunction(Index).DebugName = Name;
4789a72d3e3SSam Clegg         } else if (Type == wasm::WASM_NAMES_GLOBAL) {
4799a72d3e3SSam Clegg           nameType = wasm::NameType::GLOBAL;
4801827005cSSam Clegg           if (!SeenGlobals.insert(Index).second)
48184c6f325SSam Clegg             return make_error<GenericBinaryError>("global named more than once",
4821827005cSSam Clegg                                                   object_error::parse_failed);
4831827005cSSam Clegg           if (!isValidGlobalIndex(Index) || Name.empty())
4845a082d9cSDerek Schuff             return make_error<GenericBinaryError>("invalid global name entry",
4851827005cSSam Clegg                                                   object_error::parse_failed);
4869a72d3e3SSam Clegg         } else {
4879a72d3e3SSam Clegg           nameType = wasm::NameType::DATA_SEGMENT;
4889a72d3e3SSam Clegg           if (!SeenSegments.insert(Index).second)
4899a72d3e3SSam Clegg             return make_error<GenericBinaryError>(
49084c6f325SSam Clegg                 "segment named more than once", object_error::parse_failed);
4919a72d3e3SSam Clegg           if (Index > DataSegments.size())
4925a082d9cSDerek Schuff             return make_error<GenericBinaryError>("invalid data segment name entry",
4939a72d3e3SSam Clegg                                                   object_error::parse_failed);
4941827005cSSam Clegg         }
4959a72d3e3SSam Clegg         DebugNames.push_back(wasm::WasmDebugName{nameType, Index, Name});
496d3d84fddSDerek Schuff       }
497d3d84fddSDerek Schuff       break;
498d3d84fddSDerek Schuff     }
499d3d84fddSDerek Schuff     // Ignore local names for now
500d3d84fddSDerek Schuff     case wasm::WASM_NAMES_LOCAL:
501d3d84fddSDerek Schuff     default:
502b7c62394SSam Clegg       Ctx.Ptr += Size;
503d3d84fddSDerek Schuff       break;
504d3d84fddSDerek Schuff     }
505b7c62394SSam Clegg     if (Ctx.Ptr != SubSectionEnd)
506f208f631SHeejin Ahn       return make_error<GenericBinaryError>(
50784c6f325SSam Clegg           "name sub-section ended prematurely", object_error::parse_failed);
508d3d84fddSDerek Schuff   }
509d3d84fddSDerek Schuff 
510b7c62394SSam Clegg   if (Ctx.Ptr != Ctx.End)
51184c6f325SSam Clegg     return make_error<GenericBinaryError>("name section ended prematurely",
512d3d84fddSDerek Schuff                                           object_error::parse_failed);
513d3d84fddSDerek Schuff   return Error::success();
514d3d84fddSDerek Schuff }
515d3d84fddSDerek Schuff 
parseLinkingSection(ReadContext & Ctx)516b7c62394SSam Clegg Error WasmObjectFile::parseLinkingSection(ReadContext &Ctx) {
5174df5d764SSam Clegg   HasLinkingSection = true;
518e7a60708SSam Clegg 
519b7c62394SSam Clegg   LinkingData.Version = readVaruint32(Ctx);
5206bb5a41fSSam Clegg   if (LinkingData.Version != wasm::WasmMetadataVersion) {
5216bb5a41fSSam Clegg     return make_error<GenericBinaryError>(
52284c6f325SSam Clegg         "unexpected metadata version: " + Twine(LinkingData.Version) +
5236bb5a41fSSam Clegg             " (Expected: " + Twine(wasm::WasmMetadataVersion) + ")",
5246bb5a41fSSam Clegg         object_error::parse_failed);
5256bb5a41fSSam Clegg   }
5266bb5a41fSSam Clegg 
527b7c62394SSam Clegg   const uint8_t *OrigEnd = Ctx.End;
528b7c62394SSam Clegg   while (Ctx.Ptr < OrigEnd) {
529b7c62394SSam Clegg     Ctx.End = OrigEnd;
530b7c62394SSam Clegg     uint8_t Type = readUint8(Ctx);
531b7c62394SSam Clegg     uint32_t Size = readVaruint32(Ctx);
532f4f37509SSam Clegg     LLVM_DEBUG(dbgs() << "readSubsection type=" << int(Type) << " size=" << Size
533f4f37509SSam Clegg                       << "\n");
534b7c62394SSam Clegg     Ctx.End = Ctx.Ptr + Size;
535b7787fd0SSam Clegg     switch (Type) {
5366c899ba6SSam Clegg     case wasm::WASM_SYMBOL_TABLE:
537b7c62394SSam Clegg       if (Error Err = parseLinkingSectionSymtab(Ctx))
5386c899ba6SSam Clegg         return Err;
539b7787fd0SSam Clegg       break;
54063ebb813SSam Clegg     case wasm::WASM_SEGMENT_INFO: {
541b7c62394SSam Clegg       uint32_t Count = readVaruint32(Ctx);
542d95ed959SSam Clegg       if (Count > DataSegments.size())
54384c6f325SSam Clegg         return make_error<GenericBinaryError>("too many segment names",
544d95ed959SSam Clegg                                               object_error::parse_failed);
54518c56a07SHeejin Ahn       for (uint32_t I = 0; I < Count; I++) {
54618c56a07SHeejin Ahn         DataSegments[I].Data.Name = readString(Ctx);
54718c56a07SHeejin Ahn         DataSegments[I].Data.Alignment = readVaruint32(Ctx);
5483b8d2be5SSam Clegg         DataSegments[I].Data.LinkingFlags = readVaruint32(Ctx);
54963ebb813SSam Clegg       }
550d95ed959SSam Clegg       break;
551d95ed959SSam Clegg     }
5524273998cSSam Clegg     case wasm::WASM_INIT_FUNCS: {
553b7c62394SSam Clegg       uint32_t Count = readVaruint32(Ctx);
5544273998cSSam Clegg       LinkingData.InitFunctions.reserve(Count);
55518c56a07SHeejin Ahn       for (uint32_t I = 0; I < Count; I++) {
5564273998cSSam Clegg         wasm::WasmInitFunc Init;
557b7c62394SSam Clegg         Init.Priority = readVaruint32(Ctx);
558b7c62394SSam Clegg         Init.Symbol = readVaruint32(Ctx);
559b3748f71SNicholas Wilson         if (!isValidFunctionSymbol(Init.Symbol))
56084c6f325SSam Clegg           return make_error<GenericBinaryError>("invalid function symbol: " +
5616c899ba6SSam Clegg                                                     Twine(Init.Symbol),
5624273998cSSam Clegg                                                 object_error::parse_failed);
5634273998cSSam Clegg         LinkingData.InitFunctions.emplace_back(Init);
5644273998cSSam Clegg       }
5654273998cSSam Clegg       break;
5664273998cSSam Clegg     }
567ea7caceeSSam Clegg     case wasm::WASM_COMDAT_INFO:
568b7c62394SSam Clegg       if (Error Err = parseLinkingSectionComdat(Ctx))
569ea7caceeSSam Clegg         return Err;
570ea7caceeSSam Clegg       break;
571b7787fd0SSam Clegg     default:
572b7c62394SSam Clegg       Ctx.Ptr += Size;
573b7787fd0SSam Clegg       break;
574b7787fd0SSam Clegg     }
575b7c62394SSam Clegg     if (Ctx.Ptr != Ctx.End)
576b7787fd0SSam Clegg       return make_error<GenericBinaryError>(
57784c6f325SSam Clegg           "linking sub-section ended prematurely", object_error::parse_failed);
578b7787fd0SSam Clegg   }
579b7c62394SSam Clegg   if (Ctx.Ptr != OrigEnd)
58084c6f325SSam Clegg     return make_error<GenericBinaryError>("linking section ended prematurely",
581b7787fd0SSam Clegg                                           object_error::parse_failed);
582b7787fd0SSam Clegg   return Error::success();
583b7787fd0SSam Clegg }
584b7787fd0SSam Clegg 
parseLinkingSectionSymtab(ReadContext & Ctx)585b7c62394SSam Clegg Error WasmObjectFile::parseLinkingSectionSymtab(ReadContext &Ctx) {
586b7c62394SSam Clegg   uint32_t Count = readVaruint32(Ctx);
5876c899ba6SSam Clegg   LinkingData.SymbolTable.reserve(Count);
5886c899ba6SSam Clegg   Symbols.reserve(Count);
5896c899ba6SSam Clegg   StringSet<> SymbolNames;
5906c899ba6SSam Clegg 
5916c899ba6SSam Clegg   std::vector<wasm::WasmImport *> ImportedGlobals;
5926c899ba6SSam Clegg   std::vector<wasm::WasmImport *> ImportedFunctions;
5931d891d44SHeejin Ahn   std::vector<wasm::WasmImport *> ImportedTags;
594feac819eSAndy Wingo   std::vector<wasm::WasmImport *> ImportedTables;
5956c899ba6SSam Clegg   ImportedGlobals.reserve(Imports.size());
5966c899ba6SSam Clegg   ImportedFunctions.reserve(Imports.size());
5971d891d44SHeejin Ahn   ImportedTags.reserve(Imports.size());
598feac819eSAndy Wingo   ImportedTables.reserve(Imports.size());
5996c899ba6SSam Clegg   for (auto &I : Imports) {
6006c899ba6SSam Clegg     if (I.Kind == wasm::WASM_EXTERNAL_FUNCTION)
6016c899ba6SSam Clegg       ImportedFunctions.emplace_back(&I);
6026c899ba6SSam Clegg     else if (I.Kind == wasm::WASM_EXTERNAL_GLOBAL)
6036c899ba6SSam Clegg       ImportedGlobals.emplace_back(&I);
6041d891d44SHeejin Ahn     else if (I.Kind == wasm::WASM_EXTERNAL_TAG)
6051d891d44SHeejin Ahn       ImportedTags.emplace_back(&I);
606feac819eSAndy Wingo     else if (I.Kind == wasm::WASM_EXTERNAL_TABLE)
607feac819eSAndy Wingo       ImportedTables.emplace_back(&I);
6086c899ba6SSam Clegg   }
6096c899ba6SSam Clegg 
6106c899ba6SSam Clegg   while (Count--) {
6116c899ba6SSam Clegg     wasm::WasmSymbolInfo Info;
612a2125b8dSHeejin Ahn     const wasm::WasmSignature *Signature = nullptr;
6136c899ba6SSam Clegg     const wasm::WasmGlobalType *GlobalType = nullptr;
61453e3b81fSAndy Wingo     const wasm::WasmTableType *TableType = nullptr;
6156c899ba6SSam Clegg 
616b7c62394SSam Clegg     Info.Kind = readUint8(Ctx);
617b7c62394SSam Clegg     Info.Flags = readVaruint32(Ctx);
6186c899ba6SSam Clegg     bool IsDefined = (Info.Flags & wasm::WASM_SYMBOL_UNDEFINED) == 0;
6196c899ba6SSam Clegg 
6206c899ba6SSam Clegg     switch (Info.Kind) {
6216c899ba6SSam Clegg     case wasm::WASM_SYMBOL_TYPE_FUNCTION:
622b7c62394SSam Clegg       Info.ElementIndex = readVaruint32(Ctx);
6236c899ba6SSam Clegg       if (!isValidFunctionIndex(Info.ElementIndex) ||
6246c899ba6SSam Clegg           IsDefined != isDefinedFunctionIndex(Info.ElementIndex))
6256c899ba6SSam Clegg         return make_error<GenericBinaryError>("invalid function symbol index",
6266c899ba6SSam Clegg                                               object_error::parse_failed);
6276c899ba6SSam Clegg       if (IsDefined) {
628b7c62394SSam Clegg         Info.Name = readString(Ctx);
6296c899ba6SSam Clegg         unsigned FuncIndex = Info.ElementIndex - NumImportedFunctions;
630959e7371SNicholas Wilson         wasm::WasmFunction &Function = Functions[FuncIndex];
631c0039de2SSam Clegg         Signature = &Signatures[Function.SigIndex];
632ef90ff36SNicholas Wilson         if (Function.SymbolName.empty())
633ef90ff36SNicholas Wilson           Function.SymbolName = Info.Name;
6346c899ba6SSam Clegg       } else {
6356c899ba6SSam Clegg         wasm::WasmImport &Import = *ImportedFunctions[Info.ElementIndex];
636bd481277SSam Clegg         if ((Info.Flags & wasm::WASM_SYMBOL_EXPLICIT_NAME) != 0) {
63729874ceaSDan Gohman           Info.Name = readString(Ctx);
63829874ceaSDan Gohman           Info.ImportName = Import.Field;
639bd481277SSam Clegg         } else {
640bd481277SSam Clegg           Info.Name = Import.Field;
641bd481277SSam Clegg         }
642bd481277SSam Clegg         Signature = &Signatures[Import.SigIndex];
643bd481277SSam Clegg         if (!Import.Module.empty()) {
64429874ceaSDan Gohman           Info.ImportModule = Import.Module;
6456c899ba6SSam Clegg         }
646bd481277SSam Clegg       }
6476c899ba6SSam Clegg       break;
6486c899ba6SSam Clegg 
6496c899ba6SSam Clegg     case wasm::WASM_SYMBOL_TYPE_GLOBAL:
650b7c62394SSam Clegg       Info.ElementIndex = readVaruint32(Ctx);
6516c899ba6SSam Clegg       if (!isValidGlobalIndex(Info.ElementIndex) ||
6526c899ba6SSam Clegg           IsDefined != isDefinedGlobalIndex(Info.ElementIndex))
6536c899ba6SSam Clegg         return make_error<GenericBinaryError>("invalid global symbol index",
6546c899ba6SSam Clegg                                               object_error::parse_failed);
655f208f631SHeejin Ahn       if (!IsDefined && (Info.Flags & wasm::WASM_SYMBOL_BINDING_MASK) ==
65615f349f7SNicholas Wilson                             wasm::WASM_SYMBOL_BINDING_WEAK)
65715f349f7SNicholas Wilson         return make_error<GenericBinaryError>("undefined weak global symbol",
65815f349f7SNicholas Wilson                                               object_error::parse_failed);
6596c899ba6SSam Clegg       if (IsDefined) {
660b7c62394SSam Clegg         Info.Name = readString(Ctx);
6616c899ba6SSam Clegg         unsigned GlobalIndex = Info.ElementIndex - NumImportedGlobals;
662959e7371SNicholas Wilson         wasm::WasmGlobal &Global = Globals[GlobalIndex];
663959e7371SNicholas Wilson         GlobalType = &Global.Type;
664ef90ff36SNicholas Wilson         if (Global.SymbolName.empty())
665ef90ff36SNicholas Wilson           Global.SymbolName = Info.Name;
6666c899ba6SSam Clegg       } else {
6676c899ba6SSam Clegg         wasm::WasmImport &Import = *ImportedGlobals[Info.ElementIndex];
668bd481277SSam Clegg         if ((Info.Flags & wasm::WASM_SYMBOL_EXPLICIT_NAME) != 0) {
66929874ceaSDan Gohman           Info.Name = readString(Ctx);
670bd481277SSam Clegg           Info.ImportName = Import.Field;
671bd481277SSam Clegg         } else {
6726c899ba6SSam Clegg           Info.Name = Import.Field;
673bd481277SSam Clegg         }
6746c899ba6SSam Clegg         GlobalType = &Import.Global;
675bd481277SSam Clegg         if (!Import.Module.empty()) {
67629874ceaSDan Gohman           Info.ImportModule = Import.Module;
6776c899ba6SSam Clegg         }
678bd481277SSam Clegg       }
6796c899ba6SSam Clegg       break;
6806c899ba6SSam Clegg 
681388fb67bSPaulo Matos     case wasm::WASM_SYMBOL_TYPE_TABLE:
682388fb67bSPaulo Matos       Info.ElementIndex = readVaruint32(Ctx);
683a5a3659dSAndy Wingo       if (!isValidTableNumber(Info.ElementIndex) ||
684a5a3659dSAndy Wingo           IsDefined != isDefinedTableNumber(Info.ElementIndex))
685388fb67bSPaulo Matos         return make_error<GenericBinaryError>("invalid table symbol index",
686388fb67bSPaulo Matos                                               object_error::parse_failed);
687388fb67bSPaulo Matos       if (!IsDefined && (Info.Flags & wasm::WASM_SYMBOL_BINDING_MASK) ==
688388fb67bSPaulo Matos                             wasm::WASM_SYMBOL_BINDING_WEAK)
689388fb67bSPaulo Matos         return make_error<GenericBinaryError>("undefined weak table symbol",
690388fb67bSPaulo Matos                                               object_error::parse_failed);
691388fb67bSPaulo Matos       if (IsDefined) {
692388fb67bSPaulo Matos         Info.Name = readString(Ctx);
693a5a3659dSAndy Wingo         unsigned TableNumber = Info.ElementIndex - NumImportedTables;
694a5a3659dSAndy Wingo         wasm::WasmTable &Table = Tables[TableNumber];
69553e3b81fSAndy Wingo         TableType = &Table.Type;
6961933c9d4SAndy Wingo         if (Table.SymbolName.empty())
6971933c9d4SAndy Wingo           Table.SymbolName = Info.Name;
698388fb67bSPaulo Matos       } else {
699feac819eSAndy Wingo         wasm::WasmImport &Import = *ImportedTables[Info.ElementIndex];
700feac819eSAndy Wingo         if ((Info.Flags & wasm::WASM_SYMBOL_EXPLICIT_NAME) != 0) {
701feac819eSAndy Wingo           Info.Name = readString(Ctx);
702feac819eSAndy Wingo           Info.ImportName = Import.Field;
703feac819eSAndy Wingo         } else {
704feac819eSAndy Wingo           Info.Name = Import.Field;
705feac819eSAndy Wingo         }
70653e3b81fSAndy Wingo         TableType = &Import.Table;
707feac819eSAndy Wingo         if (!Import.Module.empty()) {
708feac819eSAndy Wingo           Info.ImportModule = Import.Module;
709feac819eSAndy Wingo         }
710388fb67bSPaulo Matos       }
711388fb67bSPaulo Matos       break;
712388fb67bSPaulo Matos 
7136c899ba6SSam Clegg     case wasm::WASM_SYMBOL_TYPE_DATA:
714b7c62394SSam Clegg       Info.Name = readString(Ctx);
7156c899ba6SSam Clegg       if (IsDefined) {
7163b29376eSWouter van Oortmerssen         auto Index = readVaruint32(Ctx);
7176c899ba6SSam Clegg         if (Index >= DataSegments.size())
7186c899ba6SSam Clegg           return make_error<GenericBinaryError>("invalid data symbol index",
7196c899ba6SSam Clegg                                                 object_error::parse_failed);
7203b29376eSWouter van Oortmerssen         auto Offset = readVaruint64(Ctx);
7213b29376eSWouter van Oortmerssen         auto Size = readVaruint64(Ctx);
722cd01430fSSam Clegg         size_t SegmentSize = DataSegments[Index].Data.Content.size();
723cd01430fSSam Clegg         if (Offset > SegmentSize)
724cd01430fSSam Clegg           return make_error<GenericBinaryError>(
725cd01430fSSam Clegg               "invalid data symbol offset: `" + Info.Name + "` (offset: " +
726cd01430fSSam Clegg                   Twine(Offset) + " segment size: " + Twine(SegmentSize) + ")",
7276c899ba6SSam Clegg               object_error::parse_failed);
7286c899ba6SSam Clegg         Info.DataRef = wasm::WasmDataReference{Index, Offset, Size};
7296c899ba6SSam Clegg       }
7306c899ba6SSam Clegg       break;
7316c899ba6SSam Clegg 
7326a31a0d6SSam Clegg     case wasm::WASM_SYMBOL_TYPE_SECTION: {
733d5504a0aSSam Clegg       if ((Info.Flags & wasm::WASM_SYMBOL_BINDING_MASK) !=
734d5504a0aSSam Clegg           wasm::WASM_SYMBOL_BINDING_LOCAL)
735e0658119SSam Clegg         return make_error<GenericBinaryError>(
73684c6f325SSam Clegg             "section symbols must have local binding",
737d5504a0aSSam Clegg             object_error::parse_failed);
738b7c62394SSam Clegg       Info.ElementIndex = readVaruint32(Ctx);
7396a31a0d6SSam Clegg       // Use somewhat unique section name as symbol name.
7406a31a0d6SSam Clegg       StringRef SectionName = Sections[Info.ElementIndex].Name;
7416a31a0d6SSam Clegg       Info.Name = SectionName;
7426a31a0d6SSam Clegg       break;
7436a31a0d6SSam Clegg     }
7446a31a0d6SSam Clegg 
7451d891d44SHeejin Ahn     case wasm::WASM_SYMBOL_TYPE_TAG: {
746da419bdbSHeejin Ahn       Info.ElementIndex = readVaruint32(Ctx);
7471d891d44SHeejin Ahn       if (!isValidTagIndex(Info.ElementIndex) ||
7481d891d44SHeejin Ahn           IsDefined != isDefinedTagIndex(Info.ElementIndex))
7491d891d44SHeejin Ahn         return make_error<GenericBinaryError>("invalid tag symbol index",
750da419bdbSHeejin Ahn                                               object_error::parse_failed);
751da419bdbSHeejin Ahn       if (!IsDefined && (Info.Flags & wasm::WASM_SYMBOL_BINDING_MASK) ==
752da419bdbSHeejin Ahn                             wasm::WASM_SYMBOL_BINDING_WEAK)
753da419bdbSHeejin Ahn         return make_error<GenericBinaryError>("undefined weak global symbol",
754da419bdbSHeejin Ahn                                               object_error::parse_failed);
755da419bdbSHeejin Ahn       if (IsDefined) {
756da419bdbSHeejin Ahn         Info.Name = readString(Ctx);
7571d891d44SHeejin Ahn         unsigned TagIndex = Info.ElementIndex - NumImportedTags;
7581d891d44SHeejin Ahn         wasm::WasmTag &Tag = Tags[TagIndex];
7593ec1760dSHeejin Ahn         Signature = &Signatures[Tag.SigIndex];
7601d891d44SHeejin Ahn         if (Tag.SymbolName.empty())
7611d891d44SHeejin Ahn           Tag.SymbolName = Info.Name;
762da419bdbSHeejin Ahn 
763da419bdbSHeejin Ahn       } else {
7641d891d44SHeejin Ahn         wasm::WasmImport &Import = *ImportedTags[Info.ElementIndex];
765bd481277SSam Clegg         if ((Info.Flags & wasm::WASM_SYMBOL_EXPLICIT_NAME) != 0) {
76629874ceaSDan Gohman           Info.Name = readString(Ctx);
767bd481277SSam Clegg           Info.ImportName = Import.Field;
768bd481277SSam Clegg         } else {
76929874ceaSDan Gohman           Info.Name = Import.Field;
770bd481277SSam Clegg         }
7713ec1760dSHeejin Ahn         Signature = &Signatures[Import.SigIndex];
772bd481277SSam Clegg         if (!Import.Module.empty()) {
77329874ceaSDan Gohman           Info.ImportModule = Import.Module;
774da419bdbSHeejin Ahn         }
775bd481277SSam Clegg       }
776da419bdbSHeejin Ahn       break;
777da419bdbSHeejin Ahn     }
778da419bdbSHeejin Ahn 
7796c899ba6SSam Clegg     default:
78084c6f325SSam Clegg       return make_error<GenericBinaryError>("invalid symbol type: " +
781d75b3719SSam Clegg                                                 Twine(unsigned(Info.Kind)),
7826c899ba6SSam Clegg                                             object_error::parse_failed);
7836c899ba6SSam Clegg     }
7846c899ba6SSam Clegg 
7856c899ba6SSam Clegg     if ((Info.Flags & wasm::WASM_SYMBOL_BINDING_MASK) !=
7866c899ba6SSam Clegg             wasm::WASM_SYMBOL_BINDING_LOCAL &&
7876c899ba6SSam Clegg         !SymbolNames.insert(Info.Name).second)
78884c6f325SSam Clegg       return make_error<GenericBinaryError>("duplicate symbol name " +
7896c899ba6SSam Clegg                                                 Twine(Info.Name),
7906c899ba6SSam Clegg                                             object_error::parse_failed);
7916c899ba6SSam Clegg     LinkingData.SymbolTable.emplace_back(Info);
792388fb67bSPaulo Matos     Symbols.emplace_back(LinkingData.SymbolTable.back(), GlobalType, TableType,
7933ec1760dSHeejin Ahn                          Signature);
794d34e60caSNicola Zaghen     LLVM_DEBUG(dbgs() << "Adding symbol: " << Symbols.back() << "\n");
7956c899ba6SSam Clegg   }
7966c899ba6SSam Clegg 
7976c899ba6SSam Clegg   return Error::success();
7986c899ba6SSam Clegg }
7996c899ba6SSam Clegg 
parseLinkingSectionComdat(ReadContext & Ctx)800b7c62394SSam Clegg Error WasmObjectFile::parseLinkingSectionComdat(ReadContext &Ctx) {
801b7c62394SSam Clegg   uint32_t ComdatCount = readVaruint32(Ctx);
802ea7caceeSSam Clegg   StringSet<> ComdatSet;
803027b9357SNicholas Wilson   for (unsigned ComdatIndex = 0; ComdatIndex < ComdatCount; ++ComdatIndex) {
804b7c62394SSam Clegg     StringRef Name = readString(Ctx);
805ea7caceeSSam Clegg     if (Name.empty() || !ComdatSet.insert(Name).second)
80684c6f325SSam Clegg       return make_error<GenericBinaryError>("bad/duplicate COMDAT name " +
807f208f631SHeejin Ahn                                                 Twine(Name),
808ea7caceeSSam Clegg                                             object_error::parse_failed);
809027b9357SNicholas Wilson     LinkingData.Comdats.emplace_back(Name);
810b7c62394SSam Clegg     uint32_t Flags = readVaruint32(Ctx);
811ea7caceeSSam Clegg     if (Flags != 0)
81284c6f325SSam Clegg       return make_error<GenericBinaryError>("unsupported COMDAT flags",
813ea7caceeSSam Clegg                                             object_error::parse_failed);
814ea7caceeSSam Clegg 
815b7c62394SSam Clegg     uint32_t EntryCount = readVaruint32(Ctx);
816ea7caceeSSam Clegg     while (EntryCount--) {
817b7c62394SSam Clegg       unsigned Kind = readVaruint32(Ctx);
818b7c62394SSam Clegg       unsigned Index = readVaruint32(Ctx);
819ea7caceeSSam Clegg       switch (Kind) {
820ea7caceeSSam Clegg       default:
82184c6f325SSam Clegg         return make_error<GenericBinaryError>("invalid COMDAT entry type",
822ea7caceeSSam Clegg                                               object_error::parse_failed);
823ea7caceeSSam Clegg       case wasm::WASM_COMDAT_DATA:
824ea7caceeSSam Clegg         if (Index >= DataSegments.size())
825f208f631SHeejin Ahn           return make_error<GenericBinaryError>(
826f208f631SHeejin Ahn               "COMDAT data index out of range", object_error::parse_failed);
827027b9357SNicholas Wilson         if (DataSegments[Index].Data.Comdat != UINT32_MAX)
82884c6f325SSam Clegg           return make_error<GenericBinaryError>("data segment in two COMDATs",
829ea7caceeSSam Clegg                                                 object_error::parse_failed);
830027b9357SNicholas Wilson         DataSegments[Index].Data.Comdat = ComdatIndex;
831ea7caceeSSam Clegg         break;
832ea7caceeSSam Clegg       case wasm::WASM_COMDAT_FUNCTION:
83323012e98SSam Clegg         if (!isDefinedFunctionIndex(Index))
834f208f631SHeejin Ahn           return make_error<GenericBinaryError>(
835f208f631SHeejin Ahn               "COMDAT function index out of range", object_error::parse_failed);
836027b9357SNicholas Wilson         if (getDefinedFunction(Index).Comdat != UINT32_MAX)
83784c6f325SSam Clegg           return make_error<GenericBinaryError>("function in two COMDATs",
838ea7caceeSSam Clegg                                                 object_error::parse_failed);
839027b9357SNicholas Wilson         getDefinedFunction(Index).Comdat = ComdatIndex;
840ea7caceeSSam Clegg         break;
8410a391060SDerek Schuff       case wasm::WASM_COMDAT_SECTION:
8420a391060SDerek Schuff         if (Index >= Sections.size())
8430a391060SDerek Schuff           return make_error<GenericBinaryError>(
8440a391060SDerek Schuff               "COMDAT section index out of range", object_error::parse_failed);
8450a391060SDerek Schuff         if (Sections[Index].Type != wasm::WASM_SEC_CUSTOM)
8460a391060SDerek Schuff           return make_error<GenericBinaryError>(
84784c6f325SSam Clegg               "non-custom section in a COMDAT", object_error::parse_failed);
8480a391060SDerek Schuff         Sections[Index].Comdat = ComdatIndex;
8490a391060SDerek Schuff         break;
850ea7caceeSSam Clegg       }
851ea7caceeSSam Clegg     }
852ea7caceeSSam Clegg   }
853ea7caceeSSam Clegg   return Error::success();
854ea7caceeSSam Clegg }
855ea7caceeSSam Clegg 
parseProducersSection(ReadContext & Ctx)856cbda16ebSThomas Lively Error WasmObjectFile::parseProducersSection(ReadContext &Ctx) {
857cbda16ebSThomas Lively   llvm::SmallSet<StringRef, 3> FieldsSeen;
858cbda16ebSThomas Lively   uint32_t Fields = readVaruint32(Ctx);
85918c56a07SHeejin Ahn   for (size_t I = 0; I < Fields; ++I) {
860cbda16ebSThomas Lively     StringRef FieldName = readString(Ctx);
861cbda16ebSThomas Lively     if (!FieldsSeen.insert(FieldName).second)
862cbda16ebSThomas Lively       return make_error<GenericBinaryError>(
86384c6f325SSam Clegg           "producers section does not have unique fields",
864cbda16ebSThomas Lively           object_error::parse_failed);
865cbda16ebSThomas Lively     std::vector<std::pair<std::string, std::string>> *ProducerVec = nullptr;
866cbda16ebSThomas Lively     if (FieldName == "language") {
867cbda16ebSThomas Lively       ProducerVec = &ProducerInfo.Languages;
868cbda16ebSThomas Lively     } else if (FieldName == "processed-by") {
869cbda16ebSThomas Lively       ProducerVec = &ProducerInfo.Tools;
870cbda16ebSThomas Lively     } else if (FieldName == "sdk") {
871cbda16ebSThomas Lively       ProducerVec = &ProducerInfo.SDKs;
872cbda16ebSThomas Lively     } else {
873cbda16ebSThomas Lively       return make_error<GenericBinaryError>(
87484c6f325SSam Clegg           "producers section field is not named one of language, processed-by, "
875cbda16ebSThomas Lively           "or sdk",
876cbda16ebSThomas Lively           object_error::parse_failed);
877cbda16ebSThomas Lively     }
878cbda16ebSThomas Lively     uint32_t ValueCount = readVaruint32(Ctx);
879cbda16ebSThomas Lively     llvm::SmallSet<StringRef, 8> ProducersSeen;
88018c56a07SHeejin Ahn     for (size_t J = 0; J < ValueCount; ++J) {
881cbda16ebSThomas Lively       StringRef Name = readString(Ctx);
882cbda16ebSThomas Lively       StringRef Version = readString(Ctx);
883cbda16ebSThomas Lively       if (!ProducersSeen.insert(Name).second) {
884cbda16ebSThomas Lively         return make_error<GenericBinaryError>(
88584c6f325SSam Clegg             "producers section contains repeated producer",
886cbda16ebSThomas Lively             object_error::parse_failed);
887cbda16ebSThomas Lively       }
888a9bc7b83SBenjamin Kramer       ProducerVec->emplace_back(std::string(Name), std::string(Version));
889cbda16ebSThomas Lively     }
890cbda16ebSThomas Lively   }
891cbda16ebSThomas Lively   if (Ctx.Ptr != Ctx.End)
89284c6f325SSam Clegg     return make_error<GenericBinaryError>("producers section ended prematurely",
893cbda16ebSThomas Lively                                           object_error::parse_failed);
894cbda16ebSThomas Lively   return Error::success();
895cbda16ebSThomas Lively }
896cbda16ebSThomas Lively 
parseTargetFeaturesSection(ReadContext & Ctx)897f6f4f843SThomas Lively Error WasmObjectFile::parseTargetFeaturesSection(ReadContext &Ctx) {
898f6f4f843SThomas Lively   llvm::SmallSet<std::string, 8> FeaturesSeen;
899f6f4f843SThomas Lively   uint32_t FeatureCount = readVaruint32(Ctx);
900f6f4f843SThomas Lively   for (size_t I = 0; I < FeatureCount; ++I) {
901f6f4f843SThomas Lively     wasm::WasmFeatureEntry Feature;
902f6f4f843SThomas Lively     Feature.Prefix = readUint8(Ctx);
903f6f4f843SThomas Lively     switch (Feature.Prefix) {
904f6f4f843SThomas Lively     case wasm::WASM_FEATURE_PREFIX_USED:
905f6f4f843SThomas Lively     case wasm::WASM_FEATURE_PREFIX_REQUIRED:
906f6f4f843SThomas Lively     case wasm::WASM_FEATURE_PREFIX_DISALLOWED:
907f6f4f843SThomas Lively       break;
908f6f4f843SThomas Lively     default:
90984c6f325SSam Clegg       return make_error<GenericBinaryError>("unknown feature policy prefix",
910f6f4f843SThomas Lively                                             object_error::parse_failed);
911f6f4f843SThomas Lively     }
912adcd0268SBenjamin Kramer     Feature.Name = std::string(readString(Ctx));
913f6f4f843SThomas Lively     if (!FeaturesSeen.insert(Feature.Name).second)
914f6f4f843SThomas Lively       return make_error<GenericBinaryError>(
91584c6f325SSam Clegg           "target features section contains repeated feature \"" +
916f6f4f843SThomas Lively               Feature.Name + "\"",
917f6f4f843SThomas Lively           object_error::parse_failed);
918f6f4f843SThomas Lively     TargetFeatures.push_back(Feature);
919f6f4f843SThomas Lively   }
920f6f4f843SThomas Lively   if (Ctx.Ptr != Ctx.End)
921f6f4f843SThomas Lively     return make_error<GenericBinaryError>(
92284c6f325SSam Clegg         "target features section ended prematurely",
923f6f4f843SThomas Lively         object_error::parse_failed);
924f6f4f843SThomas Lively   return Error::success();
925f6f4f843SThomas Lively }
926f6f4f843SThomas Lively 
parseRelocSection(StringRef Name,ReadContext & Ctx)927b7c62394SSam Clegg Error WasmObjectFile::parseRelocSection(StringRef Name, ReadContext &Ctx) {
928b7c62394SSam Clegg   uint32_t SectionIndex = readVaruint32(Ctx);
9296f08c84aSSam Clegg   if (SectionIndex >= Sections.size())
93084c6f325SSam Clegg     return make_error<GenericBinaryError>("invalid section index",
931d3d84fddSDerek Schuff                                           object_error::parse_failed);
9326f08c84aSSam Clegg   WasmSection &Section = Sections[SectionIndex];
933b7c62394SSam Clegg   uint32_t RelocCount = readVaruint32(Ctx);
9346f08c84aSSam Clegg   uint32_t EndOffset = Section.Content.size();
935f77dc2a8SSam Clegg   uint32_t PreviousOffset = 0;
936d3d84fddSDerek Schuff   while (RelocCount--) {
937b3748f71SNicholas Wilson     wasm::WasmRelocation Reloc = {};
938d75b3719SSam Clegg     uint32_t type = readVaruint32(Ctx);
939d75b3719SSam Clegg     Reloc.Type = type;
940b7c62394SSam Clegg     Reloc.Offset = readVaruint32(Ctx);
941f77dc2a8SSam Clegg     if (Reloc.Offset < PreviousOffset)
94284c6f325SSam Clegg       return make_error<GenericBinaryError>("relocations not in offset order",
943f77dc2a8SSam Clegg                                             object_error::parse_failed);
944f77dc2a8SSam Clegg     PreviousOffset = Reloc.Offset;
945b7c62394SSam Clegg     Reloc.Index = readVaruint32(Ctx);
946d75b3719SSam Clegg     switch (type) {
947d1152a26SSam Clegg     case wasm::R_WASM_FUNCTION_INDEX_LEB:
948d1152a26SSam Clegg     case wasm::R_WASM_TABLE_INDEX_SLEB:
949cc1b9b68SWouter van Oortmerssen     case wasm::R_WASM_TABLE_INDEX_SLEB64:
950d1152a26SSam Clegg     case wasm::R_WASM_TABLE_INDEX_I32:
951cc1b9b68SWouter van Oortmerssen     case wasm::R_WASM_TABLE_INDEX_I64:
9522a7cac93SSam Clegg     case wasm::R_WASM_TABLE_INDEX_REL_SLEB:
9533a293cbfSWouter van Oortmerssen     case wasm::R_WASM_TABLE_INDEX_REL_SLEB64:
954b3748f71SNicholas Wilson       if (!isValidFunctionSymbol(Reloc.Index))
95584c6f325SSam Clegg         return make_error<GenericBinaryError>(
95684c6f325SSam Clegg             "invalid relocation function index", object_error::parse_failed);
957b3748f71SNicholas Wilson       break;
95869e2797eSPaulo Matos     case wasm::R_WASM_TABLE_NUMBER_LEB:
95969e2797eSPaulo Matos       if (!isValidTableSymbol(Reloc.Index))
96084c6f325SSam Clegg         return make_error<GenericBinaryError>("invalid relocation table index",
96169e2797eSPaulo Matos                                               object_error::parse_failed);
96269e2797eSPaulo Matos       break;
963d1152a26SSam Clegg     case wasm::R_WASM_TYPE_INDEX_LEB:
964b3748f71SNicholas Wilson       if (Reloc.Index >= Signatures.size())
96584c6f325SSam Clegg         return make_error<GenericBinaryError>("invalid relocation type index",
966b3748f71SNicholas Wilson                                               object_error::parse_failed);
967b3748f71SNicholas Wilson       break;
968d1152a26SSam Clegg     case wasm::R_WASM_GLOBAL_INDEX_LEB:
969492f7529SSam Clegg       // R_WASM_GLOBAL_INDEX_LEB are can be used against function and data
9707eacefedSDavid Candler       // symbols to refer to their GOT entries.
971492f7529SSam Clegg       if (!isValidGlobalSymbol(Reloc.Index) &&
972492f7529SSam Clegg           !isValidDataSymbol(Reloc.Index) &&
973492f7529SSam Clegg           !isValidFunctionSymbol(Reloc.Index))
97484c6f325SSam Clegg         return make_error<GenericBinaryError>("invalid relocation global index",
975b3748f71SNicholas Wilson                                               object_error::parse_failed);
976d3d84fddSDerek Schuff       break;
97748139ebcSWouter van Oortmerssen     case wasm::R_WASM_GLOBAL_INDEX_I32:
97848139ebcSWouter van Oortmerssen       if (!isValidGlobalSymbol(Reloc.Index))
97984c6f325SSam Clegg         return make_error<GenericBinaryError>("invalid relocation global index",
98048139ebcSWouter van Oortmerssen                                               object_error::parse_failed);
98148139ebcSWouter van Oortmerssen       break;
9821d891d44SHeejin Ahn     case wasm::R_WASM_TAG_INDEX_LEB:
9831d891d44SHeejin Ahn       if (!isValidTagSymbol(Reloc.Index))
9841d891d44SHeejin Ahn         return make_error<GenericBinaryError>("invalid relocation tag index",
985da419bdbSHeejin Ahn                                               object_error::parse_failed);
986da419bdbSHeejin Ahn       break;
987d1152a26SSam Clegg     case wasm::R_WASM_MEMORY_ADDR_LEB:
988d1152a26SSam Clegg     case wasm::R_WASM_MEMORY_ADDR_SLEB:
989d1152a26SSam Clegg     case wasm::R_WASM_MEMORY_ADDR_I32:
9902a7cac93SSam Clegg     case wasm::R_WASM_MEMORY_ADDR_REL_SLEB:
991a28a4662SSam Clegg     case wasm::R_WASM_MEMORY_ADDR_TLS_SLEB:
992aa0c571aSYuta Saito     case wasm::R_WASM_MEMORY_ADDR_LOCREL_I32:
993b3748f71SNicholas Wilson       if (!isValidDataSymbol(Reloc.Index))
99484c6f325SSam Clegg         return make_error<GenericBinaryError>("invalid relocation data index",
995b3748f71SNicholas Wilson                                               object_error::parse_failed);
996b7c62394SSam Clegg       Reloc.Addend = readVarint32(Ctx);
997d3d84fddSDerek Schuff       break;
9983b29376eSWouter van Oortmerssen     case wasm::R_WASM_MEMORY_ADDR_LEB64:
9993b29376eSWouter van Oortmerssen     case wasm::R_WASM_MEMORY_ADDR_SLEB64:
10003b29376eSWouter van Oortmerssen     case wasm::R_WASM_MEMORY_ADDR_I64:
10013b29376eSWouter van Oortmerssen     case wasm::R_WASM_MEMORY_ADDR_REL_SLEB64:
1002670944fbSWouter van Oortmerssen     case wasm::R_WASM_MEMORY_ADDR_TLS_SLEB64:
10033b29376eSWouter van Oortmerssen       if (!isValidDataSymbol(Reloc.Index))
100484c6f325SSam Clegg         return make_error<GenericBinaryError>("invalid relocation data index",
10053b29376eSWouter van Oortmerssen                                               object_error::parse_failed);
10063b29376eSWouter van Oortmerssen       Reloc.Addend = readVarint64(Ctx);
10073b29376eSWouter van Oortmerssen       break;
1008d1152a26SSam Clegg     case wasm::R_WASM_FUNCTION_OFFSET_I32:
10096a31a0d6SSam Clegg       if (!isValidFunctionSymbol(Reloc.Index))
101084c6f325SSam Clegg         return make_error<GenericBinaryError>(
101184c6f325SSam Clegg             "invalid relocation function index", object_error::parse_failed);
1012b7c62394SSam Clegg       Reloc.Addend = readVarint32(Ctx);
10136a31a0d6SSam Clegg       break;
101416f02431SWouter van Oortmerssen     case wasm::R_WASM_FUNCTION_OFFSET_I64:
101516f02431SWouter van Oortmerssen       if (!isValidFunctionSymbol(Reloc.Index))
101684c6f325SSam Clegg         return make_error<GenericBinaryError>(
101784c6f325SSam Clegg             "invalid relocation function index", object_error::parse_failed);
101816f02431SWouter van Oortmerssen       Reloc.Addend = readVarint64(Ctx);
101916f02431SWouter van Oortmerssen       break;
1020d1152a26SSam Clegg     case wasm::R_WASM_SECTION_OFFSET_I32:
10216a31a0d6SSam Clegg       if (!isValidSectionSymbol(Reloc.Index))
102284c6f325SSam Clegg         return make_error<GenericBinaryError>(
102384c6f325SSam Clegg             "invalid relocation section index", object_error::parse_failed);
1024b7c62394SSam Clegg       Reloc.Addend = readVarint32(Ctx);
10256a31a0d6SSam Clegg       break;
1026d3d84fddSDerek Schuff     default:
102784c6f325SSam Clegg       return make_error<GenericBinaryError>("invalid relocation type: " +
102884c6f325SSam Clegg                                                 Twine(type),
102984c6f325SSam Clegg                                             object_error::parse_failed);
1030d3d84fddSDerek Schuff     }
1031b3748f71SNicholas Wilson 
1032b3748f71SNicholas Wilson     // Relocations must fit inside the section, and must appear in order.  They
1033b3748f71SNicholas Wilson     // also shouldn't overlap a function/element boundary, but we don't bother
1034b3748f71SNicholas Wilson     // to check that.
1035b3748f71SNicholas Wilson     uint64_t Size = 5;
10363b29376eSWouter van Oortmerssen     if (Reloc.Type == wasm::R_WASM_MEMORY_ADDR_LEB64 ||
10373b29376eSWouter van Oortmerssen         Reloc.Type == wasm::R_WASM_MEMORY_ADDR_SLEB64 ||
10383b29376eSWouter van Oortmerssen         Reloc.Type == wasm::R_WASM_MEMORY_ADDR_REL_SLEB64)
10393b29376eSWouter van Oortmerssen       Size = 10;
1040d1152a26SSam Clegg     if (Reloc.Type == wasm::R_WASM_TABLE_INDEX_I32 ||
1041d1152a26SSam Clegg         Reloc.Type == wasm::R_WASM_MEMORY_ADDR_I32 ||
1042aa0c571aSYuta Saito         Reloc.Type == wasm::R_WASM_MEMORY_ADDR_LOCREL_I32 ||
1043d1152a26SSam Clegg         Reloc.Type == wasm::R_WASM_SECTION_OFFSET_I32 ||
104448139ebcSWouter van Oortmerssen         Reloc.Type == wasm::R_WASM_FUNCTION_OFFSET_I32 ||
104548139ebcSWouter van Oortmerssen         Reloc.Type == wasm::R_WASM_GLOBAL_INDEX_I32)
1046b3748f71SNicholas Wilson       Size = 4;
1047cc1b9b68SWouter van Oortmerssen     if (Reloc.Type == wasm::R_WASM_TABLE_INDEX_I64 ||
104816f02431SWouter van Oortmerssen         Reloc.Type == wasm::R_WASM_MEMORY_ADDR_I64 ||
104916f02431SWouter van Oortmerssen         Reloc.Type == wasm::R_WASM_FUNCTION_OFFSET_I64)
10503b29376eSWouter van Oortmerssen       Size = 8;
1051fa5a04fbSSam Clegg     if (Reloc.Offset + Size > EndOffset)
105284c6f325SSam Clegg       return make_error<GenericBinaryError>("invalid relocation offset",
1053b3748f71SNicholas Wilson                                             object_error::parse_failed);
1054b3748f71SNicholas Wilson 
10556f08c84aSSam Clegg     Section.Relocations.push_back(Reloc);
1056d3d84fddSDerek Schuff   }
1057b7c62394SSam Clegg   if (Ctx.Ptr != Ctx.End)
105884c6f325SSam Clegg     return make_error<GenericBinaryError>("reloc section ended prematurely",
1059d3d84fddSDerek Schuff                                           object_error::parse_failed);
1060d3d84fddSDerek Schuff   return Error::success();
1061d3d84fddSDerek Schuff }
1062d3d84fddSDerek Schuff 
parseCustomSection(WasmSection & Sec,ReadContext & Ctx)1063b7c62394SSam Clegg Error WasmObjectFile::parseCustomSection(WasmSection &Sec, ReadContext &Ctx) {
1064e4afbc68SSam Clegg   if (Sec.Name == "dylink") {
1065e4afbc68SSam Clegg     if (Error Err = parseDylinkSection(Ctx))
1066e4afbc68SSam Clegg       return Err;
1067b78c85a4SSam Clegg   } else if (Sec.Name == "dylink.0") {
1068b78c85a4SSam Clegg     if (Error Err = parseDylink0Section(Ctx))
1069b78c85a4SSam Clegg       return Err;
1070e4afbc68SSam Clegg   } else if (Sec.Name == "name") {
1071b7c62394SSam Clegg     if (Error Err = parseNameSection(Ctx))
1072d3d84fddSDerek Schuff       return Err;
1073b7787fd0SSam Clegg   } else if (Sec.Name == "linking") {
1074b7c62394SSam Clegg     if (Error Err = parseLinkingSection(Ctx))
1075b7787fd0SSam Clegg       return Err;
1076cbda16ebSThomas Lively   } else if (Sec.Name == "producers") {
1077cbda16ebSThomas Lively     if (Error Err = parseProducersSection(Ctx))
1078cbda16ebSThomas Lively       return Err;
1079f6f4f843SThomas Lively   } else if (Sec.Name == "target_features") {
1080f6f4f843SThomas Lively     if (Error Err = parseTargetFeaturesSection(Ctx))
1081f6f4f843SThomas Lively       return Err;
1082d3d84fddSDerek Schuff   } else if (Sec.Name.startswith("reloc.")) {
1083b7c62394SSam Clegg     if (Error Err = parseRelocSection(Sec.Name, Ctx))
1084d3d84fddSDerek Schuff       return Err;
1085d3d84fddSDerek Schuff   }
1086d3d84fddSDerek Schuff   return Error::success();
1087d3d84fddSDerek Schuff }
1088d3d84fddSDerek Schuff 
parseTypeSection(ReadContext & Ctx)1089b7c62394SSam Clegg Error WasmObjectFile::parseTypeSection(ReadContext &Ctx) {
1090b7c62394SSam Clegg   uint32_t Count = readVaruint32(Ctx);
1091d3d84fddSDerek Schuff   Signatures.reserve(Count);
1092d3d84fddSDerek Schuff   while (Count--) {
1093d3d84fddSDerek Schuff     wasm::WasmSignature Sig;
1094b7c62394SSam Clegg     uint8_t Form = readUint8(Ctx);
1095d3d84fddSDerek Schuff     if (Form != wasm::WASM_TYPE_FUNC) {
109684c6f325SSam Clegg       return make_error<GenericBinaryError>("invalid signature type",
1097d3d84fddSDerek Schuff                                             object_error::parse_failed);
1098d3d84fddSDerek Schuff     }
1099b7c62394SSam Clegg     uint32_t ParamCount = readVaruint32(Ctx);
110077a7a380SDerek Schuff     Sig.Params.reserve(ParamCount);
1101d3d84fddSDerek Schuff     while (ParamCount--) {
1102b7c62394SSam Clegg       uint32_t ParamType = readUint8(Ctx);
110377a7a380SDerek Schuff       Sig.Params.push_back(wasm::ValType(ParamType));
1104d3d84fddSDerek Schuff     }
1105b7c62394SSam Clegg     uint32_t ReturnCount = readVaruint32(Ctx);
1106393d0f79SThomas Lively     while (ReturnCount--) {
1107393d0f79SThomas Lively       uint32_t ReturnType = readUint8(Ctx);
1108393d0f79SThomas Lively       Sig.Returns.push_back(wasm::ValType(ReturnType));
1109d3d84fddSDerek Schuff     }
111077a7a380SDerek Schuff     Signatures.push_back(std::move(Sig));
1111d3d84fddSDerek Schuff   }
1112b7c62394SSam Clegg   if (Ctx.Ptr != Ctx.End)
111384c6f325SSam Clegg     return make_error<GenericBinaryError>("type section ended prematurely",
1114d3d84fddSDerek Schuff                                           object_error::parse_failed);
1115d3d84fddSDerek Schuff   return Error::success();
1116d3d84fddSDerek Schuff }
1117d3d84fddSDerek Schuff 
parseImportSection(ReadContext & Ctx)1118b7c62394SSam Clegg Error WasmObjectFile::parseImportSection(ReadContext &Ctx) {
1119b7c62394SSam Clegg   uint32_t Count = readVaruint32(Ctx);
11203ec1760dSHeejin Ahn   uint32_t NumTypes = Signatures.size();
1121d3d84fddSDerek Schuff   Imports.reserve(Count);
112218c56a07SHeejin Ahn   for (uint32_t I = 0; I < Count; I++) {
1123d3d84fddSDerek Schuff     wasm::WasmImport Im;
1124b7c62394SSam Clegg     Im.Module = readString(Ctx);
1125b7c62394SSam Clegg     Im.Field = readString(Ctx);
1126b7c62394SSam Clegg     Im.Kind = readUint8(Ctx);
1127d3d84fddSDerek Schuff     switch (Im.Kind) {
1128d3d84fddSDerek Schuff     case wasm::WASM_EXTERNAL_FUNCTION:
1129a3b9fe6aSSam Clegg       NumImportedFunctions++;
1130b7c62394SSam Clegg       Im.SigIndex = readVaruint32(Ctx);
11313ec1760dSHeejin Ahn       if (Im.SigIndex >= NumTypes)
11323ec1760dSHeejin Ahn         return make_error<GenericBinaryError>("invalid function type",
11333ec1760dSHeejin Ahn                                               object_error::parse_failed);
1134d3d84fddSDerek Schuff       break;
1135d3d84fddSDerek Schuff     case wasm::WASM_EXTERNAL_GLOBAL:
1136a3b9fe6aSSam Clegg       NumImportedGlobals++;
1137b7c62394SSam Clegg       Im.Global.Type = readUint8(Ctx);
1138b7c62394SSam Clegg       Im.Global.Mutable = readVaruint1(Ctx);
1139d3d84fddSDerek Schuff       break;
11402ffff5afSSam Clegg     case wasm::WASM_EXTERNAL_MEMORY:
1141b7c62394SSam Clegg       Im.Memory = readLimits(Ctx);
114229f8c9f6SWouter van Oortmerssen       if (Im.Memory.Flags & wasm::WASM_LIMITS_FLAG_IS_64)
114329f8c9f6SWouter van Oortmerssen         HasMemory64 = true;
11442ffff5afSSam Clegg       break;
1145388fb67bSPaulo Matos     case wasm::WASM_EXTERNAL_TABLE: {
11461933c9d4SAndy Wingo       Im.Table = readTableType(Ctx);
1147388fb67bSPaulo Matos       NumImportedTables++;
1148388fb67bSPaulo Matos       auto ElemType = Im.Table.ElemType;
1149388fb67bSPaulo Matos       if (ElemType != wasm::WASM_TYPE_FUNCREF &&
1150388fb67bSPaulo Matos           ElemType != wasm::WASM_TYPE_EXTERNREF)
115184c6f325SSam Clegg         return make_error<GenericBinaryError>("invalid table element type",
11522ffff5afSSam Clegg                                               object_error::parse_failed);
11532ffff5afSSam Clegg       break;
1154388fb67bSPaulo Matos     }
11551d891d44SHeejin Ahn     case wasm::WASM_EXTERNAL_TAG:
11561d891d44SHeejin Ahn       NumImportedTags++;
11579261ee32SHeejin Ahn       if (readUint8(Ctx) != 0) // Reserved 'attribute' field
11589261ee32SHeejin Ahn         return make_error<GenericBinaryError>("invalid attribute",
11599261ee32SHeejin Ahn                                               object_error::parse_failed);
11603ec1760dSHeejin Ahn       Im.SigIndex = readVaruint32(Ctx);
11613ec1760dSHeejin Ahn       if (Im.SigIndex >= NumTypes)
11623ec1760dSHeejin Ahn         return make_error<GenericBinaryError>("invalid tag type",
11633ec1760dSHeejin Ahn                                               object_error::parse_failed);
1164da419bdbSHeejin Ahn       break;
1165d3d84fddSDerek Schuff     default:
116684c6f325SSam Clegg       return make_error<GenericBinaryError>("unexpected import kind",
1167f208f631SHeejin Ahn                                             object_error::parse_failed);
1168d3d84fddSDerek Schuff     }
1169d3d84fddSDerek Schuff     Imports.push_back(Im);
1170d3d84fddSDerek Schuff   }
1171b7c62394SSam Clegg   if (Ctx.Ptr != Ctx.End)
117284c6f325SSam Clegg     return make_error<GenericBinaryError>("import section ended prematurely",
1173d3d84fddSDerek Schuff                                           object_error::parse_failed);
1174d3d84fddSDerek Schuff   return Error::success();
1175d3d84fddSDerek Schuff }
1176d3d84fddSDerek Schuff 
parseFunctionSection(ReadContext & Ctx)1177b7c62394SSam Clegg Error WasmObjectFile::parseFunctionSection(ReadContext &Ctx) {
1178b7c62394SSam Clegg   uint32_t Count = readVaruint32(Ctx);
1179c0039de2SSam Clegg   Functions.reserve(Count);
11803b825105SNicholas Wilson   uint32_t NumTypes = Signatures.size();
1181d3d84fddSDerek Schuff   while (Count--) {
1182b7c62394SSam Clegg     uint32_t Type = readVaruint32(Ctx);
11833b825105SNicholas Wilson     if (Type >= NumTypes)
118484c6f325SSam Clegg       return make_error<GenericBinaryError>("invalid function type",
11853b825105SNicholas Wilson                                             object_error::parse_failed);
1186c0039de2SSam Clegg     wasm::WasmFunction F;
1187c0039de2SSam Clegg     F.SigIndex = Type;
1188c0039de2SSam Clegg     Functions.push_back(F);
1189d3d84fddSDerek Schuff   }
1190b7c62394SSam Clegg   if (Ctx.Ptr != Ctx.End)
119184c6f325SSam Clegg     return make_error<GenericBinaryError>("function section ended prematurely",
1192d3d84fddSDerek Schuff                                           object_error::parse_failed);
1193d3d84fddSDerek Schuff   return Error::success();
1194d3d84fddSDerek Schuff }
1195d3d84fddSDerek Schuff 
parseTableSection(ReadContext & Ctx)1196b7c62394SSam Clegg Error WasmObjectFile::parseTableSection(ReadContext &Ctx) {
1197feac819eSAndy Wingo   TableSection = Sections.size();
1198b7c62394SSam Clegg   uint32_t Count = readVaruint32(Ctx);
1199d3d84fddSDerek Schuff   Tables.reserve(Count);
1200d3d84fddSDerek Schuff   while (Count--) {
12011933c9d4SAndy Wingo     wasm::WasmTable T;
12021933c9d4SAndy Wingo     T.Type = readTableType(Ctx);
1203388fb67bSPaulo Matos     T.Index = NumImportedTables + Tables.size();
1204388fb67bSPaulo Matos     Tables.push_back(T);
12051933c9d4SAndy Wingo     auto ElemType = Tables.back().Type.ElemType;
1206388fb67bSPaulo Matos     if (ElemType != wasm::WASM_TYPE_FUNCREF &&
1207388fb67bSPaulo Matos         ElemType != wasm::WASM_TYPE_EXTERNREF) {
120884c6f325SSam Clegg       return make_error<GenericBinaryError>("invalid table element type",
1209d3d84fddSDerek Schuff                                             object_error::parse_failed);
1210d3d84fddSDerek Schuff     }
1211d3d84fddSDerek Schuff   }
1212b7c62394SSam Clegg   if (Ctx.Ptr != Ctx.End)
121384c6f325SSam Clegg     return make_error<GenericBinaryError>("table section ended prematurely",
1214d3d84fddSDerek Schuff                                           object_error::parse_failed);
1215d3d84fddSDerek Schuff   return Error::success();
1216d3d84fddSDerek Schuff }
1217d3d84fddSDerek Schuff 
parseMemorySection(ReadContext & Ctx)1218b7c62394SSam Clegg Error WasmObjectFile::parseMemorySection(ReadContext &Ctx) {
1219b7c62394SSam Clegg   uint32_t Count = readVaruint32(Ctx);
1220d3d84fddSDerek Schuff   Memories.reserve(Count);
1221d3d84fddSDerek Schuff   while (Count--) {
122229f8c9f6SWouter van Oortmerssen     auto Limits = readLimits(Ctx);
122329f8c9f6SWouter van Oortmerssen     if (Limits.Flags & wasm::WASM_LIMITS_FLAG_IS_64)
122429f8c9f6SWouter van Oortmerssen       HasMemory64 = true;
122529f8c9f6SWouter van Oortmerssen     Memories.push_back(Limits);
1226d3d84fddSDerek Schuff   }
1227b7c62394SSam Clegg   if (Ctx.Ptr != Ctx.End)
122884c6f325SSam Clegg     return make_error<GenericBinaryError>("memory section ended prematurely",
1229d3d84fddSDerek Schuff                                           object_error::parse_failed);
1230d3d84fddSDerek Schuff   return Error::success();
1231d3d84fddSDerek Schuff }
1232d3d84fddSDerek Schuff 
parseTagSection(ReadContext & Ctx)12331d891d44SHeejin Ahn Error WasmObjectFile::parseTagSection(ReadContext &Ctx) {
12341d891d44SHeejin Ahn   TagSection = Sections.size();
1235144ec1c3SHeejin Ahn   uint32_t Count = readVaruint32(Ctx);
12361d891d44SHeejin Ahn   Tags.reserve(Count);
12373ec1760dSHeejin Ahn   uint32_t NumTypes = Signatures.size();
1238f93426c5SHeejin Ahn   while (Count--) {
12399261ee32SHeejin Ahn     if (readUint8(Ctx) != 0) // Reserved 'attribute' field
12403ec1760dSHeejin Ahn       return make_error<GenericBinaryError>("invalid attribute",
12413ec1760dSHeejin Ahn                                             object_error::parse_failed);
12423ec1760dSHeejin Ahn     uint32_t Type = readVaruint32(Ctx);
12433ec1760dSHeejin Ahn     if (Type >= NumTypes)
12443ec1760dSHeejin Ahn       return make_error<GenericBinaryError>("invalid tag type",
12453ec1760dSHeejin Ahn                                             object_error::parse_failed);
12461d891d44SHeejin Ahn     wasm::WasmTag Tag;
12471d891d44SHeejin Ahn     Tag.Index = NumImportedTags + Tags.size();
12483ec1760dSHeejin Ahn     Tag.SigIndex = Type;
12491d891d44SHeejin Ahn     Tags.push_back(Tag);
1250f93426c5SHeejin Ahn   }
1251f93426c5SHeejin Ahn 
1252f93426c5SHeejin Ahn   if (Ctx.Ptr != Ctx.End)
12531d891d44SHeejin Ahn     return make_error<GenericBinaryError>("tag section ended prematurely",
1254f93426c5SHeejin Ahn                                           object_error::parse_failed);
1255f93426c5SHeejin Ahn   return Error::success();
1256f93426c5SHeejin Ahn }
1257f93426c5SHeejin Ahn 
parseGlobalSection(ReadContext & Ctx)1258b7c62394SSam Clegg Error WasmObjectFile::parseGlobalSection(ReadContext &Ctx) {
12596c899ba6SSam Clegg   GlobalSection = Sections.size();
1260b7c62394SSam Clegg   uint32_t Count = readVaruint32(Ctx);
1261d3d84fddSDerek Schuff   Globals.reserve(Count);
1262d3d84fddSDerek Schuff   while (Count--) {
1263d3d84fddSDerek Schuff     wasm::WasmGlobal Global;
1264e53af7f6SSam Clegg     Global.Index = NumImportedGlobals + Globals.size();
1265b7c62394SSam Clegg     Global.Type.Type = readUint8(Ctx);
1266b7c62394SSam Clegg     Global.Type.Mutable = readVaruint1(Ctx);
1267b7c62394SSam Clegg     if (Error Err = readInitExpr(Global.InitExpr, Ctx))
1268d3d84fddSDerek Schuff       return Err;
1269d3d84fddSDerek Schuff     Globals.push_back(Global);
1270d3d84fddSDerek Schuff   }
1271b7c62394SSam Clegg   if (Ctx.Ptr != Ctx.End)
127284c6f325SSam Clegg     return make_error<GenericBinaryError>("global section ended prematurely",
1273d3d84fddSDerek Schuff                                           object_error::parse_failed);
1274d3d84fddSDerek Schuff   return Error::success();
1275d3d84fddSDerek Schuff }
1276d3d84fddSDerek Schuff 
parseExportSection(ReadContext & Ctx)1277b7c62394SSam Clegg Error WasmObjectFile::parseExportSection(ReadContext &Ctx) {
1278b7c62394SSam Clegg   uint32_t Count = readVaruint32(Ctx);
1279d3d84fddSDerek Schuff   Exports.reserve(Count);
128018c56a07SHeejin Ahn   for (uint32_t I = 0; I < Count; I++) {
1281d3d84fddSDerek Schuff     wasm::WasmExport Ex;
1282b7c62394SSam Clegg     Ex.Name = readString(Ctx);
1283b7c62394SSam Clegg     Ex.Kind = readUint8(Ctx);
1284b7c62394SSam Clegg     Ex.Index = readVaruint32(Ctx);
1285d3d84fddSDerek Schuff     switch (Ex.Kind) {
1286d3d84fddSDerek Schuff     case wasm::WASM_EXTERNAL_FUNCTION:
1287881d8778SSam Clegg 
1288881d8778SSam Clegg       if (!isDefinedFunctionIndex(Ex.Index))
128984c6f325SSam Clegg         return make_error<GenericBinaryError>("invalid function export",
1290a3b9fe6aSSam Clegg                                               object_error::parse_failed);
1291881d8778SSam Clegg       getDefinedFunction(Ex.Index).ExportName = Ex.Name;
1292d3d84fddSDerek Schuff       break;
12936c899ba6SSam Clegg     case wasm::WASM_EXTERNAL_GLOBAL:
12946c899ba6SSam Clegg       if (!isValidGlobalIndex(Ex.Index))
129584c6f325SSam Clegg         return make_error<GenericBinaryError>("invalid global export",
1296a3b9fe6aSSam Clegg                                               object_error::parse_failed);
1297d3d84fddSDerek Schuff       break;
12981d891d44SHeejin Ahn     case wasm::WASM_EXTERNAL_TAG:
12991d891d44SHeejin Ahn       if (!isValidTagIndex(Ex.Index))
13001d891d44SHeejin Ahn         return make_error<GenericBinaryError>("invalid tag export",
1301da419bdbSHeejin Ahn                                               object_error::parse_failed);
1302da419bdbSHeejin Ahn       break;
13032ffff5afSSam Clegg     case wasm::WASM_EXTERNAL_MEMORY:
13042ffff5afSSam Clegg     case wasm::WASM_EXTERNAL_TABLE:
13052ffff5afSSam Clegg       break;
1306d3d84fddSDerek Schuff     default:
130784c6f325SSam Clegg       return make_error<GenericBinaryError>("unexpected export kind",
1308f208f631SHeejin Ahn                                             object_error::parse_failed);
1309d3d84fddSDerek Schuff     }
1310b7787fd0SSam Clegg     Exports.push_back(Ex);
1311d3d84fddSDerek Schuff   }
1312b7c62394SSam Clegg   if (Ctx.Ptr != Ctx.End)
131384c6f325SSam Clegg     return make_error<GenericBinaryError>("export section ended prematurely",
1314d3d84fddSDerek Schuff                                           object_error::parse_failed);
1315d3d84fddSDerek Schuff   return Error::success();
1316d3d84fddSDerek Schuff }
1317d3d84fddSDerek Schuff 
isValidFunctionIndex(uint32_t Index) const13184273998cSSam Clegg bool WasmObjectFile::isValidFunctionIndex(uint32_t Index) const {
1319c0039de2SSam Clegg   return Index < NumImportedFunctions + Functions.size();
13204273998cSSam Clegg }
13214273998cSSam Clegg 
isDefinedFunctionIndex(uint32_t Index) const132223012e98SSam Clegg bool WasmObjectFile::isDefinedFunctionIndex(uint32_t Index) const {
132323012e98SSam Clegg   return Index >= NumImportedFunctions && isValidFunctionIndex(Index);
132423012e98SSam Clegg }
132523012e98SSam Clegg 
isValidGlobalIndex(uint32_t Index) const13266c899ba6SSam Clegg bool WasmObjectFile::isValidGlobalIndex(uint32_t Index) const {
13276c899ba6SSam Clegg   return Index < NumImportedGlobals + Globals.size();
13286c899ba6SSam Clegg }
13296c899ba6SSam Clegg 
isValidTableNumber(uint32_t Index) const1330a5a3659dSAndy Wingo bool WasmObjectFile::isValidTableNumber(uint32_t Index) const {
1331388fb67bSPaulo Matos   return Index < NumImportedTables + Tables.size();
1332388fb67bSPaulo Matos }
1333388fb67bSPaulo Matos 
isDefinedGlobalIndex(uint32_t Index) const13346c899ba6SSam Clegg bool WasmObjectFile::isDefinedGlobalIndex(uint32_t Index) const {
13356c899ba6SSam Clegg   return Index >= NumImportedGlobals && isValidGlobalIndex(Index);
13366c899ba6SSam Clegg }
13376c899ba6SSam Clegg 
isDefinedTableNumber(uint32_t Index) const1338a5a3659dSAndy Wingo bool WasmObjectFile::isDefinedTableNumber(uint32_t Index) const {
1339a5a3659dSAndy Wingo   return Index >= NumImportedTables && isValidTableNumber(Index);
1340388fb67bSPaulo Matos }
1341388fb67bSPaulo Matos 
isValidTagIndex(uint32_t Index) const13421d891d44SHeejin Ahn bool WasmObjectFile::isValidTagIndex(uint32_t Index) const {
13431d891d44SHeejin Ahn   return Index < NumImportedTags + Tags.size();
1344da419bdbSHeejin Ahn }
1345da419bdbSHeejin Ahn 
isDefinedTagIndex(uint32_t Index) const13461d891d44SHeejin Ahn bool WasmObjectFile::isDefinedTagIndex(uint32_t Index) const {
13471d891d44SHeejin Ahn   return Index >= NumImportedTags && isValidTagIndex(Index);
1348da419bdbSHeejin Ahn }
1349da419bdbSHeejin Ahn 
isValidFunctionSymbol(uint32_t Index) const1350b3748f71SNicholas Wilson bool WasmObjectFile::isValidFunctionSymbol(uint32_t Index) const {
13516c899ba6SSam Clegg   return Index < Symbols.size() && Symbols[Index].isTypeFunction();
13526c899ba6SSam Clegg }
13536c899ba6SSam Clegg 
isValidTableSymbol(uint32_t Index) const135469e2797eSPaulo Matos bool WasmObjectFile::isValidTableSymbol(uint32_t Index) const {
135569e2797eSPaulo Matos   return Index < Symbols.size() && Symbols[Index].isTypeTable();
135669e2797eSPaulo Matos }
135769e2797eSPaulo Matos 
isValidGlobalSymbol(uint32_t Index) const1358b3748f71SNicholas Wilson bool WasmObjectFile::isValidGlobalSymbol(uint32_t Index) const {
1359b3748f71SNicholas Wilson   return Index < Symbols.size() && Symbols[Index].isTypeGlobal();
1360b3748f71SNicholas Wilson }
1361b3748f71SNicholas Wilson 
isValidTagSymbol(uint32_t Index) const13621d891d44SHeejin Ahn bool WasmObjectFile::isValidTagSymbol(uint32_t Index) const {
13631d891d44SHeejin Ahn   return Index < Symbols.size() && Symbols[Index].isTypeTag();
1364da419bdbSHeejin Ahn }
1365da419bdbSHeejin Ahn 
isValidDataSymbol(uint32_t Index) const1366b3748f71SNicholas Wilson bool WasmObjectFile::isValidDataSymbol(uint32_t Index) const {
1367b3748f71SNicholas Wilson   return Index < Symbols.size() && Symbols[Index].isTypeData();
1368b3748f71SNicholas Wilson }
1369b3748f71SNicholas Wilson 
isValidSectionSymbol(uint32_t Index) const13706a31a0d6SSam Clegg bool WasmObjectFile::isValidSectionSymbol(uint32_t Index) const {
13716a31a0d6SSam Clegg   return Index < Symbols.size() && Symbols[Index].isTypeSection();
13726a31a0d6SSam Clegg }
13736a31a0d6SSam Clegg 
getDefinedFunction(uint32_t Index)137423012e98SSam Clegg wasm::WasmFunction &WasmObjectFile::getDefinedFunction(uint32_t Index) {
137523012e98SSam Clegg   assert(isDefinedFunctionIndex(Index));
137623012e98SSam Clegg   return Functions[Index - NumImportedFunctions];
137723012e98SSam Clegg }
137823012e98SSam Clegg 
1379f3b762a0SWouter van Oortmerssen const wasm::WasmFunction &
getDefinedFunction(uint32_t Index) const1380f3b762a0SWouter van Oortmerssen WasmObjectFile::getDefinedFunction(uint32_t Index) const {
1381f3b762a0SWouter van Oortmerssen   assert(isDefinedFunctionIndex(Index));
1382f3b762a0SWouter van Oortmerssen   return Functions[Index - NumImportedFunctions];
1383f3b762a0SWouter van Oortmerssen }
1384f3b762a0SWouter van Oortmerssen 
getDefinedGlobal(uint32_t Index)13856c899ba6SSam Clegg wasm::WasmGlobal &WasmObjectFile::getDefinedGlobal(uint32_t Index) {
13866c899ba6SSam Clegg   assert(isDefinedGlobalIndex(Index));
13876c899ba6SSam Clegg   return Globals[Index - NumImportedGlobals];
13886c899ba6SSam Clegg }
13896c899ba6SSam Clegg 
getDefinedTag(uint32_t Index)13901d891d44SHeejin Ahn wasm::WasmTag &WasmObjectFile::getDefinedTag(uint32_t Index) {
13911d891d44SHeejin Ahn   assert(isDefinedTagIndex(Index));
13921d891d44SHeejin Ahn   return Tags[Index - NumImportedTags];
1393da419bdbSHeejin Ahn }
1394da419bdbSHeejin Ahn 
parseStartSection(ReadContext & Ctx)1395b7c62394SSam Clegg Error WasmObjectFile::parseStartSection(ReadContext &Ctx) {
1396b7c62394SSam Clegg   StartFunction = readVaruint32(Ctx);
13974273998cSSam Clegg   if (!isValidFunctionIndex(StartFunction))
139884c6f325SSam Clegg     return make_error<GenericBinaryError>("invalid start function",
1399d3d84fddSDerek Schuff                                           object_error::parse_failed);
1400d3d84fddSDerek Schuff   return Error::success();
1401d3d84fddSDerek Schuff }
1402d3d84fddSDerek Schuff 
parseCodeSection(ReadContext & Ctx)1403b7c62394SSam Clegg Error WasmObjectFile::parseCodeSection(ReadContext &Ctx) {
14040b55ccf6SSam Clegg   CodeSection = Sections.size();
1405b7c62394SSam Clegg   uint32_t FunctionCount = readVaruint32(Ctx);
1406c0039de2SSam Clegg   if (FunctionCount != Functions.size()) {
140784c6f325SSam Clegg     return make_error<GenericBinaryError>("invalid function count",
1408d3d84fddSDerek Schuff                                           object_error::parse_failed);
1409d3d84fddSDerek Schuff   }
1410d3d84fddSDerek Schuff 
1411881d8778SSam Clegg   for (uint32_t i = 0; i < FunctionCount; i++) {
1412881d8778SSam Clegg     wasm::WasmFunction& Function = Functions[i];
1413b7c62394SSam Clegg     const uint8_t *FunctionStart = Ctx.Ptr;
1414b7c62394SSam Clegg     uint32_t Size = readVaruint32(Ctx);
1415b7c62394SSam Clegg     const uint8_t *FunctionEnd = Ctx.Ptr + Size;
1416c551522dSSam Clegg 
1417b7c62394SSam Clegg     Function.CodeOffset = Ctx.Ptr - FunctionStart;
1418881d8778SSam Clegg     Function.Index = NumImportedFunctions + i;
1419b7c62394SSam Clegg     Function.CodeSectionOffset = FunctionStart - Ctx.Start;
1420c551522dSSam Clegg     Function.Size = FunctionEnd - FunctionStart;
1421d3d84fddSDerek Schuff 
1422b7c62394SSam Clegg     uint32_t NumLocalDecls = readVaruint32(Ctx);
1423d3d84fddSDerek Schuff     Function.Locals.reserve(NumLocalDecls);
1424d3d84fddSDerek Schuff     while (NumLocalDecls--) {
1425d3d84fddSDerek Schuff       wasm::WasmLocalDecl Decl;
1426b7c62394SSam Clegg       Decl.Count = readVaruint32(Ctx);
1427b7c62394SSam Clegg       Decl.Type = readUint8(Ctx);
1428d3d84fddSDerek Schuff       Function.Locals.push_back(Decl);
1429d3d84fddSDerek Schuff     }
1430d3d84fddSDerek Schuff 
1431b7c62394SSam Clegg     uint32_t BodySize = FunctionEnd - Ctx.Ptr;
1432b7c62394SSam Clegg     Function.Body = ArrayRef<uint8_t>(Ctx.Ptr, BodySize);
1433027b9357SNicholas Wilson     // This will be set later when reading in the linking metadata section.
1434027b9357SNicholas Wilson     Function.Comdat = UINT32_MAX;
1435b7c62394SSam Clegg     Ctx.Ptr += BodySize;
1436b7c62394SSam Clegg     assert(Ctx.Ptr == FunctionEnd);
1437d3d84fddSDerek Schuff   }
1438b7c62394SSam Clegg   if (Ctx.Ptr != Ctx.End)
143984c6f325SSam Clegg     return make_error<GenericBinaryError>("code section ended prematurely",
1440d3d84fddSDerek Schuff                                           object_error::parse_failed);
1441d3d84fddSDerek Schuff   return Error::success();
1442d3d84fddSDerek Schuff }
1443d3d84fddSDerek Schuff 
parseElemSection(ReadContext & Ctx)1444b7c62394SSam Clegg Error WasmObjectFile::parseElemSection(ReadContext &Ctx) {
1445b7c62394SSam Clegg   uint32_t Count = readVaruint32(Ctx);
1446d3d84fddSDerek Schuff   ElemSegments.reserve(Count);
1447d3d84fddSDerek Schuff   while (Count--) {
1448d3d84fddSDerek Schuff     wasm::WasmElemSegment Segment;
1449a5a3659dSAndy Wingo     Segment.Flags = readVaruint32(Ctx);
1450a5a3659dSAndy Wingo 
1451a5a3659dSAndy Wingo     uint32_t SupportedFlags = wasm::WASM_ELEM_SEGMENT_HAS_TABLE_NUMBER |
1452a5a3659dSAndy Wingo                               wasm::WASM_ELEM_SEGMENT_IS_PASSIVE |
1453a5a3659dSAndy Wingo                               wasm::WASM_ELEM_SEGMENT_HAS_INIT_EXPRS;
1454a5a3659dSAndy Wingo     if (Segment.Flags & ~SupportedFlags)
1455a5a3659dSAndy Wingo       return make_error<GenericBinaryError>(
1456a5a3659dSAndy Wingo           "Unsupported flags for element segment", object_error::parse_failed);
1457a5a3659dSAndy Wingo 
1458a5a3659dSAndy Wingo     if (Segment.Flags & wasm::WASM_ELEM_SEGMENT_HAS_TABLE_NUMBER)
1459a5a3659dSAndy Wingo       Segment.TableNumber = readVaruint32(Ctx);
1460a5a3659dSAndy Wingo     else
1461a5a3659dSAndy Wingo       Segment.TableNumber = 0;
1462a5a3659dSAndy Wingo     if (!isValidTableNumber(Segment.TableNumber))
1463a5a3659dSAndy Wingo       return make_error<GenericBinaryError>("invalid TableNumber",
1464d3d84fddSDerek Schuff                                             object_error::parse_failed);
1465a5a3659dSAndy Wingo 
1466a5a3659dSAndy Wingo     if (Segment.Flags & wasm::WASM_ELEM_SEGMENT_IS_PASSIVE) {
14672481adb5SSam Clegg       Segment.Offset.Extended = false;
14689504ab32SSam Clegg       Segment.Offset.Inst.Opcode = wasm::WASM_OPCODE_I32_CONST;
14699504ab32SSam Clegg       Segment.Offset.Inst.Value.Int32 = 0;
1470a5a3659dSAndy Wingo     } else {
1471b7c62394SSam Clegg       if (Error Err = readInitExpr(Segment.Offset, Ctx))
1472d3d84fddSDerek Schuff         return Err;
1473a5a3659dSAndy Wingo     }
1474a5a3659dSAndy Wingo 
1475a5a3659dSAndy Wingo     if (Segment.Flags & wasm::WASM_ELEM_SEGMENT_MASK_HAS_ELEM_KIND) {
1476a5a3659dSAndy Wingo       Segment.ElemKind = readUint8(Ctx);
1477a5a3659dSAndy Wingo       if (Segment.Flags & wasm::WASM_ELEM_SEGMENT_HAS_INIT_EXPRS) {
1478a5a3659dSAndy Wingo         if (Segment.ElemKind != uint8_t(wasm::ValType::FUNCREF) &&
1479a5a3659dSAndy Wingo             Segment.ElemKind != uint8_t(wasm::ValType::EXTERNREF)) {
1480a5a3659dSAndy Wingo           return make_error<GenericBinaryError>("invalid reference type",
1481a5a3659dSAndy Wingo                                                 object_error::parse_failed);
1482a5a3659dSAndy Wingo         }
1483a5a3659dSAndy Wingo       } else {
1484a5a3659dSAndy Wingo         if (Segment.ElemKind != 0)
1485a5a3659dSAndy Wingo           return make_error<GenericBinaryError>("invalid elemtype",
1486a5a3659dSAndy Wingo                                                 object_error::parse_failed);
1487a5a3659dSAndy Wingo         Segment.ElemKind = uint8_t(wasm::ValType::FUNCREF);
1488a5a3659dSAndy Wingo       }
1489a5a3659dSAndy Wingo     } else {
1490a5a3659dSAndy Wingo       Segment.ElemKind = uint8_t(wasm::ValType::FUNCREF);
1491a5a3659dSAndy Wingo     }
1492a5a3659dSAndy Wingo 
1493a5a3659dSAndy Wingo     if (Segment.Flags & wasm::WASM_ELEM_SEGMENT_HAS_INIT_EXPRS)
1494a5a3659dSAndy Wingo       return make_error<GenericBinaryError>(
1495a5a3659dSAndy Wingo           "elem segment init expressions not yet implemented",
1496a5a3659dSAndy Wingo           object_error::parse_failed);
1497a5a3659dSAndy Wingo 
1498b7c62394SSam Clegg     uint32_t NumElems = readVaruint32(Ctx);
1499d3d84fddSDerek Schuff     while (NumElems--) {
1500b7c62394SSam Clegg       Segment.Functions.push_back(readVaruint32(Ctx));
1501d3d84fddSDerek Schuff     }
1502d3d84fddSDerek Schuff     ElemSegments.push_back(Segment);
1503d3d84fddSDerek Schuff   }
1504b7c62394SSam Clegg   if (Ctx.Ptr != Ctx.End)
150584c6f325SSam Clegg     return make_error<GenericBinaryError>("elem section ended prematurely",
1506d3d84fddSDerek Schuff                                           object_error::parse_failed);
1507d3d84fddSDerek Schuff   return Error::success();
1508d3d84fddSDerek Schuff }
1509d3d84fddSDerek Schuff 
parseDataSection(ReadContext & Ctx)1510b7c62394SSam Clegg Error WasmObjectFile::parseDataSection(ReadContext &Ctx) {
15110b55ccf6SSam Clegg   DataSection = Sections.size();
1512b7c62394SSam Clegg   uint32_t Count = readVaruint32(Ctx);
15137a47ee51SKazu Hirata   if (DataCount && Count != *DataCount)
1514fef8de66SThomas Lively     return make_error<GenericBinaryError>(
151584c6f325SSam Clegg         "number of data segments does not match DataCount section");
1516d3d84fddSDerek Schuff   DataSegments.reserve(Count);
1517d3d84fddSDerek Schuff   while (Count--) {
15189c07f94aSSam Clegg     WasmSegment Segment;
15192e150409SThomas Lively     Segment.Data.InitFlags = readVaruint32(Ctx);
1520831a143eSAndy Wingo     Segment.Data.MemoryIndex =
1521831a143eSAndy Wingo         (Segment.Data.InitFlags & wasm::WASM_DATA_SEGMENT_HAS_MEMINDEX)
1522831a143eSAndy Wingo             ? readVaruint32(Ctx)
1523831a143eSAndy Wingo             : 0;
1524831a143eSAndy Wingo     if ((Segment.Data.InitFlags & wasm::WASM_DATA_SEGMENT_IS_PASSIVE) == 0) {
1525b7c62394SSam Clegg       if (Error Err = readInitExpr(Segment.Data.Offset, Ctx))
1526d3d84fddSDerek Schuff         return Err;
15272e150409SThomas Lively     } else {
15289b27fbd1SSam Clegg       Segment.Data.Offset.Extended = false;
15299504ab32SSam Clegg       Segment.Data.Offset.Inst.Opcode = wasm::WASM_OPCODE_I32_CONST;
15309504ab32SSam Clegg       Segment.Data.Offset.Inst.Value.Int32 = 0;
15312e150409SThomas Lively     }
1532b7c62394SSam Clegg     uint32_t Size = readVaruint32(Ctx);
1533dcf5bd27SSimon Pilgrim     if (Size > (size_t)(Ctx.End - Ctx.Ptr))
153484c6f325SSam Clegg       return make_error<GenericBinaryError>("invalid segment size",
15354bbc6b55SSam Clegg                                             object_error::parse_failed);
1536b7c62394SSam Clegg     Segment.Data.Content = ArrayRef<uint8_t>(Ctx.Ptr, Size);
1537027b9357SNicholas Wilson     // The rest of these Data fields are set later, when reading in the linking
1538027b9357SNicholas Wilson     // metadata section.
153963ebb813SSam Clegg     Segment.Data.Alignment = 0;
15403b8d2be5SSam Clegg     Segment.Data.LinkingFlags = 0;
1541027b9357SNicholas Wilson     Segment.Data.Comdat = UINT32_MAX;
1542b7c62394SSam Clegg     Segment.SectionOffset = Ctx.Ptr - Ctx.Start;
1543b7c62394SSam Clegg     Ctx.Ptr += Size;
1544d3d84fddSDerek Schuff     DataSegments.push_back(Segment);
1545d3d84fddSDerek Schuff   }
1546b7c62394SSam Clegg   if (Ctx.Ptr != Ctx.End)
154784c6f325SSam Clegg     return make_error<GenericBinaryError>("data section ended prematurely",
1548d3d84fddSDerek Schuff                                           object_error::parse_failed);
15492c6f75ddSDerek Schuff   return Error::success();
15502c6f75ddSDerek Schuff }
15512c6f75ddSDerek Schuff 
parseDataCountSection(ReadContext & Ctx)1552fef8de66SThomas Lively Error WasmObjectFile::parseDataCountSection(ReadContext &Ctx) {
1553fef8de66SThomas Lively   DataCount = readVaruint32(Ctx);
1554fef8de66SThomas Lively   return Error::success();
1555fef8de66SThomas Lively }
1556fef8de66SThomas Lively 
getHeader() const15572c6f75ddSDerek Schuff const wasm::WasmObjectHeader &WasmObjectFile::getHeader() const {
15582c6f75ddSDerek Schuff   return Header;
15592c6f75ddSDerek Schuff }
15602c6f75ddSDerek Schuff 
moveSymbolNext(DataRefImpl & Symb) const15612a193e0dSSam Clegg void WasmObjectFile::moveSymbolNext(DataRefImpl &Symb) const { Symb.d.b++; }
15622c6f75ddSDerek Schuff 
getSymbolFlags(DataRefImpl Symb) const1563ac00376aSvgxbj Expected<uint32_t> WasmObjectFile::getSymbolFlags(DataRefImpl Symb) const {
1564fc5b5cd2SSam Clegg   uint32_t Result = SymbolRef::SF_None;
1565d3d84fddSDerek Schuff   const WasmSymbol &Sym = getWasmSymbol(Symb);
1566fc5b5cd2SSam Clegg 
1567d34e60caSNicola Zaghen   LLVM_DEBUG(dbgs() << "getSymbolFlags: ptr=" << &Sym << " " << Sym << "\n");
15686e7f1826SSam Clegg   if (Sym.isBindingWeak())
1569b7787fd0SSam Clegg     Result |= SymbolRef::SF_Weak;
15706e7f1826SSam Clegg   if (!Sym.isBindingLocal())
157131a2c809SSam Clegg     Result |= SymbolRef::SF_Global;
1572a2b35dacSSam Clegg   if (Sym.isHidden())
1573a2b35dacSSam Clegg     Result |= SymbolRef::SF_Hidden;
15746c899ba6SSam Clegg   if (!Sym.isDefined())
1575fc5b5cd2SSam Clegg     Result |= SymbolRef::SF_Undefined;
15766c899ba6SSam Clegg   if (Sym.isTypeFunction())
15776c899ba6SSam Clegg     Result |= SymbolRef::SF_Executable;
1578fc5b5cd2SSam Clegg   return Result;
15792c6f75ddSDerek Schuff }
15802c6f75ddSDerek Schuff 
symbol_begin() const15812c6f75ddSDerek Schuff basic_symbol_iterator WasmObjectFile::symbol_begin() const {
1582d3d84fddSDerek Schuff   DataRefImpl Ref;
15832a193e0dSSam Clegg   Ref.d.a = 1; // Arbitrary non-zero value so that Ref.p is non-null
15842a193e0dSSam Clegg   Ref.d.b = 0; // Symbol index
1585d3d84fddSDerek Schuff   return BasicSymbolRef(Ref, this);
15862c6f75ddSDerek Schuff }
15872c6f75ddSDerek Schuff 
symbol_end() const15882c6f75ddSDerek Schuff basic_symbol_iterator WasmObjectFile::symbol_end() const {
1589d3d84fddSDerek Schuff   DataRefImpl Ref;
15902a193e0dSSam Clegg   Ref.d.a = 1; // Arbitrary non-zero value so that Ref.p is non-null
15912a193e0dSSam Clegg   Ref.d.b = Symbols.size(); // Symbol index
1592d3d84fddSDerek Schuff   return BasicSymbolRef(Ref, this);
1593d3d84fddSDerek Schuff }
1594d3d84fddSDerek Schuff 
getWasmSymbol(const DataRefImpl & Symb) const15952ffff5afSSam Clegg const WasmSymbol &WasmObjectFile::getWasmSymbol(const DataRefImpl &Symb) const {
15962a193e0dSSam Clegg   return Symbols[Symb.d.b];
15972c6f75ddSDerek Schuff }
15982c6f75ddSDerek Schuff 
getWasmSymbol(const SymbolRef & Symb) const15992ffff5afSSam Clegg const WasmSymbol &WasmObjectFile::getWasmSymbol(const SymbolRef &Symb) const {
16002ffff5afSSam Clegg   return getWasmSymbol(Symb.getRawDataRefImpl());
16012ffff5afSSam Clegg }
16022ffff5afSSam Clegg 
getSymbolName(DataRefImpl Symb) const16032c6f75ddSDerek Schuff Expected<StringRef> WasmObjectFile::getSymbolName(DataRefImpl Symb) const {
16046c899ba6SSam Clegg   return getWasmSymbol(Symb).Info.Name;
16052c6f75ddSDerek Schuff }
16062c6f75ddSDerek Schuff 
getSymbolAddress(DataRefImpl Symb) const16072c6f75ddSDerek Schuff Expected<uint64_t> WasmObjectFile::getSymbolAddress(DataRefImpl Symb) const {
1608f3b762a0SWouter van Oortmerssen   auto &Sym = getWasmSymbol(Symb);
1609f3b762a0SWouter van Oortmerssen   if (Sym.Info.Kind == wasm::WASM_SYMBOL_TYPE_FUNCTION &&
1610f3b762a0SWouter van Oortmerssen       isDefinedFunctionIndex(Sym.Info.ElementIndex))
1611f3b762a0SWouter van Oortmerssen     return getDefinedFunction(Sym.Info.ElementIndex).CodeSectionOffset;
1612f3b762a0SWouter van Oortmerssen   else
1613fc5b5cd2SSam Clegg     return getSymbolValue(Symb);
16142c6f75ddSDerek Schuff }
16152c6f75ddSDerek Schuff 
getWasmSymbolValue(const WasmSymbol & Sym) const161631a2c809SSam Clegg uint64_t WasmObjectFile::getWasmSymbolValue(const WasmSymbol &Sym) const {
16176c899ba6SSam Clegg   switch (Sym.Info.Kind) {
16186c899ba6SSam Clegg   case wasm::WASM_SYMBOL_TYPE_FUNCTION:
16196c899ba6SSam Clegg   case wasm::WASM_SYMBOL_TYPE_GLOBAL:
16201d891d44SHeejin Ahn   case wasm::WASM_SYMBOL_TYPE_TAG:
1621feac819eSAndy Wingo   case wasm::WASM_SYMBOL_TYPE_TABLE:
16226c899ba6SSam Clegg     return Sym.Info.ElementIndex;
16236c899ba6SSam Clegg   case wasm::WASM_SYMBOL_TYPE_DATA: {
16246c899ba6SSam Clegg     // The value of a data symbol is the segment offset, plus the symbol
16256c899ba6SSam Clegg     // offset within the segment.
16266c899ba6SSam Clegg     uint32_t SegmentIndex = Sym.Info.DataRef.Segment;
16276c899ba6SSam Clegg     const wasm::WasmDataSegment &Segment = DataSegments[SegmentIndex].Data;
16289504ab32SSam Clegg     if (Segment.Offset.Extended) {
16299504ab32SSam Clegg       llvm_unreachable("extended init exprs not supported");
16309504ab32SSam Clegg     } else if (Segment.Offset.Inst.Opcode == wasm::WASM_OPCODE_I32_CONST) {
16319504ab32SSam Clegg       return Segment.Offset.Inst.Value.Int32 + Sym.Info.DataRef.Offset;
16329504ab32SSam Clegg     } else if (Segment.Offset.Inst.Opcode == wasm::WASM_OPCODE_I64_CONST) {
16339504ab32SSam Clegg       return Segment.Offset.Inst.Value.Int64 + Sym.Info.DataRef.Offset;
16343b29376eSWouter van Oortmerssen     } else {
16353b29376eSWouter van Oortmerssen       llvm_unreachable("unknown init expr opcode");
16363b29376eSWouter van Oortmerssen     }
1637b09cfa51SSam Clegg   }
16386a31a0d6SSam Clegg   case wasm::WASM_SYMBOL_TYPE_SECTION:
16396a31a0d6SSam Clegg     return 0;
16402c6f75ddSDerek Schuff   }
16417f055deeSSam Clegg   llvm_unreachable("invalid symbol type");
1642b7787fd0SSam Clegg }
16432c6f75ddSDerek Schuff 
getSymbolValueImpl(DataRefImpl Symb) const164431a2c809SSam Clegg uint64_t WasmObjectFile::getSymbolValueImpl(DataRefImpl Symb) const {
164531a2c809SSam Clegg   return getWasmSymbolValue(getWasmSymbol(Symb));
164631a2c809SSam Clegg }
164731a2c809SSam Clegg 
getSymbolAlignment(DataRefImpl Symb) const16482c6f75ddSDerek Schuff uint32_t WasmObjectFile::getSymbolAlignment(DataRefImpl Symb) const {
16492c6f75ddSDerek Schuff   llvm_unreachable("not yet implemented");
16502c6f75ddSDerek Schuff   return 0;
16512c6f75ddSDerek Schuff }
16522c6f75ddSDerek Schuff 
getCommonSymbolSizeImpl(DataRefImpl Symb) const16532c6f75ddSDerek Schuff uint64_t WasmObjectFile::getCommonSymbolSizeImpl(DataRefImpl Symb) const {
16542c6f75ddSDerek Schuff   llvm_unreachable("not yet implemented");
16552c6f75ddSDerek Schuff   return 0;
16562c6f75ddSDerek Schuff }
16572c6f75ddSDerek Schuff 
16582c6f75ddSDerek Schuff Expected<SymbolRef::Type>
getSymbolType(DataRefImpl Symb) const16592c6f75ddSDerek Schuff WasmObjectFile::getSymbolType(DataRefImpl Symb) const {
1660fc5b5cd2SSam Clegg   const WasmSymbol &Sym = getWasmSymbol(Symb);
1661fc5b5cd2SSam Clegg 
16626c899ba6SSam Clegg   switch (Sym.Info.Kind) {
16636c899ba6SSam Clegg   case wasm::WASM_SYMBOL_TYPE_FUNCTION:
1664fc5b5cd2SSam Clegg     return SymbolRef::ST_Function;
16656c899ba6SSam Clegg   case wasm::WASM_SYMBOL_TYPE_GLOBAL:
16666c899ba6SSam Clegg     return SymbolRef::ST_Other;
16676c899ba6SSam Clegg   case wasm::WASM_SYMBOL_TYPE_DATA:
1668fc5b5cd2SSam Clegg     return SymbolRef::ST_Data;
16696a31a0d6SSam Clegg   case wasm::WASM_SYMBOL_TYPE_SECTION:
16706a31a0d6SSam Clegg     return SymbolRef::ST_Debug;
16711d891d44SHeejin Ahn   case wasm::WASM_SYMBOL_TYPE_TAG:
1672da419bdbSHeejin Ahn     return SymbolRef::ST_Other;
1673feac819eSAndy Wingo   case wasm::WASM_SYMBOL_TYPE_TABLE:
1674feac819eSAndy Wingo     return SymbolRef::ST_Other;
1675fc5b5cd2SSam Clegg   }
1676fc5b5cd2SSam Clegg 
167784c6f325SSam Clegg   llvm_unreachable("unknown WasmSymbol::SymbolType");
1678fc5b5cd2SSam Clegg   return SymbolRef::ST_Other;
16792c6f75ddSDerek Schuff }
16802c6f75ddSDerek Schuff 
16812c6f75ddSDerek Schuff Expected<section_iterator>
getSymbolSection(DataRefImpl Symb) const16822c6f75ddSDerek Schuff WasmObjectFile::getSymbolSection(DataRefImpl Symb) const {
16830b55ccf6SSam Clegg   const WasmSymbol &Sym = getWasmSymbol(Symb);
16846c899ba6SSam Clegg   if (Sym.isUndefined())
16850b55ccf6SSam Clegg     return section_end();
16866c899ba6SSam Clegg 
16876c899ba6SSam Clegg   DataRefImpl Ref;
1688da6a896eSDerek Schuff   Ref.d.a = getSymbolSectionIdImpl(Sym);
1689da6a896eSDerek Schuff   return section_iterator(SectionRef(Ref, this));
1690da6a896eSDerek Schuff }
1691da6a896eSDerek Schuff 
getSymbolSectionId(SymbolRef Symb) const1692da6a896eSDerek Schuff uint32_t WasmObjectFile::getSymbolSectionId(SymbolRef Symb) const {
1693da6a896eSDerek Schuff   const WasmSymbol &Sym = getWasmSymbol(Symb);
1694da6a896eSDerek Schuff   return getSymbolSectionIdImpl(Sym);
1695da6a896eSDerek Schuff }
1696da6a896eSDerek Schuff 
getSymbolSectionIdImpl(const WasmSymbol & Sym) const1697da6a896eSDerek Schuff uint32_t WasmObjectFile::getSymbolSectionIdImpl(const WasmSymbol &Sym) const {
16986c899ba6SSam Clegg   switch (Sym.Info.Kind) {
16996c899ba6SSam Clegg   case wasm::WASM_SYMBOL_TYPE_FUNCTION:
1700da6a896eSDerek Schuff     return CodeSection;
17016c899ba6SSam Clegg   case wasm::WASM_SYMBOL_TYPE_GLOBAL:
1702da6a896eSDerek Schuff     return GlobalSection;
17036c899ba6SSam Clegg   case wasm::WASM_SYMBOL_TYPE_DATA:
1704da6a896eSDerek Schuff     return DataSection;
1705da419bdbSHeejin Ahn   case wasm::WASM_SYMBOL_TYPE_SECTION:
1706da6a896eSDerek Schuff     return Sym.Info.ElementIndex;
17071d891d44SHeejin Ahn   case wasm::WASM_SYMBOL_TYPE_TAG:
17081d891d44SHeejin Ahn     return TagSection;
1709feac819eSAndy Wingo   case wasm::WASM_SYMBOL_TYPE_TABLE:
1710feac819eSAndy Wingo     return TableSection;
17116c899ba6SSam Clegg   default:
171284c6f325SSam Clegg     llvm_unreachable("unknown WasmSymbol::SymbolType");
17136c899ba6SSam Clegg   }
17142c6f75ddSDerek Schuff }
17152c6f75ddSDerek Schuff 
moveSectionNext(DataRefImpl & Sec) const17162c6f75ddSDerek Schuff void WasmObjectFile::moveSectionNext(DataRefImpl &Sec) const { Sec.d.a++; }
17172c6f75ddSDerek Schuff 
getSectionName(DataRefImpl Sec) const17188be28cdcSFangrui Song Expected<StringRef> WasmObjectFile::getSectionName(DataRefImpl Sec) const {
1719d3d84fddSDerek Schuff   const WasmSection &S = Sections[Sec.d.a];
1720a205f290SDerek Schuff   if (S.Type == wasm::WASM_SEC_CUSTOM)
17218be28cdcSFangrui Song     return S.Name;
17222ae385e5SDerek Schuff   if (S.Type > wasm::WASM_SEC_LAST_KNOWN)
17238be28cdcSFangrui Song     return createStringError(object_error::invalid_section_index, "");
1724a205f290SDerek Schuff   return wasm::sectionTypeToString(S.Type);
17252c6f75ddSDerek Schuff }
17262c6f75ddSDerek Schuff 
getSectionAddress(DataRefImpl Sec) const17272c6f75ddSDerek Schuff uint64_t WasmObjectFile::getSectionAddress(DataRefImpl Sec) const { return 0; }
17282c6f75ddSDerek Schuff 
getSectionIndex(DataRefImpl Sec) const1729a25d329bSGeorge Rimar uint64_t WasmObjectFile::getSectionIndex(DataRefImpl Sec) const {
1730a25d329bSGeorge Rimar   return Sec.d.a;
1731a25d329bSGeorge Rimar }
1732a25d329bSGeorge Rimar 
getSectionSize(DataRefImpl Sec) const17332c6f75ddSDerek Schuff uint64_t WasmObjectFile::getSectionSize(DataRefImpl Sec) const {
1734d3d84fddSDerek Schuff   const WasmSection &S = Sections[Sec.d.a];
17352c6f75ddSDerek Schuff   return S.Content.size();
17362c6f75ddSDerek Schuff }
17372c6f75ddSDerek Schuff 
1738e1cb2c0fSFangrui Song Expected<ArrayRef<uint8_t>>
getSectionContents(DataRefImpl Sec) const1739e1cb2c0fSFangrui Song WasmObjectFile::getSectionContents(DataRefImpl Sec) const {
1740d3d84fddSDerek Schuff   const WasmSection &S = Sections[Sec.d.a];
17412c6f75ddSDerek Schuff   // This will never fail since wasm sections can never be empty (user-sections
17422c6f75ddSDerek Schuff   // must have a name and non-user sections each have a defined structure).
1743e1cb2c0fSFangrui Song   return S.Content;
17442c6f75ddSDerek Schuff }
17452c6f75ddSDerek Schuff 
getSectionAlignment(DataRefImpl Sec) const17462c6f75ddSDerek Schuff uint64_t WasmObjectFile::getSectionAlignment(DataRefImpl Sec) const {
17472c6f75ddSDerek Schuff   return 1;
17482c6f75ddSDerek Schuff }
17492c6f75ddSDerek Schuff 
isSectionCompressed(DataRefImpl Sec) const17502c6f75ddSDerek Schuff bool WasmObjectFile::isSectionCompressed(DataRefImpl Sec) const {
17512c6f75ddSDerek Schuff   return false;
17522c6f75ddSDerek Schuff }
17532c6f75ddSDerek Schuff 
isSectionText(DataRefImpl Sec) const17542c6f75ddSDerek Schuff bool WasmObjectFile::isSectionText(DataRefImpl Sec) const {
1755d3d84fddSDerek Schuff   return getWasmSection(Sec).Type == wasm::WASM_SEC_CODE;
17562c6f75ddSDerek Schuff }
17572c6f75ddSDerek Schuff 
isSectionData(DataRefImpl Sec) const17582c6f75ddSDerek Schuff bool WasmObjectFile::isSectionData(DataRefImpl Sec) const {
1759d3d84fddSDerek Schuff   return getWasmSection(Sec).Type == wasm::WASM_SEC_DATA;
17602c6f75ddSDerek Schuff }
17612c6f75ddSDerek Schuff 
isSectionBSS(DataRefImpl Sec) const17622c6f75ddSDerek Schuff bool WasmObjectFile::isSectionBSS(DataRefImpl Sec) const { return false; }
17632c6f75ddSDerek Schuff 
isSectionVirtual(DataRefImpl Sec) const17642c6f75ddSDerek Schuff bool WasmObjectFile::isSectionVirtual(DataRefImpl Sec) const { return false; }
17652c6f75ddSDerek Schuff 
section_rel_begin(DataRefImpl Ref) const1766d3d84fddSDerek Schuff relocation_iterator WasmObjectFile::section_rel_begin(DataRefImpl Ref) const {
1767d3d84fddSDerek Schuff   DataRefImpl RelocRef;
1768d3d84fddSDerek Schuff   RelocRef.d.a = Ref.d.a;
1769d3d84fddSDerek Schuff   RelocRef.d.b = 0;
1770d3d84fddSDerek Schuff   return relocation_iterator(RelocationRef(RelocRef, this));
17712c6f75ddSDerek Schuff }
17722c6f75ddSDerek Schuff 
section_rel_end(DataRefImpl Ref) const1773d3d84fddSDerek Schuff relocation_iterator WasmObjectFile::section_rel_end(DataRefImpl Ref) const {
1774d3d84fddSDerek Schuff   const WasmSection &Sec = getWasmSection(Ref);
1775d3d84fddSDerek Schuff   DataRefImpl RelocRef;
1776d3d84fddSDerek Schuff   RelocRef.d.a = Ref.d.a;
1777d3d84fddSDerek Schuff   RelocRef.d.b = Sec.Relocations.size();
1778d3d84fddSDerek Schuff   return relocation_iterator(RelocationRef(RelocRef, this));
17792c6f75ddSDerek Schuff }
17802c6f75ddSDerek Schuff 
moveRelocationNext(DataRefImpl & Rel) const1781f208f631SHeejin Ahn void WasmObjectFile::moveRelocationNext(DataRefImpl &Rel) const { Rel.d.b++; }
17822c6f75ddSDerek Schuff 
getRelocationOffset(DataRefImpl Ref) const1783d3d84fddSDerek Schuff uint64_t WasmObjectFile::getRelocationOffset(DataRefImpl Ref) const {
1784d3d84fddSDerek Schuff   const wasm::WasmRelocation &Rel = getWasmRelocation(Ref);
1785d3d84fddSDerek Schuff   return Rel.Offset;
17862c6f75ddSDerek Schuff }
17872c6f75ddSDerek Schuff 
getRelocationSymbol(DataRefImpl Ref) const1788f676cdd5SSam Clegg symbol_iterator WasmObjectFile::getRelocationSymbol(DataRefImpl Ref) const {
1789f676cdd5SSam Clegg   const wasm::WasmRelocation &Rel = getWasmRelocation(Ref);
1790d1152a26SSam Clegg   if (Rel.Type == wasm::R_WASM_TYPE_INDEX_LEB)
1791f676cdd5SSam Clegg     return symbol_end();
1792f676cdd5SSam Clegg   DataRefImpl Sym;
1793c7d2e5f1SSam Clegg   Sym.d.a = 1;
1794c7d2e5f1SSam Clegg   Sym.d.b = Rel.Index;
1795f676cdd5SSam Clegg   return symbol_iterator(SymbolRef(Sym, this));
17962c6f75ddSDerek Schuff }
17972c6f75ddSDerek Schuff 
getRelocationType(DataRefImpl Ref) const1798d3d84fddSDerek Schuff uint64_t WasmObjectFile::getRelocationType(DataRefImpl Ref) const {
1799d3d84fddSDerek Schuff   const wasm::WasmRelocation &Rel = getWasmRelocation(Ref);
1800d3d84fddSDerek Schuff   return Rel.Type;
18012c6f75ddSDerek Schuff }
18022c6f75ddSDerek Schuff 
getRelocationTypeName(DataRefImpl Ref,SmallVectorImpl<char> & Result) const18032c6f75ddSDerek Schuff void WasmObjectFile::getRelocationTypeName(
1804d3d84fddSDerek Schuff     DataRefImpl Ref, SmallVectorImpl<char> &Result) const {
1805d3d84fddSDerek Schuff   const wasm::WasmRelocation &Rel = getWasmRelocation(Ref);
1806d3d84fddSDerek Schuff   StringRef Res = "Unknown";
1807d3d84fddSDerek Schuff 
1808d3d84fddSDerek Schuff #define WASM_RELOC(name, value)                                                \
1809d3d84fddSDerek Schuff   case wasm::name:                                                             \
1810d3d84fddSDerek Schuff     Res = #name;                                                               \
1811d3d84fddSDerek Schuff     break;
1812d3d84fddSDerek Schuff 
1813d3d84fddSDerek Schuff   switch (Rel.Type) {
1814c5d8bc83SSam Clegg #include "llvm/BinaryFormat/WasmRelocs.def"
1815d3d84fddSDerek Schuff   }
1816d3d84fddSDerek Schuff 
1817d3d84fddSDerek Schuff #undef WASM_RELOC
1818d3d84fddSDerek Schuff 
1819d3d84fddSDerek Schuff   Result.append(Res.begin(), Res.end());
18202c6f75ddSDerek Schuff }
18212c6f75ddSDerek Schuff 
section_begin() const18222c6f75ddSDerek Schuff section_iterator WasmObjectFile::section_begin() const {
18232c6f75ddSDerek Schuff   DataRefImpl Ref;
18242c6f75ddSDerek Schuff   Ref.d.a = 0;
18252c6f75ddSDerek Schuff   return section_iterator(SectionRef(Ref, this));
18262c6f75ddSDerek Schuff }
18272c6f75ddSDerek Schuff 
section_end() const18282c6f75ddSDerek Schuff section_iterator WasmObjectFile::section_end() const {
18292c6f75ddSDerek Schuff   DataRefImpl Ref;
18302c6f75ddSDerek Schuff   Ref.d.a = Sections.size();
18312c6f75ddSDerek Schuff   return section_iterator(SectionRef(Ref, this));
18322c6f75ddSDerek Schuff }
18332c6f75ddSDerek Schuff 
getBytesInAddress() const183429f8c9f6SWouter van Oortmerssen uint8_t WasmObjectFile::getBytesInAddress() const {
183529f8c9f6SWouter van Oortmerssen   return HasMemory64 ? 8 : 4;
183629f8c9f6SWouter van Oortmerssen }
18372c6f75ddSDerek Schuff 
getFileFormatName() const18382c6f75ddSDerek Schuff StringRef WasmObjectFile::getFileFormatName() const { return "WASM"; }
18392c6f75ddSDerek Schuff 
getArch() const184029f8c9f6SWouter van Oortmerssen Triple::ArchType WasmObjectFile::getArch() const {
184129f8c9f6SWouter van Oortmerssen   return HasMemory64 ? Triple::wasm64 : Triple::wasm32;
184229f8c9f6SWouter van Oortmerssen }
18432c6f75ddSDerek Schuff 
getFeatures() const18442c6f75ddSDerek Schuff SubtargetFeatures WasmObjectFile::getFeatures() const {
18452c6f75ddSDerek Schuff   return SubtargetFeatures();
18462c6f75ddSDerek Schuff }
18472c6f75ddSDerek Schuff 
isRelocatableObject() const1848f208f631SHeejin Ahn bool WasmObjectFile::isRelocatableObject() const { return HasLinkingSection; }
18492c6f75ddSDerek Schuff 
isSharedObject() const185003801256SSam Clegg bool WasmObjectFile::isSharedObject() const { return HasDylinkSection; }
185103801256SSam Clegg 
getWasmSection(DataRefImpl Ref) const1852d3d84fddSDerek Schuff const WasmSection &WasmObjectFile::getWasmSection(DataRefImpl Ref) const {
185316fe5822SDavide Italiano   assert(Ref.d.a < Sections.size());
1854d3d84fddSDerek Schuff   return Sections[Ref.d.a];
1855d3d84fddSDerek Schuff }
1856d3d84fddSDerek Schuff 
1857d3d84fddSDerek Schuff const WasmSection &
getWasmSection(const SectionRef & Section) const18582c6f75ddSDerek Schuff WasmObjectFile::getWasmSection(const SectionRef &Section) const {
1859d3d84fddSDerek Schuff   return getWasmSection(Section.getRawDataRefImpl());
1860d3d84fddSDerek Schuff }
1861d3d84fddSDerek Schuff 
1862d3d84fddSDerek Schuff const wasm::WasmRelocation &
getWasmRelocation(const RelocationRef & Ref) const1863d3d84fddSDerek Schuff WasmObjectFile::getWasmRelocation(const RelocationRef &Ref) const {
1864d3d84fddSDerek Schuff   return getWasmRelocation(Ref.getRawDataRefImpl());
1865d3d84fddSDerek Schuff }
1866d3d84fddSDerek Schuff 
1867d3d84fddSDerek Schuff const wasm::WasmRelocation &
getWasmRelocation(DataRefImpl Ref) const1868d3d84fddSDerek Schuff WasmObjectFile::getWasmRelocation(DataRefImpl Ref) const {
186916fe5822SDavide Italiano   assert(Ref.d.a < Sections.size());
1870d3d84fddSDerek Schuff   const WasmSection &Sec = Sections[Ref.d.a];
187116fe5822SDavide Italiano   assert(Ref.d.b < Sec.Relocations.size());
1872d3d84fddSDerek Schuff   return Sec.Relocations[Ref.d.b];
18732c6f75ddSDerek Schuff }
1874feef720bSHeejin Ahn 
getSectionOrder(unsigned ID,StringRef CustomSectionName)1875feef720bSHeejin Ahn int WasmSectionOrderChecker::getSectionOrder(unsigned ID,
1876feef720bSHeejin Ahn                                              StringRef CustomSectionName) {
1877feef720bSHeejin Ahn   switch (ID) {
1878feef720bSHeejin Ahn   case wasm::WASM_SEC_CUSTOM:
1879feef720bSHeejin Ahn     return StringSwitch<unsigned>(CustomSectionName)
1880feef720bSHeejin Ahn         .Case("dylink", WASM_SEC_ORDER_DYLINK)
1881b78c85a4SSam Clegg         .Case("dylink.0", WASM_SEC_ORDER_DYLINK)
1882feef720bSHeejin Ahn         .Case("linking", WASM_SEC_ORDER_LINKING)
1883feef720bSHeejin Ahn         .StartsWith("reloc.", WASM_SEC_ORDER_RELOC)
1884feef720bSHeejin Ahn         .Case("name", WASM_SEC_ORDER_NAME)
1885feef720bSHeejin Ahn         .Case("producers", WASM_SEC_ORDER_PRODUCERS)
1886f6f4f843SThomas Lively         .Case("target_features", WASM_SEC_ORDER_TARGET_FEATURES)
18879757bba4SThomas Lively         .Default(WASM_SEC_ORDER_NONE);
1888feef720bSHeejin Ahn   case wasm::WASM_SEC_TYPE:
1889feef720bSHeejin Ahn     return WASM_SEC_ORDER_TYPE;
1890feef720bSHeejin Ahn   case wasm::WASM_SEC_IMPORT:
1891feef720bSHeejin Ahn     return WASM_SEC_ORDER_IMPORT;
1892feef720bSHeejin Ahn   case wasm::WASM_SEC_FUNCTION:
1893feef720bSHeejin Ahn     return WASM_SEC_ORDER_FUNCTION;
1894feef720bSHeejin Ahn   case wasm::WASM_SEC_TABLE:
1895feef720bSHeejin Ahn     return WASM_SEC_ORDER_TABLE;
1896feef720bSHeejin Ahn   case wasm::WASM_SEC_MEMORY:
1897feef720bSHeejin Ahn     return WASM_SEC_ORDER_MEMORY;
1898feef720bSHeejin Ahn   case wasm::WASM_SEC_GLOBAL:
1899feef720bSHeejin Ahn     return WASM_SEC_ORDER_GLOBAL;
1900feef720bSHeejin Ahn   case wasm::WASM_SEC_EXPORT:
1901feef720bSHeejin Ahn     return WASM_SEC_ORDER_EXPORT;
1902feef720bSHeejin Ahn   case wasm::WASM_SEC_START:
1903feef720bSHeejin Ahn     return WASM_SEC_ORDER_START;
1904feef720bSHeejin Ahn   case wasm::WASM_SEC_ELEM:
1905feef720bSHeejin Ahn     return WASM_SEC_ORDER_ELEM;
1906feef720bSHeejin Ahn   case wasm::WASM_SEC_CODE:
1907feef720bSHeejin Ahn     return WASM_SEC_ORDER_CODE;
1908feef720bSHeejin Ahn   case wasm::WASM_SEC_DATA:
1909feef720bSHeejin Ahn     return WASM_SEC_ORDER_DATA;
1910feef720bSHeejin Ahn   case wasm::WASM_SEC_DATACOUNT:
1911feef720bSHeejin Ahn     return WASM_SEC_ORDER_DATACOUNT;
19121d891d44SHeejin Ahn   case wasm::WASM_SEC_TAG:
19131d891d44SHeejin Ahn     return WASM_SEC_ORDER_TAG;
1914feef720bSHeejin Ahn   default:
1915fe8aabf9SSam Clegg     return WASM_SEC_ORDER_NONE;
1916feef720bSHeejin Ahn   }
1917feef720bSHeejin Ahn }
1918feef720bSHeejin Ahn 
19199757bba4SThomas Lively // Represents the edges in a directed graph where any node B reachable from node
19209757bba4SThomas Lively // A is not allowed to appear before A in the section ordering, but may appear
19219757bba4SThomas Lively // afterward.
1922f93426c5SHeejin Ahn int WasmSectionOrderChecker::DisallowedPredecessors
1923f93426c5SHeejin Ahn     [WASM_NUM_SEC_ORDERS][WASM_NUM_SEC_ORDERS] = {
1924f93426c5SHeejin Ahn         // WASM_SEC_ORDER_NONE
1925f93426c5SHeejin Ahn         {},
1926f93426c5SHeejin Ahn         // WASM_SEC_ORDER_TYPE
1927f93426c5SHeejin Ahn         {WASM_SEC_ORDER_TYPE, WASM_SEC_ORDER_IMPORT},
1928f93426c5SHeejin Ahn         // WASM_SEC_ORDER_IMPORT
1929f93426c5SHeejin Ahn         {WASM_SEC_ORDER_IMPORT, WASM_SEC_ORDER_FUNCTION},
1930f93426c5SHeejin Ahn         // WASM_SEC_ORDER_FUNCTION
1931f93426c5SHeejin Ahn         {WASM_SEC_ORDER_FUNCTION, WASM_SEC_ORDER_TABLE},
1932f93426c5SHeejin Ahn         // WASM_SEC_ORDER_TABLE
1933f93426c5SHeejin Ahn         {WASM_SEC_ORDER_TABLE, WASM_SEC_ORDER_MEMORY},
1934f93426c5SHeejin Ahn         // WASM_SEC_ORDER_MEMORY
19351d891d44SHeejin Ahn         {WASM_SEC_ORDER_MEMORY, WASM_SEC_ORDER_TAG},
19361d891d44SHeejin Ahn         // WASM_SEC_ORDER_TAG
19371d891d44SHeejin Ahn         {WASM_SEC_ORDER_TAG, WASM_SEC_ORDER_GLOBAL},
1938f93426c5SHeejin Ahn         // WASM_SEC_ORDER_GLOBAL
1939f93426c5SHeejin Ahn         {WASM_SEC_ORDER_GLOBAL, WASM_SEC_ORDER_EXPORT},
1940f93426c5SHeejin Ahn         // WASM_SEC_ORDER_EXPORT
1941f93426c5SHeejin Ahn         {WASM_SEC_ORDER_EXPORT, WASM_SEC_ORDER_START},
1942f93426c5SHeejin Ahn         // WASM_SEC_ORDER_START
1943f93426c5SHeejin Ahn         {WASM_SEC_ORDER_START, WASM_SEC_ORDER_ELEM},
1944f93426c5SHeejin Ahn         // WASM_SEC_ORDER_ELEM
1945f93426c5SHeejin Ahn         {WASM_SEC_ORDER_ELEM, WASM_SEC_ORDER_DATACOUNT},
1946f93426c5SHeejin Ahn         // WASM_SEC_ORDER_DATACOUNT
1947f93426c5SHeejin Ahn         {WASM_SEC_ORDER_DATACOUNT, WASM_SEC_ORDER_CODE},
1948f93426c5SHeejin Ahn         // WASM_SEC_ORDER_CODE
1949f93426c5SHeejin Ahn         {WASM_SEC_ORDER_CODE, WASM_SEC_ORDER_DATA},
1950f93426c5SHeejin Ahn         // WASM_SEC_ORDER_DATA
1951f93426c5SHeejin Ahn         {WASM_SEC_ORDER_DATA, WASM_SEC_ORDER_LINKING},
19529757bba4SThomas Lively 
19539757bba4SThomas Lively         // Custom Sections
1954f93426c5SHeejin Ahn         // WASM_SEC_ORDER_DYLINK
1955f93426c5SHeejin Ahn         {WASM_SEC_ORDER_DYLINK, WASM_SEC_ORDER_TYPE},
1956f93426c5SHeejin Ahn         // WASM_SEC_ORDER_LINKING
1957f93426c5SHeejin Ahn         {WASM_SEC_ORDER_LINKING, WASM_SEC_ORDER_RELOC, WASM_SEC_ORDER_NAME},
1958f93426c5SHeejin Ahn         // WASM_SEC_ORDER_RELOC (can be repeated)
1959f93426c5SHeejin Ahn         {},
1960f93426c5SHeejin Ahn         // WASM_SEC_ORDER_NAME
1961f93426c5SHeejin Ahn         {WASM_SEC_ORDER_NAME, WASM_SEC_ORDER_PRODUCERS},
1962f93426c5SHeejin Ahn         // WASM_SEC_ORDER_PRODUCERS
1963f93426c5SHeejin Ahn         {WASM_SEC_ORDER_PRODUCERS, WASM_SEC_ORDER_TARGET_FEATURES},
1964f93426c5SHeejin Ahn         // WASM_SEC_ORDER_TARGET_FEATURES
1965f93426c5SHeejin Ahn         {WASM_SEC_ORDER_TARGET_FEATURES}};
19669757bba4SThomas Lively 
isValidSectionOrder(unsigned ID,StringRef CustomSectionName)1967feef720bSHeejin Ahn bool WasmSectionOrderChecker::isValidSectionOrder(unsigned ID,
1968feef720bSHeejin Ahn                                                   StringRef CustomSectionName) {
1969feef720bSHeejin Ahn   int Order = getSectionOrder(ID, CustomSectionName);
19709757bba4SThomas Lively   if (Order == WASM_SEC_ORDER_NONE)
1971feef720bSHeejin Ahn     return true;
19729757bba4SThomas Lively 
19739757bba4SThomas Lively   // Disallowed predecessors we need to check for
19749757bba4SThomas Lively   SmallVector<int, WASM_NUM_SEC_ORDERS> WorkList;
19759757bba4SThomas Lively 
19769757bba4SThomas Lively   // Keep track of completed checks to avoid repeating work
19779757bba4SThomas Lively   bool Checked[WASM_NUM_SEC_ORDERS] = {};
19789757bba4SThomas Lively 
19799757bba4SThomas Lively   int Curr = Order;
19809757bba4SThomas Lively   while (true) {
19819757bba4SThomas Lively     // Add new disallowed predecessors to work list
19829757bba4SThomas Lively     for (size_t I = 0;; ++I) {
19839757bba4SThomas Lively       int Next = DisallowedPredecessors[Curr][I];
19849757bba4SThomas Lively       if (Next == WASM_SEC_ORDER_NONE)
19859757bba4SThomas Lively         break;
19869757bba4SThomas Lively       if (Checked[Next])
19879757bba4SThomas Lively         continue;
19889757bba4SThomas Lively       WorkList.push_back(Next);
19899757bba4SThomas Lively       Checked[Next] = true;
19909757bba4SThomas Lively     }
19919757bba4SThomas Lively 
19929757bba4SThomas Lively     if (WorkList.empty())
19939757bba4SThomas Lively       break;
19949757bba4SThomas Lively 
19959757bba4SThomas Lively     // Consider next disallowed predecessor
19969757bba4SThomas Lively     Curr = WorkList.pop_back_val();
19979757bba4SThomas Lively     if (Seen[Curr])
19989757bba4SThomas Lively       return false;
19999757bba4SThomas Lively   }
20009757bba4SThomas Lively 
20019757bba4SThomas Lively   // Have not seen any disallowed predecessors
20029757bba4SThomas Lively   Seen[Order] = true;
20039757bba4SThomas Lively   return true;
2004feef720bSHeejin Ahn }
2005