2013-06-11 07:44:15 +08:00
|
|
|
//===- yaml2elf - Convert YAML to a ELF object file -----------------------===//
|
|
|
|
//
|
|
|
|
// The LLVM Compiler Infrastructure
|
|
|
|
//
|
|
|
|
// This file is distributed under the University of Illinois Open Source
|
|
|
|
// License. See LICENSE.TXT for details.
|
|
|
|
//
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
///
|
|
|
|
/// \file
|
|
|
|
/// \brief The ELF component of yaml2obj.
|
|
|
|
///
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
|
|
|
|
#include "yaml2obj.h"
|
2013-10-13 05:29:16 +08:00
|
|
|
#include "llvm/ADT/ArrayRef.h"
|
2014-07-03 10:01:39 +08:00
|
|
|
#include "llvm/MC/StringTableBuilder.h"
|
2013-08-09 06:27:13 +08:00
|
|
|
#include "llvm/Object/ELFObjectFile.h"
|
2013-06-11 07:44:15 +08:00
|
|
|
#include "llvm/Object/ELFYAML.h"
|
|
|
|
#include "llvm/Support/ELF.h"
|
|
|
|
#include "llvm/Support/MemoryBuffer.h"
|
|
|
|
#include "llvm/Support/YAMLTraits.h"
|
|
|
|
#include "llvm/Support/raw_ostream.h"
|
|
|
|
|
|
|
|
using namespace llvm;
|
|
|
|
|
2013-06-14 06:20:01 +08:00
|
|
|
// This class is used to build up a contiguous binary blob while keeping
|
|
|
|
// track of an offset in the output (which notionally begins at
|
|
|
|
// `InitialOffset`).
|
2013-06-15 08:31:46 +08:00
|
|
|
namespace {
|
2013-06-14 06:20:01 +08:00
|
|
|
class ContiguousBlobAccumulator {
|
|
|
|
const uint64_t InitialOffset;
|
2013-06-21 03:11:41 +08:00
|
|
|
SmallVector<char, 128> Buf;
|
2013-06-14 06:20:01 +08:00
|
|
|
raw_svector_ostream OS;
|
|
|
|
|
2013-06-22 08:47:43 +08:00
|
|
|
/// \returns The new offset.
|
|
|
|
uint64_t padToAlignment(unsigned Align) {
|
|
|
|
uint64_t CurrentOffset = InitialOffset + OS.tell();
|
|
|
|
uint64_t AlignedOffset = RoundUpToAlignment(CurrentOffset, Align);
|
|
|
|
for (; CurrentOffset != AlignedOffset; ++CurrentOffset)
|
|
|
|
OS.write('\0');
|
|
|
|
return AlignedOffset; // == CurrentOffset;
|
|
|
|
}
|
|
|
|
|
2013-06-14 06:20:01 +08:00
|
|
|
public:
|
2013-06-21 03:11:41 +08:00
|
|
|
ContiguousBlobAccumulator(uint64_t InitialOffset_)
|
|
|
|
: InitialOffset(InitialOffset_), Buf(), OS(Buf) {}
|
2013-06-22 08:47:43 +08:00
|
|
|
template <class Integer>
|
|
|
|
raw_ostream &getOSAndAlignedOffset(Integer &Offset, unsigned Align = 16) {
|
|
|
|
Offset = padToAlignment(Align);
|
|
|
|
return OS;
|
|
|
|
}
|
2013-06-14 06:20:01 +08:00
|
|
|
void writeBlobToStream(raw_ostream &Out) { Out << OS.str(); }
|
|
|
|
};
|
2013-06-15 08:31:46 +08:00
|
|
|
} // end anonymous namespace
|
2013-06-14 06:20:01 +08:00
|
|
|
|
2014-04-06 17:02:55 +08:00
|
|
|
// Used to keep track of section and symbol names, so that in the YAML file
|
|
|
|
// sections and symbols can be referenced by name instead of by index.
|
2013-06-15 08:31:46 +08:00
|
|
|
namespace {
|
2014-04-06 17:02:55 +08:00
|
|
|
class NameToIdxMap {
|
2013-06-15 08:25:26 +08:00
|
|
|
StringMap<int> Map;
|
|
|
|
public:
|
|
|
|
/// \returns true if name is already present in the map.
|
2014-04-06 17:02:55 +08:00
|
|
|
bool addName(StringRef Name, unsigned i) {
|
|
|
|
StringMapEntry<int> &Entry = Map.GetOrCreateValue(Name, -1);
|
2013-06-15 08:25:26 +08:00
|
|
|
if (Entry.getValue() != -1)
|
|
|
|
return true;
|
|
|
|
Entry.setValue((int)i);
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
/// \returns true if name is not present in the map
|
2014-04-06 17:02:55 +08:00
|
|
|
bool lookup(StringRef Name, unsigned &Idx) const {
|
|
|
|
StringMap<int>::const_iterator I = Map.find(Name);
|
2013-06-15 08:25:26 +08:00
|
|
|
if (I == Map.end())
|
|
|
|
return true;
|
|
|
|
Idx = I->getValue();
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
};
|
2013-06-15 08:31:46 +08:00
|
|
|
} // end anonymous namespace
|
2013-06-15 08:25:26 +08:00
|
|
|
|
2013-06-14 06:19:48 +08:00
|
|
|
template <class T>
|
2013-10-13 05:29:16 +08:00
|
|
|
static size_t arrayDataSize(ArrayRef<T> A) {
|
|
|
|
return A.size() * sizeof(T);
|
2013-06-14 06:19:48 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
template <class T>
|
2013-10-13 05:29:16 +08:00
|
|
|
static void writeArrayData(raw_ostream &OS, ArrayRef<T> A) {
|
|
|
|
OS.write((const char *)A.data(), arrayDataSize(A));
|
2013-06-14 06:19:48 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
template <class T>
|
|
|
|
static void zero(T &Obj) {
|
|
|
|
memset(&Obj, 0, sizeof(Obj));
|
|
|
|
}
|
|
|
|
|
2013-06-21 03:11:44 +08:00
|
|
|
namespace {
|
|
|
|
/// \brief "Single point of truth" for the ELF file construction.
|
|
|
|
/// TODO: This class still has a ways to go before it is truly a "single
|
|
|
|
/// point of truth".
|
|
|
|
template <class ELFT>
|
|
|
|
class ELFState {
|
2014-04-03 00:34:54 +08:00
|
|
|
typedef typename object::ELFFile<ELFT>::Elf_Ehdr Elf_Ehdr;
|
|
|
|
typedef typename object::ELFFile<ELFT>::Elf_Shdr Elf_Shdr;
|
|
|
|
typedef typename object::ELFFile<ELFT>::Elf_Sym Elf_Sym;
|
2014-04-11 12:13:39 +08:00
|
|
|
typedef typename object::ELFFile<ELFT>::Elf_Rel Elf_Rel;
|
|
|
|
typedef typename object::ELFFile<ELFT>::Elf_Rela Elf_Rela;
|
2014-04-03 00:34:54 +08:00
|
|
|
|
2013-06-21 03:11:44 +08:00
|
|
|
/// \brief The future ".strtab" section.
|
|
|
|
StringTableBuilder DotStrtab;
|
|
|
|
|
2014-04-03 00:34:54 +08:00
|
|
|
/// \brief The future ".shstrtab" section.
|
|
|
|
StringTableBuilder DotShStrtab;
|
|
|
|
|
2014-04-06 17:02:55 +08:00
|
|
|
NameToIdxMap SN2I;
|
2014-04-11 12:13:39 +08:00
|
|
|
NameToIdxMap SymN2I;
|
2014-04-03 00:34:40 +08:00
|
|
|
const ELFYAML::Object &Doc;
|
2013-06-21 04:59:34 +08:00
|
|
|
|
2014-04-03 00:34:54 +08:00
|
|
|
bool buildSectionIndex();
|
2014-04-11 12:13:39 +08:00
|
|
|
bool buildSymbolIndex(std::size_t &StartIndex,
|
|
|
|
const std::vector<ELFYAML::Symbol> &Symbols);
|
2014-04-03 00:34:54 +08:00
|
|
|
void initELFHeader(Elf_Ehdr &Header);
|
|
|
|
bool initSectionHeaders(std::vector<Elf_Shdr> &SHeaders,
|
|
|
|
ContiguousBlobAccumulator &CBA);
|
|
|
|
void initSymtabSectionHeader(Elf_Shdr &SHeader,
|
|
|
|
ContiguousBlobAccumulator &CBA);
|
|
|
|
void initStrtabSectionHeader(Elf_Shdr &SHeader, StringRef Name,
|
|
|
|
StringTableBuilder &STB,
|
|
|
|
ContiguousBlobAccumulator &CBA);
|
|
|
|
void addSymbols(const std::vector<ELFYAML::Symbol> &Symbols,
|
|
|
|
std::vector<Elf_Sym> &Syms, unsigned SymbolBinding);
|
2014-04-11 12:13:39 +08:00
|
|
|
void writeSectionContent(Elf_Shdr &SHeader,
|
|
|
|
const ELFYAML::RawContentSection &Section,
|
|
|
|
ContiguousBlobAccumulator &CBA);
|
|
|
|
bool writeSectionContent(Elf_Shdr &SHeader,
|
|
|
|
const ELFYAML::RelocationSection &Section,
|
|
|
|
ContiguousBlobAccumulator &CBA);
|
2014-04-03 00:34:40 +08:00
|
|
|
|
|
|
|
// - 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; }
|
2013-06-21 03:11:44 +08:00
|
|
|
|
2014-04-03 00:34:54 +08:00
|
|
|
ELFState(const ELFYAML::Object &D) : Doc(D) {}
|
|
|
|
|
|
|
|
public:
|
|
|
|
static int writeELF(raw_ostream &OS, const ELFYAML::Object &Doc);
|
2013-06-21 03:11:44 +08:00
|
|
|
};
|
|
|
|
} // end anonymous namespace
|
|
|
|
|
2013-06-21 08:33:01 +08:00
|
|
|
template <class ELFT>
|
2014-04-03 00:34:54 +08:00
|
|
|
void ELFState<ELFT>::initELFHeader(Elf_Ehdr &Header) {
|
|
|
|
using namespace llvm::ELF;
|
|
|
|
zero(Header);
|
|
|
|
Header.e_ident[EI_MAG0] = 0x7f;
|
|
|
|
Header.e_ident[EI_MAG1] = 'E';
|
|
|
|
Header.e_ident[EI_MAG2] = 'L';
|
|
|
|
Header.e_ident[EI_MAG3] = 'F';
|
|
|
|
Header.e_ident[EI_CLASS] = ELFT::Is64Bits ? ELFCLASS64 : ELFCLASS32;
|
|
|
|
bool IsLittleEndian = ELFT::TargetEndianness == support::little;
|
|
|
|
Header.e_ident[EI_DATA] = IsLittleEndian ? ELFDATA2LSB : ELFDATA2MSB;
|
|
|
|
Header.e_ident[EI_VERSION] = EV_CURRENT;
|
|
|
|
Header.e_ident[EI_OSABI] = Doc.Header.OSABI;
|
|
|
|
Header.e_ident[EI_ABIVERSION] = 0;
|
|
|
|
Header.e_type = Doc.Header.Type;
|
|
|
|
Header.e_machine = Doc.Header.Machine;
|
|
|
|
Header.e_version = EV_CURRENT;
|
|
|
|
Header.e_entry = Doc.Header.Entry;
|
|
|
|
Header.e_flags = Doc.Header.Flags;
|
|
|
|
Header.e_ehsize = sizeof(Elf_Ehdr);
|
|
|
|
Header.e_shentsize = sizeof(Elf_Shdr);
|
|
|
|
// Immediately following the ELF header.
|
|
|
|
Header.e_shoff = sizeof(Header);
|
|
|
|
Header.e_shnum = getSectionCount();
|
|
|
|
Header.e_shstrndx = getDotShStrTabSecNo();
|
|
|
|
}
|
|
|
|
|
|
|
|
template <class ELFT>
|
|
|
|
bool ELFState<ELFT>::initSectionHeaders(std::vector<Elf_Shdr> &SHeaders,
|
|
|
|
ContiguousBlobAccumulator &CBA) {
|
|
|
|
// Ensure SHN_UNDEF entry is present. An all-zero section header is a
|
|
|
|
// valid SHN_UNDEF entry since SHT_NULL == 0.
|
|
|
|
Elf_Shdr SHeader;
|
|
|
|
zero(SHeader);
|
|
|
|
SHeaders.push_back(SHeader);
|
|
|
|
|
2014-05-01 03:38:09 +08:00
|
|
|
for (const auto &Sec : Doc.Sections)
|
|
|
|
DotShStrtab.add(Sec->Name);
|
|
|
|
DotShStrtab.finalize();
|
|
|
|
|
2014-04-03 00:34:54 +08:00
|
|
|
for (const auto &Sec : Doc.Sections) {
|
|
|
|
zero(SHeader);
|
2014-05-01 03:38:09 +08:00
|
|
|
SHeader.sh_name = DotShStrtab.getOffset(Sec->Name);
|
2014-04-11 12:13:39 +08:00
|
|
|
SHeader.sh_type = Sec->Type;
|
|
|
|
SHeader.sh_flags = Sec->Flags;
|
|
|
|
SHeader.sh_addr = Sec->Address;
|
|
|
|
SHeader.sh_addralign = Sec->AddressAlign;
|
2014-04-03 00:34:54 +08:00
|
|
|
|
2014-04-11 12:13:39 +08:00
|
|
|
if (!Sec->Link.empty()) {
|
2013-06-21 09:11:48 +08:00
|
|
|
unsigned Index;
|
2014-04-11 12:13:39 +08:00
|
|
|
if (SN2I.lookup(Sec->Link, Index)) {
|
|
|
|
errs() << "error: Unknown section referenced: '" << Sec->Link
|
|
|
|
<< "' at YAML section '" << Sec->Name << "'.\n";
|
|
|
|
return false;
|
2013-06-21 09:11:48 +08:00
|
|
|
}
|
2014-04-03 00:34:54 +08:00
|
|
|
SHeader.sh_link = Index;
|
|
|
|
}
|
2014-04-11 12:13:39 +08:00
|
|
|
|
|
|
|
if (auto S = dyn_cast<ELFYAML::RawContentSection>(Sec.get()))
|
|
|
|
writeSectionContent(SHeader, *S, CBA);
|
|
|
|
else if (auto S = dyn_cast<ELFYAML::RelocationSection>(Sec.get())) {
|
|
|
|
if (S->Link.empty())
|
|
|
|
// For relocation section set link to .symtab by default.
|
|
|
|
SHeader.sh_link = getDotSymTabSecNo();
|
|
|
|
|
|
|
|
unsigned Index;
|
|
|
|
if (SN2I.lookup(S->Info, Index)) {
|
|
|
|
errs() << "error: Unknown section referenced: '" << S->Info
|
|
|
|
<< "' at YAML section '" << S->Name << "'.\n";
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
SHeader.sh_info = Index;
|
|
|
|
|
|
|
|
if (!writeSectionContent(SHeader, *S, CBA))
|
|
|
|
return false;
|
|
|
|
} else
|
|
|
|
llvm_unreachable("Unknown section type");
|
|
|
|
|
2014-04-03 00:34:54 +08:00
|
|
|
SHeaders.push_back(SHeader);
|
2013-06-21 08:27:50 +08:00
|
|
|
}
|
2014-04-03 00:34:54 +08:00
|
|
|
return true;
|
2013-06-21 08:27:50 +08:00
|
|
|
}
|
|
|
|
|
2013-06-19 07:14:03 +08:00
|
|
|
template <class ELFT>
|
2014-04-03 00:34:54 +08:00
|
|
|
void ELFState<ELFT>::initSymtabSectionHeader(Elf_Shdr &SHeader,
|
|
|
|
ContiguousBlobAccumulator &CBA) {
|
|
|
|
zero(SHeader);
|
2014-05-01 03:38:09 +08:00
|
|
|
SHeader.sh_name = DotShStrtab.getOffset(".symtab");
|
2013-06-22 09:38:00 +08:00
|
|
|
SHeader.sh_type = ELF::SHT_SYMTAB;
|
2014-04-03 00:34:54 +08:00
|
|
|
SHeader.sh_link = getDotStrTabSecNo();
|
2013-06-21 08:27:50 +08:00
|
|
|
// One greater than symbol table index of the last local symbol.
|
2014-04-03 00:34:54 +08:00
|
|
|
SHeader.sh_info = Doc.Symbols.Local.size() + 1;
|
2013-06-19 07:14:03 +08:00
|
|
|
SHeader.sh_entsize = sizeof(Elf_Sym);
|
|
|
|
|
|
|
|
std::vector<Elf_Sym> Syms;
|
2013-06-19 07:19:17 +08:00
|
|
|
{
|
|
|
|
// Ensure STN_UNDEF is present
|
|
|
|
Elf_Sym Sym;
|
|
|
|
zero(Sym);
|
|
|
|
Syms.push_back(Sym);
|
|
|
|
}
|
2014-05-01 03:38:09 +08:00
|
|
|
|
|
|
|
// Add symbol names to .strtab.
|
|
|
|
for (const auto &Sym : Doc.Symbols.Local)
|
|
|
|
DotStrtab.add(Sym.Name);
|
|
|
|
for (const auto &Sym : Doc.Symbols.Global)
|
|
|
|
DotStrtab.add(Sym.Name);
|
|
|
|
for (const auto &Sym : Doc.Symbols.Weak)
|
|
|
|
DotStrtab.add(Sym.Name);
|
|
|
|
DotStrtab.finalize();
|
|
|
|
|
2014-04-03 00:34:54 +08:00
|
|
|
addSymbols(Doc.Symbols.Local, Syms, ELF::STB_LOCAL);
|
|
|
|
addSymbols(Doc.Symbols.Global, Syms, ELF::STB_GLOBAL);
|
|
|
|
addSymbols(Doc.Symbols.Weak, Syms, ELF::STB_WEAK);
|
2013-06-19 07:14:03 +08:00
|
|
|
|
2013-10-13 05:29:16 +08:00
|
|
|
writeArrayData(CBA.getOSAndAlignedOffset(SHeader.sh_offset),
|
|
|
|
makeArrayRef(Syms));
|
|
|
|
SHeader.sh_size = arrayDataSize(makeArrayRef(Syms));
|
2013-06-19 07:14:03 +08:00
|
|
|
}
|
|
|
|
|
2013-06-11 07:44:15 +08:00
|
|
|
template <class ELFT>
|
2014-04-03 00:34:54 +08:00
|
|
|
void ELFState<ELFT>::initStrtabSectionHeader(Elf_Shdr &SHeader, StringRef Name,
|
|
|
|
StringTableBuilder &STB,
|
|
|
|
ContiguousBlobAccumulator &CBA) {
|
|
|
|
zero(SHeader);
|
2014-05-01 03:38:09 +08:00
|
|
|
SHeader.sh_name = DotShStrtab.getOffset(Name);
|
2014-04-03 00:34:54 +08:00
|
|
|
SHeader.sh_type = ELF::SHT_STRTAB;
|
2014-05-01 03:38:09 +08:00
|
|
|
CBA.getOSAndAlignedOffset(SHeader.sh_offset) << STB.data();
|
|
|
|
SHeader.sh_size = STB.data().size();
|
2014-04-03 00:34:54 +08:00
|
|
|
SHeader.sh_addralign = 1;
|
|
|
|
}
|
2013-06-14 06:19:48 +08:00
|
|
|
|
2014-04-03 00:34:54 +08:00
|
|
|
template <class ELFT>
|
|
|
|
void ELFState<ELFT>::addSymbols(const std::vector<ELFYAML::Symbol> &Symbols,
|
|
|
|
std::vector<Elf_Sym> &Syms,
|
|
|
|
unsigned SymbolBinding) {
|
|
|
|
for (const auto &Sym : Symbols) {
|
|
|
|
Elf_Sym Symbol;
|
|
|
|
zero(Symbol);
|
|
|
|
if (!Sym.Name.empty())
|
2014-05-01 03:38:09 +08:00
|
|
|
Symbol.st_name = DotStrtab.getOffset(Sym.Name);
|
2014-04-03 00:34:54 +08:00
|
|
|
Symbol.setBindingAndType(SymbolBinding, Sym.Type);
|
|
|
|
if (!Sym.Section.empty()) {
|
|
|
|
unsigned Index;
|
2014-04-06 17:02:55 +08:00
|
|
|
if (SN2I.lookup(Sym.Section, Index)) {
|
2014-04-03 00:34:54 +08:00
|
|
|
errs() << "error: Unknown section referenced: '" << Sym.Section
|
|
|
|
<< "' by YAML symbol " << Sym.Name << ".\n";
|
|
|
|
exit(1);
|
|
|
|
}
|
|
|
|
Symbol.st_shndx = Index;
|
|
|
|
} // else Symbol.st_shndex == SHN_UNDEF (== 0), since it was zero'd earlier.
|
|
|
|
Symbol.st_value = Sym.Value;
|
2014-06-06 15:41:57 +08:00
|
|
|
Symbol.st_other = Sym.Visibility;
|
2014-04-03 00:34:54 +08:00
|
|
|
Symbol.st_size = Sym.Size;
|
|
|
|
Syms.push_back(Symbol);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-04-11 12:13:39 +08:00
|
|
|
template <class ELFT>
|
|
|
|
void
|
|
|
|
ELFState<ELFT>::writeSectionContent(Elf_Shdr &SHeader,
|
|
|
|
const ELFYAML::RawContentSection &Section,
|
|
|
|
ContiguousBlobAccumulator &CBA) {
|
2014-05-17 00:01:00 +08:00
|
|
|
assert(Section.Size >= Section.Content.binary_size() &&
|
|
|
|
"Section size and section content are inconsistent");
|
|
|
|
raw_ostream &OS = CBA.getOSAndAlignedOffset(SHeader.sh_offset);
|
|
|
|
Section.Content.writeAsBinary(OS);
|
|
|
|
for (auto i = Section.Content.binary_size(); i < Section.Size; ++i)
|
|
|
|
OS.write(0);
|
2014-04-11 12:13:39 +08:00
|
|
|
SHeader.sh_entsize = 0;
|
2014-05-17 00:01:00 +08:00
|
|
|
SHeader.sh_size = Section.Size;
|
2014-04-11 12:13:39 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
template <class ELFT>
|
|
|
|
bool
|
|
|
|
ELFState<ELFT>::writeSectionContent(Elf_Shdr &SHeader,
|
|
|
|
const ELFYAML::RelocationSection &Section,
|
|
|
|
ContiguousBlobAccumulator &CBA) {
|
|
|
|
if (Section.Type != llvm::ELF::SHT_REL &&
|
|
|
|
Section.Type != llvm::ELF::SHT_RELA) {
|
|
|
|
errs() << "error: Invalid relocation section type.\n";
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool IsRela = Section.Type == llvm::ELF::SHT_RELA;
|
|
|
|
SHeader.sh_entsize = IsRela ? sizeof(Elf_Rela) : sizeof(Elf_Rel);
|
|
|
|
SHeader.sh_size = SHeader.sh_entsize * Section.Relocations.size();
|
|
|
|
|
|
|
|
auto &OS = CBA.getOSAndAlignedOffset(SHeader.sh_offset);
|
|
|
|
|
|
|
|
for (const auto &Rel : Section.Relocations) {
|
|
|
|
unsigned SymIdx;
|
|
|
|
if (SymN2I.lookup(Rel.Symbol, SymIdx)) {
|
|
|
|
errs() << "error: Unknown symbol referenced: '" << Rel.Symbol
|
|
|
|
<< "' at YAML relocation.\n";
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (IsRela) {
|
|
|
|
Elf_Rela REntry;
|
|
|
|
zero(REntry);
|
|
|
|
REntry.r_offset = Rel.Offset;
|
|
|
|
REntry.r_addend = Rel.Addend;
|
|
|
|
REntry.setSymbolAndType(SymIdx, Rel.Type);
|
|
|
|
OS.write((const char *)&REntry, sizeof(REntry));
|
|
|
|
} else {
|
|
|
|
Elf_Rel REntry;
|
|
|
|
zero(REntry);
|
|
|
|
REntry.r_offset = Rel.Offset;
|
|
|
|
REntry.setSymbolAndType(SymIdx, Rel.Type);
|
|
|
|
OS.write((const char *)&REntry, sizeof(REntry));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2014-04-03 00:34:54 +08:00
|
|
|
template <class ELFT> bool ELFState<ELFT>::buildSectionIndex() {
|
|
|
|
SN2I.addName(".symtab", getDotSymTabSecNo());
|
|
|
|
SN2I.addName(".strtab", getDotStrTabSecNo());
|
|
|
|
SN2I.addName(".shstrtab", getDotShStrTabSecNo());
|
|
|
|
|
|
|
|
for (unsigned i = 0, e = Doc.Sections.size(); i != e; ++i) {
|
2014-04-11 12:13:39 +08:00
|
|
|
StringRef Name = Doc.Sections[i]->Name;
|
2014-04-03 00:34:54 +08:00
|
|
|
if (Name.empty())
|
|
|
|
continue;
|
|
|
|
// "+ 1" to take into account the SHT_NULL entry.
|
|
|
|
if (SN2I.addName(Name, i + 1)) {
|
|
|
|
errs() << "error: Repeated section name: '" << Name
|
|
|
|
<< "' at YAML section number " << i << ".\n";
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2014-04-11 12:13:39 +08:00
|
|
|
template <class ELFT>
|
|
|
|
bool
|
|
|
|
ELFState<ELFT>::buildSymbolIndex(std::size_t &StartIndex,
|
|
|
|
const std::vector<ELFYAML::Symbol> &Symbols) {
|
|
|
|
for (const auto &Sym : Symbols) {
|
|
|
|
++StartIndex;
|
|
|
|
if (Sym.Name.empty())
|
|
|
|
continue;
|
|
|
|
if (SymN2I.addName(Sym.Name, StartIndex)) {
|
|
|
|
errs() << "error: Repeated symbol name: '" << Sym.Name << "'.\n";
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2014-04-03 00:34:54 +08:00
|
|
|
template <class ELFT>
|
|
|
|
int ELFState<ELFT>::writeELF(raw_ostream &OS, const ELFYAML::Object &Doc) {
|
2014-04-03 00:34:40 +08:00
|
|
|
ELFState<ELFT> State(Doc);
|
|
|
|
if (!State.buildSectionIndex())
|
|
|
|
return 1;
|
|
|
|
|
2014-04-11 12:13:39 +08:00
|
|
|
std::size_t StartSymIndex = 0;
|
|
|
|
if (!State.buildSymbolIndex(StartSymIndex, Doc.Symbols.Local) ||
|
|
|
|
!State.buildSymbolIndex(StartSymIndex, Doc.Symbols.Global) ||
|
|
|
|
!State.buildSymbolIndex(StartSymIndex, Doc.Symbols.Weak))
|
|
|
|
return 1;
|
|
|
|
|
2013-06-14 06:19:48 +08:00
|
|
|
Elf_Ehdr Header;
|
2014-04-03 00:34:54 +08:00
|
|
|
State.initELFHeader(Header);
|
2013-06-14 06:19:48 +08:00
|
|
|
|
|
|
|
// TODO: Flesh out section header support.
|
|
|
|
// TODO: Program headers.
|
|
|
|
|
2013-06-21 03:11:44 +08:00
|
|
|
// XXX: This offset is tightly coupled with the order that we write
|
|
|
|
// things to `OS`.
|
|
|
|
const size_t SectionContentBeginOffset =
|
|
|
|
Header.e_ehsize + Header.e_shentsize * Header.e_shnum;
|
|
|
|
ContiguousBlobAccumulator CBA(SectionContentBeginOffset);
|
2013-06-21 04:59:34 +08:00
|
|
|
|
2014-05-01 03:38:09 +08:00
|
|
|
// 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");
|
|
|
|
|
2013-06-14 06:19:48 +08:00
|
|
|
std::vector<Elf_Shdr> SHeaders;
|
2014-04-03 00:34:54 +08:00
|
|
|
if(!State.initSectionHeaders(SHeaders, CBA))
|
|
|
|
return 1;
|
2013-06-11 07:44:15 +08:00
|
|
|
|
2013-06-22 09:38:00 +08:00
|
|
|
// .symtab section.
|
|
|
|
Elf_Shdr SymtabSHeader;
|
2014-04-03 00:34:54 +08:00
|
|
|
State.initSymtabSectionHeader(SymtabSHeader, CBA);
|
2013-06-22 09:38:00 +08:00
|
|
|
SHeaders.push_back(SymtabSHeader);
|
|
|
|
|
2013-06-19 07:14:03 +08:00
|
|
|
// .strtab string table header.
|
2013-06-19 05:37:50 +08:00
|
|
|
Elf_Shdr DotStrTabSHeader;
|
2014-04-03 00:34:54 +08:00
|
|
|
State.initStrtabSectionHeader(DotStrTabSHeader, ".strtab", State.DotStrtab,
|
|
|
|
CBA);
|
2013-06-22 09:06:12 +08:00
|
|
|
SHeaders.push_back(DotStrTabSHeader);
|
2013-06-19 05:37:50 +08:00
|
|
|
|
2014-04-03 00:34:54 +08:00
|
|
|
// .shstrtab string table header.
|
|
|
|
Elf_Shdr ShStrTabSHeader;
|
|
|
|
State.initStrtabSectionHeader(ShStrTabSHeader, ".shstrtab", State.DotShStrtab,
|
|
|
|
CBA);
|
|
|
|
SHeaders.push_back(ShStrTabSHeader);
|
2013-06-11 07:44:15 +08:00
|
|
|
|
|
|
|
OS.write((const char *)&Header, sizeof(Header));
|
2013-10-13 05:29:16 +08:00
|
|
|
writeArrayData(OS, makeArrayRef(SHeaders));
|
2013-06-14 06:20:01 +08:00
|
|
|
CBA.writeBlobToStream(OS);
|
2013-06-18 04:14:59 +08:00
|
|
|
return 0;
|
2013-06-11 07:44:15 +08:00
|
|
|
}
|
|
|
|
|
2013-06-22 09:03:35 +08:00
|
|
|
static bool is64Bit(const ELFYAML::Object &Doc) {
|
|
|
|
return Doc.Header.Class == ELFYAML::ELF_ELFCLASS(ELF::ELFCLASS64);
|
|
|
|
}
|
|
|
|
|
|
|
|
static bool isLittleEndian(const ELFYAML::Object &Doc) {
|
|
|
|
return Doc.Header.Data == ELFYAML::ELF_ELFDATA(ELF::ELFDATA2LSB);
|
|
|
|
}
|
|
|
|
|
2014-05-31 12:51:07 +08:00
|
|
|
int yaml2elf(yaml::Input &YIn, raw_ostream &Out) {
|
2013-06-11 07:44:15 +08:00
|
|
|
ELFYAML::Object Doc;
|
|
|
|
YIn >> Doc;
|
|
|
|
if (YIn.error()) {
|
|
|
|
errs() << "yaml2obj: Failed to parse YAML file!\n";
|
|
|
|
return 1;
|
|
|
|
}
|
2013-06-22 09:03:35 +08:00
|
|
|
using object::ELFType;
|
|
|
|
typedef ELFType<support::little, 8, true> LE64;
|
|
|
|
typedef ELFType<support::big, 8, true> BE64;
|
|
|
|
typedef ELFType<support::little, 4, false> LE32;
|
|
|
|
typedef ELFType<support::big, 4, false> BE32;
|
|
|
|
if (is64Bit(Doc)) {
|
|
|
|
if (isLittleEndian(Doc))
|
2014-05-16 00:14:02 +08:00
|
|
|
return ELFState<LE64>::writeELF(Out, Doc);
|
2013-06-11 07:44:15 +08:00
|
|
|
else
|
2014-05-16 00:14:02 +08:00
|
|
|
return ELFState<BE64>::writeELF(Out, Doc);
|
2013-06-11 07:44:15 +08:00
|
|
|
} else {
|
2013-06-22 09:03:35 +08:00
|
|
|
if (isLittleEndian(Doc))
|
2014-05-16 00:14:02 +08:00
|
|
|
return ELFState<LE32>::writeELF(Out, Doc);
|
2013-06-11 07:44:15 +08:00
|
|
|
else
|
2014-05-16 00:14:02 +08:00
|
|
|
return ELFState<BE32>::writeELF(Out, Doc);
|
2013-06-11 07:44:15 +08:00
|
|
|
}
|
|
|
|
}
|