package/tun2socks
1
0
Fork 0
mirror of https://github.com/xjasonlyu/tun2socks.git synced 2025-11-01 20:42:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

use bufpool to optimize relay

Browse Source
This commit is contained in:
Jason
2019-08-12 14:29:48 +08:00
parent 51e02d0236
commit 371d10ea2b
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
proxy/socks/tcp.go
View File

@@ -12,6 +12,7 @@ import (
"github.com/xjasonlyu/tun2socks/common/dns" "github.com/xjasonlyu/tun2socks/common/dns"
"github.com/xjasonlyu/tun2socks/common/log" "github.com/xjasonlyu/tun2socks/common/log"
"github.com/xjasonlyu/tun2socks/common/lsof" "github.com/xjasonlyu/tun2socks/common/lsof"
"github.com/xjasonlyu/tun2socks/common/pool"
"github.com/xjasonlyu/tun2socks/common/stats" "github.com/xjasonlyu/tun2socks/common/stats"
"github.com/xjasonlyu/tun2socks/core" "github.com/xjasonlyu/tun2socks/core"
) )
@@ -51,7 +52,9 @@ func (h *tcpHandler) relay(localConn, remoteConn net.Conn) {
// Up Link // Up Link
go func() { go func() {
if _, err := io.Copy(remoteConn, localConn); err != nil { buf := pool.BufPool.Get().([]byte)
defer pool.BufPool.Put(buf[:cap(buf)])
if _, err := io.CopyBuffer(remoteConn, localConn, buf); err != nil {
closeOnce() closeOnce()
} else { } else {
localConn.SetDeadline(time.Now()) localConn.SetDeadline(time.Now())
@@ -62,13 +65,15 @@ func (h *tcpHandler) relay(localConn, remoteConn net.Conn) {
}() }()
// Down Link // Down Link
if _, err := io.Copy(localConn, remoteConn); err != nil { buf := pool.BufPool.Get().([]byte)
if _, err := io.CopyBuffer(localConn, remoteConn, buf); err != nil {
closeOnce() closeOnce()
} else { } else {
localConn.SetDeadline(time.Now()) localConn.SetDeadline(time.Now())
remoteConn.SetDeadline(time.Now()) remoteConn.SetDeadline(time.Now())
tcpCloseRead(localConn) tcpCloseRead(localConn)
} }
pool.BufPool.Put(buf[:cap(buf)])
wg.Wait() // Wait for Up Link done wg.Wait() // Wait for Up Link done