Discard notifications if we have failed to parse handshake (#8806)

paritytech · Jun 2, 2021 · 437c838 · 437c838
1 parent 538b15f
commit 437c838
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/client/network/src/protocol.rs b/client/network/src/protocol.rs
@@ -1517,6 +1517,9 @@ impl<B: BlockT> NetworkBehaviour for Protocol<B> {
 						);
 						CustomMessageOutcome::None
 					}
+					_ if self.bad_handshake_substreams.contains(&(peer_id.clone(), set_id)) => {
+						CustomMessageOutcome::None
+					}
 					_ => {
 						let protocol_name = self.notification_protocols[usize::from(set_id) - NUM_HARDCODED_PEERSETS].clone();
 						CustomMessageOutcome::NotificationsReceived {