source/Main.cpp

/*
 * Copyright (c) Meta Platforms, Inc. and affiliates.
 *
 * This source code is licensed under the MIT license found in the
 * LICENSE file in the root directory of this source tree.
 */

#include <signal.h>

#include <exception>
#include <iostream>
#include <stdexcept>

#include <boost/program_options.hpp>

#include <AggregateException.h>
#include <Debug.h>
#include <RedexContext.h>

#include <mariana-trench/MarianaTrench.h>
#include <mariana-trench/SanitizersOptions.h>

int main(int argc, char* argv[]) {
  signal(SIGSEGV, crash_backtrace_handler);
  signal(SIGABRT, crash_backtrace_handler);
#ifndef _MSC_VER
  signal(SIGBUS, crash_backtrace_handler);
#endif

  namespace program_options = boost::program_options;
  program_options::options_description options;
  options.add_options()("help,h", "Show help dialog.");

  auto tool = marianatrench::MarianaTrench();
  tool.add_options(options);

  try {
    program_options::variables_map variables;
    program_options::store(
        program_options::command_line_parser(argc, argv).options(options).run(),
        variables);

    if (variables.count("help")) {
      std::cerr << options;
      return 0;
    }

    program_options::notify(variables);

    g_redex = new RedexContext(/* allow_class_duplicates */ true);
    tool.run(variables);
    delete g_redex;
  } catch (const aggregate_exception& aggregate_exception) {
    std::cerr << "Caught multiple exceptions:" << std::endl;
    for (const auto& exception : aggregate_exception.m_exceptions) {
      try {
        std::rethrow_exception(exception);
      } catch (const std::exception& rethrown_exception) {
        std::cerr << rethrown_exception.what() << std::endl;
      }
    }
    return 1;
  } catch (const std::exception& exception) {
    std::cerr << "error: " << exception.what() << std::endl;
    return 1;
  }
}