mirror of
https://github.com/bitcoin/bitcoin.git
synced 2025-01-29 20:47:31 -03:00
112 lines
4.7 KiB
C++
112 lines
4.7 KiB
C++
// Copyright (c) 2009-2010 Satoshi Nakamoto
|
|
// Copyright (c) 2009-2022 The Bitcoin Core developers
|
|
// Distributed under the MIT software license, see the accompanying
|
|
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
|
|
|
#ifndef BITCOIN_NET_PROCESSING_H
|
|
#define BITCOIN_NET_PROCESSING_H
|
|
|
|
#include <net.h>
|
|
#include <validationinterface.h>
|
|
|
|
class AddrMan;
|
|
class CChainParams;
|
|
class CTxMemPool;
|
|
class ChainstateManager;
|
|
|
|
/** Whether transaction reconciliation protocol should be enabled by default. */
|
|
static constexpr bool DEFAULT_TXRECONCILIATION_ENABLE{false};
|
|
/** Default for -maxorphantx, maximum number of orphan transactions kept in memory */
|
|
static const uint32_t DEFAULT_MAX_ORPHAN_TRANSACTIONS{100};
|
|
/** Default number of non-mempool transactions to keep around for block reconstruction. Includes
|
|
orphan, replaced, and rejected transactions. */
|
|
static const uint32_t DEFAULT_BLOCK_RECONSTRUCTION_EXTRA_TXN{100};
|
|
static const bool DEFAULT_PEERBLOOMFILTERS = false;
|
|
static const bool DEFAULT_PEERBLOCKFILTERS = false;
|
|
/** Threshold for marking a node to be discouraged, e.g. disconnected and added to the discouragement filter. */
|
|
static const int DISCOURAGEMENT_THRESHOLD{100};
|
|
/** Maximum number of outstanding CMPCTBLOCK requests for the same block. */
|
|
static const unsigned int MAX_CMPCTBLOCKS_INFLIGHT_PER_BLOCK = 3;
|
|
|
|
struct CNodeStateStats {
|
|
int nSyncHeight = -1;
|
|
int nCommonHeight = -1;
|
|
int m_starting_height = -1;
|
|
std::chrono::microseconds m_ping_wait;
|
|
std::vector<int> vHeightInFlight;
|
|
bool m_relay_txs;
|
|
CAmount m_fee_filter_received;
|
|
uint64_t m_addr_processed = 0;
|
|
uint64_t m_addr_rate_limited = 0;
|
|
bool m_addr_relay_enabled{false};
|
|
ServiceFlags their_services;
|
|
int64_t presync_height{-1};
|
|
};
|
|
|
|
class PeerManager : public CValidationInterface, public NetEventsInterface
|
|
{
|
|
public:
|
|
struct Options {
|
|
//! Whether this node is running in -blocksonly mode
|
|
bool ignore_incoming_txs{DEFAULT_BLOCKSONLY};
|
|
//! Whether transaction reconciliation protocol is enabled
|
|
bool reconcile_txs{DEFAULT_TXRECONCILIATION_ENABLE};
|
|
//! Maximum number of orphan transactions kept in memory
|
|
uint32_t max_orphan_txs{DEFAULT_MAX_ORPHAN_TRANSACTIONS};
|
|
//! Number of non-mempool transactions to keep around for block reconstruction. Includes
|
|
//! orphan, replaced, and rejected transactions.
|
|
uint32_t max_extra_txs{DEFAULT_BLOCK_RECONSTRUCTION_EXTRA_TXN};
|
|
//! Whether all P2P messages are captured to disk
|
|
bool capture_messages{false};
|
|
};
|
|
|
|
static std::unique_ptr<PeerManager> make(CConnman& connman, AddrMan& addrman,
|
|
BanMan* banman, ChainstateManager& chainman,
|
|
CTxMemPool& pool, Options opts);
|
|
virtual ~PeerManager() { }
|
|
|
|
/**
|
|
* Attempt to manually fetch block from a given peer. We must already have the header.
|
|
*
|
|
* @param[in] peer_id The peer id
|
|
* @param[in] block_index The blockindex
|
|
* @returns std::nullopt if a request was successfully made, otherwise an error message
|
|
*/
|
|
virtual std::optional<std::string> FetchBlock(NodeId peer_id, const CBlockIndex& block_index) = 0;
|
|
|
|
/** Begin running background tasks, should only be called once */
|
|
virtual void StartScheduledTasks(CScheduler& scheduler) = 0;
|
|
|
|
/** Get statistics from node state */
|
|
virtual bool GetNodeStateStats(NodeId nodeid, CNodeStateStats& stats) const = 0;
|
|
|
|
/** Whether this node ignores txs received over p2p. */
|
|
virtual bool IgnoresIncomingTxs() = 0;
|
|
|
|
/** Relay transaction to all peers. */
|
|
virtual void RelayTransaction(const uint256& txid, const uint256& wtxid) = 0;
|
|
|
|
/** Send ping message to all peers */
|
|
virtual void SendPings() = 0;
|
|
|
|
/** Set the best height */
|
|
virtual void SetBestHeight(int height) = 0;
|
|
|
|
/* Public for unit testing. */
|
|
virtual void UnitTestMisbehaving(NodeId peer_id, int howmuch) = 0;
|
|
|
|
/**
|
|
* Evict extra outbound peers. If we think our tip may be stale, connect to an extra outbound.
|
|
* Public for unit testing.
|
|
*/
|
|
virtual void CheckForStaleTipAndEvictPeers() = 0;
|
|
|
|
/** Process a single message from a peer. Public for fuzz testing */
|
|
virtual void ProcessMessage(CNode& pfrom, const std::string& msg_type, CDataStream& vRecv,
|
|
const std::chrono::microseconds time_received, const std::atomic<bool>& interruptMsgProc) EXCLUSIVE_LOCKS_REQUIRED(g_msgproc_mutex) = 0;
|
|
|
|
/** This function is used for testing the stale tip eviction logic, see denialofservice_tests.cpp */
|
|
virtual void UpdateLastBlockAnnounceTime(NodeId node, int64_t time_in_seconds) = 0;
|
|
};
|
|
|
|
#endif // BITCOIN_NET_PROCESSING_H
|