mirror of
https://github.com/bitcoin/bitcoin.git
synced 2026-02-01 11:11:22 +00:00
3fcf78ee6a
If we're connecting to the peer which might support transaction reconciliation, we announce we want to reconcile with them. We store the reconciliation salt so that when the peer responds with their salt, we are able to compute the full reconciliation salt. This behavior is enabled with a CLI flag.
67 lines
2.0 KiB
C++
67 lines
2.0 KiB
C++
// Copyright (c) 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.
|
|
|
|
#include <node/txreconciliation.h>
|
|
|
|
#include <util/check.h>
|
|
#include <util/system.h>
|
|
|
|
#include <unordered_map>
|
|
#include <variant>
|
|
|
|
|
|
namespace {
|
|
|
|
/**
|
|
* Keeps track of txreconciliation-related per-peer state.
|
|
*/
|
|
class TxReconciliationState
|
|
{
|
|
};
|
|
|
|
} // namespace
|
|
|
|
/** Actual implementation for TxReconciliationTracker's data structure. */
|
|
class TxReconciliationTracker::Impl
|
|
{
|
|
private:
|
|
mutable Mutex m_txreconciliation_mutex;
|
|
|
|
/**
|
|
* Keeps track of txreconciliation states of eligible peers.
|
|
* For pre-registered peers, the locally generated salt is stored.
|
|
* For registered peers, the locally generated salt is forgotten, and the state (including
|
|
* "full" salt) is stored instead.
|
|
*/
|
|
std::unordered_map<NodeId, std::variant<uint64_t, TxReconciliationState>> m_states GUARDED_BY(m_txreconciliation_mutex);
|
|
|
|
public:
|
|
explicit Impl() {}
|
|
|
|
uint64_t PreRegisterPeer(NodeId peer_id) EXCLUSIVE_LOCKS_REQUIRED(!m_txreconciliation_mutex)
|
|
{
|
|
AssertLockNotHeld(m_txreconciliation_mutex);
|
|
LOCK(m_txreconciliation_mutex);
|
|
// We do not support txreconciliation salt/version updates.
|
|
assert(m_states.find(peer_id) == m_states.end());
|
|
|
|
LogPrintLevel(BCLog::TXRECONCILIATION, BCLog::Level::Debug, "Pre-register peer=%d\n", peer_id);
|
|
const uint64_t local_salt{GetRand(UINT64_MAX)};
|
|
|
|
// We do this exactly once per peer (which are unique by NodeId, see GetNewNodeId) so it's
|
|
// safe to assume we don't have this record yet.
|
|
Assert(m_states.emplace(peer_id, local_salt).second);
|
|
return local_salt;
|
|
}
|
|
};
|
|
|
|
TxReconciliationTracker::TxReconciliationTracker() : m_impl{std::make_unique<TxReconciliationTracker::Impl>()} {}
|
|
|
|
TxReconciliationTracker::~TxReconciliationTracker() = default;
|
|
|
|
uint64_t TxReconciliationTracker::PreRegisterPeer(NodeId peer_id)
|
|
{
|
|
return m_impl->PreRegisterPeer(peer_id);
|
|
}
|