hussh 0.2.2

Memory-safe SSH daemon for restricted TCP forwarding
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39

use crate::config::Config;
use crate::errors::*;
use clap::ArgAction;
use std::net::SocketAddr;
use std::path::PathBuf;
use tokio::fs;

#[derive(clap::Parser)]
#[command(version)]
pub struct Args {
    /// Increase logging output (can be used multiple times)
    #[arg(short, long, global = true, action(ArgAction::Count))]
    pub verbose: u8,
    /// Path to the data directory
    #[arg(short = 'D', long, default_value = ".", env = "HUSSH_DATA_DIR")]
    pub data_dir: PathBuf,
    /// Generate an ssh private key
    #[arg(short = 'K', long)]
    pub keygen: bool,
    /// The address to bind the daemon to
    #[arg(short = 'B', long)]
    pub bind: Option<SocketAddr>,
    /// Path to daemon config file
    #[arg(short = 'c', long)]
    pub config: Option<PathBuf>,
}

impl Args {
    pub async fn config(&self) -> Result<Config> {
        let path = self
            .config
            .as_ref()
            .context("Missing -c option for configuration file")?;
        let config = fs::read_to_string(&path)
            .await
            .with_context(|| format!("Failed to read file: {path:?}"))?;
        Config::parse(&config).with_context(|| format!("Failed to parse config file: {path:?}"))
    }
}