add multi endpoint

src/main.rs

@@ -8,6 +8,7 @@ use std::intrinsics::transmute;
 use std::io::{Read, Write};
 use std::mem::MaybeUninit;
 use std::sync::Arc;
+use std::time::{Duration, Instant};
 
 #[repr(C)]
 pub struct Meta {
@@ -93,10 +94,26 @@ fn main() -> Result<(), Box<dyn Error>> {
 
                 loop {
                     let n = router.tun_reader.read(&mut buffer[meta_size..]).unwrap();
-                    if let Some(ref addr) = *router.endpoint.read().unwrap() {
-                        router.encrypt(&mut buffer[meta_size..meta_size + n]);
-                        #[cfg(target_os = "linux")]
-                        let _ = router.socket.set_mark(router.config.mark);
+                
+                    // 加密 payload
+                    router.encrypt(&mut buffer[meta_size..meta_size + n]);
+                
+                    #[cfg(target_os = "linux")]
+                    let _ = router.socket.set_mark(router.config.mark);
+                
+                    // 计算是否超时
+                    let now = Instant::now();
+                    let last_seen = *router.last_seen.read().unwrap();
+                    let elapsed = now.duration_since(last_seen);
+                
+                    if elapsed > Duration::from_secs(5) {
+                        // ⏱ 超时，广播给所有 endpoints
+                        let endpoints = router.endpoints.read().unwrap();
+                        for addr in endpoints.iter() {
+                            let _ = router.socket.send_to(&buffer[..meta_size + n], addr);
+                        }
+                    } else if let Some(ref addr) = *router.endpoint.read().unwrap() {
+                        // ✅ 正常单播
                         let _ = router.socket.send_to(&buffer[..meta_size + n], addr);
                     }
                 }
@@ -125,6 +142,11 @@ fn main() -> Result<(), Box<dyn Error>> {
                                 .get_mut(&meta.src_id)
                                 .ok_or("missing router")?;
                             *router.endpoint.write().unwrap() = Some(addr);
+
+                            *router.last_seen.write().unwrap() = Instant::now(); // ✅ 更新时间戳
+                            let mut endpoints = router.endpoints.write().unwrap();
+                            endpoints.insert(addr); // ✅ 加入新的地址
+
                             router.decrypt(payload, &local_secret);
                             router.tun_writer.write_all(payload)?;
                         }

src/router.rs

@@ -52,6 +52,8 @@ pub struct Router<'a> {
     pub tun_writer: Writer,
     pub socket: Arc<Socket>,
     pub endpoint: Arc<RwLock<Option<SockAddr>>>,
+    pub endpoints: Arc<RwLock<HashSet<SocketAddr>>>,
+    pub last_seen: Arc<RwLock<Instant>>,
 }
 
 impl<'a> Router<'a> {
@@ -122,6 +124,13 @@ impl<'a> Router<'a> {
         let (tun_reader, tun_writer) = Self::create_tun_device(&config)?;
         Self::run_up_script(&config)?;
 
+        let mut endpoints_set = HashSet::new();
+        if let Some(addr) = *endpoint.read().unwrap() {
+            endpoints_set.insert(addr);
+        }
+        let endpoints = Arc::new(RwLock::new(endpoints_set));
+        let last_seen = Arc::new(RwLock::new(Instant::now()));
+
         let router = Router {
             config,
             secret,
@@ -129,6 +138,8 @@ impl<'a> Router<'a> {
             tun_reader,
             tun_writer,
             socket,
+            endpoints,
+            last_seen,
         };
 
         Ok(router)