xref: /llvm-project-15.0.7/lld/ELF/Writer.cpp (revision 3076db8d)
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 "OutputSections.h"
14 #include "Relocations.h"
15 #include "Strings.h"
16 #include "SymbolTable.h"
17 #include "Target.h"
18 
19 #include "llvm/ADT/StringMap.h"
20 #include "llvm/ADT/StringSwitch.h"
21 #include "llvm/Support/FileOutputBuffer.h"
22 #include "llvm/Support/StringSaver.h"
23 #include "llvm/Support/raw_ostream.h"
24 
25 using namespace llvm;
26 using namespace llvm::ELF;
27 using namespace llvm::object;
28 
29 using namespace lld;
30 using namespace lld::elf;
31 
32 namespace {
33 // The writer writes a SymbolTable result to a file.
34 template <class ELFT> class Writer {
35 public:
36   typedef typename ELFT::uint uintX_t;
37   typedef typename ELFT::Shdr Elf_Shdr;
38   typedef typename ELFT::Ehdr Elf_Ehdr;
39   typedef typename ELFT::Phdr Elf_Phdr;
40   typedef typename ELFT::Sym Elf_Sym;
41   typedef typename ELFT::SymRange Elf_Sym_Range;
42   typedef typename ELFT::Rela Elf_Rela;
43   Writer(SymbolTable<ELFT> &S) : Symtab(S) {}
44   void run();
45 
46 private:
47   typedef PhdrEntry<ELFT> Phdr;
48 
49   void copyLocalSymbols();
50   void addReservedSymbols();
51   void createSections();
52   void forEachRelSec(
53       std::function<void(InputSectionBase<ELFT> &, const typename ELFT::Shdr &)>
54           Fn);
55   void finalizeSections();
56   void addPredefinedSections();
57   bool needsGot();
58 
59   std::vector<Phdr> createPhdrs();
60   void assignAddresses();
61   void assignFileOffsets();
62   void setPhdrs();
63   void fixHeaders();
64   void fixSectionAlignments();
65   void fixAbsoluteSymbols();
66   void openFile();
67   void writeHeader();
68   void writeSections();
69   void writeBuildId();
70 
71   std::unique_ptr<FileOutputBuffer> Buffer;
72 
73   BumpPtrAllocator Alloc;
74   std::vector<OutputSectionBase<ELFT> *> OutputSections;
75   OutputSectionFactory<ELFT> Factory;
76 
77   void addRelIpltSymbols();
78   void addStartEndSymbols();
79   void addStartStopSymbols(OutputSectionBase<ELFT> *Sec);
80 
81   SymbolTable<ELFT> &Symtab;
82   std::vector<Phdr> Phdrs;
83 
84   uintX_t FileSize;
85   uintX_t SectionHeaderOff;
86 };
87 } // anonymous namespace
88 
89 template <class ELFT>
90 StringRef elf::getOutputSectionName(InputSectionBase<ELFT> *S) {
91   StringRef Name = S->getSectionName();
92   for (StringRef V : {".text.", ".rodata.", ".data.rel.ro.", ".data.", ".bss.",
93                       ".init_array.", ".fini_array.", ".ctors.", ".dtors.",
94                       ".tbss.", ".gcc_except_table.", ".tdata."})
95     if (Name.startswith(V))
96       return V.drop_back();
97   return Name;
98 }
99 
100 template <class ELFT> void elf::reportDiscarded(InputSectionBase<ELFT> *IS) {
101   if (!Config->PrintGcSections || !IS || IS->Live)
102     return;
103   errs() << "removing unused section from '" << IS->getSectionName()
104          << "' in file '" << IS->getFile()->getName() << "'\n";
105 }
106 
107 template <class ELFT> static bool needsInterpSection() {
108   return !Symtab<ELFT>::X->getSharedFiles().empty() &&
109          !Config->DynamicLinker.empty();
110 }
111 
112 template <class ELFT> void elf::writeResult(SymbolTable<ELFT> *Symtab) {
113   typedef typename ELFT::uint uintX_t;
114   typedef typename ELFT::Ehdr Elf_Ehdr;
115 
116   // Create singleton output sections.
117   OutputSection<ELFT> Bss(".bss", SHT_NOBITS, SHF_ALLOC | SHF_WRITE);
118   DynamicSection<ELFT> Dynamic;
119   EhOutputSection<ELFT> EhFrame;
120   GotSection<ELFT> Got;
121   PltSection<ELFT> Plt;
122   RelocationSection<ELFT> RelaDyn(Config->Rela ? ".rela.dyn" : ".rel.dyn",
123                                   Config->ZCombreloc);
124   StringTableSection<ELFT> DynStrTab(".dynstr", true);
125   StringTableSection<ELFT> ShStrTab(".shstrtab", false);
126   SymbolTableSection<ELFT> DynSymTab(DynStrTab);
127   VersionTableSection<ELFT> VerSym;
128   VersionNeedSection<ELFT> VerNeed;
129 
130   OutputSectionBase<ELFT> ElfHeader("", 0, SHF_ALLOC);
131   ElfHeader.setSize(sizeof(Elf_Ehdr));
132   OutputSectionBase<ELFT> ProgramHeaders("", 0, SHF_ALLOC);
133   ProgramHeaders.updateAlignment(sizeof(uintX_t));
134 
135   // Instantiate optional output sections if they are needed.
136   std::unique_ptr<InterpSection<ELFT>> Interp;
137   std::unique_ptr<BuildIdSection<ELFT>> BuildId;
138   std::unique_ptr<EhFrameHeader<ELFT>> EhFrameHdr;
139   std::unique_ptr<GnuHashTableSection<ELFT>> GnuHashTab;
140   std::unique_ptr<GotPltSection<ELFT>> GotPlt;
141   std::unique_ptr<HashTableSection<ELFT>> HashTab;
142   std::unique_ptr<RelocationSection<ELFT>> RelaPlt;
143   std::unique_ptr<StringTableSection<ELFT>> StrTab;
144   std::unique_ptr<SymbolTableSection<ELFT>> SymTabSec;
145   std::unique_ptr<OutputSection<ELFT>> MipsRldMap;
146   std::unique_ptr<VersionDefinitionSection<ELFT>> VerDef;
147 
148   if (needsInterpSection<ELFT>())
149     Interp.reset(new InterpSection<ELFT>);
150 
151   if (Config->BuildId == BuildIdKind::Fnv1)
152     BuildId.reset(new BuildIdFnv1<ELFT>);
153   else if (Config->BuildId == BuildIdKind::Md5)
154     BuildId.reset(new BuildIdMd5<ELFT>);
155   else if (Config->BuildId == BuildIdKind::Sha1)
156     BuildId.reset(new BuildIdSha1<ELFT>);
157   else if (Config->BuildId == BuildIdKind::Hexstring)
158     BuildId.reset(new BuildIdHexstring<ELFT>);
159 
160   if (Config->EhFrameHdr)
161     EhFrameHdr.reset(new EhFrameHeader<ELFT>);
162 
163   if (Config->GnuHash)
164     GnuHashTab.reset(new GnuHashTableSection<ELFT>);
165   if (Config->SysvHash)
166     HashTab.reset(new HashTableSection<ELFT>);
167   StringRef S = Config->Rela ? ".rela.plt" : ".rel.plt";
168   GotPlt.reset(new GotPltSection<ELFT>);
169   RelaPlt.reset(new RelocationSection<ELFT>(S, false /*Sort*/));
170   if (!Config->StripAll) {
171     StrTab.reset(new StringTableSection<ELFT>(".strtab", false));
172     SymTabSec.reset(new SymbolTableSection<ELFT>(*StrTab));
173   }
174   if (Config->EMachine == EM_MIPS && !Config->Shared) {
175     // This is a MIPS specific section to hold a space within the data segment
176     // of executable file which is pointed to by the DT_MIPS_RLD_MAP entry.
177     // See "Dynamic section" in Chapter 5 in the following document:
178     // ftp://www.linux-mips.org/pub/linux/mips/doc/ABI/mipsabi.pdf
179     MipsRldMap.reset(new OutputSection<ELFT>(".rld_map", SHT_PROGBITS,
180                                              SHF_ALLOC | SHF_WRITE));
181     MipsRldMap->setSize(sizeof(uintX_t));
182     MipsRldMap->updateAlignment(sizeof(uintX_t));
183   }
184   if (!Config->VersionDefinitions.empty())
185     VerDef.reset(new VersionDefinitionSection<ELFT>());
186 
187   Out<ELFT>::Bss = &Bss;
188   Out<ELFT>::BuildId = BuildId.get();
189   Out<ELFT>::DynStrTab = &DynStrTab;
190   Out<ELFT>::DynSymTab = &DynSymTab;
191   Out<ELFT>::Dynamic = &Dynamic;
192   Out<ELFT>::EhFrame = &EhFrame;
193   Out<ELFT>::EhFrameHdr = EhFrameHdr.get();
194   Out<ELFT>::GnuHashTab = GnuHashTab.get();
195   Out<ELFT>::Got = &Got;
196   Out<ELFT>::GotPlt = GotPlt.get();
197   Out<ELFT>::HashTab = HashTab.get();
198   Out<ELFT>::Interp = Interp.get();
199   Out<ELFT>::Plt = &Plt;
200   Out<ELFT>::RelaDyn = &RelaDyn;
201   Out<ELFT>::RelaPlt = RelaPlt.get();
202   Out<ELFT>::ShStrTab = &ShStrTab;
203   Out<ELFT>::StrTab = StrTab.get();
204   Out<ELFT>::SymTab = SymTabSec.get();
205   Out<ELFT>::VerDef = VerDef.get();
206   Out<ELFT>::VerSym = &VerSym;
207   Out<ELFT>::VerNeed = &VerNeed;
208   Out<ELFT>::MipsRldMap = MipsRldMap.get();
209   Out<ELFT>::Opd = nullptr;
210   Out<ELFT>::OpdBuf = nullptr;
211   Out<ELFT>::TlsPhdr = nullptr;
212   Out<ELFT>::ElfHeader = &ElfHeader;
213   Out<ELFT>::ProgramHeaders = &ProgramHeaders;
214 
215   Writer<ELFT>(*Symtab).run();
216 }
217 
218 template <class ELFT>
219 static std::vector<DefinedCommon<ELFT> *> getCommonSymbols() {
220   std::vector<DefinedCommon<ELFT> *> V;
221   for (Symbol *S : Symtab<ELFT>::X->getSymbols())
222     if (auto *B = dyn_cast<DefinedCommon<ELFT>>(S->body()))
223       V.push_back(B);
224   return V;
225 }
226 
227 // The main function of the writer.
228 template <class ELFT> void Writer<ELFT>::run() {
229   if (!Config->DiscardAll)
230     copyLocalSymbols();
231   addReservedSymbols();
232 
233   CommonInputSection<ELFT> Common(getCommonSymbols<ELFT>());
234   CommonInputSection<ELFT>::X = &Common;
235 
236   if (ScriptConfig->HasContents)
237     Script<ELFT>::X->createSections(&OutputSections, Factory);
238   else
239     createSections();
240 
241   finalizeSections();
242   if (HasError)
243     return;
244 
245   if (Config->Relocatable) {
246     assignFileOffsets();
247   } else {
248     Phdrs = Script<ELFT>::X->hasPhdrsCommands()
249                 ? Script<ELFT>::X->createPhdrs(OutputSections)
250                 : createPhdrs();
251     fixHeaders();
252     if (ScriptConfig->HasContents) {
253       Script<ELFT>::X->assignAddresses(OutputSections);
254     } else {
255       fixSectionAlignments();
256       assignAddresses();
257     }
258     assignFileOffsets();
259     setPhdrs();
260     fixAbsoluteSymbols();
261   }
262 
263   openFile();
264   if (HasError)
265     return;
266   writeHeader();
267   writeSections();
268   writeBuildId();
269   if (HasError)
270     return;
271   if (auto EC = Buffer->commit())
272     error(EC, "failed to write to the output file");
273 }
274 
275 template <class ELFT>
276 static void reportUndefined(SymbolTable<ELFT> &Symtab, SymbolBody *Sym) {
277   if (Config->UnresolvedSymbols == UnresolvedPolicy::Ignore)
278     return;
279 
280   if (Config->Shared && Sym->symbol()->Visibility == STV_DEFAULT &&
281       Config->UnresolvedSymbols != UnresolvedPolicy::NoUndef)
282     return;
283 
284   std::string Msg = "undefined symbol: " + Sym->getName().str();
285   if (Sym->File)
286     Msg += " in " + getFilename(Sym->File);
287   if (Config->UnresolvedSymbols == UnresolvedPolicy::Warn)
288     warning(Msg);
289   else
290     error(Msg);
291 }
292 
293 template <class ELFT>
294 static bool shouldKeepInSymtab(InputSectionBase<ELFT> *Sec, StringRef SymName,
295                                const SymbolBody &B) {
296   if (B.isFile())
297     return false;
298 
299   // We keep sections in symtab for relocatable output.
300   if (B.isSection())
301     return Config->Relocatable;
302 
303   // If sym references a section in a discarded group, don't keep it.
304   if (Sec == &InputSection<ELFT>::Discarded)
305     return false;
306 
307   if (Config->DiscardNone)
308     return true;
309 
310   // In ELF assembly .L symbols are normally discarded by the assembler.
311   // If the assembler fails to do so, the linker discards them if
312   // * --discard-locals is used.
313   // * The symbol is in a SHF_MERGE section, which is normally the reason for
314   //   the assembler keeping the .L symbol.
315   if (!SymName.startswith(".L") && !SymName.empty())
316     return true;
317 
318   if (Config->DiscardLocals)
319     return false;
320 
321   return !(Sec->getSectionHdr()->sh_flags & SHF_MERGE);
322 }
323 
324 template <class ELFT> static bool includeInSymtab(const SymbolBody &B) {
325   if (!B.isLocal() && !B.symbol()->IsUsedInRegularObj)
326     return false;
327 
328   if (auto *D = dyn_cast<DefinedRegular<ELFT>>(&B)) {
329     // Always include absolute symbols.
330     if (!D->Section)
331       return true;
332     // Exclude symbols pointing to garbage-collected sections.
333     if (!D->Section->Live)
334       return false;
335     if (auto *S = dyn_cast<MergeInputSection<ELFT>>(D->Section))
336       if (!S->getSectionPiece(D->Value)->Live)
337         return false;
338   }
339   return true;
340 }
341 
342 // Local symbols are not in the linker's symbol table. This function scans
343 // each object file's symbol table to copy local symbols to the output.
344 template <class ELFT> void Writer<ELFT>::copyLocalSymbols() {
345   if (!Out<ELFT>::SymTab)
346     return;
347   for (const std::unique_ptr<elf::ObjectFile<ELFT>> &F :
348        Symtab.getObjectFiles()) {
349     const char *StrTab = F->getStringTable().data();
350     for (SymbolBody *B : F->getLocalSymbols()) {
351       auto *DR = dyn_cast<DefinedRegular<ELFT>>(B);
352       // No reason to keep local undefined symbol in symtab.
353       if (!DR)
354         continue;
355       if (!includeInSymtab<ELFT>(*B))
356         continue;
357       StringRef SymName(StrTab + B->getNameOffset());
358       InputSectionBase<ELFT> *Sec = DR->Section;
359       if (!shouldKeepInSymtab<ELFT>(Sec, SymName, *B))
360         continue;
361       ++Out<ELFT>::SymTab->NumLocals;
362       if (Config->Relocatable)
363         B->DynsymIndex = Out<ELFT>::SymTab->NumLocals;
364       F->KeptLocalSyms.push_back(
365           std::make_pair(DR, Out<ELFT>::SymTab->StrTabSec.addString(SymName)));
366     }
367   }
368 }
369 
370 // PPC64 has a number of special SHT_PROGBITS+SHF_ALLOC+SHF_WRITE sections that
371 // we would like to make sure appear is a specific order to maximize their
372 // coverage by a single signed 16-bit offset from the TOC base pointer.
373 // Conversely, the special .tocbss section should be first among all SHT_NOBITS
374 // sections. This will put it next to the loaded special PPC64 sections (and,
375 // thus, within reach of the TOC base pointer).
376 static int getPPC64SectionRank(StringRef SectionName) {
377   return StringSwitch<int>(SectionName)
378       .Case(".tocbss", 0)
379       .Case(".branch_lt", 2)
380       .Case(".toc", 3)
381       .Case(".toc1", 4)
382       .Case(".opd", 5)
383       .Default(1);
384 }
385 
386 template <class ELFT> bool elf::isRelroSection(OutputSectionBase<ELFT> *Sec) {
387   if (!Config->ZRelro)
388     return false;
389   typename ELFT::uint Flags = Sec->getFlags();
390   if (!(Flags & SHF_ALLOC) || !(Flags & SHF_WRITE))
391     return false;
392   if (Flags & SHF_TLS)
393     return true;
394   uint32_t Type = Sec->getType();
395   if (Type == SHT_INIT_ARRAY || Type == SHT_FINI_ARRAY ||
396       Type == SHT_PREINIT_ARRAY)
397     return true;
398   if (Sec == Out<ELFT>::GotPlt)
399     return Config->ZNow;
400   if (Sec == Out<ELFT>::Dynamic || Sec == Out<ELFT>::Got)
401     return true;
402   StringRef S = Sec->getName();
403   return S == ".data.rel.ro" || S == ".ctors" || S == ".dtors" || S == ".jcr" ||
404          S == ".eh_frame";
405 }
406 
407 // Output section ordering is determined by this function.
408 template <class ELFT>
409 static bool compareSections(OutputSectionBase<ELFT> *A,
410                             OutputSectionBase<ELFT> *B) {
411   typedef typename ELFT::uint uintX_t;
412 
413   int Comp = Script<ELFT>::X->compareSections(A->getName(), B->getName());
414   if (Comp != 0)
415     return Comp < 0;
416 
417   uintX_t AFlags = A->getFlags();
418   uintX_t BFlags = B->getFlags();
419 
420   // Allocatable sections go first to reduce the total PT_LOAD size and
421   // so debug info doesn't change addresses in actual code.
422   bool AIsAlloc = AFlags & SHF_ALLOC;
423   bool BIsAlloc = BFlags & SHF_ALLOC;
424   if (AIsAlloc != BIsAlloc)
425     return AIsAlloc;
426 
427   // We don't have any special requirements for the relative order of
428   // two non allocatable sections.
429   if (!AIsAlloc)
430     return false;
431 
432   // We want the read only sections first so that they go in the PT_LOAD
433   // covering the program headers at the start of the file.
434   bool AIsWritable = AFlags & SHF_WRITE;
435   bool BIsWritable = BFlags & SHF_WRITE;
436   if (AIsWritable != BIsWritable)
437     return BIsWritable;
438 
439   // For a corresponding reason, put non exec sections first (the program
440   // header PT_LOAD is not executable).
441   bool AIsExec = AFlags & SHF_EXECINSTR;
442   bool BIsExec = BFlags & SHF_EXECINSTR;
443   if (AIsExec != BIsExec)
444     return BIsExec;
445 
446   // If we got here we know that both A and B are in the same PT_LOAD.
447 
448   // The TLS initialization block needs to be a single contiguous block in a R/W
449   // PT_LOAD, so stick TLS sections directly before R/W sections. The TLS NOBITS
450   // sections are placed here as they don't take up virtual address space in the
451   // PT_LOAD.
452   bool AIsTls = AFlags & SHF_TLS;
453   bool BIsTls = BFlags & SHF_TLS;
454   if (AIsTls != BIsTls)
455     return AIsTls;
456 
457   // The next requirement we have is to put nobits sections last. The
458   // reason is that the only thing the dynamic linker will see about
459   // them is a p_memsz that is larger than p_filesz. Seeing that it
460   // zeros the end of the PT_LOAD, so that has to correspond to the
461   // nobits sections.
462   bool AIsNoBits = A->getType() == SHT_NOBITS;
463   bool BIsNoBits = B->getType() == SHT_NOBITS;
464   if (AIsNoBits != BIsNoBits)
465     return BIsNoBits;
466 
467   // We place RelRo section before plain r/w ones.
468   bool AIsRelRo = isRelroSection(A);
469   bool BIsRelRo = isRelroSection(B);
470   if (AIsRelRo != BIsRelRo)
471     return AIsRelRo;
472 
473   // Some architectures have additional ordering restrictions for sections
474   // within the same PT_LOAD.
475   if (Config->EMachine == EM_PPC64)
476     return getPPC64SectionRank(A->getName()) <
477            getPPC64SectionRank(B->getName());
478 
479   return false;
480 }
481 
482 template <class ELFT> bool elf::isOutputDynamic() {
483   return !Symtab<ELFT>::X->getSharedFiles().empty() || Config->Pic;
484 }
485 
486 template <class ELFT> static bool isDiscarded(InputSectionBase<ELFT> *S) {
487   return !S || S == &InputSection<ELFT>::Discarded || !S->Live;
488 }
489 
490 // Program header entry
491 template<class ELFT>
492 PhdrEntry<ELFT>::PhdrEntry(unsigned Type, unsigned Flags) {
493   H.p_type = Type;
494   H.p_flags = Flags;
495 }
496 
497 template<class ELFT>
498 void PhdrEntry<ELFT>::add(OutputSectionBase<ELFT> *Sec) {
499   Last = Sec;
500   if (!First)
501     First = Sec;
502   H.p_align = std::max<typename ELFT::uint>(H.p_align, Sec->getAlignment());
503 }
504 
505 template <class ELFT>
506 static Symbol *addOptionalSynthetic(SymbolTable<ELFT> &Table, StringRef Name,
507                                     OutputSectionBase<ELFT> *Sec,
508                                     typename ELFT::uint Val) {
509   SymbolBody *S = Table.find(Name);
510   if (!S)
511     return nullptr;
512   if (!S->isUndefined() && !S->isShared())
513     return S->symbol();
514   return Table.addSynthetic(Name, Sec, Val);
515 }
516 
517 // The beginning and the ending of .rel[a].plt section are marked
518 // with __rel[a]_iplt_{start,end} symbols if it is a statically linked
519 // executable. The runtime needs these symbols in order to resolve
520 // all IRELATIVE relocs on startup. For dynamic executables, we don't
521 // need these symbols, since IRELATIVE relocs are resolved through GOT
522 // and PLT. For details, see http://www.airs.com/blog/archives/403.
523 template <class ELFT> void Writer<ELFT>::addRelIpltSymbols() {
524   if (isOutputDynamic<ELFT>() || !Out<ELFT>::RelaPlt)
525     return;
526   StringRef S = Config->Rela ? "__rela_iplt_start" : "__rel_iplt_start";
527   addOptionalSynthetic(Symtab, S, Out<ELFT>::RelaPlt, 0);
528 
529   S = Config->Rela ? "__rela_iplt_end" : "__rel_iplt_end";
530   addOptionalSynthetic(Symtab, S, Out<ELFT>::RelaPlt,
531                        DefinedSynthetic<ELFT>::SectionEnd);
532 }
533 
534 // The linker is expected to define some symbols depending on
535 // the linking result. This function defines such symbols.
536 template <class ELFT> void Writer<ELFT>::addReservedSymbols() {
537   if (Config->EMachine == EM_MIPS && !Config->Relocatable) {
538     // Define _gp for MIPS. st_value of _gp symbol will be updated by Writer
539     // so that it points to an absolute address which is relative to GOT.
540     // See "Global Data Symbols" in Chapter 6 in the following document:
541     // ftp://www.linux-mips.org/pub/linux/mips/doc/ABI/mipsabi.pdf
542     Symtab.addSynthetic("_gp", Out<ELFT>::Got, MipsGPOffset);
543 
544     // On MIPS O32 ABI, _gp_disp is a magic symbol designates offset between
545     // start of function and 'gp' pointer into GOT.
546     Symbol *Sym =
547         addOptionalSynthetic(Symtab, "_gp_disp", Out<ELFT>::Got, MipsGPOffset);
548     if (Sym)
549       ElfSym<ELFT>::MipsGpDisp = Sym->body();
550 
551     // The __gnu_local_gp is a magic symbol equal to the current value of 'gp'
552     // pointer. This symbol is used in the code generated by .cpload pseudo-op
553     // in case of using -mno-shared option.
554     // https://sourceware.org/ml/binutils/2004-12/msg00094.html
555     addOptionalSynthetic(Symtab, "__gnu_local_gp", Out<ELFT>::Got,
556                          MipsGPOffset);
557   }
558 
559   // In the assembly for 32 bit x86 the _GLOBAL_OFFSET_TABLE_ symbol
560   // is magical and is used to produce a R_386_GOTPC relocation.
561   // The R_386_GOTPC relocation value doesn't actually depend on the
562   // symbol value, so it could use an index of STN_UNDEF which, according
563   // to the spec, means the symbol value is 0.
564   // Unfortunately both gas and MC keep the _GLOBAL_OFFSET_TABLE_ symbol in
565   // the object file.
566   // The situation is even stranger on x86_64 where the assembly doesn't
567   // need the magical symbol, but gas still puts _GLOBAL_OFFSET_TABLE_ as
568   // an undefined symbol in the .o files.
569   // Given that the symbol is effectively unused, we just create a dummy
570   // hidden one to avoid the undefined symbol error.
571   if (!Config->Relocatable)
572     Symtab.addIgnored("_GLOBAL_OFFSET_TABLE_");
573 
574   // __tls_get_addr is defined by the dynamic linker for dynamic ELFs. For
575   // static linking the linker is required to optimize away any references to
576   // __tls_get_addr, so it's not defined anywhere. Create a hidden definition
577   // to avoid the undefined symbol error.
578   if (!isOutputDynamic<ELFT>())
579     Symtab.addIgnored("__tls_get_addr");
580 
581   auto Define = [this](StringRef S, DefinedRegular<ELFT> *&Sym1,
582                        DefinedRegular<ELFT> *&Sym2) {
583     Sym1 = Symtab.addIgnored(S, STV_DEFAULT);
584 
585     // The name without the underscore is not a reserved name,
586     // so it is defined only when there is a reference against it.
587     assert(S.startswith("_"));
588     S = S.substr(1);
589     if (SymbolBody *B = Symtab.find(S))
590       if (B->isUndefined())
591         Sym2 = Symtab.addAbsolute(S, STV_DEFAULT);
592   };
593 
594   Define("_end", ElfSym<ELFT>::End, ElfSym<ELFT>::End2);
595   Define("_etext", ElfSym<ELFT>::Etext, ElfSym<ELFT>::Etext2);
596   Define("_edata", ElfSym<ELFT>::Edata, ElfSym<ELFT>::Edata2);
597 }
598 
599 // Sort input sections by section name suffixes for
600 // __attribute__((init_priority(N))).
601 template <class ELFT> static void sortInitFini(OutputSectionBase<ELFT> *S) {
602   if (S)
603     reinterpret_cast<OutputSection<ELFT> *>(S)->sortInitFini();
604 }
605 
606 // Sort input sections by the special rule for .ctors and .dtors.
607 template <class ELFT> static void sortCtorsDtors(OutputSectionBase<ELFT> *S) {
608   if (S)
609     reinterpret_cast<OutputSection<ELFT> *>(S)->sortCtorsDtors();
610 }
611 
612 template <class ELFT>
613 void Writer<ELFT>::forEachRelSec(
614     std::function<void(InputSectionBase<ELFT> &, const typename ELFT::Shdr &)>
615         Fn) {
616   for (const std::unique_ptr<elf::ObjectFile<ELFT>> &F :
617        Symtab.getObjectFiles()) {
618     for (InputSectionBase<ELFT> *C : F->getSections()) {
619       if (isDiscarded(C))
620         continue;
621       // Scan all relocations. Each relocation goes through a series
622       // of tests to determine if it needs special treatment, such as
623       // creating GOT, PLT, copy relocations, etc.
624       // Note that relocations for non-alloc sections are directly
625       // processed by InputSection::relocateNonAlloc.
626       if (!(C->getSectionHdr()->sh_flags & SHF_ALLOC))
627         continue;
628       if (auto *S = dyn_cast<InputSection<ELFT>>(C)) {
629         for (const Elf_Shdr *RelSec : S->RelocSections)
630           Fn(*S, *RelSec);
631         continue;
632       }
633       if (auto *S = dyn_cast<EhInputSection<ELFT>>(C))
634         if (S->RelocSection)
635           Fn(*S, *S->RelocSection);
636     }
637   }
638 }
639 
640 template <class ELFT> void Writer<ELFT>::createSections() {
641   for (const std::unique_ptr<elf::ObjectFile<ELFT>> &F :
642        Symtab.getObjectFiles()) {
643     for (InputSectionBase<ELFT> *C : F->getSections()) {
644       if (isDiscarded(C)) {
645         reportDiscarded(C);
646         continue;
647       }
648       OutputSectionBase<ELFT> *Sec;
649       bool IsNew;
650       std::tie(Sec, IsNew) = Factory.create(C, getOutputSectionName(C));
651       if (IsNew)
652         OutputSections.push_back(Sec);
653       Sec->addSection(C);
654     }
655   }
656 }
657 
658 // Create output section objects and add them to OutputSections.
659 template <class ELFT> void Writer<ELFT>::finalizeSections() {
660   // Create output sections for input object file sections.
661   std::vector<OutputSectionBase<ELFT> *> RegularSections = OutputSections;
662 
663   // If we have a .opd section (used under PPC64 for function descriptors),
664   // store a pointer to it here so that we can use it later when processing
665   // relocations.
666   Out<ELFT>::Opd = Factory.lookup(".opd", SHT_PROGBITS, SHF_WRITE | SHF_ALLOC);
667 
668   Out<ELFT>::Dynamic->PreInitArraySec = Factory.lookup(
669       ".preinit_array", SHT_PREINIT_ARRAY, SHF_WRITE | SHF_ALLOC);
670   Out<ELFT>::Dynamic->InitArraySec =
671       Factory.lookup(".init_array", SHT_INIT_ARRAY, SHF_WRITE | SHF_ALLOC);
672   Out<ELFT>::Dynamic->FiniArraySec =
673       Factory.lookup(".fini_array", SHT_FINI_ARRAY, SHF_WRITE | SHF_ALLOC);
674 
675   // Sort section contents for __attribute__((init_priority(N)).
676   sortInitFini(Out<ELFT>::Dynamic->InitArraySec);
677   sortInitFini(Out<ELFT>::Dynamic->FiniArraySec);
678   sortCtorsDtors(Factory.lookup(".ctors", SHT_PROGBITS, SHF_WRITE | SHF_ALLOC));
679   sortCtorsDtors(Factory.lookup(".dtors", SHT_PROGBITS, SHF_WRITE | SHF_ALLOC));
680 
681   // The linker needs to define SECNAME_start, SECNAME_end and SECNAME_stop
682   // symbols for sections, so that the runtime can get the start and end
683   // addresses of each section by section name. Add such symbols.
684   if (!Config->Relocatable) {
685     addStartEndSymbols();
686     for (OutputSectionBase<ELFT> *Sec : RegularSections)
687       addStartStopSymbols(Sec);
688   }
689 
690   // Add _DYNAMIC symbol. Unlike GNU gold, our _DYNAMIC symbol has no type.
691   // It should be okay as no one seems to care about the type.
692   // Even the author of gold doesn't remember why gold behaves that way.
693   // https://sourceware.org/ml/binutils/2002-03/msg00360.html
694   if (isOutputDynamic<ELFT>())
695     Symtab.addSynthetic("_DYNAMIC", Out<ELFT>::Dynamic, 0);
696 
697   // Define __rel[a]_iplt_{start,end} symbols if needed.
698   addRelIpltSymbols();
699 
700   // Add scripted symbols with zero values now.
701   // Real values will be assigned later
702   Script<ELFT>::X->addScriptedSymbols();
703 
704   if (!Out<ELFT>::EhFrame->empty()) {
705     OutputSections.push_back(Out<ELFT>::EhFrame);
706     Out<ELFT>::EhFrame->finalize();
707   }
708 
709   if (Target->NeedsThunks)
710     forEachRelSec(createThunks<ELFT>);
711 
712   for (OutputSectionBase<ELFT> *Sec : OutputSections)
713     Sec->assignOffsets();
714 
715   // Scan relocations. This must be done after every symbol is declared so that
716   // we can correctly decide if a dynamic relocation is needed.
717   forEachRelSec(scanRelocations<ELFT>);
718 
719   // Now that we have defined all possible symbols including linker-
720   // synthesized ones. Visit all symbols to give the finishing touches.
721   for (Symbol *S : Symtab.getSymbols()) {
722     SymbolBody *Body = S->body();
723 
724     // We only report undefined symbols in regular objects. This means that we
725     // will accept an undefined reference in bitcode if it can be optimized out.
726     if (S->IsUsedInRegularObj && Body->isUndefined() && !S->isWeak())
727       reportUndefined<ELFT>(Symtab, Body);
728 
729     if (!includeInSymtab<ELFT>(*Body))
730       continue;
731     if (Out<ELFT>::SymTab)
732       Out<ELFT>::SymTab->addSymbol(Body);
733 
734     if (isOutputDynamic<ELFT>() && S->includeInDynsym()) {
735       Out<ELFT>::DynSymTab->addSymbol(Body);
736       if (auto *SS = dyn_cast<SharedSymbol<ELFT>>(Body))
737         if (SS->file()->isNeeded())
738           Out<ELFT>::VerNeed->addSymbol(SS);
739     }
740   }
741 
742   // Do not proceed if there was an undefined symbol.
743   if (HasError)
744     return;
745 
746   // If linker script processor hasn't added common symbol section yet,
747   // then add it to .bss now.
748   if (!CommonInputSection<ELFT>::X->OutSec) {
749     Out<ELFT>::Bss->addSection(CommonInputSection<ELFT>::X);
750     Out<ELFT>::Bss->assignOffsets();
751   }
752 
753   // So far we have added sections from input object files.
754   // This function adds linker-created Out<ELFT>::* sections.
755   addPredefinedSections();
756 
757   std::stable_sort(OutputSections.begin(), OutputSections.end(),
758                    compareSections<ELFT>);
759 
760   unsigned I = 1;
761   for (OutputSectionBase<ELFT> *Sec : OutputSections) {
762     Sec->SectionIndex = I++;
763     Sec->setSHName(Out<ELFT>::ShStrTab->addString(Sec->getName()));
764   }
765 
766   // Finalizers fix each section's size.
767   // .dynsym is finalized early since that may fill up .gnu.hash.
768   if (isOutputDynamic<ELFT>())
769     Out<ELFT>::DynSymTab->finalize();
770 
771   // Fill other section headers. The dynamic table is finalized
772   // at the end because some tags like RELSZ depend on result
773   // of finalizing other sections. The dynamic string table is
774   // finalized once the .dynamic finalizer has added a few last
775   // strings. See DynamicSection::finalize()
776   for (OutputSectionBase<ELFT> *Sec : OutputSections)
777     if (Sec != Out<ELFT>::DynStrTab && Sec != Out<ELFT>::Dynamic)
778       Sec->finalize();
779 
780   if (isOutputDynamic<ELFT>())
781     Out<ELFT>::Dynamic->finalize();
782 
783   // Now that all output offsets are fixed. Finalize mergeable sections
784   // to fix their maps from input offsets to output offsets.
785   for (OutputSectionBase<ELFT> *Sec : OutputSections)
786     Sec->finalizePieces();
787 }
788 
789 template <class ELFT> bool Writer<ELFT>::needsGot() {
790   if (!Out<ELFT>::Got->empty())
791     return true;
792 
793   // We add the .got section to the result for dynamic MIPS target because
794   // its address and properties are mentioned in the .dynamic section.
795   if (Config->EMachine == EM_MIPS && !Config->Relocatable)
796     return true;
797 
798   // If we have a relocation that is relative to GOT (such as GOTOFFREL),
799   // we need to emit a GOT even if it's empty.
800   return Out<ELFT>::Got->HasGotOffRel;
801 }
802 
803 // This function add Out<ELFT>::* sections to OutputSections.
804 template <class ELFT> void Writer<ELFT>::addPredefinedSections() {
805   auto Add = [&](OutputSectionBase<ELFT> *C) {
806     if (C)
807       OutputSections.push_back(C);
808   };
809 
810   // A core file does not usually contain unmodified segments except
811   // the first page of the executable. Add the build ID section to beginning of
812   // the file so that the section is included in the first page.
813   if (Out<ELFT>::BuildId)
814     OutputSections.insert(OutputSections.begin(), Out<ELFT>::BuildId);
815 
816   // Add .interp at first because some loaders want to see that section
817   // on the first page of the executable file when loaded into memory.
818   if (Out<ELFT>::Interp)
819     OutputSections.insert(OutputSections.begin(), Out<ELFT>::Interp);
820 
821   // This order is not the same as the final output order
822   // because we sort the sections using their attributes below.
823   Add(Out<ELFT>::SymTab);
824   Add(Out<ELFT>::ShStrTab);
825   Add(Out<ELFT>::StrTab);
826   if (isOutputDynamic<ELFT>()) {
827     Add(Out<ELFT>::DynSymTab);
828 
829     bool HasVerNeed = Out<ELFT>::VerNeed->getNeedNum() != 0;
830     if (Out<ELFT>::VerDef || HasVerNeed)
831       Add(Out<ELFT>::VerSym);
832     Add(Out<ELFT>::VerDef);
833     if (HasVerNeed)
834       Add(Out<ELFT>::VerNeed);
835 
836     Add(Out<ELFT>::GnuHashTab);
837     Add(Out<ELFT>::HashTab);
838     Add(Out<ELFT>::Dynamic);
839     Add(Out<ELFT>::DynStrTab);
840     if (Out<ELFT>::RelaDyn->hasRelocs())
841       Add(Out<ELFT>::RelaDyn);
842     Add(Out<ELFT>::MipsRldMap);
843   }
844 
845   // We always need to add rel[a].plt to output if it has entries.
846   // Even during static linking it can contain R_[*]_IRELATIVE relocations.
847   if (Out<ELFT>::RelaPlt && Out<ELFT>::RelaPlt->hasRelocs()) {
848     Add(Out<ELFT>::RelaPlt);
849     Out<ELFT>::RelaPlt->Static = !isOutputDynamic<ELFT>();
850   }
851 
852   if (needsGot())
853     Add(Out<ELFT>::Got);
854   if (Out<ELFT>::GotPlt && !Out<ELFT>::GotPlt->empty())
855     Add(Out<ELFT>::GotPlt);
856   if (!Out<ELFT>::Plt->empty())
857     Add(Out<ELFT>::Plt);
858   if (!Out<ELFT>::EhFrame->empty())
859     Add(Out<ELFT>::EhFrameHdr);
860   if (Out<ELFT>::Bss->getSize() > 0)
861     Add(Out<ELFT>::Bss);
862 }
863 
864 // The linker is expected to define SECNAME_start and SECNAME_end
865 // symbols for a few sections. This function defines them.
866 template <class ELFT> void Writer<ELFT>::addStartEndSymbols() {
867   auto Define = [&](StringRef Start, StringRef End,
868                     OutputSectionBase<ELFT> *OS) {
869     if (OS) {
870       this->Symtab.addSynthetic(Start, OS, 0);
871       this->Symtab.addSynthetic(End, OS, DefinedSynthetic<ELFT>::SectionEnd);
872     } else {
873       addOptionalSynthetic(this->Symtab, Start,
874                            (OutputSectionBase<ELFT> *)nullptr, 0);
875       addOptionalSynthetic(this->Symtab, End,
876                            (OutputSectionBase<ELFT> *)nullptr, 0);
877     }
878   };
879 
880   Define("__preinit_array_start", "__preinit_array_end",
881          Out<ELFT>::Dynamic->PreInitArraySec);
882   Define("__init_array_start", "__init_array_end",
883          Out<ELFT>::Dynamic->InitArraySec);
884   Define("__fini_array_start", "__fini_array_end",
885          Out<ELFT>::Dynamic->FiniArraySec);
886 }
887 
888 // If a section name is valid as a C identifier (which is rare because of
889 // the leading '.'), linkers are expected to define __start_<secname> and
890 // __stop_<secname> symbols. They are at beginning and end of the section,
891 // respectively. This is not requested by the ELF standard, but GNU ld and
892 // gold provide the feature, and used by many programs.
893 template <class ELFT>
894 void Writer<ELFT>::addStartStopSymbols(OutputSectionBase<ELFT> *Sec) {
895   StringRef S = Sec->getName();
896   if (!isValidCIdentifier(S))
897     return;
898   StringSaver Saver(Alloc);
899   StringRef Start = Saver.save("__start_" + S);
900   StringRef Stop = Saver.save("__stop_" + S);
901   if (SymbolBody *B = Symtab.find(Start))
902     if (B->isUndefined())
903       Symtab.addSynthetic(Start, Sec, 0);
904   if (SymbolBody *B = Symtab.find(Stop))
905     if (B->isUndefined())
906       Symtab.addSynthetic(Stop, Sec, DefinedSynthetic<ELFT>::SectionEnd);
907 }
908 
909 template <class ELFT> static bool needsPtLoad(OutputSectionBase<ELFT> *Sec) {
910   if (!(Sec->getFlags() & SHF_ALLOC))
911     return false;
912 
913   // Don't allocate VA space for TLS NOBITS sections. The PT_TLS PHDR is
914   // responsible for allocating space for them, not the PT_LOAD that
915   // contains the TLS initialization image.
916   if (Sec->getFlags() & SHF_TLS && Sec->getType() == SHT_NOBITS)
917     return false;
918   return true;
919 }
920 
921 // Decide which program headers to create and which sections to include in each
922 // one.
923 template <class ELFT>
924 std::vector<PhdrEntry<ELFT>> Writer<ELFT>::createPhdrs() {
925   std::vector<Phdr> Ret;
926 
927   auto AddHdr = [&](unsigned Type, unsigned Flags) -> Phdr * {
928     Ret.emplace_back(Type, Flags);
929     return &Ret.back();
930   };
931 
932   // The first phdr entry is PT_PHDR which describes the program header itself.
933   Phdr &Hdr = *AddHdr(PT_PHDR, PF_R);
934   Hdr.add(Out<ELFT>::ProgramHeaders);
935 
936   // PT_INTERP must be the second entry if exists.
937   if (Out<ELFT>::Interp) {
938     Phdr &Hdr = *AddHdr(PT_INTERP, Out<ELFT>::Interp->getPhdrFlags());
939     Hdr.add(Out<ELFT>::Interp);
940   }
941 
942   // Add the first PT_LOAD segment for regular output sections.
943   uintX_t Flags = PF_R;
944   Phdr *Load = AddHdr(PT_LOAD, Flags);
945   Load->add(Out<ELFT>::ElfHeader);
946   Load->add(Out<ELFT>::ProgramHeaders);
947 
948   Phdr TlsHdr(PT_TLS, PF_R);
949   Phdr RelRo(PT_GNU_RELRO, PF_R);
950   Phdr Note(PT_NOTE, PF_R);
951   for (OutputSectionBase<ELFT> *Sec : OutputSections) {
952     if (!(Sec->getFlags() & SHF_ALLOC))
953       break;
954 
955     // If we meet TLS section then we create TLS header
956     // and put all TLS sections inside for futher use when
957     // assign addresses.
958     if (Sec->getFlags() & SHF_TLS)
959       TlsHdr.add(Sec);
960 
961     if (!needsPtLoad(Sec))
962       continue;
963 
964     // If flags changed then we want new load segment.
965     uintX_t NewFlags = Sec->getPhdrFlags();
966     if (Flags != NewFlags) {
967       Load = AddHdr(PT_LOAD, NewFlags);
968       Flags = NewFlags;
969     }
970 
971     Load->add(Sec);
972 
973     if (isRelroSection(Sec))
974       RelRo.add(Sec);
975     if (Sec->getType() == SHT_NOTE)
976       Note.add(Sec);
977   }
978 
979   // Add the TLS segment unless it's empty.
980   if (TlsHdr.First)
981     Ret.push_back(std::move(TlsHdr));
982 
983   // Add an entry for .dynamic.
984   if (isOutputDynamic<ELFT>()) {
985     Phdr &H = *AddHdr(PT_DYNAMIC, Out<ELFT>::Dynamic->getPhdrFlags());
986     H.add(Out<ELFT>::Dynamic);
987   }
988 
989   // PT_GNU_RELRO includes all sections that should be marked as
990   // read-only by dynamic linker after proccessing relocations.
991   if (RelRo.First)
992     Ret.push_back(std::move(RelRo));
993 
994   // PT_GNU_EH_FRAME is a special section pointing on .eh_frame_hdr.
995   if (!Out<ELFT>::EhFrame->empty() && Out<ELFT>::EhFrameHdr) {
996     Phdr &Hdr = *AddHdr(PT_GNU_EH_FRAME, Out<ELFT>::EhFrameHdr->getPhdrFlags());
997     Hdr.add(Out<ELFT>::EhFrameHdr);
998   }
999 
1000   // PT_GNU_STACK is a special section to tell the loader to make the
1001   // pages for the stack non-executable.
1002   if (!Config->ZExecStack)
1003     AddHdr(PT_GNU_STACK, PF_R | PF_W);
1004 
1005   if (Note.First)
1006     Ret.push_back(std::move(Note));
1007   return Ret;
1008 }
1009 
1010 // The first section of each PT_LOAD and the first section after PT_GNU_RELRO
1011 // have to be page aligned so that the dynamic linker can set the permissions.
1012 template <class ELFT> void Writer<ELFT>::fixSectionAlignments() {
1013   for (const Phdr &P : Phdrs)
1014     if (P.H.p_type == PT_LOAD)
1015       P.First->PageAlign = true;
1016 
1017   for (const Phdr &P : Phdrs) {
1018     if (P.H.p_type != PT_GNU_RELRO)
1019       continue;
1020     // Find the first section after PT_GNU_RELRO. If it is in a PT_LOAD we
1021     // have to align it to a page.
1022     auto End = OutputSections.end();
1023     auto I = std::find(OutputSections.begin(), End, P.Last);
1024     if (I == End || (I + 1) == End)
1025       continue;
1026     OutputSectionBase<ELFT> *Sec = *(I + 1);
1027     if (needsPtLoad(Sec))
1028       Sec->PageAlign = true;
1029   }
1030 }
1031 
1032 // We should set file offsets and VAs for elf header and program headers
1033 // sections. These are special, we do not include them into output sections
1034 // list, but have them to simplify the code.
1035 template <class ELFT> void Writer<ELFT>::fixHeaders() {
1036   uintX_t BaseVA = ScriptConfig->HasContents ? 0 : Config->ImageBase;
1037   Out<ELFT>::ElfHeader->setVA(BaseVA);
1038   uintX_t Off = Out<ELFT>::ElfHeader->getSize();
1039   Out<ELFT>::ProgramHeaders->setVA(Off + BaseVA);
1040   Out<ELFT>::ProgramHeaders->setSize(sizeof(Elf_Phdr) * Phdrs.size());
1041 }
1042 
1043 // Assign VAs (addresses at run-time) to output sections.
1044 template <class ELFT> void Writer<ELFT>::assignAddresses() {
1045   uintX_t VA = Config->ImageBase + Out<ELFT>::ElfHeader->getSize() +
1046                Out<ELFT>::ProgramHeaders->getSize();
1047 
1048   uintX_t ThreadBssOffset = 0;
1049   for (OutputSectionBase<ELFT> *Sec : OutputSections) {
1050     uintX_t Alignment = Sec->getAlignment();
1051     if (Sec->PageAlign)
1052       Alignment = std::max<uintX_t>(Alignment, Target->PageSize);
1053 
1054     // We only assign VAs to allocated sections.
1055     if (needsPtLoad(Sec)) {
1056       VA = alignTo(VA, Alignment);
1057       Sec->setVA(VA);
1058       VA += Sec->getSize();
1059     } else if (Sec->getFlags() & SHF_TLS && Sec->getType() == SHT_NOBITS) {
1060       uintX_t TVA = VA + ThreadBssOffset;
1061       TVA = alignTo(TVA, Alignment);
1062       Sec->setVA(TVA);
1063       ThreadBssOffset = TVA - VA + Sec->getSize();
1064     }
1065   }
1066 }
1067 
1068 // Adjusts the file alignment for a given output section and returns
1069 // its new file offset. The file offset must be the same with its
1070 // virtual address (modulo the page size) so that the loader can load
1071 // executables without any address adjustment.
1072 template <class ELFT, class uintX_t>
1073 static uintX_t getFileAlignment(uintX_t Off, OutputSectionBase<ELFT> *Sec) {
1074   uintX_t Alignment = Sec->getAlignment();
1075   if (Sec->PageAlign)
1076     Alignment = std::max<uintX_t>(Alignment, Target->PageSize);
1077   Off = alignTo(Off, Alignment);
1078 
1079   // Relocatable output does not have program headers
1080   // and does not need any other offset adjusting.
1081   if (Config->Relocatable || !(Sec->getFlags() & SHF_ALLOC))
1082     return Off;
1083   return alignTo(Off, Target->PageSize, Sec->getVA());
1084 }
1085 
1086 // Assign file offsets to output sections.
1087 template <class ELFT> void Writer<ELFT>::assignFileOffsets() {
1088   uintX_t Off = 0;
1089 
1090   auto Set = [&](OutputSectionBase<ELFT> *Sec) {
1091     if (Sec->getType() == SHT_NOBITS) {
1092       Sec->setFileOffset(Off);
1093       return;
1094     }
1095 
1096     Off = getFileAlignment<ELFT>(Off, Sec);
1097     Sec->setFileOffset(Off);
1098     Off += Sec->getSize();
1099   };
1100 
1101   Set(Out<ELFT>::ElfHeader);
1102   Set(Out<ELFT>::ProgramHeaders);
1103   for (OutputSectionBase<ELFT> *Sec : OutputSections)
1104     Set(Sec);
1105 
1106   SectionHeaderOff = alignTo(Off, sizeof(uintX_t));
1107   FileSize = SectionHeaderOff + (OutputSections.size() + 1) * sizeof(Elf_Shdr);
1108 }
1109 
1110 // Finalize the program headers. We call this function after we assign
1111 // file offsets and VAs to all sections.
1112 template <class ELFT> void Writer<ELFT>::setPhdrs() {
1113   for (Phdr &P : Phdrs) {
1114     Elf_Phdr &H = P.H;
1115     OutputSectionBase<ELFT> *First = P.First;
1116     OutputSectionBase<ELFT> *Last = P.Last;
1117     if (First) {
1118       H.p_filesz = Last->getFileOff() - First->getFileOff();
1119       if (Last->getType() != SHT_NOBITS)
1120         H.p_filesz += Last->getSize();
1121       H.p_memsz = Last->getVA() + Last->getSize() - First->getVA();
1122       H.p_offset = First->getFileOff();
1123       H.p_vaddr = First->getVA();
1124     }
1125     if (H.p_type == PT_LOAD)
1126       H.p_align = Target->PageSize;
1127     else if (H.p_type == PT_GNU_RELRO)
1128       H.p_align = 1;
1129     H.p_paddr = H.p_vaddr;
1130 
1131     // The TLS pointer goes after PT_TLS. At least glibc will align it,
1132     // so round up the size to make sure the offsets are correct.
1133     if (H.p_type == PT_TLS) {
1134       Out<ELFT>::TlsPhdr = &H;
1135       if (H.p_memsz)
1136         H.p_memsz = alignTo(H.p_memsz, H.p_align);
1137     }
1138   }
1139 }
1140 
1141 template <class ELFT> static uint32_t getMipsEFlags() {
1142   // FIXME: ELF flags depends on ELF flags of all input object files and
1143   // selected emulation. For now pick the arch flag from the fisrt input file
1144   // and use hard coded values for other flags.
1145   uint32_t FirstElfFlags =
1146       cast<ELFFileBase<ELFT>>(Config->FirstElf)->getObj().getHeader()->e_flags;
1147   uint32_t ElfFlags = FirstElfFlags & EF_MIPS_ARCH;
1148   if (ELFT::Is64Bits)
1149     ElfFlags |= EF_MIPS_CPIC | EF_MIPS_PIC;
1150   else {
1151     ElfFlags |= EF_MIPS_CPIC | EF_MIPS_ABI_O32;
1152     if (Config->Shared)
1153       ElfFlags |= EF_MIPS_PIC;
1154   }
1155   return ElfFlags;
1156 }
1157 
1158 template <class ELFT> static typename ELFT::uint getEntryAddr() {
1159   if (Symbol *S = Config->EntrySym)
1160     return S->body()->getVA<ELFT>();
1161   if (Config->EntryAddr != uint64_t(-1))
1162     return Config->EntryAddr;
1163   return 0;
1164 }
1165 
1166 template <class ELFT> static uint8_t getELFEncoding() {
1167   if (ELFT::TargetEndianness == llvm::support::little)
1168     return ELFDATA2LSB;
1169   return ELFDATA2MSB;
1170 }
1171 
1172 static uint16_t getELFType() {
1173   if (Config->Pic)
1174     return ET_DYN;
1175   if (Config->Relocatable)
1176     return ET_REL;
1177   return ET_EXEC;
1178 }
1179 
1180 // This function is called after we have assigned address and size
1181 // to each section. This function fixes some predefined absolute
1182 // symbol values that depend on section address and size.
1183 template <class ELFT> void Writer<ELFT>::fixAbsoluteSymbols() {
1184   auto Set = [](DefinedRegular<ELFT> *S1, DefinedRegular<ELFT> *S2, uintX_t V) {
1185     if (S1)
1186       S1->Value = V;
1187     if (S2)
1188       S2->Value = V;
1189   };
1190 
1191   // _etext is the first location after the last read-only loadable segment.
1192   // _edata is the first location after the last read-write loadable segment.
1193   // _end is the first location after the uninitialized data region.
1194   for (Phdr &P : Phdrs) {
1195     Elf_Phdr &H = P.H;
1196     if (H.p_type != PT_LOAD)
1197       continue;
1198     Set(ElfSym<ELFT>::End, ElfSym<ELFT>::End2, H.p_vaddr + H.p_memsz);
1199 
1200     uintX_t Val = H.p_vaddr + H.p_filesz;
1201     if (H.p_flags & PF_W)
1202       Set(ElfSym<ELFT>::Edata, ElfSym<ELFT>::Edata2, Val);
1203     else
1204       Set(ElfSym<ELFT>::Etext, ElfSym<ELFT>::Etext2, Val);
1205   }
1206 }
1207 
1208 template <class ELFT> void Writer<ELFT>::writeHeader() {
1209   uint8_t *Buf = Buffer->getBufferStart();
1210   memcpy(Buf, "\177ELF", 4);
1211 
1212   auto &FirstObj = cast<ELFFileBase<ELFT>>(*Config->FirstElf);
1213 
1214   // Write the ELF header.
1215   auto *EHdr = reinterpret_cast<Elf_Ehdr *>(Buf);
1216   EHdr->e_ident[EI_CLASS] = ELFT::Is64Bits ? ELFCLASS64 : ELFCLASS32;
1217   EHdr->e_ident[EI_DATA] = getELFEncoding<ELFT>();
1218   EHdr->e_ident[EI_VERSION] = EV_CURRENT;
1219   EHdr->e_ident[EI_OSABI] = FirstObj.getOSABI();
1220   EHdr->e_type = getELFType();
1221   EHdr->e_machine = FirstObj.EMachine;
1222   EHdr->e_version = EV_CURRENT;
1223   EHdr->e_entry = getEntryAddr<ELFT>();
1224   EHdr->e_shoff = SectionHeaderOff;
1225   EHdr->e_ehsize = sizeof(Elf_Ehdr);
1226   EHdr->e_phnum = Phdrs.size();
1227   EHdr->e_shentsize = sizeof(Elf_Shdr);
1228   EHdr->e_shnum = OutputSections.size() + 1;
1229   EHdr->e_shstrndx = Out<ELFT>::ShStrTab->SectionIndex;
1230 
1231   if (Config->EMachine == EM_MIPS)
1232     EHdr->e_flags = getMipsEFlags<ELFT>();
1233 
1234   if (!Config->Relocatable) {
1235     EHdr->e_phoff = sizeof(Elf_Ehdr);
1236     EHdr->e_phentsize = sizeof(Elf_Phdr);
1237   }
1238 
1239   // Write the program header table.
1240   auto *HBuf = reinterpret_cast<Elf_Phdr *>(Buf + EHdr->e_phoff);
1241   for (Phdr &P : Phdrs)
1242     *HBuf++ = P.H;
1243 
1244   // Write the section header table. Note that the first table entry is null.
1245   auto *SHdrs = reinterpret_cast<Elf_Shdr *>(Buf + EHdr->e_shoff);
1246   for (OutputSectionBase<ELFT> *Sec : OutputSections)
1247     Sec->writeHeaderTo(++SHdrs);
1248 }
1249 
1250 template <class ELFT> void Writer<ELFT>::openFile() {
1251   ErrorOr<std::unique_ptr<FileOutputBuffer>> BufferOrErr =
1252       FileOutputBuffer::create(Config->OutputFile, FileSize,
1253                                FileOutputBuffer::F_executable);
1254   if (auto EC = BufferOrErr.getError())
1255     error(EC, "failed to open " + Config->OutputFile);
1256   else
1257     Buffer = std::move(*BufferOrErr);
1258 }
1259 
1260 // Write section contents to a mmap'ed file.
1261 template <class ELFT> void Writer<ELFT>::writeSections() {
1262   uint8_t *Buf = Buffer->getBufferStart();
1263 
1264   // PPC64 needs to process relocations in the .opd section before processing
1265   // relocations in code-containing sections.
1266   if (OutputSectionBase<ELFT> *Sec = Out<ELFT>::Opd) {
1267     Out<ELFT>::OpdBuf = Buf + Sec->getFileOff();
1268     Sec->writeTo(Buf + Sec->getFileOff());
1269   }
1270 
1271   for (OutputSectionBase<ELFT> *Sec : OutputSections)
1272     if (Sec != Out<ELFT>::Opd)
1273       Sec->writeTo(Buf + Sec->getFileOff());
1274 }
1275 
1276 template <class ELFT> void Writer<ELFT>::writeBuildId() {
1277   if (!Out<ELFT>::BuildId)
1278     return;
1279 
1280   // Compute a hash of all sections except .debug_* sections.
1281   // We skip debug sections because they tend to be very large
1282   // and their contents are very likely to be the same as long as
1283   // other sections are the same.
1284   uint8_t *Start = Buffer->getBufferStart();
1285   uint8_t *Last = Start;
1286   std::vector<ArrayRef<uint8_t>> Regions;
1287   for (OutputSectionBase<ELFT> *Sec : OutputSections) {
1288     uint8_t *End = Start + Sec->getFileOff();
1289     if (!Sec->getName().startswith(".debug_"))
1290       Regions.push_back({Last, End});
1291     Last = End;
1292   }
1293   Regions.push_back({Last, Start + FileSize});
1294   Out<ELFT>::BuildId->writeBuildId(Regions);
1295 }
1296 
1297 template void elf::writeResult<ELF32LE>(SymbolTable<ELF32LE> *Symtab);
1298 template void elf::writeResult<ELF32BE>(SymbolTable<ELF32BE> *Symtab);
1299 template void elf::writeResult<ELF64LE>(SymbolTable<ELF64LE> *Symtab);
1300 template void elf::writeResult<ELF64BE>(SymbolTable<ELF64BE> *Symtab);
1301 
1302 template struct elf::PhdrEntry<ELF32LE>;
1303 template struct elf::PhdrEntry<ELF32BE>;
1304 template struct elf::PhdrEntry<ELF64LE>;
1305 template struct elf::PhdrEntry<ELF64BE>;
1306 
1307 template bool elf::isOutputDynamic<ELF32LE>();
1308 template bool elf::isOutputDynamic<ELF32BE>();
1309 template bool elf::isOutputDynamic<ELF64LE>();
1310 template bool elf::isOutputDynamic<ELF64BE>();
1311 
1312 template bool elf::isRelroSection<ELF32LE>(OutputSectionBase<ELF32LE> *);
1313 template bool elf::isRelroSection<ELF32BE>(OutputSectionBase<ELF32BE> *);
1314 template bool elf::isRelroSection<ELF64LE>(OutputSectionBase<ELF64LE> *);
1315 template bool elf::isRelroSection<ELF64BE>(OutputSectionBase<ELF64BE> *);
1316 
1317 template StringRef elf::getOutputSectionName<ELF32LE>(InputSectionBase<ELF32LE> *);
1318 template StringRef elf::getOutputSectionName<ELF32BE>(InputSectionBase<ELF32BE> *);
1319 template StringRef elf::getOutputSectionName<ELF64LE>(InputSectionBase<ELF64LE> *);
1320 template StringRef elf::getOutputSectionName<ELF64BE>(InputSectionBase<ELF64BE> *);
1321 
1322 template void elf::reportDiscarded<ELF32LE>(InputSectionBase<ELF32LE> *);
1323 template void elf::reportDiscarded<ELF32BE>(InputSectionBase<ELF32BE> *);
1324 template void elf::reportDiscarded<ELF64LE>(InputSectionBase<ELF64LE> *);
1325 template void elf::reportDiscarded<ELF64BE>(InputSectionBase<ELF64BE> *);
1326