add tcp keep-alived (#20)

shadowsocks/go-shadowsocks2@9dfb3e5
Qv2ray · Mar 14, 2021 · 04bf6df · 04bf6df
1 parent 92ace24
commit 04bf6df
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/dispatcher/tcp/tcp.go b/dispatcher/tcp/tcp.go
@@ -79,6 +79,7 @@ func (d *TCP) handleConn(conn net.Conn) error {
 		userContext *config.UserContext
 	)
 	defer conn.Close()
+	_ = conn.(*net.TCPConn).SetKeepAlive(true)
 
 	var data = pool.Get(BasicLen)
 	defer pool.Put(data)
@@ -109,6 +110,7 @@ func (d *TCP) handleConn(conn net.Conn) error {
 	if err != nil {
 		return fmt.Errorf("[tcp] handleConn dial error: %w", err)
 	}
+	_ = rc.(*net.TCPConn).SetKeepAlive(true)
 
 	_ = rc.SetDeadline(time.Now().Add(DefaultTimeout))
 	_, err = rc.Write(data[:n])