格式化代码

switch/src/handle/tun_handler.rs

@@ -10,12 +10,12 @@ use packet::icmp::Kind;
 use packet::ip::ipv4;
 use packet::ip::ipv4::packet::IpV4Packet;
 
-use crate::ApplicationStatus;
 use crate::error::*;
 use crate::handle::{CurrentDeviceInfo, DIRECT_ROUTE_TABLE};
-use crate::protocol::{NetPacket, Protocol, Version};
 use crate::protocol::turn_packet::TurnPacket;
+use crate::protocol::{NetPacket, Protocol, Version};
 use crate::tun_device::TunReader;
+use crate::ApplicationStatus;
 
 /// 是否在一个网段
 fn check_dest(dest: Ipv4Addr, cur_info: &CurrentDeviceInfo) -> bool {
@@ -77,42 +77,51 @@ fn handle(
     if let Some(route) = DIRECT_ROUTE_TABLE.get(&dest_ip) {
         let current_time = Local::now().timestamp_millis();
         if current_time - route.recv_time < 3_000 {
-            if udp.send_to(&net_packet.buffer()[..(4 + 8 + data_len)], route.address).is_ok() {
+            if udp
+                .send_to(&net_packet.buffer()[..(4 + 8 + data_len)], route.address)
+                .is_ok()
+            {
                 return Ok(());
             }
         }
     }
-    udp.send_to(&net_packet.buffer()[..(4 + 8 + data_len)], cur_info.connect_server)?;
+    udp.send_to(
+        &net_packet.buffer()[..(4 + 8 + data_len)],
+        cur_info.connect_server,
+    )?;
     return Ok(());
 }
 
 #[cfg(target_os = "windows")]
-pub async fn handler_start<F>(mut status_watch: watch::Receiver<ApplicationStatus>,
-                              udp: UdpSocket,
-                              tun_reader: TunReader,
-                              cur_info: CurrentDeviceInfo, stop_fn: F)
-    where F: FnOnce() + Send + 'static {
+pub async fn handler_start<F>(
+    mut status_watch: watch::Receiver<ApplicationStatus>,
+    udp: UdpSocket,
+    tun_reader: TunReader,
+    cur_info: CurrentDeviceInfo,
+    stop_fn: F,
+) where
+    F: FnOnce() + Send + 'static,
+{
     let session = tun_reader.0.clone();
     tokio::spawn(async move {
         let _ = status_watch.changed().await;
         session.shutdown();
         let udp = UdpSocket::bind("0.0.0.0:0").unwrap();
-        let _ = udp.send_to(&[0],SocketAddr::new(IpAddr::V4(cur_info.virtual_gateway),10));
+        let _ = udp.send_to(
+            &[0],
+            SocketAddr::new(IpAddr::V4(cur_info.virtual_gateway), 10),
+        );
     });
     thread::spawn(move || {
         if let Err(e) = handle_loop(udp, tun_reader, cur_info) {
-            log::error!("tun数据处理线程停止 {:?}",e);
+            log::error!("tun数据处理线程停止 {:?}", e);
         }
         stop_fn();
     });
 }
 
 #[cfg(target_os = "windows")]
-fn handle_loop(
-    udp: UdpSocket,
-    tun_reader: TunReader,
-    cur_info: CurrentDeviceInfo,
-) -> Result<()> {
+fn handle_loop(udp: UdpSocket, tun_reader: TunReader, cur_info: CurrentDeviceInfo) -> Result<()> {
     let mut net_packet = NetPacket::new(vec![0u8; 4 + 8 + 1500])?;
     net_packet.set_version(Version::V1);
     net_packet.set_protocol(Protocol::Ipv4Turn);
@@ -130,11 +139,16 @@ fn handle_loop(
 }
 
 #[cfg(any(target_os = "macos", target_os = "linux", target_os = "android"))]
-pub async fn handler_start<F>(mut status_watch: watch::Receiver<ApplicationStatus>,
-                              udp: UdpSocket,
-                              tun_reader: TunReader,
-                              cur_info: CurrentDeviceInfo, stop_fn: F)
-    where F: FnOnce() + Send + 'static {
+pub async fn handler_start<F>(
+    mut status_watch: watch::Receiver<ApplicationStatus>,
+    udp: UdpSocket,
+    tun_reader: TunReader,
+    cur_info: CurrentDeviceInfo,
+    stop_fn: F,
+) where
+    F: FnOnce() + Send + 'static,
+{
+    use std::os::fd::AsRawFd;
     let raw_fd = tun_reader.0.as_raw_fd();
     tokio::spawn(async move {
         let _ = status_watch.changed().await;
@@ -143,11 +157,14 @@ pub async fn handler_start<F>(mut status_watch: watch::Receiver<ApplicationStatu
             libc::close(raw_fd);
         }
         let udp = UdpSocket::bind("0.0.0.0:0").unwrap();
-        let _ = udp.send_to(&[0],SocketAddr::new(IpAddr::V4(cur_info.virtual_gateway),10));
+        let _ = udp.send_to(
+            &[0],
+            SocketAddr::new(IpAddr::V4(cur_info.virtual_gateway), 10),
+        );
     });
     thread::spawn(move || {
         if let Err(e) = handle_loop(udp, tun_reader, cur_info) {
-            log::error!(" tun数据处理线程停止 {:?}",e);
+            log::error!(" tun数据处理线程停止 {:?}", e);
         }
         stop_fn();
     });
@@ -170,7 +187,7 @@ pub fn handle_loop(
         match handle(&udp, data, &cur_info, &mut net_packet) {
             Ok(_) => {}
             Err(e) => {
-                log::error!("{:?}",e)
+                log::error!("{:?}", e)
             }
         }
     }