package/tun2socks
1
0
Fork 0
mirror of https://github.com/xjasonlyu/tun2socks.git synced 2025-10-07 01:33:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

Chore: make set operation atomic

Browse Source
This commit is contained in:
xjasonlyu
2021-03-13 11:50:14 +08:00
parent f41b844e43
commit 094199656b
2 changed files with 17 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tunnel/udp.go
View File

@@ -14,6 +14,8 @@ import (
"github.com/xjasonlyu/tun2socks/log"
"github.com/xjasonlyu/tun2socks/proxy"
"github.com/xjasonlyu/tun2socks/tunnel/statistic"
"go.uber.org/atomic"
)
var (
@@ -23,11 +25,11 @@ var (
// _udpSessionTimeout is the default timeout for
// each UDP session.
_udpSessionTimeout = 60 * time.Second
_udpSessionTimeout = atomic.NewInt64(int64(60 * time.Second))
)
func SetUDPTimeout(v int) {
_udpSessionTimeout = time.Duration(v) * time.Second
_udpSessionTimeout.Store(int64(time.Duration(v) * time.Second))
}
func newUDPTracker(conn net.PacketConn, metadata *M.Metadata) net.PacketConn {
@@ -119,7 +121,7 @@ func handleUDPToRemote(packet core.UDPPacket, pc net.PacketConn, remote net.Addr
if _, err := pc.WriteTo(packet.Data() /* data */, remote); err != nil {
log.Warnf("[UDP] write to %s error: %v", remote, err)
}
pc.SetReadDeadline(time.Now().Add(_udpSessionTimeout)) /* reset timeout */
pc.SetReadDeadline(time.Now().Add(time.Duration(_udpSessionTimeout.Load()))) /* reset timeout */
log.Infof("[UDP] %s --> %s", packet.RemoteAddr(), remote)
}
@@ -129,7 +131,7 @@ func handleUDPToLocal(packet core.UDPPacket, pc net.PacketConn) {
defer pool.Put(buf)
for /* just loop */ {
pc.SetReadDeadline(time.Now().Add(_udpSessionTimeout))
pc.SetReadDeadline(time.Now().Add(time.Duration(_udpSessionTimeout.Load())))
n, from, err := pc.ReadFrom(buf)
if err != nil {
if !errors.Is(err, os.ErrDeadlineExceeded) /* ignore i/o timeout */ {