118eafb6cSDan Gohman //===- lib/MC/WasmObjectWriter.cpp - Wasm File Writer ---------------------===// 218eafb6cSDan Gohman // 32946cd70SChandler Carruth // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 42946cd70SChandler Carruth // See https://llvm.org/LICENSE.txt for license information. 52946cd70SChandler Carruth // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 618eafb6cSDan Gohman // 718eafb6cSDan Gohman //===----------------------------------------------------------------------===// 818eafb6cSDan Gohman // 918eafb6cSDan Gohman // This file implements Wasm object file writer information. 1018eafb6cSDan Gohman // 1118eafb6cSDan Gohman //===----------------------------------------------------------------------===// 1218eafb6cSDan Gohman 1318eafb6cSDan Gohman #include "llvm/ADT/STLExtras.h" 1418eafb6cSDan Gohman #include "llvm/ADT/SmallPtrSet.h" 15264b5d9eSZachary Turner #include "llvm/BinaryFormat/Wasm.h" 16b59dff4bSBenjamin Kramer #include "llvm/BinaryFormat/WasmTraits.h" 17432a3883SNico Weber #include "llvm/Config/llvm-config.h" 1818eafb6cSDan Gohman #include "llvm/MC/MCAsmBackend.h" 1918eafb6cSDan Gohman #include "llvm/MC/MCAsmLayout.h" 2018eafb6cSDan Gohman #include "llvm/MC/MCAssembler.h" 2118eafb6cSDan Gohman #include "llvm/MC/MCContext.h" 2218eafb6cSDan Gohman #include "llvm/MC/MCExpr.h" 2318eafb6cSDan Gohman #include "llvm/MC/MCFixupKindInfo.h" 2418eafb6cSDan Gohman #include "llvm/MC/MCObjectWriter.h" 2518eafb6cSDan Gohman #include "llvm/MC/MCSectionWasm.h" 2618eafb6cSDan Gohman #include "llvm/MC/MCSymbolWasm.h" 2718eafb6cSDan Gohman #include "llvm/MC/MCValue.h" 2818eafb6cSDan Gohman #include "llvm/MC/MCWasmObjectWriter.h" 29d934cb88SDan Gohman #include "llvm/Support/Casting.h" 3018eafb6cSDan Gohman #include "llvm/Support/Debug.h" 31bcd7f777SSimon Pilgrim #include "llvm/Support/EndianStream.h" 3218eafb6cSDan Gohman #include "llvm/Support/ErrorHandling.h" 33d934cb88SDan Gohman #include "llvm/Support/LEB128.h" 3418eafb6cSDan Gohman #include "llvm/Support/StringSaver.h" 3518eafb6cSDan Gohman #include <vector> 3618eafb6cSDan Gohman 3718eafb6cSDan Gohman using namespace llvm; 3818eafb6cSDan Gohman 395e3d33a7SSam Clegg #define DEBUG_TYPE "mc" 4018eafb6cSDan Gohman 4118eafb6cSDan Gohman namespace { 429e15f359SSam Clegg 43feac819eSAndy Wingo // When we create the indirect function table we start at 1, so that there is 44feac819eSAndy Wingo // and empty slot at 0 and therefore calling a null function pointer will trap. 4518c56a07SHeejin Ahn static const uint32_t InitialTableOffset = 1; 4630e1bbc1SSam Clegg 47d934cb88SDan Gohman // For patching purposes, we need to remember where each section starts, both 48d934cb88SDan Gohman // for patching up the section size field, and for patching up references to 49d934cb88SDan Gohman // locations within the section. 50d934cb88SDan Gohman struct SectionBookkeeping { 51d934cb88SDan Gohman // Where the size of the section is written. 52d934cb88SDan Gohman uint64_t SizeOffset; 536a31a0d6SSam Clegg // Where the section header ends (without custom section name). 546a31a0d6SSam Clegg uint64_t PayloadOffset; 556a31a0d6SSam Clegg // Where the contents of the section starts. 56d934cb88SDan Gohman uint64_t ContentsOffset; 576f08c84aSSam Clegg uint32_t Index; 58d934cb88SDan Gohman }; 59d934cb88SDan Gohman 607c395943SSam Clegg // A wasm data segment. A wasm binary contains only a single data section 617c395943SSam Clegg // but that can contain many segments, each with their own virtual location 627c395943SSam Clegg // in memory. Each MCSection data created by llvm is modeled as its own 637c395943SSam Clegg // wasm data segment. 647c395943SSam Clegg struct WasmDataSegment { 657c395943SSam Clegg MCSectionWasm *Section; 66d95ed959SSam Clegg StringRef Name; 672e150409SThomas Lively uint32_t InitFlags; 684d135b04SWouter van Oortmerssen uint64_t Offset; 6963ebb813SSam Clegg uint32_t Alignment; 702e150409SThomas Lively uint32_t LinkerFlags; 717c395943SSam Clegg SmallVector<char, 4> Data; 727c395943SSam Clegg }; 737c395943SSam Clegg 749e15f359SSam Clegg // A wasm function to be written into the function section. 759e15f359SSam Clegg struct WasmFunction { 76da419bdbSHeejin Ahn uint32_t SigIndex; 779e15f359SSam Clegg const MCSymbolWasm *Sym; 789e15f359SSam Clegg }; 799e15f359SSam Clegg 809e15f359SSam Clegg // A wasm global to be written into the global section. 819e15f359SSam Clegg struct WasmGlobal { 826e7f1826SSam Clegg wasm::WasmGlobalType Type; 839e15f359SSam Clegg uint64_t InitialValue; 849e15f359SSam Clegg }; 859e15f359SSam Clegg 86ea7caceeSSam Clegg // Information about a single item which is part of a COMDAT. For each data 87ea7caceeSSam Clegg // segment or function which is in the COMDAT, there is a corresponding 88ea7caceeSSam Clegg // WasmComdatEntry. 89ea7caceeSSam Clegg struct WasmComdatEntry { 90ea7caceeSSam Clegg unsigned Kind; 91ea7caceeSSam Clegg uint32_t Index; 92ea7caceeSSam Clegg }; 93ea7caceeSSam Clegg 946dc65e91SSam Clegg // Information about a single relocation. 956dc65e91SSam Clegg struct WasmRelocationEntry { 966dc65e91SSam Clegg uint64_t Offset; // Where is the relocation. 976dc65e91SSam Clegg const MCSymbolWasm *Symbol; // The symbol to relocate with. 986dc65e91SSam Clegg int64_t Addend; // A value to add to the symbol. 996dc65e91SSam Clegg unsigned Type; // The type of the relocation. 100fe6414b0SSam Clegg const MCSectionWasm *FixupSection; // The section the relocation is targeting. 1016dc65e91SSam Clegg 1026dc65e91SSam Clegg WasmRelocationEntry(uint64_t Offset, const MCSymbolWasm *Symbol, 1036dc65e91SSam Clegg int64_t Addend, unsigned Type, 104fe6414b0SSam Clegg const MCSectionWasm *FixupSection) 1056dc65e91SSam Clegg : Offset(Offset), Symbol(Symbol), Addend(Addend), Type(Type), 1066dc65e91SSam Clegg FixupSection(FixupSection) {} 1076dc65e91SSam Clegg 108cadcb9ebSKeno Fischer bool hasAddend() const { return wasm::relocTypeHasAddend(Type); } 109acd7d2b0SSam Clegg 1106dc65e91SSam Clegg void print(raw_ostream &Out) const { 111f208f631SHeejin Ahn Out << wasm::relocTypetoString(Type) << " Off=" << Offset 112f208f631SHeejin Ahn << ", Sym=" << *Symbol << ", Addend=" << Addend 1137d1ff446SFangrui Song << ", FixupSection=" << FixupSection->getName(); 1146dc65e91SSam Clegg } 115b7787fd0SSam Clegg 116615eb470SAaron Ballman #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP) 117b7787fd0SSam Clegg LLVM_DUMP_METHOD void dump() const { print(dbgs()); } 118b7787fd0SSam Clegg #endif 1196dc65e91SSam Clegg }; 1206dc65e91SSam Clegg 12118c56a07SHeejin Ahn static const uint32_t InvalidIndex = -1; 12225d8e68bSSam Clegg 123cfd44a2eSSam Clegg struct WasmCustomSection { 124cfd44a2eSSam Clegg 1256a31a0d6SSam Clegg StringRef Name; 1266a31a0d6SSam Clegg MCSectionWasm *Section; 1276a31a0d6SSam Clegg 1286a31a0d6SSam Clegg uint32_t OutputContentsOffset; 1296a31a0d6SSam Clegg uint32_t OutputIndex; 1306a31a0d6SSam Clegg 1316a31a0d6SSam Clegg WasmCustomSection(StringRef Name, MCSectionWasm *Section) 1326a31a0d6SSam Clegg : Name(Name), Section(Section), OutputContentsOffset(0), 13318c56a07SHeejin Ahn OutputIndex(InvalidIndex) {} 134cfd44a2eSSam Clegg }; 135cfd44a2eSSam Clegg 1361fb8daa6SSam Clegg #if !defined(NDEBUG) 1377f055deeSSam Clegg raw_ostream &operator<<(raw_ostream &OS, const WasmRelocationEntry &Rel) { 138b7787fd0SSam Clegg Rel.print(OS); 139b7787fd0SSam Clegg return OS; 140b7787fd0SSam Clegg } 1411fb8daa6SSam Clegg #endif 142b7787fd0SSam Clegg 14319e8befaSSam Clegg // Write X as an (unsigned) LEB value at offset Offset in Stream, padded 14419e8befaSSam Clegg // to allow patching. 1453b29376eSWouter van Oortmerssen template <int W> 1463b29376eSWouter van Oortmerssen void writePatchableLEB(raw_pwrite_stream &Stream, uint64_t X, uint64_t Offset) { 1473b29376eSWouter van Oortmerssen uint8_t Buffer[W]; 1483b29376eSWouter van Oortmerssen unsigned SizeLen = encodeULEB128(X, Buffer, W); 1493b29376eSWouter van Oortmerssen assert(SizeLen == W); 15019e8befaSSam Clegg Stream.pwrite((char *)Buffer, SizeLen, Offset); 15119e8befaSSam Clegg } 15219e8befaSSam Clegg 15319e8befaSSam Clegg // Write X as an signed LEB value at offset Offset in Stream, padded 15419e8befaSSam Clegg // to allow patching. 1553b29376eSWouter van Oortmerssen template <int W> 1563b29376eSWouter van Oortmerssen void writePatchableSLEB(raw_pwrite_stream &Stream, int64_t X, uint64_t Offset) { 1573b29376eSWouter van Oortmerssen uint8_t Buffer[W]; 1583b29376eSWouter van Oortmerssen unsigned SizeLen = encodeSLEB128(X, Buffer, W); 1593b29376eSWouter van Oortmerssen assert(SizeLen == W); 16019e8befaSSam Clegg Stream.pwrite((char *)Buffer, SizeLen, Offset); 16119e8befaSSam Clegg } 16219e8befaSSam Clegg 16319e8befaSSam Clegg // Write X as a plain integer value at offset Offset in Stream. 1640a6c4d8dSSam Clegg static void patchI32(raw_pwrite_stream &Stream, uint32_t X, uint64_t Offset) { 16519e8befaSSam Clegg uint8_t Buffer[4]; 16619e8befaSSam Clegg support::endian::write32le(Buffer, X); 16719e8befaSSam Clegg Stream.pwrite((char *)Buffer, sizeof(Buffer), Offset); 16819e8befaSSam Clegg } 16919e8befaSSam Clegg 1703b29376eSWouter van Oortmerssen static void patchI64(raw_pwrite_stream &Stream, uint64_t X, uint64_t Offset) { 1713b29376eSWouter van Oortmerssen uint8_t Buffer[8]; 1723b29376eSWouter van Oortmerssen support::endian::write64le(Buffer, X); 1733b29376eSWouter van Oortmerssen Stream.pwrite((char *)Buffer, sizeof(Buffer), Offset); 1743b29376eSWouter van Oortmerssen } 1753b29376eSWouter van Oortmerssen 1760ff28fa6SDerek Schuff bool isDwoSection(const MCSection &Sec) { 1770ff28fa6SDerek Schuff return Sec.getName().endswith(".dwo"); 1780ff28fa6SDerek Schuff } 1790ff28fa6SDerek Schuff 18018eafb6cSDan Gohman class WasmObjectWriter : public MCObjectWriter { 1810ff28fa6SDerek Schuff support::endian::Writer *W; 182f17b149dSPeter Collingbourne 18318eafb6cSDan Gohman /// The target specific Wasm writer instance. 18418eafb6cSDan Gohman std::unique_ptr<MCWasmObjectTargetWriter> TargetObjectWriter; 18518eafb6cSDan Gohman 186d934cb88SDan Gohman // Relocations for fixing up references in the code section. 187d934cb88SDan Gohman std::vector<WasmRelocationEntry> CodeRelocations; 188d934cb88SDan Gohman // Relocations for fixing up references in the data section. 189d934cb88SDan Gohman std::vector<WasmRelocationEntry> DataRelocations; 190d934cb88SDan Gohman 191d934cb88SDan Gohman // Index values to use for fixing up call_indirect type indices. 192acd7d2b0SSam Clegg // Maps function symbols to the index of the type of the function 193acd7d2b0SSam Clegg DenseMap<const MCSymbolWasm *, uint32_t> TypeIndices; 194d99f6078SSam Clegg // Maps function symbols to the table element index space. Used 195d99f6078SSam Clegg // for TABLE_INDEX relocation types (i.e. address taken functions). 196f9edbe95SSam Clegg DenseMap<const MCSymbolWasm *, uint32_t> TableIndices; 197388fb67bSPaulo Matos // Maps function/global/table symbols to the 198388fb67bSPaulo Matos // function/global/table/event/section index space. 1996c899ba6SSam Clegg DenseMap<const MCSymbolWasm *, uint32_t> WasmIndices; 200492f7529SSam Clegg DenseMap<const MCSymbolWasm *, uint32_t> GOTIndices; 2016c899ba6SSam Clegg // Maps data symbols to the Wasm segment and offset/size with the segment. 2026c899ba6SSam Clegg DenseMap<const MCSymbolWasm *, wasm::WasmDataReference> DataLocations; 2036a31a0d6SSam Clegg 2046a31a0d6SSam Clegg // Stores output data (index, relocations, content offset) for custom 2056a31a0d6SSam Clegg // section. 2066a31a0d6SSam Clegg std::vector<WasmCustomSection> CustomSections; 207cbda16ebSThomas Lively std::unique_ptr<WasmCustomSection> ProducersSection; 208f6f4f843SThomas Lively std::unique_ptr<WasmCustomSection> TargetFeaturesSection; 2096a31a0d6SSam Clegg // Relocations for fixing up references in the custom sections. 2106a31a0d6SSam Clegg DenseMap<const MCSectionWasm *, std::vector<WasmRelocationEntry>> 2116a31a0d6SSam Clegg CustomSectionsRelocations; 212acd7d2b0SSam Clegg 213c0d41195SSam Clegg // Map from section to defining function symbol. 2146ccb59b3SSam Clegg DenseMap<const MCSection *, const MCSymbol *> SectionFunctions; 2156ccb59b3SSam Clegg 2164c41fb5aSHeejin Ahn DenseMap<wasm::WasmSignature, uint32_t> SignatureIndices; 2174c41fb5aSHeejin Ahn SmallVector<wasm::WasmSignature, 4> Signatures; 2186c899ba6SSam Clegg SmallVector<WasmDataSegment, 4> DataSegments; 2199f3fe42eSSam Clegg unsigned NumFunctionImports = 0; 2207c395943SSam Clegg unsigned NumGlobalImports = 0; 221feac819eSAndy Wingo unsigned NumTableImports = 0; 222da419bdbSHeejin Ahn unsigned NumEventImports = 0; 2237e1c3345SChandler Carruth uint32_t SectionCount = 0; 224d934cb88SDan Gohman 2250ff28fa6SDerek Schuff enum class DwoMode { 2260ff28fa6SDerek Schuff AllSections, 2270ff28fa6SDerek Schuff NonDwoOnly, 2280ff28fa6SDerek Schuff DwoOnly, 2290ff28fa6SDerek Schuff }; 2300ff28fa6SDerek Schuff bool IsSplitDwarf = false; 2310ff28fa6SDerek Schuff raw_pwrite_stream *OS = nullptr; 2320ff28fa6SDerek Schuff raw_pwrite_stream *DwoOS = nullptr; 2330ff28fa6SDerek Schuff 2340ff28fa6SDerek Schuff // TargetObjectWriter wranppers. 23518eafb6cSDan Gohman bool is64Bit() const { return TargetObjectWriter->is64Bit(); } 236da84b688SDan Gohman bool isEmscripten() const { return TargetObjectWriter->isEmscripten(); } 23718eafb6cSDan Gohman 2382322a937SSam Clegg void startSection(SectionBookkeeping &Section, unsigned SectionId); 2392322a937SSam Clegg void startCustomSection(SectionBookkeeping &Section, StringRef Name); 240d934cb88SDan Gohman void endSection(SectionBookkeeping &Section); 241d934cb88SDan Gohman 24218eafb6cSDan Gohman public: 2431301a878SLang Hames WasmObjectWriter(std::unique_ptr<MCWasmObjectTargetWriter> MOTW, 2440ff28fa6SDerek Schuff raw_pwrite_stream &OS_) 2450ff28fa6SDerek Schuff : TargetObjectWriter(std::move(MOTW)), OS(&OS_) {} 2460ff28fa6SDerek Schuff 2470ff28fa6SDerek Schuff WasmObjectWriter(std::unique_ptr<MCWasmObjectTargetWriter> MOTW, 2480ff28fa6SDerek Schuff raw_pwrite_stream &OS_, raw_pwrite_stream &DwoOS_) 2490ff28fa6SDerek Schuff : TargetObjectWriter(std::move(MOTW)), IsSplitDwarf(true), OS(&OS_), 2500ff28fa6SDerek Schuff DwoOS(&DwoOS_) {} 25118eafb6cSDan Gohman 2520917c9e4SDan Gohman private: 253acd7d2b0SSam Clegg void reset() override { 254acd7d2b0SSam Clegg CodeRelocations.clear(); 255acd7d2b0SSam Clegg DataRelocations.clear(); 256acd7d2b0SSam Clegg TypeIndices.clear(); 2576c899ba6SSam Clegg WasmIndices.clear(); 258492f7529SSam Clegg GOTIndices.clear(); 259f9edbe95SSam Clegg TableIndices.clear(); 2606c899ba6SSam Clegg DataLocations.clear(); 261cbda16ebSThomas Lively CustomSections.clear(); 262cbda16ebSThomas Lively ProducersSection.reset(); 263f6f4f843SThomas Lively TargetFeaturesSection.reset(); 2646a31a0d6SSam Clegg CustomSectionsRelocations.clear(); 265da419bdbSHeejin Ahn SignatureIndices.clear(); 266da419bdbSHeejin Ahn Signatures.clear(); 2676c899ba6SSam Clegg DataSegments.clear(); 2686ccb59b3SSam Clegg SectionFunctions.clear(); 2699f3fe42eSSam Clegg NumFunctionImports = 0; 2707c395943SSam Clegg NumGlobalImports = 0; 271feac819eSAndy Wingo NumTableImports = 0; 272105bdc25SSam Clegg MCObjectWriter::reset(); 273acd7d2b0SSam Clegg } 274acd7d2b0SSam Clegg 27518eafb6cSDan Gohman void writeHeader(const MCAssembler &Asm); 27618eafb6cSDan Gohman 27718eafb6cSDan Gohman void recordRelocation(MCAssembler &Asm, const MCAsmLayout &Layout, 27818eafb6cSDan Gohman const MCFragment *Fragment, const MCFixup &Fixup, 279ceecfe5bSRafael Espindola MCValue Target, uint64_t &FixedValue) override; 28018eafb6cSDan Gohman 28118eafb6cSDan Gohman void executePostLayoutBinding(MCAssembler &Asm, 28218eafb6cSDan Gohman const MCAsmLayout &Layout) override; 2830ff28fa6SDerek Schuff void prepareImports(SmallVectorImpl<wasm::WasmImport> &Imports, 2840ff28fa6SDerek Schuff MCAssembler &Asm, const MCAsmLayout &Layout); 285438390faSPeter Collingbourne uint64_t writeObject(MCAssembler &Asm, const MCAsmLayout &Layout) override; 2869e15f359SSam Clegg 2870ff28fa6SDerek Schuff uint64_t writeOneObject(MCAssembler &Asm, const MCAsmLayout &Layout, 2880ff28fa6SDerek Schuff DwoMode Mode); 2890ff28fa6SDerek Schuff 290b7787fd0SSam Clegg void writeString(const StringRef Str) { 2910ff28fa6SDerek Schuff encodeULEB128(Str.size(), W->OS); 2920ff28fa6SDerek Schuff W->OS << Str; 293b7787fd0SSam Clegg } 294b7787fd0SSam Clegg 2950a6c4d8dSSam Clegg void writeI32(int32_t val) { 2960a6c4d8dSSam Clegg char Buffer[4]; 2970a6c4d8dSSam Clegg support::endian::write32le(Buffer, val); 2980ff28fa6SDerek Schuff W->OS.write(Buffer, sizeof(Buffer)); 2990a6c4d8dSSam Clegg } 3000a6c4d8dSSam Clegg 3010a6c4d8dSSam Clegg void writeI64(int64_t val) { 3020a6c4d8dSSam Clegg char Buffer[8]; 3030a6c4d8dSSam Clegg support::endian::write64le(Buffer, val); 3040ff28fa6SDerek Schuff W->OS.write(Buffer, sizeof(Buffer)); 3050a6c4d8dSSam Clegg } 3060a6c4d8dSSam Clegg 3070ff28fa6SDerek Schuff void writeValueType(wasm::ValType Ty) { W->OS << static_cast<char>(Ty); } 3089e15f359SSam Clegg 3094c41fb5aSHeejin Ahn void writeTypeSection(ArrayRef<wasm::WasmSignature> Signatures); 3104d135b04SWouter van Oortmerssen void writeImportSection(ArrayRef<wasm::WasmImport> Imports, uint64_t DataSize, 311f950b24aSSam Clegg uint32_t NumElements); 312457fb0b4SSam Clegg void writeFunctionSection(ArrayRef<WasmFunction> Functions); 3138defa952SSam Clegg void writeExportSection(ArrayRef<wasm::WasmExport> Exports); 314457fb0b4SSam Clegg void writeElemSection(ArrayRef<uint32_t> TableElems); 315fef8de66SThomas Lively void writeDataCountSection(); 316e49584a3SSam Clegg uint32_t writeCodeSection(const MCAssembler &Asm, const MCAsmLayout &Layout, 317457fb0b4SSam Clegg ArrayRef<WasmFunction> Functions); 318e49584a3SSam Clegg uint32_t writeDataSection(const MCAsmLayout &Layout); 319da419bdbSHeejin Ahn void writeEventSection(ArrayRef<wasm::WasmEventType> Events); 3200a6c4d8dSSam Clegg void writeGlobalSection(ArrayRef<wasm::WasmGlobal> Globals); 321388fb67bSPaulo Matos void writeTableSection(ArrayRef<wasm::WasmTable> Tables); 3226f08c84aSSam Clegg void writeRelocSection(uint32_t SectionIndex, StringRef Name, 323f77dc2a8SSam Clegg std::vector<WasmRelocationEntry> &Relocations); 32431a2c809SSam Clegg void writeLinkingMetaDataSection( 32586b4a09aSSam Clegg ArrayRef<wasm::WasmSymbolInfo> SymbolInfos, 326ea7caceeSSam Clegg ArrayRef<std::pair<uint16_t, uint32_t>> InitFuncs, 327ea7caceeSSam Clegg const std::map<StringRef, std::vector<WasmComdatEntry>> &Comdats); 328cbda16ebSThomas Lively void writeCustomSection(WasmCustomSection &CustomSection, 329cbda16ebSThomas Lively const MCAssembler &Asm, const MCAsmLayout &Layout); 3306a31a0d6SSam Clegg void writeCustomRelocSections(); 331acd7d2b0SSam Clegg 3327ee758d6SSam Clegg uint64_t getProvisionalValue(const WasmRelocationEntry &RelEntry, 3337ee758d6SSam Clegg const MCAsmLayout &Layout); 334acd7d2b0SSam Clegg void applyRelocations(ArrayRef<WasmRelocationEntry> Relocations, 3357ee758d6SSam Clegg uint64_t ContentsOffset, const MCAsmLayout &Layout); 336acd7d2b0SSam Clegg 337acd7d2b0SSam Clegg uint32_t getRelocationIndexValue(const WasmRelocationEntry &RelEntry); 3385e3d33a7SSam Clegg uint32_t getFunctionType(const MCSymbolWasm &Symbol); 339da419bdbSHeejin Ahn uint32_t getEventType(const MCSymbolWasm &Symbol); 34038b12f5eSHeejin Ahn void registerFunctionType(const MCSymbolWasm &Symbol); 34138b12f5eSHeejin Ahn void registerEventType(const MCSymbolWasm &Symbol); 34218eafb6cSDan Gohman }; 3439e15f359SSam Clegg 34418eafb6cSDan Gohman } // end anonymous namespace 34518eafb6cSDan Gohman 346d934cb88SDan Gohman // Write out a section header and a patchable section size field. 347d934cb88SDan Gohman void WasmObjectWriter::startSection(SectionBookkeeping &Section, 3482322a937SSam Clegg unsigned SectionId) { 349d34e60caSNicola Zaghen LLVM_DEBUG(dbgs() << "startSection " << SectionId << "\n"); 3500ff28fa6SDerek Schuff W->OS << char(SectionId); 351d934cb88SDan Gohman 3520ff28fa6SDerek Schuff Section.SizeOffset = W->OS.tell(); 353d934cb88SDan Gohman 354d934cb88SDan Gohman // The section size. We don't know the size yet, so reserve enough space 355d934cb88SDan Gohman // for any 32-bit value; we'll patch it later. 3560ff28fa6SDerek Schuff encodeULEB128(0, W->OS, 5); 357d934cb88SDan Gohman 358d934cb88SDan Gohman // The position where the section starts, for measuring its size. 3590ff28fa6SDerek Schuff Section.ContentsOffset = W->OS.tell(); 3600ff28fa6SDerek Schuff Section.PayloadOffset = W->OS.tell(); 3616f08c84aSSam Clegg Section.Index = SectionCount++; 362d934cb88SDan Gohman } 3632322a937SSam Clegg 3642322a937SSam Clegg void WasmObjectWriter::startCustomSection(SectionBookkeeping &Section, 3652322a937SSam Clegg StringRef Name) { 366d34e60caSNicola Zaghen LLVM_DEBUG(dbgs() << "startCustomSection " << Name << "\n"); 3672322a937SSam Clegg startSection(Section, wasm::WASM_SEC_CUSTOM); 3686a31a0d6SSam Clegg 3696a31a0d6SSam Clegg // The position where the section header ends, for measuring its size. 3700ff28fa6SDerek Schuff Section.PayloadOffset = W->OS.tell(); 3716a31a0d6SSam Clegg 3722322a937SSam Clegg // Custom sections in wasm also have a string identifier. 3732322a937SSam Clegg writeString(Name); 3746a31a0d6SSam Clegg 3756a31a0d6SSam Clegg // The position where the custom section starts. 3760ff28fa6SDerek Schuff Section.ContentsOffset = W->OS.tell(); 377d934cb88SDan Gohman } 378d934cb88SDan Gohman 379d934cb88SDan Gohman // Now that the section is complete and we know how big it is, patch up the 380d934cb88SDan Gohman // section size field at the start of the section. 381d934cb88SDan Gohman void WasmObjectWriter::endSection(SectionBookkeeping &Section) { 3820ff28fa6SDerek Schuff uint64_t Size = W->OS.tell(); 383c0affde8SSam Clegg // /dev/null doesn't support seek/tell and can report offset of 0. 384c0affde8SSam Clegg // Simply skip this patching in that case. 385c0affde8SSam Clegg if (!Size) 386c0affde8SSam Clegg return; 387c0affde8SSam Clegg 388c0affde8SSam Clegg Size -= Section.PayloadOffset; 389d934cb88SDan Gohman if (uint32_t(Size) != Size) 390d934cb88SDan Gohman report_fatal_error("section size does not fit in a uint32_t"); 391d934cb88SDan Gohman 392d34e60caSNicola Zaghen LLVM_DEBUG(dbgs() << "endSection size=" << Size << "\n"); 393d934cb88SDan Gohman 394d934cb88SDan Gohman // Write the final section size to the payload_len field, which follows 395d934cb88SDan Gohman // the section id byte. 3960ff28fa6SDerek Schuff writePatchableLEB<5>(static_cast<raw_pwrite_stream &>(W->OS), Size, 397f17b149dSPeter Collingbourne Section.SizeOffset); 398d934cb88SDan Gohman } 399d934cb88SDan Gohman 40018eafb6cSDan Gohman // Emit the Wasm header. 40118eafb6cSDan Gohman void WasmObjectWriter::writeHeader(const MCAssembler &Asm) { 4020ff28fa6SDerek Schuff W->OS.write(wasm::WasmMagic, sizeof(wasm::WasmMagic)); 4030ff28fa6SDerek Schuff W->write<uint32_t>(wasm::WasmVersion); 40418eafb6cSDan Gohman } 40518eafb6cSDan Gohman 40618eafb6cSDan Gohman void WasmObjectWriter::executePostLayoutBinding(MCAssembler &Asm, 40718eafb6cSDan Gohman const MCAsmLayout &Layout) { 4086ccb59b3SSam Clegg // Build a map of sections to the function that defines them, for use 4096ccb59b3SSam Clegg // in recordRelocation. 4106ccb59b3SSam Clegg for (const MCSymbol &S : Asm.symbols()) { 4116ccb59b3SSam Clegg const auto &WS = static_cast<const MCSymbolWasm &>(S); 4126ccb59b3SSam Clegg if (WS.isDefined() && WS.isFunction() && !WS.isVariable()) { 4136ccb59b3SSam Clegg const auto &Sec = static_cast<const MCSectionWasm &>(S.getSection()); 4146ccb59b3SSam Clegg auto Pair = SectionFunctions.insert(std::make_pair(&Sec, &S)); 4156ccb59b3SSam Clegg if (!Pair.second) 4166ccb59b3SSam Clegg report_fatal_error("section already has a defining function: " + 4177d1ff446SFangrui Song Sec.getName()); 4186ccb59b3SSam Clegg } 4196ccb59b3SSam Clegg } 42018eafb6cSDan Gohman } 42118eafb6cSDan Gohman 42218eafb6cSDan Gohman void WasmObjectWriter::recordRelocation(MCAssembler &Asm, 42318eafb6cSDan Gohman const MCAsmLayout &Layout, 42418eafb6cSDan Gohman const MCFragment *Fragment, 42518eafb6cSDan Gohman const MCFixup &Fixup, MCValue Target, 426ceecfe5bSRafael Espindola uint64_t &FixedValue) { 427ecc5e808SSam Clegg // The WebAssembly backend should never generate FKF_IsPCRel fixups 428e828ce1bSFangrui Song assert(!(Asm.getBackend().getFixupKindInfo(Fixup.getKind()).Flags & 429ecc5e808SSam Clegg MCFixupKindInfo::FKF_IsPCRel)); 430ecc5e808SSam Clegg 431fe6414b0SSam Clegg const auto &FixupSection = cast<MCSectionWasm>(*Fragment->getParent()); 432d934cb88SDan Gohman uint64_t C = Target.getConstant(); 433d934cb88SDan Gohman uint64_t FixupOffset = Layout.getFragmentOffset(Fragment) + Fixup.getOffset(); 434d934cb88SDan Gohman MCContext &Ctx = Asm.getContext(); 435d934cb88SDan Gohman 436d934cb88SDan Gohman if (const MCSymbolRefExpr *RefB = Target.getSymB()) { 437ecc5e808SSam Clegg // To get here the A - B expression must have failed evaluateAsRelocatable. 438ecc5e808SSam Clegg // This means either A or B must be undefined and in WebAssembly we can't 439ecc5e808SSam Clegg // support either of those cases. 440ecc5e808SSam Clegg const auto &SymB = cast<MCSymbolWasm>(RefB->getSymbol()); 441d934cb88SDan Gohman Ctx.reportError( 442d934cb88SDan Gohman Fixup.getLoc(), 443d934cb88SDan Gohman Twine("symbol '") + SymB.getName() + 444ecc5e808SSam Clegg "': unsupported subtraction expression used in relocation."); 445d934cb88SDan Gohman return; 446d934cb88SDan Gohman } 447d934cb88SDan Gohman 448d934cb88SDan Gohman // We either rejected the fixup or folded B into C at this point. 449d934cb88SDan Gohman const MCSymbolRefExpr *RefA = Target.getSymA(); 450ecc5e808SSam Clegg const auto *SymA = cast<MCSymbolWasm>(&RefA->getSymbol()); 451d934cb88SDan Gohman 4523ea1c62cSSam Clegg // The .init_array isn't translated as data, so don't do relocations in it. 4537d1ff446SFangrui Song if (FixupSection.getName().startswith(".init_array")) { 4543ea1c62cSSam Clegg SymA->setUsedInInitArray(); 4553ea1c62cSSam Clegg return; 4563ea1c62cSSam Clegg } 4573ea1c62cSSam Clegg 458ecc5e808SSam Clegg if (SymA->isVariable()) { 459d934cb88SDan Gohman const MCExpr *Expr = SymA->getVariableValue(); 4607ee758d6SSam Clegg if (const auto *Inner = dyn_cast<MCSymbolRefExpr>(Expr)) 4616ad8f190SSam Clegg if (Inner->getKind() == MCSymbolRefExpr::VK_WEAKREF) 4626ad8f190SSam Clegg llvm_unreachable("weakref used in reloc not yet implemented"); 463d934cb88SDan Gohman } 464d934cb88SDan Gohman 465d934cb88SDan Gohman // Put any constant offset in an addend. Offsets can be negative, and 466d934cb88SDan Gohman // LLVM expects wrapping, in contrast to wasm's immediates which can't 467d934cb88SDan Gohman // be negative and don't wrap. 468d934cb88SDan Gohman FixedValue = 0; 469d934cb88SDan Gohman 470a5e175c6SSam Clegg unsigned Type = TargetObjectWriter->getRelocType(Target, Fixup); 4719d24fb7fSSam Clegg 4724d57fbd0SSam Clegg // Absolute offset within a section or a function. 4734d57fbd0SSam Clegg // Currently only supported for for metadata sections. 4744d57fbd0SSam Clegg // See: test/MC/WebAssembly/blockaddress.ll 475d1152a26SSam Clegg if (Type == wasm::R_WASM_FUNCTION_OFFSET_I32 || 47616f02431SWouter van Oortmerssen Type == wasm::R_WASM_FUNCTION_OFFSET_I64 || 477d1152a26SSam Clegg Type == wasm::R_WASM_SECTION_OFFSET_I32) { 4784d57fbd0SSam Clegg if (!FixupSection.getKind().isMetadata()) 4794d57fbd0SSam Clegg report_fatal_error("relocations for function or section offsets are " 4804d57fbd0SSam Clegg "only supported in metadata sections"); 4814d57fbd0SSam Clegg 4824d57fbd0SSam Clegg const MCSymbol *SectionSymbol = nullptr; 4834d57fbd0SSam Clegg const MCSection &SecA = SymA->getSection(); 4844d57fbd0SSam Clegg if (SecA.getKind().isText()) 4856ccb59b3SSam Clegg SectionSymbol = SectionFunctions.find(&SecA)->second; 4864d57fbd0SSam Clegg else 4874d57fbd0SSam Clegg SectionSymbol = SecA.getBeginSymbol(); 4884d57fbd0SSam Clegg if (!SectionSymbol) 4894d57fbd0SSam Clegg report_fatal_error("section symbol is required for relocation"); 4904d57fbd0SSam Clegg 4914d57fbd0SSam Clegg C += Layout.getSymbolOffset(*SymA); 4924d57fbd0SSam Clegg SymA = cast<MCSymbolWasm>(SectionSymbol); 4934d57fbd0SSam Clegg } 4944d57fbd0SSam Clegg 495feac819eSAndy Wingo if (Type == wasm::R_WASM_TABLE_INDEX_REL_SLEB || 496feac819eSAndy Wingo Type == wasm::R_WASM_TABLE_INDEX_SLEB || 497feac819eSAndy Wingo Type == wasm::R_WASM_TABLE_INDEX_SLEB64 || 498feac819eSAndy Wingo Type == wasm::R_WASM_TABLE_INDEX_I32 || 499feac819eSAndy Wingo Type == wasm::R_WASM_TABLE_INDEX_I64) { 500feac819eSAndy Wingo // TABLE_INDEX relocs implicitly use the default indirect function table. 501feac819eSAndy Wingo auto TableName = "__indirect_function_table"; 502feac819eSAndy Wingo MCSymbolWasm *Sym = cast_or_null<MCSymbolWasm>(Ctx.lookupSymbol(TableName)); 503feac819eSAndy Wingo if (Sym) { 504feac819eSAndy Wingo if (!Sym->isFunctionTable()) 505feac819eSAndy Wingo Ctx.reportError( 506feac819eSAndy Wingo Fixup.getLoc(), 507feac819eSAndy Wingo "symbol '__indirect_function_table' is not a function table"); 508feac819eSAndy Wingo } else { 509feac819eSAndy Wingo Sym = cast<MCSymbolWasm>(Ctx.getOrCreateSymbol(TableName)); 510feac819eSAndy Wingo Sym->setFunctionTable(); 511feac819eSAndy Wingo // The default function table is synthesized by the linker. 512feac819eSAndy Wingo Sym->setUndefined(); 513feac819eSAndy Wingo } 514feac819eSAndy Wingo Sym->setUsedInReloc(); 515feac819eSAndy Wingo Asm.registerSymbol(*Sym); 516feac819eSAndy Wingo } 517feac819eSAndy Wingo 518d1152a26SSam Clegg // Relocation other than R_WASM_TYPE_INDEX_LEB are required to be 5194d57fbd0SSam Clegg // against a named symbol. 520d1152a26SSam Clegg if (Type != wasm::R_WASM_TYPE_INDEX_LEB) { 5214d57fbd0SSam Clegg if (SymA->getName().empty()) 5224d57fbd0SSam Clegg report_fatal_error("relocations against un-named temporaries are not yet " 5234d57fbd0SSam Clegg "supported by wasm"); 5244d57fbd0SSam Clegg 5254d57fbd0SSam Clegg SymA->setUsedInReloc(); 5264d57fbd0SSam Clegg } 527ae03c1e7SSam Clegg 528492f7529SSam Clegg if (RefA->getKind() == MCSymbolRefExpr::VK_GOT) 529492f7529SSam Clegg SymA->setUsedInGOT(); 530492f7529SSam Clegg 531d934cb88SDan Gohman WasmRelocationEntry Rec(FixupOffset, SymA, C, Type, &FixupSection); 532d34e60caSNicola Zaghen LLVM_DEBUG(dbgs() << "WasmReloc: " << Rec << "\n"); 533d934cb88SDan Gohman 5346a31a0d6SSam Clegg if (FixupSection.isWasmData()) { 535d934cb88SDan Gohman DataRelocations.push_back(Rec); 5366a31a0d6SSam Clegg } else if (FixupSection.getKind().isText()) { 53712fd3da9SSam Clegg CodeRelocations.push_back(Rec); 5386a31a0d6SSam Clegg } else if (FixupSection.getKind().isMetadata()) { 5396a31a0d6SSam Clegg CustomSectionsRelocations[&FixupSection].push_back(Rec); 5406a31a0d6SSam Clegg } else { 54112fd3da9SSam Clegg llvm_unreachable("unexpected section type"); 542d934cb88SDan Gohman } 5436a31a0d6SSam Clegg } 544d934cb88SDan Gohman 545d934cb88SDan Gohman // Compute a value to write into the code at the location covered 54660ec3034SSam Clegg // by RelEntry. This value isn't used by the static linker; it just serves 54760ec3034SSam Clegg // to make the object format more readable and more likely to be directly 54860ec3034SSam Clegg // useable. 5493b29376eSWouter van Oortmerssen uint64_t 5507ee758d6SSam Clegg WasmObjectWriter::getProvisionalValue(const WasmRelocationEntry &RelEntry, 5517ee758d6SSam Clegg const MCAsmLayout &Layout) { 55248139ebcSWouter van Oortmerssen if ((RelEntry.Type == wasm::R_WASM_GLOBAL_INDEX_LEB || 55348139ebcSWouter van Oortmerssen RelEntry.Type == wasm::R_WASM_GLOBAL_INDEX_I32) && 55448139ebcSWouter van Oortmerssen !RelEntry.Symbol->isGlobal()) { 555492f7529SSam Clegg assert(GOTIndices.count(RelEntry.Symbol) > 0 && "symbol not found in GOT index space"); 556492f7529SSam Clegg return GOTIndices[RelEntry.Symbol]; 557492f7529SSam Clegg } 558492f7529SSam Clegg 55960ec3034SSam Clegg switch (RelEntry.Type) { 5602a7cac93SSam Clegg case wasm::R_WASM_TABLE_INDEX_REL_SLEB: 561d1152a26SSam Clegg case wasm::R_WASM_TABLE_INDEX_SLEB: 562cc1b9b68SWouter van Oortmerssen case wasm::R_WASM_TABLE_INDEX_SLEB64: 563cc1b9b68SWouter van Oortmerssen case wasm::R_WASM_TABLE_INDEX_I32: 564cc1b9b68SWouter van Oortmerssen case wasm::R_WASM_TABLE_INDEX_I64: { 565f9edbe95SSam Clegg // Provisional value is table address of the resolved symbol itself 5667ee758d6SSam Clegg const MCSymbolWasm *Base = 5677ee758d6SSam Clegg cast<MCSymbolWasm>(Layout.getBaseSymbol(*RelEntry.Symbol)); 5687ee758d6SSam Clegg assert(Base->isFunction()); 56981443ac1SSam Clegg if (RelEntry.Type == wasm::R_WASM_TABLE_INDEX_REL_SLEB) 5707ee758d6SSam Clegg return TableIndices[Base] - InitialTableOffset; 57181443ac1SSam Clegg else 5727ee758d6SSam Clegg return TableIndices[Base]; 573f9edbe95SSam Clegg } 574d1152a26SSam Clegg case wasm::R_WASM_TYPE_INDEX_LEB: 5756c899ba6SSam Clegg // Provisional value is same as the index 57660ec3034SSam Clegg return getRelocationIndexValue(RelEntry); 577d1152a26SSam Clegg case wasm::R_WASM_FUNCTION_INDEX_LEB: 578d1152a26SSam Clegg case wasm::R_WASM_GLOBAL_INDEX_LEB: 57948139ebcSWouter van Oortmerssen case wasm::R_WASM_GLOBAL_INDEX_I32: 580d1152a26SSam Clegg case wasm::R_WASM_EVENT_INDEX_LEB: 58169e2797eSPaulo Matos case wasm::R_WASM_TABLE_NUMBER_LEB: 582da419bdbSHeejin Ahn // Provisional value is function/global/event Wasm index 583492f7529SSam Clegg assert(WasmIndices.count(RelEntry.Symbol) > 0 && "symbol not found in wasm index space"); 5846c899ba6SSam Clegg return WasmIndices[RelEntry.Symbol]; 585d1152a26SSam Clegg case wasm::R_WASM_FUNCTION_OFFSET_I32: 58616f02431SWouter van Oortmerssen case wasm::R_WASM_FUNCTION_OFFSET_I64: 587d1152a26SSam Clegg case wasm::R_WASM_SECTION_OFFSET_I32: { 5886a31a0d6SSam Clegg const auto &Section = 5896a31a0d6SSam Clegg static_cast<const MCSectionWasm &>(RelEntry.Symbol->getSection()); 5906a31a0d6SSam Clegg return Section.getSectionOffset() + RelEntry.Addend; 5916a31a0d6SSam Clegg } 592d1152a26SSam Clegg case wasm::R_WASM_MEMORY_ADDR_LEB: 5933b29376eSWouter van Oortmerssen case wasm::R_WASM_MEMORY_ADDR_LEB64: 5943b29376eSWouter van Oortmerssen case wasm::R_WASM_MEMORY_ADDR_SLEB: 5953b29376eSWouter van Oortmerssen case wasm::R_WASM_MEMORY_ADDR_SLEB64: 5962a7cac93SSam Clegg case wasm::R_WASM_MEMORY_ADDR_REL_SLEB: 5973b29376eSWouter van Oortmerssen case wasm::R_WASM_MEMORY_ADDR_REL_SLEB64: 5983b29376eSWouter van Oortmerssen case wasm::R_WASM_MEMORY_ADDR_I32: 599a28a4662SSam Clegg case wasm::R_WASM_MEMORY_ADDR_I64: 600a28a4662SSam Clegg case wasm::R_WASM_MEMORY_ADDR_TLS_SLEB: { 6019c7b5808SDominic Chen // Provisional value is address of the global plus the offset 6027ee758d6SSam Clegg const MCSymbolWasm *Base = 6037ee758d6SSam Clegg cast<MCSymbolWasm>(Layout.getBaseSymbol(*RelEntry.Symbol)); 604270ed1b3SSam Clegg // For undefined symbols, use zero 6057ee758d6SSam Clegg if (!Base->isDefined()) 606270ed1b3SSam Clegg return 0; 6079c7b5808SDominic Chen const wasm::WasmDataReference &BaseRef = DataLocations[Base], 6089c7b5808SDominic Chen &SymRef = DataLocations[RelEntry.Symbol]; 6099c7b5808SDominic Chen const WasmDataSegment &Segment = DataSegments[BaseRef.Segment]; 610d934cb88SDan Gohman // Ignore overflow. LLVM allows address arithmetic to silently wrap. 6119c7b5808SDominic Chen return Segment.Offset + BaseRef.Offset + SymRef.Offset + RelEntry.Addend; 61260ec3034SSam Clegg } 61360ec3034SSam Clegg default: 61460ec3034SSam Clegg llvm_unreachable("invalid relocation type"); 61560ec3034SSam Clegg } 616d934cb88SDan Gohman } 617d934cb88SDan Gohman 618759631c7SSam Clegg static void addData(SmallVectorImpl<char> &DataBytes, 61963ebb813SSam Clegg MCSectionWasm &DataSection) { 6207d1ff446SFangrui Song LLVM_DEBUG(errs() << "addData: " << DataSection.getName() << "\n"); 621759631c7SSam Clegg 62263ebb813SSam Clegg DataBytes.resize(alignTo(DataBytes.size(), DataSection.getAlignment())); 62363ebb813SSam Clegg 624759631c7SSam Clegg for (const MCFragment &Frag : DataSection) { 625759631c7SSam Clegg if (Frag.hasInstructions()) 626759631c7SSam Clegg report_fatal_error("only data supported in data sections"); 627759631c7SSam Clegg 628759631c7SSam Clegg if (auto *Align = dyn_cast<MCAlignFragment>(&Frag)) { 629759631c7SSam Clegg if (Align->getValueSize() != 1) 630759631c7SSam Clegg report_fatal_error("only byte values supported for alignment"); 631759631c7SSam Clegg // If nops are requested, use zeros, as this is the data section. 632759631c7SSam Clegg uint8_t Value = Align->hasEmitNops() ? 0 : Align->getValue(); 633f208f631SHeejin Ahn uint64_t Size = 634f208f631SHeejin Ahn std::min<uint64_t>(alignTo(DataBytes.size(), Align->getAlignment()), 635f208f631SHeejin Ahn DataBytes.size() + Align->getMaxBytesToEmit()); 636759631c7SSam Clegg DataBytes.resize(Size, Value); 637759631c7SSam Clegg } else if (auto *Fill = dyn_cast<MCFillFragment>(&Frag)) { 638588fad4dSNirav Dave int64_t NumValues; 639588fad4dSNirav Dave if (!Fill->getNumValues().evaluateAsAbsolute(NumValues)) 640d707c370SRafael Espindola llvm_unreachable("The fill should be an assembler constant"); 641588fad4dSNirav Dave DataBytes.insert(DataBytes.end(), Fill->getValueSize() * NumValues, 642588fad4dSNirav Dave Fill->getValue()); 64324faf859SHeejin Ahn } else if (auto *LEB = dyn_cast<MCLEBFragment>(&Frag)) { 64424faf859SHeejin Ahn const SmallVectorImpl<char> &Contents = LEB->getContents(); 64524faf859SHeejin Ahn DataBytes.insert(DataBytes.end(), Contents.begin(), Contents.end()); 646759631c7SSam Clegg } else { 647759631c7SSam Clegg const auto &DataFrag = cast<MCDataFragment>(Frag); 648759631c7SSam Clegg const SmallVectorImpl<char> &Contents = DataFrag.getContents(); 649759631c7SSam Clegg DataBytes.insert(DataBytes.end(), Contents.begin(), Contents.end()); 650759631c7SSam Clegg } 651759631c7SSam Clegg } 652759631c7SSam Clegg 653d34e60caSNicola Zaghen LLVM_DEBUG(dbgs() << "addData -> " << DataBytes.size() << "\n"); 654759631c7SSam Clegg } 655759631c7SSam Clegg 65660ec3034SSam Clegg uint32_t 65760ec3034SSam Clegg WasmObjectWriter::getRelocationIndexValue(const WasmRelocationEntry &RelEntry) { 658d1152a26SSam Clegg if (RelEntry.Type == wasm::R_WASM_TYPE_INDEX_LEB) { 659b7787fd0SSam Clegg if (!TypeIndices.count(RelEntry.Symbol)) 660b7787fd0SSam Clegg report_fatal_error("symbol not found in type index space: " + 661b7787fd0SSam Clegg RelEntry.Symbol->getName()); 662acd7d2b0SSam Clegg return TypeIndices[RelEntry.Symbol]; 663acd7d2b0SSam Clegg } 66460ec3034SSam Clegg 66525d8e68bSSam Clegg return RelEntry.Symbol->getIndex(); 666acd7d2b0SSam Clegg } 667acd7d2b0SSam Clegg 668d934cb88SDan Gohman // Apply the portions of the relocation records that we can handle ourselves 669d934cb88SDan Gohman // directly. 670acd7d2b0SSam Clegg void WasmObjectWriter::applyRelocations( 6717ee758d6SSam Clegg ArrayRef<WasmRelocationEntry> Relocations, uint64_t ContentsOffset, 6727ee758d6SSam Clegg const MCAsmLayout &Layout) { 6730ff28fa6SDerek Schuff auto &Stream = static_cast<raw_pwrite_stream &>(W->OS); 674d934cb88SDan Gohman for (const WasmRelocationEntry &RelEntry : Relocations) { 675d934cb88SDan Gohman uint64_t Offset = ContentsOffset + 676d934cb88SDan Gohman RelEntry.FixupSection->getSectionOffset() + 677d934cb88SDan Gohman RelEntry.Offset; 678d934cb88SDan Gohman 679d34e60caSNicola Zaghen LLVM_DEBUG(dbgs() << "applyRelocation: " << RelEntry << "\n"); 6807ee758d6SSam Clegg auto Value = getProvisionalValue(RelEntry, Layout); 68160ec3034SSam Clegg 682acd7d2b0SSam Clegg switch (RelEntry.Type) { 683d1152a26SSam Clegg case wasm::R_WASM_FUNCTION_INDEX_LEB: 684d1152a26SSam Clegg case wasm::R_WASM_TYPE_INDEX_LEB: 685d1152a26SSam Clegg case wasm::R_WASM_GLOBAL_INDEX_LEB: 686d1152a26SSam Clegg case wasm::R_WASM_MEMORY_ADDR_LEB: 687d1152a26SSam Clegg case wasm::R_WASM_EVENT_INDEX_LEB: 68869e2797eSPaulo Matos case wasm::R_WASM_TABLE_NUMBER_LEB: 6893b29376eSWouter van Oortmerssen writePatchableLEB<5>(Stream, Value, Offset); 6903b29376eSWouter van Oortmerssen break; 6913b29376eSWouter van Oortmerssen case wasm::R_WASM_MEMORY_ADDR_LEB64: 6923b29376eSWouter van Oortmerssen writePatchableLEB<10>(Stream, Value, Offset); 693d934cb88SDan Gohman break; 694d1152a26SSam Clegg case wasm::R_WASM_TABLE_INDEX_I32: 695d1152a26SSam Clegg case wasm::R_WASM_MEMORY_ADDR_I32: 696d1152a26SSam Clegg case wasm::R_WASM_FUNCTION_OFFSET_I32: 697d1152a26SSam Clegg case wasm::R_WASM_SECTION_OFFSET_I32: 69848139ebcSWouter van Oortmerssen case wasm::R_WASM_GLOBAL_INDEX_I32: 6990a6c4d8dSSam Clegg patchI32(Stream, Value, Offset); 700d934cb88SDan Gohman break; 701cc1b9b68SWouter van Oortmerssen case wasm::R_WASM_TABLE_INDEX_I64: 7023b29376eSWouter van Oortmerssen case wasm::R_WASM_MEMORY_ADDR_I64: 70316f02431SWouter van Oortmerssen case wasm::R_WASM_FUNCTION_OFFSET_I64: 7043b29376eSWouter van Oortmerssen patchI64(Stream, Value, Offset); 7053b29376eSWouter van Oortmerssen break; 706d1152a26SSam Clegg case wasm::R_WASM_TABLE_INDEX_SLEB: 7072a7cac93SSam Clegg case wasm::R_WASM_TABLE_INDEX_REL_SLEB: 708d1152a26SSam Clegg case wasm::R_WASM_MEMORY_ADDR_SLEB: 7092a7cac93SSam Clegg case wasm::R_WASM_MEMORY_ADDR_REL_SLEB: 710a28a4662SSam Clegg case wasm::R_WASM_MEMORY_ADDR_TLS_SLEB: 7113b29376eSWouter van Oortmerssen writePatchableSLEB<5>(Stream, Value, Offset); 7123b29376eSWouter van Oortmerssen break; 713cc1b9b68SWouter van Oortmerssen case wasm::R_WASM_TABLE_INDEX_SLEB64: 7143b29376eSWouter van Oortmerssen case wasm::R_WASM_MEMORY_ADDR_SLEB64: 7153b29376eSWouter van Oortmerssen case wasm::R_WASM_MEMORY_ADDR_REL_SLEB64: 7163b29376eSWouter van Oortmerssen writePatchableSLEB<10>(Stream, Value, Offset); 71760ec3034SSam Clegg break; 718d934cb88SDan Gohman default: 7199d24fb7fSSam Clegg llvm_unreachable("invalid relocation type"); 720d934cb88SDan Gohman } 721d934cb88SDan Gohman } 72218eafb6cSDan Gohman } 72318eafb6cSDan Gohman 7244c41fb5aSHeejin Ahn void WasmObjectWriter::writeTypeSection( 7254c41fb5aSHeejin Ahn ArrayRef<wasm::WasmSignature> Signatures) { 726da419bdbSHeejin Ahn if (Signatures.empty()) 7279e15f359SSam Clegg return; 7289e15f359SSam Clegg 7299e15f359SSam Clegg SectionBookkeeping Section; 7309e15f359SSam Clegg startSection(Section, wasm::WASM_SEC_TYPE); 7319e15f359SSam Clegg 7320ff28fa6SDerek Schuff encodeULEB128(Signatures.size(), W->OS); 7339e15f359SSam Clegg 7344c41fb5aSHeejin Ahn for (const wasm::WasmSignature &Sig : Signatures) { 7350ff28fa6SDerek Schuff W->OS << char(wasm::WASM_TYPE_FUNC); 7360ff28fa6SDerek Schuff encodeULEB128(Sig.Params.size(), W->OS); 737da419bdbSHeejin Ahn for (wasm::ValType Ty : Sig.Params) 7389e15f359SSam Clegg writeValueType(Ty); 7390ff28fa6SDerek Schuff encodeULEB128(Sig.Returns.size(), W->OS); 740da419bdbSHeejin Ahn for (wasm::ValType Ty : Sig.Returns) 7419e15f359SSam Clegg writeValueType(Ty); 7429e15f359SSam Clegg } 7439e15f359SSam Clegg 7449e15f359SSam Clegg endSection(Section); 7459e15f359SSam Clegg } 7469e15f359SSam Clegg 7478defa952SSam Clegg void WasmObjectWriter::writeImportSection(ArrayRef<wasm::WasmImport> Imports, 7484d135b04SWouter van Oortmerssen uint64_t DataSize, 749f950b24aSSam Clegg uint32_t NumElements) { 7509e15f359SSam Clegg if (Imports.empty()) 7519e15f359SSam Clegg return; 7529e15f359SSam Clegg 7534d135b04SWouter van Oortmerssen uint64_t NumPages = (DataSize + wasm::WasmPageSize - 1) / wasm::WasmPageSize; 754f950b24aSSam Clegg 7559e15f359SSam Clegg SectionBookkeeping Section; 7569e15f359SSam Clegg startSection(Section, wasm::WASM_SEC_IMPORT); 7579e15f359SSam Clegg 7580ff28fa6SDerek Schuff encodeULEB128(Imports.size(), W->OS); 7598defa952SSam Clegg for (const wasm::WasmImport &Import : Imports) { 7608defa952SSam Clegg writeString(Import.Module); 7618defa952SSam Clegg writeString(Import.Field); 7620ff28fa6SDerek Schuff W->OS << char(Import.Kind); 7639e15f359SSam Clegg 7649e15f359SSam Clegg switch (Import.Kind) { 7659e15f359SSam Clegg case wasm::WASM_EXTERNAL_FUNCTION: 7660ff28fa6SDerek Schuff encodeULEB128(Import.SigIndex, W->OS); 7679e15f359SSam Clegg break; 7689e15f359SSam Clegg case wasm::WASM_EXTERNAL_GLOBAL: 7690ff28fa6SDerek Schuff W->OS << char(Import.Global.Type); 7700ff28fa6SDerek Schuff W->OS << char(Import.Global.Mutable ? 1 : 0); 7719e15f359SSam Clegg break; 772f950b24aSSam Clegg case wasm::WASM_EXTERNAL_MEMORY: 7730ff28fa6SDerek Schuff encodeULEB128(Import.Memory.Flags, W->OS); 7740ff28fa6SDerek Schuff encodeULEB128(NumPages, W->OS); // initial 775f950b24aSSam Clegg break; 776f950b24aSSam Clegg case wasm::WASM_EXTERNAL_TABLE: 7770ff28fa6SDerek Schuff W->OS << char(Import.Table.ElemType); 7780ff28fa6SDerek Schuff encodeULEB128(0, W->OS); // flags 7790ff28fa6SDerek Schuff encodeULEB128(NumElements, W->OS); // initial 780f950b24aSSam Clegg break; 781da419bdbSHeejin Ahn case wasm::WASM_EXTERNAL_EVENT: 7820ff28fa6SDerek Schuff encodeULEB128(Import.Event.Attribute, W->OS); 7830ff28fa6SDerek Schuff encodeULEB128(Import.Event.SigIndex, W->OS); 784da419bdbSHeejin Ahn break; 7859e15f359SSam Clegg default: 7869e15f359SSam Clegg llvm_unreachable("unsupported import kind"); 7879e15f359SSam Clegg } 7889e15f359SSam Clegg } 7899e15f359SSam Clegg 7909e15f359SSam Clegg endSection(Section); 7919e15f359SSam Clegg } 7929e15f359SSam Clegg 793457fb0b4SSam Clegg void WasmObjectWriter::writeFunctionSection(ArrayRef<WasmFunction> Functions) { 7949e15f359SSam Clegg if (Functions.empty()) 7959e15f359SSam Clegg return; 7969e15f359SSam Clegg 7979e15f359SSam Clegg SectionBookkeeping Section; 7989e15f359SSam Clegg startSection(Section, wasm::WASM_SEC_FUNCTION); 7999e15f359SSam Clegg 8000ff28fa6SDerek Schuff encodeULEB128(Functions.size(), W->OS); 8019e15f359SSam Clegg for (const WasmFunction &Func : Functions) 8020ff28fa6SDerek Schuff encodeULEB128(Func.SigIndex, W->OS); 8039e15f359SSam Clegg 8049e15f359SSam Clegg endSection(Section); 8059e15f359SSam Clegg } 8069e15f359SSam Clegg 807da419bdbSHeejin Ahn void WasmObjectWriter::writeEventSection(ArrayRef<wasm::WasmEventType> Events) { 808da419bdbSHeejin Ahn if (Events.empty()) 809da419bdbSHeejin Ahn return; 810da419bdbSHeejin Ahn 811da419bdbSHeejin Ahn SectionBookkeeping Section; 812da419bdbSHeejin Ahn startSection(Section, wasm::WASM_SEC_EVENT); 813da419bdbSHeejin Ahn 8140ff28fa6SDerek Schuff encodeULEB128(Events.size(), W->OS); 815da419bdbSHeejin Ahn for (const wasm::WasmEventType &Event : Events) { 8160ff28fa6SDerek Schuff encodeULEB128(Event.Attribute, W->OS); 8170ff28fa6SDerek Schuff encodeULEB128(Event.SigIndex, W->OS); 818da419bdbSHeejin Ahn } 819da419bdbSHeejin Ahn 820da419bdbSHeejin Ahn endSection(Section); 821da419bdbSHeejin Ahn } 822da419bdbSHeejin Ahn 8230a6c4d8dSSam Clegg void WasmObjectWriter::writeGlobalSection(ArrayRef<wasm::WasmGlobal> Globals) { 8240a6c4d8dSSam Clegg if (Globals.empty()) 8250a6c4d8dSSam Clegg return; 8260a6c4d8dSSam Clegg 8270a6c4d8dSSam Clegg SectionBookkeeping Section; 8280a6c4d8dSSam Clegg startSection(Section, wasm::WASM_SEC_GLOBAL); 8290a6c4d8dSSam Clegg 8300ff28fa6SDerek Schuff encodeULEB128(Globals.size(), W->OS); 8310a6c4d8dSSam Clegg for (const wasm::WasmGlobal &Global : Globals) { 8320ff28fa6SDerek Schuff encodeULEB128(Global.Type.Type, W->OS); 8330ff28fa6SDerek Schuff W->OS << char(Global.Type.Mutable); 8340ff28fa6SDerek Schuff W->OS << char(Global.InitExpr.Opcode); 8350a6c4d8dSSam Clegg switch (Global.Type.Type) { 8360a6c4d8dSSam Clegg case wasm::WASM_TYPE_I32: 8370ff28fa6SDerek Schuff encodeSLEB128(0, W->OS); 8380a6c4d8dSSam Clegg break; 8390a6c4d8dSSam Clegg case wasm::WASM_TYPE_I64: 8400ff28fa6SDerek Schuff encodeSLEB128(0, W->OS); 8410a6c4d8dSSam Clegg break; 8420a6c4d8dSSam Clegg case wasm::WASM_TYPE_F32: 8430a6c4d8dSSam Clegg writeI32(0); 8440a6c4d8dSSam Clegg break; 8450a6c4d8dSSam Clegg case wasm::WASM_TYPE_F64: 8460a6c4d8dSSam Clegg writeI64(0); 8470a6c4d8dSSam Clegg break; 84879aad89dSSam Clegg case wasm::WASM_TYPE_EXTERNREF: 84979aad89dSSam Clegg writeValueType(wasm::ValType::EXTERNREF); 85079aad89dSSam Clegg break; 8510a6c4d8dSSam Clegg default: 8520a6c4d8dSSam Clegg llvm_unreachable("unexpected type"); 8530a6c4d8dSSam Clegg } 8540ff28fa6SDerek Schuff W->OS << char(wasm::WASM_OPCODE_END); 8550a6c4d8dSSam Clegg } 8560a6c4d8dSSam Clegg 8570a6c4d8dSSam Clegg endSection(Section); 8580a6c4d8dSSam Clegg } 8590a6c4d8dSSam Clegg 860388fb67bSPaulo Matos void WasmObjectWriter::writeTableSection(ArrayRef<wasm::WasmTable> Tables) { 861388fb67bSPaulo Matos if (Tables.empty()) 862388fb67bSPaulo Matos return; 863388fb67bSPaulo Matos 864388fb67bSPaulo Matos SectionBookkeeping Section; 865388fb67bSPaulo Matos startSection(Section, wasm::WASM_SEC_TABLE); 866388fb67bSPaulo Matos 867388fb67bSPaulo Matos encodeULEB128(Tables.size(), W->OS); 868388fb67bSPaulo Matos for (const wasm::WasmTable &Table : Tables) { 8691933c9d4SAndy Wingo encodeULEB128(Table.Type.ElemType, W->OS); 8701933c9d4SAndy Wingo encodeULEB128(Table.Type.Limits.Flags, W->OS); 8711933c9d4SAndy Wingo encodeULEB128(Table.Type.Limits.Initial, W->OS); 8721933c9d4SAndy Wingo if (Table.Type.Limits.Flags & wasm::WASM_LIMITS_FLAG_HAS_MAX) 8731933c9d4SAndy Wingo encodeULEB128(Table.Type.Limits.Maximum, W->OS); 874388fb67bSPaulo Matos } 875388fb67bSPaulo Matos endSection(Section); 876388fb67bSPaulo Matos } 877388fb67bSPaulo Matos 8788defa952SSam Clegg void WasmObjectWriter::writeExportSection(ArrayRef<wasm::WasmExport> Exports) { 8799e15f359SSam Clegg if (Exports.empty()) 8809e15f359SSam Clegg return; 8819e15f359SSam Clegg 8829e15f359SSam Clegg SectionBookkeeping Section; 8839e15f359SSam Clegg startSection(Section, wasm::WASM_SEC_EXPORT); 8849e15f359SSam Clegg 8850ff28fa6SDerek Schuff encodeULEB128(Exports.size(), W->OS); 8868defa952SSam Clegg for (const wasm::WasmExport &Export : Exports) { 8878defa952SSam Clegg writeString(Export.Name); 8880ff28fa6SDerek Schuff W->OS << char(Export.Kind); 8890ff28fa6SDerek Schuff encodeULEB128(Export.Index, W->OS); 8909e15f359SSam Clegg } 8919e15f359SSam Clegg 8929e15f359SSam Clegg endSection(Section); 8939e15f359SSam Clegg } 8949e15f359SSam Clegg 895457fb0b4SSam Clegg void WasmObjectWriter::writeElemSection(ArrayRef<uint32_t> TableElems) { 8969e15f359SSam Clegg if (TableElems.empty()) 8979e15f359SSam Clegg return; 8989e15f359SSam Clegg 8999e15f359SSam Clegg SectionBookkeeping Section; 9009e15f359SSam Clegg startSection(Section, wasm::WASM_SEC_ELEM); 9019e15f359SSam Clegg 9020ff28fa6SDerek Schuff encodeULEB128(1, W->OS); // number of "segments" 9030ff28fa6SDerek Schuff encodeULEB128(0, W->OS); // the table index 9049e15f359SSam Clegg 9059e15f359SSam Clegg // init expr for starting offset 9060ff28fa6SDerek Schuff W->OS << char(wasm::WASM_OPCODE_I32_CONST); 9070ff28fa6SDerek Schuff encodeSLEB128(InitialTableOffset, W->OS); 9080ff28fa6SDerek Schuff W->OS << char(wasm::WASM_OPCODE_END); 9099e15f359SSam Clegg 9100ff28fa6SDerek Schuff encodeULEB128(TableElems.size(), W->OS); 9119e15f359SSam Clegg for (uint32_t Elem : TableElems) 9120ff28fa6SDerek Schuff encodeULEB128(Elem, W->OS); 9139e15f359SSam Clegg 9149e15f359SSam Clegg endSection(Section); 9159e15f359SSam Clegg } 9169e15f359SSam Clegg 917fef8de66SThomas Lively void WasmObjectWriter::writeDataCountSection() { 918fef8de66SThomas Lively if (DataSegments.empty()) 919fef8de66SThomas Lively return; 920fef8de66SThomas Lively 921fef8de66SThomas Lively SectionBookkeeping Section; 922fef8de66SThomas Lively startSection(Section, wasm::WASM_SEC_DATACOUNT); 9230ff28fa6SDerek Schuff encodeULEB128(DataSegments.size(), W->OS); 924fef8de66SThomas Lively endSection(Section); 925fef8de66SThomas Lively } 926fef8de66SThomas Lively 927e49584a3SSam Clegg uint32_t WasmObjectWriter::writeCodeSection(const MCAssembler &Asm, 928457fb0b4SSam Clegg const MCAsmLayout &Layout, 929457fb0b4SSam Clegg ArrayRef<WasmFunction> Functions) { 9309e15f359SSam Clegg if (Functions.empty()) 931e49584a3SSam Clegg return 0; 9329e15f359SSam Clegg 9339e15f359SSam Clegg SectionBookkeeping Section; 9349e15f359SSam Clegg startSection(Section, wasm::WASM_SEC_CODE); 9359e15f359SSam Clegg 9360ff28fa6SDerek Schuff encodeULEB128(Functions.size(), W->OS); 9379e15f359SSam Clegg 9389e15f359SSam Clegg for (const WasmFunction &Func : Functions) { 939fe6414b0SSam Clegg auto &FuncSection = static_cast<MCSectionWasm &>(Func.Sym->getSection()); 9409e15f359SSam Clegg 9419e15f359SSam Clegg int64_t Size = 0; 9429e15f359SSam Clegg if (!Func.Sym->getSize()->evaluateAsAbsolute(Size, Layout)) 9439e15f359SSam Clegg report_fatal_error(".size expression must be evaluatable"); 9449e15f359SSam Clegg 9450ff28fa6SDerek Schuff encodeULEB128(Size, W->OS); 9460ff28fa6SDerek Schuff FuncSection.setSectionOffset(W->OS.tell() - Section.ContentsOffset); 9470ff28fa6SDerek Schuff Asm.writeSectionData(W->OS, &FuncSection, Layout); 9489e15f359SSam Clegg } 9499e15f359SSam Clegg 9509e15f359SSam Clegg // Apply fixups. 9517ee758d6SSam Clegg applyRelocations(CodeRelocations, Section.ContentsOffset, Layout); 9529e15f359SSam Clegg 9539e15f359SSam Clegg endSection(Section); 954e49584a3SSam Clegg return Section.Index; 9559e15f359SSam Clegg } 9569e15f359SSam Clegg 957e49584a3SSam Clegg uint32_t WasmObjectWriter::writeDataSection(const MCAsmLayout &Layout) { 9586c899ba6SSam Clegg if (DataSegments.empty()) 959e49584a3SSam Clegg return 0; 9609e15f359SSam Clegg 9619e15f359SSam Clegg SectionBookkeeping Section; 9629e15f359SSam Clegg startSection(Section, wasm::WASM_SEC_DATA); 9639e15f359SSam Clegg 9640ff28fa6SDerek Schuff encodeULEB128(DataSegments.size(), W->OS); // count 9657c395943SSam Clegg 9666c899ba6SSam Clegg for (const WasmDataSegment &Segment : DataSegments) { 9670ff28fa6SDerek Schuff encodeULEB128(Segment.InitFlags, W->OS); // flags 9682e150409SThomas Lively if (Segment.InitFlags & wasm::WASM_SEGMENT_HAS_MEMINDEX) 9690ff28fa6SDerek Schuff encodeULEB128(0, W->OS); // memory index 9702e150409SThomas Lively if ((Segment.InitFlags & wasm::WASM_SEGMENT_IS_PASSIVE) == 0) { 9710ff28fa6SDerek Schuff W->OS << char(Segment.Offset > INT32_MAX ? wasm::WASM_OPCODE_I64_CONST 9724d135b04SWouter van Oortmerssen : wasm::WASM_OPCODE_I32_CONST); 9730ff28fa6SDerek Schuff encodeSLEB128(Segment.Offset, W->OS); // offset 9740ff28fa6SDerek Schuff W->OS << char(wasm::WASM_OPCODE_END); 9752e150409SThomas Lively } 9760ff28fa6SDerek Schuff encodeULEB128(Segment.Data.size(), W->OS); // size 9770ff28fa6SDerek Schuff Segment.Section->setSectionOffset(W->OS.tell() - Section.ContentsOffset); 9780ff28fa6SDerek Schuff W->OS << Segment.Data; // data 9797c395943SSam Clegg } 9809e15f359SSam Clegg 9819e15f359SSam Clegg // Apply fixups. 9827ee758d6SSam Clegg applyRelocations(DataRelocations, Section.ContentsOffset, Layout); 9839e15f359SSam Clegg 9849e15f359SSam Clegg endSection(Section); 985e49584a3SSam Clegg return Section.Index; 9869e15f359SSam Clegg } 9879e15f359SSam Clegg 9886f08c84aSSam Clegg void WasmObjectWriter::writeRelocSection( 9896f08c84aSSam Clegg uint32_t SectionIndex, StringRef Name, 990f77dc2a8SSam Clegg std::vector<WasmRelocationEntry> &Relocs) { 9919e15f359SSam Clegg // See: https://github.com/WebAssembly/tool-conventions/blob/master/Linking.md 9929e15f359SSam Clegg // for descriptions of the reloc sections. 9939e15f359SSam Clegg 994f77dc2a8SSam Clegg if (Relocs.empty()) 9959e15f359SSam Clegg return; 9969e15f359SSam Clegg 997f77dc2a8SSam Clegg // First, ensure the relocations are sorted in offset order. In general they 998f77dc2a8SSam Clegg // should already be sorted since `recordRelocation` is called in offset 999f77dc2a8SSam Clegg // order, but for the code section we combine many MC sections into single 1000f77dc2a8SSam Clegg // wasm section, and this order is determined by the order of Asm.Symbols() 1001f77dc2a8SSam Clegg // not the sections order. 1002efd94c56SFangrui Song llvm::stable_sort( 1003efd94c56SFangrui Song Relocs, [](const WasmRelocationEntry &A, const WasmRelocationEntry &B) { 1004f77dc2a8SSam Clegg return (A.Offset + A.FixupSection->getSectionOffset()) < 1005f77dc2a8SSam Clegg (B.Offset + B.FixupSection->getSectionOffset()); 1006f77dc2a8SSam Clegg }); 1007f77dc2a8SSam Clegg 10089e15f359SSam Clegg SectionBookkeeping Section; 10096f08c84aSSam Clegg startCustomSection(Section, std::string("reloc.") + Name.str()); 10109e15f359SSam Clegg 10110ff28fa6SDerek Schuff encodeULEB128(SectionIndex, W->OS); 10120ff28fa6SDerek Schuff encodeULEB128(Relocs.size(), W->OS); 1013f77dc2a8SSam Clegg for (const WasmRelocationEntry &RelEntry : Relocs) { 1014f208f631SHeejin Ahn uint64_t Offset = 1015f208f631SHeejin Ahn RelEntry.Offset + RelEntry.FixupSection->getSectionOffset(); 10166f08c84aSSam Clegg uint32_t Index = getRelocationIndexValue(RelEntry); 10179e15f359SSam Clegg 10180ff28fa6SDerek Schuff W->OS << char(RelEntry.Type); 10190ff28fa6SDerek Schuff encodeULEB128(Offset, W->OS); 10200ff28fa6SDerek Schuff encodeULEB128(Index, W->OS); 10216f08c84aSSam Clegg if (RelEntry.hasAddend()) 10220ff28fa6SDerek Schuff encodeSLEB128(RelEntry.Addend, W->OS); 10239e15f359SSam Clegg } 10249e15f359SSam Clegg 10259e15f359SSam Clegg endSection(Section); 10269e15f359SSam Clegg } 10279e15f359SSam Clegg 10286a31a0d6SSam Clegg void WasmObjectWriter::writeCustomRelocSections() { 10296a31a0d6SSam Clegg for (const auto &Sec : CustomSections) { 10306a31a0d6SSam Clegg auto &Relocations = CustomSectionsRelocations[Sec.Section]; 10316a31a0d6SSam Clegg writeRelocSection(Sec.OutputIndex, Sec.Name, Relocations); 10326a31a0d6SSam Clegg } 10336a31a0d6SSam Clegg } 10346a31a0d6SSam Clegg 10359e15f359SSam Clegg void WasmObjectWriter::writeLinkingMetaDataSection( 103686b4a09aSSam Clegg ArrayRef<wasm::WasmSymbolInfo> SymbolInfos, 1037ea7caceeSSam Clegg ArrayRef<std::pair<uint16_t, uint32_t>> InitFuncs, 1038ea7caceeSSam Clegg const std::map<StringRef, std::vector<WasmComdatEntry>> &Comdats) { 10399e15f359SSam Clegg SectionBookkeeping Section; 10402322a937SSam Clegg startCustomSection(Section, "linking"); 10410ff28fa6SDerek Schuff encodeULEB128(wasm::WasmMetadataVersion, W->OS); 10429e15f359SSam Clegg 10436bb5a41fSSam Clegg SectionBookkeeping SubSection; 10446c899ba6SSam Clegg if (SymbolInfos.size() != 0) { 10456c899ba6SSam Clegg startSection(SubSection, wasm::WASM_SYMBOL_TABLE); 10460ff28fa6SDerek Schuff encodeULEB128(SymbolInfos.size(), W->OS); 10476c899ba6SSam Clegg for (const wasm::WasmSymbolInfo &Sym : SymbolInfos) { 10480ff28fa6SDerek Schuff encodeULEB128(Sym.Kind, W->OS); 10490ff28fa6SDerek Schuff encodeULEB128(Sym.Flags, W->OS); 10506c899ba6SSam Clegg switch (Sym.Kind) { 10516c899ba6SSam Clegg case wasm::WASM_SYMBOL_TYPE_FUNCTION: 10526c899ba6SSam Clegg case wasm::WASM_SYMBOL_TYPE_GLOBAL: 1053da419bdbSHeejin Ahn case wasm::WASM_SYMBOL_TYPE_EVENT: 1054388fb67bSPaulo Matos case wasm::WASM_SYMBOL_TYPE_TABLE: 10550ff28fa6SDerek Schuff encodeULEB128(Sym.ElementIndex, W->OS); 105629874ceaSDan Gohman if ((Sym.Flags & wasm::WASM_SYMBOL_UNDEFINED) == 0 || 105729874ceaSDan Gohman (Sym.Flags & wasm::WASM_SYMBOL_EXPLICIT_NAME) != 0) 10586c899ba6SSam Clegg writeString(Sym.Name); 10596c899ba6SSam Clegg break; 10606c899ba6SSam Clegg case wasm::WASM_SYMBOL_TYPE_DATA: 10616c899ba6SSam Clegg writeString(Sym.Name); 10626c899ba6SSam Clegg if ((Sym.Flags & wasm::WASM_SYMBOL_UNDEFINED) == 0) { 10630ff28fa6SDerek Schuff encodeULEB128(Sym.DataRef.Segment, W->OS); 10640ff28fa6SDerek Schuff encodeULEB128(Sym.DataRef.Offset, W->OS); 10650ff28fa6SDerek Schuff encodeULEB128(Sym.DataRef.Size, W->OS); 10666c899ba6SSam Clegg } 10676c899ba6SSam Clegg break; 10686a31a0d6SSam Clegg case wasm::WASM_SYMBOL_TYPE_SECTION: { 10696a31a0d6SSam Clegg const uint32_t SectionIndex = 10706a31a0d6SSam Clegg CustomSections[Sym.ElementIndex].OutputIndex; 10710ff28fa6SDerek Schuff encodeULEB128(SectionIndex, W->OS); 10726a31a0d6SSam Clegg break; 10736a31a0d6SSam Clegg } 10746c899ba6SSam Clegg default: 10756c899ba6SSam Clegg llvm_unreachable("unexpected kind"); 10766c899ba6SSam Clegg } 1077b7787fd0SSam Clegg } 1078b7787fd0SSam Clegg endSection(SubSection); 1079b7787fd0SSam Clegg } 10809e15f359SSam Clegg 10816c899ba6SSam Clegg if (DataSegments.size()) { 108263ebb813SSam Clegg startSection(SubSection, wasm::WASM_SEGMENT_INFO); 10830ff28fa6SDerek Schuff encodeULEB128(DataSegments.size(), W->OS); 10846c899ba6SSam Clegg for (const WasmDataSegment &Segment : DataSegments) { 1085d95ed959SSam Clegg writeString(Segment.Name); 10860ff28fa6SDerek Schuff encodeULEB128(Segment.Alignment, W->OS); 10870ff28fa6SDerek Schuff encodeULEB128(Segment.LinkerFlags, W->OS); 108863ebb813SSam Clegg } 1089d95ed959SSam Clegg endSection(SubSection); 1090d95ed959SSam Clegg } 1091d95ed959SSam Clegg 1092bafe6902SSam Clegg if (!InitFuncs.empty()) { 1093bafe6902SSam Clegg startSection(SubSection, wasm::WASM_INIT_FUNCS); 10940ff28fa6SDerek Schuff encodeULEB128(InitFuncs.size(), W->OS); 1095bafe6902SSam Clegg for (auto &StartFunc : InitFuncs) { 10960ff28fa6SDerek Schuff encodeULEB128(StartFunc.first, W->OS); // priority 10970ff28fa6SDerek Schuff encodeULEB128(StartFunc.second, W->OS); // function index 1098bafe6902SSam Clegg } 1099bafe6902SSam Clegg endSection(SubSection); 1100bafe6902SSam Clegg } 1101bafe6902SSam Clegg 1102ea7caceeSSam Clegg if (Comdats.size()) { 1103ea7caceeSSam Clegg startSection(SubSection, wasm::WASM_COMDAT_INFO); 11040ff28fa6SDerek Schuff encodeULEB128(Comdats.size(), W->OS); 1105ea7caceeSSam Clegg for (const auto &C : Comdats) { 1106ea7caceeSSam Clegg writeString(C.first); 11070ff28fa6SDerek Schuff encodeULEB128(0, W->OS); // flags for future use 11080ff28fa6SDerek Schuff encodeULEB128(C.second.size(), W->OS); 1109ea7caceeSSam Clegg for (const WasmComdatEntry &Entry : C.second) { 11100ff28fa6SDerek Schuff encodeULEB128(Entry.Kind, W->OS); 11110ff28fa6SDerek Schuff encodeULEB128(Entry.Index, W->OS); 1112ea7caceeSSam Clegg } 1113ea7caceeSSam Clegg } 1114ea7caceeSSam Clegg endSection(SubSection); 1115ea7caceeSSam Clegg } 1116ea7caceeSSam Clegg 11179e15f359SSam Clegg endSection(Section); 11189e15f359SSam Clegg } 11199e15f359SSam Clegg 1120cbda16ebSThomas Lively void WasmObjectWriter::writeCustomSection(WasmCustomSection &CustomSection, 1121cbda16ebSThomas Lively const MCAssembler &Asm, 11226a31a0d6SSam Clegg const MCAsmLayout &Layout) { 1123cfd44a2eSSam Clegg SectionBookkeeping Section; 11246a31a0d6SSam Clegg auto *Sec = CustomSection.Section; 11252322a937SSam Clegg startCustomSection(Section, CustomSection.Name); 11266a31a0d6SSam Clegg 11270ff28fa6SDerek Schuff Sec->setSectionOffset(W->OS.tell() - Section.ContentsOffset); 11280ff28fa6SDerek Schuff Asm.writeSectionData(W->OS, Sec, Layout); 11296a31a0d6SSam Clegg 11306a31a0d6SSam Clegg CustomSection.OutputContentsOffset = Section.ContentsOffset; 11316a31a0d6SSam Clegg CustomSection.OutputIndex = Section.Index; 11326a31a0d6SSam Clegg 1133cfd44a2eSSam Clegg endSection(Section); 11346a31a0d6SSam Clegg 11356a31a0d6SSam Clegg // Apply fixups. 11364d57fbd0SSam Clegg auto &Relocations = CustomSectionsRelocations[CustomSection.Section]; 11377ee758d6SSam Clegg applyRelocations(Relocations, CustomSection.OutputContentsOffset, Layout); 11386a31a0d6SSam Clegg } 11396a31a0d6SSam Clegg 11405e3d33a7SSam Clegg uint32_t WasmObjectWriter::getFunctionType(const MCSymbolWasm &Symbol) { 11415e3d33a7SSam Clegg assert(Symbol.isFunction()); 11425e3d33a7SSam Clegg assert(TypeIndices.count(&Symbol)); 11435e3d33a7SSam Clegg return TypeIndices[&Symbol]; 11445e3d33a7SSam Clegg } 11455e3d33a7SSam Clegg 1146da419bdbSHeejin Ahn uint32_t WasmObjectWriter::getEventType(const MCSymbolWasm &Symbol) { 1147da419bdbSHeejin Ahn assert(Symbol.isEvent()); 1148da419bdbSHeejin Ahn assert(TypeIndices.count(&Symbol)); 1149da419bdbSHeejin Ahn return TypeIndices[&Symbol]; 1150da419bdbSHeejin Ahn } 1151da419bdbSHeejin Ahn 115238b12f5eSHeejin Ahn void WasmObjectWriter::registerFunctionType(const MCSymbolWasm &Symbol) { 11535e3d33a7SSam Clegg assert(Symbol.isFunction()); 11545e3d33a7SSam Clegg 11554c41fb5aSHeejin Ahn wasm::WasmSignature S; 11567ee758d6SSam Clegg 11577ee758d6SSam Clegg if (auto *Sig = Symbol.getSignature()) { 1158da419bdbSHeejin Ahn S.Returns = Sig->Returns; 1159da419bdbSHeejin Ahn S.Params = Sig->Params; 116077a7a380SDerek Schuff } 11615e3d33a7SSam Clegg 1162da419bdbSHeejin Ahn auto Pair = SignatureIndices.insert(std::make_pair(S, Signatures.size())); 11635e3d33a7SSam Clegg if (Pair.second) 1164da419bdbSHeejin Ahn Signatures.push_back(S); 11655e3d33a7SSam Clegg TypeIndices[&Symbol] = Pair.first->second; 11665e3d33a7SSam Clegg 1167d34e60caSNicola Zaghen LLVM_DEBUG(dbgs() << "registerFunctionType: " << Symbol 1168d34e60caSNicola Zaghen << " new:" << Pair.second << "\n"); 1169d34e60caSNicola Zaghen LLVM_DEBUG(dbgs() << " -> type index: " << Pair.first->second << "\n"); 11705e3d33a7SSam Clegg } 11715e3d33a7SSam Clegg 117238b12f5eSHeejin Ahn void WasmObjectWriter::registerEventType(const MCSymbolWasm &Symbol) { 1173da419bdbSHeejin Ahn assert(Symbol.isEvent()); 1174da419bdbSHeejin Ahn 1175da419bdbSHeejin Ahn // TODO Currently we don't generate imported exceptions, but if we do, we 1176da419bdbSHeejin Ahn // should have a way of infering types of imported exceptions. 11774c41fb5aSHeejin Ahn wasm::WasmSignature S; 1178da419bdbSHeejin Ahn if (auto *Sig = Symbol.getSignature()) { 1179da419bdbSHeejin Ahn S.Returns = Sig->Returns; 1180da419bdbSHeejin Ahn S.Params = Sig->Params; 1181da419bdbSHeejin Ahn } 1182da419bdbSHeejin Ahn 1183da419bdbSHeejin Ahn auto Pair = SignatureIndices.insert(std::make_pair(S, Signatures.size())); 1184da419bdbSHeejin Ahn if (Pair.second) 1185da419bdbSHeejin Ahn Signatures.push_back(S); 1186da419bdbSHeejin Ahn TypeIndices[&Symbol] = Pair.first->second; 1187da419bdbSHeejin Ahn 1188da419bdbSHeejin Ahn LLVM_DEBUG(dbgs() << "registerEventType: " << Symbol << " new:" << Pair.second 1189da419bdbSHeejin Ahn << "\n"); 1190da419bdbSHeejin Ahn LLVM_DEBUG(dbgs() << " -> type index: " << Pair.first->second << "\n"); 1191da419bdbSHeejin Ahn } 1192da419bdbSHeejin Ahn 11934d57fbd0SSam Clegg static bool isInSymtab(const MCSymbolWasm &Sym) { 11943ea1c62cSSam Clegg if (Sym.isUsedInReloc() || Sym.isUsedInInitArray()) 11954d57fbd0SSam Clegg return true; 11964d57fbd0SSam Clegg 11974d57fbd0SSam Clegg if (Sym.isComdat() && !Sym.isDefined()) 11984d57fbd0SSam Clegg return false; 11994d57fbd0SSam Clegg 12003ea1c62cSSam Clegg if (Sym.isTemporary()) 12014d57fbd0SSam Clegg return false; 12024d57fbd0SSam Clegg 12034d57fbd0SSam Clegg if (Sym.isSection()) 12044d57fbd0SSam Clegg return false; 12054d57fbd0SSam Clegg 12064d57fbd0SSam Clegg return true; 12074d57fbd0SSam Clegg } 12080ff28fa6SDerek Schuff void WasmObjectWriter::prepareImports( 12090ff28fa6SDerek Schuff SmallVectorImpl<wasm::WasmImport> &Imports, MCAssembler &Asm, 121018eafb6cSDan Gohman const MCAsmLayout &Layout) { 1211f950b24aSSam Clegg // For now, always emit the memory import, since loads and stores are not 1212f950b24aSSam Clegg // valid without it. In the future, we could perhaps be more clever and omit 1213f950b24aSSam Clegg // it if there are no loads or stores. 12148defa952SSam Clegg wasm::WasmImport MemImport; 12151ed3a046SSam Clegg MemImport.Module = "env"; 12161ed3a046SSam Clegg MemImport.Field = "__linear_memory"; 1217f950b24aSSam Clegg MemImport.Kind = wasm::WASM_EXTERNAL_MEMORY; 1218fd0964aeSWouter van Oortmerssen MemImport.Memory.Flags = is64Bit() ? wasm::WASM_LIMITS_FLAG_IS_64 1219fd0964aeSWouter van Oortmerssen : wasm::WASM_LIMITS_FLAG_NONE; 1220f950b24aSSam Clegg Imports.push_back(MemImport); 1221f950b24aSSam Clegg 1222da419bdbSHeejin Ahn // Populate SignatureIndices, and Imports and WasmIndices for undefined 1223586320c0SNicholas Wilson // symbols. This must be done before populating WasmIndices for defined 1224586320c0SNicholas Wilson // symbols. 122532ce5ca0SDan Gohman for (const MCSymbol &S : Asm.symbols()) { 122632ce5ca0SDan Gohman const auto &WS = static_cast<const MCSymbolWasm &>(S); 122732ce5ca0SDan Gohman 122832ce5ca0SDan Gohman // Register types for all functions, including those with private linkage 12299f3fe42eSSam Clegg // (because wasm always needs a type signature). 12307ee758d6SSam Clegg if (WS.isFunction()) { 12319c7b5808SDominic Chen const auto *BS = Layout.getBaseSymbol(S); 12329c7b5808SDominic Chen if (!BS) 12339c7b5808SDominic Chen report_fatal_error(Twine(S.getName()) + 12349c7b5808SDominic Chen ": absolute addressing not supported!"); 12359c7b5808SDominic Chen registerFunctionType(*cast<MCSymbolWasm>(BS)); 12367ee758d6SSam Clegg } 123732ce5ca0SDan Gohman 1238da419bdbSHeejin Ahn if (WS.isEvent()) 1239da419bdbSHeejin Ahn registerEventType(WS); 1240da419bdbSHeejin Ahn 124132ce5ca0SDan Gohman if (WS.isTemporary()) 124232ce5ca0SDan Gohman continue; 124332ce5ca0SDan Gohman 124432ce5ca0SDan Gohman // If the symbol is not defined in this translation unit, import it. 12456c899ba6SSam Clegg if (!WS.isDefined() && !WS.isComdat()) { 12466c899ba6SSam Clegg if (WS.isFunction()) { 12478defa952SSam Clegg wasm::WasmImport Import; 1248f726e445SDan Gohman Import.Module = WS.getImportModule(); 1249f726e445SDan Gohman Import.Field = WS.getImportName(); 125032ce5ca0SDan Gohman Import.Kind = wasm::WASM_EXTERNAL_FUNCTION; 12518defa952SSam Clegg Import.SigIndex = getFunctionType(WS); 12526c899ba6SSam Clegg Imports.push_back(Import); 1253492f7529SSam Clegg assert(WasmIndices.count(&WS) == 0); 12546c899ba6SSam Clegg WasmIndices[&WS] = NumFunctionImports++; 12556c899ba6SSam Clegg } else if (WS.isGlobal()) { 125615f349f7SNicholas Wilson if (WS.isWeak()) 125715f349f7SNicholas Wilson report_fatal_error("undefined global symbol cannot be weak"); 125815f349f7SNicholas Wilson 12596c899ba6SSam Clegg wasm::WasmImport Import; 1260f726e445SDan Gohman Import.Field = WS.getImportName(); 126132ce5ca0SDan Gohman Import.Kind = wasm::WASM_EXTERNAL_GLOBAL; 1262492f7529SSam Clegg Import.Module = WS.getImportModule(); 12636c899ba6SSam Clegg Import.Global = WS.getGlobalType(); 12646c899ba6SSam Clegg Imports.push_back(Import); 1265492f7529SSam Clegg assert(WasmIndices.count(&WS) == 0); 12666c899ba6SSam Clegg WasmIndices[&WS] = NumGlobalImports++; 1267da419bdbSHeejin Ahn } else if (WS.isEvent()) { 1268da419bdbSHeejin Ahn if (WS.isWeak()) 1269da419bdbSHeejin Ahn report_fatal_error("undefined event symbol cannot be weak"); 1270da419bdbSHeejin Ahn 1271da419bdbSHeejin Ahn wasm::WasmImport Import; 1272f726e445SDan Gohman Import.Module = WS.getImportModule(); 1273f726e445SDan Gohman Import.Field = WS.getImportName(); 1274da419bdbSHeejin Ahn Import.Kind = wasm::WASM_EXTERNAL_EVENT; 1275da419bdbSHeejin Ahn Import.Event.Attribute = wasm::WASM_EVENT_ATTRIBUTE_EXCEPTION; 1276da419bdbSHeejin Ahn Import.Event.SigIndex = getEventType(WS); 1277da419bdbSHeejin Ahn Imports.push_back(Import); 1278492f7529SSam Clegg assert(WasmIndices.count(&WS) == 0); 1279da419bdbSHeejin Ahn WasmIndices[&WS] = NumEventImports++; 1280feac819eSAndy Wingo } else if (WS.isTable()) { 1281feac819eSAndy Wingo if (WS.isWeak()) 1282feac819eSAndy Wingo report_fatal_error("undefined table symbol cannot be weak"); 1283feac819eSAndy Wingo 1284feac819eSAndy Wingo wasm::WasmImport Import; 1285feac819eSAndy Wingo Import.Module = WS.getImportModule(); 1286feac819eSAndy Wingo Import.Field = WS.getImportName(); 1287feac819eSAndy Wingo Import.Kind = wasm::WASM_EXTERNAL_TABLE; 1288feac819eSAndy Wingo wasm::ValType ElemType = WS.getTableType(); 1289feac819eSAndy Wingo Import.Table.ElemType = uint8_t(ElemType); 1290feac819eSAndy Wingo // FIXME: Extend table type to include limits? For now we don't specify 1291feac819eSAndy Wingo // a min or max which does not place any restrictions on the size of the 1292feac819eSAndy Wingo // imported table. 1293feac819eSAndy Wingo Import.Table.Limits = {wasm::WASM_LIMITS_FLAG_NONE, 0, 0}; 1294feac819eSAndy Wingo Imports.push_back(Import); 1295feac819eSAndy Wingo assert(WasmIndices.count(&WS) == 0); 1296feac819eSAndy Wingo WasmIndices[&WS] = NumTableImports++; 129732ce5ca0SDan Gohman } 129832ce5ca0SDan Gohman } 129932ce5ca0SDan Gohman } 130032ce5ca0SDan Gohman 1301492f7529SSam Clegg // Add imports for GOT globals 1302492f7529SSam Clegg for (const MCSymbol &S : Asm.symbols()) { 1303492f7529SSam Clegg const auto &WS = static_cast<const MCSymbolWasm &>(S); 1304492f7529SSam Clegg if (WS.isUsedInGOT()) { 1305492f7529SSam Clegg wasm::WasmImport Import; 1306492f7529SSam Clegg if (WS.isFunction()) 1307492f7529SSam Clegg Import.Module = "GOT.func"; 1308492f7529SSam Clegg else 1309492f7529SSam Clegg Import.Module = "GOT.mem"; 1310492f7529SSam Clegg Import.Field = WS.getName(); 1311492f7529SSam Clegg Import.Kind = wasm::WASM_EXTERNAL_GLOBAL; 1312492f7529SSam Clegg Import.Global = {wasm::WASM_TYPE_I32, true}; 1313492f7529SSam Clegg Imports.push_back(Import); 1314492f7529SSam Clegg assert(GOTIndices.count(&WS) == 0); 1315492f7529SSam Clegg GOTIndices[&WS] = NumGlobalImports++; 1316492f7529SSam Clegg } 1317492f7529SSam Clegg } 13180ff28fa6SDerek Schuff } 1319492f7529SSam Clegg 13200ff28fa6SDerek Schuff uint64_t WasmObjectWriter::writeObject(MCAssembler &Asm, 13210ff28fa6SDerek Schuff const MCAsmLayout &Layout) { 13220ff28fa6SDerek Schuff support::endian::Writer MainWriter(*OS, support::little); 13230ff28fa6SDerek Schuff W = &MainWriter; 13240ff28fa6SDerek Schuff if (IsSplitDwarf) { 13250ff28fa6SDerek Schuff uint64_t TotalSize = writeOneObject(Asm, Layout, DwoMode::NonDwoOnly); 13260ff28fa6SDerek Schuff assert(DwoOS); 13270ff28fa6SDerek Schuff support::endian::Writer DwoWriter(*DwoOS, support::little); 13280ff28fa6SDerek Schuff W = &DwoWriter; 13290ff28fa6SDerek Schuff return TotalSize + writeOneObject(Asm, Layout, DwoMode::DwoOnly); 13300ff28fa6SDerek Schuff } else { 13310ff28fa6SDerek Schuff return writeOneObject(Asm, Layout, DwoMode::AllSections); 13320ff28fa6SDerek Schuff } 13330ff28fa6SDerek Schuff } 13340ff28fa6SDerek Schuff 13350ff28fa6SDerek Schuff uint64_t WasmObjectWriter::writeOneObject(MCAssembler &Asm, 13360ff28fa6SDerek Schuff const MCAsmLayout &Layout, 13370ff28fa6SDerek Schuff DwoMode Mode) { 13380ff28fa6SDerek Schuff uint64_t StartOffset = W->OS.tell(); 13390ff28fa6SDerek Schuff SectionCount = 0; 13400ff28fa6SDerek Schuff CustomSections.clear(); 13410ff28fa6SDerek Schuff 13420ff28fa6SDerek Schuff LLVM_DEBUG(dbgs() << "WasmObjectWriter::writeObject\n"); 13430ff28fa6SDerek Schuff 13440ff28fa6SDerek Schuff // Collect information from the available symbols. 13450ff28fa6SDerek Schuff SmallVector<WasmFunction, 4> Functions; 13460ff28fa6SDerek Schuff SmallVector<uint32_t, 4> TableElems; 13470ff28fa6SDerek Schuff SmallVector<wasm::WasmImport, 4> Imports; 13480ff28fa6SDerek Schuff SmallVector<wasm::WasmExport, 4> Exports; 13490ff28fa6SDerek Schuff SmallVector<wasm::WasmEventType, 1> Events; 13500ff28fa6SDerek Schuff SmallVector<wasm::WasmGlobal, 1> Globals; 1351388fb67bSPaulo Matos SmallVector<wasm::WasmTable, 1> Tables; 13520ff28fa6SDerek Schuff SmallVector<wasm::WasmSymbolInfo, 4> SymbolInfos; 13530ff28fa6SDerek Schuff SmallVector<std::pair<uint16_t, uint32_t>, 2> InitFuncs; 13540ff28fa6SDerek Schuff std::map<StringRef, std::vector<WasmComdatEntry>> Comdats; 13550ff28fa6SDerek Schuff uint64_t DataSize = 0; 13560ff28fa6SDerek Schuff if (Mode != DwoMode::DwoOnly) { 13570ff28fa6SDerek Schuff prepareImports(Imports, Asm, Layout); 13580ff28fa6SDerek Schuff } 13594d57fbd0SSam Clegg // Populate DataSegments and CustomSections, which must be done before 13604d57fbd0SSam Clegg // populating DataLocations. 1361759631c7SSam Clegg for (MCSection &Sec : Asm) { 1362759631c7SSam Clegg auto &Section = static_cast<MCSectionWasm &>(Sec); 13637d1ff446SFangrui Song StringRef SectionName = Section.getName(); 1364759631c7SSam Clegg 13650ff28fa6SDerek Schuff if (Mode == DwoMode::NonDwoOnly && isDwoSection(Sec)) 13660ff28fa6SDerek Schuff continue; 13670ff28fa6SDerek Schuff if (Mode == DwoMode::DwoOnly && !isDwoSection(Sec)) 13680ff28fa6SDerek Schuff continue; 13690ff28fa6SDerek Schuff 1370bafe6902SSam Clegg // .init_array sections are handled specially elsewhere. 13714d57fbd0SSam Clegg if (SectionName.startswith(".init_array")) 1372bafe6902SSam Clegg continue; 1373bafe6902SSam Clegg 13744d57fbd0SSam Clegg // Code is handled separately 13754d57fbd0SSam Clegg if (Section.getKind().isText()) 13764d57fbd0SSam Clegg continue; 13774d57fbd0SSam Clegg 13784d57fbd0SSam Clegg if (Section.isWasmData()) { 1379329e76d5SSam Clegg uint32_t SegmentIndex = DataSegments.size(); 1380759631c7SSam Clegg DataSize = alignTo(DataSize, Section.getAlignment()); 1381759631c7SSam Clegg DataSegments.emplace_back(); 1382759631c7SSam Clegg WasmDataSegment &Segment = DataSegments.back(); 13834d57fbd0SSam Clegg Segment.Name = SectionName; 13849b49f36aSSimon Pilgrim Segment.InitFlags = 13859b49f36aSSimon Pilgrim Section.getPassive() ? (uint32_t)wasm::WASM_SEGMENT_IS_PASSIVE : 0; 1386759631c7SSam Clegg Segment.Offset = DataSize; 1387759631c7SSam Clegg Segment.Section = &Section; 138863ebb813SSam Clegg addData(Segment.Data, Section); 138956c587adSSam Clegg Segment.Alignment = Log2_32(Section.getAlignment()); 13902e150409SThomas Lively Segment.LinkerFlags = 0; 1391759631c7SSam Clegg DataSize += Segment.Data.size(); 13926c899ba6SSam Clegg Section.setSegmentIndex(SegmentIndex); 1393ea7caceeSSam Clegg 1394ea7caceeSSam Clegg if (const MCSymbolWasm *C = Section.getGroup()) { 1395ea7caceeSSam Clegg Comdats[C->getName()].emplace_back( 1396329e76d5SSam Clegg WasmComdatEntry{wasm::WASM_COMDAT_DATA, SegmentIndex}); 1397ea7caceeSSam Clegg } 13984d57fbd0SSam Clegg } else { 13994d57fbd0SSam Clegg // Create custom sections 14004d57fbd0SSam Clegg assert(Sec.getKind().isMetadata()); 14014d57fbd0SSam Clegg 14024d57fbd0SSam Clegg StringRef Name = SectionName; 14034d57fbd0SSam Clegg 14044d57fbd0SSam Clegg // For user-defined custom sections, strip the prefix 14054d57fbd0SSam Clegg if (Name.startswith(".custom_section.")) 14064d57fbd0SSam Clegg Name = Name.substr(strlen(".custom_section.")); 14074d57fbd0SSam Clegg 14084d57fbd0SSam Clegg MCSymbol *Begin = Sec.getBeginSymbol(); 1409fb807d4dSSam Clegg if (Begin) { 14104d57fbd0SSam Clegg WasmIndices[cast<MCSymbolWasm>(Begin)] = CustomSections.size(); 1411fb807d4dSSam Clegg } 1412cbda16ebSThomas Lively 1413f6f4f843SThomas Lively // Separate out the producers and target features sections 1414cbda16ebSThomas Lively if (Name == "producers") { 14150eaee545SJonas Devlieghere ProducersSection = std::make_unique<WasmCustomSection>(Name, &Section); 1416cbda16ebSThomas Lively continue; 1417cbda16ebSThomas Lively } 1418f6f4f843SThomas Lively if (Name == "target_features") { 1419f6f4f843SThomas Lively TargetFeaturesSection = 14200eaee545SJonas Devlieghere std::make_unique<WasmCustomSection>(Name, &Section); 1421f6f4f843SThomas Lively continue; 1422f6f4f843SThomas Lively } 1423cbda16ebSThomas Lively 14244d57fbd0SSam Clegg CustomSections.emplace_back(Name, &Section); 1425759631c7SSam Clegg } 14266a31a0d6SSam Clegg } 14276a31a0d6SSam Clegg 1428586320c0SNicholas Wilson // Populate WasmIndices and DataLocations for defined symbols. 1429d934cb88SDan Gohman for (const MCSymbol &S : Asm.symbols()) { 1430d934cb88SDan Gohman // Ignore unnamed temporary symbols, which aren't ever exported, imported, 1431d934cb88SDan Gohman // or used in relocations. 1432d934cb88SDan Gohman if (S.isTemporary() && S.getName().empty()) 1433d934cb88SDan Gohman continue; 1434b7787fd0SSam Clegg 1435d934cb88SDan Gohman const auto &WS = static_cast<const MCSymbolWasm &>(S); 1436d34e60caSNicola Zaghen LLVM_DEBUG( 1437d34e60caSNicola Zaghen dbgs() << "MCSymbol: " << toString(WS.getType()) << " '" << S << "'" 1438d34e60caSNicola Zaghen << " isDefined=" << S.isDefined() << " isExternal=" 1439d34e60caSNicola Zaghen << S.isExternal() << " isTemporary=" << S.isTemporary() 1440d34e60caSNicola Zaghen << " isWeak=" << WS.isWeak() << " isHidden=" << WS.isHidden() 1441b7787fd0SSam Clegg << " isVariable=" << WS.isVariable() << "\n"); 1442b7787fd0SSam Clegg 14435e3d33a7SSam Clegg if (WS.isVariable()) 14445e3d33a7SSam Clegg continue; 14456c899ba6SSam Clegg if (WS.isComdat() && !WS.isDefined()) 14466c899ba6SSam Clegg continue; 1447b7787fd0SSam Clegg 1448d934cb88SDan Gohman if (WS.isFunction()) { 14496c899ba6SSam Clegg unsigned Index; 1450cd65f698SSam Clegg if (WS.isDefined()) { 1451b7787fd0SSam Clegg if (WS.getOffset() != 0) 1452b7787fd0SSam Clegg report_fatal_error( 1453b7787fd0SSam Clegg "function sections must contain one function each"); 1454b7787fd0SSam Clegg 145518c56a07SHeejin Ahn if (WS.getSize() == nullptr) 1456b7787fd0SSam Clegg report_fatal_error( 1457b7787fd0SSam Clegg "function symbols must have a size set with .size"); 1458b7787fd0SSam Clegg 14596c899ba6SSam Clegg // A definition. Write out the function body. 14609f3fe42eSSam Clegg Index = NumFunctionImports + Functions.size(); 1461d934cb88SDan Gohman WasmFunction Func; 1462da419bdbSHeejin Ahn Func.SigIndex = getFunctionType(WS); 1463d934cb88SDan Gohman Func.Sym = &WS; 14646c899ba6SSam Clegg WasmIndices[&WS] = Index; 1465d934cb88SDan Gohman Functions.push_back(Func); 14666c899ba6SSam Clegg 14676c899ba6SSam Clegg auto &Section = static_cast<MCSectionWasm &>(WS.getSection()); 14686c899ba6SSam Clegg if (const MCSymbolWasm *C = Section.getGroup()) { 14696c899ba6SSam Clegg Comdats[C->getName()].emplace_back( 14706c899ba6SSam Clegg WasmComdatEntry{wasm::WASM_COMDAT_FUNCTION, Index}); 14716c899ba6SSam Clegg } 1472881d8778SSam Clegg 1473881d8778SSam Clegg if (WS.hasExportName()) { 1474881d8778SSam Clegg wasm::WasmExport Export; 1475881d8778SSam Clegg Export.Name = WS.getExportName(); 1476881d8778SSam Clegg Export.Kind = wasm::WASM_EXTERNAL_FUNCTION; 1477881d8778SSam Clegg Export.Index = Index; 1478881d8778SSam Clegg Exports.push_back(Export); 1479881d8778SSam Clegg } 1480d934cb88SDan Gohman } else { 1481d934cb88SDan Gohman // An import; the index was assigned above. 14826c899ba6SSam Clegg Index = WasmIndices.find(&WS)->second; 1483d934cb88SDan Gohman } 1484d934cb88SDan Gohman 1485d34e60caSNicola Zaghen LLVM_DEBUG(dbgs() << " -> function index: " << Index << "\n"); 1486da419bdbSHeejin Ahn 14876c899ba6SSam Clegg } else if (WS.isData()) { 1488f3feb6adSWouter van Oortmerssen if (!isInSymtab(WS)) 1489c38e947eSSam Clegg continue; 1490c38e947eSSam Clegg 14916c899ba6SSam Clegg if (!WS.isDefined()) { 1492d34e60caSNicola Zaghen LLVM_DEBUG(dbgs() << " -> segment index: -1" 1493d34e60caSNicola Zaghen << "\n"); 1494fe6414b0SSam Clegg continue; 14956c899ba6SSam Clegg } 1496fe6414b0SSam Clegg 1497d934cb88SDan Gohman if (!WS.getSize()) 1498c38e947eSSam Clegg report_fatal_error("data symbols must have a size set with .size: " + 1499c38e947eSSam Clegg WS.getName()); 1500d934cb88SDan Gohman 1501d934cb88SDan Gohman int64_t Size = 0; 1502d934cb88SDan Gohman if (!WS.getSize()->evaluateAsAbsolute(Size, Layout)) 1503d934cb88SDan Gohman report_fatal_error(".size expression must be evaluatable"); 1504d934cb88SDan Gohman 1505759631c7SSam Clegg auto &DataSection = static_cast<MCSectionWasm &>(WS.getSection()); 1506079cba04SSam Clegg if (!DataSection.isWasmData()) 1507079cba04SSam Clegg report_fatal_error("data symbols must live in a data section: " + 1508079cba04SSam Clegg WS.getName()); 15097c395943SSam Clegg 15106c899ba6SSam Clegg // For each data symbol, export it in the symtab as a reference to the 15116c899ba6SSam Clegg // corresponding Wasm data segment. 15126c899ba6SSam Clegg wasm::WasmDataReference Ref = wasm::WasmDataReference{ 15133b29376eSWouter van Oortmerssen DataSection.getSegmentIndex(), Layout.getSymbolOffset(WS), 15143b29376eSWouter van Oortmerssen static_cast<uint64_t>(Size)}; 15156c899ba6SSam Clegg DataLocations[&WS] = Ref; 1516d34e60caSNicola Zaghen LLVM_DEBUG(dbgs() << " -> segment index: " << Ref.Segment << "\n"); 1517da419bdbSHeejin Ahn 1518a165f2d6SSam Clegg } else if (WS.isGlobal()) { 15196c899ba6SSam Clegg // A "true" Wasm global (currently just __stack_pointer) 15200a6c4d8dSSam Clegg if (WS.isDefined()) { 15210a6c4d8dSSam Clegg assert(WasmIndices.count(&WS) == 0); 15220a6c4d8dSSam Clegg wasm::WasmGlobal Global; 15230a6c4d8dSSam Clegg Global.Type = WS.getGlobalType(); 15240a6c4d8dSSam Clegg Global.Index = NumGlobalImports + Globals.size(); 15250a6c4d8dSSam Clegg switch (Global.Type.Type) { 15260a6c4d8dSSam Clegg case wasm::WASM_TYPE_I32: 15270a6c4d8dSSam Clegg Global.InitExpr.Opcode = wasm::WASM_OPCODE_I32_CONST; 15280a6c4d8dSSam Clegg break; 15290a6c4d8dSSam Clegg case wasm::WASM_TYPE_I64: 15300a6c4d8dSSam Clegg Global.InitExpr.Opcode = wasm::WASM_OPCODE_I64_CONST; 15310a6c4d8dSSam Clegg break; 15320a6c4d8dSSam Clegg case wasm::WASM_TYPE_F32: 15330a6c4d8dSSam Clegg Global.InitExpr.Opcode = wasm::WASM_OPCODE_F32_CONST; 15340a6c4d8dSSam Clegg break; 15350a6c4d8dSSam Clegg case wasm::WASM_TYPE_F64: 15360a6c4d8dSSam Clegg Global.InitExpr.Opcode = wasm::WASM_OPCODE_F64_CONST; 15370a6c4d8dSSam Clegg break; 153879aad89dSSam Clegg case wasm::WASM_TYPE_EXTERNREF: 153979aad89dSSam Clegg Global.InitExpr.Opcode = wasm::WASM_OPCODE_REF_NULL; 154079aad89dSSam Clegg break; 15410a6c4d8dSSam Clegg default: 15420a6c4d8dSSam Clegg llvm_unreachable("unexpected type"); 15430a6c4d8dSSam Clegg } 15440a6c4d8dSSam Clegg WasmIndices[&WS] = Global.Index; 15450a6c4d8dSSam Clegg Globals.push_back(Global); 15460a6c4d8dSSam Clegg } else { 1547545932beSEric Christopher // An import; the index was assigned above 1548d34e60caSNicola Zaghen LLVM_DEBUG(dbgs() << " -> global index: " 1549d34e60caSNicola Zaghen << WasmIndices.find(&WS)->second << "\n"); 15500a6c4d8dSSam Clegg } 1551388fb67bSPaulo Matos } else if (WS.isTable()) { 1552388fb67bSPaulo Matos if (WS.isDefined()) { 1553388fb67bSPaulo Matos assert(WasmIndices.count(&WS) == 0); 1554388fb67bSPaulo Matos wasm::WasmTable Table; 1555388fb67bSPaulo Matos Table.Index = NumTableImports + Tables.size(); 15561933c9d4SAndy Wingo Table.Type.ElemType = static_cast<uint8_t>(WS.getTableType()); 1557388fb67bSPaulo Matos // FIXME: Work on custom limits is ongoing 15581933c9d4SAndy Wingo Table.Type.Limits = {wasm::WASM_LIMITS_FLAG_NONE, 0, 0}; 1559388fb67bSPaulo Matos 1560388fb67bSPaulo Matos WasmIndices[&WS] = Table.Index; 1561388fb67bSPaulo Matos Tables.push_back(Table); 1562388fb67bSPaulo Matos } 1563388fb67bSPaulo Matos LLVM_DEBUG(dbgs() << " -> table index: " << WasmIndices.find(&WS)->second 1564388fb67bSPaulo Matos << "\n"); 1565da419bdbSHeejin Ahn } else if (WS.isEvent()) { 1566da419bdbSHeejin Ahn // C++ exception symbol (__cpp_exception) 1567da419bdbSHeejin Ahn unsigned Index; 1568da419bdbSHeejin Ahn if (WS.isDefined()) { 15690a6c4d8dSSam Clegg assert(WasmIndices.count(&WS) == 0); 1570da419bdbSHeejin Ahn Index = NumEventImports + Events.size(); 1571da419bdbSHeejin Ahn wasm::WasmEventType Event; 1572da419bdbSHeejin Ahn Event.SigIndex = getEventType(WS); 1573da419bdbSHeejin Ahn Event.Attribute = wasm::WASM_EVENT_ATTRIBUTE_EXCEPTION; 1574da419bdbSHeejin Ahn WasmIndices[&WS] = Index; 1575da419bdbSHeejin Ahn Events.push_back(Event); 1576da419bdbSHeejin Ahn } else { 1577da419bdbSHeejin Ahn // An import; the index was assigned above. 1578492f7529SSam Clegg assert(WasmIndices.count(&WS) > 0); 1579da419bdbSHeejin Ahn } 1580da419bdbSHeejin Ahn LLVM_DEBUG(dbgs() << " -> event index: " << WasmIndices.find(&WS)->second 1581da419bdbSHeejin Ahn << "\n"); 1582da419bdbSHeejin Ahn 1583a165f2d6SSam Clegg } else { 1584a165f2d6SSam Clegg assert(WS.isSection()); 1585d934cb88SDan Gohman } 1586d934cb88SDan Gohman } 1587d934cb88SDan Gohman 1588586320c0SNicholas Wilson // Populate WasmIndices and DataLocations for aliased symbols. We need to 1589586320c0SNicholas Wilson // process these in a separate pass because we need to have processed the 1590586320c0SNicholas Wilson // target of the alias before the alias itself and the symbols are not 1591586320c0SNicholas Wilson // necessarily ordered in this way. 1592b7787fd0SSam Clegg for (const MCSymbol &S : Asm.symbols()) { 1593b7787fd0SSam Clegg if (!S.isVariable()) 1594b7787fd0SSam Clegg continue; 159531a2c809SSam Clegg 1596cd65f698SSam Clegg assert(S.isDefined()); 1597b7787fd0SSam Clegg 15989c7b5808SDominic Chen const auto *BS = Layout.getBaseSymbol(S); 15999c7b5808SDominic Chen if (!BS) 16009c7b5808SDominic Chen report_fatal_error(Twine(S.getName()) + 16019c7b5808SDominic Chen ": absolute addressing not supported!"); 16029c7b5808SDominic Chen const MCSymbolWasm *Base = cast<MCSymbolWasm>(BS); 16037ee758d6SSam Clegg 16045e3d33a7SSam Clegg // Find the target symbol of this weak alias and export that index 1605aff1c4dfSSam Clegg const auto &WS = static_cast<const MCSymbolWasm &>(S); 16067ee758d6SSam Clegg LLVM_DEBUG(dbgs() << WS.getName() << ": weak alias of '" << *Base << "'\n"); 1607b7787fd0SSam Clegg 16087ee758d6SSam Clegg if (Base->isFunction()) { 16097ee758d6SSam Clegg assert(WasmIndices.count(Base) > 0); 16107ee758d6SSam Clegg uint32_t WasmIndex = WasmIndices.find(Base)->second; 1611492f7529SSam Clegg assert(WasmIndices.count(&WS) == 0); 16126c899ba6SSam Clegg WasmIndices[&WS] = WasmIndex; 1613d34e60caSNicola Zaghen LLVM_DEBUG(dbgs() << " -> index:" << WasmIndex << "\n"); 16147ee758d6SSam Clegg } else if (Base->isData()) { 16157ee758d6SSam Clegg auto &DataSection = static_cast<MCSectionWasm &>(WS.getSection()); 16167ee758d6SSam Clegg uint64_t Offset = Layout.getSymbolOffset(S); 16177ee758d6SSam Clegg int64_t Size = 0; 16187ee758d6SSam Clegg // For data symbol alias we use the size of the base symbol as the 16197ee758d6SSam Clegg // size of the alias. When an offset from the base is involved this 16207ee758d6SSam Clegg // can result in a offset + size goes past the end of the data section 16217ee758d6SSam Clegg // which out object format doesn't support. So we must clamp it. 16227ee758d6SSam Clegg if (!Base->getSize()->evaluateAsAbsolute(Size, Layout)) 16237ee758d6SSam Clegg report_fatal_error(".size expression must be evaluatable"); 16247ee758d6SSam Clegg const WasmDataSegment &Segment = 16257ee758d6SSam Clegg DataSegments[DataSection.getSegmentIndex()]; 16267ee758d6SSam Clegg Size = 16277ee758d6SSam Clegg std::min(static_cast<uint64_t>(Size), Segment.Data.size() - Offset); 16287ee758d6SSam Clegg wasm::WasmDataReference Ref = wasm::WasmDataReference{ 16297ee758d6SSam Clegg DataSection.getSegmentIndex(), 16307ee758d6SSam Clegg static_cast<uint32_t>(Layout.getSymbolOffset(S)), 16317ee758d6SSam Clegg static_cast<uint32_t>(Size)}; 16326c899ba6SSam Clegg DataLocations[&WS] = Ref; 1633d34e60caSNicola Zaghen LLVM_DEBUG(dbgs() << " -> index:" << Ref.Segment << "\n"); 16346c899ba6SSam Clegg } else { 1635da419bdbSHeejin Ahn report_fatal_error("don't yet support global/event aliases"); 16366c899ba6SSam Clegg } 1637586320c0SNicholas Wilson } 163831a2c809SSam Clegg 1639586320c0SNicholas Wilson // Finally, populate the symbol table itself, in its "natural" order. 1640586320c0SNicholas Wilson for (const MCSymbol &S : Asm.symbols()) { 1641586320c0SNicholas Wilson const auto &WS = static_cast<const MCSymbolWasm &>(S); 164225d8e68bSSam Clegg if (!isInSymtab(WS)) { 164318c56a07SHeejin Ahn WS.setIndex(InvalidIndex); 1644586320c0SNicholas Wilson continue; 164525d8e68bSSam Clegg } 1646feac819eSAndy Wingo if (WS.isTable() && WS.getName() == "__indirect_function_table") { 1647feac819eSAndy Wingo // For the moment, don't emit table symbols -- wasm-ld can't handle them. 1648feac819eSAndy Wingo continue; 1649feac819eSAndy Wingo } 1650d34e60caSNicola Zaghen LLVM_DEBUG(dbgs() << "adding to symtab: " << WS << "\n"); 1651586320c0SNicholas Wilson 1652586320c0SNicholas Wilson uint32_t Flags = 0; 1653586320c0SNicholas Wilson if (WS.isWeak()) 1654586320c0SNicholas Wilson Flags |= wasm::WASM_SYMBOL_BINDING_WEAK; 1655586320c0SNicholas Wilson if (WS.isHidden()) 1656586320c0SNicholas Wilson Flags |= wasm::WASM_SYMBOL_VISIBILITY_HIDDEN; 1657586320c0SNicholas Wilson if (!WS.isExternal() && WS.isDefined()) 1658586320c0SNicholas Wilson Flags |= wasm::WASM_SYMBOL_BINDING_LOCAL; 1659586320c0SNicholas Wilson if (WS.isUndefined()) 1660586320c0SNicholas Wilson Flags |= wasm::WASM_SYMBOL_UNDEFINED; 1661da84b688SDan Gohman if (WS.isNoStrip()) { 1662da84b688SDan Gohman Flags |= wasm::WASM_SYMBOL_NO_STRIP; 1663da84b688SDan Gohman if (isEmscripten()) { 1664d6ef8da3SSam Clegg Flags |= wasm::WASM_SYMBOL_EXPORTED; 1665da84b688SDan Gohman } 1666da84b688SDan Gohman } 1667b4f4e370SSam Clegg if (WS.hasImportName()) 166829874ceaSDan Gohman Flags |= wasm::WASM_SYMBOL_EXPLICIT_NAME; 1669881d8778SSam Clegg if (WS.hasExportName()) 1670881d8778SSam Clegg Flags |= wasm::WASM_SYMBOL_EXPORTED; 1671586320c0SNicholas Wilson 1672586320c0SNicholas Wilson wasm::WasmSymbolInfo Info; 1673586320c0SNicholas Wilson Info.Name = WS.getName(); 1674586320c0SNicholas Wilson Info.Kind = WS.getType(); 1675586320c0SNicholas Wilson Info.Flags = Flags; 16764d57fbd0SSam Clegg if (!WS.isData()) { 16774d57fbd0SSam Clegg assert(WasmIndices.count(&WS) > 0); 1678586320c0SNicholas Wilson Info.ElementIndex = WasmIndices.find(&WS)->second; 16794d57fbd0SSam Clegg } else if (WS.isDefined()) { 16804d57fbd0SSam Clegg assert(DataLocations.count(&WS) > 0); 1681586320c0SNicholas Wilson Info.DataRef = DataLocations.find(&WS)->second; 16824d57fbd0SSam Clegg } 168325d8e68bSSam Clegg WS.setIndex(SymbolInfos.size()); 1684586320c0SNicholas Wilson SymbolInfos.emplace_back(Info); 1685b7787fd0SSam Clegg } 1686b7787fd0SSam Clegg 16876006e091SSam Clegg { 16886006e091SSam Clegg auto HandleReloc = [&](const WasmRelocationEntry &Rel) { 1689f9edbe95SSam Clegg // Functions referenced by a relocation need to put in the table. This is 1690f9edbe95SSam Clegg // purely to make the object file's provisional values readable, and is 1691f9edbe95SSam Clegg // ignored by the linker, which re-calculates the relocations itself. 1692d1152a26SSam Clegg if (Rel.Type != wasm::R_WASM_TABLE_INDEX_I32 && 1693cc1b9b68SWouter van Oortmerssen Rel.Type != wasm::R_WASM_TABLE_INDEX_I64 && 169481443ac1SSam Clegg Rel.Type != wasm::R_WASM_TABLE_INDEX_SLEB && 1695cc1b9b68SWouter van Oortmerssen Rel.Type != wasm::R_WASM_TABLE_INDEX_SLEB64 && 169681443ac1SSam Clegg Rel.Type != wasm::R_WASM_TABLE_INDEX_REL_SLEB) 1697f9edbe95SSam Clegg return; 1698f9edbe95SSam Clegg assert(Rel.Symbol->isFunction()); 16997ee758d6SSam Clegg const MCSymbolWasm *Base = 17007ee758d6SSam Clegg cast<MCSymbolWasm>(Layout.getBaseSymbol(*Rel.Symbol)); 17017ee758d6SSam Clegg uint32_t FunctionIndex = WasmIndices.find(Base)->second; 170218c56a07SHeejin Ahn uint32_t TableIndex = TableElems.size() + InitialTableOffset; 17037ee758d6SSam Clegg if (TableIndices.try_emplace(Base, TableIndex).second) { 17047ee758d6SSam Clegg LLVM_DEBUG(dbgs() << " -> adding " << Base->getName() 1705f9edbe95SSam Clegg << " to table: " << TableIndex << "\n"); 17066c899ba6SSam Clegg TableElems.push_back(FunctionIndex); 17077ee758d6SSam Clegg registerFunctionType(*Base); 17086006e091SSam Clegg } 17096006e091SSam Clegg }; 1710970d02c4SDan Gohman 17116006e091SSam Clegg for (const WasmRelocationEntry &RelEntry : CodeRelocations) 17126006e091SSam Clegg HandleReloc(RelEntry); 17136006e091SSam Clegg for (const WasmRelocationEntry &RelEntry : DataRelocations) 17146006e091SSam Clegg HandleReloc(RelEntry); 1715d934cb88SDan Gohman } 1716d934cb88SDan Gohman 1717bafe6902SSam Clegg // Translate .init_array section contents into start functions. 1718bafe6902SSam Clegg for (const MCSection &S : Asm) { 1719bafe6902SSam Clegg const auto &WS = static_cast<const MCSectionWasm &>(S); 17207d1ff446SFangrui Song if (WS.getName().startswith(".fini_array")) 1721bafe6902SSam Clegg report_fatal_error(".fini_array sections are unsupported"); 17227d1ff446SFangrui Song if (!WS.getName().startswith(".init_array")) 1723bafe6902SSam Clegg continue; 1724bafe6902SSam Clegg if (WS.getFragmentList().empty()) 1725bafe6902SSam Clegg continue; 1726b210c64bSSam Clegg 1727b210c64bSSam Clegg // init_array is expected to contain a single non-empty data fragment 1728b210c64bSSam Clegg if (WS.getFragmentList().size() != 3) 1729bafe6902SSam Clegg report_fatal_error("only one .init_array section fragment supported"); 1730b210c64bSSam Clegg 1731b210c64bSSam Clegg auto IT = WS.begin(); 1732b210c64bSSam Clegg const MCFragment &EmptyFrag = *IT; 1733b210c64bSSam Clegg if (EmptyFrag.getKind() != MCFragment::FT_Data) 1734b210c64bSSam Clegg report_fatal_error(".init_array section should be aligned"); 1735b210c64bSSam Clegg 1736b210c64bSSam Clegg IT = std::next(IT); 1737b210c64bSSam Clegg const MCFragment &AlignFrag = *IT; 1738bafe6902SSam Clegg if (AlignFrag.getKind() != MCFragment::FT_Align) 1739bafe6902SSam Clegg report_fatal_error(".init_array section should be aligned"); 1740bafe6902SSam Clegg if (cast<MCAlignFragment>(AlignFrag).getAlignment() != (is64Bit() ? 8 : 4)) 1741bafe6902SSam Clegg report_fatal_error(".init_array section should be aligned for pointers"); 1742b210c64bSSam Clegg 1743b210c64bSSam Clegg const MCFragment &Frag = *std::next(IT); 1744bafe6902SSam Clegg if (Frag.hasInstructions() || Frag.getKind() != MCFragment::FT_Data) 1745bafe6902SSam Clegg report_fatal_error("only data supported in .init_array section"); 1746b210c64bSSam Clegg 1747bafe6902SSam Clegg uint16_t Priority = UINT16_MAX; 17484d57fbd0SSam Clegg unsigned PrefixLength = strlen(".init_array"); 17497d1ff446SFangrui Song if (WS.getName().size() > PrefixLength) { 17507d1ff446SFangrui Song if (WS.getName()[PrefixLength] != '.') 1751f208f631SHeejin Ahn report_fatal_error( 1752f208f631SHeejin Ahn ".init_array section priority should start with '.'"); 17537d1ff446SFangrui Song if (WS.getName().substr(PrefixLength + 1).getAsInteger(10, Priority)) 1754bafe6902SSam Clegg report_fatal_error("invalid .init_array section priority"); 1755bafe6902SSam Clegg } 1756bafe6902SSam Clegg const auto &DataFrag = cast<MCDataFragment>(Frag); 1757bafe6902SSam Clegg const SmallVectorImpl<char> &Contents = DataFrag.getContents(); 1758f208f631SHeejin Ahn for (const uint8_t * 175918c56a07SHeejin Ahn P = (const uint8_t *)Contents.data(), 176018c56a07SHeejin Ahn *End = (const uint8_t *)Contents.data() + Contents.size(); 176118c56a07SHeejin Ahn P != End; ++P) { 176218c56a07SHeejin Ahn if (*P != 0) 1763bafe6902SSam Clegg report_fatal_error("non-symbolic data in .init_array section"); 1764bafe6902SSam Clegg } 1765bafe6902SSam Clegg for (const MCFixup &Fixup : DataFrag.getFixups()) { 1766f208f631SHeejin Ahn assert(Fixup.getKind() == 1767f208f631SHeejin Ahn MCFixup::getKindForSize(is64Bit() ? 8 : 4, false)); 1768bafe6902SSam Clegg const MCExpr *Expr = Fixup.getValue(); 17698fffa1dfSSam Clegg auto *SymRef = dyn_cast<MCSymbolRefExpr>(Expr); 17708fffa1dfSSam Clegg if (!SymRef) 1771bafe6902SSam Clegg report_fatal_error("fixups in .init_array should be symbol references"); 17728fffa1dfSSam Clegg const auto &TargetSym = cast<const MCSymbolWasm>(SymRef->getSymbol()); 17738fffa1dfSSam Clegg if (TargetSym.getIndex() == InvalidIndex) 17743ea1c62cSSam Clegg report_fatal_error("symbols in .init_array should exist in symtab"); 17758fffa1dfSSam Clegg if (!TargetSym.isFunction()) 17768fffa1dfSSam Clegg report_fatal_error("symbols in .init_array should be for functions"); 177725d8e68bSSam Clegg InitFuncs.push_back( 17788fffa1dfSSam Clegg std::make_pair(Priority, TargetSym.getIndex())); 1779bafe6902SSam Clegg } 1780bafe6902SSam Clegg } 1781bafe6902SSam Clegg 178218eafb6cSDan Gohman // Write out the Wasm header. 178318eafb6cSDan Gohman writeHeader(Asm); 178418eafb6cSDan Gohman 17850ff28fa6SDerek Schuff uint32_t CodeSectionIndex, DataSectionIndex; 17860ff28fa6SDerek Schuff if (Mode != DwoMode::DwoOnly) { 1787da419bdbSHeejin Ahn writeTypeSection(Signatures); 1788f950b24aSSam Clegg writeImportSection(Imports, DataSize, TableElems.size()); 17899e15f359SSam Clegg writeFunctionSection(Functions); 1790*d823cc7cSAndy Wingo writeTableSection(Tables); 1791f950b24aSSam Clegg // Skip the "memory" section; we import the memory instead. 1792da419bdbSHeejin Ahn writeEventSection(Events); 17930a6c4d8dSSam Clegg writeGlobalSection(Globals); 17949e15f359SSam Clegg writeExportSection(Exports); 17959e15f359SSam Clegg writeElemSection(TableElems); 1796fef8de66SThomas Lively writeDataCountSection(); 17970ff28fa6SDerek Schuff 17980ff28fa6SDerek Schuff CodeSectionIndex = writeCodeSection(Asm, Layout, Functions); 17990ff28fa6SDerek Schuff DataSectionIndex = writeDataSection(Layout); 18000ff28fa6SDerek Schuff } 18010ff28fa6SDerek Schuff 18020ff28fa6SDerek Schuff for (auto &CustomSection : CustomSections) { 1803cbda16ebSThomas Lively writeCustomSection(CustomSection, Asm, Layout); 18040ff28fa6SDerek Schuff } 18050ff28fa6SDerek Schuff 18060ff28fa6SDerek Schuff if (Mode != DwoMode::DwoOnly) { 1807c22bfb6bSNicholas Wilson writeLinkingMetaDataSection(SymbolInfos, InitFuncs, Comdats); 18080ff28fa6SDerek Schuff 18096f08c84aSSam Clegg writeRelocSection(CodeSectionIndex, "CODE", CodeRelocations); 18106f08c84aSSam Clegg writeRelocSection(DataSectionIndex, "DATA", DataRelocations); 18110ff28fa6SDerek Schuff } 18126a31a0d6SSam Clegg writeCustomRelocSections(); 1813cbda16ebSThomas Lively if (ProducersSection) 1814cbda16ebSThomas Lively writeCustomSection(*ProducersSection, Asm, Layout); 1815f6f4f843SThomas Lively if (TargetFeaturesSection) 1816f6f4f843SThomas Lively writeCustomSection(*TargetFeaturesSection, Asm, Layout); 1817970d02c4SDan Gohman 1818d934cb88SDan Gohman // TODO: Translate the .comment section to the output. 18190ff28fa6SDerek Schuff return W->OS.tell() - StartOffset; 182018eafb6cSDan Gohman } 182118eafb6cSDan Gohman 182260fbc7ccSLang Hames std::unique_ptr<MCObjectWriter> 18231301a878SLang Hames llvm::createWasmObjectWriter(std::unique_ptr<MCWasmObjectTargetWriter> MOTW, 182418eafb6cSDan Gohman raw_pwrite_stream &OS) { 18250eaee545SJonas Devlieghere return std::make_unique<WasmObjectWriter>(std::move(MOTW), OS); 182618eafb6cSDan Gohman } 18270ff28fa6SDerek Schuff 18280ff28fa6SDerek Schuff std::unique_ptr<MCObjectWriter> 18290ff28fa6SDerek Schuff llvm::createWasmDwoObjectWriter(std::unique_ptr<MCWasmObjectTargetWriter> MOTW, 18300ff28fa6SDerek Schuff raw_pwrite_stream &OS, 18310ff28fa6SDerek Schuff raw_pwrite_stream &DwoOS) { 18320ff28fa6SDerek Schuff return std::make_unique<WasmObjectWriter>(std::move(MOTW), OS, DwoOS); 18330ff28fa6SDerek Schuff } 1834