51 lines
1.0 KiB
Go
51 lines
1.0 KiB
Go
package peer
|
|
|
|
import (
|
|
"log"
|
|
"time"
|
|
|
|
"vppn/peer/control"
|
|
)
|
|
|
|
func (a *App) onTick() {
|
|
wgPeers := a.devPeers()
|
|
|
|
a.nextPingID++
|
|
now := time.Now().UnixNano()
|
|
|
|
// Update Up values.
|
|
for _, p := range a.peersByKey {
|
|
p.Up = p.Alive()
|
|
}
|
|
|
|
for _, wgPeer := range wgPeers {
|
|
p, ok := a.peersByKey[wgPeer.PublicKey]
|
|
if !ok {
|
|
log.Fatalf("Wireguard peer not in index: %v", wgPeer)
|
|
}
|
|
|
|
// Send pings to peers where we're the client.
|
|
if p.Role == control.Client {
|
|
a.sendPing(p, a.nextPingID, now)
|
|
}
|
|
|
|
// Promote probing peers to direct once alive (direct path confirmed
|
|
// working).
|
|
if p.State == StateProbing && time.Since(wgPeer.LastHandshakeTime) < 2*PingInterval {
|
|
p.State = StateDirect
|
|
a.devAddDirect(p, p.WGEndpoint)
|
|
}
|
|
|
|
// Demote stale non-public direct peers back to probing.
|
|
if p.State == StateDirect && !p.IsPublic && !p.Up {
|
|
p.State = StateProbing
|
|
a.devAddProbe(p, p.WGEndpoint)
|
|
}
|
|
}
|
|
|
|
// Ensure we have a live relay.
|
|
if a.relay == nil || !a.relay.Up {
|
|
a.switchActiveRelay()
|
|
}
|
|
}
|