mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Giacomo Vercesi
ca66b4ddfc
This commit fixes command line parsing of revng through the REVNG_OPTIONS environment variable. Previously the env was prepended to the actual command-line variables, now it's appended, allowing to specify command line arguments that depend on `-load`s.
64 lines
2.0 KiB
C++
64 lines
2.0 KiB
C++
#pragma once
|
|
|
|
//
|
|
// This file is distributed under the MIT License. See LICENSE.md for details.
|
|
//
|
|
|
|
#include "llvm/Support/CommandLine.h"
|
|
#include "llvm/Support/InitLLVM.h"
|
|
#include "llvm/Support/PrettyStackTrace.h"
|
|
#include "llvm/Support/Process.h"
|
|
|
|
#include "revng/Support/Statistics.h"
|
|
|
|
namespace revng {
|
|
|
|
/// Performs initialization and shutdown steps for revng tools.
|
|
///
|
|
/// By default this performs the regular LLVM initialization steps.
|
|
/// This is required in order to initialize the stack trace printers on signal.
|
|
class InitRevng : public llvm::InitLLVM {
|
|
public:
|
|
InitRevng(int &Argc,
|
|
auto **&Argv,
|
|
const char *Overview,
|
|
llvm::ArrayRef<const llvm::cl::OptionCategory *> CategoriesToHide) :
|
|
InitLLVM(Argc, Argv, true) {
|
|
|
|
OnQuit->install();
|
|
|
|
llvm::setBugReportMsg("PLEASE submit a bug report to "
|
|
"https://github.com/revng/revng and include the "
|
|
"crash backtrace\n");
|
|
|
|
if (not CategoriesToHide.empty()) {
|
|
// NOLINTNEXTLINE
|
|
llvm::cl::HideUnrelatedOptions(CategoriesToHide);
|
|
}
|
|
|
|
// The EnvVar option in llvm::cl::ParseCommandLineOptions prepends the
|
|
// contents of the EnvVar environment variable to the argv, this is
|
|
// undesirable since we have `-load`s in the argv. Here we parse the env
|
|
// manually (if present) and append the command-line options at the end of
|
|
// the argv.
|
|
llvm::BumpPtrAllocator A;
|
|
llvm::StringSaver Saver(A);
|
|
llvm::SmallVector<const char *, 0> Arguments(Argv, Argv + Argc);
|
|
if (auto EnvValue = llvm::sys::Process::GetEnv("REVNG_OPTIONS")) {
|
|
llvm::cl::TokenizeGNUCommandLine(*EnvValue, Saver, Arguments);
|
|
}
|
|
|
|
// NOLINTNEXTLINE
|
|
bool Result = llvm::cl::ParseCommandLineOptions(Arguments.size(),
|
|
Arguments.data(),
|
|
Overview);
|
|
|
|
if (not Result)
|
|
std::exit(EXIT_FAILURE);
|
|
}
|
|
|
|
~InitRevng() { OnQuit->quit(); }
|
|
};
|
|
|
|
} // namespace revng
|