15918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silva//===- yaml2elf - Convert YAML to a ELF object file -----------------------===// 25918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silva// 35918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silva// The LLVM Compiler Infrastructure 45918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silva// 55918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silva// This file is distributed under the University of Illinois Open Source 65918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silva// License. See LICENSE.TXT for details. 75918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silva// 85918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silva//===----------------------------------------------------------------------===// 95918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silva/// 105918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silva/// \file 115918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silva/// \brief The ELF component of yaml2obj. 125918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silva/// 135918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silva//===----------------------------------------------------------------------===// 145918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silva 155918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silva#include "yaml2obj.h" 166dd7893e8c7234d65cdc47e3f3ce251fdfb35768Will Dietz#include "llvm/ADT/ArrayRef.h" 17cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines#include "llvm/MC/StringTableBuilder.h" 18081a1941b595f6294e4ce678fd61ef56a2ceb51eMichael J. Spencer#include "llvm/Object/ELFObjectFile.h" 195918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silva#include "llvm/Object/ELFYAML.h" 205918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silva#include "llvm/Support/ELF.h" 215918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silva#include "llvm/Support/MemoryBuffer.h" 225918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silva#include "llvm/Support/YAMLTraits.h" 235918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silva#include "llvm/Support/raw_ostream.h" 245918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silva 255918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silvausing namespace llvm; 265918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silva 272a7e79a30f87be4103963e6049a2f1cf171f502cSean Silva// This class is used to build up a contiguous binary blob while keeping 282a7e79a30f87be4103963e6049a2f1cf171f502cSean Silva// track of an offset in the output (which notionally begins at 292a7e79a30f87be4103963e6049a2f1cf171f502cSean Silva// `InitialOffset`). 30f3f3523b9c714d9b98f6102ce79ed3d0c9050169Sean Silvanamespace { 312a7e79a30f87be4103963e6049a2f1cf171f502cSean Silvaclass ContiguousBlobAccumulator { 322a7e79a30f87be4103963e6049a2f1cf171f502cSean Silva const uint64_t InitialOffset; 33b14f972ed6865a978402fb1c788ee7939b58680dSean Silva SmallVector<char, 128> Buf; 342a7e79a30f87be4103963e6049a2f1cf171f502cSean Silva raw_svector_ostream OS; 352a7e79a30f87be4103963e6049a2f1cf171f502cSean Silva 36e766884193afeb684fe1e14f74c316405e9d801fSean Silva /// \returns The new offset. 37e766884193afeb684fe1e14f74c316405e9d801fSean Silva uint64_t padToAlignment(unsigned Align) { 38e766884193afeb684fe1e14f74c316405e9d801fSean Silva uint64_t CurrentOffset = InitialOffset + OS.tell(); 39e766884193afeb684fe1e14f74c316405e9d801fSean Silva uint64_t AlignedOffset = RoundUpToAlignment(CurrentOffset, Align); 40e766884193afeb684fe1e14f74c316405e9d801fSean Silva for (; CurrentOffset != AlignedOffset; ++CurrentOffset) 41e766884193afeb684fe1e14f74c316405e9d801fSean Silva OS.write('\0'); 42e766884193afeb684fe1e14f74c316405e9d801fSean Silva return AlignedOffset; // == CurrentOffset; 43e766884193afeb684fe1e14f74c316405e9d801fSean Silva } 44e766884193afeb684fe1e14f74c316405e9d801fSean Silva 452a7e79a30f87be4103963e6049a2f1cf171f502cSean Silvapublic: 46b14f972ed6865a978402fb1c788ee7939b58680dSean Silva ContiguousBlobAccumulator(uint64_t InitialOffset_) 47b14f972ed6865a978402fb1c788ee7939b58680dSean Silva : InitialOffset(InitialOffset_), Buf(), OS(Buf) {} 48e766884193afeb684fe1e14f74c316405e9d801fSean Silva template <class Integer> 49e766884193afeb684fe1e14f74c316405e9d801fSean Silva raw_ostream &getOSAndAlignedOffset(Integer &Offset, unsigned Align = 16) { 50e766884193afeb684fe1e14f74c316405e9d801fSean Silva Offset = padToAlignment(Align); 51e766884193afeb684fe1e14f74c316405e9d801fSean Silva return OS; 52e766884193afeb684fe1e14f74c316405e9d801fSean Silva } 532a7e79a30f87be4103963e6049a2f1cf171f502cSean Silva void writeBlobToStream(raw_ostream &Out) { Out << OS.str(); } 542a7e79a30f87be4103963e6049a2f1cf171f502cSean Silva}; 55f3f3523b9c714d9b98f6102ce79ed3d0c9050169Sean Silva} // end anonymous namespace 562a7e79a30f87be4103963e6049a2f1cf171f502cSean Silva 57dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines// Used to keep track of section and symbol names, so that in the YAML file 58dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines// sections and symbols can be referenced by name instead of by index. 59f3f3523b9c714d9b98f6102ce79ed3d0c9050169Sean Silvanamespace { 60dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hinesclass NameToIdxMap { 61fe57e347a57d643bbbcc9c19c8267a3e8c06b5a6Sean Silva StringMap<int> Map; 62fe57e347a57d643bbbcc9c19c8267a3e8c06b5a6Sean Silvapublic: 63fe57e347a57d643bbbcc9c19c8267a3e8c06b5a6Sean Silva /// \returns true if name is already present in the map. 64dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines bool addName(StringRef Name, unsigned i) { 65dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines StringMapEntry<int> &Entry = Map.GetOrCreateValue(Name, -1); 66fe57e347a57d643bbbcc9c19c8267a3e8c06b5a6Sean Silva if (Entry.getValue() != -1) 67fe57e347a57d643bbbcc9c19c8267a3e8c06b5a6Sean Silva return true; 68fe57e347a57d643bbbcc9c19c8267a3e8c06b5a6Sean Silva Entry.setValue((int)i); 69fe57e347a57d643bbbcc9c19c8267a3e8c06b5a6Sean Silva return false; 70fe57e347a57d643bbbcc9c19c8267a3e8c06b5a6Sean Silva } 71fe57e347a57d643bbbcc9c19c8267a3e8c06b5a6Sean Silva /// \returns true if name is not present in the map 72dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines bool lookup(StringRef Name, unsigned &Idx) const { 73dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines StringMap<int>::const_iterator I = Map.find(Name); 74fe57e347a57d643bbbcc9c19c8267a3e8c06b5a6Sean Silva if (I == Map.end()) 75fe57e347a57d643bbbcc9c19c8267a3e8c06b5a6Sean Silva return true; 76fe57e347a57d643bbbcc9c19c8267a3e8c06b5a6Sean Silva Idx = I->getValue(); 77fe57e347a57d643bbbcc9c19c8267a3e8c06b5a6Sean Silva return false; 78fe57e347a57d643bbbcc9c19c8267a3e8c06b5a6Sean Silva } 79fe57e347a57d643bbbcc9c19c8267a3e8c06b5a6Sean Silva}; 80f3f3523b9c714d9b98f6102ce79ed3d0c9050169Sean Silva} // end anonymous namespace 81fe57e347a57d643bbbcc9c19c8267a3e8c06b5a6Sean Silva 82274264ce2ba93d26702f7ebfe6aea121df44113fSean Silvatemplate <class T> 836dd7893e8c7234d65cdc47e3f3ce251fdfb35768Will Dietzstatic size_t arrayDataSize(ArrayRef<T> A) { 846dd7893e8c7234d65cdc47e3f3ce251fdfb35768Will Dietz return A.size() * sizeof(T); 85274264ce2ba93d26702f7ebfe6aea121df44113fSean Silva} 86274264ce2ba93d26702f7ebfe6aea121df44113fSean Silva 87274264ce2ba93d26702f7ebfe6aea121df44113fSean Silvatemplate <class T> 886dd7893e8c7234d65cdc47e3f3ce251fdfb35768Will Dietzstatic void writeArrayData(raw_ostream &OS, ArrayRef<T> A) { 896dd7893e8c7234d65cdc47e3f3ce251fdfb35768Will Dietz OS.write((const char *)A.data(), arrayDataSize(A)); 90274264ce2ba93d26702f7ebfe6aea121df44113fSean Silva} 91274264ce2ba93d26702f7ebfe6aea121df44113fSean Silva 92274264ce2ba93d26702f7ebfe6aea121df44113fSean Silvatemplate <class T> 93274264ce2ba93d26702f7ebfe6aea121df44113fSean Silvastatic void zero(T &Obj) { 94274264ce2ba93d26702f7ebfe6aea121df44113fSean Silva memset(&Obj, 0, sizeof(Obj)); 95274264ce2ba93d26702f7ebfe6aea121df44113fSean Silva} 96274264ce2ba93d26702f7ebfe6aea121df44113fSean Silva 970382b30eb5ecdde9749841a2f39960ab8e2de87cSean Silvanamespace { 980382b30eb5ecdde9749841a2f39960ab8e2de87cSean Silva/// \brief "Single point of truth" for the ELF file construction. 990382b30eb5ecdde9749841a2f39960ab8e2de87cSean Silva/// TODO: This class still has a ways to go before it is truly a "single 1000382b30eb5ecdde9749841a2f39960ab8e2de87cSean Silva/// point of truth". 1010382b30eb5ecdde9749841a2f39960ab8e2de87cSean Silvatemplate <class ELFT> 1020382b30eb5ecdde9749841a2f39960ab8e2de87cSean Silvaclass ELFState { 10336b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines typedef typename object::ELFFile<ELFT>::Elf_Ehdr Elf_Ehdr; 10436b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines typedef typename object::ELFFile<ELFT>::Elf_Shdr Elf_Shdr; 10536b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines typedef typename object::ELFFile<ELFT>::Elf_Sym Elf_Sym; 106dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines typedef typename object::ELFFile<ELFT>::Elf_Rel Elf_Rel; 107dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines typedef typename object::ELFFile<ELFT>::Elf_Rela Elf_Rela; 10836b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines 1090382b30eb5ecdde9749841a2f39960ab8e2de87cSean Silva /// \brief The future ".strtab" section. 1100382b30eb5ecdde9749841a2f39960ab8e2de87cSean Silva StringTableBuilder DotStrtab; 1110382b30eb5ecdde9749841a2f39960ab8e2de87cSean Silva 11236b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines /// \brief The future ".shstrtab" section. 11336b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines StringTableBuilder DotShStrtab; 114c18f66e70dea8abc8c9d86d40088fc8fc8247b20Sean Silva 115dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines NameToIdxMap SN2I; 116dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines NameToIdxMap SymN2I; 11736b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines const ELFYAML::Object &Doc; 11836b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines 11936b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines bool buildSectionIndex(); 120dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines bool buildSymbolIndex(std::size_t &StartIndex, 121dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines const std::vector<ELFYAML::Symbol> &Symbols); 12236b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines void initELFHeader(Elf_Ehdr &Header); 12336b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines bool initSectionHeaders(std::vector<Elf_Shdr> &SHeaders, 12436b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines ContiguousBlobAccumulator &CBA); 12536b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines void initSymtabSectionHeader(Elf_Shdr &SHeader, 12636b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines ContiguousBlobAccumulator &CBA); 12736b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines void initStrtabSectionHeader(Elf_Shdr &SHeader, StringRef Name, 12836b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines StringTableBuilder &STB, 12936b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines ContiguousBlobAccumulator &CBA); 13036b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines void addSymbols(const std::vector<ELFYAML::Symbol> &Symbols, 13136b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines std::vector<Elf_Sym> &Syms, unsigned SymbolBinding); 132dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines void writeSectionContent(Elf_Shdr &SHeader, 133dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines const ELFYAML::RawContentSection &Section, 134dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines ContiguousBlobAccumulator &CBA); 135dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines bool writeSectionContent(Elf_Shdr &SHeader, 136dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines const ELFYAML::RelocationSection &Section, 137dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines ContiguousBlobAccumulator &CBA); 13836b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines 13936b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines // - SHT_NULL entry (placed first, i.e. 0'th entry) 14036b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines // - symbol table (.symtab) (placed third to last) 14136b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines // - string table (.strtab) (placed second to last) 14236b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines // - section header string table (.shstrtab) (placed last) 14336b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines unsigned getDotSymTabSecNo() const { return Doc.Sections.size() + 1; } 14436b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines unsigned getDotStrTabSecNo() const { return Doc.Sections.size() + 2; } 14536b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines unsigned getDotShStrTabSecNo() const { return Doc.Sections.size() + 3; } 14636b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines unsigned getSectionCount() const { return Doc.Sections.size() + 4; } 1470382b30eb5ecdde9749841a2f39960ab8e2de87cSean Silva 14836b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines ELFState(const ELFYAML::Object &D) : Doc(D) {} 1490382b30eb5ecdde9749841a2f39960ab8e2de87cSean Silva 15036b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hinespublic: 15136b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines static int writeELF(raw_ostream &OS, const ELFYAML::Object &Doc); 1520382b30eb5ecdde9749841a2f39960ab8e2de87cSean Silva}; 1530382b30eb5ecdde9749841a2f39960ab8e2de87cSean Silva} // end anonymous namespace 1540382b30eb5ecdde9749841a2f39960ab8e2de87cSean Silva 155552d7cd31a0f6eb0e1434981196f477ae741a184Sean Silvatemplate <class ELFT> 15636b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hinesvoid ELFState<ELFT>::initELFHeader(Elf_Ehdr &Header) { 15736b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines using namespace llvm::ELF; 15836b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines zero(Header); 15936b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines Header.e_ident[EI_MAG0] = 0x7f; 16036b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines Header.e_ident[EI_MAG1] = 'E'; 16136b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines Header.e_ident[EI_MAG2] = 'L'; 16236b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines Header.e_ident[EI_MAG3] = 'F'; 16336b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines Header.e_ident[EI_CLASS] = ELFT::Is64Bits ? ELFCLASS64 : ELFCLASS32; 16436b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines bool IsLittleEndian = ELFT::TargetEndianness == support::little; 16536b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines Header.e_ident[EI_DATA] = IsLittleEndian ? ELFDATA2LSB : ELFDATA2MSB; 16636b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines Header.e_ident[EI_VERSION] = EV_CURRENT; 16736b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines Header.e_ident[EI_OSABI] = Doc.Header.OSABI; 16836b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines Header.e_ident[EI_ABIVERSION] = 0; 16936b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines Header.e_type = Doc.Header.Type; 17036b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines Header.e_machine = Doc.Header.Machine; 17136b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines Header.e_version = EV_CURRENT; 17236b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines Header.e_entry = Doc.Header.Entry; 17336b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines Header.e_flags = Doc.Header.Flags; 17436b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines Header.e_ehsize = sizeof(Elf_Ehdr); 17536b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines Header.e_shentsize = sizeof(Elf_Shdr); 17636b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines // Immediately following the ELF header. 17736b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines Header.e_shoff = sizeof(Header); 17836b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines Header.e_shnum = getSectionCount(); 17936b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines Header.e_shstrndx = getDotShStrTabSecNo(); 18036b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines} 18136b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines 18236b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hinestemplate <class ELFT> 18336b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hinesbool ELFState<ELFT>::initSectionHeaders(std::vector<Elf_Shdr> &SHeaders, 18436b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines ContiguousBlobAccumulator &CBA) { 18536b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines // Ensure SHN_UNDEF entry is present. An all-zero section header is a 18636b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines // valid SHN_UNDEF entry since SHT_NULL == 0. 18736b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines Elf_Shdr SHeader; 18836b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines zero(SHeader); 18936b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines SHeaders.push_back(SHeader); 19036b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines 191dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines for (const auto &Sec : Doc.Sections) 192dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines DotShStrtab.add(Sec->Name); 193dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines DotShStrtab.finalize(); 194dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 19536b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines for (const auto &Sec : Doc.Sections) { 19636b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines zero(SHeader); 197dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines SHeader.sh_name = DotShStrtab.getOffset(Sec->Name); 198dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines SHeader.sh_type = Sec->Type; 199dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines SHeader.sh_flags = Sec->Flags; 200dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines SHeader.sh_addr = Sec->Address; 201dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines SHeader.sh_addralign = Sec->AddressAlign; 20236b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines 203dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (!Sec->Link.empty()) { 204326c193e0aa221c7836adfa16a3ec85857d7d8b7Sean Silva unsigned Index; 205dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (SN2I.lookup(Sec->Link, Index)) { 206dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines errs() << "error: Unknown section referenced: '" << Sec->Link 207dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines << "' at YAML section '" << Sec->Name << "'.\n"; 208dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return false; 209326c193e0aa221c7836adfa16a3ec85857d7d8b7Sean Silva } 21036b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines SHeader.sh_link = Index; 21136b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines } 212dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 213dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (auto S = dyn_cast<ELFYAML::RawContentSection>(Sec.get())) 214dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines writeSectionContent(SHeader, *S, CBA); 215dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines else if (auto S = dyn_cast<ELFYAML::RelocationSection>(Sec.get())) { 216dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (S->Link.empty()) 217dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines // For relocation section set link to .symtab by default. 218dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines SHeader.sh_link = getDotSymTabSecNo(); 219dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 220dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines unsigned Index; 221dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (SN2I.lookup(S->Info, Index)) { 222dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines errs() << "error: Unknown section referenced: '" << S->Info 223dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines << "' at YAML section '" << S->Name << "'.\n"; 224dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return false; 225dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines } 226dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines SHeader.sh_info = Index; 227dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 228dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (!writeSectionContent(SHeader, *S, CBA)) 229dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return false; 230dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines } else 231dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines llvm_unreachable("Unknown section type"); 232dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 23336b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines SHeaders.push_back(SHeader); 2344235ba32f2610ee2ed1e0c4bfca5c67835e9f97dSean Silva } 23536b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines return true; 2364235ba32f2610ee2ed1e0c4bfca5c67835e9f97dSean Silva} 2374235ba32f2610ee2ed1e0c4bfca5c67835e9f97dSean Silva 238afcf60fe15999ea07193118f447a34f41171e433Sean Silvatemplate <class ELFT> 23936b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hinesvoid ELFState<ELFT>::initSymtabSectionHeader(Elf_Shdr &SHeader, 24036b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines ContiguousBlobAccumulator &CBA) { 24136b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines zero(SHeader); 242dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines SHeader.sh_name = DotShStrtab.getOffset(".symtab"); 243068463b714bd53fffd41322a5654b4f359c2166cSean Silva SHeader.sh_type = ELF::SHT_SYMTAB; 24436b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines SHeader.sh_link = getDotStrTabSecNo(); 2454235ba32f2610ee2ed1e0c4bfca5c67835e9f97dSean Silva // One greater than symbol table index of the last local symbol. 24636b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines SHeader.sh_info = Doc.Symbols.Local.size() + 1; 247afcf60fe15999ea07193118f447a34f41171e433Sean Silva SHeader.sh_entsize = sizeof(Elf_Sym); 248afcf60fe15999ea07193118f447a34f41171e433Sean Silva 249afcf60fe15999ea07193118f447a34f41171e433Sean Silva std::vector<Elf_Sym> Syms; 250ca0170ffa1ba6d53f0008b345cb5ac777e81c3b1Sean Silva { 251ca0170ffa1ba6d53f0008b345cb5ac777e81c3b1Sean Silva // Ensure STN_UNDEF is present 252ca0170ffa1ba6d53f0008b345cb5ac777e81c3b1Sean Silva Elf_Sym Sym; 253ca0170ffa1ba6d53f0008b345cb5ac777e81c3b1Sean Silva zero(Sym); 254ca0170ffa1ba6d53f0008b345cb5ac777e81c3b1Sean Silva Syms.push_back(Sym); 255ca0170ffa1ba6d53f0008b345cb5ac777e81c3b1Sean Silva } 256dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 257dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines // Add symbol names to .strtab. 258dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines for (const auto &Sym : Doc.Symbols.Local) 259dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines DotStrtab.add(Sym.Name); 260dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines for (const auto &Sym : Doc.Symbols.Global) 261dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines DotStrtab.add(Sym.Name); 262dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines for (const auto &Sym : Doc.Symbols.Weak) 263dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines DotStrtab.add(Sym.Name); 264dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines DotStrtab.finalize(); 265dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 26636b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines addSymbols(Doc.Symbols.Local, Syms, ELF::STB_LOCAL); 26736b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines addSymbols(Doc.Symbols.Global, Syms, ELF::STB_GLOBAL); 26836b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines addSymbols(Doc.Symbols.Weak, Syms, ELF::STB_WEAK); 269afcf60fe15999ea07193118f447a34f41171e433Sean Silva 2706dd7893e8c7234d65cdc47e3f3ce251fdfb35768Will Dietz writeArrayData(CBA.getOSAndAlignedOffset(SHeader.sh_offset), 2716dd7893e8c7234d65cdc47e3f3ce251fdfb35768Will Dietz makeArrayRef(Syms)); 2726dd7893e8c7234d65cdc47e3f3ce251fdfb35768Will Dietz SHeader.sh_size = arrayDataSize(makeArrayRef(Syms)); 273afcf60fe15999ea07193118f447a34f41171e433Sean Silva} 274afcf60fe15999ea07193118f447a34f41171e433Sean Silva 2755918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silvatemplate <class ELFT> 27636b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hinesvoid ELFState<ELFT>::initStrtabSectionHeader(Elf_Shdr &SHeader, StringRef Name, 27736b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines StringTableBuilder &STB, 27836b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines ContiguousBlobAccumulator &CBA) { 27936b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines zero(SHeader); 280dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines SHeader.sh_name = DotShStrtab.getOffset(Name); 28136b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines SHeader.sh_type = ELF::SHT_STRTAB; 282dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines CBA.getOSAndAlignedOffset(SHeader.sh_offset) << STB.data(); 283dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines SHeader.sh_size = STB.data().size(); 28436b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines SHeader.sh_addralign = 1; 28536b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines} 286274264ce2ba93d26702f7ebfe6aea121df44113fSean Silva 28736b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hinestemplate <class ELFT> 28836b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hinesvoid ELFState<ELFT>::addSymbols(const std::vector<ELFYAML::Symbol> &Symbols, 28936b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines std::vector<Elf_Sym> &Syms, 29036b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines unsigned SymbolBinding) { 29136b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines for (const auto &Sym : Symbols) { 29236b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines Elf_Sym Symbol; 29336b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines zero(Symbol); 29436b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines if (!Sym.Name.empty()) 295dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Symbol.st_name = DotStrtab.getOffset(Sym.Name); 29636b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines Symbol.setBindingAndType(SymbolBinding, Sym.Type); 29736b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines if (!Sym.Section.empty()) { 29836b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines unsigned Index; 299dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (SN2I.lookup(Sym.Section, Index)) { 30036b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines errs() << "error: Unknown section referenced: '" << Sym.Section 30136b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines << "' by YAML symbol " << Sym.Name << ".\n"; 30236b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines exit(1); 30336b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines } 30436b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines Symbol.st_shndx = Index; 30536b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines } // else Symbol.st_shndex == SHN_UNDEF (== 0), since it was zero'd earlier. 30636b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines Symbol.st_value = Sym.Value; 307cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hines Symbol.st_other = Sym.Visibility; 30836b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines Symbol.st_size = Sym.Size; 30936b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines Syms.push_back(Symbol); 31036b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines } 31136b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines} 312274264ce2ba93d26702f7ebfe6aea121df44113fSean Silva 313dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hinestemplate <class ELFT> 314dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hinesvoid 315dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen HinesELFState<ELFT>::writeSectionContent(Elf_Shdr &SHeader, 316dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines const ELFYAML::RawContentSection &Section, 317dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines ContiguousBlobAccumulator &CBA) { 318dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines assert(Section.Size >= Section.Content.binary_size() && 319dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines "Section size and section content are inconsistent"); 320dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines raw_ostream &OS = CBA.getOSAndAlignedOffset(SHeader.sh_offset); 321dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Section.Content.writeAsBinary(OS); 322dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines for (auto i = Section.Content.binary_size(); i < Section.Size; ++i) 323dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines OS.write(0); 324dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines SHeader.sh_entsize = 0; 325dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines SHeader.sh_size = Section.Size; 326dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines} 327dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 328dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hinestemplate <class ELFT> 329dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hinesbool 330dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen HinesELFState<ELFT>::writeSectionContent(Elf_Shdr &SHeader, 331dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines const ELFYAML::RelocationSection &Section, 332dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines ContiguousBlobAccumulator &CBA) { 333dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (Section.Type != llvm::ELF::SHT_REL && 334dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Section.Type != llvm::ELF::SHT_RELA) { 335dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines errs() << "error: Invalid relocation section type.\n"; 336dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return false; 337dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines } 338dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 339dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines bool IsRela = Section.Type == llvm::ELF::SHT_RELA; 340dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines SHeader.sh_entsize = IsRela ? sizeof(Elf_Rela) : sizeof(Elf_Rel); 341dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines SHeader.sh_size = SHeader.sh_entsize * Section.Relocations.size(); 342dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 343dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines auto &OS = CBA.getOSAndAlignedOffset(SHeader.sh_offset); 344dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 345dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines for (const auto &Rel : Section.Relocations) { 346dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines unsigned SymIdx; 347dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (SymN2I.lookup(Rel.Symbol, SymIdx)) { 348dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines errs() << "error: Unknown symbol referenced: '" << Rel.Symbol 349dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines << "' at YAML relocation.\n"; 350dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return false; 351dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines } 352dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 353dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (IsRela) { 354dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Elf_Rela REntry; 355dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines zero(REntry); 356dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines REntry.r_offset = Rel.Offset; 357dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines REntry.r_addend = Rel.Addend; 358dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines REntry.setSymbolAndType(SymIdx, Rel.Type); 359dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines OS.write((const char *)&REntry, sizeof(REntry)); 360dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines } else { 361dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines Elf_Rel REntry; 362dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines zero(REntry); 363dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines REntry.r_offset = Rel.Offset; 364dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines REntry.setSymbolAndType(SymIdx, Rel.Type); 365dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines OS.write((const char *)&REntry, sizeof(REntry)); 366dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines } 367dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines } 368dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return true; 369dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines} 370dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 37136b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hinestemplate <class ELFT> bool ELFState<ELFT>::buildSectionIndex() { 37236b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines SN2I.addName(".symtab", getDotSymTabSecNo()); 37336b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines SN2I.addName(".strtab", getDotStrTabSecNo()); 37436b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines SN2I.addName(".shstrtab", getDotShStrTabSecNo()); 375274264ce2ba93d26702f7ebfe6aea121df44113fSean Silva 37636b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines for (unsigned i = 0, e = Doc.Sections.size(); i != e; ++i) { 377dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines StringRef Name = Doc.Sections[i]->Name; 378fe57e347a57d643bbbcc9c19c8267a3e8c06b5a6Sean Silva if (Name.empty()) 379fe57e347a57d643bbbcc9c19c8267a3e8c06b5a6Sean Silva continue; 380c2b6adff3e94f8e299caa31947ad970096377e40Sean Silva // "+ 1" to take into account the SHT_NULL entry. 381c2b6adff3e94f8e299caa31947ad970096377e40Sean Silva if (SN2I.addName(Name, i + 1)) { 382fe57e347a57d643bbbcc9c19c8267a3e8c06b5a6Sean Silva errs() << "error: Repeated section name: '" << Name 383fe57e347a57d643bbbcc9c19c8267a3e8c06b5a6Sean Silva << "' at YAML section number " << i << ".\n"; 38436b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines return false; 385fe57e347a57d643bbbcc9c19c8267a3e8c06b5a6Sean Silva } 386fe57e347a57d643bbbcc9c19c8267a3e8c06b5a6Sean Silva } 38736b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines return true; 38836b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines} 389fe57e347a57d643bbbcc9c19c8267a3e8c06b5a6Sean Silva 39036b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hinestemplate <class ELFT> 391dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hinesbool 392dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen HinesELFState<ELFT>::buildSymbolIndex(std::size_t &StartIndex, 393dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines const std::vector<ELFYAML::Symbol> &Symbols) { 394dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines for (const auto &Sym : Symbols) { 395dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines ++StartIndex; 396dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (Sym.Name.empty()) 397dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines continue; 398dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (SymN2I.addName(Sym.Name, StartIndex)) { 399dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines errs() << "error: Repeated symbol name: '" << Sym.Name << "'.\n"; 400dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return false; 401dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines } 402dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines } 403dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return true; 404dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines} 405dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 406dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hinestemplate <class ELFT> 40736b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hinesint ELFState<ELFT>::writeELF(raw_ostream &OS, const ELFYAML::Object &Doc) { 40836b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines ELFState<ELFT> State(Doc); 40936b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines if (!State.buildSectionIndex()) 41036b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines return 1; 411c18f66e70dea8abc8c9d86d40088fc8fc8247b20Sean Silva 412dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines std::size_t StartSymIndex = 0; 413dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (!State.buildSymbolIndex(StartSymIndex, Doc.Symbols.Local) || 414dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines !State.buildSymbolIndex(StartSymIndex, Doc.Symbols.Global) || 415dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines !State.buildSymbolIndex(StartSymIndex, Doc.Symbols.Weak)) 416dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return 1; 417dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 41836b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines Elf_Ehdr Header; 41936b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines State.initELFHeader(Header); 4202a7e79a30f87be4103963e6049a2f1cf171f502cSean Silva 42136b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines // TODO: Flesh out section header support. 42236b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines // TODO: Program headers. 4232a7e79a30f87be4103963e6049a2f1cf171f502cSean Silva 42436b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines // XXX: This offset is tightly coupled with the order that we write 42536b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines // things to `OS`. 42636b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines const size_t SectionContentBeginOffset = 42736b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines Header.e_ehsize + Header.e_shentsize * Header.e_shnum; 42836b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines ContiguousBlobAccumulator CBA(SectionContentBeginOffset); 42936b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines 430dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines // Doc might not contain .symtab, .strtab and .shstrtab sections, 431dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines // but we will emit them, so make sure to add them to ShStrTabSHeader. 432dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines State.DotShStrtab.add(".symtab"); 433dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines State.DotShStrtab.add(".strtab"); 434dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines State.DotShStrtab.add(".shstrtab"); 435dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines 43636b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines std::vector<Elf_Shdr> SHeaders; 43736b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines if(!State.initSectionHeaders(SHeaders, CBA)) 43836b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines return 1; 4395918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silva 440068463b714bd53fffd41322a5654b4f359c2166cSean Silva // .symtab section. 441068463b714bd53fffd41322a5654b4f359c2166cSean Silva Elf_Shdr SymtabSHeader; 44236b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines State.initSymtabSectionHeader(SymtabSHeader, CBA); 443068463b714bd53fffd41322a5654b4f359c2166cSean Silva SHeaders.push_back(SymtabSHeader); 444068463b714bd53fffd41322a5654b4f359c2166cSean Silva 445afcf60fe15999ea07193118f447a34f41171e433Sean Silva // .strtab string table header. 446a7bd8549fdcb18c686bdadf46000452c927d3b5dSean Silva Elf_Shdr DotStrTabSHeader; 44736b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines State.initStrtabSectionHeader(DotStrTabSHeader, ".strtab", State.DotStrtab, 44836b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines CBA); 44958495694c09540459e4eea50eef2dcffc705f3c3Sean Silva SHeaders.push_back(DotStrTabSHeader); 450a7bd8549fdcb18c686bdadf46000452c927d3b5dSean Silva 45136b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines // .shstrtab string table header. 45236b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines Elf_Shdr ShStrTabSHeader; 45336b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines State.initStrtabSectionHeader(ShStrTabSHeader, ".shstrtab", State.DotShStrtab, 45436b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines CBA); 45536b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines SHeaders.push_back(ShStrTabSHeader); 4565918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silva 4575918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silva OS.write((const char *)&Header, sizeof(Header)); 4586dd7893e8c7234d65cdc47e3f3ce251fdfb35768Will Dietz writeArrayData(OS, makeArrayRef(SHeaders)); 4592a7e79a30f87be4103963e6049a2f1cf171f502cSean Silva CBA.writeBlobToStream(OS); 4604b548ecb012ce27feff9f58aad27775df679b159Sean Silva return 0; 4615918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silva} 4625918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silva 463efc78986b3bf51572e013bfdf8cbb9e06a47c788Sean Silvastatic bool is64Bit(const ELFYAML::Object &Doc) { 464efc78986b3bf51572e013bfdf8cbb9e06a47c788Sean Silva return Doc.Header.Class == ELFYAML::ELF_ELFCLASS(ELF::ELFCLASS64); 465efc78986b3bf51572e013bfdf8cbb9e06a47c788Sean Silva} 466efc78986b3bf51572e013bfdf8cbb9e06a47c788Sean Silva 467efc78986b3bf51572e013bfdf8cbb9e06a47c788Sean Silvastatic bool isLittleEndian(const ELFYAML::Object &Doc) { 468efc78986b3bf51572e013bfdf8cbb9e06a47c788Sean Silva return Doc.Header.Data == ELFYAML::ELF_ELFDATA(ELF::ELFDATA2LSB); 469efc78986b3bf51572e013bfdf8cbb9e06a47c788Sean Silva} 470efc78986b3bf51572e013bfdf8cbb9e06a47c788Sean Silva 471cd81d94322a39503e4a3e87b6ee03d4fcb3465fbStephen Hinesint yaml2elf(yaml::Input &YIn, raw_ostream &Out) { 4725918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silva ELFYAML::Object Doc; 4735918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silva YIn >> Doc; 4745918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silva if (YIn.error()) { 4755918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silva errs() << "yaml2obj: Failed to parse YAML file!\n"; 4765918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silva return 1; 4775918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silva } 478efc78986b3bf51572e013bfdf8cbb9e06a47c788Sean Silva using object::ELFType; 479efc78986b3bf51572e013bfdf8cbb9e06a47c788Sean Silva typedef ELFType<support::little, 8, true> LE64; 480efc78986b3bf51572e013bfdf8cbb9e06a47c788Sean Silva typedef ELFType<support::big, 8, true> BE64; 481efc78986b3bf51572e013bfdf8cbb9e06a47c788Sean Silva typedef ELFType<support::little, 4, false> LE32; 482efc78986b3bf51572e013bfdf8cbb9e06a47c788Sean Silva typedef ELFType<support::big, 4, false> BE32; 483efc78986b3bf51572e013bfdf8cbb9e06a47c788Sean Silva if (is64Bit(Doc)) { 484efc78986b3bf51572e013bfdf8cbb9e06a47c788Sean Silva if (isLittleEndian(Doc)) 485dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return ELFState<LE64>::writeELF(Out, Doc); 4865918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silva else 487dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return ELFState<BE64>::writeELF(Out, Doc); 4885918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silva } else { 489efc78986b3bf51572e013bfdf8cbb9e06a47c788Sean Silva if (isLittleEndian(Doc)) 490dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return ELFState<LE32>::writeELF(Out, Doc); 4915918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silva else 492dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines return ELFState<BE32>::writeELF(Out, Doc); 4935918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silva } 4945918b7a03d4d6a52e18f7c102250c9cfd6ae52ddSean Silva} 495