adapt tun device to zerocopy (#57)

easytier/src/peers/peer.rs

@@ -1,5 +1,6 @@
 use std::sync::Arc;
 
+use crossbeam::atomic::AtomicCell;
 use dashmap::DashMap;
 
 use tokio::{
@@ -38,6 +39,8 @@ pub struct Peer {
     close_event_listener: JoinHandle<()>,
 
     shutdown_notifier: Arc<tokio::sync::Notify>,
+
+    default_conn_id: AtomicCell<PeerConnId>,
 }
 
 impl Peer {
@@ -99,6 +102,7 @@ impl Peer {
             close_event_listener,
 
             shutdown_notifier,
+            default_conn_id: AtomicCell::new(PeerConnId::default()),
         }
     }
 
@@ -112,8 +116,24 @@ impl Peer {
             .insert(conn.get_conn_id(), Arc::new(Mutex::new(conn)));
     }
 
+    async fn select_conn(&self) -> Option<ArcPeerConn> {
+        let default_conn_id = self.default_conn_id.load();
+        if let Some(conn) = self.conns.get(&default_conn_id) {
+            return Some(conn.clone());
+        }
+
+        let conn = self.conns.iter().next();
+        if conn.is_none() {
+            return None;
+        }
+
+        let conn = conn.unwrap().clone();
+        self.default_conn_id.store(conn.lock().await.get_conn_id());
+        Some(conn)
+    }
+
     pub async fn send_msg(&self, msg: ZCPacket) -> Result<(), Error> {
-        let Some(conn) = self.conns.iter().next() else {
+        let Some(conn) = self.select_conn().await else {
             return Err(Error::PeerNoConnectionError(self.peer_node_id));
         };
 

easytier/src/peers/peer_map.rs

@@ -118,12 +118,6 @@ impl PeerMap {
 
     pub async fn send_msg(&self, msg: ZCPacket, dst_peer_id: PeerId) -> Result<(), Error> {
         let Some(gateway_peer_id) = self.get_gateway_peer_id(dst_peer_id).await else {
-            tracing::trace!(
-                "no gateway for dst_peer_id: {}, peers: {:?}, my_peer_id: {}",
-                dst_peer_id,
-                self.peer_map.iter().map(|v| *v.key()).collect::<Vec<_>>(),
-                self.my_peer_id
-            );
             return Err(Error::RouteError(Some(format!(
                 "peer map sengmsg no gateway for dst_peer_id: {}",
                 dst_peer_id

easytier/src/peers/peer_ospf_route.rs

@@ -1265,7 +1265,7 @@ impl Route for PeerRoute {
             return Some(peer_id);
         }
 
-        tracing::info!("no peer id for ipv4: {}", ipv4_addr);
+        tracing::info!(?ipv4_addr, "no peer id for ipv4");
         None
     }
 }