1 //===- Binary.cpp - A generic binary file -----------------------*- 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 Binary class.
11 //
12 //===----------------------------------------------------------------------===//
13
14 #include "llvm/Object/Binary.h"
15 #include "llvm/ADT/StringRef.h"
16 #include "llvm/Support/MemoryBuffer.h"
17 #include "llvm/Support/Path.h"
18
19 // Include headers for createBinary.
20 #include "llvm/Object/Archive.h"
21 #include "llvm/Object/COFF.h"
22 #include "llvm/Object/ObjectFile.h"
23
24 using namespace llvm;
25 using namespace object;
26
~Binary()27 Binary::~Binary() {
28 delete Data;
29 }
30
Binary(unsigned int Type,MemoryBuffer * Source)31 Binary::Binary(unsigned int Type, MemoryBuffer *Source)
32 : TypeID(Type)
33 , Data(Source) {}
34
getData() const35 StringRef Binary::getData() const {
36 return Data->getBuffer();
37 }
38
getFileName() const39 StringRef Binary::getFileName() const {
40 return Data->getBufferIdentifier();
41 }
42
createBinary(MemoryBuffer * Source,OwningPtr<Binary> & Result)43 error_code object::createBinary(MemoryBuffer *Source,
44 OwningPtr<Binary> &Result) {
45 OwningPtr<MemoryBuffer> scopedSource(Source);
46 if (!Source)
47 return make_error_code(errc::invalid_argument);
48 if (Source->getBufferSize() < 64)
49 return object_error::invalid_file_type;
50 sys::LLVMFileType type = sys::IdentifyFileType(Source->getBufferStart(),
51 static_cast<unsigned>(Source->getBufferSize()));
52 error_code ec;
53 switch (type) {
54 case sys::Archive_FileType: {
55 OwningPtr<Binary> ret(new Archive(scopedSource.take(), ec));
56 if (ec) return ec;
57 Result.swap(ret);
58 return object_error::success;
59 }
60 case sys::ELF_Relocatable_FileType:
61 case sys::ELF_Executable_FileType:
62 case sys::ELF_SharedObject_FileType:
63 case sys::ELF_Core_FileType: {
64 OwningPtr<Binary> ret(
65 ObjectFile::createELFObjectFile(scopedSource.take()));
66 if (!ret)
67 return object_error::invalid_file_type;
68 Result.swap(ret);
69 return object_error::success;
70 }
71 case sys::Mach_O_Object_FileType:
72 case sys::Mach_O_Executable_FileType:
73 case sys::Mach_O_FixedVirtualMemorySharedLib_FileType:
74 case sys::Mach_O_Core_FileType:
75 case sys::Mach_O_PreloadExecutable_FileType:
76 case sys::Mach_O_DynamicallyLinkedSharedLib_FileType:
77 case sys::Mach_O_DynamicLinker_FileType:
78 case sys::Mach_O_Bundle_FileType:
79 case sys::Mach_O_DynamicallyLinkedSharedLibStub_FileType: {
80 OwningPtr<Binary> ret(
81 ObjectFile::createMachOObjectFile(scopedSource.take()));
82 if (!ret)
83 return object_error::invalid_file_type;
84 Result.swap(ret);
85 return object_error::success;
86 }
87 case sys::COFF_FileType: {
88 OwningPtr<Binary> ret(new COFFObjectFile(scopedSource.take(), ec));
89 if (ec) return ec;
90 Result.swap(ret);
91 return object_error::success;
92 }
93 default: // Unrecognized object file format.
94 return object_error::invalid_file_type;
95 }
96 }
97
createBinary(StringRef Path,OwningPtr<Binary> & Result)98 error_code object::createBinary(StringRef Path, OwningPtr<Binary> &Result) {
99 OwningPtr<MemoryBuffer> File;
100 if (error_code ec = MemoryBuffer::getFileOrSTDIN(Path, File))
101 return ec;
102 return createBinary(File.take(), Result);
103 }
104