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

cleanup

Browse Source
This commit is contained in:
Jason
2019-08-11 21:11:08 +08:00
parent 6d50f90315
commit c22db614ff
2 changed files with 12 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
proxy/socks/tcp.go
View File

@@ -51,16 +51,22 @@ func (h *tcpHandler) relay(localConn, remoteConn net.Conn) {
// Up Link
go func() {
io.Copy(remoteConn, localConn)
tcpCloseRead(localConn)
tcpCloseWrite(remoteConn)
if _, err := io.Copy(remoteConn, localConn); err != nil {
closeOnce()
} else {
tcpCloseRead(localConn)
tcpCloseWrite(remoteConn)
}
wg.Done()
}()
// Down Link
io.Copy(localConn, remoteConn)
tcpCloseRead(remoteConn)
tcpCloseWrite(localConn)
if _, err := io.Copy(localConn, remoteConn); err != nil {
closeOnce()
} else {
tcpCloseRead(remoteConn)
tcpCloseWrite(localConn)
}
wg.Wait() // Wait for Up Link done