|
| 1 | +// Copyright (c) 2023-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 | +#include <chainparams.h> |
| 6 | +#include <compat.h> |
| 7 | +#include <net.h> |
| 8 | +#include <net_processing.h> |
| 9 | +#include <netaddress.h> |
| 10 | +#include <netbase.h> |
| 11 | +#include <netgroup.h> |
| 12 | +#include <node/connection_types.h> |
| 13 | +#include <protocol.h> |
| 14 | +#include <random.h> |
| 15 | +#include <test/util/logging.h> |
| 16 | +#include <test/util/net.h> |
| 17 | +#include <test/util/setup_common.h> |
| 18 | +#include <tinyformat.h> |
| 19 | +#include <version.h> |
| 20 | + |
| 21 | +#include <algorithm> |
| 22 | +#include <cstdint> |
| 23 | +#include <memory> |
| 24 | +#include <optional> |
| 25 | +#include <string> |
| 26 | +#include <vector> |
| 27 | + |
| 28 | +#include <boost/test/unit_test.hpp> |
| 29 | + |
| 30 | +struct LogIPsTestingSetup : public TestingSetup { |
| 31 | + LogIPsTestingSetup() |
| 32 | + : TestingSetup{CBaseChainParams::MAIN, /*extra_args=*/{"-logips"}} {} |
| 33 | +}; |
| 34 | + |
| 35 | +BOOST_FIXTURE_TEST_SUITE(net_peer_connection_tests, LogIPsTestingSetup) |
| 36 | + |
| 37 | +static CService ip(uint32_t i) |
| 38 | +{ |
| 39 | + struct in_addr s; |
| 40 | + s.s_addr = i; |
| 41 | + return CService{CNetAddr{s}, Params().GetDefaultPort()}; |
| 42 | +} |
| 43 | + |
| 44 | +/** Create a peer and connect to it. If the optional `address` (IP/CJDNS only) isn't passed, a random address is created. */ |
| 45 | +static void AddPeer(NodeId& id, std::vector<CNode*>& nodes, PeerManager& peerman, ConnmanTestMsg& connman, ConnectionType conn_type, bool onion_peer = false, std::optional<std::string> address = std::nullopt) |
| 46 | +{ |
| 47 | + CAddress addr{}; |
| 48 | + |
| 49 | + if (address.has_value()) { |
| 50 | + addr = CAddress{MaybeFlipIPv6toCJDNS(LookupNumeric(address.value(), Params().GetDefaultPort())), NODE_NONE}; |
| 51 | + } else if (onion_peer) { |
| 52 | + auto tor_addr{g_insecure_rand_ctx.randbytes(ADDR_TORV3_SIZE)}; |
| 53 | + BOOST_REQUIRE(addr.SetSpecial(OnionToString(tor_addr))); |
| 54 | + } |
| 55 | + |
| 56 | + while (!addr.IsLocal() && !addr.IsRoutable()) { |
| 57 | + addr = CAddress{ip(g_insecure_rand_ctx.randbits(32)), NODE_NONE}; |
| 58 | + } |
| 59 | + |
| 60 | + BOOST_REQUIRE(addr.IsValid()); |
| 61 | + |
| 62 | + const bool inbound_onion{onion_peer && conn_type == ConnectionType::INBOUND}; |
| 63 | + |
| 64 | + nodes.emplace_back(new CNode{++id, |
| 65 | + /*sock=*/nullptr, |
| 66 | + addr, |
| 67 | + /*nKeyedNetGroupIn=*/0, |
| 68 | + /*nLocalHostNonceIn=*/0, |
| 69 | + CAddress{}, |
| 70 | + /*addrNameIn=*/"", |
| 71 | + conn_type, |
| 72 | + /*inbound_onion=*/inbound_onion}); |
| 73 | + CNode& node = *nodes.back(); |
| 74 | + node.SetCommonVersion(PROTOCOL_VERSION); |
| 75 | + |
| 76 | + peerman.InitializeNode(node, ServiceFlags(NODE_NETWORK)); |
| 77 | + node.fSuccessfullyConnected = true; |
| 78 | + |
| 79 | + connman.AddTestNode(node); |
| 80 | +} |
| 81 | + |
| 82 | +BOOST_AUTO_TEST_CASE(test_addnode_getaddednodeinfo_and_connection_detection) |
| 83 | +{ |
| 84 | + const auto& chainparams = Params(); |
| 85 | + auto connman = std::make_unique<ConnmanTestMsg>(0x1337, 0x1337, *m_node.addrman, *m_node.netgroupman); |
| 86 | + auto peerman = PeerManager::make(chainparams, *connman, *m_node.addrman, nullptr, |
| 87 | + *m_node.chainman, *m_node.mempool, *m_node.mn_metaman, *m_node.mn_sync, |
| 88 | + *m_node.govman, *m_node.sporkman, /* mn_activeman = */ nullptr, m_node.dmnman, |
| 89 | + m_node.cj_ctx, m_node.llmq_ctx, /* ignore_incoming_txs = */ false); |
| 90 | + NodeId id{0}; |
| 91 | + std::vector<CNode*> nodes; |
| 92 | + |
| 93 | + // Connect a localhost peer. |
| 94 | + { |
| 95 | + ASSERT_DEBUG_LOG("Added connection to 127.0.0.1:9999 peer=1"); |
| 96 | + AddPeer(id, nodes, *peerman, *connman, ConnectionType::MANUAL, /*onion_peer=*/false, /*address=*/"127.0.0.1"); |
| 97 | + BOOST_REQUIRE(nodes.back() != nullptr); |
| 98 | + } |
| 99 | + |
| 100 | + // Call ConnectNode(), which is also called by RPC addnode onetry, for a localhost |
| 101 | + // address that resolves to multiple IPs, including that of the connected peer. |
| 102 | + // The connection attempt should consistently fail due to the check in ConnectNode(). |
| 103 | + for (int i = 0; i < 10; ++i) { |
| 104 | + ASSERT_DEBUG_LOG("Not opening a connection to localhost, already connected to 127.0.0.1:9999"); |
| 105 | + BOOST_CHECK(!connman->ConnectNodePublic(*peerman, "localhost", ConnectionType::MANUAL)); |
| 106 | + } |
| 107 | + |
| 108 | + // Add 3 more peer connections. |
| 109 | + AddPeer(id, nodes, *peerman, *connman, ConnectionType::OUTBOUND_FULL_RELAY); |
| 110 | + AddPeer(id, nodes, *peerman, *connman, ConnectionType::BLOCK_RELAY, /*onion_peer=*/true); |
| 111 | + AddPeer(id, nodes, *peerman, *connman, ConnectionType::INBOUND); |
| 112 | + |
| 113 | + BOOST_TEST_MESSAGE("Call AddNode() for all the peers"); |
| 114 | + for (auto node : connman->TestNodes()) { |
| 115 | + BOOST_CHECK(connman->AddNode({/*m_added_node=*/node->addr.ToStringAddrPort(), /*m_use_v2transport=*/true})); |
| 116 | + BOOST_TEST_MESSAGE(strprintf("peer id=%s addr=%s", node->GetId(), node->addr.ToStringAddrPort())); |
| 117 | + } |
| 118 | + |
| 119 | + BOOST_TEST_MESSAGE("\nCall AddNode() with 2 addrs resolving to existing localhost addnode entry; neither should be added"); |
| 120 | + BOOST_CHECK(!connman->AddNode({/*m_added_node=*/"127.0.0.1", /*m_use_v2transport=*/true})); |
| 121 | + BOOST_CHECK(!connman->AddNode({/*m_added_node=*/"127.1", /*m_use_v2transport=*/true})); |
| 122 | + |
| 123 | + BOOST_TEST_MESSAGE("\nExpect GetAddedNodeInfo to return expected number of peers with `include_connected` true/false"); |
| 124 | + BOOST_CHECK_EQUAL(connman->GetAddedNodeInfo(/*include_connected=*/true).size(), nodes.size()); |
| 125 | + BOOST_CHECK(connman->GetAddedNodeInfo(/*include_connected=*/false).empty()); |
| 126 | + |
| 127 | + BOOST_TEST_MESSAGE("\nPrint GetAddedNodeInfo contents:"); |
| 128 | + for (const auto& info : connman->GetAddedNodeInfo(/*include_connected=*/true)) { |
| 129 | + BOOST_TEST_MESSAGE(strprintf("\nadded node: %s", info.m_params.m_added_node)); |
| 130 | + BOOST_TEST_MESSAGE(strprintf("connected: %s", info.fConnected)); |
| 131 | + if (info.fConnected) { |
| 132 | + BOOST_TEST_MESSAGE(strprintf("IP address: %s", info.resolvedAddress.ToStringAddrPort())); |
| 133 | + BOOST_TEST_MESSAGE(strprintf("direction: %s", info.fInbound ? "inbound" : "outbound")); |
| 134 | + } |
| 135 | + } |
| 136 | + |
| 137 | + BOOST_TEST_MESSAGE("\nCheck that all connected peers are correctly detected as connected"); |
| 138 | + for (auto node : connman->TestNodes()) { |
| 139 | + BOOST_CHECK(connman->AlreadyConnectedPublic(node->addr)); |
| 140 | + } |
| 141 | + |
| 142 | + // Clean up |
| 143 | + for (auto node : connman->TestNodes()) { |
| 144 | + peerman->FinalizeNode(*node); |
| 145 | + } |
| 146 | + connman->ClearTestNodes(); |
| 147 | +} |
| 148 | + |
| 149 | +BOOST_AUTO_TEST_SUITE_END() |
0 commit comments