1 //===- WasmObjectFile.cpp - Wasm object file implementation ---------------===// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is distributed under the University of Illinois Open Source 6 // License. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 10 #include "llvm/ADT/ArrayRef.h" 11 #include "llvm/ADT/DenseSet.h" 12 #include "llvm/ADT/STLExtras.h" 13 #include "llvm/ADT/StringRef.h" 14 #include "llvm/ADT/StringSet.h" 15 #include "llvm/ADT/Triple.h" 16 #include "llvm/BinaryFormat/Wasm.h" 17 #include "llvm/MC/SubtargetFeature.h" 18 #include "llvm/Object/Binary.h" 19 #include "llvm/Object/Error.h" 20 #include "llvm/Object/ObjectFile.h" 21 #include "llvm/Object/SymbolicFile.h" 22 #include "llvm/Object/Wasm.h" 23 #include "llvm/Support/Endian.h" 24 #include "llvm/Support/Error.h" 25 #include "llvm/Support/ErrorHandling.h" 26 #include "llvm/Support/LEB128.h" 27 #include <algorithm> 28 #include <cassert> 29 #include <cstdint> 30 #include <cstring> 31 #include <system_error> 32 33 #define DEBUG_TYPE "wasm-object" 34 35 using namespace llvm; 36 using namespace object; 37 38 Expected<std::unique_ptr<WasmObjectFile>> 39 ObjectFile::createWasmObjectFile(MemoryBufferRef Buffer) { 40 Error Err = Error::success(); 41 auto ObjectFile = llvm::make_unique<WasmObjectFile>(Buffer, Err); 42 if (Err) 43 return std::move(Err); 44 45 return std::move(ObjectFile); 46 } 47 48 #define VARINT7_MAX ((1<<7)-1) 49 #define VARINT7_MIN (-(1<<7)) 50 #define VARUINT7_MAX (1<<7) 51 #define VARUINT1_MAX (1) 52 53 static uint8_t readUint8(const uint8_t *&Ptr) { return *Ptr++; } 54 55 static uint32_t readUint32(const uint8_t *&Ptr) { 56 uint32_t Result = support::endian::read32le(Ptr); 57 Ptr += sizeof(Result); 58 return Result; 59 } 60 61 static int32_t readFloat32(const uint8_t *&Ptr) { 62 int32_t Result = 0; 63 memcpy(&Result, Ptr, sizeof(Result)); 64 Ptr += sizeof(Result); 65 return Result; 66 } 67 68 static int64_t readFloat64(const uint8_t *&Ptr) { 69 int64_t Result = 0; 70 memcpy(&Result, Ptr, sizeof(Result)); 71 Ptr += sizeof(Result); 72 return Result; 73 } 74 75 static uint64_t readULEB128(const uint8_t *&Ptr) { 76 unsigned Count; 77 uint64_t Result = decodeULEB128(Ptr, &Count); 78 Ptr += Count; 79 return Result; 80 } 81 82 static StringRef readString(const uint8_t *&Ptr) { 83 uint32_t StringLen = readULEB128(Ptr); 84 StringRef Return = StringRef(reinterpret_cast<const char *>(Ptr), StringLen); 85 Ptr += StringLen; 86 return Return; 87 } 88 89 static int64_t readLEB128(const uint8_t *&Ptr) { 90 unsigned Count; 91 uint64_t Result = decodeSLEB128(Ptr, &Count); 92 Ptr += Count; 93 return Result; 94 } 95 96 static uint8_t readVaruint1(const uint8_t *&Ptr) { 97 int64_t result = readLEB128(Ptr); 98 assert(result <= VARUINT1_MAX && result >= 0); 99 return result; 100 } 101 102 static int32_t readVarint32(const uint8_t *&Ptr) { 103 int64_t result = readLEB128(Ptr); 104 assert(result <= INT32_MAX && result >= INT32_MIN); 105 return result; 106 } 107 108 static uint32_t readVaruint32(const uint8_t *&Ptr) { 109 uint64_t result = readULEB128(Ptr); 110 assert(result <= UINT32_MAX); 111 return result; 112 } 113 114 static int64_t readVarint64(const uint8_t *&Ptr) { 115 return readLEB128(Ptr); 116 } 117 118 static uint8_t readOpcode(const uint8_t *&Ptr) { 119 return readUint8(Ptr); 120 } 121 122 static Error readInitExpr(wasm::WasmInitExpr &Expr, const uint8_t *&Ptr) { 123 Expr.Opcode = readOpcode(Ptr); 124 125 switch (Expr.Opcode) { 126 case wasm::WASM_OPCODE_I32_CONST: 127 Expr.Value.Int32 = readVarint32(Ptr); 128 break; 129 case wasm::WASM_OPCODE_I64_CONST: 130 Expr.Value.Int64 = readVarint64(Ptr); 131 break; 132 case wasm::WASM_OPCODE_F32_CONST: 133 Expr.Value.Float32 = readFloat32(Ptr); 134 break; 135 case wasm::WASM_OPCODE_F64_CONST: 136 Expr.Value.Float64 = readFloat64(Ptr); 137 break; 138 case wasm::WASM_OPCODE_GET_GLOBAL: 139 Expr.Value.Global = readULEB128(Ptr); 140 break; 141 default: 142 return make_error<GenericBinaryError>("Invalid opcode in init_expr", 143 object_error::parse_failed); 144 } 145 146 uint8_t EndOpcode = readOpcode(Ptr); 147 if (EndOpcode != wasm::WASM_OPCODE_END) { 148 return make_error<GenericBinaryError>("Invalid init_expr", 149 object_error::parse_failed); 150 } 151 return Error::success(); 152 } 153 154 static wasm::WasmLimits readLimits(const uint8_t *&Ptr) { 155 wasm::WasmLimits Result; 156 Result.Flags = readVaruint1(Ptr); 157 Result.Initial = readVaruint32(Ptr); 158 if (Result.Flags & wasm::WASM_LIMITS_FLAG_HAS_MAX) 159 Result.Maximum = readVaruint32(Ptr); 160 return Result; 161 } 162 163 static wasm::WasmTable readTable(const uint8_t *&Ptr) { 164 wasm::WasmTable Table; 165 Table.ElemType = readUint8(Ptr); 166 Table.Limits = readLimits(Ptr); 167 return Table; 168 } 169 170 static Error readSection(WasmSection &Section, const uint8_t *&Ptr, 171 const uint8_t *Start, const uint8_t *Eof) { 172 Section.Offset = Ptr - Start; 173 Section.Type = readUint8(Ptr); 174 uint32_t Size = readVaruint32(Ptr); 175 if (Size == 0) 176 return make_error<StringError>("Zero length section", 177 object_error::parse_failed); 178 if (Ptr + Size > Eof) 179 return make_error<StringError>("Section too large", 180 object_error::parse_failed); 181 if (Section.Type == wasm::WASM_SEC_CUSTOM) { 182 const uint8_t *NameStart = Ptr; 183 Section.Name = readString(Ptr); 184 Size -= Ptr - NameStart; 185 } 186 Section.Content = ArrayRef<uint8_t>(Ptr, Size); 187 Ptr += Size; 188 return Error::success(); 189 } 190 191 WasmObjectFile::WasmObjectFile(MemoryBufferRef Buffer, Error &Err) 192 : ObjectFile(Binary::ID_Wasm, Buffer) { 193 ErrorAsOutParameter ErrAsOutParam(&Err); 194 Header.Magic = getData().substr(0, 4); 195 if (Header.Magic != StringRef("\0asm", 4)) { 196 Err = make_error<StringError>("Bad magic number", 197 object_error::parse_failed); 198 return; 199 } 200 201 const uint8_t *Eof = getPtr(getData().size()); 202 const uint8_t *Ptr = getPtr(4); 203 204 if (Ptr + 4 > Eof) { 205 Err = make_error<StringError>("Missing version number", 206 object_error::parse_failed); 207 return; 208 } 209 210 Header.Version = readUint32(Ptr); 211 if (Header.Version != wasm::WasmVersion) { 212 Err = make_error<StringError>("Bad version number", 213 object_error::parse_failed); 214 return; 215 } 216 217 WasmSection Sec; 218 while (Ptr < Eof) { 219 if ((Err = readSection(Sec, Ptr, getPtr(0), Eof))) 220 return; 221 if ((Err = parseSection(Sec))) 222 return; 223 224 Sections.push_back(Sec); 225 } 226 } 227 228 Error WasmObjectFile::parseSection(WasmSection &Sec) { 229 const uint8_t* Start = Sec.Content.data(); 230 const uint8_t* End = Start + Sec.Content.size(); 231 switch (Sec.Type) { 232 case wasm::WASM_SEC_CUSTOM: 233 return parseCustomSection(Sec, Start, End); 234 case wasm::WASM_SEC_TYPE: 235 return parseTypeSection(Start, End); 236 case wasm::WASM_SEC_IMPORT: 237 return parseImportSection(Start, End); 238 case wasm::WASM_SEC_FUNCTION: 239 return parseFunctionSection(Start, End); 240 case wasm::WASM_SEC_TABLE: 241 return parseTableSection(Start, End); 242 case wasm::WASM_SEC_MEMORY: 243 return parseMemorySection(Start, End); 244 case wasm::WASM_SEC_GLOBAL: 245 return parseGlobalSection(Start, End); 246 case wasm::WASM_SEC_EXPORT: 247 return parseExportSection(Start, End); 248 case wasm::WASM_SEC_START: 249 return parseStartSection(Start, End); 250 case wasm::WASM_SEC_ELEM: 251 return parseElemSection(Start, End); 252 case wasm::WASM_SEC_CODE: 253 return parseCodeSection(Start, End); 254 case wasm::WASM_SEC_DATA: 255 return parseDataSection(Start, End); 256 default: 257 return make_error<GenericBinaryError>("Bad section type", 258 object_error::parse_failed); 259 } 260 } 261 262 Error WasmObjectFile::parseNameSection(const uint8_t *Ptr, const uint8_t *End) { 263 llvm::DenseSet<uint64_t> Seen; 264 if (Functions.size() != FunctionTypes.size()) { 265 return make_error<GenericBinaryError>("Names must come after code section", 266 object_error::parse_failed); 267 } 268 269 while (Ptr < End) { 270 uint8_t Type = readUint8(Ptr); 271 uint32_t Size = readVaruint32(Ptr); 272 const uint8_t *SubSectionEnd = Ptr + Size; 273 switch (Type) { 274 case wasm::WASM_NAMES_FUNCTION: { 275 uint32_t Count = readVaruint32(Ptr); 276 while (Count--) { 277 uint32_t Index = readVaruint32(Ptr); 278 if (!Seen.insert(Index).second) 279 return make_error<GenericBinaryError>("Function named more than once", 280 object_error::parse_failed); 281 StringRef Name = readString(Ptr); 282 if (!isValidFunctionIndex(Index) || Name.empty()) 283 return make_error<GenericBinaryError>("Invalid name entry", 284 object_error::parse_failed); 285 DebugNames.push_back(wasm::WasmFunctionName{Index, Name}); 286 if (isDefinedFunctionIndex(Index)) 287 getDefinedFunction(Index).DebugName = Name; 288 } 289 break; 290 } 291 // Ignore local names for now 292 case wasm::WASM_NAMES_LOCAL: 293 default: 294 Ptr += Size; 295 break; 296 } 297 if (Ptr != SubSectionEnd) 298 return make_error<GenericBinaryError>("Name sub-section ended prematurely", 299 object_error::parse_failed); 300 } 301 302 if (Ptr != End) 303 return make_error<GenericBinaryError>("Name section ended prematurely", 304 object_error::parse_failed); 305 return Error::success(); 306 } 307 308 Error WasmObjectFile::parseLinkingSection(const uint8_t *Ptr, 309 const uint8_t *End) { 310 HasLinkingSection = true; 311 if (Functions.size() != FunctionTypes.size()) { 312 return make_error<GenericBinaryError>( 313 "Linking data must come after code section", object_error::parse_failed); 314 } 315 316 LinkingData.Version = readVaruint32(Ptr); 317 if (LinkingData.Version != wasm::WasmMetadataVersion) { 318 return make_error<GenericBinaryError>( 319 "Unexpected metadata version: " + Twine(LinkingData.Version) + 320 " (Expected: " + Twine(wasm::WasmMetadataVersion) + ")", 321 object_error::parse_failed); 322 } 323 324 while (Ptr < End) { 325 uint8_t Type = readUint8(Ptr); 326 uint32_t Size = readVaruint32(Ptr); 327 const uint8_t *SubSectionEnd = Ptr + Size; 328 switch (Type) { 329 case wasm::WASM_SYMBOL_TABLE: 330 if (Error Err = parseLinkingSectionSymtab(Ptr, SubSectionEnd)) 331 return Err; 332 break; 333 case wasm::WASM_SEGMENT_INFO: { 334 uint32_t Count = readVaruint32(Ptr); 335 if (Count > DataSegments.size()) 336 return make_error<GenericBinaryError>("Too many segment names", 337 object_error::parse_failed); 338 for (uint32_t i = 0; i < Count; i++) { 339 DataSegments[i].Data.Name = readString(Ptr); 340 DataSegments[i].Data.Alignment = readVaruint32(Ptr); 341 DataSegments[i].Data.Flags = readVaruint32(Ptr); 342 } 343 break; 344 } 345 case wasm::WASM_INIT_FUNCS: { 346 uint32_t Count = readVaruint32(Ptr); 347 LinkingData.InitFunctions.reserve(Count); 348 for (uint32_t i = 0; i < Count; i++) { 349 wasm::WasmInitFunc Init; 350 Init.Priority = readVaruint32(Ptr); 351 Init.Symbol = readVaruint32(Ptr); 352 if (!isValidFunctionSymbol(Init.Symbol)) 353 return make_error<GenericBinaryError>("Invalid function symbol: " + 354 Twine(Init.Symbol), 355 object_error::parse_failed); 356 LinkingData.InitFunctions.emplace_back(Init); 357 } 358 break; 359 } 360 case wasm::WASM_COMDAT_INFO: 361 if (Error Err = parseLinkingSectionComdat(Ptr, SubSectionEnd)) 362 return Err; 363 break; 364 default: 365 Ptr += Size; 366 break; 367 } 368 if (Ptr != SubSectionEnd) 369 return make_error<GenericBinaryError>( 370 "Linking sub-section ended prematurely", object_error::parse_failed); 371 } 372 if (Ptr != End) 373 return make_error<GenericBinaryError>("Linking section ended prematurely", 374 object_error::parse_failed); 375 return Error::success(); 376 } 377 378 Error WasmObjectFile::parseLinkingSectionSymtab(const uint8_t *&Ptr, 379 const uint8_t *End) { 380 uint32_t Count = readVaruint32(Ptr); 381 LinkingData.SymbolTable.reserve(Count); 382 Symbols.reserve(Count); 383 StringSet<> SymbolNames; 384 385 std::vector<wasm::WasmImport *> ImportedGlobals; 386 std::vector<wasm::WasmImport *> ImportedFunctions; 387 ImportedGlobals.reserve(Imports.size()); 388 ImportedFunctions.reserve(Imports.size()); 389 for (auto &I : Imports) { 390 if (I.Kind == wasm::WASM_EXTERNAL_FUNCTION) 391 ImportedFunctions.emplace_back(&I); 392 else if (I.Kind == wasm::WASM_EXTERNAL_GLOBAL) 393 ImportedGlobals.emplace_back(&I); 394 } 395 396 while (Count--) { 397 wasm::WasmSymbolInfo Info; 398 const wasm::WasmSignature *FunctionType = nullptr; 399 const wasm::WasmGlobalType *GlobalType = nullptr; 400 401 Info.Kind = readUint8(Ptr); 402 Info.Flags = readVaruint32(Ptr); 403 bool IsDefined = (Info.Flags & wasm::WASM_SYMBOL_UNDEFINED) == 0; 404 405 switch (Info.Kind) { 406 case wasm::WASM_SYMBOL_TYPE_FUNCTION: 407 Info.ElementIndex = readVaruint32(Ptr); 408 if (!isValidFunctionIndex(Info.ElementIndex) || 409 IsDefined != isDefinedFunctionIndex(Info.ElementIndex)) 410 return make_error<GenericBinaryError>("invalid function symbol index", 411 object_error::parse_failed); 412 if (IsDefined) { 413 Info.Name = readString(Ptr); 414 unsigned FuncIndex = Info.ElementIndex - NumImportedFunctions; 415 FunctionType = &Signatures[FunctionTypes[FuncIndex]]; 416 wasm::WasmFunction &Function = Functions[FuncIndex]; 417 if (Function.SymbolName.empty()) 418 Function.SymbolName = Info.Name; 419 } else { 420 wasm::WasmImport &Import = *ImportedFunctions[Info.ElementIndex]; 421 FunctionType = &Signatures[Import.SigIndex]; 422 Info.Name = Import.Field; 423 Info.Module = Import.Module; 424 } 425 break; 426 427 case wasm::WASM_SYMBOL_TYPE_GLOBAL: 428 Info.ElementIndex = readVaruint32(Ptr); 429 if (!isValidGlobalIndex(Info.ElementIndex) || 430 IsDefined != isDefinedGlobalIndex(Info.ElementIndex)) 431 return make_error<GenericBinaryError>("invalid global symbol index", 432 object_error::parse_failed); 433 if (!IsDefined && 434 (Info.Flags & wasm::WASM_SYMBOL_BINDING_MASK) == 435 wasm::WASM_SYMBOL_BINDING_WEAK) 436 return make_error<GenericBinaryError>("undefined weak global symbol", 437 object_error::parse_failed); 438 if (IsDefined) { 439 Info.Name = readString(Ptr); 440 unsigned GlobalIndex = Info.ElementIndex - NumImportedGlobals; 441 wasm::WasmGlobal &Global = Globals[GlobalIndex]; 442 GlobalType = &Global.Type; 443 if (Global.SymbolName.empty()) 444 Global.SymbolName = Info.Name; 445 } else { 446 wasm::WasmImport &Import = *ImportedGlobals[Info.ElementIndex]; 447 Info.Name = Import.Field; 448 GlobalType = &Import.Global; 449 } 450 break; 451 452 case wasm::WASM_SYMBOL_TYPE_DATA: 453 Info.Name = readString(Ptr); 454 if (IsDefined) { 455 uint32_t Index = readVaruint32(Ptr); 456 if (Index >= DataSegments.size()) 457 return make_error<GenericBinaryError>("invalid data symbol index", 458 object_error::parse_failed); 459 uint32_t Offset = readVaruint32(Ptr); 460 uint32_t Size = readVaruint32(Ptr); 461 if (Offset + Size > DataSegments[Index].Data.Content.size()) 462 return make_error<GenericBinaryError>("invalid data symbol offset", 463 object_error::parse_failed); 464 Info.DataRef = wasm::WasmDataReference{Index, Offset, Size}; 465 } 466 break; 467 468 case wasm::WASM_SYMBOL_TYPE_SECTION: { 469 if ((Info.Flags & wasm::WASM_SYMBOL_BINDING_MASK) != 470 wasm::WASM_SYMBOL_BINDING_LOCAL) 471 return make_error<GenericBinaryError>( 472 "Section symbols must have local binding", 473 object_error::parse_failed); 474 Info.ElementIndex = readVaruint32(Ptr); 475 // Use somewhat unique section name as symbol name. 476 StringRef SectionName = Sections[Info.ElementIndex].Name; 477 Info.Name = SectionName; 478 break; 479 } 480 481 default: 482 return make_error<GenericBinaryError>("Invalid symbol type", 483 object_error::parse_failed); 484 } 485 486 if ((Info.Flags & wasm::WASM_SYMBOL_BINDING_MASK) != 487 wasm::WASM_SYMBOL_BINDING_LOCAL && 488 !SymbolNames.insert(Info.Name).second) 489 return make_error<GenericBinaryError>("Duplicate symbol name " + 490 Twine(Info.Name), 491 object_error::parse_failed); 492 LinkingData.SymbolTable.emplace_back(Info); 493 Symbols.emplace_back(LinkingData.SymbolTable.back(), FunctionType, 494 GlobalType); 495 LLVM_DEBUG(dbgs() << "Adding symbol: " << Symbols.back() << "\n"); 496 } 497 498 return Error::success(); 499 } 500 501 Error WasmObjectFile::parseLinkingSectionComdat(const uint8_t *&Ptr, 502 const uint8_t *End) 503 { 504 uint32_t ComdatCount = readVaruint32(Ptr); 505 StringSet<> ComdatSet; 506 for (unsigned ComdatIndex = 0; ComdatIndex < ComdatCount; ++ComdatIndex) { 507 StringRef Name = readString(Ptr); 508 if (Name.empty() || !ComdatSet.insert(Name).second) 509 return make_error<GenericBinaryError>("Bad/duplicate COMDAT name " + Twine(Name), 510 object_error::parse_failed); 511 LinkingData.Comdats.emplace_back(Name); 512 uint32_t Flags = readVaruint32(Ptr); 513 if (Flags != 0) 514 return make_error<GenericBinaryError>("Unsupported COMDAT flags", 515 object_error::parse_failed); 516 517 uint32_t EntryCount = readVaruint32(Ptr); 518 while (EntryCount--) { 519 unsigned Kind = readVaruint32(Ptr); 520 unsigned Index = readVaruint32(Ptr); 521 switch (Kind) { 522 default: 523 return make_error<GenericBinaryError>("Invalid COMDAT entry type", 524 object_error::parse_failed); 525 case wasm::WASM_COMDAT_DATA: 526 if (Index >= DataSegments.size()) 527 return make_error<GenericBinaryError>("COMDAT data index out of range", 528 object_error::parse_failed); 529 if (DataSegments[Index].Data.Comdat != UINT32_MAX) 530 return make_error<GenericBinaryError>("Data segment in two COMDATs", 531 object_error::parse_failed); 532 DataSegments[Index].Data.Comdat = ComdatIndex; 533 break; 534 case wasm::WASM_COMDAT_FUNCTION: 535 if (!isDefinedFunctionIndex(Index)) 536 return make_error<GenericBinaryError>("COMDAT function index out of range", 537 object_error::parse_failed); 538 if (getDefinedFunction(Index).Comdat != UINT32_MAX) 539 return make_error<GenericBinaryError>("Function in two COMDATs", 540 object_error::parse_failed); 541 getDefinedFunction(Index).Comdat = ComdatIndex; 542 break; 543 } 544 } 545 } 546 return Error::success(); 547 } 548 549 Error WasmObjectFile::parseRelocSection(StringRef Name, const uint8_t *Ptr, 550 const uint8_t *End) { 551 uint32_t SectionIndex = readVaruint32(Ptr); 552 if (SectionIndex >= Sections.size()) 553 return make_error<GenericBinaryError>("Invalid section index", 554 object_error::parse_failed); 555 WasmSection& Section = Sections[SectionIndex]; 556 uint32_t RelocCount = readVaruint32(Ptr); 557 uint32_t EndOffset = Section.Content.size(); 558 while (RelocCount--) { 559 wasm::WasmRelocation Reloc = {}; 560 Reloc.Type = readVaruint32(Ptr); 561 Reloc.Offset = readVaruint32(Ptr); 562 Reloc.Index = readVaruint32(Ptr); 563 switch (Reloc.Type) { 564 case wasm::R_WEBASSEMBLY_FUNCTION_INDEX_LEB: 565 case wasm::R_WEBASSEMBLY_TABLE_INDEX_SLEB: 566 case wasm::R_WEBASSEMBLY_TABLE_INDEX_I32: 567 if (!isValidFunctionSymbol(Reloc.Index)) 568 return make_error<GenericBinaryError>("Bad relocation function index", 569 object_error::parse_failed); 570 break; 571 case wasm::R_WEBASSEMBLY_TYPE_INDEX_LEB: 572 if (Reloc.Index >= Signatures.size()) 573 return make_error<GenericBinaryError>("Bad relocation type index", 574 object_error::parse_failed); 575 break; 576 case wasm::R_WEBASSEMBLY_GLOBAL_INDEX_LEB: 577 if (!isValidGlobalSymbol(Reloc.Index)) 578 return make_error<GenericBinaryError>("Bad relocation global index", 579 object_error::parse_failed); 580 break; 581 case wasm::R_WEBASSEMBLY_MEMORY_ADDR_LEB: 582 case wasm::R_WEBASSEMBLY_MEMORY_ADDR_SLEB: 583 case wasm::R_WEBASSEMBLY_MEMORY_ADDR_I32: 584 if (!isValidDataSymbol(Reloc.Index)) 585 return make_error<GenericBinaryError>("Bad relocation data index", 586 object_error::parse_failed); 587 Reloc.Addend = readVarint32(Ptr); 588 break; 589 case wasm::R_WEBASSEMBLY_FUNCTION_OFFSET_I32: 590 if (!isValidFunctionSymbol(Reloc.Index)) 591 return make_error<GenericBinaryError>("Bad relocation function index", 592 object_error::parse_failed); 593 Reloc.Addend = readVarint32(Ptr); 594 break; 595 case wasm::R_WEBASSEMBLY_SECTION_OFFSET_I32: 596 if (!isValidSectionSymbol(Reloc.Index)) 597 return make_error<GenericBinaryError>("Bad relocation section index", 598 object_error::parse_failed); 599 Reloc.Addend = readVarint32(Ptr); 600 break; 601 default: 602 return make_error<GenericBinaryError>("Bad relocation type: " + 603 Twine(Reloc.Type), 604 object_error::parse_failed); 605 } 606 607 // Relocations must fit inside the section, and must appear in order. They 608 // also shouldn't overlap a function/element boundary, but we don't bother 609 // to check that. 610 uint64_t Size = 5; 611 if (Reloc.Type == wasm::R_WEBASSEMBLY_TABLE_INDEX_I32 || 612 Reloc.Type == wasm::R_WEBASSEMBLY_MEMORY_ADDR_I32 || 613 Reloc.Type == wasm::R_WEBASSEMBLY_SECTION_OFFSET_I32 || 614 Reloc.Type == wasm::R_WEBASSEMBLY_FUNCTION_OFFSET_I32) 615 Size = 4; 616 if (Reloc.Offset + Size > EndOffset) 617 return make_error<GenericBinaryError>("Bad relocation offset", 618 object_error::parse_failed); 619 620 Section.Relocations.push_back(Reloc); 621 } 622 if (Ptr != End) 623 return make_error<GenericBinaryError>("Reloc section ended prematurely", 624 object_error::parse_failed); 625 return Error::success(); 626 } 627 628 Error WasmObjectFile::parseCustomSection(WasmSection &Sec, 629 const uint8_t *Ptr, const uint8_t *End) { 630 if (Sec.Name == "name") { 631 if (Error Err = parseNameSection(Ptr, End)) 632 return Err; 633 } else if (Sec.Name == "linking") { 634 if (Error Err = parseLinkingSection(Ptr, End)) 635 return Err; 636 } else if (Sec.Name.startswith("reloc.")) { 637 if (Error Err = parseRelocSection(Sec.Name, Ptr, End)) 638 return Err; 639 } 640 return Error::success(); 641 } 642 643 Error WasmObjectFile::parseTypeSection(const uint8_t *Ptr, const uint8_t *End) { 644 uint32_t Count = readVaruint32(Ptr); 645 Signatures.reserve(Count); 646 while (Count--) { 647 wasm::WasmSignature Sig; 648 Sig.ReturnType = wasm::WASM_TYPE_NORESULT; 649 uint8_t Form = readUint8(Ptr); 650 if (Form != wasm::WASM_TYPE_FUNC) { 651 return make_error<GenericBinaryError>("Invalid signature type", 652 object_error::parse_failed); 653 } 654 uint32_t ParamCount = readVaruint32(Ptr); 655 Sig.ParamTypes.reserve(ParamCount); 656 while (ParamCount--) { 657 uint32_t ParamType = readUint8(Ptr); 658 Sig.ParamTypes.push_back(ParamType); 659 } 660 uint32_t ReturnCount = readVaruint32(Ptr); 661 if (ReturnCount) { 662 if (ReturnCount != 1) { 663 return make_error<GenericBinaryError>( 664 "Multiple return types not supported", object_error::parse_failed); 665 } 666 Sig.ReturnType = readUint8(Ptr); 667 } 668 Signatures.push_back(Sig); 669 } 670 if (Ptr != End) 671 return make_error<GenericBinaryError>("Type section ended prematurely", 672 object_error::parse_failed); 673 return Error::success(); 674 } 675 676 Error WasmObjectFile::parseImportSection(const uint8_t *Ptr, const uint8_t *End) { 677 uint32_t Count = readVaruint32(Ptr); 678 Imports.reserve(Count); 679 for (uint32_t i = 0; i < Count; i++) { 680 wasm::WasmImport Im; 681 Im.Module = readString(Ptr); 682 Im.Field = readString(Ptr); 683 Im.Kind = readUint8(Ptr); 684 switch (Im.Kind) { 685 case wasm::WASM_EXTERNAL_FUNCTION: 686 NumImportedFunctions++; 687 Im.SigIndex = readVaruint32(Ptr); 688 break; 689 case wasm::WASM_EXTERNAL_GLOBAL: 690 NumImportedGlobals++; 691 Im.Global.Type = readUint8(Ptr); 692 Im.Global.Mutable = readVaruint1(Ptr); 693 break; 694 case wasm::WASM_EXTERNAL_MEMORY: 695 Im.Memory = readLimits(Ptr); 696 break; 697 case wasm::WASM_EXTERNAL_TABLE: 698 Im.Table = readTable(Ptr); 699 if (Im.Table.ElemType != wasm::WASM_TYPE_ANYFUNC) 700 return make_error<GenericBinaryError>("Invalid table element type", 701 object_error::parse_failed); 702 break; 703 default: 704 return make_error<GenericBinaryError>( 705 "Unexpected import kind", object_error::parse_failed); 706 } 707 Imports.push_back(Im); 708 } 709 if (Ptr != End) 710 return make_error<GenericBinaryError>("Import section ended prematurely", 711 object_error::parse_failed); 712 return Error::success(); 713 } 714 715 Error WasmObjectFile::parseFunctionSection(const uint8_t *Ptr, const uint8_t *End) { 716 uint32_t Count = readVaruint32(Ptr); 717 FunctionTypes.reserve(Count); 718 uint32_t NumTypes = Signatures.size(); 719 while (Count--) { 720 uint32_t Type = readVaruint32(Ptr); 721 if (Type >= NumTypes) 722 return make_error<GenericBinaryError>("Invalid function type", 723 object_error::parse_failed); 724 FunctionTypes.push_back(Type); 725 } 726 if (Ptr != End) 727 return make_error<GenericBinaryError>("Function section ended prematurely", 728 object_error::parse_failed); 729 return Error::success(); 730 } 731 732 Error WasmObjectFile::parseTableSection(const uint8_t *Ptr, const uint8_t *End) { 733 uint32_t Count = readVaruint32(Ptr); 734 Tables.reserve(Count); 735 while (Count--) { 736 Tables.push_back(readTable(Ptr)); 737 if (Tables.back().ElemType != wasm::WASM_TYPE_ANYFUNC) { 738 return make_error<GenericBinaryError>("Invalid table element type", 739 object_error::parse_failed); 740 } 741 } 742 if (Ptr != End) 743 return make_error<GenericBinaryError>("Table section ended prematurely", 744 object_error::parse_failed); 745 return Error::success(); 746 } 747 748 Error WasmObjectFile::parseMemorySection(const uint8_t *Ptr, const uint8_t *End) { 749 uint32_t Count = readVaruint32(Ptr); 750 Memories.reserve(Count); 751 while (Count--) { 752 Memories.push_back(readLimits(Ptr)); 753 } 754 if (Ptr != End) 755 return make_error<GenericBinaryError>("Memory section ended prematurely", 756 object_error::parse_failed); 757 return Error::success(); 758 } 759 760 Error WasmObjectFile::parseGlobalSection(const uint8_t *Ptr, const uint8_t *End) { 761 GlobalSection = Sections.size(); 762 uint32_t Count = readVaruint32(Ptr); 763 Globals.reserve(Count); 764 while (Count--) { 765 wasm::WasmGlobal Global; 766 Global.Index = NumImportedGlobals + Globals.size(); 767 Global.Type.Type = readUint8(Ptr); 768 Global.Type.Mutable = readVaruint1(Ptr); 769 if (Error Err = readInitExpr(Global.InitExpr, Ptr)) 770 return Err; 771 Globals.push_back(Global); 772 } 773 if (Ptr != End) 774 return make_error<GenericBinaryError>("Global section ended prematurely", 775 object_error::parse_failed); 776 return Error::success(); 777 } 778 779 Error WasmObjectFile::parseExportSection(const uint8_t *Ptr, const uint8_t *End) { 780 uint32_t Count = readVaruint32(Ptr); 781 Exports.reserve(Count); 782 for (uint32_t i = 0; i < Count; i++) { 783 wasm::WasmExport Ex; 784 Ex.Name = readString(Ptr); 785 Ex.Kind = readUint8(Ptr); 786 Ex.Index = readVaruint32(Ptr); 787 switch (Ex.Kind) { 788 case wasm::WASM_EXTERNAL_FUNCTION: 789 if (!isValidFunctionIndex(Ex.Index)) 790 return make_error<GenericBinaryError>("Invalid function export", 791 object_error::parse_failed); 792 break; 793 case wasm::WASM_EXTERNAL_GLOBAL: 794 if (!isValidGlobalIndex(Ex.Index)) 795 return make_error<GenericBinaryError>("Invalid global export", 796 object_error::parse_failed); 797 break; 798 case wasm::WASM_EXTERNAL_MEMORY: 799 case wasm::WASM_EXTERNAL_TABLE: 800 break; 801 default: 802 return make_error<GenericBinaryError>( 803 "Unexpected export kind", object_error::parse_failed); 804 } 805 Exports.push_back(Ex); 806 } 807 if (Ptr != End) 808 return make_error<GenericBinaryError>("Export section ended prematurely", 809 object_error::parse_failed); 810 return Error::success(); 811 } 812 813 bool WasmObjectFile::isValidFunctionIndex(uint32_t Index) const { 814 return Index < NumImportedFunctions + FunctionTypes.size(); 815 } 816 817 bool WasmObjectFile::isDefinedFunctionIndex(uint32_t Index) const { 818 return Index >= NumImportedFunctions && isValidFunctionIndex(Index); 819 } 820 821 bool WasmObjectFile::isValidGlobalIndex(uint32_t Index) const { 822 return Index < NumImportedGlobals + Globals.size(); 823 } 824 825 bool WasmObjectFile::isDefinedGlobalIndex(uint32_t Index) const { 826 return Index >= NumImportedGlobals && isValidGlobalIndex(Index); 827 } 828 829 bool WasmObjectFile::isValidFunctionSymbol(uint32_t Index) const { 830 return Index < Symbols.size() && Symbols[Index].isTypeFunction(); 831 } 832 833 bool WasmObjectFile::isValidGlobalSymbol(uint32_t Index) const { 834 return Index < Symbols.size() && Symbols[Index].isTypeGlobal(); 835 } 836 837 bool WasmObjectFile::isValidDataSymbol(uint32_t Index) const { 838 return Index < Symbols.size() && Symbols[Index].isTypeData(); 839 } 840 841 bool WasmObjectFile::isValidSectionSymbol(uint32_t Index) const { 842 return Index < Symbols.size() && Symbols[Index].isTypeSection(); 843 } 844 845 wasm::WasmFunction &WasmObjectFile::getDefinedFunction(uint32_t Index) { 846 assert(isDefinedFunctionIndex(Index)); 847 return Functions[Index - NumImportedFunctions]; 848 } 849 850 wasm::WasmGlobal &WasmObjectFile::getDefinedGlobal(uint32_t Index) { 851 assert(isDefinedGlobalIndex(Index)); 852 return Globals[Index - NumImportedGlobals]; 853 } 854 855 Error WasmObjectFile::parseStartSection(const uint8_t *Ptr, const uint8_t *End) { 856 StartFunction = readVaruint32(Ptr); 857 if (!isValidFunctionIndex(StartFunction)) 858 return make_error<GenericBinaryError>("Invalid start function", 859 object_error::parse_failed); 860 return Error::success(); 861 } 862 863 Error WasmObjectFile::parseCodeSection(const uint8_t *Ptr, const uint8_t *End) { 864 CodeSection = Sections.size(); 865 const uint8_t *CodeSectionStart = Ptr; 866 uint32_t FunctionCount = readVaruint32(Ptr); 867 if (FunctionCount != FunctionTypes.size()) { 868 return make_error<GenericBinaryError>("Invalid function count", 869 object_error::parse_failed); 870 } 871 872 while (FunctionCount--) { 873 wasm::WasmFunction Function; 874 const uint8_t *FunctionStart = Ptr; 875 uint32_t Size = readVaruint32(Ptr); 876 const uint8_t *FunctionEnd = Ptr + Size; 877 878 Function.Index = NumImportedFunctions + Functions.size(); 879 Function.CodeSectionOffset = FunctionStart - CodeSectionStart; 880 Function.Size = FunctionEnd - FunctionStart; 881 882 uint32_t NumLocalDecls = readVaruint32(Ptr); 883 Function.Locals.reserve(NumLocalDecls); 884 while (NumLocalDecls--) { 885 wasm::WasmLocalDecl Decl; 886 Decl.Count = readVaruint32(Ptr); 887 Decl.Type = readUint8(Ptr); 888 Function.Locals.push_back(Decl); 889 } 890 891 uint32_t BodySize = FunctionEnd - Ptr; 892 Function.Body = ArrayRef<uint8_t>(Ptr, BodySize); 893 // This will be set later when reading in the linking metadata section. 894 Function.Comdat = UINT32_MAX; 895 Ptr += BodySize; 896 assert(Ptr == FunctionEnd); 897 Functions.push_back(Function); 898 } 899 if (Ptr != End) 900 return make_error<GenericBinaryError>("Code section ended prematurely", 901 object_error::parse_failed); 902 return Error::success(); 903 } 904 905 Error WasmObjectFile::parseElemSection(const uint8_t *Ptr, const uint8_t *End) { 906 uint32_t Count = readVaruint32(Ptr); 907 ElemSegments.reserve(Count); 908 while (Count--) { 909 wasm::WasmElemSegment Segment; 910 Segment.TableIndex = readVaruint32(Ptr); 911 if (Segment.TableIndex != 0) { 912 return make_error<GenericBinaryError>("Invalid TableIndex", 913 object_error::parse_failed); 914 } 915 if (Error Err = readInitExpr(Segment.Offset, Ptr)) 916 return Err; 917 uint32_t NumElems = readVaruint32(Ptr); 918 while (NumElems--) { 919 Segment.Functions.push_back(readVaruint32(Ptr)); 920 } 921 ElemSegments.push_back(Segment); 922 } 923 if (Ptr != End) 924 return make_error<GenericBinaryError>("Elem section ended prematurely", 925 object_error::parse_failed); 926 return Error::success(); 927 } 928 929 Error WasmObjectFile::parseDataSection(const uint8_t *Ptr, const uint8_t *End) { 930 DataSection = Sections.size(); 931 const uint8_t *Start = Ptr; 932 uint32_t Count = readVaruint32(Ptr); 933 DataSegments.reserve(Count); 934 while (Count--) { 935 WasmSegment Segment; 936 Segment.Data.MemoryIndex = readVaruint32(Ptr); 937 if (Error Err = readInitExpr(Segment.Data.Offset, Ptr)) 938 return Err; 939 uint32_t Size = readVaruint32(Ptr); 940 Segment.Data.Content = ArrayRef<uint8_t>(Ptr, Size); 941 // The rest of these Data fields are set later, when reading in the linking 942 // metadata section. 943 Segment.Data.Alignment = 0; 944 Segment.Data.Flags = 0; 945 Segment.Data.Comdat = UINT32_MAX; 946 Segment.SectionOffset = Ptr - Start; 947 Ptr += Size; 948 DataSegments.push_back(Segment); 949 } 950 if (Ptr != End) 951 return make_error<GenericBinaryError>("Data section ended prematurely", 952 object_error::parse_failed); 953 return Error::success(); 954 } 955 956 const uint8_t *WasmObjectFile::getPtr(size_t Offset) const { 957 return reinterpret_cast<const uint8_t *>(getData().substr(Offset, 1).data()); 958 } 959 960 const wasm::WasmObjectHeader &WasmObjectFile::getHeader() const { 961 return Header; 962 } 963 964 void WasmObjectFile::moveSymbolNext(DataRefImpl &Symb) const { Symb.d.a++; } 965 966 uint32_t WasmObjectFile::getSymbolFlags(DataRefImpl Symb) const { 967 uint32_t Result = SymbolRef::SF_None; 968 const WasmSymbol &Sym = getWasmSymbol(Symb); 969 970 LLVM_DEBUG(dbgs() << "getSymbolFlags: ptr=" << &Sym << " " << Sym << "\n"); 971 if (Sym.isBindingWeak()) 972 Result |= SymbolRef::SF_Weak; 973 if (!Sym.isBindingLocal()) 974 Result |= SymbolRef::SF_Global; 975 if (Sym.isHidden()) 976 Result |= SymbolRef::SF_Hidden; 977 if (!Sym.isDefined()) 978 Result |= SymbolRef::SF_Undefined; 979 if (Sym.isTypeFunction()) 980 Result |= SymbolRef::SF_Executable; 981 return Result; 982 } 983 984 basic_symbol_iterator WasmObjectFile::symbol_begin() const { 985 DataRefImpl Ref; 986 Ref.d.a = 0; 987 return BasicSymbolRef(Ref, this); 988 } 989 990 basic_symbol_iterator WasmObjectFile::symbol_end() const { 991 DataRefImpl Ref; 992 Ref.d.a = Symbols.size(); 993 return BasicSymbolRef(Ref, this); 994 } 995 996 const WasmSymbol &WasmObjectFile::getWasmSymbol(const DataRefImpl &Symb) const { 997 return Symbols[Symb.d.a]; 998 } 999 1000 const WasmSymbol &WasmObjectFile::getWasmSymbol(const SymbolRef &Symb) const { 1001 return getWasmSymbol(Symb.getRawDataRefImpl()); 1002 } 1003 1004 Expected<StringRef> WasmObjectFile::getSymbolName(DataRefImpl Symb) const { 1005 return getWasmSymbol(Symb).Info.Name; 1006 } 1007 1008 Expected<uint64_t> WasmObjectFile::getSymbolAddress(DataRefImpl Symb) const { 1009 return getSymbolValue(Symb); 1010 } 1011 1012 uint64_t WasmObjectFile::getWasmSymbolValue(const WasmSymbol& Sym) const { 1013 switch (Sym.Info.Kind) { 1014 case wasm::WASM_SYMBOL_TYPE_FUNCTION: 1015 case wasm::WASM_SYMBOL_TYPE_GLOBAL: 1016 return Sym.Info.ElementIndex; 1017 case wasm::WASM_SYMBOL_TYPE_DATA: { 1018 // The value of a data symbol is the segment offset, plus the symbol 1019 // offset within the segment. 1020 uint32_t SegmentIndex = Sym.Info.DataRef.Segment; 1021 const wasm::WasmDataSegment &Segment = DataSegments[SegmentIndex].Data; 1022 assert(Segment.Offset.Opcode == wasm::WASM_OPCODE_I32_CONST); 1023 return Segment.Offset.Value.Int32 + Sym.Info.DataRef.Offset; 1024 } 1025 case wasm::WASM_SYMBOL_TYPE_SECTION: 1026 return 0; 1027 } 1028 llvm_unreachable("invalid symbol type"); 1029 } 1030 1031 uint64_t WasmObjectFile::getSymbolValueImpl(DataRefImpl Symb) const { 1032 return getWasmSymbolValue(getWasmSymbol(Symb)); 1033 } 1034 1035 uint32_t WasmObjectFile::getSymbolAlignment(DataRefImpl Symb) const { 1036 llvm_unreachable("not yet implemented"); 1037 return 0; 1038 } 1039 1040 uint64_t WasmObjectFile::getCommonSymbolSizeImpl(DataRefImpl Symb) const { 1041 llvm_unreachable("not yet implemented"); 1042 return 0; 1043 } 1044 1045 Expected<SymbolRef::Type> 1046 WasmObjectFile::getSymbolType(DataRefImpl Symb) const { 1047 const WasmSymbol &Sym = getWasmSymbol(Symb); 1048 1049 switch (Sym.Info.Kind) { 1050 case wasm::WASM_SYMBOL_TYPE_FUNCTION: 1051 return SymbolRef::ST_Function; 1052 case wasm::WASM_SYMBOL_TYPE_GLOBAL: 1053 return SymbolRef::ST_Other; 1054 case wasm::WASM_SYMBOL_TYPE_DATA: 1055 return SymbolRef::ST_Data; 1056 case wasm::WASM_SYMBOL_TYPE_SECTION: 1057 return SymbolRef::ST_Debug; 1058 } 1059 1060 llvm_unreachable("Unknown WasmSymbol::SymbolType"); 1061 return SymbolRef::ST_Other; 1062 } 1063 1064 Expected<section_iterator> 1065 WasmObjectFile::getSymbolSection(DataRefImpl Symb) const { 1066 const WasmSymbol& Sym = getWasmSymbol(Symb); 1067 if (Sym.isUndefined()) 1068 return section_end(); 1069 1070 DataRefImpl Ref; 1071 switch (Sym.Info.Kind) { 1072 case wasm::WASM_SYMBOL_TYPE_FUNCTION: 1073 Ref.d.a = CodeSection; 1074 break; 1075 case wasm::WASM_SYMBOL_TYPE_GLOBAL: 1076 Ref.d.a = GlobalSection; 1077 break; 1078 case wasm::WASM_SYMBOL_TYPE_DATA: 1079 Ref.d.a = DataSection; 1080 break; 1081 case wasm::WASM_SYMBOL_TYPE_SECTION: { 1082 Ref.d.a = Sym.Info.ElementIndex; 1083 break; 1084 } 1085 default: 1086 llvm_unreachable("Unknown WasmSymbol::SymbolType"); 1087 } 1088 return section_iterator(SectionRef(Ref, this)); 1089 } 1090 1091 void WasmObjectFile::moveSectionNext(DataRefImpl &Sec) const { Sec.d.a++; } 1092 1093 std::error_code WasmObjectFile::getSectionName(DataRefImpl Sec, 1094 StringRef &Res) const { 1095 const WasmSection &S = Sections[Sec.d.a]; 1096 #define ECase(X) \ 1097 case wasm::WASM_SEC_##X: \ 1098 Res = #X; \ 1099 break 1100 switch (S.Type) { 1101 ECase(TYPE); 1102 ECase(IMPORT); 1103 ECase(FUNCTION); 1104 ECase(TABLE); 1105 ECase(MEMORY); 1106 ECase(GLOBAL); 1107 ECase(EXPORT); 1108 ECase(START); 1109 ECase(ELEM); 1110 ECase(CODE); 1111 ECase(DATA); 1112 case wasm::WASM_SEC_CUSTOM: 1113 Res = S.Name; 1114 break; 1115 default: 1116 return object_error::invalid_section_index; 1117 } 1118 #undef ECase 1119 return std::error_code(); 1120 } 1121 1122 uint64_t WasmObjectFile::getSectionAddress(DataRefImpl Sec) const { return 0; } 1123 1124 uint64_t WasmObjectFile::getSectionIndex(DataRefImpl Sec) const { 1125 return Sec.d.a; 1126 } 1127 1128 uint64_t WasmObjectFile::getSectionSize(DataRefImpl Sec) const { 1129 const WasmSection &S = Sections[Sec.d.a]; 1130 return S.Content.size(); 1131 } 1132 1133 std::error_code WasmObjectFile::getSectionContents(DataRefImpl Sec, 1134 StringRef &Res) const { 1135 const WasmSection &S = Sections[Sec.d.a]; 1136 // This will never fail since wasm sections can never be empty (user-sections 1137 // must have a name and non-user sections each have a defined structure). 1138 Res = StringRef(reinterpret_cast<const char *>(S.Content.data()), 1139 S.Content.size()); 1140 return std::error_code(); 1141 } 1142 1143 uint64_t WasmObjectFile::getSectionAlignment(DataRefImpl Sec) const { 1144 return 1; 1145 } 1146 1147 bool WasmObjectFile::isSectionCompressed(DataRefImpl Sec) const { 1148 return false; 1149 } 1150 1151 bool WasmObjectFile::isSectionText(DataRefImpl Sec) const { 1152 return getWasmSection(Sec).Type == wasm::WASM_SEC_CODE; 1153 } 1154 1155 bool WasmObjectFile::isSectionData(DataRefImpl Sec) const { 1156 return getWasmSection(Sec).Type == wasm::WASM_SEC_DATA; 1157 } 1158 1159 bool WasmObjectFile::isSectionBSS(DataRefImpl Sec) const { return false; } 1160 1161 bool WasmObjectFile::isSectionVirtual(DataRefImpl Sec) const { return false; } 1162 1163 bool WasmObjectFile::isSectionBitcode(DataRefImpl Sec) const { return false; } 1164 1165 relocation_iterator WasmObjectFile::section_rel_begin(DataRefImpl Ref) const { 1166 DataRefImpl RelocRef; 1167 RelocRef.d.a = Ref.d.a; 1168 RelocRef.d.b = 0; 1169 return relocation_iterator(RelocationRef(RelocRef, this)); 1170 } 1171 1172 relocation_iterator WasmObjectFile::section_rel_end(DataRefImpl Ref) const { 1173 const WasmSection &Sec = getWasmSection(Ref); 1174 DataRefImpl RelocRef; 1175 RelocRef.d.a = Ref.d.a; 1176 RelocRef.d.b = Sec.Relocations.size(); 1177 return relocation_iterator(RelocationRef(RelocRef, this)); 1178 } 1179 1180 void WasmObjectFile::moveRelocationNext(DataRefImpl &Rel) const { 1181 Rel.d.b++; 1182 } 1183 1184 uint64_t WasmObjectFile::getRelocationOffset(DataRefImpl Ref) const { 1185 const wasm::WasmRelocation &Rel = getWasmRelocation(Ref); 1186 return Rel.Offset; 1187 } 1188 1189 symbol_iterator WasmObjectFile::getRelocationSymbol(DataRefImpl Ref) const { 1190 const wasm::WasmRelocation &Rel = getWasmRelocation(Ref); 1191 if (Rel.Type == wasm::R_WEBASSEMBLY_TYPE_INDEX_LEB) 1192 return symbol_end(); 1193 DataRefImpl Sym; 1194 Sym.d.a = Rel.Index; 1195 Sym.d.b = 0; 1196 return symbol_iterator(SymbolRef(Sym, this)); 1197 } 1198 1199 uint64_t WasmObjectFile::getRelocationType(DataRefImpl Ref) const { 1200 const wasm::WasmRelocation &Rel = getWasmRelocation(Ref); 1201 return Rel.Type; 1202 } 1203 1204 void WasmObjectFile::getRelocationTypeName( 1205 DataRefImpl Ref, SmallVectorImpl<char> &Result) const { 1206 const wasm::WasmRelocation& Rel = getWasmRelocation(Ref); 1207 StringRef Res = "Unknown"; 1208 1209 #define WASM_RELOC(name, value) \ 1210 case wasm::name: \ 1211 Res = #name; \ 1212 break; 1213 1214 switch (Rel.Type) { 1215 #include "llvm/BinaryFormat/WasmRelocs.def" 1216 } 1217 1218 #undef WASM_RELOC 1219 1220 Result.append(Res.begin(), Res.end()); 1221 } 1222 1223 section_iterator WasmObjectFile::section_begin() const { 1224 DataRefImpl Ref; 1225 Ref.d.a = 0; 1226 return section_iterator(SectionRef(Ref, this)); 1227 } 1228 1229 section_iterator WasmObjectFile::section_end() const { 1230 DataRefImpl Ref; 1231 Ref.d.a = Sections.size(); 1232 return section_iterator(SectionRef(Ref, this)); 1233 } 1234 1235 uint8_t WasmObjectFile::getBytesInAddress() const { return 4; } 1236 1237 StringRef WasmObjectFile::getFileFormatName() const { return "WASM"; } 1238 1239 Triple::ArchType WasmObjectFile::getArch() const { return Triple::wasm32; } 1240 1241 SubtargetFeatures WasmObjectFile::getFeatures() const { 1242 return SubtargetFeatures(); 1243 } 1244 1245 bool WasmObjectFile::isRelocatableObject() const { 1246 return HasLinkingSection; 1247 } 1248 1249 const WasmSection &WasmObjectFile::getWasmSection(DataRefImpl Ref) const { 1250 assert(Ref.d.a < Sections.size()); 1251 return Sections[Ref.d.a]; 1252 } 1253 1254 const WasmSection & 1255 WasmObjectFile::getWasmSection(const SectionRef &Section) const { 1256 return getWasmSection(Section.getRawDataRefImpl()); 1257 } 1258 1259 const wasm::WasmRelocation & 1260 WasmObjectFile::getWasmRelocation(const RelocationRef &Ref) const { 1261 return getWasmRelocation(Ref.getRawDataRefImpl()); 1262 } 1263 1264 const wasm::WasmRelocation & 1265 WasmObjectFile::getWasmRelocation(DataRefImpl Ref) const { 1266 assert(Ref.d.a < Sections.size()); 1267 const WasmSection& Sec = Sections[Ref.d.a]; 1268 assert(Ref.d.b < Sec.Relocations.size()); 1269 return Sec.Relocations[Ref.d.b]; 1270 } 1271