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