73 lines
2.2 KiB
Rust
73 lines
2.2 KiB
Rust
use std::net::Ipv4Addr;
|
|
use std::process::Command;
|
|
use std::io;
|
|
use tun::Device;
|
|
use parking_lot::Mutex;
|
|
use std::sync::Arc;
|
|
|
|
use crate::tun_device::{TunReader, TunWriter};
|
|
|
|
pub fn create_tun(
|
|
address: Ipv4Addr,
|
|
netmask: Ipv4Addr,
|
|
gateway: Ipv4Addr,
|
|
in_ips: Vec<(Ipv4Addr, Ipv4Addr)>,
|
|
) -> crate::error::Result<(TunWriter, TunReader)> {
|
|
println!("========TUN网卡配置========");
|
|
let mut config = tun::Configuration::default();
|
|
|
|
config
|
|
.destination(gateway)
|
|
.address(address)
|
|
.netmask(netmask)
|
|
.mtu(1420)
|
|
.up();
|
|
|
|
let dev = tun::create(&config).unwrap();
|
|
let name = dev.name();
|
|
config_ip(name, address, netmask, gateway)?;
|
|
add_route(name, address, netmask)?;
|
|
for (address, netmask) in in_ips {
|
|
add_route(name, address, netmask)?;
|
|
}
|
|
let packet_information = dev.has_packet_information();
|
|
let queue = dev.queue(0).unwrap();
|
|
let reader = queue.reader();
|
|
let writer = queue.writer();
|
|
println!("name:{:?}", name);
|
|
println!("========TUN网卡配置========");
|
|
Ok((
|
|
TunWriter(writer, packet_information, Arc::new(Mutex::new(dev))),
|
|
TunReader(reader, packet_information),
|
|
))
|
|
}
|
|
|
|
fn add_route(name: &str, address: Ipv4Addr, netmask: Ipv4Addr) -> io::Result<()> {
|
|
let route_add_str: String = format!(
|
|
"sudo route -n add -net {:?}/{:?} -interface {}",
|
|
address, netmask, name
|
|
);
|
|
let route_add_out = Command::new("sh")
|
|
.arg("-c")
|
|
.arg(route_add_str)
|
|
.output()
|
|
.expect("sh exec error!");
|
|
if !route_add_out.status.success() {
|
|
return Err(io::Error::new(io::ErrorKind::Other, format!("添加路由失败: {:?}", route_add_out)));
|
|
}
|
|
Ok(())
|
|
}
|
|
|
|
pub(crate) fn config_ip(name: &str, address: Ipv4Addr, netmask: Ipv4Addr, gateway: Ipv4Addr) -> io::Result<()> {
|
|
let up_eth_str: String = format!("ifconfig {} {:?} {:?} up ", name, address, gateway);
|
|
let up_eth_out = Command::new("sh")
|
|
.arg("-c")
|
|
.arg(up_eth_str)
|
|
.output()
|
|
.expect("sh exec error!");
|
|
if !up_eth_out.status.success() {
|
|
return Err(io::Error::new(io::ErrorKind::Other, format!("设置网络地址失败: {:?}", up_eth_out)));
|
|
}
|
|
Ok(())
|
|
}
|