xmengnet/mihomo
1
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Improve: Better handling of TCP connections

Browse source
This commit is contained in:
Dreamacro 2018-08-29 15:00:12 +08:00
parent a5f2bd3152
commit f4c51cdb0e
4 changed files with 10 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
adapters/remote/direct.go
View file

@ -36,7 +36,7 @@ func (d *Direct) Generator(addr *C.Addr) (adapter C.ProxyAdapter, err error) {
if err != nil {
return
}
c.(*net.TCPConn).SetKeepAlive(true)
tcpKeepAlive(c)
return &DirectAdapter{conn: c}, nil
}

2
adapters/remote/shadowsocks.go
View file

@ -46,7 +46,7 @@ func (ss *ShadowSocks) Generator(addr *C.Addr) (adapter C.ProxyAdapter, err erro
if err != nil {
return nil, fmt.Errorf("%s connect error", ss.server)
}
c.(*net.TCPConn).SetKeepAlive(true)
tcpKeepAlive(c)
c = ss.cipher.StreamConn(c)
_, err = c.Write(serializesSocksAddr(addr))
return &ShadowsocksAdapter{conn: c}, err

3
adapters/remote/socks5.go
View file

@ -44,8 +44,7 @@ func (ss *Socks5) Generator(addr *C.Addr) (adapter C.ProxyAdapter, err error) {
if err != nil {
return nil, fmt.Errorf("%s connect error", ss.addr)
}
c.(*net.TCPConn).SetKeepAlive(true)
tcpKeepAlive(c)
if err := ss.sharkHand(addr, c); err != nil {
return nil, err
}

7
adapters/remote/util.go
View file

@ -84,3 +84,10 @@ func selectFast(in chan interface{}) chan interface{} {
return out
}
func tcpKeepAlive(c net.Conn) {
if tcp, ok := c.(*net.TCPConn); ok {
tcp.SetKeepAlive(true)
tcp.SetKeepAlivePeriod(3 * time.Minute)
}
}