apps/pg
1
0
Fork 0
mirror of https://github.com/sigcn/pg.git synced 2025-10-28 21:11:48 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

disco: port scan for Symmetric NAT

Browse Source
This commit is contained in:
rkonfj
2024-02-04 16:38:55 +08:00
parent cc754ca826
commit b0d7f0bcd0
2 changed files with 129 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

90
disco/types.go
View File

@@ -5,6 +5,7 @@ import (
"log/slog"
"math/rand"
"net"
"sync"
"time"
"github.com/rkonfj/peerguard/peer"
@@ -23,13 +24,59 @@ const (
)
type PeerContext struct {
PeerID peer.PeerID
States map[string]*PeerState
CreateTime time.Time
exitSig chan struct{}
ping func(addr *net.UDPAddr)
keepaliveInterval time.Duration
statesMutex sync.RWMutex
}
func (peer *PeerContext) AddAddr(addr *net.UDPAddr) {
peer.statesMutex.Lock()
defer peer.statesMutex.Unlock()
peer.States[addr.String()] = &PeerState{Addr: addr}
}
func (peer *PeerContext) Heartbeat(addr *net.UDPAddr) {
peer.statesMutex.RLock()
defer peer.statesMutex.RUnlock()
for _, state := range peer.States {
if state.Addr.IP.Equal(addr.IP) && state.Addr.Port == addr.Port {
updated := time.Since(state.LastActiveTime) > 2*peer.keepaliveInterval+2*time.Second
if updated {
slog.Info("[UDP] AddPeer", "peer", peer.PeerID, "addr", addr)
}
state.LastActiveTime = time.Now()
}
}
}
func (peer *PeerContext) Healthcheck() {
if time.Since(peer.CreateTime) > 3*peer.keepaliveInterval {
for addr, state := range peer.States {
if time.Since(state.LastActiveTime) > 2*peer.keepaliveInterval {
if state.LastActiveTime.IsZero() {
slog.Debug("[UDP] RemovePeer", "peer", peer.PeerID, "addr", state.Addr)
} else {
slog.Info("[UDP] RemovePeer", "peer", peer.PeerID, "addr", state.Addr)
}
peer.statesMutex.Lock()
delete(peer.States, addr)
peer.statesMutex.Unlock()
}
}
}
}
func (peer *PeerContext) IPv4Ready() bool {
peer.statesMutex.RLock()
defer peer.statesMutex.RUnlock()
for _, state := range peer.States {
if state.Addr.IP.To4() != nil && time.Since(state.LastActiveTime) <= 20*time.Second {
if state.Addr.IP.To4() != nil && time.Since(state.LastActiveTime) <= peer.keepaliveInterval+2*time.Second {
return true
}
}
@@ -37,8 +84,10 @@ func (peer *PeerContext) IPv4Ready() bool {
}
func (peer *PeerContext) Ready() bool {
peer.statesMutex.RLock()
defer peer.statesMutex.RUnlock()
for _, state := range peer.States {
if time.Since(state.LastActiveTime) <= 20*time.Second {
if time.Since(state.LastActiveTime) <= peer.keepaliveInterval+2*time.Second {
return true
}
}
@@ -46,15 +95,50 @@ func (peer *PeerContext) Ready() bool {
}
func (peer *PeerContext) Select() *net.UDPAddr {
peer.statesMutex.RLock()
defer peer.statesMutex.RUnlock()
addrs := make([]*net.UDPAddr, 0, len(peer.States))
for _, state := range peer.States {
if time.Since(state.LastActiveTime) <= 20*time.Second {
if time.Since(state.LastActiveTime) <= peer.keepaliveInterval+2*time.Second {
addrs = append(addrs, state.Addr)
}
}
return addrs[rand.Intn(len(addrs))]
}
func (peer *PeerContext) Keepalive() {
ticker := time.NewTicker(peer.keepaliveInterval)
ping := func() {
addrs := make([]*net.UDPAddr, 0, len(peer.States))
peer.statesMutex.RLock()
for _, v := range peer.States {
if time.Since(v.LastActiveTime) > 2*peer.keepaliveInterval+2*time.Second {
continue
}
addrs = append(addrs, v.Addr)
}
peer.statesMutex.RUnlock()
for _, addr := range addrs {
peer.ping(addr)
}
}
for {
select {
case <-peer.exitSig:
ticker.Stop()
slog.Debug("[UDP] Ping exit", "peer", peer.PeerID)
return
case <-ticker.C:
ping()
}
}
}
func (peer *PeerContext) Close() error {
close(peer.exitSig)
return nil
}
type PeerState struct {
Addr *net.UDPAddr
LastActiveTime time.Time