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