Add control.SetKeepAlivePeriod

Author: nekohasekai

common/control/tcp_keep_alive_default.go

@@ -0,0 +1,12 @@
+//go:build !linux
+
+package control
+
+import (
+	"net"
+	"time"
+)
+
+func SetKeepAlivePeriod(conn *net.TCPConn, initial time.Duration, interval time.Duration) error {
+	return conn.SetKeepAlivePeriod(initial)
+}

common/control/tcp_keep_alive_linux.go

@@ -0,0 +1,22 @@
+package control
+
+import (
+	E "github.com/sagernet/sing/common/exceptions"
+	"net"
+	"time"
+
+	"golang.org/x/sys/unix"
+	_ "unsafe"
+)
+
+func SetKeepAlivePeriod(conn *net.TCPConn, initial time.Duration, interval time.Duration) error {
+	return Conn(conn, func(fd uintptr) error {
+		return E.Errors(
+			unix.SetsockoptInt(int(fd), unix.IPPROTO_TCP, unix.TCP_KEEPIDLE, int(roundDurationUp(initial, time.Second))),
+			unix.SetsockoptInt(int(fd), unix.IPPROTO_TCP, unix.TCP_KEEPINTVL, int(roundDurationUp(interval, time.Second))),
+		)
+	})
+}
+
+//go:linkname roundDurationUp net.roundDurationUp
+func roundDurationUp(d time.Duration, to time.Duration) time.Duration