From cb7d5eba1c3457409601329ed4889f8c6b1531f5 Mon Sep 17 00:00:00 2001
From: Age Manning <Age@AgeManning.com>
Date: Thu, 18 Apr 2019 15:26:30 +1000
Subject: [PATCH] Discovery and gossip bug fixes

---
 beacon_node/Cargo.toml                   |  2 +-
 beacon_node/eth2-libp2p/src/discovery.rs | 20 ++++----------------
 2 files changed, 5 insertions(+), 17 deletions(-)

diff --git a/beacon_node/Cargo.toml b/beacon_node/Cargo.toml
index 43e75d0a6a..783cdcda3f 100644
--- a/beacon_node/Cargo.toml
+++ b/beacon_node/Cargo.toml
@@ -22,5 +22,5 @@ tokio-timer = "0.2.10"
 futures = "0.1.25"
 exit-future = "0.1.3"
 state_processing = { path = "../eth2/state_processing" }
-slog = { version = "^2.2.3" , features = ["max_level_trace", "release_max_level_debug"] }
+slog = "^2.2.3" 
 env_logger = "0.6.1"
diff --git a/beacon_node/eth2-libp2p/src/discovery.rs b/beacon_node/eth2-libp2p/src/discovery.rs
index 232590c05a..d6fd43ef43 100644
--- a/beacon_node/eth2-libp2p/src/discovery.rs
+++ b/beacon_node/eth2-libp2p/src/discovery.rs
@@ -83,18 +83,9 @@ where
         NetworkBehaviour::new_handler(&mut self.discovery)
     }
 
-    // TODO: we store all peers in known_peers, when upgrading to discv5 we will avoid duplication
-    // of peer storage.
     fn addresses_of_peer(&mut self, peer_id: &PeerId) -> Vec<Multiaddr> {
-        if let Some(addresses) = self.known_peers.get(peer_id) {
-            addresses.clone()
-        } else {
-            debug!(
-                self.log,
-                "Tried to dial: {:?} but no address stored", peer_id
-            );
-            Vec::new()
-        }
+        // Let discovery track possible known peers.
+        self.discovery.addresses_of_peer(peer_id)
     }
 
     fn inject_connected(&mut self, peer_id: PeerId, endpoint: ConnectedPoint) {
@@ -151,11 +142,6 @@ where
                             peer_id, addresses, ..
                         } => {
                             debug!(self.log, "Kademlia peer discovered"; "Peer"=> format!("{:?}", peer_id), "Addresses" => format!("{:?}", addresses));
-                            (*self
-                                .known_peers
-                                .entry(peer_id.clone())
-                                .or_insert_with(|| vec![]))
-                            .extend(addresses.clone());
                         }
                         KademliaOut::FindNodeResult { closer_peers, .. } => {
                             debug!(
@@ -163,6 +149,8 @@ where
                                 "Kademlia query found {} peers",
                                 closer_peers.len()
                             );
+                            debug!(self.log, "Kademlia peers discovered"; "Peer"=> format!("{:?}", closer_peers));
+
                             if closer_peers.is_empty() {
                                 debug!(self.log, "Kademlia random query yielded empty results");
                             }