public PeerInfo?Allocate(PeerInfo?currentPeer, IEnumerable <PeerInfo> peers, INodeStatsManager nodeStatsManager, IBlockTree blockTree)
{
UInt256?currentDiffOrNull = blockTree.BestSuggestedHeader?.TotalDifficulty;
if (currentDiffOrNull == null)
{
return(_strategy.Allocate(currentPeer, peers, nodeStatsManager, blockTree));
}
UInt256 currentDiff = currentDiffOrNull.Value;
switch (_selectionType)
{
case TotalDiffSelectionType.Better:
currentDiff += UInt256.One;
break;
case TotalDiffSelectionType.AtLeastTheSame:
break;
case TotalDiffSelectionType.CanBeSlightlyWorse:
UInt256 lastBlockDiff = blockTree.BestSuggestedHeader?.Difficulty ?? 0;
if (currentDiff >= lastBlockDiff)
{
currentDiff -= lastBlockDiff;
}
break;
default:
throw new ArgumentOutOfRangeException();
}
return(_strategy.Allocate(currentPeer, peers.Where(p => p.TotalDifficulty >= currentDiff), nodeStatsManager, blockTree));
}
public static Block FindParent(this IBlockTree tree, BlockHeader blockHeader, BlockTreeLookupOptions options)
{
return(tree.FindBlock(blockHeader.ParentHash, options));
}
public static Block RetrieveGenesisBlock(this IBlockTree tree)
{
return(tree.FindBlock(tree.Genesis.Hash, BlockTreeLookupOptions.RequireCanonical));
}
private void PrepareHeadersResponse(HeadersSyncBatch headersSyncBatch, LatencySyncPeerMock syncPeer, IBlockTree tree)
{
if (headersSyncBatch != null)
{
long startNumber = headersSyncBatch.StartNumber;
if (_maliciousByShiftedOneBack.Contains(syncPeer))
{
startNumber++;
TestContext.WriteLine($"{_time,6} | SYNC PEER {syncPeer.Node:s} WILL SEND SHIFTED MESSAGES ({startNumber} INSTEAD OF {headersSyncBatch.StartNumber})");
}
else if (_maliciousByShiftedOneForward.Contains(syncPeer))
{
startNumber = Math.Max(0, startNumber - 1);
TestContext.WriteLine($"{_time,6} | SYNC PEER {syncPeer.Node:s} WILL SEND SHIFTED MESSAGES ({startNumber} INSTEAD OF {headersSyncBatch.StartNumber})");
}
Keccak hash = tree.FindHash(startNumber);
if (hash == null)
{
TestContext.WriteLine($"{_time,6} | SYNC PEER {syncPeer.Node:s} CANNOT FIND {headersSyncBatch.StartNumber}");
return;
}
int requestSize = headersSyncBatch.RequestSize;
if (_incorrectByTooLongMessages.Contains(syncPeer))
{
requestSize *= 2;
TestContext.WriteLine($"{_time,6} | SYNC PEER {syncPeer.Node:s} WILL SEND TOO LONG MESSAGE ({requestSize} INSTEAD OF {headersSyncBatch.RequestSize})");
}
else if (_incorrectByTooShortMessages.Contains(syncPeer))
{
requestSize = Math.Max(1, requestSize / 2);
TestContext.WriteLine($"{_time,6} | SYNC PEER {syncPeer.Node:s} WILL SEND TOO SHORT MESSAGE ({requestSize} INSTEAD OF {headersSyncBatch.RequestSize})");
}
BlockHeader[] headers = tree.FindHeaders(hash, requestSize, 0, false);
if (_invalidBlocks.ContainsKey(syncPeer))
{
for (int i = 0; i < headers.Length; i++)
{
if (_invalidBlocks[syncPeer].Contains(headers[i].Number))
{
TestContext.WriteLine($"{_time,6} | SYNC PEER {syncPeer.Node:s} WILL SEND AN INVALID BLOCK AT {headers[i].Number}");
headers[i] = Build.A.Block.WithDifficulty(1).TestObject.Header;
}
}
}
if (headers.Length > 3 && _maliciousByRepetition.Contains(syncPeer))
{
headers[headers.Length - 1] = headers[headers.Length - 3];
headers[headers.Length - 2] = headers[headers.Length - 3];
TestContext.WriteLine($"{_time,6} | SYNC PEER {syncPeer.Node:s} WILL SEND A MALICIOUS (REPEATED) MESSAGE");
}
if (_maliciousByShortAtStart.Contains(syncPeer))
{
headers[0] = null;
TestContext.WriteLine($"{_time,6} | SYNC PEER {syncPeer.Node:s} WILL SEND A MALICIOUS (SHORT AT START) MESSAGE");
}
headersSyncBatch.Response = headers;
if (_peerMaxResponseSizes.ContainsKey(syncPeer))
{
int maxResponseSize = _peerMaxResponseSizes[syncPeer];
TestContext.WriteLine($"{_time,6} | SYNC PEER {syncPeer.Node:s} WILL SEND NULLS AFTER INDEX {maxResponseSize}");
for (int i = 0; i < headers.Length; i++)
{
if (i >= maxResponseSize)
{
headers[i] = null;
}
}
}
}
}
private void PrepareBodiesResponse(BodiesSyncBatch bodiesSyncBatch, LatencySyncPeerMock syncPeer, IBlockTree tree)
{
int requestSize = bodiesSyncBatch.Request.Length;
int responseSize = bodiesSyncBatch.Request.Length;
if (_incorrectByTooLongMessages.Contains(syncPeer))
{
responseSize *= 2;
TestContext.WriteLine($"{_time,6} | SYNC PEER {syncPeer.Node:s} WILL SEND TOO LONG MESSAGE ({responseSize} INSTEAD OF {requestSize})");
}
else if (_incorrectByTooShortMessages.Contains(syncPeer))
{
responseSize = Math.Max(1, responseSize / 2);
TestContext.WriteLine($"{_time,6} | SYNC PEER {syncPeer.Node:s} WILL SEND TOO SHORT MESSAGE ({responseSize} INSTEAD OF {requestSize})");
}
bodiesSyncBatch.Response = new BlockBody[responseSize];
int maxResponseSize = _peerMaxResponseSizes.ContainsKey(syncPeer) ? Math.Min(responseSize, _peerMaxResponseSizes[syncPeer]) : responseSize;
for (int i = 0; i < Math.Min(maxResponseSize, requestSize); i++)
{
Block block = tree.FindBlock(bodiesSyncBatch.Request[i], BlockTreeLookupOptions.None);
bodiesSyncBatch.Response[i] = new BlockBody(block.Transactions, block.Ommers);
}
if (_maliciousByShortAtStart.Contains(syncPeer))
{
bodiesSyncBatch.Response[0] = null;
TestContext.WriteLine($"{_time,6} | SYNC PEER {syncPeer.Node:s} WILL SEND A MALICIOUS (SHORT AT START) MESSAGE");
}
if (_maliciousByInvalidTxs.Contains(syncPeer))
{
for (int i = 0; i < bodiesSyncBatch.Response.Length; i++)
{
BlockBody valid = bodiesSyncBatch.Response[i];
bodiesSyncBatch.Response[i] = new BlockBody(new[] { Build.A.Transaction.WithData(Bytes.FromHexString("bad")).TestObject }, valid.Ommers);
}
}
if (_maliciousByInvalidOmmers.Contains(syncPeer))
{
for (int i = 0; i < bodiesSyncBatch.Response.Length; i++)
{
BlockBody valid = bodiesSyncBatch.Response[i];
bodiesSyncBatch.Response[i] = new BlockBody(valid.Transactions, new[] { Build.A.BlockHeader.WithAuthor(new Address(Keccak.Compute("bad_ommer").Bytes.Take(20).ToArray())).TestObject });
}
}
}
public ProducedBlockSuggester(IBlockTree blockTree, IBlockProducer blockProducer)
{
_blockTree = blockTree;
_blockProducer = blockProducer;
_blockProducer.BlockProduced += OnBlockProduced;
}
public DebugBridge(IConfigProvider configProvider, IReadOnlyDbProvider dbProvider, IGethStyleTracer tracer, IBlockProcessingQueue receiptsBlockQueue, IBlockTree blockTree)
{
_configProvider = configProvider ?? throw new ArgumentNullException(nameof(configProvider));
_tracer = tracer ?? throw new ArgumentNullException(nameof(tracer));
_blockTree = blockTree ?? throw new ArgumentNullException(nameof(blockTree));
dbProvider = dbProvider ?? throw new ArgumentNullException(nameof(dbProvider));
IDb blockInfosDb = dbProvider.BlockInfosDb ?? throw new ArgumentNullException(nameof(dbProvider.BlockInfosDb));
IDb blocksDb = dbProvider.BlocksDb ?? throw new ArgumentNullException(nameof(dbProvider.BlocksDb));
IDb headersDb = dbProvider.HeadersDb ?? throw new ArgumentNullException(nameof(dbProvider.HeadersDb));
IDb receiptsDb = dbProvider.ReceiptsDb ?? throw new ArgumentNullException(nameof(dbProvider.ReceiptsDb));
IDb codeDb = dbProvider.CodeDb ?? throw new ArgumentNullException(nameof(dbProvider.CodeDb));
IDb pendingTxsDb = dbProvider.PendingTxsDb ?? throw new ArgumentNullException(nameof(dbProvider.PendingTxsDb));
_dbMappings = new Dictionary <string, IDb>(StringComparer.InvariantCultureIgnoreCase)
{
{ DbNames.State, dbProvider.StateDb },
{ DbNames.Storage, dbProvider.StateDb },
{ DbNames.BlockInfos, blockInfosDb },
{ DbNames.Blocks, blocksDb },
{ DbNames.Headers, headersDb },
{ DbNames.Code, codeDb },
{ DbNames.Receipts, receiptsDb },
{ DbNames.PendingTxs, pendingTxsDb },
};
}
public static ReadOnlyBlockTree AsReadOnly(this IBlockTree blockTree)
{
return(new ReadOnlyBlockTree(blockTree));
}
public SynchronizationPeerMock(IBlockTree blockTree, PublicKey publicKey = null)
{
_blockTree = blockTree;
Node = new Node(publicKey ?? TestItem.PublicKeyA, "127.0.0.1", 1234);
}
public ReadOnlyBlockTree(IBlockTree wrapped)
{
_wrapped = wrapped;
}
public UnprocessedBlockTreeWrapper(IBlockTree blockTree)
{
_blockTree = blockTree;
_blockTree.NewHeadBlock += (sender, args) => NewHeadBlock?.Invoke(sender, args);
_blockTree.NewBestSuggestedBlock += (sender, args) => NewBestSuggestedBlock?.Invoke(sender, args);
}
public MultiValidator(AuRaParameters.Validator validator, IAuRaValidatorProcessorFactory validatorFactory, IBlockTree blockTree, IValidatorStore validatorStore, ILogManager logManager)
{
if (validator == null)
{
throw new ArgumentNullException(nameof(validator));
}
if (validator.ValidatorType != AuRaParameters.ValidatorType.Multi)
{
throw new ArgumentException("Wrong validator type.", nameof(validator));
}
_validatorFactory = validatorFactory ?? throw new ArgumentNullException(nameof(validatorFactory));
_blockTree = blockTree ?? throw new ArgumentNullException(nameof(blockTree));
_validatorStore = validatorStore ?? throw new ArgumentNullException(nameof(validatorStore));
_logger = logManager?.GetClassLogger() ?? throw new ArgumentNullException(nameof(logManager));
_validators = validator.Validators?.Count > 0
? validator.Validators
: throw new ArgumentException("Multi validator cannot be empty.", nameof(validator.Validators));
}
public TxFilterAdapter(IBlockTree blockTree, ITxFilter txFilter, ILogManager logManager)
{
_txFilter = txFilter ?? throw new ArgumentNullException(nameof(txFilter));
_logger = logManager.GetClassLogger();
_blockTree = blockTree ?? throw new ArgumentNullException(nameof(blockTree));
}
public DebugBridge(IConfigProvider configProvider, IReadOnlyDbProvider dbProvider, ITracer tracer, IBlockchainProcessor receiptsProcessor, IBlockTree blockTree)
{
receiptsProcessor.ProcessingQueueEmpty += (sender, args) => _receiptProcessedEvent.Set();
_configProvider = configProvider ?? throw new ArgumentNullException(nameof(configProvider));
_tracer = tracer ?? throw new ArgumentNullException(nameof(tracer));
_blockTree = blockTree ?? throw new ArgumentNullException(nameof(blockTree));
dbProvider = dbProvider ?? throw new ArgumentNullException(nameof(dbProvider));
IDb blockInfosDb = dbProvider.BlockInfosDb ?? throw new ArgumentNullException(nameof(dbProvider.BlockInfosDb));
IDb blocksDb = dbProvider.BlocksDb ?? throw new ArgumentNullException(nameof(dbProvider.BlocksDb));
IDb headersDb = dbProvider.HeadersDb ?? throw new ArgumentNullException(nameof(dbProvider.HeadersDb));
IDb receiptsDb = dbProvider.ReceiptsDb ?? throw new ArgumentNullException(nameof(dbProvider.ReceiptsDb));
IDb codeDb = dbProvider.CodeDb ?? throw new ArgumentNullException(nameof(dbProvider.CodeDb));
IDb pendingTxsDb = dbProvider.PendingTxsDb ?? throw new ArgumentNullException(nameof(dbProvider.PendingTxsDb));
IDb traceDb = dbProvider.TraceDb ?? throw new ArgumentNullException(nameof(dbProvider.TraceDb));
_dbMappings = new Dictionary <string, IDb>(StringComparer.InvariantCultureIgnoreCase)
{
{ DbNames.State, dbProvider.StateDb },
{ DbNames.Storage, dbProvider.StateDb },
{ DbNames.BlockInfos, blockInfosDb },
{ DbNames.Blocks, blocksDb },
{ DbNames.Headers, headersDb },
{ DbNames.Code, codeDb },
{ DbNames.Receipts, receiptsDb },
{ DbNames.PendingTxs, pendingTxsDb },
{ DbNames.Trace, traceDb }
};
}
public GethStyleTracer(IBlockchainProcessor processor, IReceiptStorage receiptStorage, IBlockTree blockTree)
{
_processor = processor ?? throw new ArgumentNullException(nameof(processor));
_receiptStorage = receiptStorage ?? throw new ArgumentNullException(nameof(receiptStorage));
_blockTree = blockTree ?? throw new ArgumentNullException(nameof(blockTree));
}
public ProtocolValidator(INodeStatsManager nodeStatsManager, IBlockTree blockTree, ILogManager logManager)
{
_logger = logManager?.GetClassLogger();
_nodeStatsManager = nodeStatsManager ?? throw new ArgumentNullException(nameof(nodeStatsManager));
_blockTree = blockTree ?? throw new ArgumentNullException(nameof(blockTree));
}
public OmmersValidator(IBlockTree blockTree, IHeaderValidator headerValidator, ILogManager logManager)
{
_logger = logManager?.GetClassLogger() ?? throw new ArgumentNullException(nameof(logManager));
_blockTree = blockTree ?? throw new ArgumentNullException(nameof(blockTree));
_headerValidator = headerValidator ?? throw new ArgumentNullException(nameof(headerValidator));
}
public PeerInfo Allocate(PeerInfo currentPeer, IEnumerable <PeerInfo> peers, INodeStatsManager nodeStatsManager, IBlockTree blockTree)
{
return(null);
}
public Tracer(IBlockchainProcessor processor, IReceiptStorage receiptStorage, IBlockTree blockTree, IDb traceDb)
{
_processor = processor ?? throw new ArgumentNullException(nameof(processor));
_receiptStorage = receiptStorage ?? throw new ArgumentNullException(nameof(receiptStorage));
_blockTree = blockTree ?? throw new ArgumentNullException(nameof(blockTree));
_traceDb = traceDb ?? throw new ArgumentNullException(nameof(traceDb));
}
public void AllocateBestPeer(IEnumerable <PeerInfo> peers, INodeStatsManager nodeStatsManager, IBlockTree blockTree)
{
PeerInfo current = Current;
PeerInfo selected = _peerAllocationStrategy.Allocate(Current, peers, nodeStatsManager, blockTree);
if (selected == current)
{
return;
}
AllocationChangeEventArgs args;
lock (_allocationLock)
{
if (selected != null && selected.TryAllocate(Contexts))
{
Current = selected;
args = new AllocationChangeEventArgs(current, selected);
current?.Free(Contexts);
Replaced?.Invoke(this, args);
}
}
}
private void PrepareReceiptsResponse(ReceiptsSyncBatch receiptSyncBatch, LatencySyncPeerMock syncPeer, IBlockTree tree)
{
receiptSyncBatch.Response = new TxReceipt[receiptSyncBatch.Request.Length][];
for (int i = 0; i < receiptSyncBatch.Request.Length; i++)
{
Block block = tree.FindBlock(receiptSyncBatch.Request[i], BlockTreeLookupOptions.None);
receiptSyncBatch.Response[i] = new TxReceipt[block.Transactions.Length];
for (int j = 0; j < block.Transactions.Length; j++)
{
receiptSyncBatch.Response[i][j] = _remoteReceiptStorage.Find(block.Transactions[j].Hash);
if (i < 10 && j == 0 && _maliciousByInvalidReceipts.Contains(syncPeer))
{
receiptSyncBatch.Response[i][j] = new TxReceipt();
receiptSyncBatch.Response[i][j].Logs = new LogEntry[0];
receiptSyncBatch.Response[i][j].StatusCode = (byte)(1 - receiptSyncBatch.Response[i][j].StatusCode);
receiptSyncBatch.Response[i][j].PostTransactionState = Keccak.Compute(receiptSyncBatch.Response[i][j].PostTransactionState?.Bytes ?? new byte[] { 1 });
}
}
}
}
public virtual async Task <INdmCapabilityConnector> InitAsync(IConfigProvider configProvider,
IDbProvider dbProvider, string baseDbPath, IBlockTree blockTree, ITxPool txPool, ISpecProvider specProvider,
IReceiptStorage receiptStorage, IWallet wallet, IFilterStore filterStore, IFilterManager filterManager,
ITimestamper timestamper, IEthereumEcdsa ecdsa, IRpcModuleProvider rpcModuleProvider, IKeyStore keyStore,
IJsonSerializer jsonSerializer, ICryptoRandom cryptoRandom, IEnode enode,
INdmConsumerChannelManager consumerChannelManager, INdmDataPublisher dataPublisher, IGrpcServer grpcServer,
INodeStatsManager nodeStatsManager, IProtocolsManager protocolsManager,
IProtocolValidator protocolValidator, IMessageSerializationService messageSerializationService,
bool enableUnsecuredDevWallet, IWebSocketsManager webSocketsManager, ILogManager logManager,
IBlockProcessor blockProcessor, IJsonRpcClientProxy jsonRpcClientProxy,
IEthJsonRpcClientProxy ethJsonRpcClientProxy, IHttpClient httpClient, IMonitoringService monitoringService)
{
var(config, services, faucet, ethRequestService, accountService, consumerService, consumerAddress,
providerAddress) = await PreInitAsync(configProvider, dbProvider, baseDbPath, blockTree, txPool,
specProvider, receiptStorage, wallet, filterStore, filterManager, timestamper, ecdsa, rpcModuleProvider,
keyStore, jsonSerializer, cryptoRandom, enode, consumerChannelManager, dataPublisher, grpcServer,
enableUnsecuredDevWallet, webSocketsManager, logManager, blockProcessor, jsonRpcClientProxy,
ethJsonRpcClientProxy, httpClient, monitoringService);
if (!config.Enabled)
{
return(default);
public static BlockHeader FindParentHeader(this IBlockTree tree, BlockHeader header, BlockTreeLookupOptions options)
{
return(tree.FindHeader(header.ParentHash, options));
}
public Context()
{
BlockTree = Substitute.For <IBlockTree>();
Stats = Substitute.For <INodeStatsManager>();
Pool = new SyncPeerPool(BlockTree, Stats, 25, 50, LimboLogs.Instance);
}
public static Block RetrieveHeadBlock(this IBlockTree tree)
{
return(tree.FindBlock(tree.Head.Hash, BlockTreeLookupOptions.None));
}
public TestAuraProducer(ITxSource transactionSource, IBlockchainProcessor processor, IStateProvider stateProvider, ISealer sealer, IBlockTree blockTree, IBlockProcessingQueue blockProcessingQueue, ITimestamper timestamper, ILogManager logManager, IAuRaStepCalculator auRaStepCalculator, IAuraConfig config, Address nodeAddress) : base(transactionSource, processor, stateProvider, sealer, blockTree, blockProcessingQueue, timestamper, logManager, auRaStepCalculator, config, nodeAddress)
{
}
public AvroMapper(IBlockTree blockTree)
{
_blockTree = blockTree;
}
public StateSyncFeed(ISnapshotableDb codeDb, ISnapshotableDb stateDb, IDb tempDb, ISyncModeSelector syncModeSelector, IBlockTree blockTree, ILogManager logManager)
: base(logManager)
{
_codeDb = codeDb?.Innermost ?? throw new ArgumentNullException(nameof(codeDb));
_stateDb = stateDb?.Innermost ?? throw new ArgumentNullException(nameof(stateDb));
_tempDb = tempDb ?? throw new ArgumentNullException(nameof(tempDb));
_blockTree = blockTree ?? throw new ArgumentNullException(nameof(blockTree));
_syncModeSelector = syncModeSelector ?? throw new ArgumentNullException(nameof(syncModeSelector));
_syncModeSelector.Changed += SyncModeSelectorOnChanged;
_logger = logManager.GetClassLogger() ?? throw new ArgumentNullException(nameof(logManager));
byte[] progress = _codeDb.Get(_fastSyncProgressKey);
_data = new DetailedProgress(_blockTree.ChainId, progress);
_pendingItems = new PendingSyncItems();
}
