mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Pietro Fezzardi
99 lines
3.2 KiB
C++
99 lines
3.2 KiB
C++
//
|
|
// Copyright (c) rev.ng Srls. See LICENSE.md for details.
|
|
//
|
|
|
|
#include <memory>
|
|
#include <string>
|
|
|
|
#include "llvm/IR/LLVMContext.h"
|
|
#include "llvm/IR/Module.h"
|
|
#include "llvm/IRReader/IRReader.h"
|
|
#include "llvm/Support/CommandLine.h"
|
|
#include "llvm/Support/Signals.h"
|
|
#include "llvm/Support/SourceMgr.h"
|
|
|
|
#include "revng/Model/LoadModelPass.h"
|
|
#include "revng/Support/Assert.h"
|
|
|
|
#include "revng-c/ModelToHeader/ModelToHeader.h"
|
|
|
|
static llvm::cl::OptionCategory ModelToHeaderCategory("ModelToHeaderOptions");
|
|
|
|
static const char *Overview = "Standalone tool that ingests an instance of "
|
|
"revng's model, embedded into an\nllvm IR file, "
|
|
"and produces a C11 header file with all the "
|
|
"declarations of\ntypes and functions associated "
|
|
"to the model's type system.";
|
|
|
|
static llvm::cl::opt<bool> InputIsYAML("yaml",
|
|
llvm::cl::cat(ModelToHeaderCategory),
|
|
llvm::cl::Optional,
|
|
llvm::cl::init(false),
|
|
llvm::cl::desc("Set to true if input is "
|
|
"yaml, not llvm IR"));
|
|
|
|
using llvm::LLVMContext;
|
|
using llvm::Module;
|
|
|
|
int main(int Argc, const char *Argv[]) {
|
|
|
|
// Enable LLVM stack trace
|
|
llvm::sys::PrintStackTraceOnErrorSignal(Argv[0]);
|
|
|
|
// Hide options not related to this tool
|
|
llvm::cl::HideUnrelatedOptions({ &ModelToHeaderCategory });
|
|
|
|
// Parse command line
|
|
bool Ok = llvm::cl::ParseCommandLineOptions(Argc,
|
|
Argv,
|
|
Overview,
|
|
&llvm::errs());
|
|
if (not Ok)
|
|
std::exit(EXIT_FAILURE);
|
|
|
|
auto Buffer = llvm::MemoryBuffer::getSTDIN();
|
|
if (std::error_code EC = Buffer.getError())
|
|
revng_abort(EC.message().c_str());
|
|
|
|
// Try to load the module from stdin, either in yaml or bitcode format
|
|
TupleTree<model::Binary> Model;
|
|
if (InputIsYAML) {
|
|
|
|
// If input is YAML, get it from yaml
|
|
auto M = TupleTree<model::Binary>::deserialize(Buffer.get()->getBuffer());
|
|
if (std::error_code EC = M.getError())
|
|
revng_abort(EC.message().c_str());
|
|
|
|
Model = std::move(M.get());
|
|
} else {
|
|
// If we're loading from LLVM IR, do it lazily, since we don't really need
|
|
// to deserialize function bodies for converting the model to a header. We
|
|
// just need to loo, into the model metadata.
|
|
llvm::SMDiagnostic Diag;
|
|
LLVMContext Ctx;
|
|
std::unique_ptr<Module> M = llvm::getLazyIRModule(std::move(Buffer.get()),
|
|
Diag,
|
|
Ctx);
|
|
if (not M) {
|
|
Diag.print(Argv[0], llvm::errs());
|
|
std::exit(EXIT_FAILURE);
|
|
}
|
|
|
|
Model = loadModel(*M);
|
|
}
|
|
|
|
if (not Model.verify()) {
|
|
llvm::errs() << "Invalid Model\n";
|
|
std::exit(EXIT_FAILURE);
|
|
}
|
|
|
|
std::error_code EC;
|
|
llvm::raw_fd_ostream Header{ "-" /* for stdout */, EC };
|
|
if (EC)
|
|
revng_abort(EC.message().c_str());
|
|
|
|
dumpModelToHeader(*Model, Header);
|
|
|
|
return 0;
|
|
}
|