probe when reachable

1 files changed, 24 insertions, 0 deletions

diff --git a/src/swarm/peer/handshake.rs b/src/swarm/peer/handshake.rs
index 6bd4784..072545b 100644
--- a/src/swarm/peer/handshake.rs
+++ b/src/swarm/peer/handshake.rs
@@ -304,6 +304,30 @@ pub(super) async fn discover_peers(
     }
 }
 
+/// Whenever others may reach this node where they couldn't before
+/// (`Transport::on_reachable`: a relay granted it a slot), the members it can't reach
+/// are probed at once (`probe_now`): they may have given up on it meanwhile, waiting
+/// for their next probe, up to `dormant-probe-max` away. Two nodes behind NATs that
+/// restart together each fail to reach the other until it holds a slot; the second to
+/// get one reaches the first this way. Runs for the node's life.
+pub(crate) async fn probe_when_reachable(state: AppState, reachable: Arc<tokio::sync::Notify>) {
+    loop {
+        reachable.notified().await;
+        let mut probed = 0;
+        for (peer, _) in state.peers_snapshot().await {
+            if probe_now(&state, &peer).await {
+                probed += 1;
+            }
+        }
+        if probed > 0 {
+            debug!(
+                probed,
+                "Reachable anew; probing the members we can't reach now"
+            );
+        }
+    }
+}
+
 /// Brings `peer`'s next probe forward to now, if it is a member we can't reach (dormant,
 /// or unanswered HELLOs) whose probe is pending: the peer worker sends it (woken for
 /// it). Whether it was.