From ff7166cf05f3653f3cadb9619529263a28187bb2 Mon Sep 17 00:00:00 2001
From: Anton Puhach <anton@near.org>
Date: Tue, 17 Dec 2024 19:09:16 +0100
Subject: [PATCH] refactor: explicitly list all message types in
 tcp::Tier::is_allowed (#12637)

See [this
comment](https://github.com/near/nearcore/pull/12634#pullrequestreview-2508723983).
---
 .../src/peer_manager/connection/mod.rs        | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

diff --git a/chain/network/src/peer_manager/connection/mod.rs b/chain/network/src/peer_manager/connection/mod.rs
index ff7e4162991..8f4adc38e98 100644
--- a/chain/network/src/peer_manager/connection/mod.rs
+++ b/chain/network/src/peer_manager/connection/mod.rs
@@ -43,7 +43,24 @@ impl tcp::Tier {
                 self == tcp::Tier::T2 || self == tcp::Tier::T3
             }
             PeerMessage::Routed(msg) => self.is_allowed_routed(&msg.body),
-            _ => self == tcp::Tier::T2,
+            PeerMessage::SyncRoutingTable(..)
+            | PeerMessage::DistanceVector(..)
+            | PeerMessage::RequestUpdateNonce(..)
+            | PeerMessage::SyncAccountsData(..)
+            | PeerMessage::PeersRequest(..)
+            | PeerMessage::PeersResponse(..)
+            | PeerMessage::BlockHeadersRequest(..)
+            | PeerMessage::BlockHeaders(..)
+            | PeerMessage::BlockRequest(..)
+            | PeerMessage::Block(..)
+            | PeerMessage::Transaction(..)
+            | PeerMessage::Disconnect(..)
+            | PeerMessage::Challenge(..)
+            | PeerMessage::SyncSnapshotHosts(..)
+            | PeerMessage::StateRequestHeader(..)
+            | PeerMessage::StateRequestPart(..)
+            | PeerMessage::EpochSyncRequest
+            | PeerMessage::EpochSyncResponse(..) => self == tcp::Tier::T2,
         }
     }