[mio] 整理数据处理逻辑

vnt/src/handle/mod.rs

@@ -1,12 +1,15 @@
 use std::net::{Ipv4Addr, SocketAddr};
 
-pub mod handshake_handler;
-pub mod heartbeat_handler;
-pub mod punch_handler;
-pub mod recv_handler;
-pub mod registration_handler;
+pub mod callback;
+pub mod handshaker;
+pub mod maintain;
+pub mod recv_data;
+pub mod registrar;
 pub mod tun_tap;
 
+const SELF_IP: Ipv4Addr = Ipv4Addr::new(0, 0, 0, 2);
+const GATEWAY_IP: Ipv4Addr = Ipv4Addr::new(0, 0, 0, 1);
+
 pub fn now_time() -> u64 {
     let now = std::time::SystemTime::now();
     if let Ok(timestamp) = now.duration_since(std::time::UNIX_EPOCH) {
@@ -41,12 +44,48 @@ impl PeerDeviceInfo {
     }
 }
 
+#[derive(Clone, Debug)]
+pub struct BaseConfigInfo {
+    pub name: String,
+    pub token: String,
+    pub ip: Option<Ipv4Addr>,
+    pub client_secret: bool,
+    pub device_id: String,
+    pub server_addr: String,
+}
+
+impl BaseConfigInfo {
+    pub fn new(
+        name: String,
+        token: String,
+        ip: Option<Ipv4Addr>,
+        client_secret: bool,
+        device_id: String,
+        server_addr: String,
+    ) -> Self {
+        Self {
+            name,
+            token,
+            ip,
+            client_secret,
+            device_id,
+            server_addr,
+        }
+    }
+}
+
 #[derive(Copy, Clone, Debug, Eq, PartialEq, Ord, PartialOrd)]
 pub enum PeerDeviceStatus {
     Online,
     Offline,
 }
 
+impl PeerDeviceStatus {
+    pub fn is_online(&self) -> bool {
+        self == &PeerDeviceStatus::Online
+    }
+}
+
 impl Into<u8> for PeerDeviceStatus {
     fn into(self) -> u8 {
         match self {
@@ -73,27 +112,32 @@ pub enum ConnectStatus {
 
 #[derive(Copy, Clone, Debug, Eq, PartialEq)]
 pub struct CurrentDeviceInfo {
-    virtual_ip: Ipv4Addr,
-    pub virtual_gateway: Ipv4Addr,
+    //本机虚拟IP
+    pub virtual_ip: Ipv4Addr,
+    //子网掩码
     pub virtual_netmask: Ipv4Addr,
+    //虚拟网关
+    pub virtual_gateway: Ipv4Addr,
     //网络地址
     pub virtual_network: Ipv4Addr,
     //直接广播地址
-    pub broadcast_address: Ipv4Addr,
+    pub broadcast_ip: Ipv4Addr,
     //链接的服务器地址
     pub connect_server: SocketAddr,
+    //连接状态
+    pub status: ConnectStatus,
 }
 
 impl CurrentDeviceInfo {
     pub fn new(
         virtual_ip: Ipv4Addr,
-        virtual_gateway: Ipv4Addr,
         virtual_netmask: Ipv4Addr,
+        virtual_gateway: Ipv4Addr,
         connect_server: SocketAddr,
     ) -> Self {
-        let broadcast_address = (!u32::from_be_bytes(virtual_netmask.octets()))
+        let broadcast_ip = (!u32::from_be_bytes(virtual_netmask.octets()))
             | u32::from_be_bytes(virtual_gateway.octets());
-        let broadcast_address = Ipv4Addr::from(broadcast_address);
+        let broadcast_ip = Ipv4Addr::from(broadcast_ip);
         let virtual_network = u32::from_be_bytes(virtual_netmask.octets())
             & u32::from_be_bytes(virtual_gateway.octets());
         let virtual_network = Ipv4Addr::from(virtual_network);
@@ -102,10 +146,40 @@ impl CurrentDeviceInfo {
             virtual_netmask,
             virtual_gateway,
             virtual_network,
-            broadcast_address,
+            broadcast_ip,
             connect_server,
+            status: ConnectStatus::Connecting,
         }
     }
+    pub fn new0(connect_server: SocketAddr) -> Self {
+        Self {
+            virtual_ip: Ipv4Addr::UNSPECIFIED,
+            virtual_gateway: Ipv4Addr::UNSPECIFIED,
+            virtual_netmask: Ipv4Addr::UNSPECIFIED,
+            virtual_network: Ipv4Addr::UNSPECIFIED,
+            broadcast_ip: Ipv4Addr::UNSPECIFIED,
+            connect_server,
+            status: ConnectStatus::Connecting,
+        }
+    }
+    pub fn update(
+        &mut self,
+        virtual_ip: Ipv4Addr,
+        virtual_netmask: Ipv4Addr,
+        virtual_gateway: Ipv4Addr,
+    ) {
+        let broadcast_ip = (!u32::from_be_bytes(virtual_netmask.octets()))
+            | u32::from_be_bytes(virtual_gateway.octets());
+        let broadcast_ip = Ipv4Addr::from(broadcast_ip);
+        let virtual_network = u32::from_be_bytes(virtual_netmask.octets())
+            & u32::from_be_bytes(virtual_gateway.octets());
+        let virtual_network = Ipv4Addr::from(virtual_network);
+        self.virtual_ip = virtual_ip;
+        self.virtual_netmask = virtual_netmask;
+        self.virtual_gateway = virtual_gateway;
+        self.broadcast_ip = broadcast_ip;
+        self.virtual_network = virtual_network;
+    }
     #[inline]
     pub fn virtual_ip(&self) -> Ipv4Addr {
         self.virtual_ip
@@ -114,4 +188,7 @@ impl CurrentDeviceInfo {
     pub fn virtual_gateway(&self) -> Ipv4Addr {
         self.virtual_gateway
     }
+    pub fn is_gateway(&self, ip: &Ipv4Addr) -> bool {
+        &self.virtual_gateway == ip || ip == &GATEWAY_IP
+    }
 }