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