set conn deadline

This commit is contained in:
Jason
2019-08-11 21:30:32 +08:00
parent c22db614ff
commit 07a658ae0e

View File

@@ -54,6 +54,8 @@ func (h *tcpHandler) relay(localConn, remoteConn net.Conn) {
if _, err := io.Copy(remoteConn, localConn); err != nil { if _, err := io.Copy(remoteConn, localConn); err != nil {
closeOnce() closeOnce()
} else { } else {
localConn.SetDeadline(time.Now())
remoteConn.SetDeadline(time.Now())
tcpCloseRead(localConn) tcpCloseRead(localConn)
tcpCloseWrite(remoteConn) tcpCloseWrite(remoteConn)
} }
@@ -64,6 +66,8 @@ func (h *tcpHandler) relay(localConn, remoteConn net.Conn) {
if _, err := io.Copy(localConn, remoteConn); err != nil { if _, err := io.Copy(localConn, remoteConn); err != nil {
closeOnce() closeOnce()
} else { } else {
localConn.SetDeadline(time.Now())
remoteConn.SetDeadline(time.Now())
tcpCloseRead(remoteConn) tcpCloseRead(remoteConn)
tcpCloseWrite(localConn) tcpCloseWrite(localConn)
} }