ssrlive
2 years ago
committed by
GitHub
GitHub
9 changed files with 187 additions and 230 deletions
@ -0,0 +1,148 @@ |
|||||
|
#![cfg(any(target_os = "windows", target_os = "macos", target_os = "linux"))] |
||||
|
|
||||
|
use crate::{ |
||||
|
args::{ArgDns, ArgProxy}, |
||||
|
ArgVerbosity, Args, |
||||
|
}; |
||||
|
use std::os::raw::{c_char, c_int}; |
||||
|
use tproxy_config::{TproxyArgs, TUN_GATEWAY, TUN_IPV4, TUN_NETMASK}; |
||||
|
use tun2::DEFAULT_MTU as MTU; |
||||
|
|
||||
|
static TUN_QUIT: std::sync::Mutex<Option<tokio_util::sync::CancellationToken>> = std::sync::Mutex::new(None); |
||||
|
|
||||
|
/// # Safety
|
||||
|
///
|
||||
|
/// Run the tun2proxy component with some arguments.
|
||||
|
#[no_mangle] |
||||
|
pub unsafe extern "C" fn tun2proxy_run_with_name( |
||||
|
proxy_url: *const c_char, |
||||
|
tun: *const c_char, |
||||
|
bypass: *const c_char, |
||||
|
dns_strategy: ArgDns, |
||||
|
_root_privilege: bool, |
||||
|
verbosity: ArgVerbosity, |
||||
|
) -> c_int { |
||||
|
let shutdown_token = tokio_util::sync::CancellationToken::new(); |
||||
|
{ |
||||
|
if let Ok(mut lock) = TUN_QUIT.lock() { |
||||
|
if lock.is_some() { |
||||
|
return -1; |
||||
|
} |
||||
|
*lock = Some(shutdown_token.clone()); |
||||
|
} else { |
||||
|
return -2; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
log::set_max_level(verbosity.into()); |
||||
|
log::set_boxed_logger(Box::<crate::dump_logger::DumpLogger>::default()).unwrap(); |
||||
|
|
||||
|
let proxy_url = std::ffi::CStr::from_ptr(proxy_url).to_str().unwrap(); |
||||
|
let proxy = ArgProxy::from_url(proxy_url).unwrap(); |
||||
|
let tun = std::ffi::CStr::from_ptr(tun).to_str().unwrap().to_string(); |
||||
|
|
||||
|
let mut args = Args::default(); |
||||
|
args.proxy(proxy).tun(tun).dns(dns_strategy).verbosity(verbosity); |
||||
|
|
||||
|
#[cfg(target_os = "linux")] |
||||
|
args.setup(_root_privilege); |
||||
|
|
||||
|
if let Ok(bypass) = std::ffi::CStr::from_ptr(bypass).to_str() { |
||||
|
args.bypass(bypass.parse().unwrap()); |
||||
|
} |
||||
|
|
||||
|
let exit_code = match tokio::runtime::Builder::new_multi_thread().enable_all().build() { |
||||
|
Err(_e) => -3, |
||||
|
Ok(rt) => match rt.block_on(desktop_run_async(args, shutdown_token)) { |
||||
|
Ok(_) => 0, |
||||
|
Err(_e) => -4, |
||||
|
}, |
||||
|
}; |
||||
|
|
||||
|
// release shutdown token before exit.
|
||||
|
if let Ok(mut lock) = TUN_QUIT.lock() { |
||||
|
let _ = lock.take(); |
||||
|
} |
||||
|
|
||||
|
exit_code |
||||
|
} |
||||
|
|
||||
|
/// Run the tun2proxy component with some arguments.
|
||||
|
pub async fn desktop_run_async(args: Args, shutdown_token: tokio_util::sync::CancellationToken) -> std::io::Result<()> { |
||||
|
let bypass_ips = args.bypass.clone(); |
||||
|
|
||||
|
let mut config = tun2::Configuration::default(); |
||||
|
config.address(TUN_IPV4).netmask(TUN_NETMASK).mtu(MTU).up(); |
||||
|
config.destination(TUN_GATEWAY); |
||||
|
if let Some(tun_fd) = args.tun_fd { |
||||
|
config.raw_fd(tun_fd); |
||||
|
} else { |
||||
|
config.name(&args.tun); |
||||
|
} |
||||
|
|
||||
|
#[cfg(target_os = "linux")] |
||||
|
config.platform_config(|config| { |
||||
|
#[allow(deprecated)] |
||||
|
config.packet_information(true); |
||||
|
config.ensure_root_privileges(args.setup); |
||||
|
}); |
||||
|
|
||||
|
#[cfg(target_os = "windows")] |
||||
|
config.platform_config(|config| { |
||||
|
config.device_guid(Some(12324323423423434234_u128)); |
||||
|
}); |
||||
|
|
||||
|
#[allow(unused_variables)] |
||||
|
let mut tproxy_args = TproxyArgs::new() |
||||
|
.tun_dns(args.dns_addr) |
||||
|
.proxy_addr(args.proxy.addr) |
||||
|
.bypass_ips(&bypass_ips); |
||||
|
#[allow(unused_assignments)] |
||||
|
if args.tun_fd.is_none() { |
||||
|
tproxy_args = tproxy_args.tun_name(&args.tun); |
||||
|
} |
||||
|
|
||||
|
#[allow(unused_mut, unused_assignments, unused_variables)] |
||||
|
let mut setup = true; |
||||
|
|
||||
|
#[cfg(target_os = "linux")] |
||||
|
{ |
||||
|
setup = args.setup; |
||||
|
if setup { |
||||
|
tproxy_config::tproxy_setup(&tproxy_args)?; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
let device = tun2::create_as_async(&config)?; |
||||
|
|
||||
|
#[cfg(any(target_os = "windows", target_os = "macos"))] |
||||
|
if setup { |
||||
|
tproxy_config::tproxy_setup(&tproxy_args)?; |
||||
|
} |
||||
|
|
||||
|
log::info!("Proxy {} server: {}", args.proxy.proxy_type, args.proxy.addr); |
||||
|
|
||||
|
let join_handle = tokio::spawn(crate::run(device, MTU, args, shutdown_token)); |
||||
|
join_handle.await.map_err(std::io::Error::from)??; |
||||
|
|
||||
|
#[cfg(any(target_os = "linux", target_os = "windows", target_os = "macos"))] |
||||
|
if setup { |
||||
|
tproxy_config::tproxy_remove(&tproxy_args)?; |
||||
|
} |
||||
|
|
||||
|
Ok::<(), std::io::Error>(()) |
||||
|
} |
||||
|
|
||||
|
/// # Safety
|
||||
|
///
|
||||
|
/// Shutdown the tun2proxy component.
|
||||
|
#[no_mangle] |
||||
|
pub unsafe extern "C" fn tun2proxy_stop() -> c_int { |
||||
|
if let Ok(lock) = TUN_QUIT.lock() { |
||||
|
if let Some(shutdown_token) = lock.as_ref() { |
||||
|
shutdown_token.cancel(); |
||||
|
return 0; |
||||
|
} |
||||
|
} |
||||
|
-1 |
||||
|
} |
||||
@ -1,140 +0,0 @@ |
|||||
#![cfg(any(target_os = "windows", target_os = "macos", target_os = "linux"))] |
|
||||
|
|
||||
use crate::{ |
|
||||
args::{ArgDns, ArgProxy}, |
|
||||
ArgVerbosity, Args, |
|
||||
}; |
|
||||
use std::os::raw::{c_char, c_int}; |
|
||||
use tokio_util::sync::CancellationToken; |
|
||||
use tproxy_config::{TproxyArgs, TUN_GATEWAY, TUN_IPV4, TUN_NETMASK}; |
|
||||
use tun2::DEFAULT_MTU as MTU; |
|
||||
|
|
||||
static TUN_QUIT: std::sync::Mutex<Option<CancellationToken>> = std::sync::Mutex::new(None); |
|
||||
|
|
||||
/// # Safety
|
|
||||
///
|
|
||||
/// Run the tun2proxy component with some arguments.
|
|
||||
#[no_mangle] |
|
||||
pub unsafe extern "C" fn tun2proxy_run_with_name( |
|
||||
proxy_url: *const c_char, |
|
||||
tun: *const c_char, |
|
||||
bypass: *const c_char, |
|
||||
dns_strategy: ArgDns, |
|
||||
_root_privilege: bool, |
|
||||
verbosity: ArgVerbosity, |
|
||||
) -> c_int { |
|
||||
let shutdown_token = CancellationToken::new(); |
|
||||
{ |
|
||||
if let Ok(mut lock) = TUN_QUIT.lock() { |
|
||||
if lock.is_some() { |
|
||||
log::error!("tun2proxy already started"); |
|
||||
return -1; |
|
||||
} |
|
||||
*lock = Some(shutdown_token.clone()); |
|
||||
} else { |
|
||||
log::error!("tun2proxy unknown error"); |
|
||||
return -1; |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
log::set_max_level(verbosity.into()); |
|
||||
log::set_boxed_logger(Box::<crate::dump_logger::DumpLogger>::default()).unwrap(); |
|
||||
|
|
||||
let proxy_url = std::ffi::CStr::from_ptr(proxy_url).to_str().unwrap(); |
|
||||
let proxy = ArgProxy::from_url(proxy_url).unwrap(); |
|
||||
let tun = std::ffi::CStr::from_ptr(tun).to_str().unwrap().to_string(); |
|
||||
|
|
||||
let mut args = Args::default(); |
|
||||
args.proxy(proxy).tun(tun).dns(dns_strategy).verbosity(verbosity); |
|
||||
|
|
||||
#[cfg(target_os = "linux")] |
|
||||
args.setup(_root_privilege); |
|
||||
|
|
||||
if let Ok(bypass) = std::ffi::CStr::from_ptr(bypass).to_str() { |
|
||||
args.bypass(bypass.parse().unwrap()); |
|
||||
} |
|
||||
|
|
||||
let block = async move { |
|
||||
let bypass_ips = args.bypass.clone(); |
|
||||
|
|
||||
let mut config = tun2::Configuration::default(); |
|
||||
config.address(TUN_IPV4).netmask(TUN_NETMASK).mtu(MTU).up(); |
|
||||
config.destination(TUN_GATEWAY); |
|
||||
if let Some(tun_fd) = args.tun_fd { |
|
||||
config.raw_fd(tun_fd); |
|
||||
} else { |
|
||||
config.name(&args.tun); |
|
||||
} |
|
||||
|
|
||||
#[cfg(target_os = "linux")] |
|
||||
config.platform_config(|config| { |
|
||||
#[allow(deprecated)] |
|
||||
config.packet_information(true); |
|
||||
config.ensure_root_privileges(args.setup); |
|
||||
}); |
|
||||
|
|
||||
#[cfg(target_os = "windows")] |
|
||||
config.platform_config(|config| { |
|
||||
config.device_guid(Some(12324323423423434234_u128)); |
|
||||
}); |
|
||||
|
|
||||
#[allow(unused_variables)] |
|
||||
let mut tproxy_args = TproxyArgs::new() |
|
||||
.tun_dns(args.dns_addr) |
|
||||
.proxy_addr(args.proxy.addr) |
|
||||
.bypass_ips(&bypass_ips); |
|
||||
#[allow(unused_assignments)] |
|
||||
if args.tun_fd.is_none() { |
|
||||
tproxy_args = tproxy_args.tun_name(&args.tun); |
|
||||
} |
|
||||
|
|
||||
#[cfg(target_os = "linux")] |
|
||||
tproxy_config::tproxy_setup(&tproxy_args)?; |
|
||||
|
|
||||
let device = tun2::create_as_async(&config)?; |
|
||||
|
|
||||
#[cfg(any(target_os = "windows", target_os = "macos"))] |
|
||||
tproxy_config::tproxy_setup(&tproxy_args)?; |
|
||||
|
|
||||
log::info!("Proxy {} server: {}", args.proxy.proxy_type, args.proxy.addr); |
|
||||
|
|
||||
let join_handle = tokio::spawn(crate::run(device, MTU, args, shutdown_token)); |
|
||||
join_handle.await.map_err(std::io::Error::from)??; |
|
||||
|
|
||||
#[cfg(any(target_os = "linux", target_os = "windows", target_os = "macos"))] |
|
||||
if _root_privilege { |
|
||||
tproxy_config::tproxy_remove(&tproxy_args)?; |
|
||||
} |
|
||||
|
|
||||
Ok::<(), crate::BoxError>(()) |
|
||||
}; |
|
||||
|
|
||||
let exit_code = match tokio::runtime::Builder::new_multi_thread().enable_all().build() { |
|
||||
Err(_e) => -1, |
|
||||
Ok(rt) => match rt.block_on(block) { |
|
||||
Ok(_) => 0, |
|
||||
Err(_e) => -2, |
|
||||
}, |
|
||||
}; |
|
||||
|
|
||||
// release shutdown token before exit.
|
|
||||
if let Ok(mut lock) = TUN_QUIT.lock() { |
|
||||
let _ = lock.take(); |
|
||||
} |
|
||||
|
|
||||
exit_code |
|
||||
} |
|
||||
|
|
||||
/// # Safety
|
|
||||
///
|
|
||||
/// Shutdown the tun2proxy component.
|
|
||||
#[no_mangle] |
|
||||
pub unsafe extern "C" fn tun2proxy_stop() -> c_int { |
|
||||
if let Ok(lock) = TUN_QUIT.lock() { |
|
||||
if let Some(shutdown_token) = lock.as_ref() { |
|
||||
shutdown_token.cancel(); |
|
||||
return 0; |
|
||||
} |
|
||||
} |
|
||||
-1 |
|
||||
} |
|
||||
Loading…
Reference in new issue