1 //===- LinkerScript.cpp ---------------------------------------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 //
9 // This file contains the parser/evaluator of the linker script.
10 //
11 //===----------------------------------------------------------------------===//
12 
13 #include "LinkerScript.h"
14 #include "Config.h"
15 #include "InputSection.h"
16 #include "OutputSections.h"
17 #include "SymbolTable.h"
18 #include "Symbols.h"
19 #include "SyntheticSections.h"
20 #include "Target.h"
21 #include "Writer.h"
22 #include "lld/Common/Memory.h"
23 #include "lld/Common/Strings.h"
24 #include "llvm/ADT/STLExtras.h"
25 #include "llvm/ADT/StringRef.h"
26 #include "llvm/BinaryFormat/ELF.h"
27 #include "llvm/Support/Casting.h"
28 #include "llvm/Support/Endian.h"
29 #include "llvm/Support/ErrorHandling.h"
30 #include "llvm/Support/FileSystem.h"
31 #include "llvm/Support/Parallel.h"
32 #include "llvm/Support/Path.h"
33 #include "llvm/Support/TimeProfiler.h"
34 #include <algorithm>
35 #include <cassert>
36 #include <cstddef>
37 #include <cstdint>
38 #include <iterator>
39 #include <limits>
40 #include <string>
41 #include <vector>
42 
43 using namespace llvm;
44 using namespace llvm::ELF;
45 using namespace llvm::object;
46 using namespace llvm::support::endian;
47 using namespace lld;
48 using namespace lld::elf;
49 
50 std::unique_ptr<LinkerScript> elf::script;
51 
52 static bool isSectionPrefix(StringRef prefix, StringRef name) {
53   return name.consume_front(prefix) && (name.empty() || name[0] == '.');
54 }
55 
56 static StringRef getOutputSectionName(const InputSectionBase *s) {
57   if (config->relocatable)
58     return s->name;
59 
60   // This is for --emit-relocs. If .text.foo is emitted as .text.bar, we want
61   // to emit .rela.text.foo as .rela.text.bar for consistency (this is not
62   // technically required, but not doing it is odd). This code guarantees that.
63   if (auto *isec = dyn_cast<InputSection>(s)) {
64     if (InputSectionBase *rel = isec->getRelocatedSection()) {
65       OutputSection *out = rel->getOutputSection();
66       if (s->type == SHT_RELA)
67         return saver.save(".rela" + out->name);
68       return saver.save(".rel" + out->name);
69     }
70   }
71 
72   // A BssSection created for a common symbol is identified as "COMMON" in
73   // linker scripts. It should go to .bss section.
74   if (s->name == "COMMON")
75     return ".bss";
76 
77   if (script->hasSectionsCommand)
78     return s->name;
79 
80   // When no SECTIONS is specified, emulate GNU ld's internal linker scripts
81   // by grouping sections with certain prefixes.
82 
83   // GNU ld places text sections with prefix ".text.hot.", ".text.unknown.",
84   // ".text.unlikely.", ".text.startup." or ".text.exit." before others.
85   // We provide an option -z keep-text-section-prefix to group such sections
86   // into separate output sections. This is more flexible. See also
87   // sortISDBySectionOrder().
88   // ".text.unknown" means the hotness of the section is unknown. When
89   // SampleFDO is used, if a function doesn't have sample, it could be very
90   // cold or it could be a new function never being sampled. Those functions
91   // will be kept in the ".text.unknown" section.
92   // ".text.split." holds symbols which are split out from functions in other
93   // input sections. For example, with -fsplit-machine-functions, placing the
94   // cold parts in .text.split instead of .text.unlikely mitigates against poor
95   // profile inaccuracy. Techniques such as hugepage remapping can make
96   // conservative decisions at the section granularity.
97   if (isSectionPrefix(".text", s->name)) {
98     if (config->zKeepTextSectionPrefix)
99       for (StringRef v : {".text.hot", ".text.unknown", ".text.unlikely",
100                           ".text.startup", ".text.exit", ".text.split"})
101         if (isSectionPrefix(v.substr(5), s->name.substr(5)))
102           return v;
103     return ".text";
104   }
105 
106   for (StringRef v :
107        {".data.rel.ro", ".data", ".rodata", ".bss.rel.ro", ".bss",
108         ".gcc_except_table", ".init_array", ".fini_array", ".tbss", ".tdata",
109         ".ARM.exidx", ".ARM.extab", ".ctors", ".dtors"})
110     if (isSectionPrefix(v, s->name))
111       return v;
112 
113   return s->name;
114 }
115 
116 uint64_t ExprValue::getValue() const {
117   if (sec)
118     return alignTo(sec->getOutputSection()->addr + sec->getOffset(val),
119                    alignment);
120   return alignTo(val, alignment);
121 }
122 
123 uint64_t ExprValue::getSecAddr() const {
124   return sec ? sec->getOutputSection()->addr + sec->getOffset(0) : 0;
125 }
126 
127 uint64_t ExprValue::getSectionOffset() const {
128   // If the alignment is trivial, we don't have to compute the full
129   // value to know the offset. This allows this function to succeed in
130   // cases where the output section is not yet known.
131   if (alignment == 1 && !sec)
132     return val;
133   return getValue() - getSecAddr();
134 }
135 
136 OutputSection *LinkerScript::createOutputSection(StringRef name,
137                                                  StringRef location) {
138   OutputSection *&secRef = nameToOutputSection[CachedHashStringRef(name)];
139   OutputSection *sec;
140   if (secRef && secRef->location.empty()) {
141     // There was a forward reference.
142     sec = secRef;
143   } else {
144     sec = make<OutputSection>(name, SHT_PROGBITS, 0);
145     if (!secRef)
146       secRef = sec;
147   }
148   sec->location = std::string(location);
149   return sec;
150 }
151 
152 OutputSection *LinkerScript::getOrCreateOutputSection(StringRef name) {
153   OutputSection *&cmdRef = nameToOutputSection[CachedHashStringRef(name)];
154   if (!cmdRef)
155     cmdRef = make<OutputSection>(name, SHT_PROGBITS, 0);
156   return cmdRef;
157 }
158 
159 // Expands the memory region by the specified size.
160 static void expandMemoryRegion(MemoryRegion *memRegion, uint64_t size,
161                                StringRef secName) {
162   memRegion->curPos += size;
163   uint64_t newSize = memRegion->curPos - (memRegion->origin)().getValue();
164   uint64_t length = (memRegion->length)().getValue();
165   if (newSize > length)
166     error("section '" + secName + "' will not fit in region '" +
167           memRegion->name + "': overflowed by " + Twine(newSize - length) +
168           " bytes");
169 }
170 
171 void LinkerScript::expandMemoryRegions(uint64_t size) {
172   if (ctx->memRegion)
173     expandMemoryRegion(ctx->memRegion, size, ctx->outSec->name);
174   // Only expand the LMARegion if it is different from memRegion.
175   if (ctx->lmaRegion && ctx->memRegion != ctx->lmaRegion)
176     expandMemoryRegion(ctx->lmaRegion, size, ctx->outSec->name);
177 }
178 
179 void LinkerScript::expandOutputSection(uint64_t size) {
180   ctx->outSec->size += size;
181   expandMemoryRegions(size);
182 }
183 
184 void LinkerScript::setDot(Expr e, const Twine &loc, bool inSec) {
185   uint64_t val = e().getValue();
186   if (val < dot && inSec)
187     error(loc + ": unable to move location counter backward for: " +
188           ctx->outSec->name);
189 
190   // Update to location counter means update to section size.
191   if (inSec)
192     expandOutputSection(val - dot);
193 
194   dot = val;
195 }
196 
197 // Used for handling linker symbol assignments, for both finalizing
198 // their values and doing early declarations. Returns true if symbol
199 // should be defined from linker script.
200 static bool shouldDefineSym(SymbolAssignment *cmd) {
201   if (cmd->name == ".")
202     return false;
203 
204   if (!cmd->provide)
205     return true;
206 
207   // If a symbol was in PROVIDE(), we need to define it only
208   // when it is a referenced undefined symbol.
209   Symbol *b = symtab->find(cmd->name);
210   if (b && !b->isDefined())
211     return true;
212   return false;
213 }
214 
215 // Called by processSymbolAssignments() to assign definitions to
216 // linker-script-defined symbols.
217 void LinkerScript::addSymbol(SymbolAssignment *cmd) {
218   if (!shouldDefineSym(cmd))
219     return;
220 
221   // Define a symbol.
222   ExprValue value = cmd->expression();
223   SectionBase *sec = value.isAbsolute() ? nullptr : value.sec;
224   uint8_t visibility = cmd->hidden ? STV_HIDDEN : STV_DEFAULT;
225 
226   // When this function is called, section addresses have not been
227   // fixed yet. So, we may or may not know the value of the RHS
228   // expression.
229   //
230   // For example, if an expression is `x = 42`, we know x is always 42.
231   // However, if an expression is `x = .`, there's no way to know its
232   // value at the moment.
233   //
234   // We want to set symbol values early if we can. This allows us to
235   // use symbols as variables in linker scripts. Doing so allows us to
236   // write expressions like this: `alignment = 16; . = ALIGN(., alignment)`.
237   uint64_t symValue = value.sec ? 0 : value.getValue();
238 
239   Defined newSym(nullptr, cmd->name, STB_GLOBAL, visibility, value.type,
240                  symValue, 0, sec);
241 
242   Symbol *sym = symtab->insert(cmd->name);
243   sym->mergeProperties(newSym);
244   sym->replace(newSym);
245   cmd->sym = cast<Defined>(sym);
246 }
247 
248 // This function is called from LinkerScript::declareSymbols.
249 // It creates a placeholder symbol if needed.
250 static void declareSymbol(SymbolAssignment *cmd) {
251   if (!shouldDefineSym(cmd))
252     return;
253 
254   uint8_t visibility = cmd->hidden ? STV_HIDDEN : STV_DEFAULT;
255   Defined newSym(nullptr, cmd->name, STB_GLOBAL, visibility, STT_NOTYPE, 0, 0,
256                  nullptr);
257 
258   // We can't calculate final value right now.
259   Symbol *sym = symtab->insert(cmd->name);
260   sym->mergeProperties(newSym);
261   sym->replace(newSym);
262 
263   cmd->sym = cast<Defined>(sym);
264   cmd->provide = false;
265   sym->scriptDefined = true;
266 }
267 
268 using SymbolAssignmentMap =
269     DenseMap<const Defined *, std::pair<SectionBase *, uint64_t>>;
270 
271 // Collect section/value pairs of linker-script-defined symbols. This is used to
272 // check whether symbol values converge.
273 static SymbolAssignmentMap
274 getSymbolAssignmentValues(ArrayRef<SectionCommand *> sectionCommands) {
275   SymbolAssignmentMap ret;
276   for (SectionCommand *cmd : sectionCommands) {
277     if (auto *assign = dyn_cast<SymbolAssignment>(cmd)) {
278       if (assign->sym) // sym is nullptr for dot.
279         ret.try_emplace(assign->sym, std::make_pair(assign->sym->section,
280                                                     assign->sym->value));
281       continue;
282     }
283     for (SectionCommand *subCmd : cast<OutputSection>(cmd)->commands)
284       if (auto *assign = dyn_cast<SymbolAssignment>(subCmd))
285         if (assign->sym)
286           ret.try_emplace(assign->sym, std::make_pair(assign->sym->section,
287                                                       assign->sym->value));
288   }
289   return ret;
290 }
291 
292 // Returns the lexicographical smallest (for determinism) Defined whose
293 // section/value has changed.
294 static const Defined *
295 getChangedSymbolAssignment(const SymbolAssignmentMap &oldValues) {
296   const Defined *changed = nullptr;
297   for (auto &it : oldValues) {
298     const Defined *sym = it.first;
299     if (std::make_pair(sym->section, sym->value) != it.second &&
300         (!changed || sym->getName() < changed->getName()))
301       changed = sym;
302   }
303   return changed;
304 }
305 
306 // Process INSERT [AFTER|BEFORE] commands. For each command, we move the
307 // specified output section to the designated place.
308 void LinkerScript::processInsertCommands() {
309   SmallVector<OutputSection *, 0> moves;
310   for (const InsertCommand &cmd : insertCommands) {
311     for (StringRef name : cmd.names) {
312       // If base is empty, it may have been discarded by
313       // adjustSectionsBeforeSorting(). We do not handle such output sections.
314       auto from = llvm::find_if(sectionCommands, [&](SectionCommand *subCmd) {
315         return isa<OutputSection>(subCmd) &&
316                cast<OutputSection>(subCmd)->name == name;
317       });
318       if (from == sectionCommands.end())
319         continue;
320       moves.push_back(cast<OutputSection>(*from));
321       sectionCommands.erase(from);
322     }
323 
324     auto insertPos =
325         llvm::find_if(sectionCommands, [&cmd](SectionCommand *subCmd) {
326           auto *to = dyn_cast<OutputSection>(subCmd);
327           return to != nullptr && to->name == cmd.where;
328         });
329     if (insertPos == sectionCommands.end()) {
330       error("unable to insert " + cmd.names[0] +
331             (cmd.isAfter ? " after " : " before ") + cmd.where);
332     } else {
333       if (cmd.isAfter)
334         ++insertPos;
335       sectionCommands.insert(insertPos, moves.begin(), moves.end());
336     }
337     moves.clear();
338   }
339 }
340 
341 // Symbols defined in script should not be inlined by LTO. At the same time
342 // we don't know their final values until late stages of link. Here we scan
343 // over symbol assignment commands and create placeholder symbols if needed.
344 void LinkerScript::declareSymbols() {
345   assert(!ctx);
346   for (SectionCommand *cmd : sectionCommands) {
347     if (auto *assign = dyn_cast<SymbolAssignment>(cmd)) {
348       declareSymbol(assign);
349       continue;
350     }
351 
352     // If the output section directive has constraints,
353     // we can't say for sure if it is going to be included or not.
354     // Skip such sections for now. Improve the checks if we ever
355     // need symbols from that sections to be declared early.
356     auto *sec = cast<OutputSection>(cmd);
357     if (sec->constraint != ConstraintKind::NoConstraint)
358       continue;
359     for (SectionCommand *cmd : sec->commands)
360       if (auto *assign = dyn_cast<SymbolAssignment>(cmd))
361         declareSymbol(assign);
362   }
363 }
364 
365 // This function is called from assignAddresses, while we are
366 // fixing the output section addresses. This function is supposed
367 // to set the final value for a given symbol assignment.
368 void LinkerScript::assignSymbol(SymbolAssignment *cmd, bool inSec) {
369   if (cmd->name == ".") {
370     setDot(cmd->expression, cmd->location, inSec);
371     return;
372   }
373 
374   if (!cmd->sym)
375     return;
376 
377   ExprValue v = cmd->expression();
378   if (v.isAbsolute()) {
379     cmd->sym->section = nullptr;
380     cmd->sym->value = v.getValue();
381   } else {
382     cmd->sym->section = v.sec;
383     cmd->sym->value = v.getSectionOffset();
384   }
385   cmd->sym->type = v.type;
386 }
387 
388 static inline StringRef getFilename(const InputFile *file) {
389   return file ? file->getNameForScript() : StringRef();
390 }
391 
392 bool InputSectionDescription::matchesFile(const InputFile *file) const {
393   if (filePat.isTrivialMatchAll())
394     return true;
395 
396   if (!matchesFileCache || matchesFileCache->first != file)
397     matchesFileCache.emplace(file, filePat.match(getFilename(file)));
398 
399   return matchesFileCache->second;
400 }
401 
402 bool SectionPattern::excludesFile(const InputFile *file) const {
403   if (excludedFilePat.empty())
404     return false;
405 
406   if (!excludesFileCache || excludesFileCache->first != file)
407     excludesFileCache.emplace(file, excludedFilePat.match(getFilename(file)));
408 
409   return excludesFileCache->second;
410 }
411 
412 bool LinkerScript::shouldKeep(InputSectionBase *s) {
413   for (InputSectionDescription *id : keptSections)
414     if (id->matchesFile(s->file))
415       for (SectionPattern &p : id->sectionPatterns)
416         if (p.sectionPat.match(s->name) &&
417             (s->flags & id->withFlags) == id->withFlags &&
418             (s->flags & id->withoutFlags) == 0)
419           return true;
420   return false;
421 }
422 
423 // A helper function for the SORT() command.
424 static bool matchConstraints(ArrayRef<InputSectionBase *> sections,
425                              ConstraintKind kind) {
426   if (kind == ConstraintKind::NoConstraint)
427     return true;
428 
429   bool isRW = llvm::any_of(
430       sections, [](InputSectionBase *sec) { return sec->flags & SHF_WRITE; });
431 
432   return (isRW && kind == ConstraintKind::ReadWrite) ||
433          (!isRW && kind == ConstraintKind::ReadOnly);
434 }
435 
436 static void sortSections(MutableArrayRef<InputSectionBase *> vec,
437                          SortSectionPolicy k) {
438   auto alignmentComparator = [](InputSectionBase *a, InputSectionBase *b) {
439     // ">" is not a mistake. Sections with larger alignments are placed
440     // before sections with smaller alignments in order to reduce the
441     // amount of padding necessary. This is compatible with GNU.
442     return a->alignment > b->alignment;
443   };
444   auto nameComparator = [](InputSectionBase *a, InputSectionBase *b) {
445     return a->name < b->name;
446   };
447   auto priorityComparator = [](InputSectionBase *a, InputSectionBase *b) {
448     return getPriority(a->name) < getPriority(b->name);
449   };
450 
451   switch (k) {
452   case SortSectionPolicy::Default:
453   case SortSectionPolicy::None:
454     return;
455   case SortSectionPolicy::Alignment:
456     return llvm::stable_sort(vec, alignmentComparator);
457   case SortSectionPolicy::Name:
458     return llvm::stable_sort(vec, nameComparator);
459   case SortSectionPolicy::Priority:
460     return llvm::stable_sort(vec, priorityComparator);
461   }
462 }
463 
464 // Sort sections as instructed by SORT-family commands and --sort-section
465 // option. Because SORT-family commands can be nested at most two depth
466 // (e.g. SORT_BY_NAME(SORT_BY_ALIGNMENT(.text.*))) and because the command
467 // line option is respected even if a SORT command is given, the exact
468 // behavior we have here is a bit complicated. Here are the rules.
469 //
470 // 1. If two SORT commands are given, --sort-section is ignored.
471 // 2. If one SORT command is given, and if it is not SORT_NONE,
472 //    --sort-section is handled as an inner SORT command.
473 // 3. If one SORT command is given, and if it is SORT_NONE, don't sort.
474 // 4. If no SORT command is given, sort according to --sort-section.
475 static void sortInputSections(MutableArrayRef<InputSectionBase *> vec,
476                               SortSectionPolicy outer,
477                               SortSectionPolicy inner) {
478   if (outer == SortSectionPolicy::None)
479     return;
480 
481   if (inner == SortSectionPolicy::Default)
482     sortSections(vec, config->sortSection);
483   else
484     sortSections(vec, inner);
485   sortSections(vec, outer);
486 }
487 
488 // Compute and remember which sections the InputSectionDescription matches.
489 SmallVector<InputSectionBase *, 0>
490 LinkerScript::computeInputSections(const InputSectionDescription *cmd,
491                                    ArrayRef<InputSectionBase *> sections) {
492   SmallVector<InputSectionBase *, 0> ret;
493   SmallVector<size_t, 0> indexes;
494   DenseSet<size_t> seen;
495   auto sortByPositionThenCommandLine = [&](size_t begin, size_t end) {
496     llvm::sort(MutableArrayRef<size_t>(indexes).slice(begin, end - begin));
497     for (size_t i = begin; i != end; ++i)
498       ret[i] = sections[indexes[i]];
499     sortInputSections(
500         MutableArrayRef<InputSectionBase *>(ret).slice(begin, end - begin),
501         config->sortSection, SortSectionPolicy::None);
502   };
503 
504   // Collects all sections that satisfy constraints of Cmd.
505   size_t sizeAfterPrevSort = 0;
506   for (const SectionPattern &pat : cmd->sectionPatterns) {
507     size_t sizeBeforeCurrPat = ret.size();
508 
509     for (size_t i = 0, e = sections.size(); i != e; ++i) {
510       // Skip if the section is dead or has been matched by a previous input
511       // section description or a previous pattern.
512       InputSectionBase *sec = sections[i];
513       if (!sec->isLive() || sec->parent || seen.contains(i))
514         continue;
515 
516       // For --emit-relocs we have to ignore entries like
517       //   .rela.dyn : { *(.rela.data) }
518       // which are common because they are in the default bfd script.
519       // We do not ignore SHT_REL[A] linker-synthesized sections here because
520       // want to support scripts that do custom layout for them.
521       if (isa<InputSection>(sec) &&
522           cast<InputSection>(sec)->getRelocatedSection())
523         continue;
524 
525       // Check the name early to improve performance in the common case.
526       if (!pat.sectionPat.match(sec->name))
527         continue;
528 
529       if (!cmd->matchesFile(sec->file) || pat.excludesFile(sec->file) ||
530           (sec->flags & cmd->withFlags) != cmd->withFlags ||
531           (sec->flags & cmd->withoutFlags) != 0)
532         continue;
533 
534       ret.push_back(sec);
535       indexes.push_back(i);
536       seen.insert(i);
537     }
538 
539     if (pat.sortOuter == SortSectionPolicy::Default)
540       continue;
541 
542     // Matched sections are ordered by radix sort with the keys being (SORT*,
543     // --sort-section, input order), where SORT* (if present) is most
544     // significant.
545     //
546     // Matched sections between the previous SORT* and this SORT* are sorted by
547     // (--sort-alignment, input order).
548     sortByPositionThenCommandLine(sizeAfterPrevSort, sizeBeforeCurrPat);
549     // Matched sections by this SORT* pattern are sorted using all 3 keys.
550     // ret[sizeBeforeCurrPat,ret.size()) are already in the input order, so we
551     // just sort by sortOuter and sortInner.
552     sortInputSections(
553         MutableArrayRef<InputSectionBase *>(ret).slice(sizeBeforeCurrPat),
554         pat.sortOuter, pat.sortInner);
555     sizeAfterPrevSort = ret.size();
556   }
557   // Matched sections after the last SORT* are sorted by (--sort-alignment,
558   // input order).
559   sortByPositionThenCommandLine(sizeAfterPrevSort, ret.size());
560   return ret;
561 }
562 
563 void LinkerScript::discard(InputSectionBase &s) {
564   if (&s == in.shStrTab.get())
565     error("discarding " + s.name + " section is not allowed");
566 
567   s.markDead();
568   s.parent = nullptr;
569   for (InputSection *sec : s.dependentSections)
570     discard(*sec);
571 }
572 
573 void LinkerScript::discardSynthetic(OutputSection &outCmd) {
574   for (Partition &part : partitions) {
575     if (!part.armExidx || !part.armExidx->isLive())
576       continue;
577     std::vector<InputSectionBase *> secs(part.armExidx->exidxSections.begin(),
578                                          part.armExidx->exidxSections.end());
579     for (SectionCommand *cmd : outCmd.commands)
580       if (auto *isd = dyn_cast<InputSectionDescription>(cmd))
581         for (InputSectionBase *s : computeInputSections(isd, secs))
582           discard(*s);
583   }
584 }
585 
586 SmallVector<InputSectionBase *, 0>
587 LinkerScript::createInputSectionList(OutputSection &outCmd) {
588   SmallVector<InputSectionBase *, 0> ret;
589 
590   for (SectionCommand *cmd : outCmd.commands) {
591     if (auto *isd = dyn_cast<InputSectionDescription>(cmd)) {
592       isd->sectionBases = computeInputSections(isd, inputSections);
593       for (InputSectionBase *s : isd->sectionBases)
594         s->parent = &outCmd;
595       ret.insert(ret.end(), isd->sectionBases.begin(), isd->sectionBases.end());
596     }
597   }
598   return ret;
599 }
600 
601 // Create output sections described by SECTIONS commands.
602 void LinkerScript::processSectionCommands() {
603   auto process = [this](OutputSection *osec) {
604     SmallVector<InputSectionBase *, 0> v = createInputSectionList(*osec);
605 
606     // The output section name `/DISCARD/' is special.
607     // Any input section assigned to it is discarded.
608     if (osec->name == "/DISCARD/") {
609       for (InputSectionBase *s : v)
610         discard(*s);
611       discardSynthetic(*osec);
612       osec->commands.clear();
613       return false;
614     }
615 
616     // This is for ONLY_IF_RO and ONLY_IF_RW. An output section directive
617     // ".foo : ONLY_IF_R[OW] { ... }" is handled only if all member input
618     // sections satisfy a given constraint. If not, a directive is handled
619     // as if it wasn't present from the beginning.
620     //
621     // Because we'll iterate over SectionCommands many more times, the easy
622     // way to "make it as if it wasn't present" is to make it empty.
623     if (!matchConstraints(v, osec->constraint)) {
624       for (InputSectionBase *s : v)
625         s->parent = nullptr;
626       osec->commands.clear();
627       return false;
628     }
629 
630     // Handle subalign (e.g. ".foo : SUBALIGN(32) { ... }"). If subalign
631     // is given, input sections are aligned to that value, whether the
632     // given value is larger or smaller than the original section alignment.
633     if (osec->subalignExpr) {
634       uint32_t subalign = osec->subalignExpr().getValue();
635       for (InputSectionBase *s : v)
636         s->alignment = subalign;
637     }
638 
639     // Set the partition field the same way OutputSection::recordSection()
640     // does. Partitions cannot be used with the SECTIONS command, so this is
641     // always 1.
642     osec->partition = 1;
643     return true;
644   };
645 
646   // Process OVERWRITE_SECTIONS first so that it can overwrite the main script
647   // or orphans.
648   DenseMap<CachedHashStringRef, OutputSection *> map;
649   size_t i = 0;
650   for (OutputSection *osec : overwriteSections)
651     if (process(osec) &&
652         !map.try_emplace(CachedHashStringRef(osec->name), osec).second)
653       warn("OVERWRITE_SECTIONS specifies duplicate " + osec->name);
654   for (SectionCommand *&base : sectionCommands)
655     if (auto *osec = dyn_cast<OutputSection>(base)) {
656       if (OutputSection *overwrite =
657               map.lookup(CachedHashStringRef(osec->name))) {
658         log(overwrite->location + " overwrites " + osec->name);
659         overwrite->sectionIndex = i++;
660         base = overwrite;
661       } else if (process(osec)) {
662         osec->sectionIndex = i++;
663       }
664     }
665 
666   // If an OVERWRITE_SECTIONS specified output section is not in
667   // sectionCommands, append it to the end. The section will be inserted by
668   // orphan placement.
669   for (OutputSection *osec : overwriteSections)
670     if (osec->partition == 1 && osec->sectionIndex == UINT32_MAX)
671       sectionCommands.push_back(osec);
672 }
673 
674 void LinkerScript::processSymbolAssignments() {
675   // Dot outside an output section still represents a relative address, whose
676   // sh_shndx should not be SHN_UNDEF or SHN_ABS. Create a dummy aether section
677   // that fills the void outside a section. It has an index of one, which is
678   // indistinguishable from any other regular section index.
679   aether = make<OutputSection>("", 0, SHF_ALLOC);
680   aether->sectionIndex = 1;
681 
682   // ctx captures the local AddressState and makes it accessible deliberately.
683   // This is needed as there are some cases where we cannot just thread the
684   // current state through to a lambda function created by the script parser.
685   AddressState state;
686   ctx = &state;
687   ctx->outSec = aether;
688 
689   for (SectionCommand *cmd : sectionCommands) {
690     if (auto *assign = dyn_cast<SymbolAssignment>(cmd))
691       addSymbol(assign);
692     else
693       for (SectionCommand *subCmd : cast<OutputSection>(cmd)->commands)
694         if (auto *assign = dyn_cast<SymbolAssignment>(subCmd))
695           addSymbol(assign);
696   }
697 
698   ctx = nullptr;
699 }
700 
701 static OutputSection *findByName(ArrayRef<SectionCommand *> vec,
702                                  StringRef name) {
703   for (SectionCommand *cmd : vec)
704     if (auto *sec = dyn_cast<OutputSection>(cmd))
705       if (sec->name == name)
706         return sec;
707   return nullptr;
708 }
709 
710 static OutputSection *createSection(InputSectionBase *isec,
711                                     StringRef outsecName) {
712   OutputSection *sec = script->createOutputSection(outsecName, "<internal>");
713   sec->recordSection(isec);
714   return sec;
715 }
716 
717 static OutputSection *
718 addInputSec(StringMap<TinyPtrVector<OutputSection *>> &map,
719             InputSectionBase *isec, StringRef outsecName) {
720   // Sections with SHT_GROUP or SHF_GROUP attributes reach here only when the -r
721   // option is given. A section with SHT_GROUP defines a "section group", and
722   // its members have SHF_GROUP attribute. Usually these flags have already been
723   // stripped by InputFiles.cpp as section groups are processed and uniquified.
724   // However, for the -r option, we want to pass through all section groups
725   // as-is because adding/removing members or merging them with other groups
726   // change their semantics.
727   if (isec->type == SHT_GROUP || (isec->flags & SHF_GROUP))
728     return createSection(isec, outsecName);
729 
730   // Imagine .zed : { *(.foo) *(.bar) } script. Both foo and bar may have
731   // relocation sections .rela.foo and .rela.bar for example. Most tools do
732   // not allow multiple REL[A] sections for output section. Hence we
733   // should combine these relocation sections into single output.
734   // We skip synthetic sections because it can be .rela.dyn/.rela.plt or any
735   // other REL[A] sections created by linker itself.
736   if (!isa<SyntheticSection>(isec) &&
737       (isec->type == SHT_REL || isec->type == SHT_RELA)) {
738     auto *sec = cast<InputSection>(isec);
739     OutputSection *out = sec->getRelocatedSection()->getOutputSection();
740 
741     if (out->relocationSection) {
742       out->relocationSection->recordSection(sec);
743       return nullptr;
744     }
745 
746     out->relocationSection = createSection(isec, outsecName);
747     return out->relocationSection;
748   }
749 
750   //  The ELF spec just says
751   // ----------------------------------------------------------------
752   // In the first phase, input sections that match in name, type and
753   // attribute flags should be concatenated into single sections.
754   // ----------------------------------------------------------------
755   //
756   // However, it is clear that at least some flags have to be ignored for
757   // section merging. At the very least SHF_GROUP and SHF_COMPRESSED have to be
758   // ignored. We should not have two output .text sections just because one was
759   // in a group and another was not for example.
760   //
761   // It also seems that wording was a late addition and didn't get the
762   // necessary scrutiny.
763   //
764   // Merging sections with different flags is expected by some users. One
765   // reason is that if one file has
766   //
767   // int *const bar __attribute__((section(".foo"))) = (int *)0;
768   //
769   // gcc with -fPIC will produce a read only .foo section. But if another
770   // file has
771   //
772   // int zed;
773   // int *const bar __attribute__((section(".foo"))) = (int *)&zed;
774   //
775   // gcc with -fPIC will produce a read write section.
776   //
777   // Last but not least, when using linker script the merge rules are forced by
778   // the script. Unfortunately, linker scripts are name based. This means that
779   // expressions like *(.foo*) can refer to multiple input sections with
780   // different flags. We cannot put them in different output sections or we
781   // would produce wrong results for
782   //
783   // start = .; *(.foo.*) end = .; *(.bar)
784   //
785   // and a mapping of .foo1 and .bar1 to one section and .foo2 and .bar2 to
786   // another. The problem is that there is no way to layout those output
787   // sections such that the .foo sections are the only thing between the start
788   // and end symbols.
789   //
790   // Given the above issues, we instead merge sections by name and error on
791   // incompatible types and flags.
792   TinyPtrVector<OutputSection *> &v = map[outsecName];
793   for (OutputSection *sec : v) {
794     if (sec->partition != isec->partition)
795       continue;
796 
797     if (config->relocatable && (isec->flags & SHF_LINK_ORDER)) {
798       // Merging two SHF_LINK_ORDER sections with different sh_link fields will
799       // change their semantics, so we only merge them in -r links if they will
800       // end up being linked to the same output section. The casts are fine
801       // because everything in the map was created by the orphan placement code.
802       auto *firstIsec = cast<InputSectionBase>(
803           cast<InputSectionDescription>(sec->commands[0])->sectionBases[0]);
804       OutputSection *firstIsecOut =
805           firstIsec->flags & SHF_LINK_ORDER
806               ? firstIsec->getLinkOrderDep()->getOutputSection()
807               : nullptr;
808       if (firstIsecOut != isec->getLinkOrderDep()->getOutputSection())
809         continue;
810     }
811 
812     sec->recordSection(isec);
813     return nullptr;
814   }
815 
816   OutputSection *sec = createSection(isec, outsecName);
817   v.push_back(sec);
818   return sec;
819 }
820 
821 // Add sections that didn't match any sections command.
822 void LinkerScript::addOrphanSections() {
823   StringMap<TinyPtrVector<OutputSection *>> map;
824   SmallVector<OutputSection *, 0> v;
825 
826   auto add = [&](InputSectionBase *s) {
827     if (s->isLive() && !s->parent) {
828       orphanSections.push_back(s);
829 
830       StringRef name = getOutputSectionName(s);
831       if (config->unique) {
832         v.push_back(createSection(s, name));
833       } else if (OutputSection *sec = findByName(sectionCommands, name)) {
834         sec->recordSection(s);
835       } else {
836         if (OutputSection *os = addInputSec(map, s, name))
837           v.push_back(os);
838         assert(isa<MergeInputSection>(s) ||
839                s->getOutputSection()->sectionIndex == UINT32_MAX);
840       }
841     }
842   };
843 
844   // For further --emit-reloc handling code we need target output section
845   // to be created before we create relocation output section, so we want
846   // to create target sections first. We do not want priority handling
847   // for synthetic sections because them are special.
848   for (InputSectionBase *isec : inputSections) {
849     // In -r links, SHF_LINK_ORDER sections are added while adding their parent
850     // sections because we need to know the parent's output section before we
851     // can select an output section for the SHF_LINK_ORDER section.
852     if (config->relocatable && (isec->flags & SHF_LINK_ORDER))
853       continue;
854 
855     if (auto *sec = dyn_cast<InputSection>(isec))
856       if (InputSectionBase *rel = sec->getRelocatedSection())
857         if (auto *relIS = dyn_cast_or_null<InputSectionBase>(rel->parent))
858           add(relIS);
859     add(isec);
860     if (config->relocatable)
861       for (InputSectionBase *depSec : isec->dependentSections)
862         if (depSec->flags & SHF_LINK_ORDER)
863           add(depSec);
864   }
865 
866   // If no SECTIONS command was given, we should insert sections commands
867   // before others, so that we can handle scripts which refers them,
868   // for example: "foo = ABSOLUTE(ADDR(.text)));".
869   // When SECTIONS command is present we just add all orphans to the end.
870   if (hasSectionsCommand)
871     sectionCommands.insert(sectionCommands.end(), v.begin(), v.end());
872   else
873     sectionCommands.insert(sectionCommands.begin(), v.begin(), v.end());
874 }
875 
876 void LinkerScript::diagnoseOrphanHandling() const {
877   llvm::TimeTraceScope timeScope("Diagnose orphan sections");
878   if (config->orphanHandling == OrphanHandlingPolicy::Place)
879     return;
880   for (const InputSectionBase *sec : orphanSections) {
881     // Input SHT_REL[A] retained by --emit-relocs are ignored by
882     // computeInputSections(). Don't warn/error.
883     if (isa<InputSection>(sec) &&
884         cast<InputSection>(sec)->getRelocatedSection())
885       continue;
886 
887     StringRef name = getOutputSectionName(sec);
888     if (config->orphanHandling == OrphanHandlingPolicy::Error)
889       error(toString(sec) + " is being placed in '" + name + "'");
890     else
891       warn(toString(sec) + " is being placed in '" + name + "'");
892   }
893 }
894 
895 // This function searches for a memory region to place the given output
896 // section in. If found, a pointer to the appropriate memory region is
897 // returned in the first member of the pair. Otherwise, a nullptr is returned.
898 // The second member of the pair is a hint that should be passed to the
899 // subsequent call of this method.
900 std::pair<MemoryRegion *, MemoryRegion *>
901 LinkerScript::findMemoryRegion(OutputSection *sec, MemoryRegion *hint) {
902   // Non-allocatable sections are not part of the process image.
903   if (!(sec->flags & SHF_ALLOC)) {
904     if (!sec->memoryRegionName.empty())
905       warn("ignoring memory region assignment for non-allocatable section '" +
906            sec->name + "'");
907     return {nullptr, nullptr};
908   }
909 
910   // If a memory region name was specified in the output section command,
911   // then try to find that region first.
912   if (!sec->memoryRegionName.empty()) {
913     if (MemoryRegion *m = memoryRegions.lookup(sec->memoryRegionName))
914       return {m, m};
915     error("memory region '" + sec->memoryRegionName + "' not declared");
916     return {nullptr, nullptr};
917   }
918 
919   // If at least one memory region is defined, all sections must
920   // belong to some memory region. Otherwise, we don't need to do
921   // anything for memory regions.
922   if (memoryRegions.empty())
923     return {nullptr, nullptr};
924 
925   // An orphan section should continue the previous memory region.
926   if (sec->sectionIndex == UINT32_MAX && hint)
927     return {hint, hint};
928 
929   // See if a region can be found by matching section flags.
930   for (auto &pair : memoryRegions) {
931     MemoryRegion *m = pair.second;
932     if (m->compatibleWith(sec->flags))
933       return {m, nullptr};
934   }
935 
936   // Otherwise, no suitable region was found.
937   error("no memory region specified for section '" + sec->name + "'");
938   return {nullptr, nullptr};
939 }
940 
941 static OutputSection *findFirstSection(PhdrEntry *load) {
942   for (OutputSection *sec : outputSections)
943     if (sec->ptLoad == load)
944       return sec;
945   return nullptr;
946 }
947 
948 // This function assigns offsets to input sections and an output section
949 // for a single sections command (e.g. ".text { *(.text); }").
950 void LinkerScript::assignOffsets(OutputSection *sec) {
951   const bool isTbss = (sec->flags & SHF_TLS) && sec->type == SHT_NOBITS;
952   const bool sameMemRegion = ctx->memRegion == sec->memRegion;
953   const bool prevLMARegionIsDefault = ctx->lmaRegion == nullptr;
954   const uint64_t savedDot = dot;
955   ctx->memRegion = sec->memRegion;
956   ctx->lmaRegion = sec->lmaRegion;
957 
958   if (!(sec->flags & SHF_ALLOC)) {
959     // Non-SHF_ALLOC sections have zero addresses.
960     dot = 0;
961   } else if (isTbss) {
962     // Allow consecutive SHF_TLS SHT_NOBITS output sections. The address range
963     // starts from the end address of the previous tbss section.
964     if (ctx->tbssAddr == 0)
965       ctx->tbssAddr = dot;
966     else
967       dot = ctx->tbssAddr;
968   } else {
969     if (ctx->memRegion)
970       dot = ctx->memRegion->curPos;
971     if (sec->addrExpr)
972       setDot(sec->addrExpr, sec->location, false);
973 
974     // If the address of the section has been moved forward by an explicit
975     // expression so that it now starts past the current curPos of the enclosing
976     // region, we need to expand the current region to account for the space
977     // between the previous section, if any, and the start of this section.
978     if (ctx->memRegion && ctx->memRegion->curPos < dot)
979       expandMemoryRegion(ctx->memRegion, dot - ctx->memRegion->curPos,
980                          sec->name);
981   }
982 
983   ctx->outSec = sec;
984   if (sec->addrExpr && script->hasSectionsCommand) {
985     // The alignment is ignored.
986     sec->addr = dot;
987   } else {
988     // sec->alignment is the max of ALIGN and the maximum of input
989     // section alignments.
990     const uint64_t pos = dot;
991     dot = alignTo(dot, sec->alignment);
992     sec->addr = dot;
993     expandMemoryRegions(dot - pos);
994   }
995 
996   // ctx->lmaOffset is LMA minus VMA. If LMA is explicitly specified via AT() or
997   // AT>, recompute ctx->lmaOffset; otherwise, if both previous/current LMA
998   // region is the default, and the two sections are in the same memory region,
999   // reuse previous lmaOffset; otherwise, reset lmaOffset to 0. This emulates
1000   // heuristics described in
1001   // https://sourceware.org/binutils/docs/ld/Output-Section-LMA.html
1002   if (sec->lmaExpr) {
1003     ctx->lmaOffset = sec->lmaExpr().getValue() - dot;
1004   } else if (MemoryRegion *mr = sec->lmaRegion) {
1005     uint64_t lmaStart = alignTo(mr->curPos, sec->alignment);
1006     if (mr->curPos < lmaStart)
1007       expandMemoryRegion(mr, lmaStart - mr->curPos, sec->name);
1008     ctx->lmaOffset = lmaStart - dot;
1009   } else if (!sameMemRegion || !prevLMARegionIsDefault) {
1010     ctx->lmaOffset = 0;
1011   }
1012 
1013   // Propagate ctx->lmaOffset to the first "non-header" section.
1014   if (PhdrEntry *l = sec->ptLoad)
1015     if (sec == findFirstSection(l))
1016       l->lmaOffset = ctx->lmaOffset;
1017 
1018   // We can call this method multiple times during the creation of
1019   // thunks and want to start over calculation each time.
1020   sec->size = 0;
1021 
1022   // We visited SectionsCommands from processSectionCommands to
1023   // layout sections. Now, we visit SectionsCommands again to fix
1024   // section offsets.
1025   for (SectionCommand *cmd : sec->commands) {
1026     // This handles the assignments to symbol or to the dot.
1027     if (auto *assign = dyn_cast<SymbolAssignment>(cmd)) {
1028       assign->addr = dot;
1029       assignSymbol(assign, true);
1030       assign->size = dot - assign->addr;
1031       continue;
1032     }
1033 
1034     // Handle BYTE(), SHORT(), LONG(), or QUAD().
1035     if (auto *data = dyn_cast<ByteCommand>(cmd)) {
1036       data->offset = dot - sec->addr;
1037       dot += data->size;
1038       expandOutputSection(data->size);
1039       continue;
1040     }
1041 
1042     // Handle a single input section description command.
1043     // It calculates and assigns the offsets for each section and also
1044     // updates the output section size.
1045     for (InputSection *isec : cast<InputSectionDescription>(cmd)->sections) {
1046       assert(isec->getParent() == sec);
1047       const uint64_t pos = dot;
1048       dot = alignTo(dot, isec->alignment);
1049       isec->outSecOff = dot - sec->addr;
1050       dot += isec->getSize();
1051 
1052       // Update output section size after adding each section. This is so that
1053       // SIZEOF works correctly in the case below:
1054       // .foo { *(.aaa) a = SIZEOF(.foo); *(.bbb) }
1055       expandOutputSection(dot - pos);
1056     }
1057   }
1058 
1059   // Non-SHF_ALLOC sections do not affect the addresses of other OutputSections
1060   // as they are not part of the process image.
1061   if (!(sec->flags & SHF_ALLOC)) {
1062     dot = savedDot;
1063   } else if (isTbss) {
1064     // NOBITS TLS sections are similar. Additionally save the end address.
1065     ctx->tbssAddr = dot;
1066     dot = savedDot;
1067   }
1068 }
1069 
1070 static bool isDiscardable(const OutputSection &sec) {
1071   if (sec.name == "/DISCARD/")
1072     return true;
1073 
1074   // We do not want to remove OutputSections with expressions that reference
1075   // symbols even if the OutputSection is empty. We want to ensure that the
1076   // expressions can be evaluated and report an error if they cannot.
1077   if (sec.expressionsUseSymbols)
1078     return false;
1079 
1080   // OutputSections may be referenced by name in ADDR and LOADADDR expressions,
1081   // as an empty Section can has a valid VMA and LMA we keep the OutputSection
1082   // to maintain the integrity of the other Expression.
1083   if (sec.usedInExpression)
1084     return false;
1085 
1086   for (SectionCommand *cmd : sec.commands) {
1087     if (auto assign = dyn_cast<SymbolAssignment>(cmd))
1088       // Don't create empty output sections just for unreferenced PROVIDE
1089       // symbols.
1090       if (assign->name != "." && !assign->sym)
1091         continue;
1092 
1093     if (!isa<InputSectionDescription>(*cmd))
1094       return false;
1095   }
1096   return true;
1097 }
1098 
1099 bool LinkerScript::isDiscarded(const OutputSection *sec) const {
1100   return hasSectionsCommand && (getFirstInputSection(sec) == nullptr) &&
1101          isDiscardable(*sec);
1102 }
1103 
1104 static void maybePropagatePhdrs(OutputSection &sec,
1105                                 SmallVector<StringRef, 0> &phdrs) {
1106   if (sec.phdrs.empty()) {
1107     // To match the bfd linker script behaviour, only propagate program
1108     // headers to sections that are allocated.
1109     if (sec.flags & SHF_ALLOC)
1110       sec.phdrs = phdrs;
1111   } else {
1112     phdrs = sec.phdrs;
1113   }
1114 }
1115 
1116 void LinkerScript::adjustSectionsBeforeSorting() {
1117   // If the output section contains only symbol assignments, create a
1118   // corresponding output section. The issue is what to do with linker script
1119   // like ".foo : { symbol = 42; }". One option would be to convert it to
1120   // "symbol = 42;". That is, move the symbol out of the empty section
1121   // description. That seems to be what bfd does for this simple case. The
1122   // problem is that this is not completely general. bfd will give up and
1123   // create a dummy section too if there is a ". = . + 1" inside the section
1124   // for example.
1125   // Given that we want to create the section, we have to worry what impact
1126   // it will have on the link. For example, if we just create a section with
1127   // 0 for flags, it would change which PT_LOADs are created.
1128   // We could remember that particular section is dummy and ignore it in
1129   // other parts of the linker, but unfortunately there are quite a few places
1130   // that would need to change:
1131   //   * The program header creation.
1132   //   * The orphan section placement.
1133   //   * The address assignment.
1134   // The other option is to pick flags that minimize the impact the section
1135   // will have on the rest of the linker. That is why we copy the flags from
1136   // the previous sections. Only a few flags are needed to keep the impact low.
1137   uint64_t flags = SHF_ALLOC;
1138 
1139   SmallVector<StringRef, 0> defPhdrs;
1140   for (SectionCommand *&cmd : sectionCommands) {
1141     auto *sec = dyn_cast<OutputSection>(cmd);
1142     if (!sec)
1143       continue;
1144 
1145     // Handle align (e.g. ".foo : ALIGN(16) { ... }").
1146     if (sec->alignExpr)
1147       sec->alignment =
1148           std::max<uint32_t>(sec->alignment, sec->alignExpr().getValue());
1149 
1150     // The input section might have been removed (if it was an empty synthetic
1151     // section), but we at least know the flags.
1152     if (sec->hasInputSections)
1153       flags = sec->flags;
1154 
1155     // We do not want to keep any special flags for output section
1156     // in case it is empty.
1157     bool isEmpty = (getFirstInputSection(sec) == nullptr);
1158     if (isEmpty)
1159       sec->flags = flags & ((sec->nonAlloc ? 0 : (uint64_t)SHF_ALLOC) |
1160                             SHF_WRITE | SHF_EXECINSTR);
1161 
1162     // The code below may remove empty output sections. We should save the
1163     // specified program headers (if exist) and propagate them to subsequent
1164     // sections which do not specify program headers.
1165     // An example of such a linker script is:
1166     // SECTIONS { .empty : { *(.empty) } :rw
1167     //            .foo : { *(.foo) } }
1168     // Note: at this point the order of output sections has not been finalized,
1169     // because orphans have not been inserted into their expected positions. We
1170     // will handle them in adjustSectionsAfterSorting().
1171     if (sec->sectionIndex != UINT32_MAX)
1172       maybePropagatePhdrs(*sec, defPhdrs);
1173 
1174     if (isEmpty && isDiscardable(*sec)) {
1175       sec->markDead();
1176       cmd = nullptr;
1177     }
1178   }
1179 
1180   // It is common practice to use very generic linker scripts. So for any
1181   // given run some of the output sections in the script will be empty.
1182   // We could create corresponding empty output sections, but that would
1183   // clutter the output.
1184   // We instead remove trivially empty sections. The bfd linker seems even
1185   // more aggressive at removing them.
1186   llvm::erase_if(sectionCommands, [&](SectionCommand *cmd) { return !cmd; });
1187 }
1188 
1189 void LinkerScript::adjustSectionsAfterSorting() {
1190   // Try and find an appropriate memory region to assign offsets in.
1191   MemoryRegion *hint = nullptr;
1192   for (SectionCommand *cmd : sectionCommands) {
1193     if (auto *sec = dyn_cast<OutputSection>(cmd)) {
1194       if (!sec->lmaRegionName.empty()) {
1195         if (MemoryRegion *m = memoryRegions.lookup(sec->lmaRegionName))
1196           sec->lmaRegion = m;
1197         else
1198           error("memory region '" + sec->lmaRegionName + "' not declared");
1199       }
1200       std::tie(sec->memRegion, hint) = findMemoryRegion(sec, hint);
1201     }
1202   }
1203 
1204   // If output section command doesn't specify any segments,
1205   // and we haven't previously assigned any section to segment,
1206   // then we simply assign section to the very first load segment.
1207   // Below is an example of such linker script:
1208   // PHDRS { seg PT_LOAD; }
1209   // SECTIONS { .aaa : { *(.aaa) } }
1210   SmallVector<StringRef, 0> defPhdrs;
1211   auto firstPtLoad = llvm::find_if(phdrsCommands, [](const PhdrsCommand &cmd) {
1212     return cmd.type == PT_LOAD;
1213   });
1214   if (firstPtLoad != phdrsCommands.end())
1215     defPhdrs.push_back(firstPtLoad->name);
1216 
1217   // Walk the commands and propagate the program headers to commands that don't
1218   // explicitly specify them.
1219   for (SectionCommand *cmd : sectionCommands)
1220     if (auto *sec = dyn_cast<OutputSection>(cmd))
1221       maybePropagatePhdrs(*sec, defPhdrs);
1222 }
1223 
1224 static uint64_t computeBase(uint64_t min, bool allocateHeaders) {
1225   // If there is no SECTIONS or if the linkerscript is explicit about program
1226   // headers, do our best to allocate them.
1227   if (!script->hasSectionsCommand || allocateHeaders)
1228     return 0;
1229   // Otherwise only allocate program headers if that would not add a page.
1230   return alignDown(min, config->maxPageSize);
1231 }
1232 
1233 // When the SECTIONS command is used, try to find an address for the file and
1234 // program headers output sections, which can be added to the first PT_LOAD
1235 // segment when program headers are created.
1236 //
1237 // We check if the headers fit below the first allocated section. If there isn't
1238 // enough space for these sections, we'll remove them from the PT_LOAD segment,
1239 // and we'll also remove the PT_PHDR segment.
1240 void LinkerScript::allocateHeaders(SmallVector<PhdrEntry *, 0> &phdrs) {
1241   uint64_t min = std::numeric_limits<uint64_t>::max();
1242   for (OutputSection *sec : outputSections)
1243     if (sec->flags & SHF_ALLOC)
1244       min = std::min<uint64_t>(min, sec->addr);
1245 
1246   auto it = llvm::find_if(
1247       phdrs, [](const PhdrEntry *e) { return e->p_type == PT_LOAD; });
1248   if (it == phdrs.end())
1249     return;
1250   PhdrEntry *firstPTLoad = *it;
1251 
1252   bool hasExplicitHeaders =
1253       llvm::any_of(phdrsCommands, [](const PhdrsCommand &cmd) {
1254         return cmd.hasPhdrs || cmd.hasFilehdr;
1255       });
1256   bool paged = !config->omagic && !config->nmagic;
1257   uint64_t headerSize = getHeaderSize();
1258   if ((paged || hasExplicitHeaders) &&
1259       headerSize <= min - computeBase(min, hasExplicitHeaders)) {
1260     min = alignDown(min - headerSize, config->maxPageSize);
1261     Out::elfHeader->addr = min;
1262     Out::programHeaders->addr = min + Out::elfHeader->size;
1263     return;
1264   }
1265 
1266   // Error if we were explicitly asked to allocate headers.
1267   if (hasExplicitHeaders)
1268     error("could not allocate headers");
1269 
1270   Out::elfHeader->ptLoad = nullptr;
1271   Out::programHeaders->ptLoad = nullptr;
1272   firstPTLoad->firstSec = findFirstSection(firstPTLoad);
1273 
1274   llvm::erase_if(phdrs,
1275                  [](const PhdrEntry *e) { return e->p_type == PT_PHDR; });
1276 }
1277 
1278 LinkerScript::AddressState::AddressState() {
1279   for (auto &mri : script->memoryRegions) {
1280     MemoryRegion *mr = mri.second;
1281     mr->curPos = (mr->origin)().getValue();
1282   }
1283 }
1284 
1285 // Here we assign addresses as instructed by linker script SECTIONS
1286 // sub-commands. Doing that allows us to use final VA values, so here
1287 // we also handle rest commands like symbol assignments and ASSERTs.
1288 // Returns a symbol that has changed its section or value, or nullptr if no
1289 // symbol has changed.
1290 const Defined *LinkerScript::assignAddresses() {
1291   if (script->hasSectionsCommand) {
1292     // With a linker script, assignment of addresses to headers is covered by
1293     // allocateHeaders().
1294     dot = config->imageBase.getValueOr(0);
1295   } else {
1296     // Assign addresses to headers right now.
1297     dot = target->getImageBase();
1298     Out::elfHeader->addr = dot;
1299     Out::programHeaders->addr = dot + Out::elfHeader->size;
1300     dot += getHeaderSize();
1301   }
1302 
1303   AddressState state;
1304   ctx = &state;
1305   errorOnMissingSection = true;
1306   ctx->outSec = aether;
1307 
1308   SymbolAssignmentMap oldValues = getSymbolAssignmentValues(sectionCommands);
1309   for (SectionCommand *cmd : sectionCommands) {
1310     if (auto *assign = dyn_cast<SymbolAssignment>(cmd)) {
1311       assign->addr = dot;
1312       assignSymbol(assign, false);
1313       assign->size = dot - assign->addr;
1314       continue;
1315     }
1316     assignOffsets(cast<OutputSection>(cmd));
1317   }
1318 
1319   ctx = nullptr;
1320   return getChangedSymbolAssignment(oldValues);
1321 }
1322 
1323 // Creates program headers as instructed by PHDRS linker script command.
1324 SmallVector<PhdrEntry *, 0> LinkerScript::createPhdrs() {
1325   SmallVector<PhdrEntry *, 0> ret;
1326 
1327   // Process PHDRS and FILEHDR keywords because they are not
1328   // real output sections and cannot be added in the following loop.
1329   for (const PhdrsCommand &cmd : phdrsCommands) {
1330     PhdrEntry *phdr = make<PhdrEntry>(cmd.type, cmd.flags.getValueOr(PF_R));
1331 
1332     if (cmd.hasFilehdr)
1333       phdr->add(Out::elfHeader);
1334     if (cmd.hasPhdrs)
1335       phdr->add(Out::programHeaders);
1336 
1337     if (cmd.lmaExpr) {
1338       phdr->p_paddr = cmd.lmaExpr().getValue();
1339       phdr->hasLMA = true;
1340     }
1341     ret.push_back(phdr);
1342   }
1343 
1344   // Add output sections to program headers.
1345   for (OutputSection *sec : outputSections) {
1346     // Assign headers specified by linker script
1347     for (size_t id : getPhdrIndices(sec)) {
1348       ret[id]->add(sec);
1349       if (!phdrsCommands[id].flags.hasValue())
1350         ret[id]->p_flags |= sec->getPhdrFlags();
1351     }
1352   }
1353   return ret;
1354 }
1355 
1356 // Returns true if we should emit an .interp section.
1357 //
1358 // We usually do. But if PHDRS commands are given, and
1359 // no PT_INTERP is there, there's no place to emit an
1360 // .interp, so we don't do that in that case.
1361 bool LinkerScript::needsInterpSection() {
1362   if (phdrsCommands.empty())
1363     return true;
1364   for (PhdrsCommand &cmd : phdrsCommands)
1365     if (cmd.type == PT_INTERP)
1366       return true;
1367   return false;
1368 }
1369 
1370 ExprValue LinkerScript::getSymbolValue(StringRef name, const Twine &loc) {
1371   if (name == ".") {
1372     if (ctx)
1373       return {ctx->outSec, false, dot - ctx->outSec->addr, loc};
1374     error(loc + ": unable to get location counter value");
1375     return 0;
1376   }
1377 
1378   if (Symbol *sym = symtab->find(name)) {
1379     if (auto *ds = dyn_cast<Defined>(sym)) {
1380       ExprValue v{ds->section, false, ds->value, loc};
1381       // Retain the original st_type, so that the alias will get the same
1382       // behavior in relocation processing. Any operation will reset st_type to
1383       // STT_NOTYPE.
1384       v.type = ds->type;
1385       return v;
1386     }
1387     if (isa<SharedSymbol>(sym))
1388       if (!errorOnMissingSection)
1389         return {nullptr, false, 0, loc};
1390   }
1391 
1392   error(loc + ": symbol not found: " + name);
1393   return 0;
1394 }
1395 
1396 // Returns the index of the segment named Name.
1397 static Optional<size_t> getPhdrIndex(ArrayRef<PhdrsCommand> vec,
1398                                      StringRef name) {
1399   for (size_t i = 0; i < vec.size(); ++i)
1400     if (vec[i].name == name)
1401       return i;
1402   return None;
1403 }
1404 
1405 // Returns indices of ELF headers containing specific section. Each index is a
1406 // zero based number of ELF header listed within PHDRS {} script block.
1407 SmallVector<size_t, 0> LinkerScript::getPhdrIndices(OutputSection *cmd) {
1408   SmallVector<size_t, 0> ret;
1409 
1410   for (StringRef s : cmd->phdrs) {
1411     if (Optional<size_t> idx = getPhdrIndex(phdrsCommands, s))
1412       ret.push_back(*idx);
1413     else if (s != "NONE")
1414       error(cmd->location + ": program header '" + s +
1415             "' is not listed in PHDRS");
1416   }
1417   return ret;
1418 }
1419