Args class

This commit is contained in:
ssrlive 2024-04-06 23:21:50 +08:00 committed by B. Blechschmidt
parent 181497e709
commit 56be614334
2 changed files with 35 additions and 16 deletions

View file

@ -16,7 +16,7 @@ pub struct Args {
/// Name of the tun interface, such as tun0, utun4, etc. /// Name of the tun interface, such as tun0, utun4, etc.
/// If this option is not provided, the OS will generate a random one. /// If this option is not provided, the OS will generate a random one.
#[arg(short, long, value_name = "name", conflicts_with = "tun_fd")] #[arg(short, long, value_name = "name", conflicts_with = "tun_fd", value_parser = validate_tun)]
pub tun: Option<String>, pub tun: Option<String>,
/// File descriptor of the tun interface /// File descriptor of the tun interface
@ -31,7 +31,7 @@ pub struct Args {
/// File descriptor for UNIX datagram socket meant to transfer /// File descriptor for UNIX datagram socket meant to transfer
/// network sockets from global namespace to the new one. /// network sockets from global namespace to the new one.
/// See `unshare(1)`, `namespaces(7)`, `sendmsg(2)`, `unix(7)`. /// See `unshare(1)`, `namespaces(7)`, `sendmsg(2)`, `unix(7)`.
#[arg(long)] #[arg(long, value_name = "fd")]
pub socket_transfer_fd: Option<i32>, pub socket_transfer_fd: Option<i32>,
/// Specify a command to run with root-like capabilities in the new namespace. /// Specify a command to run with root-like capabilities in the new namespace.
@ -43,9 +43,9 @@ pub struct Args {
#[arg(short = '6', long)] #[arg(short = '6', long)]
pub ipv6_enabled: bool, pub ipv6_enabled: bool,
#[arg(short, long)]
/// Routing and system setup, which decides whether to setup the routing and system configuration. /// Routing and system setup, which decides whether to setup the routing and system configuration.
/// This option is only available on Linux and requires root-like privileges. See `capabilities(7)`. /// This option is only available on Linux and requires root-like privileges. See `capabilities(7)`.
#[arg(short, long, default_value = if cfg!(target_os = "linux") { "false" } else { "true" })]
pub setup: bool, pub setup: bool,
/// DNS handling strategy /// DNS handling strategy
@ -73,8 +73,20 @@ pub struct Args {
pub verbosity: ArgVerbosity, pub verbosity: ArgVerbosity,
} }
fn validate_tun(p: &str) -> Result<String> {
#[cfg(target_os = "macos")]
if p.len() <= 4 || &p[..4] != "utun" {
return Err(Error::from("Invalid tun interface name, please use utunX"));
}
Ok(p.to_string())
}
impl Default for Args { impl Default for Args {
fn default() -> Self { fn default() -> Self {
#[cfg(target_os = "linux")]
let setup = false;
#[cfg(not(target_os = "linux"))]
let setup = true;
Args { Args {
proxy: ArgProxy::default(), proxy: ArgProxy::default(),
tun: None, tun: None,
@ -83,7 +95,7 @@ impl Default for Args {
socket_transfer_fd: None, socket_transfer_fd: None,
admin_command: Vec::new(), admin_command: Vec::new(),
ipv6_enabled: false, ipv6_enabled: false,
setup: false, setup,
dns: ArgDns::default(), dns: ArgDns::default(),
dns_addr: "8.8.8.8".parse().unwrap(), dns_addr: "8.8.8.8".parse().unwrap(),
bypass: vec![], bypass: vec![],
@ -95,9 +107,16 @@ impl Default for Args {
} }
impl Args { impl Args {
#[allow(clippy::let_and_return)]
pub fn parse_args() -> Self { pub fn parse_args() -> Self {
use clap::Parser; use clap::Parser;
Self::parse() let args = Self::parse();
#[cfg(target_os = "linux")]
if !args.setup && args.tun.is_none() {
eprintln!("Missing required argument, '--tun' must present when '--setup' is not used.");
std::process::exit(-1);
}
args
} }
pub fn proxy(&mut self, proxy: ArgProxy) -> &mut Self { pub fn proxy(&mut self, proxy: ArgProxy) -> &mut Self {

View file

@ -83,25 +83,25 @@ pub unsafe extern "C" fn tun2proxy_with_name_run(
pub async fn desktop_run_async(args: Args, shutdown_token: tokio_util::sync::CancellationToken) -> std::io::Result<()> { pub async fn desktop_run_async(args: Args, shutdown_token: tokio_util::sync::CancellationToken) -> std::io::Result<()> {
let bypass_ips = args.bypass.clone(); let bypass_ips = args.bypass.clone();
let mut config = tun2::Configuration::default(); let mut tun_config = tun2::Configuration::default();
config.address(TUN_IPV4).netmask(TUN_NETMASK).mtu(MTU).up(); tun_config.address(TUN_IPV4).netmask(TUN_NETMASK).mtu(MTU).up();
config.destination(TUN_GATEWAY); tun_config.destination(TUN_GATEWAY);
if let Some(tun_fd) = args.tun_fd { if let Some(tun_fd) = args.tun_fd {
config.raw_fd(tun_fd); tun_config.raw_fd(tun_fd);
} else if let Some(ref tun) = args.tun { } else if let Some(ref tun) = args.tun {
config.tun_name(tun); tun_config.tun_name(tun);
} }
#[cfg(target_os = "linux")] #[cfg(target_os = "linux")]
config.platform_config(|config| { tun_config.platform_config(|cfg| {
#[allow(deprecated)] #[allow(deprecated)]
config.packet_information(true); cfg.packet_information(true);
config.ensure_root_privileges(args.setup); cfg.ensure_root_privileges(args.setup);
}); });
#[cfg(target_os = "windows")] #[cfg(target_os = "windows")]
config.platform_config(|config| { tun_config.platform_config(|cfg| {
config.device_guid(Some(12324323423423434234_u128)); cfg.device_guid(Some(12324323423423434234_u128));
}); });
#[allow(unused_variables)] #[allow(unused_variables)]
@ -113,7 +113,7 @@ pub async fn desktop_run_async(args: Args, shutdown_token: tokio_util::sync::Can
#[allow(unused_mut, unused_assignments, unused_variables)] #[allow(unused_mut, unused_assignments, unused_variables)]
let mut setup = true; let mut setup = true;
let device = tun2::create_as_async(&config)?; let device = tun2::create_as_async(&tun_config)?;
if let Ok(tun_name) = device.as_ref().tun_name() { if let Ok(tun_name) = device.as_ref().tun_name() {
tproxy_args = tproxy_args.tun_name(&tun_name); tproxy_args = tproxy_args.tun_name(&tun_name);