10b57cec5SDimitry Andric //===- SyntheticSections.cpp ----------------------------------------------===// 20b57cec5SDimitry Andric // 30b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 40b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information. 50b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 60b57cec5SDimitry Andric // 70b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 80b57cec5SDimitry Andric // 90b57cec5SDimitry Andric // This file contains linker-synthesized sections. Currently, 100b57cec5SDimitry Andric // synthetic sections are created either output sections or input sections, 110b57cec5SDimitry Andric // but we are rewriting code so that all synthetic sections are created as 120b57cec5SDimitry Andric // input sections. 130b57cec5SDimitry Andric // 140b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 150b57cec5SDimitry Andric 160b57cec5SDimitry Andric #include "SyntheticSections.h" 170b57cec5SDimitry Andric #include "Config.h" 1881ad6265SDimitry Andric #include "DWARF.h" 1981ad6265SDimitry Andric #include "EhFrame.h" 200b57cec5SDimitry Andric #include "InputFiles.h" 210b57cec5SDimitry Andric #include "LinkerScript.h" 220b57cec5SDimitry Andric #include "OutputSections.h" 230b57cec5SDimitry Andric #include "SymbolTable.h" 240b57cec5SDimitry Andric #include "Symbols.h" 250b57cec5SDimitry Andric #include "Target.h" 2681ad6265SDimitry Andric #include "Thunks.h" 270b57cec5SDimitry Andric #include "Writer.h" 2804eeddc0SDimitry Andric #include "lld/Common/CommonLinkerContext.h" 295ffd83dbSDimitry Andric #include "lld/Common/DWARF.h" 300b57cec5SDimitry Andric #include "lld/Common/Strings.h" 310b57cec5SDimitry Andric #include "lld/Common/Version.h" 32fcaf7f86SDimitry Andric #include "llvm/ADT/STLExtras.h" 33*0fca6ea1SDimitry Andric #include "llvm/ADT/Sequence.h" 340b57cec5SDimitry Andric #include "llvm/ADT/SetOperations.h" 350b57cec5SDimitry Andric #include "llvm/ADT/StringExtras.h" 360b57cec5SDimitry Andric #include "llvm/BinaryFormat/Dwarf.h" 3781ad6265SDimitry Andric #include "llvm/BinaryFormat/ELF.h" 38*0fca6ea1SDimitry Andric #include "llvm/DebugInfo/DWARF/DWARFAcceleratorTable.h" 390b57cec5SDimitry Andric #include "llvm/DebugInfo/DWARF/DWARFDebugPubTable.h" 40*0fca6ea1SDimitry Andric #include "llvm/Support/DJB.h" 410b57cec5SDimitry Andric #include "llvm/Support/Endian.h" 420b57cec5SDimitry Andric #include "llvm/Support/LEB128.h" 435ffd83dbSDimitry Andric #include "llvm/Support/Parallel.h" 445ffd83dbSDimitry Andric #include "llvm/Support/TimeProfiler.h" 45*0fca6ea1SDimitry Andric #include <cinttypes> 460b57cec5SDimitry Andric #include <cstdlib> 470b57cec5SDimitry Andric 480b57cec5SDimitry Andric using namespace llvm; 490b57cec5SDimitry Andric using namespace llvm::dwarf; 500b57cec5SDimitry Andric using namespace llvm::ELF; 510b57cec5SDimitry Andric using namespace llvm::object; 520b57cec5SDimitry Andric using namespace llvm::support; 535ffd83dbSDimitry Andric using namespace lld; 545ffd83dbSDimitry Andric using namespace lld::elf; 550b57cec5SDimitry Andric 560b57cec5SDimitry Andric using llvm::support::endian::read32le; 570b57cec5SDimitry Andric using llvm::support::endian::write32le; 580b57cec5SDimitry Andric using llvm::support::endian::write64le; 590b57cec5SDimitry Andric 600b57cec5SDimitry Andric constexpr size_t MergeNoTailSection::numShards; 610b57cec5SDimitry Andric 620b57cec5SDimitry Andric static uint64_t readUint(uint8_t *buf) { 630b57cec5SDimitry Andric return config->is64 ? read64(buf) : read32(buf); 640b57cec5SDimitry Andric } 650b57cec5SDimitry Andric 660b57cec5SDimitry Andric static void writeUint(uint8_t *buf, uint64_t val) { 670b57cec5SDimitry Andric if (config->is64) 680b57cec5SDimitry Andric write64(buf, val); 690b57cec5SDimitry Andric else 700b57cec5SDimitry Andric write32(buf, val); 710b57cec5SDimitry Andric } 720b57cec5SDimitry Andric 730b57cec5SDimitry Andric // Returns an LLD version string. 740b57cec5SDimitry Andric static ArrayRef<uint8_t> getVersion() { 750b57cec5SDimitry Andric // Check LLD_VERSION first for ease of testing. 760b57cec5SDimitry Andric // You can get consistent output by using the environment variable. 770b57cec5SDimitry Andric // This is only for testing. 780b57cec5SDimitry Andric StringRef s = getenv("LLD_VERSION"); 790b57cec5SDimitry Andric if (s.empty()) 8004eeddc0SDimitry Andric s = saver().save(Twine("Linker: ") + getLLDVersion()); 810b57cec5SDimitry Andric 820b57cec5SDimitry Andric // +1 to include the terminating '\0'. 830b57cec5SDimitry Andric return {(const uint8_t *)s.data(), s.size() + 1}; 840b57cec5SDimitry Andric } 850b57cec5SDimitry Andric 860b57cec5SDimitry Andric // Creates a .comment section containing LLD version info. 870b57cec5SDimitry Andric // With this feature, you can identify LLD-generated binaries easily 880b57cec5SDimitry Andric // by "readelf --string-dump .comment <file>". 890b57cec5SDimitry Andric // The returned object is a mergeable string section. 905ffd83dbSDimitry Andric MergeInputSection *elf::createCommentSection() { 911fd87a68SDimitry Andric auto *sec = make<MergeInputSection>(SHF_MERGE | SHF_STRINGS, SHT_PROGBITS, 1, 920b57cec5SDimitry Andric getVersion(), ".comment"); 931fd87a68SDimitry Andric sec->splitIntoPieces(); 941fd87a68SDimitry Andric return sec; 950b57cec5SDimitry Andric } 960b57cec5SDimitry Andric 970b57cec5SDimitry Andric // .MIPS.abiflags section. 980b57cec5SDimitry Andric template <class ELFT> 990b57cec5SDimitry Andric MipsAbiFlagsSection<ELFT>::MipsAbiFlagsSection(Elf_Mips_ABIFlags flags) 1000b57cec5SDimitry Andric : SyntheticSection(SHF_ALLOC, SHT_MIPS_ABIFLAGS, 8, ".MIPS.abiflags"), 1010b57cec5SDimitry Andric flags(flags) { 1020b57cec5SDimitry Andric this->entsize = sizeof(Elf_Mips_ABIFlags); 1030b57cec5SDimitry Andric } 1040b57cec5SDimitry Andric 1050b57cec5SDimitry Andric template <class ELFT> void MipsAbiFlagsSection<ELFT>::writeTo(uint8_t *buf) { 1060b57cec5SDimitry Andric memcpy(buf, &flags, sizeof(flags)); 1070b57cec5SDimitry Andric } 1080b57cec5SDimitry Andric 1090b57cec5SDimitry Andric template <class ELFT> 1101fd87a68SDimitry Andric std::unique_ptr<MipsAbiFlagsSection<ELFT>> MipsAbiFlagsSection<ELFT>::create() { 1110b57cec5SDimitry Andric Elf_Mips_ABIFlags flags = {}; 1120b57cec5SDimitry Andric bool create = false; 1130b57cec5SDimitry Andric 114bdd1243dSDimitry Andric for (InputSectionBase *sec : ctx.inputSections) { 1150b57cec5SDimitry Andric if (sec->type != SHT_MIPS_ABIFLAGS) 1160b57cec5SDimitry Andric continue; 1170b57cec5SDimitry Andric sec->markDead(); 1180b57cec5SDimitry Andric create = true; 1190b57cec5SDimitry Andric 1200b57cec5SDimitry Andric std::string filename = toString(sec->file); 121bdd1243dSDimitry Andric const size_t size = sec->content().size(); 1220b57cec5SDimitry Andric // Older version of BFD (such as the default FreeBSD linker) concatenate 1230b57cec5SDimitry Andric // .MIPS.abiflags instead of merging. To allow for this case (or potential 1240b57cec5SDimitry Andric // zero padding) we ignore everything after the first Elf_Mips_ABIFlags 1250b57cec5SDimitry Andric if (size < sizeof(Elf_Mips_ABIFlags)) { 1260b57cec5SDimitry Andric error(filename + ": invalid size of .MIPS.abiflags section: got " + 1270b57cec5SDimitry Andric Twine(size) + " instead of " + Twine(sizeof(Elf_Mips_ABIFlags))); 1280b57cec5SDimitry Andric return nullptr; 1290b57cec5SDimitry Andric } 130bdd1243dSDimitry Andric auto *s = 131bdd1243dSDimitry Andric reinterpret_cast<const Elf_Mips_ABIFlags *>(sec->content().data()); 1320b57cec5SDimitry Andric if (s->version != 0) { 1330b57cec5SDimitry Andric error(filename + ": unexpected .MIPS.abiflags version " + 1340b57cec5SDimitry Andric Twine(s->version)); 1350b57cec5SDimitry Andric return nullptr; 1360b57cec5SDimitry Andric } 1370b57cec5SDimitry Andric 1380b57cec5SDimitry Andric // LLD checks ISA compatibility in calcMipsEFlags(). Here we just 1390b57cec5SDimitry Andric // select the highest number of ISA/Rev/Ext. 1400b57cec5SDimitry Andric flags.isa_level = std::max(flags.isa_level, s->isa_level); 1410b57cec5SDimitry Andric flags.isa_rev = std::max(flags.isa_rev, s->isa_rev); 1420b57cec5SDimitry Andric flags.isa_ext = std::max(flags.isa_ext, s->isa_ext); 1430b57cec5SDimitry Andric flags.gpr_size = std::max(flags.gpr_size, s->gpr_size); 1440b57cec5SDimitry Andric flags.cpr1_size = std::max(flags.cpr1_size, s->cpr1_size); 1450b57cec5SDimitry Andric flags.cpr2_size = std::max(flags.cpr2_size, s->cpr2_size); 1460b57cec5SDimitry Andric flags.ases |= s->ases; 1470b57cec5SDimitry Andric flags.flags1 |= s->flags1; 1480b57cec5SDimitry Andric flags.flags2 |= s->flags2; 1495ffd83dbSDimitry Andric flags.fp_abi = elf::getMipsFpAbiFlag(flags.fp_abi, s->fp_abi, filename); 1500b57cec5SDimitry Andric }; 1510b57cec5SDimitry Andric 1520b57cec5SDimitry Andric if (create) 1531fd87a68SDimitry Andric return std::make_unique<MipsAbiFlagsSection<ELFT>>(flags); 1540b57cec5SDimitry Andric return nullptr; 1550b57cec5SDimitry Andric } 1560b57cec5SDimitry Andric 1570b57cec5SDimitry Andric // .MIPS.options section. 1580b57cec5SDimitry Andric template <class ELFT> 1590b57cec5SDimitry Andric MipsOptionsSection<ELFT>::MipsOptionsSection(Elf_Mips_RegInfo reginfo) 1600b57cec5SDimitry Andric : SyntheticSection(SHF_ALLOC, SHT_MIPS_OPTIONS, 8, ".MIPS.options"), 1610b57cec5SDimitry Andric reginfo(reginfo) { 1620b57cec5SDimitry Andric this->entsize = sizeof(Elf_Mips_Options) + sizeof(Elf_Mips_RegInfo); 1630b57cec5SDimitry Andric } 1640b57cec5SDimitry Andric 1650b57cec5SDimitry Andric template <class ELFT> void MipsOptionsSection<ELFT>::writeTo(uint8_t *buf) { 1660b57cec5SDimitry Andric auto *options = reinterpret_cast<Elf_Mips_Options *>(buf); 1670b57cec5SDimitry Andric options->kind = ODK_REGINFO; 1680b57cec5SDimitry Andric options->size = getSize(); 1690b57cec5SDimitry Andric 1700b57cec5SDimitry Andric if (!config->relocatable) 1710b57cec5SDimitry Andric reginfo.ri_gp_value = in.mipsGot->getGp(); 1720b57cec5SDimitry Andric memcpy(buf + sizeof(Elf_Mips_Options), ®info, sizeof(reginfo)); 1730b57cec5SDimitry Andric } 1740b57cec5SDimitry Andric 1750b57cec5SDimitry Andric template <class ELFT> 1761fd87a68SDimitry Andric std::unique_ptr<MipsOptionsSection<ELFT>> MipsOptionsSection<ELFT>::create() { 1770b57cec5SDimitry Andric // N64 ABI only. 1780b57cec5SDimitry Andric if (!ELFT::Is64Bits) 1790b57cec5SDimitry Andric return nullptr; 1800b57cec5SDimitry Andric 18104eeddc0SDimitry Andric SmallVector<InputSectionBase *, 0> sections; 182bdd1243dSDimitry Andric for (InputSectionBase *sec : ctx.inputSections) 1830b57cec5SDimitry Andric if (sec->type == SHT_MIPS_OPTIONS) 1840b57cec5SDimitry Andric sections.push_back(sec); 1850b57cec5SDimitry Andric 1860b57cec5SDimitry Andric if (sections.empty()) 1870b57cec5SDimitry Andric return nullptr; 1880b57cec5SDimitry Andric 1890b57cec5SDimitry Andric Elf_Mips_RegInfo reginfo = {}; 1900b57cec5SDimitry Andric for (InputSectionBase *sec : sections) { 1910b57cec5SDimitry Andric sec->markDead(); 1920b57cec5SDimitry Andric 1930b57cec5SDimitry Andric std::string filename = toString(sec->file); 194bdd1243dSDimitry Andric ArrayRef<uint8_t> d = sec->content(); 1950b57cec5SDimitry Andric 1960b57cec5SDimitry Andric while (!d.empty()) { 1970b57cec5SDimitry Andric if (d.size() < sizeof(Elf_Mips_Options)) { 1980b57cec5SDimitry Andric error(filename + ": invalid size of .MIPS.options section"); 1990b57cec5SDimitry Andric break; 2000b57cec5SDimitry Andric } 2010b57cec5SDimitry Andric 2020b57cec5SDimitry Andric auto *opt = reinterpret_cast<const Elf_Mips_Options *>(d.data()); 2030b57cec5SDimitry Andric if (opt->kind == ODK_REGINFO) { 2040b57cec5SDimitry Andric reginfo.ri_gprmask |= opt->getRegInfo().ri_gprmask; 2050b57cec5SDimitry Andric sec->getFile<ELFT>()->mipsGp0 = opt->getRegInfo().ri_gp_value; 2060b57cec5SDimitry Andric break; 2070b57cec5SDimitry Andric } 2080b57cec5SDimitry Andric 2090b57cec5SDimitry Andric if (!opt->size) 2100b57cec5SDimitry Andric fatal(filename + ": zero option descriptor size"); 2110b57cec5SDimitry Andric d = d.slice(opt->size); 2120b57cec5SDimitry Andric } 2130b57cec5SDimitry Andric }; 2140b57cec5SDimitry Andric 2151fd87a68SDimitry Andric return std::make_unique<MipsOptionsSection<ELFT>>(reginfo); 2160b57cec5SDimitry Andric } 2170b57cec5SDimitry Andric 2180b57cec5SDimitry Andric // MIPS .reginfo section. 2190b57cec5SDimitry Andric template <class ELFT> 2200b57cec5SDimitry Andric MipsReginfoSection<ELFT>::MipsReginfoSection(Elf_Mips_RegInfo reginfo) 2210b57cec5SDimitry Andric : SyntheticSection(SHF_ALLOC, SHT_MIPS_REGINFO, 4, ".reginfo"), 2220b57cec5SDimitry Andric reginfo(reginfo) { 2230b57cec5SDimitry Andric this->entsize = sizeof(Elf_Mips_RegInfo); 2240b57cec5SDimitry Andric } 2250b57cec5SDimitry Andric 2260b57cec5SDimitry Andric template <class ELFT> void MipsReginfoSection<ELFT>::writeTo(uint8_t *buf) { 2270b57cec5SDimitry Andric if (!config->relocatable) 2280b57cec5SDimitry Andric reginfo.ri_gp_value = in.mipsGot->getGp(); 2290b57cec5SDimitry Andric memcpy(buf, ®info, sizeof(reginfo)); 2300b57cec5SDimitry Andric } 2310b57cec5SDimitry Andric 2320b57cec5SDimitry Andric template <class ELFT> 2331fd87a68SDimitry Andric std::unique_ptr<MipsReginfoSection<ELFT>> MipsReginfoSection<ELFT>::create() { 2340b57cec5SDimitry Andric // Section should be alive for O32 and N32 ABIs only. 2350b57cec5SDimitry Andric if (ELFT::Is64Bits) 2360b57cec5SDimitry Andric return nullptr; 2370b57cec5SDimitry Andric 23804eeddc0SDimitry Andric SmallVector<InputSectionBase *, 0> sections; 239bdd1243dSDimitry Andric for (InputSectionBase *sec : ctx.inputSections) 2400b57cec5SDimitry Andric if (sec->type == SHT_MIPS_REGINFO) 2410b57cec5SDimitry Andric sections.push_back(sec); 2420b57cec5SDimitry Andric 2430b57cec5SDimitry Andric if (sections.empty()) 2440b57cec5SDimitry Andric return nullptr; 2450b57cec5SDimitry Andric 2460b57cec5SDimitry Andric Elf_Mips_RegInfo reginfo = {}; 2470b57cec5SDimitry Andric for (InputSectionBase *sec : sections) { 2480b57cec5SDimitry Andric sec->markDead(); 2490b57cec5SDimitry Andric 250bdd1243dSDimitry Andric if (sec->content().size() != sizeof(Elf_Mips_RegInfo)) { 2510b57cec5SDimitry Andric error(toString(sec->file) + ": invalid size of .reginfo section"); 2520b57cec5SDimitry Andric return nullptr; 2530b57cec5SDimitry Andric } 2540b57cec5SDimitry Andric 255bdd1243dSDimitry Andric auto *r = reinterpret_cast<const Elf_Mips_RegInfo *>(sec->content().data()); 2560b57cec5SDimitry Andric reginfo.ri_gprmask |= r->ri_gprmask; 2570b57cec5SDimitry Andric sec->getFile<ELFT>()->mipsGp0 = r->ri_gp_value; 2580b57cec5SDimitry Andric }; 2590b57cec5SDimitry Andric 2601fd87a68SDimitry Andric return std::make_unique<MipsReginfoSection<ELFT>>(reginfo); 2610b57cec5SDimitry Andric } 2620b57cec5SDimitry Andric 2635ffd83dbSDimitry Andric InputSection *elf::createInterpSection() { 2640b57cec5SDimitry Andric // StringSaver guarantees that the returned string ends with '\0'. 26504eeddc0SDimitry Andric StringRef s = saver().save(config->dynamicLinker); 2660b57cec5SDimitry Andric ArrayRef<uint8_t> contents = {(const uint8_t *)s.data(), s.size() + 1}; 2670b57cec5SDimitry Andric 2687a6dacacSDimitry Andric return make<InputSection>(ctx.internalFile, SHF_ALLOC, SHT_PROGBITS, 1, 2697a6dacacSDimitry Andric contents, ".interp"); 2700b57cec5SDimitry Andric } 2710b57cec5SDimitry Andric 2725ffd83dbSDimitry Andric Defined *elf::addSyntheticLocal(StringRef name, uint8_t type, uint64_t value, 2730b57cec5SDimitry Andric uint64_t size, InputSectionBase §ion) { 2740eae32dcSDimitry Andric Defined *s = makeDefined(section.file, name, STB_LOCAL, STV_DEFAULT, type, 2750b57cec5SDimitry Andric value, size, §ion); 2760b57cec5SDimitry Andric if (in.symTab) 2770b57cec5SDimitry Andric in.symTab->addSymbol(s); 27806c3fb27SDimitry Andric 27906c3fb27SDimitry Andric if (config->emachine == EM_ARM && !config->isLE && config->armBe8 && 28006c3fb27SDimitry Andric (section.flags & SHF_EXECINSTR)) 28106c3fb27SDimitry Andric // Adding Linker generated mapping symbols to the arm specific mapping 28206c3fb27SDimitry Andric // symbols list. 28306c3fb27SDimitry Andric addArmSyntheticSectionMappingSymbol(s); 28406c3fb27SDimitry Andric 2850b57cec5SDimitry Andric return s; 2860b57cec5SDimitry Andric } 2870b57cec5SDimitry Andric 2880b57cec5SDimitry Andric static size_t getHashSize() { 2890b57cec5SDimitry Andric switch (config->buildId) { 2900b57cec5SDimitry Andric case BuildIdKind::Fast: 2910b57cec5SDimitry Andric return 8; 2920b57cec5SDimitry Andric case BuildIdKind::Md5: 2930b57cec5SDimitry Andric case BuildIdKind::Uuid: 2940b57cec5SDimitry Andric return 16; 2950b57cec5SDimitry Andric case BuildIdKind::Sha1: 2960b57cec5SDimitry Andric return 20; 2970b57cec5SDimitry Andric case BuildIdKind::Hexstring: 2980b57cec5SDimitry Andric return config->buildIdVector.size(); 2990b57cec5SDimitry Andric default: 3000b57cec5SDimitry Andric llvm_unreachable("unknown BuildIdKind"); 3010b57cec5SDimitry Andric } 3020b57cec5SDimitry Andric } 3030b57cec5SDimitry Andric 3040b57cec5SDimitry Andric // This class represents a linker-synthesized .note.gnu.property section. 3050b57cec5SDimitry Andric // 3060b57cec5SDimitry Andric // In x86 and AArch64, object files may contain feature flags indicating the 3070b57cec5SDimitry Andric // features that they have used. The flags are stored in a .note.gnu.property 3080b57cec5SDimitry Andric // section. 3090b57cec5SDimitry Andric // 3100b57cec5SDimitry Andric // lld reads the sections from input files and merges them by computing AND of 3110b57cec5SDimitry Andric // the flags. The result is written as a new .note.gnu.property section. 3120b57cec5SDimitry Andric // 3130b57cec5SDimitry Andric // If the flag is zero (which indicates that the intersection of the feature 3140b57cec5SDimitry Andric // sets is empty, or some input files didn't have .note.gnu.property sections), 3150b57cec5SDimitry Andric // we don't create this section. 3160b57cec5SDimitry Andric GnuPropertySection::GnuPropertySection() 317480093f4SDimitry Andric : SyntheticSection(llvm::ELF::SHF_ALLOC, llvm::ELF::SHT_NOTE, 318480093f4SDimitry Andric config->wordsize, ".note.gnu.property") {} 3190b57cec5SDimitry Andric 3200b57cec5SDimitry Andric void GnuPropertySection::writeTo(uint8_t *buf) { 321*0fca6ea1SDimitry Andric write32(buf, 4); // Name size 322*0fca6ea1SDimitry Andric write32(buf + 4, getSize() - 16); // Content size 323*0fca6ea1SDimitry Andric write32(buf + 8, NT_GNU_PROPERTY_TYPE_0); // Type 324*0fca6ea1SDimitry Andric memcpy(buf + 12, "GNU", 4); // Name string 325*0fca6ea1SDimitry Andric 3260b57cec5SDimitry Andric uint32_t featureAndType = config->emachine == EM_AARCH64 3270b57cec5SDimitry Andric ? GNU_PROPERTY_AARCH64_FEATURE_1_AND 3280b57cec5SDimitry Andric : GNU_PROPERTY_X86_FEATURE_1_AND; 3290b57cec5SDimitry Andric 330*0fca6ea1SDimitry Andric unsigned offset = 16; 331*0fca6ea1SDimitry Andric if (config->andFeatures != 0) { 332*0fca6ea1SDimitry Andric write32(buf + offset + 0, featureAndType); // Feature type 333*0fca6ea1SDimitry Andric write32(buf + offset + 4, 4); // Feature size 334*0fca6ea1SDimitry Andric write32(buf + offset + 8, config->andFeatures); // Feature flags 3350b57cec5SDimitry Andric if (config->is64) 336*0fca6ea1SDimitry Andric write32(buf + offset + 12, 0); // Padding 337*0fca6ea1SDimitry Andric offset += 16; 3380b57cec5SDimitry Andric } 3390b57cec5SDimitry Andric 340*0fca6ea1SDimitry Andric if (!ctx.aarch64PauthAbiCoreInfo.empty()) { 341*0fca6ea1SDimitry Andric write32(buf + offset + 0, GNU_PROPERTY_AARCH64_FEATURE_PAUTH); 342*0fca6ea1SDimitry Andric write32(buf + offset + 4, ctx.aarch64PauthAbiCoreInfo.size()); 343*0fca6ea1SDimitry Andric memcpy(buf + offset + 8, ctx.aarch64PauthAbiCoreInfo.data(), 344*0fca6ea1SDimitry Andric ctx.aarch64PauthAbiCoreInfo.size()); 345*0fca6ea1SDimitry Andric } 346*0fca6ea1SDimitry Andric } 347*0fca6ea1SDimitry Andric 348*0fca6ea1SDimitry Andric size_t GnuPropertySection::getSize() const { 349*0fca6ea1SDimitry Andric uint32_t contentSize = 0; 350*0fca6ea1SDimitry Andric if (config->andFeatures != 0) 351*0fca6ea1SDimitry Andric contentSize += config->is64 ? 16 : 12; 352*0fca6ea1SDimitry Andric if (!ctx.aarch64PauthAbiCoreInfo.empty()) 353*0fca6ea1SDimitry Andric contentSize += 4 + 4 + ctx.aarch64PauthAbiCoreInfo.size(); 354*0fca6ea1SDimitry Andric assert(contentSize != 0); 355*0fca6ea1SDimitry Andric return contentSize + 16; 356*0fca6ea1SDimitry Andric } 3570b57cec5SDimitry Andric 3580b57cec5SDimitry Andric BuildIdSection::BuildIdSection() 3590b57cec5SDimitry Andric : SyntheticSection(SHF_ALLOC, SHT_NOTE, 4, ".note.gnu.build-id"), 3600b57cec5SDimitry Andric hashSize(getHashSize()) {} 3610b57cec5SDimitry Andric 3620b57cec5SDimitry Andric void BuildIdSection::writeTo(uint8_t *buf) { 3630b57cec5SDimitry Andric write32(buf, 4); // Name size 3640b57cec5SDimitry Andric write32(buf + 4, hashSize); // Content size 3650b57cec5SDimitry Andric write32(buf + 8, NT_GNU_BUILD_ID); // Type 3660b57cec5SDimitry Andric memcpy(buf + 12, "GNU", 4); // Name string 3670b57cec5SDimitry Andric hashBuf = buf + 16; 3680b57cec5SDimitry Andric } 3690b57cec5SDimitry Andric 3700b57cec5SDimitry Andric void BuildIdSection::writeBuildId(ArrayRef<uint8_t> buf) { 3710b57cec5SDimitry Andric assert(buf.size() == hashSize); 3720b57cec5SDimitry Andric memcpy(hashBuf, buf.data(), hashSize); 3730b57cec5SDimitry Andric } 3740b57cec5SDimitry Andric 3750b57cec5SDimitry Andric BssSection::BssSection(StringRef name, uint64_t size, uint32_t alignment) 3760b57cec5SDimitry Andric : SyntheticSection(SHF_ALLOC | SHF_WRITE, SHT_NOBITS, alignment, name) { 3770b57cec5SDimitry Andric this->bss = true; 3780b57cec5SDimitry Andric this->size = size; 3790b57cec5SDimitry Andric } 3800b57cec5SDimitry Andric 3810b57cec5SDimitry Andric EhFrameSection::EhFrameSection() 3820b57cec5SDimitry Andric : SyntheticSection(SHF_ALLOC, SHT_PROGBITS, 1, ".eh_frame") {} 3830b57cec5SDimitry Andric 3840b57cec5SDimitry Andric // Search for an existing CIE record or create a new one. 3850b57cec5SDimitry Andric // CIE records from input object files are uniquified by their contents 3860b57cec5SDimitry Andric // and where their relocations point to. 3870b57cec5SDimitry Andric template <class ELFT, class RelTy> 3880b57cec5SDimitry Andric CieRecord *EhFrameSection::addCie(EhSectionPiece &cie, ArrayRef<RelTy> rels) { 3890b57cec5SDimitry Andric Symbol *personality = nullptr; 3900b57cec5SDimitry Andric unsigned firstRelI = cie.firstRelocation; 3910b57cec5SDimitry Andric if (firstRelI != (unsigned)-1) 392*0fca6ea1SDimitry Andric personality = &cie.sec->file->getRelocTargetSym(rels[firstRelI]); 3930b57cec5SDimitry Andric 3940b57cec5SDimitry Andric // Search for an existing CIE by CIE contents/relocation target pair. 3950b57cec5SDimitry Andric CieRecord *&rec = cieMap[{cie.data(), personality}]; 3960b57cec5SDimitry Andric 3970b57cec5SDimitry Andric // If not found, create a new one. 3980b57cec5SDimitry Andric if (!rec) { 3990b57cec5SDimitry Andric rec = make<CieRecord>(); 4000b57cec5SDimitry Andric rec->cie = &cie; 4010b57cec5SDimitry Andric cieRecords.push_back(rec); 4020b57cec5SDimitry Andric } 4030b57cec5SDimitry Andric return rec; 4040b57cec5SDimitry Andric } 4050b57cec5SDimitry Andric 406e8d8bef9SDimitry Andric // There is one FDE per function. Returns a non-null pointer to the function 407e8d8bef9SDimitry Andric // symbol if the given FDE points to a live function. 4080b57cec5SDimitry Andric template <class ELFT, class RelTy> 409e8d8bef9SDimitry Andric Defined *EhFrameSection::isFdeLive(EhSectionPiece &fde, ArrayRef<RelTy> rels) { 4100b57cec5SDimitry Andric auto *sec = cast<EhInputSection>(fde.sec); 4110b57cec5SDimitry Andric unsigned firstRelI = fde.firstRelocation; 4120b57cec5SDimitry Andric 4130b57cec5SDimitry Andric // An FDE should point to some function because FDEs are to describe 4140b57cec5SDimitry Andric // functions. That's however not always the case due to an issue of 4150b57cec5SDimitry Andric // ld.gold with -r. ld.gold may discard only functions and leave their 4160b57cec5SDimitry Andric // corresponding FDEs, which results in creating bad .eh_frame sections. 4170b57cec5SDimitry Andric // To deal with that, we ignore such FDEs. 4180b57cec5SDimitry Andric if (firstRelI == (unsigned)-1) 419e8d8bef9SDimitry Andric return nullptr; 4200b57cec5SDimitry Andric 4210b57cec5SDimitry Andric const RelTy &rel = rels[firstRelI]; 422*0fca6ea1SDimitry Andric Symbol &b = sec->file->getRelocTargetSym(rel); 4230b57cec5SDimitry Andric 4240b57cec5SDimitry Andric // FDEs for garbage-collected or merged-by-ICF sections, or sections in 4250b57cec5SDimitry Andric // another partition, are dead. 4260b57cec5SDimitry Andric if (auto *d = dyn_cast<Defined>(&b)) 4270eae32dcSDimitry Andric if (!d->folded && d->section && d->section->partition == partition) 428e8d8bef9SDimitry Andric return d; 429e8d8bef9SDimitry Andric return nullptr; 4300b57cec5SDimitry Andric } 4310b57cec5SDimitry Andric 4320b57cec5SDimitry Andric // .eh_frame is a sequence of CIE or FDE records. In general, there 4330b57cec5SDimitry Andric // is one CIE record per input object file which is followed by 4340b57cec5SDimitry Andric // a list of FDEs. This function searches an existing CIE or create a new 4350b57cec5SDimitry Andric // one and associates FDEs to the CIE. 4360b57cec5SDimitry Andric template <class ELFT, class RelTy> 43785868e8aSDimitry Andric void EhFrameSection::addRecords(EhInputSection *sec, ArrayRef<RelTy> rels) { 4380b57cec5SDimitry Andric offsetToCie.clear(); 439bdd1243dSDimitry Andric for (EhSectionPiece &cie : sec->cies) 440bdd1243dSDimitry Andric offsetToCie[cie.inputOff] = addCie<ELFT>(cie, rels); 441bdd1243dSDimitry Andric for (EhSectionPiece &fde : sec->fdes) { 442*0fca6ea1SDimitry Andric uint32_t id = endian::read32<ELFT::Endianness>(fde.data().data() + 4); 443bdd1243dSDimitry Andric CieRecord *rec = offsetToCie[fde.inputOff + 4 - id]; 4440b57cec5SDimitry Andric if (!rec) 4450b57cec5SDimitry Andric fatal(toString(sec) + ": invalid CIE reference"); 4460b57cec5SDimitry Andric 447bdd1243dSDimitry Andric if (!isFdeLive<ELFT>(fde, rels)) 4480b57cec5SDimitry Andric continue; 449bdd1243dSDimitry Andric rec->fdes.push_back(&fde); 4500b57cec5SDimitry Andric numFdes++; 4510b57cec5SDimitry Andric } 4520b57cec5SDimitry Andric } 4530b57cec5SDimitry Andric 45485868e8aSDimitry Andric template <class ELFT> 45585868e8aSDimitry Andric void EhFrameSection::addSectionAux(EhInputSection *sec) { 45685868e8aSDimitry Andric if (!sec->isLive()) 45785868e8aSDimitry Andric return; 458349cc55cSDimitry Andric const RelsOrRelas<ELFT> rels = sec->template relsOrRelas<ELFT>(); 459349cc55cSDimitry Andric if (rels.areRelocsRel()) 460349cc55cSDimitry Andric addRecords<ELFT>(sec, rels.rels); 46185868e8aSDimitry Andric else 462349cc55cSDimitry Andric addRecords<ELFT>(sec, rels.relas); 46385868e8aSDimitry Andric } 46485868e8aSDimitry Andric 465e8d8bef9SDimitry Andric // Used by ICF<ELFT>::handleLSDA(). This function is very similar to 466e8d8bef9SDimitry Andric // EhFrameSection::addRecords(). 467e8d8bef9SDimitry Andric template <class ELFT, class RelTy> 468e8d8bef9SDimitry Andric void EhFrameSection::iterateFDEWithLSDAAux( 469e8d8bef9SDimitry Andric EhInputSection &sec, ArrayRef<RelTy> rels, DenseSet<size_t> &ciesWithLSDA, 470e8d8bef9SDimitry Andric llvm::function_ref<void(InputSection &)> fn) { 471bdd1243dSDimitry Andric for (EhSectionPiece &cie : sec.cies) 472bdd1243dSDimitry Andric if (hasLSDA(cie)) 473bdd1243dSDimitry Andric ciesWithLSDA.insert(cie.inputOff); 474bdd1243dSDimitry Andric for (EhSectionPiece &fde : sec.fdes) { 475*0fca6ea1SDimitry Andric uint32_t id = endian::read32<ELFT::Endianness>(fde.data().data() + 4); 476bdd1243dSDimitry Andric if (!ciesWithLSDA.contains(fde.inputOff + 4 - id)) 477e8d8bef9SDimitry Andric continue; 478e8d8bef9SDimitry Andric 479e8d8bef9SDimitry Andric // The CIE has a LSDA argument. Call fn with d's section. 480bdd1243dSDimitry Andric if (Defined *d = isFdeLive<ELFT>(fde, rels)) 481e8d8bef9SDimitry Andric if (auto *s = dyn_cast_or_null<InputSection>(d->section)) 482e8d8bef9SDimitry Andric fn(*s); 483e8d8bef9SDimitry Andric } 484e8d8bef9SDimitry Andric } 485e8d8bef9SDimitry Andric 486e8d8bef9SDimitry Andric template <class ELFT> 487e8d8bef9SDimitry Andric void EhFrameSection::iterateFDEWithLSDA( 488e8d8bef9SDimitry Andric llvm::function_ref<void(InputSection &)> fn) { 489e8d8bef9SDimitry Andric DenseSet<size_t> ciesWithLSDA; 490e8d8bef9SDimitry Andric for (EhInputSection *sec : sections) { 491e8d8bef9SDimitry Andric ciesWithLSDA.clear(); 492349cc55cSDimitry Andric const RelsOrRelas<ELFT> rels = sec->template relsOrRelas<ELFT>(); 493349cc55cSDimitry Andric if (rels.areRelocsRel()) 494349cc55cSDimitry Andric iterateFDEWithLSDAAux<ELFT>(*sec, rels.rels, ciesWithLSDA, fn); 495e8d8bef9SDimitry Andric else 496349cc55cSDimitry Andric iterateFDEWithLSDAAux<ELFT>(*sec, rels.relas, ciesWithLSDA, fn); 497e8d8bef9SDimitry Andric } 498e8d8bef9SDimitry Andric } 499e8d8bef9SDimitry Andric 5000b57cec5SDimitry Andric static void writeCieFde(uint8_t *buf, ArrayRef<uint8_t> d) { 5010b57cec5SDimitry Andric memcpy(buf, d.data(), d.size()); 5020b57cec5SDimitry Andric // Fix the size field. -4 since size does not include the size field itself. 503bdd1243dSDimitry Andric write32(buf, d.size() - 4); 5040b57cec5SDimitry Andric } 5050b57cec5SDimitry Andric 5060b57cec5SDimitry Andric void EhFrameSection::finalizeContents() { 5070b57cec5SDimitry Andric assert(!this->size); // Not finalized. 50885868e8aSDimitry Andric 50985868e8aSDimitry Andric switch (config->ekind) { 51085868e8aSDimitry Andric case ELFNoneKind: 51185868e8aSDimitry Andric llvm_unreachable("invalid ekind"); 51285868e8aSDimitry Andric case ELF32LEKind: 51385868e8aSDimitry Andric for (EhInputSection *sec : sections) 51485868e8aSDimitry Andric addSectionAux<ELF32LE>(sec); 51585868e8aSDimitry Andric break; 51685868e8aSDimitry Andric case ELF32BEKind: 51785868e8aSDimitry Andric for (EhInputSection *sec : sections) 51885868e8aSDimitry Andric addSectionAux<ELF32BE>(sec); 51985868e8aSDimitry Andric break; 52085868e8aSDimitry Andric case ELF64LEKind: 52185868e8aSDimitry Andric for (EhInputSection *sec : sections) 52285868e8aSDimitry Andric addSectionAux<ELF64LE>(sec); 52385868e8aSDimitry Andric break; 52485868e8aSDimitry Andric case ELF64BEKind: 52585868e8aSDimitry Andric for (EhInputSection *sec : sections) 52685868e8aSDimitry Andric addSectionAux<ELF64BE>(sec); 52785868e8aSDimitry Andric break; 52885868e8aSDimitry Andric } 52985868e8aSDimitry Andric 5300b57cec5SDimitry Andric size_t off = 0; 5310b57cec5SDimitry Andric for (CieRecord *rec : cieRecords) { 5320b57cec5SDimitry Andric rec->cie->outputOff = off; 533bdd1243dSDimitry Andric off += rec->cie->size; 5340b57cec5SDimitry Andric 5350b57cec5SDimitry Andric for (EhSectionPiece *fde : rec->fdes) { 5360b57cec5SDimitry Andric fde->outputOff = off; 537bdd1243dSDimitry Andric off += fde->size; 5380b57cec5SDimitry Andric } 5390b57cec5SDimitry Andric } 5400b57cec5SDimitry Andric 5410b57cec5SDimitry Andric // The LSB standard does not allow a .eh_frame section with zero 5420b57cec5SDimitry Andric // Call Frame Information records. glibc unwind-dw2-fde.c 5430b57cec5SDimitry Andric // classify_object_over_fdes expects there is a CIE record length 0 as a 5440b57cec5SDimitry Andric // terminator. Thus we add one unconditionally. 5450b57cec5SDimitry Andric off += 4; 5460b57cec5SDimitry Andric 5470b57cec5SDimitry Andric this->size = off; 5480b57cec5SDimitry Andric } 5490b57cec5SDimitry Andric 5500b57cec5SDimitry Andric // Returns data for .eh_frame_hdr. .eh_frame_hdr is a binary search table 5510b57cec5SDimitry Andric // to get an FDE from an address to which FDE is applied. This function 5520b57cec5SDimitry Andric // returns a list of such pairs. 55304eeddc0SDimitry Andric SmallVector<EhFrameSection::FdeData, 0> EhFrameSection::getFdeData() const { 5540b57cec5SDimitry Andric uint8_t *buf = Out::bufferStart + getParent()->offset + outSecOff; 55504eeddc0SDimitry Andric SmallVector<FdeData, 0> ret; 5560b57cec5SDimitry Andric 5570b57cec5SDimitry Andric uint64_t va = getPartition().ehFrameHdr->getVA(); 5580b57cec5SDimitry Andric for (CieRecord *rec : cieRecords) { 5590b57cec5SDimitry Andric uint8_t enc = getFdeEncoding(rec->cie); 5600b57cec5SDimitry Andric for (EhSectionPiece *fde : rec->fdes) { 5610b57cec5SDimitry Andric uint64_t pc = getFdePc(buf, fde->outputOff, enc); 5620b57cec5SDimitry Andric uint64_t fdeVA = getParent()->addr + fde->outputOff; 563*0fca6ea1SDimitry Andric if (!isInt<32>(pc - va)) { 564*0fca6ea1SDimitry Andric errorOrWarn(toString(fde->sec) + ": PC offset is too large: 0x" + 5650b57cec5SDimitry Andric Twine::utohexstr(pc - va)); 566*0fca6ea1SDimitry Andric continue; 567*0fca6ea1SDimitry Andric } 5680b57cec5SDimitry Andric ret.push_back({uint32_t(pc - va), uint32_t(fdeVA - va)}); 5690b57cec5SDimitry Andric } 5700b57cec5SDimitry Andric } 5710b57cec5SDimitry Andric 5720b57cec5SDimitry Andric // Sort the FDE list by their PC and uniqueify. Usually there is only 5730b57cec5SDimitry Andric // one FDE for a PC (i.e. function), but if ICF merges two functions 5740b57cec5SDimitry Andric // into one, there can be more than one FDEs pointing to the address. 5750b57cec5SDimitry Andric auto less = [](const FdeData &a, const FdeData &b) { 5760b57cec5SDimitry Andric return a.pcRel < b.pcRel; 5770b57cec5SDimitry Andric }; 5780b57cec5SDimitry Andric llvm::stable_sort(ret, less); 5790b57cec5SDimitry Andric auto eq = [](const FdeData &a, const FdeData &b) { 5800b57cec5SDimitry Andric return a.pcRel == b.pcRel; 5810b57cec5SDimitry Andric }; 5820b57cec5SDimitry Andric ret.erase(std::unique(ret.begin(), ret.end(), eq), ret.end()); 5830b57cec5SDimitry Andric 5840b57cec5SDimitry Andric return ret; 5850b57cec5SDimitry Andric } 5860b57cec5SDimitry Andric 5870b57cec5SDimitry Andric static uint64_t readFdeAddr(uint8_t *buf, int size) { 5880b57cec5SDimitry Andric switch (size) { 5890b57cec5SDimitry Andric case DW_EH_PE_udata2: 5900b57cec5SDimitry Andric return read16(buf); 5910b57cec5SDimitry Andric case DW_EH_PE_sdata2: 5920b57cec5SDimitry Andric return (int16_t)read16(buf); 5930b57cec5SDimitry Andric case DW_EH_PE_udata4: 5940b57cec5SDimitry Andric return read32(buf); 5950b57cec5SDimitry Andric case DW_EH_PE_sdata4: 5960b57cec5SDimitry Andric return (int32_t)read32(buf); 5970b57cec5SDimitry Andric case DW_EH_PE_udata8: 5980b57cec5SDimitry Andric case DW_EH_PE_sdata8: 5990b57cec5SDimitry Andric return read64(buf); 6000b57cec5SDimitry Andric case DW_EH_PE_absptr: 6010b57cec5SDimitry Andric return readUint(buf); 6020b57cec5SDimitry Andric } 6030b57cec5SDimitry Andric fatal("unknown FDE size encoding"); 6040b57cec5SDimitry Andric } 6050b57cec5SDimitry Andric 6060b57cec5SDimitry Andric // Returns the VA to which a given FDE (on a mmap'ed buffer) is applied to. 6070b57cec5SDimitry Andric // We need it to create .eh_frame_hdr section. 6080b57cec5SDimitry Andric uint64_t EhFrameSection::getFdePc(uint8_t *buf, size_t fdeOff, 6090b57cec5SDimitry Andric uint8_t enc) const { 6100b57cec5SDimitry Andric // The starting address to which this FDE applies is 6115f757f3fSDimitry Andric // stored at FDE + 8 byte. And this offset is within 6125f757f3fSDimitry Andric // the .eh_frame section. 6130b57cec5SDimitry Andric size_t off = fdeOff + 8; 6140b57cec5SDimitry Andric uint64_t addr = readFdeAddr(buf + off, enc & 0xf); 6150b57cec5SDimitry Andric if ((enc & 0x70) == DW_EH_PE_absptr) 616*0fca6ea1SDimitry Andric return config->is64 ? addr : uint32_t(addr); 6170b57cec5SDimitry Andric if ((enc & 0x70) == DW_EH_PE_pcrel) 6185f757f3fSDimitry Andric return addr + getParent()->addr + off + outSecOff; 6190b57cec5SDimitry Andric fatal("unknown FDE size relative encoding"); 6200b57cec5SDimitry Andric } 6210b57cec5SDimitry Andric 6220b57cec5SDimitry Andric void EhFrameSection::writeTo(uint8_t *buf) { 6230b57cec5SDimitry Andric // Write CIE and FDE records. 6240b57cec5SDimitry Andric for (CieRecord *rec : cieRecords) { 6250b57cec5SDimitry Andric size_t cieOffset = rec->cie->outputOff; 6260b57cec5SDimitry Andric writeCieFde(buf + cieOffset, rec->cie->data()); 6270b57cec5SDimitry Andric 6280b57cec5SDimitry Andric for (EhSectionPiece *fde : rec->fdes) { 6290b57cec5SDimitry Andric size_t off = fde->outputOff; 6300b57cec5SDimitry Andric writeCieFde(buf + off, fde->data()); 6310b57cec5SDimitry Andric 6320b57cec5SDimitry Andric // FDE's second word should have the offset to an associated CIE. 6330b57cec5SDimitry Andric // Write it. 6340b57cec5SDimitry Andric write32(buf + off + 4, off + 4 - cieOffset); 6350b57cec5SDimitry Andric } 6360b57cec5SDimitry Andric } 6370b57cec5SDimitry Andric 6380b57cec5SDimitry Andric // Apply relocations. .eh_frame section contents are not contiguous 6390b57cec5SDimitry Andric // in the output buffer, but relocateAlloc() still works because 6400b57cec5SDimitry Andric // getOffset() takes care of discontiguous section pieces. 6410b57cec5SDimitry Andric for (EhInputSection *s : sections) 642bdd1243dSDimitry Andric target->relocateAlloc(*s, buf); 6430b57cec5SDimitry Andric 6440b57cec5SDimitry Andric if (getPartition().ehFrameHdr && getPartition().ehFrameHdr->getParent()) 6450b57cec5SDimitry Andric getPartition().ehFrameHdr->write(); 6460b57cec5SDimitry Andric } 6470b57cec5SDimitry Andric 6480b57cec5SDimitry Andric GotSection::GotSection() 649fe6060f1SDimitry Andric : SyntheticSection(SHF_ALLOC | SHF_WRITE, SHT_PROGBITS, 650fe6060f1SDimitry Andric target->gotEntrySize, ".got") { 651fe6060f1SDimitry Andric numEntries = target->gotHeaderEntriesNum; 6520b57cec5SDimitry Andric } 6530b57cec5SDimitry Andric 654bdd1243dSDimitry Andric void GotSection::addConstant(const Relocation &r) { relocations.push_back(r); } 655*0fca6ea1SDimitry Andric void GotSection::addEntry(const Symbol &sym) { 65604eeddc0SDimitry Andric assert(sym.auxIdx == symAux.size() - 1); 65704eeddc0SDimitry Andric symAux.back().gotIdx = numEntries++; 65804eeddc0SDimitry Andric } 65904eeddc0SDimitry Andric 660*0fca6ea1SDimitry Andric bool GotSection::addTlsDescEntry(const Symbol &sym) { 66104eeddc0SDimitry Andric assert(sym.auxIdx == symAux.size() - 1); 66204eeddc0SDimitry Andric symAux.back().tlsDescIdx = numEntries; 66304eeddc0SDimitry Andric numEntries += 2; 66404eeddc0SDimitry Andric return true; 6650b57cec5SDimitry Andric } 6660b57cec5SDimitry Andric 667*0fca6ea1SDimitry Andric bool GotSection::addDynTlsEntry(const Symbol &sym) { 66804eeddc0SDimitry Andric assert(sym.auxIdx == symAux.size() - 1); 66904eeddc0SDimitry Andric symAux.back().tlsGdIdx = numEntries; 6700b57cec5SDimitry Andric // Global Dynamic TLS entries take two GOT slots. 6710b57cec5SDimitry Andric numEntries += 2; 6720b57cec5SDimitry Andric return true; 6730b57cec5SDimitry Andric } 6740b57cec5SDimitry Andric 6750b57cec5SDimitry Andric // Reserves TLS entries for a TLS module ID and a TLS block offset. 6760b57cec5SDimitry Andric // In total it takes two GOT slots. 6770b57cec5SDimitry Andric bool GotSection::addTlsIndex() { 6780b57cec5SDimitry Andric if (tlsIndexOff != uint32_t(-1)) 6790b57cec5SDimitry Andric return false; 6800b57cec5SDimitry Andric tlsIndexOff = numEntries * config->wordsize; 6810b57cec5SDimitry Andric numEntries += 2; 6820b57cec5SDimitry Andric return true; 6830b57cec5SDimitry Andric } 6840b57cec5SDimitry Andric 68504eeddc0SDimitry Andric uint32_t GotSection::getTlsDescOffset(const Symbol &sym) const { 68604eeddc0SDimitry Andric return sym.getTlsDescIdx() * config->wordsize; 68704eeddc0SDimitry Andric } 68804eeddc0SDimitry Andric 68904eeddc0SDimitry Andric uint64_t GotSection::getTlsDescAddr(const Symbol &sym) const { 69004eeddc0SDimitry Andric return getVA() + getTlsDescOffset(sym); 69104eeddc0SDimitry Andric } 69204eeddc0SDimitry Andric 6930b57cec5SDimitry Andric uint64_t GotSection::getGlobalDynAddr(const Symbol &b) const { 69404eeddc0SDimitry Andric return this->getVA() + b.getTlsGdIdx() * config->wordsize; 6950b57cec5SDimitry Andric } 6960b57cec5SDimitry Andric 6970b57cec5SDimitry Andric uint64_t GotSection::getGlobalDynOffset(const Symbol &b) const { 69804eeddc0SDimitry Andric return b.getTlsGdIdx() * config->wordsize; 6990b57cec5SDimitry Andric } 7000b57cec5SDimitry Andric 7010b57cec5SDimitry Andric void GotSection::finalizeContents() { 702fe6060f1SDimitry Andric if (config->emachine == EM_PPC64 && 703fe6060f1SDimitry Andric numEntries <= target->gotHeaderEntriesNum && !ElfSym::globalOffsetTable) 704fe6060f1SDimitry Andric size = 0; 705fe6060f1SDimitry Andric else 7060b57cec5SDimitry Andric size = numEntries * config->wordsize; 7070b57cec5SDimitry Andric } 7080b57cec5SDimitry Andric 7090b57cec5SDimitry Andric bool GotSection::isNeeded() const { 710fe6060f1SDimitry Andric // Needed if the GOT symbol is used or the number of entries is more than just 711fe6060f1SDimitry Andric // the header. A GOT with just the header may not be needed. 712fe6060f1SDimitry Andric return hasGotOffRel || numEntries > target->gotHeaderEntriesNum; 7130b57cec5SDimitry Andric } 7140b57cec5SDimitry Andric 7150b57cec5SDimitry Andric void GotSection::writeTo(uint8_t *buf) { 71661cfbce3SDimitry Andric // On PPC64 .got may be needed but empty. Skip the write. 71761cfbce3SDimitry Andric if (size == 0) 71861cfbce3SDimitry Andric return; 7190b57cec5SDimitry Andric target->writeGotHeader(buf); 720bdd1243dSDimitry Andric target->relocateAlloc(*this, buf); 7210b57cec5SDimitry Andric } 7220b57cec5SDimitry Andric 7230b57cec5SDimitry Andric static uint64_t getMipsPageAddr(uint64_t addr) { 7240b57cec5SDimitry Andric return (addr + 0x8000) & ~0xffff; 7250b57cec5SDimitry Andric } 7260b57cec5SDimitry Andric 7270b57cec5SDimitry Andric static uint64_t getMipsPageCount(uint64_t size) { 7280b57cec5SDimitry Andric return (size + 0xfffe) / 0xffff + 1; 7290b57cec5SDimitry Andric } 7300b57cec5SDimitry Andric 7310b57cec5SDimitry Andric MipsGotSection::MipsGotSection() 7320b57cec5SDimitry Andric : SyntheticSection(SHF_ALLOC | SHF_WRITE | SHF_MIPS_GPREL, SHT_PROGBITS, 16, 7330b57cec5SDimitry Andric ".got") {} 7340b57cec5SDimitry Andric 7350b57cec5SDimitry Andric void MipsGotSection::addEntry(InputFile &file, Symbol &sym, int64_t addend, 7360b57cec5SDimitry Andric RelExpr expr) { 7370b57cec5SDimitry Andric FileGot &g = getGot(file); 7380b57cec5SDimitry Andric if (expr == R_MIPS_GOT_LOCAL_PAGE) { 7390b57cec5SDimitry Andric if (const OutputSection *os = sym.getOutputSection()) 7400b57cec5SDimitry Andric g.pagesMap.insert({os, {}}); 7410b57cec5SDimitry Andric else 7420b57cec5SDimitry Andric g.local16.insert({{nullptr, getMipsPageAddr(sym.getVA(addend))}, 0}); 7430b57cec5SDimitry Andric } else if (sym.isTls()) 7440b57cec5SDimitry Andric g.tls.insert({&sym, 0}); 7450b57cec5SDimitry Andric else if (sym.isPreemptible && expr == R_ABS) 7460b57cec5SDimitry Andric g.relocs.insert({&sym, 0}); 7470b57cec5SDimitry Andric else if (sym.isPreemptible) 7480b57cec5SDimitry Andric g.global.insert({&sym, 0}); 7490b57cec5SDimitry Andric else if (expr == R_MIPS_GOT_OFF32) 7500b57cec5SDimitry Andric g.local32.insert({{&sym, addend}, 0}); 7510b57cec5SDimitry Andric else 7520b57cec5SDimitry Andric g.local16.insert({{&sym, addend}, 0}); 7530b57cec5SDimitry Andric } 7540b57cec5SDimitry Andric 7550b57cec5SDimitry Andric void MipsGotSection::addDynTlsEntry(InputFile &file, Symbol &sym) { 7560b57cec5SDimitry Andric getGot(file).dynTlsSymbols.insert({&sym, 0}); 7570b57cec5SDimitry Andric } 7580b57cec5SDimitry Andric 7590b57cec5SDimitry Andric void MipsGotSection::addTlsIndex(InputFile &file) { 7600b57cec5SDimitry Andric getGot(file).dynTlsSymbols.insert({nullptr, 0}); 7610b57cec5SDimitry Andric } 7620b57cec5SDimitry Andric 7630b57cec5SDimitry Andric size_t MipsGotSection::FileGot::getEntriesNum() const { 7640b57cec5SDimitry Andric return getPageEntriesNum() + local16.size() + global.size() + relocs.size() + 7650b57cec5SDimitry Andric tls.size() + dynTlsSymbols.size() * 2; 7660b57cec5SDimitry Andric } 7670b57cec5SDimitry Andric 7680b57cec5SDimitry Andric size_t MipsGotSection::FileGot::getPageEntriesNum() const { 7690b57cec5SDimitry Andric size_t num = 0; 7700b57cec5SDimitry Andric for (const std::pair<const OutputSection *, FileGot::PageBlock> &p : pagesMap) 7710b57cec5SDimitry Andric num += p.second.count; 7720b57cec5SDimitry Andric return num; 7730b57cec5SDimitry Andric } 7740b57cec5SDimitry Andric 7750b57cec5SDimitry Andric size_t MipsGotSection::FileGot::getIndexedEntriesNum() const { 7760b57cec5SDimitry Andric size_t count = getPageEntriesNum() + local16.size() + global.size(); 7770b57cec5SDimitry Andric // If there are relocation-only entries in the GOT, TLS entries 7780b57cec5SDimitry Andric // are allocated after them. TLS entries should be addressable 7790b57cec5SDimitry Andric // by 16-bit index so count both reloc-only and TLS entries. 7800b57cec5SDimitry Andric if (!tls.empty() || !dynTlsSymbols.empty()) 7810b57cec5SDimitry Andric count += relocs.size() + tls.size() + dynTlsSymbols.size() * 2; 7820b57cec5SDimitry Andric return count; 7830b57cec5SDimitry Andric } 7840b57cec5SDimitry Andric 7850b57cec5SDimitry Andric MipsGotSection::FileGot &MipsGotSection::getGot(InputFile &f) { 7860eae32dcSDimitry Andric if (f.mipsGotIndex == uint32_t(-1)) { 7870b57cec5SDimitry Andric gots.emplace_back(); 7880b57cec5SDimitry Andric gots.back().file = &f; 7890b57cec5SDimitry Andric f.mipsGotIndex = gots.size() - 1; 7900b57cec5SDimitry Andric } 7910eae32dcSDimitry Andric return gots[f.mipsGotIndex]; 7920b57cec5SDimitry Andric } 7930b57cec5SDimitry Andric 7940b57cec5SDimitry Andric uint64_t MipsGotSection::getPageEntryOffset(const InputFile *f, 7950b57cec5SDimitry Andric const Symbol &sym, 7960b57cec5SDimitry Andric int64_t addend) const { 7970eae32dcSDimitry Andric const FileGot &g = gots[f->mipsGotIndex]; 7980b57cec5SDimitry Andric uint64_t index = 0; 7990b57cec5SDimitry Andric if (const OutputSection *outSec = sym.getOutputSection()) { 8000b57cec5SDimitry Andric uint64_t secAddr = getMipsPageAddr(outSec->addr); 8010b57cec5SDimitry Andric uint64_t symAddr = getMipsPageAddr(sym.getVA(addend)); 8020b57cec5SDimitry Andric index = g.pagesMap.lookup(outSec).firstIndex + (symAddr - secAddr) / 0xffff; 8030b57cec5SDimitry Andric } else { 8040b57cec5SDimitry Andric index = g.local16.lookup({nullptr, getMipsPageAddr(sym.getVA(addend))}); 8050b57cec5SDimitry Andric } 8060b57cec5SDimitry Andric return index * config->wordsize; 8070b57cec5SDimitry Andric } 8080b57cec5SDimitry Andric 8090b57cec5SDimitry Andric uint64_t MipsGotSection::getSymEntryOffset(const InputFile *f, const Symbol &s, 8100b57cec5SDimitry Andric int64_t addend) const { 8110eae32dcSDimitry Andric const FileGot &g = gots[f->mipsGotIndex]; 8120b57cec5SDimitry Andric Symbol *sym = const_cast<Symbol *>(&s); 8130b57cec5SDimitry Andric if (sym->isTls()) 8140b57cec5SDimitry Andric return g.tls.lookup(sym) * config->wordsize; 8150b57cec5SDimitry Andric if (sym->isPreemptible) 8160b57cec5SDimitry Andric return g.global.lookup(sym) * config->wordsize; 8170b57cec5SDimitry Andric return g.local16.lookup({sym, addend}) * config->wordsize; 8180b57cec5SDimitry Andric } 8190b57cec5SDimitry Andric 8200b57cec5SDimitry Andric uint64_t MipsGotSection::getTlsIndexOffset(const InputFile *f) const { 8210eae32dcSDimitry Andric const FileGot &g = gots[f->mipsGotIndex]; 8220b57cec5SDimitry Andric return g.dynTlsSymbols.lookup(nullptr) * config->wordsize; 8230b57cec5SDimitry Andric } 8240b57cec5SDimitry Andric 8250b57cec5SDimitry Andric uint64_t MipsGotSection::getGlobalDynOffset(const InputFile *f, 8260b57cec5SDimitry Andric const Symbol &s) const { 8270eae32dcSDimitry Andric const FileGot &g = gots[f->mipsGotIndex]; 8280b57cec5SDimitry Andric Symbol *sym = const_cast<Symbol *>(&s); 8290b57cec5SDimitry Andric return g.dynTlsSymbols.lookup(sym) * config->wordsize; 8300b57cec5SDimitry Andric } 8310b57cec5SDimitry Andric 8320b57cec5SDimitry Andric const Symbol *MipsGotSection::getFirstGlobalEntry() const { 8330b57cec5SDimitry Andric if (gots.empty()) 8340b57cec5SDimitry Andric return nullptr; 8350b57cec5SDimitry Andric const FileGot &primGot = gots.front(); 8360b57cec5SDimitry Andric if (!primGot.global.empty()) 8370b57cec5SDimitry Andric return primGot.global.front().first; 8380b57cec5SDimitry Andric if (!primGot.relocs.empty()) 8390b57cec5SDimitry Andric return primGot.relocs.front().first; 8400b57cec5SDimitry Andric return nullptr; 8410b57cec5SDimitry Andric } 8420b57cec5SDimitry Andric 8430b57cec5SDimitry Andric unsigned MipsGotSection::getLocalEntriesNum() const { 8440b57cec5SDimitry Andric if (gots.empty()) 8450b57cec5SDimitry Andric return headerEntriesNum; 8460b57cec5SDimitry Andric return headerEntriesNum + gots.front().getPageEntriesNum() + 8470b57cec5SDimitry Andric gots.front().local16.size(); 8480b57cec5SDimitry Andric } 8490b57cec5SDimitry Andric 8500b57cec5SDimitry Andric bool MipsGotSection::tryMergeGots(FileGot &dst, FileGot &src, bool isPrimary) { 8510b57cec5SDimitry Andric FileGot tmp = dst; 8520b57cec5SDimitry Andric set_union(tmp.pagesMap, src.pagesMap); 8530b57cec5SDimitry Andric set_union(tmp.local16, src.local16); 8540b57cec5SDimitry Andric set_union(tmp.global, src.global); 8550b57cec5SDimitry Andric set_union(tmp.relocs, src.relocs); 8560b57cec5SDimitry Andric set_union(tmp.tls, src.tls); 8570b57cec5SDimitry Andric set_union(tmp.dynTlsSymbols, src.dynTlsSymbols); 8580b57cec5SDimitry Andric 8590b57cec5SDimitry Andric size_t count = isPrimary ? headerEntriesNum : 0; 8600b57cec5SDimitry Andric count += tmp.getIndexedEntriesNum(); 8610b57cec5SDimitry Andric 8620b57cec5SDimitry Andric if (count * config->wordsize > config->mipsGotSize) 8630b57cec5SDimitry Andric return false; 8640b57cec5SDimitry Andric 8650b57cec5SDimitry Andric std::swap(tmp, dst); 8660b57cec5SDimitry Andric return true; 8670b57cec5SDimitry Andric } 8680b57cec5SDimitry Andric 8690b57cec5SDimitry Andric void MipsGotSection::finalizeContents() { updateAllocSize(); } 8700b57cec5SDimitry Andric 8710b57cec5SDimitry Andric bool MipsGotSection::updateAllocSize() { 8720b57cec5SDimitry Andric size = headerEntriesNum * config->wordsize; 8730b57cec5SDimitry Andric for (const FileGot &g : gots) 8740b57cec5SDimitry Andric size += g.getEntriesNum() * config->wordsize; 8750b57cec5SDimitry Andric return false; 8760b57cec5SDimitry Andric } 8770b57cec5SDimitry Andric 8780b57cec5SDimitry Andric void MipsGotSection::build() { 8790b57cec5SDimitry Andric if (gots.empty()) 8800b57cec5SDimitry Andric return; 8810b57cec5SDimitry Andric 8820b57cec5SDimitry Andric std::vector<FileGot> mergedGots(1); 8830b57cec5SDimitry Andric 8840b57cec5SDimitry Andric // For each GOT move non-preemptible symbols from the `Global` 8850b57cec5SDimitry Andric // to `Local16` list. Preemptible symbol might become non-preemptible 8860b57cec5SDimitry Andric // one if, for example, it gets a related copy relocation. 8870b57cec5SDimitry Andric for (FileGot &got : gots) { 8880b57cec5SDimitry Andric for (auto &p: got.global) 8890b57cec5SDimitry Andric if (!p.first->isPreemptible) 8900b57cec5SDimitry Andric got.local16.insert({{p.first, 0}, 0}); 8910b57cec5SDimitry Andric got.global.remove_if([&](const std::pair<Symbol *, size_t> &p) { 8920b57cec5SDimitry Andric return !p.first->isPreemptible; 8930b57cec5SDimitry Andric }); 8940b57cec5SDimitry Andric } 8950b57cec5SDimitry Andric 8960b57cec5SDimitry Andric // For each GOT remove "reloc-only" entry if there is "global" 8970b57cec5SDimitry Andric // entry for the same symbol. And add local entries which indexed 8980b57cec5SDimitry Andric // using 32-bit value at the end of 16-bit entries. 8990b57cec5SDimitry Andric for (FileGot &got : gots) { 9000b57cec5SDimitry Andric got.relocs.remove_if([&](const std::pair<Symbol *, size_t> &p) { 9010b57cec5SDimitry Andric return got.global.count(p.first); 9020b57cec5SDimitry Andric }); 9030b57cec5SDimitry Andric set_union(got.local16, got.local32); 9040b57cec5SDimitry Andric got.local32.clear(); 9050b57cec5SDimitry Andric } 9060b57cec5SDimitry Andric 9070b57cec5SDimitry Andric // Evaluate number of "reloc-only" entries in the resulting GOT. 9080b57cec5SDimitry Andric // To do that put all unique "reloc-only" and "global" entries 9090b57cec5SDimitry Andric // from all GOTs to the future primary GOT. 9100b57cec5SDimitry Andric FileGot *primGot = &mergedGots.front(); 9110b57cec5SDimitry Andric for (FileGot &got : gots) { 9120b57cec5SDimitry Andric set_union(primGot->relocs, got.global); 9130b57cec5SDimitry Andric set_union(primGot->relocs, got.relocs); 9140b57cec5SDimitry Andric got.relocs.clear(); 9150b57cec5SDimitry Andric } 9160b57cec5SDimitry Andric 9170b57cec5SDimitry Andric // Evaluate number of "page" entries in each GOT. 9180b57cec5SDimitry Andric for (FileGot &got : gots) { 9190b57cec5SDimitry Andric for (std::pair<const OutputSection *, FileGot::PageBlock> &p : 9200b57cec5SDimitry Andric got.pagesMap) { 9210b57cec5SDimitry Andric const OutputSection *os = p.first; 9220b57cec5SDimitry Andric uint64_t secSize = 0; 9234824e7fdSDimitry Andric for (SectionCommand *cmd : os->commands) { 9240b57cec5SDimitry Andric if (auto *isd = dyn_cast<InputSectionDescription>(cmd)) 9250b57cec5SDimitry Andric for (InputSection *isec : isd->sections) { 926bdd1243dSDimitry Andric uint64_t off = alignToPowerOf2(secSize, isec->addralign); 9270b57cec5SDimitry Andric secSize = off + isec->getSize(); 9280b57cec5SDimitry Andric } 9290b57cec5SDimitry Andric } 9300b57cec5SDimitry Andric p.second.count = getMipsPageCount(secSize); 9310b57cec5SDimitry Andric } 9320b57cec5SDimitry Andric } 9330b57cec5SDimitry Andric 9340b57cec5SDimitry Andric // Merge GOTs. Try to join as much as possible GOTs but do not exceed 9350b57cec5SDimitry Andric // maximum GOT size. At first, try to fill the primary GOT because 9360b57cec5SDimitry Andric // the primary GOT can be accessed in the most effective way. If it 9370b57cec5SDimitry Andric // is not possible, try to fill the last GOT in the list, and finally 9380b57cec5SDimitry Andric // create a new GOT if both attempts failed. 9390b57cec5SDimitry Andric for (FileGot &srcGot : gots) { 9400b57cec5SDimitry Andric InputFile *file = srcGot.file; 9410b57cec5SDimitry Andric if (tryMergeGots(mergedGots.front(), srcGot, true)) { 9420b57cec5SDimitry Andric file->mipsGotIndex = 0; 9430b57cec5SDimitry Andric } else { 9440b57cec5SDimitry Andric // If this is the first time we failed to merge with the primary GOT, 9450b57cec5SDimitry Andric // MergedGots.back() will also be the primary GOT. We must make sure not 9460b57cec5SDimitry Andric // to try to merge again with isPrimary=false, as otherwise, if the 9470b57cec5SDimitry Andric // inputs are just right, we could allow the primary GOT to become 1 or 2 9480b57cec5SDimitry Andric // words bigger due to ignoring the header size. 9490b57cec5SDimitry Andric if (mergedGots.size() == 1 || 9500b57cec5SDimitry Andric !tryMergeGots(mergedGots.back(), srcGot, false)) { 9510b57cec5SDimitry Andric mergedGots.emplace_back(); 9520b57cec5SDimitry Andric std::swap(mergedGots.back(), srcGot); 9530b57cec5SDimitry Andric } 9540b57cec5SDimitry Andric file->mipsGotIndex = mergedGots.size() - 1; 9550b57cec5SDimitry Andric } 9560b57cec5SDimitry Andric } 9570b57cec5SDimitry Andric std::swap(gots, mergedGots); 9580b57cec5SDimitry Andric 9590b57cec5SDimitry Andric // Reduce number of "reloc-only" entries in the primary GOT 960480093f4SDimitry Andric // by subtracting "global" entries in the primary GOT. 9610b57cec5SDimitry Andric primGot = &gots.front(); 9620b57cec5SDimitry Andric primGot->relocs.remove_if([&](const std::pair<Symbol *, size_t> &p) { 9630b57cec5SDimitry Andric return primGot->global.count(p.first); 9640b57cec5SDimitry Andric }); 9650b57cec5SDimitry Andric 9660b57cec5SDimitry Andric // Calculate indexes for each GOT entry. 9670b57cec5SDimitry Andric size_t index = headerEntriesNum; 9680b57cec5SDimitry Andric for (FileGot &got : gots) { 9690b57cec5SDimitry Andric got.startIndex = &got == primGot ? 0 : index; 9700b57cec5SDimitry Andric for (std::pair<const OutputSection *, FileGot::PageBlock> &p : 9710b57cec5SDimitry Andric got.pagesMap) { 9720b57cec5SDimitry Andric // For each output section referenced by GOT page relocations calculate 9730b57cec5SDimitry Andric // and save into pagesMap an upper bound of MIPS GOT entries required 9740b57cec5SDimitry Andric // to store page addresses of local symbols. We assume the worst case - 9750b57cec5SDimitry Andric // each 64kb page of the output section has at least one GOT relocation 9760b57cec5SDimitry Andric // against it. And take in account the case when the section intersects 9770b57cec5SDimitry Andric // page boundaries. 9780b57cec5SDimitry Andric p.second.firstIndex = index; 9790b57cec5SDimitry Andric index += p.second.count; 9800b57cec5SDimitry Andric } 9810b57cec5SDimitry Andric for (auto &p: got.local16) 9820b57cec5SDimitry Andric p.second = index++; 9830b57cec5SDimitry Andric for (auto &p: got.global) 9840b57cec5SDimitry Andric p.second = index++; 9850b57cec5SDimitry Andric for (auto &p: got.relocs) 9860b57cec5SDimitry Andric p.second = index++; 9870b57cec5SDimitry Andric for (auto &p: got.tls) 9880b57cec5SDimitry Andric p.second = index++; 9890b57cec5SDimitry Andric for (auto &p: got.dynTlsSymbols) { 9900b57cec5SDimitry Andric p.second = index; 9910b57cec5SDimitry Andric index += 2; 9920b57cec5SDimitry Andric } 9930b57cec5SDimitry Andric } 9940b57cec5SDimitry Andric 99504eeddc0SDimitry Andric // Update SymbolAux::gotIdx field to use this 9960b57cec5SDimitry Andric // value later in the `sortMipsSymbols` function. 99704eeddc0SDimitry Andric for (auto &p : primGot->global) { 998bdd1243dSDimitry Andric if (p.first->auxIdx == 0) 99904eeddc0SDimitry Andric p.first->allocateAux(); 100004eeddc0SDimitry Andric symAux.back().gotIdx = p.second; 100104eeddc0SDimitry Andric } 100204eeddc0SDimitry Andric for (auto &p : primGot->relocs) { 1003bdd1243dSDimitry Andric if (p.first->auxIdx == 0) 100404eeddc0SDimitry Andric p.first->allocateAux(); 100504eeddc0SDimitry Andric symAux.back().gotIdx = p.second; 100604eeddc0SDimitry Andric } 10070b57cec5SDimitry Andric 10080b57cec5SDimitry Andric // Create dynamic relocations. 10090b57cec5SDimitry Andric for (FileGot &got : gots) { 10100b57cec5SDimitry Andric // Create dynamic relocations for TLS entries. 10110b57cec5SDimitry Andric for (std::pair<Symbol *, size_t> &p : got.tls) { 10120b57cec5SDimitry Andric Symbol *s = p.first; 10130b57cec5SDimitry Andric uint64_t offset = p.second * config->wordsize; 1014fe6060f1SDimitry Andric // When building a shared library we still need a dynamic relocation 1015fe6060f1SDimitry Andric // for the TP-relative offset as we don't know how much other data will 1016fe6060f1SDimitry Andric // be allocated before us in the static TLS block. 1017fe6060f1SDimitry Andric if (s->isPreemptible || config->shared) 1018fe6060f1SDimitry Andric mainPart->relaDyn->addReloc({target->tlsGotRel, this, offset, 1019fe6060f1SDimitry Andric DynamicReloc::AgainstSymbolWithTargetVA, 1020fe6060f1SDimitry Andric *s, 0, R_ABS}); 10210b57cec5SDimitry Andric } 10220b57cec5SDimitry Andric for (std::pair<Symbol *, size_t> &p : got.dynTlsSymbols) { 10230b57cec5SDimitry Andric Symbol *s = p.first; 10240b57cec5SDimitry Andric uint64_t offset = p.second * config->wordsize; 10250b57cec5SDimitry Andric if (s == nullptr) { 1026fe6060f1SDimitry Andric if (!config->shared) 10270b57cec5SDimitry Andric continue; 1028fe6060f1SDimitry Andric mainPart->relaDyn->addReloc({target->tlsModuleIndexRel, this, offset}); 10290b57cec5SDimitry Andric } else { 10300b57cec5SDimitry Andric // When building a shared library we still need a dynamic relocation 10310b57cec5SDimitry Andric // for the module index. Therefore only checking for 10320b57cec5SDimitry Andric // S->isPreemptible is not sufficient (this happens e.g. for 10330b57cec5SDimitry Andric // thread-locals that have been marked as local through a linker script) 1034fe6060f1SDimitry Andric if (!s->isPreemptible && !config->shared) 10350b57cec5SDimitry Andric continue; 10360eae32dcSDimitry Andric mainPart->relaDyn->addSymbolReloc(target->tlsModuleIndexRel, *this, 1037fe6060f1SDimitry Andric offset, *s); 10380b57cec5SDimitry Andric // However, we can skip writing the TLS offset reloc for non-preemptible 10390b57cec5SDimitry Andric // symbols since it is known even in shared libraries 10400b57cec5SDimitry Andric if (!s->isPreemptible) 10410b57cec5SDimitry Andric continue; 10420b57cec5SDimitry Andric offset += config->wordsize; 10430eae32dcSDimitry Andric mainPart->relaDyn->addSymbolReloc(target->tlsOffsetRel, *this, offset, 1044fe6060f1SDimitry Andric *s); 10450b57cec5SDimitry Andric } 10460b57cec5SDimitry Andric } 10470b57cec5SDimitry Andric 10480b57cec5SDimitry Andric // Do not create dynamic relocations for non-TLS 10490b57cec5SDimitry Andric // entries in the primary GOT. 10500b57cec5SDimitry Andric if (&got == primGot) 10510b57cec5SDimitry Andric continue; 10520b57cec5SDimitry Andric 10530b57cec5SDimitry Andric // Dynamic relocations for "global" entries. 10540b57cec5SDimitry Andric for (const std::pair<Symbol *, size_t> &p : got.global) { 10550b57cec5SDimitry Andric uint64_t offset = p.second * config->wordsize; 10560eae32dcSDimitry Andric mainPart->relaDyn->addSymbolReloc(target->relativeRel, *this, offset, 1057fe6060f1SDimitry Andric *p.first); 10580b57cec5SDimitry Andric } 10590b57cec5SDimitry Andric if (!config->isPic) 10600b57cec5SDimitry Andric continue; 10610b57cec5SDimitry Andric // Dynamic relocations for "local" entries in case of PIC. 10620b57cec5SDimitry Andric for (const std::pair<const OutputSection *, FileGot::PageBlock> &l : 10630b57cec5SDimitry Andric got.pagesMap) { 10640b57cec5SDimitry Andric size_t pageCount = l.second.count; 10650b57cec5SDimitry Andric for (size_t pi = 0; pi < pageCount; ++pi) { 10660b57cec5SDimitry Andric uint64_t offset = (l.second.firstIndex + pi) * config->wordsize; 10670b57cec5SDimitry Andric mainPart->relaDyn->addReloc({target->relativeRel, this, offset, l.first, 10680b57cec5SDimitry Andric int64_t(pi * 0x10000)}); 10690b57cec5SDimitry Andric } 10700b57cec5SDimitry Andric } 10710b57cec5SDimitry Andric for (const std::pair<GotEntry, size_t> &p : got.local16) { 10720b57cec5SDimitry Andric uint64_t offset = p.second * config->wordsize; 1073fe6060f1SDimitry Andric mainPart->relaDyn->addReloc({target->relativeRel, this, offset, 1074fe6060f1SDimitry Andric DynamicReloc::AddendOnlyWithTargetVA, 1075fe6060f1SDimitry Andric *p.first.first, p.first.second, R_ABS}); 10760b57cec5SDimitry Andric } 10770b57cec5SDimitry Andric } 10780b57cec5SDimitry Andric } 10790b57cec5SDimitry Andric 10800b57cec5SDimitry Andric bool MipsGotSection::isNeeded() const { 10810b57cec5SDimitry Andric // We add the .got section to the result for dynamic MIPS target because 10820b57cec5SDimitry Andric // its address and properties are mentioned in the .dynamic section. 10830b57cec5SDimitry Andric return !config->relocatable; 10840b57cec5SDimitry Andric } 10850b57cec5SDimitry Andric 10860b57cec5SDimitry Andric uint64_t MipsGotSection::getGp(const InputFile *f) const { 10870b57cec5SDimitry Andric // For files without related GOT or files refer a primary GOT 10880b57cec5SDimitry Andric // returns "common" _gp value. For secondary GOTs calculate 10890b57cec5SDimitry Andric // individual _gp values. 10900eae32dcSDimitry Andric if (!f || f->mipsGotIndex == uint32_t(-1) || f->mipsGotIndex == 0) 10910b57cec5SDimitry Andric return ElfSym::mipsGp->getVA(0); 10920eae32dcSDimitry Andric return getVA() + gots[f->mipsGotIndex].startIndex * config->wordsize + 0x7ff0; 10930b57cec5SDimitry Andric } 10940b57cec5SDimitry Andric 10950b57cec5SDimitry Andric void MipsGotSection::writeTo(uint8_t *buf) { 10960b57cec5SDimitry Andric // Set the MSB of the second GOT slot. This is not required by any 10970b57cec5SDimitry Andric // MIPS ABI documentation, though. 10980b57cec5SDimitry Andric // 10990b57cec5SDimitry Andric // There is a comment in glibc saying that "The MSB of got[1] of a 11000b57cec5SDimitry Andric // gnu object is set to identify gnu objects," and in GNU gold it 11010b57cec5SDimitry Andric // says "the second entry will be used by some runtime loaders". 11020b57cec5SDimitry Andric // But how this field is being used is unclear. 11030b57cec5SDimitry Andric // 11040b57cec5SDimitry Andric // We are not really willing to mimic other linkers behaviors 11050b57cec5SDimitry Andric // without understanding why they do that, but because all files 11060b57cec5SDimitry Andric // generated by GNU tools have this special GOT value, and because 11070b57cec5SDimitry Andric // we've been doing this for years, it is probably a safe bet to 11080b57cec5SDimitry Andric // keep doing this for now. We really need to revisit this to see 11090b57cec5SDimitry Andric // if we had to do this. 11100b57cec5SDimitry Andric writeUint(buf + config->wordsize, (uint64_t)1 << (config->wordsize * 8 - 1)); 11110b57cec5SDimitry Andric for (const FileGot &g : gots) { 11120b57cec5SDimitry Andric auto write = [&](size_t i, const Symbol *s, int64_t a) { 11130b57cec5SDimitry Andric uint64_t va = a; 11140b57cec5SDimitry Andric if (s) 11150b57cec5SDimitry Andric va = s->getVA(a); 11160b57cec5SDimitry Andric writeUint(buf + i * config->wordsize, va); 11170b57cec5SDimitry Andric }; 11180b57cec5SDimitry Andric // Write 'page address' entries to the local part of the GOT. 11190b57cec5SDimitry Andric for (const std::pair<const OutputSection *, FileGot::PageBlock> &l : 11200b57cec5SDimitry Andric g.pagesMap) { 11210b57cec5SDimitry Andric size_t pageCount = l.second.count; 11220b57cec5SDimitry Andric uint64_t firstPageAddr = getMipsPageAddr(l.first->addr); 11230b57cec5SDimitry Andric for (size_t pi = 0; pi < pageCount; ++pi) 11240b57cec5SDimitry Andric write(l.second.firstIndex + pi, nullptr, firstPageAddr + pi * 0x10000); 11250b57cec5SDimitry Andric } 11260b57cec5SDimitry Andric // Local, global, TLS, reloc-only entries. 11270b57cec5SDimitry Andric // If TLS entry has a corresponding dynamic relocations, leave it 11280b57cec5SDimitry Andric // initialized by zero. Write down adjusted TLS symbol's values otherwise. 11290b57cec5SDimitry Andric // To calculate the adjustments use offsets for thread-local storage. 1130fe6060f1SDimitry Andric // http://web.archive.org/web/20190324223224/https://www.linux-mips.org/wiki/NPTL 11310b57cec5SDimitry Andric for (const std::pair<GotEntry, size_t> &p : g.local16) 11320b57cec5SDimitry Andric write(p.second, p.first.first, p.first.second); 11330b57cec5SDimitry Andric // Write VA to the primary GOT only. For secondary GOTs that 11340b57cec5SDimitry Andric // will be done by REL32 dynamic relocations. 11350b57cec5SDimitry Andric if (&g == &gots.front()) 1136480093f4SDimitry Andric for (const std::pair<Symbol *, size_t> &p : g.global) 11370b57cec5SDimitry Andric write(p.second, p.first, 0); 11380b57cec5SDimitry Andric for (const std::pair<Symbol *, size_t> &p : g.relocs) 11390b57cec5SDimitry Andric write(p.second, p.first, 0); 11400b57cec5SDimitry Andric for (const std::pair<Symbol *, size_t> &p : g.tls) 1141fe6060f1SDimitry Andric write(p.second, p.first, 1142fe6060f1SDimitry Andric p.first->isPreemptible || config->shared ? 0 : -0x7000); 11430b57cec5SDimitry Andric for (const std::pair<Symbol *, size_t> &p : g.dynTlsSymbols) { 1144fe6060f1SDimitry Andric if (p.first == nullptr && !config->shared) 11450b57cec5SDimitry Andric write(p.second, nullptr, 1); 11460b57cec5SDimitry Andric else if (p.first && !p.first->isPreemptible) { 1147349cc55cSDimitry Andric // If we are emitting a shared library with relocations we mustn't write 11480b57cec5SDimitry Andric // anything to the GOT here. When using Elf_Rel relocations the value 11490b57cec5SDimitry Andric // one will be treated as an addend and will cause crashes at runtime 1150fe6060f1SDimitry Andric if (!config->shared) 11510b57cec5SDimitry Andric write(p.second, nullptr, 1); 11520b57cec5SDimitry Andric write(p.second + 1, p.first, -0x8000); 11530b57cec5SDimitry Andric } 11540b57cec5SDimitry Andric } 11550b57cec5SDimitry Andric } 11560b57cec5SDimitry Andric } 11570b57cec5SDimitry Andric 11580b57cec5SDimitry Andric // On PowerPC the .plt section is used to hold the table of function addresses 11590b57cec5SDimitry Andric // instead of the .got.plt, and the type is SHT_NOBITS similar to a .bss 11600b57cec5SDimitry Andric // section. I don't know why we have a BSS style type for the section but it is 1161480093f4SDimitry Andric // consistent across both 64-bit PowerPC ABIs as well as the 32-bit PowerPC ABI. 11620b57cec5SDimitry Andric GotPltSection::GotPltSection() 11630b57cec5SDimitry Andric : SyntheticSection(SHF_ALLOC | SHF_WRITE, SHT_PROGBITS, config->wordsize, 11640b57cec5SDimitry Andric ".got.plt") { 11650b57cec5SDimitry Andric if (config->emachine == EM_PPC) { 11660b57cec5SDimitry Andric name = ".plt"; 11670b57cec5SDimitry Andric } else if (config->emachine == EM_PPC64) { 11680b57cec5SDimitry Andric type = SHT_NOBITS; 11690b57cec5SDimitry Andric name = ".plt"; 11700b57cec5SDimitry Andric } 11710b57cec5SDimitry Andric } 11720b57cec5SDimitry Andric 11730b57cec5SDimitry Andric void GotPltSection::addEntry(Symbol &sym) { 117404eeddc0SDimitry Andric assert(sym.auxIdx == symAux.size() - 1 && 117504eeddc0SDimitry Andric symAux.back().pltIdx == entries.size()); 11760b57cec5SDimitry Andric entries.push_back(&sym); 11770b57cec5SDimitry Andric } 11780b57cec5SDimitry Andric 11790b57cec5SDimitry Andric size_t GotPltSection::getSize() const { 1180fe6060f1SDimitry Andric return (target->gotPltHeaderEntriesNum + entries.size()) * 1181fe6060f1SDimitry Andric target->gotEntrySize; 11820b57cec5SDimitry Andric } 11830b57cec5SDimitry Andric 11840b57cec5SDimitry Andric void GotPltSection::writeTo(uint8_t *buf) { 11850b57cec5SDimitry Andric target->writeGotPltHeader(buf); 1186fe6060f1SDimitry Andric buf += target->gotPltHeaderEntriesNum * target->gotEntrySize; 11870b57cec5SDimitry Andric for (const Symbol *b : entries) { 11880b57cec5SDimitry Andric target->writeGotPlt(buf, *b); 1189fe6060f1SDimitry Andric buf += target->gotEntrySize; 11900b57cec5SDimitry Andric } 11910b57cec5SDimitry Andric } 11920b57cec5SDimitry Andric 11930b57cec5SDimitry Andric bool GotPltSection::isNeeded() const { 11940b57cec5SDimitry Andric // We need to emit GOTPLT even if it's empty if there's a relocation relative 11950b57cec5SDimitry Andric // to it. 11960b57cec5SDimitry Andric return !entries.empty() || hasGotPltOffRel; 11970b57cec5SDimitry Andric } 11980b57cec5SDimitry Andric 11990b57cec5SDimitry Andric static StringRef getIgotPltName() { 12000b57cec5SDimitry Andric // On ARM the IgotPltSection is part of the GotSection. 12010b57cec5SDimitry Andric if (config->emachine == EM_ARM) 12020b57cec5SDimitry Andric return ".got"; 12030b57cec5SDimitry Andric 12040b57cec5SDimitry Andric // On PowerPC64 the GotPltSection is renamed to '.plt' so the IgotPltSection 12050b57cec5SDimitry Andric // needs to be named the same. 12060b57cec5SDimitry Andric if (config->emachine == EM_PPC64) 12070b57cec5SDimitry Andric return ".plt"; 12080b57cec5SDimitry Andric 12090b57cec5SDimitry Andric return ".got.plt"; 12100b57cec5SDimitry Andric } 12110b57cec5SDimitry Andric 12120b57cec5SDimitry Andric // On PowerPC64 the GotPltSection type is SHT_NOBITS so we have to follow suit 12130b57cec5SDimitry Andric // with the IgotPltSection. 12140b57cec5SDimitry Andric IgotPltSection::IgotPltSection() 12150b57cec5SDimitry Andric : SyntheticSection(SHF_ALLOC | SHF_WRITE, 12160b57cec5SDimitry Andric config->emachine == EM_PPC64 ? SHT_NOBITS : SHT_PROGBITS, 1217fe6060f1SDimitry Andric target->gotEntrySize, getIgotPltName()) {} 12180b57cec5SDimitry Andric 12190b57cec5SDimitry Andric void IgotPltSection::addEntry(Symbol &sym) { 122004eeddc0SDimitry Andric assert(symAux.back().pltIdx == entries.size()); 12210b57cec5SDimitry Andric entries.push_back(&sym); 12220b57cec5SDimitry Andric } 12230b57cec5SDimitry Andric 12240b57cec5SDimitry Andric size_t IgotPltSection::getSize() const { 1225fe6060f1SDimitry Andric return entries.size() * target->gotEntrySize; 12260b57cec5SDimitry Andric } 12270b57cec5SDimitry Andric 12280b57cec5SDimitry Andric void IgotPltSection::writeTo(uint8_t *buf) { 12290b57cec5SDimitry Andric for (const Symbol *b : entries) { 12300b57cec5SDimitry Andric target->writeIgotPlt(buf, *b); 1231fe6060f1SDimitry Andric buf += target->gotEntrySize; 12320b57cec5SDimitry Andric } 12330b57cec5SDimitry Andric } 12340b57cec5SDimitry Andric 12350b57cec5SDimitry Andric StringTableSection::StringTableSection(StringRef name, bool dynamic) 12360b57cec5SDimitry Andric : SyntheticSection(dynamic ? (uint64_t)SHF_ALLOC : 0, SHT_STRTAB, 1, name), 12370b57cec5SDimitry Andric dynamic(dynamic) { 12380b57cec5SDimitry Andric // ELF string tables start with a NUL byte. 12391fd87a68SDimitry Andric strings.push_back(""); 1240d781ede6SDimitry Andric stringMap.try_emplace(CachedHashStringRef(""), 0); 12411fd87a68SDimitry Andric size = 1; 12420b57cec5SDimitry Andric } 12430b57cec5SDimitry Andric 12440b57cec5SDimitry Andric // Adds a string to the string table. If `hashIt` is true we hash and check for 12450b57cec5SDimitry Andric // duplicates. It is optional because the name of global symbols are already 12460b57cec5SDimitry Andric // uniqued and hashing them again has a big cost for a small value: uniquing 12470b57cec5SDimitry Andric // them with some other string that happens to be the same. 12480b57cec5SDimitry Andric unsigned StringTableSection::addString(StringRef s, bool hashIt) { 12490b57cec5SDimitry Andric if (hashIt) { 125004eeddc0SDimitry Andric auto r = stringMap.try_emplace(CachedHashStringRef(s), size); 12510b57cec5SDimitry Andric if (!r.second) 12520b57cec5SDimitry Andric return r.first->second; 12530b57cec5SDimitry Andric } 12541fd87a68SDimitry Andric if (s.empty()) 12551fd87a68SDimitry Andric return 0; 12560b57cec5SDimitry Andric unsigned ret = this->size; 12570b57cec5SDimitry Andric this->size = this->size + s.size() + 1; 12580b57cec5SDimitry Andric strings.push_back(s); 12590b57cec5SDimitry Andric return ret; 12600b57cec5SDimitry Andric } 12610b57cec5SDimitry Andric 12620b57cec5SDimitry Andric void StringTableSection::writeTo(uint8_t *buf) { 12630b57cec5SDimitry Andric for (StringRef s : strings) { 12640b57cec5SDimitry Andric memcpy(buf, s.data(), s.size()); 12650b57cec5SDimitry Andric buf[s.size()] = '\0'; 12660b57cec5SDimitry Andric buf += s.size() + 1; 12670b57cec5SDimitry Andric } 12680b57cec5SDimitry Andric } 12690b57cec5SDimitry Andric 127085868e8aSDimitry Andric // Returns the number of entries in .gnu.version_d: the number of 127185868e8aSDimitry Andric // non-VER_NDX_LOCAL-non-VER_NDX_GLOBAL definitions, plus 1. 127285868e8aSDimitry Andric // Note that we don't support vd_cnt > 1 yet. 127385868e8aSDimitry Andric static unsigned getVerDefNum() { 127485868e8aSDimitry Andric return namedVersionDefs().size() + 1; 127585868e8aSDimitry Andric } 12760b57cec5SDimitry Andric 12770b57cec5SDimitry Andric template <class ELFT> 12780b57cec5SDimitry Andric DynamicSection<ELFT>::DynamicSection() 12790b57cec5SDimitry Andric : SyntheticSection(SHF_ALLOC | SHF_WRITE, SHT_DYNAMIC, config->wordsize, 12800b57cec5SDimitry Andric ".dynamic") { 12810b57cec5SDimitry Andric this->entsize = ELFT::Is64Bits ? 16 : 8; 12820b57cec5SDimitry Andric 12830b57cec5SDimitry Andric // .dynamic section is not writable on MIPS and on Fuchsia OS 12840b57cec5SDimitry Andric // which passes -z rodynamic. 12850b57cec5SDimitry Andric // See "Special Section" in Chapter 4 in the following document: 12860b57cec5SDimitry Andric // ftp://www.linux-mips.org/pub/linux/mips/doc/ABI/mipsabi.pdf 12870b57cec5SDimitry Andric if (config->emachine == EM_MIPS || config->zRodynamic) 12880b57cec5SDimitry Andric this->flags = SHF_ALLOC; 12890b57cec5SDimitry Andric } 12900b57cec5SDimitry Andric 129185868e8aSDimitry Andric // The output section .rela.dyn may include these synthetic sections: 129285868e8aSDimitry Andric // 129385868e8aSDimitry Andric // - part.relaDyn 129485868e8aSDimitry Andric // - in.relaPlt: this is included if a linker script places .rela.plt inside 129585868e8aSDimitry Andric // .rela.dyn 129685868e8aSDimitry Andric // 129785868e8aSDimitry Andric // DT_RELASZ is the total size of the included sections. 129804eeddc0SDimitry Andric static uint64_t addRelaSz(const RelocationBaseSection &relaDyn) { 129904eeddc0SDimitry Andric size_t size = relaDyn.getSize(); 130004eeddc0SDimitry Andric if (in.relaPlt->getParent() == relaDyn.getParent()) 130185868e8aSDimitry Andric size += in.relaPlt->getSize(); 130285868e8aSDimitry Andric return size; 130385868e8aSDimitry Andric } 130485868e8aSDimitry Andric 13050b57cec5SDimitry Andric // A Linker script may assign the RELA relocation sections to the same 13060b57cec5SDimitry Andric // output section. When this occurs we cannot just use the OutputSection 13070b57cec5SDimitry Andric // Size. Moreover the [DT_JMPREL, DT_JMPREL + DT_PLTRELSZ) is permitted to 13080b57cec5SDimitry Andric // overlap with the [DT_RELA, DT_RELA + DT_RELASZ). 1309*0fca6ea1SDimitry Andric static uint64_t addPltRelSz() { return in.relaPlt->getSize(); } 13100b57cec5SDimitry Andric 13110b57cec5SDimitry Andric // Add remaining entries to complete .dynamic contents. 13124824e7fdSDimitry Andric template <class ELFT> 13134824e7fdSDimitry Andric std::vector<std::pair<int32_t, uint64_t>> 13144824e7fdSDimitry Andric DynamicSection<ELFT>::computeContents() { 13155ffd83dbSDimitry Andric elf::Partition &part = getPartition(); 13160b57cec5SDimitry Andric bool isMain = part.name.empty(); 13174824e7fdSDimitry Andric std::vector<std::pair<int32_t, uint64_t>> entries; 13184824e7fdSDimitry Andric 13194824e7fdSDimitry Andric auto addInt = [&](int32_t tag, uint64_t val) { 13204824e7fdSDimitry Andric entries.emplace_back(tag, val); 13214824e7fdSDimitry Andric }; 13220eae32dcSDimitry Andric auto addInSec = [&](int32_t tag, const InputSection &sec) { 13230eae32dcSDimitry Andric entries.emplace_back(tag, sec.getVA()); 13244824e7fdSDimitry Andric }; 13250b57cec5SDimitry Andric 13260b57cec5SDimitry Andric for (StringRef s : config->filterList) 13270b57cec5SDimitry Andric addInt(DT_FILTER, part.dynStrTab->addString(s)); 13280b57cec5SDimitry Andric for (StringRef s : config->auxiliaryList) 13290b57cec5SDimitry Andric addInt(DT_AUXILIARY, part.dynStrTab->addString(s)); 13300b57cec5SDimitry Andric 13310b57cec5SDimitry Andric if (!config->rpath.empty()) 13320b57cec5SDimitry Andric addInt(config->enableNewDtags ? DT_RUNPATH : DT_RPATH, 13330b57cec5SDimitry Andric part.dynStrTab->addString(config->rpath)); 13340b57cec5SDimitry Andric 1335bdd1243dSDimitry Andric for (SharedFile *file : ctx.sharedFiles) 13360b57cec5SDimitry Andric if (file->isNeeded) 13370b57cec5SDimitry Andric addInt(DT_NEEDED, part.dynStrTab->addString(file->soName)); 13380b57cec5SDimitry Andric 13390b57cec5SDimitry Andric if (isMain) { 13400b57cec5SDimitry Andric if (!config->soName.empty()) 13410b57cec5SDimitry Andric addInt(DT_SONAME, part.dynStrTab->addString(config->soName)); 13420b57cec5SDimitry Andric } else { 13430b57cec5SDimitry Andric if (!config->soName.empty()) 13440b57cec5SDimitry Andric addInt(DT_NEEDED, part.dynStrTab->addString(config->soName)); 13450b57cec5SDimitry Andric addInt(DT_SONAME, part.dynStrTab->addString(part.name)); 13460b57cec5SDimitry Andric } 13470b57cec5SDimitry Andric 13480b57cec5SDimitry Andric // Set DT_FLAGS and DT_FLAGS_1. 13490b57cec5SDimitry Andric uint32_t dtFlags = 0; 13500b57cec5SDimitry Andric uint32_t dtFlags1 = 0; 13516e75b2fbSDimitry Andric if (config->bsymbolic == BsymbolicKind::All) 13520b57cec5SDimitry Andric dtFlags |= DF_SYMBOLIC; 13530b57cec5SDimitry Andric if (config->zGlobal) 13540b57cec5SDimitry Andric dtFlags1 |= DF_1_GLOBAL; 13550b57cec5SDimitry Andric if (config->zInitfirst) 13560b57cec5SDimitry Andric dtFlags1 |= DF_1_INITFIRST; 13570b57cec5SDimitry Andric if (config->zInterpose) 13580b57cec5SDimitry Andric dtFlags1 |= DF_1_INTERPOSE; 13590b57cec5SDimitry Andric if (config->zNodefaultlib) 13600b57cec5SDimitry Andric dtFlags1 |= DF_1_NODEFLIB; 13610b57cec5SDimitry Andric if (config->zNodelete) 13620b57cec5SDimitry Andric dtFlags1 |= DF_1_NODELETE; 13630b57cec5SDimitry Andric if (config->zNodlopen) 13640b57cec5SDimitry Andric dtFlags1 |= DF_1_NOOPEN; 1365dfd4db93SEd Maste if (config->pie) 1366dfd4db93SEd Maste dtFlags1 |= DF_1_PIE; 13670b57cec5SDimitry Andric if (config->zNow) { 13680b57cec5SDimitry Andric dtFlags |= DF_BIND_NOW; 13690b57cec5SDimitry Andric dtFlags1 |= DF_1_NOW; 13700b57cec5SDimitry Andric } 13710b57cec5SDimitry Andric if (config->zOrigin) { 13720b57cec5SDimitry Andric dtFlags |= DF_ORIGIN; 13730b57cec5SDimitry Andric dtFlags1 |= DF_1_ORIGIN; 13740b57cec5SDimitry Andric } 13750b57cec5SDimitry Andric if (!config->zText) 13760b57cec5SDimitry Andric dtFlags |= DF_TEXTREL; 1377bdd1243dSDimitry Andric if (ctx.hasTlsIe && config->shared) 13780b57cec5SDimitry Andric dtFlags |= DF_STATIC_TLS; 13790b57cec5SDimitry Andric 13800b57cec5SDimitry Andric if (dtFlags) 13810b57cec5SDimitry Andric addInt(DT_FLAGS, dtFlags); 13820b57cec5SDimitry Andric if (dtFlags1) 13830b57cec5SDimitry Andric addInt(DT_FLAGS_1, dtFlags1); 13840b57cec5SDimitry Andric 1385480093f4SDimitry Andric // DT_DEBUG is a pointer to debug information used by debuggers at runtime. We 13860b57cec5SDimitry Andric // need it for each process, so we don't write it for DSOs. The loader writes 13870b57cec5SDimitry Andric // the pointer into this entry. 13880b57cec5SDimitry Andric // 13890b57cec5SDimitry Andric // DT_DEBUG is the only .dynamic entry that needs to be written to. Some 13900b57cec5SDimitry Andric // systems (currently only Fuchsia OS) provide other means to give the 13910b57cec5SDimitry Andric // debugger this information. Such systems may choose make .dynamic read-only. 13920b57cec5SDimitry Andric // If the target is such a system (used -z rodynamic) don't write DT_DEBUG. 13930b57cec5SDimitry Andric if (!config->shared && !config->relocatable && !config->zRodynamic) 13940b57cec5SDimitry Andric addInt(DT_DEBUG, 0); 13950b57cec5SDimitry Andric 1396*0fca6ea1SDimitry Andric if (part.relaDyn->isNeeded()) { 13970eae32dcSDimitry Andric addInSec(part.relaDyn->dynamicTag, *part.relaDyn); 139804eeddc0SDimitry Andric entries.emplace_back(part.relaDyn->sizeDynamicTag, 139904eeddc0SDimitry Andric addRelaSz(*part.relaDyn)); 14000b57cec5SDimitry Andric 14010b57cec5SDimitry Andric bool isRela = config->isRela; 14020b57cec5SDimitry Andric addInt(isRela ? DT_RELAENT : DT_RELENT, 14030b57cec5SDimitry Andric isRela ? sizeof(Elf_Rela) : sizeof(Elf_Rel)); 14040b57cec5SDimitry Andric 14050b57cec5SDimitry Andric // MIPS dynamic loader does not support RELCOUNT tag. 14060b57cec5SDimitry Andric // The problem is in the tight relation between dynamic 14070b57cec5SDimitry Andric // relocations and GOT. So do not emit this tag on MIPS. 14080b57cec5SDimitry Andric if (config->emachine != EM_MIPS) { 14090b57cec5SDimitry Andric size_t numRelativeRels = part.relaDyn->getRelativeRelocCount(); 14100b57cec5SDimitry Andric if (config->zCombreloc && numRelativeRels) 14110b57cec5SDimitry Andric addInt(isRela ? DT_RELACOUNT : DT_RELCOUNT, numRelativeRels); 14120b57cec5SDimitry Andric } 14130b57cec5SDimitry Andric } 141404eeddc0SDimitry Andric if (part.relrDyn && part.relrDyn->getParent() && 141504eeddc0SDimitry Andric !part.relrDyn->relocs.empty()) { 14160b57cec5SDimitry Andric addInSec(config->useAndroidRelrTags ? DT_ANDROID_RELR : DT_RELR, 14170eae32dcSDimitry Andric *part.relrDyn); 14184824e7fdSDimitry Andric addInt(config->useAndroidRelrTags ? DT_ANDROID_RELRSZ : DT_RELRSZ, 14194824e7fdSDimitry Andric part.relrDyn->getParent()->size); 14200b57cec5SDimitry Andric addInt(config->useAndroidRelrTags ? DT_ANDROID_RELRENT : DT_RELRENT, 14210b57cec5SDimitry Andric sizeof(Elf_Relr)); 14220b57cec5SDimitry Andric } 1423*0fca6ea1SDimitry Andric if (part.relrAuthDyn && part.relrAuthDyn->getParent() && 1424*0fca6ea1SDimitry Andric !part.relrAuthDyn->relocs.empty()) { 1425*0fca6ea1SDimitry Andric addInSec(DT_AARCH64_AUTH_RELR, *part.relrAuthDyn); 1426*0fca6ea1SDimitry Andric addInt(DT_AARCH64_AUTH_RELRSZ, part.relrAuthDyn->getParent()->size); 1427*0fca6ea1SDimitry Andric addInt(DT_AARCH64_AUTH_RELRENT, sizeof(Elf_Relr)); 1428*0fca6ea1SDimitry Andric } 1429*0fca6ea1SDimitry Andric if (isMain && in.relaPlt->isNeeded()) { 14300eae32dcSDimitry Andric addInSec(DT_JMPREL, *in.relaPlt); 14314824e7fdSDimitry Andric entries.emplace_back(DT_PLTRELSZ, addPltRelSz()); 14320b57cec5SDimitry Andric switch (config->emachine) { 14330b57cec5SDimitry Andric case EM_MIPS: 14340eae32dcSDimitry Andric addInSec(DT_MIPS_PLTGOT, *in.gotPlt); 14350b57cec5SDimitry Andric break; 143674626c16SDimitry Andric case EM_S390: 143774626c16SDimitry Andric addInSec(DT_PLTGOT, *in.got); 143874626c16SDimitry Andric break; 14390b57cec5SDimitry Andric case EM_SPARCV9: 14400eae32dcSDimitry Andric addInSec(DT_PLTGOT, *in.plt); 14410b57cec5SDimitry Andric break; 1442e8d8bef9SDimitry Andric case EM_AARCH64: 1443e8d8bef9SDimitry Andric if (llvm::find_if(in.relaPlt->relocs, [](const DynamicReloc &r) { 1444e8d8bef9SDimitry Andric return r.type == target->pltRel && 1445e8d8bef9SDimitry Andric r.sym->stOther & STO_AARCH64_VARIANT_PCS; 1446e8d8bef9SDimitry Andric }) != in.relaPlt->relocs.end()) 1447e8d8bef9SDimitry Andric addInt(DT_AARCH64_VARIANT_PCS, 0); 1448bdd1243dSDimitry Andric addInSec(DT_PLTGOT, *in.gotPlt); 1449bdd1243dSDimitry Andric break; 1450bdd1243dSDimitry Andric case EM_RISCV: 1451bdd1243dSDimitry Andric if (llvm::any_of(in.relaPlt->relocs, [](const DynamicReloc &r) { 1452bdd1243dSDimitry Andric return r.type == target->pltRel && 1453bdd1243dSDimitry Andric (r.sym->stOther & STO_RISCV_VARIANT_CC); 1454bdd1243dSDimitry Andric })) 1455bdd1243dSDimitry Andric addInt(DT_RISCV_VARIANT_CC, 0); 1456bdd1243dSDimitry Andric [[fallthrough]]; 14570b57cec5SDimitry Andric default: 14580eae32dcSDimitry Andric addInSec(DT_PLTGOT, *in.gotPlt); 14590b57cec5SDimitry Andric break; 14600b57cec5SDimitry Andric } 14610b57cec5SDimitry Andric addInt(DT_PLTREL, config->isRela ? DT_RELA : DT_REL); 14620b57cec5SDimitry Andric } 14630b57cec5SDimitry Andric 14640b57cec5SDimitry Andric if (config->emachine == EM_AARCH64) { 14650b57cec5SDimitry Andric if (config->andFeatures & GNU_PROPERTY_AARCH64_FEATURE_1_BTI) 14660b57cec5SDimitry Andric addInt(DT_AARCH64_BTI_PLT, 0); 14675ffd83dbSDimitry Andric if (config->zPacPlt) 14680b57cec5SDimitry Andric addInt(DT_AARCH64_PAC_PLT, 0); 146906c3fb27SDimitry Andric 14701db9f3b2SDimitry Andric if (hasMemtag()) { 147106c3fb27SDimitry Andric addInt(DT_AARCH64_MEMTAG_MODE, config->androidMemtagMode == NT_MEMTAG_LEVEL_ASYNC); 147206c3fb27SDimitry Andric addInt(DT_AARCH64_MEMTAG_HEAP, config->androidMemtagHeap); 147306c3fb27SDimitry Andric addInt(DT_AARCH64_MEMTAG_STACK, config->androidMemtagStack); 14741db9f3b2SDimitry Andric if (mainPart->memtagGlobalDescriptors->isNeeded()) { 14751db9f3b2SDimitry Andric addInSec(DT_AARCH64_MEMTAG_GLOBALS, *mainPart->memtagGlobalDescriptors); 14761db9f3b2SDimitry Andric addInt(DT_AARCH64_MEMTAG_GLOBALSSZ, 14771db9f3b2SDimitry Andric mainPart->memtagGlobalDescriptors->getSize()); 14785f757f3fSDimitry Andric } 147906c3fb27SDimitry Andric } 14800b57cec5SDimitry Andric } 14810b57cec5SDimitry Andric 14820eae32dcSDimitry Andric addInSec(DT_SYMTAB, *part.dynSymTab); 14830b57cec5SDimitry Andric addInt(DT_SYMENT, sizeof(Elf_Sym)); 14840eae32dcSDimitry Andric addInSec(DT_STRTAB, *part.dynStrTab); 14850b57cec5SDimitry Andric addInt(DT_STRSZ, part.dynStrTab->getSize()); 14860b57cec5SDimitry Andric if (!config->zText) 14870b57cec5SDimitry Andric addInt(DT_TEXTREL, 0); 148804eeddc0SDimitry Andric if (part.gnuHashTab && part.gnuHashTab->getParent()) 14890eae32dcSDimitry Andric addInSec(DT_GNU_HASH, *part.gnuHashTab); 149004eeddc0SDimitry Andric if (part.hashTab && part.hashTab->getParent()) 14910eae32dcSDimitry Andric addInSec(DT_HASH, *part.hashTab); 14920b57cec5SDimitry Andric 14930b57cec5SDimitry Andric if (isMain) { 14940b57cec5SDimitry Andric if (Out::preinitArray) { 14954824e7fdSDimitry Andric addInt(DT_PREINIT_ARRAY, Out::preinitArray->addr); 14964824e7fdSDimitry Andric addInt(DT_PREINIT_ARRAYSZ, Out::preinitArray->size); 14970b57cec5SDimitry Andric } 14980b57cec5SDimitry Andric if (Out::initArray) { 14994824e7fdSDimitry Andric addInt(DT_INIT_ARRAY, Out::initArray->addr); 15004824e7fdSDimitry Andric addInt(DT_INIT_ARRAYSZ, Out::initArray->size); 15010b57cec5SDimitry Andric } 15020b57cec5SDimitry Andric if (Out::finiArray) { 15034824e7fdSDimitry Andric addInt(DT_FINI_ARRAY, Out::finiArray->addr); 15044824e7fdSDimitry Andric addInt(DT_FINI_ARRAYSZ, Out::finiArray->size); 15050b57cec5SDimitry Andric } 15060b57cec5SDimitry Andric 1507bdd1243dSDimitry Andric if (Symbol *b = symtab.find(config->init)) 15080b57cec5SDimitry Andric if (b->isDefined()) 15094824e7fdSDimitry Andric addInt(DT_INIT, b->getVA()); 1510bdd1243dSDimitry Andric if (Symbol *b = symtab.find(config->fini)) 15110b57cec5SDimitry Andric if (b->isDefined()) 15124824e7fdSDimitry Andric addInt(DT_FINI, b->getVA()); 15130b57cec5SDimitry Andric } 15140b57cec5SDimitry Andric 1515480093f4SDimitry Andric if (part.verSym && part.verSym->isNeeded()) 15160eae32dcSDimitry Andric addInSec(DT_VERSYM, *part.verSym); 1517480093f4SDimitry Andric if (part.verDef && part.verDef->isLive()) { 15180eae32dcSDimitry Andric addInSec(DT_VERDEF, *part.verDef); 15190b57cec5SDimitry Andric addInt(DT_VERDEFNUM, getVerDefNum()); 15200b57cec5SDimitry Andric } 1521480093f4SDimitry Andric if (part.verNeed && part.verNeed->isNeeded()) { 15220eae32dcSDimitry Andric addInSec(DT_VERNEED, *part.verNeed); 15230b57cec5SDimitry Andric unsigned needNum = 0; 1524bdd1243dSDimitry Andric for (SharedFile *f : ctx.sharedFiles) 15250b57cec5SDimitry Andric if (!f->vernauxs.empty()) 15260b57cec5SDimitry Andric ++needNum; 15270b57cec5SDimitry Andric addInt(DT_VERNEEDNUM, needNum); 15280b57cec5SDimitry Andric } 15290b57cec5SDimitry Andric 15300b57cec5SDimitry Andric if (config->emachine == EM_MIPS) { 15310b57cec5SDimitry Andric addInt(DT_MIPS_RLD_VERSION, 1); 15320b57cec5SDimitry Andric addInt(DT_MIPS_FLAGS, RHF_NOTPOT); 15330b57cec5SDimitry Andric addInt(DT_MIPS_BASE_ADDRESS, target->getImageBase()); 15340b57cec5SDimitry Andric addInt(DT_MIPS_SYMTABNO, part.dynSymTab->getNumSymbols()); 15354824e7fdSDimitry Andric addInt(DT_MIPS_LOCAL_GOTNO, in.mipsGot->getLocalEntriesNum()); 15360b57cec5SDimitry Andric 15370b57cec5SDimitry Andric if (const Symbol *b = in.mipsGot->getFirstGlobalEntry()) 15380b57cec5SDimitry Andric addInt(DT_MIPS_GOTSYM, b->dynsymIndex); 15390b57cec5SDimitry Andric else 15400b57cec5SDimitry Andric addInt(DT_MIPS_GOTSYM, part.dynSymTab->getNumSymbols()); 15410eae32dcSDimitry Andric addInSec(DT_PLTGOT, *in.mipsGot); 15420b57cec5SDimitry Andric if (in.mipsRldMap) { 15430b57cec5SDimitry Andric if (!config->pie) 15440eae32dcSDimitry Andric addInSec(DT_MIPS_RLD_MAP, *in.mipsRldMap); 15450b57cec5SDimitry Andric // Store the offset to the .rld_map section 15460b57cec5SDimitry Andric // relative to the address of the tag. 15474824e7fdSDimitry Andric addInt(DT_MIPS_RLD_MAP_REL, 15484824e7fdSDimitry Andric in.mipsRldMap->getVA() - (getVA() + entries.size() * entsize)); 15490b57cec5SDimitry Andric } 15500b57cec5SDimitry Andric } 15510b57cec5SDimitry Andric 15520b57cec5SDimitry Andric // DT_PPC_GOT indicates to glibc Secure PLT is used. If DT_PPC_GOT is absent, 15530b57cec5SDimitry Andric // glibc assumes the old-style BSS PLT layout which we don't support. 15540b57cec5SDimitry Andric if (config->emachine == EM_PPC) 15550eae32dcSDimitry Andric addInSec(DT_PPC_GOT, *in.got); 15560b57cec5SDimitry Andric 15570b57cec5SDimitry Andric // Glink dynamic tag is required by the V2 abi if the plt section isn't empty. 15580b57cec5SDimitry Andric if (config->emachine == EM_PPC64 && in.plt->isNeeded()) { 15590b57cec5SDimitry Andric // The Glink tag points to 32 bytes before the first lazy symbol resolution 15600b57cec5SDimitry Andric // stub, which starts directly after the header. 15614824e7fdSDimitry Andric addInt(DT_PPC64_GLINK, in.plt->getVA() + target->pltHeaderSize - 32); 15620b57cec5SDimitry Andric } 15630b57cec5SDimitry Andric 156406c3fb27SDimitry Andric if (config->emachine == EM_PPC64) 156506c3fb27SDimitry Andric addInt(DT_PPC64_OPT, getPPC64TargetInfo()->ppc64DynamicSectionOpt); 156606c3fb27SDimitry Andric 15670b57cec5SDimitry Andric addInt(DT_NULL, 0); 15684824e7fdSDimitry Andric return entries; 15694824e7fdSDimitry Andric } 15700b57cec5SDimitry Andric 15714824e7fdSDimitry Andric template <class ELFT> void DynamicSection<ELFT>::finalizeContents() { 15724824e7fdSDimitry Andric if (OutputSection *sec = getPartition().dynStrTab->getParent()) 15734824e7fdSDimitry Andric getParent()->link = sec->sectionIndex; 15744824e7fdSDimitry Andric this->size = computeContents().size() * this->entsize; 15750b57cec5SDimitry Andric } 15760b57cec5SDimitry Andric 15770b57cec5SDimitry Andric template <class ELFT> void DynamicSection<ELFT>::writeTo(uint8_t *buf) { 15780b57cec5SDimitry Andric auto *p = reinterpret_cast<Elf_Dyn *>(buf); 15790b57cec5SDimitry Andric 15804824e7fdSDimitry Andric for (std::pair<int32_t, uint64_t> kv : computeContents()) { 15810b57cec5SDimitry Andric p->d_tag = kv.first; 15824824e7fdSDimitry Andric p->d_un.d_val = kv.second; 15830b57cec5SDimitry Andric ++p; 15840b57cec5SDimitry Andric } 15850b57cec5SDimitry Andric } 15860b57cec5SDimitry Andric 15870b57cec5SDimitry Andric uint64_t DynamicReloc::getOffset() const { 15880b57cec5SDimitry Andric return inputSec->getVA(offsetInSec); 15890b57cec5SDimitry Andric } 15900b57cec5SDimitry Andric 15910b57cec5SDimitry Andric int64_t DynamicReloc::computeAddend() const { 1592fe6060f1SDimitry Andric switch (kind) { 1593fe6060f1SDimitry Andric case AddendOnly: 1594fe6060f1SDimitry Andric assert(sym == nullptr); 15950b57cec5SDimitry Andric return addend; 1596fe6060f1SDimitry Andric case AgainstSymbol: 1597fe6060f1SDimitry Andric assert(sym != nullptr); 1598fe6060f1SDimitry Andric return addend; 1599fe6060f1SDimitry Andric case AddendOnlyWithTargetVA: 160006c3fb27SDimitry Andric case AgainstSymbolWithTargetVA: { 160106c3fb27SDimitry Andric uint64_t ca = InputSection::getRelocTargetVA(inputSec->file, type, addend, 1602fe6060f1SDimitry Andric getOffset(), *sym, expr); 160306c3fb27SDimitry Andric return config->is64 ? ca : SignExtend64<32>(ca); 160406c3fb27SDimitry Andric } 1605fe6060f1SDimitry Andric case MipsMultiGotPage: 1606fe6060f1SDimitry Andric assert(sym == nullptr); 16070b57cec5SDimitry Andric return getMipsPageAddr(outputSec->addr) + addend; 16080b57cec5SDimitry Andric } 1609fe6060f1SDimitry Andric llvm_unreachable("Unknown DynamicReloc::Kind enum"); 1610fe6060f1SDimitry Andric } 16110b57cec5SDimitry Andric 16120b57cec5SDimitry Andric uint32_t DynamicReloc::getSymIndex(SymbolTableBaseSection *symTab) const { 161381ad6265SDimitry Andric if (!needsDynSymIndex()) 16140b57cec5SDimitry Andric return 0; 161581ad6265SDimitry Andric 1616*0fca6ea1SDimitry Andric size_t index = symTab->getSymbolIndex(*sym); 161781ad6265SDimitry Andric assert((index != 0 || (type != target->gotRel && type != target->pltRel) || 161881ad6265SDimitry Andric !mainPart->dynSymTab->getParent()) && 161981ad6265SDimitry Andric "GOT or PLT relocation must refer to symbol in dynamic symbol table"); 162081ad6265SDimitry Andric return index; 16210b57cec5SDimitry Andric } 16220b57cec5SDimitry Andric 16230b57cec5SDimitry Andric RelocationBaseSection::RelocationBaseSection(StringRef name, uint32_t type, 16240b57cec5SDimitry Andric int32_t dynamicTag, 16251fd87a68SDimitry Andric int32_t sizeDynamicTag, 1626bdd1243dSDimitry Andric bool combreloc, 1627bdd1243dSDimitry Andric unsigned concurrency) 16280b57cec5SDimitry Andric : SyntheticSection(SHF_ALLOC, type, config->wordsize, name), 16291fd87a68SDimitry Andric dynamicTag(dynamicTag), sizeDynamicTag(sizeDynamicTag), 1630bdd1243dSDimitry Andric relocsVec(concurrency), combreloc(combreloc) {} 16310b57cec5SDimitry Andric 1632bdd1243dSDimitry Andric void RelocationBaseSection::addSymbolReloc( 1633bdd1243dSDimitry Andric RelType dynType, InputSectionBase &isec, uint64_t offsetInSec, Symbol &sym, 1634bdd1243dSDimitry Andric int64_t addend, std::optional<RelType> addendRelType) { 1635fe6060f1SDimitry Andric addReloc(DynamicReloc::AgainstSymbol, dynType, isec, offsetInSec, sym, addend, 1636fe6060f1SDimitry Andric R_ADDEND, addendRelType ? *addendRelType : target->noneRel); 16370b57cec5SDimitry Andric } 16380b57cec5SDimitry Andric 1639fe6060f1SDimitry Andric void RelocationBaseSection::addAddendOnlyRelocIfNonPreemptible( 1640bdd1243dSDimitry Andric RelType dynType, GotSection &sec, uint64_t offsetInSec, Symbol &sym, 1641fe6060f1SDimitry Andric RelType addendRelType) { 1642fe6060f1SDimitry Andric // No need to write an addend to the section for preemptible symbols. 1643fe6060f1SDimitry Andric if (sym.isPreemptible) 1644bdd1243dSDimitry Andric addReloc({dynType, &sec, offsetInSec, DynamicReloc::AgainstSymbol, sym, 0, 1645fe6060f1SDimitry Andric R_ABS}); 1646fe6060f1SDimitry Andric else 1647bdd1243dSDimitry Andric addReloc(DynamicReloc::AddendOnlyWithTargetVA, dynType, sec, offsetInSec, 1648fe6060f1SDimitry Andric sym, 0, R_ABS, addendRelType); 1649fe6060f1SDimitry Andric } 1650fe6060f1SDimitry Andric 1651bdd1243dSDimitry Andric void RelocationBaseSection::mergeRels() { 1652bdd1243dSDimitry Andric size_t newSize = relocs.size(); 1653bdd1243dSDimitry Andric for (const auto &v : relocsVec) 1654bdd1243dSDimitry Andric newSize += v.size(); 1655bdd1243dSDimitry Andric relocs.reserve(newSize); 1656bdd1243dSDimitry Andric for (const auto &v : relocsVec) 1657bdd1243dSDimitry Andric llvm::append_range(relocs, v); 1658bdd1243dSDimitry Andric relocsVec.clear(); 16590b57cec5SDimitry Andric } 16600b57cec5SDimitry Andric 16611fd87a68SDimitry Andric void RelocationBaseSection::partitionRels() { 16621fd87a68SDimitry Andric if (!combreloc) 16631fd87a68SDimitry Andric return; 16641fd87a68SDimitry Andric const RelType relativeRel = target->relativeRel; 16651fd87a68SDimitry Andric numRelativeRelocs = 1666*0fca6ea1SDimitry Andric std::stable_partition(relocs.begin(), relocs.end(), 1667*0fca6ea1SDimitry Andric [=](auto &r) { return r.type == relativeRel; }) - 16681fd87a68SDimitry Andric relocs.begin(); 16690b57cec5SDimitry Andric } 16700b57cec5SDimitry Andric 16710b57cec5SDimitry Andric void RelocationBaseSection::finalizeContents() { 167204eeddc0SDimitry Andric SymbolTableBaseSection *symTab = getPartition().dynSymTab.get(); 16730b57cec5SDimitry Andric 16740b57cec5SDimitry Andric // When linking glibc statically, .rel{,a}.plt contains R_*_IRELATIVE 16750b57cec5SDimitry Andric // relocations due to IFUNC (e.g. strcpy). sh_link will be set to 0 in that 16760b57cec5SDimitry Andric // case. 16770b57cec5SDimitry Andric if (symTab && symTab->getParent()) 16780b57cec5SDimitry Andric getParent()->link = symTab->getParent()->sectionIndex; 16790b57cec5SDimitry Andric else 16800b57cec5SDimitry Andric getParent()->link = 0; 16810b57cec5SDimitry Andric 168204eeddc0SDimitry Andric if (in.relaPlt.get() == this && in.gotPlt->getParent()) { 1683e8d8bef9SDimitry Andric getParent()->flags |= ELF::SHF_INFO_LINK; 16840b57cec5SDimitry Andric getParent()->info = in.gotPlt->getParent()->sectionIndex; 1685e8d8bef9SDimitry Andric } 1686e8d8bef9SDimitry Andric } 16870b57cec5SDimitry Andric 16880eae32dcSDimitry Andric void DynamicReloc::computeRaw(SymbolTableBaseSection *symtab) { 16890eae32dcSDimitry Andric r_offset = getOffset(); 16900eae32dcSDimitry Andric r_sym = getSymIndex(symtab); 16910eae32dcSDimitry Andric addend = computeAddend(); 16920eae32dcSDimitry Andric kind = AddendOnly; // Catch errors 16930b57cec5SDimitry Andric } 16940b57cec5SDimitry Andric 16951fd87a68SDimitry Andric void RelocationBaseSection::computeRels() { 169604eeddc0SDimitry Andric SymbolTableBaseSection *symTab = getPartition().dynSymTab.get(); 16970eae32dcSDimitry Andric parallelForEach(relocs, 16980eae32dcSDimitry Andric [symTab](DynamicReloc &rel) { rel.computeRaw(symTab); }); 1699*0fca6ea1SDimitry Andric 1700*0fca6ea1SDimitry Andric auto irelative = std::stable_partition( 1701*0fca6ea1SDimitry Andric relocs.begin() + numRelativeRelocs, relocs.end(), 1702*0fca6ea1SDimitry Andric [t = target->iRelativeRel](auto &r) { return r.type != t; }); 1703*0fca6ea1SDimitry Andric 17040b57cec5SDimitry Andric // Sort by (!IsRelative,SymIndex,r_offset). DT_REL[A]COUNT requires us to 17050b57cec5SDimitry Andric // place R_*_RELATIVE first. SymIndex is to improve locality, while r_offset 17060b57cec5SDimitry Andric // is to make results easier to read. 17071fd87a68SDimitry Andric if (combreloc) { 17081fd87a68SDimitry Andric auto nonRelative = relocs.begin() + numRelativeRelocs; 170904eeddc0SDimitry Andric parallelSort(relocs.begin(), nonRelative, 171004eeddc0SDimitry Andric [&](auto &a, auto &b) { return a.r_offset < b.r_offset; }); 171104eeddc0SDimitry Andric // Non-relative relocations are few, so don't bother with parallelSort. 1712*0fca6ea1SDimitry Andric llvm::sort(nonRelative, irelative, [&](auto &a, auto &b) { 171304eeddc0SDimitry Andric return std::tie(a.r_sym, a.r_offset) < std::tie(b.r_sym, b.r_offset); 17140b57cec5SDimitry Andric }); 17150eae32dcSDimitry Andric } 17161fd87a68SDimitry Andric } 17170b57cec5SDimitry Andric 17181fd87a68SDimitry Andric template <class ELFT> 1719bdd1243dSDimitry Andric RelocationSection<ELFT>::RelocationSection(StringRef name, bool combreloc, 1720bdd1243dSDimitry Andric unsigned concurrency) 17211fd87a68SDimitry Andric : RelocationBaseSection(name, config->isRela ? SHT_RELA : SHT_REL, 17221fd87a68SDimitry Andric config->isRela ? DT_RELA : DT_REL, 1723bdd1243dSDimitry Andric config->isRela ? DT_RELASZ : DT_RELSZ, combreloc, 1724bdd1243dSDimitry Andric concurrency) { 17251fd87a68SDimitry Andric this->entsize = config->isRela ? sizeof(Elf_Rela) : sizeof(Elf_Rel); 17261fd87a68SDimitry Andric } 17271fd87a68SDimitry Andric 17281fd87a68SDimitry Andric template <class ELFT> void RelocationSection<ELFT>::writeTo(uint8_t *buf) { 17291fd87a68SDimitry Andric computeRels(); 17300b57cec5SDimitry Andric for (const DynamicReloc &rel : relocs) { 17311fd87a68SDimitry Andric auto *p = reinterpret_cast<Elf_Rela *>(buf); 17321fd87a68SDimitry Andric p->r_offset = rel.r_offset; 17331fd87a68SDimitry Andric p->setSymbolAndType(rel.r_sym, rel.type, config->isMips64EL); 17341fd87a68SDimitry Andric if (config->isRela) 17351fd87a68SDimitry Andric p->r_addend = rel.addend; 17360b57cec5SDimitry Andric buf += config->isRela ? sizeof(Elf_Rela) : sizeof(Elf_Rel); 17370b57cec5SDimitry Andric } 17380b57cec5SDimitry Andric } 17390b57cec5SDimitry Andric 1740*0fca6ea1SDimitry Andric RelrBaseSection::RelrBaseSection(unsigned concurrency, bool isAArch64Auth) 1741*0fca6ea1SDimitry Andric : SyntheticSection( 1742*0fca6ea1SDimitry Andric SHF_ALLOC, 1743*0fca6ea1SDimitry Andric isAArch64Auth 1744*0fca6ea1SDimitry Andric ? SHT_AARCH64_AUTH_RELR 1745*0fca6ea1SDimitry Andric : (config->useAndroidRelrTags ? SHT_ANDROID_RELR : SHT_RELR), 1746*0fca6ea1SDimitry Andric config->wordsize, isAArch64Auth ? ".relr.auth.dyn" : ".relr.dyn"), 1747bdd1243dSDimitry Andric relocsVec(concurrency) {} 1748bdd1243dSDimitry Andric 1749bdd1243dSDimitry Andric void RelrBaseSection::mergeRels() { 1750bdd1243dSDimitry Andric size_t newSize = relocs.size(); 1751bdd1243dSDimitry Andric for (const auto &v : relocsVec) 1752bdd1243dSDimitry Andric newSize += v.size(); 1753bdd1243dSDimitry Andric relocs.reserve(newSize); 1754bdd1243dSDimitry Andric for (const auto &v : relocsVec) 1755bdd1243dSDimitry Andric llvm::append_range(relocs, v); 1756bdd1243dSDimitry Andric relocsVec.clear(); 1757bdd1243dSDimitry Andric } 17581fd87a68SDimitry Andric 17590b57cec5SDimitry Andric template <class ELFT> 17600b57cec5SDimitry Andric AndroidPackedRelocationSection<ELFT>::AndroidPackedRelocationSection( 1761bdd1243dSDimitry Andric StringRef name, unsigned concurrency) 17620b57cec5SDimitry Andric : RelocationBaseSection( 17630b57cec5SDimitry Andric name, config->isRela ? SHT_ANDROID_RELA : SHT_ANDROID_REL, 17640b57cec5SDimitry Andric config->isRela ? DT_ANDROID_RELA : DT_ANDROID_REL, 17651fd87a68SDimitry Andric config->isRela ? DT_ANDROID_RELASZ : DT_ANDROID_RELSZ, 1766bdd1243dSDimitry Andric /*combreloc=*/false, concurrency) { 17670b57cec5SDimitry Andric this->entsize = 1; 17680b57cec5SDimitry Andric } 17690b57cec5SDimitry Andric 17700b57cec5SDimitry Andric template <class ELFT> 17710b57cec5SDimitry Andric bool AndroidPackedRelocationSection<ELFT>::updateAllocSize() { 17720b57cec5SDimitry Andric // This function computes the contents of an Android-format packed relocation 17730b57cec5SDimitry Andric // section. 17740b57cec5SDimitry Andric // 17750b57cec5SDimitry Andric // This format compresses relocations by using relocation groups to factor out 17760b57cec5SDimitry Andric // fields that are common between relocations and storing deltas from previous 17770b57cec5SDimitry Andric // relocations in SLEB128 format (which has a short representation for small 17780b57cec5SDimitry Andric // numbers). A good example of a relocation type with common fields is 17790b57cec5SDimitry Andric // R_*_RELATIVE, which is normally used to represent function pointers in 17800b57cec5SDimitry Andric // vtables. In the REL format, each relative relocation has the same r_info 17810b57cec5SDimitry Andric // field, and is only different from other relative relocations in terms of 17820b57cec5SDimitry Andric // the r_offset field. By sorting relocations by offset, grouping them by 17830b57cec5SDimitry Andric // r_info and representing each relocation with only the delta from the 17840b57cec5SDimitry Andric // previous offset, each 8-byte relocation can be compressed to as little as 1 17850b57cec5SDimitry Andric // byte (or less with run-length encoding). This relocation packer was able to 17860b57cec5SDimitry Andric // reduce the size of the relocation section in an Android Chromium DSO from 17870b57cec5SDimitry Andric // 2,911,184 bytes to 174,693 bytes, or 6% of the original size. 17880b57cec5SDimitry Andric // 17890b57cec5SDimitry Andric // A relocation section consists of a header containing the literal bytes 17900b57cec5SDimitry Andric // 'APS2' followed by a sequence of SLEB128-encoded integers. The first two 17910b57cec5SDimitry Andric // elements are the total number of relocations in the section and an initial 17920b57cec5SDimitry Andric // r_offset value. The remaining elements define a sequence of relocation 17930b57cec5SDimitry Andric // groups. Each relocation group starts with a header consisting of the 17940b57cec5SDimitry Andric // following elements: 17950b57cec5SDimitry Andric // 17960b57cec5SDimitry Andric // - the number of relocations in the relocation group 17970b57cec5SDimitry Andric // - flags for the relocation group 17980b57cec5SDimitry Andric // - (if RELOCATION_GROUPED_BY_OFFSET_DELTA_FLAG is set) the r_offset delta 17990b57cec5SDimitry Andric // for each relocation in the group. 18000b57cec5SDimitry Andric // - (if RELOCATION_GROUPED_BY_INFO_FLAG is set) the value of the r_info 18010b57cec5SDimitry Andric // field for each relocation in the group. 18020b57cec5SDimitry Andric // - (if RELOCATION_GROUP_HAS_ADDEND_FLAG and 18030b57cec5SDimitry Andric // RELOCATION_GROUPED_BY_ADDEND_FLAG are set) the r_addend delta for 18040b57cec5SDimitry Andric // each relocation in the group. 18050b57cec5SDimitry Andric // 18060b57cec5SDimitry Andric // Following the relocation group header are descriptions of each of the 18070b57cec5SDimitry Andric // relocations in the group. They consist of the following elements: 18080b57cec5SDimitry Andric // 18090b57cec5SDimitry Andric // - (if RELOCATION_GROUPED_BY_OFFSET_DELTA_FLAG is not set) the r_offset 18100b57cec5SDimitry Andric // delta for this relocation. 18110b57cec5SDimitry Andric // - (if RELOCATION_GROUPED_BY_INFO_FLAG is not set) the value of the r_info 18120b57cec5SDimitry Andric // field for this relocation. 18130b57cec5SDimitry Andric // - (if RELOCATION_GROUP_HAS_ADDEND_FLAG is set and 18140b57cec5SDimitry Andric // RELOCATION_GROUPED_BY_ADDEND_FLAG is not set) the r_addend delta for 18150b57cec5SDimitry Andric // this relocation. 18160b57cec5SDimitry Andric 18170b57cec5SDimitry Andric size_t oldSize = relocData.size(); 18180b57cec5SDimitry Andric 18190b57cec5SDimitry Andric relocData = {'A', 'P', 'S', '2'}; 18200b57cec5SDimitry Andric raw_svector_ostream os(relocData); 18210b57cec5SDimitry Andric auto add = [&](int64_t v) { encodeSLEB128(v, os); }; 18220b57cec5SDimitry Andric 18230b57cec5SDimitry Andric // The format header includes the number of relocations and the initial 18240b57cec5SDimitry Andric // offset (we set this to zero because the first relocation group will 18250b57cec5SDimitry Andric // perform the initial adjustment). 18260b57cec5SDimitry Andric add(relocs.size()); 18270b57cec5SDimitry Andric add(0); 18280b57cec5SDimitry Andric 18290b57cec5SDimitry Andric std::vector<Elf_Rela> relatives, nonRelatives; 18300b57cec5SDimitry Andric 18310b57cec5SDimitry Andric for (const DynamicReloc &rel : relocs) { 18320b57cec5SDimitry Andric Elf_Rela r; 18330eae32dcSDimitry Andric r.r_offset = rel.getOffset(); 183404eeddc0SDimitry Andric r.setSymbolAndType(rel.getSymIndex(getPartition().dynSymTab.get()), 183504eeddc0SDimitry Andric rel.type, false); 1836bdd1243dSDimitry Andric r.r_addend = config->isRela ? rel.computeAddend() : 0; 18370b57cec5SDimitry Andric 18380b57cec5SDimitry Andric if (r.getType(config->isMips64EL) == target->relativeRel) 18390b57cec5SDimitry Andric relatives.push_back(r); 18400b57cec5SDimitry Andric else 18410b57cec5SDimitry Andric nonRelatives.push_back(r); 18420b57cec5SDimitry Andric } 18430b57cec5SDimitry Andric 18440b57cec5SDimitry Andric llvm::sort(relatives, [](const Elf_Rel &a, const Elf_Rel &b) { 18450b57cec5SDimitry Andric return a.r_offset < b.r_offset; 18460b57cec5SDimitry Andric }); 18470b57cec5SDimitry Andric 18480b57cec5SDimitry Andric // Try to find groups of relative relocations which are spaced one word 18490b57cec5SDimitry Andric // apart from one another. These generally correspond to vtable entries. The 18500b57cec5SDimitry Andric // format allows these groups to be encoded using a sort of run-length 18510b57cec5SDimitry Andric // encoding, but each group will cost 7 bytes in addition to the offset from 18520b57cec5SDimitry Andric // the previous group, so it is only profitable to do this for groups of 18530b57cec5SDimitry Andric // size 8 or larger. 18540b57cec5SDimitry Andric std::vector<Elf_Rela> ungroupedRelatives; 18550b57cec5SDimitry Andric std::vector<std::vector<Elf_Rela>> relativeGroups; 18560b57cec5SDimitry Andric for (auto i = relatives.begin(), e = relatives.end(); i != e;) { 18570b57cec5SDimitry Andric std::vector<Elf_Rela> group; 18580b57cec5SDimitry Andric do { 18590b57cec5SDimitry Andric group.push_back(*i++); 18600b57cec5SDimitry Andric } while (i != e && (i - 1)->r_offset + config->wordsize == i->r_offset); 18610b57cec5SDimitry Andric 18620b57cec5SDimitry Andric if (group.size() < 8) 18630b57cec5SDimitry Andric ungroupedRelatives.insert(ungroupedRelatives.end(), group.begin(), 18640b57cec5SDimitry Andric group.end()); 18650b57cec5SDimitry Andric else 18660b57cec5SDimitry Andric relativeGroups.emplace_back(std::move(group)); 18670b57cec5SDimitry Andric } 18680b57cec5SDimitry Andric 186985868e8aSDimitry Andric // For non-relative relocations, we would like to: 187085868e8aSDimitry Andric // 1. Have relocations with the same symbol offset to be consecutive, so 187185868e8aSDimitry Andric // that the runtime linker can speed-up symbol lookup by implementing an 187285868e8aSDimitry Andric // 1-entry cache. 187385868e8aSDimitry Andric // 2. Group relocations by r_info to reduce the size of the relocation 187485868e8aSDimitry Andric // section. 187585868e8aSDimitry Andric // Since the symbol offset is the high bits in r_info, sorting by r_info 187685868e8aSDimitry Andric // allows us to do both. 187785868e8aSDimitry Andric // 187885868e8aSDimitry Andric // For Rela, we also want to sort by r_addend when r_info is the same. This 187985868e8aSDimitry Andric // enables us to group by r_addend as well. 1880bdd1243dSDimitry Andric llvm::sort(nonRelatives, [](const Elf_Rela &a, const Elf_Rela &b) { 188185868e8aSDimitry Andric if (a.r_info != b.r_info) 188285868e8aSDimitry Andric return a.r_info < b.r_info; 1883bdd1243dSDimitry Andric if (a.r_addend != b.r_addend) 188485868e8aSDimitry Andric return a.r_addend < b.r_addend; 1885bdd1243dSDimitry Andric return a.r_offset < b.r_offset; 188685868e8aSDimitry Andric }); 188785868e8aSDimitry Andric 188885868e8aSDimitry Andric // Group relocations with the same r_info. Note that each group emits a group 188985868e8aSDimitry Andric // header and that may make the relocation section larger. It is hard to 189085868e8aSDimitry Andric // estimate the size of a group header as the encoded size of that varies 189185868e8aSDimitry Andric // based on r_info. However, we can approximate this trade-off by the number 189285868e8aSDimitry Andric // of values encoded. Each group header contains 3 values, and each relocation 189385868e8aSDimitry Andric // in a group encodes one less value, as compared to when it is not grouped. 189485868e8aSDimitry Andric // Therefore, we only group relocations if there are 3 or more of them with 189585868e8aSDimitry Andric // the same r_info. 189685868e8aSDimitry Andric // 189785868e8aSDimitry Andric // For Rela, the addend for most non-relative relocations is zero, and thus we 189885868e8aSDimitry Andric // can usually get a smaller relocation section if we group relocations with 0 189985868e8aSDimitry Andric // addend as well. 190085868e8aSDimitry Andric std::vector<Elf_Rela> ungroupedNonRelatives; 190185868e8aSDimitry Andric std::vector<std::vector<Elf_Rela>> nonRelativeGroups; 190285868e8aSDimitry Andric for (auto i = nonRelatives.begin(), e = nonRelatives.end(); i != e;) { 190385868e8aSDimitry Andric auto j = i + 1; 190485868e8aSDimitry Andric while (j != e && i->r_info == j->r_info && 190585868e8aSDimitry Andric (!config->isRela || i->r_addend == j->r_addend)) 190685868e8aSDimitry Andric ++j; 190785868e8aSDimitry Andric if (j - i < 3 || (config->isRela && i->r_addend != 0)) 190885868e8aSDimitry Andric ungroupedNonRelatives.insert(ungroupedNonRelatives.end(), i, j); 190985868e8aSDimitry Andric else 191085868e8aSDimitry Andric nonRelativeGroups.emplace_back(i, j); 191185868e8aSDimitry Andric i = j; 191285868e8aSDimitry Andric } 191385868e8aSDimitry Andric 191485868e8aSDimitry Andric // Sort ungrouped relocations by offset to minimize the encoded length. 191585868e8aSDimitry Andric llvm::sort(ungroupedNonRelatives, [](const Elf_Rela &a, const Elf_Rela &b) { 191685868e8aSDimitry Andric return a.r_offset < b.r_offset; 191785868e8aSDimitry Andric }); 191885868e8aSDimitry Andric 19190b57cec5SDimitry Andric unsigned hasAddendIfRela = 19200b57cec5SDimitry Andric config->isRela ? RELOCATION_GROUP_HAS_ADDEND_FLAG : 0; 19210b57cec5SDimitry Andric 19220b57cec5SDimitry Andric uint64_t offset = 0; 19230b57cec5SDimitry Andric uint64_t addend = 0; 19240b57cec5SDimitry Andric 19250b57cec5SDimitry Andric // Emit the run-length encoding for the groups of adjacent relative 19260b57cec5SDimitry Andric // relocations. Each group is represented using two groups in the packed 19270b57cec5SDimitry Andric // format. The first is used to set the current offset to the start of the 19280b57cec5SDimitry Andric // group (and also encodes the first relocation), and the second encodes the 19290b57cec5SDimitry Andric // remaining relocations. 19300b57cec5SDimitry Andric for (std::vector<Elf_Rela> &g : relativeGroups) { 19310b57cec5SDimitry Andric // The first relocation in the group. 19320b57cec5SDimitry Andric add(1); 19330b57cec5SDimitry Andric add(RELOCATION_GROUPED_BY_OFFSET_DELTA_FLAG | 19340b57cec5SDimitry Andric RELOCATION_GROUPED_BY_INFO_FLAG | hasAddendIfRela); 19350b57cec5SDimitry Andric add(g[0].r_offset - offset); 19360b57cec5SDimitry Andric add(target->relativeRel); 19370b57cec5SDimitry Andric if (config->isRela) { 19380b57cec5SDimitry Andric add(g[0].r_addend - addend); 19390b57cec5SDimitry Andric addend = g[0].r_addend; 19400b57cec5SDimitry Andric } 19410b57cec5SDimitry Andric 19420b57cec5SDimitry Andric // The remaining relocations. 19430b57cec5SDimitry Andric add(g.size() - 1); 19440b57cec5SDimitry Andric add(RELOCATION_GROUPED_BY_OFFSET_DELTA_FLAG | 19450b57cec5SDimitry Andric RELOCATION_GROUPED_BY_INFO_FLAG | hasAddendIfRela); 19460b57cec5SDimitry Andric add(config->wordsize); 19470b57cec5SDimitry Andric add(target->relativeRel); 19480b57cec5SDimitry Andric if (config->isRela) { 1949bdd1243dSDimitry Andric for (const auto &i : llvm::drop_begin(g)) { 1950bdd1243dSDimitry Andric add(i.r_addend - addend); 1951bdd1243dSDimitry Andric addend = i.r_addend; 19520b57cec5SDimitry Andric } 19530b57cec5SDimitry Andric } 19540b57cec5SDimitry Andric 19550b57cec5SDimitry Andric offset = g.back().r_offset; 19560b57cec5SDimitry Andric } 19570b57cec5SDimitry Andric 19580b57cec5SDimitry Andric // Now the ungrouped relatives. 19590b57cec5SDimitry Andric if (!ungroupedRelatives.empty()) { 19600b57cec5SDimitry Andric add(ungroupedRelatives.size()); 19610b57cec5SDimitry Andric add(RELOCATION_GROUPED_BY_INFO_FLAG | hasAddendIfRela); 19620b57cec5SDimitry Andric add(target->relativeRel); 19630b57cec5SDimitry Andric for (Elf_Rela &r : ungroupedRelatives) { 19640b57cec5SDimitry Andric add(r.r_offset - offset); 19650b57cec5SDimitry Andric offset = r.r_offset; 19660b57cec5SDimitry Andric if (config->isRela) { 19670b57cec5SDimitry Andric add(r.r_addend - addend); 19680b57cec5SDimitry Andric addend = r.r_addend; 19690b57cec5SDimitry Andric } 19700b57cec5SDimitry Andric } 19710b57cec5SDimitry Andric } 19720b57cec5SDimitry Andric 197385868e8aSDimitry Andric // Grouped non-relatives. 197485868e8aSDimitry Andric for (ArrayRef<Elf_Rela> g : nonRelativeGroups) { 197585868e8aSDimitry Andric add(g.size()); 197685868e8aSDimitry Andric add(RELOCATION_GROUPED_BY_INFO_FLAG); 197785868e8aSDimitry Andric add(g[0].r_info); 197885868e8aSDimitry Andric for (const Elf_Rela &r : g) { 197985868e8aSDimitry Andric add(r.r_offset - offset); 198085868e8aSDimitry Andric offset = r.r_offset; 198185868e8aSDimitry Andric } 198285868e8aSDimitry Andric addend = 0; 198385868e8aSDimitry Andric } 198485868e8aSDimitry Andric 198585868e8aSDimitry Andric // Finally the ungrouped non-relative relocations. 198685868e8aSDimitry Andric if (!ungroupedNonRelatives.empty()) { 198785868e8aSDimitry Andric add(ungroupedNonRelatives.size()); 19880b57cec5SDimitry Andric add(hasAddendIfRela); 198985868e8aSDimitry Andric for (Elf_Rela &r : ungroupedNonRelatives) { 19900b57cec5SDimitry Andric add(r.r_offset - offset); 19910b57cec5SDimitry Andric offset = r.r_offset; 19920b57cec5SDimitry Andric add(r.r_info); 19930b57cec5SDimitry Andric if (config->isRela) { 19940b57cec5SDimitry Andric add(r.r_addend - addend); 19950b57cec5SDimitry Andric addend = r.r_addend; 19960b57cec5SDimitry Andric } 19970b57cec5SDimitry Andric } 19980b57cec5SDimitry Andric } 19990b57cec5SDimitry Andric 20000b57cec5SDimitry Andric // Don't allow the section to shrink; otherwise the size of the section can 20010b57cec5SDimitry Andric // oscillate infinitely. 20020b57cec5SDimitry Andric if (relocData.size() < oldSize) 20030b57cec5SDimitry Andric relocData.append(oldSize - relocData.size(), 0); 20040b57cec5SDimitry Andric 20050b57cec5SDimitry Andric // Returns whether the section size changed. We need to keep recomputing both 20060b57cec5SDimitry Andric // section layout and the contents of this section until the size converges 20070b57cec5SDimitry Andric // because changing this section's size can affect section layout, which in 20080b57cec5SDimitry Andric // turn can affect the sizes of the LEB-encoded integers stored in this 20090b57cec5SDimitry Andric // section. 20100b57cec5SDimitry Andric return relocData.size() != oldSize; 20110b57cec5SDimitry Andric } 20120b57cec5SDimitry Andric 2013bdd1243dSDimitry Andric template <class ELFT> 2014*0fca6ea1SDimitry Andric RelrSection<ELFT>::RelrSection(unsigned concurrency, bool isAArch64Auth) 2015*0fca6ea1SDimitry Andric : RelrBaseSection(concurrency, isAArch64Auth) { 20160b57cec5SDimitry Andric this->entsize = config->wordsize; 20170b57cec5SDimitry Andric } 20180b57cec5SDimitry Andric 20190b57cec5SDimitry Andric template <class ELFT> bool RelrSection<ELFT>::updateAllocSize() { 20200b57cec5SDimitry Andric // This function computes the contents of an SHT_RELR packed relocation 20210b57cec5SDimitry Andric // section. 20220b57cec5SDimitry Andric // 20230b57cec5SDimitry Andric // Proposal for adding SHT_RELR sections to generic-abi is here: 20240b57cec5SDimitry Andric // https://groups.google.com/forum/#!topic/generic-abi/bX460iggiKg 20250b57cec5SDimitry Andric // 20260b57cec5SDimitry Andric // The encoded sequence of Elf64_Relr entries in a SHT_RELR section looks 20270b57cec5SDimitry Andric // like [ AAAAAAAA BBBBBBB1 BBBBBBB1 ... AAAAAAAA BBBBBB1 ... ] 20280b57cec5SDimitry Andric // 20290b57cec5SDimitry Andric // i.e. start with an address, followed by any number of bitmaps. The address 20300b57cec5SDimitry Andric // entry encodes 1 relocation. The subsequent bitmap entries encode up to 63 20310b57cec5SDimitry Andric // relocations each, at subsequent offsets following the last address entry. 20320b57cec5SDimitry Andric // 20330b57cec5SDimitry Andric // The bitmap entries must have 1 in the least significant bit. The assumption 20340b57cec5SDimitry Andric // here is that an address cannot have 1 in lsb. Odd addresses are not 20350b57cec5SDimitry Andric // supported. 20360b57cec5SDimitry Andric // 20370b57cec5SDimitry Andric // Excluding the least significant bit in the bitmap, each non-zero bit in 20380b57cec5SDimitry Andric // the bitmap represents a relocation to be applied to a corresponding machine 20390b57cec5SDimitry Andric // word that follows the base address word. The second least significant bit 20400b57cec5SDimitry Andric // represents the machine word immediately following the initial address, and 20410b57cec5SDimitry Andric // each bit that follows represents the next word, in linear order. As such, 20420b57cec5SDimitry Andric // a single bitmap can encode up to 31 relocations in a 32-bit object, and 20430b57cec5SDimitry Andric // 63 relocations in a 64-bit object. 20440b57cec5SDimitry Andric // 20450b57cec5SDimitry Andric // This encoding has a couple of interesting properties: 20460b57cec5SDimitry Andric // 1. Looking at any entry, it is clear whether it's an address or a bitmap: 20470b57cec5SDimitry Andric // even means address, odd means bitmap. 20480b57cec5SDimitry Andric // 2. Just a simple list of addresses is a valid encoding. 20490b57cec5SDimitry Andric 20500b57cec5SDimitry Andric size_t oldSize = relrRelocs.size(); 20510b57cec5SDimitry Andric relrRelocs.clear(); 20520b57cec5SDimitry Andric 20530b57cec5SDimitry Andric // Same as Config->Wordsize but faster because this is a compile-time 20540b57cec5SDimitry Andric // constant. 20550b57cec5SDimitry Andric const size_t wordsize = sizeof(typename ELFT::uint); 20560b57cec5SDimitry Andric 20570b57cec5SDimitry Andric // Number of bits to use for the relocation offsets bitmap. 20580b57cec5SDimitry Andric // Must be either 63 or 31. 20590b57cec5SDimitry Andric const size_t nBits = wordsize * 8 - 1; 20600b57cec5SDimitry Andric 20610b57cec5SDimitry Andric // Get offsets for all relative relocations and sort them. 206204eeddc0SDimitry Andric std::unique_ptr<uint64_t[]> offsets(new uint64_t[relocs.size()]); 2063bdd1243dSDimitry Andric for (auto [i, r] : llvm::enumerate(relocs)) 2064bdd1243dSDimitry Andric offsets[i] = r.getOffset(); 2065fcaf7f86SDimitry Andric llvm::sort(offsets.get(), offsets.get() + relocs.size()); 20660b57cec5SDimitry Andric 20670b57cec5SDimitry Andric // For each leading relocation, find following ones that can be folded 20680b57cec5SDimitry Andric // as a bitmap and fold them. 206904eeddc0SDimitry Andric for (size_t i = 0, e = relocs.size(); i != e;) { 20700b57cec5SDimitry Andric // Add a leading relocation. 20710b57cec5SDimitry Andric relrRelocs.push_back(Elf_Relr(offsets[i])); 20720b57cec5SDimitry Andric uint64_t base = offsets[i] + wordsize; 20730b57cec5SDimitry Andric ++i; 20740b57cec5SDimitry Andric 20750b57cec5SDimitry Andric // Find foldable relocations to construct bitmaps. 207604eeddc0SDimitry Andric for (;;) { 20770b57cec5SDimitry Andric uint64_t bitmap = 0; 207804eeddc0SDimitry Andric for (; i != e; ++i) { 207904eeddc0SDimitry Andric uint64_t d = offsets[i] - base; 208004eeddc0SDimitry Andric if (d >= nBits * wordsize || d % wordsize) 20810b57cec5SDimitry Andric break; 208204eeddc0SDimitry Andric bitmap |= uint64_t(1) << (d / wordsize); 20830b57cec5SDimitry Andric } 20840b57cec5SDimitry Andric if (!bitmap) 20850b57cec5SDimitry Andric break; 20860b57cec5SDimitry Andric relrRelocs.push_back(Elf_Relr((bitmap << 1) | 1)); 20870b57cec5SDimitry Andric base += nBits * wordsize; 20880b57cec5SDimitry Andric } 20890b57cec5SDimitry Andric } 20900b57cec5SDimitry Andric 209185868e8aSDimitry Andric // Don't allow the section to shrink; otherwise the size of the section can 209285868e8aSDimitry Andric // oscillate infinitely. Trailing 1s do not decode to more relocations. 209385868e8aSDimitry Andric if (relrRelocs.size() < oldSize) { 209485868e8aSDimitry Andric log(".relr.dyn needs " + Twine(oldSize - relrRelocs.size()) + 209585868e8aSDimitry Andric " padding word(s)"); 209685868e8aSDimitry Andric relrRelocs.resize(oldSize, Elf_Relr(1)); 209785868e8aSDimitry Andric } 209885868e8aSDimitry Andric 20990b57cec5SDimitry Andric return relrRelocs.size() != oldSize; 21000b57cec5SDimitry Andric } 21010b57cec5SDimitry Andric 21020b57cec5SDimitry Andric SymbolTableBaseSection::SymbolTableBaseSection(StringTableSection &strTabSec) 21030b57cec5SDimitry Andric : SyntheticSection(strTabSec.isDynamic() ? (uint64_t)SHF_ALLOC : 0, 21040b57cec5SDimitry Andric strTabSec.isDynamic() ? SHT_DYNSYM : SHT_SYMTAB, 21050b57cec5SDimitry Andric config->wordsize, 21060b57cec5SDimitry Andric strTabSec.isDynamic() ? ".dynsym" : ".symtab"), 21070b57cec5SDimitry Andric strTabSec(strTabSec) {} 21080b57cec5SDimitry Andric 21090b57cec5SDimitry Andric // Orders symbols according to their positions in the GOT, 21100b57cec5SDimitry Andric // in compliance with MIPS ABI rules. 21110b57cec5SDimitry Andric // See "Global Offset Table" in Chapter 5 in the following document 21120b57cec5SDimitry Andric // for detailed description: 21130b57cec5SDimitry Andric // ftp://www.linux-mips.org/pub/linux/mips/doc/ABI/mipsabi.pdf 21140b57cec5SDimitry Andric static bool sortMipsSymbols(const SymbolTableEntry &l, 21150b57cec5SDimitry Andric const SymbolTableEntry &r) { 21160b57cec5SDimitry Andric // Sort entries related to non-local preemptible symbols by GOT indexes. 21170b57cec5SDimitry Andric // All other entries go to the beginning of a dynsym in arbitrary order. 21180b57cec5SDimitry Andric if (l.sym->isInGot() && r.sym->isInGot()) 211904eeddc0SDimitry Andric return l.sym->getGotIdx() < r.sym->getGotIdx(); 21200b57cec5SDimitry Andric if (!l.sym->isInGot() && !r.sym->isInGot()) 21210b57cec5SDimitry Andric return false; 21220b57cec5SDimitry Andric return !l.sym->isInGot(); 21230b57cec5SDimitry Andric } 21240b57cec5SDimitry Andric 21250b57cec5SDimitry Andric void SymbolTableBaseSection::finalizeContents() { 21260b57cec5SDimitry Andric if (OutputSection *sec = strTabSec.getParent()) 21270b57cec5SDimitry Andric getParent()->link = sec->sectionIndex; 21280b57cec5SDimitry Andric 21290b57cec5SDimitry Andric if (this->type != SHT_DYNSYM) { 21300b57cec5SDimitry Andric sortSymTabSymbols(); 21310b57cec5SDimitry Andric return; 21320b57cec5SDimitry Andric } 21330b57cec5SDimitry Andric 21340b57cec5SDimitry Andric // If it is a .dynsym, there should be no local symbols, but we need 21350b57cec5SDimitry Andric // to do a few things for the dynamic linker. 21360b57cec5SDimitry Andric 21370b57cec5SDimitry Andric // Section's Info field has the index of the first non-local symbol. 21380b57cec5SDimitry Andric // Because the first symbol entry is a null entry, 1 is the first. 21390b57cec5SDimitry Andric getParent()->info = 1; 21400b57cec5SDimitry Andric 21410b57cec5SDimitry Andric if (getPartition().gnuHashTab) { 21420b57cec5SDimitry Andric // NB: It also sorts Symbols to meet the GNU hash table requirements. 21430b57cec5SDimitry Andric getPartition().gnuHashTab->addSymbols(symbols); 21440b57cec5SDimitry Andric } else if (config->emachine == EM_MIPS) { 21450b57cec5SDimitry Andric llvm::stable_sort(symbols, sortMipsSymbols); 21460b57cec5SDimitry Andric } 21470b57cec5SDimitry Andric 21480b57cec5SDimitry Andric // Only the main partition's dynsym indexes are stored in the symbols 21490b57cec5SDimitry Andric // themselves. All other partitions use a lookup table. 215004eeddc0SDimitry Andric if (this == mainPart->dynSymTab.get()) { 21510b57cec5SDimitry Andric size_t i = 0; 21520b57cec5SDimitry Andric for (const SymbolTableEntry &s : symbols) 21530b57cec5SDimitry Andric s.sym->dynsymIndex = ++i; 21540b57cec5SDimitry Andric } 21550b57cec5SDimitry Andric } 21560b57cec5SDimitry Andric 21570b57cec5SDimitry Andric // The ELF spec requires that all local symbols precede global symbols, so we 21580b57cec5SDimitry Andric // sort symbol entries in this function. (For .dynsym, we don't do that because 21590b57cec5SDimitry Andric // symbols for dynamic linking are inherently all globals.) 21600b57cec5SDimitry Andric // 21610b57cec5SDimitry Andric // Aside from above, we put local symbols in groups starting with the STT_FILE 21620b57cec5SDimitry Andric // symbol. That is convenient for purpose of identifying where are local symbols 21630b57cec5SDimitry Andric // coming from. 21640b57cec5SDimitry Andric void SymbolTableBaseSection::sortSymTabSymbols() { 21650b57cec5SDimitry Andric // Move all local symbols before global symbols. 21660b57cec5SDimitry Andric auto e = std::stable_partition( 216704eeddc0SDimitry Andric symbols.begin(), symbols.end(), 216804eeddc0SDimitry Andric [](const SymbolTableEntry &s) { return s.sym->isLocal(); }); 21690b57cec5SDimitry Andric size_t numLocals = e - symbols.begin(); 21700b57cec5SDimitry Andric getParent()->info = numLocals + 1; 21710b57cec5SDimitry Andric 21720b57cec5SDimitry Andric // We want to group the local symbols by file. For that we rebuild the local 21730b57cec5SDimitry Andric // part of the symbols vector. We do not need to care about the STT_FILE 21740b57cec5SDimitry Andric // symbols, they are already naturally placed first in each group. That 21750b57cec5SDimitry Andric // happens because STT_FILE is always the first symbol in the object and hence 21760b57cec5SDimitry Andric // precede all other local symbols we add for a file. 217704eeddc0SDimitry Andric MapVector<InputFile *, SmallVector<SymbolTableEntry, 0>> arr; 21780b57cec5SDimitry Andric for (const SymbolTableEntry &s : llvm::make_range(symbols.begin(), e)) 21790b57cec5SDimitry Andric arr[s.sym->file].push_back(s); 21800b57cec5SDimitry Andric 21810b57cec5SDimitry Andric auto i = symbols.begin(); 218204eeddc0SDimitry Andric for (auto &p : arr) 21830b57cec5SDimitry Andric for (SymbolTableEntry &entry : p.second) 21840b57cec5SDimitry Andric *i++ = entry; 21850b57cec5SDimitry Andric } 21860b57cec5SDimitry Andric 21870b57cec5SDimitry Andric void SymbolTableBaseSection::addSymbol(Symbol *b) { 21880b57cec5SDimitry Andric // Adding a local symbol to a .dynsym is a bug. 21890b57cec5SDimitry Andric assert(this->type != SHT_DYNSYM || !b->isLocal()); 219081ad6265SDimitry Andric symbols.push_back({b, strTabSec.addString(b->getName(), false)}); 21910b57cec5SDimitry Andric } 21920b57cec5SDimitry Andric 2193*0fca6ea1SDimitry Andric size_t SymbolTableBaseSection::getSymbolIndex(const Symbol &sym) { 219404eeddc0SDimitry Andric if (this == mainPart->dynSymTab.get()) 2195*0fca6ea1SDimitry Andric return sym.dynsymIndex; 21960b57cec5SDimitry Andric 21970b57cec5SDimitry Andric // Initializes symbol lookup tables lazily. This is used only for -r, 2198349cc55cSDimitry Andric // --emit-relocs and dynsyms in partitions other than the main one. 21990b57cec5SDimitry Andric llvm::call_once(onceFlag, [&] { 22000b57cec5SDimitry Andric symbolIndexMap.reserve(symbols.size()); 22010b57cec5SDimitry Andric size_t i = 0; 22020b57cec5SDimitry Andric for (const SymbolTableEntry &e : symbols) { 22030b57cec5SDimitry Andric if (e.sym->type == STT_SECTION) 22040b57cec5SDimitry Andric sectionIndexMap[e.sym->getOutputSection()] = ++i; 22050b57cec5SDimitry Andric else 22060b57cec5SDimitry Andric symbolIndexMap[e.sym] = ++i; 22070b57cec5SDimitry Andric } 22080b57cec5SDimitry Andric }); 22090b57cec5SDimitry Andric 22100b57cec5SDimitry Andric // Section symbols are mapped based on their output sections 22110b57cec5SDimitry Andric // to maintain their semantics. 2212*0fca6ea1SDimitry Andric if (sym.type == STT_SECTION) 2213*0fca6ea1SDimitry Andric return sectionIndexMap.lookup(sym.getOutputSection()); 2214*0fca6ea1SDimitry Andric return symbolIndexMap.lookup(&sym); 22150b57cec5SDimitry Andric } 22160b57cec5SDimitry Andric 22170b57cec5SDimitry Andric template <class ELFT> 22180b57cec5SDimitry Andric SymbolTableSection<ELFT>::SymbolTableSection(StringTableSection &strTabSec) 22190b57cec5SDimitry Andric : SymbolTableBaseSection(strTabSec) { 22200b57cec5SDimitry Andric this->entsize = sizeof(Elf_Sym); 22210b57cec5SDimitry Andric } 22220b57cec5SDimitry Andric 22230b57cec5SDimitry Andric static BssSection *getCommonSec(Symbol *sym) { 222481ad6265SDimitry Andric if (config->relocatable) 22250b57cec5SDimitry Andric if (auto *d = dyn_cast<Defined>(sym)) 22260b57cec5SDimitry Andric return dyn_cast_or_null<BssSection>(d->section); 22270b57cec5SDimitry Andric return nullptr; 22280b57cec5SDimitry Andric } 22290b57cec5SDimitry Andric 22300b57cec5SDimitry Andric static uint32_t getSymSectionIndex(Symbol *sym) { 2231bdd1243dSDimitry Andric assert(!(sym->hasFlag(NEEDS_COPY) && sym->isObject())); 2232bdd1243dSDimitry Andric if (!isa<Defined>(sym) || sym->hasFlag(NEEDS_COPY)) 22330b57cec5SDimitry Andric return SHN_UNDEF; 22340b57cec5SDimitry Andric if (const OutputSection *os = sym->getOutputSection()) 22350b57cec5SDimitry Andric return os->sectionIndex >= SHN_LORESERVE ? (uint32_t)SHN_XINDEX 22360b57cec5SDimitry Andric : os->sectionIndex; 22370b57cec5SDimitry Andric return SHN_ABS; 22380b57cec5SDimitry Andric } 22390b57cec5SDimitry Andric 22400b57cec5SDimitry Andric // Write the internal symbol table contents to the output symbol table. 22410b57cec5SDimitry Andric template <class ELFT> void SymbolTableSection<ELFT>::writeTo(uint8_t *buf) { 22420b57cec5SDimitry Andric // The first entry is a null entry as per the ELF spec. 22430b57cec5SDimitry Andric buf += sizeof(Elf_Sym); 22440b57cec5SDimitry Andric 22450b57cec5SDimitry Andric auto *eSym = reinterpret_cast<Elf_Sym *>(buf); 22460b57cec5SDimitry Andric 22470b57cec5SDimitry Andric for (SymbolTableEntry &ent : symbols) { 22480b57cec5SDimitry Andric Symbol *sym = ent.sym; 22490b57cec5SDimitry Andric bool isDefinedHere = type == SHT_SYMTAB || sym->partition == partition; 22500b57cec5SDimitry Andric 225104eeddc0SDimitry Andric // Set st_name, st_info and st_other. 225204eeddc0SDimitry Andric eSym->st_name = ent.strTabOffset; 225304eeddc0SDimitry Andric eSym->setBindingAndType(sym->binding, sym->type); 2254bdd1243dSDimitry Andric eSym->st_other = sym->stOther; 22550b57cec5SDimitry Andric 225604eeddc0SDimitry Andric if (BssSection *commonSec = getCommonSec(sym)) { 225781ad6265SDimitry Andric // When -r is specified, a COMMON symbol is not allocated. Its st_shndx 225881ad6265SDimitry Andric // holds SHN_COMMON and st_value holds the alignment. 225904eeddc0SDimitry Andric eSym->st_shndx = SHN_COMMON; 2260bdd1243dSDimitry Andric eSym->st_value = commonSec->addralign; 226104eeddc0SDimitry Andric eSym->st_size = cast<Defined>(sym)->size; 226204eeddc0SDimitry Andric } else { 226304eeddc0SDimitry Andric const uint32_t shndx = getSymSectionIndex(sym); 226404eeddc0SDimitry Andric if (isDefinedHere) { 226504eeddc0SDimitry Andric eSym->st_shndx = shndx; 22660b57cec5SDimitry Andric eSym->st_value = sym->getVA(); 226704eeddc0SDimitry Andric // Copy symbol size if it is a defined symbol. st_size is not 226804eeddc0SDimitry Andric // significant for undefined symbols, so whether copying it or not is up 226904eeddc0SDimitry Andric // to us if that's the case. We'll leave it as zero because by not 227004eeddc0SDimitry Andric // setting a value, we can get the exact same outputs for two sets of 227104eeddc0SDimitry Andric // input files that differ only in undefined symbol size in DSOs. 227204eeddc0SDimitry Andric eSym->st_size = shndx != SHN_UNDEF ? cast<Defined>(sym)->size : 0; 227304eeddc0SDimitry Andric } else { 227404eeddc0SDimitry Andric eSym->st_shndx = 0; 22750b57cec5SDimitry Andric eSym->st_value = 0; 227604eeddc0SDimitry Andric eSym->st_size = 0; 227704eeddc0SDimitry Andric } 227804eeddc0SDimitry Andric } 22790b57cec5SDimitry Andric 22800b57cec5SDimitry Andric ++eSym; 22810b57cec5SDimitry Andric } 22820b57cec5SDimitry Andric 22830b57cec5SDimitry Andric // On MIPS we need to mark symbol which has a PLT entry and requires 22840b57cec5SDimitry Andric // pointer equality by STO_MIPS_PLT flag. That is necessary to help 22850b57cec5SDimitry Andric // dynamic linker distinguish such symbols and MIPS lazy-binding stubs. 22860b57cec5SDimitry Andric // https://sourceware.org/ml/binutils/2008-07/txt00000.txt 22870b57cec5SDimitry Andric if (config->emachine == EM_MIPS) { 22880b57cec5SDimitry Andric auto *eSym = reinterpret_cast<Elf_Sym *>(buf); 22890b57cec5SDimitry Andric 22900b57cec5SDimitry Andric for (SymbolTableEntry &ent : symbols) { 22910b57cec5SDimitry Andric Symbol *sym = ent.sym; 2292bdd1243dSDimitry Andric if (sym->isInPlt() && sym->hasFlag(NEEDS_COPY)) 22930b57cec5SDimitry Andric eSym->st_other |= STO_MIPS_PLT; 22940b57cec5SDimitry Andric if (isMicroMips()) { 22950b57cec5SDimitry Andric // We already set the less-significant bit for symbols 22960b57cec5SDimitry Andric // marked by the `STO_MIPS_MICROMIPS` flag and for microMIPS PLT 22970b57cec5SDimitry Andric // records. That allows us to distinguish such symbols in 22985ffd83dbSDimitry Andric // the `MIPS<ELFT>::relocate()` routine. Now we should 22990b57cec5SDimitry Andric // clear that bit for non-dynamic symbol table, so tools 23000b57cec5SDimitry Andric // like `objdump` will be able to deal with a correct 23010b57cec5SDimitry Andric // symbol position. 23020b57cec5SDimitry Andric if (sym->isDefined() && 2303bdd1243dSDimitry Andric ((sym->stOther & STO_MIPS_MICROMIPS) || sym->hasFlag(NEEDS_COPY))) { 23040b57cec5SDimitry Andric if (!strTabSec.isDynamic()) 23050b57cec5SDimitry Andric eSym->st_value &= ~1; 23060b57cec5SDimitry Andric eSym->st_other |= STO_MIPS_MICROMIPS; 23070b57cec5SDimitry Andric } 23080b57cec5SDimitry Andric } 23090b57cec5SDimitry Andric if (config->relocatable) 23100b57cec5SDimitry Andric if (auto *d = dyn_cast<Defined>(sym)) 23110b57cec5SDimitry Andric if (isMipsPIC<ELFT>(d)) 23120b57cec5SDimitry Andric eSym->st_other |= STO_MIPS_PIC; 23130b57cec5SDimitry Andric ++eSym; 23140b57cec5SDimitry Andric } 23150b57cec5SDimitry Andric } 23160b57cec5SDimitry Andric } 23170b57cec5SDimitry Andric 23180b57cec5SDimitry Andric SymtabShndxSection::SymtabShndxSection() 23190b57cec5SDimitry Andric : SyntheticSection(0, SHT_SYMTAB_SHNDX, 4, ".symtab_shndx") { 23200b57cec5SDimitry Andric this->entsize = 4; 23210b57cec5SDimitry Andric } 23220b57cec5SDimitry Andric 23230b57cec5SDimitry Andric void SymtabShndxSection::writeTo(uint8_t *buf) { 23240b57cec5SDimitry Andric // We write an array of 32 bit values, where each value has 1:1 association 23250b57cec5SDimitry Andric // with an entry in .symtab. If the corresponding entry contains SHN_XINDEX, 23260b57cec5SDimitry Andric // we need to write actual index, otherwise, we must write SHN_UNDEF(0). 23270b57cec5SDimitry Andric buf += 4; // Ignore .symtab[0] entry. 23280b57cec5SDimitry Andric for (const SymbolTableEntry &entry : in.symTab->getSymbols()) { 232904eeddc0SDimitry Andric if (!getCommonSec(entry.sym) && getSymSectionIndex(entry.sym) == SHN_XINDEX) 23300b57cec5SDimitry Andric write32(buf, entry.sym->getOutputSection()->sectionIndex); 23310b57cec5SDimitry Andric buf += 4; 23320b57cec5SDimitry Andric } 23330b57cec5SDimitry Andric } 23340b57cec5SDimitry Andric 23350b57cec5SDimitry Andric bool SymtabShndxSection::isNeeded() const { 23360b57cec5SDimitry Andric // SHT_SYMTAB can hold symbols with section indices values up to 23370b57cec5SDimitry Andric // SHN_LORESERVE. If we need more, we want to use extension SHT_SYMTAB_SHNDX 23380b57cec5SDimitry Andric // section. Problem is that we reveal the final section indices a bit too 23390b57cec5SDimitry Andric // late, and we do not know them here. For simplicity, we just always create 23400b57cec5SDimitry Andric // a .symtab_shndx section when the amount of output sections is huge. 23410b57cec5SDimitry Andric size_t size = 0; 23424824e7fdSDimitry Andric for (SectionCommand *cmd : script->sectionCommands) 234381ad6265SDimitry Andric if (isa<OutputDesc>(cmd)) 23440b57cec5SDimitry Andric ++size; 23450b57cec5SDimitry Andric return size >= SHN_LORESERVE; 23460b57cec5SDimitry Andric } 23470b57cec5SDimitry Andric 23480b57cec5SDimitry Andric void SymtabShndxSection::finalizeContents() { 23490b57cec5SDimitry Andric getParent()->link = in.symTab->getParent()->sectionIndex; 23500b57cec5SDimitry Andric } 23510b57cec5SDimitry Andric 23520b57cec5SDimitry Andric size_t SymtabShndxSection::getSize() const { 23530b57cec5SDimitry Andric return in.symTab->getNumSymbols() * 4; 23540b57cec5SDimitry Andric } 23550b57cec5SDimitry Andric 23560b57cec5SDimitry Andric // .hash and .gnu.hash sections contain on-disk hash tables that map 23570b57cec5SDimitry Andric // symbol names to their dynamic symbol table indices. Their purpose 23580b57cec5SDimitry Andric // is to help the dynamic linker resolve symbols quickly. If ELF files 23590b57cec5SDimitry Andric // don't have them, the dynamic linker has to do linear search on all 23600b57cec5SDimitry Andric // dynamic symbols, which makes programs slower. Therefore, a .hash 2361349cc55cSDimitry Andric // section is added to a DSO by default. 23620b57cec5SDimitry Andric // 23630b57cec5SDimitry Andric // The Unix semantics of resolving dynamic symbols is somewhat expensive. 23640b57cec5SDimitry Andric // Each ELF file has a list of DSOs that the ELF file depends on and a 23650b57cec5SDimitry Andric // list of dynamic symbols that need to be resolved from any of the 23660b57cec5SDimitry Andric // DSOs. That means resolving all dynamic symbols takes O(m)*O(n) 23670b57cec5SDimitry Andric // where m is the number of DSOs and n is the number of dynamic 23680b57cec5SDimitry Andric // symbols. For modern large programs, both m and n are large. So 23695ffd83dbSDimitry Andric // making each step faster by using hash tables substantially 23700b57cec5SDimitry Andric // improves time to load programs. 23710b57cec5SDimitry Andric // 23720b57cec5SDimitry Andric // (Note that this is not the only way to design the shared library. 23730b57cec5SDimitry Andric // For instance, the Windows DLL takes a different approach. On 23740b57cec5SDimitry Andric // Windows, each dynamic symbol has a name of DLL from which the symbol 23750b57cec5SDimitry Andric // has to be resolved. That makes the cost of symbol resolution O(n). 23760b57cec5SDimitry Andric // This disables some hacky techniques you can use on Unix such as 23770b57cec5SDimitry Andric // LD_PRELOAD, but this is arguably better semantics than the Unix ones.) 23780b57cec5SDimitry Andric // 23790b57cec5SDimitry Andric // Due to historical reasons, we have two different hash tables, .hash 23800b57cec5SDimitry Andric // and .gnu.hash. They are for the same purpose, and .gnu.hash is a new 23810b57cec5SDimitry Andric // and better version of .hash. .hash is just an on-disk hash table, but 23820b57cec5SDimitry Andric // .gnu.hash has a bloom filter in addition to a hash table to skip 23830b57cec5SDimitry Andric // DSOs very quickly. If you are sure that your dynamic linker knows 2384349cc55cSDimitry Andric // about .gnu.hash, you want to specify --hash-style=gnu. Otherwise, a 2385349cc55cSDimitry Andric // safe bet is to specify --hash-style=both for backward compatibility. 23860b57cec5SDimitry Andric GnuHashTableSection::GnuHashTableSection() 23870b57cec5SDimitry Andric : SyntheticSection(SHF_ALLOC, SHT_GNU_HASH, config->wordsize, ".gnu.hash") { 23880b57cec5SDimitry Andric } 23890b57cec5SDimitry Andric 23900b57cec5SDimitry Andric void GnuHashTableSection::finalizeContents() { 23910b57cec5SDimitry Andric if (OutputSection *sec = getPartition().dynSymTab->getParent()) 23920b57cec5SDimitry Andric getParent()->link = sec->sectionIndex; 23930b57cec5SDimitry Andric 23940b57cec5SDimitry Andric // Computes bloom filter size in word size. We want to allocate 12 23950b57cec5SDimitry Andric // bits for each symbol. It must be a power of two. 23960b57cec5SDimitry Andric if (symbols.empty()) { 23970b57cec5SDimitry Andric maskWords = 1; 23980b57cec5SDimitry Andric } else { 23990b57cec5SDimitry Andric uint64_t numBits = symbols.size() * 12; 24000b57cec5SDimitry Andric maskWords = NextPowerOf2(numBits / (config->wordsize * 8)); 24010b57cec5SDimitry Andric } 24020b57cec5SDimitry Andric 24030b57cec5SDimitry Andric size = 16; // Header 24040b57cec5SDimitry Andric size += config->wordsize * maskWords; // Bloom filter 24050b57cec5SDimitry Andric size += nBuckets * 4; // Hash buckets 24060b57cec5SDimitry Andric size += symbols.size() * 4; // Hash values 24070b57cec5SDimitry Andric } 24080b57cec5SDimitry Andric 24090b57cec5SDimitry Andric void GnuHashTableSection::writeTo(uint8_t *buf) { 24100b57cec5SDimitry Andric // Write a header. 24110b57cec5SDimitry Andric write32(buf, nBuckets); 24120b57cec5SDimitry Andric write32(buf + 4, getPartition().dynSymTab->getNumSymbols() - symbols.size()); 24130b57cec5SDimitry Andric write32(buf + 8, maskWords); 24140b57cec5SDimitry Andric write32(buf + 12, Shift2); 24150b57cec5SDimitry Andric buf += 16; 24160b57cec5SDimitry Andric 24174824e7fdSDimitry Andric // Write the 2-bit bloom filter. 24184824e7fdSDimitry Andric const unsigned c = config->is64 ? 64 : 32; 24190b57cec5SDimitry Andric for (const Entry &sym : symbols) { 24200b57cec5SDimitry Andric // When C = 64, we choose a word with bits [6:...] and set 1 to two bits in 24210b57cec5SDimitry Andric // the word using bits [0:5] and [26:31]. 24220b57cec5SDimitry Andric size_t i = (sym.hash / c) & (maskWords - 1); 24230b57cec5SDimitry Andric uint64_t val = readUint(buf + i * config->wordsize); 24240b57cec5SDimitry Andric val |= uint64_t(1) << (sym.hash % c); 24250b57cec5SDimitry Andric val |= uint64_t(1) << ((sym.hash >> Shift2) % c); 24260b57cec5SDimitry Andric writeUint(buf + i * config->wordsize, val); 24270b57cec5SDimitry Andric } 24284824e7fdSDimitry Andric buf += config->wordsize * maskWords; 24290b57cec5SDimitry Andric 24304824e7fdSDimitry Andric // Write the hash table. 24310b57cec5SDimitry Andric uint32_t *buckets = reinterpret_cast<uint32_t *>(buf); 24320b57cec5SDimitry Andric uint32_t oldBucket = -1; 24330b57cec5SDimitry Andric uint32_t *values = buckets + nBuckets; 24340b57cec5SDimitry Andric for (auto i = symbols.begin(), e = symbols.end(); i != e; ++i) { 24350b57cec5SDimitry Andric // Write a hash value. It represents a sequence of chains that share the 24360b57cec5SDimitry Andric // same hash modulo value. The last element of each chain is terminated by 24370b57cec5SDimitry Andric // LSB 1. 24380b57cec5SDimitry Andric uint32_t hash = i->hash; 24390b57cec5SDimitry Andric bool isLastInChain = (i + 1) == e || i->bucketIdx != (i + 1)->bucketIdx; 24400b57cec5SDimitry Andric hash = isLastInChain ? hash | 1 : hash & ~1; 24410b57cec5SDimitry Andric write32(values++, hash); 24420b57cec5SDimitry Andric 24430b57cec5SDimitry Andric if (i->bucketIdx == oldBucket) 24440b57cec5SDimitry Andric continue; 24450b57cec5SDimitry Andric // Write a hash bucket. Hash buckets contain indices in the following hash 24460b57cec5SDimitry Andric // value table. 24470b57cec5SDimitry Andric write32(buckets + i->bucketIdx, 2448*0fca6ea1SDimitry Andric getPartition().dynSymTab->getSymbolIndex(*i->sym)); 24490b57cec5SDimitry Andric oldBucket = i->bucketIdx; 24500b57cec5SDimitry Andric } 24510b57cec5SDimitry Andric } 24520b57cec5SDimitry Andric 24530b57cec5SDimitry Andric // Add symbols to this symbol hash table. Note that this function 24540b57cec5SDimitry Andric // destructively sort a given vector -- which is needed because 24550b57cec5SDimitry Andric // GNU-style hash table places some sorting requirements. 24560eae32dcSDimitry Andric void GnuHashTableSection::addSymbols(SmallVectorImpl<SymbolTableEntry> &v) { 24570b57cec5SDimitry Andric // We cannot use 'auto' for Mid because GCC 6.1 cannot deduce 24580b57cec5SDimitry Andric // its type correctly. 24590eae32dcSDimitry Andric auto mid = 24600b57cec5SDimitry Andric std::stable_partition(v.begin(), v.end(), [&](const SymbolTableEntry &s) { 24610b57cec5SDimitry Andric return !s.sym->isDefined() || s.sym->partition != partition; 24620b57cec5SDimitry Andric }); 24630b57cec5SDimitry Andric 24640b57cec5SDimitry Andric // We chose load factor 4 for the on-disk hash table. For each hash 24650b57cec5SDimitry Andric // collision, the dynamic linker will compare a uint32_t hash value. 24660b57cec5SDimitry Andric // Since the integer comparison is quite fast, we believe we can 24670b57cec5SDimitry Andric // make the load factor even larger. 4 is just a conservative choice. 24680b57cec5SDimitry Andric // 24690b57cec5SDimitry Andric // Note that we don't want to create a zero-sized hash table because 24700b57cec5SDimitry Andric // Android loader as of 2018 doesn't like a .gnu.hash containing such 24710b57cec5SDimitry Andric // table. If that's the case, we create a hash table with one unused 24720b57cec5SDimitry Andric // dummy slot. 24730b57cec5SDimitry Andric nBuckets = std::max<size_t>((v.end() - mid) / 4, 1); 24740b57cec5SDimitry Andric 24750b57cec5SDimitry Andric if (mid == v.end()) 24760b57cec5SDimitry Andric return; 24770b57cec5SDimitry Andric 24780b57cec5SDimitry Andric for (SymbolTableEntry &ent : llvm::make_range(mid, v.end())) { 24790b57cec5SDimitry Andric Symbol *b = ent.sym; 24800b57cec5SDimitry Andric uint32_t hash = hashGnu(b->getName()); 24810b57cec5SDimitry Andric uint32_t bucketIdx = hash % nBuckets; 24820b57cec5SDimitry Andric symbols.push_back({b, ent.strTabOffset, hash, bucketIdx}); 24830b57cec5SDimitry Andric } 24840b57cec5SDimitry Andric 248504eeddc0SDimitry Andric llvm::sort(symbols, [](const Entry &l, const Entry &r) { 248604eeddc0SDimitry Andric return std::tie(l.bucketIdx, l.strTabOffset) < 248704eeddc0SDimitry Andric std::tie(r.bucketIdx, r.strTabOffset); 24880b57cec5SDimitry Andric }); 24890b57cec5SDimitry Andric 24900b57cec5SDimitry Andric v.erase(mid, v.end()); 24910b57cec5SDimitry Andric for (const Entry &ent : symbols) 24920b57cec5SDimitry Andric v.push_back({ent.sym, ent.strTabOffset}); 24930b57cec5SDimitry Andric } 24940b57cec5SDimitry Andric 24950b57cec5SDimitry Andric HashTableSection::HashTableSection() 24960b57cec5SDimitry Andric : SyntheticSection(SHF_ALLOC, SHT_HASH, 4, ".hash") { 24970b57cec5SDimitry Andric this->entsize = 4; 24980b57cec5SDimitry Andric } 24990b57cec5SDimitry Andric 25000b57cec5SDimitry Andric void HashTableSection::finalizeContents() { 250104eeddc0SDimitry Andric SymbolTableBaseSection *symTab = getPartition().dynSymTab.get(); 25020b57cec5SDimitry Andric 25030b57cec5SDimitry Andric if (OutputSection *sec = symTab->getParent()) 25040b57cec5SDimitry Andric getParent()->link = sec->sectionIndex; 25050b57cec5SDimitry Andric 25060b57cec5SDimitry Andric unsigned numEntries = 2; // nbucket and nchain. 25070b57cec5SDimitry Andric numEntries += symTab->getNumSymbols(); // The chain entries. 25080b57cec5SDimitry Andric 25090b57cec5SDimitry Andric // Create as many buckets as there are symbols. 25100b57cec5SDimitry Andric numEntries += symTab->getNumSymbols(); 25110b57cec5SDimitry Andric this->size = numEntries * 4; 25120b57cec5SDimitry Andric } 25130b57cec5SDimitry Andric 25140b57cec5SDimitry Andric void HashTableSection::writeTo(uint8_t *buf) { 251504eeddc0SDimitry Andric SymbolTableBaseSection *symTab = getPartition().dynSymTab.get(); 25160b57cec5SDimitry Andric unsigned numSymbols = symTab->getNumSymbols(); 25170b57cec5SDimitry Andric 25180b57cec5SDimitry Andric uint32_t *p = reinterpret_cast<uint32_t *>(buf); 25190b57cec5SDimitry Andric write32(p++, numSymbols); // nbucket 25200b57cec5SDimitry Andric write32(p++, numSymbols); // nchain 25210b57cec5SDimitry Andric 25220b57cec5SDimitry Andric uint32_t *buckets = p; 25230b57cec5SDimitry Andric uint32_t *chains = p + numSymbols; 25240b57cec5SDimitry Andric 25250b57cec5SDimitry Andric for (const SymbolTableEntry &s : symTab->getSymbols()) { 25260b57cec5SDimitry Andric Symbol *sym = s.sym; 25270b57cec5SDimitry Andric StringRef name = sym->getName(); 25280b57cec5SDimitry Andric unsigned i = sym->dynsymIndex; 25290b57cec5SDimitry Andric uint32_t hash = hashSysV(name) % numSymbols; 25300b57cec5SDimitry Andric chains[i] = buckets[hash]; 25310b57cec5SDimitry Andric write32(buckets + hash, i); 25320b57cec5SDimitry Andric } 25330b57cec5SDimitry Andric } 25340b57cec5SDimitry Andric 2535480093f4SDimitry Andric PltSection::PltSection() 2536480093f4SDimitry Andric : SyntheticSection(SHF_ALLOC | SHF_EXECINSTR, SHT_PROGBITS, 16, ".plt"), 2537480093f4SDimitry Andric headerSize(target->pltHeaderSize) { 2538480093f4SDimitry Andric // On PowerPC, this section contains lazy symbol resolvers. 253992c0d181SDimitry Andric if (config->emachine == EM_PPC64) { 2540480093f4SDimitry Andric name = ".glink"; 2541bdd1243dSDimitry Andric addralign = 4; 2542480093f4SDimitry Andric } 2543480093f4SDimitry Andric 2544480093f4SDimitry Andric // On x86 when IBT is enabled, this section contains the second PLT (lazy 2545480093f4SDimitry Andric // symbol resolvers). 2546480093f4SDimitry Andric if ((config->emachine == EM_386 || config->emachine == EM_X86_64) && 2547480093f4SDimitry Andric (config->andFeatures & GNU_PROPERTY_X86_FEATURE_1_IBT)) 2548480093f4SDimitry Andric name = ".plt.sec"; 2549480093f4SDimitry Andric 25500b57cec5SDimitry Andric // The PLT needs to be writable on SPARC as the dynamic linker will 25510b57cec5SDimitry Andric // modify the instructions in the PLT entries. 25520b57cec5SDimitry Andric if (config->emachine == EM_SPARCV9) 25530b57cec5SDimitry Andric this->flags |= SHF_WRITE; 25540b57cec5SDimitry Andric } 25550b57cec5SDimitry Andric 25560b57cec5SDimitry Andric void PltSection::writeTo(uint8_t *buf) { 2557480093f4SDimitry Andric // At beginning of PLT, we have code to call the dynamic 25580b57cec5SDimitry Andric // linker to resolve dynsyms at runtime. Write such code. 25590b57cec5SDimitry Andric target->writePltHeader(buf); 25600b57cec5SDimitry Andric size_t off = headerSize; 25610b57cec5SDimitry Andric 2562480093f4SDimitry Andric for (const Symbol *sym : entries) { 2563480093f4SDimitry Andric target->writePlt(buf + off, *sym, getVA() + off); 25640b57cec5SDimitry Andric off += target->pltEntrySize; 25650b57cec5SDimitry Andric } 25660b57cec5SDimitry Andric } 25670b57cec5SDimitry Andric 2568480093f4SDimitry Andric void PltSection::addEntry(Symbol &sym) { 256904eeddc0SDimitry Andric assert(sym.auxIdx == symAux.size() - 1); 257004eeddc0SDimitry Andric symAux.back().pltIdx = entries.size(); 25710b57cec5SDimitry Andric entries.push_back(&sym); 25720b57cec5SDimitry Andric } 25730b57cec5SDimitry Andric 25740b57cec5SDimitry Andric size_t PltSection::getSize() const { 257592c0d181SDimitry Andric return headerSize + entries.size() * target->pltEntrySize; 25760b57cec5SDimitry Andric } 25770b57cec5SDimitry Andric 2578480093f4SDimitry Andric bool PltSection::isNeeded() const { 2579480093f4SDimitry Andric // For -z retpolineplt, .iplt needs the .plt header. 2580480093f4SDimitry Andric return !entries.empty() || (config->zRetpolineplt && in.iplt->isNeeded()); 2581480093f4SDimitry Andric } 2582480093f4SDimitry Andric 2583480093f4SDimitry Andric // Used by ARM to add mapping symbols in the PLT section, which aid 2584480093f4SDimitry Andric // disassembly. 25850b57cec5SDimitry Andric void PltSection::addSymbols() { 25860b57cec5SDimitry Andric target->addPltHeaderSymbols(*this); 25870b57cec5SDimitry Andric 25880b57cec5SDimitry Andric size_t off = headerSize; 25890b57cec5SDimitry Andric for (size_t i = 0; i < entries.size(); ++i) { 25900b57cec5SDimitry Andric target->addPltSymbols(*this, off); 25910b57cec5SDimitry Andric off += target->pltEntrySize; 25920b57cec5SDimitry Andric } 25930b57cec5SDimitry Andric } 25940b57cec5SDimitry Andric 2595480093f4SDimitry Andric IpltSection::IpltSection() 2596480093f4SDimitry Andric : SyntheticSection(SHF_ALLOC | SHF_EXECINSTR, SHT_PROGBITS, 16, ".iplt") { 2597480093f4SDimitry Andric if (config->emachine == EM_PPC || config->emachine == EM_PPC64) { 2598480093f4SDimitry Andric name = ".glink"; 2599bdd1243dSDimitry Andric addralign = 4; 2600480093f4SDimitry Andric } 2601480093f4SDimitry Andric } 2602480093f4SDimitry Andric 2603480093f4SDimitry Andric void IpltSection::writeTo(uint8_t *buf) { 2604480093f4SDimitry Andric uint32_t off = 0; 2605480093f4SDimitry Andric for (const Symbol *sym : entries) { 2606480093f4SDimitry Andric target->writeIplt(buf + off, *sym, getVA() + off); 2607480093f4SDimitry Andric off += target->ipltEntrySize; 2608480093f4SDimitry Andric } 2609480093f4SDimitry Andric } 2610480093f4SDimitry Andric 2611480093f4SDimitry Andric size_t IpltSection::getSize() const { 2612480093f4SDimitry Andric return entries.size() * target->ipltEntrySize; 2613480093f4SDimitry Andric } 2614480093f4SDimitry Andric 2615480093f4SDimitry Andric void IpltSection::addEntry(Symbol &sym) { 261604eeddc0SDimitry Andric assert(sym.auxIdx == symAux.size() - 1); 261704eeddc0SDimitry Andric symAux.back().pltIdx = entries.size(); 2618480093f4SDimitry Andric entries.push_back(&sym); 2619480093f4SDimitry Andric } 2620480093f4SDimitry Andric 2621480093f4SDimitry Andric // ARM uses mapping symbols to aid disassembly. 2622480093f4SDimitry Andric void IpltSection::addSymbols() { 2623480093f4SDimitry Andric size_t off = 0; 2624480093f4SDimitry Andric for (size_t i = 0, e = entries.size(); i != e; ++i) { 2625480093f4SDimitry Andric target->addPltSymbols(*this, off); 2626480093f4SDimitry Andric off += target->pltEntrySize; 2627480093f4SDimitry Andric } 2628480093f4SDimitry Andric } 2629480093f4SDimitry Andric 263092c0d181SDimitry Andric PPC32GlinkSection::PPC32GlinkSection() { 263192c0d181SDimitry Andric name = ".glink"; 2632bdd1243dSDimitry Andric addralign = 4; 263392c0d181SDimitry Andric } 263492c0d181SDimitry Andric 263592c0d181SDimitry Andric void PPC32GlinkSection::writeTo(uint8_t *buf) { 263692c0d181SDimitry Andric writePPC32GlinkSection(buf, entries.size()); 263792c0d181SDimitry Andric } 263892c0d181SDimitry Andric 263992c0d181SDimitry Andric size_t PPC32GlinkSection::getSize() const { 264092c0d181SDimitry Andric return headerSize + entries.size() * target->pltEntrySize + footerSize; 264192c0d181SDimitry Andric } 264292c0d181SDimitry Andric 2643480093f4SDimitry Andric // This is an x86-only extra PLT section and used only when a security 2644480093f4SDimitry Andric // enhancement feature called CET is enabled. In this comment, I'll explain what 2645480093f4SDimitry Andric // the feature is and why we have two PLT sections if CET is enabled. 2646480093f4SDimitry Andric // 2647480093f4SDimitry Andric // So, what does CET do? CET introduces a new restriction to indirect jump 2648480093f4SDimitry Andric // instructions. CET works this way. Assume that CET is enabled. Then, if you 2649480093f4SDimitry Andric // execute an indirect jump instruction, the processor verifies that a special 2650480093f4SDimitry Andric // "landing pad" instruction (which is actually a repurposed NOP instruction and 2651480093f4SDimitry Andric // now called "endbr32" or "endbr64") is at the jump target. If the jump target 2652480093f4SDimitry Andric // does not start with that instruction, the processor raises an exception 2653480093f4SDimitry Andric // instead of continuing executing code. 2654480093f4SDimitry Andric // 2655480093f4SDimitry Andric // If CET is enabled, the compiler emits endbr to all locations where indirect 2656480093f4SDimitry Andric // jumps may jump to. 2657480093f4SDimitry Andric // 2658480093f4SDimitry Andric // This mechanism makes it extremely hard to transfer the control to a middle of 2659480093f4SDimitry Andric // a function that is not supporsed to be a indirect jump target, preventing 2660480093f4SDimitry Andric // certain types of attacks such as ROP or JOP. 2661480093f4SDimitry Andric // 2662480093f4SDimitry Andric // Note that the processors in the market as of 2019 don't actually support the 2663480093f4SDimitry Andric // feature. Only the spec is available at the moment. 2664480093f4SDimitry Andric // 2665480093f4SDimitry Andric // Now, I'll explain why we have this extra PLT section for CET. 2666480093f4SDimitry Andric // 2667480093f4SDimitry Andric // Since you can indirectly jump to a PLT entry, we have to make PLT entries 2668480093f4SDimitry Andric // start with endbr. The problem is there's no extra space for endbr (which is 4 2669480093f4SDimitry Andric // bytes long), as the PLT entry is only 16 bytes long and all bytes are already 2670480093f4SDimitry Andric // used. 2671480093f4SDimitry Andric // 2672480093f4SDimitry Andric // In order to deal with the issue, we split a PLT entry into two PLT entries. 2673480093f4SDimitry Andric // Remember that each PLT entry contains code to jump to an address read from 2674480093f4SDimitry Andric // .got.plt AND code to resolve a dynamic symbol lazily. With the 2-PLT scheme, 2675480093f4SDimitry Andric // the former code is written to .plt.sec, and the latter code is written to 2676480093f4SDimitry Andric // .plt. 2677480093f4SDimitry Andric // 2678480093f4SDimitry Andric // Lazy symbol resolution in the 2-PLT scheme works in the usual way, except 2679480093f4SDimitry Andric // that the regular .plt is now called .plt.sec and .plt is repurposed to 2680480093f4SDimitry Andric // contain only code for lazy symbol resolution. 2681480093f4SDimitry Andric // 2682480093f4SDimitry Andric // In other words, this is how the 2-PLT scheme works. Application code is 2683480093f4SDimitry Andric // supposed to jump to .plt.sec to call an external function. Each .plt.sec 2684480093f4SDimitry Andric // entry contains code to read an address from a corresponding .got.plt entry 2685480093f4SDimitry Andric // and jump to that address. Addresses in .got.plt initially point to .plt, so 2686480093f4SDimitry Andric // when an application calls an external function for the first time, the 2687480093f4SDimitry Andric // control is transferred to a function that resolves a symbol name from 2688480093f4SDimitry Andric // external shared object files. That function then rewrites a .got.plt entry 2689480093f4SDimitry Andric // with a resolved address, so that the subsequent function calls directly jump 2690480093f4SDimitry Andric // to a desired location from .plt.sec. 2691480093f4SDimitry Andric // 2692480093f4SDimitry Andric // There is an open question as to whether the 2-PLT scheme was desirable or 2693480093f4SDimitry Andric // not. We could have simply extended the PLT entry size to 32-bytes to 2694480093f4SDimitry Andric // accommodate endbr, and that scheme would have been much simpler than the 2695480093f4SDimitry Andric // 2-PLT scheme. One reason to split PLT was, by doing that, we could keep hot 2696480093f4SDimitry Andric // code (.plt.sec) from cold code (.plt). But as far as I know no one proved 2697480093f4SDimitry Andric // that the optimization actually makes a difference. 2698480093f4SDimitry Andric // 2699480093f4SDimitry Andric // That said, the 2-PLT scheme is a part of the ABI, debuggers and other tools 2700480093f4SDimitry Andric // depend on it, so we implement the ABI. 2701480093f4SDimitry Andric IBTPltSection::IBTPltSection() 2702480093f4SDimitry Andric : SyntheticSection(SHF_ALLOC | SHF_EXECINSTR, SHT_PROGBITS, 16, ".plt") {} 2703480093f4SDimitry Andric 2704480093f4SDimitry Andric void IBTPltSection::writeTo(uint8_t *buf) { 2705480093f4SDimitry Andric target->writeIBTPlt(buf, in.plt->getNumEntries()); 2706480093f4SDimitry Andric } 2707480093f4SDimitry Andric 2708480093f4SDimitry Andric size_t IBTPltSection::getSize() const { 2709480093f4SDimitry Andric // 16 is the header size of .plt. 2710480093f4SDimitry Andric return 16 + in.plt->getNumEntries() * target->pltEntrySize; 2711480093f4SDimitry Andric } 2712480093f4SDimitry Andric 2713d781ede6SDimitry Andric bool IBTPltSection::isNeeded() const { return in.plt->getNumEntries() > 0; } 2714d781ede6SDimitry Andric 27155f757f3fSDimitry Andric RelroPaddingSection::RelroPaddingSection() 27165f757f3fSDimitry Andric : SyntheticSection(SHF_ALLOC | SHF_WRITE, SHT_NOBITS, 1, ".relro_padding") { 27175f757f3fSDimitry Andric } 27185f757f3fSDimitry Andric 27190b57cec5SDimitry Andric // The string hash function for .gdb_index. 27200b57cec5SDimitry Andric static uint32_t computeGdbHash(StringRef s) { 27210b57cec5SDimitry Andric uint32_t h = 0; 27220b57cec5SDimitry Andric for (uint8_t c : s) 27230b57cec5SDimitry Andric h = h * 67 + toLower(c) - 113; 27240b57cec5SDimitry Andric return h; 27250b57cec5SDimitry Andric } 27260b57cec5SDimitry Andric 2727*0fca6ea1SDimitry Andric // 4-byte alignment ensures that values in the hash lookup table and the name 2728*0fca6ea1SDimitry Andric // table are aligned. 2729*0fca6ea1SDimitry Andric DebugNamesBaseSection::DebugNamesBaseSection() 2730*0fca6ea1SDimitry Andric : SyntheticSection(0, SHT_PROGBITS, 4, ".debug_names") {} 2731*0fca6ea1SDimitry Andric 2732*0fca6ea1SDimitry Andric // Get the size of the .debug_names section header in bytes for DWARF32: 2733*0fca6ea1SDimitry Andric static uint32_t getDebugNamesHeaderSize(uint32_t augmentationStringSize) { 2734*0fca6ea1SDimitry Andric return /* unit length */ 4 + 2735*0fca6ea1SDimitry Andric /* version */ 2 + 2736*0fca6ea1SDimitry Andric /* padding */ 2 + 2737*0fca6ea1SDimitry Andric /* CU count */ 4 + 2738*0fca6ea1SDimitry Andric /* TU count */ 4 + 2739*0fca6ea1SDimitry Andric /* Foreign TU count */ 4 + 2740*0fca6ea1SDimitry Andric /* Bucket Count */ 4 + 2741*0fca6ea1SDimitry Andric /* Name Count */ 4 + 2742*0fca6ea1SDimitry Andric /* Abbrev table size */ 4 + 2743*0fca6ea1SDimitry Andric /* Augmentation string size */ 4 + 2744*0fca6ea1SDimitry Andric /* Augmentation string */ augmentationStringSize; 2745*0fca6ea1SDimitry Andric } 2746*0fca6ea1SDimitry Andric 2747*0fca6ea1SDimitry Andric static Expected<DebugNamesBaseSection::IndexEntry *> 2748*0fca6ea1SDimitry Andric readEntry(uint64_t &offset, const DWARFDebugNames::NameIndex &ni, 2749*0fca6ea1SDimitry Andric uint64_t entriesBase, DWARFDataExtractor &namesExtractor, 2750*0fca6ea1SDimitry Andric const LLDDWARFSection &namesSec) { 2751*0fca6ea1SDimitry Andric auto ie = makeThreadLocal<DebugNamesBaseSection::IndexEntry>(); 2752*0fca6ea1SDimitry Andric ie->poolOffset = offset; 2753*0fca6ea1SDimitry Andric Error err = Error::success(); 2754*0fca6ea1SDimitry Andric uint64_t ulebVal = namesExtractor.getULEB128(&offset, &err); 2755*0fca6ea1SDimitry Andric if (err) 2756*0fca6ea1SDimitry Andric return createStringError(inconvertibleErrorCode(), 2757*0fca6ea1SDimitry Andric "invalid abbrev code: %s", 2758*0fca6ea1SDimitry Andric toString(std::move(err)).c_str()); 2759*0fca6ea1SDimitry Andric if (!isUInt<32>(ulebVal)) 2760*0fca6ea1SDimitry Andric return createStringError(inconvertibleErrorCode(), 2761*0fca6ea1SDimitry Andric "abbrev code too large for DWARF32: %" PRIu64, 2762*0fca6ea1SDimitry Andric ulebVal); 2763*0fca6ea1SDimitry Andric ie->abbrevCode = static_cast<uint32_t>(ulebVal); 2764*0fca6ea1SDimitry Andric auto it = ni.getAbbrevs().find_as(ie->abbrevCode); 2765*0fca6ea1SDimitry Andric if (it == ni.getAbbrevs().end()) 2766*0fca6ea1SDimitry Andric return createStringError(inconvertibleErrorCode(), 2767*0fca6ea1SDimitry Andric "abbrev code not found in abbrev table: %" PRIu32, 2768*0fca6ea1SDimitry Andric ie->abbrevCode); 2769*0fca6ea1SDimitry Andric 2770*0fca6ea1SDimitry Andric DebugNamesBaseSection::AttrValue attr, cuAttr = {0, 0}; 2771*0fca6ea1SDimitry Andric for (DWARFDebugNames::AttributeEncoding a : it->Attributes) { 2772*0fca6ea1SDimitry Andric if (a.Index == dwarf::DW_IDX_parent) { 2773*0fca6ea1SDimitry Andric if (a.Form == dwarf::DW_FORM_ref4) { 2774*0fca6ea1SDimitry Andric attr.attrValue = namesExtractor.getU32(&offset, &err); 2775*0fca6ea1SDimitry Andric attr.attrSize = 4; 2776*0fca6ea1SDimitry Andric ie->parentOffset = entriesBase + attr.attrValue; 2777*0fca6ea1SDimitry Andric } else if (a.Form != DW_FORM_flag_present) 2778*0fca6ea1SDimitry Andric return createStringError(inconvertibleErrorCode(), 2779*0fca6ea1SDimitry Andric "invalid form for DW_IDX_parent"); 2780*0fca6ea1SDimitry Andric } else { 2781*0fca6ea1SDimitry Andric switch (a.Form) { 2782*0fca6ea1SDimitry Andric case DW_FORM_data1: 2783*0fca6ea1SDimitry Andric case DW_FORM_ref1: { 2784*0fca6ea1SDimitry Andric attr.attrValue = namesExtractor.getU8(&offset, &err); 2785*0fca6ea1SDimitry Andric attr.attrSize = 1; 2786*0fca6ea1SDimitry Andric break; 2787*0fca6ea1SDimitry Andric } 2788*0fca6ea1SDimitry Andric case DW_FORM_data2: 2789*0fca6ea1SDimitry Andric case DW_FORM_ref2: { 2790*0fca6ea1SDimitry Andric attr.attrValue = namesExtractor.getU16(&offset, &err); 2791*0fca6ea1SDimitry Andric attr.attrSize = 2; 2792*0fca6ea1SDimitry Andric break; 2793*0fca6ea1SDimitry Andric } 2794*0fca6ea1SDimitry Andric case DW_FORM_data4: 2795*0fca6ea1SDimitry Andric case DW_FORM_ref4: { 2796*0fca6ea1SDimitry Andric attr.attrValue = namesExtractor.getU32(&offset, &err); 2797*0fca6ea1SDimitry Andric attr.attrSize = 4; 2798*0fca6ea1SDimitry Andric break; 2799*0fca6ea1SDimitry Andric } 2800*0fca6ea1SDimitry Andric default: 2801*0fca6ea1SDimitry Andric return createStringError( 2802*0fca6ea1SDimitry Andric inconvertibleErrorCode(), 2803*0fca6ea1SDimitry Andric "unrecognized form encoding %d in abbrev table", a.Form); 2804*0fca6ea1SDimitry Andric } 2805*0fca6ea1SDimitry Andric } 2806*0fca6ea1SDimitry Andric if (err) 2807*0fca6ea1SDimitry Andric return createStringError(inconvertibleErrorCode(), 2808*0fca6ea1SDimitry Andric "error while reading attributes: %s", 2809*0fca6ea1SDimitry Andric toString(std::move(err)).c_str()); 2810*0fca6ea1SDimitry Andric if (a.Index == DW_IDX_compile_unit) 2811*0fca6ea1SDimitry Andric cuAttr = attr; 2812*0fca6ea1SDimitry Andric else if (a.Form != DW_FORM_flag_present) 2813*0fca6ea1SDimitry Andric ie->attrValues.push_back(attr); 2814*0fca6ea1SDimitry Andric } 2815*0fca6ea1SDimitry Andric // Canonicalize abbrev by placing the CU/TU index at the end. 2816*0fca6ea1SDimitry Andric ie->attrValues.push_back(cuAttr); 2817*0fca6ea1SDimitry Andric return ie; 2818*0fca6ea1SDimitry Andric } 2819*0fca6ea1SDimitry Andric 2820*0fca6ea1SDimitry Andric void DebugNamesBaseSection::parseDebugNames( 2821*0fca6ea1SDimitry Andric InputChunk &inputChunk, OutputChunk &chunk, 2822*0fca6ea1SDimitry Andric DWARFDataExtractor &namesExtractor, DataExtractor &strExtractor, 2823*0fca6ea1SDimitry Andric function_ref<SmallVector<uint32_t, 0>( 2824*0fca6ea1SDimitry Andric uint32_t numCus, const DWARFDebugNames::Header &, 2825*0fca6ea1SDimitry Andric const DWARFDebugNames::DWARFDebugNamesOffsets &)> 2826*0fca6ea1SDimitry Andric readOffsets) { 2827*0fca6ea1SDimitry Andric const LLDDWARFSection &namesSec = inputChunk.section; 2828*0fca6ea1SDimitry Andric DenseMap<uint32_t, IndexEntry *> offsetMap; 2829*0fca6ea1SDimitry Andric // Number of CUs seen in previous NameIndex sections within current chunk. 2830*0fca6ea1SDimitry Andric uint32_t numCus = 0; 2831*0fca6ea1SDimitry Andric for (const DWARFDebugNames::NameIndex &ni : *inputChunk.llvmDebugNames) { 2832*0fca6ea1SDimitry Andric NameData &nd = inputChunk.nameData.emplace_back(); 2833*0fca6ea1SDimitry Andric nd.hdr = ni.getHeader(); 2834*0fca6ea1SDimitry Andric if (nd.hdr.Format != DwarfFormat::DWARF32) { 2835*0fca6ea1SDimitry Andric errorOrWarn(toString(namesSec.sec) + 2836*0fca6ea1SDimitry Andric Twine(": found DWARF64, which is currently unsupported")); 2837*0fca6ea1SDimitry Andric return; 2838*0fca6ea1SDimitry Andric } 2839*0fca6ea1SDimitry Andric if (nd.hdr.Version != 5) { 2840*0fca6ea1SDimitry Andric errorOrWarn(toString(namesSec.sec) + Twine(": unsupported version: ") + 2841*0fca6ea1SDimitry Andric Twine(nd.hdr.Version)); 2842*0fca6ea1SDimitry Andric return; 2843*0fca6ea1SDimitry Andric } 2844*0fca6ea1SDimitry Andric uint32_t dwarfSize = dwarf::getDwarfOffsetByteSize(DwarfFormat::DWARF32); 2845*0fca6ea1SDimitry Andric DWARFDebugNames::DWARFDebugNamesOffsets locs = ni.getOffsets(); 2846*0fca6ea1SDimitry Andric if (locs.EntriesBase > namesExtractor.getData().size()) { 2847*0fca6ea1SDimitry Andric errorOrWarn(toString(namesSec.sec) + 2848*0fca6ea1SDimitry Andric Twine(": entry pool start is beyond end of section")); 2849*0fca6ea1SDimitry Andric return; 2850*0fca6ea1SDimitry Andric } 2851*0fca6ea1SDimitry Andric 2852*0fca6ea1SDimitry Andric SmallVector<uint32_t, 0> entryOffsets = readOffsets(numCus, nd.hdr, locs); 2853*0fca6ea1SDimitry Andric 2854*0fca6ea1SDimitry Andric // Read the entry pool. 2855*0fca6ea1SDimitry Andric offsetMap.clear(); 2856*0fca6ea1SDimitry Andric nd.nameEntries.resize(nd.hdr.NameCount); 2857*0fca6ea1SDimitry Andric for (auto i : seq(nd.hdr.NameCount)) { 2858*0fca6ea1SDimitry Andric NameEntry &ne = nd.nameEntries[i]; 2859*0fca6ea1SDimitry Andric uint64_t strOffset = locs.StringOffsetsBase + i * dwarfSize; 2860*0fca6ea1SDimitry Andric ne.stringOffset = strOffset; 2861*0fca6ea1SDimitry Andric uint64_t strp = namesExtractor.getRelocatedValue(dwarfSize, &strOffset); 2862*0fca6ea1SDimitry Andric StringRef name = strExtractor.getCStrRef(&strp); 2863*0fca6ea1SDimitry Andric ne.name = name.data(); 2864*0fca6ea1SDimitry Andric ne.hashValue = caseFoldingDjbHash(name); 2865*0fca6ea1SDimitry Andric 2866*0fca6ea1SDimitry Andric // Read a series of index entries that end with abbreviation code 0. 2867*0fca6ea1SDimitry Andric uint64_t offset = locs.EntriesBase + entryOffsets[i]; 2868*0fca6ea1SDimitry Andric while (offset < namesSec.Data.size() && namesSec.Data[offset] != 0) { 2869*0fca6ea1SDimitry Andric // Read & store all entries (for the same string). 2870*0fca6ea1SDimitry Andric Expected<IndexEntry *> ieOrErr = 2871*0fca6ea1SDimitry Andric readEntry(offset, ni, locs.EntriesBase, namesExtractor, namesSec); 2872*0fca6ea1SDimitry Andric if (!ieOrErr) { 2873*0fca6ea1SDimitry Andric errorOrWarn(toString(namesSec.sec) + ": " + 2874*0fca6ea1SDimitry Andric toString(ieOrErr.takeError())); 2875*0fca6ea1SDimitry Andric return; 2876*0fca6ea1SDimitry Andric } 2877*0fca6ea1SDimitry Andric ne.indexEntries.push_back(std::move(*ieOrErr)); 2878*0fca6ea1SDimitry Andric } 2879*0fca6ea1SDimitry Andric if (offset >= namesSec.Data.size()) 2880*0fca6ea1SDimitry Andric errorOrWarn(toString(namesSec.sec) + 2881*0fca6ea1SDimitry Andric Twine(": index entry is out of bounds")); 2882*0fca6ea1SDimitry Andric 2883*0fca6ea1SDimitry Andric for (IndexEntry &ie : ne.entries()) 2884*0fca6ea1SDimitry Andric offsetMap[ie.poolOffset] = &ie; 2885*0fca6ea1SDimitry Andric } 2886*0fca6ea1SDimitry Andric 2887*0fca6ea1SDimitry Andric // Assign parent pointers, which will be used to update DW_IDX_parent index 2888*0fca6ea1SDimitry Andric // attributes. Note: offsetMap[0] does not exist, so parentOffset == 0 will 2889*0fca6ea1SDimitry Andric // get parentEntry == null as well. 2890*0fca6ea1SDimitry Andric for (NameEntry &ne : nd.nameEntries) 2891*0fca6ea1SDimitry Andric for (IndexEntry &ie : ne.entries()) 2892*0fca6ea1SDimitry Andric ie.parentEntry = offsetMap.lookup(ie.parentOffset); 2893*0fca6ea1SDimitry Andric numCus += nd.hdr.CompUnitCount; 2894*0fca6ea1SDimitry Andric } 2895*0fca6ea1SDimitry Andric } 2896*0fca6ea1SDimitry Andric 2897*0fca6ea1SDimitry Andric // Compute the form for output DW_IDX_compile_unit attributes, similar to 2898*0fca6ea1SDimitry Andric // DIEInteger::BestForm. The input form (often DW_FORM_data1) may not hold all 2899*0fca6ea1SDimitry Andric // the merged CU indices. 2900*0fca6ea1SDimitry Andric std::pair<uint8_t, dwarf::Form> static getMergedCuCountForm( 2901*0fca6ea1SDimitry Andric uint32_t compUnitCount) { 2902*0fca6ea1SDimitry Andric if (compUnitCount > UINT16_MAX) 2903*0fca6ea1SDimitry Andric return {4, DW_FORM_data4}; 2904*0fca6ea1SDimitry Andric if (compUnitCount > UINT8_MAX) 2905*0fca6ea1SDimitry Andric return {2, DW_FORM_data2}; 2906*0fca6ea1SDimitry Andric return {1, DW_FORM_data1}; 2907*0fca6ea1SDimitry Andric } 2908*0fca6ea1SDimitry Andric 2909*0fca6ea1SDimitry Andric void DebugNamesBaseSection::computeHdrAndAbbrevTable( 2910*0fca6ea1SDimitry Andric MutableArrayRef<InputChunk> inputChunks) { 2911*0fca6ea1SDimitry Andric TimeTraceScope timeScope("Merge .debug_names", "hdr and abbrev table"); 2912*0fca6ea1SDimitry Andric size_t numCu = 0; 2913*0fca6ea1SDimitry Andric hdr.Format = DwarfFormat::DWARF32; 2914*0fca6ea1SDimitry Andric hdr.Version = 5; 2915*0fca6ea1SDimitry Andric hdr.CompUnitCount = 0; 2916*0fca6ea1SDimitry Andric hdr.LocalTypeUnitCount = 0; 2917*0fca6ea1SDimitry Andric hdr.ForeignTypeUnitCount = 0; 2918*0fca6ea1SDimitry Andric hdr.AugmentationStringSize = 0; 2919*0fca6ea1SDimitry Andric 2920*0fca6ea1SDimitry Andric // Compute CU and TU counts. 2921*0fca6ea1SDimitry Andric for (auto i : seq(numChunks)) { 2922*0fca6ea1SDimitry Andric InputChunk &inputChunk = inputChunks[i]; 2923*0fca6ea1SDimitry Andric inputChunk.baseCuIdx = numCu; 2924*0fca6ea1SDimitry Andric numCu += chunks[i].compUnits.size(); 2925*0fca6ea1SDimitry Andric for (const NameData &nd : inputChunk.nameData) { 2926*0fca6ea1SDimitry Andric hdr.CompUnitCount += nd.hdr.CompUnitCount; 2927*0fca6ea1SDimitry Andric // TODO: We don't handle type units yet, so LocalTypeUnitCount & 2928*0fca6ea1SDimitry Andric // ForeignTypeUnitCount are left as 0. 2929*0fca6ea1SDimitry Andric if (nd.hdr.LocalTypeUnitCount || nd.hdr.ForeignTypeUnitCount) 2930*0fca6ea1SDimitry Andric warn(toString(inputChunk.section.sec) + 2931*0fca6ea1SDimitry Andric Twine(": type units are not implemented")); 2932*0fca6ea1SDimitry Andric // If augmentation strings are not identical, use an empty string. 2933*0fca6ea1SDimitry Andric if (i == 0) { 2934*0fca6ea1SDimitry Andric hdr.AugmentationStringSize = nd.hdr.AugmentationStringSize; 2935*0fca6ea1SDimitry Andric hdr.AugmentationString = nd.hdr.AugmentationString; 2936*0fca6ea1SDimitry Andric } else if (hdr.AugmentationString != nd.hdr.AugmentationString) { 2937*0fca6ea1SDimitry Andric // There are conflicting augmentation strings, so it's best for the 2938*0fca6ea1SDimitry Andric // merged index to not use an augmentation string. 2939*0fca6ea1SDimitry Andric hdr.AugmentationStringSize = 0; 2940*0fca6ea1SDimitry Andric hdr.AugmentationString.clear(); 2941*0fca6ea1SDimitry Andric } 2942*0fca6ea1SDimitry Andric } 2943*0fca6ea1SDimitry Andric } 2944*0fca6ea1SDimitry Andric 2945*0fca6ea1SDimitry Andric // Create the merged abbrev table, uniquifyinng the input abbrev tables and 2946*0fca6ea1SDimitry Andric // computing mapping from old (per-cu) abbrev codes to new (merged) abbrev 2947*0fca6ea1SDimitry Andric // codes. 2948*0fca6ea1SDimitry Andric FoldingSet<Abbrev> abbrevSet; 2949*0fca6ea1SDimitry Andric // Determine the form for the DW_IDX_compile_unit attributes in the merged 2950*0fca6ea1SDimitry Andric // index. The input form may not be big enough for all CU indices. 2951*0fca6ea1SDimitry Andric dwarf::Form cuAttrForm = getMergedCuCountForm(hdr.CompUnitCount).second; 2952*0fca6ea1SDimitry Andric for (InputChunk &inputChunk : inputChunks) { 2953*0fca6ea1SDimitry Andric for (auto [i, ni] : enumerate(*inputChunk.llvmDebugNames)) { 2954*0fca6ea1SDimitry Andric for (const DWARFDebugNames::Abbrev &oldAbbrev : ni.getAbbrevs()) { 2955*0fca6ea1SDimitry Andric // Canonicalize abbrev by placing the CU/TU index at the end, 2956*0fca6ea1SDimitry Andric // similar to 'parseDebugNames'. 2957*0fca6ea1SDimitry Andric Abbrev abbrev; 2958*0fca6ea1SDimitry Andric DWARFDebugNames::AttributeEncoding cuAttr(DW_IDX_compile_unit, 2959*0fca6ea1SDimitry Andric cuAttrForm); 2960*0fca6ea1SDimitry Andric abbrev.code = oldAbbrev.Code; 2961*0fca6ea1SDimitry Andric abbrev.tag = oldAbbrev.Tag; 2962*0fca6ea1SDimitry Andric for (DWARFDebugNames::AttributeEncoding a : oldAbbrev.Attributes) { 2963*0fca6ea1SDimitry Andric if (a.Index == DW_IDX_compile_unit) 2964*0fca6ea1SDimitry Andric cuAttr.Index = a.Index; 2965*0fca6ea1SDimitry Andric else 2966*0fca6ea1SDimitry Andric abbrev.attributes.push_back({a.Index, a.Form}); 2967*0fca6ea1SDimitry Andric } 2968*0fca6ea1SDimitry Andric // Put the CU/TU index at the end of the attributes list. 2969*0fca6ea1SDimitry Andric abbrev.attributes.push_back(cuAttr); 2970*0fca6ea1SDimitry Andric 2971*0fca6ea1SDimitry Andric // Profile the abbrev, get or assign a new code, then record the abbrev 2972*0fca6ea1SDimitry Andric // code mapping. 2973*0fca6ea1SDimitry Andric FoldingSetNodeID id; 2974*0fca6ea1SDimitry Andric abbrev.Profile(id); 2975*0fca6ea1SDimitry Andric uint32_t newCode; 2976*0fca6ea1SDimitry Andric void *insertPos; 2977*0fca6ea1SDimitry Andric if (Abbrev *existing = abbrevSet.FindNodeOrInsertPos(id, insertPos)) { 2978*0fca6ea1SDimitry Andric // Found it; we've already seen an identical abbreviation. 2979*0fca6ea1SDimitry Andric newCode = existing->code; 2980*0fca6ea1SDimitry Andric } else { 2981*0fca6ea1SDimitry Andric Abbrev *abbrev2 = 2982*0fca6ea1SDimitry Andric new (abbrevAlloc.Allocate()) Abbrev(std::move(abbrev)); 2983*0fca6ea1SDimitry Andric abbrevSet.InsertNode(abbrev2, insertPos); 2984*0fca6ea1SDimitry Andric abbrevTable.push_back(abbrev2); 2985*0fca6ea1SDimitry Andric newCode = abbrevTable.size(); 2986*0fca6ea1SDimitry Andric abbrev2->code = newCode; 2987*0fca6ea1SDimitry Andric } 2988*0fca6ea1SDimitry Andric inputChunk.nameData[i].abbrevCodeMap[oldAbbrev.Code] = newCode; 2989*0fca6ea1SDimitry Andric } 2990*0fca6ea1SDimitry Andric } 2991*0fca6ea1SDimitry Andric } 2992*0fca6ea1SDimitry Andric 2993*0fca6ea1SDimitry Andric // Compute the merged abbrev table. 2994*0fca6ea1SDimitry Andric raw_svector_ostream os(abbrevTableBuf); 2995*0fca6ea1SDimitry Andric for (Abbrev *abbrev : abbrevTable) { 2996*0fca6ea1SDimitry Andric encodeULEB128(abbrev->code, os); 2997*0fca6ea1SDimitry Andric encodeULEB128(abbrev->tag, os); 2998*0fca6ea1SDimitry Andric for (DWARFDebugNames::AttributeEncoding a : abbrev->attributes) { 2999*0fca6ea1SDimitry Andric encodeULEB128(a.Index, os); 3000*0fca6ea1SDimitry Andric encodeULEB128(a.Form, os); 3001*0fca6ea1SDimitry Andric } 3002*0fca6ea1SDimitry Andric os.write("\0", 2); // attribute specification end 3003*0fca6ea1SDimitry Andric } 3004*0fca6ea1SDimitry Andric os.write(0); // abbrev table end 3005*0fca6ea1SDimitry Andric hdr.AbbrevTableSize = abbrevTableBuf.size(); 3006*0fca6ea1SDimitry Andric } 3007*0fca6ea1SDimitry Andric 3008*0fca6ea1SDimitry Andric void DebugNamesBaseSection::Abbrev::Profile(FoldingSetNodeID &id) const { 3009*0fca6ea1SDimitry Andric id.AddInteger(tag); 3010*0fca6ea1SDimitry Andric for (const DWARFDebugNames::AttributeEncoding &attr : attributes) { 3011*0fca6ea1SDimitry Andric id.AddInteger(attr.Index); 3012*0fca6ea1SDimitry Andric id.AddInteger(attr.Form); 3013*0fca6ea1SDimitry Andric } 3014*0fca6ea1SDimitry Andric } 3015*0fca6ea1SDimitry Andric 3016*0fca6ea1SDimitry Andric std::pair<uint32_t, uint32_t> DebugNamesBaseSection::computeEntryPool( 3017*0fca6ea1SDimitry Andric MutableArrayRef<InputChunk> inputChunks) { 3018*0fca6ea1SDimitry Andric TimeTraceScope timeScope("Merge .debug_names", "entry pool"); 3019*0fca6ea1SDimitry Andric // Collect and de-duplicate all the names (preserving all the entries). 3020*0fca6ea1SDimitry Andric // Speed it up using multithreading, as the number of symbols can be in the 3021*0fca6ea1SDimitry Andric // order of millions. 3022*0fca6ea1SDimitry Andric const size_t concurrency = 3023*0fca6ea1SDimitry Andric bit_floor(std::min<size_t>(config->threadCount, numShards)); 3024*0fca6ea1SDimitry Andric const size_t shift = 32 - countr_zero(numShards); 3025*0fca6ea1SDimitry Andric const uint8_t cuAttrSize = getMergedCuCountForm(hdr.CompUnitCount).first; 3026*0fca6ea1SDimitry Andric DenseMap<CachedHashStringRef, size_t> maps[numShards]; 3027*0fca6ea1SDimitry Andric 3028*0fca6ea1SDimitry Andric parallelFor(0, concurrency, [&](size_t threadId) { 3029*0fca6ea1SDimitry Andric for (auto i : seq(numChunks)) { 3030*0fca6ea1SDimitry Andric InputChunk &inputChunk = inputChunks[i]; 3031*0fca6ea1SDimitry Andric for (auto j : seq(inputChunk.nameData.size())) { 3032*0fca6ea1SDimitry Andric NameData &nd = inputChunk.nameData[j]; 3033*0fca6ea1SDimitry Andric // Deduplicate the NameEntry records (based on the string/name), 3034*0fca6ea1SDimitry Andric // appending all IndexEntries from duplicate NameEntry records to 3035*0fca6ea1SDimitry Andric // the single preserved copy. 3036*0fca6ea1SDimitry Andric for (NameEntry &ne : nd.nameEntries) { 3037*0fca6ea1SDimitry Andric auto shardId = ne.hashValue >> shift; 3038*0fca6ea1SDimitry Andric if ((shardId & (concurrency - 1)) != threadId) 3039*0fca6ea1SDimitry Andric continue; 3040*0fca6ea1SDimitry Andric 3041*0fca6ea1SDimitry Andric ne.chunkIdx = i; 3042*0fca6ea1SDimitry Andric for (IndexEntry &ie : ne.entries()) { 3043*0fca6ea1SDimitry Andric // Update the IndexEntry's abbrev code to match the merged 3044*0fca6ea1SDimitry Andric // abbreviations. 3045*0fca6ea1SDimitry Andric ie.abbrevCode = nd.abbrevCodeMap[ie.abbrevCode]; 3046*0fca6ea1SDimitry Andric // Update the DW_IDX_compile_unit attribute (the last one after 3047*0fca6ea1SDimitry Andric // canonicalization) to have correct merged offset value and size. 3048*0fca6ea1SDimitry Andric auto &back = ie.attrValues.back(); 3049*0fca6ea1SDimitry Andric back.attrValue += inputChunk.baseCuIdx + j; 3050*0fca6ea1SDimitry Andric back.attrSize = cuAttrSize; 3051*0fca6ea1SDimitry Andric } 3052*0fca6ea1SDimitry Andric 3053*0fca6ea1SDimitry Andric auto &nameVec = nameVecs[shardId]; 3054*0fca6ea1SDimitry Andric auto [it, inserted] = maps[shardId].try_emplace( 3055*0fca6ea1SDimitry Andric CachedHashStringRef(ne.name, ne.hashValue), nameVec.size()); 3056*0fca6ea1SDimitry Andric if (inserted) 3057*0fca6ea1SDimitry Andric nameVec.push_back(std::move(ne)); 3058*0fca6ea1SDimitry Andric else 3059*0fca6ea1SDimitry Andric nameVec[it->second].indexEntries.append(std::move(ne.indexEntries)); 3060*0fca6ea1SDimitry Andric } 3061*0fca6ea1SDimitry Andric } 3062*0fca6ea1SDimitry Andric } 3063*0fca6ea1SDimitry Andric }); 3064*0fca6ea1SDimitry Andric 3065*0fca6ea1SDimitry Andric // Compute entry offsets in parallel. First, compute offsets relative to the 3066*0fca6ea1SDimitry Andric // current shard. 3067*0fca6ea1SDimitry Andric uint32_t offsets[numShards]; 3068*0fca6ea1SDimitry Andric parallelFor(0, numShards, [&](size_t shard) { 3069*0fca6ea1SDimitry Andric uint32_t offset = 0; 3070*0fca6ea1SDimitry Andric for (NameEntry &ne : nameVecs[shard]) { 3071*0fca6ea1SDimitry Andric ne.entryOffset = offset; 3072*0fca6ea1SDimitry Andric for (IndexEntry &ie : ne.entries()) { 3073*0fca6ea1SDimitry Andric ie.poolOffset = offset; 3074*0fca6ea1SDimitry Andric offset += getULEB128Size(ie.abbrevCode); 3075*0fca6ea1SDimitry Andric for (AttrValue value : ie.attrValues) 3076*0fca6ea1SDimitry Andric offset += value.attrSize; 3077*0fca6ea1SDimitry Andric } 3078*0fca6ea1SDimitry Andric ++offset; // index entry sentinel 3079*0fca6ea1SDimitry Andric } 3080*0fca6ea1SDimitry Andric offsets[shard] = offset; 3081*0fca6ea1SDimitry Andric }); 3082*0fca6ea1SDimitry Andric // Then add shard offsets. 3083*0fca6ea1SDimitry Andric std::partial_sum(offsets, std::end(offsets), offsets); 3084*0fca6ea1SDimitry Andric parallelFor(1, numShards, [&](size_t shard) { 3085*0fca6ea1SDimitry Andric uint32_t offset = offsets[shard - 1]; 3086*0fca6ea1SDimitry Andric for (NameEntry &ne : nameVecs[shard]) { 3087*0fca6ea1SDimitry Andric ne.entryOffset += offset; 3088*0fca6ea1SDimitry Andric for (IndexEntry &ie : ne.entries()) 3089*0fca6ea1SDimitry Andric ie.poolOffset += offset; 3090*0fca6ea1SDimitry Andric } 3091*0fca6ea1SDimitry Andric }); 3092*0fca6ea1SDimitry Andric 3093*0fca6ea1SDimitry Andric // Update the DW_IDX_parent entries that refer to real parents (have 3094*0fca6ea1SDimitry Andric // DW_FORM_ref4). 3095*0fca6ea1SDimitry Andric parallelFor(0, numShards, [&](size_t shard) { 3096*0fca6ea1SDimitry Andric for (NameEntry &ne : nameVecs[shard]) { 3097*0fca6ea1SDimitry Andric for (IndexEntry &ie : ne.entries()) { 3098*0fca6ea1SDimitry Andric if (!ie.parentEntry) 3099*0fca6ea1SDimitry Andric continue; 3100*0fca6ea1SDimitry Andric // Abbrevs are indexed starting at 1; vector starts at 0. (abbrevCode 3101*0fca6ea1SDimitry Andric // corresponds to position in the merged table vector). 3102*0fca6ea1SDimitry Andric const Abbrev *abbrev = abbrevTable[ie.abbrevCode - 1]; 3103*0fca6ea1SDimitry Andric for (const auto &[a, v] : zip_equal(abbrev->attributes, ie.attrValues)) 3104*0fca6ea1SDimitry Andric if (a.Index == DW_IDX_parent && a.Form == DW_FORM_ref4) 3105*0fca6ea1SDimitry Andric v.attrValue = ie.parentEntry->poolOffset; 3106*0fca6ea1SDimitry Andric } 3107*0fca6ea1SDimitry Andric } 3108*0fca6ea1SDimitry Andric }); 3109*0fca6ea1SDimitry Andric 3110*0fca6ea1SDimitry Andric // Return (entry pool size, number of entries). 3111*0fca6ea1SDimitry Andric uint32_t num = 0; 3112*0fca6ea1SDimitry Andric for (auto &map : maps) 3113*0fca6ea1SDimitry Andric num += map.size(); 3114*0fca6ea1SDimitry Andric return {offsets[numShards - 1], num}; 3115*0fca6ea1SDimitry Andric } 3116*0fca6ea1SDimitry Andric 3117*0fca6ea1SDimitry Andric void DebugNamesBaseSection::init( 3118*0fca6ea1SDimitry Andric function_ref<void(InputFile *, InputChunk &, OutputChunk &)> parseFile) { 3119*0fca6ea1SDimitry Andric TimeTraceScope timeScope("Merge .debug_names"); 3120*0fca6ea1SDimitry Andric // Collect and remove input .debug_names sections. Save InputSection pointers 3121*0fca6ea1SDimitry Andric // to relocate string offsets in `writeTo`. 3122*0fca6ea1SDimitry Andric SetVector<InputFile *> files; 3123*0fca6ea1SDimitry Andric for (InputSectionBase *s : ctx.inputSections) { 3124*0fca6ea1SDimitry Andric InputSection *isec = dyn_cast<InputSection>(s); 3125*0fca6ea1SDimitry Andric if (!isec) 3126*0fca6ea1SDimitry Andric continue; 3127*0fca6ea1SDimitry Andric if (!(s->flags & SHF_ALLOC) && s->name == ".debug_names") { 3128*0fca6ea1SDimitry Andric s->markDead(); 3129*0fca6ea1SDimitry Andric inputSections.push_back(isec); 3130*0fca6ea1SDimitry Andric files.insert(isec->file); 3131*0fca6ea1SDimitry Andric } 3132*0fca6ea1SDimitry Andric } 3133*0fca6ea1SDimitry Andric 3134*0fca6ea1SDimitry Andric // Parse input .debug_names sections and extract InputChunk and OutputChunk 3135*0fca6ea1SDimitry Andric // data. OutputChunk contains CU information, which will be needed by 3136*0fca6ea1SDimitry Andric // `writeTo`. 3137*0fca6ea1SDimitry Andric auto inputChunksPtr = std::make_unique<InputChunk[]>(files.size()); 3138*0fca6ea1SDimitry Andric MutableArrayRef<InputChunk> inputChunks(inputChunksPtr.get(), files.size()); 3139*0fca6ea1SDimitry Andric numChunks = files.size(); 3140*0fca6ea1SDimitry Andric chunks = std::make_unique<OutputChunk[]>(files.size()); 3141*0fca6ea1SDimitry Andric { 3142*0fca6ea1SDimitry Andric TimeTraceScope timeScope("Merge .debug_names", "parse"); 3143*0fca6ea1SDimitry Andric parallelFor(0, files.size(), [&](size_t i) { 3144*0fca6ea1SDimitry Andric parseFile(files[i], inputChunks[i], chunks[i]); 3145*0fca6ea1SDimitry Andric }); 3146*0fca6ea1SDimitry Andric } 3147*0fca6ea1SDimitry Andric 3148*0fca6ea1SDimitry Andric // Compute section header (except unit_length), abbrev table, and entry pool. 3149*0fca6ea1SDimitry Andric computeHdrAndAbbrevTable(inputChunks); 3150*0fca6ea1SDimitry Andric uint32_t entryPoolSize; 3151*0fca6ea1SDimitry Andric std::tie(entryPoolSize, hdr.NameCount) = computeEntryPool(inputChunks); 3152*0fca6ea1SDimitry Andric hdr.BucketCount = dwarf::getDebugNamesBucketCount(hdr.NameCount); 3153*0fca6ea1SDimitry Andric 3154*0fca6ea1SDimitry Andric // Compute the section size. Subtract 4 to get the unit_length for DWARF32. 3155*0fca6ea1SDimitry Andric uint32_t hdrSize = getDebugNamesHeaderSize(hdr.AugmentationStringSize); 3156*0fca6ea1SDimitry Andric size = findDebugNamesOffsets(hdrSize, hdr).EntriesBase + entryPoolSize; 3157*0fca6ea1SDimitry Andric hdr.UnitLength = size - 4; 3158*0fca6ea1SDimitry Andric } 3159*0fca6ea1SDimitry Andric 3160*0fca6ea1SDimitry Andric template <class ELFT> DebugNamesSection<ELFT>::DebugNamesSection() { 3161*0fca6ea1SDimitry Andric init([](InputFile *f, InputChunk &inputChunk, OutputChunk &chunk) { 3162*0fca6ea1SDimitry Andric auto *file = cast<ObjFile<ELFT>>(f); 3163*0fca6ea1SDimitry Andric DWARFContext dwarf(std::make_unique<LLDDwarfObj<ELFT>>(file)); 3164*0fca6ea1SDimitry Andric auto &dobj = static_cast<const LLDDwarfObj<ELFT> &>(dwarf.getDWARFObj()); 3165*0fca6ea1SDimitry Andric chunk.infoSec = dobj.getInfoSection(); 3166*0fca6ea1SDimitry Andric DWARFDataExtractor namesExtractor(dobj, dobj.getNamesSection(), 3167*0fca6ea1SDimitry Andric ELFT::Endianness == endianness::little, 3168*0fca6ea1SDimitry Andric ELFT::Is64Bits ? 8 : 4); 3169*0fca6ea1SDimitry Andric // .debug_str is needed to get symbol names from string offsets. 3170*0fca6ea1SDimitry Andric DataExtractor strExtractor(dobj.getStrSection(), 3171*0fca6ea1SDimitry Andric ELFT::Endianness == endianness::little, 3172*0fca6ea1SDimitry Andric ELFT::Is64Bits ? 8 : 4); 3173*0fca6ea1SDimitry Andric inputChunk.section = dobj.getNamesSection(); 3174*0fca6ea1SDimitry Andric 3175*0fca6ea1SDimitry Andric inputChunk.llvmDebugNames.emplace(namesExtractor, strExtractor); 3176*0fca6ea1SDimitry Andric if (Error e = inputChunk.llvmDebugNames->extract()) { 3177*0fca6ea1SDimitry Andric errorOrWarn(toString(dobj.getNamesSection().sec) + Twine(": ") + 3178*0fca6ea1SDimitry Andric toString(std::move(e))); 3179*0fca6ea1SDimitry Andric } 3180*0fca6ea1SDimitry Andric parseDebugNames( 3181*0fca6ea1SDimitry Andric inputChunk, chunk, namesExtractor, strExtractor, 3182*0fca6ea1SDimitry Andric [&chunk, namesData = dobj.getNamesSection().Data.data()]( 3183*0fca6ea1SDimitry Andric uint32_t numCus, const DWARFDebugNames::Header &hdr, 3184*0fca6ea1SDimitry Andric const DWARFDebugNames::DWARFDebugNamesOffsets &locs) { 3185*0fca6ea1SDimitry Andric // Read CU offsets, which are relocated by .debug_info + X 3186*0fca6ea1SDimitry Andric // relocations. Record the section offset to be relocated by 3187*0fca6ea1SDimitry Andric // `finalizeContents`. 3188*0fca6ea1SDimitry Andric chunk.compUnits.resize_for_overwrite(numCus + hdr.CompUnitCount); 3189*0fca6ea1SDimitry Andric for (auto i : seq(hdr.CompUnitCount)) 3190*0fca6ea1SDimitry Andric chunk.compUnits[numCus + i] = locs.CUsBase + i * 4; 3191*0fca6ea1SDimitry Andric 3192*0fca6ea1SDimitry Andric // Read entry offsets. 3193*0fca6ea1SDimitry Andric const char *p = namesData + locs.EntryOffsetsBase; 3194*0fca6ea1SDimitry Andric SmallVector<uint32_t, 0> entryOffsets; 3195*0fca6ea1SDimitry Andric entryOffsets.resize_for_overwrite(hdr.NameCount); 3196*0fca6ea1SDimitry Andric for (uint32_t &offset : entryOffsets) 3197*0fca6ea1SDimitry Andric offset = endian::readNext<uint32_t, ELFT::Endianness, unaligned>(p); 3198*0fca6ea1SDimitry Andric return entryOffsets; 3199*0fca6ea1SDimitry Andric }); 3200*0fca6ea1SDimitry Andric }); 3201*0fca6ea1SDimitry Andric } 3202*0fca6ea1SDimitry Andric 3203*0fca6ea1SDimitry Andric template <class ELFT> 3204*0fca6ea1SDimitry Andric template <class RelTy> 3205*0fca6ea1SDimitry Andric void DebugNamesSection<ELFT>::getNameRelocs( 3206*0fca6ea1SDimitry Andric InputSection *sec, ArrayRef<RelTy> rels, 3207*0fca6ea1SDimitry Andric DenseMap<uint32_t, uint32_t> &relocs) { 3208*0fca6ea1SDimitry Andric for (const RelTy &rel : rels) { 3209*0fca6ea1SDimitry Andric Symbol &sym = sec->file->getRelocTargetSym(rel); 3210*0fca6ea1SDimitry Andric relocs[rel.r_offset] = sym.getVA(getAddend<ELFT>(rel)); 3211*0fca6ea1SDimitry Andric } 3212*0fca6ea1SDimitry Andric } 3213*0fca6ea1SDimitry Andric 3214*0fca6ea1SDimitry Andric template <class ELFT> void DebugNamesSection<ELFT>::finalizeContents() { 3215*0fca6ea1SDimitry Andric // Get relocations of .debug_names sections. 3216*0fca6ea1SDimitry Andric auto relocs = std::make_unique<DenseMap<uint32_t, uint32_t>[]>(numChunks); 3217*0fca6ea1SDimitry Andric parallelFor(0, numChunks, [&](size_t i) { 3218*0fca6ea1SDimitry Andric InputSection *sec = inputSections[i]; 3219*0fca6ea1SDimitry Andric auto rels = sec->template relsOrRelas<ELFT>(); 3220*0fca6ea1SDimitry Andric if (rels.areRelocsRel()) 3221*0fca6ea1SDimitry Andric getNameRelocs(sec, rels.rels, relocs.get()[i]); 3222*0fca6ea1SDimitry Andric else 3223*0fca6ea1SDimitry Andric getNameRelocs(sec, rels.relas, relocs.get()[i]); 3224*0fca6ea1SDimitry Andric 3225*0fca6ea1SDimitry Andric // Relocate CU offsets with .debug_info + X relocations. 3226*0fca6ea1SDimitry Andric OutputChunk &chunk = chunks.get()[i]; 3227*0fca6ea1SDimitry Andric for (auto [j, cuOffset] : enumerate(chunk.compUnits)) 3228*0fca6ea1SDimitry Andric cuOffset = relocs.get()[i].lookup(cuOffset); 3229*0fca6ea1SDimitry Andric }); 3230*0fca6ea1SDimitry Andric 3231*0fca6ea1SDimitry Andric // Relocate string offsets in the name table with .debug_str + X relocations. 3232*0fca6ea1SDimitry Andric parallelForEach(nameVecs, [&](auto &nameVec) { 3233*0fca6ea1SDimitry Andric for (NameEntry &ne : nameVec) 3234*0fca6ea1SDimitry Andric ne.stringOffset = relocs.get()[ne.chunkIdx].lookup(ne.stringOffset); 3235*0fca6ea1SDimitry Andric }); 3236*0fca6ea1SDimitry Andric } 3237*0fca6ea1SDimitry Andric 3238*0fca6ea1SDimitry Andric template <class ELFT> void DebugNamesSection<ELFT>::writeTo(uint8_t *buf) { 3239*0fca6ea1SDimitry Andric [[maybe_unused]] const uint8_t *const beginBuf = buf; 3240*0fca6ea1SDimitry Andric // Write the header. 3241*0fca6ea1SDimitry Andric endian::writeNext<uint32_t, ELFT::Endianness>(buf, hdr.UnitLength); 3242*0fca6ea1SDimitry Andric endian::writeNext<uint16_t, ELFT::Endianness>(buf, hdr.Version); 3243*0fca6ea1SDimitry Andric buf += 2; // padding 3244*0fca6ea1SDimitry Andric endian::writeNext<uint32_t, ELFT::Endianness>(buf, hdr.CompUnitCount); 3245*0fca6ea1SDimitry Andric endian::writeNext<uint32_t, ELFT::Endianness>(buf, hdr.LocalTypeUnitCount); 3246*0fca6ea1SDimitry Andric endian::writeNext<uint32_t, ELFT::Endianness>(buf, hdr.ForeignTypeUnitCount); 3247*0fca6ea1SDimitry Andric endian::writeNext<uint32_t, ELFT::Endianness>(buf, hdr.BucketCount); 3248*0fca6ea1SDimitry Andric endian::writeNext<uint32_t, ELFT::Endianness>(buf, hdr.NameCount); 3249*0fca6ea1SDimitry Andric endian::writeNext<uint32_t, ELFT::Endianness>(buf, hdr.AbbrevTableSize); 3250*0fca6ea1SDimitry Andric endian::writeNext<uint32_t, ELFT::Endianness>(buf, 3251*0fca6ea1SDimitry Andric hdr.AugmentationStringSize); 3252*0fca6ea1SDimitry Andric memcpy(buf, hdr.AugmentationString.c_str(), hdr.AugmentationString.size()); 3253*0fca6ea1SDimitry Andric buf += hdr.AugmentationStringSize; 3254*0fca6ea1SDimitry Andric 3255*0fca6ea1SDimitry Andric // Write the CU list. 3256*0fca6ea1SDimitry Andric for (auto &chunk : getChunks()) 3257*0fca6ea1SDimitry Andric for (uint32_t cuOffset : chunk.compUnits) 3258*0fca6ea1SDimitry Andric endian::writeNext<uint32_t, ELFT::Endianness>(buf, cuOffset); 3259*0fca6ea1SDimitry Andric 3260*0fca6ea1SDimitry Andric // TODO: Write the local TU list, then the foreign TU list.. 3261*0fca6ea1SDimitry Andric 3262*0fca6ea1SDimitry Andric // Write the hash lookup table. 3263*0fca6ea1SDimitry Andric SmallVector<SmallVector<NameEntry *, 0>, 0> buckets(hdr.BucketCount); 3264*0fca6ea1SDimitry Andric // Symbols enter into a bucket whose index is the hash modulo bucket_count. 3265*0fca6ea1SDimitry Andric for (auto &nameVec : nameVecs) 3266*0fca6ea1SDimitry Andric for (NameEntry &ne : nameVec) 3267*0fca6ea1SDimitry Andric buckets[ne.hashValue % hdr.BucketCount].push_back(&ne); 3268*0fca6ea1SDimitry Andric 3269*0fca6ea1SDimitry Andric // Write buckets (accumulated bucket counts). 3270*0fca6ea1SDimitry Andric uint32_t bucketIdx = 1; 3271*0fca6ea1SDimitry Andric for (const SmallVector<NameEntry *, 0> &bucket : buckets) { 3272*0fca6ea1SDimitry Andric if (!bucket.empty()) 3273*0fca6ea1SDimitry Andric endian::write32<ELFT::Endianness>(buf, bucketIdx); 3274*0fca6ea1SDimitry Andric buf += 4; 3275*0fca6ea1SDimitry Andric bucketIdx += bucket.size(); 3276*0fca6ea1SDimitry Andric } 3277*0fca6ea1SDimitry Andric // Write the hashes. 3278*0fca6ea1SDimitry Andric for (const SmallVector<NameEntry *, 0> &bucket : buckets) 3279*0fca6ea1SDimitry Andric for (const NameEntry *e : bucket) 3280*0fca6ea1SDimitry Andric endian::writeNext<uint32_t, ELFT::Endianness>(buf, e->hashValue); 3281*0fca6ea1SDimitry Andric 3282*0fca6ea1SDimitry Andric // Write the name table. The name entries are ordered by bucket_idx and 3283*0fca6ea1SDimitry Andric // correspond one-to-one with the hash lookup table. 3284*0fca6ea1SDimitry Andric // 3285*0fca6ea1SDimitry Andric // First, write the relocated string offsets. 3286*0fca6ea1SDimitry Andric for (const SmallVector<NameEntry *, 0> &bucket : buckets) 3287*0fca6ea1SDimitry Andric for (const NameEntry *ne : bucket) 3288*0fca6ea1SDimitry Andric endian::writeNext<uint32_t, ELFT::Endianness>(buf, ne->stringOffset); 3289*0fca6ea1SDimitry Andric 3290*0fca6ea1SDimitry Andric // Then write the entry offsets. 3291*0fca6ea1SDimitry Andric for (const SmallVector<NameEntry *, 0> &bucket : buckets) 3292*0fca6ea1SDimitry Andric for (const NameEntry *ne : bucket) 3293*0fca6ea1SDimitry Andric endian::writeNext<uint32_t, ELFT::Endianness>(buf, ne->entryOffset); 3294*0fca6ea1SDimitry Andric 3295*0fca6ea1SDimitry Andric // Write the abbrev table. 3296*0fca6ea1SDimitry Andric buf = llvm::copy(abbrevTableBuf, buf); 3297*0fca6ea1SDimitry Andric 3298*0fca6ea1SDimitry Andric // Write the entry pool. Unlike the name table, the name entries follow the 3299*0fca6ea1SDimitry Andric // nameVecs order computed by `computeEntryPool`. 3300*0fca6ea1SDimitry Andric for (auto &nameVec : nameVecs) { 3301*0fca6ea1SDimitry Andric for (NameEntry &ne : nameVec) { 3302*0fca6ea1SDimitry Andric // Write all the entries for the string. 3303*0fca6ea1SDimitry Andric for (const IndexEntry &ie : ne.entries()) { 3304*0fca6ea1SDimitry Andric buf += encodeULEB128(ie.abbrevCode, buf); 3305*0fca6ea1SDimitry Andric for (AttrValue value : ie.attrValues) { 3306*0fca6ea1SDimitry Andric switch (value.attrSize) { 3307*0fca6ea1SDimitry Andric case 1: 3308*0fca6ea1SDimitry Andric *buf++ = value.attrValue; 3309*0fca6ea1SDimitry Andric break; 3310*0fca6ea1SDimitry Andric case 2: 3311*0fca6ea1SDimitry Andric endian::writeNext<uint16_t, ELFT::Endianness>(buf, value.attrValue); 3312*0fca6ea1SDimitry Andric break; 3313*0fca6ea1SDimitry Andric case 4: 3314*0fca6ea1SDimitry Andric endian::writeNext<uint32_t, ELFT::Endianness>(buf, value.attrValue); 3315*0fca6ea1SDimitry Andric break; 3316*0fca6ea1SDimitry Andric default: 3317*0fca6ea1SDimitry Andric llvm_unreachable("invalid attrSize"); 3318*0fca6ea1SDimitry Andric } 3319*0fca6ea1SDimitry Andric } 3320*0fca6ea1SDimitry Andric } 3321*0fca6ea1SDimitry Andric ++buf; // index entry sentinel 3322*0fca6ea1SDimitry Andric } 3323*0fca6ea1SDimitry Andric } 3324*0fca6ea1SDimitry Andric assert(uint64_t(buf - beginBuf) == size); 3325*0fca6ea1SDimitry Andric } 3326*0fca6ea1SDimitry Andric 33270b57cec5SDimitry Andric GdbIndexSection::GdbIndexSection() 33280b57cec5SDimitry Andric : SyntheticSection(0, SHT_PROGBITS, 1, ".gdb_index") {} 33290b57cec5SDimitry Andric 33300b57cec5SDimitry Andric // Returns the desired size of an on-disk hash table for a .gdb_index section. 33310b57cec5SDimitry Andric // There's a tradeoff between size and collision rate. We aim 75% utilization. 33320b57cec5SDimitry Andric size_t GdbIndexSection::computeSymtabSize() const { 33330b57cec5SDimitry Andric return std::max<size_t>(NextPowerOf2(symbols.size() * 4 / 3), 1024); 33340b57cec5SDimitry Andric } 33350b57cec5SDimitry Andric 33360eae32dcSDimitry Andric static SmallVector<GdbIndexSection::CuEntry, 0> 33370eae32dcSDimitry Andric readCuList(DWARFContext &dwarf) { 33380eae32dcSDimitry Andric SmallVector<GdbIndexSection::CuEntry, 0> ret; 33390b57cec5SDimitry Andric for (std::unique_ptr<DWARFUnit> &cu : dwarf.compile_units()) 33400b57cec5SDimitry Andric ret.push_back({cu->getOffset(), cu->getLength() + 4}); 33410b57cec5SDimitry Andric return ret; 33420b57cec5SDimitry Andric } 33430b57cec5SDimitry Andric 33440eae32dcSDimitry Andric static SmallVector<GdbIndexSection::AddressEntry, 0> 33450b57cec5SDimitry Andric readAddressAreas(DWARFContext &dwarf, InputSection *sec) { 33460eae32dcSDimitry Andric SmallVector<GdbIndexSection::AddressEntry, 0> ret; 33470b57cec5SDimitry Andric 33480b57cec5SDimitry Andric uint32_t cuIdx = 0; 33490b57cec5SDimitry Andric for (std::unique_ptr<DWARFUnit> &cu : dwarf.compile_units()) { 335085868e8aSDimitry Andric if (Error e = cu->tryExtractDIEsIfNeeded(false)) { 33515ffd83dbSDimitry Andric warn(toString(sec) + ": " + toString(std::move(e))); 335285868e8aSDimitry Andric return {}; 335385868e8aSDimitry Andric } 33540b57cec5SDimitry Andric Expected<DWARFAddressRangesVector> ranges = cu->collectAddressRanges(); 33550b57cec5SDimitry Andric if (!ranges) { 33565ffd83dbSDimitry Andric warn(toString(sec) + ": " + toString(ranges.takeError())); 33570b57cec5SDimitry Andric return {}; 33580b57cec5SDimitry Andric } 33590b57cec5SDimitry Andric 33600b57cec5SDimitry Andric ArrayRef<InputSectionBase *> sections = sec->file->getSections(); 33610b57cec5SDimitry Andric for (DWARFAddressRange &r : *ranges) { 33620b57cec5SDimitry Andric if (r.SectionIndex == -1ULL) 33630b57cec5SDimitry Andric continue; 33640b57cec5SDimitry Andric // Range list with zero size has no effect. 33655ffd83dbSDimitry Andric InputSectionBase *s = sections[r.SectionIndex]; 33665ffd83dbSDimitry Andric if (s && s != &InputSection::discarded && s->isLive()) 33675ffd83dbSDimitry Andric if (r.LowPC != r.HighPC) 33685ffd83dbSDimitry Andric ret.push_back({cast<InputSection>(s), r.LowPC, r.HighPC, cuIdx}); 33690b57cec5SDimitry Andric } 33700b57cec5SDimitry Andric ++cuIdx; 33710b57cec5SDimitry Andric } 33720b57cec5SDimitry Andric 33730b57cec5SDimitry Andric return ret; 33740b57cec5SDimitry Andric } 33750b57cec5SDimitry Andric 33760b57cec5SDimitry Andric template <class ELFT> 33771fd87a68SDimitry Andric static SmallVector<GdbIndexSection::NameAttrEntry, 0> 33780b57cec5SDimitry Andric readPubNamesAndTypes(const LLDDwarfObj<ELFT> &obj, 33790eae32dcSDimitry Andric const SmallVectorImpl<GdbIndexSection::CuEntry> &cus) { 33805ffd83dbSDimitry Andric const LLDDWARFSection &pubNames = obj.getGnuPubnamesSection(); 33815ffd83dbSDimitry Andric const LLDDWARFSection &pubTypes = obj.getGnuPubtypesSection(); 33820b57cec5SDimitry Andric 33831fd87a68SDimitry Andric SmallVector<GdbIndexSection::NameAttrEntry, 0> ret; 33845ffd83dbSDimitry Andric for (const LLDDWARFSection *pub : {&pubNames, &pubTypes}) { 3385*0fca6ea1SDimitry Andric DWARFDataExtractor data(obj, *pub, ELFT::Endianness == endianness::little, 3386*0fca6ea1SDimitry Andric ELFT::Is64Bits ? 8 : 4); 33875ffd83dbSDimitry Andric DWARFDebugPubTable table; 33885ffd83dbSDimitry Andric table.extract(data, /*GnuStyle=*/true, [&](Error e) { 33895ffd83dbSDimitry Andric warn(toString(pub->sec) + ": " + toString(std::move(e))); 33905ffd83dbSDimitry Andric }); 33910b57cec5SDimitry Andric for (const DWARFDebugPubTable::Set &set : table.getData()) { 33920b57cec5SDimitry Andric // The value written into the constant pool is kind << 24 | cuIndex. As we 33930b57cec5SDimitry Andric // don't know how many compilation units precede this object to compute 33940b57cec5SDimitry Andric // cuIndex, we compute (kind << 24 | cuIndexInThisObject) instead, and add 33950b57cec5SDimitry Andric // the number of preceding compilation units later. 339685868e8aSDimitry Andric uint32_t i = llvm::partition_point(cus, 339785868e8aSDimitry Andric [&](GdbIndexSection::CuEntry cu) { 339885868e8aSDimitry Andric return cu.cuOffset < set.Offset; 33990b57cec5SDimitry Andric }) - 340085868e8aSDimitry Andric cus.begin(); 34010b57cec5SDimitry Andric for (const DWARFDebugPubTable::Entry &ent : set.Entries) 34020b57cec5SDimitry Andric ret.push_back({{ent.Name, computeGdbHash(ent.Name)}, 34030b57cec5SDimitry Andric (ent.Descriptor.toBits() << 24) | i}); 34040b57cec5SDimitry Andric } 34050b57cec5SDimitry Andric } 34060b57cec5SDimitry Andric return ret; 34070b57cec5SDimitry Andric } 34080b57cec5SDimitry Andric 34090b57cec5SDimitry Andric // Create a list of symbols from a given list of symbol names and types 34100b57cec5SDimitry Andric // by uniquifying them by name. 3411bdd1243dSDimitry Andric static std::pair<SmallVector<GdbIndexSection::GdbSymbol, 0>, size_t> 3412bdd1243dSDimitry Andric createSymbols( 34131fd87a68SDimitry Andric ArrayRef<SmallVector<GdbIndexSection::NameAttrEntry, 0>> nameAttrs, 34141fd87a68SDimitry Andric const SmallVector<GdbIndexSection::GdbChunk, 0> &chunks) { 34150b57cec5SDimitry Andric using GdbSymbol = GdbIndexSection::GdbSymbol; 34160b57cec5SDimitry Andric using NameAttrEntry = GdbIndexSection::NameAttrEntry; 34170b57cec5SDimitry Andric 34180b57cec5SDimitry Andric // For each chunk, compute the number of compilation units preceding it. 34190b57cec5SDimitry Andric uint32_t cuIdx = 0; 342004eeddc0SDimitry Andric std::unique_ptr<uint32_t[]> cuIdxs(new uint32_t[chunks.size()]); 34210b57cec5SDimitry Andric for (uint32_t i = 0, e = chunks.size(); i != e; ++i) { 34220b57cec5SDimitry Andric cuIdxs[i] = cuIdx; 34230b57cec5SDimitry Andric cuIdx += chunks[i].compilationUnits.size(); 34240b57cec5SDimitry Andric } 34250b57cec5SDimitry Andric 3426*0fca6ea1SDimitry Andric // Collect the compilation unitss for each unique name. Speed it up using 3427*0fca6ea1SDimitry Andric // multi-threading as the number of symbols can be in the order of millions. 3428*0fca6ea1SDimitry Andric // Shard GdbSymbols by hash's high bits. 34295ffd83dbSDimitry Andric constexpr size_t numShards = 32; 3430bdd1243dSDimitry Andric const size_t concurrency = 343106c3fb27SDimitry Andric llvm::bit_floor(std::min<size_t>(config->threadCount, numShards)); 3432*0fca6ea1SDimitry Andric const size_t shift = 32 - llvm::countr_zero(numShards); 343304eeddc0SDimitry Andric auto map = 343404eeddc0SDimitry Andric std::make_unique<DenseMap<CachedHashStringRef, size_t>[]>(numShards); 34351fd87a68SDimitry Andric auto symbols = std::make_unique<SmallVector<GdbSymbol, 0>[]>(numShards); 343681ad6265SDimitry Andric parallelFor(0, concurrency, [&](size_t threadId) { 34370b57cec5SDimitry Andric uint32_t i = 0; 34380b57cec5SDimitry Andric for (ArrayRef<NameAttrEntry> entries : nameAttrs) { 34390b57cec5SDimitry Andric for (const NameAttrEntry &ent : entries) { 34400b57cec5SDimitry Andric size_t shardId = ent.name.hash() >> shift; 34410b57cec5SDimitry Andric if ((shardId & (concurrency - 1)) != threadId) 34420b57cec5SDimitry Andric continue; 34430b57cec5SDimitry Andric 34440b57cec5SDimitry Andric uint32_t v = ent.cuIndexAndAttrs + cuIdxs[i]; 3445*0fca6ea1SDimitry Andric auto [it, inserted] = 3446*0fca6ea1SDimitry Andric map[shardId].try_emplace(ent.name, symbols[shardId].size()); 3447*0fca6ea1SDimitry Andric if (inserted) 34480b57cec5SDimitry Andric symbols[shardId].push_back({ent.name, {v}, 0, 0}); 3449*0fca6ea1SDimitry Andric else 3450*0fca6ea1SDimitry Andric symbols[shardId][it->second].cuVector.push_back(v); 34510b57cec5SDimitry Andric } 34520b57cec5SDimitry Andric ++i; 34530b57cec5SDimitry Andric } 34540b57cec5SDimitry Andric }); 34550b57cec5SDimitry Andric 34560b57cec5SDimitry Andric size_t numSymbols = 0; 3457bdd1243dSDimitry Andric for (ArrayRef<GdbSymbol> v : ArrayRef(symbols.get(), numShards)) 34580b57cec5SDimitry Andric numSymbols += v.size(); 34590b57cec5SDimitry Andric 34600b57cec5SDimitry Andric // The return type is a flattened vector, so we'll copy each vector 34610b57cec5SDimitry Andric // contents to Ret. 34621fd87a68SDimitry Andric SmallVector<GdbSymbol, 0> ret; 34630b57cec5SDimitry Andric ret.reserve(numSymbols); 34641fd87a68SDimitry Andric for (SmallVector<GdbSymbol, 0> &vec : 3465bdd1243dSDimitry Andric MutableArrayRef(symbols.get(), numShards)) 34660b57cec5SDimitry Andric for (GdbSymbol &sym : vec) 34670b57cec5SDimitry Andric ret.push_back(std::move(sym)); 34680b57cec5SDimitry Andric 34690b57cec5SDimitry Andric // CU vectors and symbol names are adjacent in the output file. 34700b57cec5SDimitry Andric // We can compute their offsets in the output file now. 34710b57cec5SDimitry Andric size_t off = 0; 34720b57cec5SDimitry Andric for (GdbSymbol &sym : ret) { 34730b57cec5SDimitry Andric sym.cuVectorOff = off; 34740b57cec5SDimitry Andric off += (sym.cuVector.size() + 1) * 4; 34750b57cec5SDimitry Andric } 34760b57cec5SDimitry Andric for (GdbSymbol &sym : ret) { 34770b57cec5SDimitry Andric sym.nameOff = off; 34780b57cec5SDimitry Andric off += sym.name.size() + 1; 34790b57cec5SDimitry Andric } 3480bdd1243dSDimitry Andric // If off overflows, the last symbol's nameOff likely overflows. 3481bdd1243dSDimitry Andric if (!isUInt<32>(off)) 3482bdd1243dSDimitry Andric errorOrWarn("--gdb-index: constant pool size (" + Twine(off) + 3483bdd1243dSDimitry Andric ") exceeds UINT32_MAX"); 34840b57cec5SDimitry Andric 3485bdd1243dSDimitry Andric return {ret, off}; 34860b57cec5SDimitry Andric } 34870b57cec5SDimitry Andric 34880b57cec5SDimitry Andric // Returns a newly-created .gdb_index section. 3489*0fca6ea1SDimitry Andric template <class ELFT> 3490*0fca6ea1SDimitry Andric std::unique_ptr<GdbIndexSection> GdbIndexSection::create() { 3491bdd1243dSDimitry Andric llvm::TimeTraceScope timeScope("Create gdb index"); 3492bdd1243dSDimitry Andric 349316d6b3b3SDimitry Andric // Collect InputFiles with .debug_info. See the comment in 349416d6b3b3SDimitry Andric // LLDDwarfObj<ELFT>::LLDDwarfObj. If we do lightweight parsing in the future, 349516d6b3b3SDimitry Andric // note that isec->data() may uncompress the full content, which should be 349616d6b3b3SDimitry Andric // parallelized. 349716d6b3b3SDimitry Andric SetVector<InputFile *> files; 3498bdd1243dSDimitry Andric for (InputSectionBase *s : ctx.inputSections) { 349916d6b3b3SDimitry Andric InputSection *isec = dyn_cast<InputSection>(s); 350016d6b3b3SDimitry Andric if (!isec) 350116d6b3b3SDimitry Andric continue; 35020b57cec5SDimitry Andric // .debug_gnu_pub{names,types} are useless in executables. 35030b57cec5SDimitry Andric // They are present in input object files solely for creating 35040b57cec5SDimitry Andric // a .gdb_index. So we can remove them from the output. 35050b57cec5SDimitry Andric if (s->name == ".debug_gnu_pubnames" || s->name == ".debug_gnu_pubtypes") 35060b57cec5SDimitry Andric s->markDead(); 350716d6b3b3SDimitry Andric else if (isec->name == ".debug_info") 350816d6b3b3SDimitry Andric files.insert(isec->file); 350916d6b3b3SDimitry Andric } 3510e8d8bef9SDimitry Andric // Drop .rel[a].debug_gnu_pub{names,types} for --emit-relocs. 3511bdd1243dSDimitry Andric llvm::erase_if(ctx.inputSections, [](InputSectionBase *s) { 3512e8d8bef9SDimitry Andric if (auto *isec = dyn_cast<InputSection>(s)) 3513e8d8bef9SDimitry Andric if (InputSectionBase *rel = isec->getRelocatedSection()) 3514e8d8bef9SDimitry Andric return !rel->isLive(); 3515e8d8bef9SDimitry Andric return !s->isLive(); 3516e8d8bef9SDimitry Andric }); 35170b57cec5SDimitry Andric 35181fd87a68SDimitry Andric SmallVector<GdbChunk, 0> chunks(files.size()); 35191fd87a68SDimitry Andric SmallVector<SmallVector<NameAttrEntry, 0>, 0> nameAttrs(files.size()); 35200b57cec5SDimitry Andric 352181ad6265SDimitry Andric parallelFor(0, files.size(), [&](size_t i) { 35225ffd83dbSDimitry Andric // To keep memory usage low, we don't want to keep cached DWARFContext, so 35235ffd83dbSDimitry Andric // avoid getDwarf() here. 352416d6b3b3SDimitry Andric ObjFile<ELFT> *file = cast<ObjFile<ELFT>>(files[i]); 352585868e8aSDimitry Andric DWARFContext dwarf(std::make_unique<LLDDwarfObj<ELFT>>(file)); 352616d6b3b3SDimitry Andric auto &dobj = static_cast<const LLDDwarfObj<ELFT> &>(dwarf.getDWARFObj()); 35270b57cec5SDimitry Andric 352816d6b3b3SDimitry Andric // If the are multiple compile units .debug_info (very rare ld -r --unique), 352916d6b3b3SDimitry Andric // this only picks the last one. Other address ranges are lost. 353016d6b3b3SDimitry Andric chunks[i].sec = dobj.getInfoSection(); 35310b57cec5SDimitry Andric chunks[i].compilationUnits = readCuList(dwarf); 353216d6b3b3SDimitry Andric chunks[i].addressAreas = readAddressAreas(dwarf, chunks[i].sec); 353316d6b3b3SDimitry Andric nameAttrs[i] = readPubNamesAndTypes<ELFT>(dobj, chunks[i].compilationUnits); 35340b57cec5SDimitry Andric }); 35350b57cec5SDimitry Andric 3536*0fca6ea1SDimitry Andric auto ret = std::make_unique<GdbIndexSection>(); 35370b57cec5SDimitry Andric ret->chunks = std::move(chunks); 3538bdd1243dSDimitry Andric std::tie(ret->symbols, ret->size) = createSymbols(nameAttrs, ret->chunks); 3539bdd1243dSDimitry Andric 3540bdd1243dSDimitry Andric // Count the areas other than the constant pool. 3541bdd1243dSDimitry Andric ret->size += sizeof(GdbIndexHeader) + ret->computeSymtabSize() * 8; 3542bdd1243dSDimitry Andric for (GdbChunk &chunk : ret->chunks) 3543bdd1243dSDimitry Andric ret->size += 3544bdd1243dSDimitry Andric chunk.compilationUnits.size() * 16 + chunk.addressAreas.size() * 20; 3545bdd1243dSDimitry Andric 35460b57cec5SDimitry Andric return ret; 35470b57cec5SDimitry Andric } 35480b57cec5SDimitry Andric 35490b57cec5SDimitry Andric void GdbIndexSection::writeTo(uint8_t *buf) { 35500b57cec5SDimitry Andric // Write the header. 35510b57cec5SDimitry Andric auto *hdr = reinterpret_cast<GdbIndexHeader *>(buf); 35520b57cec5SDimitry Andric uint8_t *start = buf; 35530b57cec5SDimitry Andric hdr->version = 7; 35540b57cec5SDimitry Andric buf += sizeof(*hdr); 35550b57cec5SDimitry Andric 35560b57cec5SDimitry Andric // Write the CU list. 35570b57cec5SDimitry Andric hdr->cuListOff = buf - start; 35580b57cec5SDimitry Andric for (GdbChunk &chunk : chunks) { 35590b57cec5SDimitry Andric for (CuEntry &cu : chunk.compilationUnits) { 35600b57cec5SDimitry Andric write64le(buf, chunk.sec->outSecOff + cu.cuOffset); 35610b57cec5SDimitry Andric write64le(buf + 8, cu.cuLength); 35620b57cec5SDimitry Andric buf += 16; 35630b57cec5SDimitry Andric } 35640b57cec5SDimitry Andric } 35650b57cec5SDimitry Andric 35660b57cec5SDimitry Andric // Write the address area. 35670b57cec5SDimitry Andric hdr->cuTypesOff = buf - start; 35680b57cec5SDimitry Andric hdr->addressAreaOff = buf - start; 35690b57cec5SDimitry Andric uint32_t cuOff = 0; 35700b57cec5SDimitry Andric for (GdbChunk &chunk : chunks) { 35710b57cec5SDimitry Andric for (AddressEntry &e : chunk.addressAreas) { 3572e8d8bef9SDimitry Andric // In the case of ICF there may be duplicate address range entries. 3573e8d8bef9SDimitry Andric const uint64_t baseAddr = e.section->repl->getVA(0); 35740b57cec5SDimitry Andric write64le(buf, baseAddr + e.lowAddress); 35750b57cec5SDimitry Andric write64le(buf + 8, baseAddr + e.highAddress); 35760b57cec5SDimitry Andric write32le(buf + 16, e.cuIndex + cuOff); 35770b57cec5SDimitry Andric buf += 20; 35780b57cec5SDimitry Andric } 35790b57cec5SDimitry Andric cuOff += chunk.compilationUnits.size(); 35800b57cec5SDimitry Andric } 35810b57cec5SDimitry Andric 35820b57cec5SDimitry Andric // Write the on-disk open-addressing hash table containing symbols. 35830b57cec5SDimitry Andric hdr->symtabOff = buf - start; 35840b57cec5SDimitry Andric size_t symtabSize = computeSymtabSize(); 35850b57cec5SDimitry Andric uint32_t mask = symtabSize - 1; 35860b57cec5SDimitry Andric 35870b57cec5SDimitry Andric for (GdbSymbol &sym : symbols) { 35880b57cec5SDimitry Andric uint32_t h = sym.name.hash(); 35890b57cec5SDimitry Andric uint32_t i = h & mask; 35900b57cec5SDimitry Andric uint32_t step = ((h * 17) & mask) | 1; 35910b57cec5SDimitry Andric 35920b57cec5SDimitry Andric while (read32le(buf + i * 8)) 35930b57cec5SDimitry Andric i = (i + step) & mask; 35940b57cec5SDimitry Andric 35950b57cec5SDimitry Andric write32le(buf + i * 8, sym.nameOff); 35960b57cec5SDimitry Andric write32le(buf + i * 8 + 4, sym.cuVectorOff); 35970b57cec5SDimitry Andric } 35980b57cec5SDimitry Andric 35990b57cec5SDimitry Andric buf += symtabSize * 8; 36000b57cec5SDimitry Andric 36010b57cec5SDimitry Andric // Write the string pool. 36020b57cec5SDimitry Andric hdr->constantPoolOff = buf - start; 36030b57cec5SDimitry Andric parallelForEach(symbols, [&](GdbSymbol &sym) { 36040b57cec5SDimitry Andric memcpy(buf + sym.nameOff, sym.name.data(), sym.name.size()); 36050b57cec5SDimitry Andric }); 36060b57cec5SDimitry Andric 36070b57cec5SDimitry Andric // Write the CU vectors. 36080b57cec5SDimitry Andric for (GdbSymbol &sym : symbols) { 36090b57cec5SDimitry Andric write32le(buf, sym.cuVector.size()); 36100b57cec5SDimitry Andric buf += 4; 36110b57cec5SDimitry Andric for (uint32_t val : sym.cuVector) { 36120b57cec5SDimitry Andric write32le(buf, val); 36130b57cec5SDimitry Andric buf += 4; 36140b57cec5SDimitry Andric } 36150b57cec5SDimitry Andric } 36160b57cec5SDimitry Andric } 36170b57cec5SDimitry Andric 36180b57cec5SDimitry Andric bool GdbIndexSection::isNeeded() const { return !chunks.empty(); } 36190b57cec5SDimitry Andric 36200b57cec5SDimitry Andric EhFrameHeader::EhFrameHeader() 36210b57cec5SDimitry Andric : SyntheticSection(SHF_ALLOC, SHT_PROGBITS, 4, ".eh_frame_hdr") {} 36220b57cec5SDimitry Andric 36230b57cec5SDimitry Andric void EhFrameHeader::writeTo(uint8_t *buf) { 36240b57cec5SDimitry Andric // Unlike most sections, the EhFrameHeader section is written while writing 36250b57cec5SDimitry Andric // another section, namely EhFrameSection, which calls the write() function 36260b57cec5SDimitry Andric // below from its writeTo() function. This is necessary because the contents 36270b57cec5SDimitry Andric // of EhFrameHeader depend on the relocated contents of EhFrameSection and we 36280b57cec5SDimitry Andric // don't know which order the sections will be written in. 36290b57cec5SDimitry Andric } 36300b57cec5SDimitry Andric 36310b57cec5SDimitry Andric // .eh_frame_hdr contains a binary search table of pointers to FDEs. 36320b57cec5SDimitry Andric // Each entry of the search table consists of two values, 36330b57cec5SDimitry Andric // the starting PC from where FDEs covers, and the FDE's address. 36340b57cec5SDimitry Andric // It is sorted by PC. 36350b57cec5SDimitry Andric void EhFrameHeader::write() { 36360b57cec5SDimitry Andric uint8_t *buf = Out::bufferStart + getParent()->offset + outSecOff; 36370b57cec5SDimitry Andric using FdeData = EhFrameSection::FdeData; 363804eeddc0SDimitry Andric SmallVector<FdeData, 0> fdes = getPartition().ehFrame->getFdeData(); 36390b57cec5SDimitry Andric 36400b57cec5SDimitry Andric buf[0] = 1; 36410b57cec5SDimitry Andric buf[1] = DW_EH_PE_pcrel | DW_EH_PE_sdata4; 36420b57cec5SDimitry Andric buf[2] = DW_EH_PE_udata4; 36430b57cec5SDimitry Andric buf[3] = DW_EH_PE_datarel | DW_EH_PE_sdata4; 36440b57cec5SDimitry Andric write32(buf + 4, 36450b57cec5SDimitry Andric getPartition().ehFrame->getParent()->addr - this->getVA() - 4); 36460b57cec5SDimitry Andric write32(buf + 8, fdes.size()); 36470b57cec5SDimitry Andric buf += 12; 36480b57cec5SDimitry Andric 36490b57cec5SDimitry Andric for (FdeData &fde : fdes) { 36500b57cec5SDimitry Andric write32(buf, fde.pcRel); 36510b57cec5SDimitry Andric write32(buf + 4, fde.fdeVARel); 36520b57cec5SDimitry Andric buf += 8; 36530b57cec5SDimitry Andric } 36540b57cec5SDimitry Andric } 36550b57cec5SDimitry Andric 36560b57cec5SDimitry Andric size_t EhFrameHeader::getSize() const { 36570b57cec5SDimitry Andric // .eh_frame_hdr has a 12 bytes header followed by an array of FDEs. 36580b57cec5SDimitry Andric return 12 + getPartition().ehFrame->numFdes * 8; 36590b57cec5SDimitry Andric } 36600b57cec5SDimitry Andric 36610b57cec5SDimitry Andric bool EhFrameHeader::isNeeded() const { 36620b57cec5SDimitry Andric return isLive() && getPartition().ehFrame->isNeeded(); 36630b57cec5SDimitry Andric } 36640b57cec5SDimitry Andric 36650b57cec5SDimitry Andric VersionDefinitionSection::VersionDefinitionSection() 36660b57cec5SDimitry Andric : SyntheticSection(SHF_ALLOC, SHT_GNU_verdef, sizeof(uint32_t), 36670b57cec5SDimitry Andric ".gnu.version_d") {} 36680b57cec5SDimitry Andric 36690b57cec5SDimitry Andric StringRef VersionDefinitionSection::getFileDefName() { 36700b57cec5SDimitry Andric if (!getPartition().name.empty()) 36710b57cec5SDimitry Andric return getPartition().name; 36720b57cec5SDimitry Andric if (!config->soName.empty()) 36730b57cec5SDimitry Andric return config->soName; 36740b57cec5SDimitry Andric return config->outputFile; 36750b57cec5SDimitry Andric } 36760b57cec5SDimitry Andric 36770b57cec5SDimitry Andric void VersionDefinitionSection::finalizeContents() { 36780b57cec5SDimitry Andric fileDefNameOff = getPartition().dynStrTab->addString(getFileDefName()); 367985868e8aSDimitry Andric for (const VersionDefinition &v : namedVersionDefs()) 36800b57cec5SDimitry Andric verDefNameOffs.push_back(getPartition().dynStrTab->addString(v.name)); 36810b57cec5SDimitry Andric 36820b57cec5SDimitry Andric if (OutputSection *sec = getPartition().dynStrTab->getParent()) 36830b57cec5SDimitry Andric getParent()->link = sec->sectionIndex; 36840b57cec5SDimitry Andric 36850b57cec5SDimitry Andric // sh_info should be set to the number of definitions. This fact is missed in 36860b57cec5SDimitry Andric // documentation, but confirmed by binutils community: 36870b57cec5SDimitry Andric // https://sourceware.org/ml/binutils/2014-11/msg00355.html 36880b57cec5SDimitry Andric getParent()->info = getVerDefNum(); 36890b57cec5SDimitry Andric } 36900b57cec5SDimitry Andric 36910b57cec5SDimitry Andric void VersionDefinitionSection::writeOne(uint8_t *buf, uint32_t index, 36920b57cec5SDimitry Andric StringRef name, size_t nameOff) { 36930b57cec5SDimitry Andric uint16_t flags = index == 1 ? VER_FLG_BASE : 0; 36940b57cec5SDimitry Andric 36950b57cec5SDimitry Andric // Write a verdef. 36960b57cec5SDimitry Andric write16(buf, 1); // vd_version 36970b57cec5SDimitry Andric write16(buf + 2, flags); // vd_flags 36980b57cec5SDimitry Andric write16(buf + 4, index); // vd_ndx 36990b57cec5SDimitry Andric write16(buf + 6, 1); // vd_cnt 37000b57cec5SDimitry Andric write32(buf + 8, hashSysV(name)); // vd_hash 37010b57cec5SDimitry Andric write32(buf + 12, 20); // vd_aux 37020b57cec5SDimitry Andric write32(buf + 16, 28); // vd_next 37030b57cec5SDimitry Andric 37040b57cec5SDimitry Andric // Write a veraux. 37050b57cec5SDimitry Andric write32(buf + 20, nameOff); // vda_name 37060b57cec5SDimitry Andric write32(buf + 24, 0); // vda_next 37070b57cec5SDimitry Andric } 37080b57cec5SDimitry Andric 37090b57cec5SDimitry Andric void VersionDefinitionSection::writeTo(uint8_t *buf) { 37100b57cec5SDimitry Andric writeOne(buf, 1, getFileDefName(), fileDefNameOff); 37110b57cec5SDimitry Andric 37120b57cec5SDimitry Andric auto nameOffIt = verDefNameOffs.begin(); 371385868e8aSDimitry Andric for (const VersionDefinition &v : namedVersionDefs()) { 37140b57cec5SDimitry Andric buf += EntrySize; 37150b57cec5SDimitry Andric writeOne(buf, v.id, v.name, *nameOffIt++); 37160b57cec5SDimitry Andric } 37170b57cec5SDimitry Andric 37180b57cec5SDimitry Andric // Need to terminate the last version definition. 37190b57cec5SDimitry Andric write32(buf + 16, 0); // vd_next 37200b57cec5SDimitry Andric } 37210b57cec5SDimitry Andric 37220b57cec5SDimitry Andric size_t VersionDefinitionSection::getSize() const { 37230b57cec5SDimitry Andric return EntrySize * getVerDefNum(); 37240b57cec5SDimitry Andric } 37250b57cec5SDimitry Andric 37260b57cec5SDimitry Andric // .gnu.version is a table where each entry is 2 byte long. 37270b57cec5SDimitry Andric VersionTableSection::VersionTableSection() 37280b57cec5SDimitry Andric : SyntheticSection(SHF_ALLOC, SHT_GNU_versym, sizeof(uint16_t), 37290b57cec5SDimitry Andric ".gnu.version") { 37300b57cec5SDimitry Andric this->entsize = 2; 37310b57cec5SDimitry Andric } 37320b57cec5SDimitry Andric 37330b57cec5SDimitry Andric void VersionTableSection::finalizeContents() { 37340b57cec5SDimitry Andric // At the moment of june 2016 GNU docs does not mention that sh_link field 37350b57cec5SDimitry Andric // should be set, but Sun docs do. Also readelf relies on this field. 37360b57cec5SDimitry Andric getParent()->link = getPartition().dynSymTab->getParent()->sectionIndex; 37370b57cec5SDimitry Andric } 37380b57cec5SDimitry Andric 37390b57cec5SDimitry Andric size_t VersionTableSection::getSize() const { 37400b57cec5SDimitry Andric return (getPartition().dynSymTab->getSymbols().size() + 1) * 2; 37410b57cec5SDimitry Andric } 37420b57cec5SDimitry Andric 37430b57cec5SDimitry Andric void VersionTableSection::writeTo(uint8_t *buf) { 37440b57cec5SDimitry Andric buf += 2; 37450b57cec5SDimitry Andric for (const SymbolTableEntry &s : getPartition().dynSymTab->getSymbols()) { 37464824e7fdSDimitry Andric // For an unextracted lazy symbol (undefined weak), it must have been 3747349cc55cSDimitry Andric // converted to Undefined and have VER_NDX_GLOBAL version here. 3748349cc55cSDimitry Andric assert(!s.sym->isLazy()); 3749349cc55cSDimitry Andric write16(buf, s.sym->versionId); 37500b57cec5SDimitry Andric buf += 2; 37510b57cec5SDimitry Andric } 37520b57cec5SDimitry Andric } 37530b57cec5SDimitry Andric 37540b57cec5SDimitry Andric bool VersionTableSection::isNeeded() const { 3755480093f4SDimitry Andric return isLive() && 3756480093f4SDimitry Andric (getPartition().verDef || getPartition().verNeed->isNeeded()); 37570b57cec5SDimitry Andric } 37580b57cec5SDimitry Andric 37595ffd83dbSDimitry Andric void elf::addVerneed(Symbol *ss) { 37600b57cec5SDimitry Andric auto &file = cast<SharedFile>(*ss->file); 37615f757f3fSDimitry Andric if (ss->versionId == VER_NDX_GLOBAL) 37620b57cec5SDimitry Andric return; 37630b57cec5SDimitry Andric 37640b57cec5SDimitry Andric if (file.vernauxs.empty()) 37650b57cec5SDimitry Andric file.vernauxs.resize(file.verdefs.size()); 37660b57cec5SDimitry Andric 37670b57cec5SDimitry Andric // Select a version identifier for the vernaux data structure, if we haven't 37680b57cec5SDimitry Andric // already allocated one. The verdef identifiers cover the range 37690b57cec5SDimitry Andric // [1..getVerDefNum()]; this causes the vernaux identifiers to start from 37700b57cec5SDimitry Andric // getVerDefNum()+1. 37715f757f3fSDimitry Andric if (file.vernauxs[ss->versionId] == 0) 37725f757f3fSDimitry Andric file.vernauxs[ss->versionId] = ++SharedFile::vernauxNum + getVerDefNum(); 37730b57cec5SDimitry Andric 37745f757f3fSDimitry Andric ss->versionId = file.vernauxs[ss->versionId]; 37750b57cec5SDimitry Andric } 37760b57cec5SDimitry Andric 37770b57cec5SDimitry Andric template <class ELFT> 37780b57cec5SDimitry Andric VersionNeedSection<ELFT>::VersionNeedSection() 37790b57cec5SDimitry Andric : SyntheticSection(SHF_ALLOC, SHT_GNU_verneed, sizeof(uint32_t), 37800b57cec5SDimitry Andric ".gnu.version_r") {} 37810b57cec5SDimitry Andric 37820b57cec5SDimitry Andric template <class ELFT> void VersionNeedSection<ELFT>::finalizeContents() { 3783bdd1243dSDimitry Andric for (SharedFile *f : ctx.sharedFiles) { 37840b57cec5SDimitry Andric if (f->vernauxs.empty()) 37850b57cec5SDimitry Andric continue; 37860b57cec5SDimitry Andric verneeds.emplace_back(); 37870b57cec5SDimitry Andric Verneed &vn = verneeds.back(); 37880b57cec5SDimitry Andric vn.nameStrTab = getPartition().dynStrTab->addString(f->soName); 378906c3fb27SDimitry Andric bool isLibc = config->relrGlibc && f->soName.starts_with("libc.so."); 379081ad6265SDimitry Andric bool isGlibc2 = false; 37910b57cec5SDimitry Andric for (unsigned i = 0; i != f->vernauxs.size(); ++i) { 37920b57cec5SDimitry Andric if (f->vernauxs[i] == 0) 37930b57cec5SDimitry Andric continue; 37940b57cec5SDimitry Andric auto *verdef = 37950b57cec5SDimitry Andric reinterpret_cast<const typename ELFT::Verdef *>(f->verdefs[i]); 379681ad6265SDimitry Andric StringRef ver(f->getStringTable().data() + verdef->getAux()->vda_name); 379706c3fb27SDimitry Andric if (isLibc && ver.starts_with("GLIBC_2.")) 379881ad6265SDimitry Andric isGlibc2 = true; 379981ad6265SDimitry Andric vn.vernauxs.push_back({verdef->vd_hash, f->vernauxs[i], 380081ad6265SDimitry Andric getPartition().dynStrTab->addString(ver)}); 380181ad6265SDimitry Andric } 380281ad6265SDimitry Andric if (isGlibc2) { 380381ad6265SDimitry Andric const char *ver = "GLIBC_ABI_DT_RELR"; 380481ad6265SDimitry Andric vn.vernauxs.push_back({hashSysV(ver), 380581ad6265SDimitry Andric ++SharedFile::vernauxNum + getVerDefNum(), 380681ad6265SDimitry Andric getPartition().dynStrTab->addString(ver)}); 38070b57cec5SDimitry Andric } 38080b57cec5SDimitry Andric } 38090b57cec5SDimitry Andric 38100b57cec5SDimitry Andric if (OutputSection *sec = getPartition().dynStrTab->getParent()) 38110b57cec5SDimitry Andric getParent()->link = sec->sectionIndex; 38120b57cec5SDimitry Andric getParent()->info = verneeds.size(); 38130b57cec5SDimitry Andric } 38140b57cec5SDimitry Andric 38150b57cec5SDimitry Andric template <class ELFT> void VersionNeedSection<ELFT>::writeTo(uint8_t *buf) { 38160b57cec5SDimitry Andric // The Elf_Verneeds need to appear first, followed by the Elf_Vernauxs. 38170b57cec5SDimitry Andric auto *verneed = reinterpret_cast<Elf_Verneed *>(buf); 38180b57cec5SDimitry Andric auto *vernaux = reinterpret_cast<Elf_Vernaux *>(verneed + verneeds.size()); 38190b57cec5SDimitry Andric 38200b57cec5SDimitry Andric for (auto &vn : verneeds) { 38210b57cec5SDimitry Andric // Create an Elf_Verneed for this DSO. 38220b57cec5SDimitry Andric verneed->vn_version = 1; 38230b57cec5SDimitry Andric verneed->vn_cnt = vn.vernauxs.size(); 38240b57cec5SDimitry Andric verneed->vn_file = vn.nameStrTab; 38250b57cec5SDimitry Andric verneed->vn_aux = 38260b57cec5SDimitry Andric reinterpret_cast<char *>(vernaux) - reinterpret_cast<char *>(verneed); 38270b57cec5SDimitry Andric verneed->vn_next = sizeof(Elf_Verneed); 38280b57cec5SDimitry Andric ++verneed; 38290b57cec5SDimitry Andric 38300b57cec5SDimitry Andric // Create the Elf_Vernauxs for this Elf_Verneed. 38310b57cec5SDimitry Andric for (auto &vna : vn.vernauxs) { 38320b57cec5SDimitry Andric vernaux->vna_hash = vna.hash; 38330b57cec5SDimitry Andric vernaux->vna_flags = 0; 38340b57cec5SDimitry Andric vernaux->vna_other = vna.verneedIndex; 38350b57cec5SDimitry Andric vernaux->vna_name = vna.nameStrTab; 38360b57cec5SDimitry Andric vernaux->vna_next = sizeof(Elf_Vernaux); 38370b57cec5SDimitry Andric ++vernaux; 38380b57cec5SDimitry Andric } 38390b57cec5SDimitry Andric 38400b57cec5SDimitry Andric vernaux[-1].vna_next = 0; 38410b57cec5SDimitry Andric } 38420b57cec5SDimitry Andric verneed[-1].vn_next = 0; 38430b57cec5SDimitry Andric } 38440b57cec5SDimitry Andric 38450b57cec5SDimitry Andric template <class ELFT> size_t VersionNeedSection<ELFT>::getSize() const { 38460b57cec5SDimitry Andric return verneeds.size() * sizeof(Elf_Verneed) + 38470b57cec5SDimitry Andric SharedFile::vernauxNum * sizeof(Elf_Vernaux); 38480b57cec5SDimitry Andric } 38490b57cec5SDimitry Andric 38500b57cec5SDimitry Andric template <class ELFT> bool VersionNeedSection<ELFT>::isNeeded() const { 3851480093f4SDimitry Andric return isLive() && SharedFile::vernauxNum != 0; 38520b57cec5SDimitry Andric } 38530b57cec5SDimitry Andric 38540b57cec5SDimitry Andric void MergeSyntheticSection::addSection(MergeInputSection *ms) { 38550b57cec5SDimitry Andric ms->parent = this; 38560b57cec5SDimitry Andric sections.push_back(ms); 3857bdd1243dSDimitry Andric assert(addralign == ms->addralign || !(ms->flags & SHF_STRINGS)); 3858bdd1243dSDimitry Andric addralign = std::max(addralign, ms->addralign); 38590b57cec5SDimitry Andric } 38600b57cec5SDimitry Andric 38610b57cec5SDimitry Andric MergeTailSection::MergeTailSection(StringRef name, uint32_t type, 38620b57cec5SDimitry Andric uint64_t flags, uint32_t alignment) 38630b57cec5SDimitry Andric : MergeSyntheticSection(name, type, flags, alignment), 3864bdd1243dSDimitry Andric builder(StringTableBuilder::RAW, llvm::Align(alignment)) {} 38650b57cec5SDimitry Andric 38660b57cec5SDimitry Andric size_t MergeTailSection::getSize() const { return builder.getSize(); } 38670b57cec5SDimitry Andric 38680b57cec5SDimitry Andric void MergeTailSection::writeTo(uint8_t *buf) { builder.write(buf); } 38690b57cec5SDimitry Andric 38700b57cec5SDimitry Andric void MergeTailSection::finalizeContents() { 38710b57cec5SDimitry Andric // Add all string pieces to the string table builder to create section 38720b57cec5SDimitry Andric // contents. 38730b57cec5SDimitry Andric for (MergeInputSection *sec : sections) 38740b57cec5SDimitry Andric for (size_t i = 0, e = sec->pieces.size(); i != e; ++i) 38750b57cec5SDimitry Andric if (sec->pieces[i].live) 38760b57cec5SDimitry Andric builder.add(sec->getData(i)); 38770b57cec5SDimitry Andric 38780b57cec5SDimitry Andric // Fix the string table content. After this, the contents will never change. 38790b57cec5SDimitry Andric builder.finalize(); 38800b57cec5SDimitry Andric 38810b57cec5SDimitry Andric // finalize() fixed tail-optimized strings, so we can now get 38820b57cec5SDimitry Andric // offsets of strings. Get an offset for each string and save it 38830b57cec5SDimitry Andric // to a corresponding SectionPiece for easy access. 38840b57cec5SDimitry Andric for (MergeInputSection *sec : sections) 38850b57cec5SDimitry Andric for (size_t i = 0, e = sec->pieces.size(); i != e; ++i) 38860b57cec5SDimitry Andric if (sec->pieces[i].live) 38870b57cec5SDimitry Andric sec->pieces[i].outputOff = builder.getOffset(sec->getData(i)); 38880b57cec5SDimitry Andric } 38890b57cec5SDimitry Andric 38900b57cec5SDimitry Andric void MergeNoTailSection::writeTo(uint8_t *buf) { 389181ad6265SDimitry Andric parallelFor(0, numShards, 38920eae32dcSDimitry Andric [&](size_t i) { shards[i].write(buf + shardOffsets[i]); }); 38930b57cec5SDimitry Andric } 38940b57cec5SDimitry Andric 38950b57cec5SDimitry Andric // This function is very hot (i.e. it can take several seconds to finish) 38960b57cec5SDimitry Andric // because sometimes the number of inputs is in an order of magnitude of 38970b57cec5SDimitry Andric // millions. So, we use multi-threading. 38980b57cec5SDimitry Andric // 38990b57cec5SDimitry Andric // For any strings S and T, we know S is not mergeable with T if S's hash 39000b57cec5SDimitry Andric // value is different from T's. If that's the case, we can safely put S and 39010b57cec5SDimitry Andric // T into different string builders without worrying about merge misses. 39020b57cec5SDimitry Andric // We do it in parallel. 39030b57cec5SDimitry Andric void MergeNoTailSection::finalizeContents() { 39040b57cec5SDimitry Andric // Initializes string table builders. 39050b57cec5SDimitry Andric for (size_t i = 0; i < numShards; ++i) 3906bdd1243dSDimitry Andric shards.emplace_back(StringTableBuilder::RAW, llvm::Align(addralign)); 39070b57cec5SDimitry Andric 39080b57cec5SDimitry Andric // Concurrency level. Must be a power of 2 to avoid expensive modulo 39090b57cec5SDimitry Andric // operations in the following tight loop. 3910bdd1243dSDimitry Andric const size_t concurrency = 391106c3fb27SDimitry Andric llvm::bit_floor(std::min<size_t>(config->threadCount, numShards)); 39120b57cec5SDimitry Andric 39130b57cec5SDimitry Andric // Add section pieces to the builders. 391481ad6265SDimitry Andric parallelFor(0, concurrency, [&](size_t threadId) { 39150b57cec5SDimitry Andric for (MergeInputSection *sec : sections) { 39160b57cec5SDimitry Andric for (size_t i = 0, e = sec->pieces.size(); i != e; ++i) { 39170b57cec5SDimitry Andric if (!sec->pieces[i].live) 39180b57cec5SDimitry Andric continue; 39190b57cec5SDimitry Andric size_t shardId = getShardId(sec->pieces[i].hash); 39200b57cec5SDimitry Andric if ((shardId & (concurrency - 1)) == threadId) 39210b57cec5SDimitry Andric sec->pieces[i].outputOff = shards[shardId].add(sec->getData(i)); 39220b57cec5SDimitry Andric } 39230b57cec5SDimitry Andric } 39240b57cec5SDimitry Andric }); 39250b57cec5SDimitry Andric 39260b57cec5SDimitry Andric // Compute an in-section offset for each shard. 39270b57cec5SDimitry Andric size_t off = 0; 39280b57cec5SDimitry Andric for (size_t i = 0; i < numShards; ++i) { 39290b57cec5SDimitry Andric shards[i].finalizeInOrder(); 39300b57cec5SDimitry Andric if (shards[i].getSize() > 0) 3931bdd1243dSDimitry Andric off = alignToPowerOf2(off, addralign); 39320b57cec5SDimitry Andric shardOffsets[i] = off; 39330b57cec5SDimitry Andric off += shards[i].getSize(); 39340b57cec5SDimitry Andric } 39350b57cec5SDimitry Andric size = off; 39360b57cec5SDimitry Andric 39370b57cec5SDimitry Andric // So far, section pieces have offsets from beginning of shards, but 39380b57cec5SDimitry Andric // we want offsets from beginning of the whole section. Fix them. 39390b57cec5SDimitry Andric parallelForEach(sections, [&](MergeInputSection *sec) { 39400b57cec5SDimitry Andric for (size_t i = 0, e = sec->pieces.size(); i != e; ++i) 39410b57cec5SDimitry Andric if (sec->pieces[i].live) 39420b57cec5SDimitry Andric sec->pieces[i].outputOff += 39430b57cec5SDimitry Andric shardOffsets[getShardId(sec->pieces[i].hash)]; 39440b57cec5SDimitry Andric }); 39450b57cec5SDimitry Andric } 39460b57cec5SDimitry Andric 39475ffd83dbSDimitry Andric template <class ELFT> void elf::splitSections() { 39485ffd83dbSDimitry Andric llvm::TimeTraceScope timeScope("Split sections"); 39490b57cec5SDimitry Andric // splitIntoPieces needs to be called on each MergeInputSection 39500b57cec5SDimitry Andric // before calling finalizeContents(). 3951bdd1243dSDimitry Andric parallelForEach(ctx.objectFiles, [](ELFFileBase *file) { 39521fd87a68SDimitry Andric for (InputSectionBase *sec : file->getSections()) { 39531fd87a68SDimitry Andric if (!sec) 39541fd87a68SDimitry Andric continue; 39550b57cec5SDimitry Andric if (auto *s = dyn_cast<MergeInputSection>(sec)) 39560b57cec5SDimitry Andric s->splitIntoPieces(); 39570b57cec5SDimitry Andric else if (auto *eh = dyn_cast<EhInputSection>(sec)) 39580b57cec5SDimitry Andric eh->split<ELFT>(); 39591fd87a68SDimitry Andric } 39600b57cec5SDimitry Andric }); 39610b57cec5SDimitry Andric } 39620b57cec5SDimitry Andric 3963bdd1243dSDimitry Andric void elf::combineEhSections() { 3964bdd1243dSDimitry Andric llvm::TimeTraceScope timeScope("Combine EH sections"); 3965bdd1243dSDimitry Andric for (EhInputSection *sec : ctx.ehInputSections) { 3966bdd1243dSDimitry Andric EhFrameSection &eh = *sec->getPartition().ehFrame; 3967bdd1243dSDimitry Andric sec->parent = &eh; 3968bdd1243dSDimitry Andric eh.addralign = std::max(eh.addralign, sec->addralign); 3969bdd1243dSDimitry Andric eh.sections.push_back(sec); 3970bdd1243dSDimitry Andric llvm::append_range(eh.dependentSections, sec->dependentSections); 3971bdd1243dSDimitry Andric } 3972bdd1243dSDimitry Andric 3973bdd1243dSDimitry Andric if (!mainPart->armExidx) 3974bdd1243dSDimitry Andric return; 3975bdd1243dSDimitry Andric llvm::erase_if(ctx.inputSections, [](InputSectionBase *s) { 3976bdd1243dSDimitry Andric // Ignore dead sections and the partition end marker (.part.end), 3977bdd1243dSDimitry Andric // whose partition number is out of bounds. 3978bdd1243dSDimitry Andric if (!s->isLive() || s->partition == 255) 3979bdd1243dSDimitry Andric return false; 3980bdd1243dSDimitry Andric Partition &part = s->getPartition(); 3981bdd1243dSDimitry Andric return s->kind() == SectionBase::Regular && part.armExidx && 3982bdd1243dSDimitry Andric part.armExidx->addSection(cast<InputSection>(s)); 3983bdd1243dSDimitry Andric }); 3984bdd1243dSDimitry Andric } 3985bdd1243dSDimitry Andric 39860b57cec5SDimitry Andric MipsRldMapSection::MipsRldMapSection() 39870b57cec5SDimitry Andric : SyntheticSection(SHF_ALLOC | SHF_WRITE, SHT_PROGBITS, config->wordsize, 39880b57cec5SDimitry Andric ".rld_map") {} 39890b57cec5SDimitry Andric 39900b57cec5SDimitry Andric ARMExidxSyntheticSection::ARMExidxSyntheticSection() 39910b57cec5SDimitry Andric : SyntheticSection(SHF_ALLOC | SHF_LINK_ORDER, SHT_ARM_EXIDX, 39920b57cec5SDimitry Andric config->wordsize, ".ARM.exidx") {} 39930b57cec5SDimitry Andric 39940b57cec5SDimitry Andric static InputSection *findExidxSection(InputSection *isec) { 39950b57cec5SDimitry Andric for (InputSection *d : isec->dependentSections) 39965ffd83dbSDimitry Andric if (d->type == SHT_ARM_EXIDX && d->isLive()) 39970b57cec5SDimitry Andric return d; 39980b57cec5SDimitry Andric return nullptr; 39990b57cec5SDimitry Andric } 40000b57cec5SDimitry Andric 400185868e8aSDimitry Andric static bool isValidExidxSectionDep(InputSection *isec) { 400285868e8aSDimitry Andric return (isec->flags & SHF_ALLOC) && (isec->flags & SHF_EXECINSTR) && 400385868e8aSDimitry Andric isec->getSize() > 0; 400485868e8aSDimitry Andric } 400585868e8aSDimitry Andric 40060b57cec5SDimitry Andric bool ARMExidxSyntheticSection::addSection(InputSection *isec) { 40070b57cec5SDimitry Andric if (isec->type == SHT_ARM_EXIDX) { 400885868e8aSDimitry Andric if (InputSection *dep = isec->getLinkOrderDep()) 40095ffd83dbSDimitry Andric if (isValidExidxSectionDep(dep)) { 40100b57cec5SDimitry Andric exidxSections.push_back(isec); 40115ffd83dbSDimitry Andric // Every exidxSection is 8 bytes, we need an estimate of 40125ffd83dbSDimitry Andric // size before assignAddresses can be called. Final size 40135ffd83dbSDimitry Andric // will only be known after finalize is called. 40145ffd83dbSDimitry Andric size += 8; 40155ffd83dbSDimitry Andric } 40160b57cec5SDimitry Andric return true; 40170b57cec5SDimitry Andric } 40180b57cec5SDimitry Andric 401985868e8aSDimitry Andric if (isValidExidxSectionDep(isec)) { 40200b57cec5SDimitry Andric executableSections.push_back(isec); 40210b57cec5SDimitry Andric return false; 40220b57cec5SDimitry Andric } 40230b57cec5SDimitry Andric 40240b57cec5SDimitry Andric // FIXME: we do not output a relocation section when --emit-relocs is used 40250b57cec5SDimitry Andric // as we do not have relocation sections for linker generated table entries 40260b57cec5SDimitry Andric // and we would have to erase at a late stage relocations from merged entries. 40270b57cec5SDimitry Andric // Given that exception tables are already position independent and a binary 40280b57cec5SDimitry Andric // analyzer could derive the relocations we choose to erase the relocations. 40290b57cec5SDimitry Andric if (config->emitRelocs && isec->type == SHT_REL) 40300b57cec5SDimitry Andric if (InputSectionBase *ex = isec->getRelocatedSection()) 40310b57cec5SDimitry Andric if (isa<InputSection>(ex) && ex->type == SHT_ARM_EXIDX) 40320b57cec5SDimitry Andric return true; 40330b57cec5SDimitry Andric 40340b57cec5SDimitry Andric return false; 40350b57cec5SDimitry Andric } 40360b57cec5SDimitry Andric 40370b57cec5SDimitry Andric // References to .ARM.Extab Sections have bit 31 clear and are not the 40380b57cec5SDimitry Andric // special EXIDX_CANTUNWIND bit-pattern. 40390b57cec5SDimitry Andric static bool isExtabRef(uint32_t unwind) { 40400b57cec5SDimitry Andric return (unwind & 0x80000000) == 0 && unwind != 0x1; 40410b57cec5SDimitry Andric } 40420b57cec5SDimitry Andric 40430b57cec5SDimitry Andric // Return true if the .ARM.exidx section Cur can be merged into the .ARM.exidx 40440b57cec5SDimitry Andric // section Prev, where Cur follows Prev in the table. This can be done if the 40450b57cec5SDimitry Andric // unwinding instructions in Cur are identical to Prev. Linker generated 40460b57cec5SDimitry Andric // EXIDX_CANTUNWIND entries are represented by nullptr as they do not have an 40470b57cec5SDimitry Andric // InputSection. 40480b57cec5SDimitry Andric static bool isDuplicateArmExidxSec(InputSection *prev, InputSection *cur) { 40490b57cec5SDimitry Andric // Get the last table Entry from the previous .ARM.exidx section. If Prev is 40500b57cec5SDimitry Andric // nullptr then it will be a synthesized EXIDX_CANTUNWIND entry. 405106c3fb27SDimitry Andric uint32_t prevUnwind = 1; 40520b57cec5SDimitry Andric if (prev) 405306c3fb27SDimitry Andric prevUnwind = read32(prev->content().data() + prev->content().size() - 4); 405406c3fb27SDimitry Andric if (isExtabRef(prevUnwind)) 40550b57cec5SDimitry Andric return false; 40560b57cec5SDimitry Andric 40570b57cec5SDimitry Andric // We consider the unwind instructions of an .ARM.exidx table entry 40580b57cec5SDimitry Andric // a duplicate if the previous unwind instructions if: 40590b57cec5SDimitry Andric // - Both are the special EXIDX_CANTUNWIND. 40600b57cec5SDimitry Andric // - Both are the same inline unwind instructions. 40610b57cec5SDimitry Andric // We do not attempt to follow and check links into .ARM.extab tables as 40620b57cec5SDimitry Andric // consecutive identical entries are rare and the effort to check that they 40630b57cec5SDimitry Andric // are identical is high. 40640b57cec5SDimitry Andric 40650b57cec5SDimitry Andric // If Cur is nullptr then this is synthesized EXIDX_CANTUNWIND entry. 40660b57cec5SDimitry Andric if (cur == nullptr) 406706c3fb27SDimitry Andric return prevUnwind == 1; 40680b57cec5SDimitry Andric 406906c3fb27SDimitry Andric for (uint32_t offset = 4; offset < (uint32_t)cur->content().size(); offset +=8) { 407006c3fb27SDimitry Andric uint32_t curUnwind = read32(cur->content().data() + offset); 407106c3fb27SDimitry Andric if (isExtabRef(curUnwind) || curUnwind != prevUnwind) 40720b57cec5SDimitry Andric return false; 407306c3fb27SDimitry Andric } 40740b57cec5SDimitry Andric // All table entries in this .ARM.exidx Section can be merged into the 40750b57cec5SDimitry Andric // previous Section. 40760b57cec5SDimitry Andric return true; 40770b57cec5SDimitry Andric } 40780b57cec5SDimitry Andric 40790b57cec5SDimitry Andric // The .ARM.exidx table must be sorted in ascending order of the address of the 4080bdd1243dSDimitry Andric // functions the table describes. std::optionally duplicate adjacent table 4081bdd1243dSDimitry Andric // entries can be removed. At the end of the function the executableSections 4082bdd1243dSDimitry Andric // must be sorted in ascending order of address, Sentinel is set to the 4083bdd1243dSDimitry Andric // InputSection with the highest address and any InputSections that have 4084bdd1243dSDimitry Andric // mergeable .ARM.exidx table entries are removed from it. 40850b57cec5SDimitry Andric void ARMExidxSyntheticSection::finalizeContents() { 4086*0fca6ea1SDimitry Andric // Ensure that any fixed-point iterations after the first see the original set 4087*0fca6ea1SDimitry Andric // of sections. 4088*0fca6ea1SDimitry Andric if (!originalExecutableSections.empty()) 4089*0fca6ea1SDimitry Andric executableSections = originalExecutableSections; 4090*0fca6ea1SDimitry Andric else if (config->enableNonContiguousRegions) 4091*0fca6ea1SDimitry Andric originalExecutableSections = executableSections; 4092*0fca6ea1SDimitry Andric 409385868e8aSDimitry Andric // The executableSections and exidxSections that we use to derive the final 409485868e8aSDimitry Andric // contents of this SyntheticSection are populated before 409585868e8aSDimitry Andric // processSectionCommands() and ICF. A /DISCARD/ entry in SECTIONS command or 409685868e8aSDimitry Andric // ICF may remove executable InputSections and their dependent .ARM.exidx 409785868e8aSDimitry Andric // section that we recorded earlier. 40980b57cec5SDimitry Andric auto isDiscarded = [](const InputSection *isec) { return !isec->isLive(); }; 40990b57cec5SDimitry Andric llvm::erase_if(exidxSections, isDiscarded); 41005ffd83dbSDimitry Andric // We need to remove discarded InputSections and InputSections without 41015ffd83dbSDimitry Andric // .ARM.exidx sections that if we generated the .ARM.exidx it would be out 41025ffd83dbSDimitry Andric // of range. 41035ffd83dbSDimitry Andric auto isDiscardedOrOutOfRange = [this](InputSection *isec) { 41045ffd83dbSDimitry Andric if (!isec->isLive()) 41055ffd83dbSDimitry Andric return true; 41065ffd83dbSDimitry Andric if (findExidxSection(isec)) 41075ffd83dbSDimitry Andric return false; 41085ffd83dbSDimitry Andric int64_t off = static_cast<int64_t>(isec->getVA() - getVA()); 41095ffd83dbSDimitry Andric return off != llvm::SignExtend64(off, 31); 41105ffd83dbSDimitry Andric }; 41115ffd83dbSDimitry Andric llvm::erase_if(executableSections, isDiscardedOrOutOfRange); 41120b57cec5SDimitry Andric 41130b57cec5SDimitry Andric // Sort the executable sections that may or may not have associated 41140b57cec5SDimitry Andric // .ARM.exidx sections by order of ascending address. This requires the 41155ffd83dbSDimitry Andric // relative positions of InputSections and OutputSections to be known. 41160b57cec5SDimitry Andric auto compareByFilePosition = [](const InputSection *a, 41170b57cec5SDimitry Andric const InputSection *b) { 41180b57cec5SDimitry Andric OutputSection *aOut = a->getParent(); 41190b57cec5SDimitry Andric OutputSection *bOut = b->getParent(); 41200b57cec5SDimitry Andric 41210b57cec5SDimitry Andric if (aOut != bOut) 41225ffd83dbSDimitry Andric return aOut->addr < bOut->addr; 41230b57cec5SDimitry Andric return a->outSecOff < b->outSecOff; 41240b57cec5SDimitry Andric }; 41250b57cec5SDimitry Andric llvm::stable_sort(executableSections, compareByFilePosition); 41260b57cec5SDimitry Andric sentinel = executableSections.back(); 4127bdd1243dSDimitry Andric // std::optionally merge adjacent duplicate entries. 41280b57cec5SDimitry Andric if (config->mergeArmExidx) { 41291fd87a68SDimitry Andric SmallVector<InputSection *, 0> selectedSections; 41300b57cec5SDimitry Andric selectedSections.reserve(executableSections.size()); 41310b57cec5SDimitry Andric selectedSections.push_back(executableSections[0]); 41320b57cec5SDimitry Andric size_t prev = 0; 41330b57cec5SDimitry Andric for (size_t i = 1; i < executableSections.size(); ++i) { 41340b57cec5SDimitry Andric InputSection *ex1 = findExidxSection(executableSections[prev]); 41350b57cec5SDimitry Andric InputSection *ex2 = findExidxSection(executableSections[i]); 41360b57cec5SDimitry Andric if (!isDuplicateArmExidxSec(ex1, ex2)) { 41370b57cec5SDimitry Andric selectedSections.push_back(executableSections[i]); 41380b57cec5SDimitry Andric prev = i; 41390b57cec5SDimitry Andric } 41400b57cec5SDimitry Andric } 41410b57cec5SDimitry Andric executableSections = std::move(selectedSections); 41420b57cec5SDimitry Andric } 414306c3fb27SDimitry Andric // offset is within the SyntheticSection. 41440b57cec5SDimitry Andric size_t offset = 0; 41450b57cec5SDimitry Andric size = 0; 41460b57cec5SDimitry Andric for (InputSection *isec : executableSections) { 41470b57cec5SDimitry Andric if (InputSection *d = findExidxSection(isec)) { 41480b57cec5SDimitry Andric d->outSecOff = offset; 41490b57cec5SDimitry Andric d->parent = getParent(); 41500b57cec5SDimitry Andric offset += d->getSize(); 41510b57cec5SDimitry Andric } else { 41520b57cec5SDimitry Andric offset += 8; 41530b57cec5SDimitry Andric } 41540b57cec5SDimitry Andric } 41550b57cec5SDimitry Andric // Size includes Sentinel. 41560b57cec5SDimitry Andric size = offset + 8; 41570b57cec5SDimitry Andric } 41580b57cec5SDimitry Andric 41590b57cec5SDimitry Andric InputSection *ARMExidxSyntheticSection::getLinkOrderDep() const { 41600b57cec5SDimitry Andric return executableSections.front(); 41610b57cec5SDimitry Andric } 41620b57cec5SDimitry Andric 41630b57cec5SDimitry Andric // To write the .ARM.exidx table from the ExecutableSections we have three cases 41640b57cec5SDimitry Andric // 1.) The InputSection has a .ARM.exidx InputSection in its dependent sections. 41650b57cec5SDimitry Andric // We write the .ARM.exidx section contents and apply its relocations. 41660b57cec5SDimitry Andric // 2.) The InputSection does not have a dependent .ARM.exidx InputSection. We 41670b57cec5SDimitry Andric // must write the contents of an EXIDX_CANTUNWIND directly. We use the 41680b57cec5SDimitry Andric // start of the InputSection as the purpose of the linker generated 41690b57cec5SDimitry Andric // section is to terminate the address range of the previous entry. 41700b57cec5SDimitry Andric // 3.) A trailing EXIDX_CANTUNWIND sentinel section is required at the end of 41710b57cec5SDimitry Andric // the table to terminate the address range of the final entry. 41720b57cec5SDimitry Andric void ARMExidxSyntheticSection::writeTo(uint8_t *buf) { 41730b57cec5SDimitry Andric 417406c3fb27SDimitry Andric // A linker generated CANTUNWIND entry is made up of two words: 417506c3fb27SDimitry Andric // 0x0 with R_ARM_PREL31 relocation to target. 417606c3fb27SDimitry Andric // 0x1 with EXIDX_CANTUNWIND. 41770b57cec5SDimitry Andric uint64_t offset = 0; 41780b57cec5SDimitry Andric for (InputSection *isec : executableSections) { 41790b57cec5SDimitry Andric assert(isec->getParent() != nullptr); 41800b57cec5SDimitry Andric if (InputSection *d = findExidxSection(isec)) { 418106c3fb27SDimitry Andric for (int dataOffset = 0; dataOffset != (int)d->content().size(); 418206c3fb27SDimitry Andric dataOffset += 4) 418306c3fb27SDimitry Andric write32(buf + offset + dataOffset, 418406c3fb27SDimitry Andric read32(d->content().data() + dataOffset)); 418506c3fb27SDimitry Andric // Recalculate outSecOff as finalizeAddressDependentContent() 418606c3fb27SDimitry Andric // may have altered syntheticSection outSecOff. 418706c3fb27SDimitry Andric d->outSecOff = offset + outSecOff; 418806c3fb27SDimitry Andric target->relocateAlloc(*d, buf + offset); 41890b57cec5SDimitry Andric offset += d->getSize(); 41900b57cec5SDimitry Andric } else { 41910b57cec5SDimitry Andric // A Linker generated CANTUNWIND section. 419206c3fb27SDimitry Andric write32(buf + offset + 0, 0x0); 419306c3fb27SDimitry Andric write32(buf + offset + 4, 0x1); 41940b57cec5SDimitry Andric uint64_t s = isec->getVA(); 41950b57cec5SDimitry Andric uint64_t p = getVA() + offset; 41965ffd83dbSDimitry Andric target->relocateNoSym(buf + offset, R_ARM_PREL31, s - p); 41970b57cec5SDimitry Andric offset += 8; 41980b57cec5SDimitry Andric } 41990b57cec5SDimitry Andric } 420006c3fb27SDimitry Andric // Write Sentinel CANTUNWIND entry. 420106c3fb27SDimitry Andric write32(buf + offset + 0, 0x0); 420206c3fb27SDimitry Andric write32(buf + offset + 4, 0x1); 42030b57cec5SDimitry Andric uint64_t s = sentinel->getVA(sentinel->getSize()); 42040b57cec5SDimitry Andric uint64_t p = getVA() + offset; 42055ffd83dbSDimitry Andric target->relocateNoSym(buf + offset, R_ARM_PREL31, s - p); 42060b57cec5SDimitry Andric assert(size == offset + 8); 42070b57cec5SDimitry Andric } 42080b57cec5SDimitry Andric 420985868e8aSDimitry Andric bool ARMExidxSyntheticSection::isNeeded() const { 4210349cc55cSDimitry Andric return llvm::any_of(exidxSections, 4211349cc55cSDimitry Andric [](InputSection *isec) { return isec->isLive(); }); 421285868e8aSDimitry Andric } 421385868e8aSDimitry Andric 42140b57cec5SDimitry Andric ThunkSection::ThunkSection(OutputSection *os, uint64_t off) 4215e8d8bef9SDimitry Andric : SyntheticSection(SHF_ALLOC | SHF_EXECINSTR, SHT_PROGBITS, 4216e8d8bef9SDimitry Andric config->emachine == EM_PPC64 ? 16 : 4, ".text.thunk") { 42170b57cec5SDimitry Andric this->parent = os; 42180b57cec5SDimitry Andric this->outSecOff = off; 42190b57cec5SDimitry Andric } 42200b57cec5SDimitry Andric 4221480093f4SDimitry Andric size_t ThunkSection::getSize() const { 422213138422SDimitry Andric if (roundUpSizeForErrata) 4223480093f4SDimitry Andric return alignTo(size, 4096); 4224480093f4SDimitry Andric return size; 4225480093f4SDimitry Andric } 4226480093f4SDimitry Andric 42270b57cec5SDimitry Andric void ThunkSection::addThunk(Thunk *t) { 42280b57cec5SDimitry Andric thunks.push_back(t); 42290b57cec5SDimitry Andric t->addSymbols(*this); 42300b57cec5SDimitry Andric } 42310b57cec5SDimitry Andric 42320b57cec5SDimitry Andric void ThunkSection::writeTo(uint8_t *buf) { 42330b57cec5SDimitry Andric for (Thunk *t : thunks) 42340b57cec5SDimitry Andric t->writeTo(buf + t->offset); 42350b57cec5SDimitry Andric } 42360b57cec5SDimitry Andric 42370b57cec5SDimitry Andric InputSection *ThunkSection::getTargetInputSection() const { 42380b57cec5SDimitry Andric if (thunks.empty()) 42390b57cec5SDimitry Andric return nullptr; 42400b57cec5SDimitry Andric const Thunk *t = thunks.front(); 42410b57cec5SDimitry Andric return t->getTargetInputSection(); 42420b57cec5SDimitry Andric } 42430b57cec5SDimitry Andric 42440b57cec5SDimitry Andric bool ThunkSection::assignOffsets() { 42450b57cec5SDimitry Andric uint64_t off = 0; 42460b57cec5SDimitry Andric for (Thunk *t : thunks) { 4247972a253aSDimitry Andric off = alignToPowerOf2(off, t->alignment); 42480b57cec5SDimitry Andric t->setOffset(off); 42490b57cec5SDimitry Andric uint32_t size = t->size(); 42500b57cec5SDimitry Andric t->getThunkTargetSym()->size = size; 42510b57cec5SDimitry Andric off += size; 42520b57cec5SDimitry Andric } 42530b57cec5SDimitry Andric bool changed = off != size; 42540b57cec5SDimitry Andric size = off; 42550b57cec5SDimitry Andric return changed; 42560b57cec5SDimitry Andric } 42570b57cec5SDimitry Andric 42580b57cec5SDimitry Andric PPC32Got2Section::PPC32Got2Section() 42590b57cec5SDimitry Andric : SyntheticSection(SHF_ALLOC | SHF_WRITE, SHT_PROGBITS, 4, ".got2") {} 42600b57cec5SDimitry Andric 42610b57cec5SDimitry Andric bool PPC32Got2Section::isNeeded() const { 42620b57cec5SDimitry Andric // See the comment below. This is not needed if there is no other 42630b57cec5SDimitry Andric // InputSection. 42644824e7fdSDimitry Andric for (SectionCommand *cmd : getParent()->commands) 42654824e7fdSDimitry Andric if (auto *isd = dyn_cast<InputSectionDescription>(cmd)) 42660b57cec5SDimitry Andric for (InputSection *isec : isd->sections) 42670b57cec5SDimitry Andric if (isec != this) 42680b57cec5SDimitry Andric return true; 42690b57cec5SDimitry Andric return false; 42700b57cec5SDimitry Andric } 42710b57cec5SDimitry Andric 42720b57cec5SDimitry Andric void PPC32Got2Section::finalizeContents() { 42730b57cec5SDimitry Andric // PPC32 may create multiple GOT sections for -fPIC/-fPIE, one per file in 42740b57cec5SDimitry Andric // .got2 . This function computes outSecOff of each .got2 to be used in 42750b57cec5SDimitry Andric // PPC32PltCallStub::writeTo(). The purpose of this empty synthetic section is 42760b57cec5SDimitry Andric // to collect input sections named ".got2". 42774824e7fdSDimitry Andric for (SectionCommand *cmd : getParent()->commands) 42784824e7fdSDimitry Andric if (auto *isd = dyn_cast<InputSectionDescription>(cmd)) { 42790b57cec5SDimitry Andric for (InputSection *isec : isd->sections) { 42800eae32dcSDimitry Andric // isec->file may be nullptr for MergeSyntheticSection. 42810eae32dcSDimitry Andric if (isec != this && isec->file) 42820eae32dcSDimitry Andric isec->file->ppc32Got2 = isec; 42830b57cec5SDimitry Andric } 42840b57cec5SDimitry Andric } 42850b57cec5SDimitry Andric } 42860b57cec5SDimitry Andric 42870b57cec5SDimitry Andric // If linking position-dependent code then the table will store the addresses 42880b57cec5SDimitry Andric // directly in the binary so the section has type SHT_PROGBITS. If linking 42890b57cec5SDimitry Andric // position-independent code the section has type SHT_NOBITS since it will be 42900b57cec5SDimitry Andric // allocated and filled in by the dynamic linker. 42910b57cec5SDimitry Andric PPC64LongBranchTargetSection::PPC64LongBranchTargetSection() 42920b57cec5SDimitry Andric : SyntheticSection(SHF_ALLOC | SHF_WRITE, 42930b57cec5SDimitry Andric config->isPic ? SHT_NOBITS : SHT_PROGBITS, 8, 42940b57cec5SDimitry Andric ".branch_lt") {} 42950b57cec5SDimitry Andric 4296480093f4SDimitry Andric uint64_t PPC64LongBranchTargetSection::getEntryVA(const Symbol *sym, 4297480093f4SDimitry Andric int64_t addend) { 4298480093f4SDimitry Andric return getVA() + entry_index.find({sym, addend})->second * 8; 4299480093f4SDimitry Andric } 4300480093f4SDimitry Andric 4301bdd1243dSDimitry Andric std::optional<uint32_t> 4302bdd1243dSDimitry Andric PPC64LongBranchTargetSection::addEntry(const Symbol *sym, int64_t addend) { 4303480093f4SDimitry Andric auto res = 4304480093f4SDimitry Andric entry_index.try_emplace(std::make_pair(sym, addend), entries.size()); 4305480093f4SDimitry Andric if (!res.second) 4306bdd1243dSDimitry Andric return std::nullopt; 4307480093f4SDimitry Andric entries.emplace_back(sym, addend); 4308480093f4SDimitry Andric return res.first->second; 43090b57cec5SDimitry Andric } 43100b57cec5SDimitry Andric 43110b57cec5SDimitry Andric size_t PPC64LongBranchTargetSection::getSize() const { 43120b57cec5SDimitry Andric return entries.size() * 8; 43130b57cec5SDimitry Andric } 43140b57cec5SDimitry Andric 43150b57cec5SDimitry Andric void PPC64LongBranchTargetSection::writeTo(uint8_t *buf) { 43160b57cec5SDimitry Andric // If linking non-pic we have the final addresses of the targets and they get 43170b57cec5SDimitry Andric // written to the table directly. For pic the dynamic linker will allocate 4318bdd1243dSDimitry Andric // the section and fill it. 43190b57cec5SDimitry Andric if (config->isPic) 43200b57cec5SDimitry Andric return; 43210b57cec5SDimitry Andric 4322480093f4SDimitry Andric for (auto entry : entries) { 4323480093f4SDimitry Andric const Symbol *sym = entry.first; 4324480093f4SDimitry Andric int64_t addend = entry.second; 43250b57cec5SDimitry Andric assert(sym->getVA()); 43260b57cec5SDimitry Andric // Need calls to branch to the local entry-point since a long-branch 43270b57cec5SDimitry Andric // must be a local-call. 4328480093f4SDimitry Andric write64(buf, sym->getVA(addend) + 4329480093f4SDimitry Andric getPPC64GlobalEntryToLocalEntryOffset(sym->stOther)); 43300b57cec5SDimitry Andric buf += 8; 43310b57cec5SDimitry Andric } 43320b57cec5SDimitry Andric } 43330b57cec5SDimitry Andric 43340b57cec5SDimitry Andric bool PPC64LongBranchTargetSection::isNeeded() const { 43350b57cec5SDimitry Andric // `removeUnusedSyntheticSections()` is called before thunk allocation which 43360b57cec5SDimitry Andric // is too early to determine if this section will be empty or not. We need 43370b57cec5SDimitry Andric // Finalized to keep the section alive until after thunk creation. Finalized 43380b57cec5SDimitry Andric // only gets set to true once `finalizeSections()` is called after thunk 4339480093f4SDimitry Andric // creation. Because of this, if we don't create any long-branch thunks we end 43400b57cec5SDimitry Andric // up with an empty .branch_lt section in the binary. 43410b57cec5SDimitry Andric return !finalized || !entries.empty(); 43420b57cec5SDimitry Andric } 43430b57cec5SDimitry Andric 43440b57cec5SDimitry Andric static uint8_t getAbiVersion() { 43450b57cec5SDimitry Andric // MIPS non-PIC executable gets ABI version 1. 43460b57cec5SDimitry Andric if (config->emachine == EM_MIPS) { 43470b57cec5SDimitry Andric if (!config->isPic && !config->relocatable && 43480b57cec5SDimitry Andric (config->eflags & (EF_MIPS_PIC | EF_MIPS_CPIC)) == EF_MIPS_CPIC) 43490b57cec5SDimitry Andric return 1; 43500b57cec5SDimitry Andric return 0; 43510b57cec5SDimitry Andric } 43520b57cec5SDimitry Andric 4353bdd1243dSDimitry Andric if (config->emachine == EM_AMDGPU && !ctx.objectFiles.empty()) { 4354bdd1243dSDimitry Andric uint8_t ver = ctx.objectFiles[0]->abiVersion; 4355bdd1243dSDimitry Andric for (InputFile *file : ArrayRef(ctx.objectFiles).slice(1)) 43560b57cec5SDimitry Andric if (file->abiVersion != ver) 43570b57cec5SDimitry Andric error("incompatible ABI version: " + toString(file)); 43580b57cec5SDimitry Andric return ver; 43590b57cec5SDimitry Andric } 43600b57cec5SDimitry Andric 43610b57cec5SDimitry Andric return 0; 43620b57cec5SDimitry Andric } 43630b57cec5SDimitry Andric 43645ffd83dbSDimitry Andric template <typename ELFT> void elf::writeEhdr(uint8_t *buf, Partition &part) { 43650b57cec5SDimitry Andric memcpy(buf, "\177ELF", 4); 43660b57cec5SDimitry Andric 43670b57cec5SDimitry Andric auto *eHdr = reinterpret_cast<typename ELFT::Ehdr *>(buf); 4368*0fca6ea1SDimitry Andric eHdr->e_ident[EI_CLASS] = ELFT::Is64Bits ? ELFCLASS64 : ELFCLASS32; 4369*0fca6ea1SDimitry Andric eHdr->e_ident[EI_DATA] = 4370*0fca6ea1SDimitry Andric ELFT::Endianness == endianness::little ? ELFDATA2LSB : ELFDATA2MSB; 43710b57cec5SDimitry Andric eHdr->e_ident[EI_VERSION] = EV_CURRENT; 43720b57cec5SDimitry Andric eHdr->e_ident[EI_OSABI] = config->osabi; 43730b57cec5SDimitry Andric eHdr->e_ident[EI_ABIVERSION] = getAbiVersion(); 43740b57cec5SDimitry Andric eHdr->e_machine = config->emachine; 43750b57cec5SDimitry Andric eHdr->e_version = EV_CURRENT; 43760b57cec5SDimitry Andric eHdr->e_flags = config->eflags; 43770b57cec5SDimitry Andric eHdr->e_ehsize = sizeof(typename ELFT::Ehdr); 43780b57cec5SDimitry Andric eHdr->e_phnum = part.phdrs.size(); 43790b57cec5SDimitry Andric eHdr->e_shentsize = sizeof(typename ELFT::Shdr); 43800b57cec5SDimitry Andric 43810b57cec5SDimitry Andric if (!config->relocatable) { 43820b57cec5SDimitry Andric eHdr->e_phoff = sizeof(typename ELFT::Ehdr); 43830b57cec5SDimitry Andric eHdr->e_phentsize = sizeof(typename ELFT::Phdr); 43840b57cec5SDimitry Andric } 43850b57cec5SDimitry Andric } 43860b57cec5SDimitry Andric 43875ffd83dbSDimitry Andric template <typename ELFT> void elf::writePhdrs(uint8_t *buf, Partition &part) { 43880b57cec5SDimitry Andric // Write the program header table. 43890b57cec5SDimitry Andric auto *hBuf = reinterpret_cast<typename ELFT::Phdr *>(buf); 43900b57cec5SDimitry Andric for (PhdrEntry *p : part.phdrs) { 43910b57cec5SDimitry Andric hBuf->p_type = p->p_type; 43920b57cec5SDimitry Andric hBuf->p_flags = p->p_flags; 43930b57cec5SDimitry Andric hBuf->p_offset = p->p_offset; 43940b57cec5SDimitry Andric hBuf->p_vaddr = p->p_vaddr; 43950b57cec5SDimitry Andric hBuf->p_paddr = p->p_paddr; 43960b57cec5SDimitry Andric hBuf->p_filesz = p->p_filesz; 43970b57cec5SDimitry Andric hBuf->p_memsz = p->p_memsz; 43980b57cec5SDimitry Andric hBuf->p_align = p->p_align; 43990b57cec5SDimitry Andric ++hBuf; 44000b57cec5SDimitry Andric } 44010b57cec5SDimitry Andric } 44020b57cec5SDimitry Andric 44030b57cec5SDimitry Andric template <typename ELFT> 44040b57cec5SDimitry Andric PartitionElfHeaderSection<ELFT>::PartitionElfHeaderSection() 44050b57cec5SDimitry Andric : SyntheticSection(SHF_ALLOC, SHT_LLVM_PART_EHDR, 1, "") {} 44060b57cec5SDimitry Andric 44070b57cec5SDimitry Andric template <typename ELFT> 44080b57cec5SDimitry Andric size_t PartitionElfHeaderSection<ELFT>::getSize() const { 44090b57cec5SDimitry Andric return sizeof(typename ELFT::Ehdr); 44100b57cec5SDimitry Andric } 44110b57cec5SDimitry Andric 44120b57cec5SDimitry Andric template <typename ELFT> 44130b57cec5SDimitry Andric void PartitionElfHeaderSection<ELFT>::writeTo(uint8_t *buf) { 44140b57cec5SDimitry Andric writeEhdr<ELFT>(buf, getPartition()); 44150b57cec5SDimitry Andric 44160b57cec5SDimitry Andric // Loadable partitions are always ET_DYN. 44170b57cec5SDimitry Andric auto *eHdr = reinterpret_cast<typename ELFT::Ehdr *>(buf); 44180b57cec5SDimitry Andric eHdr->e_type = ET_DYN; 44190b57cec5SDimitry Andric } 44200b57cec5SDimitry Andric 44210b57cec5SDimitry Andric template <typename ELFT> 44220b57cec5SDimitry Andric PartitionProgramHeadersSection<ELFT>::PartitionProgramHeadersSection() 44230b57cec5SDimitry Andric : SyntheticSection(SHF_ALLOC, SHT_LLVM_PART_PHDR, 1, ".phdrs") {} 44240b57cec5SDimitry Andric 44250b57cec5SDimitry Andric template <typename ELFT> 44260b57cec5SDimitry Andric size_t PartitionProgramHeadersSection<ELFT>::getSize() const { 44270b57cec5SDimitry Andric return sizeof(typename ELFT::Phdr) * getPartition().phdrs.size(); 44280b57cec5SDimitry Andric } 44290b57cec5SDimitry Andric 44300b57cec5SDimitry Andric template <typename ELFT> 44310b57cec5SDimitry Andric void PartitionProgramHeadersSection<ELFT>::writeTo(uint8_t *buf) { 44320b57cec5SDimitry Andric writePhdrs<ELFT>(buf, getPartition()); 44330b57cec5SDimitry Andric } 44340b57cec5SDimitry Andric 44350b57cec5SDimitry Andric PartitionIndexSection::PartitionIndexSection() 44360b57cec5SDimitry Andric : SyntheticSection(SHF_ALLOC, SHT_PROGBITS, 4, ".rodata") {} 44370b57cec5SDimitry Andric 44380b57cec5SDimitry Andric size_t PartitionIndexSection::getSize() const { 44390b57cec5SDimitry Andric return 12 * (partitions.size() - 1); 44400b57cec5SDimitry Andric } 44410b57cec5SDimitry Andric 44420b57cec5SDimitry Andric void PartitionIndexSection::finalizeContents() { 44430b57cec5SDimitry Andric for (size_t i = 1; i != partitions.size(); ++i) 44440b57cec5SDimitry Andric partitions[i].nameStrTab = mainPart->dynStrTab->addString(partitions[i].name); 44450b57cec5SDimitry Andric } 44460b57cec5SDimitry Andric 44470b57cec5SDimitry Andric void PartitionIndexSection::writeTo(uint8_t *buf) { 44480b57cec5SDimitry Andric uint64_t va = getVA(); 44490b57cec5SDimitry Andric for (size_t i = 1; i != partitions.size(); ++i) { 44500b57cec5SDimitry Andric write32(buf, mainPart->dynStrTab->getVA() + partitions[i].nameStrTab - va); 44510b57cec5SDimitry Andric write32(buf + 4, partitions[i].elfHeader->getVA() - (va + 4)); 44520b57cec5SDimitry Andric 445304eeddc0SDimitry Andric SyntheticSection *next = i == partitions.size() - 1 445404eeddc0SDimitry Andric ? in.partEnd.get() 445504eeddc0SDimitry Andric : partitions[i + 1].elfHeader.get(); 44560b57cec5SDimitry Andric write32(buf + 8, next->getVA() - partitions[i].elfHeader->getVA()); 44570b57cec5SDimitry Andric 44580b57cec5SDimitry Andric va += 12; 44590b57cec5SDimitry Andric buf += 12; 44600b57cec5SDimitry Andric } 44610b57cec5SDimitry Andric } 44620b57cec5SDimitry Andric 446304eeddc0SDimitry Andric void InStruct::reset() { 446404eeddc0SDimitry Andric attributes.reset(); 4465bdd1243dSDimitry Andric riscvAttributes.reset(); 446604eeddc0SDimitry Andric bss.reset(); 446704eeddc0SDimitry Andric bssRelRo.reset(); 446804eeddc0SDimitry Andric got.reset(); 446904eeddc0SDimitry Andric gotPlt.reset(); 447004eeddc0SDimitry Andric igotPlt.reset(); 44715f757f3fSDimitry Andric relroPadding.reset(); 447206c3fb27SDimitry Andric armCmseSGSection.reset(); 447304eeddc0SDimitry Andric ppc64LongBranchTarget.reset(); 44741fd87a68SDimitry Andric mipsAbiFlags.reset(); 447504eeddc0SDimitry Andric mipsGot.reset(); 44761fd87a68SDimitry Andric mipsOptions.reset(); 44771fd87a68SDimitry Andric mipsReginfo.reset(); 447804eeddc0SDimitry Andric mipsRldMap.reset(); 447904eeddc0SDimitry Andric partEnd.reset(); 448004eeddc0SDimitry Andric partIndex.reset(); 448104eeddc0SDimitry Andric plt.reset(); 448204eeddc0SDimitry Andric iplt.reset(); 448304eeddc0SDimitry Andric ppc32Got2.reset(); 448404eeddc0SDimitry Andric ibtPlt.reset(); 448504eeddc0SDimitry Andric relaPlt.reset(); 4486*0fca6ea1SDimitry Andric debugNames.reset(); 4487*0fca6ea1SDimitry Andric gdbIndex.reset(); 448804eeddc0SDimitry Andric shStrTab.reset(); 448904eeddc0SDimitry Andric strTab.reset(); 449004eeddc0SDimitry Andric symTab.reset(); 449104eeddc0SDimitry Andric symTabShndx.reset(); 449204eeddc0SDimitry Andric } 449304eeddc0SDimitry Andric 4494*0fca6ea1SDimitry Andric static bool needsInterpSection() { 4495*0fca6ea1SDimitry Andric return !config->relocatable && !config->shared && 4496*0fca6ea1SDimitry Andric !config->dynamicLinker.empty() && script->needsInterpSection(); 4497*0fca6ea1SDimitry Andric } 4498*0fca6ea1SDimitry Andric 4499*0fca6ea1SDimitry Andric bool elf::hasMemtag() { 4500*0fca6ea1SDimitry Andric return config->emachine == EM_AARCH64 && 4501*0fca6ea1SDimitry Andric config->androidMemtagMode != ELF::NT_MEMTAG_LEVEL_NONE; 4502*0fca6ea1SDimitry Andric } 4503*0fca6ea1SDimitry Andric 4504*0fca6ea1SDimitry Andric // Fully static executables don't support MTE globals at this point in time, as 4505*0fca6ea1SDimitry Andric // we currently rely on: 4506*0fca6ea1SDimitry Andric // - A dynamic loader to process relocations, and 4507*0fca6ea1SDimitry Andric // - Dynamic entries. 4508*0fca6ea1SDimitry Andric // This restriction could be removed in future by re-using some of the ideas 4509*0fca6ea1SDimitry Andric // that ifuncs use in fully static executables. 4510*0fca6ea1SDimitry Andric bool elf::canHaveMemtagGlobals() { 4511*0fca6ea1SDimitry Andric return hasMemtag() && 4512*0fca6ea1SDimitry Andric (config->relocatable || config->shared || needsInterpSection()); 4513*0fca6ea1SDimitry Andric } 4514*0fca6ea1SDimitry Andric 451581ad6265SDimitry Andric constexpr char kMemtagAndroidNoteName[] = "Android"; 451681ad6265SDimitry Andric void MemtagAndroidNote::writeTo(uint8_t *buf) { 451706c3fb27SDimitry Andric static_assert( 451806c3fb27SDimitry Andric sizeof(kMemtagAndroidNoteName) == 8, 451906c3fb27SDimitry Andric "Android 11 & 12 have an ABI that the note name is 8 bytes long. Keep it " 452006c3fb27SDimitry Andric "that way for backwards compatibility."); 452181ad6265SDimitry Andric 452281ad6265SDimitry Andric write32(buf, sizeof(kMemtagAndroidNoteName)); 452381ad6265SDimitry Andric write32(buf + 4, sizeof(uint32_t)); 452481ad6265SDimitry Andric write32(buf + 8, ELF::NT_ANDROID_TYPE_MEMTAG); 452581ad6265SDimitry Andric memcpy(buf + 12, kMemtagAndroidNoteName, sizeof(kMemtagAndroidNoteName)); 452606c3fb27SDimitry Andric buf += 12 + alignTo(sizeof(kMemtagAndroidNoteName), 4); 452781ad6265SDimitry Andric 452881ad6265SDimitry Andric uint32_t value = 0; 452981ad6265SDimitry Andric value |= config->androidMemtagMode; 453081ad6265SDimitry Andric if (config->androidMemtagHeap) 453181ad6265SDimitry Andric value |= ELF::NT_MEMTAG_HEAP; 453281ad6265SDimitry Andric // Note, MTE stack is an ABI break. Attempting to run an MTE stack-enabled 453381ad6265SDimitry Andric // binary on Android 11 or 12 will result in a checkfail in the loader. 453481ad6265SDimitry Andric if (config->androidMemtagStack) 453581ad6265SDimitry Andric value |= ELF::NT_MEMTAG_STACK; 453681ad6265SDimitry Andric write32(buf, value); // note value 453781ad6265SDimitry Andric } 453881ad6265SDimitry Andric 453981ad6265SDimitry Andric size_t MemtagAndroidNote::getSize() const { 454081ad6265SDimitry Andric return sizeof(llvm::ELF::Elf64_Nhdr) + 454106c3fb27SDimitry Andric /*namesz=*/alignTo(sizeof(kMemtagAndroidNoteName), 4) + 454281ad6265SDimitry Andric /*descsz=*/sizeof(uint32_t); 454381ad6265SDimitry Andric } 454481ad6265SDimitry Andric 454561cfbce3SDimitry Andric void PackageMetadataNote::writeTo(uint8_t *buf) { 454661cfbce3SDimitry Andric write32(buf, 4); 454761cfbce3SDimitry Andric write32(buf + 4, config->packageMetadata.size() + 1); 454861cfbce3SDimitry Andric write32(buf + 8, FDO_PACKAGING_METADATA); 454961cfbce3SDimitry Andric memcpy(buf + 12, "FDO", 4); 455061cfbce3SDimitry Andric memcpy(buf + 16, config->packageMetadata.data(), 455161cfbce3SDimitry Andric config->packageMetadata.size()); 455261cfbce3SDimitry Andric } 455361cfbce3SDimitry Andric 455461cfbce3SDimitry Andric size_t PackageMetadataNote::getSize() const { 455561cfbce3SDimitry Andric return sizeof(llvm::ELF::Elf64_Nhdr) + 4 + 455661cfbce3SDimitry Andric alignTo(config->packageMetadata.size() + 1, 4); 455761cfbce3SDimitry Andric } 455861cfbce3SDimitry Andric 45595f757f3fSDimitry Andric // Helper function, return the size of the ULEB128 for 'v', optionally writing 45605f757f3fSDimitry Andric // it to `*(buf + offset)` if `buf` is non-null. 45615f757f3fSDimitry Andric static size_t computeOrWriteULEB128(uint64_t v, uint8_t *buf, size_t offset) { 45625f757f3fSDimitry Andric if (buf) 45635f757f3fSDimitry Andric return encodeULEB128(v, buf + offset); 45645f757f3fSDimitry Andric return getULEB128Size(v); 45655f757f3fSDimitry Andric } 45665f757f3fSDimitry Andric 45675f757f3fSDimitry Andric // https://github.com/ARM-software/abi-aa/blob/main/memtagabielf64/memtagabielf64.rst#83encoding-of-sht_aarch64_memtag_globals_dynamic 45685f757f3fSDimitry Andric constexpr uint64_t kMemtagStepSizeBits = 3; 45695f757f3fSDimitry Andric constexpr uint64_t kMemtagGranuleSize = 16; 45701db9f3b2SDimitry Andric static size_t 45711db9f3b2SDimitry Andric createMemtagGlobalDescriptors(const SmallVector<const Symbol *, 0> &symbols, 45725f757f3fSDimitry Andric uint8_t *buf = nullptr) { 45735f757f3fSDimitry Andric size_t sectionSize = 0; 45745f757f3fSDimitry Andric uint64_t lastGlobalEnd = 0; 45755f757f3fSDimitry Andric 45765f757f3fSDimitry Andric for (const Symbol *sym : symbols) { 45775f757f3fSDimitry Andric if (!includeInSymtab(*sym)) 45785f757f3fSDimitry Andric continue; 45795f757f3fSDimitry Andric const uint64_t addr = sym->getVA(); 45805f757f3fSDimitry Andric const uint64_t size = sym->getSize(); 45815f757f3fSDimitry Andric 45825f757f3fSDimitry Andric if (addr <= kMemtagGranuleSize && buf != nullptr) 45835f757f3fSDimitry Andric errorOrWarn("address of the tagged symbol \"" + sym->getName() + 45845f757f3fSDimitry Andric "\" falls in the ELF header. This is indicative of a " 45855f757f3fSDimitry Andric "compiler/linker bug"); 45865f757f3fSDimitry Andric if (addr % kMemtagGranuleSize != 0) 45875f757f3fSDimitry Andric errorOrWarn("address of the tagged symbol \"" + sym->getName() + 45885f757f3fSDimitry Andric "\" at 0x" + Twine::utohexstr(addr) + 45895f757f3fSDimitry Andric "\" is not granule (16-byte) aligned"); 45905f757f3fSDimitry Andric if (size == 0) 45915f757f3fSDimitry Andric errorOrWarn("size of the tagged symbol \"" + sym->getName() + 45925f757f3fSDimitry Andric "\" is not allowed to be zero"); 45935f757f3fSDimitry Andric if (size % kMemtagGranuleSize != 0) 45945f757f3fSDimitry Andric errorOrWarn("size of the tagged symbol \"" + sym->getName() + 45955f757f3fSDimitry Andric "\" (size 0x" + Twine::utohexstr(size) + 45965f757f3fSDimitry Andric ") is not granule (16-byte) aligned"); 45975f757f3fSDimitry Andric 45985f757f3fSDimitry Andric const uint64_t sizeToEncode = size / kMemtagGranuleSize; 45995f757f3fSDimitry Andric const uint64_t stepToEncode = ((addr - lastGlobalEnd) / kMemtagGranuleSize) 46005f757f3fSDimitry Andric << kMemtagStepSizeBits; 46015f757f3fSDimitry Andric if (sizeToEncode < (1 << kMemtagStepSizeBits)) { 46025f757f3fSDimitry Andric sectionSize += computeOrWriteULEB128(stepToEncode | sizeToEncode, buf, sectionSize); 46035f757f3fSDimitry Andric } else { 46045f757f3fSDimitry Andric sectionSize += computeOrWriteULEB128(stepToEncode, buf, sectionSize); 46055f757f3fSDimitry Andric sectionSize += computeOrWriteULEB128(sizeToEncode - 1, buf, sectionSize); 46065f757f3fSDimitry Andric } 46075f757f3fSDimitry Andric lastGlobalEnd = addr + size; 46085f757f3fSDimitry Andric } 46095f757f3fSDimitry Andric 46105f757f3fSDimitry Andric return sectionSize; 46115f757f3fSDimitry Andric } 46125f757f3fSDimitry Andric 46131db9f3b2SDimitry Andric bool MemtagGlobalDescriptors::updateAllocSize() { 46145f757f3fSDimitry Andric size_t oldSize = getSize(); 46155f757f3fSDimitry Andric std::stable_sort(symbols.begin(), symbols.end(), 46165f757f3fSDimitry Andric [](const Symbol *s1, const Symbol *s2) { 46175f757f3fSDimitry Andric return s1->getVA() < s2->getVA(); 46185f757f3fSDimitry Andric }); 46195f757f3fSDimitry Andric return oldSize != getSize(); 46205f757f3fSDimitry Andric } 46215f757f3fSDimitry Andric 46221db9f3b2SDimitry Andric void MemtagGlobalDescriptors::writeTo(uint8_t *buf) { 46231db9f3b2SDimitry Andric createMemtagGlobalDescriptors(symbols, buf); 46245f757f3fSDimitry Andric } 46255f757f3fSDimitry Andric 46261db9f3b2SDimitry Andric size_t MemtagGlobalDescriptors::getSize() const { 46271db9f3b2SDimitry Andric return createMemtagGlobalDescriptors(symbols); 46285f757f3fSDimitry Andric } 46295f757f3fSDimitry Andric 4630*0fca6ea1SDimitry Andric static OutputSection *findSection(StringRef name) { 4631*0fca6ea1SDimitry Andric for (SectionCommand *cmd : script->sectionCommands) 4632*0fca6ea1SDimitry Andric if (auto *osd = dyn_cast<OutputDesc>(cmd)) 4633*0fca6ea1SDimitry Andric if (osd->osec.name == name) 4634*0fca6ea1SDimitry Andric return &osd->osec; 4635*0fca6ea1SDimitry Andric return nullptr; 4636*0fca6ea1SDimitry Andric } 4637*0fca6ea1SDimitry Andric 4638*0fca6ea1SDimitry Andric static Defined *addOptionalRegular(StringRef name, SectionBase *sec, 4639*0fca6ea1SDimitry Andric uint64_t val, uint8_t stOther = STV_HIDDEN) { 4640*0fca6ea1SDimitry Andric Symbol *s = symtab.find(name); 4641*0fca6ea1SDimitry Andric if (!s || s->isDefined() || s->isCommon()) 4642*0fca6ea1SDimitry Andric return nullptr; 4643*0fca6ea1SDimitry Andric 4644*0fca6ea1SDimitry Andric s->resolve(Defined{ctx.internalFile, StringRef(), STB_GLOBAL, stOther, 4645*0fca6ea1SDimitry Andric STT_NOTYPE, val, 4646*0fca6ea1SDimitry Andric /*size=*/0, sec}); 4647*0fca6ea1SDimitry Andric s->isUsedInRegularObj = true; 4648*0fca6ea1SDimitry Andric return cast<Defined>(s); 4649*0fca6ea1SDimitry Andric } 4650*0fca6ea1SDimitry Andric 4651*0fca6ea1SDimitry Andric template <class ELFT> void elf::createSyntheticSections() { 4652*0fca6ea1SDimitry Andric // Initialize all pointers with NULL. This is needed because 4653*0fca6ea1SDimitry Andric // you can call lld::elf::main more than once as a library. 4654*0fca6ea1SDimitry Andric Out::tlsPhdr = nullptr; 4655*0fca6ea1SDimitry Andric Out::preinitArray = nullptr; 4656*0fca6ea1SDimitry Andric Out::initArray = nullptr; 4657*0fca6ea1SDimitry Andric Out::finiArray = nullptr; 4658*0fca6ea1SDimitry Andric 4659*0fca6ea1SDimitry Andric // Add the .interp section first because it is not a SyntheticSection. 4660*0fca6ea1SDimitry Andric // The removeUnusedSyntheticSections() function relies on the 4661*0fca6ea1SDimitry Andric // SyntheticSections coming last. 4662*0fca6ea1SDimitry Andric if (needsInterpSection()) { 4663*0fca6ea1SDimitry Andric for (size_t i = 1; i <= partitions.size(); ++i) { 4664*0fca6ea1SDimitry Andric InputSection *sec = createInterpSection(); 4665*0fca6ea1SDimitry Andric sec->partition = i; 4666*0fca6ea1SDimitry Andric ctx.inputSections.push_back(sec); 4667*0fca6ea1SDimitry Andric } 4668*0fca6ea1SDimitry Andric } 4669*0fca6ea1SDimitry Andric 4670*0fca6ea1SDimitry Andric auto add = [](SyntheticSection &sec) { ctx.inputSections.push_back(&sec); }; 4671*0fca6ea1SDimitry Andric 4672*0fca6ea1SDimitry Andric in.shStrTab = std::make_unique<StringTableSection>(".shstrtab", false); 4673*0fca6ea1SDimitry Andric 4674*0fca6ea1SDimitry Andric Out::programHeaders = make<OutputSection>("", 0, SHF_ALLOC); 4675*0fca6ea1SDimitry Andric Out::programHeaders->addralign = config->wordsize; 4676*0fca6ea1SDimitry Andric 4677*0fca6ea1SDimitry Andric if (config->strip != StripPolicy::All) { 4678*0fca6ea1SDimitry Andric in.strTab = std::make_unique<StringTableSection>(".strtab", false); 4679*0fca6ea1SDimitry Andric in.symTab = std::make_unique<SymbolTableSection<ELFT>>(*in.strTab); 4680*0fca6ea1SDimitry Andric in.symTabShndx = std::make_unique<SymtabShndxSection>(); 4681*0fca6ea1SDimitry Andric } 4682*0fca6ea1SDimitry Andric 4683*0fca6ea1SDimitry Andric in.bss = std::make_unique<BssSection>(".bss", 0, 1); 4684*0fca6ea1SDimitry Andric add(*in.bss); 4685*0fca6ea1SDimitry Andric 4686*0fca6ea1SDimitry Andric // If there is a SECTIONS command and a .data.rel.ro section name use name 4687*0fca6ea1SDimitry Andric // .data.rel.ro.bss so that we match in the .data.rel.ro output section. 4688*0fca6ea1SDimitry Andric // This makes sure our relro is contiguous. 4689*0fca6ea1SDimitry Andric bool hasDataRelRo = script->hasSectionsCommand && findSection(".data.rel.ro"); 4690*0fca6ea1SDimitry Andric in.bssRelRo = std::make_unique<BssSection>( 4691*0fca6ea1SDimitry Andric hasDataRelRo ? ".data.rel.ro.bss" : ".bss.rel.ro", 0, 1); 4692*0fca6ea1SDimitry Andric add(*in.bssRelRo); 4693*0fca6ea1SDimitry Andric 4694*0fca6ea1SDimitry Andric // Add MIPS-specific sections. 4695*0fca6ea1SDimitry Andric if (config->emachine == EM_MIPS) { 4696*0fca6ea1SDimitry Andric if (!config->shared && config->hasDynSymTab) { 4697*0fca6ea1SDimitry Andric in.mipsRldMap = std::make_unique<MipsRldMapSection>(); 4698*0fca6ea1SDimitry Andric add(*in.mipsRldMap); 4699*0fca6ea1SDimitry Andric } 4700*0fca6ea1SDimitry Andric if ((in.mipsAbiFlags = MipsAbiFlagsSection<ELFT>::create())) 4701*0fca6ea1SDimitry Andric add(*in.mipsAbiFlags); 4702*0fca6ea1SDimitry Andric if ((in.mipsOptions = MipsOptionsSection<ELFT>::create())) 4703*0fca6ea1SDimitry Andric add(*in.mipsOptions); 4704*0fca6ea1SDimitry Andric if ((in.mipsReginfo = MipsReginfoSection<ELFT>::create())) 4705*0fca6ea1SDimitry Andric add(*in.mipsReginfo); 4706*0fca6ea1SDimitry Andric } 4707*0fca6ea1SDimitry Andric 4708*0fca6ea1SDimitry Andric StringRef relaDynName = config->isRela ? ".rela.dyn" : ".rel.dyn"; 4709*0fca6ea1SDimitry Andric 4710*0fca6ea1SDimitry Andric const unsigned threadCount = config->threadCount; 4711*0fca6ea1SDimitry Andric for (Partition &part : partitions) { 4712*0fca6ea1SDimitry Andric auto add = [&](SyntheticSection &sec) { 4713*0fca6ea1SDimitry Andric sec.partition = part.getNumber(); 4714*0fca6ea1SDimitry Andric ctx.inputSections.push_back(&sec); 4715*0fca6ea1SDimitry Andric }; 4716*0fca6ea1SDimitry Andric 4717*0fca6ea1SDimitry Andric if (!part.name.empty()) { 4718*0fca6ea1SDimitry Andric part.elfHeader = std::make_unique<PartitionElfHeaderSection<ELFT>>(); 4719*0fca6ea1SDimitry Andric part.elfHeader->name = part.name; 4720*0fca6ea1SDimitry Andric add(*part.elfHeader); 4721*0fca6ea1SDimitry Andric 4722*0fca6ea1SDimitry Andric part.programHeaders = 4723*0fca6ea1SDimitry Andric std::make_unique<PartitionProgramHeadersSection<ELFT>>(); 4724*0fca6ea1SDimitry Andric add(*part.programHeaders); 4725*0fca6ea1SDimitry Andric } 4726*0fca6ea1SDimitry Andric 4727*0fca6ea1SDimitry Andric if (config->buildId != BuildIdKind::None) { 4728*0fca6ea1SDimitry Andric part.buildId = std::make_unique<BuildIdSection>(); 4729*0fca6ea1SDimitry Andric add(*part.buildId); 4730*0fca6ea1SDimitry Andric } 4731*0fca6ea1SDimitry Andric 4732*0fca6ea1SDimitry Andric // dynSymTab is always present to simplify sym->includeInDynsym() in 4733*0fca6ea1SDimitry Andric // finalizeSections. 4734*0fca6ea1SDimitry Andric part.dynStrTab = std::make_unique<StringTableSection>(".dynstr", true); 4735*0fca6ea1SDimitry Andric part.dynSymTab = 4736*0fca6ea1SDimitry Andric std::make_unique<SymbolTableSection<ELFT>>(*part.dynStrTab); 4737*0fca6ea1SDimitry Andric 4738*0fca6ea1SDimitry Andric if (config->relocatable) 4739*0fca6ea1SDimitry Andric continue; 4740*0fca6ea1SDimitry Andric part.dynamic = std::make_unique<DynamicSection<ELFT>>(); 4741*0fca6ea1SDimitry Andric 4742*0fca6ea1SDimitry Andric if (hasMemtag()) { 4743*0fca6ea1SDimitry Andric part.memtagAndroidNote = std::make_unique<MemtagAndroidNote>(); 4744*0fca6ea1SDimitry Andric add(*part.memtagAndroidNote); 4745*0fca6ea1SDimitry Andric if (canHaveMemtagGlobals()) { 4746*0fca6ea1SDimitry Andric part.memtagGlobalDescriptors = 4747*0fca6ea1SDimitry Andric std::make_unique<MemtagGlobalDescriptors>(); 4748*0fca6ea1SDimitry Andric add(*part.memtagGlobalDescriptors); 4749*0fca6ea1SDimitry Andric } 4750*0fca6ea1SDimitry Andric } 4751*0fca6ea1SDimitry Andric 4752*0fca6ea1SDimitry Andric if (config->androidPackDynRelocs) 4753*0fca6ea1SDimitry Andric part.relaDyn = std::make_unique<AndroidPackedRelocationSection<ELFT>>( 4754*0fca6ea1SDimitry Andric relaDynName, threadCount); 4755*0fca6ea1SDimitry Andric else 4756*0fca6ea1SDimitry Andric part.relaDyn = std::make_unique<RelocationSection<ELFT>>( 4757*0fca6ea1SDimitry Andric relaDynName, config->zCombreloc, threadCount); 4758*0fca6ea1SDimitry Andric 4759*0fca6ea1SDimitry Andric if (config->hasDynSymTab) { 4760*0fca6ea1SDimitry Andric add(*part.dynSymTab); 4761*0fca6ea1SDimitry Andric 4762*0fca6ea1SDimitry Andric part.verSym = std::make_unique<VersionTableSection>(); 4763*0fca6ea1SDimitry Andric add(*part.verSym); 4764*0fca6ea1SDimitry Andric 4765*0fca6ea1SDimitry Andric if (!namedVersionDefs().empty()) { 4766*0fca6ea1SDimitry Andric part.verDef = std::make_unique<VersionDefinitionSection>(); 4767*0fca6ea1SDimitry Andric add(*part.verDef); 4768*0fca6ea1SDimitry Andric } 4769*0fca6ea1SDimitry Andric 4770*0fca6ea1SDimitry Andric part.verNeed = std::make_unique<VersionNeedSection<ELFT>>(); 4771*0fca6ea1SDimitry Andric add(*part.verNeed); 4772*0fca6ea1SDimitry Andric 4773*0fca6ea1SDimitry Andric if (config->gnuHash) { 4774*0fca6ea1SDimitry Andric part.gnuHashTab = std::make_unique<GnuHashTableSection>(); 4775*0fca6ea1SDimitry Andric add(*part.gnuHashTab); 4776*0fca6ea1SDimitry Andric } 4777*0fca6ea1SDimitry Andric 4778*0fca6ea1SDimitry Andric if (config->sysvHash) { 4779*0fca6ea1SDimitry Andric part.hashTab = std::make_unique<HashTableSection>(); 4780*0fca6ea1SDimitry Andric add(*part.hashTab); 4781*0fca6ea1SDimitry Andric } 4782*0fca6ea1SDimitry Andric 4783*0fca6ea1SDimitry Andric add(*part.dynamic); 4784*0fca6ea1SDimitry Andric add(*part.dynStrTab); 4785*0fca6ea1SDimitry Andric } 4786*0fca6ea1SDimitry Andric add(*part.relaDyn); 4787*0fca6ea1SDimitry Andric 4788*0fca6ea1SDimitry Andric if (config->relrPackDynRelocs) { 4789*0fca6ea1SDimitry Andric part.relrDyn = std::make_unique<RelrSection<ELFT>>(threadCount); 4790*0fca6ea1SDimitry Andric add(*part.relrDyn); 4791*0fca6ea1SDimitry Andric part.relrAuthDyn = std::make_unique<RelrSection<ELFT>>( 4792*0fca6ea1SDimitry Andric threadCount, /*isAArch64Auth=*/true); 4793*0fca6ea1SDimitry Andric add(*part.relrAuthDyn); 4794*0fca6ea1SDimitry Andric } 4795*0fca6ea1SDimitry Andric 4796*0fca6ea1SDimitry Andric if (config->ehFrameHdr) { 4797*0fca6ea1SDimitry Andric part.ehFrameHdr = std::make_unique<EhFrameHeader>(); 4798*0fca6ea1SDimitry Andric add(*part.ehFrameHdr); 4799*0fca6ea1SDimitry Andric } 4800*0fca6ea1SDimitry Andric part.ehFrame = std::make_unique<EhFrameSection>(); 4801*0fca6ea1SDimitry Andric add(*part.ehFrame); 4802*0fca6ea1SDimitry Andric 4803*0fca6ea1SDimitry Andric if (config->emachine == EM_ARM) { 4804*0fca6ea1SDimitry Andric // This section replaces all the individual .ARM.exidx InputSections. 4805*0fca6ea1SDimitry Andric part.armExidx = std::make_unique<ARMExidxSyntheticSection>(); 4806*0fca6ea1SDimitry Andric add(*part.armExidx); 4807*0fca6ea1SDimitry Andric } 4808*0fca6ea1SDimitry Andric 4809*0fca6ea1SDimitry Andric if (!config->packageMetadata.empty()) { 4810*0fca6ea1SDimitry Andric part.packageMetadataNote = std::make_unique<PackageMetadataNote>(); 4811*0fca6ea1SDimitry Andric add(*part.packageMetadataNote); 4812*0fca6ea1SDimitry Andric } 4813*0fca6ea1SDimitry Andric } 4814*0fca6ea1SDimitry Andric 4815*0fca6ea1SDimitry Andric if (partitions.size() != 1) { 4816*0fca6ea1SDimitry Andric // Create the partition end marker. This needs to be in partition number 255 4817*0fca6ea1SDimitry Andric // so that it is sorted after all other partitions. It also has other 4818*0fca6ea1SDimitry Andric // special handling (see createPhdrs() and combineEhSections()). 4819*0fca6ea1SDimitry Andric in.partEnd = 4820*0fca6ea1SDimitry Andric std::make_unique<BssSection>(".part.end", config->maxPageSize, 1); 4821*0fca6ea1SDimitry Andric in.partEnd->partition = 255; 4822*0fca6ea1SDimitry Andric add(*in.partEnd); 4823*0fca6ea1SDimitry Andric 4824*0fca6ea1SDimitry Andric in.partIndex = std::make_unique<PartitionIndexSection>(); 4825*0fca6ea1SDimitry Andric addOptionalRegular("__part_index_begin", in.partIndex.get(), 0); 4826*0fca6ea1SDimitry Andric addOptionalRegular("__part_index_end", in.partIndex.get(), 4827*0fca6ea1SDimitry Andric in.partIndex->getSize()); 4828*0fca6ea1SDimitry Andric add(*in.partIndex); 4829*0fca6ea1SDimitry Andric } 4830*0fca6ea1SDimitry Andric 4831*0fca6ea1SDimitry Andric // Add .got. MIPS' .got is so different from the other archs, 4832*0fca6ea1SDimitry Andric // it has its own class. 4833*0fca6ea1SDimitry Andric if (config->emachine == EM_MIPS) { 4834*0fca6ea1SDimitry Andric in.mipsGot = std::make_unique<MipsGotSection>(); 4835*0fca6ea1SDimitry Andric add(*in.mipsGot); 4836*0fca6ea1SDimitry Andric } else { 4837*0fca6ea1SDimitry Andric in.got = std::make_unique<GotSection>(); 4838*0fca6ea1SDimitry Andric add(*in.got); 4839*0fca6ea1SDimitry Andric } 4840*0fca6ea1SDimitry Andric 4841*0fca6ea1SDimitry Andric if (config->emachine == EM_PPC) { 4842*0fca6ea1SDimitry Andric in.ppc32Got2 = std::make_unique<PPC32Got2Section>(); 4843*0fca6ea1SDimitry Andric add(*in.ppc32Got2); 4844*0fca6ea1SDimitry Andric } 4845*0fca6ea1SDimitry Andric 4846*0fca6ea1SDimitry Andric if (config->emachine == EM_PPC64) { 4847*0fca6ea1SDimitry Andric in.ppc64LongBranchTarget = std::make_unique<PPC64LongBranchTargetSection>(); 4848*0fca6ea1SDimitry Andric add(*in.ppc64LongBranchTarget); 4849*0fca6ea1SDimitry Andric } 4850*0fca6ea1SDimitry Andric 4851*0fca6ea1SDimitry Andric in.gotPlt = std::make_unique<GotPltSection>(); 4852*0fca6ea1SDimitry Andric add(*in.gotPlt); 4853*0fca6ea1SDimitry Andric in.igotPlt = std::make_unique<IgotPltSection>(); 4854*0fca6ea1SDimitry Andric add(*in.igotPlt); 4855*0fca6ea1SDimitry Andric // Add .relro_padding if DATA_SEGMENT_RELRO_END is used; otherwise, add the 4856*0fca6ea1SDimitry Andric // section in the absence of PHDRS/SECTIONS commands. 4857*0fca6ea1SDimitry Andric if (config->zRelro && 4858*0fca6ea1SDimitry Andric ((script->phdrsCommands.empty() && !script->hasSectionsCommand) || 4859*0fca6ea1SDimitry Andric script->seenRelroEnd)) { 4860*0fca6ea1SDimitry Andric in.relroPadding = std::make_unique<RelroPaddingSection>(); 4861*0fca6ea1SDimitry Andric add(*in.relroPadding); 4862*0fca6ea1SDimitry Andric } 4863*0fca6ea1SDimitry Andric 4864*0fca6ea1SDimitry Andric if (config->emachine == EM_ARM) { 4865*0fca6ea1SDimitry Andric in.armCmseSGSection = std::make_unique<ArmCmseSGSection>(); 4866*0fca6ea1SDimitry Andric add(*in.armCmseSGSection); 4867*0fca6ea1SDimitry Andric } 4868*0fca6ea1SDimitry Andric 4869*0fca6ea1SDimitry Andric // _GLOBAL_OFFSET_TABLE_ is defined relative to either .got.plt or .got. Treat 4870*0fca6ea1SDimitry Andric // it as a relocation and ensure the referenced section is created. 4871*0fca6ea1SDimitry Andric if (ElfSym::globalOffsetTable && config->emachine != EM_MIPS) { 4872*0fca6ea1SDimitry Andric if (target->gotBaseSymInGotPlt) 4873*0fca6ea1SDimitry Andric in.gotPlt->hasGotPltOffRel = true; 4874*0fca6ea1SDimitry Andric else 4875*0fca6ea1SDimitry Andric in.got->hasGotOffRel = true; 4876*0fca6ea1SDimitry Andric } 4877*0fca6ea1SDimitry Andric 4878*0fca6ea1SDimitry Andric // We always need to add rel[a].plt to output if it has entries. 4879*0fca6ea1SDimitry Andric // Even for static linking it can contain R_[*]_IRELATIVE relocations. 4880*0fca6ea1SDimitry Andric in.relaPlt = std::make_unique<RelocationSection<ELFT>>( 4881*0fca6ea1SDimitry Andric config->isRela ? ".rela.plt" : ".rel.plt", /*sort=*/false, 4882*0fca6ea1SDimitry Andric /*threadCount=*/1); 4883*0fca6ea1SDimitry Andric add(*in.relaPlt); 4884*0fca6ea1SDimitry Andric 4885*0fca6ea1SDimitry Andric if ((config->emachine == EM_386 || config->emachine == EM_X86_64) && 4886*0fca6ea1SDimitry Andric (config->andFeatures & GNU_PROPERTY_X86_FEATURE_1_IBT)) { 4887*0fca6ea1SDimitry Andric in.ibtPlt = std::make_unique<IBTPltSection>(); 4888*0fca6ea1SDimitry Andric add(*in.ibtPlt); 4889*0fca6ea1SDimitry Andric } 4890*0fca6ea1SDimitry Andric 4891*0fca6ea1SDimitry Andric if (config->emachine == EM_PPC) 4892*0fca6ea1SDimitry Andric in.plt = std::make_unique<PPC32GlinkSection>(); 4893*0fca6ea1SDimitry Andric else 4894*0fca6ea1SDimitry Andric in.plt = std::make_unique<PltSection>(); 4895*0fca6ea1SDimitry Andric add(*in.plt); 4896*0fca6ea1SDimitry Andric in.iplt = std::make_unique<IpltSection>(); 4897*0fca6ea1SDimitry Andric add(*in.iplt); 4898*0fca6ea1SDimitry Andric 4899*0fca6ea1SDimitry Andric if (config->andFeatures || !ctx.aarch64PauthAbiCoreInfo.empty()) 4900*0fca6ea1SDimitry Andric add(*make<GnuPropertySection>()); 4901*0fca6ea1SDimitry Andric 4902*0fca6ea1SDimitry Andric if (config->debugNames) { 4903*0fca6ea1SDimitry Andric in.debugNames = std::make_unique<DebugNamesSection<ELFT>>(); 4904*0fca6ea1SDimitry Andric add(*in.debugNames); 4905*0fca6ea1SDimitry Andric } 4906*0fca6ea1SDimitry Andric 4907*0fca6ea1SDimitry Andric if (config->gdbIndex) { 4908*0fca6ea1SDimitry Andric in.gdbIndex = GdbIndexSection::create<ELFT>(); 4909*0fca6ea1SDimitry Andric add(*in.gdbIndex); 4910*0fca6ea1SDimitry Andric } 4911*0fca6ea1SDimitry Andric 4912*0fca6ea1SDimitry Andric // .note.GNU-stack is always added when we are creating a re-linkable 4913*0fca6ea1SDimitry Andric // object file. Other linkers are using the presence of this marker 4914*0fca6ea1SDimitry Andric // section to control the executable-ness of the stack area, but that 4915*0fca6ea1SDimitry Andric // is irrelevant these days. Stack area should always be non-executable 4916*0fca6ea1SDimitry Andric // by default. So we emit this section unconditionally. 4917*0fca6ea1SDimitry Andric if (config->relocatable) 4918*0fca6ea1SDimitry Andric add(*make<GnuStackSection>()); 4919*0fca6ea1SDimitry Andric 4920*0fca6ea1SDimitry Andric if (in.symTab) 4921*0fca6ea1SDimitry Andric add(*in.symTab); 4922*0fca6ea1SDimitry Andric if (in.symTabShndx) 4923*0fca6ea1SDimitry Andric add(*in.symTabShndx); 4924*0fca6ea1SDimitry Andric add(*in.shStrTab); 4925*0fca6ea1SDimitry Andric if (in.strTab) 4926*0fca6ea1SDimitry Andric add(*in.strTab); 4927*0fca6ea1SDimitry Andric } 4928*0fca6ea1SDimitry Andric 49295ffd83dbSDimitry Andric InStruct elf::in; 49300b57cec5SDimitry Andric 49315ffd83dbSDimitry Andric std::vector<Partition> elf::partitions; 49325ffd83dbSDimitry Andric Partition *elf::mainPart; 49330b57cec5SDimitry Andric 49345ffd83dbSDimitry Andric template void elf::splitSections<ELF32LE>(); 49355ffd83dbSDimitry Andric template void elf::splitSections<ELF32BE>(); 49365ffd83dbSDimitry Andric template void elf::splitSections<ELF64LE>(); 49375ffd83dbSDimitry Andric template void elf::splitSections<ELF64BE>(); 49380b57cec5SDimitry Andric 4939e8d8bef9SDimitry Andric template void EhFrameSection::iterateFDEWithLSDA<ELF32LE>( 4940e8d8bef9SDimitry Andric function_ref<void(InputSection &)>); 4941e8d8bef9SDimitry Andric template void EhFrameSection::iterateFDEWithLSDA<ELF32BE>( 4942e8d8bef9SDimitry Andric function_ref<void(InputSection &)>); 4943e8d8bef9SDimitry Andric template void EhFrameSection::iterateFDEWithLSDA<ELF64LE>( 4944e8d8bef9SDimitry Andric function_ref<void(InputSection &)>); 4945e8d8bef9SDimitry Andric template void EhFrameSection::iterateFDEWithLSDA<ELF64BE>( 4946e8d8bef9SDimitry Andric function_ref<void(InputSection &)>); 4947e8d8bef9SDimitry Andric 49485ffd83dbSDimitry Andric template class elf::SymbolTableSection<ELF32LE>; 49495ffd83dbSDimitry Andric template class elf::SymbolTableSection<ELF32BE>; 49505ffd83dbSDimitry Andric template class elf::SymbolTableSection<ELF64LE>; 49515ffd83dbSDimitry Andric template class elf::SymbolTableSection<ELF64BE>; 49520b57cec5SDimitry Andric 49535ffd83dbSDimitry Andric template void elf::writeEhdr<ELF32LE>(uint8_t *Buf, Partition &Part); 49545ffd83dbSDimitry Andric template void elf::writeEhdr<ELF32BE>(uint8_t *Buf, Partition &Part); 49555ffd83dbSDimitry Andric template void elf::writeEhdr<ELF64LE>(uint8_t *Buf, Partition &Part); 49565ffd83dbSDimitry Andric template void elf::writeEhdr<ELF64BE>(uint8_t *Buf, Partition &Part); 49570b57cec5SDimitry Andric 49585ffd83dbSDimitry Andric template void elf::writePhdrs<ELF32LE>(uint8_t *Buf, Partition &Part); 49595ffd83dbSDimitry Andric template void elf::writePhdrs<ELF32BE>(uint8_t *Buf, Partition &Part); 49605ffd83dbSDimitry Andric template void elf::writePhdrs<ELF64LE>(uint8_t *Buf, Partition &Part); 49615ffd83dbSDimitry Andric template void elf::writePhdrs<ELF64BE>(uint8_t *Buf, Partition &Part); 49620b57cec5SDimitry Andric 4963*0fca6ea1SDimitry Andric template void elf::createSyntheticSections<ELF32LE>(); 4964*0fca6ea1SDimitry Andric template void elf::createSyntheticSections<ELF32BE>(); 4965*0fca6ea1SDimitry Andric template void elf::createSyntheticSections<ELF64LE>(); 4966*0fca6ea1SDimitry Andric template void elf::createSyntheticSections<ELF64BE>(); 4967