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/FileSystem.h"
18 #include "llvm/Support/Path.h"
19
20 // Include headers for createBinary.
21 #include "llvm/Object/Archive.h"
22 #include "llvm/Object/COFF.h"
23 #include "llvm/Object/MachOUniversal.h"
24 #include "llvm/Object/ObjectFile.h"
25
26 using namespace llvm;
27 using namespace object;
28
~Binary()29 Binary::~Binary() {
30 delete Data;
31 }
32
Binary(unsigned int Type,MemoryBuffer * Source)33 Binary::Binary(unsigned int Type, MemoryBuffer *Source)
34 : TypeID(Type)
35 , Data(Source) {}
36
getData() const37 StringRef Binary::getData() const {
38 return Data->getBuffer();
39 }
40
getFileName() const41 StringRef Binary::getFileName() const {
42 return Data->getBufferIdentifier();
43 }
44
createBinary(MemoryBuffer * Source,OwningPtr<Binary> & Result)45 error_code object::createBinary(MemoryBuffer *Source,
46 OwningPtr<Binary> &Result) {
47 OwningPtr<MemoryBuffer> scopedSource(Source);
48 if (!Source)
49 return make_error_code(errc::invalid_argument);
50 sys::fs::file_magic type = sys::fs::identify_magic(Source->getBuffer());
51 error_code ec;
52 switch (type) {
53 case sys::fs::file_magic::archive: {
54 OwningPtr<Binary> ret(new Archive(scopedSource.take(), ec));
55 if (ec) return ec;
56 Result.swap(ret);
57 return object_error::success;
58 }
59 case sys::fs::file_magic::elf_relocatable:
60 case sys::fs::file_magic::elf_executable:
61 case sys::fs::file_magic::elf_shared_object:
62 case sys::fs::file_magic::elf_core: {
63 OwningPtr<Binary> ret(
64 ObjectFile::createELFObjectFile(scopedSource.take()));
65 if (!ret)
66 return object_error::invalid_file_type;
67 Result.swap(ret);
68 return object_error::success;
69 }
70 case sys::fs::file_magic::macho_object:
71 case sys::fs::file_magic::macho_executable:
72 case sys::fs::file_magic::macho_fixed_virtual_memory_shared_lib:
73 case sys::fs::file_magic::macho_core:
74 case sys::fs::file_magic::macho_preload_executable:
75 case sys::fs::file_magic::macho_dynamically_linked_shared_lib:
76 case sys::fs::file_magic::macho_dynamic_linker:
77 case sys::fs::file_magic::macho_bundle:
78 case sys::fs::file_magic::macho_dynamically_linked_shared_lib_stub:
79 case sys::fs::file_magic::macho_dsym_companion: {
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::fs::file_magic::macho_universal_binary: {
88 OwningPtr<Binary> ret(new MachOUniversalBinary(scopedSource.take(), ec));
89 if (ec) return ec;
90 Result.swap(ret);
91 return object_error::success;
92 }
93 case sys::fs::file_magic::coff_object:
94 case sys::fs::file_magic::pecoff_executable: {
95 OwningPtr<Binary> ret(
96 ObjectFile::createCOFFObjectFile(scopedSource.take()));
97 if (!ret)
98 return object_error::invalid_file_type;
99 Result.swap(ret);
100 return object_error::success;
101 }
102 case sys::fs::file_magic::unknown:
103 case sys::fs::file_magic::bitcode: {
104 // Unrecognized object file format.
105 return object_error::invalid_file_type;
106 }
107 }
108 llvm_unreachable("Unexpected Binary File Type");
109 }
110
createBinary(StringRef Path,OwningPtr<Binary> & Result)111 error_code object::createBinary(StringRef Path, OwningPtr<Binary> &Result) {
112 OwningPtr<MemoryBuffer> File;
113 if (error_code ec = MemoryBuffer::getFileOrSTDIN(Path, File))
114 return ec;
115 return createBinary(File.take(), Result);
116 }
117