From 0f0a0558391ba391c089806c23f369651f6c9eeb Mon Sep 17 00:00:00 2001
From: "sijie.sun" <sijie.sun@smartx.com>
Date: Sat, 8 Feb 2025 23:32:11 +0800
Subject: [PATCH] all incoming packet should update pong time

---
 src/endpoint.rs | 19 +++++++++----------
 1 file changed, 9 insertions(+), 10 deletions(-)

diff --git a/src/endpoint.rs b/src/endpoint.rs
index e6dccf1..2bff65a 100644
--- a/src/endpoint.rs
+++ b/src/endpoint.rs
@@ -465,11 +465,9 @@ impl KcpEndpoint {
         packet: &KcpPacket,
         output_sender: &KcpPakcetSender,
     ) -> bool {
-        if !packet.header().is_ping() {
-            return false;
-        }
+        let hdr = packet.header();
 
-        if !packet.header().is_pong() {
+        if hdr.is_ping() && !hdr.is_pong() {
             let conn_id = ConnId::from(packet);
             let need_send_pong = data
                 .state_map
@@ -490,14 +488,15 @@ impl KcpEndpoint {
             if let Err(e) = ret {
                 tracing::error!(?e, "send pong packet failed");
             }
-        } else {
-            let conv = ConnId::from(packet);
-            if let Some(mut state) = data.state_map.get_mut(&conv) {
-                state.notify_pong();
-            }
         }
 
-        true
+        // all incoming packet should update pong time
+        let conv = ConnId::from(packet);
+        if let Some(mut state) = data.state_map.get_mut(&conv) {
+            state.notify_pong();
+        }
+
+        packet.header().is_ping()
     }
 
     pub async fn run(&mut self) {