Branch data Line data Source code
1 : : // Copyright (c) 2009-2010 Satoshi Nakamoto
2 : : // Copyright (c) 2009-present The Bitcoin Core developers
3 : : // Distributed under the MIT software license, see the accompanying
4 : : // file COPYING or http://www.opensource.org/licenses/mit-license.php.
5 : :
6 : : #ifndef BITCOIN_VALIDATIONINTERFACE_H
7 : : #define BITCOIN_VALIDATIONINTERFACE_H
8 : :
9 : : #include <kernel/chain.h>
10 : : #include <kernel/cs_main.h>
11 : : #include <primitives/transaction.h>
12 : : #include <sync.h>
13 : :
14 : : #include <cstddef>
15 : : #include <cstdint>
16 : : #include <functional>
17 : : #include <memory>
18 : : #include <vector>
19 : :
20 : : namespace util {
21 : : class TaskRunnerInterface;
22 : : } // namespace util
23 : :
24 : : class BlockValidationState;
25 : : class CBlock;
26 : : class CBlockIndex;
27 : : struct CBlockLocator;
28 : : enum class MemPoolRemovalReason;
29 : : struct RemovedMempoolTransactionInfo;
30 : : struct NewMempoolTransactionInfo;
31 : :
32 : : /**
33 : : * Implement this to subscribe to events generated in validation and mempool
34 : : *
35 : : * Each CValidationInterface() subscriber will receive event callbacks
36 : : * in the order in which the events were generated by validation and mempool.
37 : : * Furthermore, each ValidationInterface() subscriber may assume that
38 : : * callbacks effectively run in a single thread with single-threaded
39 : : * memory consistency. That is, for a given ValidationInterface()
40 : : * instantiation, each callback will complete before the next one is
41 : : * invoked. This means, for example when a block is connected that the
42 : : * UpdatedBlockTip() callback may depend on an operation performed in
43 : : * the BlockConnected() callback without worrying about explicit
44 : : * synchronization. No ordering should be assumed across
45 : : * ValidationInterface() subscribers.
46 : : */
47 [ + - ]: 250210 : class CValidationInterface {
48 : : protected:
49 : : /**
50 : : * Protected destructor so that instances can only be deleted by derived classes.
51 : : * If that restriction is no longer desired, this should be made public and virtual.
52 : : */
53 : : ~CValidationInterface() = default;
54 : : /**
55 : : * Notifies listeners when the block chain tip advances.
56 : : *
57 : : * When multiple blocks are connected at once, UpdatedBlockTip will be called on the final tip
58 : : * but may not be called on every intermediate tip. If the latter behavior is desired,
59 : : * subscribe to BlockConnected() instead.
60 : : *
61 : : * Called on a background thread. Only called for the active chainstate.
62 : : */
63 : 15 : virtual void UpdatedBlockTip(const CBlockIndex *pindexNew, const CBlockIndex *pindexFork, bool fInitialDownload) {}
64 : : /**
65 : : * Notifies listeners any time the block chain tip changes, synchronously.
66 : : */
67 : 75 : virtual void ActiveTipChange(const CBlockIndex& new_tip, bool is_ibd) {};
68 : : /**
69 : : * Notifies listeners of a transaction having been added to mempool.
70 : : *
71 : : * Called on a background thread.
72 : : */
73 : 0 : virtual void TransactionAddedToMempool(const NewMempoolTransactionInfo& tx, uint64_t mempool_sequence) {}
74 : :
75 : : /**
76 : : * Notifies listeners of a transaction leaving mempool.
77 : : *
78 : : * This notification fires for transactions that are removed from the
79 : : * mempool for the following reasons:
80 : : *
81 : : * - EXPIRY (expired from mempool after -mempoolexpiry hours)
82 : : * - SIZELIMIT (removed in size limiting if the mempool exceeds -maxmempool megabytes)
83 : : * - REORG (removed during a reorg)
84 : : * - CONFLICT (removed because it conflicts with in-block transaction)
85 : : * - REPLACED (removed due to RBF replacement)
86 : : *
87 : : * This does not fire for transactions that are removed from the mempool
88 : : * because they have been included in a block. Any client that is interested
89 : : * in transactions removed from the mempool for inclusion in a block can learn
90 : : * about those transactions from the MempoolTransactionsRemovedForBlock notification.
91 : : *
92 : : * Transactions that are removed from the mempool because they conflict
93 : : * with a transaction in the new block will have
94 : : * TransactionRemovedFromMempool events fired *before* the BlockConnected
95 : : * event is fired. If multiple blocks are connected in one step, then the
96 : : * ordering could be:
97 : : *
98 : : * - TransactionRemovedFromMempool(tx1 from block A)
99 : : * - TransactionRemovedFromMempool(tx2 from block A)
100 : : * - TransactionRemovedFromMempool(tx1 from block B)
101 : : * - TransactionRemovedFromMempool(tx2 from block B)
102 : : * - BlockConnected(A)
103 : : * - BlockConnected(B)
104 : : *
105 : : * Called on a background thread.
106 : : */
107 : 0 : virtual void TransactionRemovedFromMempool(const CTransactionRef& tx, MemPoolRemovalReason reason, uint64_t mempool_sequence) {}
108 : : /*
109 : : * Notifies listeners of transactions removed from the mempool as
110 : : * as a result of new block being connected.
111 : : * MempoolTransactionsRemovedForBlock will be fired before BlockConnected.
112 : : *
113 : : * Called on a background thread.
114 : : */
115 : 111 : virtual void MempoolTransactionsRemovedForBlock(const std::vector<RemovedMempoolTransactionInfo>& txs_removed_for_block, unsigned int nBlockHeight) {}
116 : : /**
117 : : * Notifies listeners of a block being connected.
118 : : *
119 : : * Called on a background thread.
120 : : */
121 : 0 : virtual void BlockConnected(ChainstateRole role, const std::shared_ptr<const CBlock> &block, const CBlockIndex *pindex) {}
122 : : /**
123 : : * Notifies listeners of a block being disconnected
124 : : * Provides the block that was disconnected.
125 : : *
126 : : * Called on a background thread. Only called for the active chainstate, since
127 : : * background chainstates should never disconnect blocks.
128 : : */
129 : 5 : virtual void BlockDisconnected(const std::shared_ptr<const CBlock> &block, const CBlockIndex* pindex) {}
130 : : /**
131 : : * Notifies listeners of the new active block chain on-disk.
132 : : *
133 : : * Prior to this callback, any updates are not guaranteed to persist on disk
134 : : * (ie clients need to handle shutdown/restart safety by being able to
135 : : * understand when some updates were lost due to unclean shutdown).
136 : : *
137 : : * When this callback is invoked, the validation changes done by any prior
138 : : * callback are guaranteed to exist on disk and survive a restart, including
139 : : * an unclean shutdown.
140 : : *
141 : : * Provides a locator describing the best chain, which is likely useful for
142 : : * storing current state on disk in client DBs.
143 : : *
144 : : * Called on a background thread.
145 : : */
146 : 0 : virtual void ChainStateFlushed(ChainstateRole role, const CBlockLocator &locator) {}
147 : : /**
148 : : * Notifies listeners of a block validation result.
149 : : * If the provided BlockValidationState IsValid, the provided block
150 : : * is guaranteed to be the current best block at the time the
151 : : * callback was generated (not necessarily now).
152 : : */
153 : 176 : virtual void BlockChecked(const CBlock&, const BlockValidationState&) {}
154 : : /**
155 : : * Notifies listeners that a block which builds directly on our current tip
156 : : * has been received and connected to the headers tree, though not validated yet.
157 : : */
158 : 18 : virtual void NewPoWValidBlock(const CBlockIndex *pindex, const std::shared_ptr<const CBlock>& block) {};
159 : : friend class ValidationSignals;
160 : : friend class ValidationInterfaceTest;
161 : : };
162 : :
163 : : class ValidationSignalsImpl;
164 : : class ValidationSignals {
165 : : private:
166 : : std::unique_ptr<ValidationSignalsImpl> m_internals;
167 : :
168 : : public:
169 : : // The task runner will block validation if it calls its insert method's
170 : : // func argument synchronously. In this class func contains a loop that
171 : : // dispatches a single validation event to all subscribers sequentially.
172 : : explicit ValidationSignals(std::unique_ptr<util::TaskRunnerInterface> task_runner);
173 : :
174 : : ~ValidationSignals();
175 : :
176 : : /** Call any remaining callbacks on the calling thread */
177 : : void FlushBackgroundCallbacks();
178 : :
179 : : size_t CallbacksPending();
180 : :
181 : : /** Register subscriber */
182 : : void RegisterValidationInterface(CValidationInterface* callbacks);
183 : : /** Unregister subscriber. DEPRECATED. This is not safe to use when the RPC server or main message handler thread is running. */
184 : : void UnregisterValidationInterface(CValidationInterface* callbacks);
185 : : /** Unregister all subscribers */
186 : : void UnregisterAllValidationInterfaces();
187 : :
188 : : // Alternate registration functions that release a shared_ptr after the last
189 : : // notification is sent. These are useful for race-free cleanup, since
190 : : // unregistration is nonblocking and can return before the last notification is
191 : : // processed.
192 : : /** Register subscriber */
193 : : void RegisterSharedValidationInterface(std::shared_ptr<CValidationInterface> callbacks);
194 : : /** Unregister subscriber */
195 : : void UnregisterSharedValidationInterface(std::shared_ptr<CValidationInterface> callbacks);
196 : :
197 : : /**
198 : : * Pushes a function to callback onto the notification queue, guaranteeing any
199 : : * callbacks generated prior to now are finished when the function is called.
200 : : *
201 : : * Be very careful blocking on func to be called if any locks are held -
202 : : * validation interface clients may not be able to make progress as they often
203 : : * wait for things like cs_main, so blocking until func is called with cs_main
204 : : * will result in a deadlock (that DEBUG_LOCKORDER will miss).
205 : : */
206 : : void CallFunctionInValidationInterfaceQueue(std::function<void ()> func);
207 : :
208 : : /**
209 : : * This is a synonym for the following, which asserts certain locks are not
210 : : * held:
211 : : * std::promise<void> promise;
212 : : * CallFunctionInValidationInterfaceQueue([&promise] {
213 : : * promise.set_value();
214 : : * });
215 : : * promise.get_future().wait();
216 : : */
217 : : void SyncWithValidationInterfaceQueue() LOCKS_EXCLUDED(cs_main);
218 : :
219 : : void UpdatedBlockTip(const CBlockIndex *, const CBlockIndex *, bool fInitialDownload);
220 : : void ActiveTipChange(const CBlockIndex&, bool);
221 : : void TransactionAddedToMempool(const NewMempoolTransactionInfo&, uint64_t mempool_sequence);
222 : : void TransactionRemovedFromMempool(const CTransactionRef&, MemPoolRemovalReason, uint64_t mempool_sequence);
223 : : void MempoolTransactionsRemovedForBlock(const std::vector<RemovedMempoolTransactionInfo>&, unsigned int nBlockHeight);
224 : : void BlockConnected(ChainstateRole, const std::shared_ptr<const CBlock> &, const CBlockIndex *pindex);
225 : : void BlockDisconnected(const std::shared_ptr<const CBlock> &, const CBlockIndex* pindex);
226 : : void ChainStateFlushed(ChainstateRole, const CBlockLocator &);
227 : : void BlockChecked(const CBlock&, const BlockValidationState&);
228 : : void NewPoWValidBlock(const CBlockIndex *, const std::shared_ptr<const CBlock>&);
229 : : };
230 : :
231 : : #endif // BITCOIN_VALIDATIONINTERFACE_H
|