use tokio::io::AsyncReadExt;
use tokio::sync::mpsc::Receiver;
use tun2::{AbstractDevice, BoxError};
#[tokio::main]
async fn main() -> Result<(), BoxError> {
env_logger::Builder::from_env(env_logger::Env::default().default_filter_or("trace")).init();
let (tx, rx) = tokio::sync::mpsc::channel::<()>(1);
ctrlc2::set_async_handler(async move {
tx.send(()).await.expect("Signal error");
})
.await;
main_entry(rx).await?;
Ok(())
}
async fn main_entry(mut quit: Receiver<()>) -> Result<(), BoxError> {
let mut config = tun2::Configuration::default();
config
.address((10, 0, 0, 9))
.netmask((255, 255, 255, 0))
.destination((10, 0, 0, 1))
.mtu(tun2::DEFAULT_MTU)
.up();
#[cfg(target_os = "linux")]
config.platform_config(|config| {
config.ensure_root_privileges(true);
});
let mut dev = tun2::create_as_async(&config)?;
let size = dev.mtu()? as usize + tun2::PACKET_INFORMATION_LENGTH;
let mut buf = vec![0; size];
loop {
tokio::select! {
_ = quit.recv() => {
println!("Quit...");
break;
}
len = dev.read(&mut buf) => {
println!("pkt: {:?}", &buf[..len?]);
}
};
}
Ok(())
}