From 15705b3fa17fdf7c35fa549e212c96bf50d7ef75 Mon Sep 17 00:00:00 2001
From: Ignotus Peverell <igno.peverell@protonmail.com>
Date: Fri, 3 Nov 2017 16:47:35 -0400
Subject: [PATCH] Detect unspecified IP and use TCP conn IP instead

---
 p2p/src/handshake.rs | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/p2p/src/handshake.rs b/p2p/src/handshake.rs
index 1f355b231..836f0e9d4 100644
--- a/p2p/src/handshake.rs
+++ b/p2p/src/handshake.rs
@@ -172,14 +172,16 @@ impl Handshake {
 fn extract_ip(advertised: &SocketAddr, conn: &TcpStream) -> SocketAddr {
   match advertised {
     &SocketAddr::V4(v4sock) => {
-      if v4sock.ip().is_loopback() {
+      let ip = v4sock.ip();
+      if ip.is_loopback() || ip.is_unspecified() {
         if let Ok(addr) =  conn.peer_addr() {
           return SocketAddr::new(addr.ip(), advertised.port());
         }
       }
     }
     &SocketAddr::V6(v6sock) => {
-      if v6sock.ip().is_loopback() {
+      let ip = v6sock.ip();
+      if ip.is_loopback() || ip.is_unspecified() {
         if let Ok(addr) =  conn.peer_addr() {
           return SocketAddr::new(addr.ip(), advertised.port());
         }