mirror of
https://github.com/tun2proxy/tun2proxy.git
synced 2025-06-21 00:20:53 +00:00
Ffi2 (#90)
This commit is contained in:
parent
a26621bbcd
commit
a1083273ee
9 changed files with 187 additions and 230 deletions
71
src/api.rs
71
src/api.rs
|
@ -1,71 +0,0 @@
|
|||
#![cfg(any(target_os = "ios", target_os = "android"))]
|
||||
|
||||
use crate::Args;
|
||||
use std::{os::raw::c_int, sync::Mutex};
|
||||
use tokio_util::sync::CancellationToken;
|
||||
|
||||
static TUN_QUIT: Mutex<Option<CancellationToken>> = Mutex::new(None);
|
||||
|
||||
pub(crate) fn tun2proxy_internal_run(args: Args, tun_mtu: u16) -> c_int {
|
||||
let shutdown_token = CancellationToken::new();
|
||||
{
|
||||
let mut lock = TUN_QUIT.lock().unwrap();
|
||||
if lock.is_some() {
|
||||
log::error!("tun2proxy already started");
|
||||
return -1;
|
||||
}
|
||||
*lock = Some(shutdown_token.clone());
|
||||
}
|
||||
|
||||
let block = async move {
|
||||
log::info!("Proxy {} server: {}", args.proxy.proxy_type, args.proxy.addr);
|
||||
|
||||
let mut config = tun2::Configuration::default();
|
||||
|
||||
#[cfg(unix)]
|
||||
if let Some(fd) = args.tun_fd {
|
||||
config.raw_fd(fd);
|
||||
} else {
|
||||
config.name(&args.tun);
|
||||
}
|
||||
#[cfg(windows)]
|
||||
config.name(&args.tun);
|
||||
|
||||
let device = tun2::create_as_async(&config).map_err(std::io::Error::from)?;
|
||||
let join_handle = tokio::spawn(crate::run(device, tun_mtu, args, shutdown_token));
|
||||
|
||||
join_handle.await.map_err(std::io::Error::from)?
|
||||
};
|
||||
|
||||
let exit_code = match tokio::runtime::Builder::new_multi_thread().enable_all().build() {
|
||||
Err(e) => {
|
||||
log::error!("failed to create tokio runtime with error: {:?}", e);
|
||||
-1
|
||||
}
|
||||
Ok(rt) => match rt.block_on(block) {
|
||||
Ok(_) => 0,
|
||||
Err(e) => {
|
||||
log::error!("failed to run tun2proxy with error: {:?}", e);
|
||||
-2
|
||||
}
|
||||
},
|
||||
};
|
||||
|
||||
// release shutdown token before exit.
|
||||
let mut lock = TUN_QUIT.lock().unwrap();
|
||||
let _ = lock.take();
|
||||
|
||||
exit_code
|
||||
}
|
||||
|
||||
pub(crate) fn tun2proxy_internal_stop() -> c_int {
|
||||
let lock = TUN_QUIT.lock().unwrap();
|
||||
|
||||
if let Some(shutdown_token) = lock.as_ref() {
|
||||
shutdown_token.cancel();
|
||||
0
|
||||
} else {
|
||||
log::error!("tun2proxy not started");
|
||||
-1
|
||||
}
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue