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
24using namespace llvm;
25using namespace object;
26
27Binary::~Binary() {
28  delete Data;
29}
30
31Binary::Binary(unsigned int Type, MemoryBuffer *Source)
32  : TypeID(Type)
33  , Data(Source) {}
34
35StringRef Binary::getData() const {
36  return Data->getBuffer();
37}
38
39StringRef Binary::getFileName() const {
40  return Data->getBufferIdentifier();
41}
42
43error_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
98error_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