mirror of
https://github.com/carbon-language/carbon-lang.git
synced 2026-09-24 21:20:11 +01:00
Chandler Carruth
437d3b9af6
We end up needing to do this in any driver subcommand that reaches into external code that may not be fully fuzz-clean. No need to grow multiple different diagnostics for each, we can use a common diagnostic.
65 lines
2.1 KiB
C++
65 lines
2.1 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
|
|
|
|
#ifndef CARBON_TOOLCHAIN_DRIVER_DRIVER_SUBCOMMAND_H_
|
|
#define CARBON_TOOLCHAIN_DRIVER_DRIVER_SUBCOMMAND_H_
|
|
|
|
#include "common/command_line.h"
|
|
#include "common/ostream.h"
|
|
#include "llvm/Support/VirtualFileSystem.h"
|
|
#include "toolchain/driver/driver_env.h"
|
|
#include "toolchain/install/install_paths.h"
|
|
|
|
namespace Carbon {
|
|
|
|
// The result of a driver run.
|
|
struct DriverResult {
|
|
// Overall success result.
|
|
bool success;
|
|
|
|
// Per-file success results. May be empty if files aren't individually
|
|
// processed.
|
|
llvm::SmallVector<std::pair<std::string, bool>> per_file_success = {};
|
|
};
|
|
|
|
// A subcommand for the driver.
|
|
class DriverSubcommand {
|
|
public:
|
|
explicit DriverSubcommand(CommandLine::CommandInfo info) : info_(info) {}
|
|
|
|
// Adds the subcommand to the main command, assigning `selected_command` when
|
|
// the subcommand is in use.
|
|
auto AddTo(CommandLine::CommandBuilder& b,
|
|
DriverSubcommand** selected_subcommand) -> void {
|
|
b.AddSubcommand(info_, [this, selected_subcommand](
|
|
CommandLine::CommandBuilder& sub_b) {
|
|
BuildOptions(sub_b);
|
|
sub_b.Do([this, selected_subcommand] { *selected_subcommand = this; });
|
|
});
|
|
}
|
|
|
|
// Adds command line options.
|
|
virtual auto BuildOptions(CommandLine::CommandBuilder& b) -> void = 0;
|
|
|
|
// Runs the command.
|
|
virtual auto Run(DriverEnv& driver_env) -> DriverResult = 0;
|
|
|
|
protected:
|
|
// Diagnoses and returns false if currently fuzzing.
|
|
//
|
|
// This should be used in subcommands to check and diagnose rather than
|
|
// entering them during fuzzing when they use external libraries that we can't
|
|
// keep fuzz-clean.
|
|
auto DisableFuzzingExternalLibraries(DriverEnv& driver_env,
|
|
llvm::StringRef name) -> bool;
|
|
|
|
private:
|
|
// Subcommand information.
|
|
CommandLine::CommandInfo info_;
|
|
};
|
|
|
|
} // namespace Carbon
|
|
|
|
#endif // CARBON_TOOLCHAIN_DRIVER_DRIVER_SUBCOMMAND_H_
|