device: add SendKeepalivesToPeersWithCurrentKeypair for handover

diff --git a/device/device.go b/device/device.go
index fa7ed88fa720070fda57b66223dfb48f2728e488..77758447c366c756d1a2d8298e083c4e2cf74fb2 100644 (file)
--- a/device/device.go
+++ b/device/device.go
@@ -396,3 +396,20 @@ func (device *Device) Close() {
 func (device *Device) Wait() chan struct{} {
        return device.signals.stop
 }
+
+func (device *Device) SendKeepalivesToPeersWithCurrentKeypair() {
+       if device.isClosed.Get() {
+               return
+       }
+
+       device.peers.RLock()
+       for _, peer := range device.peers.keyMap {
+               peer.keypairs.RLock()
+               sendKeepalive := peer.keypairs.current != nil && !peer.keypairs.current.created.Add(RejectAfterTime).Before(time.Now())
+               peer.keypairs.RUnlock()
+               if sendKeepalive {
+                       peer.SendKeepalive()
+               }
+       }
+       device.peers.RUnlock()
+}