mirror of
https://github.com/RPCS3/llvm-mirror.git
synced 2024-11-23 11:13:28 +01:00
676775660f
This is a continuation of the YAML library error reporting refactoring/improvement and the idea by itself was mentioned in the following thread: https://reviews.llvm.org/D67182?id=218714#inline-603404 This performs a cleanup of all object emitters in the library. It allows using the custom one provided by the caller. One of the nice things is that each tool can now print its tool name, e.g: "yaml2obj: error: <text>" Also, the code became a bit simpler. Differential revision: https://reviews.llvm.org/D67445 llvm-svn: 371865
73 lines
2.2 KiB
C++
73 lines
2.2 KiB
C++
//===- yaml2obj - Convert YAML to a binary object file --------------------===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// 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 "llvm/ObjectYAML/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/InitLLVM.h"
|
|
#include "llvm/Support/MemoryBuffer.h"
|
|
#include "llvm/Support/ToolOutputFile.h"
|
|
#include "llvm/Support/WithColor.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"));
|
|
|
|
int main(int argc, char **argv) {
|
|
InitLLVM X(argc, argv);
|
|
cl::ParseCommandLineOptions(argc, argv);
|
|
|
|
if (OutputFilename.empty())
|
|
OutputFilename = "-";
|
|
|
|
auto ErrHandler = [](const Twine &Msg) {
|
|
WithColor::error(errs(), "yaml2obj") << Msg << "\n";
|
|
};
|
|
|
|
std::error_code EC;
|
|
std::unique_ptr<ToolOutputFile> Out(
|
|
new ToolOutputFile(OutputFilename, EC, sys::fs::OF_None));
|
|
if (EC) {
|
|
ErrHandler("failed to open '" + OutputFilename + "': " + EC.message());
|
|
return 1;
|
|
}
|
|
|
|
ErrorOr<std::unique_ptr<MemoryBuffer>> Buf =
|
|
MemoryBuffer::getFileOrSTDIN(Input);
|
|
if (!Buf)
|
|
return 1;
|
|
|
|
yaml::Input YIn(Buf.get()->getBuffer());
|
|
if (!convertYAML(YIn, Out->os(), ErrHandler, DocNum))
|
|
return 1;
|
|
|
|
Out->keep();
|
|
Out->os().flush();
|
|
return 0;
|
|
}
|