120 lines
4.4 KiB
Rust
120 lines
4.4 KiB
Rust
use std::{io, thread};
|
|
/// 接收tun数据，并且转发到udp上
|
|
use std::net::Ipv4Addr;
|
|
use std::sync::Arc;
|
|
use crossbeam::atomic::AtomicCell;
|
|
|
|
use p2p_channel::channel::sender::Sender;
|
|
use packet::icmp::icmp::IcmpPacket;
|
|
use packet::icmp::Kind;
|
|
use packet::ip::ipv4;
|
|
use packet::ip::ipv4::packet::IpV4Packet;
|
|
|
|
use crate::error::*;
|
|
use crate::handle::{check_dest, CurrentDeviceInfo};
|
|
use crate::protocol::{MAX_TTL, NetPacket, Protocol, Version};
|
|
use crate::tun_device::{TunReader, TunWriter};
|
|
|
|
|
|
fn icmp(tun_writer: &TunWriter, mut ipv4_packet: IpV4Packet<&mut [u8]>) -> Result<()> {
|
|
if ipv4_packet.protocol() == ipv4::protocol::Protocol::Icmp {
|
|
let mut icmp = IcmpPacket::new(ipv4_packet.payload_mut())?;
|
|
if icmp.kind() == Kind::EchoRequest {
|
|
icmp.set_kind(Kind::EchoReply);
|
|
icmp.update_checksum();
|
|
let src = ipv4_packet.source_ip();
|
|
ipv4_packet.set_source_ip(ipv4_packet.destination_ip());
|
|
ipv4_packet.set_destination_ip(src);
|
|
ipv4_packet.update_checksum();
|
|
tun_writer.write(ipv4_packet.buffer)?;
|
|
}
|
|
}
|
|
Ok(())
|
|
}
|
|
|
|
#[inline]
|
|
fn handle(sender: &Sender<Ipv4Addr>, data: &mut [u8], tun_writer: &TunWriter, current_device: CurrentDeviceInfo, net_packet: &mut NetPacket<Vec<u8>>) -> Result<()> {
|
|
let data_len = data.len();
|
|
let ipv4_packet = match IpV4Packet::new(data) {
|
|
Ok(ipv4_packet) => ipv4_packet,
|
|
Err(packet::error::Error::Unimplemented) => {
|
|
return Ok(());
|
|
}
|
|
Err(e) => Err(e)?,
|
|
};
|
|
let src_ip = ipv4_packet.source_ip();
|
|
let dest_ip = ipv4_packet.destination_ip();
|
|
// if dest_ip == cur_info.broadcast_address {
|
|
// // 启动服务后会收到对137端口的广播
|
|
// // 137端口是在局域网中提供计算机的名字或IP地址查询服务
|
|
// return Ok(());
|
|
// }
|
|
if src_ip != current_device.virtual_ip() || (!check_dest(dest_ip, current_device.virtual_netmask, current_device.virtual_network) && !dest_ip.is_broadcast()) {
|
|
return Ok(());
|
|
}
|
|
if src_ip == dest_ip {
|
|
return icmp(&tun_writer, ipv4_packet);
|
|
}
|
|
net_packet.set_source(src_ip);
|
|
net_packet.set_destination(dest_ip);
|
|
net_packet.set_payload(ipv4_packet.buffer);
|
|
//优先发到直连到地址
|
|
if sender.send_to_id(&net_packet.buffer()[..(12 + data_len)], &dest_ip).is_err() {
|
|
sender.send_to_addr(&net_packet.buffer()[..(12 + data_len)], current_device.connect_server)?;
|
|
}
|
|
return Ok(());
|
|
}
|
|
|
|
pub fn start(sender: Sender<Ipv4Addr>,
|
|
tun_reader: TunReader,
|
|
tun_writer: TunWriter,
|
|
current_device: Arc<AtomicCell<CurrentDeviceInfo>>, ) {
|
|
thread::Builder::new().name("tun-handler".into()).spawn(move || {
|
|
if let Err(e) = start_(sender, tun_reader, tun_writer, current_device) {
|
|
log::warn!("{:?}",e);
|
|
}
|
|
}).unwrap();
|
|
}
|
|
|
|
#[cfg(target_os = "windows")]
|
|
fn start_(sender: Sender<Ipv4Addr>,
|
|
tun_reader: TunReader,
|
|
tun_writer: TunWriter,
|
|
current_device: Arc<AtomicCell<CurrentDeviceInfo>>, ) -> io::Result<()> {
|
|
let mut net_packet = NetPacket::new(vec![0u8; 4 + 8 + 1500])?;
|
|
net_packet.set_version(Version::V1);
|
|
net_packet.set_protocol(Protocol::Ipv4Turn);
|
|
net_packet.set_transport_protocol(ipv4::protocol::Protocol::Ipv4.into());
|
|
net_packet.set_ttl(MAX_TTL);
|
|
loop {
|
|
let mut data = tun_reader.next()?;
|
|
match handle(&sender, data.bytes_mut(), &tun_writer, current_device.load(), &mut net_packet) {
|
|
Ok(_) => {}
|
|
Err(e) => {
|
|
log::warn!("{:?}", e)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
#[cfg(any(target_os = "linux", target_os = "macos"))]
|
|
fn start_(sender: Sender<Ipv4Addr>,
|
|
tun_reader: TunReader,
|
|
tun_writer: TunWriter,
|
|
current_device: Arc<AtomicCell<CurrentDeviceInfo>>, ) -> io::Result<()> {
|
|
let mut net_packet = NetPacket::new(vec![0u8; 4 + 8 + 1500])?;
|
|
net_packet.set_version(Version::V1);
|
|
net_packet.set_protocol(Protocol::Ipv4Turn);
|
|
net_packet.set_transport_protocol(ipv4::protocol::Protocol::Ipv4.into());
|
|
net_packet.set_ttl(MAX_TTL);
|
|
let mut buf = [0; 4096];
|
|
loop {
|
|
let data = tun_reader.read(&mut buf)?;
|
|
match handle(&sender, data, &tun_writer, current_device.load(), &mut net_packet) {
|
|
Ok(_) => {}
|
|
Err(e) => {
|
|
log::warn!("{:?}", e)
|
|
}
|
|
}
|
|
}
|
|
} |