1//===- Objcopy.cpp --------------------------------------------------------===//
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#include "llvm/ObjCopy/ObjCopy.h"
10#include "llvm/ObjCopy/COFF/COFFConfig.h"
11#include "llvm/ObjCopy/COFF/COFFObjcopy.h"
12#include "llvm/ObjCopy/DXContainer/DXContainerConfig.h"
13#include "llvm/ObjCopy/DXContainer/DXContainerObjcopy.h"
14#include "llvm/ObjCopy/ELF/ELFConfig.h"
15#include "llvm/ObjCopy/ELF/ELFObjcopy.h"
16#include "llvm/ObjCopy/MachO/MachOConfig.h"
17#include "llvm/ObjCopy/MachO/MachOObjcopy.h"
18#include "llvm/ObjCopy/MultiFormatConfig.h"
19#include "llvm/ObjCopy/XCOFF/XCOFFConfig.h"
20#include "llvm/ObjCopy/XCOFF/XCOFFObjcopy.h"
21#include "llvm/ObjCopy/wasm/WasmConfig.h"
22#include "llvm/ObjCopy/wasm/WasmObjcopy.h"
23#include "llvm/Object/COFF.h"
24#include "llvm/Object/DXContainer.h"
25#include "llvm/Object/ELFObjectFile.h"
26#include "llvm/Object/Error.h"
27#include "llvm/Object/MachO.h"
28#include "llvm/Object/MachOUniversal.h"
29#include "llvm/Object/Wasm.h"
30#include "llvm/Object/XCOFFObjectFile.h"
31
32using namespace llvm;
33using namespace llvm::object;
34
35StringRef objcopy::getObjectFormatName(const object::Binary &B) {
36 if (const auto *OF = dyn_cast<ObjectFile>(Val: &B))
37 return OF->getFileFormatName();
38 return {};
39}
40
41void objcopy::printCopyMessage(StringRef InPath, StringRef InFormatName,
42 StringRef OutPath, StringRef OutFormatName) {
43 if (OutFormatName.empty())
44 OutFormatName = InFormatName;
45 outs() << "copy from '" << InPath << "' [" << InFormatName << "] to '"
46 << OutPath << "' [" << OutFormatName << "]\n";
47}
48
49/// The function executeObjcopyOnBinary does the dispatch based on the format
50/// of the input binary (ELF, MachO or COFF).
51Error objcopy::executeObjcopyOnBinary(const MultiFormatConfig &Config,
52 object::Binary &In, raw_ostream &Out) {
53 if (auto *ELFBinary = dyn_cast<object::ELFObjectFileBase>(Val: &In)) {
54 Expected<const ELFConfig &> ELFConfig = Config.getELFConfig();
55 if (!ELFConfig)
56 return ELFConfig.takeError();
57
58 return elf::executeObjcopyOnBinary(Config: Config.getCommonConfig(), ELFConfig: *ELFConfig,
59 In&: *ELFBinary, Out);
60 }
61 if (auto *COFFBinary = dyn_cast<object::COFFObjectFile>(Val: &In)) {
62 Expected<const COFFConfig &> COFFConfig = Config.getCOFFConfig();
63 if (!COFFConfig)
64 return COFFConfig.takeError();
65
66 return coff::executeObjcopyOnBinary(Config: Config.getCommonConfig(), *COFFConfig,
67 In&: *COFFBinary, Out);
68 }
69 if (auto *MachOBinary = dyn_cast<object::MachOObjectFile>(Val: &In)) {
70 Expected<const MachOConfig &> MachOConfig = Config.getMachOConfig();
71 if (!MachOConfig)
72 return MachOConfig.takeError();
73
74 return macho::executeObjcopyOnBinary(Config: Config.getCommonConfig(), MachOConfig: *MachOConfig,
75 In&: *MachOBinary, Out);
76 }
77 if (auto *MachOUniversalBinary =
78 dyn_cast<object::MachOUniversalBinary>(Val: &In)) {
79 return macho::executeObjcopyOnMachOUniversalBinary(
80 Config, In: *MachOUniversalBinary, Out);
81 }
82 if (auto *WasmBinary = dyn_cast<object::WasmObjectFile>(Val: &In)) {
83 Expected<const WasmConfig &> WasmConfig = Config.getWasmConfig();
84 if (!WasmConfig)
85 return WasmConfig.takeError();
86
87 return objcopy::wasm::executeObjcopyOnBinary(Config: Config.getCommonConfig(),
88 *WasmConfig, In&: *WasmBinary, Out);
89 }
90 if (auto *XCOFFBinary = dyn_cast<object::XCOFFObjectFile>(Val: &In)) {
91 Expected<const XCOFFConfig &> XCOFFConfig = Config.getXCOFFConfig();
92 if (!XCOFFConfig)
93 return XCOFFConfig.takeError();
94
95 return xcoff::executeObjcopyOnBinary(Config: Config.getCommonConfig(), *XCOFFConfig,
96 In&: *XCOFFBinary, Out);
97 }
98 if (auto *DXContainerBinary = dyn_cast<object::DXContainerObjectFile>(Val: &In)) {
99 Expected<const DXContainerConfig &> DXContainerConfig =
100 Config.getDXContainerConfig();
101 if (!DXContainerConfig)
102 return DXContainerConfig.takeError();
103
104 return dxbc::executeObjcopyOnBinary(
105 Config: Config.getCommonConfig(), *DXContainerConfig, In&: *DXContainerBinary, Out);
106 }
107 return createStringError(EC: object_error::invalid_file_type,
108 S: "unsupported object file format");
109}
110