rose_cat707
0d84b07f68
CI / docker (push) Successful in 1m58s
Go + Vue 管理台：Agent 隧道、域名反代、IP 安全、访客验证与监控大盘。 Gitea Actions CI 多架构镜像；纯 Go SQLite、无 CGO Docker 构建。 Co-authored-by: Cursor <cursoragent@cursor.com>
47 lines
1022 B
Go
47 lines
1022 B
Go
package bridge
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"net"
|
|
"time"
|
|
|
|
"github.com/xtaci/smux"
|
|
)
|
|
|
|
const (
|
|
// HeartbeatInterval is the application-level ping interval (agent and server probe).
|
|
HeartbeatInterval = 30 * time.Second
|
|
// HeartbeatTimeout evicts a session when no ping/pong for this duration.
|
|
HeartbeatTimeout = 90 * time.Second
|
|
// WatchdogInterval is how often the server checks agent heartbeats.
|
|
WatchdogInterval = 15 * time.Second
|
|
// ServerProbeAfter triggers a server-initiated ping when agent has been silent.
|
|
ServerProbeAfter = 45 * time.Second
|
|
)
|
|
|
|
func SmuxConfig() *smux.Config {
|
|
return smuxConfig()
|
|
}
|
|
|
|
func smuxConfig() *smux.Config {
|
|
cfg := smux.DefaultConfig()
|
|
cfg.KeepAliveInterval = 10 * time.Second
|
|
cfg.KeepAliveTimeout = 30 * time.Second
|
|
return cfg
|
|
}
|
|
|
|
func SetTCPKeepAlive(conn net.Conn) {
|
|
for {
|
|
switch c := conn.(type) {
|
|
case *net.TCPConn:
|
|
_ = c.SetKeepAlive(true)
|
|
_ = c.SetKeepAlivePeriod(30 * time.Second)
|
|
return
|
|
case *tls.Conn:
|
|
conn = c.NetConn()
|
|
default:
|
|
return
|
|
}
|
|
}
|
|
}
|