READ THE CODE

Inside Bitcoin

The code behind the network, with room to understand it.

Stable release v32.1bitcoin/bitcoin · f0de87f04062 ↗3,020 source files · 47.5 MB stored on the server440 file guides

Release published 2026-10-28 · Snapshot fetched 2026-11-01. Latest stable release at fetch time; development branches and Git history are excluded. Only the selected code window and visible folders load here.

18,557 bytes · SHA-256 59acf297552bf0dd… · View upstream ↗

1// Copyright (c) 2011-present The Bitcoin Core developers
2// Distributed under the MIT software license, see the accompanying
3// file COPYING or http://www.opensource.org/licenses/mit-license.php.
4
5// Unit tests for denial-of-service detection/prevention code
6
7#include <banman.h>
8#include <chainparams.h>
9#include <common/args.h>
10#include <net.h>
11#include <net_processing.h>
12#include <pubkey.h>
13#include <script/sign.h>
14#include <script/signingprovider.h>
15#include <serialize.h>
16#include <test/util/net.h>
17#include <test/util/random.h>
18#include <test/util/setup_common.h>
19#include <test/util/time.h>
20#include <util/string.h>
21#include <util/time.h>
22#include <validation.h>
23
24#include <array>
25#include <cstdint>
26
27#include <boost/test/unit_test.hpp>
28
29static CService ip(uint32_t i)
30{
31    struct in_addr s;
32    s.s_addr = i;
33    return CService(CNetAddr(s), Params().GetDefaultPort());
34}
35
36BOOST_FIXTURE_TEST_SUITE(denialofservice_tests, TestingSetup)
37
38// Test eviction of an outbound peer whose chain never advances
39// Mock a node connection, and use mocktime to simulate a peer
40// which never sends any headers messages.  PeerLogic should
41// decide to evict that outbound peer, after the appropriate timeouts.
42// Note that we protect 4 outbound nodes from being subject to
43// this logic; this test takes advantage of that protection only
44// being applied to nodes which send headers with sufficient
45// work.
46BOOST_AUTO_TEST_CASE(outbound_slow_chain_eviction)
47{
48    LOCK(NetEventsInterface::g_msgproc_mutex);
49
50    ConnmanTestMsg& connman = static_cast<ConnmanTestMsg&>(*m_node.connman);
51    // Disable inactivity checks for this test to avoid interference
52    connman.SetPeerConnectTimeout(99999s);
53    PeerManager& peerman = *m_node.peerman;
54
55    // Mock an outbound peer
56    CAddress addr1(ip(0xa0b0c001), NODE_NONE);
57    NodeId id{0};
58    CNode dummyNode1{id++,
59                     /*sock=*/nullptr,
60                     addr1,
61                     /*nKeyedNetGroupIn=*/0,
62                     /*nLocalHostNonceIn=*/0,
63                     CAddress(),
64                     /*addrNameIn=*/"",
65                     ConnectionType::OUTBOUND_FULL_RELAY,
66                     /*inbound_onion=*/false,
67                     /*network_key=*/0};
68
69    connman.Handshake(
70        /*node=*/dummyNode1,
71        /*successfully_connected=*/true,
72        /*remote_services=*/ServiceFlags(NODE_NETWORK | NODE_WITNESS),
73        /*local_services=*/ServiceFlags(NODE_NETWORK | NODE_WITNESS),
74        /*version=*/PROTOCOL_VERSION,
75        /*relay_txs=*/true);
76
77    // This test requires that we have a chain with non-zero work.
78    {
79        LOCK(cs_main);
80        BOOST_CHECK(m_node.chainman->ActiveChain().Tip() != nullptr);
81        BOOST_CHECK(m_node.chainman->ActiveChain().Tip()->nChainWork > 0);
82    }
83
84    // Test starts here
85    BOOST_CHECK(peerman.SendMessages(dummyNode1)); // should result in getheaders
86
87    {
88        LOCK(dummyNode1.cs_vSend);
89        const auto& [to_send, _more, _msg_type] = dummyNode1.m_transport->GetBytesToSend(false);
90        BOOST_CHECK(!to_send.empty());
91    }
92    connman.FlushSendBuffer(dummyNode1);
93
94    FakeNodeClock clock{};
95    clock += 21min;
96
97    BOOST_CHECK(peerman.SendMessages(dummyNode1)); // should result in getheaders
98    {
99        LOCK(dummyNode1.cs_vSend);
100        const auto& [to_send, _more, _msg_type] = dummyNode1.m_transport->GetBytesToSend(false);
101        BOOST_CHECK(!to_send.empty());
102    }
103
104    clock += 3min;
105    BOOST_CHECK(peerman.SendMessages(dummyNode1)); // should result in disconnect
106    BOOST_CHECK(dummyNode1.fDisconnect == true);
107
108    peerman.FinalizeNode(dummyNode1);
109}
110
111struct OutboundTest : TestingSetup {
112void AddRandomOutboundPeer(NodeId& id, std::vector<CNode*>& vNodes, PeerManager& peerLogic, ConnmanTestMsg& connman, ConnectionType connType, bool onion_peer = false)
113{
114    CAddress addr;
115
116    if (onion_peer) {
117        auto tor_addr{m_rng.randbytes(ADDR_TORV3_SIZE)};
118        BOOST_REQUIRE(addr.SetSpecial(OnionToString(tor_addr)));
119    }
120
121    while (!addr.IsRoutable()) {
122        addr = CAddress(ip(m_rng.randbits(32)), NODE_NONE);
123    }
124
125    vNodes.emplace_back(new CNode{id++,
126                                  /*sock=*/nullptr,
127                                  addr,
128                                  /*nKeyedNetGroupIn=*/0,
129                                  /*nLocalHostNonceIn=*/0,
130                                  CAddress(),
131                                  /*addrNameIn=*/"",
132                                  connType,
133                                  /*inbound_onion=*/false,
134                                  /*network_key=*/0});
135    CNode &node = *vNodes.back();
136    node.SetCommonVersion(PROTOCOL_VERSION);
137
138    peerLogic.InitializeNode(node, ServiceFlags(NODE_NETWORK | NODE_WITNESS));
139    node.fSuccessfullyConnected = true;
140
141    connman.AddTestNode(node);
142}
143}; // struct OutboundTest
144
145BOOST_FIXTURE_TEST_CASE(stale_tip_peer_management, OutboundTest)
146{
147    NodeId id{0};
148    auto connman = std::make_unique<ConnmanTestMsg>(0x1337, 0x1337, *m_node.addrman, *m_node.netgroupman, Params());
149    auto peerLogic = PeerManager::make(*connman, *m_node.addrman, nullptr, *m_node.chainman, *m_node.mempool, *m_node.warnings, {});
150
151    constexpr int max_outbound_full_relay = MAX_OUTBOUND_FULL_RELAY_CONNECTIONS;
152    CConnman::Options options;
153    options.m_max_automatic_connections = DEFAULT_MAX_PEER_CONNECTIONS;
154
155    const auto time_init{Now<NodeSeconds>()};
156    FakeNodeClock clock{time_init};
157    const auto delta{3 * std::chrono::seconds{m_node.chainman->GetConsensus().nPowTargetSpacing} + 1s};
158    connman->Init(options);
159    std::vector<CNode *> vNodes;
160
161    // Mock some outbound peers
162    for (int i = 0; i < max_outbound_full_relay; ++i) {
163        AddRandomOutboundPeer(id, vNodes, *peerLogic, *connman, ConnectionType::OUTBOUND_FULL_RELAY);
164    }
165
166    peerLogic->CheckForStaleTipAndEvictPeers();
167
168    // No nodes should be marked for disconnection while we have no extra peers
169    for (const CNode *node : vNodes) {
170        BOOST_CHECK(node->fDisconnect == false);
171    }
172
173    clock += delta;
174
175    // Now tip should definitely be stale, and we should look for an extra
176    // outbound peer
177    peerLogic->CheckForStaleTipAndEvictPeers();
178    BOOST_CHECK(connman->GetTryNewOutboundPeer());
179
180    // Still no peers should be marked for disconnection
181    for (const CNode *node : vNodes) {
182        BOOST_CHECK(node->fDisconnect == false);
183    }
184
185    // If we add one more peer, something should get marked for eviction
186    // on the next check (since we're mocking the time to be in the future, the
187    // required time connected check should be satisfied).
188    clock.set(time_init);
189    AddRandomOutboundPeer(id, vNodes, *peerLogic, *connman, ConnectionType::OUTBOUND_FULL_RELAY);
190    clock += delta;
191
192    peerLogic->CheckForStaleTipAndEvictPeers();
193    for (int i = 0; i < max_outbound_full_relay; ++i) {
194        BOOST_CHECK(vNodes[i]->fDisconnect == false);
195    }
196    // Last added node should get marked for eviction
197    BOOST_CHECK(vNodes.back()->fDisconnect == true);
198
199    vNodes.back()->fDisconnect = false;
200
201    // Update the last announced block time for the last
202    // peer, and check that the next newest node gets evicted.
203    peerLogic->UpdateLastBlockAnnounceTime(vNodes.back()->GetId(), GetTime());
204
205    peerLogic->CheckForStaleTipAndEvictPeers();
206    for (int i = 0; i < max_outbound_full_relay - 1; ++i) {
207        BOOST_CHECK(vNodes[i]->fDisconnect == false);
208    }
209    BOOST_CHECK(vNodes[max_outbound_full_relay-1]->fDisconnect == true);
210    BOOST_CHECK(vNodes.back()->fDisconnect == false);
211
212    vNodes[max_outbound_full_relay - 1]->fDisconnect = false;
213
214    // Add an onion peer, that will be protected because it is the only one for
215    // its network, so another peer gets disconnected instead.
216    clock.set(time_init);
217    AddRandomOutboundPeer(id, vNodes, *peerLogic, *connman, ConnectionType::OUTBOUND_FULL_RELAY, /*onion_peer=*/true);
218    clock += delta;
219    peerLogic->CheckForStaleTipAndEvictPeers();
220
221    for (int i = 0; i < max_outbound_full_relay - 2; ++i) {
222        BOOST_CHECK(vNodes[i]->fDisconnect == false);
223    }
224    BOOST_CHECK(vNodes[max_outbound_full_relay - 2]->fDisconnect == false);
225    BOOST_CHECK(vNodes[max_outbound_full_relay - 1]->fDisconnect == true);
226    BOOST_CHECK(vNodes[max_outbound_full_relay]->fDisconnect == false);
227
228    // Add a second onion peer which won't be protected
229    clock.set(time_init);
230    AddRandomOutboundPeer(id, vNodes, *peerLogic, *connman, ConnectionType::OUTBOUND_FULL_RELAY, /*onion_peer=*/true);
231    clock += delta;
232    peerLogic->CheckForStaleTipAndEvictPeers();
233
234    BOOST_CHECK(vNodes.back()->fDisconnect == true);
235
236    for (const CNode *node : vNodes) {
237        peerLogic->FinalizeNode(*node);
238    }
239
240    connman->ClearTestNodes();
241}
242
243BOOST_FIXTURE_TEST_CASE(block_relay_only_eviction, OutboundTest)
244{
245    NodeId id{0};
246    FakeNodeClock clock{};
247    auto connman = std::make_unique<ConnmanTestMsg>(0x1337, 0x1337, *m_node.addrman, *m_node.netgroupman, Params());
248    auto peerLogic = PeerManager::make(*connman, *m_node.addrman, nullptr, *m_node.chainman, *m_node.mempool, *m_node.warnings, {});
249
250    constexpr int max_outbound_block_relay{MAX_BLOCK_RELAY_ONLY_CONNECTIONS};

src/wallet/receive.cpp

src/wallet/coinselection.cpp

src/consensus/validation.h

src/primitives/transaction.cpp

src/script/descriptor.cpp

src/bitcoin-cli.cpp

src/rpc/blockchain.cpp

src/rpc/rawtransaction.cpp

src/init.cpp

src/crypto/sha256.cpp

src/consensus/merkle.h

src/consensus/tx_check.h

src/consensus/tx_verify.h

src/policy/feerate.h

src/policy/feerate.cpp

src/primitives/transaction_identifier.h

src/chainparamsbase.h

src/chainparamsbase.cpp

src/util/moneystr.cpp

src/crypto/ripemd160.h

src/crypto/sha1.h

src/crypto/sha512.h

src/crypto/hmac_sha256.h

src/crypto/hmac_sha512.h

src/crypto/hmac_sha256.cpp

src/crypto/hmac_sha512.cpp

src/crypto/hkdf_sha256_32.h

src/crypto/hkdf_sha256_32.cpp

src/support/cleanse.h

src/support/cleanse.cpp

src/crypto/siphash.cpp

src/crypto/muhash.h

src/crypto/chacha20.h

src/crypto/poly1305.h

src/crypto/chacha20poly1305.h

src/crypto/chacha20poly1305.cpp

src/util/chaintype.cpp

src/util/obfuscation.h

src/util/byte_units.h

src/util/overflow.h

src/blockencodings.cpp

src/net_permissions.cpp

src/policy/packages.cpp

src/policy/rbf.cpp

src/policy/truc_policy.cpp

src/policy/ephemeral_policy.cpp

src/policy/settings.cpp

src/policy/policy.cpp

src/node/transaction.cpp

src/node/coin.cpp

src/rpc/client.cpp

src/rpc/external_signer.cpp

src/rpc/request.cpp

src/rpc/server_util.cpp

src/rpc/fees.cpp

src/rpc/output_script.cpp

src/rpc/rawtransaction_util.cpp

src/rpc/rawtransaction_util.h

src/rpc/register.h

src/rpc/request.h

src/rpc/server_util.h

src/rpc/protocol.h

src/rpc/server.h

src/rpc/signmessage.cpp

src/rpc/txoutproof.cpp

src/rpc/util.cpp

src/rpc/util.h

src/wallet/rpc/addresses.cpp

src/wallet/rpc/backup.cpp

src/wallet/rpc/coins.cpp

src/wallet/rpc/encrypt.cpp

src/wallet/rpc/signmessage.cpp

src/wallet/rpc/util.cpp

src/wallet/rpc/util.h

src/wallet/coincontrol.cpp

src/wallet/coincontrol.h

src/wallet/context.cpp

src/wallet/context.h

src/wallet/crypter.cpp

src/wallet/crypter.h

src/wallet/db.cpp

src/wallet/export.cpp

src/wallet/external_signer_scriptpubkeyman.cpp

src/wallet/rpc/spend.cpp

src/wallet/rpc/transactions.cpp

src/wallet/rpc/wallet.cpp

src/wallet/feebumper.cpp

src/wallet/feebumper.h

src/index/base.h

src/index/base.cpp

src/index/txindex.h

src/index/txindex.cpp

src/index/txindex_key.h

src/index/txospenderindex.h

src/index/txospenderindex.cpp

src/index/blockfilterindex.h

src/index/blockfilterindex.cpp

src/index/coinstatsindex.h

src/index/coinstatsindex.cpp

src/index/db_key.h

src/index/disktxpos.h

src/kernel/chain.cpp

src/kernel/chain.h

src/kernel/checks.cpp

src/kernel/checks.h

src/kernel/context.cpp

src/kernel/context.h

src/kernel/cs_main.cpp

src/kernel/cs_main.h

src/kernel/disconnected_transactions.cpp

src/kernel/disconnected_transactions.h

src/kernel/mempool_removal_reason.cpp

src/kernel/mempool_removal_reason.h

src/kernel/coinstats.cpp

src/kernel/coinstats.h

src/kernel/chainparams.cpp

src/kernel/chainparams.h

src/wallet/sqlite.cpp

src/wallet/sqlite.h

src/wallet/transaction.cpp

src/wallet/transaction.h

src/wallet/dump.cpp

src/wallet/dump.h

src/wallet/init.cpp

src/wallet/walletutil.cpp

src/wallet/walletutil.h

src/wallet/wallettool.cpp

src/wallet/wallettool.h

src/wallet/export.h

src/wallet/migrate.cpp

src/wallet/migrate.h

src/wallet/db.h

src/wallet/walletdb.cpp

src/wallet/interfaces.cpp

src/wallet/scriptpubkeyman.cpp

src/wallet/scriptpubkeyman.h

src/kernel/bitcoinkernel.cpp

src/kernel/bitcoinkernel.h

src/kernel/bitcoinkernel_wrapper.h

src/node/chainstate.cpp

src/node/chainstate.h

src/node/blockstorage.cpp

src/node/blockstorage.h

src/node/mempool_persist.cpp

src/node/mempool_persist.h

src/node/utxo_snapshot.cpp

src/node/utxo_snapshot.h

src/node/kernel_notifications.cpp

src/node/kernel_notifications.h

src/node/blockmanager_args.cpp

src/kernel/blockmanager_opts.h

src/util/expected.cpp

src/policy/settings.h

src/util/types.h

src/kernel/messagestartchars.h

src/kernel/warning.h

src/util/exception.h

src/common/license_info.h

src/node/coins_view_args.h

src/node/database_args.h

src/util/batchpriority.h

src/util/ui_change_type.h

src/node/coins_view_args.cpp

src/node/peerman_args.h

src/util/serfloat.h

src/node/blockmanager_args.h

src/node/database_args.cpp

src/util/overloaded.h

src/interfaces/types.h

src/node/chainstatemanager_args.h

src/node/minisketchwrapper.h

src/util/macros.h

src/util/syserror.h

src/common/run_command.h

src/util/bytevectorhash.cpp

src/node/mempool_persist_args.cpp

src/util/exec.h

src/util/rbf.h

src/crypto/hex_base.h

src/node/mempool_persist_args.h

src/interfaces/echo.h

src/util/any.h

src/util/batchpriority.cpp

src/util/threadnames.h

src/node/coin.h

src/node/mempool_args.h

src/util/asmap.h

src/util/bytevectorhash.h

src/node/context.cpp

src/util/insert.h

src/util/readwritefile.h

src/node/peerman_args.cpp

src/node/interface_ui.cpp

src/interfaces/handler.h

src/node/mining_args.h

src/util/hasher.cpp

src/common/init.h

src/kernel/mempool_limits.h

src/common/system.h

src/crypto/sha3.h

src/util/threadinterrupt.cpp

src/node/protocol_version.h

src/util/exception.cpp

src/common/messages.h

src/node/connection_types.cpp

src/util/bip32.h

src/common/license_info.cpp

src/util/stdmutex.h

src/common/run_command.cpp

src/util/check.cpp

src/util/golombrice.h

src/node/types.h

src/node/timeoffsets.h

src/util/readwritefile.cpp

src/util/fastrange.h

src/util/signalinterrupt.h

src/util/task_runner.h

src/util/syserror.cpp

src/common/interfaces.cpp

src/common/types.h

src/util/threadinterrupt.h

src/kernel/chainstatemanager_opts.h

src/node/psbt.h

src/policy/ephemeral_policy.h

src/util/trace.h

src/common/url.cpp

src/kernel/mempool_options.h

src/util/serfloat.cpp

src/kernel/notifications_interface.h

src/node/timeoffsets.cpp

src/node/warnings.cpp

src/util/hash_type.h

src/crypto/hex_base.cpp

src/node/eviction.h

src/util/tokenbucket.h

src/crypto/aes.h

src/common/pcp.h

src/util/feefrac.cpp

src/node/transaction.h

src/util/threadnames.cpp

src/common/signmessage.h

src/util/exec.cpp

src/util/signalinterrupt.cpp

src/clientversion.cpp

src/common/signmessage.cpp

src/compat/assumptions.h

src/compat/byteswap.h

src/deploymentinfo.cpp

src/deploymentinfo.h

src/deploymentstatus.cpp

src/deploymentstatus.h

src/dummywallet.cpp

src/external_signer.h

src/init/bitcoin-gui.cpp

src/init/bitcoin-node.cpp

src/init/bitcoin-qt.cpp

src/init/bitcoin-wallet.cpp

src/init/bitcoind.cpp

src/init/common.h

src/interfaces/init.h

src/interfaces/ipc.h

src/key_io.h

src/logging/categories.h

src/logging/timer.h

src/mapport.h

src/musig.h

src/net_permissions.h

src/netgroup.h

src/netmessagemaker.h

src/node/chainstatemanager_args.cpp

src/node/connection_types.h

src/node/context.h

src/node/interface_ui.h

src/node/mempool_args.cpp

src/node/mining_args.cpp

src/node/minisketchwrapper.cpp

src/node/txreconciliation.h

src/policy/fees/estimator_args.cpp

src/policy/fees/estimator_args.h

src/policy/fees/estimator_man.cpp

src/policy/fees/estimator_man.h

src/policy/packages.h

src/policy/rbf.h

src/policy/truc_policy.h

src/pow.h

src/primitives/block.cpp

src/randomenv.h

src/rest.h

src/script/keyorigin.h

src/script/parsing.cpp

src/script/parsing.h

src/script/script_error.h

src/script/sigcache.cpp

src/script/sigcache.h

src/script/solver.h

src/script/verify_flags.h

src/semaphore_grant.h

src/signet.h

src/span.h

src/support/allocators/secure.h

src/support/allocators/zeroafterfree.h

src/util/tokenpipe.cpp

src/util/translation.h

src/util/vector.h

src/versionbits.h

src/versionbits_impl.h

src/wallet/external_signer_scriptpubkeyman.h

src/wallet/fees.h

src/wallet/load.h

src/wallet/receive.h

src/wallet/rpc/wallet.h

src/wallet/types.h

src/walletinitinterface.h

src/zmq/zmqabstractnotifier.cpp

Independent explanations by MyBitcoinExplorer. Not affiliated with or endorsed by Bitcoin Core. Guides are educational and do not replace the source or its tests.

Bitcoin Core source © its contributors, distributed under the MIT license. Original notices and bundled third-party licenses are retained in the source tree.