highperfocused/bitcoin
1
0
Fork 0
mirror of https://github.com/bitcoin/bitcoin.git synced 2025-11-11 22:50:59 +01:00
Code Issues Packages Projects Releases Wiki Activity

p2p: Forget peer's reconciliation state on disconnect

Browse Source
This commit is contained in:
Gleb Naumenko
2021-03-20 12:50:02 +02:00
parent 3fcf78ee6a
commit 4470acf076
3 changed files with 21 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/node/txreconciliation.cpp
View File

@@ -54,6 +54,15 @@ public:
Assert(m_states.emplace(peer_id, local_salt).second);
return local_salt;
}
void ForgetPeer(NodeId peer_id) EXCLUSIVE_LOCKS_REQUIRED(!m_txreconciliation_mutex)
{
AssertLockNotHeld(m_txreconciliation_mutex);
LOCK(m_txreconciliation_mutex);
if (m_states.erase(peer_id)) {
LogPrintLevel(BCLog::TXRECONCILIATION, BCLog::Level::Debug, "Forget txreconciliation state of peer=%d\n", peer_id);
}
}
};
TxReconciliationTracker::TxReconciliationTracker() : m_impl{std::make_unique<TxReconciliationTracker::Impl>()} {}
@@ -64,3 +73,8 @@ uint64_t TxReconciliationTracker::PreRegisterPeer(NodeId peer_id)
{
return m_impl->PreRegisterPeer(peer_id);
}
void TxReconciliationTracker::ForgetPeer(NodeId peer_id)
{
m_impl->ForgetPeer(peer_id);
}

6
src/node/txreconciliation.h
View File

@@ -61,6 +61,12 @@ public:
* This function must be called only once per peer.
*/
uint64_t PreRegisterPeer(NodeId peer_id);
/**
* Attempts to forget txreconciliation-related state of the peer (if we previously stored any).
* After this, we won't be able to reconcile transactions with the peer.
*/
void ForgetPeer(NodeId peer_id);
};
#endif // BITCOIN_NODE_TXRECONCILIATION_H