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