From 152587bcb9a9e5961f4f08bf7ea5736ce410b5d0 Mon Sep 17 00:00:00 2001
From: Pierre Krieger <pierre.krieger1708@gmail.com>
Date: Wed, 2 Jun 2021 20:41:21 +0200
Subject: [PATCH] Discard notifications if we have failed to parse handshake
 (#8806)

---
 substrate/client/network/src/protocol.rs | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/substrate/client/network/src/protocol.rs b/substrate/client/network/src/protocol.rs
index 6dafd8b85f..6431250c96 100644
--- a/substrate/client/network/src/protocol.rs
+++ b/substrate/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 {