| 1 | //===- SymbolicFile.cpp - Interface that only provides symbols ------------===// |
| 2 | // |
| 3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
| 4 | // See https://llvm.org/LICENSE.txt for license information. |
| 5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| 6 | // |
| 7 | //===----------------------------------------------------------------------===// |
| 8 | // |
| 9 | // This file defines a file format independent SymbolicFile class. |
| 10 | // |
| 11 | //===----------------------------------------------------------------------===// |
| 12 | |
| 13 | #include "llvm/Object/SymbolicFile.h" |
| 14 | #include "llvm/ADT/StringRef.h" |
| 15 | #include "llvm/BinaryFormat/Magic.h" |
| 16 | #include "llvm/Object/COFFImportFile.h" |
| 17 | #include "llvm/Object/Error.h" |
| 18 | #include "llvm/Object/IRObjectFile.h" |
| 19 | #include "llvm/Object/ObjectFile.h" |
| 20 | #include "llvm/Support/Error.h" |
| 21 | #include "llvm/Support/ErrorHandling.h" |
| 22 | #include <memory> |
| 23 | |
| 24 | using namespace llvm; |
| 25 | using namespace object; |
| 26 | |
| 27 | namespace llvm { |
| 28 | class LLVMContext; |
| 29 | } |
| 30 | |
| 31 | SymbolicFile::SymbolicFile(unsigned int Type, MemoryBufferRef Source) |
| 32 | : Binary(Type, Source) {} |
| 33 | |
| 34 | SymbolicFile::~SymbolicFile() = default; |
| 35 | |
| 36 | Expected<std::unique_ptr<SymbolicFile>> |
| 37 | SymbolicFile::createSymbolicFile(MemoryBufferRef Object, file_magic Type, |
| 38 | LLVMContext *Context, bool InitContent) { |
| 39 | StringRef Data = Object.getBuffer(); |
| 40 | if (Type == file_magic::unknown) |
| 41 | Type = identify_magic(magic: Data); |
| 42 | |
| 43 | if (!isSymbolicFile(Type, Context)) |
| 44 | return errorCodeToError(EC: object_error::invalid_file_type); |
| 45 | |
| 46 | switch (Type) { |
| 47 | case file_magic::bitcode: |
| 48 | // Context is guaranteed to be non-null here, because bitcode magic only |
| 49 | // indicates a symbolic file when Context is non-null. |
| 50 | return IRObjectFile::create(Object, Context&: *Context); |
| 51 | case file_magic::elf: |
| 52 | case file_magic::elf_executable: |
| 53 | case file_magic::elf_shared_object: |
| 54 | case file_magic::elf_core: |
| 55 | case file_magic::goff_object: |
| 56 | case file_magic::macho_executable: |
| 57 | case file_magic::macho_fixed_virtual_memory_shared_lib: |
| 58 | case file_magic::macho_core: |
| 59 | case file_magic::macho_preload_executable: |
| 60 | case file_magic::macho_dynamically_linked_shared_lib: |
| 61 | case file_magic::macho_dynamic_linker: |
| 62 | case file_magic::macho_bundle: |
| 63 | case file_magic::macho_dynamically_linked_shared_lib_stub: |
| 64 | case file_magic::macho_dsym_companion: |
| 65 | case file_magic::macho_kext_bundle: |
| 66 | case file_magic::macho_file_set: |
| 67 | case file_magic::pecoff_executable: |
| 68 | case file_magic::xcoff_object_32: |
| 69 | case file_magic::xcoff_object_64: |
| 70 | case file_magic::wasm_object: |
| 71 | return ObjectFile::createObjectFile(Object, Type, InitContent); |
| 72 | case file_magic::coff_import_library: |
| 73 | return std::unique_ptr<SymbolicFile>(new COFFImportFile(Object)); |
| 74 | case file_magic::elf_relocatable: |
| 75 | case file_magic::macho_object: |
| 76 | case file_magic::coff_object: { |
| 77 | Expected<std::unique_ptr<ObjectFile>> Obj = |
| 78 | ObjectFile::createObjectFile(Object, Type, InitContent); |
| 79 | if (!Obj || !Context) |
| 80 | return std::move(Obj); |
| 81 | |
| 82 | Expected<MemoryBufferRef> BCData = |
| 83 | IRObjectFile::findBitcodeInObject(Obj: *Obj->get()); |
| 84 | if (!BCData) { |
| 85 | consumeError(Err: BCData.takeError()); |
| 86 | return std::move(Obj); |
| 87 | } |
| 88 | |
| 89 | return IRObjectFile::create( |
| 90 | Object: MemoryBufferRef(BCData->getBuffer(), Object.getBufferIdentifier()), |
| 91 | Context&: *Context); |
| 92 | } |
| 93 | default: |
| 94 | llvm_unreachable("Unexpected Binary File Type" ); |
| 95 | } |
| 96 | } |
| 97 | |
| 98 | bool SymbolicFile::isSymbolicFile(file_magic Type, const LLVMContext *Context) { |
| 99 | switch (Type) { |
| 100 | case file_magic::bitcode: |
| 101 | return Context != nullptr; |
| 102 | case file_magic::elf: |
| 103 | case file_magic::elf_executable: |
| 104 | case file_magic::elf_shared_object: |
| 105 | case file_magic::elf_core: |
| 106 | case file_magic::goff_object: |
| 107 | case file_magic::macho_executable: |
| 108 | case file_magic::macho_fixed_virtual_memory_shared_lib: |
| 109 | case file_magic::macho_core: |
| 110 | case file_magic::macho_preload_executable: |
| 111 | case file_magic::macho_dynamically_linked_shared_lib: |
| 112 | case file_magic::macho_dynamic_linker: |
| 113 | case file_magic::macho_bundle: |
| 114 | case file_magic::macho_dynamically_linked_shared_lib_stub: |
| 115 | case file_magic::macho_dsym_companion: |
| 116 | case file_magic::macho_kext_bundle: |
| 117 | case file_magic::macho_file_set: |
| 118 | case file_magic::pecoff_executable: |
| 119 | case file_magic::xcoff_object_32: |
| 120 | case file_magic::xcoff_object_64: |
| 121 | case file_magic::wasm_object: |
| 122 | case file_magic::coff_import_library: |
| 123 | case file_magic::elf_relocatable: |
| 124 | case file_magic::macho_object: |
| 125 | case file_magic::coff_object: |
| 126 | return true; |
| 127 | default: |
| 128 | return false; |
| 129 | } |
| 130 | } |
| 131 | |