carbon-lang/explorer/fuzzing/fuzzverter.cpp

// 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

// An utility for converting between fuzzer protos and Carbon sources.
//
// For example, to convert a crashing input in text proto to carbon source:
// `fuzzverter --mode=proto_to_carbon --input file.textproto`
//
// To generate a new text proto from carbon source for seeding the corpus:
// `fuzzverter --mode=carbon_to_proto --input file.carbon`

#include <google/protobuf/text_format.h>

#include <cstdlib>
#include <fstream>
#include <ios>
#include <sstream>

#include "common/error.h"
#include "common/fuzzing/carbon.pb.h"
#include "explorer/common/error_builders.h"
#include "explorer/fuzzing/ast_to_proto.h"
#include "explorer/fuzzing/fuzzer_util.h"
#include "explorer/syntax/parse.h"
#include "llvm/Support/CommandLine.h"
#include "llvm/Support/InitLLVM.h"

namespace Carbon {

namespace cl = llvm::cl;

// Reads a file and returns its contents as a string.
static auto ReadFile(std::string_view file_name) -> ErrorOr<std::string> {
  std::ifstream file(file_name, std::ios::in);
  if (!file.is_open()) {
    return ErrorBuilder() << "Could not open " << file_name << " for reading";
  }
  std::stringstream ss;
  ss << file.rdbuf();
  return ss.str();
}

// Writes string `s` to `file_name`.
static auto WriteFile(std::string_view s, std::string_view file_name)
    -> ErrorOr<Success> {
  std::ofstream file(file_name, std::ios::out);
  if (!file.is_open()) {
    return ErrorBuilder() << "Could not open " << file_name << " for writing";
  }
  file << s;
  return Success();
}

// Converts text proto to Carbon source.
static auto TextProtoToCarbon(std::string_view input_file_name,
                              std::string_view output_file_name)
    -> ErrorOr<Success> {
  CARBON_ASSIGN_OR_RETURN(const std::string input_contents,
                          ReadFile(input_file_name));
  CARBON_ASSIGN_OR_RETURN(const Fuzzing::Carbon carbon_proto,
                          ParseCarbonTextProto(input_contents));
  const std::string carbon_source =
      ProtoToCarbonWithMain(carbon_proto.compilation_unit());
  return WriteFile(carbon_source, output_file_name);
}

// Converts Carbon source to text proto.
static auto CarbonToTextProto(std::string_view input_file_name,
                              std::string_view output_file_name)
    -> ErrorOr<Success> {
  Carbon::Arena arena;
  const ErrorOr<AST> ast = Carbon::Parse(&arena, input_file_name,
                                         /*parser_debug=*/false);
  if (!ast.ok()) {
    return ErrorBuilder() << "Parsing failed: " << ast.error().message();
  }
  Fuzzing::Carbon carbon_proto;
  *carbon_proto.mutable_compilation_unit() = AstToProto(*ast);

  std::string proto_string;
  google::protobuf::TextFormat::Printer p;
  if (!p.PrintToString(carbon_proto, &proto_string)) {
    return Error("Failed to convert to text proto");
  }
  return WriteFile(proto_string, output_file_name);
}

// Command line options for defining input/output format.
enum class ConversionMode { TextProtoToCarbon, CarbonToTextProto };

auto Main(int argc, char* argv[]) -> ErrorOr<Success> {
  llvm::InitLLVM init_llvm(argc, argv);

  cl::opt<ConversionMode> mode(
      "mode", cl::desc("Conversion mode"),
      cl::values(
          clEnumValN(ConversionMode::TextProtoToCarbon, "proto_to_carbon",
                     "Convert text proto to Carbon source"),
          clEnumValN(ConversionMode::CarbonToTextProto, "carbon_to_proto",
                     "Convert Carbon source to text proto")),
      cl::Required);
  cl::opt<std::string> input_file_name("input", cl::desc("<input file>"),
                                       cl::init("/dev/stdin"));
  cl::opt<std::string> output_file_name("output", cl::desc("<output file>"),
                                        cl::init("/dev/stdout"));
  cl::ParseCommandLineOptions(argc, argv);

  switch (mode) {
    case ConversionMode::TextProtoToCarbon:
      return TextProtoToCarbon(input_file_name, output_file_name);
    case ConversionMode::CarbonToTextProto:
      return CarbonToTextProto(input_file_name, output_file_name);
  }
}

}  // namespace Carbon

auto main(int argc, char* argv[]) -> int {
  if (const auto result = Carbon::Main(argc, argv); !result.ok()) {
    llvm::errs() << result.error().message() << "\n";
    return EXIT_FAILURE;
  }
  return EXIT_SUCCESS;
}