Line data    Source code

       1             : // Copyright (c) 2020-2022 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 <test/util/net.h>
       6             : 
       7             : #include <chainparams.h>
       8             : #include <node/eviction.h>
       9             : #include <net.h>
      10             : #include <net_processing.h>
      11             : #include <netmessagemaker.h>
      12             : #include <span.h>
      13             : 
      14             : #include <vector>
      15             : 
      16           0 : void ConnmanTestMsg::Handshake(CNode& node,
      17           2 :                                bool successfully_connected,
      18           2 :                                ServiceFlags remote_services,
      19             :                                ServiceFlags local_services,
      20             :                                int32_t version,
      21             :                                bool relay_txs)
      22             : {
      23           0 :     auto& peerman{static_cast<PeerManager&>(*m_msgproc)};
      24           0 :     auto& connman{*this};
      25           0 :     const CNetMsgMaker mm{0};
      26             : 
      27           0 :     peerman.InitializeNode(node, local_services);
      28           0 :     FlushSendBuffer(node); // Drop the version message added by InitializeNode.
      29             : 
      30           0 :     CSerializedNetMsg msg_version{
      31           0 :         mm.Make(NetMsgType::VERSION,
      32             :                 version,                                        //
      33           0 :                 Using<CustomUintFormatter<8>>(remote_services), //
      34           0 :                 int64_t{},                                      // dummy time
      35           0 :                 int64_t{},                                      // ignored service bits
      36           0 :                 CNetAddr::V1(CService{}),                       // dummy
      37           0 :                 int64_t{},                                      // ignored service bits
      38           0 :                 CNetAddr::V1(CService{}),                       // ignored
      39           0 :                 uint64_t{1},                                    // dummy nonce
      40           0 :                 std::string{},                                  // dummy subver
      41           0 :                 int32_t{},                                      // dummy starting_height
      42             :                 relay_txs),
      43             :     };
      44             : 
      45           0 :     (void)connman.ReceiveMsgFrom(node, std::move(msg_version));
      46           0 :     node.fPauseSend = false;
      47           0 :     connman.ProcessMessagesOnce(node);
      48           0 :     peerman.SendMessages(&node);
      49           0 :     FlushSendBuffer(node); // Drop the verack message added by SendMessages.
      50           0 :     if (node.fDisconnect) return;
      51           0 :     assert(node.nVersion == version);
      52           0 :     assert(node.GetCommonVersion() == std::min(version, PROTOCOL_VERSION));
      53           0 :     CNodeStateStats statestats;
      54           0 :     assert(peerman.GetNodeStateStats(node.GetId(), statestats));
      55           0 :     assert(statestats.m_relay_txs == (relay_txs && !node.IsBlockOnlyConn()));
      56           0 :     assert(statestats.their_services == remote_services);
      57           0 :     if (successfully_connected) {
      58           0 :         CSerializedNetMsg msg_verack{mm.Make(NetMsgType::VERACK)};
      59           0 :         (void)connman.ReceiveMsgFrom(node, std::move(msg_verack));
      60           0 :         node.fPauseSend = false;
      61           0 :         connman.ProcessMessagesOnce(node);
      62           0 :         peerman.SendMessages(&node);
      63           0 :         assert(node.fSuccessfullyConnected == true);
      64           0 :     }
      65           0 : }
      66             : 
      67           0 : void ConnmanTestMsg::NodeReceiveMsgBytes(CNode& node, Span<const uint8_t> msg_bytes, bool& complete) const
      68             : {
      69           0 :     assert(node.ReceiveMsgBytes(msg_bytes, complete));
      70           0 :     if (complete) {
      71           0 :         node.MarkReceivedMsgsForProcessing();
      72           0 :     }
      73           0 : }
      74           2 : 
      75           0 : void ConnmanTestMsg::FlushSendBuffer(CNode& node) const
      76             : {
      77           0 :     LOCK(node.cs_vSend);
      78           0 :     node.vSendMsg.clear();
      79           0 :     node.m_send_memusage = 0;
      80           0 :     while (true) {
      81           0 :         const auto& [to_send, _more, _msg_type] = node.m_transport->GetBytesToSend(false);
      82           0 :         if (to_send.empty()) break;
      83           2 :         node.m_transport->MarkBytesSent(to_send.size());
      84             :     }
      85           0 : }
      86             : 
      87           0 : bool ConnmanTestMsg::ReceiveMsgFrom(CNode& node, CSerializedNetMsg&& ser_msg) const
      88             : {
      89           0 :     bool queued = node.m_transport->SetMessageToSend(ser_msg);
      90           0 :     assert(queued);
      91           0 :     bool complete{false};
      92           0 :     while (true) {
      93           0 :         const auto& [to_send, _more, _msg_type] = node.m_transport->GetBytesToSend(false);
      94           0 :         if (to_send.empty()) break;
      95           0 :         NodeReceiveMsgBytes(node, to_send, complete);
      96           0 :         node.m_transport->MarkBytesSent(to_send.size());
      97             :     }
      98           0 :     return complete;
      99             : }
     100             : 
     101           0 : std::vector<NodeEvictionCandidate> GetRandomNodeEvictionCandidates(int n_candidates, FastRandomContext& random_context)
     102             : {
     103           0 :     std::vector<NodeEvictionCandidate> candidates;
     104           0 :     candidates.reserve(n_candidates);
     105           0 :     for (int id = 0; id < n_candidates; ++id) {
     106           0 :         candidates.push_back({
     107           0 :             /*id=*/id,
     108           0 :             /*m_connected=*/std::chrono::seconds{random_context.randrange(100)},
     109           0 :             /*m_min_ping_time=*/std::chrono::microseconds{random_context.randrange(100)},
     110           0 :             /*m_last_block_time=*/std::chrono::seconds{random_context.randrange(100)},
     111           0 :             /*m_last_tx_time=*/std::chrono::seconds{random_context.randrange(100)},
     112           0 :             /*fRelevantServices=*/random_context.randbool(),
     113           0 :             /*m_relay_txs=*/random_context.randbool(),
     114           0 :             /*fBloomFilter=*/random_context.randbool(),
     115           0 :             /*nKeyedNetGroup=*/random_context.randrange(100),
     116           0 :             /*prefer_evict=*/random_context.randbool(),
     117           0 :             /*m_is_local=*/random_context.randbool(),
     118           0 :             /*m_network=*/ALL_NETWORKS[random_context.randrange(ALL_NETWORKS.size())],
     119             :             /*m_noban=*/false,
     120             :             /*m_conn_type=*/ConnectionType::INBOUND,
     121             :         });
     122           0 :     }
     123           0 :     return candidates;
     124           0 : }