WIP
This commit is contained in:
50
peer/on_tick.go
Normal file
50
peer/on_tick.go
Normal file
@@ -0,0 +1,50 @@
|
||||
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()
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user