Page MenuHomePhabricator
Differential D7146 Diff 22790 src/test/denialofservice_tests.cpp

Changeset View

Standalone View

View Options

src/test/denialofservice_tests.cpp

Show First 20 Lines • Show All 257 Lines • ▼ Show 20 Lines dummyNode1.fSuccessfullyConnected = true;
// Should get banned. // Should get banned.
Misbehaving(dummyNode1.GetId(), 100, ""); Misbehaving(dummyNode1.GetId(), 100, "");
} }
{ {
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->IsBanned(addr1)); BOOST_CHECK(banman->IsDiscouraged(addr1));
// Different IP, not banned. // Different IP, not banned
BOOST_CHECK(!banman->IsBanned(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(), "", true); CAddress(), "", true);
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); LOCK(cs_main);
Misbehaving(dummyNode2.GetId(), 50, ""); Misbehaving(dummyNode2.GetId(), 50, "");
} }
{ {
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 banned yet... // 2 not banned yet...
BOOST_CHECK(!banman->IsBanned(addr2)); BOOST_CHECK(!banman->IsDiscouraged(addr2));
// ... but 1 still should be. // ... but 1 still should be
BOOST_CHECK(banman->IsBanned(addr1)); BOOST_CHECK(banman->IsDiscouraged(addr1));
{ {
LOCK(cs_main); LOCK(cs_main);
Misbehaving(dummyNode2.GetId(), 50, ""); Misbehaving(dummyNode2.GetId(), 50, "");
} }
{ {
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->IsBanned(addr2)); BOOST_CHECK(banman->IsDiscouraged(addr2));
bool dummy; bool dummy;
peerLogic->FinalizeNode(config, dummyNode1.GetId(), dummy); peerLogic->FinalizeNode(config, dummyNode1.GetId(), dummy);
peerLogic->FinalizeNode(config, dummyNode2.GetId(), dummy); peerLogic->FinalizeNode(config, dummyNode2.GetId(), dummy);
} }
BOOST_AUTO_TEST_CASE(DoS_banscore) { BOOST_AUTO_TEST_CASE(DoS_banscore) {
const Config &config = GetConfig(); const Config &config = GetConfig();
Show All 20 Lines dummyNode1.fSuccessfullyConnected = true;
LOCK(cs_main); LOCK(cs_main);
Misbehaving(dummyNode1.GetId(), 100, ""); Misbehaving(dummyNode1.GetId(), 100, "");
} }
{ {
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->IsBanned(addr1)); BOOST_CHECK(!banman->IsDiscouraged(addr1));
{ {
LOCK(cs_main); LOCK(cs_main);
Misbehaving(dummyNode1.GetId(), 10, ""); Misbehaving(dummyNode1.GetId(), 10, "");
} }
{ {
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->IsBanned(addr1)); BOOST_CHECK(!banman->IsDiscouraged(addr1));
{ {
LOCK(cs_main); LOCK(cs_main);
Misbehaving(dummyNode1.GetId(), 1, ""); Misbehaving(dummyNode1.GetId(), 1, "");
} }
{ {
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->IsBanned(addr1)); BOOST_CHECK(banman->IsDiscouraged(addr1));
gArgs.ForceSetArg("-banscore", std::to_string(DEFAULT_BANSCORE_THRESHOLD)); gArgs.ForceSetArg("-banscore", std::to_string(DEFAULT_BANSCORE_THRESHOLD));
bool dummy; bool dummy;
peerLogic->FinalizeNode(config, dummyNode1.GetId(), dummy); peerLogic->FinalizeNode(config, dummyNode1.GetId(), dummy);
} }
BOOST_AUTO_TEST_CASE(DoS_bantime) { BOOST_AUTO_TEST_CASE(DoS_bantime) {
const Config &config = GetConfig(); const Config &config = GetConfig();
Show All 23 Lines dummyNode.fSuccessfullyConnected = true;
LOCK(cs_main); LOCK(cs_main);
Misbehaving(dummyNode.GetId(), 100, ""); Misbehaving(dummyNode.GetId(), 100, "");
} }
{ {
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->IsBanned(addr)); BOOST_CHECK(banman->IsDiscouraged(addr));
bool dummy; bool dummy;
peerLogic->FinalizeNode(config, dummyNode.GetId(), dummy); peerLogic->FinalizeNode(config, dummyNode.GetId(), dummy);
} }
static CTransactionRef RandomOrphan() { static CTransactionRef RandomOrphan() {
std::map<uint256, COrphanTx>::iterator it; std::map<uint256, COrphanTx>::iterator it;
LOCK2(cs_main, g_cs_orphans); LOCK2(cs_main, g_cs_orphans);
▲ Show 20 Lines • Show All 86 Lines • Show Last 20 Lines