xref: /llvm-project-15.0.7/lld/ELF/Writer.cpp (revision 7d982509)
1 //===- 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 #include "Config.h"
12 #include "LinkerScript.h"
13 #include "Memory.h"
14 #include "OutputSections.h"
15 #include "Relocations.h"
16 #include "Strings.h"
17 #include "SymbolTable.h"
18 #include "SyntheticSections.h"
19 #include "Target.h"
20 #include "llvm/ADT/StringMap.h"
21 #include "llvm/ADT/StringSwitch.h"
22 #include "llvm/Support/FileOutputBuffer.h"
23 #include "llvm/Support/FileSystem.h"
24 #include "llvm/Support/raw_ostream.h"
25 #include <climits>
26 #include <thread>
27 
28 using namespace llvm;
29 using namespace llvm::ELF;
30 using namespace llvm::object;
31 using namespace llvm::support;
32 using namespace llvm::support::endian;
33 
34 using namespace lld;
35 using namespace lld::elf;
36 
37 namespace {
38 // The writer writes a SymbolTable result to a file.
39 template <class ELFT> class Writer {
40 public:
41   typedef typename ELFT::uint uintX_t;
42   typedef typename ELFT::Shdr Elf_Shdr;
43   typedef typename ELFT::Ehdr Elf_Ehdr;
44   typedef typename ELFT::Phdr Elf_Phdr;
45   typedef typename ELFT::Sym Elf_Sym;
46   typedef typename ELFT::SymRange Elf_Sym_Range;
47   typedef typename ELFT::Rela Elf_Rela;
48   void run();
49 
50 private:
51   void createSyntheticSections();
52   void copyLocalSymbols();
53   void addReservedSymbols();
54   void addInputSec(InputSectionBase<ELFT> *S);
55   void createSections();
56   void forEachRelSec(std::function<void(InputSectionBase<ELFT> &)> Fn);
57   void sortSections();
58   void finalizeSections();
59   void addPredefinedSections();
60 
61   std::vector<PhdrEntry> createPhdrs();
62   void removeEmptyPTLoad();
63   void addPtArmExid(std::vector<PhdrEntry> &Phdrs);
64   void assignAddresses();
65   void assignFileOffsets();
66   void assignFileOffsetsBinary();
67   void setPhdrs();
68   void fixHeaders();
69   void fixSectionAlignments();
70   void fixAbsoluteSymbols();
71   void openFile();
72   void writeHeader();
73   void writeSections();
74   void writeSectionsBinary();
75   void writeBuildId();
76 
77   std::unique_ptr<FileOutputBuffer> Buffer;
78 
79   std::vector<OutputSectionBase *> OutputSections;
80   OutputSectionFactory<ELFT> Factory;
81 
82   void addRelIpltSymbols();
83   void addStartEndSymbols();
84   void addStartStopSymbols(OutputSectionBase *Sec);
85   uintX_t getEntryAddr();
86   OutputSectionBase *findSection(StringRef Name);
87 
88   std::vector<PhdrEntry> Phdrs;
89 
90   uintX_t FileSize;
91   uintX_t SectionHeaderOff;
92   bool AllocateHeader = true;
93 };
94 } // anonymous namespace
95 
96 StringRef elf::getOutputSectionName(StringRef Name) {
97   if (Config->Relocatable)
98     return Name;
99 
100   for (StringRef V :
101        {".text.", ".rodata.", ".data.rel.ro.", ".data.", ".bss.",
102         ".init_array.", ".fini_array.", ".ctors.", ".dtors.", ".tbss.",
103         ".gcc_except_table.", ".tdata.", ".ARM.exidx."}) {
104     StringRef Prefix = V.drop_back();
105     if (Name.startswith(V) || Name == Prefix)
106       return Prefix;
107   }
108 
109   // CommonSection is identified as "COMMON" in linker scripts.
110   // By default, it should go to .bss section.
111   if (Name == "COMMON")
112     return ".bss";
113 
114   // ".zdebug_" is a prefix for ZLIB-compressed sections.
115   // Because we decompressed input sections, we want to remove 'z'.
116   if (Name.startswith(".zdebug_"))
117     return Saver.save(Twine(".") + Name.substr(2));
118   return Name;
119 }
120 
121 template <class ELFT> void elf::reportDiscarded(InputSectionBase<ELFT> *IS) {
122   if (!Config->PrintGcSections)
123     return;
124   errs() << "removing unused section from '" << IS->Name << "' in file '"
125          << IS->getFile()->getName() << "'\n";
126 }
127 
128 template <class ELFT> static bool needsInterpSection() {
129   return !Symtab<ELFT>::X->getSharedFiles().empty() &&
130          !Config->DynamicLinker.empty() &&
131          !Script<ELFT>::X->ignoreInterpSection();
132 }
133 
134 template <class ELFT> void elf::writeResult() { Writer<ELFT>().run(); }
135 
136 template <class ELFT> void Writer<ELFT>::removeEmptyPTLoad() {
137   auto I = std::remove_if(Phdrs.begin(), Phdrs.end(), [&](const PhdrEntry &P) {
138     if (P.p_type != PT_LOAD)
139       return false;
140     if (!P.First)
141       return true;
142     uintX_t Size = P.Last->Addr + P.Last->Size - P.First->Addr;
143     return Size == 0;
144   });
145   Phdrs.erase(I, Phdrs.end());
146 }
147 
148 // The main function of the writer.
149 template <class ELFT> void Writer<ELFT>::run() {
150   // Create linker-synthesized sections such as .got or .plt.
151   // Such sections are of type input section.
152   createSyntheticSections();
153 
154   // We need to create some reserved symbols such as _end. Create them.
155   if (!Config->Relocatable)
156     addReservedSymbols();
157 
158   // Some architectures use small displacements for jump instructions.
159   // It is linker's responsibility to create thunks containing long
160   // jump instructions if jump targets are too far. Create thunks.
161   if (Target->NeedsThunks)
162     forEachRelSec(createThunks<ELFT>);
163 
164   // Create output sections.
165   Script<ELFT>::X->OutputSections = &OutputSections;
166   if (ScriptConfig->HasSections) {
167     // If linker script contains SECTIONS commands, let it create sections.
168     Script<ELFT>::X->processCommands(Factory);
169 
170     // Linker scripts may have left some input sections unassigned.
171     // Assign such sections using the default rule.
172     Script<ELFT>::X->addOrphanSections(Factory);
173   } else {
174     // If linker script does not contain SECTIONS commands, create
175     // output sections by default rules. We still need to give the
176     // linker script a chance to run, because it might contain
177     // non-SECTIONS commands such as ASSERT.
178     createSections();
179     Script<ELFT>::X->processCommands(Factory);
180   }
181 
182   if (Config->Discard != DiscardPolicy::All)
183     copyLocalSymbols();
184 
185   // Now that we have a complete set of output sections. This function
186   // completes section contents. For example, we need to add strings
187   // to the string table, and add entries to .got and .plt.
188   // finalizeSections does that.
189   finalizeSections();
190   if (ErrorCount)
191     return;
192 
193   if (Config->Relocatable) {
194     assignFileOffsets();
195   } else {
196     if (ScriptConfig->HasSections) {
197       Script<ELFT>::X->assignAddresses(Phdrs);
198     } else {
199       fixSectionAlignments();
200       assignAddresses();
201     }
202 
203     // Remove empty PT_LOAD to avoid causing the dynamic linker to try to mmap a
204     // 0 sized region. This has to be done late since only after assignAddresses
205     // we know the size of the sections.
206     removeEmptyPTLoad();
207 
208     if (!Config->OFormatBinary)
209       assignFileOffsets();
210     else
211       assignFileOffsetsBinary();
212 
213     setPhdrs();
214     fixAbsoluteSymbols();
215   }
216 
217   // Write the result down to a file.
218   openFile();
219   if (ErrorCount)
220     return;
221   if (!Config->OFormatBinary) {
222     writeHeader();
223     writeSections();
224   } else {
225     writeSectionsBinary();
226   }
227 
228   // Backfill .note.gnu.build-id section content. This is done at last
229   // because the content is usually a hash value of the entire output file.
230   writeBuildId();
231   if (ErrorCount)
232     return;
233 
234   if (auto EC = Buffer->commit())
235     error(EC, "failed to write to the output file");
236 
237   // Flush the output streams and exit immediately. A full shutdown
238   // is a good test that we are keeping track of all allocated memory,
239   // but actually freeing it is a waste of time in a regular linker run.
240   if (Config->ExitEarly)
241     exitLld(0);
242 }
243 
244 // Initialize Out<ELFT> members.
245 template <class ELFT> void Writer<ELFT>::createSyntheticSections() {
246   // Initialize all pointers with NULL. This is needed because
247   // you can call lld::elf::main more than once as a library.
248   memset(&Out<ELFT>::First, 0, sizeof(Out<ELFT>));
249 
250   // Create singleton output sections.
251   Out<ELFT>::Bss =
252       make<OutputSection<ELFT>>(".bss", SHT_NOBITS, SHF_ALLOC | SHF_WRITE);
253   In<ELFT>::DynStrTab = make<StringTableSection<ELFT>>(".dynstr", true);
254   In<ELFT>::Dynamic = make<DynamicSection<ELFT>>();
255   Out<ELFT>::EhFrame = make<EhOutputSection<ELFT>>();
256   In<ELFT>::RelaDyn = make<RelocationSection<ELFT>>(
257       Config->Rela ? ".rela.dyn" : ".rel.dyn", Config->ZCombreloc);
258   In<ELFT>::ShStrTab = make<StringTableSection<ELFT>>(".shstrtab", false);
259 
260   Out<ELFT>::ElfHeader = make<OutputSectionBase>("", 0, SHF_ALLOC);
261   Out<ELFT>::ElfHeader->Size = sizeof(Elf_Ehdr);
262   Out<ELFT>::ProgramHeaders = make<OutputSectionBase>("", 0, SHF_ALLOC);
263   Out<ELFT>::ProgramHeaders->updateAlignment(sizeof(uintX_t));
264 
265   if (needsInterpSection<ELFT>()) {
266     In<ELFT>::Interp = createInterpSection<ELFT>();
267     Symtab<ELFT>::X->Sections.push_back(In<ELFT>::Interp);
268   } else {
269     In<ELFT>::Interp = nullptr;
270   }
271 
272   if (!Config->Relocatable)
273     Symtab<ELFT>::X->Sections.push_back(createCommentSection<ELFT>());
274 
275   if (Config->Strip != StripPolicy::All) {
276     In<ELFT>::StrTab = make<StringTableSection<ELFT>>(".strtab", false);
277     In<ELFT>::SymTab = make<SymbolTableSection<ELFT>>(*In<ELFT>::StrTab);
278   }
279 
280   if (Config->BuildId != BuildIdKind::None) {
281     In<ELFT>::BuildId = make<BuildIdSection<ELFT>>();
282     Symtab<ELFT>::X->Sections.push_back(In<ELFT>::BuildId);
283   }
284 
285   InputSection<ELFT> *Common = createCommonSection<ELFT>();
286   if (!Common->Data.empty()) {
287     In<ELFT>::Common = Common;
288     Symtab<ELFT>::X->Sections.push_back(Common);
289   }
290 
291   // Add MIPS-specific sections.
292   bool HasDynSymTab = !Symtab<ELFT>::X->getSharedFiles().empty() || Config->Pic;
293   if (Config->EMachine == EM_MIPS) {
294     if (!Config->Shared && HasDynSymTab) {
295       In<ELFT>::MipsRldMap = make<MipsRldMapSection<ELFT>>();
296       Symtab<ELFT>::X->Sections.push_back(In<ELFT>::MipsRldMap);
297     }
298     if (auto *Sec = MipsAbiFlagsSection<ELFT>::create())
299       Symtab<ELFT>::X->Sections.push_back(Sec);
300     if (auto *Sec = MipsOptionsSection<ELFT>::create())
301       Symtab<ELFT>::X->Sections.push_back(Sec);
302     if (auto *Sec = MipsReginfoSection<ELFT>::create())
303       Symtab<ELFT>::X->Sections.push_back(Sec);
304   }
305 
306   if (HasDynSymTab) {
307     In<ELFT>::DynSymTab = make<SymbolTableSection<ELFT>>(*In<ELFT>::DynStrTab);
308     Symtab<ELFT>::X->Sections.push_back(In<ELFT>::DynSymTab);
309 
310     In<ELFT>::VerSym = make<VersionTableSection<ELFT>>();
311     Symtab<ELFT>::X->Sections.push_back(In<ELFT>::VerSym);
312 
313     if (!Config->VersionDefinitions.empty()) {
314       In<ELFT>::VerDef = make<VersionDefinitionSection<ELFT>>();
315       Symtab<ELFT>::X->Sections.push_back(In<ELFT>::VerDef);
316     }
317 
318     In<ELFT>::VerNeed = make<VersionNeedSection<ELFT>>();
319     Symtab<ELFT>::X->Sections.push_back(In<ELFT>::VerNeed);
320 
321     if (Config->GnuHash) {
322       In<ELFT>::GnuHashTab = make<GnuHashTableSection<ELFT>>();
323       Symtab<ELFT>::X->Sections.push_back(In<ELFT>::GnuHashTab);
324     }
325 
326     if (Config->SysvHash) {
327       In<ELFT>::HashTab = make<HashTableSection<ELFT>>();
328       Symtab<ELFT>::X->Sections.push_back(In<ELFT>::HashTab);
329     }
330 
331     Symtab<ELFT>::X->Sections.push_back(In<ELFT>::Dynamic);
332     Symtab<ELFT>::X->Sections.push_back(In<ELFT>::DynStrTab);
333     Symtab<ELFT>::X->Sections.push_back(In<ELFT>::RelaDyn);
334   }
335 
336   // Add .got. MIPS' .got is so different from the other archs,
337   // it has its own class.
338   if (Config->EMachine == EM_MIPS) {
339     In<ELFT>::MipsGot = make<MipsGotSection<ELFT>>();
340     Symtab<ELFT>::X->Sections.push_back(In<ELFT>::MipsGot);
341   } else {
342     In<ELFT>::Got = make<GotSection<ELFT>>();
343     Symtab<ELFT>::X->Sections.push_back(In<ELFT>::Got);
344   }
345 
346   In<ELFT>::GotPlt = make<GotPltSection<ELFT>>();
347   Symtab<ELFT>::X->Sections.push_back(In<ELFT>::GotPlt);
348   In<ELFT>::IgotPlt = make<IgotPltSection<ELFT>>();
349   Symtab<ELFT>::X->Sections.push_back(In<ELFT>::IgotPlt);
350 
351   if (Config->GdbIndex) {
352     In<ELFT>::GdbIndex = make<GdbIndexSection<ELFT>>();
353     Symtab<ELFT>::X->Sections.push_back(In<ELFT>::GdbIndex);
354   }
355 
356   // We always need to add rel[a].plt to output if it has entries.
357   // Even for static linking it can contain R_[*]_IRELATIVE relocations.
358   In<ELFT>::RelaPlt = make<RelocationSection<ELFT>>(
359       Config->Rela ? ".rela.plt" : ".rel.plt", false /*Sort*/);
360   Symtab<ELFT>::X->Sections.push_back(In<ELFT>::RelaPlt);
361 
362   // The RelaIplt immediately follows .rel.plt (.rel.dyn for ARM) to ensure
363   // that the IRelative relocations are processed last by the dynamic loader
364   In<ELFT>::RelaIplt = make<RelocationSection<ELFT>>(
365       (Config->EMachine == EM_ARM) ? ".rel.dyn" : In<ELFT>::RelaPlt->Name,
366       false /*Sort*/);
367   Symtab<ELFT>::X->Sections.push_back(In<ELFT>::RelaIplt);
368 
369   In<ELFT>::Plt = make<PltSection<ELFT>>();
370   Symtab<ELFT>::X->Sections.push_back(In<ELFT>::Plt);
371   In<ELFT>::Iplt = make<IpltSection<ELFT>>();
372   Symtab<ELFT>::X->Sections.push_back(In<ELFT>::Iplt);
373 
374   if (Config->EhFrameHdr) {
375     In<ELFT>::EhFrameHdr = make<EhFrameHeader<ELFT>>();
376     Symtab<ELFT>::X->Sections.push_back(In<ELFT>::EhFrameHdr);
377   }
378 }
379 
380 template <class ELFT>
381 static bool shouldKeepInSymtab(InputSectionBase<ELFT> *Sec, StringRef SymName,
382                                const SymbolBody &B) {
383   if (B.isFile())
384     return false;
385 
386   // We keep sections in symtab for relocatable output.
387   if (B.isSection())
388     return Config->Relocatable;
389 
390   // If sym references a section in a discarded group, don't keep it.
391   if (Sec == &InputSection<ELFT>::Discarded)
392     return false;
393 
394   if (Config->Discard == DiscardPolicy::None)
395     return true;
396 
397   // In ELF assembly .L symbols are normally discarded by the assembler.
398   // If the assembler fails to do so, the linker discards them if
399   // * --discard-locals is used.
400   // * The symbol is in a SHF_MERGE section, which is normally the reason for
401   //   the assembler keeping the .L symbol.
402   if (!SymName.startswith(".L") && !SymName.empty())
403     return true;
404 
405   if (Config->Discard == DiscardPolicy::Locals)
406     return false;
407 
408   return !Sec || !(Sec->Flags & SHF_MERGE);
409 }
410 
411 template <class ELFT> static bool includeInSymtab(const SymbolBody &B) {
412   if (!B.isLocal() && !B.symbol()->IsUsedInRegularObj)
413     return false;
414 
415   // If --retain-symbols-file is given, we'll keep only symbols listed in that
416   // file.
417   if (Config->Discard == DiscardPolicy::RetainFile &&
418       !Config->RetainSymbolsFile.count(B.getName()))
419     return false;
420 
421   if (auto *D = dyn_cast<DefinedRegular<ELFT>>(&B)) {
422     // Always include absolute symbols.
423     if (!D->Section)
424       return true;
425     // Exclude symbols pointing to garbage-collected sections.
426     if (!D->Section->Live)
427       return false;
428     if (auto *S = dyn_cast<MergeInputSection<ELFT>>(D->Section))
429       if (!S->getSectionPiece(D->Value)->Live)
430         return false;
431   }
432   return true;
433 }
434 
435 // Local symbols are not in the linker's symbol table. This function scans
436 // each object file's symbol table to copy local symbols to the output.
437 template <class ELFT> void Writer<ELFT>::copyLocalSymbols() {
438   if (!In<ELFT>::SymTab)
439     return;
440   for (elf::ObjectFile<ELFT> *F : Symtab<ELFT>::X->getObjectFiles()) {
441     for (SymbolBody *B : F->getLocalSymbols()) {
442       if (!B->IsLocal)
443         fatal(toString(F) +
444               ": broken object: getLocalSymbols returns a non-local symbol");
445       auto *DR = dyn_cast<DefinedRegular<ELFT>>(B);
446 
447       // No reason to keep local undefined symbol in symtab.
448       if (!DR)
449         continue;
450       if (!includeInSymtab<ELFT>(*B))
451         continue;
452 
453       InputSectionBase<ELFT> *Sec = DR->Section;
454       if (!shouldKeepInSymtab<ELFT>(Sec, B->getName(), *B))
455         continue;
456       ++In<ELFT>::SymTab->NumLocals;
457       if (Config->Relocatable)
458         B->DynsymIndex = In<ELFT>::SymTab->NumLocals;
459       F->KeptLocalSyms.push_back(std::make_pair(
460           DR, In<ELFT>::SymTab->StrTabSec.addString(B->getName())));
461     }
462   }
463 }
464 
465 // PPC64 has a number of special SHT_PROGBITS+SHF_ALLOC+SHF_WRITE sections that
466 // we would like to make sure appear is a specific order to maximize their
467 // coverage by a single signed 16-bit offset from the TOC base pointer.
468 // Conversely, the special .tocbss section should be first among all SHT_NOBITS
469 // sections. This will put it next to the loaded special PPC64 sections (and,
470 // thus, within reach of the TOC base pointer).
471 static int getPPC64SectionRank(StringRef SectionName) {
472   return StringSwitch<int>(SectionName)
473       .Case(".tocbss", 0)
474       .Case(".branch_lt", 2)
475       .Case(".toc", 3)
476       .Case(".toc1", 4)
477       .Case(".opd", 5)
478       .Default(1);
479 }
480 
481 template <class ELFT> bool elf::isRelroSection(const OutputSectionBase *Sec) {
482   if (!Config->ZRelro)
483     return false;
484   uint64_t Flags = Sec->Flags;
485   if (!(Flags & SHF_ALLOC) || !(Flags & SHF_WRITE))
486     return false;
487   if (Flags & SHF_TLS)
488     return true;
489   uint32_t Type = Sec->Type;
490   if (Type == SHT_INIT_ARRAY || Type == SHT_FINI_ARRAY ||
491       Type == SHT_PREINIT_ARRAY)
492     return true;
493   if (Sec == In<ELFT>::GotPlt->OutSec)
494     return Config->ZNow;
495   if (Sec == In<ELFT>::Dynamic->OutSec)
496     return true;
497   if (In<ELFT>::Got && Sec == In<ELFT>::Got->OutSec)
498     return true;
499   if (In<ELFT>::MipsGot && Sec == In<ELFT>::MipsGot->OutSec)
500     return true;
501   StringRef S = Sec->getName();
502   return S == ".data.rel.ro" || S == ".ctors" || S == ".dtors" || S == ".jcr" ||
503          S == ".eh_frame" || S == ".openbsd.randomdata";
504 }
505 
506 template <class ELFT>
507 static bool compareSectionsNonScript(const OutputSectionBase *A,
508                                      const OutputSectionBase *B) {
509   // Put .interp first because some loaders want to see that section
510   // on the first page of the executable file when loaded into memory.
511   bool AIsInterp = A->getName() == ".interp";
512   bool BIsInterp = B->getName() == ".interp";
513   if (AIsInterp != BIsInterp)
514     return AIsInterp;
515 
516   // Allocatable sections go first to reduce the total PT_LOAD size and
517   // so debug info doesn't change addresses in actual code.
518   bool AIsAlloc = A->Flags & SHF_ALLOC;
519   bool BIsAlloc = B->Flags & SHF_ALLOC;
520   if (AIsAlloc != BIsAlloc)
521     return AIsAlloc;
522 
523   // We don't have any special requirements for the relative order of two non
524   // allocatable sections.
525   if (!AIsAlloc)
526     return false;
527 
528   // We want to put section specified by -T option first, so we
529   // can start assigning VA starting from them later.
530   auto AAddrSetI = Config->SectionStartMap.find(A->getName());
531   auto BAddrSetI = Config->SectionStartMap.find(B->getName());
532   bool AHasAddrSet = AAddrSetI != Config->SectionStartMap.end();
533   bool BHasAddrSet = BAddrSetI != Config->SectionStartMap.end();
534   if (AHasAddrSet != BHasAddrSet)
535     return AHasAddrSet;
536   if (AHasAddrSet)
537     return AAddrSetI->second < BAddrSetI->second;
538 
539   // We want the read only sections first so that they go in the PT_LOAD
540   // covering the program headers at the start of the file.
541   bool AIsWritable = A->Flags & SHF_WRITE;
542   bool BIsWritable = B->Flags & SHF_WRITE;
543   if (AIsWritable != BIsWritable)
544     return BIsWritable;
545 
546   if (!Config->SingleRoRx) {
547     // For a corresponding reason, put non exec sections first (the program
548     // header PT_LOAD is not executable).
549     // We only do that if we are not using linker scripts, since with linker
550     // scripts ro and rx sections are in the same PT_LOAD, so their relative
551     // order is not important. The same applies for -no-rosegment.
552     bool AIsExec = A->Flags & SHF_EXECINSTR;
553     bool BIsExec = B->Flags & SHF_EXECINSTR;
554     if (AIsExec != BIsExec)
555       return BIsExec;
556   }
557 
558   // If we got here we know that both A and B are in the same PT_LOAD.
559 
560   // The TLS initialization block needs to be a single contiguous block in a R/W
561   // PT_LOAD, so stick TLS sections directly before R/W sections. The TLS NOBITS
562   // sections are placed here as they don't take up virtual address space in the
563   // PT_LOAD.
564   bool AIsTls = A->Flags & SHF_TLS;
565   bool BIsTls = B->Flags & SHF_TLS;
566   if (AIsTls != BIsTls)
567     return AIsTls;
568 
569   // The next requirement we have is to put nobits sections last. The
570   // reason is that the only thing the dynamic linker will see about
571   // them is a p_memsz that is larger than p_filesz. Seeing that it
572   // zeros the end of the PT_LOAD, so that has to correspond to the
573   // nobits sections.
574   bool AIsNoBits = A->Type == SHT_NOBITS;
575   bool BIsNoBits = B->Type == SHT_NOBITS;
576   if (AIsNoBits != BIsNoBits)
577     return BIsNoBits;
578 
579   // We place RelRo section before plain r/w ones.
580   bool AIsRelRo = isRelroSection<ELFT>(A);
581   bool BIsRelRo = isRelroSection<ELFT>(B);
582   if (AIsRelRo != BIsRelRo)
583     return AIsRelRo;
584 
585   // Some architectures have additional ordering restrictions for sections
586   // within the same PT_LOAD.
587   if (Config->EMachine == EM_PPC64)
588     return getPPC64SectionRank(A->getName()) <
589            getPPC64SectionRank(B->getName());
590 
591   return false;
592 }
593 
594 // Output section ordering is determined by this function.
595 template <class ELFT>
596 static bool compareSections(const OutputSectionBase *A,
597                             const OutputSectionBase *B) {
598   // For now, put sections mentioned in a linker script first.
599   int AIndex = Script<ELFT>::X->getSectionIndex(A->getName());
600   int BIndex = Script<ELFT>::X->getSectionIndex(B->getName());
601   bool AInScript = AIndex != INT_MAX;
602   bool BInScript = BIndex != INT_MAX;
603   if (AInScript != BInScript)
604     return AInScript;
605   // If both are in the script, use that order.
606   if (AInScript)
607     return AIndex < BIndex;
608 
609   return compareSectionsNonScript<ELFT>(A, B);
610 }
611 
612 // Program header entry
613 PhdrEntry::PhdrEntry(unsigned Type, unsigned Flags) {
614   p_type = Type;
615   p_flags = Flags;
616 }
617 
618 void PhdrEntry::add(OutputSectionBase *Sec) {
619   Last = Sec;
620   if (!First)
621     First = Sec;
622   p_align = std::max(p_align, Sec->Addralign);
623   if (p_type == PT_LOAD)
624     Sec->FirstInPtLoad = First;
625 }
626 
627 template <class ELFT>
628 static Symbol *addOptionalSynthetic(StringRef Name, OutputSectionBase *Sec,
629                                     typename ELFT::uint Val,
630                                     uint8_t StOther = STV_HIDDEN) {
631   SymbolBody *S = Symtab<ELFT>::X->find(Name);
632   if (!S)
633     return nullptr;
634   if (!S->isUndefined() && !S->isShared())
635     return S->symbol();
636   return Symtab<ELFT>::X->addSynthetic(Name, Sec, Val, StOther);
637 }
638 
639 template <class ELFT>
640 static Symbol *addRegular(StringRef Name, InputSectionBase<ELFT> *Sec,
641                           typename ELFT::uint Value) {
642   // The linker generated symbols are added as STB_WEAK to allow user defined
643   // ones to override them.
644   return Symtab<ELFT>::X->addRegular(Name, STV_HIDDEN, STT_NOTYPE, Value,
645                                      /*Size=*/0, STB_WEAK, Sec,
646                                      /*File=*/nullptr);
647 }
648 
649 template <class ELFT>
650 static Symbol *addOptionalRegular(StringRef Name, InputSectionBase<ELFT> *IS,
651                                   typename ELFT::uint Value) {
652   SymbolBody *S = Symtab<ELFT>::X->find(Name);
653   if (!S)
654     return nullptr;
655   if (!S->isUndefined() && !S->isShared())
656     return S->symbol();
657   return addRegular(Name, IS, Value);
658 }
659 
660 // The beginning and the ending of .rel[a].plt section are marked
661 // with __rel[a]_iplt_{start,end} symbols if it is a statically linked
662 // executable. The runtime needs these symbols in order to resolve
663 // all IRELATIVE relocs on startup. For dynamic executables, we don't
664 // need these symbols, since IRELATIVE relocs are resolved through GOT
665 // and PLT. For details, see http://www.airs.com/blog/archives/403.
666 template <class ELFT> void Writer<ELFT>::addRelIpltSymbols() {
667   if (In<ELFT>::DynSymTab)
668     return;
669   StringRef S = Config->Rela ? "__rela_iplt_start" : "__rel_iplt_start";
670   addOptionalRegular<ELFT>(S, In<ELFT>::RelaIplt, 0);
671 
672   S = Config->Rela ? "__rela_iplt_end" : "__rel_iplt_end";
673   addOptionalRegular<ELFT>(S, In<ELFT>::RelaIplt, -1);
674 }
675 
676 // The linker is expected to define some symbols depending on
677 // the linking result. This function defines such symbols.
678 template <class ELFT> void Writer<ELFT>::addReservedSymbols() {
679   if (Config->EMachine == EM_MIPS) {
680     // Define _gp for MIPS. st_value of _gp symbol will be updated by Writer
681     // so that it points to an absolute address which by default is relative
682     // to GOT. Default offset is 0x7ff0.
683     // See "Global Data Symbols" in Chapter 6 in the following document:
684     // ftp://www.linux-mips.org/pub/linux/mips/doc/ABI/mipsabi.pdf
685     ElfSym<ELFT>::MipsGp =
686         Symtab<ELFT>::X->addAbsolute("_gp", STV_HIDDEN, STB_LOCAL);
687 
688     // On MIPS O32 ABI, _gp_disp is a magic symbol designates offset between
689     // start of function and 'gp' pointer into GOT. To simplify relocation
690     // calculation we assign _gp value to it and calculate corresponding
691     // relocations as relative to this value.
692     if (Symtab<ELFT>::X->find("_gp_disp"))
693       ElfSym<ELFT>::MipsGpDisp =
694           Symtab<ELFT>::X->addAbsolute("_gp_disp", STV_HIDDEN, STB_LOCAL);
695 
696     // The __gnu_local_gp is a magic symbol equal to the current value of 'gp'
697     // pointer. This symbol is used in the code generated by .cpload pseudo-op
698     // in case of using -mno-shared option.
699     // https://sourceware.org/ml/binutils/2004-12/msg00094.html
700     if (Symtab<ELFT>::X->find("__gnu_local_gp"))
701       ElfSym<ELFT>::MipsLocalGp =
702           Symtab<ELFT>::X->addAbsolute("__gnu_local_gp", STV_HIDDEN, STB_LOCAL);
703   }
704 
705   // In the assembly for 32 bit x86 the _GLOBAL_OFFSET_TABLE_ symbol
706   // is magical and is used to produce a R_386_GOTPC relocation.
707   // The R_386_GOTPC relocation value doesn't actually depend on the
708   // symbol value, so it could use an index of STN_UNDEF which, according
709   // to the spec, means the symbol value is 0.
710   // Unfortunately both gas and MC keep the _GLOBAL_OFFSET_TABLE_ symbol in
711   // the object file.
712   // The situation is even stranger on x86_64 where the assembly doesn't
713   // need the magical symbol, but gas still puts _GLOBAL_OFFSET_TABLE_ as
714   // an undefined symbol in the .o files.
715   // Given that the symbol is effectively unused, we just create a dummy
716   // hidden one to avoid the undefined symbol error.
717   Symtab<ELFT>::X->addIgnored("_GLOBAL_OFFSET_TABLE_");
718 
719   // __tls_get_addr is defined by the dynamic linker for dynamic ELFs. For
720   // static linking the linker is required to optimize away any references to
721   // __tls_get_addr, so it's not defined anywhere. Create a hidden definition
722   // to avoid the undefined symbol error. As usual special cases are ARM and
723   // MIPS - the libc for these targets defines __tls_get_addr itself because
724   // there are no TLS optimizations for these targets.
725   if (!In<ELFT>::DynSymTab &&
726       (Config->EMachine != EM_MIPS && Config->EMachine != EM_ARM))
727     Symtab<ELFT>::X->addIgnored("__tls_get_addr");
728 
729   // If linker script do layout we do not need to create any standart symbols.
730   if (ScriptConfig->HasSections)
731     return;
732 
733   ElfSym<ELFT>::EhdrStart = Symtab<ELFT>::X->addIgnored("__ehdr_start");
734 
735   auto Define = [this](StringRef S, DefinedRegular<ELFT> *&Sym1,
736                        DefinedRegular<ELFT> *&Sym2) {
737     Sym1 = Symtab<ELFT>::X->addIgnored(S, STV_DEFAULT);
738 
739     // The name without the underscore is not a reserved name,
740     // so it is defined only when there is a reference against it.
741     assert(S.startswith("_"));
742     S = S.substr(1);
743     if (SymbolBody *B = Symtab<ELFT>::X->find(S))
744       if (B->isUndefined())
745         Sym2 = Symtab<ELFT>::X->addAbsolute(S, STV_DEFAULT);
746   };
747 
748   Define("_end", ElfSym<ELFT>::End, ElfSym<ELFT>::End2);
749   Define("_etext", ElfSym<ELFT>::Etext, ElfSym<ELFT>::Etext2);
750   Define("_edata", ElfSym<ELFT>::Edata, ElfSym<ELFT>::Edata2);
751 }
752 
753 // Sort input sections by section name suffixes for
754 // __attribute__((init_priority(N))).
755 template <class ELFT> static void sortInitFini(OutputSectionBase *S) {
756   if (S)
757     reinterpret_cast<OutputSection<ELFT> *>(S)->sortInitFini();
758 }
759 
760 // Sort input sections by the special rule for .ctors and .dtors.
761 template <class ELFT> static void sortCtorsDtors(OutputSectionBase *S) {
762   if (S)
763     reinterpret_cast<OutputSection<ELFT> *>(S)->sortCtorsDtors();
764 }
765 
766 // Sort input sections using the list provided by --symbol-ordering-file.
767 template <class ELFT>
768 static void sortBySymbolsOrder(ArrayRef<OutputSectionBase *> OutputSections) {
769   if (Config->SymbolOrderingFile.empty())
770     return;
771 
772   // Build a map from symbols to their priorities. Symbols that didn't
773   // appear in the symbol ordering file have the lowest priority 0.
774   // All explicitly mentioned symbols have negative (higher) priorities.
775   DenseMap<StringRef, int> SymbolOrder;
776   int Priority = -Config->SymbolOrderingFile.size();
777   for (StringRef S : Config->SymbolOrderingFile)
778     SymbolOrder.insert({S, Priority++});
779 
780   // Build a map from sections to their priorities.
781   DenseMap<InputSectionBase<ELFT> *, int> SectionOrder;
782   for (elf::ObjectFile<ELFT> *File : Symtab<ELFT>::X->getObjectFiles()) {
783     for (SymbolBody *Body : File->getSymbols()) {
784       auto *D = dyn_cast<DefinedRegular<ELFT>>(Body);
785       if (!D || !D->Section)
786         continue;
787       int &Priority = SectionOrder[D->Section];
788       Priority = std::min(Priority, SymbolOrder.lookup(D->getName()));
789     }
790   }
791 
792   // Sort sections by priority.
793   for (OutputSectionBase *Base : OutputSections)
794     if (auto *Sec = dyn_cast<OutputSection<ELFT>>(Base))
795       Sec->sort([&](InputSection<ELFT> *S) { return SectionOrder.lookup(S); });
796 }
797 
798 template <class ELFT>
799 void Writer<ELFT>::forEachRelSec(
800     std::function<void(InputSectionBase<ELFT> &)> Fn) {
801   for (InputSectionBase<ELFT> *IS : Symtab<ELFT>::X->Sections) {
802     if (!IS->Live)
803       continue;
804     // Scan all relocations. Each relocation goes through a series
805     // of tests to determine if it needs special treatment, such as
806     // creating GOT, PLT, copy relocations, etc.
807     // Note that relocations for non-alloc sections are directly
808     // processed by InputSection::relocateNonAlloc.
809     if (!(IS->Flags & SHF_ALLOC))
810       continue;
811     if (isa<InputSection<ELFT>>(IS) || isa<EhInputSection<ELFT>>(IS))
812       Fn(*IS);
813   }
814 }
815 
816 template <class ELFT>
817 void Writer<ELFT>::addInputSec(InputSectionBase<ELFT> *IS) {
818   if (!IS)
819     return;
820 
821   if (!IS->Live) {
822     reportDiscarded(IS);
823     return;
824   }
825   OutputSectionBase *Sec;
826   bool IsNew;
827   StringRef OutsecName = getOutputSectionName(IS->Name);
828   std::tie(Sec, IsNew) = Factory.create(IS, OutsecName);
829   if (IsNew)
830     OutputSections.push_back(Sec);
831   Sec->addSection(IS);
832 }
833 
834 template <class ELFT> void Writer<ELFT>::createSections() {
835   for (InputSectionBase<ELFT> *IS : Symtab<ELFT>::X->Sections)
836     addInputSec(IS);
837 
838   sortBySymbolsOrder<ELFT>(OutputSections);
839   sortInitFini<ELFT>(findSection(".init_array"));
840   sortInitFini<ELFT>(findSection(".fini_array"));
841   sortCtorsDtors<ELFT>(findSection(".ctors"));
842   sortCtorsDtors<ELFT>(findSection(".dtors"));
843 
844   for (OutputSectionBase *Sec : OutputSections)
845     Sec->assignOffsets();
846 }
847 
848 template <class ELFT>
849 static bool canSharePtLoad(const OutputSectionBase &S1,
850                            const OutputSectionBase &S2) {
851   if (!(S1.Flags & SHF_ALLOC) || !(S2.Flags & SHF_ALLOC))
852     return false;
853 
854   bool S1IsWrite = S1.Flags & SHF_WRITE;
855   bool S2IsWrite = S2.Flags & SHF_WRITE;
856   if (S1IsWrite != S2IsWrite)
857     return false;
858 
859   if (!S1IsWrite)
860     return true; // RO and RX share a PT_LOAD with linker scripts.
861   return (S1.Flags & SHF_EXECINSTR) == (S2.Flags & SHF_EXECINSTR);
862 }
863 
864 template <class ELFT> void Writer<ELFT>::sortSections() {
865   // Don't sort if using -r. It is not necessary and we want to preserve the
866   // relative order for SHF_LINK_ORDER sections.
867   if (Config->Relocatable)
868     return;
869   if (!ScriptConfig->HasSections) {
870     std::stable_sort(OutputSections.begin(), OutputSections.end(),
871                      compareSectionsNonScript<ELFT>);
872     return;
873   }
874   Script<ELFT>::X->adjustSectionsBeforeSorting();
875 
876   // The order of the sections in the script is arbitrary and may not agree with
877   // compareSectionsNonScript. This means that we cannot easily define a
878   // strict weak ordering. To see why, consider a comparison of a section in the
879   // script and one not in the script. We have a two simple options:
880   // * Make them equivalent (a is not less than b, and b is not less than a).
881   //   The problem is then that equivalence has to be transitive and we can
882   //   have sections a, b and c with only b in a script and a less than c
883   //   which breaks this property.
884   // * Use compareSectionsNonScript. Given that the script order doesn't have
885   //   to match, we can end up with sections a, b, c, d where b and c are in the
886   //   script and c is compareSectionsNonScript less than b. In which case d
887   //   can be equivalent to c, a to b and d < a. As a concrete example:
888   //   .a (rx) # not in script
889   //   .b (rx) # in script
890   //   .c (ro) # in script
891   //   .d (ro) # not in script
892   //
893   // The way we define an order then is:
894   // *  First put script sections at the start and sort the script and
895   //    non-script sections independently.
896   // *  Move each non-script section to its preferred position. We try
897   //    to put each section in the last position where it it can share
898   //    a PT_LOAD.
899 
900   std::stable_sort(OutputSections.begin(), OutputSections.end(),
901                    compareSections<ELFT>);
902 
903   auto I = OutputSections.begin();
904   auto E = OutputSections.end();
905   auto NonScriptI =
906       std::find_if(OutputSections.begin(), E, [](OutputSectionBase *S) {
907         return Script<ELFT>::X->getSectionIndex(S->getName()) == INT_MAX;
908       });
909   while (NonScriptI != E) {
910     auto BestPos = std::max_element(
911         I, NonScriptI, [&](OutputSectionBase *&A, OutputSectionBase *&B) {
912           bool ACanSharePtLoad = canSharePtLoad<ELFT>(**NonScriptI, *A);
913           bool BCanSharePtLoad = canSharePtLoad<ELFT>(**NonScriptI, *B);
914           if (ACanSharePtLoad != BCanSharePtLoad)
915             return BCanSharePtLoad;
916 
917           bool ACmp = compareSectionsNonScript<ELFT>(*NonScriptI, A);
918           bool BCmp = compareSectionsNonScript<ELFT>(*NonScriptI, B);
919           if (ACmp != BCmp)
920             return BCmp; // FIXME: missing test
921 
922           size_t PosA = &A - &OutputSections[0];
923           size_t PosB = &B - &OutputSections[0];
924           return ACmp ? PosA > PosB : PosA < PosB;
925         });
926 
927     // max_element only returns NonScriptI if the range is empty. If the range
928     // is not empty we should consider moving the the element forward one
929     // position.
930     if (BestPos != NonScriptI &&
931         !compareSectionsNonScript<ELFT>(*NonScriptI, *BestPos))
932       ++BestPos;
933     std::rotate(BestPos, NonScriptI, NonScriptI + 1);
934     ++NonScriptI;
935   }
936 
937   Script<ELFT>::X->adjustSectionsAfterSorting();
938 }
939 
940 template <class ELFT>
941 static void
942 finalizeSynthetic(const std::vector<SyntheticSection<ELFT> *> &Sections) {
943   for (SyntheticSection<ELFT> *SS : Sections)
944     if (SS && SS->OutSec && !SS->empty()) {
945       SS->finalize();
946       SS->OutSec->Size = 0;
947       SS->OutSec->assignOffsets();
948     }
949 }
950 
951 // We need to add input synthetic sections early in createSyntheticSections()
952 // to make them visible from linkescript side. But not all sections are always
953 // required to be in output. For example we don't need dynamic section content
954 // sometimes. This function filters out such unused sections from output.
955 template <class ELFT>
956 static void removeUnusedSyntheticSections(std::vector<OutputSectionBase *> &V) {
957   // Input synthetic sections are placed after all regular ones. We iterate over
958   // them all and exit at first non-synthetic.
959   for (InputSectionBase<ELFT> *S : llvm::reverse(Symtab<ELFT>::X->Sections)) {
960     SyntheticSection<ELFT> *SS = dyn_cast<SyntheticSection<ELFT>>(S);
961     if (!SS)
962       return;
963     if (!SS->empty() || !SS->OutSec)
964       continue;
965 
966     OutputSection<ELFT> *OutSec = cast<OutputSection<ELFT>>(SS->OutSec);
967     OutSec->Sections.erase(
968         std::find(OutSec->Sections.begin(), OutSec->Sections.end(), SS));
969     // If there is no other sections in output section, remove it from output.
970     if (OutSec->Sections.empty())
971       V.erase(std::find(V.begin(), V.end(), OutSec));
972   }
973 }
974 
975 // Create output section objects and add them to OutputSections.
976 template <class ELFT> void Writer<ELFT>::finalizeSections() {
977   Out<ELFT>::DebugInfo = findSection(".debug_info");
978   Out<ELFT>::PreinitArray = findSection(".preinit_array");
979   Out<ELFT>::InitArray = findSection(".init_array");
980   Out<ELFT>::FiniArray = findSection(".fini_array");
981 
982   // The linker needs to define SECNAME_start, SECNAME_end and SECNAME_stop
983   // symbols for sections, so that the runtime can get the start and end
984   // addresses of each section by section name. Add such symbols.
985   if (!Config->Relocatable) {
986     addStartEndSymbols();
987     for (OutputSectionBase *Sec : OutputSections)
988       addStartStopSymbols(Sec);
989   }
990 
991   // Add _DYNAMIC symbol. Unlike GNU gold, our _DYNAMIC symbol has no type.
992   // It should be okay as no one seems to care about the type.
993   // Even the author of gold doesn't remember why gold behaves that way.
994   // https://sourceware.org/ml/binutils/2002-03/msg00360.html
995   if (In<ELFT>::DynSymTab)
996     addRegular("_DYNAMIC", In<ELFT>::Dynamic, 0);
997 
998   // Define __rel[a]_iplt_{start,end} symbols if needed.
999   addRelIpltSymbols();
1000 
1001   if (!Out<ELFT>::EhFrame->empty()) {
1002     OutputSections.push_back(Out<ELFT>::EhFrame);
1003     Out<ELFT>::EhFrame->finalize();
1004   }
1005 
1006   // Scan relocations. This must be done after every symbol is declared so that
1007   // we can correctly decide if a dynamic relocation is needed.
1008   forEachRelSec(scanRelocations<ELFT>);
1009 
1010   // Now that we have defined all possible symbols including linker-
1011   // synthesized ones. Visit all symbols to give the finishing touches.
1012   for (Symbol *S : Symtab<ELFT>::X->getSymbols()) {
1013     SymbolBody *Body = S->body();
1014 
1015     if (!includeInSymtab<ELFT>(*Body))
1016       continue;
1017     if (In<ELFT>::SymTab)
1018       In<ELFT>::SymTab->addSymbol(Body);
1019 
1020     if (In<ELFT>::DynSymTab && S->includeInDynsym()) {
1021       In<ELFT>::DynSymTab->addSymbol(Body);
1022       if (auto *SS = dyn_cast<SharedSymbol<ELFT>>(Body))
1023         if (SS->file()->isNeeded())
1024           In<ELFT>::VerNeed->addSymbol(SS);
1025     }
1026   }
1027 
1028   // Do not proceed if there was an undefined symbol.
1029   if (ErrorCount)
1030     return;
1031 
1032   // So far we have added sections from input object files.
1033   // This function adds linker-created Out<ELFT>::* sections.
1034   addPredefinedSections();
1035   removeUnusedSyntheticSections<ELFT>(OutputSections);
1036 
1037   sortSections();
1038 
1039   unsigned I = 1;
1040   for (OutputSectionBase *Sec : OutputSections) {
1041     Sec->SectionIndex = I++;
1042     Sec->ShName = In<ELFT>::ShStrTab->addString(Sec->getName());
1043   }
1044 
1045   // Binary and relocatable output does not have PHDRS.
1046   // The headers have to be created before finalize as that can influence the
1047   // image base and the dynamic section on mips includes the image base.
1048   if (!Config->Relocatable && !Config->OFormatBinary) {
1049     Phdrs = Script<ELFT>::X->hasPhdrsCommands() ? Script<ELFT>::X->createPhdrs()
1050                                                 : createPhdrs();
1051     addPtArmExid(Phdrs);
1052     fixHeaders();
1053   }
1054 
1055   // Fill other section headers. The dynamic table is finalized
1056   // at the end because some tags like RELSZ depend on result
1057   // of finalizing other sections.
1058   for (OutputSectionBase *Sec : OutputSections)
1059     Sec->finalize();
1060 
1061   // Dynamic section must be the last one in this list and dynamic
1062   // symbol table section (DynSymTab) must be the first one.
1063   finalizeSynthetic<ELFT>(
1064       {In<ELFT>::DynSymTab, In<ELFT>::GnuHashTab, In<ELFT>::HashTab,
1065        In<ELFT>::SymTab,    In<ELFT>::ShStrTab,   In<ELFT>::StrTab,
1066        In<ELFT>::VerDef,    In<ELFT>::DynStrTab,  In<ELFT>::GdbIndex,
1067        In<ELFT>::Got,       In<ELFT>::MipsGot,    In<ELFT>::IgotPlt,
1068        In<ELFT>::GotPlt,    In<ELFT>::RelaDyn,    In<ELFT>::RelaIplt,
1069        In<ELFT>::RelaPlt,   In<ELFT>::Plt,        In<ELFT>::Iplt,
1070        In<ELFT>::Plt,       In<ELFT>::EhFrameHdr, In<ELFT>::VerSym,
1071        In<ELFT>::VerNeed,   In<ELFT>::Dynamic});
1072 }
1073 
1074 template <class ELFT> void Writer<ELFT>::addPredefinedSections() {
1075   if (Out<ELFT>::Bss->Size > 0)
1076     OutputSections.push_back(Out<ELFT>::Bss);
1077 
1078   auto OS = dyn_cast_or_null<OutputSection<ELFT>>(findSection(".ARM.exidx"));
1079   if (OS && !OS->Sections.empty() && !Config->Relocatable)
1080     OS->addSection(make<ARMExidxSentinelSection<ELFT>>());
1081 
1082   addInputSec(In<ELFT>::SymTab);
1083   addInputSec(In<ELFT>::ShStrTab);
1084   addInputSec(In<ELFT>::StrTab);
1085 }
1086 
1087 // The linker is expected to define SECNAME_start and SECNAME_end
1088 // symbols for a few sections. This function defines them.
1089 template <class ELFT> void Writer<ELFT>::addStartEndSymbols() {
1090   auto Define = [&](StringRef Start, StringRef End, OutputSectionBase *OS) {
1091     // These symbols resolve to the image base if the section does not exist.
1092     // A special value -1 indicates end of the section.
1093     addOptionalSynthetic<ELFT>(Start, OS, 0);
1094     addOptionalSynthetic<ELFT>(End, OS, OS ? -1 : 0);
1095   };
1096 
1097   Define("__preinit_array_start", "__preinit_array_end",
1098          Out<ELFT>::PreinitArray);
1099   Define("__init_array_start", "__init_array_end", Out<ELFT>::InitArray);
1100   Define("__fini_array_start", "__fini_array_end", Out<ELFT>::FiniArray);
1101 
1102   if (OutputSectionBase *Sec = findSection(".ARM.exidx"))
1103     Define("__exidx_start", "__exidx_end", Sec);
1104 }
1105 
1106 // If a section name is valid as a C identifier (which is rare because of
1107 // the leading '.'), linkers are expected to define __start_<secname> and
1108 // __stop_<secname> symbols. They are at beginning and end of the section,
1109 // respectively. This is not requested by the ELF standard, but GNU ld and
1110 // gold provide the feature, and used by many programs.
1111 template <class ELFT>
1112 void Writer<ELFT>::addStartStopSymbols(OutputSectionBase *Sec) {
1113   StringRef S = Sec->getName();
1114   if (!isValidCIdentifier(S))
1115     return;
1116   addOptionalSynthetic<ELFT>(Saver.save("__start_" + S), Sec, 0, STV_DEFAULT);
1117   addOptionalSynthetic<ELFT>(Saver.save("__stop_" + S), Sec, -1, STV_DEFAULT);
1118 }
1119 
1120 template <class ELFT>
1121 OutputSectionBase *Writer<ELFT>::findSection(StringRef Name) {
1122   for (OutputSectionBase *Sec : OutputSections)
1123     if (Sec->getName() == Name)
1124       return Sec;
1125   return nullptr;
1126 }
1127 
1128 template <class ELFT> static bool needsPtLoad(OutputSectionBase *Sec) {
1129   if (!(Sec->Flags & SHF_ALLOC))
1130     return false;
1131 
1132   // Don't allocate VA space for TLS NOBITS sections. The PT_TLS PHDR is
1133   // responsible for allocating space for them, not the PT_LOAD that
1134   // contains the TLS initialization image.
1135   if (Sec->Flags & SHF_TLS && Sec->Type == SHT_NOBITS)
1136     return false;
1137   return true;
1138 }
1139 
1140 // Linker scripts are responsible for aligning addresses. Unfortunately, most
1141 // linker scripts are designed for creating two PT_LOADs only, one RX and one
1142 // RW. This means that there is no alignment in the RO to RX transition and we
1143 // cannot create a PT_LOAD there.
1144 template <class ELFT>
1145 static typename ELFT::uint computeFlags(typename ELFT::uint F) {
1146   if (Config->OMagic)
1147     return PF_R | PF_W | PF_X;
1148   if (Config->SingleRoRx && !(F & PF_W))
1149     return F | PF_X;
1150   return F;
1151 }
1152 
1153 // Decide which program headers to create and which sections to include in each
1154 // one.
1155 template <class ELFT> std::vector<PhdrEntry> Writer<ELFT>::createPhdrs() {
1156   std::vector<PhdrEntry> Ret;
1157   auto AddHdr = [&](unsigned Type, unsigned Flags) -> PhdrEntry * {
1158     Ret.emplace_back(Type, Flags);
1159     return &Ret.back();
1160   };
1161 
1162   // The first phdr entry is PT_PHDR which describes the program header itself.
1163   PhdrEntry &Hdr = *AddHdr(PT_PHDR, PF_R);
1164   Hdr.add(Out<ELFT>::ProgramHeaders);
1165 
1166   // PT_INTERP must be the second entry if exists.
1167   if (OutputSectionBase *Sec = findSection(".interp")) {
1168     PhdrEntry &Hdr = *AddHdr(PT_INTERP, Sec->getPhdrFlags());
1169     Hdr.add(Sec);
1170   }
1171 
1172   // Add the first PT_LOAD segment for regular output sections.
1173   uintX_t Flags = computeFlags<ELFT>(PF_R);
1174   PhdrEntry *Load = AddHdr(PT_LOAD, Flags);
1175 
1176   PhdrEntry TlsHdr(PT_TLS, PF_R);
1177   PhdrEntry RelRo(PT_GNU_RELRO, PF_R);
1178   PhdrEntry Note(PT_NOTE, PF_R);
1179   for (OutputSectionBase *Sec : OutputSections) {
1180     if (!(Sec->Flags & SHF_ALLOC))
1181       break;
1182 
1183     // If we meet TLS section then we create TLS header
1184     // and put all TLS sections inside for further use when
1185     // assign addresses.
1186     if (Sec->Flags & SHF_TLS)
1187       TlsHdr.add(Sec);
1188 
1189     if (!needsPtLoad<ELFT>(Sec))
1190       continue;
1191 
1192     // Segments are contiguous memory regions that has the same attributes
1193     // (e.g. executable or writable). There is one phdr for each segment.
1194     // Therefore, we need to create a new phdr when the next section has
1195     // different flags or is loaded at a discontiguous address using AT linker
1196     // script command.
1197     uintX_t NewFlags = computeFlags<ELFT>(Sec->getPhdrFlags());
1198     if (Script<ELFT>::X->hasLMA(Sec->getName()) || Flags != NewFlags) {
1199       Load = AddHdr(PT_LOAD, NewFlags);
1200       Flags = NewFlags;
1201     }
1202 
1203     Load->add(Sec);
1204 
1205     if (isRelroSection<ELFT>(Sec))
1206       RelRo.add(Sec);
1207     if (Sec->Type == SHT_NOTE)
1208       Note.add(Sec);
1209   }
1210 
1211   // Add the TLS segment unless it's empty.
1212   if (TlsHdr.First)
1213     Ret.push_back(std::move(TlsHdr));
1214 
1215   // Add an entry for .dynamic.
1216   if (In<ELFT>::DynSymTab) {
1217     PhdrEntry &H =
1218         *AddHdr(PT_DYNAMIC, In<ELFT>::Dynamic->OutSec->getPhdrFlags());
1219     H.add(In<ELFT>::Dynamic->OutSec);
1220   }
1221 
1222   // PT_GNU_RELRO includes all sections that should be marked as
1223   // read-only by dynamic linker after proccessing relocations.
1224   if (RelRo.First)
1225     Ret.push_back(std::move(RelRo));
1226 
1227   // PT_GNU_EH_FRAME is a special section pointing on .eh_frame_hdr.
1228   if (!Out<ELFT>::EhFrame->empty() && In<ELFT>::EhFrameHdr) {
1229     PhdrEntry &Hdr =
1230         *AddHdr(PT_GNU_EH_FRAME, In<ELFT>::EhFrameHdr->OutSec->getPhdrFlags());
1231     Hdr.add(In<ELFT>::EhFrameHdr->OutSec);
1232   }
1233 
1234   // PT_OPENBSD_RANDOMIZE specifies the location and size of a part of the
1235   // memory image of the program that must be filled with random data before any
1236   // code in the object is executed.
1237   if (OutputSectionBase *Sec = findSection(".openbsd.randomdata")) {
1238     PhdrEntry &Hdr = *AddHdr(PT_OPENBSD_RANDOMIZE, Sec->getPhdrFlags());
1239     Hdr.add(Sec);
1240   }
1241 
1242   // PT_GNU_STACK is a special section to tell the loader to make the
1243   // pages for the stack non-executable.
1244   if (!Config->ZExecstack) {
1245     PhdrEntry &Hdr = *AddHdr(PT_GNU_STACK, PF_R | PF_W);
1246     if (Config->ZStackSize != uint64_t(-1))
1247       Hdr.p_memsz = Config->ZStackSize;
1248   }
1249 
1250   // PT_OPENBSD_WXNEEDED is a OpenBSD-specific header to mark the executable
1251   // is expected to perform W^X violations, such as calling mprotect(2) or
1252   // mmap(2) with PROT_WRITE | PROT_EXEC, which is prohibited by default on
1253   // OpenBSD.
1254   if (Config->ZWxneeded)
1255     AddHdr(PT_OPENBSD_WXNEEDED, PF_X);
1256 
1257   if (Note.First)
1258     Ret.push_back(std::move(Note));
1259   return Ret;
1260 }
1261 
1262 template <class ELFT>
1263 void Writer<ELFT>::addPtArmExid(std::vector<PhdrEntry> &Phdrs) {
1264   if (Config->EMachine != EM_ARM)
1265     return;
1266   auto I = std::find_if(
1267       OutputSections.begin(), OutputSections.end(),
1268       [](OutputSectionBase *Sec) { return Sec->Type == SHT_ARM_EXIDX; });
1269   if (I == OutputSections.end())
1270     return;
1271 
1272   // PT_ARM_EXIDX is the ARM EHABI equivalent of PT_GNU_EH_FRAME
1273   PhdrEntry ARMExidx(PT_ARM_EXIDX, PF_R);
1274   ARMExidx.add(*I);
1275   Phdrs.push_back(ARMExidx);
1276 }
1277 
1278 // The first section of each PT_LOAD and the first section after PT_GNU_RELRO
1279 // have to be page aligned so that the dynamic linker can set the permissions.
1280 template <class ELFT> void Writer<ELFT>::fixSectionAlignments() {
1281   for (const PhdrEntry &P : Phdrs)
1282     if (P.p_type == PT_LOAD && P.First)
1283       P.First->PageAlign = true;
1284 
1285   for (const PhdrEntry &P : Phdrs) {
1286     if (P.p_type != PT_GNU_RELRO)
1287       continue;
1288     // Find the first section after PT_GNU_RELRO. If it is in a PT_LOAD we
1289     // have to align it to a page.
1290     auto End = OutputSections.end();
1291     auto I = std::find(OutputSections.begin(), End, P.Last);
1292     if (I == End || (I + 1) == End)
1293       continue;
1294     OutputSectionBase *Sec = *(I + 1);
1295     if (needsPtLoad<ELFT>(Sec))
1296       Sec->PageAlign = true;
1297   }
1298 }
1299 
1300 template <class ELFT>
1301 void elf::allocateHeaders(MutableArrayRef<PhdrEntry> Phdrs,
1302                           ArrayRef<OutputSectionBase *> OutputSections) {
1303   auto FirstPTLoad =
1304       std::find_if(Phdrs.begin(), Phdrs.end(),
1305                    [](const PhdrEntry &E) { return E.p_type == PT_LOAD; });
1306   if (FirstPTLoad == Phdrs.end())
1307     return;
1308   if (FirstPTLoad->First)
1309     for (OutputSectionBase *Sec : OutputSections)
1310       if (Sec->FirstInPtLoad == FirstPTLoad->First)
1311         Sec->FirstInPtLoad = Out<ELFT>::ElfHeader;
1312   FirstPTLoad->First = Out<ELFT>::ElfHeader;
1313   if (!FirstPTLoad->Last)
1314     FirstPTLoad->Last = Out<ELFT>::ProgramHeaders;
1315 }
1316 
1317 // We should set file offsets and VAs for elf header and program headers
1318 // sections. These are special, we do not include them into output sections
1319 // list, but have them to simplify the code.
1320 template <class ELFT> void Writer<ELFT>::fixHeaders() {
1321   Out<ELFT>::ProgramHeaders->Size = sizeof(Elf_Phdr) * Phdrs.size();
1322   // If the script has SECTIONS, assignAddresses will compute the values.
1323   if (ScriptConfig->HasSections)
1324     return;
1325 
1326   uintX_t HeaderSize = getHeaderSize<ELFT>();
1327   // When -T<section> option is specified, lower the base to make room for those
1328   // sections.
1329   if (!Config->SectionStartMap.empty()) {
1330     uint64_t Min = -1;
1331     for (const auto &P : Config->SectionStartMap)
1332       Min = std::min(Min, P.second);
1333     if (HeaderSize < Min)
1334       Min -= HeaderSize;
1335     else
1336       AllocateHeader = false;
1337     if (Min < Config->ImageBase)
1338       Config->ImageBase = alignDown(Min, Config->MaxPageSize);
1339   }
1340 
1341   if (AllocateHeader)
1342     allocateHeaders<ELFT>(Phdrs, OutputSections);
1343 
1344   uintX_t BaseVA = Config->ImageBase;
1345   Out<ELFT>::ElfHeader->Addr = BaseVA;
1346   Out<ELFT>::ProgramHeaders->Addr = BaseVA + Out<ELFT>::ElfHeader->Size;
1347 }
1348 
1349 // Assign VAs (addresses at run-time) to output sections.
1350 template <class ELFT> void Writer<ELFT>::assignAddresses() {
1351   uintX_t VA = Config->ImageBase;
1352   if (AllocateHeader)
1353     VA += getHeaderSize<ELFT>();
1354   uintX_t ThreadBssOffset = 0;
1355   for (OutputSectionBase *Sec : OutputSections) {
1356     uintX_t Alignment = Sec->Addralign;
1357     if (Sec->PageAlign)
1358       Alignment = std::max<uintX_t>(Alignment, Config->MaxPageSize);
1359 
1360     auto I = Config->SectionStartMap.find(Sec->getName());
1361     if (I != Config->SectionStartMap.end())
1362       VA = I->second;
1363 
1364     // We only assign VAs to allocated sections.
1365     if (needsPtLoad<ELFT>(Sec)) {
1366       VA = alignTo(VA, Alignment);
1367       Sec->Addr = VA;
1368       VA += Sec->Size;
1369     } else if (Sec->Flags & SHF_TLS && Sec->Type == SHT_NOBITS) {
1370       uintX_t TVA = VA + ThreadBssOffset;
1371       TVA = alignTo(TVA, Alignment);
1372       Sec->Addr = TVA;
1373       ThreadBssOffset = TVA - VA + Sec->Size;
1374     }
1375   }
1376 }
1377 
1378 // Adjusts the file alignment for a given output section and returns
1379 // its new file offset. The file offset must be the same with its
1380 // virtual address (modulo the page size) so that the loader can load
1381 // executables without any address adjustment.
1382 template <class ELFT, class uintX_t>
1383 static uintX_t getFileAlignment(uintX_t Off, OutputSectionBase *Sec) {
1384   OutputSectionBase *First = Sec->FirstInPtLoad;
1385   // If the section is not in a PT_LOAD, we just have to align it.
1386   if (!First)
1387     return alignTo(Off, Sec->Addralign);
1388 
1389   // The first section in a PT_LOAD has to have congruent offset and address
1390   // module the page size.
1391   if (Sec == First)
1392     return alignTo(Off, Config->MaxPageSize, Sec->Addr);
1393 
1394   // If two sections share the same PT_LOAD the file offset is calculated
1395   // using this formula: Off2 = Off1 + (VA2 - VA1).
1396   return First->Offset + Sec->Addr - First->Addr;
1397 }
1398 
1399 template <class ELFT, class uintX_t>
1400 void setOffset(OutputSectionBase *Sec, uintX_t &Off) {
1401   if (Sec->Type == SHT_NOBITS) {
1402     Sec->Offset = Off;
1403     return;
1404   }
1405 
1406   Off = getFileAlignment<ELFT>(Off, Sec);
1407   Sec->Offset = Off;
1408   Off += Sec->Size;
1409 }
1410 
1411 template <class ELFT> void Writer<ELFT>::assignFileOffsetsBinary() {
1412   uintX_t Off = 0;
1413   for (OutputSectionBase *Sec : OutputSections)
1414     if (Sec->Flags & SHF_ALLOC)
1415       setOffset<ELFT>(Sec, Off);
1416   FileSize = alignTo(Off, sizeof(uintX_t));
1417 }
1418 
1419 // Assign file offsets to output sections.
1420 template <class ELFT> void Writer<ELFT>::assignFileOffsets() {
1421   uintX_t Off = 0;
1422   setOffset<ELFT>(Out<ELFT>::ElfHeader, Off);
1423   setOffset<ELFT>(Out<ELFT>::ProgramHeaders, Off);
1424 
1425   for (OutputSectionBase *Sec : OutputSections)
1426     setOffset<ELFT>(Sec, Off);
1427 
1428   SectionHeaderOff = alignTo(Off, sizeof(uintX_t));
1429   FileSize = SectionHeaderOff + (OutputSections.size() + 1) * sizeof(Elf_Shdr);
1430 }
1431 
1432 // Finalize the program headers. We call this function after we assign
1433 // file offsets and VAs to all sections.
1434 template <class ELFT> void Writer<ELFT>::setPhdrs() {
1435   for (PhdrEntry &P : Phdrs) {
1436     OutputSectionBase *First = P.First;
1437     OutputSectionBase *Last = P.Last;
1438     if (First) {
1439       P.p_filesz = Last->Offset - First->Offset;
1440       if (Last->Type != SHT_NOBITS)
1441         P.p_filesz += Last->Size;
1442       P.p_memsz = Last->Addr + Last->Size - First->Addr;
1443       P.p_offset = First->Offset;
1444       P.p_vaddr = First->Addr;
1445       if (!P.HasLMA)
1446         P.p_paddr = First->getLMA();
1447     }
1448     if (P.p_type == PT_LOAD)
1449       P.p_align = Config->MaxPageSize;
1450     else if (P.p_type == PT_GNU_RELRO)
1451       P.p_align = 1;
1452 
1453     // The TLS pointer goes after PT_TLS. At least glibc will align it,
1454     // so round up the size to make sure the offsets are correct.
1455     if (P.p_type == PT_TLS) {
1456       Out<ELFT>::TlsPhdr = &P;
1457       if (P.p_memsz)
1458         P.p_memsz = alignTo(P.p_memsz, P.p_align);
1459     }
1460   }
1461 }
1462 
1463 // The entry point address is chosen in the following ways.
1464 //
1465 // 1. the '-e' entry command-line option;
1466 // 2. the ENTRY(symbol) command in a linker control script;
1467 // 3. the value of the symbol start, if present;
1468 // 4. the address of the first byte of the .text section, if present;
1469 // 5. the address 0.
1470 template <class ELFT> typename ELFT::uint Writer<ELFT>::getEntryAddr() {
1471   // Case 1, 2 or 3. As a special case, if the symbol is actually
1472   // a number, we'll use that number as an address.
1473   if (SymbolBody *B = Symtab<ELFT>::X->find(Config->Entry))
1474     return B->getVA<ELFT>();
1475   uint64_t Addr;
1476   if (!Config->Entry.getAsInteger(0, Addr))
1477     return Addr;
1478 
1479   // Case 4
1480   if (OutputSectionBase *Sec = findSection(".text")) {
1481     if (Config->WarnMissingEntry)
1482       warn("cannot find entry symbol " + Config->Entry + "; defaulting to 0x" +
1483            utohexstr(Sec->Addr));
1484     return Sec->Addr;
1485   }
1486 
1487   // Case 5
1488   if (Config->WarnMissingEntry)
1489     warn("cannot find entry symbol " + Config->Entry +
1490          "; not setting start address");
1491   return 0;
1492 }
1493 
1494 template <class ELFT> static uint8_t getELFEncoding() {
1495   if (ELFT::TargetEndianness == llvm::support::little)
1496     return ELFDATA2LSB;
1497   return ELFDATA2MSB;
1498 }
1499 
1500 static uint16_t getELFType() {
1501   if (Config->Pic)
1502     return ET_DYN;
1503   if (Config->Relocatable)
1504     return ET_REL;
1505   return ET_EXEC;
1506 }
1507 
1508 // This function is called after we have assigned address and size
1509 // to each section. This function fixes some predefined absolute
1510 // symbol values that depend on section address and size.
1511 template <class ELFT> void Writer<ELFT>::fixAbsoluteSymbols() {
1512   // __ehdr_start is the location of program headers.
1513   if (ElfSym<ELFT>::EhdrStart)
1514     ElfSym<ELFT>::EhdrStart->Value = Out<ELFT>::ProgramHeaders->Addr;
1515 
1516   auto Set = [](DefinedRegular<ELFT> *S1, DefinedRegular<ELFT> *S2, uintX_t V) {
1517     if (S1)
1518       S1->Value = V;
1519     if (S2)
1520       S2->Value = V;
1521   };
1522 
1523   // _etext is the first location after the last read-only loadable segment.
1524   // _edata is the first location after the last read-write loadable segment.
1525   // _end is the first location after the uninitialized data region.
1526   for (PhdrEntry &P : Phdrs) {
1527     if (P.p_type != PT_LOAD)
1528       continue;
1529     Set(ElfSym<ELFT>::End, ElfSym<ELFT>::End2, P.p_vaddr + P.p_memsz);
1530 
1531     uintX_t Val = P.p_vaddr + P.p_filesz;
1532     if (P.p_flags & PF_W)
1533       Set(ElfSym<ELFT>::Edata, ElfSym<ELFT>::Edata2, Val);
1534     else
1535       Set(ElfSym<ELFT>::Etext, ElfSym<ELFT>::Etext2, Val);
1536   }
1537 
1538   // Setup MIPS _gp_disp/__gnu_local_gp symbols which should
1539   // be equal to the _gp symbol's value.
1540   if (Config->EMachine == EM_MIPS) {
1541     if (!ElfSym<ELFT>::MipsGp->Value) {
1542       // Find GP-relative section with the lowest address
1543       // and use this address to calculate default _gp value.
1544       uintX_t Gp = -1;
1545       for (const OutputSectionBase * OS : OutputSections)
1546         if ((OS->Flags & SHF_MIPS_GPREL) && OS->Addr < Gp)
1547           Gp = OS->Addr;
1548       if (Gp != (uintX_t)-1)
1549         ElfSym<ELFT>::MipsGp->Value = Gp + 0x7ff0;
1550     }
1551     if (ElfSym<ELFT>::MipsGpDisp)
1552       ElfSym<ELFT>::MipsGpDisp->Value = ElfSym<ELFT>::MipsGp->Value;
1553     if (ElfSym<ELFT>::MipsLocalGp)
1554       ElfSym<ELFT>::MipsLocalGp->Value = ElfSym<ELFT>::MipsGp->Value;
1555   }
1556 }
1557 
1558 template <class ELFT> void Writer<ELFT>::writeHeader() {
1559   uint8_t *Buf = Buffer->getBufferStart();
1560   memcpy(Buf, "\177ELF", 4);
1561 
1562   // Write the ELF header.
1563   auto *EHdr = reinterpret_cast<Elf_Ehdr *>(Buf);
1564   EHdr->e_ident[EI_CLASS] = ELFT::Is64Bits ? ELFCLASS64 : ELFCLASS32;
1565   EHdr->e_ident[EI_DATA] = getELFEncoding<ELFT>();
1566   EHdr->e_ident[EI_VERSION] = EV_CURRENT;
1567   EHdr->e_ident[EI_OSABI] = Config->OSABI;
1568   EHdr->e_type = getELFType();
1569   EHdr->e_machine = Config->EMachine;
1570   EHdr->e_version = EV_CURRENT;
1571   EHdr->e_entry = getEntryAddr();
1572   EHdr->e_shoff = SectionHeaderOff;
1573   EHdr->e_ehsize = sizeof(Elf_Ehdr);
1574   EHdr->e_phnum = Phdrs.size();
1575   EHdr->e_shentsize = sizeof(Elf_Shdr);
1576   EHdr->e_shnum = OutputSections.size() + 1;
1577   EHdr->e_shstrndx = In<ELFT>::ShStrTab->OutSec->SectionIndex;
1578 
1579   if (Config->EMachine == EM_ARM)
1580     // We don't currently use any features incompatible with EF_ARM_EABI_VER5,
1581     // but we don't have any firm guarantees of conformance. Linux AArch64
1582     // kernels (as of 2016) require an EABI version to be set.
1583     EHdr->e_flags = EF_ARM_EABI_VER5;
1584   else if (Config->EMachine == EM_MIPS)
1585     EHdr->e_flags = getMipsEFlags<ELFT>();
1586 
1587   if (!Config->Relocatable) {
1588     EHdr->e_phoff = sizeof(Elf_Ehdr);
1589     EHdr->e_phentsize = sizeof(Elf_Phdr);
1590   }
1591 
1592   // Write the program header table.
1593   auto *HBuf = reinterpret_cast<Elf_Phdr *>(Buf + EHdr->e_phoff);
1594   for (PhdrEntry &P : Phdrs) {
1595     HBuf->p_type = P.p_type;
1596     HBuf->p_flags = P.p_flags;
1597     HBuf->p_offset = P.p_offset;
1598     HBuf->p_vaddr = P.p_vaddr;
1599     HBuf->p_paddr = P.p_paddr;
1600     HBuf->p_filesz = P.p_filesz;
1601     HBuf->p_memsz = P.p_memsz;
1602     HBuf->p_align = P.p_align;
1603     ++HBuf;
1604   }
1605 
1606   // Write the section header table. Note that the first table entry is null.
1607   auto *SHdrs = reinterpret_cast<Elf_Shdr *>(Buf + EHdr->e_shoff);
1608   for (OutputSectionBase *Sec : OutputSections)
1609     Sec->writeHeaderTo<ELFT>(++SHdrs);
1610 }
1611 
1612 // Removes a given file asynchronously. This is a performance hack,
1613 // so remove this when operating systems are improved.
1614 //
1615 // On Linux (and probably on other Unix-like systems), unlink(2) is a
1616 // noticeably slow system call. As of 2016, unlink takes 250
1617 // milliseconds to remove a 1 GB file on ext4 filesystem on my machine.
1618 //
1619 // To create a new result file, we first remove existing file. So, if
1620 // you repeatedly link a 1 GB program in a regular compile-link-debug
1621 // cycle, every cycle wastes 250 milliseconds only to remove a file.
1622 // Since LLD can link a 1 GB binary in about 5 seconds, that waste
1623 // actually counts.
1624 //
1625 // This function spawns a background thread to call unlink.
1626 // The calling thread returns almost immediately.
1627 static void unlinkAsync(StringRef Path) {
1628   if (!Config->Threads || !sys::fs::exists(Config->OutputFile))
1629     return;
1630 
1631   // First, rename Path to avoid race condition. We cannot remomve
1632   // Path from a different thread because we are now going to create
1633   // Path as a new file. If we do that in a different thread, the new
1634   // thread can remove the new file.
1635   SmallString<128> TempPath;
1636   if (auto EC = sys::fs::createUniqueFile(Path + "tmp%%%%%%%%", TempPath))
1637     fatal(EC, "createUniqueFile failed");
1638   if (auto EC = sys::fs::rename(Path, TempPath))
1639     fatal(EC, "rename failed");
1640 
1641   // Remove TempPath in background.
1642   std::thread([=] { ::remove(TempPath.str().str().c_str()); }).detach();
1643 }
1644 
1645 // Open a result file.
1646 template <class ELFT> void Writer<ELFT>::openFile() {
1647   unlinkAsync(Config->OutputFile);
1648   ErrorOr<std::unique_ptr<FileOutputBuffer>> BufferOrErr =
1649       FileOutputBuffer::create(Config->OutputFile, FileSize,
1650                                FileOutputBuffer::F_executable);
1651 
1652   if (auto EC = BufferOrErr.getError())
1653     error(EC, "failed to open " + Config->OutputFile);
1654   else
1655     Buffer = std::move(*BufferOrErr);
1656 }
1657 
1658 template <class ELFT> void Writer<ELFT>::writeSectionsBinary() {
1659   uint8_t *Buf = Buffer->getBufferStart();
1660   for (OutputSectionBase *Sec : OutputSections)
1661     if (Sec->Flags & SHF_ALLOC)
1662       Sec->writeTo(Buf + Sec->Offset);
1663 }
1664 
1665 // Write section contents to a mmap'ed file.
1666 template <class ELFT> void Writer<ELFT>::writeSections() {
1667   uint8_t *Buf = Buffer->getBufferStart();
1668 
1669   // PPC64 needs to process relocations in the .opd section
1670   // before processing relocations in code-containing sections.
1671   Out<ELFT>::Opd = findSection(".opd");
1672   if (Out<ELFT>::Opd) {
1673     Out<ELFT>::OpdBuf = Buf + Out<ELFT>::Opd->Offset;
1674     Out<ELFT>::Opd->writeTo(Buf + Out<ELFT>::Opd->Offset);
1675   }
1676 
1677   OutputSectionBase *EhFrameHdr =
1678       In<ELFT>::EhFrameHdr ? In<ELFT>::EhFrameHdr->OutSec : nullptr;
1679   for (OutputSectionBase *Sec : OutputSections)
1680     if (Sec != Out<ELFT>::Opd && Sec != EhFrameHdr)
1681       Sec->writeTo(Buf + Sec->Offset);
1682 
1683   // The .eh_frame_hdr depends on .eh_frame section contents, therefore
1684   // it should be written after .eh_frame is written.
1685   if (!Out<ELFT>::EhFrame->empty() && EhFrameHdr)
1686     EhFrameHdr->writeTo(Buf + EhFrameHdr->Offset);
1687 }
1688 
1689 template <class ELFT> void Writer<ELFT>::writeBuildId() {
1690   if (!In<ELFT>::BuildId || !In<ELFT>::BuildId->OutSec)
1691     return;
1692 
1693   // Compute a hash of all sections of the output file.
1694   uint8_t *Start = Buffer->getBufferStart();
1695   uint8_t *End = Start + FileSize;
1696   In<ELFT>::BuildId->writeBuildId({Start, End});
1697 }
1698 
1699 template void elf::writeResult<ELF32LE>();
1700 template void elf::writeResult<ELF32BE>();
1701 template void elf::writeResult<ELF64LE>();
1702 template void elf::writeResult<ELF64BE>();
1703 
1704 template void elf::allocateHeaders<ELF32LE>(MutableArrayRef<PhdrEntry>,
1705                                             ArrayRef<OutputSectionBase *>);
1706 template void elf::allocateHeaders<ELF32BE>(MutableArrayRef<PhdrEntry>,
1707                                             ArrayRef<OutputSectionBase *>);
1708 template void elf::allocateHeaders<ELF64LE>(MutableArrayRef<PhdrEntry>,
1709                                             ArrayRef<OutputSectionBase *>);
1710 template void elf::allocateHeaders<ELF64BE>(MutableArrayRef<PhdrEntry>,
1711                                             ArrayRef<OutputSectionBase *>);
1712 
1713 template bool elf::isRelroSection<ELF32LE>(const OutputSectionBase *);
1714 template bool elf::isRelroSection<ELF32BE>(const OutputSectionBase *);
1715 template bool elf::isRelroSection<ELF64LE>(const OutputSectionBase *);
1716 template bool elf::isRelroSection<ELF64BE>(const OutputSectionBase *);
1717 
1718 template void elf::reportDiscarded<ELF32LE>(InputSectionBase<ELF32LE> *);
1719 template void elf::reportDiscarded<ELF32BE>(InputSectionBase<ELF32BE> *);
1720 template void elf::reportDiscarded<ELF64LE>(InputSectionBase<ELF64LE> *);
1721 template void elf::reportDiscarded<ELF64BE>(InputSectionBase<ELF64BE> *);
1722