From ecebbecd3b0c162a4901f9e9918c5cecd8192183 Mon Sep 17 00:00:00 2001
From: BlackLuny <602814112@qq.com>
Date: Mon, 9 Jun 2025 19:35:29 +0800
Subject: [PATCH] add check for rpc packet fix #963 (#969)

---
 easytier/src/proto/rpc_impl/bidirect.rs | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/easytier/src/proto/rpc_impl/bidirect.rs b/easytier/src/proto/rpc_impl/bidirect.rs
index a673949f..231c3041 100644
--- a/easytier/src/proto/rpc_impl/bidirect.rs
+++ b/easytier/src/proto/rpc_impl/bidirect.rs
@@ -131,11 +131,14 @@ impl BidirectRpcManager {
                     }
                 };
 
-                if o.peer_manager_header().unwrap().packet_type == PacketType::RpcReq as u8 {
+                let Some(peer_manager_header) = o.peer_manager_header() else {
+                    tracing::error!("peer manager header not found");
+                    continue;
+                };
+                if peer_manager_header.packet_type == PacketType::RpcReq as u8 {
                     server_tx.send(o).await.unwrap();
                     continue;
-                } else if o.peer_manager_header().unwrap().packet_type == PacketType::RpcResp as u8
-                {
+                } else if peer_manager_header.packet_type == PacketType::RpcResp as u8 {
                     client_tx.send(o).await.unwrap();
                     continue;
                 }