public async Task NodeIsSynced_PeerSendsABadBlockAndPeerIsWhitelisted_ThePeerIsNotBanned_Async()
{
string dataDir = Path.Combine("TestData", nameof(PeerBanningTest), nameof(this.NodeIsSynced_PeerSendsABadBlockAndPeerIsWhitelisted_ThePeerIsNotBanned_Async));
Directory.CreateDirectory(dataDir);
TestChainContext context = await TestChainFactory.CreateAsync(Network.RegTest, dataDir);
var peerEndPoint = new IPEndPoint(IPAddress.Parse("1.2.3.4"), context.Network.DefaultPort);
var connectionManagerBehavior = new ConnectionManagerBehavior(false, context.ConnectionManager, context.LoggerFactory)
{
Whitelisted = true
};
var peer = new Mock <INetworkPeer>();
peer.Setup(p => p.Behavior <ConnectionManagerBehavior>()).Returns(connectionManagerBehavior);
context.MockReadOnlyNodesCollection.Setup(s => s.FindByEndpoint(It.IsAny <IPEndPoint>())).Returns(peer.Object);
var blocks = await TestChainFactory.MineBlocksAsync(context, 2, new Key().ScriptPubKey);
// create a new block that breaks consensus.
var block = blocks.First();
block.Header.HashPrevBlock = context.Chain.Tip.HashBlock;
await context.Consensus.AcceptBlockAsync(new BlockValidationContext { Block = block, Peer = peerEndPoint });
Assert.False(context.PeerBanning.IsBanned(peerEndPoint));
}
public async Task NodeIsSynced_PeerSendsABadBlockAndPeerIsBandAndBanIsExpired_ThePeerIsNotBanned_Async()
{
string dataDir = Path.Combine("TestData", nameof(PeerBanningTest), nameof(this.NodeIsSynced_PeerSendsABadBlockAndPeerIsBandAndBanIsExpired_ThePeerIsNotBanned_Async));
Directory.CreateDirectory(dataDir);
TestChainContext context = await TestChainFactory.CreateAsync(Network.RegTest, dataDir);
var peer = new IPEndPoint(IPAddress.Parse("1.2.3.4"), context.Network.DefaultPort);
var connectionManagerBehavior = new ConnectionManagerBehavior(false, context.ConnectionManager, context.LoggerFactory)
{
Whitelisted = true
};
var node = new NetworkPeer(context.DateTimeProvider, context.LoggerFactory);
node.Behaviors.Add(connectionManagerBehavior);
context.MockReadOnlyNodesCollection.Setup(s => s.FindByEndpoint(It.IsAny <IPEndPoint>())).Returns(node);
var blocks = await TestChainFactory.MineBlocksAsync(context, 2, new Key().ScriptPubKey);
// create a new block that breaks consensus.
var block = blocks.First();
block.Header.HashPrevBlock = context.Chain.Tip.HashBlock;
await context.Consensus.AcceptBlockAsync(new BlockValidationContext { Block = block, Peer = peer, BanDurationSeconds = 1 }); // ban for 1 second
// wait 1 sec for ban to expire.
Thread.Sleep(1000);
Assert.False(context.PeerBanning.IsBanned(peer));
}
private static async Task <Block> Mine2BlocksAndCreateABlockWithBadPrevHashAsync(TestChainContext context)
{
var blocks = await TestChainFactory.MineBlocksAsync(context, 2, MinerScriptPubKey);
var block = blocks.First();
block.Header.HashPrevBlock = context.Chain.Tip.HashBlock;
return(block);
}
public async Task NodeIsSynced_PeerSendsABadBlockAndPeerDiconnected_ThePeerGetsBanned_Async()
{
string dataDir = Path.Combine("TestData", nameof(PeerBanningTest), nameof(this.NodeIsSynced_PeerSendsABadBlockAndPeerDiconnected_ThePeerGetsBanned_Async));
Directory.CreateDirectory(dataDir);
TestChainContext context = await TestChainFactory.CreateAsync(Network.RegTest, dataDir);
var peer = new IPEndPoint(IPAddress.Parse("1.2.3.4"), context.Network.DefaultPort);
context.MockReadOnlyNodesCollection.Setup(s => s.FindByEndpoint(It.IsAny <IPEndPoint>())).Returns((NetworkPeer)null);
var blocks = await TestChainFactory.MineBlocksAsync(context, 2, new Key().ScriptPubKey);
var block = blocks.First();
block.Header.HashPrevBlock = context.Chain.Tip.HashBlock;
await context.Consensus.AcceptBlockAsync(new BlockValidationContext { Block = block, Peer = peer });
Assert.True(context.PeerBanning.IsBanned(peer));
}