public async Task PeerBanning_ResettingExpiredBannedPeerAsync()
{
// Arrange
string dataDir = GetTestDirectoryPath(this);
TestChainContext context = await TestChainFactory.CreateAsync(Network.RegTest, dataDir);
IPAddress ipAddress = IPAddress.Parse("::ffff:192.168.0.1");
var endpoint = new IPEndPoint(ipAddress, 80);
context.PeerAddressManager.AddPeer(endpoint, endpoint.Address.MapToIPv6());
// Act
context.PeerBanning.BanPeer(endpoint, 1, nameof(PeerBanningTest));
context.PeerAddressManager.SavePeers();
// Wait one second for ban to expire.
Thread.Sleep(1000);
context.PeerAddressManager.Peers.Clear();
context.PeerAddressManager.LoadPeers();
// Assert
PeerAddress peer = context.PeerAddressManager.FindPeer(endpoint);
Assert.Null(peer.BanTimeStamp);
Assert.Null(peer.BanUntil);
Assert.Empty(peer.BanReason);
}
public async Task PeerBanning_SavingAndLoadingBannedPeerToAddressManagerStoreAsync()
{
// Arrange
string dataDir = GetTestDirectoryPath(this);
TestChainContext context = await TestChainFactory.CreateAsync(Network.RegTest, dataDir);
IPAddress ipAddress = IPAddress.Parse("::ffff:192.168.0.1");
var endpoint = new IPEndPoint(ipAddress, 80);
context.PeerAddressManager.AddPeer(endpoint, endpoint.Address.MapToIPv6());
// Act - Ban Peer, save store, clear current Peers, load store
context.PeerBanning.BanPeer(endpoint, context.ConnectionManager.ConnectionSettings.BanTimeSeconds, nameof(PeerBanningTest));
context.PeerAddressManager.SavePeers();
context.PeerAddressManager.Peers.Clear();
context.PeerAddressManager.LoadPeers();
// Assert
PeerAddress peer = context.PeerAddressManager.FindPeer(endpoint);
Assert.NotNull(peer.BanTimeStamp);
Assert.NotNull(peer.BanUntil);
Assert.NotEmpty(peer.BanReason);
}
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 async Task <(TestChainContext context, IPEndPoint peerEndPoint)> InitialiseContextAndPeerEndpointAsync()
{
string dataDir = GetTestDirectoryPath(this);
TestChainContext context = await TestChainFactory.CreateAsync(Network.RegTest, dataDir);
var peerEndPoint = new IPEndPoint(IPAddress.Parse("1.2.3.4"), context.Network.DefaultPort);
context.PeerAddressManager.AddPeer(peerEndPoint, peerEndPoint.Address.MapToIPv6());
return(context, peerEndPoint);
}
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));
}
public async Task PeerBanning_AddingBannedPeerToAddressManagerStoreAsync()
{
// Arrange
string dataDir = GetTestDirectoryPath(this);
TestChainContext context = await TestChainFactory.CreateAsync(Network.RegTest, dataDir);
IPAddress ipAddress = IPAddress.Parse("::ffff:192.168.0.1");
var endpoint = new IPEndPoint(ipAddress, 80);
context.PeerAddressManager.AddPeer(endpoint, endpoint.Address.MapToIPv6());
// Act
context.PeerBanning.BanAndDisconnectPeer(endpoint, context.ConnectionManager.ConnectionSettings.BanTimeSeconds, nameof(PeerBanningTest));
// Assert
PeerAddress peer = context.PeerAddressManager.FindPeer(endpoint);
Assert.True(peer.BanUntil.HasValue);
Assert.NotNull(peer.BanUntil);
Assert.NotEmpty(peer.BanReason);
}