• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 //===- ELFObjectFile.cpp - ELF object file implementation -------*- C++ -*-===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // This file defines the ELFObjectFile class.
11 //
12 //===----------------------------------------------------------------------===//
13 
14 #include "llvm/ADT/SmallVector.h"
15 #include "llvm/ADT/StringSwitch.h"
16 #include "llvm/ADT/Triple.h"
17 #include "llvm/ADT/DenseMap.h"
18 #include "llvm/Object/ObjectFile.h"
19 #include "llvm/Support/ELF.h"
20 #include "llvm/Support/Endian.h"
21 #include "llvm/Support/ErrorHandling.h"
22 #include "llvm/Support/MemoryBuffer.h"
23 #include "llvm/Support/raw_ostream.h"
24 #include <algorithm>
25 #include <limits>
26 #include <utility>
27 
28 using namespace llvm;
29 using namespace object;
30 
31 // Templates to choose Elf_Addr and Elf_Off depending on is64Bits.
32 namespace {
33 template<support::endianness target_endianness>
34 struct ELFDataTypeTypedefHelperCommon {
35   typedef support::detail::packed_endian_specific_integral
36     <uint16_t, target_endianness, support::aligned> Elf_Half;
37   typedef support::detail::packed_endian_specific_integral
38     <uint32_t, target_endianness, support::aligned> Elf_Word;
39   typedef support::detail::packed_endian_specific_integral
40     <int32_t, target_endianness, support::aligned> Elf_Sword;
41   typedef support::detail::packed_endian_specific_integral
42     <uint64_t, target_endianness, support::aligned> Elf_Xword;
43   typedef support::detail::packed_endian_specific_integral
44     <int64_t, target_endianness, support::aligned> Elf_Sxword;
45 };
46 }
47 
48 namespace {
49 template<support::endianness target_endianness, bool is64Bits>
50 struct ELFDataTypeTypedefHelper;
51 
52 /// ELF 32bit types.
53 template<support::endianness target_endianness>
54 struct ELFDataTypeTypedefHelper<target_endianness, false>
55   : ELFDataTypeTypedefHelperCommon<target_endianness> {
56   typedef support::detail::packed_endian_specific_integral
57     <uint32_t, target_endianness, support::aligned> Elf_Addr;
58   typedef support::detail::packed_endian_specific_integral
59     <uint32_t, target_endianness, support::aligned> Elf_Off;
60 };
61 
62 /// ELF 64bit types.
63 template<support::endianness target_endianness>
64 struct ELFDataTypeTypedefHelper<target_endianness, true>
65   : ELFDataTypeTypedefHelperCommon<target_endianness>{
66   typedef support::detail::packed_endian_specific_integral
67     <uint64_t, target_endianness, support::aligned> Elf_Addr;
68   typedef support::detail::packed_endian_specific_integral
69     <uint64_t, target_endianness, support::aligned> Elf_Off;
70 };
71 }
72 
73 // I really don't like doing this, but the alternative is copypasta.
74 #define LLVM_ELF_IMPORT_TYPES(target_endianness, is64Bits) \
75 typedef typename \
76   ELFDataTypeTypedefHelper<target_endianness, is64Bits>::Elf_Addr Elf_Addr; \
77 typedef typename \
78   ELFDataTypeTypedefHelper<target_endianness, is64Bits>::Elf_Off Elf_Off; \
79 typedef typename \
80   ELFDataTypeTypedefHelper<target_endianness, is64Bits>::Elf_Half Elf_Half; \
81 typedef typename \
82   ELFDataTypeTypedefHelper<target_endianness, is64Bits>::Elf_Word Elf_Word; \
83 typedef typename \
84   ELFDataTypeTypedefHelper<target_endianness, is64Bits>::Elf_Sword Elf_Sword; \
85 typedef typename \
86   ELFDataTypeTypedefHelper<target_endianness, is64Bits>::Elf_Xword Elf_Xword; \
87 typedef typename \
88   ELFDataTypeTypedefHelper<target_endianness, is64Bits>::Elf_Sxword Elf_Sxword;
89 
90   // Section header.
91 namespace {
92 template<support::endianness target_endianness, bool is64Bits>
93 struct Elf_Shdr_Base;
94 
95 template<support::endianness target_endianness>
96 struct Elf_Shdr_Base<target_endianness, false> {
97   LLVM_ELF_IMPORT_TYPES(target_endianness, false)
98   Elf_Word sh_name;     // Section name (index into string table)
99   Elf_Word sh_type;     // Section type (SHT_*)
100   Elf_Word sh_flags;    // Section flags (SHF_*)
101   Elf_Addr sh_addr;     // Address where section is to be loaded
102   Elf_Off  sh_offset;   // File offset of section data, in bytes
103   Elf_Word sh_size;     // Size of section, in bytes
104   Elf_Word sh_link;     // Section type-specific header table index link
105   Elf_Word sh_info;     // Section type-specific extra information
106   Elf_Word sh_addralign;// Section address alignment
107   Elf_Word sh_entsize;  // Size of records contained within the section
108 };
109 
110 template<support::endianness target_endianness>
111 struct Elf_Shdr_Base<target_endianness, true> {
112   LLVM_ELF_IMPORT_TYPES(target_endianness, true)
113   Elf_Word  sh_name;     // Section name (index into string table)
114   Elf_Word  sh_type;     // Section type (SHT_*)
115   Elf_Xword sh_flags;    // Section flags (SHF_*)
116   Elf_Addr  sh_addr;     // Address where section is to be loaded
117   Elf_Off   sh_offset;   // File offset of section data, in bytes
118   Elf_Xword sh_size;     // Size of section, in bytes
119   Elf_Word  sh_link;     // Section type-specific header table index link
120   Elf_Word  sh_info;     // Section type-specific extra information
121   Elf_Xword sh_addralign;// Section address alignment
122   Elf_Xword sh_entsize;  // Size of records contained within the section
123 };
124 
125 template<support::endianness target_endianness, bool is64Bits>
126 struct Elf_Shdr_Impl : Elf_Shdr_Base<target_endianness, is64Bits> {
127   using Elf_Shdr_Base<target_endianness, is64Bits>::sh_entsize;
128   using Elf_Shdr_Base<target_endianness, is64Bits>::sh_size;
129 
130   /// @brief Get the number of entities this section contains if it has any.
getEntityCount__anonb69d08910311::Elf_Shdr_Impl131   unsigned getEntityCount() const {
132     if (sh_entsize == 0)
133       return 0;
134     return sh_size / sh_entsize;
135   }
136 };
137 }
138 
139 namespace {
140 template<support::endianness target_endianness, bool is64Bits>
141 struct Elf_Sym_Base;
142 
143 template<support::endianness target_endianness>
144 struct Elf_Sym_Base<target_endianness, false> {
145   LLVM_ELF_IMPORT_TYPES(target_endianness, false)
146   Elf_Word      st_name;  // Symbol name (index into string table)
147   Elf_Addr      st_value; // Value or address associated with the symbol
148   Elf_Word      st_size;  // Size of the symbol
149   unsigned char st_info;  // Symbol's type and binding attributes
150   unsigned char st_other; // Must be zero; reserved
151   Elf_Half      st_shndx; // Which section (header table index) it's defined in
152 };
153 
154 template<support::endianness target_endianness>
155 struct Elf_Sym_Base<target_endianness, true> {
156   LLVM_ELF_IMPORT_TYPES(target_endianness, true)
157   Elf_Word      st_name;  // Symbol name (index into string table)
158   unsigned char st_info;  // Symbol's type and binding attributes
159   unsigned char st_other; // Must be zero; reserved
160   Elf_Half      st_shndx; // Which section (header table index) it's defined in
161   Elf_Addr      st_value; // Value or address associated with the symbol
162   Elf_Xword     st_size;  // Size of the symbol
163 };
164 
165 template<support::endianness target_endianness, bool is64Bits>
166 struct Elf_Sym_Impl : Elf_Sym_Base<target_endianness, is64Bits> {
167   using Elf_Sym_Base<target_endianness, is64Bits>::st_info;
168 
169   // These accessors and mutators correspond to the ELF32_ST_BIND,
170   // ELF32_ST_TYPE, and ELF32_ST_INFO macros defined in the ELF specification:
getBinding__anonb69d08910411::Elf_Sym_Impl171   unsigned char getBinding() const { return st_info >> 4; }
getType__anonb69d08910411::Elf_Sym_Impl172   unsigned char getType() const { return st_info & 0x0f; }
setBinding__anonb69d08910411::Elf_Sym_Impl173   void setBinding(unsigned char b) { setBindingAndType(b, getType()); }
setType__anonb69d08910411::Elf_Sym_Impl174   void setType(unsigned char t) { setBindingAndType(getBinding(), t); }
setBindingAndType__anonb69d08910411::Elf_Sym_Impl175   void setBindingAndType(unsigned char b, unsigned char t) {
176     st_info = (b << 4) + (t & 0x0f);
177   }
178 };
179 }
180 
181 namespace {
182 template<support::endianness target_endianness, bool is64Bits, bool isRela>
183 struct Elf_Rel_Base;
184 
185 template<support::endianness target_endianness>
186 struct Elf_Rel_Base<target_endianness, false, false> {
187   LLVM_ELF_IMPORT_TYPES(target_endianness, false)
188   Elf_Addr      r_offset; // Location (file byte offset, or program virtual addr)
189   Elf_Word      r_info;  // Symbol table index and type of relocation to apply
190 };
191 
192 template<support::endianness target_endianness>
193 struct Elf_Rel_Base<target_endianness, true, false> {
194   LLVM_ELF_IMPORT_TYPES(target_endianness, true)
195   Elf_Addr      r_offset; // Location (file byte offset, or program virtual addr)
196   Elf_Xword     r_info;   // Symbol table index and type of relocation to apply
197 };
198 
199 template<support::endianness target_endianness>
200 struct Elf_Rel_Base<target_endianness, false, true> {
201   LLVM_ELF_IMPORT_TYPES(target_endianness, false)
202   Elf_Addr      r_offset; // Location (file byte offset, or program virtual addr)
203   Elf_Word      r_info;   // Symbol table index and type of relocation to apply
204   Elf_Sword     r_addend; // Compute value for relocatable field by adding this
205 };
206 
207 template<support::endianness target_endianness>
208 struct Elf_Rel_Base<target_endianness, true, true> {
209   LLVM_ELF_IMPORT_TYPES(target_endianness, true)
210   Elf_Addr      r_offset; // Location (file byte offset, or program virtual addr)
211   Elf_Xword     r_info;   // Symbol table index and type of relocation to apply
212   Elf_Sxword    r_addend; // Compute value for relocatable field by adding this.
213 };
214 
215 template<support::endianness target_endianness, bool is64Bits, bool isRela>
216 struct Elf_Rel_Impl;
217 
218 template<support::endianness target_endianness, bool isRela>
219 struct Elf_Rel_Impl<target_endianness, true, isRela>
220        : Elf_Rel_Base<target_endianness, true, isRela> {
221   using Elf_Rel_Base<target_endianness, true, isRela>::r_info;
LLVM_ELF_IMPORT_TYPES__anonb69d08910511::Elf_Rel_Impl222   LLVM_ELF_IMPORT_TYPES(target_endianness, true)
223 
224   // These accessors and mutators correspond to the ELF64_R_SYM, ELF64_R_TYPE,
225   // and ELF64_R_INFO macros defined in the ELF specification:
226   uint64_t getSymbol() const { return (r_info >> 32); }
getType__anonb69d08910511::Elf_Rel_Impl227   unsigned char getType() const {
228     return (unsigned char) (r_info & 0xffffffffL);
229   }
setSymbol__anonb69d08910511::Elf_Rel_Impl230   void setSymbol(uint64_t s) { setSymbolAndType(s, getType()); }
setType__anonb69d08910511::Elf_Rel_Impl231   void setType(unsigned char t) { setSymbolAndType(getSymbol(), t); }
setSymbolAndType__anonb69d08910511::Elf_Rel_Impl232   void setSymbolAndType(uint64_t s, unsigned char t) {
233     r_info = (s << 32) + (t&0xffffffffL);
234   }
235 };
236 
237 template<support::endianness target_endianness, bool isRela>
238 struct Elf_Rel_Impl<target_endianness, false, isRela>
239        : Elf_Rel_Base<target_endianness, false, isRela> {
240   using Elf_Rel_Base<target_endianness, false, isRela>::r_info;
LLVM_ELF_IMPORT_TYPES__anonb69d08910511::Elf_Rel_Impl241   LLVM_ELF_IMPORT_TYPES(target_endianness, false)
242 
243   // These accessors and mutators correspond to the ELF32_R_SYM, ELF32_R_TYPE,
244   // and ELF32_R_INFO macros defined in the ELF specification:
245   uint32_t getSymbol() const { return (r_info >> 8); }
getType__anonb69d08910511::Elf_Rel_Impl246   unsigned char getType() const { return (unsigned char) (r_info & 0x0ff); }
setSymbol__anonb69d08910511::Elf_Rel_Impl247   void setSymbol(uint32_t s) { setSymbolAndType(s, getType()); }
setType__anonb69d08910511::Elf_Rel_Impl248   void setType(unsigned char t) { setSymbolAndType(getSymbol(), t); }
setSymbolAndType__anonb69d08910511::Elf_Rel_Impl249   void setSymbolAndType(uint32_t s, unsigned char t) {
250     r_info = (s << 8) + t;
251   }
252 };
253 
254 }
255 
256 namespace {
257 template<support::endianness target_endianness, bool is64Bits>
258 class ELFObjectFile : public ObjectFile {
259   LLVM_ELF_IMPORT_TYPES(target_endianness, is64Bits)
260 
261   typedef Elf_Shdr_Impl<target_endianness, is64Bits> Elf_Shdr;
262   typedef Elf_Sym_Impl<target_endianness, is64Bits> Elf_Sym;
263   typedef Elf_Rel_Impl<target_endianness, is64Bits, false> Elf_Rel;
264   typedef Elf_Rel_Impl<target_endianness, is64Bits, true> Elf_Rela;
265 
266   struct Elf_Ehdr {
267     unsigned char e_ident[ELF::EI_NIDENT]; // ELF Identification bytes
268     Elf_Half e_type;     // Type of file (see ET_*)
269     Elf_Half e_machine;  // Required architecture for this file (see EM_*)
270     Elf_Word e_version;  // Must be equal to 1
271     Elf_Addr e_entry;    // Address to jump to in order to start program
272     Elf_Off  e_phoff;    // Program header table's file offset, in bytes
273     Elf_Off  e_shoff;    // Section header table's file offset, in bytes
274     Elf_Word e_flags;    // Processor-specific flags
275     Elf_Half e_ehsize;   // Size of ELF header, in bytes
276     Elf_Half e_phentsize;// Size of an entry in the program header table
277     Elf_Half e_phnum;    // Number of entries in the program header table
278     Elf_Half e_shentsize;// Size of an entry in the section header table
279     Elf_Half e_shnum;    // Number of entries in the section header table
280     Elf_Half e_shstrndx; // Section header table index of section name
281                                   // string table
checkMagic__anonb69d08910611::ELFObjectFile::Elf_Ehdr282     bool checkMagic() const {
283       return (memcmp(e_ident, ELF::ElfMagic, strlen(ELF::ElfMagic))) == 0;
284     }
getFileClass__anonb69d08910611::ELFObjectFile::Elf_Ehdr285     unsigned char getFileClass() const { return e_ident[ELF::EI_CLASS]; }
getDataEncoding__anonb69d08910611::ELFObjectFile::Elf_Ehdr286     unsigned char getDataEncoding() const { return e_ident[ELF::EI_DATA]; }
287   };
288 
289   typedef SmallVector<const Elf_Shdr*, 1> Sections_t;
290   typedef DenseMap<unsigned, unsigned> IndexMap_t;
291   typedef DenseMap<const Elf_Shdr*, SmallVector<uint32_t, 1> > RelocMap_t;
292 
293   const Elf_Ehdr *Header;
294   const Elf_Shdr *SectionHeaderTable;
295   const Elf_Shdr *dot_shstrtab_sec; // Section header string table.
296   const Elf_Shdr *dot_strtab_sec;   // Symbol header string table.
297   Sections_t SymbolTableSections;
298   IndexMap_t SymbolTableSectionsIndexMap;
299   DenseMap<const Elf_Sym*, ELF::Elf64_Word> ExtendedSymbolTable;
300 
301   /// @brief Map sections to an array of relocation sections that reference
302   ///        them sorted by section index.
303   RelocMap_t SectionRelocMap;
304 
305   /// @brief Get the relocation section that contains \a Rel.
getRelSection(DataRefImpl Rel) const306   const Elf_Shdr *getRelSection(DataRefImpl Rel) const {
307     return getSection(Rel.w.b);
308   }
309 
310   void            validateSymbol(DataRefImpl Symb) const;
311   bool            isRelocationHasAddend(DataRefImpl Rel) const;
312   template<typename T>
313   const T        *getEntry(uint16_t Section, uint32_t Entry) const;
314   template<typename T>
315   const T        *getEntry(const Elf_Shdr *Section, uint32_t Entry) const;
316   const Elf_Sym  *getSymbol(DataRefImpl Symb) const;
317   const Elf_Shdr *getSection(DataRefImpl index) const;
318   const Elf_Shdr *getSection(uint32_t index) const;
319   const Elf_Rel  *getRel(DataRefImpl Rel) const;
320   const Elf_Rela *getRela(DataRefImpl Rela) const;
321   const char     *getString(uint32_t section, uint32_t offset) const;
322   const char     *getString(const Elf_Shdr *section, uint32_t offset) const;
323   error_code      getSymbolName(const Elf_Sym *Symb, StringRef &Res) const;
324 
325 protected:
326   virtual error_code getSymbolNext(DataRefImpl Symb, SymbolRef &Res) const;
327   virtual error_code getSymbolName(DataRefImpl Symb, StringRef &Res) const;
328   virtual error_code getSymbolOffset(DataRefImpl Symb, uint64_t &Res) const;
329   virtual error_code getSymbolAddress(DataRefImpl Symb, uint64_t &Res) const;
330   virtual error_code getSymbolSize(DataRefImpl Symb, uint64_t &Res) const;
331   virtual error_code getSymbolNMTypeChar(DataRefImpl Symb, char &Res) const;
332   virtual error_code isSymbolInternal(DataRefImpl Symb, bool &Res) const;
333   virtual error_code isSymbolGlobal(DataRefImpl Symb, bool &Res) const;
334   virtual error_code getSymbolType(DataRefImpl Symb, SymbolRef::SymbolType &Res) const;
335 
336   virtual error_code getSectionNext(DataRefImpl Sec, SectionRef &Res) const;
337   virtual error_code getSectionName(DataRefImpl Sec, StringRef &Res) const;
338   virtual error_code getSectionAddress(DataRefImpl Sec, uint64_t &Res) const;
339   virtual error_code getSectionSize(DataRefImpl Sec, uint64_t &Res) const;
340   virtual error_code getSectionContents(DataRefImpl Sec, StringRef &Res) const;
341   virtual error_code getSectionAlignment(DataRefImpl Sec, uint64_t &Res) const;
342   virtual error_code isSectionText(DataRefImpl Sec, bool &Res) const;
343   virtual error_code isSectionData(DataRefImpl Sec, bool &Res) const;
344   virtual error_code isSectionBSS(DataRefImpl Sec, bool &Res) const;
345   virtual error_code sectionContainsSymbol(DataRefImpl Sec, DataRefImpl Symb,
346                                            bool &Result) const;
347   virtual relocation_iterator getSectionRelBegin(DataRefImpl Sec) const;
348   virtual relocation_iterator getSectionRelEnd(DataRefImpl Sec) const;
349 
350   virtual error_code getRelocationNext(DataRefImpl Rel,
351                                        RelocationRef &Res) const;
352   virtual error_code getRelocationAddress(DataRefImpl Rel,
353                                           uint64_t &Res) const;
354   virtual error_code getRelocationSymbol(DataRefImpl Rel,
355                                          SymbolRef &Res) const;
356   virtual error_code getRelocationType(DataRefImpl Rel,
357                                        uint32_t &Res) const;
358   virtual error_code getRelocationTypeName(DataRefImpl Rel,
359                                            SmallVectorImpl<char> &Result) const;
360   virtual error_code getRelocationAdditionalInfo(DataRefImpl Rel,
361                                                  int64_t &Res) const;
362   virtual error_code getRelocationValueString(DataRefImpl Rel,
363                                            SmallVectorImpl<char> &Result) const;
364 
365 public:
366   ELFObjectFile(MemoryBuffer *Object, error_code &ec);
367   virtual symbol_iterator begin_symbols() const;
368   virtual symbol_iterator end_symbols() const;
369   virtual section_iterator begin_sections() const;
370   virtual section_iterator end_sections() const;
371 
372   virtual uint8_t getBytesInAddress() const;
373   virtual StringRef getFileFormatName() const;
374   virtual unsigned getArch() const;
375 
376   uint64_t getNumSections() const;
377   uint64_t getStringTableIndex() const;
378   ELF::Elf64_Word getSymbolTableIndex(const Elf_Sym *symb) const;
379   const Elf_Shdr *getSection(const Elf_Sym *symb) const;
380 };
381 } // end namespace
382 
383 template<support::endianness target_endianness, bool is64Bits>
384 void ELFObjectFile<target_endianness, is64Bits>
validateSymbol(DataRefImpl Symb) const385                   ::validateSymbol(DataRefImpl Symb) const {
386   const Elf_Sym  *symb = getSymbol(Symb);
387   const Elf_Shdr *SymbolTableSection = SymbolTableSections[Symb.d.b];
388   // FIXME: We really need to do proper error handling in the case of an invalid
389   //        input file. Because we don't use exceptions, I think we'll just pass
390   //        an error object around.
391   if (!(  symb
392         && SymbolTableSection
393         && symb >= (const Elf_Sym*)(base()
394                    + SymbolTableSection->sh_offset)
395         && symb <  (const Elf_Sym*)(base()
396                    + SymbolTableSection->sh_offset
397                    + SymbolTableSection->sh_size)))
398     // FIXME: Proper error handling.
399     report_fatal_error("Symb must point to a valid symbol!");
400 }
401 
402 template<support::endianness target_endianness, bool is64Bits>
403 error_code ELFObjectFile<target_endianness, is64Bits>
getSymbolNext(DataRefImpl Symb,SymbolRef & Result) const404                         ::getSymbolNext(DataRefImpl Symb,
405                                         SymbolRef &Result) const {
406   validateSymbol(Symb);
407   const Elf_Shdr *SymbolTableSection = SymbolTableSections[Symb.d.b];
408 
409   ++Symb.d.a;
410   // Check to see if we are at the end of this symbol table.
411   if (Symb.d.a >= SymbolTableSection->getEntityCount()) {
412     // We are at the end. If there are other symbol tables, jump to them.
413     ++Symb.d.b;
414     Symb.d.a = 1; // The 0th symbol in ELF is fake.
415     // Otherwise return the terminator.
416     if (Symb.d.b >= SymbolTableSections.size()) {
417       Symb.d.a = std::numeric_limits<uint32_t>::max();
418       Symb.d.b = std::numeric_limits<uint32_t>::max();
419     }
420   }
421 
422   Result = SymbolRef(Symb, this);
423   return object_error::success;
424 }
425 
426 template<support::endianness target_endianness, bool is64Bits>
427 error_code ELFObjectFile<target_endianness, is64Bits>
getSymbolName(DataRefImpl Symb,StringRef & Result) const428                         ::getSymbolName(DataRefImpl Symb,
429                                         StringRef &Result) const {
430   validateSymbol(Symb);
431   const Elf_Sym *symb = getSymbol(Symb);
432   return getSymbolName(symb, Result);
433 }
434 
435 template<support::endianness target_endianness, bool is64Bits>
436 ELF::Elf64_Word ELFObjectFile<target_endianness, is64Bits>
getSymbolTableIndex(const Elf_Sym * symb) const437                       ::getSymbolTableIndex(const Elf_Sym *symb) const {
438   if (symb->st_shndx == ELF::SHN_XINDEX)
439     return ExtendedSymbolTable.lookup(symb);
440   return symb->st_shndx;
441 }
442 
443 template<support::endianness target_endianness, bool is64Bits>
444 const typename ELFObjectFile<target_endianness, is64Bits>::Elf_Shdr *
445 ELFObjectFile<target_endianness, is64Bits>
getSection(const Elf_Sym * symb) const446                              ::getSection(const Elf_Sym *symb) const {
447   if (symb->st_shndx == ELF::SHN_XINDEX)
448     return getSection(ExtendedSymbolTable.lookup(symb));
449   if (symb->st_shndx >= ELF::SHN_LORESERVE)
450     return 0;
451   return getSection(symb->st_shndx);
452 }
453 
454 template<support::endianness target_endianness, bool is64Bits>
455 error_code ELFObjectFile<target_endianness, is64Bits>
getSymbolOffset(DataRefImpl Symb,uint64_t & Result) const456                         ::getSymbolOffset(DataRefImpl Symb,
457                                           uint64_t &Result) const {
458   validateSymbol(Symb);
459   const Elf_Sym  *symb = getSymbol(Symb);
460   const Elf_Shdr *Section;
461   switch (getSymbolTableIndex(symb)) {
462   case ELF::SHN_COMMON:
463    // Undefined symbols have no address yet.
464   case ELF::SHN_UNDEF:
465     Result = UnknownAddressOrSize;
466     return object_error::success;
467   case ELF::SHN_ABS:
468     Result = symb->st_value;
469     return object_error::success;
470   default: Section = getSection(symb);
471   }
472 
473   switch (symb->getType()) {
474   case ELF::STT_SECTION:
475     Result = Section ? Section->sh_addr : UnknownAddressOrSize;
476     return object_error::success;
477   case ELF::STT_FUNC:
478   case ELF::STT_OBJECT:
479   case ELF::STT_NOTYPE:
480     Result = symb->st_value;
481     return object_error::success;
482   default:
483     Result = UnknownAddressOrSize;
484     return object_error::success;
485   }
486 }
487 
488 template<support::endianness target_endianness, bool is64Bits>
489 error_code ELFObjectFile<target_endianness, is64Bits>
getSymbolAddress(DataRefImpl Symb,uint64_t & Result) const490                         ::getSymbolAddress(DataRefImpl Symb,
491                                            uint64_t &Result) const {
492   validateSymbol(Symb);
493   const Elf_Sym  *symb = getSymbol(Symb);
494   const Elf_Shdr *Section;
495   switch (getSymbolTableIndex(symb)) {
496   case ELF::SHN_COMMON: // Fall through.
497    // Undefined symbols have no address yet.
498   case ELF::SHN_UNDEF:
499     Result = UnknownAddressOrSize;
500     return object_error::success;
501   case ELF::SHN_ABS:
502     Result = reinterpret_cast<uintptr_t>(base()+symb->st_value);
503     return object_error::success;
504   default: Section = getSection(symb);
505   }
506   const uint8_t* addr = base();
507   if (Section)
508     addr += Section->sh_offset;
509   switch (symb->getType()) {
510   case ELF::STT_SECTION:
511     Result = reinterpret_cast<uintptr_t>(addr);
512     return object_error::success;
513   case ELF::STT_FUNC: // Fall through.
514   case ELF::STT_OBJECT: // Fall through.
515   case ELF::STT_NOTYPE:
516     addr += symb->st_value;
517     Result = reinterpret_cast<uintptr_t>(addr);
518     return object_error::success;
519   default:
520     Result = UnknownAddressOrSize;
521     return object_error::success;
522   }
523 }
524 
525 template<support::endianness target_endianness, bool is64Bits>
526 error_code ELFObjectFile<target_endianness, is64Bits>
getSymbolSize(DataRefImpl Symb,uint64_t & Result) const527                         ::getSymbolSize(DataRefImpl Symb,
528                                         uint64_t &Result) const {
529   validateSymbol(Symb);
530   const Elf_Sym  *symb = getSymbol(Symb);
531   if (symb->st_size == 0)
532     Result = UnknownAddressOrSize;
533   Result = symb->st_size;
534   return object_error::success;
535 }
536 
537 template<support::endianness target_endianness, bool is64Bits>
538 error_code ELFObjectFile<target_endianness, is64Bits>
getSymbolNMTypeChar(DataRefImpl Symb,char & Result) const539                         ::getSymbolNMTypeChar(DataRefImpl Symb,
540                                               char &Result) const {
541   validateSymbol(Symb);
542   const Elf_Sym  *symb = getSymbol(Symb);
543   const Elf_Shdr *Section = getSection(symb);
544 
545   char ret = '?';
546 
547   if (Section) {
548     switch (Section->sh_type) {
549     case ELF::SHT_PROGBITS:
550     case ELF::SHT_DYNAMIC:
551       switch (Section->sh_flags) {
552       case (ELF::SHF_ALLOC | ELF::SHF_EXECINSTR):
553         ret = 't'; break;
554       case (ELF::SHF_ALLOC | ELF::SHF_WRITE):
555         ret = 'd'; break;
556       case ELF::SHF_ALLOC:
557       case (ELF::SHF_ALLOC | ELF::SHF_MERGE):
558       case (ELF::SHF_ALLOC | ELF::SHF_MERGE | ELF::SHF_STRINGS):
559         ret = 'r'; break;
560       }
561       break;
562     case ELF::SHT_NOBITS: ret = 'b';
563     }
564   }
565 
566   switch (getSymbolTableIndex(symb)) {
567   case ELF::SHN_UNDEF:
568     if (ret == '?')
569       ret = 'U';
570     break;
571   case ELF::SHN_ABS: ret = 'a'; break;
572   case ELF::SHN_COMMON: ret = 'c'; break;
573   }
574 
575   switch (symb->getBinding()) {
576   case ELF::STB_GLOBAL: ret = ::toupper(ret); break;
577   case ELF::STB_WEAK:
578     if (getSymbolTableIndex(symb) == ELF::SHN_UNDEF)
579       ret = 'w';
580     else
581       if (symb->getType() == ELF::STT_OBJECT)
582         ret = 'V';
583       else
584         ret = 'W';
585   }
586 
587   if (ret == '?' && symb->getType() == ELF::STT_SECTION) {
588     StringRef name;
589     if (error_code ec = getSymbolName(Symb, name))
590       return ec;
591     Result = StringSwitch<char>(name)
592       .StartsWith(".debug", 'N')
593       .StartsWith(".note", 'n')
594       .Default('?');
595     return object_error::success;
596   }
597 
598   Result = ret;
599   return object_error::success;
600 }
601 
602 template<support::endianness target_endianness, bool is64Bits>
603 error_code ELFObjectFile<target_endianness, is64Bits>
getSymbolType(DataRefImpl Symb,SymbolRef::SymbolType & Result) const604                         ::getSymbolType(DataRefImpl Symb,
605                                         SymbolRef::SymbolType &Result) const {
606   validateSymbol(Symb);
607   const Elf_Sym  *symb = getSymbol(Symb);
608 
609   if (getSymbolTableIndex(symb) == ELF::SHN_UNDEF) {
610     Result = SymbolRef::ST_External;
611     return object_error::success;
612   }
613 
614   switch (symb->getType()) {
615   case ELF::STT_FUNC:
616     Result = SymbolRef::ST_Function;
617     break;
618   case ELF::STT_OBJECT:
619     Result = SymbolRef::ST_Data;
620     break;
621   default:
622     Result = SymbolRef::ST_Other;
623     break;
624   }
625   return object_error::success;
626 }
627 
628 template<support::endianness target_endianness, bool is64Bits>
629 error_code ELFObjectFile<target_endianness, is64Bits>
isSymbolGlobal(DataRefImpl Symb,bool & Result) const630                         ::isSymbolGlobal(DataRefImpl Symb,
631                                         bool &Result) const {
632   validateSymbol(Symb);
633   const Elf_Sym  *symb = getSymbol(Symb);
634 
635   Result = symb->getBinding() == ELF::STB_GLOBAL;
636   return object_error::success;
637 }
638 
639 template<support::endianness target_endianness, bool is64Bits>
640 error_code ELFObjectFile<target_endianness, is64Bits>
isSymbolInternal(DataRefImpl Symb,bool & Result) const641                         ::isSymbolInternal(DataRefImpl Symb,
642                                            bool &Result) const {
643   validateSymbol(Symb);
644   const Elf_Sym  *symb = getSymbol(Symb);
645 
646   if (  symb->getType() == ELF::STT_FILE
647      || symb->getType() == ELF::STT_SECTION)
648     Result = true;
649   Result = false;
650   return object_error::success;
651 }
652 
653 template<support::endianness target_endianness, bool is64Bits>
654 error_code ELFObjectFile<target_endianness, is64Bits>
getSectionNext(DataRefImpl Sec,SectionRef & Result) const655                         ::getSectionNext(DataRefImpl Sec, SectionRef &Result) const {
656   const uint8_t *sec = reinterpret_cast<const uint8_t *>(Sec.p);
657   sec += Header->e_shentsize;
658   Sec.p = reinterpret_cast<intptr_t>(sec);
659   Result = SectionRef(Sec, this);
660   return object_error::success;
661 }
662 
663 template<support::endianness target_endianness, bool is64Bits>
664 error_code ELFObjectFile<target_endianness, is64Bits>
getSectionName(DataRefImpl Sec,StringRef & Result) const665                         ::getSectionName(DataRefImpl Sec,
666                                          StringRef &Result) const {
667   const Elf_Shdr *sec = reinterpret_cast<const Elf_Shdr *>(Sec.p);
668   Result = StringRef(getString(dot_shstrtab_sec, sec->sh_name));
669   return object_error::success;
670 }
671 
672 template<support::endianness target_endianness, bool is64Bits>
673 error_code ELFObjectFile<target_endianness, is64Bits>
getSectionAddress(DataRefImpl Sec,uint64_t & Result) const674                         ::getSectionAddress(DataRefImpl Sec,
675                                             uint64_t &Result) const {
676   const Elf_Shdr *sec = reinterpret_cast<const Elf_Shdr *>(Sec.p);
677   Result = sec->sh_addr;
678   return object_error::success;
679 }
680 
681 template<support::endianness target_endianness, bool is64Bits>
682 error_code ELFObjectFile<target_endianness, is64Bits>
getSectionSize(DataRefImpl Sec,uint64_t & Result) const683                         ::getSectionSize(DataRefImpl Sec,
684                                          uint64_t &Result) const {
685   const Elf_Shdr *sec = reinterpret_cast<const Elf_Shdr *>(Sec.p);
686   Result = sec->sh_size;
687   return object_error::success;
688 }
689 
690 template<support::endianness target_endianness, bool is64Bits>
691 error_code ELFObjectFile<target_endianness, is64Bits>
getSectionContents(DataRefImpl Sec,StringRef & Result) const692                         ::getSectionContents(DataRefImpl Sec,
693                                              StringRef &Result) const {
694   const Elf_Shdr *sec = reinterpret_cast<const Elf_Shdr *>(Sec.p);
695   const char *start = (const char*)base() + sec->sh_offset;
696   Result = StringRef(start, sec->sh_size);
697   return object_error::success;
698 }
699 
700 template<support::endianness target_endianness, bool is64Bits>
701 error_code ELFObjectFile<target_endianness, is64Bits>
getSectionAlignment(DataRefImpl Sec,uint64_t & Result) const702                         ::getSectionAlignment(DataRefImpl Sec,
703                                               uint64_t &Result) const {
704   const Elf_Shdr *sec = reinterpret_cast<const Elf_Shdr *>(Sec.p);
705   Result = sec->sh_addralign;
706   return object_error::success;
707 }
708 
709 template<support::endianness target_endianness, bool is64Bits>
710 error_code ELFObjectFile<target_endianness, is64Bits>
isSectionText(DataRefImpl Sec,bool & Result) const711                         ::isSectionText(DataRefImpl Sec,
712                                         bool &Result) const {
713   const Elf_Shdr *sec = reinterpret_cast<const Elf_Shdr *>(Sec.p);
714   if (sec->sh_flags & ELF::SHF_EXECINSTR)
715     Result = true;
716   else
717     Result = false;
718   return object_error::success;
719 }
720 
721 template<support::endianness target_endianness, bool is64Bits>
722 error_code ELFObjectFile<target_endianness, is64Bits>
isSectionData(DataRefImpl Sec,bool & Result) const723                         ::isSectionData(DataRefImpl Sec,
724                                         bool &Result) const {
725   const Elf_Shdr *sec = reinterpret_cast<const Elf_Shdr *>(Sec.p);
726   if (sec->sh_flags & (ELF::SHF_ALLOC | ELF::SHF_WRITE)
727       && sec->sh_type == ELF::SHT_PROGBITS)
728     Result = true;
729   else
730     Result = false;
731   return object_error::success;
732 }
733 
734 template<support::endianness target_endianness, bool is64Bits>
735 error_code ELFObjectFile<target_endianness, is64Bits>
isSectionBSS(DataRefImpl Sec,bool & Result) const736                         ::isSectionBSS(DataRefImpl Sec,
737                                        bool &Result) const {
738   const Elf_Shdr *sec = reinterpret_cast<const Elf_Shdr *>(Sec.p);
739   if (sec->sh_flags & (ELF::SHF_ALLOC | ELF::SHF_WRITE)
740       && sec->sh_type == ELF::SHT_NOBITS)
741     Result = true;
742   else
743     Result = false;
744   return object_error::success;
745 }
746 
747 template<support::endianness target_endianness, bool is64Bits>
748 error_code ELFObjectFile<target_endianness, is64Bits>
sectionContainsSymbol(DataRefImpl Sec,DataRefImpl Symb,bool & Result) const749                           ::sectionContainsSymbol(DataRefImpl Sec,
750                                                   DataRefImpl Symb,
751                                                   bool &Result) const {
752   // FIXME: Unimplemented.
753   Result = false;
754   return object_error::success;
755 }
756 
757 template<support::endianness target_endianness, bool is64Bits>
758 relocation_iterator ELFObjectFile<target_endianness, is64Bits>
getSectionRelBegin(DataRefImpl Sec) const759                                  ::getSectionRelBegin(DataRefImpl Sec) const {
760   DataRefImpl RelData;
761   memset(&RelData, 0, sizeof(RelData));
762   const Elf_Shdr *sec = reinterpret_cast<const Elf_Shdr *>(Sec.p);
763   typename RelocMap_t::const_iterator ittr = SectionRelocMap.find(sec);
764   if (sec != 0 && ittr != SectionRelocMap.end()) {
765     RelData.w.a = getSection(ittr->second[0])->sh_info;
766     RelData.w.b = ittr->second[0];
767     RelData.w.c = 0;
768   }
769   return relocation_iterator(RelocationRef(RelData, this));
770 }
771 
772 template<support::endianness target_endianness, bool is64Bits>
773 relocation_iterator ELFObjectFile<target_endianness, is64Bits>
getSectionRelEnd(DataRefImpl Sec) const774                                  ::getSectionRelEnd(DataRefImpl Sec) const {
775   DataRefImpl RelData;
776   memset(&RelData, 0, sizeof(RelData));
777   const Elf_Shdr *sec = reinterpret_cast<const Elf_Shdr *>(Sec.p);
778   typename RelocMap_t::const_iterator ittr = SectionRelocMap.find(sec);
779   if (sec != 0 && ittr != SectionRelocMap.end()) {
780     // Get the index of the last relocation section for this section.
781     std::size_t relocsecindex = ittr->second[ittr->second.size() - 1];
782     const Elf_Shdr *relocsec = getSection(relocsecindex);
783     RelData.w.a = relocsec->sh_info;
784     RelData.w.b = relocsecindex;
785     RelData.w.c = relocsec->sh_size / relocsec->sh_entsize;
786   }
787   return relocation_iterator(RelocationRef(RelData, this));
788 }
789 
790 // Relocations
791 template<support::endianness target_endianness, bool is64Bits>
792 error_code ELFObjectFile<target_endianness, is64Bits>
getRelocationNext(DataRefImpl Rel,RelocationRef & Result) const793                         ::getRelocationNext(DataRefImpl Rel,
794                                             RelocationRef &Result) const {
795   ++Rel.w.c;
796   const Elf_Shdr *relocsec = getSection(Rel.w.b);
797   if (Rel.w.c >= (relocsec->sh_size / relocsec->sh_entsize)) {
798     // We have reached the end of the relocations for this section. See if there
799     // is another relocation section.
800     typename RelocMap_t::mapped_type relocseclist =
801       SectionRelocMap.lookup(getSection(Rel.w.a));
802 
803     // Do a binary search for the current reloc section index (which must be
804     // present). Then get the next one.
805     typename RelocMap_t::mapped_type::const_iterator loc =
806       std::lower_bound(relocseclist.begin(), relocseclist.end(), Rel.w.b);
807     ++loc;
808 
809     // If there is no next one, don't do anything. The ++Rel.w.c above sets Rel
810     // to the end iterator.
811     if (loc != relocseclist.end()) {
812       Rel.w.b = *loc;
813       Rel.w.a = 0;
814     }
815   }
816   Result = RelocationRef(Rel, this);
817   return object_error::success;
818 }
819 
820 template<support::endianness target_endianness, bool is64Bits>
821 error_code ELFObjectFile<target_endianness, is64Bits>
getRelocationSymbol(DataRefImpl Rel,SymbolRef & Result) const822                         ::getRelocationSymbol(DataRefImpl Rel,
823                                               SymbolRef &Result) const {
824   uint32_t symbolIdx;
825   const Elf_Shdr *sec = getSection(Rel.w.b);
826   switch (sec->sh_type) {
827     default :
828       report_fatal_error("Invalid section type in Rel!");
829     case ELF::SHT_REL : {
830       symbolIdx = getRel(Rel)->getSymbol();
831       break;
832     }
833     case ELF::SHT_RELA : {
834       symbolIdx = getRela(Rel)->getSymbol();
835       break;
836     }
837   }
838   DataRefImpl SymbolData;
839   IndexMap_t::const_iterator it = SymbolTableSectionsIndexMap.find(sec->sh_link);
840   if (it == SymbolTableSectionsIndexMap.end())
841     report_fatal_error("Relocation symbol table not found!");
842   SymbolData.d.a = symbolIdx;
843   SymbolData.d.b = it->second;
844   Result = SymbolRef(SymbolData, this);
845   return object_error::success;
846 }
847 
848 template<support::endianness target_endianness, bool is64Bits>
849 error_code ELFObjectFile<target_endianness, is64Bits>
getRelocationAddress(DataRefImpl Rel,uint64_t & Result) const850                         ::getRelocationAddress(DataRefImpl Rel,
851                                                uint64_t &Result) const {
852   uint64_t offset;
853   const Elf_Shdr *sec = getSection(Rel.w.b);
854   switch (sec->sh_type) {
855     default :
856       report_fatal_error("Invalid section type in Rel!");
857     case ELF::SHT_REL : {
858       offset = getRel(Rel)->r_offset;
859       break;
860     }
861     case ELF::SHT_RELA : {
862       offset = getRela(Rel)->r_offset;
863       break;
864     }
865   }
866 
867   Result = offset;
868   return object_error::success;
869 }
870 
871 template<support::endianness target_endianness, bool is64Bits>
872 error_code ELFObjectFile<target_endianness, is64Bits>
getRelocationType(DataRefImpl Rel,uint32_t & Result) const873                         ::getRelocationType(DataRefImpl Rel,
874                                             uint32_t &Result) const {
875   const Elf_Shdr *sec = getSection(Rel.w.b);
876   switch (sec->sh_type) {
877     default :
878       report_fatal_error("Invalid section type in Rel!");
879     case ELF::SHT_REL : {
880       Result = getRel(Rel)->getType();
881       break;
882     }
883     case ELF::SHT_RELA : {
884       Result = getRela(Rel)->getType();
885       break;
886     }
887   }
888   return object_error::success;
889 }
890 
891 #define LLVM_ELF_SWITCH_RELOC_TYPE_NAME(enum) \
892   case ELF::enum: res = #enum; break;
893 
894 template<support::endianness target_endianness, bool is64Bits>
895 error_code ELFObjectFile<target_endianness, is64Bits>
getRelocationTypeName(DataRefImpl Rel,SmallVectorImpl<char> & Result) const896                         ::getRelocationTypeName(DataRefImpl Rel,
897                                           SmallVectorImpl<char> &Result) const {
898   const Elf_Shdr *sec = getSection(Rel.w.b);
899   uint8_t type;
900   StringRef res;
901   switch (sec->sh_type) {
902     default :
903       return object_error::parse_failed;
904     case ELF::SHT_REL : {
905       type = getRel(Rel)->getType();
906       break;
907     }
908     case ELF::SHT_RELA : {
909       type = getRela(Rel)->getType();
910       break;
911     }
912   }
913   switch (Header->e_machine) {
914   case ELF::EM_X86_64:
915     switch (type) {
916       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_X86_64_NONE);
917       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_X86_64_64);
918       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_X86_64_PC32);
919       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_X86_64_GOT32);
920       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_X86_64_PLT32);
921       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_X86_64_COPY);
922       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_X86_64_GLOB_DAT);
923       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_X86_64_JUMP_SLOT);
924       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_X86_64_RELATIVE);
925       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_X86_64_GOTPCREL);
926       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_X86_64_32);
927       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_X86_64_32S);
928       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_X86_64_16);
929       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_X86_64_PC16);
930       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_X86_64_8);
931       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_X86_64_PC8);
932       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_X86_64_DTPMOD64);
933       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_X86_64_DTPOFF64);
934       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_X86_64_TPOFF64);
935       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_X86_64_TLSGD);
936       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_X86_64_TLSLD);
937       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_X86_64_DTPOFF32);
938       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_X86_64_GOTTPOFF);
939       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_X86_64_TPOFF32);
940       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_X86_64_PC64);
941       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_X86_64_GOTOFF64);
942       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_X86_64_GOTPC32);
943       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_X86_64_SIZE32);
944       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_X86_64_SIZE64);
945       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_X86_64_GOTPC32_TLSDESC);
946       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_X86_64_TLSDESC_CALL);
947       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_X86_64_TLSDESC);
948     default:
949       res = "Unknown";
950     }
951     break;
952   case ELF::EM_386:
953     switch (type) {
954       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_NONE);
955       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_32);
956       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_PC32);
957       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_GOT32);
958       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_PLT32);
959       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_COPY);
960       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_GLOB_DAT);
961       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_JUMP_SLOT);
962       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_RELATIVE);
963       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_GOTOFF);
964       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_GOTPC);
965       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_32PLT);
966       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_TLS_TPOFF);
967       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_TLS_IE);
968       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_TLS_GOTIE);
969       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_TLS_LE);
970       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_TLS_GD);
971       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_TLS_LDM);
972       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_16);
973       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_PC16);
974       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_8);
975       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_PC8);
976       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_TLS_GD_32);
977       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_TLS_GD_PUSH);
978       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_TLS_GD_CALL);
979       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_TLS_GD_POP);
980       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_TLS_LDM_32);
981       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_TLS_LDM_PUSH);
982       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_TLS_LDM_CALL);
983       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_TLS_LDM_POP);
984       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_TLS_LDO_32);
985       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_TLS_IE_32);
986       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_TLS_LE_32);
987       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_TLS_DTPMOD32);
988       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_TLS_DTPOFF32);
989       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_TLS_TPOFF32);
990       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_TLS_GOTDESC);
991       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_TLS_DESC_CALL);
992       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_TLS_DESC);
993       LLVM_ELF_SWITCH_RELOC_TYPE_NAME(R_386_IRELATIVE);
994     default:
995       res = "Unknown";
996     }
997     break;
998   default:
999     res = "Unknown";
1000   }
1001   Result.append(res.begin(), res.end());
1002   return object_error::success;
1003 }
1004 
1005 #undef LLVM_ELF_SWITCH_RELOC_TYPE_NAME
1006 
1007 template<support::endianness target_endianness, bool is64Bits>
1008 error_code ELFObjectFile<target_endianness, is64Bits>
getRelocationAdditionalInfo(DataRefImpl Rel,int64_t & Result) const1009                         ::getRelocationAdditionalInfo(DataRefImpl Rel,
1010                                                       int64_t &Result) const {
1011   const Elf_Shdr *sec = getSection(Rel.w.b);
1012   switch (sec->sh_type) {
1013     default :
1014       report_fatal_error("Invalid section type in Rel!");
1015     case ELF::SHT_REL : {
1016       Result = 0;
1017       return object_error::success;
1018     }
1019     case ELF::SHT_RELA : {
1020       Result = getRela(Rel)->r_addend;
1021       return object_error::success;
1022     }
1023   }
1024 }
1025 
1026 template<support::endianness target_endianness, bool is64Bits>
1027 error_code ELFObjectFile<target_endianness, is64Bits>
getRelocationValueString(DataRefImpl Rel,SmallVectorImpl<char> & Result) const1028                         ::getRelocationValueString(DataRefImpl Rel,
1029                                           SmallVectorImpl<char> &Result) const {
1030   const Elf_Shdr *sec = getSection(Rel.w.b);
1031   uint8_t type;
1032   StringRef res;
1033   int64_t addend = 0;
1034   uint16_t symbol_index = 0;
1035   switch (sec->sh_type) {
1036     default :
1037       return object_error::parse_failed;
1038     case ELF::SHT_REL : {
1039       type = getRel(Rel)->getType();
1040       symbol_index = getRel(Rel)->getSymbol();
1041       // TODO: Read implicit addend from section data.
1042       break;
1043     }
1044     case ELF::SHT_RELA : {
1045       type = getRela(Rel)->getType();
1046       symbol_index = getRela(Rel)->getSymbol();
1047       addend = getRela(Rel)->r_addend;
1048       break;
1049     }
1050   }
1051   const Elf_Sym *symb = getEntry<Elf_Sym>(sec->sh_link, symbol_index);
1052   StringRef symname;
1053   if (error_code ec = getSymbolName(symb, symname))
1054     return ec;
1055   switch (Header->e_machine) {
1056   case ELF::EM_X86_64:
1057     switch (type) {
1058     case ELF::R_X86_64_32S:
1059       res = symname;
1060       break;
1061     case ELF::R_X86_64_PC32: {
1062         std::string fmtbuf;
1063         raw_string_ostream fmt(fmtbuf);
1064         fmt << symname << (addend < 0 ? "" : "+") << addend << "-P";
1065         fmt.flush();
1066         Result.append(fmtbuf.begin(), fmtbuf.end());
1067       }
1068       break;
1069     default:
1070       res = "Unknown";
1071     }
1072     break;
1073   default:
1074     res = "Unknown";
1075   }
1076   if (Result.empty())
1077     Result.append(res.begin(), res.end());
1078   return object_error::success;
1079 }
1080 
1081 template<support::endianness target_endianness, bool is64Bits>
ELFObjectFile(MemoryBuffer * Object,error_code & ec)1082 ELFObjectFile<target_endianness, is64Bits>::ELFObjectFile(MemoryBuffer *Object
1083                                                           , error_code &ec)
1084   : ObjectFile(Binary::isELF, Object, ec)
1085   , SectionHeaderTable(0)
1086   , dot_shstrtab_sec(0)
1087   , dot_strtab_sec(0) {
1088   Header = reinterpret_cast<const Elf_Ehdr *>(base());
1089 
1090   if (Header->e_shoff == 0)
1091     return;
1092 
1093   SectionHeaderTable =
1094     reinterpret_cast<const Elf_Shdr *>(base() + Header->e_shoff);
1095   uint64_t SectionTableSize = getNumSections() * Header->e_shentsize;
1096   if (!(  (const uint8_t *)SectionHeaderTable + SectionTableSize
1097          <= base() + Data->getBufferSize()))
1098     // FIXME: Proper error handling.
1099     report_fatal_error("Section table goes past end of file!");
1100 
1101 
1102   // To find the symbol tables we walk the section table to find SHT_SYMTAB.
1103   const Elf_Shdr* SymbolTableSectionHeaderIndex = 0;
1104   const Elf_Shdr* sh = reinterpret_cast<const Elf_Shdr*>(SectionHeaderTable);
1105   for (uint64_t i = 0, e = getNumSections(); i != e; ++i) {
1106     if (sh->sh_type == ELF::SHT_SYMTAB_SHNDX) {
1107       if (SymbolTableSectionHeaderIndex)
1108         // FIXME: Proper error handling.
1109         report_fatal_error("More than one .symtab_shndx!");
1110       SymbolTableSectionHeaderIndex = sh;
1111     }
1112     if (sh->sh_type == ELF::SHT_SYMTAB) {
1113       SymbolTableSectionsIndexMap[i] = SymbolTableSections.size();
1114       SymbolTableSections.push_back(sh);
1115     }
1116     if (sh->sh_type == ELF::SHT_REL || sh->sh_type == ELF::SHT_RELA) {
1117       SectionRelocMap[getSection(sh->sh_info)].push_back(i);
1118     }
1119     ++sh;
1120   }
1121 
1122   // Sort section relocation lists by index.
1123   for (typename RelocMap_t::iterator i = SectionRelocMap.begin(),
1124                                      e = SectionRelocMap.end(); i != e; ++i) {
1125     std::sort(i->second.begin(), i->second.end());
1126   }
1127 
1128   // Get string table sections.
1129   dot_shstrtab_sec = getSection(getStringTableIndex());
1130   if (dot_shstrtab_sec) {
1131     // Verify that the last byte in the string table in a null.
1132     if (((const char*)base() + dot_shstrtab_sec->sh_offset)
1133         [dot_shstrtab_sec->sh_size - 1] != 0)
1134       // FIXME: Proper error handling.
1135       report_fatal_error("String table must end with a null terminator!");
1136   }
1137 
1138   // Merge this into the above loop.
1139   for (const char *i = reinterpret_cast<const char *>(SectionHeaderTable),
1140                   *e = i + getNumSections() * Header->e_shentsize;
1141                    i != e; i += Header->e_shentsize) {
1142     const Elf_Shdr *sh = reinterpret_cast<const Elf_Shdr*>(i);
1143     if (sh->sh_type == ELF::SHT_STRTAB) {
1144       StringRef SectionName(getString(dot_shstrtab_sec, sh->sh_name));
1145       if (SectionName == ".strtab") {
1146         if (dot_strtab_sec != 0)
1147           // FIXME: Proper error handling.
1148           report_fatal_error("Already found section named .strtab!");
1149         dot_strtab_sec = sh;
1150         const char *dot_strtab = (const char*)base() + sh->sh_offset;
1151           if (dot_strtab[sh->sh_size - 1] != 0)
1152             // FIXME: Proper error handling.
1153             report_fatal_error("String table must end with a null terminator!");
1154       }
1155     }
1156   }
1157 
1158   // Build symbol name side-mapping if there is one.
1159   if (SymbolTableSectionHeaderIndex) {
1160     const Elf_Word *ShndxTable = reinterpret_cast<const Elf_Word*>(base() +
1161                                       SymbolTableSectionHeaderIndex->sh_offset);
1162     error_code ec;
1163     for (symbol_iterator si = begin_symbols(),
1164                          se = end_symbols(); si != se; si.increment(ec)) {
1165       if (ec)
1166         report_fatal_error("Fewer extended symbol table entries than symbols!");
1167       if (*ShndxTable != ELF::SHN_UNDEF)
1168         ExtendedSymbolTable[getSymbol(si->getRawDataRefImpl())] = *ShndxTable;
1169       ++ShndxTable;
1170     }
1171   }
1172 }
1173 
1174 template<support::endianness target_endianness, bool is64Bits>
1175 symbol_iterator ELFObjectFile<target_endianness, is64Bits>
begin_symbols() const1176                              ::begin_symbols() const {
1177   DataRefImpl SymbolData;
1178   memset(&SymbolData, 0, sizeof(SymbolData));
1179   if (SymbolTableSections.size() == 0) {
1180     SymbolData.d.a = std::numeric_limits<uint32_t>::max();
1181     SymbolData.d.b = std::numeric_limits<uint32_t>::max();
1182   } else {
1183     SymbolData.d.a = 1; // The 0th symbol in ELF is fake.
1184     SymbolData.d.b = 0;
1185   }
1186   return symbol_iterator(SymbolRef(SymbolData, this));
1187 }
1188 
1189 template<support::endianness target_endianness, bool is64Bits>
1190 symbol_iterator ELFObjectFile<target_endianness, is64Bits>
end_symbols() const1191                              ::end_symbols() const {
1192   DataRefImpl SymbolData;
1193   memset(&SymbolData, 0, sizeof(SymbolData));
1194   SymbolData.d.a = std::numeric_limits<uint32_t>::max();
1195   SymbolData.d.b = std::numeric_limits<uint32_t>::max();
1196   return symbol_iterator(SymbolRef(SymbolData, this));
1197 }
1198 
1199 template<support::endianness target_endianness, bool is64Bits>
1200 section_iterator ELFObjectFile<target_endianness, is64Bits>
begin_sections() const1201                               ::begin_sections() const {
1202   DataRefImpl ret;
1203   memset(&ret, 0, sizeof(DataRefImpl));
1204   ret.p = reinterpret_cast<intptr_t>(base() + Header->e_shoff);
1205   return section_iterator(SectionRef(ret, this));
1206 }
1207 
1208 template<support::endianness target_endianness, bool is64Bits>
1209 section_iterator ELFObjectFile<target_endianness, is64Bits>
end_sections() const1210                               ::end_sections() const {
1211   DataRefImpl ret;
1212   memset(&ret, 0, sizeof(DataRefImpl));
1213   ret.p = reinterpret_cast<intptr_t>(base()
1214                                      + Header->e_shoff
1215                                      + (Header->e_shentsize*getNumSections()));
1216   return section_iterator(SectionRef(ret, this));
1217 }
1218 
1219 template<support::endianness target_endianness, bool is64Bits>
getBytesInAddress() const1220 uint8_t ELFObjectFile<target_endianness, is64Bits>::getBytesInAddress() const {
1221   return is64Bits ? 8 : 4;
1222 }
1223 
1224 template<support::endianness target_endianness, bool is64Bits>
1225 StringRef ELFObjectFile<target_endianness, is64Bits>
getFileFormatName() const1226                        ::getFileFormatName() const {
1227   switch(Header->e_ident[ELF::EI_CLASS]) {
1228   case ELF::ELFCLASS32:
1229     switch(Header->e_machine) {
1230     case ELF::EM_386:
1231       return "ELF32-i386";
1232     case ELF::EM_X86_64:
1233       return "ELF32-x86-64";
1234     case ELF::EM_ARM:
1235       return "ELF32-arm";
1236     default:
1237       return "ELF32-unknown";
1238     }
1239   case ELF::ELFCLASS64:
1240     switch(Header->e_machine) {
1241     case ELF::EM_386:
1242       return "ELF64-i386";
1243     case ELF::EM_X86_64:
1244       return "ELF64-x86-64";
1245     default:
1246       return "ELF64-unknown";
1247     }
1248   default:
1249     // FIXME: Proper error handling.
1250     report_fatal_error("Invalid ELFCLASS!");
1251   }
1252 }
1253 
1254 template<support::endianness target_endianness, bool is64Bits>
getArch() const1255 unsigned ELFObjectFile<target_endianness, is64Bits>::getArch() const {
1256   switch(Header->e_machine) {
1257   case ELF::EM_386:
1258     return Triple::x86;
1259   case ELF::EM_X86_64:
1260     return Triple::x86_64;
1261   case ELF::EM_ARM:
1262     return Triple::arm;
1263   default:
1264     return Triple::UnknownArch;
1265   }
1266 }
1267 
1268 template<support::endianness target_endianness, bool is64Bits>
getNumSections() const1269 uint64_t ELFObjectFile<target_endianness, is64Bits>::getNumSections() const {
1270   if (Header->e_shnum == ELF::SHN_UNDEF)
1271     return SectionHeaderTable->sh_size;
1272   return Header->e_shnum;
1273 }
1274 
1275 template<support::endianness target_endianness, bool is64Bits>
1276 uint64_t
getStringTableIndex() const1277 ELFObjectFile<target_endianness, is64Bits>::getStringTableIndex() const {
1278   if (Header->e_shnum == ELF::SHN_UNDEF) {
1279     if (Header->e_shstrndx == ELF::SHN_HIRESERVE)
1280       return SectionHeaderTable->sh_link;
1281     if (Header->e_shstrndx >= getNumSections())
1282       return 0;
1283   }
1284   return Header->e_shstrndx;
1285 }
1286 
1287 
1288 template<support::endianness target_endianness, bool is64Bits>
1289 template<typename T>
1290 inline const T *
getEntry(uint16_t Section,uint32_t Entry) const1291 ELFObjectFile<target_endianness, is64Bits>::getEntry(uint16_t Section,
1292                                                      uint32_t Entry) const {
1293   return getEntry<T>(getSection(Section), Entry);
1294 }
1295 
1296 template<support::endianness target_endianness, bool is64Bits>
1297 template<typename T>
1298 inline const T *
getEntry(const Elf_Shdr * Section,uint32_t Entry) const1299 ELFObjectFile<target_endianness, is64Bits>::getEntry(const Elf_Shdr * Section,
1300                                                      uint32_t Entry) const {
1301   return reinterpret_cast<const T *>(
1302            base()
1303            + Section->sh_offset
1304            + (Entry * Section->sh_entsize));
1305 }
1306 
1307 template<support::endianness target_endianness, bool is64Bits>
1308 const typename ELFObjectFile<target_endianness, is64Bits>::Elf_Sym *
getSymbol(DataRefImpl Symb) const1309 ELFObjectFile<target_endianness, is64Bits>::getSymbol(DataRefImpl Symb) const {
1310   return getEntry<Elf_Sym>(SymbolTableSections[Symb.d.b], Symb.d.a);
1311 }
1312 
1313 template<support::endianness target_endianness, bool is64Bits>
1314 const typename ELFObjectFile<target_endianness, is64Bits>::Elf_Rel *
getRel(DataRefImpl Rel) const1315 ELFObjectFile<target_endianness, is64Bits>::getRel(DataRefImpl Rel) const {
1316   return getEntry<Elf_Rel>(Rel.w.b, Rel.w.c);
1317 }
1318 
1319 template<support::endianness target_endianness, bool is64Bits>
1320 const typename ELFObjectFile<target_endianness, is64Bits>::Elf_Rela *
getRela(DataRefImpl Rela) const1321 ELFObjectFile<target_endianness, is64Bits>::getRela(DataRefImpl Rela) const {
1322   return getEntry<Elf_Rela>(Rela.w.b, Rela.w.c);
1323 }
1324 
1325 template<support::endianness target_endianness, bool is64Bits>
1326 const typename ELFObjectFile<target_endianness, is64Bits>::Elf_Shdr *
getSection(DataRefImpl Symb) const1327 ELFObjectFile<target_endianness, is64Bits>::getSection(DataRefImpl Symb) const {
1328   const Elf_Shdr *sec = getSection(Symb.d.b);
1329   if (sec->sh_type != ELF::SHT_SYMTAB || sec->sh_type != ELF::SHT_DYNSYM)
1330     // FIXME: Proper error handling.
1331     report_fatal_error("Invalid symbol table section!");
1332   return sec;
1333 }
1334 
1335 template<support::endianness target_endianness, bool is64Bits>
1336 const typename ELFObjectFile<target_endianness, is64Bits>::Elf_Shdr *
getSection(uint32_t index) const1337 ELFObjectFile<target_endianness, is64Bits>::getSection(uint32_t index) const {
1338   if (index == 0)
1339     return 0;
1340   if (!SectionHeaderTable || index >= getNumSections())
1341     // FIXME: Proper error handling.
1342     report_fatal_error("Invalid section index!");
1343 
1344   return reinterpret_cast<const Elf_Shdr *>(
1345          reinterpret_cast<const char *>(SectionHeaderTable)
1346          + (index * Header->e_shentsize));
1347 }
1348 
1349 template<support::endianness target_endianness, bool is64Bits>
1350 const char *ELFObjectFile<target_endianness, is64Bits>
getString(uint32_t section,ELF::Elf32_Word offset) const1351                          ::getString(uint32_t section,
1352                                      ELF::Elf32_Word offset) const {
1353   return getString(getSection(section), offset);
1354 }
1355 
1356 template<support::endianness target_endianness, bool is64Bits>
1357 const char *ELFObjectFile<target_endianness, is64Bits>
getString(const Elf_Shdr * section,ELF::Elf32_Word offset) const1358                          ::getString(const Elf_Shdr *section,
1359                                      ELF::Elf32_Word offset) const {
1360   assert(section && section->sh_type == ELF::SHT_STRTAB && "Invalid section!");
1361   if (offset >= section->sh_size)
1362     // FIXME: Proper error handling.
1363     report_fatal_error("Symbol name offset outside of string table!");
1364   return (const char *)base() + section->sh_offset + offset;
1365 }
1366 
1367 template<support::endianness target_endianness, bool is64Bits>
1368 error_code ELFObjectFile<target_endianness, is64Bits>
getSymbolName(const Elf_Sym * symb,StringRef & Result) const1369                         ::getSymbolName(const Elf_Sym *symb,
1370                                         StringRef &Result) const {
1371   if (symb->st_name == 0) {
1372     const Elf_Shdr *section = getSection(symb);
1373     if (!section)
1374       Result = "";
1375     else
1376       Result = getString(dot_shstrtab_sec, section->sh_name);
1377     return object_error::success;
1378   }
1379 
1380   // Use the default symbol table name section.
1381   Result = getString(dot_strtab_sec, symb->st_name);
1382   return object_error::success;
1383 }
1384 
1385 // EI_CLASS, EI_DATA.
1386 static std::pair<unsigned char, unsigned char>
getElfArchType(MemoryBuffer * Object)1387 getElfArchType(MemoryBuffer *Object) {
1388   if (Object->getBufferSize() < ELF::EI_NIDENT)
1389     return std::make_pair((uint8_t)ELF::ELFCLASSNONE,(uint8_t)ELF::ELFDATANONE);
1390   return std::make_pair( (uint8_t)Object->getBufferStart()[ELF::EI_CLASS]
1391                        , (uint8_t)Object->getBufferStart()[ELF::EI_DATA]);
1392 }
1393 
1394 namespace llvm {
1395 
createELFObjectFile(MemoryBuffer * Object)1396   ObjectFile *ObjectFile::createELFObjectFile(MemoryBuffer *Object) {
1397     std::pair<unsigned char, unsigned char> Ident = getElfArchType(Object);
1398     error_code ec;
1399     if (Ident.first == ELF::ELFCLASS32 && Ident.second == ELF::ELFDATA2LSB)
1400       return new ELFObjectFile<support::little, false>(Object, ec);
1401     else if (Ident.first == ELF::ELFCLASS32 && Ident.second == ELF::ELFDATA2MSB)
1402       return new ELFObjectFile<support::big, false>(Object, ec);
1403     else if (Ident.first == ELF::ELFCLASS64 && Ident.second == ELF::ELFDATA2LSB)
1404       return new ELFObjectFile<support::little, true>(Object, ec);
1405     else if (Ident.first == ELF::ELFCLASS64 && Ident.second == ELF::ELFDATA2MSB)
1406       return new ELFObjectFile<support::big, true>(Object, ec);
1407     // FIXME: Proper error handling.
1408     report_fatal_error("Not an ELF object file!");
1409   }
1410 
1411 } // end namespace llvm
1412