144d95122SEugene Zelenko //===- WasmObjectFile.cpp - Wasm object file implementation ---------------===//
22c6f75ddSDerek Schuff //
32c6f75ddSDerek Schuff //                     The LLVM Compiler Infrastructure
42c6f75ddSDerek Schuff //
52c6f75ddSDerek Schuff // This file is distributed under the University of Illinois Open Source
62c6f75ddSDerek Schuff // License. See LICENSE.TXT for details.
72c6f75ddSDerek Schuff //
82c6f75ddSDerek Schuff //===----------------------------------------------------------------------===//
92c6f75ddSDerek Schuff 
1044d95122SEugene Zelenko #include "llvm/ADT/ArrayRef.h"
114710ed7aSSam Clegg #include "llvm/ADT/DenseSet.h"
1244d95122SEugene Zelenko #include "llvm/ADT/STLExtras.h"
1344d95122SEugene Zelenko #include "llvm/ADT/StringRef.h"
14ea7caceeSSam Clegg #include "llvm/ADT/StringSet.h"
1544d95122SEugene Zelenko #include "llvm/ADT/Triple.h"
16264b5d9eSZachary Turner #include "llvm/BinaryFormat/Wasm.h"
179f5094dfSEugene Zelenko #include "llvm/MC/SubtargetFeature.h"
1844d95122SEugene Zelenko #include "llvm/Object/Binary.h"
1944d95122SEugene Zelenko #include "llvm/Object/Error.h"
2044d95122SEugene Zelenko #include "llvm/Object/ObjectFile.h"
2144d95122SEugene Zelenko #include "llvm/Object/SymbolicFile.h"
222c6f75ddSDerek Schuff #include "llvm/Object/Wasm.h"
232c6f75ddSDerek Schuff #include "llvm/Support/Endian.h"
2444d95122SEugene Zelenko #include "llvm/Support/Error.h"
2544d95122SEugene Zelenko #include "llvm/Support/ErrorHandling.h"
262c6f75ddSDerek Schuff #include "llvm/Support/LEB128.h"
2744d95122SEugene Zelenko #include <algorithm>
289f5094dfSEugene Zelenko #include <cassert>
2944d95122SEugene Zelenko #include <cstdint>
309f5094dfSEugene Zelenko #include <cstring>
3144d95122SEugene Zelenko #include <system_error>
322c6f75ddSDerek Schuff 
33b7787fd0SSam Clegg #define DEBUG_TYPE "wasm-object"
34b7787fd0SSam Clegg 
3544d95122SEugene Zelenko using namespace llvm;
3644d95122SEugene Zelenko using namespace object;
372c6f75ddSDerek Schuff 
38659932b0SJF Bastien void WasmSymbol::print(raw_ostream &Out) const {
39659932b0SJF Bastien   Out << "Name=" << Info.Name
40659932b0SJF Bastien   << ", Kind=" << toString(wasm::WasmSymbolType(Info.Kind))
41659932b0SJF Bastien   << ", Flags=" << Info.Flags;
42659932b0SJF Bastien   if (!isTypeData()) {
43659932b0SJF Bastien     Out << ", ElemIndex=" << Info.ElementIndex;
44659932b0SJF Bastien   } else if (isDefined()) {
45659932b0SJF Bastien     Out << ", Segment=" << Info.DataRef.Segment;
46659932b0SJF Bastien     Out << ", Offset=" << Info.DataRef.Offset;
47659932b0SJF Bastien     Out << ", Size=" << Info.DataRef.Size;
48659932b0SJF Bastien   }
49659932b0SJF Bastien }
50659932b0SJF Bastien 
51*ddc84bf7SJF Bastien #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
52*ddc84bf7SJF Bastien LLVM_DUMP_METHOD void WasmSymbol::dump() const { print(dbgs()); }
53*ddc84bf7SJF Bastien #endif
54*ddc84bf7SJF Bastien 
552c6f75ddSDerek Schuff Expected<std::unique_ptr<WasmObjectFile>>
562c6f75ddSDerek Schuff ObjectFile::createWasmObjectFile(MemoryBufferRef Buffer) {
572c6f75ddSDerek Schuff   Error Err = Error::success();
582c6f75ddSDerek Schuff   auto ObjectFile = llvm::make_unique<WasmObjectFile>(Buffer, Err);
592c6f75ddSDerek Schuff   if (Err)
602c6f75ddSDerek Schuff     return std::move(Err);
612c6f75ddSDerek Schuff 
622c6f75ddSDerek Schuff   return std::move(ObjectFile);
632c6f75ddSDerek Schuff }
642c6f75ddSDerek Schuff 
65d3d84fddSDerek Schuff #define VARINT7_MAX ((1<<7)-1)
66d3d84fddSDerek Schuff #define VARINT7_MIN (-(1<<7))
67d3d84fddSDerek Schuff #define VARUINT7_MAX (1<<7)
68d3d84fddSDerek Schuff #define VARUINT1_MAX (1)
69d3d84fddSDerek Schuff 
70d3d84fddSDerek Schuff static uint8_t readUint8(const uint8_t *&Ptr) { return *Ptr++; }
71d3d84fddSDerek Schuff 
7244d95122SEugene Zelenko static uint32_t readUint32(const uint8_t *&Ptr) {
732c6f75ddSDerek Schuff   uint32_t Result = support::endian::read32le(Ptr);
742c6f75ddSDerek Schuff   Ptr += sizeof(Result);
752c6f75ddSDerek Schuff   return Result;
762c6f75ddSDerek Schuff }
772c6f75ddSDerek Schuff 
78d3d84fddSDerek Schuff static int32_t readFloat32(const uint8_t *&Ptr) {
79d3d84fddSDerek Schuff   int32_t Result = 0;
80d3d84fddSDerek Schuff   memcpy(&Result, Ptr, sizeof(Result));
81d3d84fddSDerek Schuff   Ptr += sizeof(Result);
82d3d84fddSDerek Schuff   return Result;
83d3d84fddSDerek Schuff }
84d3d84fddSDerek Schuff 
85d3d84fddSDerek Schuff static int64_t readFloat64(const uint8_t *&Ptr) {
86d3d84fddSDerek Schuff   int64_t Result = 0;
87d3d84fddSDerek Schuff   memcpy(&Result, Ptr, sizeof(Result));
88d3d84fddSDerek Schuff   Ptr += sizeof(Result);
89d3d84fddSDerek Schuff   return Result;
90d3d84fddSDerek Schuff }
91d3d84fddSDerek Schuff 
9244d95122SEugene Zelenko static uint64_t readULEB128(const uint8_t *&Ptr) {
932c6f75ddSDerek Schuff   unsigned Count;
942c6f75ddSDerek Schuff   uint64_t Result = decodeULEB128(Ptr, &Count);
952c6f75ddSDerek Schuff   Ptr += Count;
962c6f75ddSDerek Schuff   return Result;
972c6f75ddSDerek Schuff }
982c6f75ddSDerek Schuff 
9944d95122SEugene Zelenko static StringRef readString(const uint8_t *&Ptr) {
1002c6f75ddSDerek Schuff   uint32_t StringLen = readULEB128(Ptr);
1012c6f75ddSDerek Schuff   StringRef Return = StringRef(reinterpret_cast<const char *>(Ptr), StringLen);
1022c6f75ddSDerek Schuff   Ptr += StringLen;
1032c6f75ddSDerek Schuff   return Return;
1042c6f75ddSDerek Schuff }
1052c6f75ddSDerek Schuff 
106d3d84fddSDerek Schuff static int64_t readLEB128(const uint8_t *&Ptr) {
107d3d84fddSDerek Schuff   unsigned Count;
108d3d84fddSDerek Schuff   uint64_t Result = decodeSLEB128(Ptr, &Count);
109d3d84fddSDerek Schuff   Ptr += Count;
110d3d84fddSDerek Schuff   return Result;
111d3d84fddSDerek Schuff }
112d3d84fddSDerek Schuff 
113d3d84fddSDerek Schuff static uint8_t readVaruint1(const uint8_t *&Ptr) {
114d3d84fddSDerek Schuff   int64_t result = readLEB128(Ptr);
115d3d84fddSDerek Schuff   assert(result <= VARUINT1_MAX && result >= 0);
116d3d84fddSDerek Schuff   return result;
117d3d84fddSDerek Schuff }
118d3d84fddSDerek Schuff 
119d3d84fddSDerek Schuff static int32_t readVarint32(const uint8_t *&Ptr) {
120d3d84fddSDerek Schuff   int64_t result = readLEB128(Ptr);
121d3d84fddSDerek Schuff   assert(result <= INT32_MAX && result >= INT32_MIN);
122d3d84fddSDerek Schuff   return result;
123d3d84fddSDerek Schuff }
124d3d84fddSDerek Schuff 
125d3d84fddSDerek Schuff static uint32_t readVaruint32(const uint8_t *&Ptr) {
126d3d84fddSDerek Schuff   uint64_t result = readULEB128(Ptr);
12754376021SDavide Italiano   assert(result <= UINT32_MAX);
128d3d84fddSDerek Schuff   return result;
129d3d84fddSDerek Schuff }
130d3d84fddSDerek Schuff 
131d3d84fddSDerek Schuff static int64_t readVarint64(const uint8_t *&Ptr) {
132d3d84fddSDerek Schuff   return readLEB128(Ptr);
133d3d84fddSDerek Schuff }
134d3d84fddSDerek Schuff 
135d3d84fddSDerek Schuff static uint8_t readOpcode(const uint8_t *&Ptr) {
136d3d84fddSDerek Schuff   return readUint8(Ptr);
137d3d84fddSDerek Schuff }
138d3d84fddSDerek Schuff 
139d3d84fddSDerek Schuff static Error readInitExpr(wasm::WasmInitExpr &Expr, const uint8_t *&Ptr) {
140d3d84fddSDerek Schuff   Expr.Opcode = readOpcode(Ptr);
141d3d84fddSDerek Schuff 
142d3d84fddSDerek Schuff   switch (Expr.Opcode) {
143d3d84fddSDerek Schuff   case wasm::WASM_OPCODE_I32_CONST:
144d3d84fddSDerek Schuff     Expr.Value.Int32 = readVarint32(Ptr);
145d3d84fddSDerek Schuff     break;
146d3d84fddSDerek Schuff   case wasm::WASM_OPCODE_I64_CONST:
147d3d84fddSDerek Schuff     Expr.Value.Int64 = readVarint64(Ptr);
148d3d84fddSDerek Schuff     break;
149d3d84fddSDerek Schuff   case wasm::WASM_OPCODE_F32_CONST:
150d3d84fddSDerek Schuff     Expr.Value.Float32 = readFloat32(Ptr);
151d3d84fddSDerek Schuff     break;
152d3d84fddSDerek Schuff   case wasm::WASM_OPCODE_F64_CONST:
153d3d84fddSDerek Schuff     Expr.Value.Float64 = readFloat64(Ptr);
154d3d84fddSDerek Schuff     break;
155d3d84fddSDerek Schuff   case wasm::WASM_OPCODE_GET_GLOBAL:
1567fb391feSSam Clegg     Expr.Value.Global = readULEB128(Ptr);
157d3d84fddSDerek Schuff     break;
158d3d84fddSDerek Schuff   default:
159d3d84fddSDerek Schuff     return make_error<GenericBinaryError>("Invalid opcode in init_expr",
160d3d84fddSDerek Schuff                                           object_error::parse_failed);
161d3d84fddSDerek Schuff   }
162d3d84fddSDerek Schuff 
163d3d84fddSDerek Schuff   uint8_t EndOpcode = readOpcode(Ptr);
164d3d84fddSDerek Schuff   if (EndOpcode != wasm::WASM_OPCODE_END) {
165d3d84fddSDerek Schuff     return make_error<GenericBinaryError>("Invalid init_expr",
166d3d84fddSDerek Schuff                                           object_error::parse_failed);
167d3d84fddSDerek Schuff   }
168d3d84fddSDerek Schuff   return Error::success();
169d3d84fddSDerek Schuff }
170d3d84fddSDerek Schuff 
171d3d84fddSDerek Schuff static wasm::WasmLimits readLimits(const uint8_t *&Ptr) {
172d3d84fddSDerek Schuff   wasm::WasmLimits Result;
173d3d84fddSDerek Schuff   Result.Flags = readVaruint1(Ptr);
174d3d84fddSDerek Schuff   Result.Initial = readVaruint32(Ptr);
175d3d84fddSDerek Schuff   if (Result.Flags & wasm::WASM_LIMITS_FLAG_HAS_MAX)
176d3d84fddSDerek Schuff     Result.Maximum = readVaruint32(Ptr);
177d3d84fddSDerek Schuff   return Result;
178d3d84fddSDerek Schuff }
179d3d84fddSDerek Schuff 
1802ffff5afSSam Clegg static wasm::WasmTable readTable(const uint8_t *&Ptr) {
1812ffff5afSSam Clegg   wasm::WasmTable Table;
18203e101f1SSam Clegg   Table.ElemType = readUint8(Ptr);
1832ffff5afSSam Clegg   Table.Limits = readLimits(Ptr);
1842ffff5afSSam Clegg   return Table;
1852ffff5afSSam Clegg }
1862ffff5afSSam Clegg 
187d3d84fddSDerek Schuff static Error readSection(WasmSection &Section, const uint8_t *&Ptr,
18835b50a83SVedant Kumar                          const uint8_t *Start, const uint8_t *Eof) {
1892c6f75ddSDerek Schuff   Section.Offset = Ptr - Start;
19003e101f1SSam Clegg   Section.Type = readUint8(Ptr);
191d3d84fddSDerek Schuff   uint32_t Size = readVaruint32(Ptr);
1922c6f75ddSDerek Schuff   if (Size == 0)
1932c6f75ddSDerek Schuff     return make_error<StringError>("Zero length section",
1942c6f75ddSDerek Schuff                                    object_error::parse_failed);
19535b50a83SVedant Kumar   if (Ptr + Size > Eof)
19635b50a83SVedant Kumar     return make_error<StringError>("Section too large",
19735b50a83SVedant Kumar                                    object_error::parse_failed);
1989745afa6SSam Clegg   if (Section.Type == wasm::WASM_SEC_CUSTOM) {
1999745afa6SSam Clegg     const uint8_t *NameStart = Ptr;
2009745afa6SSam Clegg     Section.Name = readString(Ptr);
2019745afa6SSam Clegg     Size -= Ptr - NameStart;
2029745afa6SSam Clegg   }
2032c6f75ddSDerek Schuff   Section.Content = ArrayRef<uint8_t>(Ptr, Size);
2042c6f75ddSDerek Schuff   Ptr += Size;
2052c6f75ddSDerek Schuff   return Error::success();
2062c6f75ddSDerek Schuff }
2072c6f75ddSDerek Schuff 
2082c6f75ddSDerek Schuff WasmObjectFile::WasmObjectFile(MemoryBufferRef Buffer, Error &Err)
2099f5094dfSEugene Zelenko     : ObjectFile(Binary::ID_Wasm, Buffer) {
2102c6f75ddSDerek Schuff   ErrorAsOutParameter ErrAsOutParam(&Err);
2112c6f75ddSDerek Schuff   Header.Magic = getData().substr(0, 4);
2122c6f75ddSDerek Schuff   if (Header.Magic != StringRef("\0asm", 4)) {
2132c6f75ddSDerek Schuff     Err = make_error<StringError>("Bad magic number",
2142c6f75ddSDerek Schuff                                   object_error::parse_failed);
2152c6f75ddSDerek Schuff     return;
2162c6f75ddSDerek Schuff   }
217a845167dSJonas Devlieghere 
218a845167dSJonas Devlieghere   const uint8_t *Eof = getPtr(getData().size());
2192c6f75ddSDerek Schuff   const uint8_t *Ptr = getPtr(4);
220a845167dSJonas Devlieghere 
221a845167dSJonas Devlieghere   if (Ptr + 4 > Eof) {
222a845167dSJonas Devlieghere     Err = make_error<StringError>("Missing version number",
223a845167dSJonas Devlieghere                                   object_error::parse_failed);
224a845167dSJonas Devlieghere     return;
225a845167dSJonas Devlieghere   }
226a845167dSJonas Devlieghere 
2272c6f75ddSDerek Schuff   Header.Version = readUint32(Ptr);
2282c6f75ddSDerek Schuff   if (Header.Version != wasm::WasmVersion) {
2292c6f75ddSDerek Schuff     Err = make_error<StringError>("Bad version number",
2302c6f75ddSDerek Schuff                                   object_error::parse_failed);
2312c6f75ddSDerek Schuff     return;
2322c6f75ddSDerek Schuff   }
2332c6f75ddSDerek Schuff 
234d3d84fddSDerek Schuff   WasmSection Sec;
2352c6f75ddSDerek Schuff   while (Ptr < Eof) {
23635b50a83SVedant Kumar     if ((Err = readSection(Sec, Ptr, getPtr(0), Eof)))
2372c6f75ddSDerek Schuff       return;
238d3d84fddSDerek Schuff     if ((Err = parseSection(Sec)))
2392c6f75ddSDerek Schuff       return;
240d3d84fddSDerek Schuff 
2412c6f75ddSDerek Schuff     Sections.push_back(Sec);
2422c6f75ddSDerek Schuff   }
2432c6f75ddSDerek Schuff }
2442c6f75ddSDerek Schuff 
245d3d84fddSDerek Schuff Error WasmObjectFile::parseSection(WasmSection &Sec) {
246d3d84fddSDerek Schuff   const uint8_t* Start = Sec.Content.data();
247d3d84fddSDerek Schuff   const uint8_t* End = Start + Sec.Content.size();
248d3d84fddSDerek Schuff   switch (Sec.Type) {
249d3d84fddSDerek Schuff   case wasm::WASM_SEC_CUSTOM:
250d3d84fddSDerek Schuff     return parseCustomSection(Sec, Start, End);
251d3d84fddSDerek Schuff   case wasm::WASM_SEC_TYPE:
252d3d84fddSDerek Schuff     return parseTypeSection(Start, End);
253d3d84fddSDerek Schuff   case wasm::WASM_SEC_IMPORT:
254d3d84fddSDerek Schuff     return parseImportSection(Start, End);
255d3d84fddSDerek Schuff   case wasm::WASM_SEC_FUNCTION:
256d3d84fddSDerek Schuff     return parseFunctionSection(Start, End);
257d3d84fddSDerek Schuff   case wasm::WASM_SEC_TABLE:
258d3d84fddSDerek Schuff     return parseTableSection(Start, End);
259d3d84fddSDerek Schuff   case wasm::WASM_SEC_MEMORY:
260d3d84fddSDerek Schuff     return parseMemorySection(Start, End);
261d3d84fddSDerek Schuff   case wasm::WASM_SEC_GLOBAL:
262d3d84fddSDerek Schuff     return parseGlobalSection(Start, End);
263d3d84fddSDerek Schuff   case wasm::WASM_SEC_EXPORT:
264d3d84fddSDerek Schuff     return parseExportSection(Start, End);
265d3d84fddSDerek Schuff   case wasm::WASM_SEC_START:
266d3d84fddSDerek Schuff     return parseStartSection(Start, End);
267d3d84fddSDerek Schuff   case wasm::WASM_SEC_ELEM:
268d3d84fddSDerek Schuff     return parseElemSection(Start, End);
269d3d84fddSDerek Schuff   case wasm::WASM_SEC_CODE:
270d3d84fddSDerek Schuff     return parseCodeSection(Start, End);
271d3d84fddSDerek Schuff   case wasm::WASM_SEC_DATA:
272d3d84fddSDerek Schuff     return parseDataSection(Start, End);
273d3d84fddSDerek Schuff   default:
274d3d84fddSDerek Schuff     return make_error<GenericBinaryError>("Bad section type",
275d3d84fddSDerek Schuff                                           object_error::parse_failed);
276d3d84fddSDerek Schuff   }
277d3d84fddSDerek Schuff }
278d3d84fddSDerek Schuff 
279d3d84fddSDerek Schuff Error WasmObjectFile::parseNameSection(const uint8_t *Ptr, const uint8_t *End) {
2804710ed7aSSam Clegg   llvm::DenseSet<uint64_t> Seen;
2819f3fe42eSSam Clegg   if (Functions.size() != FunctionTypes.size()) {
2829f3fe42eSSam Clegg     return make_error<GenericBinaryError>("Names must come after code section",
2839f3fe42eSSam Clegg                                           object_error::parse_failed);
2849f3fe42eSSam Clegg   }
2854710ed7aSSam Clegg 
286d3d84fddSDerek Schuff   while (Ptr < End) {
28703e101f1SSam Clegg     uint8_t Type = readUint8(Ptr);
288d3d84fddSDerek Schuff     uint32_t Size = readVaruint32(Ptr);
289b7787fd0SSam Clegg     const uint8_t *SubSectionEnd = Ptr + Size;
290d3d84fddSDerek Schuff     switch (Type) {
291d3d84fddSDerek Schuff     case wasm::WASM_NAMES_FUNCTION: {
292d3d84fddSDerek Schuff       uint32_t Count = readVaruint32(Ptr);
293d3d84fddSDerek Schuff       while (Count--) {
294fc5b5cd2SSam Clegg         uint32_t Index = readVaruint32(Ptr);
2954710ed7aSSam Clegg         if (!Seen.insert(Index).second)
2964710ed7aSSam Clegg           return make_error<GenericBinaryError>("Function named more than once",
2974710ed7aSSam Clegg                                                 object_error::parse_failed);
298d3d84fddSDerek Schuff         StringRef Name = readString(Ptr);
2999f3fe42eSSam Clegg         if (!isValidFunctionIndex(Index) || Name.empty())
3009f3fe42eSSam Clegg           return make_error<GenericBinaryError>("Invalid name entry",
3019f3fe42eSSam Clegg                                                 object_error::parse_failed);
3029f3fe42eSSam Clegg         DebugNames.push_back(wasm::WasmFunctionName{Index, Name});
303ef90ff36SNicholas Wilson         if (isDefinedFunctionIndex(Index))
304ef90ff36SNicholas Wilson           getDefinedFunction(Index).DebugName = Name;
305d3d84fddSDerek Schuff       }
306d3d84fddSDerek Schuff       break;
307d3d84fddSDerek Schuff     }
308d3d84fddSDerek Schuff     // Ignore local names for now
309d3d84fddSDerek Schuff     case wasm::WASM_NAMES_LOCAL:
310d3d84fddSDerek Schuff     default:
311d3d84fddSDerek Schuff       Ptr += Size;
312d3d84fddSDerek Schuff       break;
313d3d84fddSDerek Schuff     }
314b7787fd0SSam Clegg     if (Ptr != SubSectionEnd)
315b7787fd0SSam Clegg       return make_error<GenericBinaryError>("Name sub-section ended prematurely",
316b7787fd0SSam Clegg                                             object_error::parse_failed);
317d3d84fddSDerek Schuff   }
318d3d84fddSDerek Schuff 
319d3d84fddSDerek Schuff   if (Ptr != End)
320d3d84fddSDerek Schuff     return make_error<GenericBinaryError>("Name section ended prematurely",
321d3d84fddSDerek Schuff                                           object_error::parse_failed);
322d3d84fddSDerek Schuff   return Error::success();
323d3d84fddSDerek Schuff }
324d3d84fddSDerek Schuff 
325b7787fd0SSam Clegg Error WasmObjectFile::parseLinkingSection(const uint8_t *Ptr,
326b7787fd0SSam Clegg                                           const uint8_t *End) {
3274df5d764SSam Clegg   HasLinkingSection = true;
3289f3fe42eSSam Clegg   if (Functions.size() != FunctionTypes.size()) {
3299f3fe42eSSam Clegg     return make_error<GenericBinaryError>(
3309f3fe42eSSam Clegg         "Linking data must come after code section", object_error::parse_failed);
3319f3fe42eSSam Clegg   }
332e7a60708SSam Clegg 
3336bb5a41fSSam Clegg   LinkingData.Version = readVaruint32(Ptr);
3346bb5a41fSSam Clegg   if (LinkingData.Version != wasm::WasmMetadataVersion) {
3356bb5a41fSSam Clegg     return make_error<GenericBinaryError>(
3366bb5a41fSSam Clegg         "Unexpected metadata version: " + Twine(LinkingData.Version) +
3376bb5a41fSSam Clegg             " (Expected: " + Twine(wasm::WasmMetadataVersion) + ")",
3386bb5a41fSSam Clegg         object_error::parse_failed);
3396bb5a41fSSam Clegg   }
3406bb5a41fSSam Clegg 
341b7787fd0SSam Clegg   while (Ptr < End) {
34203e101f1SSam Clegg     uint8_t Type = readUint8(Ptr);
343b7787fd0SSam Clegg     uint32_t Size = readVaruint32(Ptr);
344b7787fd0SSam Clegg     const uint8_t *SubSectionEnd = Ptr + Size;
345b7787fd0SSam Clegg     switch (Type) {
3466c899ba6SSam Clegg     case wasm::WASM_SYMBOL_TABLE:
3476c899ba6SSam Clegg       if (Error Err = parseLinkingSectionSymtab(Ptr, SubSectionEnd))
3486c899ba6SSam Clegg         return Err;
349b7787fd0SSam Clegg       break;
35063ebb813SSam Clegg     case wasm::WASM_SEGMENT_INFO: {
351d95ed959SSam Clegg       uint32_t Count = readVaruint32(Ptr);
352d95ed959SSam Clegg       if (Count > DataSegments.size())
353d95ed959SSam Clegg         return make_error<GenericBinaryError>("Too many segment names",
354d95ed959SSam Clegg                                               object_error::parse_failed);
35563ebb813SSam Clegg       for (uint32_t i = 0; i < Count; i++) {
356d95ed959SSam Clegg         DataSegments[i].Data.Name = readString(Ptr);
35763ebb813SSam Clegg         DataSegments[i].Data.Alignment = readVaruint32(Ptr);
35863ebb813SSam Clegg         DataSegments[i].Data.Flags = readVaruint32(Ptr);
35963ebb813SSam Clegg       }
360d95ed959SSam Clegg       break;
361d95ed959SSam Clegg     }
3624273998cSSam Clegg     case wasm::WASM_INIT_FUNCS: {
3634273998cSSam Clegg       uint32_t Count = readVaruint32(Ptr);
3644273998cSSam Clegg       LinkingData.InitFunctions.reserve(Count);
3654273998cSSam Clegg       for (uint32_t i = 0; i < Count; i++) {
3664273998cSSam Clegg         wasm::WasmInitFunc Init;
3674273998cSSam Clegg         Init.Priority = readVaruint32(Ptr);
3686c899ba6SSam Clegg         Init.Symbol = readVaruint32(Ptr);
369b3748f71SNicholas Wilson         if (!isValidFunctionSymbol(Init.Symbol))
3706c899ba6SSam Clegg           return make_error<GenericBinaryError>("Invalid function symbol: " +
3716c899ba6SSam Clegg                                                     Twine(Init.Symbol),
3724273998cSSam Clegg                                                 object_error::parse_failed);
3734273998cSSam Clegg         LinkingData.InitFunctions.emplace_back(Init);
3744273998cSSam Clegg       }
3754273998cSSam Clegg       break;
3764273998cSSam Clegg     }
377ea7caceeSSam Clegg     case wasm::WASM_COMDAT_INFO:
378ea7caceeSSam Clegg       if (Error Err = parseLinkingSectionComdat(Ptr, SubSectionEnd))
379ea7caceeSSam Clegg         return Err;
380ea7caceeSSam Clegg       break;
381b7787fd0SSam Clegg     default:
382b7787fd0SSam Clegg       Ptr += Size;
383b7787fd0SSam Clegg       break;
384b7787fd0SSam Clegg     }
385b7787fd0SSam Clegg     if (Ptr != SubSectionEnd)
386b7787fd0SSam Clegg       return make_error<GenericBinaryError>(
387b7787fd0SSam Clegg           "Linking sub-section ended prematurely", object_error::parse_failed);
388b7787fd0SSam Clegg   }
389b7787fd0SSam Clegg   if (Ptr != End)
390b7787fd0SSam Clegg     return make_error<GenericBinaryError>("Linking section ended prematurely",
391b7787fd0SSam Clegg                                           object_error::parse_failed);
392b7787fd0SSam Clegg   return Error::success();
393b7787fd0SSam Clegg }
394b7787fd0SSam Clegg 
3956c899ba6SSam Clegg Error WasmObjectFile::parseLinkingSectionSymtab(const uint8_t *&Ptr,
3966c899ba6SSam Clegg                                                 const uint8_t *End) {
3976c899ba6SSam Clegg   uint32_t Count = readVaruint32(Ptr);
3986c899ba6SSam Clegg   LinkingData.SymbolTable.reserve(Count);
3996c899ba6SSam Clegg   Symbols.reserve(Count);
4006c899ba6SSam Clegg   StringSet<> SymbolNames;
4016c899ba6SSam Clegg 
4026c899ba6SSam Clegg   std::vector<wasm::WasmImport *> ImportedGlobals;
4036c899ba6SSam Clegg   std::vector<wasm::WasmImport *> ImportedFunctions;
4046c899ba6SSam Clegg   ImportedGlobals.reserve(Imports.size());
4056c899ba6SSam Clegg   ImportedFunctions.reserve(Imports.size());
4066c899ba6SSam Clegg   for (auto &I : Imports) {
4076c899ba6SSam Clegg     if (I.Kind == wasm::WASM_EXTERNAL_FUNCTION)
4086c899ba6SSam Clegg       ImportedFunctions.emplace_back(&I);
4096c899ba6SSam Clegg     else if (I.Kind == wasm::WASM_EXTERNAL_GLOBAL)
4106c899ba6SSam Clegg       ImportedGlobals.emplace_back(&I);
4116c899ba6SSam Clegg   }
4126c899ba6SSam Clegg 
4136c899ba6SSam Clegg   while (Count--) {
4146c899ba6SSam Clegg     wasm::WasmSymbolInfo Info;
4156c899ba6SSam Clegg     const wasm::WasmSignature *FunctionType = nullptr;
4166c899ba6SSam Clegg     const wasm::WasmGlobalType *GlobalType = nullptr;
4176c899ba6SSam Clegg 
4186c899ba6SSam Clegg     Info.Kind = readUint8(Ptr);
4196c899ba6SSam Clegg     Info.Flags = readVaruint32(Ptr);
4206c899ba6SSam Clegg     bool IsDefined = (Info.Flags & wasm::WASM_SYMBOL_UNDEFINED) == 0;
4216c899ba6SSam Clegg 
4226c899ba6SSam Clegg     switch (Info.Kind) {
4236c899ba6SSam Clegg     case wasm::WASM_SYMBOL_TYPE_FUNCTION:
4246c899ba6SSam Clegg       Info.ElementIndex = readVaruint32(Ptr);
4256c899ba6SSam Clegg       if (!isValidFunctionIndex(Info.ElementIndex) ||
4266c899ba6SSam Clegg           IsDefined != isDefinedFunctionIndex(Info.ElementIndex))
4276c899ba6SSam Clegg         return make_error<GenericBinaryError>("invalid function symbol index",
4286c899ba6SSam Clegg                                               object_error::parse_failed);
4296c899ba6SSam Clegg       if (IsDefined) {
4306c899ba6SSam Clegg         Info.Name = readString(Ptr);
4316c899ba6SSam Clegg         unsigned FuncIndex = Info.ElementIndex - NumImportedFunctions;
4326c899ba6SSam Clegg         FunctionType = &Signatures[FunctionTypes[FuncIndex]];
433959e7371SNicholas Wilson         wasm::WasmFunction &Function = Functions[FuncIndex];
434ef90ff36SNicholas Wilson         if (Function.SymbolName.empty())
435ef90ff36SNicholas Wilson           Function.SymbolName = Info.Name;
4366c899ba6SSam Clegg       } else {
4376c899ba6SSam Clegg         wasm::WasmImport &Import = *ImportedFunctions[Info.ElementIndex];
4386c899ba6SSam Clegg         FunctionType = &Signatures[Import.SigIndex];
4396c899ba6SSam Clegg         Info.Name = Import.Field;
4409067b46eSSam Clegg         Info.Module = Import.Module;
4416c899ba6SSam Clegg       }
4426c899ba6SSam Clegg       break;
4436c899ba6SSam Clegg 
4446c899ba6SSam Clegg     case wasm::WASM_SYMBOL_TYPE_GLOBAL:
4456c899ba6SSam Clegg       Info.ElementIndex = readVaruint32(Ptr);
4466c899ba6SSam Clegg       if (!isValidGlobalIndex(Info.ElementIndex) ||
4476c899ba6SSam Clegg           IsDefined != isDefinedGlobalIndex(Info.ElementIndex))
4486c899ba6SSam Clegg         return make_error<GenericBinaryError>("invalid global symbol index",
4496c899ba6SSam Clegg                                               object_error::parse_failed);
45015f349f7SNicholas Wilson       if (!IsDefined &&
45115f349f7SNicholas Wilson           (Info.Flags & wasm::WASM_SYMBOL_BINDING_MASK) ==
45215f349f7SNicholas Wilson               wasm::WASM_SYMBOL_BINDING_WEAK)
45315f349f7SNicholas Wilson         return make_error<GenericBinaryError>("undefined weak global symbol",
45415f349f7SNicholas Wilson                                               object_error::parse_failed);
4556c899ba6SSam Clegg       if (IsDefined) {
4566c899ba6SSam Clegg         Info.Name = readString(Ptr);
4576c899ba6SSam Clegg         unsigned GlobalIndex = Info.ElementIndex - NumImportedGlobals;
458959e7371SNicholas Wilson         wasm::WasmGlobal &Global = Globals[GlobalIndex];
459959e7371SNicholas Wilson         GlobalType = &Global.Type;
460ef90ff36SNicholas Wilson         if (Global.SymbolName.empty())
461ef90ff36SNicholas Wilson           Global.SymbolName = Info.Name;
4626c899ba6SSam Clegg       } else {
4636c899ba6SSam Clegg         wasm::WasmImport &Import = *ImportedGlobals[Info.ElementIndex];
4646c899ba6SSam Clegg         Info.Name = Import.Field;
4656c899ba6SSam Clegg         GlobalType = &Import.Global;
4666c899ba6SSam Clegg       }
4676c899ba6SSam Clegg       break;
4686c899ba6SSam Clegg 
4696c899ba6SSam Clegg     case wasm::WASM_SYMBOL_TYPE_DATA:
4706c899ba6SSam Clegg       Info.Name = readString(Ptr);
4716c899ba6SSam Clegg       if (IsDefined) {
4726c899ba6SSam Clegg         uint32_t Index = readVaruint32(Ptr);
4736c899ba6SSam Clegg         if (Index >= DataSegments.size())
4746c899ba6SSam Clegg           return make_error<GenericBinaryError>("invalid data symbol index",
4756c899ba6SSam Clegg                                                 object_error::parse_failed);
4766c899ba6SSam Clegg         uint32_t Offset = readVaruint32(Ptr);
4776c899ba6SSam Clegg         uint32_t Size = readVaruint32(Ptr);
4786c899ba6SSam Clegg         if (Offset + Size > DataSegments[Index].Data.Content.size())
4797e4eee98SNicholas Wilson           return make_error<GenericBinaryError>("invalid data symbol offset",
4806c899ba6SSam Clegg                                                 object_error::parse_failed);
4816c899ba6SSam Clegg         Info.DataRef = wasm::WasmDataReference{Index, Offset, Size};
4826c899ba6SSam Clegg       }
4836c899ba6SSam Clegg       break;
4846c899ba6SSam Clegg 
4856a31a0d6SSam Clegg     case wasm::WASM_SYMBOL_TYPE_SECTION: {
486d5504a0aSSam Clegg       if ((Info.Flags & wasm::WASM_SYMBOL_BINDING_MASK) !=
487d5504a0aSSam Clegg           wasm::WASM_SYMBOL_BINDING_LOCAL)
488e0658119SSam Clegg         return make_error<GenericBinaryError>(
489e0658119SSam Clegg             "Section symbols must have local binding",
490d5504a0aSSam Clegg             object_error::parse_failed);
4916a31a0d6SSam Clegg       Info.ElementIndex = readVaruint32(Ptr);
4926a31a0d6SSam Clegg       // Use somewhat unique section name as symbol name.
4936a31a0d6SSam Clegg       StringRef SectionName = Sections[Info.ElementIndex].Name;
4946a31a0d6SSam Clegg       Info.Name = SectionName;
4956a31a0d6SSam Clegg       break;
4966a31a0d6SSam Clegg     }
4976a31a0d6SSam Clegg 
4986c899ba6SSam Clegg     default:
4996c899ba6SSam Clegg       return make_error<GenericBinaryError>("Invalid symbol type",
5006c899ba6SSam Clegg                                             object_error::parse_failed);
5016c899ba6SSam Clegg     }
5026c899ba6SSam Clegg 
5036c899ba6SSam Clegg     if ((Info.Flags & wasm::WASM_SYMBOL_BINDING_MASK) !=
5046c899ba6SSam Clegg             wasm::WASM_SYMBOL_BINDING_LOCAL &&
5056c899ba6SSam Clegg         !SymbolNames.insert(Info.Name).second)
5066c899ba6SSam Clegg       return make_error<GenericBinaryError>("Duplicate symbol name " +
5076c899ba6SSam Clegg                                                 Twine(Info.Name),
5086c899ba6SSam Clegg                                             object_error::parse_failed);
5096c899ba6SSam Clegg     LinkingData.SymbolTable.emplace_back(Info);
5106c899ba6SSam Clegg     Symbols.emplace_back(LinkingData.SymbolTable.back(), FunctionType,
5116c899ba6SSam Clegg                          GlobalType);
512d34e60caSNicola Zaghen     LLVM_DEBUG(dbgs() << "Adding symbol: " << Symbols.back() << "\n");
5136c899ba6SSam Clegg   }
5146c899ba6SSam Clegg 
5156c899ba6SSam Clegg   return Error::success();
5166c899ba6SSam Clegg }
5176c899ba6SSam Clegg 
518ea7caceeSSam Clegg Error WasmObjectFile::parseLinkingSectionComdat(const uint8_t *&Ptr,
519ea7caceeSSam Clegg                                                 const uint8_t *End)
520ea7caceeSSam Clegg {
521ea7caceeSSam Clegg   uint32_t ComdatCount = readVaruint32(Ptr);
522ea7caceeSSam Clegg   StringSet<> ComdatSet;
523027b9357SNicholas Wilson   for (unsigned ComdatIndex = 0; ComdatIndex < ComdatCount; ++ComdatIndex) {
524ea7caceeSSam Clegg     StringRef Name = readString(Ptr);
525ea7caceeSSam Clegg     if (Name.empty() || !ComdatSet.insert(Name).second)
526ea7caceeSSam Clegg       return make_error<GenericBinaryError>("Bad/duplicate COMDAT name " + Twine(Name),
527ea7caceeSSam Clegg                                             object_error::parse_failed);
528027b9357SNicholas Wilson     LinkingData.Comdats.emplace_back(Name);
529ea7caceeSSam Clegg     uint32_t Flags = readVaruint32(Ptr);
530ea7caceeSSam Clegg     if (Flags != 0)
531ea7caceeSSam Clegg       return make_error<GenericBinaryError>("Unsupported COMDAT flags",
532ea7caceeSSam Clegg                                             object_error::parse_failed);
533ea7caceeSSam Clegg 
534ea7caceeSSam Clegg     uint32_t EntryCount = readVaruint32(Ptr);
535ea7caceeSSam Clegg     while (EntryCount--) {
536ea7caceeSSam Clegg       unsigned Kind = readVaruint32(Ptr);
537ea7caceeSSam Clegg       unsigned Index = readVaruint32(Ptr);
538ea7caceeSSam Clegg       switch (Kind) {
539ea7caceeSSam Clegg       default:
540ea7caceeSSam Clegg         return make_error<GenericBinaryError>("Invalid COMDAT entry type",
541ea7caceeSSam Clegg                                               object_error::parse_failed);
542ea7caceeSSam Clegg       case wasm::WASM_COMDAT_DATA:
543ea7caceeSSam Clegg         if (Index >= DataSegments.size())
544ea7caceeSSam Clegg           return make_error<GenericBinaryError>("COMDAT data index out of range",
545ea7caceeSSam Clegg                                                 object_error::parse_failed);
546027b9357SNicholas Wilson         if (DataSegments[Index].Data.Comdat != UINT32_MAX)
547ea7caceeSSam Clegg           return make_error<GenericBinaryError>("Data segment in two COMDATs",
548ea7caceeSSam Clegg                                                 object_error::parse_failed);
549027b9357SNicholas Wilson         DataSegments[Index].Data.Comdat = ComdatIndex;
550ea7caceeSSam Clegg         break;
551ea7caceeSSam Clegg       case wasm::WASM_COMDAT_FUNCTION:
55223012e98SSam Clegg         if (!isDefinedFunctionIndex(Index))
553ea7caceeSSam Clegg           return make_error<GenericBinaryError>("COMDAT function index out of range",
554ea7caceeSSam Clegg                                                 object_error::parse_failed);
555027b9357SNicholas Wilson         if (getDefinedFunction(Index).Comdat != UINT32_MAX)
556ea7caceeSSam Clegg           return make_error<GenericBinaryError>("Function in two COMDATs",
557ea7caceeSSam Clegg                                                 object_error::parse_failed);
558027b9357SNicholas Wilson         getDefinedFunction(Index).Comdat = ComdatIndex;
559ea7caceeSSam Clegg         break;
560ea7caceeSSam Clegg       }
561ea7caceeSSam Clegg     }
562ea7caceeSSam Clegg   }
563ea7caceeSSam Clegg   return Error::success();
564ea7caceeSSam Clegg }
565ea7caceeSSam Clegg 
566d3d84fddSDerek Schuff Error WasmObjectFile::parseRelocSection(StringRef Name, const uint8_t *Ptr,
567d3d84fddSDerek Schuff                                         const uint8_t *End) {
5686f08c84aSSam Clegg   uint32_t SectionIndex = readVaruint32(Ptr);
5696f08c84aSSam Clegg   if (SectionIndex >= Sections.size())
5706f08c84aSSam Clegg     return make_error<GenericBinaryError>("Invalid section index",
571d3d84fddSDerek Schuff                                           object_error::parse_failed);
5726f08c84aSSam Clegg   WasmSection& Section = Sections[SectionIndex];
573d3d84fddSDerek Schuff   uint32_t RelocCount = readVaruint32(Ptr);
5746f08c84aSSam Clegg   uint32_t EndOffset = Section.Content.size();
575d3d84fddSDerek Schuff   while (RelocCount--) {
576b3748f71SNicholas Wilson     wasm::WasmRelocation Reloc = {};
577d3d84fddSDerek Schuff     Reloc.Type = readVaruint32(Ptr);
578d3d84fddSDerek Schuff     Reloc.Offset = readVaruint32(Ptr);
579d3d84fddSDerek Schuff     Reloc.Index = readVaruint32(Ptr);
580d3d84fddSDerek Schuff     switch (Reloc.Type) {
581d3d84fddSDerek Schuff     case wasm::R_WEBASSEMBLY_FUNCTION_INDEX_LEB:
582d3d84fddSDerek Schuff     case wasm::R_WEBASSEMBLY_TABLE_INDEX_SLEB:
583d3d84fddSDerek Schuff     case wasm::R_WEBASSEMBLY_TABLE_INDEX_I32:
584b3748f71SNicholas Wilson       if (!isValidFunctionSymbol(Reloc.Index))
585b3748f71SNicholas Wilson         return make_error<GenericBinaryError>("Bad relocation function index",
586b3748f71SNicholas Wilson                                               object_error::parse_failed);
587b3748f71SNicholas Wilson       break;
588cc182aaaSSam Clegg     case wasm::R_WEBASSEMBLY_TYPE_INDEX_LEB:
589b3748f71SNicholas Wilson       if (Reloc.Index >= Signatures.size())
590b3748f71SNicholas Wilson         return make_error<GenericBinaryError>("Bad relocation type index",
591b3748f71SNicholas Wilson                                               object_error::parse_failed);
592b3748f71SNicholas Wilson       break;
5939d24fb7fSSam Clegg     case wasm::R_WEBASSEMBLY_GLOBAL_INDEX_LEB:
594b3748f71SNicholas Wilson       if (!isValidGlobalSymbol(Reloc.Index))
595b3748f71SNicholas Wilson         return make_error<GenericBinaryError>("Bad relocation global index",
596b3748f71SNicholas Wilson                                               object_error::parse_failed);
597d3d84fddSDerek Schuff       break;
59813a2e899SSam Clegg     case wasm::R_WEBASSEMBLY_MEMORY_ADDR_LEB:
59913a2e899SSam Clegg     case wasm::R_WEBASSEMBLY_MEMORY_ADDR_SLEB:
60013a2e899SSam Clegg     case wasm::R_WEBASSEMBLY_MEMORY_ADDR_I32:
601b3748f71SNicholas Wilson       if (!isValidDataSymbol(Reloc.Index))
602b3748f71SNicholas Wilson         return make_error<GenericBinaryError>("Bad relocation data index",
603b3748f71SNicholas Wilson                                               object_error::parse_failed);
604cc182aaaSSam Clegg       Reloc.Addend = readVarint32(Ptr);
605d3d84fddSDerek Schuff       break;
6066a31a0d6SSam Clegg     case wasm::R_WEBASSEMBLY_FUNCTION_OFFSET_I32:
6076a31a0d6SSam Clegg       if (!isValidFunctionSymbol(Reloc.Index))
6086a31a0d6SSam Clegg         return make_error<GenericBinaryError>("Bad relocation function index",
6096a31a0d6SSam Clegg                                               object_error::parse_failed);
6106a31a0d6SSam Clegg       Reloc.Addend = readVarint32(Ptr);
6116a31a0d6SSam Clegg       break;
6126a31a0d6SSam Clegg     case wasm::R_WEBASSEMBLY_SECTION_OFFSET_I32:
6136a31a0d6SSam Clegg       if (!isValidSectionSymbol(Reloc.Index))
6146a31a0d6SSam Clegg         return make_error<GenericBinaryError>("Bad relocation section index",
6156a31a0d6SSam Clegg                                               object_error::parse_failed);
6166a31a0d6SSam Clegg       Reloc.Addend = readVarint32(Ptr);
6176a31a0d6SSam Clegg       break;
618d3d84fddSDerek Schuff     default:
6199d24fb7fSSam Clegg       return make_error<GenericBinaryError>("Bad relocation type: " +
6209d24fb7fSSam Clegg                                                 Twine(Reloc.Type),
621d3d84fddSDerek Schuff                                             object_error::parse_failed);
622d3d84fddSDerek Schuff     }
623b3748f71SNicholas Wilson 
624b3748f71SNicholas Wilson     // Relocations must fit inside the section, and must appear in order.  They
625b3748f71SNicholas Wilson     // also shouldn't overlap a function/element boundary, but we don't bother
626b3748f71SNicholas Wilson     // to check that.
627b3748f71SNicholas Wilson     uint64_t Size = 5;
628b3748f71SNicholas Wilson     if (Reloc.Type == wasm::R_WEBASSEMBLY_TABLE_INDEX_I32 ||
6296a31a0d6SSam Clegg         Reloc.Type == wasm::R_WEBASSEMBLY_MEMORY_ADDR_I32 ||
6306a31a0d6SSam Clegg         Reloc.Type == wasm::R_WEBASSEMBLY_SECTION_OFFSET_I32 ||
6316a31a0d6SSam Clegg         Reloc.Type == wasm::R_WEBASSEMBLY_FUNCTION_OFFSET_I32)
632b3748f71SNicholas Wilson       Size = 4;
633fa5a04fbSSam Clegg     if (Reloc.Offset + Size > EndOffset)
634b3748f71SNicholas Wilson       return make_error<GenericBinaryError>("Bad relocation offset",
635b3748f71SNicholas Wilson                                             object_error::parse_failed);
636b3748f71SNicholas Wilson 
6376f08c84aSSam Clegg     Section.Relocations.push_back(Reloc);
638d3d84fddSDerek Schuff   }
639d3d84fddSDerek Schuff   if (Ptr != End)
640d3d84fddSDerek Schuff     return make_error<GenericBinaryError>("Reloc section ended prematurely",
641d3d84fddSDerek Schuff                                           object_error::parse_failed);
642d3d84fddSDerek Schuff   return Error::success();
643d3d84fddSDerek Schuff }
644d3d84fddSDerek Schuff 
645d3d84fddSDerek Schuff Error WasmObjectFile::parseCustomSection(WasmSection &Sec,
646d3d84fddSDerek Schuff                                          const uint8_t *Ptr, const uint8_t *End) {
647d3d84fddSDerek Schuff   if (Sec.Name == "name") {
648d3d84fddSDerek Schuff     if (Error Err = parseNameSection(Ptr, End))
649d3d84fddSDerek Schuff       return Err;
650b7787fd0SSam Clegg   } else if (Sec.Name == "linking") {
651b7787fd0SSam Clegg     if (Error Err = parseLinkingSection(Ptr, End))
652b7787fd0SSam Clegg       return Err;
653d3d84fddSDerek Schuff   } else if (Sec.Name.startswith("reloc.")) {
654d3d84fddSDerek Schuff     if (Error Err = parseRelocSection(Sec.Name, Ptr, End))
655d3d84fddSDerek Schuff       return Err;
656d3d84fddSDerek Schuff   }
657d3d84fddSDerek Schuff   return Error::success();
658d3d84fddSDerek Schuff }
659d3d84fddSDerek Schuff 
660d3d84fddSDerek Schuff Error WasmObjectFile::parseTypeSection(const uint8_t *Ptr, const uint8_t *End) {
661d3d84fddSDerek Schuff   uint32_t Count = readVaruint32(Ptr);
662d3d84fddSDerek Schuff   Signatures.reserve(Count);
663d3d84fddSDerek Schuff   while (Count--) {
664d3d84fddSDerek Schuff     wasm::WasmSignature Sig;
665d3d84fddSDerek Schuff     Sig.ReturnType = wasm::WASM_TYPE_NORESULT;
66603e101f1SSam Clegg     uint8_t Form = readUint8(Ptr);
667d3d84fddSDerek Schuff     if (Form != wasm::WASM_TYPE_FUNC) {
668d3d84fddSDerek Schuff       return make_error<GenericBinaryError>("Invalid signature type",
669d3d84fddSDerek Schuff                                             object_error::parse_failed);
670d3d84fddSDerek Schuff     }
671d3d84fddSDerek Schuff     uint32_t ParamCount = readVaruint32(Ptr);
672d3d84fddSDerek Schuff     Sig.ParamTypes.reserve(ParamCount);
673d3d84fddSDerek Schuff     while (ParamCount--) {
67403e101f1SSam Clegg       uint32_t ParamType = readUint8(Ptr);
675d3d84fddSDerek Schuff       Sig.ParamTypes.push_back(ParamType);
676d3d84fddSDerek Schuff     }
677d3d84fddSDerek Schuff     uint32_t ReturnCount = readVaruint32(Ptr);
678d3d84fddSDerek Schuff     if (ReturnCount) {
679d3d84fddSDerek Schuff       if (ReturnCount != 1) {
680d3d84fddSDerek Schuff         return make_error<GenericBinaryError>(
681d3d84fddSDerek Schuff             "Multiple return types not supported", object_error::parse_failed);
682d3d84fddSDerek Schuff       }
68303e101f1SSam Clegg       Sig.ReturnType = readUint8(Ptr);
684d3d84fddSDerek Schuff     }
685d3d84fddSDerek Schuff     Signatures.push_back(Sig);
686d3d84fddSDerek Schuff   }
687d3d84fddSDerek Schuff   if (Ptr != End)
688d3d84fddSDerek Schuff     return make_error<GenericBinaryError>("Type section ended prematurely",
689d3d84fddSDerek Schuff                                           object_error::parse_failed);
690d3d84fddSDerek Schuff   return Error::success();
691d3d84fddSDerek Schuff }
692d3d84fddSDerek Schuff 
693d3d84fddSDerek Schuff Error WasmObjectFile::parseImportSection(const uint8_t *Ptr, const uint8_t *End) {
694d3d84fddSDerek Schuff   uint32_t Count = readVaruint32(Ptr);
695d3d84fddSDerek Schuff   Imports.reserve(Count);
696fc5b5cd2SSam Clegg   for (uint32_t i = 0; i < Count; i++) {
697d3d84fddSDerek Schuff     wasm::WasmImport Im;
698d3d84fddSDerek Schuff     Im.Module = readString(Ptr);
699d3d84fddSDerek Schuff     Im.Field = readString(Ptr);
700d3d84fddSDerek Schuff     Im.Kind = readUint8(Ptr);
701d3d84fddSDerek Schuff     switch (Im.Kind) {
702d3d84fddSDerek Schuff     case wasm::WASM_EXTERNAL_FUNCTION:
703a3b9fe6aSSam Clegg       NumImportedFunctions++;
704d3d84fddSDerek Schuff       Im.SigIndex = readVaruint32(Ptr);
705d3d84fddSDerek Schuff       break;
706d3d84fddSDerek Schuff     case wasm::WASM_EXTERNAL_GLOBAL:
707a3b9fe6aSSam Clegg       NumImportedGlobals++;
70803e101f1SSam Clegg       Im.Global.Type = readUint8(Ptr);
7092ffff5afSSam Clegg       Im.Global.Mutable = readVaruint1(Ptr);
710d3d84fddSDerek Schuff       break;
7112ffff5afSSam Clegg     case wasm::WASM_EXTERNAL_MEMORY:
7122ffff5afSSam Clegg       Im.Memory = readLimits(Ptr);
7132ffff5afSSam Clegg       break;
7142ffff5afSSam Clegg     case wasm::WASM_EXTERNAL_TABLE:
7152ffff5afSSam Clegg       Im.Table = readTable(Ptr);
716e7a60708SSam Clegg       if (Im.Table.ElemType != wasm::WASM_TYPE_ANYFUNC)
7172ffff5afSSam Clegg         return make_error<GenericBinaryError>("Invalid table element type",
7182ffff5afSSam Clegg                                               object_error::parse_failed);
7192ffff5afSSam Clegg       break;
720d3d84fddSDerek Schuff     default:
721d3d84fddSDerek Schuff       return make_error<GenericBinaryError>(
722d3d84fddSDerek Schuff           "Unexpected import kind", object_error::parse_failed);
723d3d84fddSDerek Schuff     }
724d3d84fddSDerek Schuff     Imports.push_back(Im);
725d3d84fddSDerek Schuff   }
726d3d84fddSDerek Schuff   if (Ptr != End)
727d3d84fddSDerek Schuff     return make_error<GenericBinaryError>("Import section ended prematurely",
728d3d84fddSDerek Schuff                                           object_error::parse_failed);
729d3d84fddSDerek Schuff   return Error::success();
730d3d84fddSDerek Schuff }
731d3d84fddSDerek Schuff 
732d3d84fddSDerek Schuff Error WasmObjectFile::parseFunctionSection(const uint8_t *Ptr, const uint8_t *End) {
733d3d84fddSDerek Schuff   uint32_t Count = readVaruint32(Ptr);
734d3d84fddSDerek Schuff   FunctionTypes.reserve(Count);
7353b825105SNicholas Wilson   uint32_t NumTypes = Signatures.size();
736d3d84fddSDerek Schuff   while (Count--) {
7373b825105SNicholas Wilson     uint32_t Type = readVaruint32(Ptr);
7383b825105SNicholas Wilson     if (Type >= NumTypes)
7393b825105SNicholas Wilson       return make_error<GenericBinaryError>("Invalid function type",
7403b825105SNicholas Wilson                                             object_error::parse_failed);
7413b825105SNicholas Wilson     FunctionTypes.push_back(Type);
742d3d84fddSDerek Schuff   }
743d3d84fddSDerek Schuff   if (Ptr != End)
744d3d84fddSDerek Schuff     return make_error<GenericBinaryError>("Function section ended prematurely",
745d3d84fddSDerek Schuff                                           object_error::parse_failed);
746d3d84fddSDerek Schuff   return Error::success();
747d3d84fddSDerek Schuff }
748d3d84fddSDerek Schuff 
749d3d84fddSDerek Schuff Error WasmObjectFile::parseTableSection(const uint8_t *Ptr, const uint8_t *End) {
750d3d84fddSDerek Schuff   uint32_t Count = readVaruint32(Ptr);
751d3d84fddSDerek Schuff   Tables.reserve(Count);
752d3d84fddSDerek Schuff   while (Count--) {
7532ffff5afSSam Clegg     Tables.push_back(readTable(Ptr));
7542ffff5afSSam Clegg     if (Tables.back().ElemType != wasm::WASM_TYPE_ANYFUNC) {
755d3d84fddSDerek Schuff       return make_error<GenericBinaryError>("Invalid table element type",
756d3d84fddSDerek Schuff                                             object_error::parse_failed);
757d3d84fddSDerek Schuff     }
758d3d84fddSDerek Schuff   }
759d3d84fddSDerek Schuff   if (Ptr != End)
760d3d84fddSDerek Schuff     return make_error<GenericBinaryError>("Table section ended prematurely",
761d3d84fddSDerek Schuff                                           object_error::parse_failed);
762d3d84fddSDerek Schuff   return Error::success();
763d3d84fddSDerek Schuff }
764d3d84fddSDerek Schuff 
765d3d84fddSDerek Schuff Error WasmObjectFile::parseMemorySection(const uint8_t *Ptr, const uint8_t *End) {
766d3d84fddSDerek Schuff   uint32_t Count = readVaruint32(Ptr);
767d3d84fddSDerek Schuff   Memories.reserve(Count);
768d3d84fddSDerek Schuff   while (Count--) {
769d3d84fddSDerek Schuff     Memories.push_back(readLimits(Ptr));
770d3d84fddSDerek Schuff   }
771d3d84fddSDerek Schuff   if (Ptr != End)
772d3d84fddSDerek Schuff     return make_error<GenericBinaryError>("Memory section ended prematurely",
773d3d84fddSDerek Schuff                                           object_error::parse_failed);
774d3d84fddSDerek Schuff   return Error::success();
775d3d84fddSDerek Schuff }
776d3d84fddSDerek Schuff 
777d3d84fddSDerek Schuff Error WasmObjectFile::parseGlobalSection(const uint8_t *Ptr, const uint8_t *End) {
7786c899ba6SSam Clegg   GlobalSection = Sections.size();
779d3d84fddSDerek Schuff   uint32_t Count = readVaruint32(Ptr);
780d3d84fddSDerek Schuff   Globals.reserve(Count);
781d3d84fddSDerek Schuff   while (Count--) {
782d3d84fddSDerek Schuff     wasm::WasmGlobal Global;
783e53af7f6SSam Clegg     Global.Index = NumImportedGlobals + Globals.size();
78403e101f1SSam Clegg     Global.Type.Type = readUint8(Ptr);
7856e7f1826SSam Clegg     Global.Type.Mutable = readVaruint1(Ptr);
786deede839SDavide Italiano     if (Error Err = readInitExpr(Global.InitExpr, Ptr))
787d3d84fddSDerek Schuff       return Err;
788d3d84fddSDerek Schuff     Globals.push_back(Global);
789d3d84fddSDerek Schuff   }
790d3d84fddSDerek Schuff   if (Ptr != End)
791d3d84fddSDerek Schuff     return make_error<GenericBinaryError>("Global section ended prematurely",
792d3d84fddSDerek Schuff                                           object_error::parse_failed);
793d3d84fddSDerek Schuff   return Error::success();
794d3d84fddSDerek Schuff }
795d3d84fddSDerek Schuff 
796d3d84fddSDerek Schuff Error WasmObjectFile::parseExportSection(const uint8_t *Ptr, const uint8_t *End) {
797d3d84fddSDerek Schuff   uint32_t Count = readVaruint32(Ptr);
798d3d84fddSDerek Schuff   Exports.reserve(Count);
799fc5b5cd2SSam Clegg   for (uint32_t i = 0; i < Count; i++) {
800d3d84fddSDerek Schuff     wasm::WasmExport Ex;
801d3d84fddSDerek Schuff     Ex.Name = readString(Ptr);
802d3d84fddSDerek Schuff     Ex.Kind = readUint8(Ptr);
803d3d84fddSDerek Schuff     Ex.Index = readVaruint32(Ptr);
804d3d84fddSDerek Schuff     switch (Ex.Kind) {
805d3d84fddSDerek Schuff     case wasm::WASM_EXTERNAL_FUNCTION:
80623012e98SSam Clegg       if (!isValidFunctionIndex(Ex.Index))
807a3b9fe6aSSam Clegg         return make_error<GenericBinaryError>("Invalid function export",
808a3b9fe6aSSam Clegg                                               object_error::parse_failed);
809d3d84fddSDerek Schuff       break;
8106c899ba6SSam Clegg     case wasm::WASM_EXTERNAL_GLOBAL:
8116c899ba6SSam Clegg       if (!isValidGlobalIndex(Ex.Index))
812a3b9fe6aSSam Clegg         return make_error<GenericBinaryError>("Invalid global export",
813a3b9fe6aSSam Clegg                                               object_error::parse_failed);
814d3d84fddSDerek Schuff       break;
8152ffff5afSSam Clegg     case wasm::WASM_EXTERNAL_MEMORY:
8162ffff5afSSam Clegg     case wasm::WASM_EXTERNAL_TABLE:
8172ffff5afSSam Clegg       break;
818d3d84fddSDerek Schuff     default:
819d3d84fddSDerek Schuff       return make_error<GenericBinaryError>(
820d3d84fddSDerek Schuff           "Unexpected export kind", object_error::parse_failed);
821d3d84fddSDerek Schuff     }
822b7787fd0SSam Clegg     Exports.push_back(Ex);
823d3d84fddSDerek Schuff   }
824d3d84fddSDerek Schuff   if (Ptr != End)
825d3d84fddSDerek Schuff     return make_error<GenericBinaryError>("Export section ended prematurely",
826d3d84fddSDerek Schuff                                           object_error::parse_failed);
827d3d84fddSDerek Schuff   return Error::success();
828d3d84fddSDerek Schuff }
829d3d84fddSDerek Schuff 
8304273998cSSam Clegg bool WasmObjectFile::isValidFunctionIndex(uint32_t Index) const {
8316c899ba6SSam Clegg   return Index < NumImportedFunctions + FunctionTypes.size();
8324273998cSSam Clegg }
8334273998cSSam Clegg 
83423012e98SSam Clegg bool WasmObjectFile::isDefinedFunctionIndex(uint32_t Index) const {
83523012e98SSam Clegg   return Index >= NumImportedFunctions && isValidFunctionIndex(Index);
83623012e98SSam Clegg }
83723012e98SSam Clegg 
8386c899ba6SSam Clegg bool WasmObjectFile::isValidGlobalIndex(uint32_t Index) const {
8396c899ba6SSam Clegg   return Index < NumImportedGlobals + Globals.size();
8406c899ba6SSam Clegg }
8416c899ba6SSam Clegg 
8426c899ba6SSam Clegg bool WasmObjectFile::isDefinedGlobalIndex(uint32_t Index) const {
8436c899ba6SSam Clegg   return Index >= NumImportedGlobals && isValidGlobalIndex(Index);
8446c899ba6SSam Clegg }
8456c899ba6SSam Clegg 
846b3748f71SNicholas Wilson bool WasmObjectFile::isValidFunctionSymbol(uint32_t Index) const {
8476c899ba6SSam Clegg   return Index < Symbols.size() && Symbols[Index].isTypeFunction();
8486c899ba6SSam Clegg }
8496c899ba6SSam Clegg 
850b3748f71SNicholas Wilson bool WasmObjectFile::isValidGlobalSymbol(uint32_t Index) const {
851b3748f71SNicholas Wilson   return Index < Symbols.size() && Symbols[Index].isTypeGlobal();
852b3748f71SNicholas Wilson }
853b3748f71SNicholas Wilson 
854b3748f71SNicholas Wilson bool WasmObjectFile::isValidDataSymbol(uint32_t Index) const {
855b3748f71SNicholas Wilson   return Index < Symbols.size() && Symbols[Index].isTypeData();
856b3748f71SNicholas Wilson }
857b3748f71SNicholas Wilson 
8586a31a0d6SSam Clegg bool WasmObjectFile::isValidSectionSymbol(uint32_t Index) const {
8596a31a0d6SSam Clegg   return Index < Symbols.size() && Symbols[Index].isTypeSection();
8606a31a0d6SSam Clegg }
8616a31a0d6SSam Clegg 
86223012e98SSam Clegg wasm::WasmFunction &WasmObjectFile::getDefinedFunction(uint32_t Index) {
86323012e98SSam Clegg   assert(isDefinedFunctionIndex(Index));
86423012e98SSam Clegg   return Functions[Index - NumImportedFunctions];
86523012e98SSam Clegg }
86623012e98SSam Clegg 
8676c899ba6SSam Clegg wasm::WasmGlobal &WasmObjectFile::getDefinedGlobal(uint32_t Index) {
8686c899ba6SSam Clegg   assert(isDefinedGlobalIndex(Index));
8696c899ba6SSam Clegg   return Globals[Index - NumImportedGlobals];
8706c899ba6SSam Clegg }
8716c899ba6SSam Clegg 
872d3d84fddSDerek Schuff Error WasmObjectFile::parseStartSection(const uint8_t *Ptr, const uint8_t *End) {
873d3d84fddSDerek Schuff   StartFunction = readVaruint32(Ptr);
8744273998cSSam Clegg   if (!isValidFunctionIndex(StartFunction))
875d3d84fddSDerek Schuff     return make_error<GenericBinaryError>("Invalid start function",
876d3d84fddSDerek Schuff                                           object_error::parse_failed);
877d3d84fddSDerek Schuff   return Error::success();
878d3d84fddSDerek Schuff }
879d3d84fddSDerek Schuff 
880d3d84fddSDerek Schuff Error WasmObjectFile::parseCodeSection(const uint8_t *Ptr, const uint8_t *End) {
8810b55ccf6SSam Clegg   CodeSection = Sections.size();
882c551522dSSam Clegg   const uint8_t *CodeSectionStart = Ptr;
883d3d84fddSDerek Schuff   uint32_t FunctionCount = readVaruint32(Ptr);
884d3d84fddSDerek Schuff   if (FunctionCount != FunctionTypes.size()) {
885d3d84fddSDerek Schuff     return make_error<GenericBinaryError>("Invalid function count",
886d3d84fddSDerek Schuff                                           object_error::parse_failed);
887d3d84fddSDerek Schuff   }
888d3d84fddSDerek Schuff 
889d3d84fddSDerek Schuff   while (FunctionCount--) {
890d3d84fddSDerek Schuff     wasm::WasmFunction Function;
891c551522dSSam Clegg     const uint8_t *FunctionStart = Ptr;
892c551522dSSam Clegg     uint32_t Size = readVaruint32(Ptr);
893c551522dSSam Clegg     const uint8_t *FunctionEnd = Ptr + Size;
894c551522dSSam Clegg 
8951be4bf9aSSam Clegg     Function.CodeOffset = Ptr - FunctionStart;
896e53af7f6SSam Clegg     Function.Index = NumImportedFunctions + Functions.size();
897c551522dSSam Clegg     Function.CodeSectionOffset = FunctionStart - CodeSectionStart;
898c551522dSSam Clegg     Function.Size = FunctionEnd - FunctionStart;
899d3d84fddSDerek Schuff 
900d3d84fddSDerek Schuff     uint32_t NumLocalDecls = readVaruint32(Ptr);
901d3d84fddSDerek Schuff     Function.Locals.reserve(NumLocalDecls);
902d3d84fddSDerek Schuff     while (NumLocalDecls--) {
903d3d84fddSDerek Schuff       wasm::WasmLocalDecl Decl;
904d3d84fddSDerek Schuff       Decl.Count = readVaruint32(Ptr);
90503e101f1SSam Clegg       Decl.Type = readUint8(Ptr);
906d3d84fddSDerek Schuff       Function.Locals.push_back(Decl);
907d3d84fddSDerek Schuff     }
908d3d84fddSDerek Schuff 
909d3d84fddSDerek Schuff     uint32_t BodySize = FunctionEnd - Ptr;
910d3d84fddSDerek Schuff     Function.Body = ArrayRef<uint8_t>(Ptr, BodySize);
911027b9357SNicholas Wilson     // This will be set later when reading in the linking metadata section.
912027b9357SNicholas Wilson     Function.Comdat = UINT32_MAX;
913d3d84fddSDerek Schuff     Ptr += BodySize;
914d3d84fddSDerek Schuff     assert(Ptr == FunctionEnd);
915d3d84fddSDerek Schuff     Functions.push_back(Function);
916d3d84fddSDerek Schuff   }
917d3d84fddSDerek Schuff   if (Ptr != End)
918d3d84fddSDerek Schuff     return make_error<GenericBinaryError>("Code section ended prematurely",
919d3d84fddSDerek Schuff                                           object_error::parse_failed);
920d3d84fddSDerek Schuff   return Error::success();
921d3d84fddSDerek Schuff }
922d3d84fddSDerek Schuff 
923d3d84fddSDerek Schuff Error WasmObjectFile::parseElemSection(const uint8_t *Ptr, const uint8_t *End) {
924d3d84fddSDerek Schuff   uint32_t Count = readVaruint32(Ptr);
925d3d84fddSDerek Schuff   ElemSegments.reserve(Count);
926d3d84fddSDerek Schuff   while (Count--) {
927d3d84fddSDerek Schuff     wasm::WasmElemSegment Segment;
928d3d84fddSDerek Schuff     Segment.TableIndex = readVaruint32(Ptr);
929d3d84fddSDerek Schuff     if (Segment.TableIndex != 0) {
930d3d84fddSDerek Schuff       return make_error<GenericBinaryError>("Invalid TableIndex",
931d3d84fddSDerek Schuff                                             object_error::parse_failed);
932d3d84fddSDerek Schuff     }
933d3d84fddSDerek Schuff     if (Error Err = readInitExpr(Segment.Offset, Ptr))
934d3d84fddSDerek Schuff       return Err;
935d3d84fddSDerek Schuff     uint32_t NumElems = readVaruint32(Ptr);
936d3d84fddSDerek Schuff     while (NumElems--) {
937d3d84fddSDerek Schuff       Segment.Functions.push_back(readVaruint32(Ptr));
938d3d84fddSDerek Schuff     }
939d3d84fddSDerek Schuff     ElemSegments.push_back(Segment);
940d3d84fddSDerek Schuff   }
941d3d84fddSDerek Schuff   if (Ptr != End)
942d3d84fddSDerek Schuff     return make_error<GenericBinaryError>("Elem section ended prematurely",
943d3d84fddSDerek Schuff                                           object_error::parse_failed);
944d3d84fddSDerek Schuff   return Error::success();
945d3d84fddSDerek Schuff }
946d3d84fddSDerek Schuff 
947d3d84fddSDerek Schuff Error WasmObjectFile::parseDataSection(const uint8_t *Ptr, const uint8_t *End) {
9480b55ccf6SSam Clegg   DataSection = Sections.size();
9499c07f94aSSam Clegg   const uint8_t *Start = Ptr;
950d3d84fddSDerek Schuff   uint32_t Count = readVaruint32(Ptr);
951d3d84fddSDerek Schuff   DataSegments.reserve(Count);
952d3d84fddSDerek Schuff   while (Count--) {
9539c07f94aSSam Clegg     WasmSegment Segment;
9549c07f94aSSam Clegg     Segment.Data.MemoryIndex = readVaruint32(Ptr);
9559c07f94aSSam Clegg     if (Error Err = readInitExpr(Segment.Data.Offset, Ptr))
956d3d84fddSDerek Schuff       return Err;
957d3d84fddSDerek Schuff     uint32_t Size = readVaruint32(Ptr);
9589c07f94aSSam Clegg     Segment.Data.Content = ArrayRef<uint8_t>(Ptr, Size);
959027b9357SNicholas Wilson     // The rest of these Data fields are set later, when reading in the linking
960027b9357SNicholas Wilson     // metadata section.
96163ebb813SSam Clegg     Segment.Data.Alignment = 0;
96263ebb813SSam Clegg     Segment.Data.Flags = 0;
963027b9357SNicholas Wilson     Segment.Data.Comdat = UINT32_MAX;
9649c07f94aSSam Clegg     Segment.SectionOffset = Ptr - Start;
965d3d84fddSDerek Schuff     Ptr += Size;
966d3d84fddSDerek Schuff     DataSegments.push_back(Segment);
967d3d84fddSDerek Schuff   }
968d3d84fddSDerek Schuff   if (Ptr != End)
969d3d84fddSDerek Schuff     return make_error<GenericBinaryError>("Data section ended prematurely",
970d3d84fddSDerek Schuff                                           object_error::parse_failed);
9712c6f75ddSDerek Schuff   return Error::success();
9722c6f75ddSDerek Schuff }
9732c6f75ddSDerek Schuff 
9742c6f75ddSDerek Schuff const uint8_t *WasmObjectFile::getPtr(size_t Offset) const {
9752c6f75ddSDerek Schuff   return reinterpret_cast<const uint8_t *>(getData().substr(Offset, 1).data());
9762c6f75ddSDerek Schuff }
9772c6f75ddSDerek Schuff 
9782c6f75ddSDerek Schuff const wasm::WasmObjectHeader &WasmObjectFile::getHeader() const {
9792c6f75ddSDerek Schuff   return Header;
9802c6f75ddSDerek Schuff }
9812c6f75ddSDerek Schuff 
982d3d84fddSDerek Schuff void WasmObjectFile::moveSymbolNext(DataRefImpl &Symb) const { Symb.d.a++; }
9832c6f75ddSDerek Schuff 
9842c6f75ddSDerek Schuff uint32_t WasmObjectFile::getSymbolFlags(DataRefImpl Symb) const {
985fc5b5cd2SSam Clegg   uint32_t Result = SymbolRef::SF_None;
986d3d84fddSDerek Schuff   const WasmSymbol &Sym = getWasmSymbol(Symb);
987fc5b5cd2SSam Clegg 
988d34e60caSNicola Zaghen   LLVM_DEBUG(dbgs() << "getSymbolFlags: ptr=" << &Sym << " " << Sym << "\n");
9896e7f1826SSam Clegg   if (Sym.isBindingWeak())
990b7787fd0SSam Clegg     Result |= SymbolRef::SF_Weak;
9916e7f1826SSam Clegg   if (!Sym.isBindingLocal())
99231a2c809SSam Clegg     Result |= SymbolRef::SF_Global;
993a2b35dacSSam Clegg   if (Sym.isHidden())
994a2b35dacSSam Clegg     Result |= SymbolRef::SF_Hidden;
9956c899ba6SSam Clegg   if (!Sym.isDefined())
996fc5b5cd2SSam Clegg     Result |= SymbolRef::SF_Undefined;
9976c899ba6SSam Clegg   if (Sym.isTypeFunction())
9986c899ba6SSam Clegg     Result |= SymbolRef::SF_Executable;
999fc5b5cd2SSam Clegg   return Result;
10002c6f75ddSDerek Schuff }
10012c6f75ddSDerek Schuff 
10022c6f75ddSDerek Schuff basic_symbol_iterator WasmObjectFile::symbol_begin() const {
1003d3d84fddSDerek Schuff   DataRefImpl Ref;
1004d3d84fddSDerek Schuff   Ref.d.a = 0;
1005d3d84fddSDerek Schuff   return BasicSymbolRef(Ref, this);
10062c6f75ddSDerek Schuff }
10072c6f75ddSDerek Schuff 
10082c6f75ddSDerek Schuff basic_symbol_iterator WasmObjectFile::symbol_end() const {
1009d3d84fddSDerek Schuff   DataRefImpl Ref;
1010d3d84fddSDerek Schuff   Ref.d.a = Symbols.size();
1011d3d84fddSDerek Schuff   return BasicSymbolRef(Ref, this);
1012d3d84fddSDerek Schuff }
1013d3d84fddSDerek Schuff 
10142ffff5afSSam Clegg const WasmSymbol &WasmObjectFile::getWasmSymbol(const DataRefImpl &Symb) const {
1015d3d84fddSDerek Schuff   return Symbols[Symb.d.a];
10162c6f75ddSDerek Schuff }
10172c6f75ddSDerek Schuff 
10182ffff5afSSam Clegg const WasmSymbol &WasmObjectFile::getWasmSymbol(const SymbolRef &Symb) const {
10192ffff5afSSam Clegg   return getWasmSymbol(Symb.getRawDataRefImpl());
10202ffff5afSSam Clegg }
10212ffff5afSSam Clegg 
10222c6f75ddSDerek Schuff Expected<StringRef> WasmObjectFile::getSymbolName(DataRefImpl Symb) const {
10236c899ba6SSam Clegg   return getWasmSymbol(Symb).Info.Name;
10242c6f75ddSDerek Schuff }
10252c6f75ddSDerek Schuff 
10262c6f75ddSDerek Schuff Expected<uint64_t> WasmObjectFile::getSymbolAddress(DataRefImpl Symb) const {
1027fc5b5cd2SSam Clegg   return getSymbolValue(Symb);
10282c6f75ddSDerek Schuff }
10292c6f75ddSDerek Schuff 
103031a2c809SSam Clegg uint64_t WasmObjectFile::getWasmSymbolValue(const WasmSymbol& Sym) const {
10316c899ba6SSam Clegg   switch (Sym.Info.Kind) {
10326c899ba6SSam Clegg   case wasm::WASM_SYMBOL_TYPE_FUNCTION:
10336c899ba6SSam Clegg   case wasm::WASM_SYMBOL_TYPE_GLOBAL:
10346c899ba6SSam Clegg     return Sym.Info.ElementIndex;
10356c899ba6SSam Clegg   case wasm::WASM_SYMBOL_TYPE_DATA: {
10366c899ba6SSam Clegg     // The value of a data symbol is the segment offset, plus the symbol
10376c899ba6SSam Clegg     // offset within the segment.
10386c899ba6SSam Clegg     uint32_t SegmentIndex = Sym.Info.DataRef.Segment;
10396c899ba6SSam Clegg     const wasm::WasmDataSegment &Segment = DataSegments[SegmentIndex].Data;
10406c899ba6SSam Clegg     assert(Segment.Offset.Opcode == wasm::WASM_OPCODE_I32_CONST);
10416c899ba6SSam Clegg     return Segment.Offset.Value.Int32 + Sym.Info.DataRef.Offset;
1042b09cfa51SSam Clegg   }
10436a31a0d6SSam Clegg   case wasm::WASM_SYMBOL_TYPE_SECTION:
10446a31a0d6SSam Clegg     return 0;
10452c6f75ddSDerek Schuff   }
10467f055deeSSam Clegg   llvm_unreachable("invalid symbol type");
1047b7787fd0SSam Clegg }
10482c6f75ddSDerek Schuff 
104931a2c809SSam Clegg uint64_t WasmObjectFile::getSymbolValueImpl(DataRefImpl Symb) const {
105031a2c809SSam Clegg   return getWasmSymbolValue(getWasmSymbol(Symb));
105131a2c809SSam Clegg }
105231a2c809SSam Clegg 
10532c6f75ddSDerek Schuff uint32_t WasmObjectFile::getSymbolAlignment(DataRefImpl Symb) const {
10542c6f75ddSDerek Schuff   llvm_unreachable("not yet implemented");
10552c6f75ddSDerek Schuff   return 0;
10562c6f75ddSDerek Schuff }
10572c6f75ddSDerek Schuff 
10582c6f75ddSDerek Schuff uint64_t WasmObjectFile::getCommonSymbolSizeImpl(DataRefImpl Symb) const {
10592c6f75ddSDerek Schuff   llvm_unreachable("not yet implemented");
10602c6f75ddSDerek Schuff   return 0;
10612c6f75ddSDerek Schuff }
10622c6f75ddSDerek Schuff 
10632c6f75ddSDerek Schuff Expected<SymbolRef::Type>
10642c6f75ddSDerek Schuff WasmObjectFile::getSymbolType(DataRefImpl Symb) const {
1065fc5b5cd2SSam Clegg   const WasmSymbol &Sym = getWasmSymbol(Symb);
1066fc5b5cd2SSam Clegg 
10676c899ba6SSam Clegg   switch (Sym.Info.Kind) {
10686c899ba6SSam Clegg   case wasm::WASM_SYMBOL_TYPE_FUNCTION:
1069fc5b5cd2SSam Clegg     return SymbolRef::ST_Function;
10706c899ba6SSam Clegg   case wasm::WASM_SYMBOL_TYPE_GLOBAL:
10716c899ba6SSam Clegg     return SymbolRef::ST_Other;
10726c899ba6SSam Clegg   case wasm::WASM_SYMBOL_TYPE_DATA:
1073fc5b5cd2SSam Clegg     return SymbolRef::ST_Data;
10746a31a0d6SSam Clegg   case wasm::WASM_SYMBOL_TYPE_SECTION:
10756a31a0d6SSam Clegg     return SymbolRef::ST_Debug;
1076fc5b5cd2SSam Clegg   }
1077fc5b5cd2SSam Clegg 
1078fc5b5cd2SSam Clegg   llvm_unreachable("Unknown WasmSymbol::SymbolType");
1079fc5b5cd2SSam Clegg   return SymbolRef::ST_Other;
10802c6f75ddSDerek Schuff }
10812c6f75ddSDerek Schuff 
10822c6f75ddSDerek Schuff Expected<section_iterator>
10832c6f75ddSDerek Schuff WasmObjectFile::getSymbolSection(DataRefImpl Symb) const {
10840b55ccf6SSam Clegg   const WasmSymbol& Sym = getWasmSymbol(Symb);
10856c899ba6SSam Clegg   if (Sym.isUndefined())
10860b55ccf6SSam Clegg     return section_end();
10876c899ba6SSam Clegg 
10886c899ba6SSam Clegg   DataRefImpl Ref;
10896c899ba6SSam Clegg   switch (Sym.Info.Kind) {
10906c899ba6SSam Clegg   case wasm::WASM_SYMBOL_TYPE_FUNCTION:
10916c899ba6SSam Clegg     Ref.d.a = CodeSection;
10926c899ba6SSam Clegg     break;
10936c899ba6SSam Clegg   case wasm::WASM_SYMBOL_TYPE_GLOBAL:
10946c899ba6SSam Clegg     Ref.d.a = GlobalSection;
10956c899ba6SSam Clegg     break;
10966c899ba6SSam Clegg   case wasm::WASM_SYMBOL_TYPE_DATA:
10976c899ba6SSam Clegg     Ref.d.a = DataSection;
10986c899ba6SSam Clegg     break;
10996a31a0d6SSam Clegg   case wasm::WASM_SYMBOL_TYPE_SECTION: {
11006a31a0d6SSam Clegg     Ref.d.a = Sym.Info.ElementIndex;
11016a31a0d6SSam Clegg     break;
11026a31a0d6SSam Clegg   }
11036c899ba6SSam Clegg   default:
11046c899ba6SSam Clegg     llvm_unreachable("Unknown WasmSymbol::SymbolType");
11056c899ba6SSam Clegg   }
1106fc5b5cd2SSam Clegg   return section_iterator(SectionRef(Ref, this));
11072c6f75ddSDerek Schuff }
11082c6f75ddSDerek Schuff 
11092c6f75ddSDerek Schuff void WasmObjectFile::moveSectionNext(DataRefImpl &Sec) const { Sec.d.a++; }
11102c6f75ddSDerek Schuff 
11112c6f75ddSDerek Schuff std::error_code WasmObjectFile::getSectionName(DataRefImpl Sec,
11122c6f75ddSDerek Schuff                                                StringRef &Res) const {
1113d3d84fddSDerek Schuff   const WasmSection &S = Sections[Sec.d.a];
11142c6f75ddSDerek Schuff #define ECase(X)                                                               \
11152c6f75ddSDerek Schuff   case wasm::WASM_SEC_##X:                                                     \
11162c6f75ddSDerek Schuff     Res = #X;                                                                  \
11172c6f75ddSDerek Schuff     break
11182c6f75ddSDerek Schuff   switch (S.Type) {
11192c6f75ddSDerek Schuff     ECase(TYPE);
11202c6f75ddSDerek Schuff     ECase(IMPORT);
11212c6f75ddSDerek Schuff     ECase(FUNCTION);
11222c6f75ddSDerek Schuff     ECase(TABLE);
11232c6f75ddSDerek Schuff     ECase(MEMORY);
11242c6f75ddSDerek Schuff     ECase(GLOBAL);
11252c6f75ddSDerek Schuff     ECase(EXPORT);
11262c6f75ddSDerek Schuff     ECase(START);
11272c6f75ddSDerek Schuff     ECase(ELEM);
11282c6f75ddSDerek Schuff     ECase(CODE);
11292c6f75ddSDerek Schuff     ECase(DATA);
11306d76b7b4SDerek Schuff   case wasm::WASM_SEC_CUSTOM:
11312c6f75ddSDerek Schuff     Res = S.Name;
11322c6f75ddSDerek Schuff     break;
11332c6f75ddSDerek Schuff   default:
11342c6f75ddSDerek Schuff     return object_error::invalid_section_index;
11352c6f75ddSDerek Schuff   }
11362c6f75ddSDerek Schuff #undef ECase
11372c6f75ddSDerek Schuff   return std::error_code();
11382c6f75ddSDerek Schuff }
11392c6f75ddSDerek Schuff 
11402c6f75ddSDerek Schuff uint64_t WasmObjectFile::getSectionAddress(DataRefImpl Sec) const { return 0; }
11412c6f75ddSDerek Schuff 
1142a25d329bSGeorge Rimar uint64_t WasmObjectFile::getSectionIndex(DataRefImpl Sec) const {
1143a25d329bSGeorge Rimar   return Sec.d.a;
1144a25d329bSGeorge Rimar }
1145a25d329bSGeorge Rimar 
11462c6f75ddSDerek Schuff uint64_t WasmObjectFile::getSectionSize(DataRefImpl Sec) const {
1147d3d84fddSDerek Schuff   const WasmSection &S = Sections[Sec.d.a];
11482c6f75ddSDerek Schuff   return S.Content.size();
11492c6f75ddSDerek Schuff }
11502c6f75ddSDerek Schuff 
11512c6f75ddSDerek Schuff std::error_code WasmObjectFile::getSectionContents(DataRefImpl Sec,
11522c6f75ddSDerek Schuff                                                    StringRef &Res) const {
1153d3d84fddSDerek Schuff   const WasmSection &S = Sections[Sec.d.a];
11542c6f75ddSDerek Schuff   // This will never fail since wasm sections can never be empty (user-sections
11552c6f75ddSDerek Schuff   // must have a name and non-user sections each have a defined structure).
11562c6f75ddSDerek Schuff   Res = StringRef(reinterpret_cast<const char *>(S.Content.data()),
11572c6f75ddSDerek Schuff                   S.Content.size());
11582c6f75ddSDerek Schuff   return std::error_code();
11592c6f75ddSDerek Schuff }
11602c6f75ddSDerek Schuff 
11612c6f75ddSDerek Schuff uint64_t WasmObjectFile::getSectionAlignment(DataRefImpl Sec) const {
11622c6f75ddSDerek Schuff   return 1;
11632c6f75ddSDerek Schuff }
11642c6f75ddSDerek Schuff 
11652c6f75ddSDerek Schuff bool WasmObjectFile::isSectionCompressed(DataRefImpl Sec) const {
11662c6f75ddSDerek Schuff   return false;
11672c6f75ddSDerek Schuff }
11682c6f75ddSDerek Schuff 
11692c6f75ddSDerek Schuff bool WasmObjectFile::isSectionText(DataRefImpl Sec) const {
1170d3d84fddSDerek Schuff   return getWasmSection(Sec).Type == wasm::WASM_SEC_CODE;
11712c6f75ddSDerek Schuff }
11722c6f75ddSDerek Schuff 
11732c6f75ddSDerek Schuff bool WasmObjectFile::isSectionData(DataRefImpl Sec) const {
1174d3d84fddSDerek Schuff   return getWasmSection(Sec).Type == wasm::WASM_SEC_DATA;
11752c6f75ddSDerek Schuff }
11762c6f75ddSDerek Schuff 
11772c6f75ddSDerek Schuff bool WasmObjectFile::isSectionBSS(DataRefImpl Sec) const { return false; }
11782c6f75ddSDerek Schuff 
11792c6f75ddSDerek Schuff bool WasmObjectFile::isSectionVirtual(DataRefImpl Sec) const { return false; }
11802c6f75ddSDerek Schuff 
11812c6f75ddSDerek Schuff bool WasmObjectFile::isSectionBitcode(DataRefImpl Sec) const { return false; }
11822c6f75ddSDerek Schuff 
1183d3d84fddSDerek Schuff relocation_iterator WasmObjectFile::section_rel_begin(DataRefImpl Ref) const {
1184d3d84fddSDerek Schuff   DataRefImpl RelocRef;
1185d3d84fddSDerek Schuff   RelocRef.d.a = Ref.d.a;
1186d3d84fddSDerek Schuff   RelocRef.d.b = 0;
1187d3d84fddSDerek Schuff   return relocation_iterator(RelocationRef(RelocRef, this));
11882c6f75ddSDerek Schuff }
11892c6f75ddSDerek Schuff 
1190d3d84fddSDerek Schuff relocation_iterator WasmObjectFile::section_rel_end(DataRefImpl Ref) const {
1191d3d84fddSDerek Schuff   const WasmSection &Sec = getWasmSection(Ref);
1192d3d84fddSDerek Schuff   DataRefImpl RelocRef;
1193d3d84fddSDerek Schuff   RelocRef.d.a = Ref.d.a;
1194d3d84fddSDerek Schuff   RelocRef.d.b = Sec.Relocations.size();
1195d3d84fddSDerek Schuff   return relocation_iterator(RelocationRef(RelocRef, this));
11962c6f75ddSDerek Schuff }
11972c6f75ddSDerek Schuff 
11982c6f75ddSDerek Schuff void WasmObjectFile::moveRelocationNext(DataRefImpl &Rel) const {
1199d3d84fddSDerek Schuff   Rel.d.b++;
12002c6f75ddSDerek Schuff }
12012c6f75ddSDerek Schuff 
1202d3d84fddSDerek Schuff uint64_t WasmObjectFile::getRelocationOffset(DataRefImpl Ref) const {
1203d3d84fddSDerek Schuff   const wasm::WasmRelocation &Rel = getWasmRelocation(Ref);
1204d3d84fddSDerek Schuff   return Rel.Offset;
12052c6f75ddSDerek Schuff }
12062c6f75ddSDerek Schuff 
1207f676cdd5SSam Clegg symbol_iterator WasmObjectFile::getRelocationSymbol(DataRefImpl Ref) const {
1208f676cdd5SSam Clegg   const wasm::WasmRelocation &Rel = getWasmRelocation(Ref);
1209f676cdd5SSam Clegg   if (Rel.Type == wasm::R_WEBASSEMBLY_TYPE_INDEX_LEB)
1210f676cdd5SSam Clegg     return symbol_end();
1211f676cdd5SSam Clegg   DataRefImpl Sym;
1212f676cdd5SSam Clegg   Sym.d.a = Rel.Index;
1213f676cdd5SSam Clegg   Sym.d.b = 0;
1214f676cdd5SSam Clegg   return symbol_iterator(SymbolRef(Sym, this));
12152c6f75ddSDerek Schuff }
12162c6f75ddSDerek Schuff 
1217d3d84fddSDerek Schuff uint64_t WasmObjectFile::getRelocationType(DataRefImpl Ref) const {
1218d3d84fddSDerek Schuff   const wasm::WasmRelocation &Rel = getWasmRelocation(Ref);
1219d3d84fddSDerek Schuff   return Rel.Type;
12202c6f75ddSDerek Schuff }
12212c6f75ddSDerek Schuff 
12222c6f75ddSDerek Schuff void WasmObjectFile::getRelocationTypeName(
1223d3d84fddSDerek Schuff     DataRefImpl Ref, SmallVectorImpl<char> &Result) const {
1224d3d84fddSDerek Schuff   const wasm::WasmRelocation& Rel = getWasmRelocation(Ref);
1225d3d84fddSDerek Schuff   StringRef Res = "Unknown";
1226d3d84fddSDerek Schuff 
1227d3d84fddSDerek Schuff #define WASM_RELOC(name, value)  \
1228d3d84fddSDerek Schuff   case wasm::name:              \
1229d3d84fddSDerek Schuff     Res = #name;               \
1230d3d84fddSDerek Schuff     break;
1231d3d84fddSDerek Schuff 
1232d3d84fddSDerek Schuff   switch (Rel.Type) {
1233c5d8bc83SSam Clegg #include "llvm/BinaryFormat/WasmRelocs.def"
1234d3d84fddSDerek Schuff   }
1235d3d84fddSDerek Schuff 
1236d3d84fddSDerek Schuff #undef WASM_RELOC
1237d3d84fddSDerek Schuff 
1238d3d84fddSDerek Schuff   Result.append(Res.begin(), Res.end());
12392c6f75ddSDerek Schuff }
12402c6f75ddSDerek Schuff 
12412c6f75ddSDerek Schuff section_iterator WasmObjectFile::section_begin() const {
12422c6f75ddSDerek Schuff   DataRefImpl Ref;
12432c6f75ddSDerek Schuff   Ref.d.a = 0;
12442c6f75ddSDerek Schuff   return section_iterator(SectionRef(Ref, this));
12452c6f75ddSDerek Schuff }
12462c6f75ddSDerek Schuff 
12472c6f75ddSDerek Schuff section_iterator WasmObjectFile::section_end() const {
12482c6f75ddSDerek Schuff   DataRefImpl Ref;
12492c6f75ddSDerek Schuff   Ref.d.a = Sections.size();
12502c6f75ddSDerek Schuff   return section_iterator(SectionRef(Ref, this));
12512c6f75ddSDerek Schuff }
12522c6f75ddSDerek Schuff 
12532c6f75ddSDerek Schuff uint8_t WasmObjectFile::getBytesInAddress() const { return 4; }
12542c6f75ddSDerek Schuff 
12552c6f75ddSDerek Schuff StringRef WasmObjectFile::getFileFormatName() const { return "WASM"; }
12562c6f75ddSDerek Schuff 
1257260fe3ecSZachary Turner Triple::ArchType WasmObjectFile::getArch() const { return Triple::wasm32; }
12582c6f75ddSDerek Schuff 
12592c6f75ddSDerek Schuff SubtargetFeatures WasmObjectFile::getFeatures() const {
12602c6f75ddSDerek Schuff   return SubtargetFeatures();
12612c6f75ddSDerek Schuff }
12622c6f75ddSDerek Schuff 
12634df5d764SSam Clegg bool WasmObjectFile::isRelocatableObject() const {
12644df5d764SSam Clegg   return HasLinkingSection;
12654df5d764SSam Clegg }
12662c6f75ddSDerek Schuff 
1267d3d84fddSDerek Schuff const WasmSection &WasmObjectFile::getWasmSection(DataRefImpl Ref) const {
126816fe5822SDavide Italiano   assert(Ref.d.a < Sections.size());
1269d3d84fddSDerek Schuff   return Sections[Ref.d.a];
1270d3d84fddSDerek Schuff }
1271d3d84fddSDerek Schuff 
1272d3d84fddSDerek Schuff const WasmSection &
12732c6f75ddSDerek Schuff WasmObjectFile::getWasmSection(const SectionRef &Section) const {
1274d3d84fddSDerek Schuff   return getWasmSection(Section.getRawDataRefImpl());
1275d3d84fddSDerek Schuff }
1276d3d84fddSDerek Schuff 
1277d3d84fddSDerek Schuff const wasm::WasmRelocation &
1278d3d84fddSDerek Schuff WasmObjectFile::getWasmRelocation(const RelocationRef &Ref) const {
1279d3d84fddSDerek Schuff   return getWasmRelocation(Ref.getRawDataRefImpl());
1280d3d84fddSDerek Schuff }
1281d3d84fddSDerek Schuff 
1282d3d84fddSDerek Schuff const wasm::WasmRelocation &
1283d3d84fddSDerek Schuff WasmObjectFile::getWasmRelocation(DataRefImpl Ref) const {
128416fe5822SDavide Italiano   assert(Ref.d.a < Sections.size());
1285d3d84fddSDerek Schuff   const WasmSection& Sec = Sections[Ref.d.a];
128616fe5822SDavide Italiano   assert(Ref.d.b < Sec.Relocations.size());
1287d3d84fddSDerek Schuff   return Sec.Relocations[Ref.d.b];
12882c6f75ddSDerek Schuff }
1289