validationinterface.h

Go to the documentation of this file.

// 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.
 
#ifndef BITCOIN_VALIDATIONINTERFACE_H
#define BITCOIN_VALIDATIONINTERFACE_H
 
#include <kernel/cs_main.h>
#include <primitives/transaction.h> // CTransaction(Ref)
#include <sync.h>
 
#include <functional>
#include <memory>
 
class BlockValidationState;
class CBlock;
class CBlockIndex;
struct CBlockLocator;
class Coin;
class CValidationInterface;
class CScheduler;
enum class MemPoolRemovalReason;
 
void RegisterValidationInterface(CValidationInterface *callbacks);
void UnregisterValidationInterface(CValidationInterface *callbacks);
void UnregisterAllValidationInterfaces();
 
// Alternate registration functions that release a shared_ptr after the last
// notification is sent. These are useful for race-free cleanup, since
// unregistration is nonblocking and can return before the last notification is
// processed.
void RegisterSharedValidationInterface(
    std::shared_ptr<CValidationInterface> callbacks);
void UnregisterSharedValidationInterface(
    std::shared_ptr<CValidationInterface> callbacks);
 
void CallFunctionInValidationInterfaceQueue(std::function<void()> func);
void SyncWithValidationInterfaceQueue() LOCKS_EXCLUDED(cs_main);
 
class CValidationInterface {
protected:
    ~CValidationInterface() = default;
    virtual void UpdatedBlockTip(const CBlockIndex *pindexNew,
                                 const CBlockIndex *pindexFork,
                                 bool fInitialDownload) {}
    virtual void TransactionAddedToMempool(
        const CTransactionRef &tx,
        std::shared_ptr<const std::vector<Coin>> spent_coins,
        uint64_t mempool_sequence) {}
 
    virtual void TransactionRemovedFromMempool(const CTransactionRef &tx,
                                               MemPoolRemovalReason reason,
                                               uint64_t mempool_sequence) {}
    virtual void BlockConnected(const std::shared_ptr<const CBlock> &block,
                                const CBlockIndex *pindex) {}
    virtual void BlockDisconnected(const std::shared_ptr<const CBlock> &block,
                                   const CBlockIndex *pindex) {}
    virtual void ChainStateFlushed(const CBlockLocator &locator) {}
    virtual void BlockChecked(const CBlock &, const BlockValidationState &) {}
    virtual void NewPoWValidBlock(const CBlockIndex *pindex,
                                  const std::shared_ptr<const CBlock> &block){};
 
    virtual void BlockFinalized(const CBlockIndex *pindex){};
    virtual void BlockInvalidated(const CBlockIndex *pindex,
                                  const std::shared_ptr<const CBlock> &block){};
 
    friend class CMainSignals;
    friend class ValidationInterfaceTest;
};
 
class MainSignalsImpl;
class CMainSignals {
private:
    std::unique_ptr<MainSignalsImpl> m_internals;
 
    friend void ::RegisterSharedValidationInterface(
        std::shared_ptr<CValidationInterface>);
    friend void ::UnregisterValidationInterface(CValidationInterface *);
    friend void ::UnregisterAllValidationInterfaces();
    friend void ::CallFunctionInValidationInterfaceQueue(
        std::function<void()> func);
 
public:
    void RegisterBackgroundSignalScheduler(CScheduler &scheduler);
    void UnregisterBackgroundSignalScheduler();
    void FlushBackgroundCallbacks();
 
    size_t CallbacksPending();
 
    void UpdatedBlockTip(const CBlockIndex *, const CBlockIndex *,
                         bool fInitialDownload);
    void TransactionAddedToMempool(const CTransactionRef &,
                                   std::shared_ptr<const std::vector<Coin>>,
                                   uint64_t mempool_sequence);
    void TransactionRemovedFromMempool(const CTransactionRef &,
                                       MemPoolRemovalReason,
                                       uint64_t mempool_sequence);
    void BlockConnected(const std::shared_ptr<const CBlock> &,
                        const CBlockIndex *pindex);
    void BlockDisconnected(const std::shared_ptr<const CBlock> &,
                           const CBlockIndex *pindex);
    void ChainStateFlushed(const CBlockLocator &);
    void BlockChecked(const CBlock &, const BlockValidationState &);
    void NewPoWValidBlock(const CBlockIndex *,
                          const std::shared_ptr<const CBlock> &);
    void BlockFinalized(const CBlockIndex *pindex);
    void BlockInvalidated(const CBlockIndex *pindex,
                          const std::shared_ptr<const CBlock> &block);
};
 
CMainSignals &GetMainSignals();
 
#endif // BITCOIN_VALIDATIONINTERFACE_H