From 595a4486c103fb5761e1f1354c9b8c7958781f77 Mon Sep 17 00:00:00 2001 From: Dave Lee Date: Tue, 7 Nov 2017 22:05:24 +0000 Subject: Allow yaml2obj to order implicit sections for ELF Summary: This change allows yaml input to control the order of implicitly added sections (`.symtab`, `.strtab`, `.shstrtab`). The order is controlled by adding a placeholder section of the given name to the Sections field. This change is to support changes in D39582, where it is desirable to control the location of the `.dynsym` section. Reviewers: compnerd, jakehehrlich Reviewed By: jakehehrlich Subscribers: llvm-commits Differential Revision: https://reviews.llvm.org/D39749 git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@317622 91177308-0d34-0410-b5e6-96231b3b80d8 --- tools/yaml2obj/yaml2elf.cpp | 78 ++++++++++++++++++++++++--------------------- 1 file changed, 41 insertions(+), 37 deletions(-) (limited to 'tools/yaml2obj') diff --git a/tools/yaml2obj/yaml2elf.cpp b/tools/yaml2obj/yaml2elf.cpp index 913f19050dc..1792b128882 100644 --- a/tools/yaml2obj/yaml2elf.cpp +++ b/tools/yaml2obj/yaml2elf.cpp @@ -74,6 +74,13 @@ public: Idx = I->getValue(); return false; } + /// asserts if name is not present in the map + unsigned get(StringRef Name) const { + unsigned Idx; + assert(!lookup(Name, Idx) && "Expected section not found in index"); + return Idx; + } + unsigned size() const { return Map.size(); } }; } // end anonymous namespace @@ -144,19 +151,21 @@ class ELFState { ContiguousBlobAccumulator &CBA); // - SHT_NULL entry (placed first, i.e. 0'th entry) - // - symbol table (.symtab) (placed third to last) - // - string table (.strtab) (placed second to last) - // - section header string table (.shstrtab) (placed last) - unsigned getDotSymTabSecNo() const { return Doc.Sections.size() + 1; } - unsigned getDotStrTabSecNo() const { return Doc.Sections.size() + 2; } - unsigned getDotShStrTabSecNo() const { return Doc.Sections.size() + 3; } - unsigned getSectionCount() const { return Doc.Sections.size() + 4; } + // - symbol table (.symtab) (defaults to third to last) + // - string table (.strtab) (defaults to second to last) + // - section header string table (.shstrtab) (defaults to last) + unsigned getDotSymTabSecNo() const { return SN2I.get(".symtab"); } + unsigned getDotStrTabSecNo() const { return SN2I.get(".strtab"); } + unsigned getDotShStrTabSecNo() const { return SN2I.get(".shstrtab"); } + unsigned getSectionCount() const { return SN2I.size() + 1; } ELFState(const ELFYAML::Object &D) : Doc(D) {} public: static int writeELF(raw_ostream &OS, const ELFYAML::Object &Doc); }; + +static const char * const ImplicitSecNames[] = {".symtab", ".strtab", ".shstrtab"}; } // end anonymous namespace template @@ -211,10 +220,6 @@ bool ELFState::initSectionHeaders(std::vector &SHeaders, zero(SHeader); SHeaders.push_back(SHeader); - for (const auto &Sec : Doc.Sections) - DotShStrtab.add(Sec->Name); - DotShStrtab.finalize(); - for (const auto &Sec : Doc.Sections) { zero(SHeader); SHeader.sh_name = DotShStrtab.getOffset(Sec->Name); @@ -547,10 +552,6 @@ bool ELFState::writeSectionContent(Elf_Shdr &SHeader, } template bool ELFState::buildSectionIndex() { - SN2I.addName(".symtab", getDotSymTabSecNo()); - SN2I.addName(".strtab", getDotStrTabSecNo()); - SN2I.addName(".shstrtab", getDotShStrTabSecNo()); - for (unsigned i = 0, e = Doc.Sections.size(); i != e; ++i) { StringRef Name = Doc.Sections[i]->Name; if (Name.empty()) @@ -561,7 +562,19 @@ template bool ELFState::buildSectionIndex() { << "' at YAML section number " << i << ".\n"; return false; } + DotShStrtab.add(Name); } + + auto SecNo = 1 + Doc.Sections.size(); + // Add special sections after input sections, if necessary. + for (const auto &Name : ImplicitSecNames) + if (!SN2I.addName(Name, SecNo)) { + // Account for this section, since it wasn't in the Doc + ++SecNo; + DotShStrtab.add(Name); + } + + DotShStrtab.finalize(); return true; } @@ -608,32 +621,23 @@ int ELFState::writeELF(raw_ostream &OS, const ELFYAML::Object &Doc) { Header.e_shentsize * Header.e_shnum; ContiguousBlobAccumulator CBA(SectionContentBeginOffset); - // Doc might not contain .symtab, .strtab and .shstrtab sections, - // but we will emit them, so make sure to add them to ShStrTabSHeader. - State.DotShStrtab.add(".symtab"); - State.DotShStrtab.add(".strtab"); - State.DotShStrtab.add(".shstrtab"); - std::vector SHeaders; + SHeaders.reserve(State.SN2I.size()); if(!State.initSectionHeaders(SHeaders, CBA)) return 1; - // .symtab section. - Elf_Shdr SymtabSHeader; - State.initSymtabSectionHeader(SymtabSHeader, CBA); - SHeaders.push_back(SymtabSHeader); - - // .strtab string table header. - Elf_Shdr DotStrTabSHeader; - State.initStrtabSectionHeader(DotStrTabSHeader, ".strtab", State.DotStrtab, - CBA); - SHeaders.push_back(DotStrTabSHeader); - - // .shstrtab string table header. - Elf_Shdr ShStrTabSHeader; - State.initStrtabSectionHeader(ShStrTabSHeader, ".shstrtab", State.DotShStrtab, - CBA); - SHeaders.push_back(ShStrTabSHeader); + // Populate SHeaders with implicit sections not present in the Doc + for (const auto &Name : ImplicitSecNames) + if (State.SN2I.get(Name) >= SHeaders.size()) + SHeaders.push_back({}); + + // Initialize the implicit sections + auto Index = State.SN2I.get(".symtab"); + State.initSymtabSectionHeader(SHeaders[Index], CBA); + Index = State.SN2I.get(".strtab"); + State.initStrtabSectionHeader(SHeaders[Index], ".strtab", State.DotStrtab, CBA); + Index = State.SN2I.get(".shstrtab"); + State.initStrtabSectionHeader(SHeaders[Index], ".shstrtab", State.DotShStrtab, CBA); // Now we can decide segment offsets State.setProgramHeaderLayout(PHeaders, SHeaders); -- cgit v1.2.3