refactor: explicitly list all message types in tcp::Tier::is_allowed (#…

…12637) See [this comment](#12634 (review)).
near · Dec 17, 2024 · ff7166c · ff7166c
1 parent f867ea7
commit ff7166c
Showing 1 changed file with 18 additions and 1 deletion.
diff --git 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,
         }
     }