fix 🐛: Added case for peer disconnection, additional testing needed

ISSUE: #60
AnduroProject · Jan 14, 2025 · 1245b7a · 1245b7a
1 parent 954f5ed
commit 1245b7a
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/app/src/chain.rs b/app/src/chain.rs
@@ -1183,6 +1183,9 @@ impl<DB: ItemStore<MainnetEthSpec>> Chain<DB> {
                         let payload_prev_hash = x.message.execution_payload.parent_hash;
 
                         info!("Received payload at height {number} {payload_prev_hash} -> {payload_hash}");
+                        let head_hash = self.head.read().await.as_ref().unwrap().hash;
+                        let head_height = self.head.read().await.as_ref().unwrap().height;
+                        debug!("Local head: {:#?}, height: {}", head_hash, head_height);
 
                         // sync first then process block so we don't skip and trigger a re-sync
                         if matches!(self.get_parent(&x), Err(Error::MissingParent)) {
@@ -1292,7 +1295,7 @@ impl<DB: ItemStore<MainnetEthSpec>> Chain<DB> {
                 peer_id,
                 OutboundRequest::BlocksByRange(
                     crate::network::rpc::methods::BlocksByRangeRequest {
-                        start_height: head + 1,
+                        start_height: head,
                         count: 1024,
                     },
                 ),

diff --git a/app/src/network/mod.rs b/app/src/network/mod.rs
@@ -362,6 +362,11 @@ impl NetworkBackend {
                         peers.insert(peer_id);
                         let _ = peers_connected_tx.send(peers.clone());
                     }
+                    SwarmEvent::ConnectionClosed { peer_id, connection_id, endpoint, num_established, cause } => {
+                        debug!("Connection closed: peer_id: {peer_id}, connection_id: {connection_id}, endpoint: {endpoint:?}, num_established: {num_established}, cause: {cause:?}");
+                        peers.remove(&peer_id);
+                        let _ = peers_connected_tx.send(peers.clone());
+                    }
                     x => {
                         trace!("Unhandled message {x:?}");
                     }