feat: ban a peer if its last state isn't changed after timeout

nervosnetwork · Dec 29, 2023 · b553660 · b553660
1 parent 74cbc3c
commit b553660
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/protocols/light_client/peers.rs b/src/protocols/light_client/peers.rs
@@ -1856,6 +1856,15 @@ impl Peers {
                             None
                         }
                     })
+                    .or_else(|| {
+                        peer.state.get_last_state().and_then(|state| {
+                            if now > state.update_ts + MESSAGE_TIMEOUT {
+                                Some(*peer_index)
+                            } else {
+                                None
+                            }
+                        })
+                    })
                     .or_else(|| {
                         peer.get_blocks_proof_request().and_then(|req| {
                             if now > req.when_sent + MESSAGE_TIMEOUT {