105 lines
3.0 KiB
Rust
Executable File
105 lines
3.0 KiB
Rust
Executable File
use std::path::PathBuf;
|
|
|
|
use clap::{Parser, ValueEnum};
|
|
use memflow::plugins::Inventory;
|
|
|
|
use crate::dma::Connector;
|
|
const PORT_RANGE: std::ops::RangeInclusive<usize> = 8000..=65535;
|
|
|
|
#[derive(Parser)]
|
|
#[command(author, version = version(), about, long_about = None)]
|
|
pub struct Cli {
|
|
/// Specifies the connector type for DMA
|
|
#[clap(value_enum, short, long, ignore_case = true, default_value_t = Connector::Qemu)]
|
|
pub connector: Connector,
|
|
|
|
/// Name of the Pcileech device
|
|
#[clap(long, default_value_t = String::from("FPGA"))]
|
|
pub pcileech_device: String,
|
|
|
|
/// Port number for the Webserver to run on
|
|
#[arg(short, long, default_value_t = 8000, value_parser = port_in_range)]
|
|
pub port: u16,
|
|
|
|
/// Path to the directory served by the Webserver
|
|
#[arg(short, long, default_value = "./webradar", value_parser = valid_path)]
|
|
pub web_path: PathBuf,
|
|
|
|
/// Verbosity level for logging to the console
|
|
#[arg(value_enum, long, short, ignore_case = true, default_value_t = Loglevel::Warn)]
|
|
pub loglevel: Loglevel,
|
|
|
|
/// Skip the dwBuildNumber check, allows for running with *possibly* outdated offsets.
|
|
#[arg(long)]
|
|
pub skip_version: bool,
|
|
}
|
|
|
|
fn version() -> String {
|
|
let pkg_ver = env!("CARGO_PKG_VERSION");
|
|
let git_hash = option_env!("VERGEN_GIT_SHA").unwrap_or("unknown");
|
|
let commit_date = option_env!("VERGEN_GIT_COMMIT_DATE").unwrap_or("unknown");
|
|
let avail_cons = {
|
|
let inventory = Inventory::scan();
|
|
let mut avail = inventory.available_connectors();
|
|
avail.push("native".into());
|
|
avail.join(", ")
|
|
};
|
|
|
|
format!(
|
|
"{pkg_ver} (rev {git_hash})\n\
|
|
Commit Date: {commit_date}\n\
|
|
Available Connectors: {avail_cons}\n"
|
|
)
|
|
}
|
|
|
|
fn port_in_range(s: &str) -> Result<u16, String> {
|
|
let port: usize = s
|
|
.parse()
|
|
.map_err(|_| format!("`{s}` isn't a port number"))?;
|
|
if PORT_RANGE.contains(&port) {
|
|
Ok(port as u16)
|
|
} else {
|
|
Err(format!(
|
|
"port not in range {}-{}",
|
|
PORT_RANGE.start(),
|
|
PORT_RANGE.end()
|
|
))
|
|
}
|
|
}
|
|
|
|
fn valid_path(s: &str) -> Result<PathBuf, String> {
|
|
let path = PathBuf::from(s);
|
|
|
|
if !path.exists() {
|
|
return Err("Path does not exist".to_string())
|
|
}
|
|
|
|
if !path.is_dir() {
|
|
return Err("Path is not a directory".to_string())
|
|
}
|
|
|
|
Ok(path)
|
|
}
|
|
|
|
/// Wrapper because log::LevelFilter doesn't implement ValueEnum
|
|
#[derive(Copy, Clone, PartialEq, Eq, PartialOrd, Ord, ValueEnum, Default)]
|
|
pub enum Loglevel {
|
|
Error,
|
|
#[default]
|
|
Warn,
|
|
Info,
|
|
Debug,
|
|
Trace,
|
|
}
|
|
|
|
impl From<Loglevel> for log::LevelFilter {
|
|
fn from(val: Loglevel) -> Self {
|
|
match val {
|
|
Loglevel::Error => log::LevelFilter::Error,
|
|
Loglevel::Warn => log::LevelFilter::Warn,
|
|
Loglevel::Info => log::LevelFilter::Info,
|
|
Loglevel::Debug => log::LevelFilter::Debug,
|
|
Loglevel::Trace => log::LevelFilter::Trace,
|
|
}
|
|
}
|
|
} |