blob: 169b234abe087c5e0206b400174b5438a0f3270e [file] [log] [blame]
Sam Cleggc94d3932017-11-17 18:14:09 +00001//===- Writer.cpp ---------------------------------------------------------===//
2//
3// The LLVM Linker
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9
10#include "Writer.h"
11
12#include "Config.h"
Sam Cleggc94d3932017-11-17 18:14:09 +000013#include "OutputSections.h"
14#include "OutputSegment.h"
15#include "SymbolTable.h"
16#include "WriterUtils.h"
17#include "lld/Common/ErrorHandler.h"
Rui Ueyama2017d522017-11-28 20:39:17 +000018#include "lld/Common/Memory.h"
Sam Cleggc94d3932017-11-17 18:14:09 +000019#include "lld/Common/Threads.h"
20#include "llvm/Support/FileOutputBuffer.h"
21#include "llvm/Support/Format.h"
22#include "llvm/Support/FormatVariadic.h"
23#include "llvm/Support/LEB128.h"
24
25#include <cstdarg>
26
27#define DEBUG_TYPE "lld"
28
29using namespace llvm;
30using namespace llvm::wasm;
31using namespace lld;
32using namespace lld::wasm;
33
34static constexpr int kStackAlignment = 16;
35
36namespace {
37
Sam Cleggc94d3932017-11-17 18:14:09 +000038// Traits for using WasmSignature in a DenseMap.
39struct WasmSignatureDenseMapInfo {
40 static WasmSignature getEmptyKey() {
41 WasmSignature Sig;
42 Sig.ReturnType = 1;
43 return Sig;
44 }
45 static WasmSignature getTombstoneKey() {
46 WasmSignature Sig;
47 Sig.ReturnType = 2;
48 return Sig;
49 }
50 static unsigned getHashValue(const WasmSignature &Sig) {
51 uintptr_t Value = 0;
52 Value += DenseMapInfo<int32_t>::getHashValue(Sig.ReturnType);
53 for (int32_t Param : Sig.ParamTypes)
54 Value += DenseMapInfo<int32_t>::getHashValue(Param);
55 return Value;
56 }
57 static bool isEqual(const WasmSignature &LHS, const WasmSignature &RHS) {
58 return LHS == RHS;
59 }
60};
61
62// The writer writes a SymbolTable result to a file.
63class Writer {
64public:
65 void run();
66
67private:
68 void openFile();
69
Sam Cleggb8621592017-11-30 01:40:08 +000070 uint32_t getTypeIndex(const WasmSignature &Sig);
Sam Cleggc94d3932017-11-17 18:14:09 +000071 void assignSymbolIndexes();
72 void calculateImports();
73 void calculateOffsets();
74 void calculateTypes();
75 void createOutputSegments();
76 void layoutMemory();
77 void createHeader();
78 void createSections();
79 SyntheticSection *createSyntheticSection(uint32_t Type,
80 std::string Name = "");
81
82 // Builtin sections
83 void createTypeSection();
84 void createFunctionSection();
85 void createTableSection();
86 void createGlobalSection();
87 void createExportSection();
88 void createImportSection();
89 void createMemorySection();
90 void createElemSection();
91 void createStartSection();
92 void createCodeSection();
93 void createDataSection();
94
95 // Custom sections
96 void createRelocSections();
97 void createLinkingSection();
98 void createNameSection();
99
100 void writeHeader();
101 void writeSections();
102
103 uint64_t FileSize = 0;
104 uint32_t DataSize = 0;
105 uint32_t NumFunctions = 0;
106 uint32_t NumGlobals = 0;
107 uint32_t NumMemoryPages = 0;
108 uint32_t NumTableElems = 0;
109 uint32_t NumElements = 0;
110 uint32_t InitialTableOffset = 0;
111
112 std::vector<const WasmSignature *> Types;
113 DenseMap<WasmSignature, int32_t, WasmSignatureDenseMapInfo> TypeIndices;
114 std::vector<Symbol *> FunctionImports;
115 std::vector<Symbol *> GlobalImports;
116
117 // Elements that are used to construct the final output
118 std::string Header;
119 std::vector<OutputSection *> OutputSections;
120
121 std::unique_ptr<FileOutputBuffer> Buffer;
122
123 std::vector<OutputSegment *> Segments;
124 llvm::SmallDenseMap<StringRef, OutputSegment *> SegmentMap;
125};
126
127} // anonymous namespace
128
129static void debugPrint(const char *fmt, ...) {
130 if (!errorHandler().Verbose)
131 return;
132 fprintf(stderr, "lld: ");
133 va_list ap;
134 va_start(ap, fmt);
135 vfprintf(stderr, fmt, ap);
136 va_end(ap);
137}
138
139void Writer::createImportSection() {
140 uint32_t NumImports = FunctionImports.size() + GlobalImports.size();
141 if (Config->ImportMemory)
142 ++NumImports;
143
144 if (NumImports == 0)
145 return;
146
147 SyntheticSection *Section = createSyntheticSection(WASM_SEC_IMPORT);
148 raw_ostream &OS = Section->getStream();
149
150 writeUleb128(OS, NumImports, "import count");
151
152 for (Symbol *Sym : FunctionImports) {
153 WasmImport Import;
154 Import.Module = "env";
155 Import.Field = Sym->getName();
156 Import.Kind = WASM_EXTERNAL_FUNCTION;
Sam Cleggb8621592017-11-30 01:40:08 +0000157 assert(TypeIndices.count(Sym->getFunctionType()) > 0);
158 Import.SigIndex = TypeIndices.lookup(Sym->getFunctionType());
Sam Cleggc94d3932017-11-17 18:14:09 +0000159 writeImport(OS, Import);
160 }
161
162 if (Config->ImportMemory) {
163 WasmImport Import;
164 Import.Module = "env";
165 Import.Field = "memory";
166 Import.Kind = WASM_EXTERNAL_MEMORY;
167 Import.Memory.Flags = 0;
168 Import.Memory.Initial = NumMemoryPages;
169 writeImport(OS, Import);
170 }
171
172 for (Symbol *Sym : GlobalImports) {
173 WasmImport Import;
174 Import.Module = "env";
175 Import.Field = Sym->getName();
176 Import.Kind = WASM_EXTERNAL_GLOBAL;
177 Import.Global.Mutable = false;
Sam Cleggc94d3932017-11-17 18:14:09 +0000178 Import.Global.Type = WASM_TYPE_I32; // Sym->getGlobalType();
179 writeImport(OS, Import);
180 }
181}
182
183void Writer::createTypeSection() {
184 SyntheticSection *Section = createSyntheticSection(WASM_SEC_TYPE);
185 raw_ostream &OS = Section->getStream();
186 writeUleb128(OS, Types.size(), "type count");
187 for (const WasmSignature *Sig : Types) {
188 writeSig(OS, *Sig);
189 }
190}
191
192void Writer::createFunctionSection() {
193 if (!NumFunctions)
194 return;
195
196 SyntheticSection *Section = createSyntheticSection(WASM_SEC_FUNCTION);
197 raw_ostream &OS = Section->getStream();
198
199 writeUleb128(OS, NumFunctions, "function count");
200 for (ObjFile *File : Symtab->ObjectFiles) {
201 for (uint32_t Sig : File->getWasmObj()->functionTypes()) {
202 writeUleb128(OS, File->relocateTypeIndex(Sig), "sig index");
203 }
204 }
205}
206
207void Writer::createMemorySection() {
208 if (Config->ImportMemory)
209 return;
210
211 SyntheticSection *Section = createSyntheticSection(WASM_SEC_MEMORY);
212 raw_ostream &OS = Section->getStream();
213
214 writeUleb128(OS, 1, "memory count");
215 writeUleb128(OS, 0, "memory limits flags");
216 writeUleb128(OS, NumMemoryPages, "initial pages");
217}
218
219void Writer::createGlobalSection() {
220 SyntheticSection *Section = createSyntheticSection(WASM_SEC_GLOBAL);
221 raw_ostream &OS = Section->getStream();
222
223 writeUleb128(OS, NumGlobals, "global count");
224 for (auto &Pair : Config->SyntheticGlobals) {
225 WasmGlobal &Global = Pair.second;
226 writeGlobal(OS, Global);
227 }
228
229 if (Config->Relocatable || Config->EmitRelocs) {
230 for (ObjFile *File : Symtab->ObjectFiles) {
231 uint32_t GlobalIndex = File->NumGlobalImports();
232 for (const WasmGlobal &Global : File->getWasmObj()->globals()) {
233 WasmGlobal RelocatedGlobal(Global);
234 if (Global.Type != WASM_TYPE_I32)
235 fatal("unsupported global type: " + Twine(Global.Type));
236 if (Global.InitExpr.Opcode != WASM_OPCODE_I32_CONST)
237 fatal("unsupported global init opcode: " +
238 Twine(Global.InitExpr.Opcode));
239 RelocatedGlobal.InitExpr.Value.Int32 =
240 File->getRelocatedAddress(GlobalIndex);
241 writeGlobal(OS, RelocatedGlobal);
242 ++GlobalIndex;
243 }
244 }
245 }
246}
247
248void Writer::createTableSection() {
249 SyntheticSection *Section = createSyntheticSection(WASM_SEC_TABLE);
250 raw_ostream &OS = Section->getStream();
251
252 writeUleb128(OS, 1, "table count");
253 writeSleb128(OS, WASM_TYPE_ANYFUNC, "table type");
254 writeUleb128(OS, WASM_LIMITS_FLAG_HAS_MAX, "table flags");
255 writeUleb128(OS, NumTableElems, "table initial size");
256 writeUleb128(OS, NumTableElems, "table max size");
257}
258
259void Writer::createExportSection() {
260 // Memory is and main function are exported for executables.
261 bool ExportMemory = !Config->Relocatable && !Config->ImportMemory;
262 bool ExportMain = !Config->Relocatable;
263 bool ExportOther = true; // Config->Relocatable;
264
265 uint32_t NumExports = 0;
266
267 if (ExportMemory)
268 ++NumExports;
269
270 if (ExportMain && !ExportOther)
271 ++NumExports;
272
273 if (ExportOther) {
274 for (ObjFile *File : Symtab->ObjectFiles) {
275 for (Symbol *Sym : File->getSymbols()) {
276 if (!Sym->isFunction() || Sym->isLocal() || Sym->isUndefined() ||
277 Sym->WrittenToSymtab)
278 continue;
279 Sym->WrittenToSymtab = true;
280 ++NumExports;
281 }
282 }
283 }
284
285 if (!NumExports)
286 return;
287
288 SyntheticSection *Section = createSyntheticSection(WASM_SEC_EXPORT);
289 raw_ostream &OS = Section->getStream();
290
291 writeUleb128(OS, NumExports, "export count");
292
293 if (ExportMemory) {
294 WasmExport MemoryExport;
295 MemoryExport.Name = "memory";
296 MemoryExport.Kind = WASM_EXTERNAL_MEMORY;
297 MemoryExport.Index = 0;
298 writeExport(OS, MemoryExport);
299 }
300
301 if (ExportMain) {
302 Symbol *Sym = Symtab->find(Config->Entry);
303 if (Sym->isDefined()) {
304 if (!Sym->isFunction())
305 fatal("entry point is not a function: " + Sym->getName());
306
307 if (!ExportOther) {
308 WasmExport MainExport;
309 MainExport.Name = Config->Entry;
310 MainExport.Kind = WASM_EXTERNAL_FUNCTION;
311 MainExport.Index = Sym->getOutputIndex();
312 writeExport(OS, MainExport);
313 }
314 }
315 }
316
317 if (ExportOther) {
318 for (ObjFile *File : Symtab->ObjectFiles) {
319 for (Symbol *Sym : File->getSymbols()) {
320 if (!Sym->isFunction() || Sym->isLocal() | Sym->isUndefined() ||
321 !Sym->WrittenToSymtab)
322 continue;
323 Sym->WrittenToSymtab = false;
324 log("Export: " + Sym->getName());
325 WasmExport Export;
326 Export.Name = Sym->getName();
327 Export.Index = Sym->getOutputIndex();
328 if (Sym->isFunction())
329 Export.Kind = WASM_EXTERNAL_FUNCTION;
330 else
331 Export.Kind = WASM_EXTERNAL_GLOBAL;
332 writeExport(OS, Export);
333 }
334 }
335
336 // TODO(sbc): Export local symbols too, Even though they are not part
337 // of the symbol table?
338 }
339}
340
341void Writer::createStartSection() {}
342
343void Writer::createElemSection() {
344 if (!NumElements)
345 return;
346
347 SyntheticSection *Section = createSyntheticSection(WASM_SEC_ELEM);
348 raw_ostream &OS = Section->getStream();
349
350 writeUleb128(OS, 1, "segment count");
351 writeUleb128(OS, 0, "table index");
352 WasmInitExpr InitExpr;
353 InitExpr.Opcode = WASM_OPCODE_I32_CONST;
354 InitExpr.Value.Int32 = InitialTableOffset;
355 writeInitExpr(OS, InitExpr);
356 writeUleb128(OS, NumElements, "elem count");
357
358 for (ObjFile *File : Symtab->ObjectFiles)
359 for (const WasmElemSegment &Segment : File->getWasmObj()->elements())
360 for (uint64_t FunctionIndex : Segment.Functions)
361 writeUleb128(OS, File->relocateFunctionIndex(FunctionIndex),
362 "function index");
363}
364
365void Writer::createCodeSection() {
366 if (!NumFunctions)
367 return;
368
369 log("createCodeSection");
370
371 auto Section = make<CodeSection>(NumFunctions, Symtab->ObjectFiles);
372 OutputSections.push_back(Section);
373}
374
375void Writer::createDataSection() {
376 if (!Segments.size())
377 return;
378
379 log("createDataSection");
380 auto Section = make<DataSection>(Segments);
381 OutputSections.push_back(Section);
382}
383
384// Create reloctions sections in the final output.
385// These are only created when relocatable output is requested.
386void Writer::createRelocSections() {
387 log("createRelocSections");
388 // Don't use iterator here since we are adding to OutputSection
389 size_t OrigSize = OutputSections.size();
390 for (size_t i = 0; i < OrigSize; i++) {
391 OutputSection *S = OutputSections[i];
392 const char *name;
393 uint32_t Count = S->numRelocations();
394 if (!Count)
395 continue;
396
397 if (S->Type == WASM_SEC_DATA)
398 name = "reloc.DATA";
399 else if (S->Type == WASM_SEC_CODE)
400 name = "reloc.CODE";
401 else
402 llvm_unreachable("relocations only support for code and data");
403
404 SyntheticSection *Section = createSyntheticSection(WASM_SEC_CUSTOM, name);
405 raw_ostream &OS = Section->getStream();
406 writeUleb128(OS, S->Type, "reloc section");
407 writeUleb128(OS, Count, "reloc count");
408 S->writeRelocations(OS);
409 }
410}
411
412// Create the custome "linking" section containing linker metadata.
413// This is only created when relocatable output is requested.
414void Writer::createLinkingSection() {
415 SyntheticSection *Section =
416 createSyntheticSection(WASM_SEC_CUSTOM, "linking");
417 raw_ostream &OS = Section->getStream();
418
419 SubSection DataSizeSubSection(WASM_DATA_SIZE);
420 writeUleb128(DataSizeSubSection.getStream(), DataSize, "data size");
421 DataSizeSubSection.finalizeContents();
422 DataSizeSubSection.writeToStream(OS);
423
424 if (Segments.size() && Config->Relocatable) {
425 SubSection SubSection(WASM_SEGMENT_INFO);
426 writeUleb128(SubSection.getStream(), Segments.size(), "num data segments");
427 for (const OutputSegment *S : Segments) {
428 writeStr(SubSection.getStream(), S->Name, "segment name");
429 writeUleb128(SubSection.getStream(), S->Alignment, "alignment");
430 writeUleb128(SubSection.getStream(), 0, "flags");
431 }
432 SubSection.finalizeContents();
433 SubSection.writeToStream(OS);
434 }
435}
436
437// Create the custom "name" section containing debug symbol names.
438void Writer::createNameSection() {
439 // Create an array of all function sorted by function index space
440 std::vector<const Symbol *> Names;
441
442 for (ObjFile *File : Symtab->ObjectFiles) {
443 Names.reserve(Names.size() + File->getSymbols().size());
444 for (Symbol *S : File->getSymbols()) {
445 if (!S->isFunction() || S->isWeak() || S->WrittenToNameSec)
446 continue;
447 S->WrittenToNameSec = true;
448 Names.emplace_back(S);
449 }
450 }
451
452 SyntheticSection *Section = createSyntheticSection(WASM_SEC_CUSTOM, "name");
453
454 std::sort(Names.begin(), Names.end(), [](const Symbol *A, const Symbol *B) {
455 return A->getOutputIndex() < B->getOutputIndex();
456 });
457
458 SubSection FunctionSubsection(WASM_NAMES_FUNCTION);
459 raw_ostream &OS = FunctionSubsection.getStream();
460 writeUleb128(OS, Names.size(), "name count");
461
462 // We have to iterate through the inputs twice so that all the imports
463 // appear first before any of the local function names.
464 for (const Symbol *S : Names) {
465 writeUleb128(OS, S->getOutputIndex(), "func index");
466 writeStr(OS, S->getName(), "symbol name");
467 }
468
469 FunctionSubsection.finalizeContents();
470 FunctionSubsection.writeToStream(Section->getStream());
471}
472
473void Writer::writeHeader() {
474 memcpy(Buffer->getBufferStart(), Header.data(), Header.size());
475}
476
477void Writer::writeSections() {
478 uint8_t *Buf = Buffer->getBufferStart();
479 parallelForEach(OutputSections, [Buf](OutputSection *S) { S->writeTo(Buf); });
480}
481
482// Fix the memory layout of the output binary. This assigns memory offsets
483// to each of the intput data sections as well as the explicit stack region.
484void Writer::layoutMemory() {
485 uint32_t MemoryPtr = 0;
486 if (!Config->Relocatable) {
487 MemoryPtr = Config->GlobalBase;
488 debugPrint("mem: global base = %d\n", Config->GlobalBase);
489 }
490
491 createOutputSegments();
492
493 // Static data comes first
494 for (OutputSegment *Seg : Segments) {
495 MemoryPtr = alignTo(MemoryPtr, Seg->Alignment);
496 Seg->StartVA = MemoryPtr;
497 debugPrint("mem: %-10s offset=%-8d size=%-4d align=%d\n",
498 Seg->Name.str().c_str(), MemoryPtr, Seg->Size, Seg->Alignment);
499 MemoryPtr += Seg->Size;
500 }
501
502 DataSize = MemoryPtr;
503 if (!Config->Relocatable)
504 DataSize -= Config->GlobalBase;
505 debugPrint("mem: static data = %d\n", DataSize);
506
507 // Stack comes after static data
508 if (!Config->Relocatable) {
509 MemoryPtr = alignTo(MemoryPtr, kStackAlignment);
510 if (Config->ZStackSize != alignTo(Config->ZStackSize, kStackAlignment))
511 error("stack size must be " + Twine(kStackAlignment) + "-byte aligned");
512 debugPrint("mem: stack size = %d\n", Config->ZStackSize);
513 debugPrint("mem: stack base = %d\n", MemoryPtr);
514 MemoryPtr += Config->ZStackSize;
515 Config->SyntheticGlobals[0].second.InitExpr.Value.Int32 = MemoryPtr;
516 debugPrint("mem: stack top = %d\n", MemoryPtr);
517 }
518
519 uint32_t MemSize = alignTo(MemoryPtr, WasmPageSize);
520 NumMemoryPages = MemSize / WasmPageSize;
521 debugPrint("mem: total pages = %d\n", NumMemoryPages);
522}
523
524SyntheticSection *Writer::createSyntheticSection(uint32_t Type,
525 std::string Name) {
526 auto Sec = make<SyntheticSection>(Type, Name);
527 log("createSection: " + toString(Sec));
528 OutputSections.push_back(Sec);
529 return Sec;
530}
531
532void Writer::createSections() {
533 // Known sections
534 createTypeSection();
535 createImportSection();
536 createFunctionSection();
537 createTableSection();
538 createMemorySection();
539 createGlobalSection();
540 createExportSection();
541 createStartSection();
542 createElemSection();
543 createCodeSection();
544 createDataSection();
545
546 // Custom sections
547 if (Config->EmitRelocs || Config->Relocatable)
548 createRelocSections();
549 createLinkingSection();
550 if (!Config->StripDebug && !Config->StripAll)
551 createNameSection();
552
553 for (OutputSection *S : OutputSections) {
554 S->setOffset(FileSize);
555 S->finalizeContents();
556 FileSize += S->getSize();
557 }
558}
559
560void Writer::calculateOffsets() {
561 NumGlobals = Config->SyntheticGlobals.size();
562 NumTableElems = InitialTableOffset;
563
564 for (ObjFile *File : Symtab->ObjectFiles) {
565 const WasmObjectFile *WasmFile = File->getWasmObj();
566
567 // Function Index
568 File->FunctionIndexOffset =
569 FunctionImports.size() - File->NumFunctionImports() + NumFunctions;
570 NumFunctions += WasmFile->functions().size();
571
572 // Global Index
573 if (Config->Relocatable || Config->EmitRelocs) {
574 File->GlobalIndexOffset =
575 GlobalImports.size() - File->NumGlobalImports() + NumGlobals;
576 NumGlobals += WasmFile->globals().size();
577 }
578
579 // Memory
580 if (WasmFile->memories().size()) {
581 if (WasmFile->memories().size() > 1) {
582 fatal(File->getName() + ": contains more than one memory");
583 }
584 }
585
586 // Table
587 uint32_t TableCount = WasmFile->tables().size();
588 if (TableCount) {
589 if (TableCount > 1)
590 fatal(File->getName() + ": contains more than one table");
591 File->TableIndexOffset = NumTableElems;
592 NumTableElems += WasmFile->tables()[0].Limits.Initial;
593 }
594
595 // Elem
596 uint32_t SegmentCount = WasmFile->elements().size();
597 if (SegmentCount) {
Rui Ueyamae48f2232017-11-29 20:45:58 +0000598 if (SegmentCount > 1)
Sam Cleggc94d3932017-11-17 18:14:09 +0000599 fatal(File->getName() + ": contains more than element segment");
Rui Ueyamae48f2232017-11-29 20:45:58 +0000600
601 const WasmElemSegment &Segment = WasmFile->elements()[0];
602 if (Segment.TableIndex != 0)
603 fatal(File->getName() + ": unsupported table index");
604 if (Segment.Offset.Value.Int32 != 0)
605 fatal(File->getName() + ": unsupported segment offset");
606 NumElements += Segment.Functions.size();
Sam Cleggc94d3932017-11-17 18:14:09 +0000607 }
608 }
609}
610
611void Writer::calculateImports() {
612 for (ObjFile *File : Symtab->ObjectFiles) {
613 for (Symbol *Sym : File->getSymbols()) {
614 if (Sym->hasOutputIndex() || Sym->isDefined() || Sym->isWeak())
615 continue;
616
617 if (Sym->isFunction()) {
618 Sym->setOutputIndex(FunctionImports.size());
619 FunctionImports.push_back(Sym);
620 } else {
621 Sym->setOutputIndex(GlobalImports.size());
622 GlobalImports.push_back(Sym);
623 }
624 }
625 }
626}
627
Sam Cleggb8621592017-11-30 01:40:08 +0000628uint32_t Writer::getTypeIndex(const WasmSignature &Sig) {
629 auto Pair = TypeIndices.insert(std::make_pair(Sig, Types.size()));
630 if (Pair.second)
631 Types.push_back(&Sig);
632 return Pair.first->second;
633}
634
Sam Cleggc94d3932017-11-17 18:14:09 +0000635void Writer::calculateTypes() {
636 for (ObjFile *File : Symtab->ObjectFiles) {
637 File->TypeMap.reserve(File->getWasmObj()->types().size());
Sam Cleggb8621592017-11-30 01:40:08 +0000638 for (const WasmSignature &Sig : File->getWasmObj()->types())
639 File->TypeMap.push_back(getTypeIndex(Sig));
Sam Cleggc94d3932017-11-17 18:14:09 +0000640 }
641}
642
643void Writer::assignSymbolIndexes() {
644 for (ObjFile *File : Symtab->ObjectFiles) {
645 DEBUG(dbgs() << "assignSymbolIndexes: " << File->getName() << "\n");
646 for (Symbol *Sym : File->getSymbols()) {
647 if (Sym->hasOutputIndex() || !Sym->isDefined())
648 continue;
649
650 if (Sym->getFile() && isa<ObjFile>(Sym->getFile())) {
651 auto *Obj = cast<ObjFile>(Sym->getFile());
652 if (Sym->isFunction())
653 Sym->setOutputIndex(Obj->FunctionIndexOffset +
654 Sym->getFunctionIndex());
655 else
656 Sym->setOutputIndex(Obj->GlobalIndexOffset + Sym->getGlobalIndex());
657 }
658 }
659 }
660}
661
662static StringRef getOutputDataSegmentName(StringRef Name) {
663 if (Config->Relocatable)
664 return Name;
665
666 for (StringRef V :
667 {".text.", ".rodata.", ".data.rel.ro.", ".data.", ".bss.rel.ro.",
668 ".bss.", ".init_array.", ".fini_array.", ".ctors.", ".dtors.", ".tbss.",
669 ".gcc_except_table.", ".tdata.", ".ARM.exidx.", ".ARM.extab."}) {
670 StringRef Prefix = V.drop_back();
671 if (Name.startswith(V) || Name == Prefix)
672 return Prefix;
673 }
674
675 return Name;
676}
677
678void Writer::createOutputSegments() {
679 for (ObjFile *File : Symtab->ObjectFiles) {
680 for (InputSegment *Segment : File->Segments) {
681 StringRef Name = getOutputDataSegmentName(Segment->getName());
682 OutputSegment *&S = SegmentMap[Name];
683 if (S == nullptr) {
684 DEBUG(dbgs() << "new segment: " << Name << "\n");
685 S = make<OutputSegment>(Name);
686 Segments.push_back(S);
687 }
688 S->addInputSegment(Segment);
689 DEBUG(dbgs() << "added data: " << Name << ": " << S->Size << "\n");
690 for (const WasmRelocation &R : File->DataSection->Relocations) {
691 if (R.Offset >= Segment->getInputSectionOffset() &&
692 R.Offset < Segment->getInputSectionOffset() + Segment->getSize()) {
693 Segment->Relocations.push_back(R);
694 }
695 }
696 }
697 }
698}
699
700void Writer::run() {
701 if (!Config->Relocatable)
702 InitialTableOffset = 1;
703
704 log("-- calculateTypes");
705 calculateTypes();
706 log("-- calculateImports");
707 calculateImports();
708 log("-- calculateOffsets");
709 calculateOffsets();
710
711 if (errorHandler().Verbose) {
712 log("NumFunctions : " + Twine(NumFunctions));
713 log("NumGlobals : " + Twine(NumGlobals));
714 log("NumImports : " +
715 Twine(FunctionImports.size() + GlobalImports.size()));
716 log("FunctionImports : " + Twine(FunctionImports.size()));
717 log("GlobalImports : " + Twine(GlobalImports.size()));
718 for (ObjFile *File : Symtab->ObjectFiles)
719 File->dumpInfo();
720 }
721
722 log("-- assignSymbolIndexes");
723 assignSymbolIndexes();
724 log("-- layoutMemory");
725 layoutMemory();
726
727 createHeader();
728 log("-- createSections");
729 createSections();
730
731 log("-- openFile");
732 openFile();
733 if (errorCount())
734 return;
735
736 writeHeader();
737
738 log("-- writeSections");
739 writeSections();
740 if (errorCount())
741 return;
742
743 if (Error E = Buffer->commit())
744 fatal("failed to write the output file: " + toString(std::move(E)));
745}
746
747// Open a result file.
748void Writer::openFile() {
749 log("writing: " + Config->OutputFile);
750 ::remove(Config->OutputFile.str().c_str());
751
752 Expected<std::unique_ptr<FileOutputBuffer>> BufferOrErr =
753 FileOutputBuffer::create(Config->OutputFile, FileSize,
754 FileOutputBuffer::F_executable);
755
756 if (!BufferOrErr)
757 error("failed to open " + Config->OutputFile + ": " +
758 toString(BufferOrErr.takeError()));
759 else
760 Buffer = std::move(*BufferOrErr);
761}
762
763void Writer::createHeader() {
764 raw_string_ostream OS(Header);
765 writeBytes(OS, WasmMagic, sizeof(WasmMagic), "wasm magic");
766 writeU32(OS, WasmVersion, "wasm version");
767 OS.flush();
768 FileSize += Header.size();
769}
770
771void lld::wasm::writeResult() { Writer().run(); }