Update host ENR when receiving Pong with new endpoint in destination

ethereum · May 15, 2019 · 7dc369f · 7dc369f
1 parent 351a38d
commit 7dc369f
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/libp2p/NodeTable.cpp b/libp2p/NodeTable.cpp
@@ -544,6 +544,19 @@ shared_ptr<NodeEntry> NodeTable::handlePong(
             isPublicAddress(pong.destination.address()))
             m_hostNodeEndpoint.setAddress(pong.destination.address());
         m_hostNodeEndpoint.setUdpPort(pong.destination.udpPort());
+
+        if (m_hostENRInfo.ip != m_hostNodeEndpoint.address() ||
+            m_hostENRInfo.udpPort != pong.destination.udpPort())
+        {
+            {
+                Guard l(m_hostENRMutex);
+                m_hostENR =
+                    IdentitySchemeV4::updateENR(m_hostENR, m_secret, m_hostNodeEndpoint.address(),
+                        m_hostNodeEndpoint.tcpPort(), pong.destination.udpPort());
+                m_hostENRInfo = IdentitySchemeV4::info(m_hostENR);
+            }
+            clog(VerbosityInfo, "net") << "ENR updated: " << m_hostENR;
+        }
     }
 
     return sourceNodeEntry;