Refactor - now wireguard based. (#7)

peer/on_tick.go

@@ -0,0 +1,52 @@
+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:
+			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()
+	}
+}