53 lines
1.2 KiB
Go
53 lines
1.2 KiB
Go
package peer
|
|
|
|
import (
|
|
"log"
|
|
"time"
|
|
|
|
"vppn/peer/control"
|
|
"vppn/peer/wginterface"
|
|
)
|
|
|
|
func (a *App) onTick() {
|
|
wgPeers := a.devPeers()
|
|
|
|
now := time.Now().UnixNano()
|
|
|
|
for _, wgPeer := range wgPeers {
|
|
p, ok := a.peersByKey[wgPeer.PublicKey]
|
|
if !ok {
|
|
log.Printf("Wireguard peer not in index, removing: %v", wgPeer)
|
|
a.devRemove(&Peer{wgPeer: wgPeer})
|
|
continue
|
|
}
|
|
p.wgPeer = wgPeer
|
|
|
|
// Send pings to peers where we're the client.
|
|
if p.Role == control.Client {
|
|
a.sendPing(p, now)
|
|
}
|
|
|
|
switch p.State {
|
|
case StateProbing:
|
|
// Promote probing peers to direct once alive (direct path confirmed
|
|
// working).
|
|
if time.Since(p.LastHandshakeTime()) < 2*wginterface.ProbeKeepalive {
|
|
a.devPromote(p)
|
|
}
|
|
|
|
case StateDirect:
|
|
// Demote stale non-public direct peers back to probing.
|
|
// Public nodes skip this: the non-public peer will re-initiate the
|
|
// handshake, and WireGuard will update the endpoint automatically.
|
|
if !p.IsPublic && !p.Up() && !a.isPublic {
|
|
a.devAddProbe(p, p.WGEndpoint())
|
|
}
|
|
}
|
|
}
|
|
|
|
// Ensure we have a live relay (if we're not public).
|
|
if !a.isPublic && (a.relay == nil || !a.relay.Up()) {
|
|
a.switchActiveRelay()
|
|
}
|
|
}
|