From da315a18186e3949fd26cb44a791201c57939165 Mon Sep 17 00:00:00 2001
From: jaysunxiao <jaysunxiao@gmail.com>
Date: Fri, 8 Aug 2025 17:31:22 +0800
Subject: [PATCH] chore[tunnel]: enhance the robustness of the code

---
 .../net/core/proxy/handler/TunnelServerCodecHandler.java    | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/net/src/main/java/com/zfoo/net/core/proxy/handler/TunnelServerCodecHandler.java b/net/src/main/java/com/zfoo/net/core/proxy/handler/TunnelServerCodecHandler.java
index 9f996e6b..52b35d75 100644
--- a/net/src/main/java/com/zfoo/net/core/proxy/handler/TunnelServerCodecHandler.java
+++ b/net/src/main/java/com/zfoo/net/core/proxy/handler/TunnelServerCodecHandler.java
@@ -92,7 +92,11 @@ public class TunnelServerCodecHandler extends ByteToMessageCodec<TunnelProtocolS
         }
 
         var retainedByteBuf = in.readRetainedSlice(length);
-        session.getChannel().writeAndFlush(TunnelProtocolServer2Client.valueOf(sid, uid, retainedByteBuf));
+        try {
+            session.getChannel().writeAndFlush(TunnelProtocolServer2Client.valueOf(sid, uid, retainedByteBuf));
+        } catch (Throwable t) {
+            ReferenceCountUtil.release(retainedByteBuf);
+        }
     }
 
     @Override