Index: lib/ObjectYAML/ELFEmitter.cpp =================================================================== --- lib/ObjectYAML/ELFEmitter.cpp +++ lib/ObjectYAML/ELFEmitter.cpp @@ -118,11 +118,9 @@ bool buildSymbolIndexes(); void initELFHeader(Elf_Ehdr &Header); void initProgramHeaders(std::vector &PHeaders); - bool initImplicitHeader(ELFState &State, ContiguousBlobAccumulator &CBA, - Elf_Shdr &Header, StringRef SecName, - ELFYAML::Section *YAMLSec); - bool initSectionHeaders(ELFState &State, - std::vector &SHeaders, + bool initImplicitHeader(ContiguousBlobAccumulator &CBA, Elf_Shdr &Header, + StringRef SecName, ELFYAML::Section *YAMLSec); + bool initSectionHeaders(std::vector &SHeaders, ContiguousBlobAccumulator &CBA); void initSymtabSectionHeader(Elf_Shdr &SHeader, SymtabType STType, ContiguousBlobAccumulator &CBA, @@ -133,6 +131,7 @@ ELFYAML::Section *YAMLSec); void setProgramHeaderLayout(std::vector &PHeaders, std::vector &SHeaders); + void finalizeStrings(); bool writeSectionContent(Elf_Shdr &SHeader, const ELFYAML::RawContentSection &Section, ContiguousBlobAccumulator &CBA); @@ -163,9 +162,6 @@ public: static int writeELF(raw_ostream &OS, ELFYAML::Object &Doc); - -private: - void finalizeStrings(); }; } // end anonymous namespace @@ -267,8 +263,7 @@ } template -bool ELFState::initImplicitHeader(ELFState &State, - ContiguousBlobAccumulator &CBA, +bool ELFState::initImplicitHeader(ContiguousBlobAccumulator &CBA, Elf_Shdr &Header, StringRef SecName, ELFYAML::Section *YAMLSec) { // Check if the header was already initialized. @@ -276,19 +271,15 @@ return false; if (SecName == ".symtab") - State.initSymtabSectionHeader(Header, SymtabType::Static, CBA, YAMLSec); + initSymtabSectionHeader(Header, SymtabType::Static, CBA, YAMLSec); else if (SecName == ".strtab") - State.initStrtabSectionHeader(Header, SecName, State.DotStrtab, CBA, - YAMLSec); + initStrtabSectionHeader(Header, SecName, DotStrtab, CBA, YAMLSec); else if (SecName == ".shstrtab") - State.initStrtabSectionHeader(Header, SecName, State.DotShStrtab, CBA, - YAMLSec); - + initStrtabSectionHeader(Header, SecName, DotShStrtab, CBA, YAMLSec); else if (SecName == ".dynsym") - State.initSymtabSectionHeader(Header, SymtabType::Dynamic, CBA, YAMLSec); + initSymtabSectionHeader(Header, SymtabType::Dynamic, CBA, YAMLSec); else if (SecName == ".dynstr") - State.initStrtabSectionHeader(Header, SecName, State.DotDynstr, CBA, - YAMLSec); + initStrtabSectionHeader(Header, SecName, DotDynstr, CBA, YAMLSec); else return false; @@ -313,8 +304,7 @@ } template -bool ELFState::initSectionHeaders(ELFState &State, - std::vector &SHeaders, +bool ELFState::initSectionHeaders(std::vector &SHeaders, ContiguousBlobAccumulator &CBA) { // Ensure SHN_UNDEF entry is present. An all-zero section header is a // valid SHN_UNDEF entry since SHT_NULL == 0. @@ -330,7 +320,7 @@ // in the YAML, we need to write them here. This ensures the file offset // remains correct. Elf_Shdr &SHeader = SHeaders[I]; - if (initImplicitHeader(State, CBA, SHeader, Sec->Name, + if (initImplicitHeader(CBA, SHeader, Sec->Name, Sec->IsImplicit ? nullptr : Sec)) continue; @@ -1066,7 +1056,7 @@ ContiguousBlobAccumulator CBA(SectionContentBeginOffset); std::vector SHeaders; - if (!State.initSectionHeaders(State, SHeaders, CBA)) + if (!State.initSectionHeaders(SHeaders, CBA)) return 1; // Now we can decide segment offsets