Page MenuHomePhabricator
Differential D380 Diff 956 src/test/DoS_tests.cpp

Changeset View

Standalone View

View Options

src/test/DoS_tests.cpp

Show First 20 Lines • Show All 51 Lines • ▼ Show 20 LinesBOOST_AUTO_TEST_CASE(DoS_banning) {
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, "",
true); true);
dummyNode1.SetSendVersion(PROTOCOL_VERSION); dummyNode1.SetSendVersion(PROTOCOL_VERSION);
GetNodeSignals().InitializeNode(config, &dummyNode1, *connman); GetNodeSignals().InitializeNode(config, &dummyNode1, *connman);
dummyNode1.nVersion = 1; dummyNode1.nVersion = 1;
dummyNode1.fSuccessfullyConnected = true; dummyNode1.fSuccessfullyConnected = true;
// Should get banned. // Should get banned.
Misbehaving(dummyNode1.GetId(), 100); Misbehaving(dummyNode1.GetId(), 100, "");
SendMessages(config, &dummyNode1, *connman, interruptDummy); SendMessages(config, &dummyNode1, *connman, interruptDummy);
BOOST_CHECK(connman->IsBanned(addr1)); BOOST_CHECK(connman->IsBanned(addr1));
// Different IP, not banned. // Different IP, not banned.
BOOST_CHECK(!connman->IsBanned(ip(0xa0b0c001 | 0x0000ff00))); BOOST_CHECK(!connman->IsBanned(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, "",
true); true);
dummyNode2.SetSendVersion(PROTOCOL_VERSION); dummyNode2.SetSendVersion(PROTOCOL_VERSION);
GetNodeSignals().InitializeNode(config, &dummyNode2, *connman); GetNodeSignals().InitializeNode(config, &dummyNode2, *connman);
dummyNode2.nVersion = 1; dummyNode2.nVersion = 1;
dummyNode2.fSuccessfullyConnected = true; dummyNode2.fSuccessfullyConnected = true;
Misbehaving(dummyNode2.GetId(), 50); Misbehaving(dummyNode2.GetId(), 50, "");
SendMessages(config, &dummyNode2, *connman, interruptDummy); SendMessages(config, &dummyNode2, *connman, interruptDummy);
// 2 not banned yet... // 2 not banned yet...
BOOST_CHECK(!connman->IsBanned(addr2)); BOOST_CHECK(!connman->IsBanned(addr2));
// ... but 1 still should be. // ... but 1 still should be.
BOOST_CHECK(connman->IsBanned(addr1)); BOOST_CHECK(connman->IsBanned(addr1));
Misbehaving(dummyNode2.GetId(), 50); Misbehaving(dummyNode2.GetId(), 50, "");
SendMessages(config, &dummyNode2, *connman, interruptDummy); SendMessages(config, &dummyNode2, *connman, interruptDummy);
BOOST_CHECK(connman->IsBanned(addr2)); BOOST_CHECK(connman->IsBanned(addr2));
} }
BOOST_AUTO_TEST_CASE(DoS_banscore) { BOOST_AUTO_TEST_CASE(DoS_banscore) {
const Config &config = GetConfig(); const Config &config = GetConfig();
std::atomic<bool> interruptDummy(false); std::atomic<bool> interruptDummy(false);
connman->ClearBanned(); connman->ClearBanned();
// because 11 is my favorite number. // because 11 is my favorite number.
ForceSetArg("-banscore", "111"); ForceSetArg("-banscore", "111");
CAddress addr1(ip(0xa0b0c001), NODE_NONE); CAddress addr1(ip(0xa0b0c001), NODE_NONE);
CNode dummyNode1(id++, NODE_NETWORK, 0, INVALID_SOCKET, addr1, 3, 1, "", CNode dummyNode1(id++, NODE_NETWORK, 0, INVALID_SOCKET, addr1, 3, 1, "",
true); true);
dummyNode1.SetSendVersion(PROTOCOL_VERSION); dummyNode1.SetSendVersion(PROTOCOL_VERSION);
GetNodeSignals().InitializeNode(config, &dummyNode1, *connman); GetNodeSignals().InitializeNode(config, &dummyNode1, *connman);
dummyNode1.nVersion = 1; dummyNode1.nVersion = 1;
dummyNode1.fSuccessfullyConnected = true; dummyNode1.fSuccessfullyConnected = true;
Misbehaving(dummyNode1.GetId(), 100); Misbehaving(dummyNode1.GetId(), 100, "");
SendMessages(config, &dummyNode1, *connman, interruptDummy); SendMessages(config, &dummyNode1, *connman, interruptDummy);
BOOST_CHECK(!connman->IsBanned(addr1)); BOOST_CHECK(!connman->IsBanned(addr1));
Misbehaving(dummyNode1.GetId(), 10); Misbehaving(dummyNode1.GetId(), 10, "");
SendMessages(config, &dummyNode1, *connman, interruptDummy); SendMessages(config, &dummyNode1, *connman, interruptDummy);
BOOST_CHECK(!connman->IsBanned(addr1)); BOOST_CHECK(!connman->IsBanned(addr1));
Misbehaving(dummyNode1.GetId(), 1); Misbehaving(dummyNode1.GetId(), 1, "");
SendMessages(config, &dummyNode1, *connman, interruptDummy); SendMessages(config, &dummyNode1, *connman, interruptDummy);
BOOST_CHECK(connman->IsBanned(addr1)); BOOST_CHECK(connman->IsBanned(addr1));
ForceSetArg("-banscore", std::to_string(DEFAULT_BANSCORE_THRESHOLD)); ForceSetArg("-banscore", std::to_string(DEFAULT_BANSCORE_THRESHOLD));
} }
BOOST_AUTO_TEST_CASE(DoS_bantime) { BOOST_AUTO_TEST_CASE(DoS_bantime) {
const Config &config = GetConfig(); const Config &config = GetConfig();
std::atomic<bool> interruptDummy(false); std::atomic<bool> interruptDummy(false);
connman->ClearBanned(); connman->ClearBanned();
int64_t nStartTime = GetTime(); int64_t nStartTime = GetTime();
// Overrides future calls to GetTime() // Overrides future calls to GetTime()
SetMockTime(nStartTime); SetMockTime(nStartTime);
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, "",
true); true);
dummyNode.SetSendVersion(PROTOCOL_VERSION); dummyNode.SetSendVersion(PROTOCOL_VERSION);
GetNodeSignals().InitializeNode(config, &dummyNode, *connman); GetNodeSignals().InitializeNode(config, &dummyNode, *connman);
dummyNode.nVersion = 1; dummyNode.nVersion = 1;
dummyNode.fSuccessfullyConnected = true; dummyNode.fSuccessfullyConnected = true;
Misbehaving(dummyNode.GetId(), 100); Misbehaving(dummyNode.GetId(), 100, "");
SendMessages(config, &dummyNode, *connman, interruptDummy); SendMessages(config, &dummyNode, *connman, interruptDummy);
BOOST_CHECK(connman->IsBanned(addr)); BOOST_CHECK(connman->IsBanned(addr));
SetMockTime(nStartTime + 60 * 60); SetMockTime(nStartTime + 60 * 60);
BOOST_CHECK(connman->IsBanned(addr)); BOOST_CHECK(connman->IsBanned(addr));
SetMockTime(nStartTime + 60 * 60 * 24 + 1); SetMockTime(nStartTime + 60 * 60 * 24 + 1);
BOOST_CHECK(!connman->IsBanned(addr)); BOOST_CHECK(!connman->IsBanned(addr));
▲ Show 20 Lines • Show All 87 Lines • Show Last 20 Lines