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"
17fe5ee802SMichael J. Spencer #include "llvm/Support/MemoryBuffer.h"
18447762daSMichael J. Spencer #include "llvm/Support/Path.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 
26*717c4d44SRafael Espindola ObjectFile::ObjectFile(unsigned int Type, MemoryBuffer *source)
27ec29b121SMichael J. Spencer   : Binary(Type, source) {
28fe5ee802SMichael J. Spencer }
29fe5ee802SMichael J. Spencer 
30fe5ee802SMichael J. Spencer ObjectFile *ObjectFile::createObjectFile(MemoryBuffer *Object) {
31fe5ee802SMichael J. Spencer   if (!Object || Object->getBufferSize() < 64)
32fe5ee802SMichael J. Spencer     return 0;
33fe5ee802SMichael J. Spencer   sys::LLVMFileType type = sys::IdentifyFileType(Object->getBufferStart(),
34fe5ee802SMichael J. Spencer                                 static_cast<unsigned>(Object->getBufferSize()));
35fe5ee802SMichael J. Spencer   switch (type) {
36a5a4ff5aSEli Bendersky     case sys::Unknown_FileType:
37a5a4ff5aSEli Bendersky       return 0;
38fe5ee802SMichael J. Spencer     case sys::ELF_Relocatable_FileType:
39fe5ee802SMichael J. Spencer     case sys::ELF_Executable_FileType:
40fe5ee802SMichael J. Spencer     case sys::ELF_SharedObject_FileType:
41fe5ee802SMichael J. Spencer     case sys::ELF_Core_FileType:
42b60a18deSMichael J. Spencer       return createELFObjectFile(Object);
43fe5ee802SMichael J. Spencer     case sys::Mach_O_Object_FileType:
44fe5ee802SMichael J. Spencer     case sys::Mach_O_Executable_FileType:
45fe5ee802SMichael J. Spencer     case sys::Mach_O_FixedVirtualMemorySharedLib_FileType:
46fe5ee802SMichael J. Spencer     case sys::Mach_O_Core_FileType:
47ee5ba319SPeter Collingbourne     case sys::Mach_O_PreloadExecutable_FileType:
48fe5ee802SMichael J. Spencer     case sys::Mach_O_DynamicallyLinkedSharedLib_FileType:
49fe5ee802SMichael J. Spencer     case sys::Mach_O_DynamicLinker_FileType:
50fe5ee802SMichael J. Spencer     case sys::Mach_O_Bundle_FileType:
51fe5ee802SMichael J. Spencer     case sys::Mach_O_DynamicallyLinkedSharedLibStub_FileType:
52557a81e9SBenjamin Kramer     case sys::Mach_O_DSYMCompanion_FileType:
53bafb9347SEric Christopher       return createMachOObjectFile(Object);
54fe5ee802SMichael J. Spencer     case sys::COFF_FileType:
558e90adafSMichael J. Spencer       return createCOFFObjectFile(Object);
56fe5ee802SMichael J. Spencer     default:
57a5a4ff5aSEli Bendersky       llvm_unreachable("Unexpected Object File Type");
58fe5ee802SMichael J. Spencer   }
59fe5ee802SMichael J. Spencer }
60fe5ee802SMichael J. Spencer 
61f177d16fSMichael J. Spencer ObjectFile *ObjectFile::createObjectFile(StringRef ObjectPath) {
6239a0ffc3SMichael J. Spencer   OwningPtr<MemoryBuffer> File;
632d3dac3eSRafael Espindola   if (MemoryBuffer::getFile(ObjectPath, File))
6439a0ffc3SMichael J. Spencer     return NULL;
6539a0ffc3SMichael J. Spencer   return createObjectFile(File.take());
66fe5ee802SMichael J. Spencer }
67