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"
1144d95122SEugene Zelenko #include "llvm/ADT/STLExtras.h"
1244d95122SEugene Zelenko #include "llvm/ADT/StringRef.h"
1344d95122SEugene Zelenko #include "llvm/ADT/Triple.h"
149f5094dfSEugene Zelenko #include "llvm/MC/SubtargetFeature.h"
1544d95122SEugene Zelenko #include "llvm/Object/Binary.h"
1644d95122SEugene Zelenko #include "llvm/Object/Error.h"
1744d95122SEugene Zelenko #include "llvm/Object/ObjectFile.h"
1844d95122SEugene Zelenko #include "llvm/Object/SymbolicFile.h"
192c6f75ddSDerek Schuff #include "llvm/Object/Wasm.h"
202c6f75ddSDerek Schuff #include "llvm/Support/Endian.h"
2144d95122SEugene Zelenko #include "llvm/Support/Error.h"
2244d95122SEugene Zelenko #include "llvm/Support/ErrorHandling.h"
232c6f75ddSDerek Schuff #include "llvm/Support/LEB128.h"
2444d95122SEugene Zelenko #include "llvm/Support/Wasm.h"
2544d95122SEugene Zelenko #include <algorithm>
269f5094dfSEugene Zelenko #include <cassert>
2744d95122SEugene Zelenko #include <cstdint>
289f5094dfSEugene Zelenko #include <cstring>
2944d95122SEugene Zelenko #include <system_error>
302c6f75ddSDerek Schuff 
3144d95122SEugene Zelenko using namespace llvm;
3244d95122SEugene Zelenko using namespace object;
332c6f75ddSDerek Schuff 
342c6f75ddSDerek Schuff Expected<std::unique_ptr<WasmObjectFile>>
352c6f75ddSDerek Schuff ObjectFile::createWasmObjectFile(MemoryBufferRef Buffer) {
362c6f75ddSDerek Schuff   Error Err = Error::success();
372c6f75ddSDerek Schuff   auto ObjectFile = llvm::make_unique<WasmObjectFile>(Buffer, Err);
382c6f75ddSDerek Schuff   if (Err)
392c6f75ddSDerek Schuff     return std::move(Err);
402c6f75ddSDerek Schuff 
412c6f75ddSDerek Schuff   return std::move(ObjectFile);
422c6f75ddSDerek Schuff }
432c6f75ddSDerek Schuff 
44d3d84fddSDerek Schuff #define VARINT7_MAX ((1<<7)-1)
45d3d84fddSDerek Schuff #define VARINT7_MIN (-(1<<7))
46d3d84fddSDerek Schuff #define VARUINT7_MAX (1<<7)
47d3d84fddSDerek Schuff #define VARUINT1_MAX (1)
48d3d84fddSDerek Schuff 
49d3d84fddSDerek Schuff static uint8_t readUint8(const uint8_t *&Ptr) { return *Ptr++; }
50d3d84fddSDerek Schuff 
5144d95122SEugene Zelenko static uint32_t readUint32(const uint8_t *&Ptr) {
522c6f75ddSDerek Schuff   uint32_t Result = support::endian::read32le(Ptr);
532c6f75ddSDerek Schuff   Ptr += sizeof(Result);
542c6f75ddSDerek Schuff   return Result;
552c6f75ddSDerek Schuff }
562c6f75ddSDerek Schuff 
57d3d84fddSDerek Schuff static int32_t readFloat32(const uint8_t *&Ptr) {
58d3d84fddSDerek Schuff   int32_t Result = 0;
59d3d84fddSDerek Schuff   memcpy(&Result, Ptr, sizeof(Result));
60d3d84fddSDerek Schuff   Ptr += sizeof(Result);
61d3d84fddSDerek Schuff   return Result;
62d3d84fddSDerek Schuff }
63d3d84fddSDerek Schuff 
64d3d84fddSDerek Schuff static int64_t readFloat64(const uint8_t *&Ptr) {
65d3d84fddSDerek Schuff   int64_t Result = 0;
66d3d84fddSDerek Schuff   memcpy(&Result, Ptr, sizeof(Result));
67d3d84fddSDerek Schuff   Ptr += sizeof(Result);
68d3d84fddSDerek Schuff   return Result;
69d3d84fddSDerek Schuff }
70d3d84fddSDerek Schuff 
7144d95122SEugene Zelenko static uint64_t readULEB128(const uint8_t *&Ptr) {
722c6f75ddSDerek Schuff   unsigned Count;
732c6f75ddSDerek Schuff   uint64_t Result = decodeULEB128(Ptr, &Count);
742c6f75ddSDerek Schuff   Ptr += Count;
752c6f75ddSDerek Schuff   return Result;
762c6f75ddSDerek Schuff }
772c6f75ddSDerek Schuff 
7844d95122SEugene Zelenko static StringRef readString(const uint8_t *&Ptr) {
792c6f75ddSDerek Schuff   uint32_t StringLen = readULEB128(Ptr);
802c6f75ddSDerek Schuff   StringRef Return = StringRef(reinterpret_cast<const char *>(Ptr), StringLen);
812c6f75ddSDerek Schuff   Ptr += StringLen;
822c6f75ddSDerek Schuff   return Return;
832c6f75ddSDerek Schuff }
842c6f75ddSDerek Schuff 
85d3d84fddSDerek Schuff static int64_t readLEB128(const uint8_t *&Ptr) {
86d3d84fddSDerek Schuff   unsigned Count;
87d3d84fddSDerek Schuff   uint64_t Result = decodeSLEB128(Ptr, &Count);
88d3d84fddSDerek Schuff   Ptr += Count;
89d3d84fddSDerek Schuff   return Result;
90d3d84fddSDerek Schuff }
91d3d84fddSDerek Schuff 
92d3d84fddSDerek Schuff static uint8_t readVaruint1(const uint8_t *&Ptr) {
93d3d84fddSDerek Schuff   int64_t result = readLEB128(Ptr);
94d3d84fddSDerek Schuff   assert(result <= VARUINT1_MAX && result >= 0);
95d3d84fddSDerek Schuff   return result;
96d3d84fddSDerek Schuff }
97d3d84fddSDerek Schuff 
98d3d84fddSDerek Schuff static int8_t readVarint7(const uint8_t *&Ptr) {
99d3d84fddSDerek Schuff   int64_t result = readLEB128(Ptr);
100d3d84fddSDerek Schuff   assert(result <= VARINT7_MAX && result >= VARINT7_MIN);
101d3d84fddSDerek Schuff   return result;
102d3d84fddSDerek Schuff }
103d3d84fddSDerek Schuff 
104d3d84fddSDerek Schuff static uint8_t readVaruint7(const uint8_t *&Ptr) {
105d3d84fddSDerek Schuff   uint64_t result = readULEB128(Ptr);
10654376021SDavide Italiano   assert(result <= VARUINT7_MAX);
107d3d84fddSDerek Schuff   return result;
108d3d84fddSDerek Schuff }
109d3d84fddSDerek Schuff 
110d3d84fddSDerek Schuff static int32_t readVarint32(const uint8_t *&Ptr) {
111d3d84fddSDerek Schuff   int64_t result = readLEB128(Ptr);
112d3d84fddSDerek Schuff   assert(result <= INT32_MAX && result >= INT32_MIN);
113d3d84fddSDerek Schuff   return result;
114d3d84fddSDerek Schuff }
115d3d84fddSDerek Schuff 
116d3d84fddSDerek Schuff static uint32_t readVaruint32(const uint8_t *&Ptr) {
117d3d84fddSDerek Schuff   uint64_t result = readULEB128(Ptr);
11854376021SDavide Italiano   assert(result <= UINT32_MAX);
119d3d84fddSDerek Schuff   return result;
120d3d84fddSDerek Schuff }
121d3d84fddSDerek Schuff 
122d3d84fddSDerek Schuff static int64_t readVarint64(const uint8_t *&Ptr) {
123d3d84fddSDerek Schuff   return readLEB128(Ptr);
124d3d84fddSDerek Schuff }
125d3d84fddSDerek Schuff 
126d3d84fddSDerek Schuff static uint8_t readOpcode(const uint8_t *&Ptr) {
127d3d84fddSDerek Schuff   return readUint8(Ptr);
128d3d84fddSDerek Schuff }
129d3d84fddSDerek Schuff 
130d3d84fddSDerek Schuff static Error readInitExpr(wasm::WasmInitExpr &Expr, const uint8_t *&Ptr) {
131d3d84fddSDerek Schuff   Expr.Opcode = readOpcode(Ptr);
132d3d84fddSDerek Schuff 
133d3d84fddSDerek Schuff   switch (Expr.Opcode) {
134d3d84fddSDerek Schuff   case wasm::WASM_OPCODE_I32_CONST:
135d3d84fddSDerek Schuff     Expr.Value.Int32 = readVarint32(Ptr);
136d3d84fddSDerek Schuff     break;
137d3d84fddSDerek Schuff   case wasm::WASM_OPCODE_I64_CONST:
138d3d84fddSDerek Schuff     Expr.Value.Int64 = readVarint64(Ptr);
139d3d84fddSDerek Schuff     break;
140d3d84fddSDerek Schuff   case wasm::WASM_OPCODE_F32_CONST:
141d3d84fddSDerek Schuff     Expr.Value.Float32 = readFloat32(Ptr);
142d3d84fddSDerek Schuff     break;
143d3d84fddSDerek Schuff   case wasm::WASM_OPCODE_F64_CONST:
144d3d84fddSDerek Schuff     Expr.Value.Float64 = readFloat64(Ptr);
145d3d84fddSDerek Schuff     break;
146d3d84fddSDerek Schuff   case wasm::WASM_OPCODE_GET_GLOBAL:
1477fb391feSSam Clegg     Expr.Value.Global = readULEB128(Ptr);
148d3d84fddSDerek Schuff     break;
149d3d84fddSDerek Schuff   default:
150d3d84fddSDerek Schuff     return make_error<GenericBinaryError>("Invalid opcode in init_expr",
151d3d84fddSDerek Schuff                                           object_error::parse_failed);
152d3d84fddSDerek Schuff   }
153d3d84fddSDerek Schuff 
154d3d84fddSDerek Schuff   uint8_t EndOpcode = readOpcode(Ptr);
155d3d84fddSDerek Schuff   if (EndOpcode != wasm::WASM_OPCODE_END) {
156d3d84fddSDerek Schuff     return make_error<GenericBinaryError>("Invalid init_expr",
157d3d84fddSDerek Schuff                                           object_error::parse_failed);
158d3d84fddSDerek Schuff   }
159d3d84fddSDerek Schuff   return Error::success();
160d3d84fddSDerek Schuff }
161d3d84fddSDerek Schuff 
162d3d84fddSDerek Schuff static wasm::WasmLimits readLimits(const uint8_t *&Ptr) {
163d3d84fddSDerek Schuff   wasm::WasmLimits Result;
164d3d84fddSDerek Schuff   Result.Flags = readVaruint1(Ptr);
165d3d84fddSDerek Schuff   Result.Initial = readVaruint32(Ptr);
166d3d84fddSDerek Schuff   if (Result.Flags & wasm::WASM_LIMITS_FLAG_HAS_MAX)
167d3d84fddSDerek Schuff     Result.Maximum = readVaruint32(Ptr);
168d3d84fddSDerek Schuff   return Result;
169d3d84fddSDerek Schuff }
170d3d84fddSDerek Schuff 
171d3d84fddSDerek Schuff static Error readSection(WasmSection &Section, const uint8_t *&Ptr,
1722c6f75ddSDerek Schuff                          const uint8_t *Start) {
1732c6f75ddSDerek Schuff   // TODO(sbc): Avoid reading past EOF in the case of malformed files.
1742c6f75ddSDerek Schuff   Section.Offset = Ptr - Start;
175d3d84fddSDerek Schuff   Section.Type = readVaruint7(Ptr);
176d3d84fddSDerek Schuff   uint32_t Size = readVaruint32(Ptr);
1772c6f75ddSDerek Schuff   if (Size == 0)
1782c6f75ddSDerek Schuff     return make_error<StringError>("Zero length section",
1792c6f75ddSDerek Schuff                                    object_error::parse_failed);
1802c6f75ddSDerek Schuff   Section.Content = ArrayRef<uint8_t>(Ptr, Size);
1812c6f75ddSDerek Schuff   Ptr += Size;
1822c6f75ddSDerek Schuff   return Error::success();
1832c6f75ddSDerek Schuff }
1842c6f75ddSDerek Schuff 
1852c6f75ddSDerek Schuff WasmObjectFile::WasmObjectFile(MemoryBufferRef Buffer, Error &Err)
1869f5094dfSEugene Zelenko     : ObjectFile(Binary::ID_Wasm, Buffer) {
1872c6f75ddSDerek Schuff   ErrorAsOutParameter ErrAsOutParam(&Err);
1882c6f75ddSDerek Schuff   Header.Magic = getData().substr(0, 4);
1892c6f75ddSDerek Schuff   if (Header.Magic != StringRef("\0asm", 4)) {
1902c6f75ddSDerek Schuff     Err = make_error<StringError>("Bad magic number",
1912c6f75ddSDerek Schuff                                   object_error::parse_failed);
1922c6f75ddSDerek Schuff     return;
1932c6f75ddSDerek Schuff   }
1942c6f75ddSDerek Schuff   const uint8_t *Ptr = getPtr(4);
1952c6f75ddSDerek Schuff   Header.Version = readUint32(Ptr);
1962c6f75ddSDerek Schuff   if (Header.Version != wasm::WasmVersion) {
1972c6f75ddSDerek Schuff     Err = make_error<StringError>("Bad version number",
1982c6f75ddSDerek Schuff                                   object_error::parse_failed);
1992c6f75ddSDerek Schuff     return;
2002c6f75ddSDerek Schuff   }
2012c6f75ddSDerek Schuff 
2022c6f75ddSDerek Schuff   const uint8_t *Eof = getPtr(getData().size());
203d3d84fddSDerek Schuff   WasmSection Sec;
2042c6f75ddSDerek Schuff   while (Ptr < Eof) {
2052c6f75ddSDerek Schuff     if ((Err = readSection(Sec, Ptr, getPtr(0))))
2062c6f75ddSDerek Schuff       return;
207d3d84fddSDerek Schuff     if ((Err = parseSection(Sec)))
2082c6f75ddSDerek Schuff       return;
209d3d84fddSDerek Schuff 
2102c6f75ddSDerek Schuff     Sections.push_back(Sec);
2112c6f75ddSDerek Schuff   }
2122c6f75ddSDerek Schuff }
2132c6f75ddSDerek Schuff 
214d3d84fddSDerek Schuff Error WasmObjectFile::parseSection(WasmSection &Sec) {
215d3d84fddSDerek Schuff   const uint8_t* Start = Sec.Content.data();
216d3d84fddSDerek Schuff   const uint8_t* End = Start + Sec.Content.size();
217d3d84fddSDerek Schuff   switch (Sec.Type) {
218d3d84fddSDerek Schuff   case wasm::WASM_SEC_CUSTOM:
219d3d84fddSDerek Schuff     return parseCustomSection(Sec, Start, End);
220d3d84fddSDerek Schuff   case wasm::WASM_SEC_TYPE:
221d3d84fddSDerek Schuff     return parseTypeSection(Start, End);
222d3d84fddSDerek Schuff   case wasm::WASM_SEC_IMPORT:
223d3d84fddSDerek Schuff     return parseImportSection(Start, End);
224d3d84fddSDerek Schuff   case wasm::WASM_SEC_FUNCTION:
225d3d84fddSDerek Schuff     return parseFunctionSection(Start, End);
226d3d84fddSDerek Schuff   case wasm::WASM_SEC_TABLE:
227d3d84fddSDerek Schuff     return parseTableSection(Start, End);
228d3d84fddSDerek Schuff   case wasm::WASM_SEC_MEMORY:
229d3d84fddSDerek Schuff     return parseMemorySection(Start, End);
230d3d84fddSDerek Schuff   case wasm::WASM_SEC_GLOBAL:
231d3d84fddSDerek Schuff     return parseGlobalSection(Start, End);
232d3d84fddSDerek Schuff   case wasm::WASM_SEC_EXPORT:
233d3d84fddSDerek Schuff     return parseExportSection(Start, End);
234d3d84fddSDerek Schuff   case wasm::WASM_SEC_START:
235d3d84fddSDerek Schuff     return parseStartSection(Start, End);
236d3d84fddSDerek Schuff   case wasm::WASM_SEC_ELEM:
237d3d84fddSDerek Schuff     return parseElemSection(Start, End);
238d3d84fddSDerek Schuff   case wasm::WASM_SEC_CODE:
239d3d84fddSDerek Schuff     return parseCodeSection(Start, End);
240d3d84fddSDerek Schuff   case wasm::WASM_SEC_DATA:
241d3d84fddSDerek Schuff     return parseDataSection(Start, End);
242d3d84fddSDerek Schuff   default:
243d3d84fddSDerek Schuff     return make_error<GenericBinaryError>("Bad section type",
244d3d84fddSDerek Schuff                                           object_error::parse_failed);
245d3d84fddSDerek Schuff   }
246d3d84fddSDerek Schuff }
247d3d84fddSDerek Schuff 
248d3d84fddSDerek Schuff Error WasmObjectFile::parseNameSection(const uint8_t *Ptr, const uint8_t *End) {
249d3d84fddSDerek Schuff   while (Ptr < End) {
250d3d84fddSDerek Schuff     uint8_t Type = readVarint7(Ptr);
251d3d84fddSDerek Schuff     uint32_t Size = readVaruint32(Ptr);
252d3d84fddSDerek Schuff     switch (Type) {
253d3d84fddSDerek Schuff     case wasm::WASM_NAMES_FUNCTION: {
254d3d84fddSDerek Schuff       uint32_t Count = readVaruint32(Ptr);
255d3d84fddSDerek Schuff       while (Count--) {
256d3d84fddSDerek Schuff         /*uint32_t Index =*/readVaruint32(Ptr);
257d3d84fddSDerek Schuff         StringRef Name = readString(Ptr);
2589f5094dfSEugene Zelenko         if (!Name.empty())
259d3d84fddSDerek Schuff           Symbols.emplace_back(Name,
260d3d84fddSDerek Schuff                                WasmSymbol::SymbolType::DEBUG_FUNCTION_NAME);
261d3d84fddSDerek Schuff       }
262d3d84fddSDerek Schuff       break;
263d3d84fddSDerek Schuff     }
264d3d84fddSDerek Schuff     // Ignore local names for now
265d3d84fddSDerek Schuff     case wasm::WASM_NAMES_LOCAL:
266d3d84fddSDerek Schuff     default:
267d3d84fddSDerek Schuff       Ptr += Size;
268d3d84fddSDerek Schuff       break;
269d3d84fddSDerek Schuff     }
270d3d84fddSDerek Schuff   }
271d3d84fddSDerek Schuff 
272d3d84fddSDerek Schuff   if (Ptr != End)
273d3d84fddSDerek Schuff     return make_error<GenericBinaryError>("Name section ended prematurely",
274d3d84fddSDerek Schuff                                           object_error::parse_failed);
275d3d84fddSDerek Schuff   return Error::success();
276d3d84fddSDerek Schuff }
277d3d84fddSDerek Schuff 
278d3d84fddSDerek Schuff WasmSection* WasmObjectFile::findCustomSectionByName(StringRef Name) {
279d3d84fddSDerek Schuff   for (WasmSection& Section : Sections) {
280d3d84fddSDerek Schuff     if (Section.Type == wasm::WASM_SEC_CUSTOM && Section.Name == Name)
281d3d84fddSDerek Schuff       return &Section;
282d3d84fddSDerek Schuff   }
283d3d84fddSDerek Schuff   return nullptr;
284d3d84fddSDerek Schuff }
285d3d84fddSDerek Schuff 
286d3d84fddSDerek Schuff WasmSection* WasmObjectFile::findSectionByType(uint32_t Type) {
287d3d84fddSDerek Schuff   assert(Type != wasm::WASM_SEC_CUSTOM);
288d3d84fddSDerek Schuff   for (WasmSection& Section : Sections) {
289d3d84fddSDerek Schuff     if (Section.Type == Type)
290d3d84fddSDerek Schuff       return &Section;
291d3d84fddSDerek Schuff   }
292d3d84fddSDerek Schuff   return nullptr;
293d3d84fddSDerek Schuff }
294d3d84fddSDerek Schuff 
295d3d84fddSDerek Schuff Error WasmObjectFile::parseRelocSection(StringRef Name, const uint8_t *Ptr,
296d3d84fddSDerek Schuff                                         const uint8_t *End) {
297d3d84fddSDerek Schuff   uint8_t SectionCode = readVarint7(Ptr);
298d3d84fddSDerek Schuff   WasmSection* Section = nullptr;
299d3d84fddSDerek Schuff   if (SectionCode == wasm::WASM_SEC_CUSTOM) {
300d3d84fddSDerek Schuff     StringRef Name = readString(Ptr);
301d3d84fddSDerek Schuff     Section = findCustomSectionByName(Name);
302d3d84fddSDerek Schuff   } else {
303d3d84fddSDerek Schuff     Section = findSectionByType(SectionCode);
304d3d84fddSDerek Schuff   }
305d3d84fddSDerek Schuff   if (!Section)
306d3d84fddSDerek Schuff     return make_error<GenericBinaryError>("Invalid section code",
307d3d84fddSDerek Schuff                                           object_error::parse_failed);
308d3d84fddSDerek Schuff   uint32_t RelocCount = readVaruint32(Ptr);
309d3d84fddSDerek Schuff   while (RelocCount--) {
310d3d84fddSDerek Schuff     wasm::WasmRelocation Reloc;
311d3d84fddSDerek Schuff     memset(&Reloc, 0, sizeof(Reloc));
312d3d84fddSDerek Schuff     Reloc.Type = readVaruint32(Ptr);
313d3d84fddSDerek Schuff     Reloc.Offset = readVaruint32(Ptr);
314d3d84fddSDerek Schuff     Reloc.Index = readVaruint32(Ptr);
315d3d84fddSDerek Schuff     switch (Reloc.Type) {
316d3d84fddSDerek Schuff     case wasm::R_WEBASSEMBLY_FUNCTION_INDEX_LEB:
317d3d84fddSDerek Schuff     case wasm::R_WEBASSEMBLY_TABLE_INDEX_SLEB:
318d3d84fddSDerek Schuff     case wasm::R_WEBASSEMBLY_TABLE_INDEX_I32:
319*cc182aaaSSam Clegg     case wasm::R_WEBASSEMBLY_TYPE_INDEX_LEB:
320d3d84fddSDerek Schuff       break;
321d3d84fddSDerek Schuff     case wasm::R_WEBASSEMBLY_GLOBAL_ADDR_LEB:
322d3d84fddSDerek Schuff     case wasm::R_WEBASSEMBLY_GLOBAL_ADDR_SLEB:
323d3d84fddSDerek Schuff     case wasm::R_WEBASSEMBLY_GLOBAL_ADDR_I32:
324*cc182aaaSSam Clegg       Reloc.Addend = readVarint32(Ptr);
325d3d84fddSDerek Schuff       break;
326d3d84fddSDerek Schuff     default:
327*cc182aaaSSam Clegg       return make_error<GenericBinaryError>("Bad relocation type: " + Twine(Reloc.Type),
328d3d84fddSDerek Schuff                                             object_error::parse_failed);
329d3d84fddSDerek Schuff     }
330d3d84fddSDerek Schuff     Section->Relocations.push_back(Reloc);
331d3d84fddSDerek Schuff   }
332d3d84fddSDerek Schuff   if (Ptr != End)
333d3d84fddSDerek Schuff     return make_error<GenericBinaryError>("Reloc section ended prematurely",
334d3d84fddSDerek Schuff                                           object_error::parse_failed);
335d3d84fddSDerek Schuff   return Error::success();
336d3d84fddSDerek Schuff }
337d3d84fddSDerek Schuff 
338d3d84fddSDerek Schuff Error WasmObjectFile::parseCustomSection(WasmSection &Sec,
339d3d84fddSDerek Schuff                                          const uint8_t *Ptr, const uint8_t *End) {
3402c6f75ddSDerek Schuff   Sec.Name = readString(Ptr);
341d3d84fddSDerek Schuff   if (Sec.Name == "name") {
342d3d84fddSDerek Schuff     if (Error Err = parseNameSection(Ptr, End))
343d3d84fddSDerek Schuff       return Err;
344d3d84fddSDerek Schuff   } else if (Sec.Name.startswith("reloc.")) {
345d3d84fddSDerek Schuff     if (Error Err = parseRelocSection(Sec.Name, Ptr, End))
346d3d84fddSDerek Schuff       return Err;
347d3d84fddSDerek Schuff   }
348d3d84fddSDerek Schuff   return Error::success();
349d3d84fddSDerek Schuff }
350d3d84fddSDerek Schuff 
351d3d84fddSDerek Schuff Error WasmObjectFile::parseTypeSection(const uint8_t *Ptr, const uint8_t *End) {
352d3d84fddSDerek Schuff   uint32_t Count = readVaruint32(Ptr);
353d3d84fddSDerek Schuff   Signatures.reserve(Count);
354d3d84fddSDerek Schuff   while (Count--) {
355d3d84fddSDerek Schuff     wasm::WasmSignature Sig;
356d3d84fddSDerek Schuff     Sig.ReturnType = wasm::WASM_TYPE_NORESULT;
357d3d84fddSDerek Schuff     int8_t Form = readVarint7(Ptr);
358d3d84fddSDerek Schuff     if (Form != wasm::WASM_TYPE_FUNC) {
359d3d84fddSDerek Schuff       return make_error<GenericBinaryError>("Invalid signature type",
360d3d84fddSDerek Schuff                                             object_error::parse_failed);
361d3d84fddSDerek Schuff     }
362d3d84fddSDerek Schuff     uint32_t ParamCount = readVaruint32(Ptr);
363d3d84fddSDerek Schuff     Sig.ParamTypes.reserve(ParamCount);
364d3d84fddSDerek Schuff     while (ParamCount--) {
365d3d84fddSDerek Schuff       uint32_t ParamType = readVarint7(Ptr);
366d3d84fddSDerek Schuff       Sig.ParamTypes.push_back(ParamType);
367d3d84fddSDerek Schuff     }
368d3d84fddSDerek Schuff     uint32_t ReturnCount = readVaruint32(Ptr);
369d3d84fddSDerek Schuff     if (ReturnCount) {
370d3d84fddSDerek Schuff       if (ReturnCount != 1) {
371d3d84fddSDerek Schuff         return make_error<GenericBinaryError>(
372d3d84fddSDerek Schuff             "Multiple return types not supported", object_error::parse_failed);
373d3d84fddSDerek Schuff       }
374d3d84fddSDerek Schuff       Sig.ReturnType = readVarint7(Ptr);
375d3d84fddSDerek Schuff     }
376d3d84fddSDerek Schuff     Signatures.push_back(Sig);
377d3d84fddSDerek Schuff   }
378d3d84fddSDerek Schuff   if (Ptr != End)
379d3d84fddSDerek Schuff     return make_error<GenericBinaryError>("Type section ended prematurely",
380d3d84fddSDerek Schuff                                           object_error::parse_failed);
381d3d84fddSDerek Schuff   return Error::success();
382d3d84fddSDerek Schuff }
383d3d84fddSDerek Schuff 
384d3d84fddSDerek Schuff Error WasmObjectFile::parseImportSection(const uint8_t *Ptr, const uint8_t *End) {
385d3d84fddSDerek Schuff   uint32_t Count = readVaruint32(Ptr);
386d3d84fddSDerek Schuff   Imports.reserve(Count);
387d3d84fddSDerek Schuff   while (Count--) {
388d3d84fddSDerek Schuff     wasm::WasmImport Im;
389d3d84fddSDerek Schuff     Im.Module = readString(Ptr);
390d3d84fddSDerek Schuff     Im.Field = readString(Ptr);
391d3d84fddSDerek Schuff     Im.Kind = readUint8(Ptr);
392d3d84fddSDerek Schuff     switch (Im.Kind) {
393d3d84fddSDerek Schuff     case wasm::WASM_EXTERNAL_FUNCTION:
394d3d84fddSDerek Schuff       Im.SigIndex = readVaruint32(Ptr);
395d3d84fddSDerek Schuff       Symbols.emplace_back(Im.Field, WasmSymbol::SymbolType::FUNCTION_IMPORT);
396d3d84fddSDerek Schuff       break;
397d3d84fddSDerek Schuff     case wasm::WASM_EXTERNAL_GLOBAL:
398d3d84fddSDerek Schuff       Im.GlobalType = readVarint7(Ptr);
399d3d84fddSDerek Schuff       Im.GlobalMutable = readVaruint1(Ptr);
400d3d84fddSDerek Schuff       Symbols.emplace_back(Im.Field, WasmSymbol::SymbolType::GLOBAL_IMPORT);
401d3d84fddSDerek Schuff       break;
402d3d84fddSDerek Schuff     default:
403d3d84fddSDerek Schuff       // TODO(sbc): Handle other kinds of imports
404d3d84fddSDerek Schuff       return make_error<GenericBinaryError>(
405d3d84fddSDerek Schuff           "Unexpected import kind", object_error::parse_failed);
406d3d84fddSDerek Schuff     }
407d3d84fddSDerek Schuff     Imports.push_back(Im);
408d3d84fddSDerek Schuff   }
409d3d84fddSDerek Schuff   if (Ptr != End)
410d3d84fddSDerek Schuff     return make_error<GenericBinaryError>("Import section ended prematurely",
411d3d84fddSDerek Schuff                                           object_error::parse_failed);
412d3d84fddSDerek Schuff   return Error::success();
413d3d84fddSDerek Schuff }
414d3d84fddSDerek Schuff 
415d3d84fddSDerek Schuff Error WasmObjectFile::parseFunctionSection(const uint8_t *Ptr, const uint8_t *End) {
416d3d84fddSDerek Schuff   uint32_t Count = readVaruint32(Ptr);
417d3d84fddSDerek Schuff   FunctionTypes.reserve(Count);
418d3d84fddSDerek Schuff   while (Count--) {
419d3d84fddSDerek Schuff     FunctionTypes.push_back(readVaruint32(Ptr));
420d3d84fddSDerek Schuff   }
421d3d84fddSDerek Schuff   if (Ptr != End)
422d3d84fddSDerek Schuff     return make_error<GenericBinaryError>("Function section ended prematurely",
423d3d84fddSDerek Schuff                                           object_error::parse_failed);
424d3d84fddSDerek Schuff   return Error::success();
425d3d84fddSDerek Schuff }
426d3d84fddSDerek Schuff 
427d3d84fddSDerek Schuff Error WasmObjectFile::parseTableSection(const uint8_t *Ptr, const uint8_t *End) {
428d3d84fddSDerek Schuff   uint32_t Count = readVaruint32(Ptr);
429d3d84fddSDerek Schuff   Tables.reserve(Count);
430d3d84fddSDerek Schuff   while (Count--) {
431d3d84fddSDerek Schuff     wasm::WasmTable Table;
432d3d84fddSDerek Schuff     Table.ElemType = readVarint7(Ptr);
433d3d84fddSDerek Schuff     if (Table.ElemType != wasm::WASM_TYPE_ANYFUNC) {
434d3d84fddSDerek Schuff       return make_error<GenericBinaryError>("Invalid table element type",
435d3d84fddSDerek Schuff                                             object_error::parse_failed);
436d3d84fddSDerek Schuff     }
437d3d84fddSDerek Schuff     Table.Limits = readLimits(Ptr);
438d3d84fddSDerek Schuff     Tables.push_back(Table);
439d3d84fddSDerek Schuff   }
440d3d84fddSDerek Schuff   if (Ptr != End)
441d3d84fddSDerek Schuff     return make_error<GenericBinaryError>("Table section ended prematurely",
442d3d84fddSDerek Schuff                                           object_error::parse_failed);
443d3d84fddSDerek Schuff   return Error::success();
444d3d84fddSDerek Schuff }
445d3d84fddSDerek Schuff 
446d3d84fddSDerek Schuff Error WasmObjectFile::parseMemorySection(const uint8_t *Ptr, const uint8_t *End) {
447d3d84fddSDerek Schuff   uint32_t Count = readVaruint32(Ptr);
448d3d84fddSDerek Schuff   Memories.reserve(Count);
449d3d84fddSDerek Schuff   while (Count--) {
450d3d84fddSDerek Schuff     Memories.push_back(readLimits(Ptr));
451d3d84fddSDerek Schuff   }
452d3d84fddSDerek Schuff   if (Ptr != End)
453d3d84fddSDerek Schuff     return make_error<GenericBinaryError>("Memory section ended prematurely",
454d3d84fddSDerek Schuff                                           object_error::parse_failed);
455d3d84fddSDerek Schuff   return Error::success();
456d3d84fddSDerek Schuff }
457d3d84fddSDerek Schuff 
458d3d84fddSDerek Schuff Error WasmObjectFile::parseGlobalSection(const uint8_t *Ptr, const uint8_t *End) {
459d3d84fddSDerek Schuff   uint32_t Count = readVaruint32(Ptr);
460d3d84fddSDerek Schuff   Globals.reserve(Count);
461d3d84fddSDerek Schuff   while (Count--) {
462d3d84fddSDerek Schuff     wasm::WasmGlobal Global;
463d3d84fddSDerek Schuff     Global.Type = readVarint7(Ptr);
464d3d84fddSDerek Schuff     Global.Mutable = readVaruint1(Ptr);
465deede839SDavide Italiano     if (Error Err = readInitExpr(Global.InitExpr, Ptr))
466d3d84fddSDerek Schuff       return Err;
467d3d84fddSDerek Schuff     Globals.push_back(Global);
468d3d84fddSDerek Schuff   }
469d3d84fddSDerek Schuff   if (Ptr != End)
470d3d84fddSDerek Schuff     return make_error<GenericBinaryError>("Global section ended prematurely",
471d3d84fddSDerek Schuff                                           object_error::parse_failed);
472d3d84fddSDerek Schuff   return Error::success();
473d3d84fddSDerek Schuff }
474d3d84fddSDerek Schuff 
475d3d84fddSDerek Schuff Error WasmObjectFile::parseExportSection(const uint8_t *Ptr, const uint8_t *End) {
476d3d84fddSDerek Schuff   uint32_t Count = readVaruint32(Ptr);
477d3d84fddSDerek Schuff   Exports.reserve(Count);
478d3d84fddSDerek Schuff   while (Count--) {
479d3d84fddSDerek Schuff     wasm::WasmExport Ex;
480d3d84fddSDerek Schuff     Ex.Name = readString(Ptr);
481d3d84fddSDerek Schuff     Ex.Kind = readUint8(Ptr);
482d3d84fddSDerek Schuff     Ex.Index = readVaruint32(Ptr);
483d3d84fddSDerek Schuff     Exports.push_back(Ex);
484d3d84fddSDerek Schuff     switch (Ex.Kind) {
485d3d84fddSDerek Schuff     case wasm::WASM_EXTERNAL_FUNCTION:
486d3d84fddSDerek Schuff       Symbols.emplace_back(Ex.Name, WasmSymbol::SymbolType::FUNCTION_EXPORT);
487d3d84fddSDerek Schuff       break;
488d3d84fddSDerek Schuff     case wasm::WASM_EXTERNAL_GLOBAL:
489d3d84fddSDerek Schuff       Symbols.emplace_back(Ex.Name, WasmSymbol::SymbolType::GLOBAL_EXPORT);
490d3d84fddSDerek Schuff       break;
491d3d84fddSDerek Schuff     default:
492d3d84fddSDerek Schuff       // TODO(sbc): Handle other kinds of exports
493d3d84fddSDerek Schuff       return make_error<GenericBinaryError>(
494d3d84fddSDerek Schuff           "Unexpected export kind", object_error::parse_failed);
495d3d84fddSDerek Schuff     }
496d3d84fddSDerek Schuff   }
497d3d84fddSDerek Schuff   if (Ptr != End)
498d3d84fddSDerek Schuff     return make_error<GenericBinaryError>("Export section ended prematurely",
499d3d84fddSDerek Schuff                                           object_error::parse_failed);
500d3d84fddSDerek Schuff   return Error::success();
501d3d84fddSDerek Schuff }
502d3d84fddSDerek Schuff 
503d3d84fddSDerek Schuff Error WasmObjectFile::parseStartSection(const uint8_t *Ptr, const uint8_t *End) {
504d3d84fddSDerek Schuff   StartFunction = readVaruint32(Ptr);
505d3d84fddSDerek Schuff   if (StartFunction < FunctionTypes.size())
506d3d84fddSDerek Schuff     return make_error<GenericBinaryError>("Invalid start function",
507d3d84fddSDerek Schuff                                           object_error::parse_failed);
508d3d84fddSDerek Schuff   return Error::success();
509d3d84fddSDerek Schuff }
510d3d84fddSDerek Schuff 
511d3d84fddSDerek Schuff Error WasmObjectFile::parseCodeSection(const uint8_t *Ptr, const uint8_t *End) {
512d3d84fddSDerek Schuff   uint32_t FunctionCount = readVaruint32(Ptr);
513d3d84fddSDerek Schuff   if (FunctionCount != FunctionTypes.size()) {
514d3d84fddSDerek Schuff     return make_error<GenericBinaryError>("Invalid function count",
515d3d84fddSDerek Schuff                                           object_error::parse_failed);
516d3d84fddSDerek Schuff   }
517d3d84fddSDerek Schuff 
518d3d84fddSDerek Schuff   CodeSection = ArrayRef<uint8_t>(Ptr, End - Ptr);
519d3d84fddSDerek Schuff 
520d3d84fddSDerek Schuff   while (FunctionCount--) {
521d3d84fddSDerek Schuff     wasm::WasmFunction Function;
522d3d84fddSDerek Schuff     uint32_t FunctionSize = readVaruint32(Ptr);
523d3d84fddSDerek Schuff     const uint8_t *FunctionEnd = Ptr + FunctionSize;
524d3d84fddSDerek Schuff 
525d3d84fddSDerek Schuff     uint32_t NumLocalDecls = readVaruint32(Ptr);
526d3d84fddSDerek Schuff     Function.Locals.reserve(NumLocalDecls);
527d3d84fddSDerek Schuff     while (NumLocalDecls--) {
528d3d84fddSDerek Schuff       wasm::WasmLocalDecl Decl;
529d3d84fddSDerek Schuff       Decl.Count = readVaruint32(Ptr);
530d3d84fddSDerek Schuff       Decl.Type = readVarint7(Ptr);
531d3d84fddSDerek Schuff       Function.Locals.push_back(Decl);
532d3d84fddSDerek Schuff     }
533d3d84fddSDerek Schuff 
534d3d84fddSDerek Schuff     uint32_t BodySize = FunctionEnd - Ptr;
535d3d84fddSDerek Schuff     Function.Body = ArrayRef<uint8_t>(Ptr, BodySize);
536d3d84fddSDerek Schuff     Ptr += BodySize;
537d3d84fddSDerek Schuff     assert(Ptr == FunctionEnd);
538d3d84fddSDerek Schuff     Functions.push_back(Function);
539d3d84fddSDerek Schuff   }
540d3d84fddSDerek Schuff   if (Ptr != End)
541d3d84fddSDerek Schuff     return make_error<GenericBinaryError>("Code section ended prematurely",
542d3d84fddSDerek Schuff                                           object_error::parse_failed);
543d3d84fddSDerek Schuff   return Error::success();
544d3d84fddSDerek Schuff }
545d3d84fddSDerek Schuff 
546d3d84fddSDerek Schuff Error WasmObjectFile::parseElemSection(const uint8_t *Ptr, const uint8_t *End) {
547d3d84fddSDerek Schuff   uint32_t Count = readVaruint32(Ptr);
548d3d84fddSDerek Schuff   ElemSegments.reserve(Count);
549d3d84fddSDerek Schuff   while (Count--) {
550d3d84fddSDerek Schuff     wasm::WasmElemSegment Segment;
551d3d84fddSDerek Schuff     Segment.TableIndex = readVaruint32(Ptr);
552d3d84fddSDerek Schuff     if (Segment.TableIndex != 0) {
553d3d84fddSDerek Schuff       return make_error<GenericBinaryError>("Invalid TableIndex",
554d3d84fddSDerek Schuff                                             object_error::parse_failed);
555d3d84fddSDerek Schuff     }
556d3d84fddSDerek Schuff     if (Error Err = readInitExpr(Segment.Offset, Ptr))
557d3d84fddSDerek Schuff       return Err;
558d3d84fddSDerek Schuff     uint32_t NumElems = readVaruint32(Ptr);
559d3d84fddSDerek Schuff     while (NumElems--) {
560d3d84fddSDerek Schuff       Segment.Functions.push_back(readVaruint32(Ptr));
561d3d84fddSDerek Schuff     }
562d3d84fddSDerek Schuff     ElemSegments.push_back(Segment);
563d3d84fddSDerek Schuff   }
564d3d84fddSDerek Schuff   if (Ptr != End)
565d3d84fddSDerek Schuff     return make_error<GenericBinaryError>("Elem section ended prematurely",
566d3d84fddSDerek Schuff                                           object_error::parse_failed);
567d3d84fddSDerek Schuff   return Error::success();
568d3d84fddSDerek Schuff }
569d3d84fddSDerek Schuff 
570d3d84fddSDerek Schuff Error WasmObjectFile::parseDataSection(const uint8_t *Ptr, const uint8_t *End) {
571d3d84fddSDerek Schuff   uint32_t Count = readVaruint32(Ptr);
572d3d84fddSDerek Schuff   DataSegments.reserve(Count);
573d3d84fddSDerek Schuff   while (Count--) {
574d3d84fddSDerek Schuff     wasm::WasmDataSegment Segment;
575d3d84fddSDerek Schuff     Segment.Index = readVaruint32(Ptr);
576d3d84fddSDerek Schuff     if (Error Err = readInitExpr(Segment.Offset, Ptr))
577d3d84fddSDerek Schuff       return Err;
578d3d84fddSDerek Schuff     uint32_t Size = readVaruint32(Ptr);
579d3d84fddSDerek Schuff     Segment.Content = ArrayRef<uint8_t>(Ptr, Size);
580d3d84fddSDerek Schuff     Ptr += Size;
581d3d84fddSDerek Schuff     DataSegments.push_back(Segment);
582d3d84fddSDerek Schuff   }
583d3d84fddSDerek Schuff   if (Ptr != End)
584d3d84fddSDerek Schuff     return make_error<GenericBinaryError>("Data section ended prematurely",
585d3d84fddSDerek Schuff                                           object_error::parse_failed);
5862c6f75ddSDerek Schuff   return Error::success();
5872c6f75ddSDerek Schuff }
5882c6f75ddSDerek Schuff 
5892c6f75ddSDerek Schuff const uint8_t *WasmObjectFile::getPtr(size_t Offset) const {
5902c6f75ddSDerek Schuff   return reinterpret_cast<const uint8_t *>(getData().substr(Offset, 1).data());
5912c6f75ddSDerek Schuff }
5922c6f75ddSDerek Schuff 
5932c6f75ddSDerek Schuff const wasm::WasmObjectHeader &WasmObjectFile::getHeader() const {
5942c6f75ddSDerek Schuff   return Header;
5952c6f75ddSDerek Schuff }
5962c6f75ddSDerek Schuff 
597d3d84fddSDerek Schuff void WasmObjectFile::moveSymbolNext(DataRefImpl &Symb) const { Symb.d.a++; }
5982c6f75ddSDerek Schuff 
5992c6f75ddSDerek Schuff uint32_t WasmObjectFile::getSymbolFlags(DataRefImpl Symb) const {
600d3d84fddSDerek Schuff   const WasmSymbol &Sym = getWasmSymbol(Symb);
601d3d84fddSDerek Schuff   switch (Sym.Type) {
602d3d84fddSDerek Schuff   case WasmSymbol::SymbolType::FUNCTION_IMPORT:
603d3d84fddSDerek Schuff     return object::SymbolRef::SF_Undefined | SymbolRef::SF_Executable;
604d3d84fddSDerek Schuff   case WasmSymbol::SymbolType::FUNCTION_EXPORT:
605d3d84fddSDerek Schuff     return object::SymbolRef::SF_Global | SymbolRef::SF_Executable;
606d3d84fddSDerek Schuff   case WasmSymbol::SymbolType::DEBUG_FUNCTION_NAME:
607d3d84fddSDerek Schuff     return object::SymbolRef::SF_Executable;
608d3d84fddSDerek Schuff   case WasmSymbol::SymbolType::GLOBAL_IMPORT:
609d3d84fddSDerek Schuff     return object::SymbolRef::SF_Undefined;
610d3d84fddSDerek Schuff   case WasmSymbol::SymbolType::GLOBAL_EXPORT:
611d3d84fddSDerek Schuff     return object::SymbolRef::SF_Global;
612d3d84fddSDerek Schuff   }
613e32a833aSSimon Pilgrim   llvm_unreachable("Unknown WasmSymbol::SymbolType");
6142c6f75ddSDerek Schuff }
6152c6f75ddSDerek Schuff 
6162c6f75ddSDerek Schuff basic_symbol_iterator WasmObjectFile::symbol_begin() const {
617d3d84fddSDerek Schuff   DataRefImpl Ref;
618d3d84fddSDerek Schuff   Ref.d.a = 0;
619d3d84fddSDerek Schuff   return BasicSymbolRef(Ref, this);
6202c6f75ddSDerek Schuff }
6212c6f75ddSDerek Schuff 
6222c6f75ddSDerek Schuff basic_symbol_iterator WasmObjectFile::symbol_end() const {
623d3d84fddSDerek Schuff   DataRefImpl Ref;
624d3d84fddSDerek Schuff   Ref.d.a = Symbols.size();
625d3d84fddSDerek Schuff   return BasicSymbolRef(Ref, this);
626d3d84fddSDerek Schuff }
627d3d84fddSDerek Schuff 
628d3d84fddSDerek Schuff const WasmSymbol &WasmObjectFile::getWasmSymbol(DataRefImpl Symb) const {
629d3d84fddSDerek Schuff   return Symbols[Symb.d.a];
6302c6f75ddSDerek Schuff }
6312c6f75ddSDerek Schuff 
6322c6f75ddSDerek Schuff Expected<StringRef> WasmObjectFile::getSymbolName(DataRefImpl Symb) const {
633d3d84fddSDerek Schuff   const WasmSymbol &Sym = getWasmSymbol(Symb);
634d3d84fddSDerek Schuff   return Sym.Name;
6352c6f75ddSDerek Schuff }
6362c6f75ddSDerek Schuff 
6372c6f75ddSDerek Schuff Expected<uint64_t> WasmObjectFile::getSymbolAddress(DataRefImpl Symb) const {
638d3d84fddSDerek Schuff   return (uint64_t)Symb.d.a;
6392c6f75ddSDerek Schuff }
6402c6f75ddSDerek Schuff 
6412c6f75ddSDerek Schuff uint64_t WasmObjectFile::getSymbolValueImpl(DataRefImpl Symb) const {
6422c6f75ddSDerek Schuff   llvm_unreachable("not yet implemented");
6432c6f75ddSDerek Schuff   return 0;
6442c6f75ddSDerek Schuff }
6452c6f75ddSDerek Schuff 
6462c6f75ddSDerek Schuff uint32_t WasmObjectFile::getSymbolAlignment(DataRefImpl Symb) const {
6472c6f75ddSDerek Schuff   llvm_unreachable("not yet implemented");
6482c6f75ddSDerek Schuff   return 0;
6492c6f75ddSDerek Schuff }
6502c6f75ddSDerek Schuff 
6512c6f75ddSDerek Schuff uint64_t WasmObjectFile::getCommonSymbolSizeImpl(DataRefImpl Symb) const {
6522c6f75ddSDerek Schuff   llvm_unreachable("not yet implemented");
6532c6f75ddSDerek Schuff   return 0;
6542c6f75ddSDerek Schuff }
6552c6f75ddSDerek Schuff 
6562c6f75ddSDerek Schuff Expected<SymbolRef::Type>
6572c6f75ddSDerek Schuff WasmObjectFile::getSymbolType(DataRefImpl Symb) const {
6582c6f75ddSDerek Schuff   llvm_unreachable("not yet implemented");
6592c6f75ddSDerek Schuff   return errorCodeToError(object_error::invalid_symbol_index);
6602c6f75ddSDerek Schuff }
6612c6f75ddSDerek Schuff 
6622c6f75ddSDerek Schuff Expected<section_iterator>
6632c6f75ddSDerek Schuff WasmObjectFile::getSymbolSection(DataRefImpl Symb) const {
6642c6f75ddSDerek Schuff   llvm_unreachable("not yet implemented");
6652c6f75ddSDerek Schuff   return errorCodeToError(object_error::invalid_symbol_index);
6662c6f75ddSDerek Schuff }
6672c6f75ddSDerek Schuff 
6682c6f75ddSDerek Schuff void WasmObjectFile::moveSectionNext(DataRefImpl &Sec) const { Sec.d.a++; }
6692c6f75ddSDerek Schuff 
6702c6f75ddSDerek Schuff std::error_code WasmObjectFile::getSectionName(DataRefImpl Sec,
6712c6f75ddSDerek Schuff                                                StringRef &Res) const {
672d3d84fddSDerek Schuff   const WasmSection &S = Sections[Sec.d.a];
6732c6f75ddSDerek Schuff #define ECase(X)                                                               \
6742c6f75ddSDerek Schuff   case wasm::WASM_SEC_##X:                                                     \
6752c6f75ddSDerek Schuff     Res = #X;                                                                  \
6762c6f75ddSDerek Schuff     break
6772c6f75ddSDerek Schuff   switch (S.Type) {
6782c6f75ddSDerek Schuff     ECase(TYPE);
6792c6f75ddSDerek Schuff     ECase(IMPORT);
6802c6f75ddSDerek Schuff     ECase(FUNCTION);
6812c6f75ddSDerek Schuff     ECase(TABLE);
6822c6f75ddSDerek Schuff     ECase(MEMORY);
6832c6f75ddSDerek Schuff     ECase(GLOBAL);
6842c6f75ddSDerek Schuff     ECase(EXPORT);
6852c6f75ddSDerek Schuff     ECase(START);
6862c6f75ddSDerek Schuff     ECase(ELEM);
6872c6f75ddSDerek Schuff     ECase(CODE);
6882c6f75ddSDerek Schuff     ECase(DATA);
6896d76b7b4SDerek Schuff   case wasm::WASM_SEC_CUSTOM:
6902c6f75ddSDerek Schuff     Res = S.Name;
6912c6f75ddSDerek Schuff     break;
6922c6f75ddSDerek Schuff   default:
6932c6f75ddSDerek Schuff     return object_error::invalid_section_index;
6942c6f75ddSDerek Schuff   }
6952c6f75ddSDerek Schuff #undef ECase
6962c6f75ddSDerek Schuff   return std::error_code();
6972c6f75ddSDerek Schuff }
6982c6f75ddSDerek Schuff 
6992c6f75ddSDerek Schuff uint64_t WasmObjectFile::getSectionAddress(DataRefImpl Sec) const { return 0; }
7002c6f75ddSDerek Schuff 
7012c6f75ddSDerek Schuff uint64_t WasmObjectFile::getSectionSize(DataRefImpl Sec) const {
702d3d84fddSDerek Schuff   const WasmSection &S = Sections[Sec.d.a];
7032c6f75ddSDerek Schuff   return S.Content.size();
7042c6f75ddSDerek Schuff }
7052c6f75ddSDerek Schuff 
7062c6f75ddSDerek Schuff std::error_code WasmObjectFile::getSectionContents(DataRefImpl Sec,
7072c6f75ddSDerek Schuff                                                    StringRef &Res) const {
708d3d84fddSDerek Schuff   const WasmSection &S = Sections[Sec.d.a];
7092c6f75ddSDerek Schuff   // This will never fail since wasm sections can never be empty (user-sections
7102c6f75ddSDerek Schuff   // must have a name and non-user sections each have a defined structure).
7112c6f75ddSDerek Schuff   Res = StringRef(reinterpret_cast<const char *>(S.Content.data()),
7122c6f75ddSDerek Schuff                   S.Content.size());
7132c6f75ddSDerek Schuff   return std::error_code();
7142c6f75ddSDerek Schuff }
7152c6f75ddSDerek Schuff 
7162c6f75ddSDerek Schuff uint64_t WasmObjectFile::getSectionAlignment(DataRefImpl Sec) const {
7172c6f75ddSDerek Schuff   return 1;
7182c6f75ddSDerek Schuff }
7192c6f75ddSDerek Schuff 
7202c6f75ddSDerek Schuff bool WasmObjectFile::isSectionCompressed(DataRefImpl Sec) const {
7212c6f75ddSDerek Schuff   return false;
7222c6f75ddSDerek Schuff }
7232c6f75ddSDerek Schuff 
7242c6f75ddSDerek Schuff bool WasmObjectFile::isSectionText(DataRefImpl Sec) const {
725d3d84fddSDerek Schuff   return getWasmSection(Sec).Type == wasm::WASM_SEC_CODE;
7262c6f75ddSDerek Schuff }
7272c6f75ddSDerek Schuff 
7282c6f75ddSDerek Schuff bool WasmObjectFile::isSectionData(DataRefImpl Sec) const {
729d3d84fddSDerek Schuff   return getWasmSection(Sec).Type == wasm::WASM_SEC_DATA;
7302c6f75ddSDerek Schuff }
7312c6f75ddSDerek Schuff 
7322c6f75ddSDerek Schuff bool WasmObjectFile::isSectionBSS(DataRefImpl Sec) const { return false; }
7332c6f75ddSDerek Schuff 
7342c6f75ddSDerek Schuff bool WasmObjectFile::isSectionVirtual(DataRefImpl Sec) const { return false; }
7352c6f75ddSDerek Schuff 
7362c6f75ddSDerek Schuff bool WasmObjectFile::isSectionBitcode(DataRefImpl Sec) const { return false; }
7372c6f75ddSDerek Schuff 
738d3d84fddSDerek Schuff relocation_iterator WasmObjectFile::section_rel_begin(DataRefImpl Ref) const {
739d3d84fddSDerek Schuff   DataRefImpl RelocRef;
740d3d84fddSDerek Schuff   RelocRef.d.a = Ref.d.a;
741d3d84fddSDerek Schuff   RelocRef.d.b = 0;
742d3d84fddSDerek Schuff   return relocation_iterator(RelocationRef(RelocRef, this));
7432c6f75ddSDerek Schuff }
7442c6f75ddSDerek Schuff 
745d3d84fddSDerek Schuff relocation_iterator WasmObjectFile::section_rel_end(DataRefImpl Ref) const {
746d3d84fddSDerek Schuff   const WasmSection &Sec = getWasmSection(Ref);
747d3d84fddSDerek Schuff   DataRefImpl RelocRef;
748d3d84fddSDerek Schuff   RelocRef.d.a = Ref.d.a;
749d3d84fddSDerek Schuff   RelocRef.d.b = Sec.Relocations.size();
750d3d84fddSDerek Schuff   return relocation_iterator(RelocationRef(RelocRef, this));
7512c6f75ddSDerek Schuff }
7522c6f75ddSDerek Schuff 
7532c6f75ddSDerek Schuff void WasmObjectFile::moveRelocationNext(DataRefImpl &Rel) const {
754d3d84fddSDerek Schuff   Rel.d.b++;
7552c6f75ddSDerek Schuff }
7562c6f75ddSDerek Schuff 
757d3d84fddSDerek Schuff uint64_t WasmObjectFile::getRelocationOffset(DataRefImpl Ref) const {
758d3d84fddSDerek Schuff   const wasm::WasmRelocation &Rel = getWasmRelocation(Ref);
759d3d84fddSDerek Schuff   return Rel.Offset;
7602c6f75ddSDerek Schuff }
7612c6f75ddSDerek Schuff 
7622c6f75ddSDerek Schuff symbol_iterator WasmObjectFile::getRelocationSymbol(DataRefImpl Rel) const {
7632c6f75ddSDerek Schuff   llvm_unreachable("not yet implemented");
7642c6f75ddSDerek Schuff   SymbolRef Ref;
7652c6f75ddSDerek Schuff   return symbol_iterator(Ref);
7662c6f75ddSDerek Schuff }
7672c6f75ddSDerek Schuff 
768d3d84fddSDerek Schuff uint64_t WasmObjectFile::getRelocationType(DataRefImpl Ref) const {
769d3d84fddSDerek Schuff   const wasm::WasmRelocation &Rel = getWasmRelocation(Ref);
770d3d84fddSDerek Schuff   return Rel.Type;
7712c6f75ddSDerek Schuff }
7722c6f75ddSDerek Schuff 
7732c6f75ddSDerek Schuff void WasmObjectFile::getRelocationTypeName(
774d3d84fddSDerek Schuff     DataRefImpl Ref, SmallVectorImpl<char> &Result) const {
775d3d84fddSDerek Schuff   const wasm::WasmRelocation& Rel = getWasmRelocation(Ref);
776d3d84fddSDerek Schuff   StringRef Res = "Unknown";
777d3d84fddSDerek Schuff 
778d3d84fddSDerek Schuff #define WASM_RELOC(name, value)  \
779d3d84fddSDerek Schuff   case wasm::name:              \
780d3d84fddSDerek Schuff     Res = #name;               \
781d3d84fddSDerek Schuff     break;
782d3d84fddSDerek Schuff 
783d3d84fddSDerek Schuff   switch (Rel.Type) {
784d3d84fddSDerek Schuff #include "llvm/Support/WasmRelocs/WebAssembly.def"
785d3d84fddSDerek Schuff   }
786d3d84fddSDerek Schuff 
787d3d84fddSDerek Schuff #undef WASM_RELOC
788d3d84fddSDerek Schuff 
789d3d84fddSDerek Schuff   Result.append(Res.begin(), Res.end());
7902c6f75ddSDerek Schuff }
7912c6f75ddSDerek Schuff 
7922c6f75ddSDerek Schuff section_iterator WasmObjectFile::section_begin() const {
7932c6f75ddSDerek Schuff   DataRefImpl Ref;
7942c6f75ddSDerek Schuff   Ref.d.a = 0;
7952c6f75ddSDerek Schuff   return section_iterator(SectionRef(Ref, this));
7962c6f75ddSDerek Schuff }
7972c6f75ddSDerek Schuff 
7982c6f75ddSDerek Schuff section_iterator WasmObjectFile::section_end() const {
7992c6f75ddSDerek Schuff   DataRefImpl Ref;
8002c6f75ddSDerek Schuff   Ref.d.a = Sections.size();
8012c6f75ddSDerek Schuff   return section_iterator(SectionRef(Ref, this));
8022c6f75ddSDerek Schuff }
8032c6f75ddSDerek Schuff 
8042c6f75ddSDerek Schuff uint8_t WasmObjectFile::getBytesInAddress() const { return 4; }
8052c6f75ddSDerek Schuff 
8062c6f75ddSDerek Schuff StringRef WasmObjectFile::getFileFormatName() const { return "WASM"; }
8072c6f75ddSDerek Schuff 
8082c6f75ddSDerek Schuff unsigned WasmObjectFile::getArch() const { return Triple::wasm32; }
8092c6f75ddSDerek Schuff 
8102c6f75ddSDerek Schuff SubtargetFeatures WasmObjectFile::getFeatures() const {
8112c6f75ddSDerek Schuff   return SubtargetFeatures();
8122c6f75ddSDerek Schuff }
8132c6f75ddSDerek Schuff 
8142c6f75ddSDerek Schuff bool WasmObjectFile::isRelocatableObject() const { return false; }
8152c6f75ddSDerek Schuff 
816d3d84fddSDerek Schuff const WasmSection &WasmObjectFile::getWasmSection(DataRefImpl Ref) const {
81716fe5822SDavide Italiano   assert(Ref.d.a < Sections.size());
818d3d84fddSDerek Schuff   return Sections[Ref.d.a];
819d3d84fddSDerek Schuff }
820d3d84fddSDerek Schuff 
821d3d84fddSDerek Schuff const WasmSection &
8222c6f75ddSDerek Schuff WasmObjectFile::getWasmSection(const SectionRef &Section) const {
823d3d84fddSDerek Schuff   return getWasmSection(Section.getRawDataRefImpl());
824d3d84fddSDerek Schuff }
825d3d84fddSDerek Schuff 
826d3d84fddSDerek Schuff const wasm::WasmRelocation &
827d3d84fddSDerek Schuff WasmObjectFile::getWasmRelocation(const RelocationRef &Ref) const {
828d3d84fddSDerek Schuff   return getWasmRelocation(Ref.getRawDataRefImpl());
829d3d84fddSDerek Schuff }
830d3d84fddSDerek Schuff 
831d3d84fddSDerek Schuff const wasm::WasmRelocation &
832d3d84fddSDerek Schuff WasmObjectFile::getWasmRelocation(DataRefImpl Ref) const {
83316fe5822SDavide Italiano   assert(Ref.d.a < Sections.size());
834d3d84fddSDerek Schuff   const WasmSection& Sec = Sections[Ref.d.a];
83516fe5822SDavide Italiano   assert(Ref.d.b < Sec.Relocations.size());
836d3d84fddSDerek Schuff   return Sec.Relocations[Ref.d.b];
8372c6f75ddSDerek Schuff }
838