Branch data Line data Source code
1 : : // Copyright (c) 2024-present The Bitcoin Core developers
2 : : // Distributed under the MIT software license, see the accompanying
3 : : // file COPYING or http://www.opensource.org/licenses/mit-license.php.
4 : :
5 : : #include <node/txdownloadman_impl.h>
6 : : #include <node/txdownloadman.h>
7 : :
8 : : #include <chain.h>
9 : : #include <consensus/validation.h>
10 : : #include <txmempool.h>
11 : : #include <util/log.h>
12 : : #include <validation.h>
13 : : #include <validationinterface.h>
14 : :
15 : : namespace node {
16 : : // TxDownloadManager wrappers
17 : 1273 : TxDownloadManager::TxDownloadManager(const TxDownloadOptions& options) :
18 : 1273 : m_impl{std::make_unique<TxDownloadManagerImpl>(options)}
19 : 1273 : {}
20 : 1273 : TxDownloadManager::~TxDownloadManager() = default;
21 : :
22 : 106226 : void TxDownloadManager::ActiveTipChange()
23 : : {
24 : 106226 : m_impl->ActiveTipChange();
25 : 106226 : }
26 : 114099 : void TxDownloadManager::BlockConnected(const std::shared_ptr<const CBlock>& pblock)
27 : : {
28 : 114099 : m_impl->BlockConnected(pblock);
29 : 114099 : }
30 : 13605 : void TxDownloadManager::BlockDisconnected()
31 : : {
32 : 13605 : m_impl->BlockDisconnected();
33 : 13605 : }
34 : 1681 : void TxDownloadManager::ConnectedPeer(NodeId nodeid, const TxDownloadConnectionInfo& info)
35 : : {
36 : 1681 : m_impl->ConnectedPeer(nodeid, info);
37 : 1681 : }
38 : 1863 : void TxDownloadManager::DisconnectedPeer(NodeId nodeid)
39 : : {
40 : 1863 : m_impl->DisconnectedPeer(nodeid);
41 : 1863 : }
42 : 25979 : bool TxDownloadManager::AddTxAnnouncement(NodeId peer, const GenTxid& gtxid, std::chrono::microseconds now)
43 : : {
44 : 25979 : return m_impl->AddTxAnnouncement(peer, gtxid, now);
45 : : }
46 : 415415 : std::vector<GenTxid> TxDownloadManager::GetRequestsToSend(NodeId nodeid, std::chrono::microseconds current_time)
47 : : {
48 : 415415 : return m_impl->GetRequestsToSend(nodeid, current_time);
49 : : }
50 : 8 : void TxDownloadManager::ReceivedNotFound(NodeId nodeid, const std::vector<GenTxid>& gtxids)
51 : : {
52 : 8 : m_impl->ReceivedNotFound(nodeid, gtxids);
53 : 8 : }
54 : 8128 : void TxDownloadManager::MempoolAcceptedTx(const CTransactionRef& tx)
55 : : {
56 : 8128 : m_impl->MempoolAcceptedTx(tx);
57 : 8128 : }
58 : 844 : RejectedTxTodo TxDownloadManager::MempoolRejectedTx(const CTransactionRef& ptx, const TxValidationState& state, NodeId nodeid, bool first_time_failure)
59 : : {
60 : 844 : return m_impl->MempoolRejectedTx(ptx, state, nodeid, first_time_failure);
61 : : }
62 : 3 : void TxDownloadManager::MempoolRejectedPackage(const Package& package)
63 : : {
64 : 3 : m_impl->MempoolRejectedPackage(package);
65 : 3 : }
66 : 12730 : std::pair<bool, std::optional<PackageToValidate>> TxDownloadManager::ReceivedTx(NodeId nodeid, const CTransactionRef& ptx)
67 : : {
68 : 12730 : return m_impl->ReceivedTx(nodeid, ptx);
69 : : }
70 : 158030 : bool TxDownloadManager::HaveMoreWork(NodeId nodeid) const
71 : : {
72 : 158030 : return m_impl->HaveMoreWork(nodeid);
73 : : }
74 : 420729 : CTransactionRef TxDownloadManager::GetTxToReconsider(NodeId nodeid)
75 : : {
76 : 420729 : return m_impl->GetTxToReconsider(nodeid);
77 : : }
78 : 952 : void TxDownloadManager::CheckIsEmpty() const
79 : : {
80 : 952 : m_impl->CheckIsEmpty();
81 : 952 : }
82 : 1864 : void TxDownloadManager::CheckIsEmpty(NodeId nodeid) const
83 : : {
84 : 1864 : m_impl->CheckIsEmpty(nodeid);
85 : 1864 : }
86 : 228 : std::vector<TxOrphanage::OrphanInfo> TxDownloadManager::GetOrphanTransactions() const
87 : : {
88 : 228 : return m_impl->GetOrphanTransactions();
89 : : }
90 : :
91 : : // TxDownloadManagerImpl
92 : 106226 : void TxDownloadManagerImpl::ActiveTipChange()
93 : : {
94 : 106226 : RecentRejectsFilter().reset();
95 : 106226 : RecentRejectsReconsiderableFilter().reset();
96 : 106226 : }
97 : :
98 : 114099 : void TxDownloadManagerImpl::BlockConnected(const std::shared_ptr<const CBlock>& pblock)
99 : : {
100 : 114099 : m_orphanage->EraseForBlock(*pblock);
101 : :
102 [ + + ]: 273388 : for (const auto& ptx : pblock->vtx) {
103 : : // Reconsider potential child transactions.
104 : 159289 : m_orphanage->AddChildrenToWorkSet(*ptx, m_opts.m_rng);
105 : :
106 : 159289 : RecentConfirmedTransactionsFilter().insert(ptx->GetHash().ToUint256());
107 [ + + ]: 159289 : if (ptx->HasWitness()) {
108 : 124060 : RecentConfirmedTransactionsFilter().insert(ptx->GetWitnessHash().ToUint256());
109 : : }
110 : 159289 : m_txrequest.ForgetTxHash(ptx->GetHash().ToUint256());
111 : 159289 : m_txrequest.ForgetTxHash(ptx->GetWitnessHash().ToUint256());
112 : : }
113 : 114099 : }
114 : :
115 : 13605 : void TxDownloadManagerImpl::BlockDisconnected()
116 : : {
117 : : // To avoid relay problems with transactions that were previously
118 : : // confirmed, clear our filter of recently confirmed transactions whenever
119 : : // there's a reorg.
120 : : // This means that in a 1-block reorg (where 1 block is disconnected and
121 : : // then another block reconnected), our filter will drop to having only one
122 : : // block's worth of transactions in it, but that should be fine, since
123 : : // presumably the most common case of relaying a confirmed transaction
124 : : // should be just after a new block containing it is found.
125 : 13605 : RecentConfirmedTransactionsFilter().reset();
126 : 13605 : }
127 : :
128 : 58840 : bool TxDownloadManagerImpl::AlreadyHaveTx(const GenTxid& gtxid, bool include_reconsiderable)
129 : : {
130 [ + + - ]: 58840 : const uint256& hash = gtxid.ToUint256();
131 : :
132 : : // Never query by txid: it is possible that the transaction in the orphanage has the same
133 : : // txid but a different witness, which would give us a false positive result. If we decided
134 : : // not to request the transaction based on this result, an attacker could prevent us from
135 : : // downloading a transaction by intentionally creating a malleated version of it. While
136 : : // only one (or none!) of these transactions can ultimately be confirmed, we have no way of
137 : : // discerning which one that is, so the orphanage can store multiple transactions with the
138 : : // same txid.
139 : : //
140 : : // While we won't query by txid, we can try to "guess" what the wtxid is based on the txid.
141 : : // A non-segwit transaction's txid == wtxid. Query this txhash "casted" to a wtxid. This will
142 : : // help us find non-segwit transactions, saving bandwidth, and should have no false positives.
143 [ + + ]: 58840 : if (m_orphanage->HaveTx(Wtxid::FromUint256(hash))) return true;
144 : :
145 [ + + + + ]: 55768 : if (include_reconsiderable && RecentRejectsReconsiderableFilter().contains(hash)) return true;
146 : :
147 [ + + ]: 55755 : if (RecentConfirmedTransactionsFilter().contains(hash)) return true;
148 : :
149 [ + + + + ]: 111343 : return RecentRejectsFilter().contains(hash) || std::visit([&](const auto& id) { return m_opts.m_mempool.exists(id); }, gtxid);
150 : : }
151 : :
152 : 1750 : void TxDownloadManagerImpl::ConnectedPeer(NodeId nodeid, const TxDownloadConnectionInfo& info)
153 : : {
154 : : // If already connected (shouldn't happen in practice), exit early.
155 [ + - ]: 1750 : if (m_peer_info.contains(nodeid)) return;
156 : :
157 : 1750 : m_peer_info.try_emplace(nodeid, info);
158 [ + + ]: 1750 : if (info.m_wtxid_relay) m_num_wtxid_peers += 1;
159 : : }
160 : :
161 : 1863 : void TxDownloadManagerImpl::DisconnectedPeer(NodeId nodeid)
162 : : {
163 : 1863 : m_orphanage->EraseForPeer(nodeid);
164 : 1863 : m_txrequest.DisconnectedPeer(nodeid);
165 : :
166 [ + + ]: 1863 : if (auto it = m_peer_info.find(nodeid); it != m_peer_info.end()) {
167 [ + + ]: 1680 : if (it->second.m_connection_info.m_wtxid_relay) m_num_wtxid_peers -= 1;
168 : 1680 : m_peer_info.erase(it);
169 : : }
170 : :
171 : 1863 : }
172 : :
173 : 26075 : bool TxDownloadManagerImpl::AddTxAnnouncement(NodeId peer, const GenTxid& gtxid, std::chrono::microseconds now)
174 : : {
175 : : // If this is an orphan we are trying to resolve, consider this peer as a orphan resolution candidate instead.
176 : : // - is wtxid matching something in orphanage
177 : : // - exists in orphanage
178 : : // - peer can be an orphan resolution candidate
179 [ + + ]: 26075 : if (const auto* wtxid = std::get_if<Wtxid>(>xid)) {
180 [ + + ]: 26005 : if (auto orphan_tx{m_orphanage->GetTx(*wtxid)}) {
181 [ + - ]: 7 : auto unique_parents{GetUniqueParents(*orphan_tx)};
182 [ + - ]: 7 : std::erase_if(unique_parents, [&](const auto& txid) {
183 : 8 : return AlreadyHaveTx(txid, /*include_reconsiderable=*/false);
184 : : });
185 : :
186 : : // The missing parents may have all been rejected or accepted since the orphan was added to the orphanage.
187 : : // Do not delete from the orphanage, as it may be queued for processing.
188 [ + - ]: 7 : if (unique_parents.empty()) {
189 : : return true;
190 : : }
191 : :
192 [ + - + + ]: 7 : if (MaybeAddOrphanResolutionCandidate(unique_parents, *wtxid, peer, now)) {
193 [ + - ]: 3 : m_orphanage->AddAnnouncer(orphan_tx->GetWitnessHash(), peer);
194 : : }
195 : :
196 : : // Return even if the peer isn't an orphan resolution candidate. This would be caught by AlreadyHaveTx.
197 : 7 : return true;
198 [ + - ]: 26012 : }
199 : : }
200 : :
201 : : // If this is an inv received from a peer and we already have it, we can drop it.
202 [ + + ]: 26068 : if (AlreadyHaveTx(gtxid, /*include_reconsiderable=*/true)) return true;
203 : :
204 : 19550 : auto it = m_peer_info.find(peer);
205 [ + - ]: 19550 : if (it == m_peer_info.end()) return false;
206 [ + + ]: 19550 : const auto& info = it->second.m_connection_info;
207 [ + + + + ]: 19550 : if (!info.m_relay_permissions && m_txrequest.Count(peer) >= MAX_PEER_TX_ANNOUNCEMENTS) {
208 : : // Too many queued announcements for this peer
209 : : return false;
210 : : }
211 : : // Decide the TxRequestTracker parameters for this announcement:
212 : : // - "preferred": if fPreferredDownload is set (= outbound, or NetPermissionFlags::NoBan permission)
213 : : // - "reqtime": current time plus delays for:
214 : : // - NONPREF_PEER_TX_DELAY for announcements from non-preferred connections
215 : : // - TXID_RELAY_DELAY for txid announcements while wtxid peers are available
216 : : // - OVERLOADED_PEER_TX_DELAY for announcements from peers which have at least
217 : : // MAX_PEER_TX_REQUEST_IN_FLIGHT requests in flight (and don't have NetPermissionFlags::Relay).
218 : 19549 : auto delay{0us};
219 [ + + ]: 19549 : if (!info.m_preferred) delay += NONPREF_PEER_TX_DELAY;
220 [ + + + + ]: 19549 : if (!gtxid.IsWtxid() && m_num_wtxid_peers > 0) delay += TXID_RELAY_DELAY;
221 [ + + + + ]: 19549 : const bool overloaded = !info.m_relay_permissions && m_txrequest.CountInFlight(peer) >= MAX_PEER_TX_REQUEST_IN_FLIGHT;
222 : 2 : if (overloaded) delay += OVERLOADED_PEER_TX_DELAY;
223 : :
224 : 19549 : m_txrequest.ReceivedInv(peer, gtxid, info.m_preferred, now + delay);
225 : :
226 : 19549 : return false;
227 : : }
228 : :
229 : 670 : bool TxDownloadManagerImpl::MaybeAddOrphanResolutionCandidate(const std::vector<Txid>& unique_parents, const Wtxid& wtxid, NodeId nodeid, std::chrono::microseconds now)
230 : : {
231 : 670 : auto it_peer = m_peer_info.find(nodeid);
232 [ + - ]: 670 : if (it_peer == m_peer_info.end()) return false;
233 [ + + ]: 670 : if (m_orphanage->HaveTxFromPeer(wtxid, nodeid)) return false;
234 : :
235 : 666 : const auto& peer_entry = m_peer_info.at(nodeid);
236 : 666 : const auto& info = peer_entry.m_connection_info;
237 : :
238 : : // TODO: add delays and limits based on the amount of orphan resolution we are already doing
239 : : // with this peer, how much they are using the orphanage, etc.
240 [ + - ]: 666 : if (!info.m_relay_permissions) {
241 : : // This mirrors the delaying and dropping behavior in AddTxAnnouncement in order to preserve
242 : : // existing behavior: drop if we are tracking too many invs for this peer already. Each
243 : : // orphan resolution involves at least 1 transaction request which may or may not be
244 : : // currently tracked in m_txrequest, so we include that in the count.
245 [ - + + - ]: 666 : if (m_txrequest.Count(nodeid) + unique_parents.size() > MAX_PEER_TX_ANNOUNCEMENTS) return false;
246 : : }
247 : :
248 : 666 : std::chrono::seconds delay{0s};
249 [ + + ]: 666 : if (!info.m_preferred) delay += NONPREF_PEER_TX_DELAY;
250 : : // The orphan wtxid is used, but resolution entails requesting the parents by txid. Sometimes
251 : : // parent and child are announced and thus requested around the same time, and we happen to
252 : : // receive child sooner. Waiting a few seconds may allow us to cancel the orphan resolution
253 : : // request if the parent arrives in that time.
254 [ + - ]: 666 : if (m_num_wtxid_peers > 0) delay += TXID_RELAY_DELAY;
255 [ + - - + ]: 666 : const bool overloaded = !info.m_relay_permissions && m_txrequest.CountInFlight(nodeid) >= MAX_PEER_TX_REQUEST_IN_FLIGHT;
256 : 0 : if (overloaded) delay += OVERLOADED_PEER_TX_DELAY;
257 : :
258 : : // Treat finding orphan resolution candidate as equivalent to the peer announcing all missing parents.
259 : : // In the future, orphan resolution may include more explicit steps
260 [ + + ]: 1352 : for (const auto& parent_txid : unique_parents) {
261 : 686 : m_txrequest.ReceivedInv(nodeid, parent_txid, info.m_preferred, now + delay);
262 : : }
263 [ + - + - ]: 666 : LogDebug(BCLog::TXPACKAGES, "added peer=%d as a candidate for resolving orphan %s\n", nodeid, wtxid.ToString());
264 : : return true;
265 : : }
266 : :
267 : 415415 : std::vector<GenTxid> TxDownloadManagerImpl::GetRequestsToSend(NodeId nodeid, std::chrono::microseconds current_time)
268 : : {
269 : 415415 : std::vector<GenTxid> requests;
270 : 415415 : std::vector<std::pair<NodeId, GenTxid>> expired;
271 [ + - ]: 415415 : auto requestable = m_txrequest.GetRequestable(nodeid, current_time, &expired);
272 [ + - + + ]: 415444 : for (const auto& [expired_nodeid, gtxid] : expired) {
273 [ + - + - : 75 : LogDebug(BCLog::NET, "timeout of inflight %s %s from peer=%d\n", gtxid.IsWtxid() ? "wtx" : "tx",
+ + - + -
+ + + - ]
274 : : gtxid.ToUint256().ToString(), expired_nodeid);
275 : : }
276 [ + + ]: 434728 : for (const GenTxid& gtxid : requestable) {
277 [ + - + + ]: 19313 : if (!AlreadyHaveTx(gtxid, /*include_reconsiderable=*/false)) {
278 [ + - + - : 38744 : LogDebug(BCLog::NET, "Requesting %s %s peer=%d\n", gtxid.IsWtxid() ? "wtx" : "tx",
+ + - + -
+ + + - ]
279 : : gtxid.ToUint256().ToString(), nodeid);
280 [ + - ]: 19257 : requests.emplace_back(gtxid);
281 [ + + - + : 38514 : m_txrequest.RequestedTx(nodeid, gtxid.ToUint256(), current_time + GETDATA_TX_INTERVAL);
- ]
282 : : } else {
283 : : // We have already seen this transaction, no need to download. This is just a belt-and-suspenders, as
284 : : // this should already be called whenever a transaction becomes AlreadyHaveTx().
285 [ - + - + : 112 : m_txrequest.ForgetTxHash(gtxid.ToUint256());
- ]
286 : : }
287 : : }
288 : 415415 : return requests;
289 : 415415 : }
290 : :
291 : 8 : void TxDownloadManagerImpl::ReceivedNotFound(NodeId nodeid, const std::vector<GenTxid>& gtxids)
292 : : {
293 [ + + ]: 16 : for (const auto& gtxid : gtxids) {
294 : : // If we receive a NOTFOUND message for a tx we requested, mark the announcement for it as
295 : : // completed in TxRequestTracker.
296 [ + + - ]: 16 : m_txrequest.ReceivedResponse(nodeid, gtxid.ToUint256());
297 : : }
298 : 8 : }
299 : :
300 : 52 : std::optional<PackageToValidate> TxDownloadManagerImpl::Find1P1CPackage(const CTransactionRef& ptx, NodeId nodeid)
301 : : {
302 : 52 : const auto& parent_wtxid{ptx->GetWitnessHash()};
303 : :
304 : 52 : Assume(RecentRejectsReconsiderableFilter().contains(parent_wtxid.ToUint256()));
305 : :
306 : : // Only consider children from this peer. This helps prevent censorship attempts in which an attacker
307 : : // sends lots of fake children for the parent, and we (unluckily) keep selecting the fake
308 : : // children instead of the real one provided by the honest peer. Since we track all announcers
309 : : // of an orphan, this does not exclude parent + orphan pairs that we happened to request from
310 : : // different peers.
311 : 52 : const auto cpfp_candidates_same_peer{m_orphanage->GetChildrenFromSamePeer(ptx, nodeid)};
312 : :
313 : : // These children should be sorted from newest to oldest. In the (probably uncommon) case
314 : : // of children that replace each other, this helps us accept the highest feerate (probably the
315 : : // most recent) one efficiently.
316 [ + + ]: 52 : for (const auto& child : cpfp_candidates_same_peer) {
317 [ + + + - : 90 : Package maybe_cpfp_package{ptx, child};
- - - - ]
318 [ + - + - : 60 : if (!RecentRejectsReconsiderableFilter().contains(GetPackageHash(maybe_cpfp_package)) &&
+ - + - +
- ]
319 [ + - + - ]: 30 : !RecentRejectsFilter().contains(child->GetHash().ToUint256())) {
320 [ + - ]: 30 : return PackageToValidate{ptx, child, nodeid, nodeid};
321 : : }
322 : 30 : }
323 : 22 : return std::nullopt;
324 [ + - + - : 142 : }
+ - ]
325 : :
326 : 8128 : void TxDownloadManagerImpl::MempoolAcceptedTx(const CTransactionRef& tx)
327 : : {
328 : : // As this version of the transaction was acceptable, we can forget about any requests for it.
329 : : // No-op if the tx is not in txrequest.
330 : 8128 : m_txrequest.ForgetTxHash(tx->GetHash().ToUint256());
331 : 8128 : m_txrequest.ForgetTxHash(tx->GetWitnessHash().ToUint256());
332 : :
333 : 8128 : m_orphanage->AddChildrenToWorkSet(*tx, m_opts.m_rng);
334 : : // If it came from the orphanage, remove it. No-op if the tx is not in txorphanage.
335 : 8128 : m_orphanage->EraseTx(tx->GetWitnessHash());
336 : 8128 : }
337 : :
338 : 704 : std::vector<Txid> TxDownloadManagerImpl::GetUniqueParents(const CTransaction& tx)
339 : : {
340 : 704 : std::vector<Txid> unique_parents;
341 [ - + + - ]: 704 : unique_parents.reserve(tx.vin.size());
342 [ + + ]: 1529 : for (const CTxIn& txin : tx.vin) {
343 : : // We start with all parents, and then remove duplicates below.
344 [ + - ]: 825 : unique_parents.push_back(txin.prevout.hash);
345 : : }
346 : :
347 : 704 : std::sort(unique_parents.begin(), unique_parents.end());
348 : 704 : unique_parents.erase(std::unique(unique_parents.begin(), unique_parents.end()), unique_parents.end());
349 : :
350 : 704 : return unique_parents;
351 : 0 : }
352 : :
353 : 961 : node::RejectedTxTodo TxDownloadManagerImpl::MempoolRejectedTx(const CTransactionRef& ptx, const TxValidationState& state, NodeId nodeid, bool first_time_failure)
354 : : {
355 [ + + ]: 961 : const CTransaction& tx{*ptx};
356 : : // Results returned to caller
357 : : // Whether we should call AddToCompactExtraTransactions at the end
358 : 961 : bool add_extra_compact_tx{first_time_failure};
359 : : // Hashes to pass to AddKnownTx later
360 : 961 : std::vector<Txid> unique_parents;
361 : : // Populated if failure is reconsiderable and eligible package is found.
362 : 961 : std::optional<node::PackageToValidate> package_to_validate;
363 : :
364 [ + + ]: 961 : if (state.GetResult() == TxValidationResult::TX_MISSING_INPUTS) {
365 : : // Only process a new orphan if this is a first time failure, as otherwise it must be either
366 : : // already in orphanage or from 1p1c processing.
367 [ + - + - : 697 : if (first_time_failure && !RecentRejectsFilter().contains(ptx->GetWitnessHash().ToUint256())) {
+ - + - ]
368 : 697 : bool fRejectedParents = false; // It may be the case that the orphans parents have all been rejected
369 : :
370 : : // Deduplicate parent txids, so that we don't have to loop over
371 : : // the same parent txid more than once down below.
372 [ + - ]: 1394 : unique_parents = GetUniqueParents(tx);
373 : :
374 : : // Distinguish between parents in m_lazy_recent_rejects and m_lazy_recent_rejects_reconsiderable.
375 : : // We can tolerate having up to 1 parent in m_lazy_recent_rejects_reconsiderable since we
376 : : // submit 1p1c packages. However, fail immediately if any are in m_lazy_recent_rejects.
377 : 697 : std::optional<Txid> rejected_parent_reconsiderable;
378 [ + + ]: 1440 : for (const Txid& parent_txid : unique_parents) {
379 [ + - + - : 781 : if (RecentRejectsFilter().contains(parent_txid.ToUint256())) {
+ + ]
380 : : fRejectedParents = true;
381 : : break;
382 [ + - + - : 762 : } else if (RecentRejectsReconsiderableFilter().contains(parent_txid.ToUint256()) &&
+ + + + ]
383 [ + - ]: 17 : !m_opts.m_mempool.exists(parent_txid)) {
384 : : // More than 1 parent in m_lazy_recent_rejects_reconsiderable: 1p1c will not be
385 : : // sufficient to accept this package, so just give up here.
386 [ + + ]: 15 : if (rejected_parent_reconsiderable.has_value()) {
387 : : fRejectedParents = true;
388 : : break;
389 : : }
390 : 13 : rejected_parent_reconsiderable = parent_txid;
391 : : }
392 : : }
393 [ + + ]: 697 : if (!fRejectedParents) {
394 : : // Filter parents that we already have.
395 : : // Exclude m_lazy_recent_rejects_reconsiderable: the missing parent may have been
396 : : // previously rejected for being too low feerate. This orphan might CPFP it.
397 [ + - ]: 659 : std::erase_if(unique_parents, [&](const auto& txid) {
398 : 721 : return AlreadyHaveTx(txid, /*include_reconsiderable=*/false);
399 : : });
400 : 659 : const auto now{GetTime<std::chrono::microseconds>()};
401 [ + - ]: 659 : const auto& wtxid = ptx->GetWitnessHash();
402 : : // Potentially flip add_extra_compact_tx to false if tx is already in orphanage, which
403 : : // means it was already added to vExtraTxnForCompact.
404 [ + - ]: 659 : add_extra_compact_tx &= !m_orphanage->HaveTx(wtxid);
405 : :
406 : : // If there is no candidate for orphan resolution, AddTx will not be called. This means
407 : : // that if a peer is overloading us with invs and orphans, they will eventually not be
408 : : // able to add any more transactions to the orphanage.
409 : : //
410 : : // Search by txid and, if the tx has a witness, wtxid
411 [ + - ]: 659 : std::vector<NodeId> orphan_resolution_candidates{nodeid};
412 [ + - ]: 659 : m_txrequest.GetCandidatePeers(ptx->GetHash().ToUint256(), orphan_resolution_candidates);
413 [ + + + - ]: 659 : if (ptx->HasWitness()) m_txrequest.GetCandidatePeers(ptx->GetWitnessHash().ToUint256(), orphan_resolution_candidates);
414 : :
415 [ + + ]: 1322 : for (const auto& nodeid : orphan_resolution_candidates) {
416 [ + - + - ]: 663 : if (MaybeAddOrphanResolutionCandidate(unique_parents, ptx->GetWitnessHash(), nodeid, now)) {
417 [ + - ]: 663 : m_orphanage->AddTx(ptx, nodeid);
418 : : }
419 : : }
420 : :
421 : : // Once added to the orphan pool, a tx is considered AlreadyHave, and we shouldn't request it anymore.
422 [ + - ]: 659 : m_txrequest.ForgetTxHash(tx.GetHash().ToUint256());
423 [ + - ]: 659 : m_txrequest.ForgetTxHash(tx.GetWitnessHash().ToUint256());
424 : 659 : } else {
425 [ + - ]: 38 : unique_parents.clear();
426 [ + - + - : 38 : LogDebug(BCLog::MEMPOOL, "not keeping orphan with rejected parents %s (wtxid=%s)\n",
+ - + - +
- ]
427 : : tx.GetHash().ToString(),
428 : : tx.GetWitnessHash().ToString());
429 : : // We will continue to reject this tx since it has rejected
430 : : // parents so avoid re-requesting it from other peers.
431 : : // Here we add both the txid and the wtxid, as we know that
432 : : // regardless of what witness is provided, we will not accept
433 : : // this, so we don't need to allow for redownload of this txid
434 : : // from any of our non-wtxidrelay peers.
435 [ + - + - ]: 38 : RecentRejectsFilter().insert(tx.GetHash().ToUint256());
436 [ + - + - ]: 38 : RecentRejectsFilter().insert(tx.GetWitnessHash().ToUint256());
437 [ + - ]: 38 : m_txrequest.ForgetTxHash(tx.GetHash().ToUint256());
438 [ + - ]: 38 : m_txrequest.ForgetTxHash(tx.GetWitnessHash().ToUint256());
439 : : }
440 : : }
441 [ + + ]: 264 : } else if (state.GetResult() == TxValidationResult::TX_WITNESS_STRIPPED) {
442 : : add_extra_compact_tx = false;
443 : : } else {
444 : : // We can add the wtxid of this transaction to our reject filter.
445 : : // Do not add txids of witness transactions or witness-stripped
446 : : // transactions to the filter, as they can have been malleated;
447 : : // adding such txids to the reject filter would potentially
448 : : // interfere with relay of valid transactions from peers that
449 : : // do not support wtxid-based relay. See
450 : : // https://github.com/bitcoin/bitcoin/issues/8279 for details.
451 : : // We can remove this restriction (and always add wtxids to
452 : : // the filter even for witness stripped transactions) once
453 : : // wtxid-based relay is broadly deployed.
454 : : // See also comments in https://github.com/bitcoin/bitcoin/pull/18044#discussion_r443419034
455 : : // for concerns around weakening security of unupgraded nodes
456 : : // if we start doing this too early.
457 [ + + ]: 255 : if (state.GetResult() == TxValidationResult::TX_RECONSIDERABLE) {
458 : : // If the result is TX_RECONSIDERABLE, add it to m_lazy_recent_rejects_reconsiderable
459 : : // because we should not download or submit this transaction by itself again, but may
460 : : // submit it as part of a package later.
461 [ + - + - ]: 44 : RecentRejectsReconsiderableFilter().insert(ptx->GetWitnessHash().ToUint256());
462 : :
463 [ + + ]: 44 : if (first_time_failure) {
464 : : // When a transaction fails for TX_RECONSIDERABLE, look for a matching child in the
465 : : // orphanage, as it is possible that they succeed as a package.
466 [ + - + - : 37 : LogDebug(BCLog::TXPACKAGES, "tx %s (wtxid=%s) failed but reconsiderable, looking for child in orphanage\n",
+ - + - +
- ]
467 : : ptx->GetHash().ToString(), ptx->GetWitnessHash().ToString());
468 [ + - + + ]: 111 : package_to_validate = Find1P1CPackage(ptx, nodeid);
469 : : }
470 : : } else {
471 [ + - + - ]: 211 : RecentRejectsFilter().insert(ptx->GetWitnessHash().ToUint256());
472 : : }
473 [ + - ]: 255 : m_txrequest.ForgetTxHash(ptx->GetWitnessHash().ToUint256());
474 : : // If the transaction failed for TX_INPUTS_NOT_STANDARD,
475 : : // then we know that the witness was irrelevant to the policy
476 : : // failure, since this check depends only on the txid
477 : : // (the scriptPubKey being spent is covered by the txid).
478 : : // Add the txid to the reject filter to prevent repeated
479 : : // processing of this transaction in the event that child
480 : : // transactions are later received (resulting in
481 : : // parent-fetching by txid via the orphan-handling logic).
482 : : // We only add the txid if it differs from the wtxid, to avoid wasting entries in the
483 : : // rolling bloom filter.
484 [ + + + + ]: 255 : if (state.GetResult() == TxValidationResult::TX_INPUTS_NOT_STANDARD && ptx->HasWitness()) {
485 [ + - + - ]: 3 : RecentRejectsFilter().insert(ptx->GetHash().ToUint256());
486 [ + - ]: 3 : m_txrequest.ForgetTxHash(ptx->GetHash().ToUint256());
487 : : }
488 : : }
489 : :
490 : : // If the tx failed in ProcessOrphanTx, it should be removed from the orphanage unless the
491 : : // tx was still missing inputs. If the tx was not in the orphanage, EraseTx does nothing and returns 0.
492 [ + + + - : 961 : if (state.GetResult() != TxValidationResult::TX_MISSING_INPUTS && m_orphanage->EraseTx(ptx->GetWitnessHash())) {
+ + ]
493 [ + - + - : 10 : LogDebug(BCLog::TXPACKAGES, " removed orphan tx %s (wtxid=%s)\n", ptx->GetHash().ToString(), ptx->GetWitnessHash().ToString());
+ - + - +
- ]
494 : : }
495 : :
496 [ + + + + ]: 980 : return RejectedTxTodo{
497 : : .m_should_add_extra_compact_tx = add_extra_compact_tx,
498 : : .m_unique_parents = std::move(unique_parents),
499 : : .m_package_to_validate = std::move(package_to_validate)
500 : 961 : };
501 : 961 : }
502 : :
503 : 3 : void TxDownloadManagerImpl::MempoolRejectedPackage(const Package& package)
504 : : {
505 : 3 : RecentRejectsReconsiderableFilter().insert(GetPackageHash(package));
506 : 3 : }
507 : :
508 : 12730 : std::pair<bool, std::optional<PackageToValidate>> TxDownloadManagerImpl::ReceivedTx(NodeId nodeid, const CTransactionRef& ptx)
509 : : {
510 : 12730 : const Txid& txid = ptx->GetHash();
511 : 12730 : const Wtxid& wtxid = ptx->GetWitnessHash();
512 : :
513 : : // Mark that we have received a response
514 : 12730 : m_txrequest.ReceivedResponse(nodeid, txid.ToUint256());
515 [ + + ]: 12730 : if (ptx->HasWitness()) m_txrequest.ReceivedResponse(nodeid, wtxid.ToUint256());
516 : :
517 : : // First check if we should drop this tx.
518 : : // We do the AlreadyHaveTx() check using wtxid, rather than txid - in the
519 : : // absence of witness malleation, this is strictly better, because the
520 : : // recent rejects filter may contain the wtxid but rarely contains
521 : : // the txid of a segwit transaction that has been rejected.
522 : : // In the presence of witness malleation, it's possible that by only
523 : : // doing the check with wtxid, we could overlook a transaction which
524 : : // was confirmed with a different witness, or exists in our mempool
525 : : // with a different witness, but this has limited downside:
526 : : // mempool validation does its own lookup of whether we have the txid
527 : : // already; and an adversary can already relay us old transactions
528 : : // (older than our recency filter) if trying to DoS us, without any need
529 : : // for witness malleation.
530 [ + + ]: 12730 : if (AlreadyHaveTx(wtxid, /*include_reconsiderable=*/false)) {
531 : : // If a tx is detected by m_lazy_recent_rejects it is ignored. Because we haven't
532 : : // submitted the tx to our mempool, we won't have computed a DoS
533 : : // score for it or determined exactly why we consider it invalid.
534 : : //
535 : : // This means we won't penalize any peer subsequently relaying a DoSy
536 : : // tx (even if we penalized the first peer who gave it to us) because
537 : : // we have to account for m_lazy_recent_rejects showing false positives. In
538 : : // other words, we shouldn't penalize a peer if we aren't *sure* they
539 : : // submitted a DoSy tx.
540 : : //
541 : : // Note that m_lazy_recent_rejects doesn't just record DoSy or invalid
542 : : // transactions, but any tx not accepted by the mempool, which may be
543 : : // due to node policy (vs. consensus). So we can't blanket penalize a
544 : : // peer simply for relaying a tx that our m_lazy_recent_rejects has caught,
545 : : // regardless of false positives.
546 : 3851 : return {false, std::nullopt};
547 [ + + ]: 8879 : } else if (RecentRejectsReconsiderableFilter().contains(wtxid.ToUint256())) {
548 : : // When a transaction is already in m_lazy_recent_rejects_reconsiderable, we shouldn't submit
549 : : // it by itself again. However, look for a matching child in the orphanage, as it is
550 : : // possible that they succeed as a package.
551 [ + - + - : 15 : LogDebug(BCLog::TXPACKAGES, "found tx %s (wtxid=%s) in reconsiderable rejects, looking for child in orphanage\n",
+ - ]
552 : : txid.ToString(), wtxid.ToString());
553 [ + + + + ]: 41 : return {false, Find1P1CPackage(ptx, nodeid)};
554 : : }
555 : :
556 : :
557 : 8864 : return {true, std::nullopt};
558 : : }
559 : :
560 : 158030 : bool TxDownloadManagerImpl::HaveMoreWork(NodeId nodeid)
561 : : {
562 : 158030 : return m_orphanage->HaveTxToReconsider(nodeid);
563 : : }
564 : :
565 : 420729 : CTransactionRef TxDownloadManagerImpl::GetTxToReconsider(NodeId nodeid)
566 : : {
567 : 420729 : return m_orphanage->GetTxToReconsider(nodeid);
568 : : }
569 : :
570 : 1864 : void TxDownloadManagerImpl::CheckIsEmpty(NodeId nodeid)
571 : : {
572 [ - + ]: 1864 : assert(m_txrequest.Count(nodeid) == 0);
573 [ - + ]: 1864 : assert(m_orphanage->UsageByPeer(nodeid) == 0);
574 : 1864 : }
575 : 952 : void TxDownloadManagerImpl::CheckIsEmpty()
576 : : {
577 [ - + ]: 952 : assert(m_orphanage->TotalOrphanUsage() == 0);
578 [ - + ]: 952 : assert(m_orphanage->CountUniqueOrphans() == 0);
579 [ - + ]: 952 : assert(m_txrequest.Size() == 0);
580 [ - + ]: 952 : assert(m_num_wtxid_peers == 0);
581 : 952 : }
582 : 228 : std::vector<TxOrphanage::OrphanInfo> TxDownloadManagerImpl::GetOrphanTransactions() const
583 : : {
584 : 228 : return m_orphanage->GetOrphanTransactions();
585 : : }
586 : : } // namespace node
|