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"
1144d95122SEugene Zelenko #include "llvm/ADT/STLExtras.h"
12cbda16ebSThomas Lively #include "llvm/ADT/SmallSet.h"
1344d95122SEugene Zelenko #include "llvm/ADT/StringRef.h"
14ea7caceeSSam Clegg #include "llvm/ADT/StringSet.h"
150ea52537SSimon Pilgrim #include "llvm/ADT/StringSwitch.h"
1644d95122SEugene Zelenko #include "llvm/ADT/Triple.h"
17264b5d9eSZachary Turner #include "llvm/BinaryFormat/Wasm.h"
189f5094dfSEugene Zelenko #include "llvm/MC/SubtargetFeature.h"
1944d95122SEugene Zelenko #include "llvm/Object/Binary.h"
2044d95122SEugene Zelenko #include "llvm/Object/Error.h"
2144d95122SEugene Zelenko #include "llvm/Object/ObjectFile.h"
2244d95122SEugene Zelenko #include "llvm/Object/SymbolicFile.h"
232c6f75ddSDerek Schuff #include "llvm/Object/Wasm.h"
242c6f75ddSDerek Schuff #include "llvm/Support/Endian.h"
2544d95122SEugene Zelenko #include "llvm/Support/Error.h"
2644d95122SEugene Zelenko #include "llvm/Support/ErrorHandling.h"
272c6f75ddSDerek Schuff #include "llvm/Support/LEB128.h"
28feef720bSHeejin Ahn #include "llvm/Support/ScopedPrinter.h"
2944d95122SEugene Zelenko #include <algorithm>
309f5094dfSEugene Zelenko #include <cassert>
3144d95122SEugene Zelenko #include <cstdint>
329f5094dfSEugene Zelenko #include <cstring>
3344d95122SEugene Zelenko #include <system_error>
342c6f75ddSDerek Schuff 
35b7787fd0SSam Clegg #define DEBUG_TYPE "wasm-object"
36b7787fd0SSam Clegg 
3744d95122SEugene Zelenko using namespace llvm;
3844d95122SEugene Zelenko using namespace object;
392c6f75ddSDerek Schuff 
40659932b0SJF Bastien void WasmSymbol::print(raw_ostream &Out) const {
41659932b0SJF Bastien   Out << "Name=" << Info.Name
4273332d73SSam Clegg       << ", Kind=" << toString(wasm::WasmSymbolType(Info.Kind)) << ", Flags=0x"
4373332d73SSam Clegg       << Twine::utohexstr(Info.Flags);
44659932b0SJF Bastien   if (!isTypeData()) {
45659932b0SJF Bastien     Out << ", ElemIndex=" << Info.ElementIndex;
46659932b0SJF Bastien   } else if (isDefined()) {
47659932b0SJF Bastien     Out << ", Segment=" << Info.DataRef.Segment;
48659932b0SJF Bastien     Out << ", Offset=" << Info.DataRef.Offset;
49659932b0SJF Bastien     Out << ", Size=" << Info.DataRef.Size;
50659932b0SJF Bastien   }
51659932b0SJF Bastien }
52659932b0SJF Bastien 
53ddc84bf7SJF Bastien #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
54ddc84bf7SJF Bastien LLVM_DUMP_METHOD void WasmSymbol::dump() const { print(dbgs()); }
55ddc84bf7SJF Bastien #endif
56ddc84bf7SJF Bastien 
572c6f75ddSDerek Schuff Expected<std::unique_ptr<WasmObjectFile>>
582c6f75ddSDerek Schuff ObjectFile::createWasmObjectFile(MemoryBufferRef Buffer) {
592c6f75ddSDerek Schuff   Error Err = Error::success();
600eaee545SJonas Devlieghere   auto ObjectFile = std::make_unique<WasmObjectFile>(Buffer, Err);
612c6f75ddSDerek Schuff   if (Err)
62c55cf4afSBill Wendling     return std::move(Err);
632c6f75ddSDerek Schuff 
64c55cf4afSBill Wendling   return std::move(ObjectFile);
652c6f75ddSDerek Schuff }
662c6f75ddSDerek Schuff 
67d3d84fddSDerek Schuff #define VARINT7_MAX ((1 << 7) - 1)
68d3d84fddSDerek Schuff #define VARINT7_MIN (-(1 << 7))
69d3d84fddSDerek Schuff #define VARUINT7_MAX (1 << 7)
70d3d84fddSDerek Schuff #define VARUINT1_MAX (1)
71d3d84fddSDerek Schuff 
72b7c62394SSam Clegg static uint8_t readUint8(WasmObjectFile::ReadContext &Ctx) {
73b7c62394SSam Clegg   if (Ctx.Ptr == Ctx.End)
74b7c62394SSam Clegg     report_fatal_error("EOF while reading uint8");
75b7c62394SSam Clegg   return *Ctx.Ptr++;
76b7c62394SSam Clegg }
77d3d84fddSDerek Schuff 
78b7c62394SSam Clegg static uint32_t readUint32(WasmObjectFile::ReadContext &Ctx) {
79b7c62394SSam Clegg   if (Ctx.Ptr + 4 > Ctx.End)
80b7c62394SSam Clegg     report_fatal_error("EOF while reading uint32");
81b7c62394SSam Clegg   uint32_t Result = support::endian::read32le(Ctx.Ptr);
82b7c62394SSam Clegg   Ctx.Ptr += 4;
832c6f75ddSDerek Schuff   return Result;
842c6f75ddSDerek Schuff }
852c6f75ddSDerek Schuff 
86b7c62394SSam Clegg static int32_t readFloat32(WasmObjectFile::ReadContext &Ctx) {
87e3d6b978SJonas Devlieghere   if (Ctx.Ptr + 4 > Ctx.End)
88e3d6b978SJonas Devlieghere     report_fatal_error("EOF while reading float64");
89d3d84fddSDerek Schuff   int32_t Result = 0;
90b7c62394SSam Clegg   memcpy(&Result, Ctx.Ptr, sizeof(Result));
91b7c62394SSam Clegg   Ctx.Ptr += sizeof(Result);
92d3d84fddSDerek Schuff   return Result;
93d3d84fddSDerek Schuff }
94d3d84fddSDerek Schuff 
95b7c62394SSam Clegg static int64_t readFloat64(WasmObjectFile::ReadContext &Ctx) {
96e3d6b978SJonas Devlieghere   if (Ctx.Ptr + 8 > Ctx.End)
97e3d6b978SJonas Devlieghere     report_fatal_error("EOF while reading float64");
98d3d84fddSDerek Schuff   int64_t Result = 0;
99b7c62394SSam Clegg   memcpy(&Result, Ctx.Ptr, sizeof(Result));
100b7c62394SSam Clegg   Ctx.Ptr += sizeof(Result);
101d3d84fddSDerek Schuff   return Result;
102d3d84fddSDerek Schuff }
103d3d84fddSDerek Schuff 
104b7c62394SSam Clegg static uint64_t readULEB128(WasmObjectFile::ReadContext &Ctx) {
1052c6f75ddSDerek Schuff   unsigned Count;
106b7c62394SSam Clegg   const char *Error = nullptr;
107b7c62394SSam Clegg   uint64_t Result = decodeULEB128(Ctx.Ptr, &Count, Ctx.End, &Error);
108b7c62394SSam Clegg   if (Error)
109b7c62394SSam Clegg     report_fatal_error(Error);
110b7c62394SSam Clegg   Ctx.Ptr += Count;
1112c6f75ddSDerek Schuff   return Result;
1122c6f75ddSDerek Schuff }
1132c6f75ddSDerek Schuff 
114b7c62394SSam Clegg static StringRef readString(WasmObjectFile::ReadContext &Ctx) {
115b7c62394SSam Clegg   uint32_t StringLen = readULEB128(Ctx);
116b7c62394SSam Clegg   if (Ctx.Ptr + StringLen > Ctx.End)
117b7c62394SSam Clegg     report_fatal_error("EOF while reading string");
118b7c62394SSam Clegg   StringRef Return =
119b7c62394SSam Clegg       StringRef(reinterpret_cast<const char *>(Ctx.Ptr), StringLen);
120b7c62394SSam Clegg   Ctx.Ptr += StringLen;
1212c6f75ddSDerek Schuff   return Return;
1222c6f75ddSDerek Schuff }
1232c6f75ddSDerek Schuff 
124b7c62394SSam Clegg static int64_t readLEB128(WasmObjectFile::ReadContext &Ctx) {
125d3d84fddSDerek Schuff   unsigned Count;
126b7c62394SSam Clegg   const char *Error = nullptr;
127b7c62394SSam Clegg   uint64_t Result = decodeSLEB128(Ctx.Ptr, &Count, Ctx.End, &Error);
128b7c62394SSam Clegg   if (Error)
129b7c62394SSam Clegg     report_fatal_error(Error);
130b7c62394SSam Clegg   Ctx.Ptr += Count;
131d3d84fddSDerek Schuff   return Result;
132d3d84fddSDerek Schuff }
133d3d84fddSDerek Schuff 
134b7c62394SSam Clegg static uint8_t readVaruint1(WasmObjectFile::ReadContext &Ctx) {
13518c56a07SHeejin Ahn   int64_t Result = readLEB128(Ctx);
13618c56a07SHeejin Ahn   if (Result > VARUINT1_MAX || Result < 0)
1374bbc6b55SSam Clegg     report_fatal_error("LEB is outside Varuint1 range");
13818c56a07SHeejin Ahn   return Result;
139d3d84fddSDerek Schuff }
140d3d84fddSDerek Schuff 
141b7c62394SSam Clegg static int32_t readVarint32(WasmObjectFile::ReadContext &Ctx) {
14218c56a07SHeejin Ahn   int64_t Result = readLEB128(Ctx);
14318c56a07SHeejin Ahn   if (Result > INT32_MAX || Result < INT32_MIN)
1444bbc6b55SSam Clegg     report_fatal_error("LEB is outside Varint32 range");
14518c56a07SHeejin Ahn   return Result;
146d3d84fddSDerek Schuff }
147d3d84fddSDerek Schuff 
148b7c62394SSam Clegg static uint32_t readVaruint32(WasmObjectFile::ReadContext &Ctx) {
14918c56a07SHeejin Ahn   uint64_t Result = readULEB128(Ctx);
15018c56a07SHeejin Ahn   if (Result > UINT32_MAX)
1514bbc6b55SSam Clegg     report_fatal_error("LEB is outside Varuint32 range");
15218c56a07SHeejin Ahn   return Result;
153d3d84fddSDerek Schuff }
154d3d84fddSDerek Schuff 
155b7c62394SSam Clegg static int64_t readVarint64(WasmObjectFile::ReadContext &Ctx) {
156b7c62394SSam Clegg   return readLEB128(Ctx);
157d3d84fddSDerek Schuff }
158d3d84fddSDerek Schuff 
1593b29376eSWouter van Oortmerssen static uint64_t readVaruint64(WasmObjectFile::ReadContext &Ctx) {
1603b29376eSWouter van Oortmerssen   return readULEB128(Ctx);
1613b29376eSWouter van Oortmerssen }
1623b29376eSWouter van Oortmerssen 
163b7c62394SSam Clegg static uint8_t readOpcode(WasmObjectFile::ReadContext &Ctx) {
164b7c62394SSam Clegg   return readUint8(Ctx);
165d3d84fddSDerek Schuff }
166d3d84fddSDerek Schuff 
167b7c62394SSam Clegg static Error readInitExpr(wasm::WasmInitExpr &Expr,
168b7c62394SSam Clegg                           WasmObjectFile::ReadContext &Ctx) {
169b7c62394SSam Clegg   Expr.Opcode = readOpcode(Ctx);
170d3d84fddSDerek Schuff 
171d3d84fddSDerek Schuff   switch (Expr.Opcode) {
172d3d84fddSDerek Schuff   case wasm::WASM_OPCODE_I32_CONST:
173b7c62394SSam Clegg     Expr.Value.Int32 = readVarint32(Ctx);
174d3d84fddSDerek Schuff     break;
175d3d84fddSDerek Schuff   case wasm::WASM_OPCODE_I64_CONST:
176b7c62394SSam Clegg     Expr.Value.Int64 = readVarint64(Ctx);
177d3d84fddSDerek Schuff     break;
178d3d84fddSDerek Schuff   case wasm::WASM_OPCODE_F32_CONST:
179b7c62394SSam Clegg     Expr.Value.Float32 = readFloat32(Ctx);
180d3d84fddSDerek Schuff     break;
181d3d84fddSDerek Schuff   case wasm::WASM_OPCODE_F64_CONST:
182b7c62394SSam Clegg     Expr.Value.Float64 = readFloat64(Ctx);
183d3d84fddSDerek Schuff     break;
1846a87ddacSThomas Lively   case wasm::WASM_OPCODE_GLOBAL_GET:
185b7c62394SSam Clegg     Expr.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:
19684c6f325SSam Clegg     return make_error<GenericBinaryError>("invalid opcode in init_expr",
197d3d84fddSDerek Schuff                                           object_error::parse_failed);
198d3d84fddSDerek Schuff   }
199d3d84fddSDerek Schuff 
200b7c62394SSam Clegg   uint8_t EndOpcode = readOpcode(Ctx);
201d3d84fddSDerek Schuff   if (EndOpcode != wasm::WASM_OPCODE_END) {
20284c6f325SSam Clegg     return make_error<GenericBinaryError>("invalid init_expr",
203d3d84fddSDerek Schuff                                           object_error::parse_failed);
204d3d84fddSDerek Schuff   }
205d3d84fddSDerek Schuff   return Error::success();
206d3d84fddSDerek Schuff }
207d3d84fddSDerek Schuff 
208b7c62394SSam Clegg static wasm::WasmLimits readLimits(WasmObjectFile::ReadContext &Ctx) {
209d3d84fddSDerek Schuff   wasm::WasmLimits Result;
21068818062SDerek Schuff   Result.Flags = readVaruint32(Ctx);
2119ac5620cSAndy Wingo   Result.Minimum = readVaruint64(Ctx);
212d3d84fddSDerek Schuff   if (Result.Flags & wasm::WASM_LIMITS_FLAG_HAS_MAX)
2134d135b04SWouter van Oortmerssen     Result.Maximum = readVaruint64(Ctx);
214d3d84fddSDerek Schuff   return Result;
215d3d84fddSDerek Schuff }
216d3d84fddSDerek Schuff 
2171933c9d4SAndy Wingo static wasm::WasmTableType readTableType(WasmObjectFile::ReadContext &Ctx) {
2181933c9d4SAndy Wingo   wasm::WasmTableType TableType;
2191933c9d4SAndy Wingo   TableType.ElemType = readUint8(Ctx);
2201933c9d4SAndy Wingo   TableType.Limits = readLimits(Ctx);
2211933c9d4SAndy Wingo   return TableType;
2222ffff5afSSam Clegg }
2232ffff5afSSam Clegg 
224feef720bSHeejin Ahn static Error readSection(WasmSection &Section, WasmObjectFile::ReadContext &Ctx,
225feef720bSHeejin Ahn                          WasmSectionOrderChecker &Checker) {
226b7c62394SSam Clegg   Section.Offset = Ctx.Ptr - Ctx.Start;
227b7c62394SSam Clegg   Section.Type = readUint8(Ctx);
228f4f37509SSam Clegg   LLVM_DEBUG(dbgs() << "readSection type=" << Section.Type << "\n");
229b7c62394SSam Clegg   uint32_t Size = readVaruint32(Ctx);
2302c6f75ddSDerek Schuff   if (Size == 0)
23184c6f325SSam Clegg     return make_error<StringError>("zero length section",
2322c6f75ddSDerek Schuff                                    object_error::parse_failed);
233b7c62394SSam Clegg   if (Ctx.Ptr + Size > Ctx.End)
23484c6f325SSam Clegg     return make_error<StringError>("section too large",
23535b50a83SVedant Kumar                                    object_error::parse_failed);
2369745afa6SSam Clegg   if (Section.Type == wasm::WASM_SEC_CUSTOM) {
2378511777dSJonas Devlieghere     WasmObjectFile::ReadContext SectionCtx;
2388511777dSJonas Devlieghere     SectionCtx.Start = Ctx.Ptr;
2398511777dSJonas Devlieghere     SectionCtx.Ptr = Ctx.Ptr;
2408511777dSJonas Devlieghere     SectionCtx.End = Ctx.Ptr + Size;
2418511777dSJonas Devlieghere 
2428511777dSJonas Devlieghere     Section.Name = readString(SectionCtx);
2438511777dSJonas Devlieghere 
2448511777dSJonas Devlieghere     uint32_t SectionNameSize = SectionCtx.Ptr - SectionCtx.Start;
2458511777dSJonas Devlieghere     Ctx.Ptr += SectionNameSize;
2468511777dSJonas Devlieghere     Size -= SectionNameSize;
2479745afa6SSam Clegg   }
248feef720bSHeejin Ahn 
249feef720bSHeejin Ahn   if (!Checker.isValidSectionOrder(Section.Type, Section.Name)) {
25084c6f325SSam Clegg     return make_error<StringError>("out of order section type: " +
251feef720bSHeejin Ahn                                        llvm::to_string(Section.Type),
252feef720bSHeejin Ahn                                    object_error::parse_failed);
253feef720bSHeejin Ahn   }
254feef720bSHeejin Ahn 
255b7c62394SSam Clegg   Section.Content = ArrayRef<uint8_t>(Ctx.Ptr, Size);
256b7c62394SSam Clegg   Ctx.Ptr += Size;
2572c6f75ddSDerek Schuff   return Error::success();
2582c6f75ddSDerek Schuff }
2592c6f75ddSDerek Schuff 
2602c6f75ddSDerek Schuff WasmObjectFile::WasmObjectFile(MemoryBufferRef Buffer, Error &Err)
2619f5094dfSEugene Zelenko     : ObjectFile(Binary::ID_Wasm, Buffer) {
2622c6f75ddSDerek Schuff   ErrorAsOutParameter ErrAsOutParam(&Err);
2632c6f75ddSDerek Schuff   Header.Magic = getData().substr(0, 4);
2642c6f75ddSDerek Schuff   if (Header.Magic != StringRef("\0asm", 4)) {
26584c6f325SSam Clegg     Err = make_error<StringError>("invalid magic number",
26684c6f325SSam Clegg                                   object_error::parse_failed);
2672c6f75ddSDerek Schuff     return;
2682c6f75ddSDerek Schuff   }
269a845167dSJonas Devlieghere 
270b7c62394SSam Clegg   ReadContext Ctx;
2716a0746a9SFangrui Song   Ctx.Start = getData().bytes_begin();
272b7c62394SSam Clegg   Ctx.Ptr = Ctx.Start + 4;
273b7c62394SSam Clegg   Ctx.End = Ctx.Start + getData().size();
274a845167dSJonas Devlieghere 
275b7c62394SSam Clegg   if (Ctx.Ptr + 4 > Ctx.End) {
27684c6f325SSam Clegg     Err = make_error<StringError>("missing version number",
277a845167dSJonas Devlieghere                                   object_error::parse_failed);
278a845167dSJonas Devlieghere     return;
279a845167dSJonas Devlieghere   }
280a845167dSJonas Devlieghere 
281b7c62394SSam Clegg   Header.Version = readUint32(Ctx);
2822c6f75ddSDerek Schuff   if (Header.Version != wasm::WasmVersion) {
28384c6f325SSam Clegg     Err = make_error<StringError>("invalid version number: " +
28484c6f325SSam Clegg                                       Twine(Header.Version),
2852c6f75ddSDerek Schuff                                   object_error::parse_failed);
2862c6f75ddSDerek Schuff     return;
2872c6f75ddSDerek Schuff   }
2882c6f75ddSDerek Schuff 
289feef720bSHeejin Ahn   WasmSectionOrderChecker Checker;
290b7c62394SSam Clegg   while (Ctx.Ptr < Ctx.End) {
291e4b2f305SSam Clegg     WasmSection Sec;
292feef720bSHeejin Ahn     if ((Err = readSection(Sec, Ctx, Checker)))
2932c6f75ddSDerek Schuff       return;
294d3d84fddSDerek Schuff     if ((Err = parseSection(Sec)))
2952c6f75ddSDerek Schuff       return;
296d3d84fddSDerek Schuff 
2972c6f75ddSDerek Schuff     Sections.push_back(Sec);
2982c6f75ddSDerek Schuff   }
2992c6f75ddSDerek Schuff }
3002c6f75ddSDerek Schuff 
301d3d84fddSDerek Schuff Error WasmObjectFile::parseSection(WasmSection &Sec) {
302b7c62394SSam Clegg   ReadContext Ctx;
303b7c62394SSam Clegg   Ctx.Start = Sec.Content.data();
304b7c62394SSam Clegg   Ctx.End = Ctx.Start + Sec.Content.size();
305b7c62394SSam Clegg   Ctx.Ptr = Ctx.Start;
306d3d84fddSDerek Schuff   switch (Sec.Type) {
307d3d84fddSDerek Schuff   case wasm::WASM_SEC_CUSTOM:
308b7c62394SSam Clegg     return parseCustomSection(Sec, Ctx);
309d3d84fddSDerek Schuff   case wasm::WASM_SEC_TYPE:
310b7c62394SSam Clegg     return parseTypeSection(Ctx);
311d3d84fddSDerek Schuff   case wasm::WASM_SEC_IMPORT:
312b7c62394SSam Clegg     return parseImportSection(Ctx);
313d3d84fddSDerek Schuff   case wasm::WASM_SEC_FUNCTION:
314b7c62394SSam Clegg     return parseFunctionSection(Ctx);
315d3d84fddSDerek Schuff   case wasm::WASM_SEC_TABLE:
316b7c62394SSam Clegg     return parseTableSection(Ctx);
317d3d84fddSDerek Schuff   case wasm::WASM_SEC_MEMORY:
318b7c62394SSam Clegg     return parseMemorySection(Ctx);
3191d891d44SHeejin Ahn   case wasm::WASM_SEC_TAG:
3201d891d44SHeejin Ahn     return parseTagSection(Ctx);
321f93426c5SHeejin Ahn   case wasm::WASM_SEC_GLOBAL:
322f93426c5SHeejin Ahn     return parseGlobalSection(Ctx);
323d3d84fddSDerek Schuff   case wasm::WASM_SEC_EXPORT:
324b7c62394SSam Clegg     return parseExportSection(Ctx);
325d3d84fddSDerek Schuff   case wasm::WASM_SEC_START:
326b7c62394SSam Clegg     return parseStartSection(Ctx);
327d3d84fddSDerek Schuff   case wasm::WASM_SEC_ELEM:
328b7c62394SSam Clegg     return parseElemSection(Ctx);
329d3d84fddSDerek Schuff   case wasm::WASM_SEC_CODE:
330b7c62394SSam Clegg     return parseCodeSection(Ctx);
331d3d84fddSDerek Schuff   case wasm::WASM_SEC_DATA:
332b7c62394SSam Clegg     return parseDataSection(Ctx);
333fef8de66SThomas Lively   case wasm::WASM_SEC_DATACOUNT:
334fef8de66SThomas Lively     return parseDataCountSection(Ctx);
335d3d84fddSDerek Schuff   default:
336fe8aabf9SSam Clegg     return make_error<GenericBinaryError>(
33784c6f325SSam Clegg         "invalid section type: " + Twine(Sec.Type), object_error::parse_failed);
338d3d84fddSDerek Schuff   }
339d3d84fddSDerek Schuff }
340d3d84fddSDerek Schuff 
341e4afbc68SSam Clegg Error WasmObjectFile::parseDylinkSection(ReadContext &Ctx) {
342b78c85a4SSam Clegg   // Legacy "dylink" section support.
343b78c85a4SSam Clegg   // See parseDylink0Section for the current "dylink.0" section parsing.
344a688a42cSSam Clegg   HasDylinkSection = true;
345e4afbc68SSam Clegg   DylinkInfo.MemorySize = readVaruint32(Ctx);
346e4afbc68SSam Clegg   DylinkInfo.MemoryAlignment = readVaruint32(Ctx);
347e4afbc68SSam Clegg   DylinkInfo.TableSize = readVaruint32(Ctx);
348e4afbc68SSam Clegg   DylinkInfo.TableAlignment = readVaruint32(Ctx);
34903801256SSam Clegg   uint32_t Count = readVaruint32(Ctx);
35003801256SSam Clegg   while (Count--) {
35103801256SSam Clegg     DylinkInfo.Needed.push_back(readString(Ctx));
35203801256SSam Clegg   }
353b78c85a4SSam Clegg 
354e4afbc68SSam Clegg   if (Ctx.Ptr != Ctx.End)
355e4afbc68SSam Clegg     return make_error<GenericBinaryError>("dylink section ended prematurely",
356e4afbc68SSam Clegg                                           object_error::parse_failed);
357e4afbc68SSam Clegg   return Error::success();
358e4afbc68SSam Clegg }
359e4afbc68SSam Clegg 
360b78c85a4SSam Clegg Error WasmObjectFile::parseDylink0Section(ReadContext &Ctx) {
361b78c85a4SSam Clegg   // See
362*c71fbdd8SQuinn Pham   // https://github.com/WebAssembly/tool-conventions/blob/main/DynamicLinking.md
363b78c85a4SSam Clegg   HasDylinkSection = true;
364b78c85a4SSam Clegg 
365b78c85a4SSam Clegg   const uint8_t *OrigEnd = Ctx.End;
366b78c85a4SSam Clegg   while (Ctx.Ptr < OrigEnd) {
367b78c85a4SSam Clegg     Ctx.End = OrigEnd;
368b78c85a4SSam Clegg     uint8_t Type = readUint8(Ctx);
369b78c85a4SSam Clegg     uint32_t Size = readVaruint32(Ctx);
370b78c85a4SSam Clegg     LLVM_DEBUG(dbgs() << "readSubsection type=" << int(Type) << " size=" << Size
371b78c85a4SSam Clegg                       << "\n");
372b78c85a4SSam Clegg     Ctx.End = Ctx.Ptr + Size;
373b78c85a4SSam Clegg     uint32_t Count;
374b78c85a4SSam Clegg     switch (Type) {
375b78c85a4SSam Clegg     case wasm::WASM_DYLINK_MEM_INFO:
376b78c85a4SSam Clegg       DylinkInfo.MemorySize = readVaruint32(Ctx);
377b78c85a4SSam Clegg       DylinkInfo.MemoryAlignment = readVaruint32(Ctx);
378b78c85a4SSam Clegg       DylinkInfo.TableSize = readVaruint32(Ctx);
379b78c85a4SSam Clegg       DylinkInfo.TableAlignment = readVaruint32(Ctx);
380b78c85a4SSam Clegg       break;
381b78c85a4SSam Clegg     case wasm::WASM_DYLINK_NEEDED:
382b78c85a4SSam Clegg       Count = readVaruint32(Ctx);
383b78c85a4SSam Clegg       while (Count--) {
384b78c85a4SSam Clegg         DylinkInfo.Needed.push_back(readString(Ctx));
385b78c85a4SSam Clegg       }
386b78c85a4SSam Clegg       break;
387ef8c9135SSam Clegg     case wasm::WASM_DYLINK_EXPORT_INFO: {
388ef8c9135SSam Clegg       uint32_t Count = readVaruint32(Ctx);
389ef8c9135SSam Clegg       while (Count--) {
390ef8c9135SSam Clegg         DylinkInfo.ExportInfo.push_back({readString(Ctx), readVaruint32(Ctx)});
391ef8c9135SSam Clegg       }
392ef8c9135SSam Clegg       break;
393ef8c9135SSam Clegg     }
394659a0839SSam Clegg     case wasm::WASM_DYLINK_IMPORT_INFO: {
395659a0839SSam Clegg       uint32_t Count = readVaruint32(Ctx);
396659a0839SSam Clegg       while (Count--) {
397659a0839SSam Clegg         DylinkInfo.ImportInfo.push_back(
398659a0839SSam Clegg             {readString(Ctx), readString(Ctx), readVaruint32(Ctx)});
399659a0839SSam Clegg       }
400659a0839SSam Clegg       break;
401659a0839SSam Clegg     }
402b78c85a4SSam Clegg     default:
403210cbcf4SSam Clegg       LLVM_DEBUG(dbgs() << "unknown dylink.0 sub-section: " << Type << "\n");
404b78c85a4SSam Clegg       Ctx.Ptr += Size;
405b78c85a4SSam Clegg       break;
406b78c85a4SSam Clegg     }
407b78c85a4SSam Clegg     if (Ctx.Ptr != Ctx.End) {
408b78c85a4SSam Clegg       return make_error<GenericBinaryError>(
409b78c85a4SSam Clegg           "dylink.0 sub-section ended prematurely", object_error::parse_failed);
410b78c85a4SSam Clegg     }
411b78c85a4SSam Clegg   }
412b78c85a4SSam Clegg 
413b78c85a4SSam Clegg   if (Ctx.Ptr != Ctx.End)
414b78c85a4SSam Clegg     return make_error<GenericBinaryError>("dylink.0 section ended prematurely",
415b78c85a4SSam Clegg                                           object_error::parse_failed);
416b78c85a4SSam Clegg   return Error::success();
417b78c85a4SSam Clegg }
418b78c85a4SSam Clegg 
419b7c62394SSam Clegg Error WasmObjectFile::parseNameSection(ReadContext &Ctx) {
4201827005cSSam Clegg   llvm::DenseSet<uint64_t> SeenFunctions;
4211827005cSSam Clegg   llvm::DenseSet<uint64_t> SeenGlobals;
4229a72d3e3SSam Clegg   llvm::DenseSet<uint64_t> SeenSegments;
423c0039de2SSam Clegg   if (Functions.size() && !SeenCodeSection) {
42484c6f325SSam Clegg     return make_error<GenericBinaryError>("names must come after code section",
4259f3fe42eSSam Clegg                                           object_error::parse_failed);
4269f3fe42eSSam Clegg   }
4274710ed7aSSam Clegg 
428b7c62394SSam Clegg   while (Ctx.Ptr < Ctx.End) {
429b7c62394SSam Clegg     uint8_t Type = readUint8(Ctx);
430b7c62394SSam Clegg     uint32_t Size = readVaruint32(Ctx);
431b7c62394SSam Clegg     const uint8_t *SubSectionEnd = Ctx.Ptr + Size;
432d3d84fddSDerek Schuff     switch (Type) {
4331827005cSSam Clegg     case wasm::WASM_NAMES_FUNCTION:
4349a72d3e3SSam Clegg     case wasm::WASM_NAMES_GLOBAL:
4359a72d3e3SSam Clegg     case wasm::WASM_NAMES_DATA_SEGMENT: {
436b7c62394SSam Clegg       uint32_t Count = readVaruint32(Ctx);
437d3d84fddSDerek Schuff       while (Count--) {
438b7c62394SSam Clegg         uint32_t Index = readVaruint32(Ctx);
439b7c62394SSam Clegg         StringRef Name = readString(Ctx);
4409a72d3e3SSam Clegg         wasm::NameType nameType = wasm::NameType::FUNCTION;
4411827005cSSam Clegg         if (Type == wasm::WASM_NAMES_FUNCTION) {
4421827005cSSam Clegg           if (!SeenFunctions.insert(Index).second)
4431827005cSSam Clegg             return make_error<GenericBinaryError>(
44484c6f325SSam Clegg                 "function named more than once", object_error::parse_failed);
4459f3fe42eSSam Clegg           if (!isValidFunctionIndex(Index) || Name.empty())
44684c6f325SSam Clegg             return make_error<GenericBinaryError>("invalid name entry",
4479f3fe42eSSam Clegg                                                   object_error::parse_failed);
4481827005cSSam Clegg 
449ef90ff36SNicholas Wilson           if (isDefinedFunctionIndex(Index))
450ef90ff36SNicholas Wilson             getDefinedFunction(Index).DebugName = Name;
4519a72d3e3SSam Clegg         } else if (Type == wasm::WASM_NAMES_GLOBAL) {
4529a72d3e3SSam Clegg           nameType = wasm::NameType::GLOBAL;
4531827005cSSam Clegg           if (!SeenGlobals.insert(Index).second)
45484c6f325SSam Clegg             return make_error<GenericBinaryError>("global named more than once",
4551827005cSSam Clegg                                                   object_error::parse_failed);
4561827005cSSam Clegg           if (!isValidGlobalIndex(Index) || Name.empty())
45784c6f325SSam Clegg             return make_error<GenericBinaryError>("invalid name entry",
4581827005cSSam Clegg                                                   object_error::parse_failed);
4599a72d3e3SSam Clegg         } else {
4609a72d3e3SSam Clegg           nameType = wasm::NameType::DATA_SEGMENT;
4619a72d3e3SSam Clegg           if (!SeenSegments.insert(Index).second)
4629a72d3e3SSam Clegg             return make_error<GenericBinaryError>(
46384c6f325SSam Clegg                 "segment named more than once", object_error::parse_failed);
4649a72d3e3SSam Clegg           if (Index > DataSegments.size())
46584c6f325SSam Clegg             return make_error<GenericBinaryError>("invalid named data segment",
4669a72d3e3SSam Clegg                                                   object_error::parse_failed);
4671827005cSSam Clegg         }
4689a72d3e3SSam Clegg         DebugNames.push_back(wasm::WasmDebugName{nameType, Index, Name});
469d3d84fddSDerek Schuff       }
470d3d84fddSDerek Schuff       break;
471d3d84fddSDerek Schuff     }
472d3d84fddSDerek Schuff     // Ignore local names for now
473d3d84fddSDerek Schuff     case wasm::WASM_NAMES_LOCAL:
474d3d84fddSDerek Schuff     default:
475b7c62394SSam Clegg       Ctx.Ptr += Size;
476d3d84fddSDerek Schuff       break;
477d3d84fddSDerek Schuff     }
478b7c62394SSam Clegg     if (Ctx.Ptr != SubSectionEnd)
479f208f631SHeejin Ahn       return make_error<GenericBinaryError>(
48084c6f325SSam Clegg           "name sub-section ended prematurely", object_error::parse_failed);
481d3d84fddSDerek Schuff   }
482d3d84fddSDerek Schuff 
483b7c62394SSam Clegg   if (Ctx.Ptr != Ctx.End)
48484c6f325SSam Clegg     return make_error<GenericBinaryError>("name section ended prematurely",
485d3d84fddSDerek Schuff                                           object_error::parse_failed);
486d3d84fddSDerek Schuff   return Error::success();
487d3d84fddSDerek Schuff }
488d3d84fddSDerek Schuff 
489b7c62394SSam Clegg Error WasmObjectFile::parseLinkingSection(ReadContext &Ctx) {
4904df5d764SSam Clegg   HasLinkingSection = true;
491c0039de2SSam Clegg   if (Functions.size() && !SeenCodeSection) {
4929f3fe42eSSam Clegg     return make_error<GenericBinaryError>(
49384c6f325SSam Clegg         "linking data must come after code section",
494f208f631SHeejin Ahn         object_error::parse_failed);
4959f3fe42eSSam Clegg   }
496e7a60708SSam Clegg 
497b7c62394SSam Clegg   LinkingData.Version = readVaruint32(Ctx);
4986bb5a41fSSam Clegg   if (LinkingData.Version != wasm::WasmMetadataVersion) {
4996bb5a41fSSam Clegg     return make_error<GenericBinaryError>(
50084c6f325SSam Clegg         "unexpected metadata version: " + Twine(LinkingData.Version) +
5016bb5a41fSSam Clegg             " (Expected: " + Twine(wasm::WasmMetadataVersion) + ")",
5026bb5a41fSSam Clegg         object_error::parse_failed);
5036bb5a41fSSam Clegg   }
5046bb5a41fSSam Clegg 
505b7c62394SSam Clegg   const uint8_t *OrigEnd = Ctx.End;
506b7c62394SSam Clegg   while (Ctx.Ptr < OrigEnd) {
507b7c62394SSam Clegg     Ctx.End = OrigEnd;
508b7c62394SSam Clegg     uint8_t Type = readUint8(Ctx);
509b7c62394SSam Clegg     uint32_t Size = readVaruint32(Ctx);
510f4f37509SSam Clegg     LLVM_DEBUG(dbgs() << "readSubsection type=" << int(Type) << " size=" << Size
511f4f37509SSam Clegg                       << "\n");
512b7c62394SSam Clegg     Ctx.End = Ctx.Ptr + Size;
513b7787fd0SSam Clegg     switch (Type) {
5146c899ba6SSam Clegg     case wasm::WASM_SYMBOL_TABLE:
515b7c62394SSam Clegg       if (Error Err = parseLinkingSectionSymtab(Ctx))
5166c899ba6SSam Clegg         return Err;
517b7787fd0SSam Clegg       break;
51863ebb813SSam Clegg     case wasm::WASM_SEGMENT_INFO: {
519b7c62394SSam Clegg       uint32_t Count = readVaruint32(Ctx);
520d95ed959SSam Clegg       if (Count > DataSegments.size())
52184c6f325SSam Clegg         return make_error<GenericBinaryError>("too many segment names",
522d95ed959SSam Clegg                                               object_error::parse_failed);
52318c56a07SHeejin Ahn       for (uint32_t I = 0; I < Count; I++) {
52418c56a07SHeejin Ahn         DataSegments[I].Data.Name = readString(Ctx);
52518c56a07SHeejin Ahn         DataSegments[I].Data.Alignment = readVaruint32(Ctx);
5263b8d2be5SSam Clegg         DataSegments[I].Data.LinkingFlags = readVaruint32(Ctx);
52763ebb813SSam Clegg       }
528d95ed959SSam Clegg       break;
529d95ed959SSam Clegg     }
5304273998cSSam Clegg     case wasm::WASM_INIT_FUNCS: {
531b7c62394SSam Clegg       uint32_t Count = readVaruint32(Ctx);
5324273998cSSam Clegg       LinkingData.InitFunctions.reserve(Count);
53318c56a07SHeejin Ahn       for (uint32_t I = 0; I < Count; I++) {
5344273998cSSam Clegg         wasm::WasmInitFunc Init;
535b7c62394SSam Clegg         Init.Priority = readVaruint32(Ctx);
536b7c62394SSam Clegg         Init.Symbol = readVaruint32(Ctx);
537b3748f71SNicholas Wilson         if (!isValidFunctionSymbol(Init.Symbol))
53884c6f325SSam Clegg           return make_error<GenericBinaryError>("invalid function symbol: " +
5396c899ba6SSam Clegg                                                     Twine(Init.Symbol),
5404273998cSSam Clegg                                                 object_error::parse_failed);
5414273998cSSam Clegg         LinkingData.InitFunctions.emplace_back(Init);
5424273998cSSam Clegg       }
5434273998cSSam Clegg       break;
5444273998cSSam Clegg     }
545ea7caceeSSam Clegg     case wasm::WASM_COMDAT_INFO:
546b7c62394SSam Clegg       if (Error Err = parseLinkingSectionComdat(Ctx))
547ea7caceeSSam Clegg         return Err;
548ea7caceeSSam Clegg       break;
549b7787fd0SSam Clegg     default:
550b7c62394SSam Clegg       Ctx.Ptr += Size;
551b7787fd0SSam Clegg       break;
552b7787fd0SSam Clegg     }
553b7c62394SSam Clegg     if (Ctx.Ptr != Ctx.End)
554b7787fd0SSam Clegg       return make_error<GenericBinaryError>(
55584c6f325SSam Clegg           "linking sub-section ended prematurely", object_error::parse_failed);
556b7787fd0SSam Clegg   }
557b7c62394SSam Clegg   if (Ctx.Ptr != OrigEnd)
55884c6f325SSam Clegg     return make_error<GenericBinaryError>("linking section ended prematurely",
559b7787fd0SSam Clegg                                           object_error::parse_failed);
560b7787fd0SSam Clegg   return Error::success();
561b7787fd0SSam Clegg }
562b7787fd0SSam Clegg 
563b7c62394SSam Clegg Error WasmObjectFile::parseLinkingSectionSymtab(ReadContext &Ctx) {
564b7c62394SSam Clegg   uint32_t Count = readVaruint32(Ctx);
5656c899ba6SSam Clegg   LinkingData.SymbolTable.reserve(Count);
5666c899ba6SSam Clegg   Symbols.reserve(Count);
5676c899ba6SSam Clegg   StringSet<> SymbolNames;
5686c899ba6SSam Clegg 
5696c899ba6SSam Clegg   std::vector<wasm::WasmImport *> ImportedGlobals;
5706c899ba6SSam Clegg   std::vector<wasm::WasmImport *> ImportedFunctions;
5711d891d44SHeejin Ahn   std::vector<wasm::WasmImport *> ImportedTags;
572feac819eSAndy Wingo   std::vector<wasm::WasmImport *> ImportedTables;
5736c899ba6SSam Clegg   ImportedGlobals.reserve(Imports.size());
5746c899ba6SSam Clegg   ImportedFunctions.reserve(Imports.size());
5751d891d44SHeejin Ahn   ImportedTags.reserve(Imports.size());
576feac819eSAndy Wingo   ImportedTables.reserve(Imports.size());
5776c899ba6SSam Clegg   for (auto &I : Imports) {
5786c899ba6SSam Clegg     if (I.Kind == wasm::WASM_EXTERNAL_FUNCTION)
5796c899ba6SSam Clegg       ImportedFunctions.emplace_back(&I);
5806c899ba6SSam Clegg     else if (I.Kind == wasm::WASM_EXTERNAL_GLOBAL)
5816c899ba6SSam Clegg       ImportedGlobals.emplace_back(&I);
5821d891d44SHeejin Ahn     else if (I.Kind == wasm::WASM_EXTERNAL_TAG)
5831d891d44SHeejin Ahn       ImportedTags.emplace_back(&I);
584feac819eSAndy Wingo     else if (I.Kind == wasm::WASM_EXTERNAL_TABLE)
585feac819eSAndy Wingo       ImportedTables.emplace_back(&I);
5866c899ba6SSam Clegg   }
5876c899ba6SSam Clegg 
5886c899ba6SSam Clegg   while (Count--) {
5896c899ba6SSam Clegg     wasm::WasmSymbolInfo Info;
590a2125b8dSHeejin Ahn     const wasm::WasmSignature *Signature = nullptr;
5916c899ba6SSam Clegg     const wasm::WasmGlobalType *GlobalType = nullptr;
59253e3b81fSAndy Wingo     const wasm::WasmTableType *TableType = nullptr;
5936c899ba6SSam Clegg 
594b7c62394SSam Clegg     Info.Kind = readUint8(Ctx);
595b7c62394SSam Clegg     Info.Flags = readVaruint32(Ctx);
5966c899ba6SSam Clegg     bool IsDefined = (Info.Flags & wasm::WASM_SYMBOL_UNDEFINED) == 0;
5976c899ba6SSam Clegg 
5986c899ba6SSam Clegg     switch (Info.Kind) {
5996c899ba6SSam Clegg     case wasm::WASM_SYMBOL_TYPE_FUNCTION:
600b7c62394SSam Clegg       Info.ElementIndex = readVaruint32(Ctx);
6016c899ba6SSam Clegg       if (!isValidFunctionIndex(Info.ElementIndex) ||
6026c899ba6SSam Clegg           IsDefined != isDefinedFunctionIndex(Info.ElementIndex))
6036c899ba6SSam Clegg         return make_error<GenericBinaryError>("invalid function symbol index",
6046c899ba6SSam Clegg                                               object_error::parse_failed);
6056c899ba6SSam Clegg       if (IsDefined) {
606b7c62394SSam Clegg         Info.Name = readString(Ctx);
6076c899ba6SSam Clegg         unsigned FuncIndex = Info.ElementIndex - NumImportedFunctions;
608959e7371SNicholas Wilson         wasm::WasmFunction &Function = Functions[FuncIndex];
609c0039de2SSam Clegg         Signature = &Signatures[Function.SigIndex];
610ef90ff36SNicholas Wilson         if (Function.SymbolName.empty())
611ef90ff36SNicholas Wilson           Function.SymbolName = Info.Name;
6126c899ba6SSam Clegg       } else {
6136c899ba6SSam Clegg         wasm::WasmImport &Import = *ImportedFunctions[Info.ElementIndex];
614bd481277SSam Clegg         if ((Info.Flags & wasm::WASM_SYMBOL_EXPLICIT_NAME) != 0) {
61529874ceaSDan Gohman           Info.Name = readString(Ctx);
61629874ceaSDan Gohman           Info.ImportName = Import.Field;
617bd481277SSam Clegg         } else {
618bd481277SSam Clegg           Info.Name = Import.Field;
619bd481277SSam Clegg         }
620bd481277SSam Clegg         Signature = &Signatures[Import.SigIndex];
621bd481277SSam Clegg         if (!Import.Module.empty()) {
62229874ceaSDan Gohman           Info.ImportModule = Import.Module;
6236c899ba6SSam Clegg         }
624bd481277SSam Clegg       }
6256c899ba6SSam Clegg       break;
6266c899ba6SSam Clegg 
6276c899ba6SSam Clegg     case wasm::WASM_SYMBOL_TYPE_GLOBAL:
628b7c62394SSam Clegg       Info.ElementIndex = readVaruint32(Ctx);
6296c899ba6SSam Clegg       if (!isValidGlobalIndex(Info.ElementIndex) ||
6306c899ba6SSam Clegg           IsDefined != isDefinedGlobalIndex(Info.ElementIndex))
6316c899ba6SSam Clegg         return make_error<GenericBinaryError>("invalid global symbol index",
6326c899ba6SSam Clegg                                               object_error::parse_failed);
633f208f631SHeejin Ahn       if (!IsDefined && (Info.Flags & wasm::WASM_SYMBOL_BINDING_MASK) ==
63415f349f7SNicholas Wilson                             wasm::WASM_SYMBOL_BINDING_WEAK)
63515f349f7SNicholas Wilson         return make_error<GenericBinaryError>("undefined weak global symbol",
63615f349f7SNicholas Wilson                                               object_error::parse_failed);
6376c899ba6SSam Clegg       if (IsDefined) {
638b7c62394SSam Clegg         Info.Name = readString(Ctx);
6396c899ba6SSam Clegg         unsigned GlobalIndex = Info.ElementIndex - NumImportedGlobals;
640959e7371SNicholas Wilson         wasm::WasmGlobal &Global = Globals[GlobalIndex];
641959e7371SNicholas Wilson         GlobalType = &Global.Type;
642ef90ff36SNicholas Wilson         if (Global.SymbolName.empty())
643ef90ff36SNicholas Wilson           Global.SymbolName = Info.Name;
6446c899ba6SSam Clegg       } else {
6456c899ba6SSam Clegg         wasm::WasmImport &Import = *ImportedGlobals[Info.ElementIndex];
646bd481277SSam Clegg         if ((Info.Flags & wasm::WASM_SYMBOL_EXPLICIT_NAME) != 0) {
64729874ceaSDan Gohman           Info.Name = readString(Ctx);
648bd481277SSam Clegg           Info.ImportName = Import.Field;
649bd481277SSam Clegg         } else {
6506c899ba6SSam Clegg           Info.Name = Import.Field;
651bd481277SSam Clegg         }
6526c899ba6SSam Clegg         GlobalType = &Import.Global;
653bd481277SSam Clegg         if (!Import.Module.empty()) {
65429874ceaSDan Gohman           Info.ImportModule = Import.Module;
6556c899ba6SSam Clegg         }
656bd481277SSam Clegg       }
6576c899ba6SSam Clegg       break;
6586c899ba6SSam Clegg 
659388fb67bSPaulo Matos     case wasm::WASM_SYMBOL_TYPE_TABLE:
660388fb67bSPaulo Matos       Info.ElementIndex = readVaruint32(Ctx);
661a5a3659dSAndy Wingo       if (!isValidTableNumber(Info.ElementIndex) ||
662a5a3659dSAndy Wingo           IsDefined != isDefinedTableNumber(Info.ElementIndex))
663388fb67bSPaulo Matos         return make_error<GenericBinaryError>("invalid table symbol index",
664388fb67bSPaulo Matos                                               object_error::parse_failed);
665388fb67bSPaulo Matos       if (!IsDefined && (Info.Flags & wasm::WASM_SYMBOL_BINDING_MASK) ==
666388fb67bSPaulo Matos                             wasm::WASM_SYMBOL_BINDING_WEAK)
667388fb67bSPaulo Matos         return make_error<GenericBinaryError>("undefined weak table symbol",
668388fb67bSPaulo Matos                                               object_error::parse_failed);
669388fb67bSPaulo Matos       if (IsDefined) {
670388fb67bSPaulo Matos         Info.Name = readString(Ctx);
671a5a3659dSAndy Wingo         unsigned TableNumber = Info.ElementIndex - NumImportedTables;
672a5a3659dSAndy Wingo         wasm::WasmTable &Table = Tables[TableNumber];
67353e3b81fSAndy Wingo         TableType = &Table.Type;
6741933c9d4SAndy Wingo         if (Table.SymbolName.empty())
6751933c9d4SAndy Wingo           Table.SymbolName = Info.Name;
676388fb67bSPaulo Matos       } else {
677feac819eSAndy Wingo         wasm::WasmImport &Import = *ImportedTables[Info.ElementIndex];
678feac819eSAndy Wingo         if ((Info.Flags & wasm::WASM_SYMBOL_EXPLICIT_NAME) != 0) {
679feac819eSAndy Wingo           Info.Name = readString(Ctx);
680feac819eSAndy Wingo           Info.ImportName = Import.Field;
681feac819eSAndy Wingo         } else {
682feac819eSAndy Wingo           Info.Name = Import.Field;
683feac819eSAndy Wingo         }
68453e3b81fSAndy Wingo         TableType = &Import.Table;
685feac819eSAndy Wingo         if (!Import.Module.empty()) {
686feac819eSAndy Wingo           Info.ImportModule = Import.Module;
687feac819eSAndy Wingo         }
688388fb67bSPaulo Matos       }
689388fb67bSPaulo Matos       break;
690388fb67bSPaulo Matos 
6916c899ba6SSam Clegg     case wasm::WASM_SYMBOL_TYPE_DATA:
692b7c62394SSam Clegg       Info.Name = readString(Ctx);
6936c899ba6SSam Clegg       if (IsDefined) {
6943b29376eSWouter van Oortmerssen         auto Index = readVaruint32(Ctx);
6956c899ba6SSam Clegg         if (Index >= DataSegments.size())
6966c899ba6SSam Clegg           return make_error<GenericBinaryError>("invalid data symbol index",
6976c899ba6SSam Clegg                                                 object_error::parse_failed);
6983b29376eSWouter van Oortmerssen         auto Offset = readVaruint64(Ctx);
6993b29376eSWouter van Oortmerssen         auto Size = readVaruint64(Ctx);
700cd01430fSSam Clegg         size_t SegmentSize = DataSegments[Index].Data.Content.size();
701cd01430fSSam Clegg         if (Offset > SegmentSize)
702cd01430fSSam Clegg           return make_error<GenericBinaryError>(
703cd01430fSSam Clegg               "invalid data symbol offset: `" + Info.Name + "` (offset: " +
704cd01430fSSam Clegg                   Twine(Offset) + " segment size: " + Twine(SegmentSize) + ")",
7056c899ba6SSam Clegg               object_error::parse_failed);
7066c899ba6SSam Clegg         Info.DataRef = wasm::WasmDataReference{Index, Offset, Size};
7076c899ba6SSam Clegg       }
7086c899ba6SSam Clegg       break;
7096c899ba6SSam Clegg 
7106a31a0d6SSam Clegg     case wasm::WASM_SYMBOL_TYPE_SECTION: {
711d5504a0aSSam Clegg       if ((Info.Flags & wasm::WASM_SYMBOL_BINDING_MASK) !=
712d5504a0aSSam Clegg           wasm::WASM_SYMBOL_BINDING_LOCAL)
713e0658119SSam Clegg         return make_error<GenericBinaryError>(
71484c6f325SSam Clegg             "section symbols must have local binding",
715d5504a0aSSam Clegg             object_error::parse_failed);
716b7c62394SSam Clegg       Info.ElementIndex = readVaruint32(Ctx);
7176a31a0d6SSam Clegg       // Use somewhat unique section name as symbol name.
7186a31a0d6SSam Clegg       StringRef SectionName = Sections[Info.ElementIndex].Name;
7196a31a0d6SSam Clegg       Info.Name = SectionName;
7206a31a0d6SSam Clegg       break;
7216a31a0d6SSam Clegg     }
7226a31a0d6SSam Clegg 
7231d891d44SHeejin Ahn     case wasm::WASM_SYMBOL_TYPE_TAG: {
724da419bdbSHeejin Ahn       Info.ElementIndex = readVaruint32(Ctx);
7251d891d44SHeejin Ahn       if (!isValidTagIndex(Info.ElementIndex) ||
7261d891d44SHeejin Ahn           IsDefined != isDefinedTagIndex(Info.ElementIndex))
7271d891d44SHeejin Ahn         return make_error<GenericBinaryError>("invalid tag symbol index",
728da419bdbSHeejin Ahn                                               object_error::parse_failed);
729da419bdbSHeejin Ahn       if (!IsDefined && (Info.Flags & wasm::WASM_SYMBOL_BINDING_MASK) ==
730da419bdbSHeejin Ahn                             wasm::WASM_SYMBOL_BINDING_WEAK)
731da419bdbSHeejin Ahn         return make_error<GenericBinaryError>("undefined weak global symbol",
732da419bdbSHeejin Ahn                                               object_error::parse_failed);
733da419bdbSHeejin Ahn       if (IsDefined) {
734da419bdbSHeejin Ahn         Info.Name = readString(Ctx);
7351d891d44SHeejin Ahn         unsigned TagIndex = Info.ElementIndex - NumImportedTags;
7361d891d44SHeejin Ahn         wasm::WasmTag &Tag = Tags[TagIndex];
7373ec1760dSHeejin Ahn         Signature = &Signatures[Tag.SigIndex];
7381d891d44SHeejin Ahn         if (Tag.SymbolName.empty())
7391d891d44SHeejin Ahn           Tag.SymbolName = Info.Name;
740da419bdbSHeejin Ahn 
741da419bdbSHeejin Ahn       } else {
7421d891d44SHeejin Ahn         wasm::WasmImport &Import = *ImportedTags[Info.ElementIndex];
743bd481277SSam Clegg         if ((Info.Flags & wasm::WASM_SYMBOL_EXPLICIT_NAME) != 0) {
74429874ceaSDan Gohman           Info.Name = readString(Ctx);
745bd481277SSam Clegg           Info.ImportName = Import.Field;
746bd481277SSam Clegg         } else {
74729874ceaSDan Gohman           Info.Name = Import.Field;
748bd481277SSam Clegg         }
7493ec1760dSHeejin Ahn         Signature = &Signatures[Import.SigIndex];
750bd481277SSam Clegg         if (!Import.Module.empty()) {
75129874ceaSDan Gohman           Info.ImportModule = Import.Module;
752da419bdbSHeejin Ahn         }
753bd481277SSam Clegg       }
754da419bdbSHeejin Ahn       break;
755da419bdbSHeejin Ahn     }
756da419bdbSHeejin Ahn 
7576c899ba6SSam Clegg     default:
75884c6f325SSam Clegg       return make_error<GenericBinaryError>("invalid symbol type: " +
759d75b3719SSam Clegg                                                 Twine(unsigned(Info.Kind)),
7606c899ba6SSam Clegg                                             object_error::parse_failed);
7616c899ba6SSam Clegg     }
7626c899ba6SSam Clegg 
7636c899ba6SSam Clegg     if ((Info.Flags & wasm::WASM_SYMBOL_BINDING_MASK) !=
7646c899ba6SSam Clegg             wasm::WASM_SYMBOL_BINDING_LOCAL &&
7656c899ba6SSam Clegg         !SymbolNames.insert(Info.Name).second)
76684c6f325SSam Clegg       return make_error<GenericBinaryError>("duplicate symbol name " +
7676c899ba6SSam Clegg                                                 Twine(Info.Name),
7686c899ba6SSam Clegg                                             object_error::parse_failed);
7696c899ba6SSam Clegg     LinkingData.SymbolTable.emplace_back(Info);
770388fb67bSPaulo Matos     Symbols.emplace_back(LinkingData.SymbolTable.back(), GlobalType, TableType,
7713ec1760dSHeejin Ahn                          Signature);
772d34e60caSNicola Zaghen     LLVM_DEBUG(dbgs() << "Adding symbol: " << Symbols.back() << "\n");
7736c899ba6SSam Clegg   }
7746c899ba6SSam Clegg 
7756c899ba6SSam Clegg   return Error::success();
7766c899ba6SSam Clegg }
7776c899ba6SSam Clegg 
778b7c62394SSam Clegg Error WasmObjectFile::parseLinkingSectionComdat(ReadContext &Ctx) {
779b7c62394SSam Clegg   uint32_t ComdatCount = readVaruint32(Ctx);
780ea7caceeSSam Clegg   StringSet<> ComdatSet;
781027b9357SNicholas Wilson   for (unsigned ComdatIndex = 0; ComdatIndex < ComdatCount; ++ComdatIndex) {
782b7c62394SSam Clegg     StringRef Name = readString(Ctx);
783ea7caceeSSam Clegg     if (Name.empty() || !ComdatSet.insert(Name).second)
78484c6f325SSam Clegg       return make_error<GenericBinaryError>("bad/duplicate COMDAT name " +
785f208f631SHeejin Ahn                                                 Twine(Name),
786ea7caceeSSam Clegg                                             object_error::parse_failed);
787027b9357SNicholas Wilson     LinkingData.Comdats.emplace_back(Name);
788b7c62394SSam Clegg     uint32_t Flags = readVaruint32(Ctx);
789ea7caceeSSam Clegg     if (Flags != 0)
79084c6f325SSam Clegg       return make_error<GenericBinaryError>("unsupported COMDAT flags",
791ea7caceeSSam Clegg                                             object_error::parse_failed);
792ea7caceeSSam Clegg 
793b7c62394SSam Clegg     uint32_t EntryCount = readVaruint32(Ctx);
794ea7caceeSSam Clegg     while (EntryCount--) {
795b7c62394SSam Clegg       unsigned Kind = readVaruint32(Ctx);
796b7c62394SSam Clegg       unsigned Index = readVaruint32(Ctx);
797ea7caceeSSam Clegg       switch (Kind) {
798ea7caceeSSam Clegg       default:
79984c6f325SSam Clegg         return make_error<GenericBinaryError>("invalid COMDAT entry type",
800ea7caceeSSam Clegg                                               object_error::parse_failed);
801ea7caceeSSam Clegg       case wasm::WASM_COMDAT_DATA:
802ea7caceeSSam Clegg         if (Index >= DataSegments.size())
803f208f631SHeejin Ahn           return make_error<GenericBinaryError>(
804f208f631SHeejin Ahn               "COMDAT data index out of range", object_error::parse_failed);
805027b9357SNicholas Wilson         if (DataSegments[Index].Data.Comdat != UINT32_MAX)
80684c6f325SSam Clegg           return make_error<GenericBinaryError>("data segment in two COMDATs",
807ea7caceeSSam Clegg                                                 object_error::parse_failed);
808027b9357SNicholas Wilson         DataSegments[Index].Data.Comdat = ComdatIndex;
809ea7caceeSSam Clegg         break;
810ea7caceeSSam Clegg       case wasm::WASM_COMDAT_FUNCTION:
81123012e98SSam Clegg         if (!isDefinedFunctionIndex(Index))
812f208f631SHeejin Ahn           return make_error<GenericBinaryError>(
813f208f631SHeejin Ahn               "COMDAT function index out of range", object_error::parse_failed);
814027b9357SNicholas Wilson         if (getDefinedFunction(Index).Comdat != UINT32_MAX)
81584c6f325SSam Clegg           return make_error<GenericBinaryError>("function in two COMDATs",
816ea7caceeSSam Clegg                                                 object_error::parse_failed);
817027b9357SNicholas Wilson         getDefinedFunction(Index).Comdat = ComdatIndex;
818ea7caceeSSam Clegg         break;
8190a391060SDerek Schuff       case wasm::WASM_COMDAT_SECTION:
8200a391060SDerek Schuff         if (Index >= Sections.size())
8210a391060SDerek Schuff           return make_error<GenericBinaryError>(
8220a391060SDerek Schuff               "COMDAT section index out of range", object_error::parse_failed);
8230a391060SDerek Schuff         if (Sections[Index].Type != wasm::WASM_SEC_CUSTOM)
8240a391060SDerek Schuff           return make_error<GenericBinaryError>(
82584c6f325SSam Clegg               "non-custom section in a COMDAT", object_error::parse_failed);
8260a391060SDerek Schuff         Sections[Index].Comdat = ComdatIndex;
8270a391060SDerek Schuff         break;
828ea7caceeSSam Clegg       }
829ea7caceeSSam Clegg     }
830ea7caceeSSam Clegg   }
831ea7caceeSSam Clegg   return Error::success();
832ea7caceeSSam Clegg }
833ea7caceeSSam Clegg 
834cbda16ebSThomas Lively Error WasmObjectFile::parseProducersSection(ReadContext &Ctx) {
835cbda16ebSThomas Lively   llvm::SmallSet<StringRef, 3> FieldsSeen;
836cbda16ebSThomas Lively   uint32_t Fields = readVaruint32(Ctx);
83718c56a07SHeejin Ahn   for (size_t I = 0; I < Fields; ++I) {
838cbda16ebSThomas Lively     StringRef FieldName = readString(Ctx);
839cbda16ebSThomas Lively     if (!FieldsSeen.insert(FieldName).second)
840cbda16ebSThomas Lively       return make_error<GenericBinaryError>(
84184c6f325SSam Clegg           "producers section does not have unique fields",
842cbda16ebSThomas Lively           object_error::parse_failed);
843cbda16ebSThomas Lively     std::vector<std::pair<std::string, std::string>> *ProducerVec = nullptr;
844cbda16ebSThomas Lively     if (FieldName == "language") {
845cbda16ebSThomas Lively       ProducerVec = &ProducerInfo.Languages;
846cbda16ebSThomas Lively     } else if (FieldName == "processed-by") {
847cbda16ebSThomas Lively       ProducerVec = &ProducerInfo.Tools;
848cbda16ebSThomas Lively     } else if (FieldName == "sdk") {
849cbda16ebSThomas Lively       ProducerVec = &ProducerInfo.SDKs;
850cbda16ebSThomas Lively     } else {
851cbda16ebSThomas Lively       return make_error<GenericBinaryError>(
85284c6f325SSam Clegg           "producers section field is not named one of language, processed-by, "
853cbda16ebSThomas Lively           "or sdk",
854cbda16ebSThomas Lively           object_error::parse_failed);
855cbda16ebSThomas Lively     }
856cbda16ebSThomas Lively     uint32_t ValueCount = readVaruint32(Ctx);
857cbda16ebSThomas Lively     llvm::SmallSet<StringRef, 8> ProducersSeen;
85818c56a07SHeejin Ahn     for (size_t J = 0; J < ValueCount; ++J) {
859cbda16ebSThomas Lively       StringRef Name = readString(Ctx);
860cbda16ebSThomas Lively       StringRef Version = readString(Ctx);
861cbda16ebSThomas Lively       if (!ProducersSeen.insert(Name).second) {
862cbda16ebSThomas Lively         return make_error<GenericBinaryError>(
86384c6f325SSam Clegg             "producers section contains repeated producer",
864cbda16ebSThomas Lively             object_error::parse_failed);
865cbda16ebSThomas Lively       }
866a9bc7b83SBenjamin Kramer       ProducerVec->emplace_back(std::string(Name), std::string(Version));
867cbda16ebSThomas Lively     }
868cbda16ebSThomas Lively   }
869cbda16ebSThomas Lively   if (Ctx.Ptr != Ctx.End)
87084c6f325SSam Clegg     return make_error<GenericBinaryError>("producers section ended prematurely",
871cbda16ebSThomas Lively                                           object_error::parse_failed);
872cbda16ebSThomas Lively   return Error::success();
873cbda16ebSThomas Lively }
874cbda16ebSThomas Lively 
875f6f4f843SThomas Lively Error WasmObjectFile::parseTargetFeaturesSection(ReadContext &Ctx) {
876f6f4f843SThomas Lively   llvm::SmallSet<std::string, 8> FeaturesSeen;
877f6f4f843SThomas Lively   uint32_t FeatureCount = readVaruint32(Ctx);
878f6f4f843SThomas Lively   for (size_t I = 0; I < FeatureCount; ++I) {
879f6f4f843SThomas Lively     wasm::WasmFeatureEntry Feature;
880f6f4f843SThomas Lively     Feature.Prefix = readUint8(Ctx);
881f6f4f843SThomas Lively     switch (Feature.Prefix) {
882f6f4f843SThomas Lively     case wasm::WASM_FEATURE_PREFIX_USED:
883f6f4f843SThomas Lively     case wasm::WASM_FEATURE_PREFIX_REQUIRED:
884f6f4f843SThomas Lively     case wasm::WASM_FEATURE_PREFIX_DISALLOWED:
885f6f4f843SThomas Lively       break;
886f6f4f843SThomas Lively     default:
88784c6f325SSam Clegg       return make_error<GenericBinaryError>("unknown feature policy prefix",
888f6f4f843SThomas Lively                                             object_error::parse_failed);
889f6f4f843SThomas Lively     }
890adcd0268SBenjamin Kramer     Feature.Name = std::string(readString(Ctx));
891f6f4f843SThomas Lively     if (!FeaturesSeen.insert(Feature.Name).second)
892f6f4f843SThomas Lively       return make_error<GenericBinaryError>(
89384c6f325SSam Clegg           "target features section contains repeated feature \"" +
894f6f4f843SThomas Lively               Feature.Name + "\"",
895f6f4f843SThomas Lively           object_error::parse_failed);
896f6f4f843SThomas Lively     TargetFeatures.push_back(Feature);
897f6f4f843SThomas Lively   }
898f6f4f843SThomas Lively   if (Ctx.Ptr != Ctx.End)
899f6f4f843SThomas Lively     return make_error<GenericBinaryError>(
90084c6f325SSam Clegg         "target features section ended prematurely",
901f6f4f843SThomas Lively         object_error::parse_failed);
902f6f4f843SThomas Lively   return Error::success();
903f6f4f843SThomas Lively }
904f6f4f843SThomas Lively 
905b7c62394SSam Clegg Error WasmObjectFile::parseRelocSection(StringRef Name, ReadContext &Ctx) {
906b7c62394SSam Clegg   uint32_t SectionIndex = readVaruint32(Ctx);
9076f08c84aSSam Clegg   if (SectionIndex >= Sections.size())
90884c6f325SSam Clegg     return make_error<GenericBinaryError>("invalid section index",
909d3d84fddSDerek Schuff                                           object_error::parse_failed);
9106f08c84aSSam Clegg   WasmSection &Section = Sections[SectionIndex];
911b7c62394SSam Clegg   uint32_t RelocCount = readVaruint32(Ctx);
9126f08c84aSSam Clegg   uint32_t EndOffset = Section.Content.size();
913f77dc2a8SSam Clegg   uint32_t PreviousOffset = 0;
914d3d84fddSDerek Schuff   while (RelocCount--) {
915b3748f71SNicholas Wilson     wasm::WasmRelocation Reloc = {};
916d75b3719SSam Clegg     uint32_t type = readVaruint32(Ctx);
917d75b3719SSam Clegg     Reloc.Type = type;
918b7c62394SSam Clegg     Reloc.Offset = readVaruint32(Ctx);
919f77dc2a8SSam Clegg     if (Reloc.Offset < PreviousOffset)
92084c6f325SSam Clegg       return make_error<GenericBinaryError>("relocations not in offset order",
921f77dc2a8SSam Clegg                                             object_error::parse_failed);
922f77dc2a8SSam Clegg     PreviousOffset = Reloc.Offset;
923b7c62394SSam Clegg     Reloc.Index = readVaruint32(Ctx);
924d75b3719SSam Clegg     switch (type) {
925d1152a26SSam Clegg     case wasm::R_WASM_FUNCTION_INDEX_LEB:
926d1152a26SSam Clegg     case wasm::R_WASM_TABLE_INDEX_SLEB:
927cc1b9b68SWouter van Oortmerssen     case wasm::R_WASM_TABLE_INDEX_SLEB64:
928d1152a26SSam Clegg     case wasm::R_WASM_TABLE_INDEX_I32:
929cc1b9b68SWouter van Oortmerssen     case wasm::R_WASM_TABLE_INDEX_I64:
9302a7cac93SSam Clegg     case wasm::R_WASM_TABLE_INDEX_REL_SLEB:
9313a293cbfSWouter van Oortmerssen     case wasm::R_WASM_TABLE_INDEX_REL_SLEB64:
932b3748f71SNicholas Wilson       if (!isValidFunctionSymbol(Reloc.Index))
93384c6f325SSam Clegg         return make_error<GenericBinaryError>(
93484c6f325SSam Clegg             "invalid relocation function index", object_error::parse_failed);
935b3748f71SNicholas Wilson       break;
93669e2797eSPaulo Matos     case wasm::R_WASM_TABLE_NUMBER_LEB:
93769e2797eSPaulo Matos       if (!isValidTableSymbol(Reloc.Index))
93884c6f325SSam Clegg         return make_error<GenericBinaryError>("invalid relocation table index",
93969e2797eSPaulo Matos                                               object_error::parse_failed);
94069e2797eSPaulo Matos       break;
941d1152a26SSam Clegg     case wasm::R_WASM_TYPE_INDEX_LEB:
942b3748f71SNicholas Wilson       if (Reloc.Index >= Signatures.size())
94384c6f325SSam Clegg         return make_error<GenericBinaryError>("invalid relocation type index",
944b3748f71SNicholas Wilson                                               object_error::parse_failed);
945b3748f71SNicholas Wilson       break;
946d1152a26SSam Clegg     case wasm::R_WASM_GLOBAL_INDEX_LEB:
947492f7529SSam Clegg       // R_WASM_GLOBAL_INDEX_LEB are can be used against function and data
9487eacefedSDavid Candler       // symbols to refer to their GOT entries.
949492f7529SSam Clegg       if (!isValidGlobalSymbol(Reloc.Index) &&
950492f7529SSam Clegg           !isValidDataSymbol(Reloc.Index) &&
951492f7529SSam Clegg           !isValidFunctionSymbol(Reloc.Index))
95284c6f325SSam Clegg         return make_error<GenericBinaryError>("invalid relocation global index",
953b3748f71SNicholas Wilson                                               object_error::parse_failed);
954d3d84fddSDerek Schuff       break;
95548139ebcSWouter van Oortmerssen     case wasm::R_WASM_GLOBAL_INDEX_I32:
95648139ebcSWouter van Oortmerssen       if (!isValidGlobalSymbol(Reloc.Index))
95784c6f325SSam Clegg         return make_error<GenericBinaryError>("invalid relocation global index",
95848139ebcSWouter van Oortmerssen                                               object_error::parse_failed);
95948139ebcSWouter van Oortmerssen       break;
9601d891d44SHeejin Ahn     case wasm::R_WASM_TAG_INDEX_LEB:
9611d891d44SHeejin Ahn       if (!isValidTagSymbol(Reloc.Index))
9621d891d44SHeejin Ahn         return make_error<GenericBinaryError>("invalid relocation tag index",
963da419bdbSHeejin Ahn                                               object_error::parse_failed);
964da419bdbSHeejin Ahn       break;
965d1152a26SSam Clegg     case wasm::R_WASM_MEMORY_ADDR_LEB:
966d1152a26SSam Clegg     case wasm::R_WASM_MEMORY_ADDR_SLEB:
967d1152a26SSam Clegg     case wasm::R_WASM_MEMORY_ADDR_I32:
9682a7cac93SSam Clegg     case wasm::R_WASM_MEMORY_ADDR_REL_SLEB:
969a28a4662SSam Clegg     case wasm::R_WASM_MEMORY_ADDR_TLS_SLEB:
970aa0c571aSYuta Saito     case wasm::R_WASM_MEMORY_ADDR_LOCREL_I32:
971b3748f71SNicholas Wilson       if (!isValidDataSymbol(Reloc.Index))
97284c6f325SSam Clegg         return make_error<GenericBinaryError>("invalid relocation data index",
973b3748f71SNicholas Wilson                                               object_error::parse_failed);
974b7c62394SSam Clegg       Reloc.Addend = readVarint32(Ctx);
975d3d84fddSDerek Schuff       break;
9763b29376eSWouter van Oortmerssen     case wasm::R_WASM_MEMORY_ADDR_LEB64:
9773b29376eSWouter van Oortmerssen     case wasm::R_WASM_MEMORY_ADDR_SLEB64:
9783b29376eSWouter van Oortmerssen     case wasm::R_WASM_MEMORY_ADDR_I64:
9793b29376eSWouter van Oortmerssen     case wasm::R_WASM_MEMORY_ADDR_REL_SLEB64:
980670944fbSWouter van Oortmerssen     case wasm::R_WASM_MEMORY_ADDR_TLS_SLEB64:
9813b29376eSWouter van Oortmerssen       if (!isValidDataSymbol(Reloc.Index))
98284c6f325SSam Clegg         return make_error<GenericBinaryError>("invalid relocation data index",
9833b29376eSWouter van Oortmerssen                                               object_error::parse_failed);
9843b29376eSWouter van Oortmerssen       Reloc.Addend = readVarint64(Ctx);
9853b29376eSWouter van Oortmerssen       break;
986d1152a26SSam Clegg     case wasm::R_WASM_FUNCTION_OFFSET_I32:
9876a31a0d6SSam Clegg       if (!isValidFunctionSymbol(Reloc.Index))
98884c6f325SSam Clegg         return make_error<GenericBinaryError>(
98984c6f325SSam Clegg             "invalid relocation function index", object_error::parse_failed);
990b7c62394SSam Clegg       Reloc.Addend = readVarint32(Ctx);
9916a31a0d6SSam Clegg       break;
99216f02431SWouter van Oortmerssen     case wasm::R_WASM_FUNCTION_OFFSET_I64:
99316f02431SWouter van Oortmerssen       if (!isValidFunctionSymbol(Reloc.Index))
99484c6f325SSam Clegg         return make_error<GenericBinaryError>(
99584c6f325SSam Clegg             "invalid relocation function index", object_error::parse_failed);
99616f02431SWouter van Oortmerssen       Reloc.Addend = readVarint64(Ctx);
99716f02431SWouter van Oortmerssen       break;
998d1152a26SSam Clegg     case wasm::R_WASM_SECTION_OFFSET_I32:
9996a31a0d6SSam Clegg       if (!isValidSectionSymbol(Reloc.Index))
100084c6f325SSam Clegg         return make_error<GenericBinaryError>(
100184c6f325SSam Clegg             "invalid relocation section index", object_error::parse_failed);
1002b7c62394SSam Clegg       Reloc.Addend = readVarint32(Ctx);
10036a31a0d6SSam Clegg       break;
1004d3d84fddSDerek Schuff     default:
100584c6f325SSam Clegg       return make_error<GenericBinaryError>("invalid relocation type: " +
100684c6f325SSam Clegg                                                 Twine(type),
100784c6f325SSam Clegg                                             object_error::parse_failed);
1008d3d84fddSDerek Schuff     }
1009b3748f71SNicholas Wilson 
1010b3748f71SNicholas Wilson     // Relocations must fit inside the section, and must appear in order.  They
1011b3748f71SNicholas Wilson     // also shouldn't overlap a function/element boundary, but we don't bother
1012b3748f71SNicholas Wilson     // to check that.
1013b3748f71SNicholas Wilson     uint64_t Size = 5;
10143b29376eSWouter van Oortmerssen     if (Reloc.Type == wasm::R_WASM_MEMORY_ADDR_LEB64 ||
10153b29376eSWouter van Oortmerssen         Reloc.Type == wasm::R_WASM_MEMORY_ADDR_SLEB64 ||
10163b29376eSWouter van Oortmerssen         Reloc.Type == wasm::R_WASM_MEMORY_ADDR_REL_SLEB64)
10173b29376eSWouter van Oortmerssen       Size = 10;
1018d1152a26SSam Clegg     if (Reloc.Type == wasm::R_WASM_TABLE_INDEX_I32 ||
1019d1152a26SSam Clegg         Reloc.Type == wasm::R_WASM_MEMORY_ADDR_I32 ||
1020aa0c571aSYuta Saito         Reloc.Type == wasm::R_WASM_MEMORY_ADDR_LOCREL_I32 ||
1021d1152a26SSam Clegg         Reloc.Type == wasm::R_WASM_SECTION_OFFSET_I32 ||
102248139ebcSWouter van Oortmerssen         Reloc.Type == wasm::R_WASM_FUNCTION_OFFSET_I32 ||
102348139ebcSWouter van Oortmerssen         Reloc.Type == wasm::R_WASM_GLOBAL_INDEX_I32)
1024b3748f71SNicholas Wilson       Size = 4;
1025cc1b9b68SWouter van Oortmerssen     if (Reloc.Type == wasm::R_WASM_TABLE_INDEX_I64 ||
102616f02431SWouter van Oortmerssen         Reloc.Type == wasm::R_WASM_MEMORY_ADDR_I64 ||
102716f02431SWouter van Oortmerssen         Reloc.Type == wasm::R_WASM_FUNCTION_OFFSET_I64)
10283b29376eSWouter van Oortmerssen       Size = 8;
1029fa5a04fbSSam Clegg     if (Reloc.Offset + Size > EndOffset)
103084c6f325SSam Clegg       return make_error<GenericBinaryError>("invalid relocation offset",
1031b3748f71SNicholas Wilson                                             object_error::parse_failed);
1032b3748f71SNicholas Wilson 
10336f08c84aSSam Clegg     Section.Relocations.push_back(Reloc);
1034d3d84fddSDerek Schuff   }
1035b7c62394SSam Clegg   if (Ctx.Ptr != Ctx.End)
103684c6f325SSam Clegg     return make_error<GenericBinaryError>("reloc section ended prematurely",
1037d3d84fddSDerek Schuff                                           object_error::parse_failed);
1038d3d84fddSDerek Schuff   return Error::success();
1039d3d84fddSDerek Schuff }
1040d3d84fddSDerek Schuff 
1041b7c62394SSam Clegg Error WasmObjectFile::parseCustomSection(WasmSection &Sec, ReadContext &Ctx) {
1042e4afbc68SSam Clegg   if (Sec.Name == "dylink") {
1043e4afbc68SSam Clegg     if (Error Err = parseDylinkSection(Ctx))
1044e4afbc68SSam Clegg       return Err;
1045b78c85a4SSam Clegg   } else if (Sec.Name == "dylink.0") {
1046b78c85a4SSam Clegg     if (Error Err = parseDylink0Section(Ctx))
1047b78c85a4SSam Clegg       return Err;
1048e4afbc68SSam Clegg   } else if (Sec.Name == "name") {
1049b7c62394SSam Clegg     if (Error Err = parseNameSection(Ctx))
1050d3d84fddSDerek Schuff       return Err;
1051b7787fd0SSam Clegg   } else if (Sec.Name == "linking") {
1052b7c62394SSam Clegg     if (Error Err = parseLinkingSection(Ctx))
1053b7787fd0SSam Clegg       return Err;
1054cbda16ebSThomas Lively   } else if (Sec.Name == "producers") {
1055cbda16ebSThomas Lively     if (Error Err = parseProducersSection(Ctx))
1056cbda16ebSThomas Lively       return Err;
1057f6f4f843SThomas Lively   } else if (Sec.Name == "target_features") {
1058f6f4f843SThomas Lively     if (Error Err = parseTargetFeaturesSection(Ctx))
1059f6f4f843SThomas Lively       return Err;
1060d3d84fddSDerek Schuff   } else if (Sec.Name.startswith("reloc.")) {
1061b7c62394SSam Clegg     if (Error Err = parseRelocSection(Sec.Name, Ctx))
1062d3d84fddSDerek Schuff       return Err;
1063d3d84fddSDerek Schuff   }
1064d3d84fddSDerek Schuff   return Error::success();
1065d3d84fddSDerek Schuff }
1066d3d84fddSDerek Schuff 
1067b7c62394SSam Clegg Error WasmObjectFile::parseTypeSection(ReadContext &Ctx) {
1068b7c62394SSam Clegg   uint32_t Count = readVaruint32(Ctx);
1069d3d84fddSDerek Schuff   Signatures.reserve(Count);
1070d3d84fddSDerek Schuff   while (Count--) {
1071d3d84fddSDerek Schuff     wasm::WasmSignature Sig;
1072b7c62394SSam Clegg     uint8_t Form = readUint8(Ctx);
1073d3d84fddSDerek Schuff     if (Form != wasm::WASM_TYPE_FUNC) {
107484c6f325SSam Clegg       return make_error<GenericBinaryError>("invalid signature type",
1075d3d84fddSDerek Schuff                                             object_error::parse_failed);
1076d3d84fddSDerek Schuff     }
1077b7c62394SSam Clegg     uint32_t ParamCount = readVaruint32(Ctx);
107877a7a380SDerek Schuff     Sig.Params.reserve(ParamCount);
1079d3d84fddSDerek Schuff     while (ParamCount--) {
1080b7c62394SSam Clegg       uint32_t ParamType = readUint8(Ctx);
108177a7a380SDerek Schuff       Sig.Params.push_back(wasm::ValType(ParamType));
1082d3d84fddSDerek Schuff     }
1083b7c62394SSam Clegg     uint32_t ReturnCount = readVaruint32(Ctx);
1084393d0f79SThomas Lively     while (ReturnCount--) {
1085393d0f79SThomas Lively       uint32_t ReturnType = readUint8(Ctx);
1086393d0f79SThomas Lively       Sig.Returns.push_back(wasm::ValType(ReturnType));
1087d3d84fddSDerek Schuff     }
108877a7a380SDerek Schuff     Signatures.push_back(std::move(Sig));
1089d3d84fddSDerek Schuff   }
1090b7c62394SSam Clegg   if (Ctx.Ptr != Ctx.End)
109184c6f325SSam Clegg     return make_error<GenericBinaryError>("type section ended prematurely",
1092d3d84fddSDerek Schuff                                           object_error::parse_failed);
1093d3d84fddSDerek Schuff   return Error::success();
1094d3d84fddSDerek Schuff }
1095d3d84fddSDerek Schuff 
1096b7c62394SSam Clegg Error WasmObjectFile::parseImportSection(ReadContext &Ctx) {
1097b7c62394SSam Clegg   uint32_t Count = readVaruint32(Ctx);
10983ec1760dSHeejin Ahn   uint32_t NumTypes = Signatures.size();
1099d3d84fddSDerek Schuff   Imports.reserve(Count);
110018c56a07SHeejin Ahn   for (uint32_t I = 0; I < Count; I++) {
1101d3d84fddSDerek Schuff     wasm::WasmImport Im;
1102b7c62394SSam Clegg     Im.Module = readString(Ctx);
1103b7c62394SSam Clegg     Im.Field = readString(Ctx);
1104b7c62394SSam Clegg     Im.Kind = readUint8(Ctx);
1105d3d84fddSDerek Schuff     switch (Im.Kind) {
1106d3d84fddSDerek Schuff     case wasm::WASM_EXTERNAL_FUNCTION:
1107a3b9fe6aSSam Clegg       NumImportedFunctions++;
1108b7c62394SSam Clegg       Im.SigIndex = readVaruint32(Ctx);
11093ec1760dSHeejin Ahn       if (Im.SigIndex >= NumTypes)
11103ec1760dSHeejin Ahn         return make_error<GenericBinaryError>("invalid function type",
11113ec1760dSHeejin Ahn                                               object_error::parse_failed);
1112d3d84fddSDerek Schuff       break;
1113d3d84fddSDerek Schuff     case wasm::WASM_EXTERNAL_GLOBAL:
1114a3b9fe6aSSam Clegg       NumImportedGlobals++;
1115b7c62394SSam Clegg       Im.Global.Type = readUint8(Ctx);
1116b7c62394SSam Clegg       Im.Global.Mutable = readVaruint1(Ctx);
1117d3d84fddSDerek Schuff       break;
11182ffff5afSSam Clegg     case wasm::WASM_EXTERNAL_MEMORY:
1119b7c62394SSam Clegg       Im.Memory = readLimits(Ctx);
112029f8c9f6SWouter van Oortmerssen       if (Im.Memory.Flags & wasm::WASM_LIMITS_FLAG_IS_64)
112129f8c9f6SWouter van Oortmerssen         HasMemory64 = true;
11222ffff5afSSam Clegg       break;
1123388fb67bSPaulo Matos     case wasm::WASM_EXTERNAL_TABLE: {
11241933c9d4SAndy Wingo       Im.Table = readTableType(Ctx);
1125388fb67bSPaulo Matos       NumImportedTables++;
1126388fb67bSPaulo Matos       auto ElemType = Im.Table.ElemType;
1127388fb67bSPaulo Matos       if (ElemType != wasm::WASM_TYPE_FUNCREF &&
1128388fb67bSPaulo Matos           ElemType != wasm::WASM_TYPE_EXTERNREF)
112984c6f325SSam Clegg         return make_error<GenericBinaryError>("invalid table element type",
11302ffff5afSSam Clegg                                               object_error::parse_failed);
11312ffff5afSSam Clegg       break;
1132388fb67bSPaulo Matos     }
11331d891d44SHeejin Ahn     case wasm::WASM_EXTERNAL_TAG:
11341d891d44SHeejin Ahn       NumImportedTags++;
11359261ee32SHeejin Ahn       if (readUint8(Ctx) != 0) // Reserved 'attribute' field
11369261ee32SHeejin Ahn         return make_error<GenericBinaryError>("invalid attribute",
11379261ee32SHeejin Ahn                                               object_error::parse_failed);
11383ec1760dSHeejin Ahn       Im.SigIndex = readVaruint32(Ctx);
11393ec1760dSHeejin Ahn       if (Im.SigIndex >= NumTypes)
11403ec1760dSHeejin Ahn         return make_error<GenericBinaryError>("invalid tag type",
11413ec1760dSHeejin Ahn                                               object_error::parse_failed);
1142da419bdbSHeejin Ahn       break;
1143d3d84fddSDerek Schuff     default:
114484c6f325SSam Clegg       return make_error<GenericBinaryError>("unexpected import kind",
1145f208f631SHeejin Ahn                                             object_error::parse_failed);
1146d3d84fddSDerek Schuff     }
1147d3d84fddSDerek Schuff     Imports.push_back(Im);
1148d3d84fddSDerek Schuff   }
1149b7c62394SSam Clegg   if (Ctx.Ptr != Ctx.End)
115084c6f325SSam Clegg     return make_error<GenericBinaryError>("import section ended prematurely",
1151d3d84fddSDerek Schuff                                           object_error::parse_failed);
1152d3d84fddSDerek Schuff   return Error::success();
1153d3d84fddSDerek Schuff }
1154d3d84fddSDerek Schuff 
1155b7c62394SSam Clegg Error WasmObjectFile::parseFunctionSection(ReadContext &Ctx) {
1156b7c62394SSam Clegg   uint32_t Count = readVaruint32(Ctx);
1157c0039de2SSam Clegg   Functions.reserve(Count);
11583b825105SNicholas Wilson   uint32_t NumTypes = Signatures.size();
1159d3d84fddSDerek Schuff   while (Count--) {
1160b7c62394SSam Clegg     uint32_t Type = readVaruint32(Ctx);
11613b825105SNicholas Wilson     if (Type >= NumTypes)
116284c6f325SSam Clegg       return make_error<GenericBinaryError>("invalid function type",
11633b825105SNicholas Wilson                                             object_error::parse_failed);
1164c0039de2SSam Clegg     wasm::WasmFunction F;
1165c0039de2SSam Clegg     F.SigIndex = Type;
1166c0039de2SSam Clegg     Functions.push_back(F);
1167d3d84fddSDerek Schuff   }
1168b7c62394SSam Clegg   if (Ctx.Ptr != Ctx.End)
116984c6f325SSam Clegg     return make_error<GenericBinaryError>("function section ended prematurely",
1170d3d84fddSDerek Schuff                                           object_error::parse_failed);
1171d3d84fddSDerek Schuff   return Error::success();
1172d3d84fddSDerek Schuff }
1173d3d84fddSDerek Schuff 
1174b7c62394SSam Clegg Error WasmObjectFile::parseTableSection(ReadContext &Ctx) {
1175feac819eSAndy Wingo   TableSection = Sections.size();
1176b7c62394SSam Clegg   uint32_t Count = readVaruint32(Ctx);
1177d3d84fddSDerek Schuff   Tables.reserve(Count);
1178d3d84fddSDerek Schuff   while (Count--) {
11791933c9d4SAndy Wingo     wasm::WasmTable T;
11801933c9d4SAndy Wingo     T.Type = readTableType(Ctx);
1181388fb67bSPaulo Matos     T.Index = NumImportedTables + Tables.size();
1182388fb67bSPaulo Matos     Tables.push_back(T);
11831933c9d4SAndy Wingo     auto ElemType = Tables.back().Type.ElemType;
1184388fb67bSPaulo Matos     if (ElemType != wasm::WASM_TYPE_FUNCREF &&
1185388fb67bSPaulo Matos         ElemType != wasm::WASM_TYPE_EXTERNREF) {
118684c6f325SSam Clegg       return make_error<GenericBinaryError>("invalid table element type",
1187d3d84fddSDerek Schuff                                             object_error::parse_failed);
1188d3d84fddSDerek Schuff     }
1189d3d84fddSDerek Schuff   }
1190b7c62394SSam Clegg   if (Ctx.Ptr != Ctx.End)
119184c6f325SSam Clegg     return make_error<GenericBinaryError>("table section ended prematurely",
1192d3d84fddSDerek Schuff                                           object_error::parse_failed);
1193d3d84fddSDerek Schuff   return Error::success();
1194d3d84fddSDerek Schuff }
1195d3d84fddSDerek Schuff 
1196b7c62394SSam Clegg Error WasmObjectFile::parseMemorySection(ReadContext &Ctx) {
1197b7c62394SSam Clegg   uint32_t Count = readVaruint32(Ctx);
1198d3d84fddSDerek Schuff   Memories.reserve(Count);
1199d3d84fddSDerek Schuff   while (Count--) {
120029f8c9f6SWouter van Oortmerssen     auto Limits = readLimits(Ctx);
120129f8c9f6SWouter van Oortmerssen     if (Limits.Flags & wasm::WASM_LIMITS_FLAG_IS_64)
120229f8c9f6SWouter van Oortmerssen       HasMemory64 = true;
120329f8c9f6SWouter van Oortmerssen     Memories.push_back(Limits);
1204d3d84fddSDerek Schuff   }
1205b7c62394SSam Clegg   if (Ctx.Ptr != Ctx.End)
120684c6f325SSam Clegg     return make_error<GenericBinaryError>("memory section ended prematurely",
1207d3d84fddSDerek Schuff                                           object_error::parse_failed);
1208d3d84fddSDerek Schuff   return Error::success();
1209d3d84fddSDerek Schuff }
1210d3d84fddSDerek Schuff 
12111d891d44SHeejin Ahn Error WasmObjectFile::parseTagSection(ReadContext &Ctx) {
12121d891d44SHeejin Ahn   TagSection = Sections.size();
1213144ec1c3SHeejin Ahn   uint32_t Count = readVaruint32(Ctx);
12141d891d44SHeejin Ahn   Tags.reserve(Count);
12153ec1760dSHeejin Ahn   uint32_t NumTypes = Signatures.size();
1216f93426c5SHeejin Ahn   while (Count--) {
12179261ee32SHeejin Ahn     if (readUint8(Ctx) != 0) // Reserved 'attribute' field
12183ec1760dSHeejin Ahn       return make_error<GenericBinaryError>("invalid attribute",
12193ec1760dSHeejin Ahn                                             object_error::parse_failed);
12203ec1760dSHeejin Ahn     uint32_t Type = readVaruint32(Ctx);
12213ec1760dSHeejin Ahn     if (Type >= NumTypes)
12223ec1760dSHeejin Ahn       return make_error<GenericBinaryError>("invalid tag type",
12233ec1760dSHeejin Ahn                                             object_error::parse_failed);
12241d891d44SHeejin Ahn     wasm::WasmTag Tag;
12251d891d44SHeejin Ahn     Tag.Index = NumImportedTags + Tags.size();
12263ec1760dSHeejin Ahn     Tag.SigIndex = Type;
12271d891d44SHeejin Ahn     Tags.push_back(Tag);
1228f93426c5SHeejin Ahn   }
1229f93426c5SHeejin Ahn 
1230f93426c5SHeejin Ahn   if (Ctx.Ptr != Ctx.End)
12311d891d44SHeejin Ahn     return make_error<GenericBinaryError>("tag section ended prematurely",
1232f93426c5SHeejin Ahn                                           object_error::parse_failed);
1233f93426c5SHeejin Ahn   return Error::success();
1234f93426c5SHeejin Ahn }
1235f93426c5SHeejin Ahn 
1236b7c62394SSam Clegg Error WasmObjectFile::parseGlobalSection(ReadContext &Ctx) {
12376c899ba6SSam Clegg   GlobalSection = Sections.size();
1238b7c62394SSam Clegg   uint32_t Count = readVaruint32(Ctx);
1239d3d84fddSDerek Schuff   Globals.reserve(Count);
1240d3d84fddSDerek Schuff   while (Count--) {
1241d3d84fddSDerek Schuff     wasm::WasmGlobal Global;
1242e53af7f6SSam Clegg     Global.Index = NumImportedGlobals + Globals.size();
1243b7c62394SSam Clegg     Global.Type.Type = readUint8(Ctx);
1244b7c62394SSam Clegg     Global.Type.Mutable = readVaruint1(Ctx);
1245b7c62394SSam Clegg     if (Error Err = readInitExpr(Global.InitExpr, Ctx))
1246d3d84fddSDerek Schuff       return Err;
1247d3d84fddSDerek Schuff     Globals.push_back(Global);
1248d3d84fddSDerek Schuff   }
1249b7c62394SSam Clegg   if (Ctx.Ptr != Ctx.End)
125084c6f325SSam Clegg     return make_error<GenericBinaryError>("global section ended prematurely",
1251d3d84fddSDerek Schuff                                           object_error::parse_failed);
1252d3d84fddSDerek Schuff   return Error::success();
1253d3d84fddSDerek Schuff }
1254d3d84fddSDerek Schuff 
1255b7c62394SSam Clegg Error WasmObjectFile::parseExportSection(ReadContext &Ctx) {
1256b7c62394SSam Clegg   uint32_t Count = readVaruint32(Ctx);
1257d3d84fddSDerek Schuff   Exports.reserve(Count);
125818c56a07SHeejin Ahn   for (uint32_t I = 0; I < Count; I++) {
1259d3d84fddSDerek Schuff     wasm::WasmExport Ex;
1260b7c62394SSam Clegg     Ex.Name = readString(Ctx);
1261b7c62394SSam Clegg     Ex.Kind = readUint8(Ctx);
1262b7c62394SSam Clegg     Ex.Index = readVaruint32(Ctx);
1263d3d84fddSDerek Schuff     switch (Ex.Kind) {
1264d3d84fddSDerek Schuff     case wasm::WASM_EXTERNAL_FUNCTION:
1265881d8778SSam Clegg 
1266881d8778SSam Clegg       if (!isDefinedFunctionIndex(Ex.Index))
126784c6f325SSam Clegg         return make_error<GenericBinaryError>("invalid function export",
1268a3b9fe6aSSam Clegg                                               object_error::parse_failed);
1269881d8778SSam Clegg       getDefinedFunction(Ex.Index).ExportName = Ex.Name;
1270d3d84fddSDerek Schuff       break;
12716c899ba6SSam Clegg     case wasm::WASM_EXTERNAL_GLOBAL:
12726c899ba6SSam Clegg       if (!isValidGlobalIndex(Ex.Index))
127384c6f325SSam Clegg         return make_error<GenericBinaryError>("invalid global export",
1274a3b9fe6aSSam Clegg                                               object_error::parse_failed);
1275d3d84fddSDerek Schuff       break;
12761d891d44SHeejin Ahn     case wasm::WASM_EXTERNAL_TAG:
12771d891d44SHeejin Ahn       if (!isValidTagIndex(Ex.Index))
12781d891d44SHeejin Ahn         return make_error<GenericBinaryError>("invalid tag export",
1279da419bdbSHeejin Ahn                                               object_error::parse_failed);
1280da419bdbSHeejin Ahn       break;
12812ffff5afSSam Clegg     case wasm::WASM_EXTERNAL_MEMORY:
12822ffff5afSSam Clegg     case wasm::WASM_EXTERNAL_TABLE:
12832ffff5afSSam Clegg       break;
1284d3d84fddSDerek Schuff     default:
128584c6f325SSam Clegg       return make_error<GenericBinaryError>("unexpected export kind",
1286f208f631SHeejin Ahn                                             object_error::parse_failed);
1287d3d84fddSDerek Schuff     }
1288b7787fd0SSam Clegg     Exports.push_back(Ex);
1289d3d84fddSDerek Schuff   }
1290b7c62394SSam Clegg   if (Ctx.Ptr != Ctx.End)
129184c6f325SSam Clegg     return make_error<GenericBinaryError>("export section ended prematurely",
1292d3d84fddSDerek Schuff                                           object_error::parse_failed);
1293d3d84fddSDerek Schuff   return Error::success();
1294d3d84fddSDerek Schuff }
1295d3d84fddSDerek Schuff 
12964273998cSSam Clegg bool WasmObjectFile::isValidFunctionIndex(uint32_t Index) const {
1297c0039de2SSam Clegg   return Index < NumImportedFunctions + Functions.size();
12984273998cSSam Clegg }
12994273998cSSam Clegg 
130023012e98SSam Clegg bool WasmObjectFile::isDefinedFunctionIndex(uint32_t Index) const {
130123012e98SSam Clegg   return Index >= NumImportedFunctions && isValidFunctionIndex(Index);
130223012e98SSam Clegg }
130323012e98SSam Clegg 
13046c899ba6SSam Clegg bool WasmObjectFile::isValidGlobalIndex(uint32_t Index) const {
13056c899ba6SSam Clegg   return Index < NumImportedGlobals + Globals.size();
13066c899ba6SSam Clegg }
13076c899ba6SSam Clegg 
1308a5a3659dSAndy Wingo bool WasmObjectFile::isValidTableNumber(uint32_t Index) const {
1309388fb67bSPaulo Matos   return Index < NumImportedTables + Tables.size();
1310388fb67bSPaulo Matos }
1311388fb67bSPaulo Matos 
13126c899ba6SSam Clegg bool WasmObjectFile::isDefinedGlobalIndex(uint32_t Index) const {
13136c899ba6SSam Clegg   return Index >= NumImportedGlobals && isValidGlobalIndex(Index);
13146c899ba6SSam Clegg }
13156c899ba6SSam Clegg 
1316a5a3659dSAndy Wingo bool WasmObjectFile::isDefinedTableNumber(uint32_t Index) const {
1317a5a3659dSAndy Wingo   return Index >= NumImportedTables && isValidTableNumber(Index);
1318388fb67bSPaulo Matos }
1319388fb67bSPaulo Matos 
13201d891d44SHeejin Ahn bool WasmObjectFile::isValidTagIndex(uint32_t Index) const {
13211d891d44SHeejin Ahn   return Index < NumImportedTags + Tags.size();
1322da419bdbSHeejin Ahn }
1323da419bdbSHeejin Ahn 
13241d891d44SHeejin Ahn bool WasmObjectFile::isDefinedTagIndex(uint32_t Index) const {
13251d891d44SHeejin Ahn   return Index >= NumImportedTags && isValidTagIndex(Index);
1326da419bdbSHeejin Ahn }
1327da419bdbSHeejin Ahn 
1328b3748f71SNicholas Wilson bool WasmObjectFile::isValidFunctionSymbol(uint32_t Index) const {
13296c899ba6SSam Clegg   return Index < Symbols.size() && Symbols[Index].isTypeFunction();
13306c899ba6SSam Clegg }
13316c899ba6SSam Clegg 
133269e2797eSPaulo Matos bool WasmObjectFile::isValidTableSymbol(uint32_t Index) const {
133369e2797eSPaulo Matos   return Index < Symbols.size() && Symbols[Index].isTypeTable();
133469e2797eSPaulo Matos }
133569e2797eSPaulo Matos 
1336b3748f71SNicholas Wilson bool WasmObjectFile::isValidGlobalSymbol(uint32_t Index) const {
1337b3748f71SNicholas Wilson   return Index < Symbols.size() && Symbols[Index].isTypeGlobal();
1338b3748f71SNicholas Wilson }
1339b3748f71SNicholas Wilson 
13401d891d44SHeejin Ahn bool WasmObjectFile::isValidTagSymbol(uint32_t Index) const {
13411d891d44SHeejin Ahn   return Index < Symbols.size() && Symbols[Index].isTypeTag();
1342da419bdbSHeejin Ahn }
1343da419bdbSHeejin Ahn 
1344b3748f71SNicholas Wilson bool WasmObjectFile::isValidDataSymbol(uint32_t Index) const {
1345b3748f71SNicholas Wilson   return Index < Symbols.size() && Symbols[Index].isTypeData();
1346b3748f71SNicholas Wilson }
1347b3748f71SNicholas Wilson 
13486a31a0d6SSam Clegg bool WasmObjectFile::isValidSectionSymbol(uint32_t Index) const {
13496a31a0d6SSam Clegg   return Index < Symbols.size() && Symbols[Index].isTypeSection();
13506a31a0d6SSam Clegg }
13516a31a0d6SSam Clegg 
135223012e98SSam Clegg wasm::WasmFunction &WasmObjectFile::getDefinedFunction(uint32_t Index) {
135323012e98SSam Clegg   assert(isDefinedFunctionIndex(Index));
135423012e98SSam Clegg   return Functions[Index - NumImportedFunctions];
135523012e98SSam Clegg }
135623012e98SSam Clegg 
1357f3b762a0SWouter van Oortmerssen const wasm::WasmFunction &
1358f3b762a0SWouter van Oortmerssen WasmObjectFile::getDefinedFunction(uint32_t Index) const {
1359f3b762a0SWouter van Oortmerssen   assert(isDefinedFunctionIndex(Index));
1360f3b762a0SWouter van Oortmerssen   return Functions[Index - NumImportedFunctions];
1361f3b762a0SWouter van Oortmerssen }
1362f3b762a0SWouter van Oortmerssen 
13636c899ba6SSam Clegg wasm::WasmGlobal &WasmObjectFile::getDefinedGlobal(uint32_t Index) {
13646c899ba6SSam Clegg   assert(isDefinedGlobalIndex(Index));
13656c899ba6SSam Clegg   return Globals[Index - NumImportedGlobals];
13666c899ba6SSam Clegg }
13676c899ba6SSam Clegg 
13681d891d44SHeejin Ahn wasm::WasmTag &WasmObjectFile::getDefinedTag(uint32_t Index) {
13691d891d44SHeejin Ahn   assert(isDefinedTagIndex(Index));
13701d891d44SHeejin Ahn   return Tags[Index - NumImportedTags];
1371da419bdbSHeejin Ahn }
1372da419bdbSHeejin Ahn 
1373b7c62394SSam Clegg Error WasmObjectFile::parseStartSection(ReadContext &Ctx) {
1374b7c62394SSam Clegg   StartFunction = readVaruint32(Ctx);
13754273998cSSam Clegg   if (!isValidFunctionIndex(StartFunction))
137684c6f325SSam Clegg     return make_error<GenericBinaryError>("invalid start function",
1377d3d84fddSDerek Schuff                                           object_error::parse_failed);
1378d3d84fddSDerek Schuff   return Error::success();
1379d3d84fddSDerek Schuff }
1380d3d84fddSDerek Schuff 
1381b7c62394SSam Clegg Error WasmObjectFile::parseCodeSection(ReadContext &Ctx) {
1382881d8778SSam Clegg   SeenCodeSection = true;
13830b55ccf6SSam Clegg   CodeSection = Sections.size();
1384b7c62394SSam Clegg   uint32_t FunctionCount = readVaruint32(Ctx);
1385c0039de2SSam Clegg   if (FunctionCount != Functions.size()) {
138684c6f325SSam Clegg     return make_error<GenericBinaryError>("invalid function count",
1387d3d84fddSDerek Schuff                                           object_error::parse_failed);
1388d3d84fddSDerek Schuff   }
1389d3d84fddSDerek Schuff 
1390881d8778SSam Clegg   for (uint32_t i = 0; i < FunctionCount; i++) {
1391881d8778SSam Clegg     wasm::WasmFunction& Function = Functions[i];
1392b7c62394SSam Clegg     const uint8_t *FunctionStart = Ctx.Ptr;
1393b7c62394SSam Clegg     uint32_t Size = readVaruint32(Ctx);
1394b7c62394SSam Clegg     const uint8_t *FunctionEnd = Ctx.Ptr + Size;
1395c551522dSSam Clegg 
1396b7c62394SSam Clegg     Function.CodeOffset = Ctx.Ptr - FunctionStart;
1397881d8778SSam Clegg     Function.Index = NumImportedFunctions + i;
1398b7c62394SSam Clegg     Function.CodeSectionOffset = FunctionStart - Ctx.Start;
1399c551522dSSam Clegg     Function.Size = FunctionEnd - FunctionStart;
1400d3d84fddSDerek Schuff 
1401b7c62394SSam Clegg     uint32_t NumLocalDecls = readVaruint32(Ctx);
1402d3d84fddSDerek Schuff     Function.Locals.reserve(NumLocalDecls);
1403d3d84fddSDerek Schuff     while (NumLocalDecls--) {
1404d3d84fddSDerek Schuff       wasm::WasmLocalDecl Decl;
1405b7c62394SSam Clegg       Decl.Count = readVaruint32(Ctx);
1406b7c62394SSam Clegg       Decl.Type = readUint8(Ctx);
1407d3d84fddSDerek Schuff       Function.Locals.push_back(Decl);
1408d3d84fddSDerek Schuff     }
1409d3d84fddSDerek Schuff 
1410b7c62394SSam Clegg     uint32_t BodySize = FunctionEnd - Ctx.Ptr;
1411b7c62394SSam Clegg     Function.Body = ArrayRef<uint8_t>(Ctx.Ptr, BodySize);
1412027b9357SNicholas Wilson     // This will be set later when reading in the linking metadata section.
1413027b9357SNicholas Wilson     Function.Comdat = UINT32_MAX;
1414b7c62394SSam Clegg     Ctx.Ptr += BodySize;
1415b7c62394SSam Clegg     assert(Ctx.Ptr == FunctionEnd);
1416d3d84fddSDerek Schuff   }
1417b7c62394SSam Clegg   if (Ctx.Ptr != Ctx.End)
141884c6f325SSam Clegg     return make_error<GenericBinaryError>("code section ended prematurely",
1419d3d84fddSDerek Schuff                                           object_error::parse_failed);
1420d3d84fddSDerek Schuff   return Error::success();
1421d3d84fddSDerek Schuff }
1422d3d84fddSDerek Schuff 
1423b7c62394SSam Clegg Error WasmObjectFile::parseElemSection(ReadContext &Ctx) {
1424b7c62394SSam Clegg   uint32_t Count = readVaruint32(Ctx);
1425d3d84fddSDerek Schuff   ElemSegments.reserve(Count);
1426d3d84fddSDerek Schuff   while (Count--) {
1427d3d84fddSDerek Schuff     wasm::WasmElemSegment Segment;
1428a5a3659dSAndy Wingo     Segment.Flags = readVaruint32(Ctx);
1429a5a3659dSAndy Wingo 
1430a5a3659dSAndy Wingo     uint32_t SupportedFlags = wasm::WASM_ELEM_SEGMENT_HAS_TABLE_NUMBER |
1431a5a3659dSAndy Wingo                               wasm::WASM_ELEM_SEGMENT_IS_PASSIVE |
1432a5a3659dSAndy Wingo                               wasm::WASM_ELEM_SEGMENT_HAS_INIT_EXPRS;
1433a5a3659dSAndy Wingo     if (Segment.Flags & ~SupportedFlags)
1434a5a3659dSAndy Wingo       return make_error<GenericBinaryError>(
1435a5a3659dSAndy Wingo           "Unsupported flags for element segment", object_error::parse_failed);
1436a5a3659dSAndy Wingo 
1437a5a3659dSAndy Wingo     if (Segment.Flags & wasm::WASM_ELEM_SEGMENT_HAS_TABLE_NUMBER)
1438a5a3659dSAndy Wingo       Segment.TableNumber = readVaruint32(Ctx);
1439a5a3659dSAndy Wingo     else
1440a5a3659dSAndy Wingo       Segment.TableNumber = 0;
1441a5a3659dSAndy Wingo     if (!isValidTableNumber(Segment.TableNumber))
1442a5a3659dSAndy Wingo       return make_error<GenericBinaryError>("invalid TableNumber",
1443d3d84fddSDerek Schuff                                             object_error::parse_failed);
1444a5a3659dSAndy Wingo 
1445a5a3659dSAndy Wingo     if (Segment.Flags & wasm::WASM_ELEM_SEGMENT_IS_PASSIVE) {
1446a5a3659dSAndy Wingo       Segment.Offset.Opcode = wasm::WASM_OPCODE_I32_CONST;
1447a5a3659dSAndy Wingo       Segment.Offset.Value.Int32 = 0;
1448a5a3659dSAndy Wingo     } else {
1449b7c62394SSam Clegg       if (Error Err = readInitExpr(Segment.Offset, Ctx))
1450d3d84fddSDerek Schuff         return Err;
1451a5a3659dSAndy Wingo     }
1452a5a3659dSAndy Wingo 
1453a5a3659dSAndy Wingo     if (Segment.Flags & wasm::WASM_ELEM_SEGMENT_MASK_HAS_ELEM_KIND) {
1454a5a3659dSAndy Wingo       Segment.ElemKind = readUint8(Ctx);
1455a5a3659dSAndy Wingo       if (Segment.Flags & wasm::WASM_ELEM_SEGMENT_HAS_INIT_EXPRS) {
1456a5a3659dSAndy Wingo         if (Segment.ElemKind != uint8_t(wasm::ValType::FUNCREF) &&
1457a5a3659dSAndy Wingo             Segment.ElemKind != uint8_t(wasm::ValType::EXTERNREF)) {
1458a5a3659dSAndy Wingo           return make_error<GenericBinaryError>("invalid reference type",
1459a5a3659dSAndy Wingo                                                 object_error::parse_failed);
1460a5a3659dSAndy Wingo         }
1461a5a3659dSAndy Wingo       } else {
1462a5a3659dSAndy Wingo         if (Segment.ElemKind != 0)
1463a5a3659dSAndy Wingo           return make_error<GenericBinaryError>("invalid elemtype",
1464a5a3659dSAndy Wingo                                                 object_error::parse_failed);
1465a5a3659dSAndy Wingo         Segment.ElemKind = uint8_t(wasm::ValType::FUNCREF);
1466a5a3659dSAndy Wingo       }
1467a5a3659dSAndy Wingo     } else {
1468a5a3659dSAndy Wingo       Segment.ElemKind = uint8_t(wasm::ValType::FUNCREF);
1469a5a3659dSAndy Wingo     }
1470a5a3659dSAndy Wingo 
1471a5a3659dSAndy Wingo     if (Segment.Flags & wasm::WASM_ELEM_SEGMENT_HAS_INIT_EXPRS)
1472a5a3659dSAndy Wingo       return make_error<GenericBinaryError>(
1473a5a3659dSAndy Wingo           "elem segment init expressions not yet implemented",
1474a5a3659dSAndy Wingo           object_error::parse_failed);
1475a5a3659dSAndy Wingo 
1476b7c62394SSam Clegg     uint32_t NumElems = readVaruint32(Ctx);
1477d3d84fddSDerek Schuff     while (NumElems--) {
1478b7c62394SSam Clegg       Segment.Functions.push_back(readVaruint32(Ctx));
1479d3d84fddSDerek Schuff     }
1480d3d84fddSDerek Schuff     ElemSegments.push_back(Segment);
1481d3d84fddSDerek Schuff   }
1482b7c62394SSam Clegg   if (Ctx.Ptr != Ctx.End)
148384c6f325SSam Clegg     return make_error<GenericBinaryError>("elem section ended prematurely",
1484d3d84fddSDerek Schuff                                           object_error::parse_failed);
1485d3d84fddSDerek Schuff   return Error::success();
1486d3d84fddSDerek Schuff }
1487d3d84fddSDerek Schuff 
1488b7c62394SSam Clegg Error WasmObjectFile::parseDataSection(ReadContext &Ctx) {
14890b55ccf6SSam Clegg   DataSection = Sections.size();
1490b7c62394SSam Clegg   uint32_t Count = readVaruint32(Ctx);
1491fef8de66SThomas Lively   if (DataCount && Count != DataCount.getValue())
1492fef8de66SThomas Lively     return make_error<GenericBinaryError>(
149384c6f325SSam Clegg         "number of data segments does not match DataCount section");
1494d3d84fddSDerek Schuff   DataSegments.reserve(Count);
1495d3d84fddSDerek Schuff   while (Count--) {
14969c07f94aSSam Clegg     WasmSegment Segment;
14972e150409SThomas Lively     Segment.Data.InitFlags = readVaruint32(Ctx);
1498831a143eSAndy Wingo     Segment.Data.MemoryIndex =
1499831a143eSAndy Wingo         (Segment.Data.InitFlags & wasm::WASM_DATA_SEGMENT_HAS_MEMINDEX)
1500831a143eSAndy Wingo             ? readVaruint32(Ctx)
1501831a143eSAndy Wingo             : 0;
1502831a143eSAndy Wingo     if ((Segment.Data.InitFlags & wasm::WASM_DATA_SEGMENT_IS_PASSIVE) == 0) {
1503b7c62394SSam Clegg       if (Error Err = readInitExpr(Segment.Data.Offset, Ctx))
1504d3d84fddSDerek Schuff         return Err;
15052e150409SThomas Lively     } else {
15062e150409SThomas Lively       Segment.Data.Offset.Opcode = wasm::WASM_OPCODE_I32_CONST;
15072e150409SThomas Lively       Segment.Data.Offset.Value.Int32 = 0;
15082e150409SThomas Lively     }
1509b7c62394SSam Clegg     uint32_t Size = readVaruint32(Ctx);
1510dcf5bd27SSimon Pilgrim     if (Size > (size_t)(Ctx.End - Ctx.Ptr))
151184c6f325SSam Clegg       return make_error<GenericBinaryError>("invalid segment size",
15124bbc6b55SSam Clegg                                             object_error::parse_failed);
1513b7c62394SSam Clegg     Segment.Data.Content = ArrayRef<uint8_t>(Ctx.Ptr, Size);
1514027b9357SNicholas Wilson     // The rest of these Data fields are set later, when reading in the linking
1515027b9357SNicholas Wilson     // metadata section.
151663ebb813SSam Clegg     Segment.Data.Alignment = 0;
15173b8d2be5SSam Clegg     Segment.Data.LinkingFlags = 0;
1518027b9357SNicholas Wilson     Segment.Data.Comdat = UINT32_MAX;
1519b7c62394SSam Clegg     Segment.SectionOffset = Ctx.Ptr - Ctx.Start;
1520b7c62394SSam Clegg     Ctx.Ptr += Size;
1521d3d84fddSDerek Schuff     DataSegments.push_back(Segment);
1522d3d84fddSDerek Schuff   }
1523b7c62394SSam Clegg   if (Ctx.Ptr != Ctx.End)
152484c6f325SSam Clegg     return make_error<GenericBinaryError>("data section ended prematurely",
1525d3d84fddSDerek Schuff                                           object_error::parse_failed);
15262c6f75ddSDerek Schuff   return Error::success();
15272c6f75ddSDerek Schuff }
15282c6f75ddSDerek Schuff 
1529fef8de66SThomas Lively Error WasmObjectFile::parseDataCountSection(ReadContext &Ctx) {
1530fef8de66SThomas Lively   DataCount = readVaruint32(Ctx);
1531fef8de66SThomas Lively   return Error::success();
1532fef8de66SThomas Lively }
1533fef8de66SThomas Lively 
15342c6f75ddSDerek Schuff const wasm::WasmObjectHeader &WasmObjectFile::getHeader() const {
15352c6f75ddSDerek Schuff   return Header;
15362c6f75ddSDerek Schuff }
15372c6f75ddSDerek Schuff 
15382a193e0dSSam Clegg void WasmObjectFile::moveSymbolNext(DataRefImpl &Symb) const { Symb.d.b++; }
15392c6f75ddSDerek Schuff 
1540ac00376aSvgxbj Expected<uint32_t> WasmObjectFile::getSymbolFlags(DataRefImpl Symb) const {
1541fc5b5cd2SSam Clegg   uint32_t Result = SymbolRef::SF_None;
1542d3d84fddSDerek Schuff   const WasmSymbol &Sym = getWasmSymbol(Symb);
1543fc5b5cd2SSam Clegg 
1544d34e60caSNicola Zaghen   LLVM_DEBUG(dbgs() << "getSymbolFlags: ptr=" << &Sym << " " << Sym << "\n");
15456e7f1826SSam Clegg   if (Sym.isBindingWeak())
1546b7787fd0SSam Clegg     Result |= SymbolRef::SF_Weak;
15476e7f1826SSam Clegg   if (!Sym.isBindingLocal())
154831a2c809SSam Clegg     Result |= SymbolRef::SF_Global;
1549a2b35dacSSam Clegg   if (Sym.isHidden())
1550a2b35dacSSam Clegg     Result |= SymbolRef::SF_Hidden;
15516c899ba6SSam Clegg   if (!Sym.isDefined())
1552fc5b5cd2SSam Clegg     Result |= SymbolRef::SF_Undefined;
15536c899ba6SSam Clegg   if (Sym.isTypeFunction())
15546c899ba6SSam Clegg     Result |= SymbolRef::SF_Executable;
1555fc5b5cd2SSam Clegg   return Result;
15562c6f75ddSDerek Schuff }
15572c6f75ddSDerek Schuff 
15582c6f75ddSDerek Schuff basic_symbol_iterator WasmObjectFile::symbol_begin() const {
1559d3d84fddSDerek Schuff   DataRefImpl Ref;
15602a193e0dSSam Clegg   Ref.d.a = 1; // Arbitrary non-zero value so that Ref.p is non-null
15612a193e0dSSam Clegg   Ref.d.b = 0; // Symbol index
1562d3d84fddSDerek Schuff   return BasicSymbolRef(Ref, this);
15632c6f75ddSDerek Schuff }
15642c6f75ddSDerek Schuff 
15652c6f75ddSDerek Schuff basic_symbol_iterator WasmObjectFile::symbol_end() const {
1566d3d84fddSDerek Schuff   DataRefImpl Ref;
15672a193e0dSSam Clegg   Ref.d.a = 1; // Arbitrary non-zero value so that Ref.p is non-null
15682a193e0dSSam Clegg   Ref.d.b = Symbols.size(); // Symbol index
1569d3d84fddSDerek Schuff   return BasicSymbolRef(Ref, this);
1570d3d84fddSDerek Schuff }
1571d3d84fddSDerek Schuff 
15722ffff5afSSam Clegg const WasmSymbol &WasmObjectFile::getWasmSymbol(const DataRefImpl &Symb) const {
15732a193e0dSSam Clegg   return Symbols[Symb.d.b];
15742c6f75ddSDerek Schuff }
15752c6f75ddSDerek Schuff 
15762ffff5afSSam Clegg const WasmSymbol &WasmObjectFile::getWasmSymbol(const SymbolRef &Symb) const {
15772ffff5afSSam Clegg   return getWasmSymbol(Symb.getRawDataRefImpl());
15782ffff5afSSam Clegg }
15792ffff5afSSam Clegg 
15802c6f75ddSDerek Schuff Expected<StringRef> WasmObjectFile::getSymbolName(DataRefImpl Symb) const {
15816c899ba6SSam Clegg   return getWasmSymbol(Symb).Info.Name;
15822c6f75ddSDerek Schuff }
15832c6f75ddSDerek Schuff 
15842c6f75ddSDerek Schuff Expected<uint64_t> WasmObjectFile::getSymbolAddress(DataRefImpl Symb) const {
1585f3b762a0SWouter van Oortmerssen   auto &Sym = getWasmSymbol(Symb);
1586f3b762a0SWouter van Oortmerssen   if (Sym.Info.Kind == wasm::WASM_SYMBOL_TYPE_FUNCTION &&
1587f3b762a0SWouter van Oortmerssen       isDefinedFunctionIndex(Sym.Info.ElementIndex))
1588f3b762a0SWouter van Oortmerssen     return getDefinedFunction(Sym.Info.ElementIndex).CodeSectionOffset;
1589f3b762a0SWouter van Oortmerssen   else
1590fc5b5cd2SSam Clegg     return getSymbolValue(Symb);
15912c6f75ddSDerek Schuff }
15922c6f75ddSDerek Schuff 
159331a2c809SSam Clegg uint64_t WasmObjectFile::getWasmSymbolValue(const WasmSymbol &Sym) const {
15946c899ba6SSam Clegg   switch (Sym.Info.Kind) {
15956c899ba6SSam Clegg   case wasm::WASM_SYMBOL_TYPE_FUNCTION:
15966c899ba6SSam Clegg   case wasm::WASM_SYMBOL_TYPE_GLOBAL:
15971d891d44SHeejin Ahn   case wasm::WASM_SYMBOL_TYPE_TAG:
1598feac819eSAndy Wingo   case wasm::WASM_SYMBOL_TYPE_TABLE:
15996c899ba6SSam Clegg     return Sym.Info.ElementIndex;
16006c899ba6SSam Clegg   case wasm::WASM_SYMBOL_TYPE_DATA: {
16016c899ba6SSam Clegg     // The value of a data symbol is the segment offset, plus the symbol
16026c899ba6SSam Clegg     // offset within the segment.
16036c899ba6SSam Clegg     uint32_t SegmentIndex = Sym.Info.DataRef.Segment;
16046c899ba6SSam Clegg     const wasm::WasmDataSegment &Segment = DataSegments[SegmentIndex].Data;
16053b29376eSWouter van Oortmerssen     if (Segment.Offset.Opcode == wasm::WASM_OPCODE_I32_CONST) {
16066c899ba6SSam Clegg       return Segment.Offset.Value.Int32 + Sym.Info.DataRef.Offset;
16073b29376eSWouter van Oortmerssen     } else if (Segment.Offset.Opcode == wasm::WASM_OPCODE_I64_CONST) {
16083b29376eSWouter van Oortmerssen       return Segment.Offset.Value.Int64 + Sym.Info.DataRef.Offset;
16093b29376eSWouter van Oortmerssen     } else {
16103b29376eSWouter van Oortmerssen       llvm_unreachable("unknown init expr opcode");
16113b29376eSWouter van Oortmerssen     }
1612b09cfa51SSam Clegg   }
16136a31a0d6SSam Clegg   case wasm::WASM_SYMBOL_TYPE_SECTION:
16146a31a0d6SSam Clegg     return 0;
16152c6f75ddSDerek Schuff   }
16167f055deeSSam Clegg   llvm_unreachable("invalid symbol type");
1617b7787fd0SSam Clegg }
16182c6f75ddSDerek Schuff 
161931a2c809SSam Clegg uint64_t WasmObjectFile::getSymbolValueImpl(DataRefImpl Symb) const {
162031a2c809SSam Clegg   return getWasmSymbolValue(getWasmSymbol(Symb));
162131a2c809SSam Clegg }
162231a2c809SSam Clegg 
16232c6f75ddSDerek Schuff uint32_t WasmObjectFile::getSymbolAlignment(DataRefImpl Symb) const {
16242c6f75ddSDerek Schuff   llvm_unreachable("not yet implemented");
16252c6f75ddSDerek Schuff   return 0;
16262c6f75ddSDerek Schuff }
16272c6f75ddSDerek Schuff 
16282c6f75ddSDerek Schuff uint64_t WasmObjectFile::getCommonSymbolSizeImpl(DataRefImpl Symb) const {
16292c6f75ddSDerek Schuff   llvm_unreachable("not yet implemented");
16302c6f75ddSDerek Schuff   return 0;
16312c6f75ddSDerek Schuff }
16322c6f75ddSDerek Schuff 
16332c6f75ddSDerek Schuff Expected<SymbolRef::Type>
16342c6f75ddSDerek Schuff WasmObjectFile::getSymbolType(DataRefImpl Symb) const {
1635fc5b5cd2SSam Clegg   const WasmSymbol &Sym = getWasmSymbol(Symb);
1636fc5b5cd2SSam Clegg 
16376c899ba6SSam Clegg   switch (Sym.Info.Kind) {
16386c899ba6SSam Clegg   case wasm::WASM_SYMBOL_TYPE_FUNCTION:
1639fc5b5cd2SSam Clegg     return SymbolRef::ST_Function;
16406c899ba6SSam Clegg   case wasm::WASM_SYMBOL_TYPE_GLOBAL:
16416c899ba6SSam Clegg     return SymbolRef::ST_Other;
16426c899ba6SSam Clegg   case wasm::WASM_SYMBOL_TYPE_DATA:
1643fc5b5cd2SSam Clegg     return SymbolRef::ST_Data;
16446a31a0d6SSam Clegg   case wasm::WASM_SYMBOL_TYPE_SECTION:
16456a31a0d6SSam Clegg     return SymbolRef::ST_Debug;
16461d891d44SHeejin Ahn   case wasm::WASM_SYMBOL_TYPE_TAG:
1647da419bdbSHeejin Ahn     return SymbolRef::ST_Other;
1648feac819eSAndy Wingo   case wasm::WASM_SYMBOL_TYPE_TABLE:
1649feac819eSAndy Wingo     return SymbolRef::ST_Other;
1650fc5b5cd2SSam Clegg   }
1651fc5b5cd2SSam Clegg 
165284c6f325SSam Clegg   llvm_unreachable("unknown WasmSymbol::SymbolType");
1653fc5b5cd2SSam Clegg   return SymbolRef::ST_Other;
16542c6f75ddSDerek Schuff }
16552c6f75ddSDerek Schuff 
16562c6f75ddSDerek Schuff Expected<section_iterator>
16572c6f75ddSDerek Schuff WasmObjectFile::getSymbolSection(DataRefImpl Symb) const {
16580b55ccf6SSam Clegg   const WasmSymbol &Sym = getWasmSymbol(Symb);
16596c899ba6SSam Clegg   if (Sym.isUndefined())
16600b55ccf6SSam Clegg     return section_end();
16616c899ba6SSam Clegg 
16626c899ba6SSam Clegg   DataRefImpl Ref;
1663da6a896eSDerek Schuff   Ref.d.a = getSymbolSectionIdImpl(Sym);
1664da6a896eSDerek Schuff   return section_iterator(SectionRef(Ref, this));
1665da6a896eSDerek Schuff }
1666da6a896eSDerek Schuff 
1667da6a896eSDerek Schuff uint32_t WasmObjectFile::getSymbolSectionId(SymbolRef Symb) const {
1668da6a896eSDerek Schuff   const WasmSymbol &Sym = getWasmSymbol(Symb);
1669da6a896eSDerek Schuff   return getSymbolSectionIdImpl(Sym);
1670da6a896eSDerek Schuff }
1671da6a896eSDerek Schuff 
1672da6a896eSDerek Schuff uint32_t WasmObjectFile::getSymbolSectionIdImpl(const WasmSymbol &Sym) const {
16736c899ba6SSam Clegg   switch (Sym.Info.Kind) {
16746c899ba6SSam Clegg   case wasm::WASM_SYMBOL_TYPE_FUNCTION:
1675da6a896eSDerek Schuff     return CodeSection;
16766c899ba6SSam Clegg   case wasm::WASM_SYMBOL_TYPE_GLOBAL:
1677da6a896eSDerek Schuff     return GlobalSection;
16786c899ba6SSam Clegg   case wasm::WASM_SYMBOL_TYPE_DATA:
1679da6a896eSDerek Schuff     return DataSection;
1680da419bdbSHeejin Ahn   case wasm::WASM_SYMBOL_TYPE_SECTION:
1681da6a896eSDerek Schuff     return Sym.Info.ElementIndex;
16821d891d44SHeejin Ahn   case wasm::WASM_SYMBOL_TYPE_TAG:
16831d891d44SHeejin Ahn     return TagSection;
1684feac819eSAndy Wingo   case wasm::WASM_SYMBOL_TYPE_TABLE:
1685feac819eSAndy Wingo     return TableSection;
16866c899ba6SSam Clegg   default:
168784c6f325SSam Clegg     llvm_unreachable("unknown WasmSymbol::SymbolType");
16886c899ba6SSam Clegg   }
16892c6f75ddSDerek Schuff }
16902c6f75ddSDerek Schuff 
16912c6f75ddSDerek Schuff void WasmObjectFile::moveSectionNext(DataRefImpl &Sec) const { Sec.d.a++; }
16922c6f75ddSDerek Schuff 
16938be28cdcSFangrui Song Expected<StringRef> WasmObjectFile::getSectionName(DataRefImpl Sec) const {
1694d3d84fddSDerek Schuff   const WasmSection &S = Sections[Sec.d.a];
16952c6f75ddSDerek Schuff #define ECase(X)                                                               \
16962c6f75ddSDerek Schuff   case wasm::WASM_SEC_##X:                                                     \
16978be28cdcSFangrui Song     return #X;
16982c6f75ddSDerek Schuff   switch (S.Type) {
16992c6f75ddSDerek Schuff     ECase(TYPE);
17002c6f75ddSDerek Schuff     ECase(IMPORT);
17012c6f75ddSDerek Schuff     ECase(FUNCTION);
17022c6f75ddSDerek Schuff     ECase(TABLE);
17032c6f75ddSDerek Schuff     ECase(MEMORY);
17042c6f75ddSDerek Schuff     ECase(GLOBAL);
17051d891d44SHeejin Ahn     ECase(TAG);
17062c6f75ddSDerek Schuff     ECase(EXPORT);
17072c6f75ddSDerek Schuff     ECase(START);
17082c6f75ddSDerek Schuff     ECase(ELEM);
17092c6f75ddSDerek Schuff     ECase(CODE);
17102c6f75ddSDerek Schuff     ECase(DATA);
1711fef8de66SThomas Lively     ECase(DATACOUNT);
17126d76b7b4SDerek Schuff   case wasm::WASM_SEC_CUSTOM:
17138be28cdcSFangrui Song     return S.Name;
17142c6f75ddSDerek Schuff   default:
17158be28cdcSFangrui Song     return createStringError(object_error::invalid_section_index, "");
17162c6f75ddSDerek Schuff   }
17172c6f75ddSDerek Schuff #undef ECase
17182c6f75ddSDerek Schuff }
17192c6f75ddSDerek Schuff 
17202c6f75ddSDerek Schuff uint64_t WasmObjectFile::getSectionAddress(DataRefImpl Sec) const { return 0; }
17212c6f75ddSDerek Schuff 
1722a25d329bSGeorge Rimar uint64_t WasmObjectFile::getSectionIndex(DataRefImpl Sec) const {
1723a25d329bSGeorge Rimar   return Sec.d.a;
1724a25d329bSGeorge Rimar }
1725a25d329bSGeorge Rimar 
17262c6f75ddSDerek Schuff uint64_t WasmObjectFile::getSectionSize(DataRefImpl Sec) const {
1727d3d84fddSDerek Schuff   const WasmSection &S = Sections[Sec.d.a];
17282c6f75ddSDerek Schuff   return S.Content.size();
17292c6f75ddSDerek Schuff }
17302c6f75ddSDerek Schuff 
1731e1cb2c0fSFangrui Song Expected<ArrayRef<uint8_t>>
1732e1cb2c0fSFangrui Song WasmObjectFile::getSectionContents(DataRefImpl Sec) const {
1733d3d84fddSDerek Schuff   const WasmSection &S = Sections[Sec.d.a];
17342c6f75ddSDerek Schuff   // This will never fail since wasm sections can never be empty (user-sections
17352c6f75ddSDerek Schuff   // must have a name and non-user sections each have a defined structure).
1736e1cb2c0fSFangrui Song   return S.Content;
17372c6f75ddSDerek Schuff }
17382c6f75ddSDerek Schuff 
17392c6f75ddSDerek Schuff uint64_t WasmObjectFile::getSectionAlignment(DataRefImpl Sec) const {
17402c6f75ddSDerek Schuff   return 1;
17412c6f75ddSDerek Schuff }
17422c6f75ddSDerek Schuff 
17432c6f75ddSDerek Schuff bool WasmObjectFile::isSectionCompressed(DataRefImpl Sec) const {
17442c6f75ddSDerek Schuff   return false;
17452c6f75ddSDerek Schuff }
17462c6f75ddSDerek Schuff 
17472c6f75ddSDerek Schuff bool WasmObjectFile::isSectionText(DataRefImpl Sec) const {
1748d3d84fddSDerek Schuff   return getWasmSection(Sec).Type == wasm::WASM_SEC_CODE;
17492c6f75ddSDerek Schuff }
17502c6f75ddSDerek Schuff 
17512c6f75ddSDerek Schuff bool WasmObjectFile::isSectionData(DataRefImpl Sec) const {
1752d3d84fddSDerek Schuff   return getWasmSection(Sec).Type == wasm::WASM_SEC_DATA;
17532c6f75ddSDerek Schuff }
17542c6f75ddSDerek Schuff 
17552c6f75ddSDerek Schuff bool WasmObjectFile::isSectionBSS(DataRefImpl Sec) const { return false; }
17562c6f75ddSDerek Schuff 
17572c6f75ddSDerek Schuff bool WasmObjectFile::isSectionVirtual(DataRefImpl Sec) const { return false; }
17582c6f75ddSDerek Schuff 
1759d3d84fddSDerek Schuff relocation_iterator WasmObjectFile::section_rel_begin(DataRefImpl Ref) const {
1760d3d84fddSDerek Schuff   DataRefImpl RelocRef;
1761d3d84fddSDerek Schuff   RelocRef.d.a = Ref.d.a;
1762d3d84fddSDerek Schuff   RelocRef.d.b = 0;
1763d3d84fddSDerek Schuff   return relocation_iterator(RelocationRef(RelocRef, this));
17642c6f75ddSDerek Schuff }
17652c6f75ddSDerek Schuff 
1766d3d84fddSDerek Schuff relocation_iterator WasmObjectFile::section_rel_end(DataRefImpl Ref) const {
1767d3d84fddSDerek Schuff   const WasmSection &Sec = getWasmSection(Ref);
1768d3d84fddSDerek Schuff   DataRefImpl RelocRef;
1769d3d84fddSDerek Schuff   RelocRef.d.a = Ref.d.a;
1770d3d84fddSDerek Schuff   RelocRef.d.b = Sec.Relocations.size();
1771d3d84fddSDerek Schuff   return relocation_iterator(RelocationRef(RelocRef, this));
17722c6f75ddSDerek Schuff }
17732c6f75ddSDerek Schuff 
1774f208f631SHeejin Ahn void WasmObjectFile::moveRelocationNext(DataRefImpl &Rel) const { Rel.d.b++; }
17752c6f75ddSDerek Schuff 
1776d3d84fddSDerek Schuff uint64_t WasmObjectFile::getRelocationOffset(DataRefImpl Ref) const {
1777d3d84fddSDerek Schuff   const wasm::WasmRelocation &Rel = getWasmRelocation(Ref);
1778d3d84fddSDerek Schuff   return Rel.Offset;
17792c6f75ddSDerek Schuff }
17802c6f75ddSDerek Schuff 
1781f676cdd5SSam Clegg symbol_iterator WasmObjectFile::getRelocationSymbol(DataRefImpl Ref) const {
1782f676cdd5SSam Clegg   const wasm::WasmRelocation &Rel = getWasmRelocation(Ref);
1783d1152a26SSam Clegg   if (Rel.Type == wasm::R_WASM_TYPE_INDEX_LEB)
1784f676cdd5SSam Clegg     return symbol_end();
1785f676cdd5SSam Clegg   DataRefImpl Sym;
1786c7d2e5f1SSam Clegg   Sym.d.a = 1;
1787c7d2e5f1SSam Clegg   Sym.d.b = Rel.Index;
1788f676cdd5SSam Clegg   return symbol_iterator(SymbolRef(Sym, this));
17892c6f75ddSDerek Schuff }
17902c6f75ddSDerek Schuff 
1791d3d84fddSDerek Schuff uint64_t WasmObjectFile::getRelocationType(DataRefImpl Ref) const {
1792d3d84fddSDerek Schuff   const wasm::WasmRelocation &Rel = getWasmRelocation(Ref);
1793d3d84fddSDerek Schuff   return Rel.Type;
17942c6f75ddSDerek Schuff }
17952c6f75ddSDerek Schuff 
17962c6f75ddSDerek Schuff void WasmObjectFile::getRelocationTypeName(
1797d3d84fddSDerek Schuff     DataRefImpl Ref, SmallVectorImpl<char> &Result) const {
1798d3d84fddSDerek Schuff   const wasm::WasmRelocation &Rel = getWasmRelocation(Ref);
1799d3d84fddSDerek Schuff   StringRef Res = "Unknown";
1800d3d84fddSDerek Schuff 
1801d3d84fddSDerek Schuff #define WASM_RELOC(name, value)                                                \
1802d3d84fddSDerek Schuff   case wasm::name:                                                             \
1803d3d84fddSDerek Schuff     Res = #name;                                                               \
1804d3d84fddSDerek Schuff     break;
1805d3d84fddSDerek Schuff 
1806d3d84fddSDerek Schuff   switch (Rel.Type) {
1807c5d8bc83SSam Clegg #include "llvm/BinaryFormat/WasmRelocs.def"
1808d3d84fddSDerek Schuff   }
1809d3d84fddSDerek Schuff 
1810d3d84fddSDerek Schuff #undef WASM_RELOC
1811d3d84fddSDerek Schuff 
1812d3d84fddSDerek Schuff   Result.append(Res.begin(), Res.end());
18132c6f75ddSDerek Schuff }
18142c6f75ddSDerek Schuff 
18152c6f75ddSDerek Schuff section_iterator WasmObjectFile::section_begin() const {
18162c6f75ddSDerek Schuff   DataRefImpl Ref;
18172c6f75ddSDerek Schuff   Ref.d.a = 0;
18182c6f75ddSDerek Schuff   return section_iterator(SectionRef(Ref, this));
18192c6f75ddSDerek Schuff }
18202c6f75ddSDerek Schuff 
18212c6f75ddSDerek Schuff section_iterator WasmObjectFile::section_end() const {
18222c6f75ddSDerek Schuff   DataRefImpl Ref;
18232c6f75ddSDerek Schuff   Ref.d.a = Sections.size();
18242c6f75ddSDerek Schuff   return section_iterator(SectionRef(Ref, this));
18252c6f75ddSDerek Schuff }
18262c6f75ddSDerek Schuff 
182729f8c9f6SWouter van Oortmerssen uint8_t WasmObjectFile::getBytesInAddress() const {
182829f8c9f6SWouter van Oortmerssen   return HasMemory64 ? 8 : 4;
182929f8c9f6SWouter van Oortmerssen }
18302c6f75ddSDerek Schuff 
18312c6f75ddSDerek Schuff StringRef WasmObjectFile::getFileFormatName() const { return "WASM"; }
18322c6f75ddSDerek Schuff 
183329f8c9f6SWouter van Oortmerssen Triple::ArchType WasmObjectFile::getArch() const {
183429f8c9f6SWouter van Oortmerssen   return HasMemory64 ? Triple::wasm64 : Triple::wasm32;
183529f8c9f6SWouter van Oortmerssen }
18362c6f75ddSDerek Schuff 
18372c6f75ddSDerek Schuff SubtargetFeatures WasmObjectFile::getFeatures() const {
18382c6f75ddSDerek Schuff   return SubtargetFeatures();
18392c6f75ddSDerek Schuff }
18402c6f75ddSDerek Schuff 
1841f208f631SHeejin Ahn bool WasmObjectFile::isRelocatableObject() const { return HasLinkingSection; }
18422c6f75ddSDerek Schuff 
184303801256SSam Clegg bool WasmObjectFile::isSharedObject() const { return HasDylinkSection; }
184403801256SSam Clegg 
1845d3d84fddSDerek Schuff const WasmSection &WasmObjectFile::getWasmSection(DataRefImpl Ref) const {
184616fe5822SDavide Italiano   assert(Ref.d.a < Sections.size());
1847d3d84fddSDerek Schuff   return Sections[Ref.d.a];
1848d3d84fddSDerek Schuff }
1849d3d84fddSDerek Schuff 
1850d3d84fddSDerek Schuff const WasmSection &
18512c6f75ddSDerek Schuff WasmObjectFile::getWasmSection(const SectionRef &Section) const {
1852d3d84fddSDerek Schuff   return getWasmSection(Section.getRawDataRefImpl());
1853d3d84fddSDerek Schuff }
1854d3d84fddSDerek Schuff 
1855d3d84fddSDerek Schuff const wasm::WasmRelocation &
1856d3d84fddSDerek Schuff WasmObjectFile::getWasmRelocation(const RelocationRef &Ref) const {
1857d3d84fddSDerek Schuff   return getWasmRelocation(Ref.getRawDataRefImpl());
1858d3d84fddSDerek Schuff }
1859d3d84fddSDerek Schuff 
1860d3d84fddSDerek Schuff const wasm::WasmRelocation &
1861d3d84fddSDerek Schuff WasmObjectFile::getWasmRelocation(DataRefImpl Ref) const {
186216fe5822SDavide Italiano   assert(Ref.d.a < Sections.size());
1863d3d84fddSDerek Schuff   const WasmSection &Sec = Sections[Ref.d.a];
186416fe5822SDavide Italiano   assert(Ref.d.b < Sec.Relocations.size());
1865d3d84fddSDerek Schuff   return Sec.Relocations[Ref.d.b];
18662c6f75ddSDerek Schuff }
1867feef720bSHeejin Ahn 
1868feef720bSHeejin Ahn int WasmSectionOrderChecker::getSectionOrder(unsigned ID,
1869feef720bSHeejin Ahn                                              StringRef CustomSectionName) {
1870feef720bSHeejin Ahn   switch (ID) {
1871feef720bSHeejin Ahn   case wasm::WASM_SEC_CUSTOM:
1872feef720bSHeejin Ahn     return StringSwitch<unsigned>(CustomSectionName)
1873feef720bSHeejin Ahn         .Case("dylink", WASM_SEC_ORDER_DYLINK)
1874b78c85a4SSam Clegg         .Case("dylink.0", WASM_SEC_ORDER_DYLINK)
1875feef720bSHeejin Ahn         .Case("linking", WASM_SEC_ORDER_LINKING)
1876feef720bSHeejin Ahn         .StartsWith("reloc.", WASM_SEC_ORDER_RELOC)
1877feef720bSHeejin Ahn         .Case("name", WASM_SEC_ORDER_NAME)
1878feef720bSHeejin Ahn         .Case("producers", WASM_SEC_ORDER_PRODUCERS)
1879f6f4f843SThomas Lively         .Case("target_features", WASM_SEC_ORDER_TARGET_FEATURES)
18809757bba4SThomas Lively         .Default(WASM_SEC_ORDER_NONE);
1881feef720bSHeejin Ahn   case wasm::WASM_SEC_TYPE:
1882feef720bSHeejin Ahn     return WASM_SEC_ORDER_TYPE;
1883feef720bSHeejin Ahn   case wasm::WASM_SEC_IMPORT:
1884feef720bSHeejin Ahn     return WASM_SEC_ORDER_IMPORT;
1885feef720bSHeejin Ahn   case wasm::WASM_SEC_FUNCTION:
1886feef720bSHeejin Ahn     return WASM_SEC_ORDER_FUNCTION;
1887feef720bSHeejin Ahn   case wasm::WASM_SEC_TABLE:
1888feef720bSHeejin Ahn     return WASM_SEC_ORDER_TABLE;
1889feef720bSHeejin Ahn   case wasm::WASM_SEC_MEMORY:
1890feef720bSHeejin Ahn     return WASM_SEC_ORDER_MEMORY;
1891feef720bSHeejin Ahn   case wasm::WASM_SEC_GLOBAL:
1892feef720bSHeejin Ahn     return WASM_SEC_ORDER_GLOBAL;
1893feef720bSHeejin Ahn   case wasm::WASM_SEC_EXPORT:
1894feef720bSHeejin Ahn     return WASM_SEC_ORDER_EXPORT;
1895feef720bSHeejin Ahn   case wasm::WASM_SEC_START:
1896feef720bSHeejin Ahn     return WASM_SEC_ORDER_START;
1897feef720bSHeejin Ahn   case wasm::WASM_SEC_ELEM:
1898feef720bSHeejin Ahn     return WASM_SEC_ORDER_ELEM;
1899feef720bSHeejin Ahn   case wasm::WASM_SEC_CODE:
1900feef720bSHeejin Ahn     return WASM_SEC_ORDER_CODE;
1901feef720bSHeejin Ahn   case wasm::WASM_SEC_DATA:
1902feef720bSHeejin Ahn     return WASM_SEC_ORDER_DATA;
1903feef720bSHeejin Ahn   case wasm::WASM_SEC_DATACOUNT:
1904feef720bSHeejin Ahn     return WASM_SEC_ORDER_DATACOUNT;
19051d891d44SHeejin Ahn   case wasm::WASM_SEC_TAG:
19061d891d44SHeejin Ahn     return WASM_SEC_ORDER_TAG;
1907feef720bSHeejin Ahn   default:
1908fe8aabf9SSam Clegg     return WASM_SEC_ORDER_NONE;
1909feef720bSHeejin Ahn   }
1910feef720bSHeejin Ahn }
1911feef720bSHeejin Ahn 
19129757bba4SThomas Lively // Represents the edges in a directed graph where any node B reachable from node
19139757bba4SThomas Lively // A is not allowed to appear before A in the section ordering, but may appear
19149757bba4SThomas Lively // afterward.
1915f93426c5SHeejin Ahn int WasmSectionOrderChecker::DisallowedPredecessors
1916f93426c5SHeejin Ahn     [WASM_NUM_SEC_ORDERS][WASM_NUM_SEC_ORDERS] = {
1917f93426c5SHeejin Ahn         // WASM_SEC_ORDER_NONE
1918f93426c5SHeejin Ahn         {},
1919f93426c5SHeejin Ahn         // WASM_SEC_ORDER_TYPE
1920f93426c5SHeejin Ahn         {WASM_SEC_ORDER_TYPE, WASM_SEC_ORDER_IMPORT},
1921f93426c5SHeejin Ahn         // WASM_SEC_ORDER_IMPORT
1922f93426c5SHeejin Ahn         {WASM_SEC_ORDER_IMPORT, WASM_SEC_ORDER_FUNCTION},
1923f93426c5SHeejin Ahn         // WASM_SEC_ORDER_FUNCTION
1924f93426c5SHeejin Ahn         {WASM_SEC_ORDER_FUNCTION, WASM_SEC_ORDER_TABLE},
1925f93426c5SHeejin Ahn         // WASM_SEC_ORDER_TABLE
1926f93426c5SHeejin Ahn         {WASM_SEC_ORDER_TABLE, WASM_SEC_ORDER_MEMORY},
1927f93426c5SHeejin Ahn         // WASM_SEC_ORDER_MEMORY
19281d891d44SHeejin Ahn         {WASM_SEC_ORDER_MEMORY, WASM_SEC_ORDER_TAG},
19291d891d44SHeejin Ahn         // WASM_SEC_ORDER_TAG
19301d891d44SHeejin Ahn         {WASM_SEC_ORDER_TAG, WASM_SEC_ORDER_GLOBAL},
1931f93426c5SHeejin Ahn         // WASM_SEC_ORDER_GLOBAL
1932f93426c5SHeejin Ahn         {WASM_SEC_ORDER_GLOBAL, WASM_SEC_ORDER_EXPORT},
1933f93426c5SHeejin Ahn         // WASM_SEC_ORDER_EXPORT
1934f93426c5SHeejin Ahn         {WASM_SEC_ORDER_EXPORT, WASM_SEC_ORDER_START},
1935f93426c5SHeejin Ahn         // WASM_SEC_ORDER_START
1936f93426c5SHeejin Ahn         {WASM_SEC_ORDER_START, WASM_SEC_ORDER_ELEM},
1937f93426c5SHeejin Ahn         // WASM_SEC_ORDER_ELEM
1938f93426c5SHeejin Ahn         {WASM_SEC_ORDER_ELEM, WASM_SEC_ORDER_DATACOUNT},
1939f93426c5SHeejin Ahn         // WASM_SEC_ORDER_DATACOUNT
1940f93426c5SHeejin Ahn         {WASM_SEC_ORDER_DATACOUNT, WASM_SEC_ORDER_CODE},
1941f93426c5SHeejin Ahn         // WASM_SEC_ORDER_CODE
1942f93426c5SHeejin Ahn         {WASM_SEC_ORDER_CODE, WASM_SEC_ORDER_DATA},
1943f93426c5SHeejin Ahn         // WASM_SEC_ORDER_DATA
1944f93426c5SHeejin Ahn         {WASM_SEC_ORDER_DATA, WASM_SEC_ORDER_LINKING},
19459757bba4SThomas Lively 
19469757bba4SThomas Lively         // Custom Sections
1947f93426c5SHeejin Ahn         // WASM_SEC_ORDER_DYLINK
1948f93426c5SHeejin Ahn         {WASM_SEC_ORDER_DYLINK, WASM_SEC_ORDER_TYPE},
1949f93426c5SHeejin Ahn         // WASM_SEC_ORDER_LINKING
1950f93426c5SHeejin Ahn         {WASM_SEC_ORDER_LINKING, WASM_SEC_ORDER_RELOC, WASM_SEC_ORDER_NAME},
1951f93426c5SHeejin Ahn         // WASM_SEC_ORDER_RELOC (can be repeated)
1952f93426c5SHeejin Ahn         {},
1953f93426c5SHeejin Ahn         // WASM_SEC_ORDER_NAME
1954f93426c5SHeejin Ahn         {WASM_SEC_ORDER_NAME, WASM_SEC_ORDER_PRODUCERS},
1955f93426c5SHeejin Ahn         // WASM_SEC_ORDER_PRODUCERS
1956f93426c5SHeejin Ahn         {WASM_SEC_ORDER_PRODUCERS, WASM_SEC_ORDER_TARGET_FEATURES},
1957f93426c5SHeejin Ahn         // WASM_SEC_ORDER_TARGET_FEATURES
1958f93426c5SHeejin Ahn         {WASM_SEC_ORDER_TARGET_FEATURES}};
19599757bba4SThomas Lively 
1960feef720bSHeejin Ahn bool WasmSectionOrderChecker::isValidSectionOrder(unsigned ID,
1961feef720bSHeejin Ahn                                                   StringRef CustomSectionName) {
1962feef720bSHeejin Ahn   int Order = getSectionOrder(ID, CustomSectionName);
19639757bba4SThomas Lively   if (Order == WASM_SEC_ORDER_NONE)
1964feef720bSHeejin Ahn     return true;
19659757bba4SThomas Lively 
19669757bba4SThomas Lively   // Disallowed predecessors we need to check for
19679757bba4SThomas Lively   SmallVector<int, WASM_NUM_SEC_ORDERS> WorkList;
19689757bba4SThomas Lively 
19699757bba4SThomas Lively   // Keep track of completed checks to avoid repeating work
19709757bba4SThomas Lively   bool Checked[WASM_NUM_SEC_ORDERS] = {};
19719757bba4SThomas Lively 
19729757bba4SThomas Lively   int Curr = Order;
19739757bba4SThomas Lively   while (true) {
19749757bba4SThomas Lively     // Add new disallowed predecessors to work list
19759757bba4SThomas Lively     for (size_t I = 0;; ++I) {
19769757bba4SThomas Lively       int Next = DisallowedPredecessors[Curr][I];
19779757bba4SThomas Lively       if (Next == WASM_SEC_ORDER_NONE)
19789757bba4SThomas Lively         break;
19799757bba4SThomas Lively       if (Checked[Next])
19809757bba4SThomas Lively         continue;
19819757bba4SThomas Lively       WorkList.push_back(Next);
19829757bba4SThomas Lively       Checked[Next] = true;
19839757bba4SThomas Lively     }
19849757bba4SThomas Lively 
19859757bba4SThomas Lively     if (WorkList.empty())
19869757bba4SThomas Lively       break;
19879757bba4SThomas Lively 
19889757bba4SThomas Lively     // Consider next disallowed predecessor
19899757bba4SThomas Lively     Curr = WorkList.pop_back_val();
19909757bba4SThomas Lively     if (Seen[Curr])
19919757bba4SThomas Lively       return false;
19929757bba4SThomas Lively   }
19939757bba4SThomas Lively 
19949757bba4SThomas Lively   // Have not seen any disallowed predecessors
19959757bba4SThomas Lively   Seen[Order] = true;
19969757bba4SThomas Lively   return true;
1997feef720bSHeejin Ahn }
1998