75 lines
2.2 KiB
C++
75 lines
2.2 KiB
C++
//===-- llvm-driver.cpp ---------------------------------------------------===//
|
|
//
|
|
// 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 "llvm/ADT/StringExtras.h"
|
|
#include "llvm/ADT/StringRef.h"
|
|
#include "llvm/ADT/StringSwitch.h"
|
|
#include "llvm/Support/CommandLine.h"
|
|
#include "llvm/Support/ErrorHandling.h"
|
|
#include "llvm/Support/Path.h"
|
|
#include "llvm/Support/WithColor.h"
|
|
|
|
using namespace llvm;
|
|
|
|
#define LLVM_DRIVER_TOOL(tool, entry) int entry##_main(int argc, char **argv);
|
|
#include "LLVMDriverTools.def"
|
|
|
|
constexpr char subcommands[] =
|
|
#define LLVM_DRIVER_TOOL(tool, entry) " " tool "\n"
|
|
#include "LLVMDriverTools.def"
|
|
;
|
|
|
|
static void printHelpMessage() {
|
|
llvm::outs() << "OVERVIEW: llvm toolchain driver\n\n"
|
|
<< "USAGE: llvm [subcommand] [options]\n\n"
|
|
<< "SUBCOMMANDS:\n\n"
|
|
<< subcommands
|
|
<< "\n Type \"llvm <subcommand> --help\" to get more help on a "
|
|
"specific subcommand\n\n"
|
|
<< "OPTIONS:\n\n --help - Display this message";
|
|
}
|
|
|
|
static int findTool(int Argc, char **Argv) {
|
|
if (!Argc) {
|
|
printHelpMessage();
|
|
return 1;
|
|
}
|
|
|
|
StringRef ToolName = Argv[0];
|
|
|
|
if (ToolName == "--help") {
|
|
printHelpMessage();
|
|
return 0;
|
|
}
|
|
|
|
StringRef Stem = sys::path::stem(ToolName);
|
|
auto Is = [=](StringRef Tool) {
|
|
auto I = Stem.rfind_insensitive(Tool);
|
|
return I != StringRef::npos && (I + Tool.size() == Stem.size() ||
|
|
!llvm::isAlnum(Stem[I + Tool.size()]));
|
|
};
|
|
|
|
#define LLVM_DRIVER_TOOL(tool, entry) \
|
|
if (Is(tool)) \
|
|
return entry##_main(Argc, Argv);
|
|
#include "LLVMDriverTools.def"
|
|
|
|
if (Is("llvm"))
|
|
return findTool(Argc - 1, Argv + 1);
|
|
|
|
printHelpMessage();
|
|
return 1;
|
|
}
|
|
|
|
extern bool IsLLVMDriver;
|
|
|
|
int main(int Argc, char **Argv) {
|
|
IsLLVMDriver = true;
|
|
return findTool(Argc, Argv);
|
|
}
|