mirror of
https://github.com/bitcoin/bitcoin.git
synced 2025-01-10 20:03:34 -03:00
net_processing: split PeerManager into interface and implementation classes
This commit is contained in:
parent
0df3d3fd6b
commit
a568b82feb
6 changed files with 109 additions and 77 deletions
|
@ -1410,8 +1410,8 @@ bool AppInitMain(const util::Ref& context, NodeContext& node, interfaces::BlockA
|
||||||
ChainstateManager& chainman = *Assert(node.chainman);
|
ChainstateManager& chainman = *Assert(node.chainman);
|
||||||
|
|
||||||
assert(!node.peerman);
|
assert(!node.peerman);
|
||||||
node.peerman = std::make_unique<PeerManager>(chainparams, *node.connman, node.banman.get(),
|
node.peerman = PeerManager::make(chainparams, *node.connman, node.banman.get(),
|
||||||
*node.scheduler, chainman, *node.mempool, ignores_incoming_txs);
|
*node.scheduler, chainman, *node.mempool, ignores_incoming_txs);
|
||||||
RegisterValidationInterface(node.peerman.get());
|
RegisterValidationInterface(node.peerman.get());
|
||||||
|
|
||||||
// sanitize comments per BIP-0014, format user agent and check total size
|
// sanitize comments per BIP-0014, format user agent and check total size
|
||||||
|
|
|
@ -683,7 +683,7 @@ static void FindNextBlocksToDownload(NodeId nodeid, unsigned int count, std::vec
|
||||||
|
|
||||||
} // namespace
|
} // namespace
|
||||||
|
|
||||||
void PeerManager::PushNodeVersion(CNode& pnode, int64_t nTime)
|
void PeerManagerImpl::PushNodeVersion(CNode& pnode, int64_t nTime)
|
||||||
{
|
{
|
||||||
// Note that pnode->GetLocalServices() is a reflection of the local
|
// Note that pnode->GetLocalServices() is a reflection of the local
|
||||||
// services we were offering when the CNode object was created for this
|
// services we were offering when the CNode object was created for this
|
||||||
|
@ -709,7 +709,7 @@ void PeerManager::PushNodeVersion(CNode& pnode, int64_t nTime)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
void PeerManager::AddTxAnnouncement(const CNode& node, const GenTxid& gtxid, std::chrono::microseconds current_time)
|
void PeerManagerImpl::AddTxAnnouncement(const CNode& node, const GenTxid& gtxid, std::chrono::microseconds current_time)
|
||||||
{
|
{
|
||||||
AssertLockHeld(::cs_main); // For m_txrequest
|
AssertLockHeld(::cs_main); // For m_txrequest
|
||||||
NodeId nodeid = node.GetId();
|
NodeId nodeid = node.GetId();
|
||||||
|
@ -745,7 +745,8 @@ void UpdateLastBlockAnnounceTime(NodeId node, int64_t time_in_seconds)
|
||||||
if (state) state->m_last_block_announcement = time_in_seconds;
|
if (state) state->m_last_block_announcement = time_in_seconds;
|
||||||
}
|
}
|
||||||
|
|
||||||
void PeerManager::InitializeNode(CNode *pnode) {
|
void PeerManagerImpl::InitializeNode(CNode *pnode)
|
||||||
|
{
|
||||||
CAddress addr = pnode->addr;
|
CAddress addr = pnode->addr;
|
||||||
std::string addrName = pnode->GetAddrName();
|
std::string addrName = pnode->GetAddrName();
|
||||||
NodeId nodeid = pnode->GetId();
|
NodeId nodeid = pnode->GetId();
|
||||||
|
@ -764,7 +765,7 @@ void PeerManager::InitializeNode(CNode *pnode) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
void PeerManager::ReattemptInitialBroadcast(CScheduler& scheduler) const
|
void PeerManagerImpl::ReattemptInitialBroadcast(CScheduler& scheduler) const
|
||||||
{
|
{
|
||||||
std::set<uint256> unbroadcast_txids = m_mempool.GetUnbroadcastTxs();
|
std::set<uint256> unbroadcast_txids = m_mempool.GetUnbroadcastTxs();
|
||||||
|
|
||||||
|
@ -785,7 +786,8 @@ void PeerManager::ReattemptInitialBroadcast(CScheduler& scheduler) const
|
||||||
scheduler.scheduleFromNow([&] { ReattemptInitialBroadcast(scheduler); }, delta);
|
scheduler.scheduleFromNow([&] { ReattemptInitialBroadcast(scheduler); }, delta);
|
||||||
}
|
}
|
||||||
|
|
||||||
void PeerManager::FinalizeNode(const CNode& node, bool& fUpdateConnectionTime) {
|
void PeerManagerImpl::FinalizeNode(const CNode& node, bool& fUpdateConnectionTime)
|
||||||
|
{
|
||||||
NodeId nodeid = node.GetId();
|
NodeId nodeid = node.GetId();
|
||||||
fUpdateConnectionTime = false;
|
fUpdateConnectionTime = false;
|
||||||
LOCK(cs_main);
|
LOCK(cs_main);
|
||||||
|
@ -839,14 +841,14 @@ void PeerManager::FinalizeNode(const CNode& node, bool& fUpdateConnectionTime) {
|
||||||
LogPrint(BCLog::NET, "Cleared nodestate for peer=%d\n", nodeid);
|
LogPrint(BCLog::NET, "Cleared nodestate for peer=%d\n", nodeid);
|
||||||
}
|
}
|
||||||
|
|
||||||
PeerRef PeerManager::GetPeerRef(NodeId id) const
|
PeerRef PeerManagerImpl::GetPeerRef(NodeId id) const
|
||||||
{
|
{
|
||||||
LOCK(m_peer_mutex);
|
LOCK(m_peer_mutex);
|
||||||
auto it = m_peer_map.find(id);
|
auto it = m_peer_map.find(id);
|
||||||
return it != m_peer_map.end() ? it->second : nullptr;
|
return it != m_peer_map.end() ? it->second : nullptr;
|
||||||
}
|
}
|
||||||
|
|
||||||
PeerRef PeerManager::RemovePeer(NodeId id)
|
PeerRef PeerManagerImpl::RemovePeer(NodeId id)
|
||||||
{
|
{
|
||||||
PeerRef ret;
|
PeerRef ret;
|
||||||
LOCK(m_peer_mutex);
|
LOCK(m_peer_mutex);
|
||||||
|
@ -858,7 +860,8 @@ PeerRef PeerManager::RemovePeer(NodeId id)
|
||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool PeerManager::GetNodeStateStats(NodeId nodeid, CNodeStateStats &stats) {
|
bool PeerManagerImpl::GetNodeStateStats(NodeId nodeid, CNodeStateStats &stats)
|
||||||
|
{
|
||||||
{
|
{
|
||||||
LOCK(cs_main);
|
LOCK(cs_main);
|
||||||
CNodeState* state = State(nodeid);
|
CNodeState* state = State(nodeid);
|
||||||
|
@ -1015,7 +1018,7 @@ unsigned int LimitOrphanTxSize(unsigned int nMaxOrphans)
|
||||||
return nEvicted;
|
return nEvicted;
|
||||||
}
|
}
|
||||||
|
|
||||||
void PeerManager::Misbehaving(const NodeId pnode, const int howmuch, const std::string& message)
|
void PeerManagerImpl::Misbehaving(const NodeId pnode, const int howmuch, const std::string& message)
|
||||||
{
|
{
|
||||||
assert(howmuch > 0);
|
assert(howmuch > 0);
|
||||||
|
|
||||||
|
@ -1033,8 +1036,8 @@ void PeerManager::Misbehaving(const NodeId pnode, const int howmuch, const std::
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
bool PeerManager::MaybePunishNodeForBlock(NodeId nodeid, const BlockValidationState& state,
|
bool PeerManagerImpl::MaybePunishNodeForBlock(NodeId nodeid, const BlockValidationState& state,
|
||||||
bool via_compact_block, const std::string& message)
|
bool via_compact_block, const std::string& message)
|
||||||
{
|
{
|
||||||
switch (state.GetResult()) {
|
switch (state.GetResult()) {
|
||||||
case BlockValidationResult::BLOCK_RESULT_UNSET:
|
case BlockValidationResult::BLOCK_RESULT_UNSET:
|
||||||
|
@ -1083,7 +1086,7 @@ bool PeerManager::MaybePunishNodeForBlock(NodeId nodeid, const BlockValidationSt
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool PeerManager::MaybePunishNodeForTx(NodeId nodeid, const TxValidationState& state, const std::string& message)
|
bool PeerManagerImpl::MaybePunishNodeForTx(NodeId nodeid, const TxValidationState& state, const std::string& message)
|
||||||
{
|
{
|
||||||
switch (state.GetResult()) {
|
switch (state.GetResult()) {
|
||||||
case TxValidationResult::TX_RESULT_UNSET:
|
case TxValidationResult::TX_RESULT_UNSET:
|
||||||
|
@ -1129,9 +1132,16 @@ static bool BlockRequestAllowed(const CBlockIndex* pindex, const Consensus::Para
|
||||||
(GetBlockProofEquivalentTime(*pindexBestHeader, *pindex, *pindexBestHeader, consensusParams) < STALE_RELAY_AGE_LIMIT);
|
(GetBlockProofEquivalentTime(*pindexBestHeader, *pindex, *pindexBestHeader, consensusParams) < STALE_RELAY_AGE_LIMIT);
|
||||||
}
|
}
|
||||||
|
|
||||||
PeerManager::PeerManager(const CChainParams& chainparams, CConnman& connman, BanMan* banman,
|
std::unique_ptr<PeerManager> PeerManager::make(const CChainParams& chainparams, CConnman& connman, BanMan* banman,
|
||||||
CScheduler& scheduler, ChainstateManager& chainman, CTxMemPool& pool,
|
CScheduler& scheduler, ChainstateManager& chainman, CTxMemPool& pool,
|
||||||
bool ignore_incoming_txs)
|
bool ignore_incoming_txs)
|
||||||
|
{
|
||||||
|
return std::make_unique<PeerManagerImpl>(chainparams, connman, banman, scheduler, chainman, pool, ignore_incoming_txs);
|
||||||
|
}
|
||||||
|
|
||||||
|
PeerManagerImpl::PeerManagerImpl(const CChainParams& chainparams, CConnman& connman, BanMan* banman,
|
||||||
|
CScheduler& scheduler, ChainstateManager& chainman, CTxMemPool& pool,
|
||||||
|
bool ignore_incoming_txs)
|
||||||
: m_chainparams(chainparams),
|
: m_chainparams(chainparams),
|
||||||
m_connman(connman),
|
m_connman(connman),
|
||||||
m_banman(banman),
|
m_banman(banman),
|
||||||
|
@ -1171,7 +1181,7 @@ PeerManager::PeerManager(const CChainParams& chainparams, CConnman& connman, Ban
|
||||||
* block, remember the recently confirmed transactions, and delete tracked
|
* block, remember the recently confirmed transactions, and delete tracked
|
||||||
* announcements for them. Also save the time of the last tip update.
|
* announcements for them. Also save the time of the last tip update.
|
||||||
*/
|
*/
|
||||||
void PeerManager::BlockConnected(const std::shared_ptr<const CBlock>& pblock, const CBlockIndex* pindex)
|
void PeerManagerImpl::BlockConnected(const std::shared_ptr<const CBlock>& pblock, const CBlockIndex* pindex)
|
||||||
{
|
{
|
||||||
{
|
{
|
||||||
LOCK(g_cs_orphans);
|
LOCK(g_cs_orphans);
|
||||||
|
@ -1222,7 +1232,7 @@ void PeerManager::BlockConnected(const std::shared_ptr<const CBlock>& pblock, co
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
void PeerManager::BlockDisconnected(const std::shared_ptr<const CBlock> &block, const CBlockIndex* pindex)
|
void PeerManagerImpl::BlockDisconnected(const std::shared_ptr<const CBlock> &block, const CBlockIndex* pindex)
|
||||||
{
|
{
|
||||||
// To avoid relay problems with transactions that were previously
|
// To avoid relay problems with transactions that were previously
|
||||||
// confirmed, clear our filter of recently confirmed transactions whenever
|
// confirmed, clear our filter of recently confirmed transactions whenever
|
||||||
|
@ -1247,7 +1257,8 @@ static bool fWitnessesPresentInMostRecentCompactBlock GUARDED_BY(cs_most_recent_
|
||||||
* Maintain state about the best-seen block and fast-announce a compact block
|
* Maintain state about the best-seen block and fast-announce a compact block
|
||||||
* to compatible peers.
|
* to compatible peers.
|
||||||
*/
|
*/
|
||||||
void PeerManager::NewPoWValidBlock(const CBlockIndex *pindex, const std::shared_ptr<const CBlock>& pblock) {
|
void PeerManagerImpl::NewPoWValidBlock(const CBlockIndex *pindex, const std::shared_ptr<const CBlock>& pblock)
|
||||||
|
{
|
||||||
std::shared_ptr<const CBlockHeaderAndShortTxIDs> pcmpctblock = std::make_shared<const CBlockHeaderAndShortTxIDs> (*pblock, true);
|
std::shared_ptr<const CBlockHeaderAndShortTxIDs> pcmpctblock = std::make_shared<const CBlockHeaderAndShortTxIDs> (*pblock, true);
|
||||||
const CNetMsgMaker msgMaker(PROTOCOL_VERSION);
|
const CNetMsgMaker msgMaker(PROTOCOL_VERSION);
|
||||||
|
|
||||||
|
@ -1294,9 +1305,9 @@ void PeerManager::NewPoWValidBlock(const CBlockIndex *pindex, const std::shared_
|
||||||
* Update our best height and announce any block hashes which weren't previously
|
* Update our best height and announce any block hashes which weren't previously
|
||||||
* in ::ChainActive() to our peers.
|
* in ::ChainActive() to our peers.
|
||||||
*/
|
*/
|
||||||
void PeerManager::UpdatedBlockTip(const CBlockIndex* pindexNew, const CBlockIndex* pindexFork, bool fInitialDownload)
|
void PeerManagerImpl::UpdatedBlockTip(const CBlockIndex *pindexNew, const CBlockIndex *pindexFork, bool fInitialDownload)
|
||||||
{
|
{
|
||||||
m_best_height = pindexNew->nHeight;
|
SetBestHeight(pindexNew->nHeight);
|
||||||
SetServiceFlagsIBDCache(!fInitialDownload);
|
SetServiceFlagsIBDCache(!fInitialDownload);
|
||||||
|
|
||||||
// Don't relay inventory during initial block download.
|
// Don't relay inventory during initial block download.
|
||||||
|
@ -1333,7 +1344,8 @@ void PeerManager::UpdatedBlockTip(const CBlockIndex* pindexNew, const CBlockInde
|
||||||
* Handle invalid block rejection and consequent peer discouragement, maintain which
|
* Handle invalid block rejection and consequent peer discouragement, maintain which
|
||||||
* peers announce compact blocks.
|
* peers announce compact blocks.
|
||||||
*/
|
*/
|
||||||
void PeerManager::BlockChecked(const CBlock& block, const BlockValidationState& state) {
|
void PeerManagerImpl::BlockChecked(const CBlock& block, const BlockValidationState& state)
|
||||||
|
{
|
||||||
LOCK(cs_main);
|
LOCK(cs_main);
|
||||||
|
|
||||||
const uint256 hash(block.GetHash());
|
const uint256 hash(block.GetHash());
|
||||||
|
@ -1761,7 +1773,8 @@ static uint32_t GetFetchFlags(const CNode& pfrom) EXCLUSIVE_LOCKS_REQUIRED(cs_ma
|
||||||
return nFetchFlags;
|
return nFetchFlags;
|
||||||
}
|
}
|
||||||
|
|
||||||
void PeerManager::SendBlockTransactions(CNode& pfrom, const CBlock& block, const BlockTransactionsRequest& req) {
|
void PeerManagerImpl::SendBlockTransactions(CNode& pfrom, const CBlock& block, const BlockTransactionsRequest& req)
|
||||||
|
{
|
||||||
BlockTransactions resp(req);
|
BlockTransactions resp(req);
|
||||||
for (size_t i = 0; i < req.indexes.size(); i++) {
|
for (size_t i = 0; i < req.indexes.size(); i++) {
|
||||||
if (req.indexes[i] >= block.vtx.size()) {
|
if (req.indexes[i] >= block.vtx.size()) {
|
||||||
|
@ -1776,9 +1789,9 @@ void PeerManager::SendBlockTransactions(CNode& pfrom, const CBlock& block, const
|
||||||
m_connman.PushMessage(&pfrom, msgMaker.Make(nSendFlags, NetMsgType::BLOCKTXN, resp));
|
m_connman.PushMessage(&pfrom, msgMaker.Make(nSendFlags, NetMsgType::BLOCKTXN, resp));
|
||||||
}
|
}
|
||||||
|
|
||||||
void PeerManager::ProcessHeadersMessage(CNode& pfrom, const Peer& peer,
|
void PeerManagerImpl::ProcessHeadersMessage(CNode& pfrom, const Peer& peer,
|
||||||
const std::vector<CBlockHeader>& headers,
|
const std::vector<CBlockHeader>& headers,
|
||||||
bool via_compact_block)
|
bool via_compact_block)
|
||||||
{
|
{
|
||||||
const CNetMsgMaker msgMaker(pfrom.GetCommonVersion());
|
const CNetMsgMaker msgMaker(pfrom.GetCommonVersion());
|
||||||
size_t nCount = headers.size();
|
size_t nCount = headers.size();
|
||||||
|
@ -1970,7 +1983,7 @@ void PeerManager::ProcessHeadersMessage(CNode& pfrom, const Peer& peer,
|
||||||
* may be added to if accepting an orphan causes its children to be
|
* may be added to if accepting an orphan causes its children to be
|
||||||
* reconsidered.
|
* reconsidered.
|
||||||
*/
|
*/
|
||||||
void PeerManager::ProcessOrphanTx(std::set<uint256>& orphan_work_set)
|
void PeerManagerImpl::ProcessOrphanTx(std::set<uint256>& orphan_work_set)
|
||||||
{
|
{
|
||||||
AssertLockHeld(cs_main);
|
AssertLockHeld(cs_main);
|
||||||
AssertLockHeld(g_cs_orphans);
|
AssertLockHeld(g_cs_orphans);
|
||||||
|
@ -2267,9 +2280,9 @@ static void ProcessGetCFCheckPt(CNode& peer, CDataStream& vRecv, const CChainPar
|
||||||
connman.PushMessage(&peer, std::move(msg));
|
connman.PushMessage(&peer, std::move(msg));
|
||||||
}
|
}
|
||||||
|
|
||||||
void PeerManager::ProcessMessage(CNode& pfrom, const std::string& msg_type, CDataStream& vRecv,
|
void PeerManagerImpl::ProcessMessage(CNode& pfrom, const std::string& msg_type, CDataStream& vRecv,
|
||||||
const std::chrono::microseconds time_received,
|
const std::chrono::microseconds time_received,
|
||||||
const std::atomic<bool>& interruptMsgProc)
|
const std::atomic<bool>& interruptMsgProc)
|
||||||
{
|
{
|
||||||
LogPrint(BCLog::NET, "received: %s (%u bytes) peer=%d\n", SanitizeString(msg_type), vRecv.size(), pfrom.GetId());
|
LogPrint(BCLog::NET, "received: %s (%u bytes) peer=%d\n", SanitizeString(msg_type), vRecv.size(), pfrom.GetId());
|
||||||
|
|
||||||
|
@ -3756,7 +3769,7 @@ void PeerManager::ProcessMessage(CNode& pfrom, const std::string& msg_type, CDat
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool PeerManager::MaybeDiscourageAndDisconnect(CNode& pnode)
|
bool PeerManagerImpl::MaybeDiscourageAndDisconnect(CNode& pnode)
|
||||||
{
|
{
|
||||||
const NodeId peer_id{pnode.GetId()};
|
const NodeId peer_id{pnode.GetId()};
|
||||||
PeerRef peer = GetPeerRef(peer_id);
|
PeerRef peer = GetPeerRef(peer_id);
|
||||||
|
@ -3798,7 +3811,7 @@ bool PeerManager::MaybeDiscourageAndDisconnect(CNode& pnode)
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool PeerManager::ProcessMessages(CNode* pfrom, std::atomic<bool>& interruptMsgProc)
|
bool PeerManagerImpl::ProcessMessages(CNode* pfrom, std::atomic<bool>& interruptMsgProc)
|
||||||
{
|
{
|
||||||
bool fMoreWork = false;
|
bool fMoreWork = false;
|
||||||
|
|
||||||
|
@ -3873,7 +3886,7 @@ bool PeerManager::ProcessMessages(CNode* pfrom, std::atomic<bool>& interruptMsgP
|
||||||
return fMoreWork;
|
return fMoreWork;
|
||||||
}
|
}
|
||||||
|
|
||||||
void PeerManager::ConsiderEviction(CNode& pto, int64_t time_in_seconds)
|
void PeerManagerImpl::ConsiderEviction(CNode& pto, int64_t time_in_seconds)
|
||||||
{
|
{
|
||||||
AssertLockHeld(cs_main);
|
AssertLockHeld(cs_main);
|
||||||
|
|
||||||
|
@ -3926,7 +3939,7 @@ void PeerManager::ConsiderEviction(CNode& pto, int64_t time_in_seconds)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
void PeerManager::EvictExtraOutboundPeers(int64_t time_in_seconds)
|
void PeerManagerImpl::EvictExtraOutboundPeers(int64_t time_in_seconds)
|
||||||
{
|
{
|
||||||
// If we have any extra block-relay-only peers, disconnect the youngest unless
|
// If we have any extra block-relay-only peers, disconnect the youngest unless
|
||||||
// it's given us a block -- in which case, compare with the second-youngest, and
|
// it's given us a block -- in which case, compare with the second-youngest, and
|
||||||
|
@ -4027,7 +4040,7 @@ void PeerManager::EvictExtraOutboundPeers(int64_t time_in_seconds)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
void PeerManager::CheckForStaleTipAndEvictPeers()
|
void PeerManagerImpl::CheckForStaleTipAndEvictPeers()
|
||||||
{
|
{
|
||||||
LOCK(cs_main);
|
LOCK(cs_main);
|
||||||
|
|
||||||
|
@ -4074,7 +4087,7 @@ public:
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
bool PeerManager::SendMessages(CNode* pto)
|
bool PeerManagerImpl::SendMessages(CNode* pto)
|
||||||
{
|
{
|
||||||
PeerRef peer = GetPeerRef(pto->GetId());
|
PeerRef peer = GetPeerRef(pto->GetId());
|
||||||
const Consensus::Params& consensusParams = m_chainparams.GetConsensus();
|
const Consensus::Params& consensusParams = m_chainparams.GetConsensus();
|
||||||
|
|
|
@ -94,11 +94,47 @@ struct Peer {
|
||||||
|
|
||||||
using PeerRef = std::shared_ptr<Peer>;
|
using PeerRef = std::shared_ptr<Peer>;
|
||||||
|
|
||||||
class PeerManager final : public CValidationInterface, public NetEventsInterface {
|
class PeerManager : public CValidationInterface, public NetEventsInterface
|
||||||
|
{
|
||||||
public:
|
public:
|
||||||
PeerManager(const CChainParams& chainparams, CConnman& connman, BanMan* banman,
|
static std::unique_ptr<PeerManager> make(const CChainParams& chainparams, CConnman& connman, BanMan* banman,
|
||||||
CScheduler& scheduler, ChainstateManager& chainman, CTxMemPool& pool,
|
CScheduler& scheduler, ChainstateManager& chainman, CTxMemPool& pool,
|
||||||
bool ignore_incoming_txs);
|
bool ignore_incoming_txs);
|
||||||
|
virtual ~PeerManager() { }
|
||||||
|
|
||||||
|
/** Get statistics from node state */
|
||||||
|
virtual bool GetNodeStateStats(NodeId nodeid, CNodeStateStats& stats) = 0;
|
||||||
|
|
||||||
|
/** Whether this node ignores txs received over p2p. */
|
||||||
|
virtual bool IgnoresIncomingTxs() = 0;
|
||||||
|
|
||||||
|
/** Set the best height */
|
||||||
|
virtual void SetBestHeight(int height) = 0;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Increment peer's misbehavior score. If the new value >= DISCOURAGEMENT_THRESHOLD, mark the node
|
||||||
|
* to be discouraged, meaning the peer might be disconnected and added to the discouragement filter.
|
||||||
|
* Public for unit testing.
|
||||||
|
*/
|
||||||
|
virtual void Misbehaving(const NodeId pnode, const int howmuch, const std::string& message) = 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) = 0;
|
||||||
|
};
|
||||||
|
|
||||||
|
class PeerManagerImpl final : public PeerManager
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
PeerManagerImpl(const CChainParams& chainparams, CConnman& connman, BanMan* banman,
|
||||||
|
CScheduler& scheduler, ChainstateManager& chainman, CTxMemPool& pool,
|
||||||
|
bool ignore_incoming_txs);
|
||||||
|
|
||||||
/** Overridden from CValidationInterface. */
|
/** Overridden from CValidationInterface. */
|
||||||
void BlockConnected(const std::shared_ptr<const CBlock>& pblock, const CBlockIndex* pindexConnected) override;
|
void BlockConnected(const std::shared_ptr<const CBlock>& pblock, const CBlockIndex* pindexConnected) override;
|
||||||
|
@ -113,31 +149,14 @@ public:
|
||||||
bool ProcessMessages(CNode* pfrom, std::atomic<bool>& interrupt) override;
|
bool ProcessMessages(CNode* pfrom, std::atomic<bool>& interrupt) override;
|
||||||
bool SendMessages(CNode* pto) override EXCLUSIVE_LOCKS_REQUIRED(pto->cs_sendProcessing);
|
bool SendMessages(CNode* pto) override EXCLUSIVE_LOCKS_REQUIRED(pto->cs_sendProcessing);
|
||||||
|
|
||||||
/** Get statistics from node state */
|
/** Implement PeerManager */
|
||||||
bool GetNodeStateStats(NodeId nodeid, CNodeStateStats& stats);
|
void CheckForStaleTipAndEvictPeers() override;
|
||||||
|
bool GetNodeStateStats(NodeId nodeid, CNodeStateStats& stats) override;
|
||||||
/** Whether this node ignores txs received over p2p. */
|
bool IgnoresIncomingTxs() override { return m_ignore_incoming_txs; }
|
||||||
bool IgnoresIncomingTxs() { return m_ignore_incoming_txs; };
|
void SetBestHeight(int height) override { m_best_height = height; };
|
||||||
|
void Misbehaving(const NodeId pnode, const int howmuch, const std::string& message) override;
|
||||||
/** Set the best height */
|
|
||||||
void SetBestHeight(int height) { m_best_height = height; };
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Increment peer's misbehavior score. If the new value >= DISCOURAGEMENT_THRESHOLD, mark the node
|
|
||||||
* to be discouraged, meaning the peer might be disconnected and added to the discouragement filter.
|
|
||||||
* Public for unit testing.
|
|
||||||
*/
|
|
||||||
void Misbehaving(const NodeId pnode, const int howmuch, const std::string& message);
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Evict extra outbound peers. If we think our tip may be stale, connect to an extra outbound.
|
|
||||||
* Public for unit testing.
|
|
||||||
*/
|
|
||||||
void CheckForStaleTipAndEvictPeers();
|
|
||||||
|
|
||||||
/** Process a single message from a peer. Public for fuzz testing */
|
|
||||||
void ProcessMessage(CNode& pfrom, const std::string& msg_type, CDataStream& vRecv,
|
void ProcessMessage(CNode& pfrom, const std::string& msg_type, CDataStream& vRecv,
|
||||||
const std::chrono::microseconds time_received, const std::atomic<bool>& interruptMsgProc);
|
const std::chrono::microseconds time_received, const std::atomic<bool>& interruptMsgProc) override;
|
||||||
|
|
||||||
private:
|
private:
|
||||||
/** Consider evicting an outbound peer based on the amount of time they've been behind our tip */
|
/** Consider evicting an outbound peer based on the amount of time they've been behind our tip */
|
||||||
|
|
|
@ -80,8 +80,8 @@ BOOST_AUTO_TEST_CASE(outbound_slow_chain_eviction)
|
||||||
{
|
{
|
||||||
const CChainParams& chainparams = Params();
|
const CChainParams& chainparams = Params();
|
||||||
auto connman = MakeUnique<CConnman>(0x1337, 0x1337);
|
auto connman = MakeUnique<CConnman>(0x1337, 0x1337);
|
||||||
auto peerLogic = std::make_unique<PeerManager>(chainparams, *connman, nullptr, *m_node.scheduler,
|
auto peerLogic = PeerManager::make(chainparams, *connman, nullptr, *m_node.scheduler,
|
||||||
*m_node.chainman, *m_node.mempool, false);
|
*m_node.chainman, *m_node.mempool, false);
|
||||||
|
|
||||||
// Mock an outbound peer
|
// Mock an outbound peer
|
||||||
CAddress addr1(ip(0xa0b0c001), NODE_NONE);
|
CAddress addr1(ip(0xa0b0c001), NODE_NONE);
|
||||||
|
@ -150,8 +150,8 @@ BOOST_AUTO_TEST_CASE(stale_tip_peer_management)
|
||||||
{
|
{
|
||||||
const CChainParams& chainparams = Params();
|
const CChainParams& chainparams = Params();
|
||||||
auto connman = MakeUnique<CConnmanTest>(0x1337, 0x1337);
|
auto connman = MakeUnique<CConnmanTest>(0x1337, 0x1337);
|
||||||
auto peerLogic = std::make_unique<PeerManager>(chainparams, *connman, nullptr, *m_node.scheduler,
|
auto peerLogic = PeerManager::make(chainparams, *connman, nullptr, *m_node.scheduler,
|
||||||
*m_node.chainman, *m_node.mempool, false);
|
*m_node.chainman, *m_node.mempool, false);
|
||||||
|
|
||||||
constexpr int max_outbound_full_relay = MAX_OUTBOUND_FULL_RELAY_CONNECTIONS;
|
constexpr int max_outbound_full_relay = MAX_OUTBOUND_FULL_RELAY_CONNECTIONS;
|
||||||
CConnman::Options options;
|
CConnman::Options options;
|
||||||
|
@ -224,8 +224,8 @@ BOOST_AUTO_TEST_CASE(peer_discouragement)
|
||||||
const CChainParams& chainparams = Params();
|
const CChainParams& chainparams = Params();
|
||||||
auto banman = MakeUnique<BanMan>(GetDataDir() / "banlist.dat", nullptr, DEFAULT_MISBEHAVING_BANTIME);
|
auto banman = MakeUnique<BanMan>(GetDataDir() / "banlist.dat", nullptr, DEFAULT_MISBEHAVING_BANTIME);
|
||||||
auto connman = MakeUnique<CConnman>(0x1337, 0x1337);
|
auto connman = MakeUnique<CConnman>(0x1337, 0x1337);
|
||||||
auto peerLogic = std::make_unique<PeerManager>(chainparams, *connman, banman.get(), *m_node.scheduler,
|
auto peerLogic = PeerManager::make(chainparams, *connman, banman.get(), *m_node.scheduler,
|
||||||
*m_node.chainman, *m_node.mempool, false);
|
*m_node.chainman, *m_node.mempool, false);
|
||||||
|
|
||||||
banman->ClearBanned();
|
banman->ClearBanned();
|
||||||
CAddress addr1(ip(0xa0b0c001), NODE_NONE);
|
CAddress addr1(ip(0xa0b0c001), NODE_NONE);
|
||||||
|
@ -271,8 +271,8 @@ BOOST_AUTO_TEST_CASE(DoS_bantime)
|
||||||
const CChainParams& chainparams = Params();
|
const CChainParams& chainparams = Params();
|
||||||
auto banman = MakeUnique<BanMan>(GetDataDir() / "banlist.dat", nullptr, DEFAULT_MISBEHAVING_BANTIME);
|
auto banman = MakeUnique<BanMan>(GetDataDir() / "banlist.dat", nullptr, DEFAULT_MISBEHAVING_BANTIME);
|
||||||
auto connman = MakeUnique<CConnman>(0x1337, 0x1337);
|
auto connman = MakeUnique<CConnman>(0x1337, 0x1337);
|
||||||
auto peerLogic = std::make_unique<PeerManager>(chainparams, *connman, banman.get(), *m_node.scheduler,
|
auto peerLogic = PeerManager::make(chainparams, *connman, banman.get(), *m_node.scheduler,
|
||||||
*m_node.chainman, *m_node.mempool, false);
|
*m_node.chainman, *m_node.mempool, false);
|
||||||
|
|
||||||
banman->ClearBanned();
|
banman->ClearBanned();
|
||||||
int64_t nStartTime = GetTime();
|
int64_t nStartTime = GetTime();
|
||||||
|
|
|
@ -192,9 +192,9 @@ TestingSetup::TestingSetup(const std::string& chainName, const std::vector<const
|
||||||
|
|
||||||
m_node.banman = MakeUnique<BanMan>(GetDataDir() / "banlist.dat", nullptr, DEFAULT_MISBEHAVING_BANTIME);
|
m_node.banman = MakeUnique<BanMan>(GetDataDir() / "banlist.dat", nullptr, DEFAULT_MISBEHAVING_BANTIME);
|
||||||
m_node.connman = MakeUnique<CConnman>(0x1337, 0x1337); // Deterministic randomness for tests.
|
m_node.connman = MakeUnique<CConnman>(0x1337, 0x1337); // Deterministic randomness for tests.
|
||||||
m_node.peerman = std::make_unique<PeerManager>(chainparams, *m_node.connman, m_node.banman.get(),
|
m_node.peerman = PeerManager::make(chainparams, *m_node.connman, m_node.banman.get(),
|
||||||
*m_node.scheduler, *m_node.chainman, *m_node.mempool,
|
*m_node.scheduler, *m_node.chainman, *m_node.mempool,
|
||||||
false);
|
false);
|
||||||
{
|
{
|
||||||
CConnman::Options options;
|
CConnman::Options options;
|
||||||
options.m_msgproc = m_node.peerman.get();
|
options.m_msgproc = m_node.peerman.get();
|
||||||
|
|
|
@ -13,7 +13,7 @@ mutex:CConnman::ThreadOpenConnections
|
||||||
mutex:CConnman::ThreadOpenAddedConnections
|
mutex:CConnman::ThreadOpenAddedConnections
|
||||||
mutex:CConnman::SocketHandler
|
mutex:CConnman::SocketHandler
|
||||||
mutex:UpdateTip
|
mutex:UpdateTip
|
||||||
mutex:PeerManager::UpdatedBlockTip
|
mutex:PeerManagerImpl::UpdatedBlockTip
|
||||||
mutex:g_best_block_mutex
|
mutex:g_best_block_mutex
|
||||||
|
|
||||||
# race (TODO fix)
|
# race (TODO fix)
|
||||||
|
|
Loading…
Reference in a new issue