fix(autonat): close dial-back connections once probes complete

Author: imabdulbasit

protocols/autonat/src/v1/behaviour.rs

@@ -39,8 +39,8 @@ use libp2p_request_response::{
     self as request_response, InboundRequestId, OutboundRequestId, ProtocolSupport, ResponseChannel,
 };
 use libp2p_swarm::{
-    ConnectionDenied, ConnectionId, ListenAddresses, NetworkBehaviour, THandler, THandlerInEvent,
-    THandlerOutEvent, ToSwarm,
+    CloseConnection, ConnectionDenied, ConnectionId, ListenAddresses, NetworkBehaviour, THandler,
+    THandlerInEvent, THandlerOutEvent, ToSwarm,
     behaviour::{AddressChange, ConnectionClosed, ConnectionEstablished, DialFailure, FromSwarm},
 };
 use web_time::Instant;
@@ -349,6 +349,15 @@ impl Behaviour {
                 if let Some(event) = self.as_server().on_outbound_connection(&peer, address) {
                     self.pending_actions
                         .push_back(ToSwarm::GenerateEvent(Event::InboundProbe(event)));
+                    // This connection was dialed solely to probe the peer's reachability;
+                    // the probe is resolved and the response is sent over the connection
+                    // the peer dialed us on. Close the probe connection so it does not
+                    // linger alongside the peer's primary connection and hold on to
+                    // transport resources.
+                    self.pending_actions.push_back(ToSwarm::CloseConnection {
+                        peer_id: peer,
+                        connection: CloseConnection::One(conn),
+                    });
                 }
             }
             ConnectedPoint::Dialer {

protocols/autonat/src/v2/server/behaviour.rs

@@ -8,8 +8,8 @@ use either::Either;
 use libp2p_core::{Endpoint, Multiaddr, transport::PortUse};
 use libp2p_identity::PeerId;
 use libp2p_swarm::{
-    ConnectionDenied, ConnectionHandler, ConnectionId, DialFailure, FromSwarm, NetworkBehaviour,
-    ToSwarm,
+    CloseConnection, ConnectionDenied, ConnectionHandler, ConnectionId, DialFailure, FromSwarm,
+    NetworkBehaviour, ToSwarm,
     dial_opts::{DialOpts, PeerCondition},
     dummy,
 };
@@ -102,13 +102,23 @@ where
     fn on_connection_handler_event(
         &mut self,
         peer_id: PeerId,
-        _connection_id: ConnectionId,
+        connection_id: ConnectionId,
         event: <Handler<R> as ConnectionHandler>::ToBehaviour,
     ) {
         match event {
-            Either::Left(Either::Left(Ok(_))) => {}
-            Either::Left(Either::Left(Err(e))) => {
-                tracing::debug!("dial back error: {e:?}");
+            Either::Left(Either::Left(result)) => {
+                if let Err(e) = result {
+                    tracing::debug!("dial back error: {e:?}");
+                }
+                // This connection was dialed solely to perform the dial-back, which
+                // has now run to completion; the outcome reaches the client over the
+                // connection it dialed us on. Close the dial-back connection so it
+                // does not linger alongside the client's primary connection and hold
+                // on to transport resources.
+                self.pending_events.push_back(ToSwarm::CloseConnection {
+                    peer_id,
+                    connection: CloseConnection::One(connection_id),
+                });
             }
             Either::Left(Either::Right(v)) => libp2p_core::util::unreachable(v),
             Either::Right(Either::Left(cmd)) => {