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

Feature: cli tcp-wait-timeout option (#156)

Browse Source 
Co-authored-by: xjasonlyu <xjasonlyu@gmail.com>
This commit is contained in:
Solyn
2023-03-29 17:41:01 +08:00
committed by GitHub
parent fb9ca95909
commit 2c51a65685
4 changed files with 13 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tunnel/tcp.go
View File

@@ -16,9 +16,11 @@ import (
"github.com/xjasonlyu/tun2socks/v2/tunnel/statistic"
)
const (
tcpWaitTimeout = 5 * time.Second
)
var _tcpWaitTimeout = 5 * time.Second
func SetTCPWaitTimeout(t time.Duration) {
_tcpWaitTimeout = t
}
func handleTCPConn(localConn adapter.TCPConn) {
defer localConn.Close()
@@ -60,7 +62,7 @@ func relay(left, right net.Conn) error {
if err := copyBuffer(right, left); err != nil {
leftErr = errors.Join(leftErr, err)
}
right.SetReadDeadline(time.Now().Add(tcpWaitTimeout))
right.SetReadDeadline(time.Now().Add(_tcpWaitTimeout))
}()
go func() {
@@ -68,7 +70,7 @@ func relay(left, right net.Conn) error {
if err := copyBuffer(left, right); err != nil {
rightErr = errors.Join(rightErr, err)
}
left.SetReadDeadline(time.Now().Add(tcpWaitTimeout))
left.SetReadDeadline(time.Now().Add(_tcpWaitTimeout))
}()
wg.Wait()