1b60a18deSMichael J. Spencer //===- ELFObjectFile.cpp - ELF object file implementation -------*- C++ -*-===// 2b60a18deSMichael J. Spencer // 3b60a18deSMichael J. Spencer // The LLVM Compiler Infrastructure 4b60a18deSMichael J. Spencer // 5b60a18deSMichael J. Spencer // This file is distributed under the University of Illinois Open Source 6b60a18deSMichael J. Spencer // License. See LICENSE.TXT for details. 7b60a18deSMichael J. Spencer // 8b60a18deSMichael J. Spencer //===----------------------------------------------------------------------===// 9b60a18deSMichael J. Spencer // 10b60a18deSMichael J. Spencer // This file defines the ELFObjectFile class. 11b60a18deSMichael J. Spencer // 12b60a18deSMichael J. Spencer //===----------------------------------------------------------------------===// 13b60a18deSMichael J. Spencer 14b60a18deSMichael J. Spencer #include "llvm/ADT/SmallVector.h" 15b60a18deSMichael J. Spencer #include "llvm/ADT/StringSwitch.h" 16b60a18deSMichael J. Spencer #include "llvm/ADT/Triple.h" 17b60a18deSMichael J. Spencer #include "llvm/Object/ObjectFile.h" 18b60a18deSMichael J. Spencer #include "llvm/Support/ELF.h" 19b60a18deSMichael J. Spencer #include "llvm/Support/Endian.h" 20b60a18deSMichael J. Spencer #include "llvm/Support/ErrorHandling.h" 21b60a18deSMichael J. Spencer #include "llvm/Support/MemoryBuffer.h" 22b60a18deSMichael J. Spencer #include <limits> 23b60a18deSMichael J. Spencer #include <utility> 24b60a18deSMichael J. Spencer 25b60a18deSMichael J. Spencer using namespace llvm; 26b60a18deSMichael J. Spencer using namespace object; 27b60a18deSMichael J. Spencer 28b60a18deSMichael J. Spencer // Templates to choose Elf_Addr and Elf_Off depending on is64Bits. 29b60a18deSMichael J. Spencer namespace { 30b60a18deSMichael J. Spencer template<support::endianness target_endianness> 31b60a18deSMichael J. Spencer struct ELFDataTypeTypedefHelperCommon { 32b60a18deSMichael J. Spencer typedef support::detail::packed_endian_specific_integral 33b60a18deSMichael J. Spencer <uint16_t, target_endianness, support::aligned> Elf_Half; 34b60a18deSMichael J. Spencer typedef support::detail::packed_endian_specific_integral 35b60a18deSMichael J. Spencer <uint32_t, target_endianness, support::aligned> Elf_Word; 36b60a18deSMichael J. Spencer typedef support::detail::packed_endian_specific_integral 37b60a18deSMichael J. Spencer <int32_t, target_endianness, support::aligned> Elf_Sword; 38b60a18deSMichael J. Spencer typedef support::detail::packed_endian_specific_integral 39b60a18deSMichael J. Spencer <uint64_t, target_endianness, support::aligned> Elf_Xword; 40b60a18deSMichael J. Spencer typedef support::detail::packed_endian_specific_integral 41b60a18deSMichael J. Spencer <int64_t, target_endianness, support::aligned> Elf_Sxword; 42b60a18deSMichael J. Spencer }; 43b60a18deSMichael J. Spencer } 44b60a18deSMichael J. Spencer 45b60a18deSMichael J. Spencer namespace { 46b60a18deSMichael J. Spencer template<support::endianness target_endianness, bool is64Bits> 47b60a18deSMichael J. Spencer struct ELFDataTypeTypedefHelper; 48b60a18deSMichael J. Spencer 49b60a18deSMichael J. Spencer /// ELF 32bit types. 50b60a18deSMichael J. Spencer template<support::endianness target_endianness> 51b60a18deSMichael J. Spencer struct ELFDataTypeTypedefHelper<target_endianness, false> 52b60a18deSMichael J. Spencer : ELFDataTypeTypedefHelperCommon<target_endianness> { 53b60a18deSMichael J. Spencer typedef support::detail::packed_endian_specific_integral 54b60a18deSMichael J. Spencer <uint32_t, target_endianness, support::aligned> Elf_Addr; 55b60a18deSMichael J. Spencer typedef support::detail::packed_endian_specific_integral 56b60a18deSMichael J. Spencer <uint32_t, target_endianness, support::aligned> Elf_Off; 57b60a18deSMichael J. Spencer }; 58b60a18deSMichael J. Spencer 59b60a18deSMichael J. Spencer /// ELF 64bit types. 60b60a18deSMichael J. Spencer template<support::endianness target_endianness> 61b60a18deSMichael J. Spencer struct ELFDataTypeTypedefHelper<target_endianness, true> 62b60a18deSMichael J. Spencer : ELFDataTypeTypedefHelperCommon<target_endianness>{ 63b60a18deSMichael J. Spencer typedef support::detail::packed_endian_specific_integral 64b60a18deSMichael J. Spencer <uint64_t, target_endianness, support::aligned> Elf_Addr; 65b60a18deSMichael J. Spencer typedef support::detail::packed_endian_specific_integral 66b60a18deSMichael J. Spencer <uint64_t, target_endianness, support::aligned> Elf_Off; 67b60a18deSMichael J. Spencer }; 68b60a18deSMichael J. Spencer } 69b60a18deSMichael J. Spencer 70b60a18deSMichael J. Spencer // I really don't like doing this, but the alternative is copypasta. 71b60a18deSMichael J. Spencer #define LLVM_ELF_IMPORT_TYPES(target_endianness, is64Bits) \ 72b60a18deSMichael J. Spencer typedef typename \ 73b60a18deSMichael J. Spencer ELFDataTypeTypedefHelper<target_endianness, is64Bits>::Elf_Addr Elf_Addr; \ 74b60a18deSMichael J. Spencer typedef typename \ 75b60a18deSMichael J. Spencer ELFDataTypeTypedefHelper<target_endianness, is64Bits>::Elf_Off Elf_Off; \ 76b60a18deSMichael J. Spencer typedef typename \ 77b60a18deSMichael J. Spencer ELFDataTypeTypedefHelper<target_endianness, is64Bits>::Elf_Half Elf_Half; \ 78b60a18deSMichael J. Spencer typedef typename \ 79b60a18deSMichael J. Spencer ELFDataTypeTypedefHelper<target_endianness, is64Bits>::Elf_Word Elf_Word; \ 80b60a18deSMichael J. Spencer typedef typename \ 81b60a18deSMichael J. Spencer ELFDataTypeTypedefHelper<target_endianness, is64Bits>::Elf_Sword Elf_Sword; \ 82b60a18deSMichael J. Spencer typedef typename \ 83b60a18deSMichael J. Spencer ELFDataTypeTypedefHelper<target_endianness, is64Bits>::Elf_Xword Elf_Xword; \ 84b60a18deSMichael J. Spencer typedef typename \ 85b60a18deSMichael J. Spencer ELFDataTypeTypedefHelper<target_endianness, is64Bits>::Elf_Sxword Elf_Sxword; 86b60a18deSMichael J. Spencer 87b60a18deSMichael J. Spencer // Section header. 88b60a18deSMichael J. Spencer namespace { 89b60a18deSMichael J. Spencer template<support::endianness target_endianness, bool is64Bits> 90b60a18deSMichael J. Spencer struct Elf_Shdr_Base; 91b60a18deSMichael J. Spencer 92b60a18deSMichael J. Spencer template<support::endianness target_endianness> 93b60a18deSMichael J. Spencer struct Elf_Shdr_Base<target_endianness, false> { 94b60a18deSMichael J. Spencer LLVM_ELF_IMPORT_TYPES(target_endianness, false) 95b60a18deSMichael J. Spencer Elf_Word sh_name; // Section name (index into string table) 96b60a18deSMichael J. Spencer Elf_Word sh_type; // Section type (SHT_*) 97b60a18deSMichael J. Spencer Elf_Word sh_flags; // Section flags (SHF_*) 98b60a18deSMichael J. Spencer Elf_Addr sh_addr; // Address where section is to be loaded 99b60a18deSMichael J. Spencer Elf_Off sh_offset; // File offset of section data, in bytes 100b60a18deSMichael J. Spencer Elf_Word sh_size; // Size of section, in bytes 101b60a18deSMichael J. Spencer Elf_Word sh_link; // Section type-specific header table index link 102b60a18deSMichael J. Spencer Elf_Word sh_info; // Section type-specific extra information 103b60a18deSMichael J. Spencer Elf_Word sh_addralign;// Section address alignment 104b60a18deSMichael J. Spencer Elf_Word sh_entsize; // Size of records contained within the section 105b60a18deSMichael J. Spencer }; 106b60a18deSMichael J. Spencer 107b60a18deSMichael J. Spencer template<support::endianness target_endianness> 108b60a18deSMichael J. Spencer struct Elf_Shdr_Base<target_endianness, true> { 109b60a18deSMichael J. Spencer LLVM_ELF_IMPORT_TYPES(target_endianness, true) 110b60a18deSMichael J. Spencer Elf_Word sh_name; // Section name (index into string table) 111b60a18deSMichael J. Spencer Elf_Word sh_type; // Section type (SHT_*) 112b60a18deSMichael J. Spencer Elf_Xword sh_flags; // Section flags (SHF_*) 113b60a18deSMichael J. Spencer Elf_Addr sh_addr; // Address where section is to be loaded 114b60a18deSMichael J. Spencer Elf_Off sh_offset; // File offset of section data, in bytes 115b60a18deSMichael J. Spencer Elf_Xword sh_size; // Size of section, in bytes 116b60a18deSMichael J. Spencer Elf_Word sh_link; // Section type-specific header table index link 117b60a18deSMichael J. Spencer Elf_Word sh_info; // Section type-specific extra information 118b60a18deSMichael J. Spencer Elf_Xword sh_addralign;// Section address alignment 119b60a18deSMichael J. Spencer Elf_Xword sh_entsize; // Size of records contained within the section 120b60a18deSMichael J. Spencer }; 121b60a18deSMichael J. Spencer 122b60a18deSMichael J. Spencer template<support::endianness target_endianness, bool is64Bits> 123b60a18deSMichael J. Spencer struct Elf_Shdr_Impl : Elf_Shdr_Base<target_endianness, is64Bits> { 124b60a18deSMichael J. Spencer using Elf_Shdr_Base<target_endianness, is64Bits>::sh_entsize; 125b60a18deSMichael J. Spencer using Elf_Shdr_Base<target_endianness, is64Bits>::sh_size; 126b60a18deSMichael J. Spencer 127b60a18deSMichael J. Spencer /// @brief Get the number of entities this section contains if it has any. 128b60a18deSMichael J. Spencer unsigned getEntityCount() const { 129b60a18deSMichael J. Spencer if (sh_entsize == 0) 130b60a18deSMichael J. Spencer return 0; 131b60a18deSMichael J. Spencer return sh_size / sh_entsize; 132b60a18deSMichael J. Spencer } 133b60a18deSMichael J. Spencer }; 134b60a18deSMichael J. Spencer } 135b60a18deSMichael J. Spencer 136b60a18deSMichael J. Spencer namespace { 137b60a18deSMichael J. Spencer template<support::endianness target_endianness, bool is64Bits> 138b60a18deSMichael J. Spencer struct Elf_Sym_Base; 139b60a18deSMichael J. Spencer 140b60a18deSMichael J. Spencer template<support::endianness target_endianness> 141b60a18deSMichael J. Spencer struct Elf_Sym_Base<target_endianness, false> { 142b60a18deSMichael J. Spencer LLVM_ELF_IMPORT_TYPES(target_endianness, false) 143b60a18deSMichael J. Spencer Elf_Word st_name; // Symbol name (index into string table) 144b60a18deSMichael J. Spencer Elf_Addr st_value; // Value or address associated with the symbol 145b60a18deSMichael J. Spencer Elf_Word st_size; // Size of the symbol 146b60a18deSMichael J. Spencer unsigned char st_info; // Symbol's type and binding attributes 147b60a18deSMichael J. Spencer unsigned char st_other; // Must be zero; reserved 148b60a18deSMichael J. Spencer Elf_Half st_shndx; // Which section (header table index) it's defined in 149b60a18deSMichael J. Spencer }; 150b60a18deSMichael J. Spencer 151b60a18deSMichael J. Spencer template<support::endianness target_endianness> 152b60a18deSMichael J. Spencer struct Elf_Sym_Base<target_endianness, true> { 153b60a18deSMichael J. Spencer LLVM_ELF_IMPORT_TYPES(target_endianness, true) 154b60a18deSMichael J. Spencer Elf_Word st_name; // Symbol name (index into string table) 155b60a18deSMichael J. Spencer unsigned char st_info; // Symbol's type and binding attributes 156b60a18deSMichael J. Spencer unsigned char st_other; // Must be zero; reserved 157b60a18deSMichael J. Spencer Elf_Half st_shndx; // Which section (header table index) it's defined in 158b60a18deSMichael J. Spencer Elf_Addr st_value; // Value or address associated with the symbol 159b60a18deSMichael J. Spencer Elf_Xword st_size; // Size of the symbol 160b60a18deSMichael J. Spencer }; 161b60a18deSMichael J. Spencer 162b60a18deSMichael J. Spencer template<support::endianness target_endianness, bool is64Bits> 163b60a18deSMichael J. Spencer struct Elf_Sym_Impl : Elf_Sym_Base<target_endianness, is64Bits> { 164b60a18deSMichael J. Spencer using Elf_Sym_Base<target_endianness, is64Bits>::st_info; 165b60a18deSMichael J. Spencer 166b60a18deSMichael J. Spencer // These accessors and mutators correspond to the ELF32_ST_BIND, 167b60a18deSMichael J. Spencer // ELF32_ST_TYPE, and ELF32_ST_INFO macros defined in the ELF specification: 168b60a18deSMichael J. Spencer unsigned char getBinding() const { return st_info >> 4; } 169b60a18deSMichael J. Spencer unsigned char getType() const { return st_info & 0x0f; } 170b60a18deSMichael J. Spencer void setBinding(unsigned char b) { setBindingAndType(b, getType()); } 171b60a18deSMichael J. Spencer void setType(unsigned char t) { setBindingAndType(getBinding(), t); } 172b60a18deSMichael J. Spencer void setBindingAndType(unsigned char b, unsigned char t) { 173b60a18deSMichael J. Spencer st_info = (b << 4) + (t & 0x0f); 174b60a18deSMichael J. Spencer } 175b60a18deSMichael J. Spencer }; 176b60a18deSMichael J. Spencer } 177b60a18deSMichael J. Spencer 178b60a18deSMichael J. Spencer namespace { 179b60a18deSMichael J. Spencer template<support::endianness target_endianness, bool is64Bits> 180b60a18deSMichael J. Spencer class ELFObjectFile : public ObjectFile { 181b60a18deSMichael J. Spencer LLVM_ELF_IMPORT_TYPES(target_endianness, is64Bits) 182b60a18deSMichael J. Spencer 183b60a18deSMichael J. Spencer typedef Elf_Shdr_Impl<target_endianness, is64Bits> Elf_Shdr; 184b60a18deSMichael J. Spencer typedef Elf_Sym_Impl<target_endianness, is64Bits> Elf_Sym; 185b60a18deSMichael J. Spencer 186b60a18deSMichael J. Spencer struct Elf_Ehdr { 187b60a18deSMichael J. Spencer unsigned char e_ident[ELF::EI_NIDENT]; // ELF Identification bytes 188b60a18deSMichael J. Spencer Elf_Half e_type; // Type of file (see ET_*) 189b60a18deSMichael J. Spencer Elf_Half e_machine; // Required architecture for this file (see EM_*) 190b60a18deSMichael J. Spencer Elf_Word e_version; // Must be equal to 1 191b60a18deSMichael J. Spencer Elf_Addr e_entry; // Address to jump to in order to start program 192b60a18deSMichael J. Spencer Elf_Off e_phoff; // Program header table's file offset, in bytes 193b60a18deSMichael J. Spencer Elf_Off e_shoff; // Section header table's file offset, in bytes 194b60a18deSMichael J. Spencer Elf_Word e_flags; // Processor-specific flags 195b60a18deSMichael J. Spencer Elf_Half e_ehsize; // Size of ELF header, in bytes 196b60a18deSMichael J. Spencer Elf_Half e_phentsize;// Size of an entry in the program header table 197b60a18deSMichael J. Spencer Elf_Half e_phnum; // Number of entries in the program header table 198b60a18deSMichael J. Spencer Elf_Half e_shentsize;// Size of an entry in the section header table 199b60a18deSMichael J. Spencer Elf_Half e_shnum; // Number of entries in the section header table 200b60a18deSMichael J. Spencer Elf_Half e_shstrndx; // Section header table index of section name 201b60a18deSMichael J. Spencer // string table 202b60a18deSMichael J. Spencer bool checkMagic() const { 203b60a18deSMichael J. Spencer return (memcmp(e_ident, ELF::ElfMagic, strlen(ELF::ElfMagic))) == 0; 204b60a18deSMichael J. Spencer } 205b60a18deSMichael J. Spencer unsigned char getFileClass() const { return e_ident[ELF::EI_CLASS]; } 206b60a18deSMichael J. Spencer unsigned char getDataEncoding() const { return e_ident[ELF::EI_DATA]; } 207b60a18deSMichael J. Spencer }; 208b60a18deSMichael J. Spencer 209b60a18deSMichael J. Spencer typedef SmallVector<const Elf_Shdr*, 1> SymbolTableSections_t; 210b60a18deSMichael J. Spencer 211b60a18deSMichael J. Spencer const Elf_Ehdr *Header; 212b60a18deSMichael J. Spencer const Elf_Shdr *SectionHeaderTable; 213b60a18deSMichael J. Spencer const Elf_Shdr *dot_shstrtab_sec; // Section header string table. 214b60a18deSMichael J. Spencer const Elf_Shdr *dot_strtab_sec; // Symbol header string table. 215b60a18deSMichael J. Spencer SymbolTableSections_t SymbolTableSections; 216b60a18deSMichael J. Spencer 217b60a18deSMichael J. Spencer void validateSymbol(DataRefImpl Symb) const; 218b60a18deSMichael J. Spencer const Elf_Sym *getSymbol(DataRefImpl Symb) const; 219b60a18deSMichael J. Spencer const Elf_Shdr *getSection(DataRefImpl index) const; 220b60a18deSMichael J. Spencer const Elf_Shdr *getSection(uint16_t index) const; 221b60a18deSMichael J. Spencer const char *getString(uint16_t section, uint32_t offset) const; 222b60a18deSMichael J. Spencer const char *getString(const Elf_Shdr *section, uint32_t offset) const; 223b60a18deSMichael J. Spencer 224b60a18deSMichael J. Spencer protected: 225b60a18deSMichael J. Spencer virtual SymbolRef getSymbolNext(DataRefImpl Symb) const; 226b60a18deSMichael J. Spencer virtual StringRef getSymbolName(DataRefImpl Symb) const; 227b60a18deSMichael J. Spencer virtual uint64_t getSymbolAddress(DataRefImpl Symb) const; 228b60a18deSMichael J. Spencer virtual uint64_t getSymbolSize(DataRefImpl Symb) const; 229b60a18deSMichael J. Spencer virtual char getSymbolNMTypeChar(DataRefImpl Symb) const; 230b60a18deSMichael J. Spencer virtual bool isSymbolInternal(DataRefImpl Symb) const; 231b60a18deSMichael J. Spencer 232b60a18deSMichael J. Spencer virtual SectionRef getSectionNext(DataRefImpl Sec) const; 233b60a18deSMichael J. Spencer virtual StringRef getSectionName(DataRefImpl Sec) const; 234b60a18deSMichael J. Spencer virtual uint64_t getSectionAddress(DataRefImpl Sec) const; 235b60a18deSMichael J. Spencer virtual uint64_t getSectionSize(DataRefImpl Sec) const; 236b60a18deSMichael J. Spencer virtual StringRef getSectionContents(DataRefImpl Sec) const; 237b60a18deSMichael J. Spencer virtual bool isSectionText(DataRefImpl Sec) const; 238b60a18deSMichael J. Spencer 239b60a18deSMichael J. Spencer public: 240b60a18deSMichael J. Spencer ELFObjectFile(MemoryBuffer *Object); 241b60a18deSMichael J. Spencer virtual symbol_iterator begin_symbols() const; 242b60a18deSMichael J. Spencer virtual symbol_iterator end_symbols() const; 243b60a18deSMichael J. Spencer virtual section_iterator begin_sections() const; 244b60a18deSMichael J. Spencer virtual section_iterator end_sections() const; 245b60a18deSMichael J. Spencer 246b60a18deSMichael J. Spencer virtual uint8_t getBytesInAddress() const; 247b60a18deSMichael J. Spencer virtual StringRef getFileFormatName() const; 248b60a18deSMichael J. Spencer virtual unsigned getArch() const; 249b60a18deSMichael J. Spencer }; 250b60a18deSMichael J. Spencer } // end namespace 251b60a18deSMichael J. Spencer 252b60a18deSMichael J. Spencer template<support::endianness target_endianness, bool is64Bits> 253b60a18deSMichael J. Spencer void ELFObjectFile<target_endianness, is64Bits> 254b60a18deSMichael J. Spencer ::validateSymbol(DataRefImpl Symb) const { 255b60a18deSMichael J. Spencer const Elf_Sym *symb = getSymbol(Symb); 2560324b672SMichael J. Spencer const Elf_Shdr *SymbolTableSection = SymbolTableSections[Symb.d.b]; 257b60a18deSMichael J. Spencer // FIXME: We really need to do proper error handling in the case of an invalid 258b60a18deSMichael J. Spencer // input file. Because we don't use exceptions, I think we'll just pass 259b60a18deSMichael J. Spencer // an error object around. 260b60a18deSMichael J. Spencer if (!( symb 261b60a18deSMichael J. Spencer && SymbolTableSection 262b60a18deSMichael J. Spencer && symb >= (const Elf_Sym*)(base 263b60a18deSMichael J. Spencer + SymbolTableSection->sh_offset) 264b60a18deSMichael J. Spencer && symb < (const Elf_Sym*)(base 265b60a18deSMichael J. Spencer + SymbolTableSection->sh_offset 266b60a18deSMichael J. Spencer + SymbolTableSection->sh_size))) 267b60a18deSMichael J. Spencer // FIXME: Proper error handling. 268b60a18deSMichael J. Spencer report_fatal_error("Symb must point to a valid symbol!"); 269b60a18deSMichael J. Spencer } 270b60a18deSMichael J. Spencer 271b60a18deSMichael J. Spencer template<support::endianness target_endianness, bool is64Bits> 272b60a18deSMichael J. Spencer SymbolRef ELFObjectFile<target_endianness, is64Bits> 273b60a18deSMichael J. Spencer ::getSymbolNext(DataRefImpl Symb) const { 274b60a18deSMichael J. Spencer validateSymbol(Symb); 2750324b672SMichael J. Spencer const Elf_Shdr *SymbolTableSection = SymbolTableSections[Symb.d.b]; 276b60a18deSMichael J. Spencer 2770324b672SMichael J. Spencer ++Symb.d.a; 278b60a18deSMichael J. Spencer // Check to see if we are at the end of this symbol table. 2790324b672SMichael J. Spencer if (Symb.d.a >= SymbolTableSection->getEntityCount()) { 280b60a18deSMichael J. Spencer // We are at the end. If there are other symbol tables, jump to them. 2810324b672SMichael J. Spencer ++Symb.d.b; 2820324b672SMichael J. Spencer Symb.d.a = 1; // The 0th symbol in ELF is fake. 283b60a18deSMichael J. Spencer // Otherwise return the terminator. 2840324b672SMichael J. Spencer if (Symb.d.b >= SymbolTableSections.size()) { 2850324b672SMichael J. Spencer Symb.d.a = std::numeric_limits<uint32_t>::max(); 2860324b672SMichael J. Spencer Symb.d.b = std::numeric_limits<uint32_t>::max(); 287b60a18deSMichael J. Spencer } 288b60a18deSMichael J. Spencer } 289b60a18deSMichael J. Spencer 290b60a18deSMichael J. Spencer return SymbolRef(Symb, this); 291b60a18deSMichael J. Spencer } 292b60a18deSMichael J. Spencer 293b60a18deSMichael J. Spencer template<support::endianness target_endianness, bool is64Bits> 294b60a18deSMichael J. Spencer StringRef ELFObjectFile<target_endianness, is64Bits> 295b60a18deSMichael J. Spencer ::getSymbolName(DataRefImpl Symb) const { 296b60a18deSMichael J. Spencer validateSymbol(Symb); 297b60a18deSMichael J. Spencer const Elf_Sym *symb = getSymbol(Symb); 298b60a18deSMichael J. Spencer if (symb->st_name == 0) { 299b60a18deSMichael J. Spencer const Elf_Shdr *section = getSection(symb->st_shndx); 300b60a18deSMichael J. Spencer if (!section) 301b60a18deSMichael J. Spencer return ""; 302b60a18deSMichael J. Spencer return getString(dot_shstrtab_sec, section->sh_name); 303b60a18deSMichael J. Spencer } 304b60a18deSMichael J. Spencer 305b60a18deSMichael J. Spencer // Use the default symbol table name section. 306b60a18deSMichael J. Spencer return getString(dot_strtab_sec, symb->st_name); 307b60a18deSMichael J. Spencer } 308b60a18deSMichael J. Spencer 309b60a18deSMichael J. Spencer template<support::endianness target_endianness, bool is64Bits> 310b60a18deSMichael J. Spencer uint64_t ELFObjectFile<target_endianness, is64Bits> 311b60a18deSMichael J. Spencer ::getSymbolAddress(DataRefImpl Symb) const { 312b60a18deSMichael J. Spencer validateSymbol(Symb); 313b60a18deSMichael J. Spencer const Elf_Sym *symb = getSymbol(Symb); 314b60a18deSMichael J. Spencer const Elf_Shdr *Section; 315b60a18deSMichael J. Spencer switch (symb->st_shndx) { 316b60a18deSMichael J. Spencer case ELF::SHN_COMMON: 317b60a18deSMichael J. Spencer // Undefined symbols have no address yet. 318b60a18deSMichael J. Spencer case ELF::SHN_UNDEF: return UnknownAddressOrSize; 319b60a18deSMichael J. Spencer case ELF::SHN_ABS: return symb->st_value; 320b60a18deSMichael J. Spencer default: Section = getSection(symb->st_shndx); 321b60a18deSMichael J. Spencer } 322b60a18deSMichael J. Spencer 323b60a18deSMichael J. Spencer switch (symb->getType()) { 324b60a18deSMichael J. Spencer case ELF::STT_SECTION: return Section ? Section->sh_addr 325b60a18deSMichael J. Spencer : UnknownAddressOrSize; 326b60a18deSMichael J. Spencer case ELF::STT_FUNC: 327b60a18deSMichael J. Spencer case ELF::STT_OBJECT: 328b60a18deSMichael J. Spencer case ELF::STT_NOTYPE: 329b60a18deSMichael J. Spencer return symb->st_value; 330b60a18deSMichael J. Spencer default: return UnknownAddressOrSize; 331b60a18deSMichael J. Spencer } 332b60a18deSMichael J. Spencer } 333b60a18deSMichael J. Spencer 334b60a18deSMichael J. Spencer template<support::endianness target_endianness, bool is64Bits> 335b60a18deSMichael J. Spencer uint64_t ELFObjectFile<target_endianness, is64Bits> 336b60a18deSMichael J. Spencer ::getSymbolSize(DataRefImpl Symb) const { 337b60a18deSMichael J. Spencer validateSymbol(Symb); 338b60a18deSMichael J. Spencer const Elf_Sym *symb = getSymbol(Symb); 339b60a18deSMichael J. Spencer if (symb->st_size == 0) 340b60a18deSMichael J. Spencer return UnknownAddressOrSize; 341b60a18deSMichael J. Spencer return symb->st_size; 342b60a18deSMichael J. Spencer } 343b60a18deSMichael J. Spencer 344b60a18deSMichael J. Spencer template<support::endianness target_endianness, bool is64Bits> 345b60a18deSMichael J. Spencer char ELFObjectFile<target_endianness, is64Bits> 346b60a18deSMichael J. Spencer ::getSymbolNMTypeChar(DataRefImpl Symb) const { 347b60a18deSMichael J. Spencer validateSymbol(Symb); 348b60a18deSMichael J. Spencer const Elf_Sym *symb = getSymbol(Symb); 349b60a18deSMichael J. Spencer const Elf_Shdr *Section = getSection(symb->st_shndx); 350b60a18deSMichael J. Spencer 351b60a18deSMichael J. Spencer char ret = '?'; 352b60a18deSMichael J. Spencer 353b60a18deSMichael J. Spencer if (Section) { 354b60a18deSMichael J. Spencer switch (Section->sh_type) { 355b60a18deSMichael J. Spencer case ELF::SHT_PROGBITS: 356b60a18deSMichael J. Spencer case ELF::SHT_DYNAMIC: 357b60a18deSMichael J. Spencer switch (Section->sh_flags) { 358b60a18deSMichael J. Spencer case (ELF::SHF_ALLOC | ELF::SHF_EXECINSTR): 359b60a18deSMichael J. Spencer ret = 't'; break; 360b60a18deSMichael J. Spencer case (ELF::SHF_ALLOC | ELF::SHF_WRITE): 361b60a18deSMichael J. Spencer ret = 'd'; break; 362b60a18deSMichael J. Spencer case ELF::SHF_ALLOC: 363b60a18deSMichael J. Spencer case (ELF::SHF_ALLOC | ELF::SHF_MERGE): 364b60a18deSMichael J. Spencer case (ELF::SHF_ALLOC | ELF::SHF_MERGE | ELF::SHF_STRINGS): 365b60a18deSMichael J. Spencer ret = 'r'; break; 366b60a18deSMichael J. Spencer } 367b60a18deSMichael J. Spencer break; 368b60a18deSMichael J. Spencer case ELF::SHT_NOBITS: ret = 'b'; 369b60a18deSMichael J. Spencer } 370b60a18deSMichael J. Spencer } 371b60a18deSMichael J. Spencer 372b60a18deSMichael J. Spencer switch (symb->st_shndx) { 373b60a18deSMichael J. Spencer case ELF::SHN_UNDEF: 374b60a18deSMichael J. Spencer if (ret == '?') 375b60a18deSMichael J. Spencer ret = 'U'; 376b60a18deSMichael J. Spencer break; 377b60a18deSMichael J. Spencer case ELF::SHN_ABS: ret = 'a'; break; 378b60a18deSMichael J. Spencer case ELF::SHN_COMMON: ret = 'c'; break; 379b60a18deSMichael J. Spencer } 380b60a18deSMichael J. Spencer 381b60a18deSMichael J. Spencer switch (symb->getBinding()) { 382b60a18deSMichael J. Spencer case ELF::STB_GLOBAL: ret = ::toupper(ret); break; 383b60a18deSMichael J. Spencer case ELF::STB_WEAK: 384b60a18deSMichael J. Spencer if (symb->st_shndx == ELF::SHN_UNDEF) 385b60a18deSMichael J. Spencer ret = 'w'; 386b60a18deSMichael J. Spencer else 387b60a18deSMichael J. Spencer if (symb->getType() == ELF::STT_OBJECT) 388b60a18deSMichael J. Spencer ret = 'V'; 389b60a18deSMichael J. Spencer else 390b60a18deSMichael J. Spencer ret = 'W'; 391b60a18deSMichael J. Spencer } 392b60a18deSMichael J. Spencer 393b60a18deSMichael J. Spencer if (ret == '?' && symb->getType() == ELF::STT_SECTION) 394b60a18deSMichael J. Spencer return StringSwitch<char>(getSymbolName(Symb)) 395b60a18deSMichael J. Spencer .StartsWith(".debug", 'N') 396b60a18deSMichael J. Spencer .StartsWith(".note", 'n'); 397b60a18deSMichael J. Spencer 398b60a18deSMichael J. Spencer return ret; 399b60a18deSMichael J. Spencer } 400b60a18deSMichael J. Spencer 401b60a18deSMichael J. Spencer template<support::endianness target_endianness, bool is64Bits> 402b60a18deSMichael J. Spencer bool ELFObjectFile<target_endianness, is64Bits> 403b60a18deSMichael J. Spencer ::isSymbolInternal(DataRefImpl Symb) const { 404b60a18deSMichael J. Spencer validateSymbol(Symb); 405b60a18deSMichael J. Spencer const Elf_Sym *symb = getSymbol(Symb); 406b60a18deSMichael J. Spencer 407b60a18deSMichael J. Spencer if ( symb->getType() == ELF::STT_FILE 408b60a18deSMichael J. Spencer || symb->getType() == ELF::STT_SECTION) 409b60a18deSMichael J. Spencer return true; 410b60a18deSMichael J. Spencer return false; 411b60a18deSMichael J. Spencer } 412b60a18deSMichael J. Spencer 413b60a18deSMichael J. Spencer template<support::endianness target_endianness, bool is64Bits> 414b60a18deSMichael J. Spencer SectionRef ELFObjectFile<target_endianness, is64Bits> 415b60a18deSMichael J. Spencer ::getSectionNext(DataRefImpl Sec) const { 4160324b672SMichael J. Spencer const uint8_t *sec = reinterpret_cast<const uint8_t *>(Sec.p); 417b60a18deSMichael J. Spencer sec += Header->e_shentsize; 4180324b672SMichael J. Spencer Sec.p = reinterpret_cast<intptr_t>(sec); 4190324b672SMichael J. Spencer return SectionRef(Sec, this); 420b60a18deSMichael J. Spencer } 421b60a18deSMichael J. Spencer 422b60a18deSMichael J. Spencer template<support::endianness target_endianness, bool is64Bits> 423b60a18deSMichael J. Spencer StringRef ELFObjectFile<target_endianness, is64Bits> 424b60a18deSMichael J. Spencer ::getSectionName(DataRefImpl Sec) const { 4250324b672SMichael J. Spencer const Elf_Shdr *sec = reinterpret_cast<const Elf_Shdr *>(Sec.p); 426b60a18deSMichael J. Spencer return StringRef(getString(dot_shstrtab_sec, sec->sh_name)); 427b60a18deSMichael J. Spencer } 428b60a18deSMichael J. Spencer 429b60a18deSMichael J. Spencer template<support::endianness target_endianness, bool is64Bits> 430b60a18deSMichael J. Spencer uint64_t ELFObjectFile<target_endianness, is64Bits> 431b60a18deSMichael J. Spencer ::getSectionAddress(DataRefImpl Sec) const { 4320324b672SMichael J. Spencer const Elf_Shdr *sec = reinterpret_cast<const Elf_Shdr *>(Sec.p); 433b60a18deSMichael J. Spencer return sec->sh_addr; 434b60a18deSMichael J. Spencer } 435b60a18deSMichael J. Spencer 436b60a18deSMichael J. Spencer template<support::endianness target_endianness, bool is64Bits> 437b60a18deSMichael J. Spencer uint64_t ELFObjectFile<target_endianness, is64Bits> 438b60a18deSMichael J. Spencer ::getSectionSize(DataRefImpl Sec) const { 4390324b672SMichael J. Spencer const Elf_Shdr *sec = reinterpret_cast<const Elf_Shdr *>(Sec.p); 440b60a18deSMichael J. Spencer return sec->sh_size; 441b60a18deSMichael J. Spencer } 442b60a18deSMichael J. Spencer 443b60a18deSMichael J. Spencer template<support::endianness target_endianness, bool is64Bits> 444b60a18deSMichael J. Spencer StringRef ELFObjectFile<target_endianness, is64Bits> 445b60a18deSMichael J. Spencer ::getSectionContents(DataRefImpl Sec) const { 4460324b672SMichael J. Spencer const Elf_Shdr *sec = reinterpret_cast<const Elf_Shdr *>(Sec.p); 447b60a18deSMichael J. Spencer const char *start = (char*)base + sec->sh_offset; 448b60a18deSMichael J. Spencer return StringRef(start, sec->sh_size); 449b60a18deSMichael J. Spencer } 450b60a18deSMichael J. Spencer 451b60a18deSMichael J. Spencer template<support::endianness target_endianness, bool is64Bits> 452b60a18deSMichael J. Spencer bool ELFObjectFile<target_endianness, is64Bits> 453b60a18deSMichael J. Spencer ::isSectionText(DataRefImpl Sec) const { 4540324b672SMichael J. Spencer const Elf_Shdr *sec = reinterpret_cast<const Elf_Shdr *>(Sec.p); 455b60a18deSMichael J. Spencer if (sec->sh_flags & ELF::SHF_EXECINSTR) 456b60a18deSMichael J. Spencer return true; 457b60a18deSMichael J. Spencer return false; 458b60a18deSMichael J. Spencer } 459b60a18deSMichael J. Spencer 460b60a18deSMichael J. Spencer template<support::endianness target_endianness, bool is64Bits> 461b60a18deSMichael J. Spencer ELFObjectFile<target_endianness, is64Bits>::ELFObjectFile(MemoryBuffer *Object) 462b60a18deSMichael J. Spencer : ObjectFile(Object) 463b60a18deSMichael J. Spencer , SectionHeaderTable(0) 464b60a18deSMichael J. Spencer , dot_shstrtab_sec(0) 465b60a18deSMichael J. Spencer , dot_strtab_sec(0) { 466b60a18deSMichael J. Spencer Header = reinterpret_cast<const Elf_Ehdr *>(base); 467b60a18deSMichael J. Spencer 468b60a18deSMichael J. Spencer if (Header->e_shoff == 0) 469b60a18deSMichael J. Spencer return; 470b60a18deSMichael J. Spencer 471b60a18deSMichael J. Spencer SectionHeaderTable = 472b60a18deSMichael J. Spencer reinterpret_cast<const Elf_Shdr *>(base + Header->e_shoff); 473b60a18deSMichael J. Spencer uint32_t SectionTableSize = Header->e_shnum * Header->e_shentsize; 474b60a18deSMichael J. Spencer if (!( (const uint8_t *)SectionHeaderTable + SectionTableSize 475b60a18deSMichael J. Spencer <= base + MapFile->getBufferSize())) 476b60a18deSMichael J. Spencer // FIXME: Proper error handling. 477b60a18deSMichael J. Spencer report_fatal_error("Section table goes past end of file!"); 478b60a18deSMichael J. Spencer 479b60a18deSMichael J. Spencer 480b60a18deSMichael J. Spencer // To find the symbol tables we walk the section table to find SHT_STMTAB. 481b60a18deSMichael J. Spencer for (const char *i = reinterpret_cast<const char *>(SectionHeaderTable), 482b60a18deSMichael J. Spencer *e = i + Header->e_shnum * Header->e_shentsize; 483b60a18deSMichael J. Spencer i != e; i += Header->e_shentsize) { 484b60a18deSMichael J. Spencer const Elf_Shdr *sh = reinterpret_cast<const Elf_Shdr*>(i); 485b60a18deSMichael J. Spencer if (sh->sh_type == ELF::SHT_SYMTAB) { 486b60a18deSMichael J. Spencer SymbolTableSections.push_back(sh); 487b60a18deSMichael J. Spencer } 488b60a18deSMichael J. Spencer } 489b60a18deSMichael J. Spencer 490b60a18deSMichael J. Spencer // Get string table sections. 491b60a18deSMichael J. Spencer dot_shstrtab_sec = getSection(Header->e_shstrndx); 492b60a18deSMichael J. Spencer if (dot_shstrtab_sec) { 493b60a18deSMichael J. Spencer // Verify that the last byte in the string table in a null. 494b60a18deSMichael J. Spencer if (((const char*)base + dot_shstrtab_sec->sh_offset) 495b60a18deSMichael J. Spencer [dot_shstrtab_sec->sh_size - 1] != 0) 496b60a18deSMichael J. Spencer // FIXME: Proper error handling. 497b60a18deSMichael J. Spencer report_fatal_error("String table must end with a null terminator!"); 498b60a18deSMichael J. Spencer } 499b60a18deSMichael J. Spencer 500b60a18deSMichael J. Spencer // Merge this into the above loop. 501b60a18deSMichael J. Spencer for (const char *i = reinterpret_cast<const char *>(SectionHeaderTable), 502b60a18deSMichael J. Spencer *e = i + Header->e_shnum * Header->e_shentsize; 503b60a18deSMichael J. Spencer i != e; i += Header->e_shentsize) { 504b60a18deSMichael J. Spencer const Elf_Shdr *sh = reinterpret_cast<const Elf_Shdr*>(i); 505b60a18deSMichael J. Spencer if (sh->sh_type == ELF::SHT_STRTAB) { 506b60a18deSMichael J. Spencer StringRef SectionName(getString(dot_shstrtab_sec, sh->sh_name)); 507b60a18deSMichael J. Spencer if (SectionName == ".strtab") { 508b60a18deSMichael J. Spencer if (dot_strtab_sec != 0) 509b60a18deSMichael J. Spencer // FIXME: Proper error handling. 510b60a18deSMichael J. Spencer report_fatal_error("Already found section named .strtab!"); 511b60a18deSMichael J. Spencer dot_strtab_sec = sh; 512b60a18deSMichael J. Spencer const char *dot_strtab = (const char*)base + sh->sh_offset; 513b60a18deSMichael J. Spencer if (dot_strtab[sh->sh_size - 1] != 0) 514b60a18deSMichael J. Spencer // FIXME: Proper error handling. 515b60a18deSMichael J. Spencer report_fatal_error("String table must end with a null terminator!"); 516b60a18deSMichael J. Spencer } 517b60a18deSMichael J. Spencer } 518b60a18deSMichael J. Spencer } 519b60a18deSMichael J. Spencer } 520b60a18deSMichael J. Spencer 521b60a18deSMichael J. Spencer template<support::endianness target_endianness, bool is64Bits> 522b60a18deSMichael J. Spencer ObjectFile::symbol_iterator ELFObjectFile<target_endianness, is64Bits> 523b60a18deSMichael J. Spencer ::begin_symbols() const { 5240324b672SMichael J. Spencer DataRefImpl SymbolData; 525b60a18deSMichael J. Spencer memset(&SymbolData, 0, sizeof(SymbolData)); 526b60a18deSMichael J. Spencer if (SymbolTableSections.size() == 0) { 5270324b672SMichael J. Spencer SymbolData.d.a = std::numeric_limits<uint32_t>::max(); 5280324b672SMichael J. Spencer SymbolData.d.b = std::numeric_limits<uint32_t>::max(); 529b60a18deSMichael J. Spencer } else { 5300324b672SMichael J. Spencer SymbolData.d.a = 1; // The 0th symbol in ELF is fake. 5310324b672SMichael J. Spencer SymbolData.d.b = 0; 532b60a18deSMichael J. Spencer } 5330324b672SMichael J. Spencer return symbol_iterator(SymbolRef(SymbolData, this)); 534b60a18deSMichael J. Spencer } 535b60a18deSMichael J. Spencer 536b60a18deSMichael J. Spencer template<support::endianness target_endianness, bool is64Bits> 537b60a18deSMichael J. Spencer ObjectFile::symbol_iterator ELFObjectFile<target_endianness, is64Bits> 538b60a18deSMichael J. Spencer ::end_symbols() const { 5390324b672SMichael J. Spencer DataRefImpl SymbolData; 540b60a18deSMichael J. Spencer memset(&SymbolData, 0, sizeof(SymbolData)); 5410324b672SMichael J. Spencer SymbolData.d.a = std::numeric_limits<uint32_t>::max(); 5420324b672SMichael J. Spencer SymbolData.d.b = std::numeric_limits<uint32_t>::max(); 5430324b672SMichael J. Spencer return symbol_iterator(SymbolRef(SymbolData, this)); 544b60a18deSMichael J. Spencer } 545b60a18deSMichael J. Spencer 546b60a18deSMichael J. Spencer template<support::endianness target_endianness, bool is64Bits> 547b60a18deSMichael J. Spencer ObjectFile::section_iterator ELFObjectFile<target_endianness, is64Bits> 548b60a18deSMichael J. Spencer ::begin_sections() const { 5490324b672SMichael J. Spencer DataRefImpl ret; 550*ee066fc4SEric Christopher memset(&ret, 0, sizeof(DataRefImpl)); 5510324b672SMichael J. Spencer ret.p = reinterpret_cast<intptr_t>(base + Header->e_shoff); 5520324b672SMichael J. Spencer return section_iterator(SectionRef(ret, this)); 553b60a18deSMichael J. Spencer } 554b60a18deSMichael J. Spencer 555b60a18deSMichael J. Spencer template<support::endianness target_endianness, bool is64Bits> 556b60a18deSMichael J. Spencer ObjectFile::section_iterator ELFObjectFile<target_endianness, is64Bits> 557b60a18deSMichael J. Spencer ::end_sections() const { 5580324b672SMichael J. Spencer DataRefImpl ret; 559*ee066fc4SEric Christopher memset(&ret, 0, sizeof(DataRefImpl)); 5600324b672SMichael J. Spencer ret.p = reinterpret_cast<intptr_t>(base 561b60a18deSMichael J. Spencer + Header->e_shoff 5620324b672SMichael J. Spencer + (Header->e_shentsize * Header->e_shnum)); 5630324b672SMichael J. Spencer return section_iterator(SectionRef(ret, this)); 564b60a18deSMichael J. Spencer } 565b60a18deSMichael J. Spencer 566b60a18deSMichael J. Spencer template<support::endianness target_endianness, bool is64Bits> 567b60a18deSMichael J. Spencer uint8_t ELFObjectFile<target_endianness, is64Bits>::getBytesInAddress() const { 5680324b672SMichael J. Spencer return is64Bits ? 8 : 4; 569b60a18deSMichael J. Spencer } 570b60a18deSMichael J. Spencer 571b60a18deSMichael J. Spencer template<support::endianness target_endianness, bool is64Bits> 572b60a18deSMichael J. Spencer StringRef ELFObjectFile<target_endianness, is64Bits> 573b60a18deSMichael J. Spencer ::getFileFormatName() const { 574b60a18deSMichael J. Spencer switch(Header->e_ident[ELF::EI_CLASS]) { 575b60a18deSMichael J. Spencer case ELF::ELFCLASS32: 576b60a18deSMichael J. Spencer switch(Header->e_machine) { 577b60a18deSMichael J. Spencer case ELF::EM_386: 578b60a18deSMichael J. Spencer return "ELF32-i386"; 579b60a18deSMichael J. Spencer case ELF::EM_X86_64: 580b60a18deSMichael J. Spencer return "ELF32-x86-64"; 581b60a18deSMichael J. Spencer default: 582b60a18deSMichael J. Spencer return "ELF32-unknown"; 583b60a18deSMichael J. Spencer } 584b60a18deSMichael J. Spencer case ELF::ELFCLASS64: 585b60a18deSMichael J. Spencer switch(Header->e_machine) { 586b60a18deSMichael J. Spencer case ELF::EM_386: 587b60a18deSMichael J. Spencer return "ELF64-i386"; 588b60a18deSMichael J. Spencer case ELF::EM_X86_64: 589b60a18deSMichael J. Spencer return "ELF64-x86-64"; 590b60a18deSMichael J. Spencer default: 591b60a18deSMichael J. Spencer return "ELF64-unknown"; 592b60a18deSMichael J. Spencer } 593b60a18deSMichael J. Spencer default: 594b60a18deSMichael J. Spencer // FIXME: Proper error handling. 595b60a18deSMichael J. Spencer report_fatal_error("Invalid ELFCLASS!"); 596b60a18deSMichael J. Spencer } 597b60a18deSMichael J. Spencer } 598b60a18deSMichael J. Spencer 599b60a18deSMichael J. Spencer template<support::endianness target_endianness, bool is64Bits> 600b60a18deSMichael J. Spencer unsigned ELFObjectFile<target_endianness, is64Bits>::getArch() const { 601b60a18deSMichael J. Spencer switch(Header->e_machine) { 602b60a18deSMichael J. Spencer case ELF::EM_386: 603b60a18deSMichael J. Spencer return Triple::x86; 604b60a18deSMichael J. Spencer case ELF::EM_X86_64: 605b60a18deSMichael J. Spencer return Triple::x86_64; 606b60a18deSMichael J. Spencer default: 607b60a18deSMichael J. Spencer return Triple::UnknownArch; 608b60a18deSMichael J. Spencer } 609b60a18deSMichael J. Spencer } 610b60a18deSMichael J. Spencer 611b60a18deSMichael J. Spencer template<support::endianness target_endianness, bool is64Bits> 612b60a18deSMichael J. Spencer const typename ELFObjectFile<target_endianness, is64Bits>::Elf_Sym * 613b60a18deSMichael J. Spencer ELFObjectFile<target_endianness, is64Bits>::getSymbol(DataRefImpl Symb) const { 6140324b672SMichael J. Spencer const Elf_Shdr *sec = SymbolTableSections[Symb.d.b]; 615b60a18deSMichael J. Spencer return reinterpret_cast<const Elf_Sym *>( 616b60a18deSMichael J. Spencer base 617b60a18deSMichael J. Spencer + sec->sh_offset 6180324b672SMichael J. Spencer + (Symb.d.a * sec->sh_entsize)); 619b60a18deSMichael J. Spencer } 620b60a18deSMichael J. Spencer 621b60a18deSMichael J. Spencer template<support::endianness target_endianness, bool is64Bits> 622b60a18deSMichael J. Spencer const typename ELFObjectFile<target_endianness, is64Bits>::Elf_Shdr * 623b60a18deSMichael J. Spencer ELFObjectFile<target_endianness, is64Bits>::getSection(DataRefImpl Symb) const { 6240324b672SMichael J. Spencer const Elf_Shdr *sec = getSection(Symb.d.b); 625b60a18deSMichael J. Spencer if (sec->sh_type != ELF::SHT_SYMTAB) 626b60a18deSMichael J. Spencer // FIXME: Proper error handling. 627b60a18deSMichael J. Spencer report_fatal_error("Invalid symbol table section!"); 628b60a18deSMichael J. Spencer return sec; 629b60a18deSMichael J. Spencer } 630b60a18deSMichael J. Spencer 631b60a18deSMichael J. Spencer template<support::endianness target_endianness, bool is64Bits> 632b60a18deSMichael J. Spencer const typename ELFObjectFile<target_endianness, is64Bits>::Elf_Shdr * 633b60a18deSMichael J. Spencer ELFObjectFile<target_endianness, is64Bits>::getSection(uint16_t index) const { 634b60a18deSMichael J. Spencer if (index == 0 || index >= ELF::SHN_LORESERVE) 635b60a18deSMichael J. Spencer return 0; 636b60a18deSMichael J. Spencer if (!SectionHeaderTable || index >= Header->e_shnum) 637b60a18deSMichael J. Spencer // FIXME: Proper error handling. 638b60a18deSMichael J. Spencer report_fatal_error("Invalid section index!"); 639b60a18deSMichael J. Spencer 640b60a18deSMichael J. Spencer return reinterpret_cast<const Elf_Shdr *>( 641b60a18deSMichael J. Spencer reinterpret_cast<const char *>(SectionHeaderTable) 642b60a18deSMichael J. Spencer + (index * Header->e_shentsize)); 643b60a18deSMichael J. Spencer } 644b60a18deSMichael J. Spencer 645b60a18deSMichael J. Spencer template<support::endianness target_endianness, bool is64Bits> 646b60a18deSMichael J. Spencer const char *ELFObjectFile<target_endianness, is64Bits> 647b60a18deSMichael J. Spencer ::getString(uint16_t section, 648b60a18deSMichael J. Spencer ELF::Elf32_Word offset) const { 649b60a18deSMichael J. Spencer return getString(getSection(section), offset); 650b60a18deSMichael J. Spencer } 651b60a18deSMichael J. Spencer 652b60a18deSMichael J. Spencer template<support::endianness target_endianness, bool is64Bits> 653b60a18deSMichael J. Spencer const char *ELFObjectFile<target_endianness, is64Bits> 654b60a18deSMichael J. Spencer ::getString(const Elf_Shdr *section, 655b60a18deSMichael J. Spencer ELF::Elf32_Word offset) const { 656b60a18deSMichael J. Spencer assert(section && section->sh_type == ELF::SHT_STRTAB && "Invalid section!"); 657b60a18deSMichael J. Spencer if (offset >= section->sh_size) 658b60a18deSMichael J. Spencer // FIXME: Proper error handling. 659b60a18deSMichael J. Spencer report_fatal_error("Sybol name offset outside of string table!"); 660b60a18deSMichael J. Spencer return (const char *)base + section->sh_offset + offset; 661b60a18deSMichael J. Spencer } 662b60a18deSMichael J. Spencer 663b60a18deSMichael J. Spencer // EI_CLASS, EI_DATA. 664b60a18deSMichael J. Spencer static std::pair<unsigned char, unsigned char> 665b60a18deSMichael J. Spencer getElfArchType(MemoryBuffer *Object) { 666b60a18deSMichael J. Spencer if (Object->getBufferSize() < ELF::EI_NIDENT) 667b60a18deSMichael J. Spencer return std::make_pair((uint8_t)ELF::ELFCLASSNONE,(uint8_t)ELF::ELFDATANONE); 668b60a18deSMichael J. Spencer return std::make_pair( (uint8_t)Object->getBufferStart()[ELF::EI_CLASS] 669b60a18deSMichael J. Spencer , (uint8_t)Object->getBufferStart()[ELF::EI_DATA]); 670b60a18deSMichael J. Spencer } 671b60a18deSMichael J. Spencer 672b60a18deSMichael J. Spencer namespace llvm { 673b60a18deSMichael J. Spencer 674b60a18deSMichael J. Spencer ObjectFile *ObjectFile::createELFObjectFile(MemoryBuffer *Object) { 675b60a18deSMichael J. Spencer std::pair<unsigned char, unsigned char> Ident = getElfArchType(Object); 676b60a18deSMichael J. Spencer if (Ident.first == ELF::ELFCLASS32 && Ident.second == ELF::ELFDATA2LSB) 677b60a18deSMichael J. Spencer return new ELFObjectFile<support::little, false>(Object); 678b60a18deSMichael J. Spencer else if (Ident.first == ELF::ELFCLASS32 && Ident.second == ELF::ELFDATA2MSB) 679b60a18deSMichael J. Spencer return new ELFObjectFile<support::big, false>(Object); 680b60a18deSMichael J. Spencer else if (Ident.first == ELF::ELFCLASS64 && Ident.second == ELF::ELFDATA2LSB) 681b60a18deSMichael J. Spencer return new ELFObjectFile<support::little, true>(Object); 682b60a18deSMichael J. Spencer else if (Ident.first == ELF::ELFCLASS64 && Ident.second == ELF::ELFDATA2MSB) 683b60a18deSMichael J. Spencer return new ELFObjectFile<support::big, true>(Object); 684b60a18deSMichael J. Spencer // FIXME: Proper error handling. 685b60a18deSMichael J. Spencer report_fatal_error("Not an ELF object file!"); 686b60a18deSMichael J. Spencer } 687b60a18deSMichael J. Spencer 688b60a18deSMichael J. Spencer } // end namespace llvm 689