Fix: return pooled buffer when simple-obfs tls read error (#2643)

sketchrrr · Mar 26, 2023 · 7e2974f · 7e2974f
1 parent 8f9b39c
commit 7e2974f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/transport/simple-obfs/tls.go b/transport/simple-obfs/tls.go
@@ -27,10 +27,10 @@ type TLSObfs struct {
 func (to *TLSObfs) read(b []byte, discardN int) (int, error) {
 	buf := pool.Get(discardN)
 	_, err := io.ReadFull(to.Conn, buf)
+	pool.Put(buf)
 	if err != nil {
 		return 0, err
 	}
-	pool.Put(buf)
 
 	sizeBuf := make([]byte, 2)
 	_, err = io.ReadFull(to.Conn, sizeBuf)