Function main

runners/KeyManagerRunner.cpp:12–73

}

int main(int argc, char **argv) {
  SET_VERBOSITY_LEVEL(verbosity_INFO);

  td::set_default_failure_signal_handler().ensure();

  std::string engine_config_filename = "key-manager-config.json";
  std::string pseudo_config_filename = "";
  bool check_hash = false;

  td::OptionParser option_parser;
  option_parser.set_description("key manager runner: run COCOON key manager");
  option_parser.add_option('c', "config", "key manager config",
                           [&](td::Slice opt) { engine_config_filename = opt.str(); });
  option_parser.add_checked_option('v', "verbosity", "set verbosity level", [&](td::Slice opt) -> td::Status {
    TRY_RESULT(level, td::to_integer_safe<td::int32>(opt));
    SET_VERBOSITY_LEVEL(level);
    return td::Status::OK();
  });
  option_parser.add_option('C', "disable-ton", "disable ton and use fake ton config",
                           [&](td::Slice opt) { pseudo_config_filename = opt.str(); });
  option_parser.add_option('p', "check-hashes", "check hashes of requesters", [&]() { check_hash = true; });
  option_parser.run(argc, argv, 0).ensure();

  td::actor::set_debug(true);
  td::actor::Scheduler scheduler({7});

  td::actor::ActorOwn<cocoon::KeyManagerRunner> key_manager_runner;

  scheduler.run_in_context([&] {
    key_manager_runner =
        td::actor::create_actor<cocoon::KeyManagerRunner>("key-manager", std::move(engine_config_filename), &scheduler);
    td::actor::send_lambda(key_manager_runner, [&]() {
      auto &ptr = key_manager_runner.get_actor_unsafe();
      if (pseudo_config_filename.size() > 0) {
        ptr.disable_ton(pseudo_config_filename);
      }
      if (check_hash) {
        ptr.enable_check_hashes();
      }
      ptr.initialize();
    });
  });

  while (scheduler.run(1)) {
    if (need_stats_flag.exchange(false)) {
      dump_stats();
    }
    if (need_scheduler_status_flag.exchange(false)) {
      LOG(ERROR) << "DUMPING SCHEDULER STATISTICS";
      td::StringBuilder sb;
      scheduler.get_debug().dump(sb);
      LOG(ERROR) << "GOT SCHEDULER STATISTICS\n" << sb.as_cslice();
    }
    if (rotate_logs_flags.exchange(false)) {
      if (td::log_interface) {
        td::log_interface->rotate();
      }
    }