48 lines
1.4 KiB
Rust
48 lines
1.4 KiB
Rust
use std::sync::Arc;
|
|
use std::error::Error;
|
|
use tokio_tun::Tun;
|
|
use std::net::Ipv4Addr;
|
|
use std::os::unix::io::AsRawFd;
|
|
use crate::device::AbstractDevice;
|
|
|
|
pub fn create(cfg: AbstractDevice) -> (DeviceReader, DeviceWriter) {
|
|
let tun = Arc::new(
|
|
Tun::builder()
|
|
.name(&cfg.tun_name.unwrap()) // if name is empty, then it is set by kernel.
|
|
.mtu(cfg.mtu.unwrap().into())
|
|
.address(cfg.address.unwrap())
|
|
.netmask(cfg.netmask.unwrap())
|
|
.destination(cfg.destination.unwrap())
|
|
.up() // or set it up manually using `sudo ip link set <tun-name> up`.
|
|
.try_build() // or `.try_build_mq(queues)` for multi-queue support.
|
|
.unwrap(),
|
|
);
|
|
|
|
println!("tun created, name: {}, fd: {}", tun.name(), tun.as_raw_fd());
|
|
|
|
let tun_writer = tun.clone();
|
|
|
|
(DeviceReader {reader: tun}, DeviceWriter {writer: tun_writer})
|
|
}
|
|
|
|
pub struct DeviceWriter {
|
|
writer: Arc<Tun>
|
|
}
|
|
|
|
pub struct DeviceReader {
|
|
reader: Arc<Tun>
|
|
}
|
|
|
|
impl DeviceWriter {
|
|
pub async fn write(&self, buf: &Vec<u8>) -> Result<usize, Box<dyn Error>> {
|
|
self.writer.send_all(buf).await?;
|
|
Ok(0)
|
|
}
|
|
}
|
|
|
|
impl DeviceReader {
|
|
pub async fn read(&self, buf: &mut Vec<u8>) -> Result<usize, Box<dyn Error>> {
|
|
let n = self.reader.recv(buf).await?;
|
|
Ok(n)
|
|
}
|
|
} |