Files
carbon-lang/toolchain/driver/driver_main.cpp
T
Richard Smith db35d17b6c Don't canonicalize prefixes when locating the driver binary. (#3912)
The approach here directly follows the one used by Clang's driver.
2024-04-24 05:24:29 +00:00

50 lines
1.6 KiB
C++

// Part of the Carbon Language project, under the Apache License v2.0 with LLVM
// Exceptions. See /LICENSE for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
#include <cstdlib>
#include "common/bazel_working_dir.h"
#include "common/init_llvm.h"
#include "llvm/ADT/SmallVector.h"
#include "llvm/ADT/StringRef.h"
#include "llvm/Support/Path.h"
#include "llvm/Support/Program.h"
#include "toolchain/driver/driver.h"
auto main(int argc, char** argv) -> int {
Carbon::InitLLVM init_llvm(argc, argv);
if (argc < 1) {
return EXIT_FAILURE;
}
// Find the executable without resolving symlinks. Do a PATH lookup if argv[0]
// isn't a valid path.
llvm::SmallString<128> exe_path(argv[0]);
if (!llvm::sys::fs::exists(exe_path)) {
if (llvm::ErrorOr<std::string> path =
llvm::sys::findProgramByName(exe_path)) {
exe_path = *path;
}
}
Carbon::SetWorkingDirForBazel();
// Printing to stderr should flush stdout. This is most noticeable when stderr
// is piped to stdout.
llvm::errs().tie(&llvm::outs());
llvm::SmallVector<llvm::StringRef> args(argv + 1, argv + argc);
auto fs = llvm::vfs::getRealFileSystem();
// Construct the data directory relative to the executable location.
llvm::SmallString<256> data_dir(llvm::sys::path::parent_path(exe_path));
llvm::sys::path::append(data_dir, llvm::sys::path::Style::posix,
"carbon.runfiles/_main/");
Carbon::Driver driver(*fs, data_dir, llvm::outs(), llvm::errs());
bool success = driver.RunCommand(args).success;
return success ? EXIT_SUCCESS : EXIT_FAILURE;
}