mirror of
https://github.com/rosenpass/rosenpass.git
synced 2025-12-12 07:40:30 -08:00
f4e8e4314b
This, for now, disables correct handling of program termination, but not because the RAII does not work. Instead, we need to implement a proper signal handling concept. We also removed some teardown handlers which are not covered by RAII, like removing the routes we set up. The reason for this is, that this is going to be taken care of by removing the wireguard device anyway.
63 lines
1.9 KiB
Rust
63 lines
1.9 KiB
Rust
use std::{fs, process::exit};
|
|
|
|
use rosenpass_util::tokio::janitor::ensure_janitor;
|
|
|
|
use rosenpass_secret_memory::policy;
|
|
|
|
use crate::cli::{Cli, Command};
|
|
use crate::exchange::exchange;
|
|
use crate::key::{genkey, pubkey};
|
|
|
|
mod cli;
|
|
mod exchange;
|
|
mod key;
|
|
|
|
#[tokio::main]
|
|
async fn main() -> anyhow::Result<()> {
|
|
#[cfg(feature = "experiment_memfd_secret")]
|
|
policy::secret_policy_try_use_memfd_secrets();
|
|
#[cfg(not(feature = "experiment_memfd_secret"))]
|
|
policy::secret_policy_use_only_malloc_secrets();
|
|
|
|
ensure_janitor(async move { main_impl().await }).await
|
|
}
|
|
|
|
async fn main_impl() -> anyhow::Result<()> {
|
|
let cli = match Cli::parse(std::env::args().peekable()) {
|
|
Ok(cli) => cli,
|
|
Err(err) => {
|
|
eprintln!("{}", err);
|
|
exit(1);
|
|
}
|
|
};
|
|
|
|
// init logging
|
|
// TODO: Taken from rosenpass; we should deduplicate the code.
|
|
env_logger::Builder::from_default_env().init(); // sets log level filter from environment (or defaults)
|
|
|
|
let command = cli.command.unwrap();
|
|
|
|
match command {
|
|
Command::GenKey { private_keys_dir } => genkey(&private_keys_dir),
|
|
Command::PubKey {
|
|
private_keys_dir,
|
|
public_keys_dir,
|
|
} => pubkey(&private_keys_dir, &public_keys_dir),
|
|
Command::Exchange(mut options) => {
|
|
options.verbose = cli.verbose;
|
|
exchange(options).await
|
|
}
|
|
Command::ExchangeConfig { config_file } => {
|
|
let s: String = fs::read_to_string(config_file).expect("cannot read config");
|
|
let mut options: exchange::ExchangeOptions =
|
|
toml::from_str::<exchange::ExchangeOptions>(&s).expect("cannot parse config");
|
|
options.verbose = options.verbose || cli.verbose;
|
|
exchange(options).await
|
|
}
|
|
Command::Help => {
|
|
println!("Usage: rp [verbose] genkey|pubkey|exchange [ARGS]...");
|
|
Ok(())
|
|
}
|
|
}
|
|
}
|