2014-03-18 02:11:00 -07:00
|
|
|
// Copyright (c) 2011-2014 The Bitcoin Core developers
|
2023-01-23 10:31:54 -08:00
|
|
|
// Copyright (c) 2018-2023 The Zcash developers
|
2014-12-12 20:09:33 -08:00
|
|
|
// Distributed under the MIT software license, see the accompanying
|
2019-07-18 07:16:09 -07:00
|
|
|
// file COPYING or https://www.opensource.org/licenses/mit-license.php .
|
2014-03-18 02:11:00 -07:00
|
|
|
|
2011-09-06 13:09:04 -07:00
|
|
|
//
|
|
|
|
// Unit tests for denial-of-service detection/prevention code
|
|
|
|
//
|
2012-05-16 12:57:04 -07:00
|
|
|
|
2011-09-06 13:09:04 -07:00
|
|
|
|
2013-04-12 22:13:08 -07:00
|
|
|
|
2018-02-01 17:49:42 -08:00
|
|
|
#include "consensus/upgrades.h"
|
2013-04-12 22:13:08 -07:00
|
|
|
#include "keystore.h"
|
2011-10-11 16:50:06 -07:00
|
|
|
#include "main.h"
|
|
|
|
#include "net.h"
|
2014-03-10 08:46:53 -07:00
|
|
|
#include "pow.h"
|
2014-08-27 08:22:33 -07:00
|
|
|
#include "script/sign.h"
|
2013-04-12 22:13:08 -07:00
|
|
|
#include "serialize.h"
|
2018-10-22 15:51:11 -07:00
|
|
|
#include "util/system.h"
|
2011-09-06 13:09:04 -07:00
|
|
|
|
2015-03-03 07:49:12 -08:00
|
|
|
#include "test/test_bitcoin.h"
|
|
|
|
|
2012-01-03 14:33:31 -08:00
|
|
|
#include <stdint.h>
|
|
|
|
|
2013-04-12 22:13:08 -07:00
|
|
|
#include <boost/assign/list_of.hpp> // for 'map_list_of()'
|
|
|
|
#include <boost/date_time/posix_time/posix_time_types.hpp>
|
|
|
|
#include <boost/test/unit_test.hpp>
|
2018-04-09 16:17:30 -07:00
|
|
|
#include <boost/test/data/test_case.hpp>
|
2013-04-12 22:13:08 -07:00
|
|
|
|
2012-02-29 07:14:18 -08:00
|
|
|
// Tests this internal-to-main.cpp method:
|
2014-08-28 10:23:24 -07:00
|
|
|
extern bool AddOrphanTx(const CTransaction& tx, NodeId peer);
|
|
|
|
extern void EraseOrphansFor(NodeId peer);
|
2012-04-23 11:14:03 -07:00
|
|
|
extern unsigned int LimitOrphanTxSize(unsigned int nMaxOrphans);
|
2014-09-20 00:53:50 -07:00
|
|
|
struct COrphanTx {
|
|
|
|
CTransaction tx;
|
|
|
|
NodeId fromPeer;
|
|
|
|
};
|
|
|
|
extern std::map<uint256, COrphanTx> mapOrphanTransactions;
|
2013-08-01 22:14:44 -07:00
|
|
|
extern std::map<uint256, std::set<uint256> > mapOrphanTransactionsByPrev;
|
2012-02-29 07:14:18 -08:00
|
|
|
|
2012-01-03 14:33:31 -08:00
|
|
|
CService ip(uint32_t i)
|
|
|
|
{
|
|
|
|
struct in_addr s;
|
|
|
|
s.s_addr = i;
|
2013-05-07 06:16:25 -07:00
|
|
|
return CService(CNetAddr(s), Params().GetDefaultPort());
|
2012-01-03 14:33:31 -08:00
|
|
|
}
|
2011-09-06 13:09:04 -07:00
|
|
|
|
2015-03-03 07:49:12 -08:00
|
|
|
BOOST_FIXTURE_TEST_SUITE(DoS_tests, TestingSetup)
|
2011-09-06 13:09:04 -07:00
|
|
|
|
|
|
|
BOOST_AUTO_TEST_CASE(DoS_banning)
|
|
|
|
{
|
2020-07-10 10:34:53 -07:00
|
|
|
const Consensus::Params& params = Params().GetConsensus();
|
2011-09-06 13:09:04 -07:00
|
|
|
CNode::ClearBanned();
|
2012-01-03 14:33:31 -08:00
|
|
|
CAddress addr1(ip(0xa0b0c001));
|
2012-02-12 04:45:24 -08:00
|
|
|
CNode dummyNode1(INVALID_SOCKET, addr1, "", true);
|
2013-11-17 16:25:17 -08:00
|
|
|
dummyNode1.nVersion = 1;
|
|
|
|
Misbehaving(dummyNode1.GetId(), 100); // Should get banned
|
2015-04-08 11:20:00 -07:00
|
|
|
SendMessages(params, &dummyNode1);
|
2012-01-03 14:33:31 -08:00
|
|
|
BOOST_CHECK(CNode::IsBanned(addr1));
|
2012-07-25 17:48:39 -07:00
|
|
|
BOOST_CHECK(!CNode::IsBanned(ip(0xa0b0c001|0x0000ff00))); // Different IP, not banned
|
2011-09-06 13:09:04 -07:00
|
|
|
|
2012-01-03 14:33:31 -08:00
|
|
|
CAddress addr2(ip(0xa0b0c002));
|
2012-02-12 04:45:24 -08:00
|
|
|
CNode dummyNode2(INVALID_SOCKET, addr2, "", true);
|
2013-11-17 16:25:17 -08:00
|
|
|
dummyNode2.nVersion = 1;
|
|
|
|
Misbehaving(dummyNode2.GetId(), 50);
|
2015-04-08 11:20:00 -07:00
|
|
|
SendMessages(params, &dummyNode2);
|
2012-01-03 14:33:31 -08:00
|
|
|
BOOST_CHECK(!CNode::IsBanned(addr2)); // 2 not banned yet...
|
|
|
|
BOOST_CHECK(CNode::IsBanned(addr1)); // ... but 1 still should be
|
2013-11-17 16:25:17 -08:00
|
|
|
Misbehaving(dummyNode2.GetId(), 50);
|
2015-04-08 11:20:00 -07:00
|
|
|
SendMessages(params, &dummyNode2);
|
2012-01-03 14:33:31 -08:00
|
|
|
BOOST_CHECK(CNode::IsBanned(addr2));
|
2012-10-05 10:22:21 -07:00
|
|
|
}
|
2011-09-06 13:09:04 -07:00
|
|
|
|
|
|
|
BOOST_AUTO_TEST_CASE(DoS_banscore)
|
|
|
|
{
|
2020-07-10 10:34:53 -07:00
|
|
|
const Consensus::Params& params = Params().GetConsensus();
|
2011-09-06 13:09:04 -07:00
|
|
|
CNode::ClearBanned();
|
|
|
|
mapArgs["-banscore"] = "111"; // because 11 is my favorite number
|
2012-01-03 14:33:31 -08:00
|
|
|
CAddress addr1(ip(0xa0b0c001));
|
2012-02-12 04:45:24 -08:00
|
|
|
CNode dummyNode1(INVALID_SOCKET, addr1, "", true);
|
2013-11-17 16:25:17 -08:00
|
|
|
dummyNode1.nVersion = 1;
|
|
|
|
Misbehaving(dummyNode1.GetId(), 100);
|
2015-04-08 11:20:00 -07:00
|
|
|
SendMessages(params, &dummyNode1);
|
2012-01-03 14:33:31 -08:00
|
|
|
BOOST_CHECK(!CNode::IsBanned(addr1));
|
2013-11-17 16:25:17 -08:00
|
|
|
Misbehaving(dummyNode1.GetId(), 10);
|
2015-04-08 11:20:00 -07:00
|
|
|
SendMessages(params, &dummyNode1);
|
2012-01-03 14:33:31 -08:00
|
|
|
BOOST_CHECK(!CNode::IsBanned(addr1));
|
2013-11-17 16:25:17 -08:00
|
|
|
Misbehaving(dummyNode1.GetId(), 1);
|
2015-04-08 11:20:00 -07:00
|
|
|
SendMessages(params, &dummyNode1);
|
2012-01-03 14:33:31 -08:00
|
|
|
BOOST_CHECK(CNode::IsBanned(addr1));
|
2012-05-16 12:57:04 -07:00
|
|
|
mapArgs.erase("-banscore");
|
2011-09-06 13:09:04 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
BOOST_AUTO_TEST_CASE(DoS_bantime)
|
|
|
|
{
|
2022-06-28 16:18:36 -07:00
|
|
|
FixedClock::SetGlobal();
|
|
|
|
|
2020-07-10 10:34:53 -07:00
|
|
|
const Consensus::Params& params = Params().GetConsensus();
|
2011-09-06 13:09:04 -07:00
|
|
|
CNode::ClearBanned();
|
2022-06-28 16:18:36 -07:00
|
|
|
std::chrono::seconds nStartTime(GetTime());
|
|
|
|
FixedClock::Instance()->Set(nStartTime); // Overrides future calls to GetTime()
|
2011-09-06 13:09:04 -07:00
|
|
|
|
2012-01-03 14:33:31 -08:00
|
|
|
CAddress addr(ip(0xa0b0c001));
|
2012-02-12 04:45:24 -08:00
|
|
|
CNode dummyNode(INVALID_SOCKET, addr, "", true);
|
2013-11-17 16:25:17 -08:00
|
|
|
dummyNode.nVersion = 1;
|
2011-09-06 13:09:04 -07:00
|
|
|
|
2013-11-17 16:25:17 -08:00
|
|
|
Misbehaving(dummyNode.GetId(), 100);
|
2015-04-08 11:20:00 -07:00
|
|
|
SendMessages(params, &dummyNode);
|
2012-01-03 14:33:31 -08:00
|
|
|
BOOST_CHECK(CNode::IsBanned(addr));
|
2011-09-06 13:09:04 -07:00
|
|
|
|
2022-06-28 16:18:36 -07:00
|
|
|
FixedClock::Instance()->Set(nStartTime + std::chrono::seconds(60*60));
|
2012-01-03 14:33:31 -08:00
|
|
|
BOOST_CHECK(CNode::IsBanned(addr));
|
2011-09-06 13:09:04 -07:00
|
|
|
|
2022-06-28 16:18:36 -07:00
|
|
|
FixedClock::Instance()->Set(nStartTime + std::chrono::seconds(60*60*24+1));
|
2012-01-03 14:33:31 -08:00
|
|
|
BOOST_CHECK(!CNode::IsBanned(addr));
|
2022-06-28 16:18:36 -07:00
|
|
|
|
|
|
|
SystemClock::SetGlobal();
|
2012-01-03 14:33:31 -08:00
|
|
|
}
|
2011-09-06 13:09:04 -07:00
|
|
|
|
2012-02-29 07:14:18 -08:00
|
|
|
CTransaction RandomOrphan()
|
|
|
|
{
|
2014-09-20 00:53:50 -07:00
|
|
|
std::map<uint256, COrphanTx>::iterator it;
|
2017-06-07 12:03:17 -07:00
|
|
|
it = mapOrphanTransactions.lower_bound(InsecureRand256());
|
2012-02-29 07:14:18 -08:00
|
|
|
if (it == mapOrphanTransactions.end())
|
|
|
|
it = mapOrphanTransactions.begin();
|
2014-09-20 00:53:50 -07:00
|
|
|
return it->second.tx;
|
2012-02-29 07:14:18 -08:00
|
|
|
}
|
|
|
|
|
2018-04-09 16:17:30 -07:00
|
|
|
// Parameterized testing over consensus branch ids
|
|
|
|
BOOST_DATA_TEST_CASE(DoS_mapOrphans, boost::unit_test::data::xrange(static_cast<int>(Consensus::MAX_NETWORK_UPGRADES)))
|
2012-02-29 07:14:18 -08:00
|
|
|
{
|
2018-04-09 16:17:30 -07:00
|
|
|
uint32_t consensusBranchId = NetworkUpgradeInfo[sample].nBranchId;
|
2018-02-01 17:49:42 -08:00
|
|
|
|
2021-10-18 18:47:28 -07:00
|
|
|
CKey key = CKey::TestOnlyRandomKey(true);
|
2012-02-29 07:14:18 -08:00
|
|
|
CBasicKeyStore keystore;
|
|
|
|
keystore.AddKey(key);
|
|
|
|
|
|
|
|
// 50 orphan transactions:
|
|
|
|
for (int i = 0; i < 50; i++)
|
|
|
|
{
|
2014-06-07 04:53:27 -07:00
|
|
|
CMutableTransaction tx;
|
2012-02-29 07:14:18 -08:00
|
|
|
tx.vin.resize(1);
|
|
|
|
tx.vin[0].prevout.n = 0;
|
2017-06-07 12:03:17 -07:00
|
|
|
tx.vin[0].prevout.hash = InsecureRand256();
|
2012-02-29 07:14:18 -08:00
|
|
|
tx.vin[0].scriptSig << OP_1;
|
|
|
|
tx.vout.resize(1);
|
|
|
|
tx.vout[0].nValue = 1*CENT;
|
2014-09-11 10:15:29 -07:00
|
|
|
tx.vout[0].scriptPubKey = GetScriptForDestination(key.GetPubKey().GetID());
|
2012-02-29 07:14:18 -08:00
|
|
|
|
2014-08-28 10:23:24 -07:00
|
|
|
AddOrphanTx(tx, i);
|
2012-02-29 07:14:18 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
// ... and 50 that depend on other orphans:
|
|
|
|
for (int i = 0; i < 50; i++)
|
|
|
|
{
|
|
|
|
CTransaction txPrev = RandomOrphan();
|
|
|
|
|
2014-06-07 04:53:27 -07:00
|
|
|
CMutableTransaction tx;
|
2012-02-29 07:14:18 -08:00
|
|
|
tx.vin.resize(1);
|
|
|
|
tx.vin[0].prevout.n = 0;
|
2016-08-30 12:49:38 -07:00
|
|
|
tx.vin[0].prevout.hash = txPrev.GetHash();
|
2012-02-29 07:14:18 -08:00
|
|
|
tx.vout.resize(1);
|
|
|
|
tx.vout[0].nValue = 1*CENT;
|
2014-09-11 10:15:29 -07:00
|
|
|
tx.vout[0].scriptPubKey = GetScriptForDestination(key.GetPubKey().GetID());
|
2022-01-22 18:37:10 -08:00
|
|
|
const PrecomputedTransactionData txdata(tx, {txPrev.vout[0]});
|
2022-01-22 14:59:28 -08:00
|
|
|
SignSignature(keystore, txPrev, tx, txdata, 0, SIGHASH_ALL, consensusBranchId);
|
2012-02-29 07:14:18 -08:00
|
|
|
|
2014-08-28 10:23:24 -07:00
|
|
|
AddOrphanTx(tx, i);
|
2012-02-29 07:14:18 -08:00
|
|
|
}
|
|
|
|
|
2012-05-15 12:53:30 -07:00
|
|
|
// This really-big orphan should be ignored:
|
|
|
|
for (int i = 0; i < 10; i++)
|
|
|
|
{
|
|
|
|
CTransaction txPrev = RandomOrphan();
|
|
|
|
|
2014-06-07 04:53:27 -07:00
|
|
|
CMutableTransaction tx;
|
2012-05-15 12:53:30 -07:00
|
|
|
tx.vout.resize(1);
|
|
|
|
tx.vout[0].nValue = 1*CENT;
|
2014-09-11 10:15:29 -07:00
|
|
|
tx.vout[0].scriptPubKey = GetScriptForDestination(key.GetPubKey().GetID());
|
2012-05-15 12:53:30 -07:00
|
|
|
tx.vin.resize(500);
|
2012-05-24 09:18:50 -07:00
|
|
|
for (unsigned int j = 0; j < tx.vin.size(); j++)
|
2012-05-15 12:53:30 -07:00
|
|
|
{
|
|
|
|
tx.vin[j].prevout.n = j;
|
2016-08-30 12:49:38 -07:00
|
|
|
tx.vin[j].prevout.hash = txPrev.GetHash();
|
2012-05-15 12:53:30 -07:00
|
|
|
}
|
2022-01-22 18:37:10 -08:00
|
|
|
// Fake the coins being spent (the random orphan doesn't actually have them all).
|
|
|
|
std::vector<CTxOut> allPrevOutputs;
|
|
|
|
allPrevOutputs.resize(tx.vin.size(), txPrev.vout[0]);
|
|
|
|
const PrecomputedTransactionData txdata(tx, allPrevOutputs);
|
2022-01-22 14:59:28 -08:00
|
|
|
SignSignature(keystore, txPrev, tx, txdata, 0, SIGHASH_ALL, consensusBranchId);
|
2012-05-15 12:53:30 -07:00
|
|
|
// Re-use same signature for other inputs
|
|
|
|
// (they don't have to be valid for this test)
|
2012-05-24 09:18:50 -07:00
|
|
|
for (unsigned int j = 1; j < tx.vin.size(); j++)
|
2012-05-15 12:53:30 -07:00
|
|
|
tx.vin[j].scriptSig = tx.vin[0].scriptSig;
|
|
|
|
|
2014-08-28 10:23:24 -07:00
|
|
|
BOOST_CHECK(!AddOrphanTx(tx, i));
|
|
|
|
}
|
|
|
|
|
|
|
|
// Test EraseOrphansFor:
|
|
|
|
for (NodeId i = 0; i < 3; i++)
|
|
|
|
{
|
|
|
|
size_t sizeBefore = mapOrphanTransactions.size();
|
|
|
|
EraseOrphansFor(i);
|
|
|
|
BOOST_CHECK(mapOrphanTransactions.size() < sizeBefore);
|
2012-05-15 12:53:30 -07:00
|
|
|
}
|
|
|
|
|
2012-02-29 07:14:18 -08:00
|
|
|
// Test LimitOrphanTxSize() function:
|
|
|
|
LimitOrphanTxSize(40);
|
|
|
|
BOOST_CHECK(mapOrphanTransactions.size() <= 40);
|
|
|
|
LimitOrphanTxSize(10);
|
|
|
|
BOOST_CHECK(mapOrphanTransactions.size() <= 10);
|
|
|
|
LimitOrphanTxSize(0);
|
|
|
|
BOOST_CHECK(mapOrphanTransactions.empty());
|
|
|
|
BOOST_CHECK(mapOrphanTransactionsByPrev.empty());
|
|
|
|
}
|
|
|
|
|
2011-09-06 13:09:04 -07:00
|
|
|
BOOST_AUTO_TEST_SUITE_END()
|