Page MenuHomePhabricator
Differential D8794 Diff 26682 src/test/denialofservice_tests.cpp

Changeset View

Standalone View

View Options

src/test/denialofservice_tests.cpp

Show First 20 Lines • Show All 248 Lines • ▼ Show 20 LinesBOOST_AUTO_TEST_CASE(peer_discouragement) {
banman->ClearBanned(); banman->ClearBanned();
CAddress addr1(ip(0xa0b0c001), NODE_NONE); CAddress addr1(ip(0xa0b0c001), NODE_NONE);
CNode dummyNode1(id++, NODE_NETWORK, 0, INVALID_SOCKET, addr1, 0, 0, CNode dummyNode1(id++, NODE_NETWORK, 0, INVALID_SOCKET, addr1, 0, 0,
CAddress(), "", ConnectionType::INBOUND); CAddress(), "", ConnectionType::INBOUND);
dummyNode1.SetSendVersion(PROTOCOL_VERSION); dummyNode1.SetSendVersion(PROTOCOL_VERSION);
peerLogic->InitializeNode(config, &dummyNode1); peerLogic->InitializeNode(config, &dummyNode1);
dummyNode1.nVersion = 1; dummyNode1.nVersion = 1;
dummyNode1.fSuccessfullyConnected = true; dummyNode1.fSuccessfullyConnected = true;
{
LOCK(cs_main);
// Should be discouraged // Should be discouraged
Misbehaving(dummyNode1.GetId(), DISCOURAGEMENT_THRESHOLD); Misbehaving(dummyNode1.GetId(), DISCOURAGEMENT_THRESHOLD);
}
{ {
LOCK2(cs_main, dummyNode1.cs_sendProcessing); LOCK2(cs_main, dummyNode1.cs_sendProcessing);
BOOST_CHECK( BOOST_CHECK(
peerLogic->SendMessages(config, &dummyNode1, interruptDummy)); peerLogic->SendMessages(config, &dummyNode1, interruptDummy));
} }
BOOST_CHECK(banman->IsDiscouraged(addr1)); BOOST_CHECK(banman->IsDiscouraged(addr1));
// Different IP, not discouraged // Different IP, not discouraged
BOOST_CHECK(!banman->IsDiscouraged(ip(0xa0b0c001 | 0x0000ff00))); BOOST_CHECK(!banman->IsDiscouraged(ip(0xa0b0c001 | 0x0000ff00)));
CAddress addr2(ip(0xa0b0c002), NODE_NONE); CAddress addr2(ip(0xa0b0c002), NODE_NONE);
CNode dummyNode2(id++, NODE_NETWORK, 0, INVALID_SOCKET, addr2, 1, 1, CNode dummyNode2(id++, NODE_NETWORK, 0, INVALID_SOCKET, addr2, 1, 1,
CAddress(), "", ConnectionType::INBOUND); CAddress(), "", ConnectionType::INBOUND);
dummyNode2.SetSendVersion(PROTOCOL_VERSION); dummyNode2.SetSendVersion(PROTOCOL_VERSION);
peerLogic->InitializeNode(config, &dummyNode2); peerLogic->InitializeNode(config, &dummyNode2);
dummyNode2.nVersion = 1; dummyNode2.nVersion = 1;
dummyNode2.fSuccessfullyConnected = true; dummyNode2.fSuccessfullyConnected = true;
{
LOCK(cs_main);
Misbehaving(dummyNode2.GetId(), DISCOURAGEMENT_THRESHOLD - 1); Misbehaving(dummyNode2.GetId(), DISCOURAGEMENT_THRESHOLD - 1);
}
{ {
LOCK2(cs_main, dummyNode2.cs_sendProcessing); LOCK2(cs_main, dummyNode2.cs_sendProcessing);
BOOST_CHECK( BOOST_CHECK(
peerLogic->SendMessages(config, &dummyNode2, interruptDummy)); peerLogic->SendMessages(config, &dummyNode2, interruptDummy));
} }
// 2 not discouraged yet... // 2 not discouraged yet...
BOOST_CHECK(!banman->IsDiscouraged(addr2)); BOOST_CHECK(!banman->IsDiscouraged(addr2));
// ... but 1 still should be // ... but 1 still should be
BOOST_CHECK(banman->IsDiscouraged(addr1)); BOOST_CHECK(banman->IsDiscouraged(addr1));
{
LOCK(cs_main);
// 2 reaches discouragement threshold // 2 reaches discouragement threshold
Misbehaving(dummyNode2.GetId(), 1); Misbehaving(dummyNode2.GetId(), 1);
}
{ {
LOCK2(cs_main, dummyNode2.cs_sendProcessing); LOCK2(cs_main, dummyNode2.cs_sendProcessing);
BOOST_CHECK( BOOST_CHECK(
peerLogic->SendMessages(config, &dummyNode2, interruptDummy)); peerLogic->SendMessages(config, &dummyNode2, interruptDummy));
} }
BOOST_CHECK(banman->IsDiscouraged(addr1)); // Expect both 1 and 2 BOOST_CHECK(banman->IsDiscouraged(addr1)); // Expect both 1 and 2
BOOST_CHECK(banman->IsDiscouraged(addr2)); // to be discouraged now BOOST_CHECK(banman->IsDiscouraged(addr2)); // to be discouraged now
Show All 22 LinesBOOST_AUTO_TEST_CASE(DoS_bantime) {
CAddress addr(ip(0xa0b0c001), NODE_NONE); CAddress addr(ip(0xa0b0c001), NODE_NONE);
CNode dummyNode(id++, NODE_NETWORK, 0, INVALID_SOCKET, addr, 4, 4, CNode dummyNode(id++, NODE_NETWORK, 0, INVALID_SOCKET, addr, 4, 4,
CAddress(), "", ConnectionType::INBOUND); CAddress(), "", ConnectionType::INBOUND);
dummyNode.SetSendVersion(PROTOCOL_VERSION); dummyNode.SetSendVersion(PROTOCOL_VERSION);
peerLogic->InitializeNode(config, &dummyNode); peerLogic->InitializeNode(config, &dummyNode);
dummyNode.nVersion = 1; dummyNode.nVersion = 1;
dummyNode.fSuccessfullyConnected = true; dummyNode.fSuccessfullyConnected = true;
{
LOCK(cs_main);
Misbehaving(dummyNode.GetId(), DISCOURAGEMENT_THRESHOLD); Misbehaving(dummyNode.GetId(), DISCOURAGEMENT_THRESHOLD);
}
{ {
LOCK2(cs_main, dummyNode.cs_sendProcessing); LOCK2(cs_main, dummyNode.cs_sendProcessing);
BOOST_CHECK( BOOST_CHECK(
peerLogic->SendMessages(config, &dummyNode, interruptDummy)); peerLogic->SendMessages(config, &dummyNode, interruptDummy));
} }
BOOST_CHECK(banman->IsDiscouraged(addr)); BOOST_CHECK(banman->IsDiscouraged(addr));
bool dummy; bool dummy;
▲ Show 20 Lines • Show All 92 Lines • Show Last 20 Lines