mirror of
https://github.com/RPCS3/llvm-mirror.git
synced 2024-11-23 03:02:36 +01:00
5ef47e6205
This teaches obj2yaml to dump valid regular (not thin) archives. This also teaches yaml2obj to recognize archives YAML descriptions, what allows to craft all different kinds of archives (valid and broken ones). Differential revision: https://reviews.llvm.org/D89949
91 lines
2.9 KiB
C++
91 lines
2.9 KiB
C++
//===------ utils/obj2yaml.cpp - obj2yaml conversion tool -------*- C++ -*-===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "obj2yaml.h"
|
|
#include "llvm/BinaryFormat/Magic.h"
|
|
#include "llvm/Object/Archive.h"
|
|
#include "llvm/Object/COFF.h"
|
|
#include "llvm/Object/Minidump.h"
|
|
#include "llvm/Support/CommandLine.h"
|
|
#include "llvm/Support/Errc.h"
|
|
#include "llvm/Support/InitLLVM.h"
|
|
|
|
using namespace llvm;
|
|
using namespace llvm::object;
|
|
|
|
static Error dumpObject(const ObjectFile &Obj) {
|
|
if (Obj.isCOFF())
|
|
return errorCodeToError(coff2yaml(outs(), cast<COFFObjectFile>(Obj)));
|
|
|
|
if (Obj.isXCOFF())
|
|
return errorCodeToError(xcoff2yaml(outs(), cast<XCOFFObjectFile>(Obj)));
|
|
|
|
if (Obj.isELF())
|
|
return elf2yaml(outs(), Obj);
|
|
|
|
if (Obj.isWasm())
|
|
return errorCodeToError(wasm2yaml(outs(), cast<WasmObjectFile>(Obj)));
|
|
|
|
llvm_unreachable("unexpected object file format");
|
|
}
|
|
|
|
static Error dumpInput(StringRef File) {
|
|
ErrorOr<std::unique_ptr<MemoryBuffer>> FileOrErr =
|
|
MemoryBuffer::getFileOrSTDIN(File, /*FileSize=*/-1,
|
|
/*RequiresNullTerminator=*/false);
|
|
if (std::error_code EC = FileOrErr.getError())
|
|
return errorCodeToError(EC);
|
|
std::unique_ptr<MemoryBuffer> &Buffer = FileOrErr.get();
|
|
MemoryBufferRef MemBuf = Buffer->getMemBufferRef();
|
|
if (file_magic::archive == identify_magic(MemBuf.getBuffer()))
|
|
return archive2yaml(outs(), MemBuf);
|
|
|
|
Expected<std::unique_ptr<Binary>> BinOrErr =
|
|
createBinary(MemBuf, /*Context=*/nullptr);
|
|
if (!BinOrErr)
|
|
return BinOrErr.takeError();
|
|
|
|
Binary &Binary = *BinOrErr->get();
|
|
// Universal MachO is not a subclass of ObjectFile, so it needs to be handled
|
|
// here with the other binary types.
|
|
if (Binary.isMachO() || Binary.isMachOUniversalBinary())
|
|
return macho2yaml(outs(), Binary);
|
|
if (ObjectFile *Obj = dyn_cast<ObjectFile>(&Binary))
|
|
return dumpObject(*Obj);
|
|
if (MinidumpFile *Minidump = dyn_cast<MinidumpFile>(&Binary))
|
|
return minidump2yaml(outs(), *Minidump);
|
|
|
|
return Error::success();
|
|
}
|
|
|
|
static void reportError(StringRef Input, Error Err) {
|
|
if (Input == "-")
|
|
Input = "<stdin>";
|
|
std::string ErrMsg;
|
|
raw_string_ostream OS(ErrMsg);
|
|
logAllUnhandledErrors(std::move(Err), OS);
|
|
OS.flush();
|
|
errs() << "Error reading file: " << Input << ": " << ErrMsg;
|
|
errs().flush();
|
|
}
|
|
|
|
cl::opt<std::string> InputFilename(cl::Positional, cl::desc("<input file>"),
|
|
cl::init("-"));
|
|
|
|
int main(int argc, char *argv[]) {
|
|
InitLLVM X(argc, argv);
|
|
cl::ParseCommandLineOptions(argc, argv);
|
|
|
|
if (Error Err = dumpInput(InputFilename)) {
|
|
reportError(InputFilename, std::move(Err));
|
|
return 1;
|
|
}
|
|
|
|
return 0;
|
|
}
|