Use leaky buffer to when piping data.

shadowsocks/pipe.go

@@ -2,6 +2,7 @@ package shadowsocks
 
 import (
 	// "io"
+	"github.com/cyfdecyf/leakybuf"
 	"net"
 	"time"
 )
@@ -17,10 +18,16 @@ func SetReadTimeout(c net.Conn) {
 	}
 }
 
+const bufSize = 4096
+const nBuf = 2048
+
+var pipeBuf = leakybuf.NewLeakyBuf(nBuf, bufSize)
+
 // PipeThenClose copies data from src to dst, closes dst when done.
 func PipeThenClose(src, dst net.Conn, timeoutOpt int) {
 	defer dst.Close()
-	buf := make([]byte, 4096)
+	buf := pipeBuf.Get()
+	defer pipeBuf.Put(buf)
 	for {
 		if timeoutOpt == SET_TIMEOUT {
 			SetReadTimeout(src)