mirror of
https://github.com/RPCS3/llvm.git
synced 2026-07-19 15:13:49 -04:00
Derek Schuff
349a48fadd
Mostly this change adds support converting to and from YAML which will allow us to write more test cases for the WebAssembly MC and lld ports. Better support for objdump, readelf, and nm will be in followup CLs. I had to update the two wasm test binaries because they used the old style 'name' section which is no longer supported. Differential Revision: https://reviews.llvm.org/D31099 Patch by Sam Clegg git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@299101 91177308-0d34-0410-b5e6-96231b3b80d8
102 lines
2.9 KiB
C++
102 lines
2.9 KiB
C++
//===- yaml2obj - Convert YAML to a binary object file --------------------===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This program takes a YAML description of an object file and outputs the
|
|
// binary equivalent.
|
|
//
|
|
// This is used for writing tests that require binary files.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "yaml2obj.h"
|
|
#include "llvm/ADT/StringExtras.h"
|
|
#include "llvm/ObjectYAML/ObjectYAML.h"
|
|
#include "llvm/Support/CommandLine.h"
|
|
#include "llvm/Support/FileSystem.h"
|
|
#include "llvm/Support/ManagedStatic.h"
|
|
#include "llvm/Support/MemoryBuffer.h"
|
|
#include "llvm/Support/PrettyStackTrace.h"
|
|
#include "llvm/Support/Signals.h"
|
|
#include "llvm/Support/ToolOutputFile.h"
|
|
#include "llvm/Support/YAMLTraits.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
#include <system_error>
|
|
|
|
using namespace llvm;
|
|
|
|
static cl::opt<std::string>
|
|
Input(cl::Positional, cl::desc("<input>"), cl::init("-"));
|
|
|
|
cl::opt<unsigned>
|
|
DocNum("docnum", cl::init(1),
|
|
cl::desc("Read specified document from input (default = 1)"));
|
|
|
|
static cl::opt<std::string> OutputFilename("o", cl::desc("Output filename"),
|
|
cl::value_desc("filename"));
|
|
|
|
static int convertYAML(yaml::Input &YIn, raw_ostream &Out) {
|
|
unsigned CurDocNum = 0;
|
|
do {
|
|
if (++CurDocNum == DocNum) {
|
|
yaml::YamlObjectFile Doc;
|
|
YIn >> Doc;
|
|
if (YIn.error()) {
|
|
errs() << "yaml2obj: Failed to parse YAML file!\n";
|
|
return 1;
|
|
}
|
|
|
|
if (Doc.Elf)
|
|
return yaml2elf(*Doc.Elf, Out);
|
|
if (Doc.Coff)
|
|
return yaml2coff(*Doc.Coff, Out);
|
|
if (Doc.MachO || Doc.FatMachO)
|
|
return yaml2macho(Doc, Out);
|
|
if (Doc.Wasm)
|
|
return yaml2wasm(*Doc.Wasm, Out);
|
|
errs() << "yaml2obj: Unknown document type!\n";
|
|
return 1;
|
|
}
|
|
} while (YIn.nextDocument());
|
|
|
|
errs() << "yaml2obj: Cannot find the " << DocNum
|
|
<< llvm::getOrdinalSuffix(DocNum) << " document\n";
|
|
return 1;
|
|
}
|
|
|
|
int main(int argc, char **argv) {
|
|
cl::ParseCommandLineOptions(argc, argv);
|
|
sys::PrintStackTraceOnErrorSignal(argv[0]);
|
|
PrettyStackTraceProgram X(argc, argv);
|
|
llvm_shutdown_obj Y; // Call llvm_shutdown() on exit.
|
|
|
|
if (OutputFilename.empty())
|
|
OutputFilename = "-";
|
|
|
|
std::error_code EC;
|
|
std::unique_ptr<tool_output_file> Out(
|
|
new tool_output_file(OutputFilename, EC, sys::fs::F_None));
|
|
if (EC) {
|
|
errs() << EC.message() << '\n';
|
|
return 1;
|
|
}
|
|
|
|
ErrorOr<std::unique_ptr<MemoryBuffer>> Buf =
|
|
MemoryBuffer::getFileOrSTDIN(Input);
|
|
if (!Buf)
|
|
return 1;
|
|
|
|
yaml::Input YIn(Buf.get()->getBuffer());
|
|
|
|
int Res = convertYAML(YIn, Out->os());
|
|
if (Res == 0)
|
|
Out->keep();
|
|
|
|
return Res;
|
|
}
|