fix(listener/tproxy): possilly using released buffer in tproxy

MetaCubeX · May 21, 2024 · 0c47e73 · 0c47e73
1 parent 43bdc76
commit 0c47e73
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/listener/tproxy/packet.go b/listener/tproxy/packet.go
@@ -105,11 +105,12 @@ func listenLocalConn(rAddr, lAddr net.Addr, tunnel C.Tunnel) (*net.UDPConn, erro
 			buf := pool.Get(pool.UDPBufferSize)
 			br, err := lc.Read(buf)
 			if err != nil {
-				pool.Put(buf)
-				if errors.Is(err, net.ErrClosed) {
-					log.Debugln("TProxy local conn listener exit.. rAddr=%s lAddr=%s", rAddr.String(), lAddr.String())
-					return
+				if !errors.Is(err, net.ErrClosed) {
+					log.Debugln("TProxy local conn Read err=%v", err)
 				}
+				log.Debugln("TProxy local conn listener exit.. rAddr=%s lAddr=%s", rAddr.String(), lAddr.String())
+				pool.Put(buf)
+				return
 			}
 			// since following localPackets are pass through this socket which listen rAddr
 			// I choose current listener as packet's packet conn