chore: Add inbound and outbound probe debug logs

ipvm-wg · Apr 2, 2024 · 87ca02f · 87ca02f
1 parent 871ba53
commit 87ca02f
Showing 1 changed file with 59 additions and 12 deletions.
diff --git a/homestar-runtime/src/event_handler/swarm_event.rs b/homestar-runtime/src/event_handler/swarm_event.rs
@@ -109,18 +109,65 @@ async fn handle_swarm_event<DB: Database>(
     match event {
         SwarmEvent::Behaviour(ComposedEvent::Autonat(autonat_event)) => {
             match autonat_event {
-                autonat::Event::InboundProbe(event) => {
-                    // TODO Add log
-                    println!("INBOUND PROBE EVENT: {event:?}")
-                }
-                autonat::Event::OutboundProbe(event) => {
-                    // TODO Add log
-                    println!("OUTBOUND PROBE EVENT: {event:?}");
-                    println!(
-                        "CONFIDENCE: {}",
-                        event_handler.swarm.behaviour().autonat.confidence()
-                    )
-                }
+                autonat::Event::InboundProbe(event) => match event {
+                    autonat::InboundProbeEvent::Request {
+                        peer, addresses, ..
+                    } => {
+                        debug!(
+                            subject = "libp2p.autonat.inbound_probe",
+                            category = "handle_swarm_event",
+                            peer_id = peer.to_string(),
+                            addresses = ?addresses,
+                            "Received a probe request",
+                        );
+                    }
+                    autonat::InboundProbeEvent::Response { peer, address, .. } => {
+                        debug!(
+                            subject = "libp2p.autonat.inbound_probe",
+                            category = "handle_swarm_event",
+                            peer_id = peer.to_string(),
+                            address = address.to_string(),
+                            "Successfully probed an external address for a peer",
+                        );
+                    }
+                    autonat::InboundProbeEvent::Error { peer, error, .. } => {
+                        debug!(
+                            subject = "libp2p.autonat.inbound_probe",
+                            category = "handle_swarm_event",
+                            peer_id = peer.to_string(),
+                            error = ?error,
+                            "Unable to probe a peer",
+                        );
+                    }
+                },
+                autonat::Event::OutboundProbe(event) => match event {
+                    autonat::OutboundProbeEvent::Request { peer, .. } => {
+                        debug!(
+                            subject = "libp2p.autonat.outbound_probe",
+                            category = "handle_swarm_event",
+                            peer_id = peer.to_string(),
+                            "Requested a probe from a peer",
+                        );
+                    }
+                    autonat::OutboundProbeEvent::Response { peer, address, .. } => {
+                        debug!(
+                            subject = "libp2p.autonat.outbound_probe",
+                            category = "handle_swarm_event",
+                            peer_id = peer.to_string(),
+                            address = address.to_string(),
+                            "Peer successfully probed an external address",
+                        );
+                    }
+                    autonat::OutboundProbeEvent::Error { peer, error, .. } => {
+                        debug!(
+                            subject = "libp2p.autonat.outbound_probe",
+                            category = "handle_swarm_event",
+                            peer_id = peer.map(|p| p.to_string()).unwrap_or("<none>".to_string()),
+                            error = ?error,
+                            "Requested probe failed",
+                        );
+                    }
+                },
                 autonat::Event::StatusChanged { old, new } => {
                     match &new {
                         NatStatus::Public(address) => {