From b2a297135ecd40d715dec615c757bee19f752c4b Mon Sep 17 00:00:00 2001
From: Lionel Faber <lionel1704@gmail.com>
Date: Tue, 13 Apr 2021 15:01:30 +0530
Subject: [PATCH] fix(igd): run igd even if echo service succeedes

- also addresses some review comments
---
 src/endpoint.rs     | 22 ++++++++++++++--------
 src/tests/common.rs |  5 ++---
 2 files changed, 16 insertions(+), 11 deletions(-)

diff --git a/src/endpoint.rs b/src/endpoint.rs
index bffd74e6..a1a0129b 100644
--- a/src/endpoint.rs
+++ b/src/endpoint.rs
@@ -248,7 +248,7 @@ impl Endpoint {
         }
 
         #[cfg(not(feature = "no-igd"))]
-        if addr.is_none() && self.qp2p_config.forward_port {
+        if self.qp2p_config.forward_port {
             // Attempt to use IGD for port forwarding
             match timeout(
                 Duration::from_secs(PORT_FORWARD_TIMEOUT),
@@ -268,12 +268,10 @@ impl Endpoint {
                     }
                     Err(e) => {
                         info!("IGD request failed: {} - {:?}", e, e);
-                        return Err(Error::IgdNotSupported);
                     }
                 },
                 Err(e) => {
                     info!("IGD request timeout: {:?}", e);
-                    return Err(Error::IgdNotSupported);
                 }
             }
         }
@@ -380,19 +378,27 @@ impl Endpoint {
         Ok(())
     }
 
-    /// Verify if an address is publicly rechable. This will attempt to create
+    /// Verify if an address is publicly reachable. This will attempt to create
     /// a new connection and use it to exchange a message and verify that the node
-    /// can be reached
+    /// can be reached.
     pub async fn is_reachable(&self, peer_addr: &SocketAddr) -> Result<()> {
         let new_connection = self.create_new_connection(peer_addr).await?;
         let (mut send_stream, mut recv_stream) = new_connection.connection.open_bi().await?;
         let message = WireMsg::EndpointEchoReq;
         message.write_to_stream(&mut send_stream).await?;
-        
-        match timeout(Duration::from_secs(ECHO_SERVICE_QUERY_TIMEOUT), WireMsg::read_from_stream(&mut recv_stream)).await {
+
+        match timeout(
+            Duration::from_secs(ECHO_SERVICE_QUERY_TIMEOUT),
+            WireMsg::read_from_stream(&mut recv_stream),
+        )
+        .await
+        {
             Ok(Ok(WireMsg::EndpointEchoResp(_))) => Ok(()),
             Ok(Ok(other)) => {
-                info!("Unexpected message type when verifying reachability: {}", &other);
+                info!(
+                    "Unexpected message type when verifying reachability: {}",
+                    &other
+                );
                 Ok(())
             }
             Ok(Err(err)) => {
diff --git a/src/tests/common.rs b/src/tests/common.rs
index 6c81a29d..600cb94f 100644
--- a/src/tests/common.rs
+++ b/src/tests/common.rs
@@ -457,6 +457,5 @@ async fn reachability() -> Result<()> {
     };
     let reachable_addr = ep2.socket_addr();
     ep1.is_reachable(&reachable_addr).await?;
-    Ok(())    
-
-}
\ No newline at end of file
+    Ok(())
+}