From ed2d05ab518ab85d85b00ad02c7566c25badde8c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E4=B8=96=E7=95=8C?= <i@sekai.icu>
Date: Thu, 27 Apr 2023 23:17:30 +0800
Subject: [PATCH] Fix copyPacketWaitWithPool

---
 common/bufio/copy.go              | 2 +-
 common/bufio/copy_direct_posix.go | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/common/bufio/copy.go b/common/bufio/copy.go
index 00d78b0..bc1b1d8 100644
--- a/common/bufio/copy.go
+++ b/common/bufio/copy.go
@@ -71,7 +71,7 @@ func CopyExtended(originDestination io.Writer, destination N.ExtendedWriter, sou
 			return
 		}
 	}
-	if N.IsUnsafeWriter(destination) {
+	if !common.UnsafeBuffer || N.IsUnsafeWriter(destination) {
 		return CopyExtendedWithPool(originDestination, destination, source, readCounters, writeCounters)
 	}
 	bufferSize := N.CalculateMTU(source, destination)
diff --git a/common/bufio/copy_direct_posix.go b/common/bufio/copy_direct_posix.go
index 8f79c94..870097c 100644
--- a/common/bufio/copy_direct_posix.go
+++ b/common/bufio/copy_direct_posix.go
@@ -113,6 +113,8 @@ func copyPacketWaitWithPool(destinationConn N.PacketWriter, source N.PacketReadW
 		if err != nil {
 			buffer.Release()
 			return
+		} else {
+			buffer = nil
 		}
 		n += int64(dataLen)
 		for _, counter := range readCounters {