1fe5ee802SMichael J. Spencer //===- ObjectFile.cpp - File format independent object file -----*- C++ -*-===//
2fe5ee802SMichael J. Spencer //
3fe5ee802SMichael J. Spencer //                     The LLVM Compiler Infrastructure
4fe5ee802SMichael J. Spencer //
5fe5ee802SMichael J. Spencer // This file is distributed under the University of Illinois Open Source
6fe5ee802SMichael J. Spencer // License. See LICENSE.TXT for details.
7fe5ee802SMichael J. Spencer //
8fe5ee802SMichael J. Spencer //===----------------------------------------------------------------------===//
9fe5ee802SMichael J. Spencer //
10fe5ee802SMichael J. Spencer // This file defines a file format independent ObjectFile class.
11fe5ee802SMichael J. Spencer //
12fe5ee802SMichael J. Spencer //===----------------------------------------------------------------------===//
13fe5ee802SMichael J. Spencer 
14fe5ee802SMichael J. Spencer #include "llvm/Object/ObjectFile.h"
1539a0ffc3SMichael J. Spencer #include "llvm/ADT/OwningPtr.h"
16fe5ee802SMichael J. Spencer #include "llvm/Support/ErrorHandling.h"
17cd81b909SRafael Espindola #include "llvm/Support/FileSystem.h"
18fe5ee802SMichael J. Spencer #include "llvm/Support/MemoryBuffer.h"
197b6fef82SMichael J. Spencer #include "llvm/Support/system_error.h"
20fe5ee802SMichael J. Spencer 
21f177d16fSMichael J. Spencer using namespace llvm;
22f177d16fSMichael J. Spencer using namespace object;
23fe5ee802SMichael J. Spencer 
24a379b181SDavid Blaikie void ObjectFile::anchor() { }
25a379b181SDavid Blaikie 
26717c4d44SRafael Espindola ObjectFile::ObjectFile(unsigned int Type, MemoryBuffer *source)
27ec29b121SMichael J. Spencer   : Binary(Type, source) {
28fe5ee802SMichael J. Spencer }
29fe5ee802SMichael J. Spencer 
30e4dd2e01SRafael Espindola error_code ObjectFile::getSymbolAlignment(DataRefImpl DRI,
31e4dd2e01SRafael Espindola                                           uint32_t &Result) const {
32e4dd2e01SRafael Espindola   Result = 0;
33e4dd2e01SRafael Espindola   return object_error::success;
34e4dd2e01SRafael Espindola }
35e4dd2e01SRafael Espindola 
364f60a38fSRafael Espindola section_iterator ObjectFile::getRelocatedSection(DataRefImpl Sec) const {
374f60a38fSRafael Espindola   return section_iterator(SectionRef(Sec, this));
384f60a38fSRafael Espindola }
394f60a38fSRafael Espindola 
40fe5ee802SMichael J. Spencer ObjectFile *ObjectFile::createObjectFile(MemoryBuffer *Object) {
414d76a21bSRafael Espindola   if (Object->getBufferSize() < 64) {
424d76a21bSRafael Espindola     delete Object;
43fe5ee802SMichael J. Spencer     return 0;
444d76a21bSRafael Espindola   }
45cd81b909SRafael Espindola 
46cd81b909SRafael Espindola   sys::fs::file_magic Type = sys::fs::identify_magic(Object->getBuffer());
47cd81b909SRafael Espindola   switch (Type) {
48cd81b909SRafael Espindola   case sys::fs::file_magic::unknown:
49447d2d12SRafael Espindola   case sys::fs::file_magic::bitcode:
50447d2d12SRafael Espindola   case sys::fs::file_magic::archive:
51e6388e62SAlexey Samsonov   case sys::fs::file_magic::macho_universal_binary:
52*fc149a69SRui Ueyama   case sys::fs::file_magic::windows_resource:
534d76a21bSRafael Espindola     delete Object;
54a5a4ff5aSEli Bendersky     return 0;
55cd81b909SRafael Espindola   case sys::fs::file_magic::elf_relocatable:
56cd81b909SRafael Espindola   case sys::fs::file_magic::elf_executable:
57cd81b909SRafael Espindola   case sys::fs::file_magic::elf_shared_object:
58cd81b909SRafael Espindola   case sys::fs::file_magic::elf_core:
59b60a18deSMichael J. Spencer     return createELFObjectFile(Object);
60cd81b909SRafael Espindola   case sys::fs::file_magic::macho_object:
61cd81b909SRafael Espindola   case sys::fs::file_magic::macho_executable:
62cd81b909SRafael Espindola   case sys::fs::file_magic::macho_fixed_virtual_memory_shared_lib:
63cd81b909SRafael Espindola   case sys::fs::file_magic::macho_core:
64cd81b909SRafael Espindola   case sys::fs::file_magic::macho_preload_executable:
65cd81b909SRafael Espindola   case sys::fs::file_magic::macho_dynamically_linked_shared_lib:
66cd81b909SRafael Espindola   case sys::fs::file_magic::macho_dynamic_linker:
67cd81b909SRafael Espindola   case sys::fs::file_magic::macho_bundle:
68cd81b909SRafael Espindola   case sys::fs::file_magic::macho_dynamically_linked_shared_lib_stub:
69cd81b909SRafael Espindola   case sys::fs::file_magic::macho_dsym_companion:
70bafb9347SEric Christopher     return createMachOObjectFile(Object);
71cd81b909SRafael Espindola   case sys::fs::file_magic::coff_object:
72cd81b909SRafael Espindola   case sys::fs::file_magic::pecoff_executable:
738e90adafSMichael J. Spencer     return createCOFFObjectFile(Object);
74fe5ee802SMichael J. Spencer   }
75cd81b909SRafael Espindola   llvm_unreachable("Unexpected Object File Type");
76fe5ee802SMichael J. Spencer }
77fe5ee802SMichael J. Spencer 
78f177d16fSMichael J. Spencer ObjectFile *ObjectFile::createObjectFile(StringRef ObjectPath) {
7939a0ffc3SMichael J. Spencer   OwningPtr<MemoryBuffer> File;
802d3dac3eSRafael Espindola   if (MemoryBuffer::getFile(ObjectPath, File))
8139a0ffc3SMichael J. Spencer     return NULL;
8239a0ffc3SMichael J. Spencer   return createObjectFile(File.take());
83fe5ee802SMichael J. Spencer }
84