83 lines
2.6 KiB
Rust
83 lines
2.6 KiB
Rust
use clap::Parser;
|
|
use std::sync::Arc;
|
|
use std::{env, path::PathBuf, process::exit};
|
|
use tetanus::AppState;
|
|
use tetanus::funcs::*;
|
|
use tokio::sync::Mutex;
|
|
|
|
mod install;
|
|
|
|
#[derive(Parser, Debug)]
|
|
#[command(
|
|
author,
|
|
version,
|
|
about = "The Tetanus Redteaming tool. This will start the client, server, or both"
|
|
)]
|
|
struct Args {
|
|
#[arg(short, long, help = "start in client mode")]
|
|
client: bool,
|
|
|
|
#[arg(short, long, help = "start in server mode")]
|
|
server: bool,
|
|
|
|
#[arg(short, long, help = "start gui client")]
|
|
gui: bool,
|
|
|
|
#[arg(short, long, help = "for testing stuff...")]
|
|
test: bool,
|
|
}
|
|
|
|
#[tokio::main]
|
|
async fn main() {
|
|
let args = Args::parse();
|
|
if args.server {
|
|
let server = Arc::new(Mutex::new(tetanus::server::ServerState {
|
|
id: 0,
|
|
running: true,
|
|
address: String::from("127.0.0.1:31337"),
|
|
clients: Vec::new(),
|
|
key: String::new(),
|
|
}));
|
|
tetanus::server::start_server(server).await.unwrap();
|
|
} else {
|
|
println!("checking for server or client config files...");
|
|
let mut config_path = env::home_dir().unwrap();
|
|
config_path.push(".config/tetanus");
|
|
if !config_path.exists() {
|
|
println!("no config directory found in home directory...");
|
|
config_path = PathBuf::from("/etc/tetanus");
|
|
if !config_path.exists() {
|
|
println!("no config directory found in /etc/tetanus... Installing!");
|
|
let install_result = install::install();
|
|
if install_result.is_ok() {
|
|
install_result.unwrap();
|
|
println!("install succeeded!");
|
|
exit(0);
|
|
}
|
|
}
|
|
}
|
|
let args = Args::parse();
|
|
let mut client_config_path = config_path.clone();
|
|
client_config_path.push("client.conf");
|
|
let mut server_config_path = config_path.clone();
|
|
server_config_path.push("server.conf");
|
|
if args.client {
|
|
if !client_config_path.exists() {
|
|
eprintln!(
|
|
"error: no client config path found at {}",
|
|
client_config_path.display()
|
|
);
|
|
exit(1);
|
|
}
|
|
let (mut appstate, rx) = AppState::new();
|
|
let config_load = appstate.load_config(client_config_path.clone());
|
|
if config_load.is_err() {
|
|
eprintln!("error loading config!");
|
|
exit(1);
|
|
}
|
|
appstate.initialize_modules();
|
|
let _res = run_tui(appstate, rx).unwrap();
|
|
}
|
|
}
|
|
}
|