53 lines
1.1 KiB
Go
53 lines
1.1 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 known. Removing: %v", wgPeer.PublicKey)
|
|
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:
|
|
if p.IsPublic || a.isPublic || p.Up() {
|
|
break
|
|
}
|
|
// Stale non-public direct peer: demote to probing so WireGuard
|
|
// resumes handshake attempts on the direct path.
|
|
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()
|
|
}
|
|
}
|