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.

5,758 bytes · SHA-256 2309e65bc57c2fd6… · View upstream ↗

1#!/usr/bin/env python3
2# Copyright (c) 2024-present The Bitcoin Core developers
3# Distributed under the MIT software license, see the accompanying
4# file COPYING or http://www.opensource.org/licenses/mit-license.php.
5"""
6Test P2P behaviour during the handshake phase (VERSION, VERACK messages).
7"""
8import itertools
9import time
10
11from test_framework.test_framework import BitcoinTestFramework
12from test_framework.util import assert_equal, assert_not_equal
13from test_framework.messages import (
14    NODE_NETWORK,
15    NODE_NETWORK_LIMITED,
16    NODE_NONE,
17    NODE_P2P_V2,
18    NODE_WITNESS,
19    msg_verack,
20)
21from test_framework.p2p import P2PInterface
22from test_framework.util import p2p_port
23
24
25# Desirable service flags for outbound non-pruned and pruned peers. Note that
26# the desirable service flags for pruned peers are dynamic and only apply if
27#  1. the peer's service flag NODE_NETWORK_LIMITED is set *and*
28#  2. the local chain is close to the tip (<24h)
29DESIRABLE_SERVICE_FLAGS_FULL = NODE_NETWORK | NODE_WITNESS
30DESIRABLE_SERVICE_FLAGS_PRUNED = NODE_NETWORK_LIMITED | NODE_WITNESS
31
32
33class P2PHandshakeTest(BitcoinTestFramework):
34    def set_test_params(self):
35        self.num_nodes = 1
36
37    def add_outbound_connection(self, node, connection_type, services, wait_for_disconnect):
38        peer = node.add_outbound_p2p_connection(
39            P2PInterface(), p2p_idx=0, wait_for_disconnect=wait_for_disconnect,
40            connection_type=connection_type, services=services,
41            supports_v2_p2p=self.options.v2transport, advertise_v2_p2p=self.options.v2transport)
42        if not wait_for_disconnect:
43            # check that connection is alive past the version handshake and disconnect manually
44            peer.sync_with_ping()
45            peer.peer_disconnect()
46            peer.wait_for_disconnect()
47        self.wait_until(lambda: len(node.getpeerinfo()) == 0)
48
49    def test_desirable_service_flags(self, node, service_flag_tests, desirable_service_flags, expect_disconnect):
50        """Check that connecting to a peer either fails or succeeds depending on its offered
51           service flags in the VERSION message. The test is exercised for all relevant
52           outbound connection types where the desirable service flags check is done."""
53        CONNECTION_TYPES = ["outbound-full-relay", "block-relay-only", "addr-fetch"]
54        for conn_type, services in itertools.product(CONNECTION_TYPES, service_flag_tests):
55            if self.options.v2transport:
56                services |= NODE_P2P_V2
57            expected_result = "disconnect" if expect_disconnect else "connect"
58            self.log.info(f'    - services 0x{services:08x}, type "{conn_type}" [{expected_result}]')
59            if expect_disconnect:
60                assert_not_equal((services & desirable_service_flags), desirable_service_flags)
61                expected_debug_log = f'does not offer the expected services ' \
62                        f'({services:08x} offered, {desirable_service_flags:08x} expected)'
63                with node.assert_debug_log([expected_debug_log]):
64                    self.add_outbound_connection(node, conn_type, services, wait_for_disconnect=True)
65            else:
66                assert_equal((services & desirable_service_flags), desirable_service_flags)
67                self.add_outbound_connection(node, conn_type, services, wait_for_disconnect=False)
68
69    def generate_at_mocktime(self, time):
70        self.nodes[0].setmocktime(time)
71        self.generate(self.nodes[0], 1)
72        self.nodes[0].setmocktime(0)
73
74    def run_test(self):
75        node = self.nodes[0]
76
77        self.log.info("Check that redundant verack message is ignored")
78        verack_conn = node.add_p2p_connection(P2PInterface())
79        with node.assert_debug_log(["ignoring redundant verack message"]):
80            verack_conn.send_and_ping(msg_verack())
81        node.disconnect_p2ps()
82
83        self.log.info("Check that lacking desired service flags leads to disconnect (non-pruned peers)")
84        self.test_desirable_service_flags(node, [NODE_NONE, NODE_NETWORK, NODE_WITNESS],
85                                          DESIRABLE_SERVICE_FLAGS_FULL, expect_disconnect=True)
86        self.test_desirable_service_flags(node, [NODE_NETWORK | NODE_WITNESS],
87                                          DESIRABLE_SERVICE_FLAGS_FULL, expect_disconnect=False)
88
89        self.log.info("Check that limited peers are only desired if the local chain is close to the tip (<24h)")
90        self.generate_at_mocktime(int(time.time()) - 25 * 3600)  # tip outside the 24h window, should fail
91        self.test_desirable_service_flags(node, [NODE_NETWORK_LIMITED | NODE_WITNESS],
92                                          DESIRABLE_SERVICE_FLAGS_FULL, expect_disconnect=True)
93        self.generate_at_mocktime(int(time.time()) - 23 * 3600)  # tip inside the 24h window, should succeed
94        self.test_desirable_service_flags(node, [NODE_NETWORK_LIMITED | NODE_WITNESS],
95                                          DESIRABLE_SERVICE_FLAGS_PRUNED, expect_disconnect=False)
96
97        self.log.info("Check that feeler connections get disconnected immediately")
98        with node.assert_debug_log(["feeler connection completed"]):
99            self.add_outbound_connection(node, "feeler", NODE_NONE, wait_for_disconnect=True)
100
101        self.log.info("Check that connecting to ourself leads to immediate disconnect")
102        with node.assert_debug_log(["connected to self", "disconnecting"]):
103            node_listen_addr = f"127.0.0.1:{p2p_port(0)}"
104            node.addconnection(node_listen_addr, "outbound-full-relay", self.options.v2transport)
105            self.wait_until(lambda: len(node.getpeerinfo()) == 0)
106
107
108if __name__ == '__main__':
109    P2PHandshakeTest(__file__).main()
110

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.