public ProtocolsManager(
ISyncPeerPool syncPeerPool,
ISyncServer syncServer,
ITxPool txPool,
IDiscoveryApp discoveryApp,
IMessageSerializationService serializationService,
IRlpxPeer localPeer,
INodeStatsManager nodeStatsManager,
IProtocolValidator protocolValidator,
INetworkStorage peerStorage,
ISpecProvider specProvider,
ILogManager logManager)
{
_syncPool = syncPeerPool ?? throw new ArgumentNullException(nameof(syncPeerPool));
_syncServer = syncServer ?? throw new ArgumentNullException(nameof(syncServer));
_txPool = txPool ?? throw new ArgumentNullException(nameof(txPool));
_discoveryApp = discoveryApp ?? throw new ArgumentNullException(nameof(discoveryApp));
_serializer = serializationService ?? throw new ArgumentNullException(nameof(serializationService));
_localPeer = localPeer ?? throw new ArgumentNullException(nameof(localPeer));
_stats = nodeStatsManager ?? throw new ArgumentNullException(nameof(nodeStatsManager));
_protocolValidator = protocolValidator ?? throw new ArgumentNullException(nameof(protocolValidator));
_peerStorage = peerStorage ?? throw new ArgumentNullException(nameof(peerStorage));
_specProvider = specProvider ?? throw new ArgumentNullException(nameof(specProvider));
_logManager = logManager ?? throw new ArgumentNullException(nameof(logManager));
_logger = _logManager?.GetClassLogger() ?? throw new ArgumentNullException(nameof(logManager));
_protocolFactories = GetProtocolFactories();
localPeer.SessionCreated += SessionCreated;
}
public ProtocolsManager(
ISynchronizationManager synchronizationManager,
ITransactionPool transactionPool,
IDiscoveryApp discoveryApp,
IMessageSerializationService serializationService,
IRlpxPeer localPeer,
INodeStatsManager nodeStatsManager,
IProtocolValidator protocolValidator,
INetworkStorage peerStorage,
IPerfService perfService,
ILogManager logManager)
{
_syncManager = synchronizationManager ?? throw new ArgumentNullException(nameof(synchronizationManager));
_transactionPool = transactionPool ?? throw new ArgumentNullException(nameof(transactionPool));
_discoveryApp = discoveryApp ?? throw new ArgumentNullException(nameof(discoveryApp));
_serializer = serializationService ?? throw new ArgumentNullException(nameof(serializationService));
_localPeer = localPeer ?? throw new ArgumentNullException(nameof(localPeer));
_stats = nodeStatsManager ?? throw new ArgumentNullException(nameof(nodeStatsManager));
_protocolValidator = protocolValidator ?? throw new ArgumentNullException(nameof(protocolValidator));
_perfService = perfService ?? throw new ArgumentNullException(nameof(perfService));
_peerStorage = peerStorage ?? throw new ArgumentNullException(nameof(peerStorage));
_logManager = logManager ?? throw new ArgumentNullException(nameof(logManager));
_logger = _logManager.GetClassLogger();
_syncManager.SyncEvent += OnSyncEvent;
localPeer.SessionCreated += SessionCreated;
}
private static void Start()
{
var privateKeyProvider = new PrivateKeyProvider(PrivateKey);
var config = new DiscoveryConfigurationProvider(new NetworkHelper(Logger));
var signer = new Signer();
var cryptoRandom = new CryptoRandom();
var configProvider = new ConfigurationProvider();
var nodeFactory = new NodeFactory();
var calculator = new NodeDistanceCalculator(config);
var nodeTable = new NodeTable(config, nodeFactory, new FileKeyStore(configProvider, new JsonSerializer(Logger), new AesEncrypter(configProvider, Logger), cryptoRandom, Logger), Logger, calculator);
var evictionManager = new EvictionManager(nodeTable, Logger);
var lifecycleFactory = new NodeLifecycleManagerFactory(nodeFactory, nodeTable, Logger, config, new DiscoveryMessageFactory(config), evictionManager);
var discoveryManager = new DiscoveryManager(Logger, config, lifecycleFactory, nodeFactory, nodeTable);
var nodesLocator = new NodesLocator(nodeTable, discoveryManager, config, Logger);
var discoveryMesageFactory = new DiscoveryMessageFactory(config);
var nodeIdResolver = new NodeIdResolver(signer);
var pingSerializer = new PingMessageSerializer(signer, privateKeyProvider, discoveryMesageFactory, nodeIdResolver, nodeFactory);
var pongSerializer = new PongMessageSerializer(signer, privateKeyProvider, discoveryMesageFactory, nodeIdResolver, nodeFactory);
var findNodeSerializer = new FindNodeMessageSerializer(signer, privateKeyProvider, discoveryMesageFactory, nodeIdResolver, nodeFactory);
var neighborsSerializer = new NeighborsMessageSerializer(signer, privateKeyProvider, discoveryMesageFactory, nodeIdResolver, nodeFactory);
var messageSerializationService = new MessageSerializationService();
messageSerializationService.Register(pingSerializer);
messageSerializationService.Register(pongSerializer);
messageSerializationService.Register(findNodeSerializer);
messageSerializationService.Register(neighborsSerializer);
//P2P initialization
IMessagePad eip8Pad = new Eip8MessagePad(cryptoRandom);
messageSerializationService.Register(new AuthEip8MessageSerializer(eip8Pad));
messageSerializationService.Register(new AckEip8MessageSerializer(eip8Pad));
messageSerializationService.Register(new HelloMessageSerializer());
messageSerializationService.Register(new DisconnectMessageSerializer());
messageSerializationService.Register(new Nethermind.Network.P2P.PingMessageSerializer());
messageSerializationService.Register(new Nethermind.Network.P2P.PongMessageSerializer());
messageSerializationService.Register(new StatusMessageSerializer());
IEciesCipher eciesCipher = new EciesCipher(cryptoRandom);
IEncryptionHandshakeService encryptionHandshakeService = new EncryptionHandshakeService(messageSerializationService, eciesCipher, cryptoRandom, signer, PrivateKey, Logger);
var p2pManager = new P2PManager(encryptionHandshakeService, Logger, messageSerializationService, Substitute.For <ISynchronizationManager>());
//Connect discovery with P2P
discoveryManager.RegisterDiscoveryListener(p2pManager);
_discoveryApp = new DiscoveryApp(config, nodesLocator, Logger, discoveryManager, nodeFactory, nodeTable, messageSerializationService, cryptoRandom);
_discoveryApp.Start(PrivateKey.PublicKey);
}
public void SetUp()
{
_rlpxPeer = new RlpxMock(_sessions);
_discoveryApp = Substitute.For <IDiscoveryApp>();
_stats = new NodeStatsManager(new StatsConfig(), LimboLogs.Instance);
_storage = new InMemoryStorage();
_peerLoader = new PeerLoader(new NetworkConfig(), _stats, _storage, LimboLogs.Instance);
_networkConfig = new NetworkConfig();
_networkConfig.PeersPersistenceInterval = 50;
_peerManager = new PeerManager(_rlpxPeer, _discoveryApp, _stats, _storage, _peerLoader, _networkConfig, LimboLogs.Instance);
}
public Context()
{
_channel = Substitute.For <IChannel>();
_channelHandlerContext = Substitute.For <IChannelHandlerContext>();
_pipeline = Substitute.For <IChannelPipeline>();
_channelHandlerContext.Channel.Returns(_channel);
_channel.Pipeline.Returns(_pipeline);
_pipeline.Get <ZeroPacketSplitter>().Returns(new ZeroPacketSplitter(LimboLogs.Instance));
_packetSender = Substitute.For <IPacketSender>();
_syncServer = Substitute.For <ISyncServer>();
_syncServer = Substitute.For <ISyncServer>();
_syncServer.Genesis.Returns(Build.A.Block.Genesis.TestObject.Header);
_syncServer.Head.Returns(Build.A.BlockHeader.TestObject);
_txPool = Substitute.For <ITxPool>();
_pooledTxsRequestor = Substitute.For <IPooledTxsRequestor>();
_discoveryApp = Substitute.For <IDiscoveryApp>();
_serializer = new MessageSerializationService();
_rlpxHost = Substitute.For <IRlpxHost>();
_rlpxHost.LocalPort.Returns(_localPort);
_rlpxHost.LocalNodeId.Returns(TestItem.PublicKeyA);
ITimerFactory timerFactory = Substitute.For <ITimerFactory>();
_nodeStatsManager = new NodeStatsManager(timerFactory, LimboLogs.Instance);
_blockTree = Substitute.For <IBlockTree>();
_blockTree.ChainId.Returns(1ul);
_blockTree.Genesis.Returns(Build.A.Block.Genesis.TestObject.Header);
_protocolValidator = new ProtocolValidator(_nodeStatsManager, _blockTree, LimboLogs.Instance);
_peerStorage = Substitute.For <INetworkStorage>();
_syncPeerPool = Substitute.For <ISyncPeerPool>();
_gossipPolicy = Substitute.For <IGossipPolicy>();
_manager = new ProtocolsManager(
_syncPeerPool,
_syncServer,
_txPool,
_pooledTxsRequestor,
_discoveryApp,
_serializer,
_rlpxHost,
_nodeStatsManager,
_protocolValidator,
_peerStorage,
MainnetSpecProvider.Instance,
_gossipPolicy,
LimboLogs.Instance);
_serializer.Register(new HelloMessageSerializer());
_serializer.Register(new StatusMessageSerializer());
_serializer.Register(new DisconnectMessageSerializer());
}
public Context()
{
_channel = Substitute.For <IChannel>();
_channelHandlerContext = Substitute.For <IChannelHandlerContext>();
_pipeline = Substitute.For <IChannelPipeline>();
_channelHandlerContext.Channel.Returns(_channel);
_channel.Pipeline.Returns(_pipeline);
_pipeline.Get <ZeroPacketSplitter>().Returns(new ZeroPacketSplitter(LimboLogs.Instance));
_packetSender = Substitute.For <IPacketSender>();
_syncServer = Substitute.For <ISyncServer>();
_syncServer = Substitute.For <ISyncServer>();
_syncServer.Genesis.Returns(Build.A.Block.Genesis.TestObject.Header);
_syncServer.Head.Returns(Build.A.BlockHeader.TestObject);
_txPool = Substitute.For <ITxPool>();
_discoveryApp = Substitute.For <IDiscoveryApp>();
_serializer = new MessageSerializationService();
_localPeer = Substitute.For <IRlpxPeer>();
_localPeer.LocalPort.Returns(_localPort);
_localPeer.LocalNodeId.Returns(TestItem.PublicKeyA);
_nodeStatsManager = new NodeStatsManager(new StatsConfig(), LimboLogs.Instance);
_blockTree = Substitute.For <IBlockTree>();
_blockTree.ChainId.Returns(1);
_blockTree.Genesis.Returns(Build.A.Block.Genesis.TestObject.Header);
_protocolValidator = new ProtocolValidator(_nodeStatsManager, _blockTree, LimboLogs.Instance);
_peerStorage = Substitute.For <INetworkStorage>();
_perfService = new PerfService(LimboLogs.Instance);
_syncPeerPool = Substitute.For <IEthSyncPeerPool>();
_manager = new ProtocolsManager(
_syncPeerPool,
_syncServer,
_txPool,
_discoveryApp,
_serializer,
_localPeer,
_nodeStatsManager,
_protocolValidator,
_peerStorage,
_perfService,
LimboLogs.Instance);
_serializer.Register(new HelloMessageSerializer());
_serializer.Register(new StatusMessageSerializer());
_serializer.Register(new DisconnectMessageSerializer());
}
public PeerManager(
IRlpxPeer rlpxPeer,
IDiscoveryApp discoveryApp,
INodeStatsManager stats,
INetworkStorage peerStorage,
IPeerLoader peerLoader,
INetworkConfig networkConfig,
ILogManager logManager)
{
_logger = logManager.GetClassLogger();
_rlpxPeer = rlpxPeer ?? throw new ArgumentNullException(nameof(rlpxPeer));
_stats = stats ?? throw new ArgumentNullException(nameof(stats));
_discoveryApp = discoveryApp ?? throw new ArgumentNullException(nameof(discoveryApp));
_networkConfig = networkConfig ?? throw new ArgumentNullException(nameof(networkConfig));
_peerStorage = peerStorage ?? throw new ArgumentNullException(nameof(peerStorage));
_peerLoader = peerLoader ?? throw new ArgumentNullException(nameof(peerLoader));
_peerStorage.StartBatch();
_peerComparer = new PeerComparer(_stats);
}
private Task InitDiscovery(InitParams initParams)
{
_logger.Info("Initializing Discovery");
if (initParams.DiscoveryPort.HasValue)
{
_configProvider.GetConfig <NetworkConfig>().MasterPort = initParams.DiscoveryPort.Value;
}
var privateKeyProvider = new PrivateKeyProvider(_privateKey);
var discoveryMessageFactory = new DiscoveryMessageFactory(_configProvider);
var nodeIdResolver = new NodeIdResolver(_signer);
var msgSerializersProvider = new DiscoveryMsgSerializersProvider(_messageSerializationService, _signer, privateKeyProvider, discoveryMessageFactory, nodeIdResolver, _nodeFactory);
msgSerializersProvider.RegisterDiscoverySerializers();
var nodeDistanceCalculator = new NodeDistanceCalculator(_configProvider);
var nodeTable = new NodeTable(_nodeFactory, _keyStore, nodeDistanceCalculator, _configProvider, _logManager);
var evictionManager = new EvictionManager(nodeTable, _logManager);
var nodeLifeCycleFactory = new NodeLifecycleManagerFactory(_nodeFactory, nodeTable, discoveryMessageFactory, evictionManager, _nodeStatsProvider, _configProvider, _logManager);
var discoveryStorage = new DiscoveryStorage(_configProvider, _nodeFactory, _logManager, _perfService);
_discoveryManager = new DiscoveryManager(nodeLifeCycleFactory, _nodeFactory, nodeTable, discoveryStorage, _configProvider, _logManager);
var nodesLocator = new NodesLocator(nodeTable, _discoveryManager, _configProvider, _logManager);
_discoveryApp = new DiscoveryApp(nodesLocator, _discoveryManager, _nodeFactory, nodeTable, _messageSerializationService, _cryptoRandom, discoveryStorage, _configProvider, _logManager);
_discoveryApp.Start(_privateKey.PublicKey);
_logger.Info("Discovery initialization completed");
return(Task.CompletedTask);
}
private void InitDiscovery()
{
if (!_initConfig.DiscoveryEnabled)
{
_discoveryApp = new NullDiscoveryApp();
return;
}
INetworkConfig networkConfig = _configProvider.GetConfig <INetworkConfig>();
networkConfig.MasterPort = _initConfig.DiscoveryPort;
var privateKeyProvider = new SameKeyGenerator(_nodeKey);
var discoveryMessageFactory = new DiscoveryMessageFactory(networkConfig, _timestamp);
var nodeIdResolver = new NodeIdResolver(_ecdsa);
IDiscoveryMsgSerializersProvider msgSerializersProvider = new DiscoveryMsgSerializersProvider(
_messageSerializationService,
_ecdsa,
privateKeyProvider,
discoveryMessageFactory,
nodeIdResolver);
msgSerializersProvider.RegisterDiscoverySerializers();
var nodeDistanceCalculator = new NodeDistanceCalculator(networkConfig);
var nodeTable = new NodeTable(
_keyStore,
nodeDistanceCalculator,
networkConfig,
_logManager);
var evictionManager = new EvictionManager(
nodeTable,
_logManager);
var nodeLifeCycleFactory = new NodeLifecycleManagerFactory(
nodeTable,
discoveryMessageFactory,
evictionManager,
_nodeStatsManager,
networkConfig,
_logManager);
var discoveryStorage = new NetworkStorage(
DiscoveryNodesDbPath,
networkConfig,
_logManager,
_perfService);
var discoveryManager = new DiscoveryManager(
nodeLifeCycleFactory,
nodeTable,
discoveryStorage,
networkConfig,
_logManager);
var nodesLocator = new NodesLocator(
nodeTable,
discoveryManager,
_configProvider,
_logManager);
_discoveryApp = new DiscoveryApp(
nodesLocator,
discoveryManager,
nodeTable,
_messageSerializationService,
_cryptoRandom,
discoveryStorage,
networkConfig,
_logManager, _perfService);
_discoveryApp.Initialize(_nodeKey.PublicKey);
}
private void InitDiscovery()
{
_configProvider.GetConfig <INetworkConfig>().MasterPort = _initConfig.DiscoveryPort;
var privateKeyProvider = new PrivateKeyProvider(_privateKey);
var discoveryMessageFactory = new DiscoveryMessageFactory(_configProvider);
var nodeIdResolver = new NodeIdResolver(_signer);
IDiscoveryMsgSerializersProvider msgSerializersProvider = new DiscoveryMsgSerializersProvider(
_messageSerializationService,
_signer,
privateKeyProvider,
discoveryMessageFactory,
nodeIdResolver,
_nodeFactory);
msgSerializersProvider.RegisterDiscoverySerializers();
var nodeDistanceCalculator = new NodeDistanceCalculator(_configProvider);
var nodeTable = new NodeTable(
_nodeFactory,
_keyStore,
nodeDistanceCalculator,
_configProvider,
_logManager);
var evictionManager = new EvictionManager(
nodeTable,
_logManager);
var nodeLifeCycleFactory = new NodeLifecycleManagerFactory(
_nodeFactory,
nodeTable,
discoveryMessageFactory,
evictionManager,
_nodeStatsProvider,
_configProvider,
_logManager);
var discoveryStorage = new NetworkStorage(
DiscoveryNodesDbPath,
_configProvider.GetConfig <INetworkConfig>(),
_logManager,
_perfService);
_discoveryManager = new DiscoveryManager(
nodeLifeCycleFactory,
_nodeFactory,
nodeTable,
discoveryStorage,
_configProvider,
_logManager);
var nodesLocator = new NodesLocator(
nodeTable,
_discoveryManager,
_configProvider,
_logManager);
_discoveryApp = new DiscoveryApp(
nodesLocator,
_discoveryManager,
_nodeFactory,
nodeTable,
_messageSerializationService,
_cryptoRandom,
discoveryStorage,
_configProvider,
_logManager, _perfService);
_discoveryApp.Initialize(_privateKey.PublicKey);
}
private void InitDiscovery()
{
if (!_initConfig.DiscoveryEnabled)
{
_discoveryApp = new NullDiscoveryApp();
return;
}
IDiscoveryConfig discoveryConfig = _configProvider.GetConfig <IDiscoveryConfig>();
var privateKeyProvider = new SameKeyGenerator(_nodeKey);
var discoveryMessageFactory = new DiscoveryMessageFactory(_timestamper);
var nodeIdResolver = new NodeIdResolver(_ethereumEcdsa);
IDiscoveryMsgSerializersProvider msgSerializersProvider = new DiscoveryMsgSerializersProvider(
_messageSerializationService,
_ethereumEcdsa,
privateKeyProvider,
discoveryMessageFactory,
nodeIdResolver);
msgSerializersProvider.RegisterDiscoverySerializers();
var nodeDistanceCalculator = new NodeDistanceCalculator(discoveryConfig);
var nodeTable = new NodeTable(nodeDistanceCalculator, discoveryConfig, _networkConfig, _logManager);
var evictionManager = new EvictionManager(nodeTable, _logManager);
var nodeLifeCycleFactory = new NodeLifecycleManagerFactory(
nodeTable,
discoveryMessageFactory,
evictionManager,
_nodeStatsManager,
discoveryConfig,
_logManager);
var discoveryDb = new SimpleFilePublicKeyDb("DiscoveryDB", Path.Combine(_initConfig.BaseDbPath, DiscoveryNodesDbPath), _logManager);
var discoveryStorage = new NetworkStorage(
discoveryDb,
_logManager);
var discoveryManager = new DiscoveryManager(
nodeLifeCycleFactory,
nodeTable,
discoveryStorage,
discoveryConfig,
_logManager);
var nodesLocator = new NodesLocator(
nodeTable,
discoveryManager,
discoveryConfig,
_logManager);
_discoveryApp = new DiscoveryApp(
nodesLocator,
discoveryManager,
nodeTable,
_messageSerializationService,
_cryptoRandom,
discoveryStorage,
_networkConfig,
discoveryConfig,
_timestamper,
_logManager, _perfService);
_discoveryApp.Initialize(_nodeKey.PublicKey);
}
