You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
165 lines
8.2 KiB
165 lines
8.2 KiB
// Copyright (c) 2009-2010 Satoshi Nakamoto |
|
// Copyright (c) 2009-2016 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 <validationinterface.h> |
|
|
|
#include <init.h> |
|
#include <primitives/block.h> |
|
#include <scheduler.h> |
|
#include <sync.h> |
|
#include <txmempool.h> |
|
#include <util.h> |
|
|
|
#include <list> |
|
#include <atomic> |
|
|
|
#include <boost/signals2/signal.hpp> |
|
|
|
struct MainSignalsInstance { |
|
boost::signals2::signal<void (const CBlockIndex *, const CBlockIndex *, bool fInitialDownload)> UpdatedBlockTip; |
|
boost::signals2::signal<void (const CTransactionRef &)> TransactionAddedToMempool; |
|
boost::signals2::signal<void (const std::shared_ptr<const CBlock> &, const CBlockIndex *pindex, const std::vector<CTransactionRef>&)> BlockConnected; |
|
boost::signals2::signal<void (const std::shared_ptr<const CBlock> &)> BlockDisconnected; |
|
boost::signals2::signal<void (const CTransactionRef &)> TransactionRemovedFromMempool; |
|
boost::signals2::signal<void (const CBlockLocator &)> SetBestChain; |
|
boost::signals2::signal<void (const uint256 &)> Inventory; |
|
boost::signals2::signal<void (int64_t nBestBlockTime, CConnman* connman)> Broadcast; |
|
boost::signals2::signal<void (const CBlock&, const CValidationState&)> BlockChecked; |
|
boost::signals2::signal<void (const CBlockIndex *, const std::shared_ptr<const CBlock>&)> NewPoWValidBlock; |
|
|
|
// We are not allowed to assume the scheduler only runs in one thread, |
|
// but must ensure all callbacks happen in-order, so we end up creating |
|
// our own queue here :( |
|
SingleThreadedSchedulerClient m_schedulerClient; |
|
|
|
explicit MainSignalsInstance(CScheduler *pscheduler) : m_schedulerClient(pscheduler) {} |
|
}; |
|
|
|
static CMainSignals g_signals; |
|
|
|
void CMainSignals::RegisterBackgroundSignalScheduler(CScheduler& scheduler) { |
|
assert(!m_internals); |
|
m_internals.reset(new MainSignalsInstance(&scheduler)); |
|
} |
|
|
|
void CMainSignals::UnregisterBackgroundSignalScheduler() { |
|
m_internals.reset(nullptr); |
|
} |
|
|
|
void CMainSignals::FlushBackgroundCallbacks() { |
|
m_internals->m_schedulerClient.EmptyQueue(); |
|
} |
|
|
|
void CMainSignals::RegisterWithMempoolSignals(CTxMemPool& pool) { |
|
pool.NotifyEntryRemoved.connect(boost::bind(&CMainSignals::MempoolEntryRemoved, this, _1, _2)); |
|
} |
|
|
|
void CMainSignals::UnregisterWithMempoolSignals(CTxMemPool& pool) { |
|
pool.NotifyEntryRemoved.disconnect(boost::bind(&CMainSignals::MempoolEntryRemoved, this, _1, _2)); |
|
} |
|
|
|
CMainSignals& GetMainSignals() |
|
{ |
|
return g_signals; |
|
} |
|
|
|
void RegisterValidationInterface(CValidationInterface* pwalletIn) { |
|
g_signals.m_internals->UpdatedBlockTip.connect(boost::bind(&CValidationInterface::UpdatedBlockTip, pwalletIn, _1, _2, _3)); |
|
g_signals.m_internals->TransactionAddedToMempool.connect(boost::bind(&CValidationInterface::TransactionAddedToMempool, pwalletIn, _1)); |
|
g_signals.m_internals->BlockConnected.connect(boost::bind(&CValidationInterface::BlockConnected, pwalletIn, _1, _2, _3)); |
|
g_signals.m_internals->BlockDisconnected.connect(boost::bind(&CValidationInterface::BlockDisconnected, pwalletIn, _1)); |
|
g_signals.m_internals->TransactionRemovedFromMempool.connect(boost::bind(&CValidationInterface::TransactionRemovedFromMempool, pwalletIn, _1)); |
|
g_signals.m_internals->SetBestChain.connect(boost::bind(&CValidationInterface::SetBestChain, pwalletIn, _1)); |
|
g_signals.m_internals->Inventory.connect(boost::bind(&CValidationInterface::Inventory, pwalletIn, _1)); |
|
g_signals.m_internals->Broadcast.connect(boost::bind(&CValidationInterface::ResendWalletTransactions, pwalletIn, _1, _2)); |
|
g_signals.m_internals->BlockChecked.connect(boost::bind(&CValidationInterface::BlockChecked, pwalletIn, _1, _2)); |
|
g_signals.m_internals->NewPoWValidBlock.connect(boost::bind(&CValidationInterface::NewPoWValidBlock, pwalletIn, _1, _2)); |
|
} |
|
|
|
void UnregisterValidationInterface(CValidationInterface* pwalletIn) { |
|
g_signals.m_internals->BlockChecked.disconnect(boost::bind(&CValidationInterface::BlockChecked, pwalletIn, _1, _2)); |
|
g_signals.m_internals->Broadcast.disconnect(boost::bind(&CValidationInterface::ResendWalletTransactions, pwalletIn, _1, _2)); |
|
g_signals.m_internals->Inventory.disconnect(boost::bind(&CValidationInterface::Inventory, pwalletIn, _1)); |
|
g_signals.m_internals->SetBestChain.disconnect(boost::bind(&CValidationInterface::SetBestChain, pwalletIn, _1)); |
|
g_signals.m_internals->TransactionAddedToMempool.disconnect(boost::bind(&CValidationInterface::TransactionAddedToMempool, pwalletIn, _1)); |
|
g_signals.m_internals->BlockConnected.disconnect(boost::bind(&CValidationInterface::BlockConnected, pwalletIn, _1, _2, _3)); |
|
g_signals.m_internals->BlockDisconnected.disconnect(boost::bind(&CValidationInterface::BlockDisconnected, pwalletIn, _1)); |
|
g_signals.m_internals->TransactionRemovedFromMempool.disconnect(boost::bind(&CValidationInterface::TransactionRemovedFromMempool, pwalletIn, _1)); |
|
g_signals.m_internals->UpdatedBlockTip.disconnect(boost::bind(&CValidationInterface::UpdatedBlockTip, pwalletIn, _1, _2, _3)); |
|
g_signals.m_internals->NewPoWValidBlock.disconnect(boost::bind(&CValidationInterface::NewPoWValidBlock, pwalletIn, _1, _2)); |
|
} |
|
|
|
void UnregisterAllValidationInterfaces() { |
|
g_signals.m_internals->BlockChecked.disconnect_all_slots(); |
|
g_signals.m_internals->Broadcast.disconnect_all_slots(); |
|
g_signals.m_internals->Inventory.disconnect_all_slots(); |
|
g_signals.m_internals->SetBestChain.disconnect_all_slots(); |
|
g_signals.m_internals->TransactionAddedToMempool.disconnect_all_slots(); |
|
g_signals.m_internals->BlockConnected.disconnect_all_slots(); |
|
g_signals.m_internals->BlockDisconnected.disconnect_all_slots(); |
|
g_signals.m_internals->TransactionRemovedFromMempool.disconnect_all_slots(); |
|
g_signals.m_internals->UpdatedBlockTip.disconnect_all_slots(); |
|
g_signals.m_internals->NewPoWValidBlock.disconnect_all_slots(); |
|
} |
|
|
|
void CallFunctionInValidationInterfaceQueue(std::function<void ()> func) { |
|
g_signals.m_internals->m_schedulerClient.AddToProcessQueue(std::move(func)); |
|
} |
|
|
|
void CMainSignals::MempoolEntryRemoved(CTransactionRef ptx, MemPoolRemovalReason reason) { |
|
if (reason != MemPoolRemovalReason::BLOCK && reason != MemPoolRemovalReason::CONFLICT) { |
|
m_internals->m_schedulerClient.AddToProcessQueue([ptx, this] { |
|
m_internals->TransactionRemovedFromMempool(ptx); |
|
}); |
|
} |
|
} |
|
|
|
void CMainSignals::UpdatedBlockTip(const CBlockIndex *pindexNew, const CBlockIndex *pindexFork, bool fInitialDownload) { |
|
m_internals->m_schedulerClient.AddToProcessQueue([pindexNew, pindexFork, fInitialDownload, this] { |
|
m_internals->UpdatedBlockTip(pindexNew, pindexFork, fInitialDownload); |
|
}); |
|
} |
|
|
|
void CMainSignals::TransactionAddedToMempool(const CTransactionRef &ptx) { |
|
m_internals->m_schedulerClient.AddToProcessQueue([ptx, this] { |
|
m_internals->TransactionAddedToMempool(ptx); |
|
}); |
|
} |
|
|
|
void CMainSignals::BlockConnected(const std::shared_ptr<const CBlock> &pblock, const CBlockIndex *pindex, const std::shared_ptr<const std::vector<CTransactionRef>>& pvtxConflicted) { |
|
m_internals->m_schedulerClient.AddToProcessQueue([pblock, pindex, pvtxConflicted, this] { |
|
m_internals->BlockConnected(pblock, pindex, *pvtxConflicted); |
|
}); |
|
} |
|
|
|
void CMainSignals::BlockDisconnected(const std::shared_ptr<const CBlock> &pblock) { |
|
m_internals->m_schedulerClient.AddToProcessQueue([pblock, this] { |
|
m_internals->BlockDisconnected(pblock); |
|
}); |
|
} |
|
|
|
void CMainSignals::SetBestChain(const CBlockLocator &locator) { |
|
m_internals->m_schedulerClient.AddToProcessQueue([locator, this] { |
|
m_internals->SetBestChain(locator); |
|
}); |
|
} |
|
|
|
void CMainSignals::Inventory(const uint256 &hash) { |
|
m_internals->m_schedulerClient.AddToProcessQueue([hash, this] { |
|
m_internals->Inventory(hash); |
|
}); |
|
} |
|
|
|
void CMainSignals::Broadcast(int64_t nBestBlockTime, CConnman* connman) { |
|
m_internals->Broadcast(nBestBlockTime, connman); |
|
} |
|
|
|
void CMainSignals::BlockChecked(const CBlock& block, const CValidationState& state) { |
|
m_internals->BlockChecked(block, state); |
|
} |
|
|
|
void CMainSignals::NewPoWValidBlock(const CBlockIndex *pindex, const std::shared_ptr<const CBlock> &block) { |
|
m_internals->NewPoWValidBlock(pindex, block); |
|
}
|
|
|