From 1ed9e7a90ace4f0c58fb2c34cb66c57d87fcd14c Mon Sep 17 00:00:00 2001 From: ak88 Date: Thu, 27 Nov 2025 14:58:07 +0100 Subject: [PATCH 1/2] make IProtocolsManager customizable --- .../Steps/InitializeNetwork.cs | 66 ++++++++++--------- 1 file changed, 36 insertions(+), 30 deletions(-) diff --git a/src/Nethermind/Nethermind.Init/Steps/InitializeNetwork.cs b/src/Nethermind/Nethermind.Init/Steps/InitializeNetwork.cs index 32f65f11c8f..95d7977b268 100644 --- a/src/Nethermind/Nethermind.Init/Steps/InitializeNetwork.cs +++ b/src/Nethermind/Nethermind.Init/Steps/InitializeNetwork.cs @@ -264,35 +264,8 @@ private async Task InitPeer() await _api.StaticNodesManager.InitAsync(); await _api.TrustedNodesManager.InitAsync(); - - ISyncServer syncServer = _api.SyncServer!; - - ProtocolValidator protocolValidator = new( - _nodeStatsManager!, - _api.BlockTree, - _forkInfo, - _api.PeerManager!, - _networkConfig, - _api.LogManager); - - _api.ProtocolsManager = new ProtocolsManager( - _api.SyncPeerPool!, - syncServer, - _api.BackgroundTaskScheduler, - _api.TxPool, - _discoveryApp, - _api.MessageSerializationService, - _api.RlpxPeer, - _nodeStatsManager, - protocolValidator, - _peerStorage, - _forkInfo, - _api.GossipPolicy, - _api.WorldStateManager!, - _api.LogManager, - _api.Config(), - _api.SpecProvider, - _api.TxGossipPolicy); + + CreateProtocolManager(); if (_syncConfig.SnapServingEnabled == true) { @@ -303,7 +276,6 @@ private async Task InitPeer() _api.ProtocolsManager!.RemoveSupportedCapability(new Capability(Protocol.NodeData, 1)); } - _api.ProtocolValidator = protocolValidator; if (!_networkConfig.DisableDiscV4DnsFeeder) { @@ -316,4 +288,38 @@ private async Task InitPeer() await plugin.InitNetworkProtocol(); } } + + protected virtual IProtocolsManager CreateProtocolManager() + { + ISyncServer syncServer = _api.SyncServer!; + + ProtocolValidator protocolValidator = new( + _nodeStatsManager!, + _api.BlockTree, + _forkInfo, + _api.PeerManager!, + _networkConfig, + _api.LogManager); + + _api.ProtocolValidator = protocolValidator; + + return new ProtocolsManager( + _api.SyncPeerPool!, + syncServer, + _api.BackgroundTaskScheduler, + _api.TxPool!, + _discoveryApp, + _api.MessageSerializationService, + _api.RlpxPeer, + _nodeStatsManager, + protocolValidator, + _peerStorage, + _forkInfo, + _api.GossipPolicy, + _api.WorldStateManager!, + _api.LogManager, + _api.Config(), + _api.SpecProvider!, + _api.TxGossipPolicy); + } } From 73d7daace1c9fe5f2e7283fb2fd8d4718e680f77 Mon Sep 17 00:00:00 2001 From: ak88 Date: Thu, 27 Nov 2025 14:59:05 +0100 Subject: [PATCH 2/2] format --- src/Nethermind/Nethermind.Init/Steps/InitializeNetwork.cs | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/Nethermind/Nethermind.Init/Steps/InitializeNetwork.cs b/src/Nethermind/Nethermind.Init/Steps/InitializeNetwork.cs index 95d7977b268..a37cc947d71 100644 --- a/src/Nethermind/Nethermind.Init/Steps/InitializeNetwork.cs +++ b/src/Nethermind/Nethermind.Init/Steps/InitializeNetwork.cs @@ -264,7 +264,7 @@ private async Task InitPeer() await _api.StaticNodesManager.InitAsync(); await _api.TrustedNodesManager.InitAsync(); - + CreateProtocolManager(); if (_syncConfig.SnapServingEnabled == true)