Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Introduce evicted-at/last-evicted timestamps #1811

Draft
wants to merge 16 commits into
base: master
Choose a base branch
from

Conversation

evanlinjin
Copy link
Member

@evanlinjin evanlinjin commented Jan 24, 2025

Fixes #1740.

Description

This PR replaces #1765. For context and the original discussion that led to this change, please refer to this comment.

This PR addresses a potential malicious double-spending issue by introducing improvements to unconfirmed transaction tracking. Key changes include the addition of TxUpdate::missing that tracks transactions that have been replaced and are no longer in the mempool, and the inclusion of last_evicted and evicted_at timestamps in TxGraph to track when a transaction was last deemed missing.

SpkWithExpectedTxids is introduced in SpkClient to track expected Txids for each spk. During a sync, if any Txids from SpkWithExpectedTxids are not in the current history of an spk obtained from the chain source, those Txids are considered missing. Support for SpkWithExpectedTxids has been added to both bdk_electrum and bdk_esplora chain source crates.

The canonicalization algorithm is updated to disregard transactions with a last_evicted timestamp greater than or equal to their last_seen timestamp, except in cases where transitivity rules apply.

Changelog notice

  • TxUpdate::missing tracks transactions that have been replaced and are no longer present in mempool.
  • last_evicted and evicted_at timestamps in TxGraph track when a transaction was last marked as missing from the mempool.
  • The canonicalization algorithm now disregards transactions with a last_evicted timestamp greater than or equal to it's last_seen timestamp, except when a canonical descendant exists due to rules of transitivity.
  • SpkWithExpectedTxids in SpkClient keeps track of expected Txids for each spk.
  • SpkWithExpectedTxids support added for bdk_electrum and bdk_esplora.
  • SyncRequestBuilder::expected_txids_of_spk adds an association between Txid and spk.
  • SyncRequestExt::check_unconfirmed_statuses adds unconfirmed transactions alongside their spks during sync.

Checklists

All Submissions:

  • I've signed all my commits
  • I followed the contribution guidelines
  • I ran cargo fmt and cargo clippy before committing

New Features:

  • I've added tests for the new feature
  • I've added docs for the new feature

Bugfixes:

  • This pull request breaks the existing API
  • I've added tests to reproduce the issue which are now passing
  • I'm linking the issue being fixed by this PR

@evanlinjin evanlinjin changed the title Introduce MissingMarker Introduce missing-at/last-missing timestamps Jan 24, 2025
@LagginTimes LagginTimes force-pushed the missing_marker branch 2 times, most recently from 0f52ead to 2e31bc7 Compare January 24, 2025 15:27
@notmandatory notmandatory added the bug Something isn't working label Jan 24, 2025
@notmandatory
Copy link
Member

Looks like only breaking changes are on core and chain, so we'll be able to roll this out in a non-breaking 1.x wallet release?

@LLFourn
Copy link
Contributor

LLFourn commented Jan 28, 2025

@notmandatory core breaking change is also a wallet crate breaking change unfortunately. It is a minor breaking change (some struct from core is changing that no one inspects directly in their code probably) but it technically is a breaking change so it would have to be in bdk_wallet v2. This is a pretty important improvement so I'd say it justifies a v2 release by itself when it's merged.

Copy link
Contributor

@LLFourn LLFourn left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

ConceptACK. Nice ideas here. I think chain API needs a clean up.

crates/chain/src/indexer/keychain_txout.rs Outdated Show resolved Hide resolved
crates/chain/src/indexer/keychain_txout.rs Outdated Show resolved Hide resolved
crates/chain/src/indexer/spk_txout.rs Outdated Show resolved Hide resolved
crates/chain/src/indexer/spk_txout.rs Outdated Show resolved Hide resolved
crates/chain/src/tx_graph.rs Outdated Show resolved Hide resolved
@evanlinjin evanlinjin marked this pull request as ready for review January 30, 2025 12:33
@@ -145,6 +145,7 @@ pub struct TxGraph<A = ConfirmationBlockTime> {
spends: BTreeMap<OutPoint, HashSet<Txid>>,
anchors: HashMap<Txid, BTreeSet<A>>,
last_seen: HashMap<Txid, u64>,
last_missing: HashMap<Txid, u64>,
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think it would be nice if we had a data structure like this and then last_seen would become HashMap<Txid, LastSeen>

#[derive(Debug, Clone, Copy, Default, PartialEq)]
struct LastSeen {
    /// seen at
    seen_at: u64,
    /// evicted at
    evicted_at: Option<u64>,
}

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I quite like this idea since there has been talks about adding more timestamps.

I.e. first-seen timestamp, which will help with tx ordering when listing transactions.

struct MempoolTimestamps {
    pub first_seen: u64,
    pub last_seen: u64,
    pub last_evicted: Option<u64>,
}

I also like the term evicted as it's more descriptive than missing.

Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I agree, and +1 on this idea, it makes it clear, and easy to "derive" its lifecycle and it's and can be useful info to expose to the users.

Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

After some discussion with @evanlinjin, it was decided that this was best done in a separate PR as it may be too breaking of a change.

Copy link
Member Author

@evanlinjin evanlinjin Feb 5, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Yes. On second thought, this introduces an invariant that last_evicted must exist only if we have had a seen_at value. Because of the existence of this invariant, this invariant must be reflected in tx_graph::ChangeSet by changing ChangeSet::seen_at to have values of MempoolTimestamps. This change breaks our policy of changes to changeset which was to only add new fields.

Edit: Maybe we can make it a user-facing API.

.flat_map(|(txid, anchors)| anchors.into_iter().map(move |a| (a, txid)))
.collect();
tx_update.seen_ats = graph.last_seen.into_iter().collect();
tx_update
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

It seems like TxUpdate would need another field evicted_at in order to maintain roundtrip convertibility between TxUpdate and TxGraph.

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Do you think the convertibility between TxGraph and TxUpdate is important? I'm thinking maybe we should just get rid of it. One can just construct an empty TxGraph and apply an TxUpdate.

Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

It's probably ok to forego complete fungibility between graphs and updates, but at the same time I wonder how useful are the seen-ats without the corresponding evictions?

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

but at the same time I wonder how useful are the seen-ats without the corresponding evictions?

@ValuedMammal what do you mean by this?

I think it makes sense to completely remove convertibility of TxGraph -> TxUpdate and TxUpdate -> TxGraph.

Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Just pointing out that we lose information going from TxGraph to TxUpdate. Since both last-seen and last-evicted can influence canonicalization, it might make sense to include both (or neither).

Maybe it depends on the use case for TxUpdate. If updating a tx graph from a spk_client based chain source, the seen_ats don't seem to be as important compared to treating graphs and updates as interchangeable. If we remove convertibility, would it make sense to change this From impl to a method on TxGraph that returns a TxUpdate? We could potentially develop different kinds of updates, like one that is only relevant to a subset of keychains.

crates/chain/src/tx_graph.rs Outdated Show resolved Hide resolved
crates/chain/src/tx_graph.rs Outdated Show resolved Hide resolved
crates/chain/src/indexer/keychain_txout.rs Outdated Show resolved Hide resolved
crates/core/src/spk_client.rs Outdated Show resolved Hide resolved
crates/core/src/spk_client.rs Outdated Show resolved Hide resolved
@@ -145,6 +145,7 @@ pub struct TxGraph<A = ConfirmationBlockTime> {
spends: BTreeMap<OutPoint, HashSet<Txid>>,
anchors: HashMap<Txid, BTreeSet<A>>,
last_seen: HashMap<Txid, u64>,
last_missing: HashMap<Txid, u64>,
Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I quite like this idea since there has been talks about adding more timestamps.

I.e. first-seen timestamp, which will help with tx ordering when listing transactions.

struct MempoolTimestamps {
    pub first_seen: u64,
    pub last_seen: u64,
    pub last_evicted: Option<u64>,
}

I also like the term evicted as it's more descriptive than missing.

.flat_map(|(txid, anchors)| anchors.into_iter().map(move |a| (a, txid)))
.collect();
tx_update.seen_ats = graph.last_seen.into_iter().collect();
tx_update
Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Do you think the convertibility between TxGraph and TxUpdate is important? I'm thinking maybe we should just get rid of it. One can just construct an empty TxGraph and apply an TxUpdate.

crates/chain/src/indexer/keychain_txout.rs Outdated Show resolved Hide resolved
Copy link
Contributor

@oleonardolima oleonardolima left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

A general ConceptACK, some small nits.

crates/chain/src/tx_graph.rs Outdated Show resolved Hide resolved
crates/chain/src/tx_graph.rs Outdated Show resolved Hide resolved
@@ -145,6 +145,7 @@ pub struct TxGraph<A = ConfirmationBlockTime> {
spends: BTreeMap<OutPoint, HashSet<Txid>>,
anchors: HashMap<Txid, BTreeSet<A>>,
last_seen: HashMap<Txid, u64>,
last_missing: HashMap<Txid, u64>,
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I agree, and +1 on this idea, it makes it clear, and easy to "derive" its lifecycle and it's and can be useful info to expose to the users.

@LagginTimes LagginTimes force-pushed the missing_marker branch 3 times, most recently from 7f3b6d4 to be60ada Compare February 6, 2025 08:47
@LagginTimes LagginTimes closed this Feb 6, 2025
@LagginTimes LagginTimes reopened this Feb 6, 2025
@evanlinjin evanlinjin changed the title Introduce missing-at/last-missing timestamps Introduce evicted-at/last-evicted timestamps Feb 7, 2025
crates/bitcoind_rpc/tests/test_emitter.rs Outdated Show resolved Hide resolved
crates/chain/src/indexer/spk_txout.rs Outdated Show resolved Hide resolved
.flat_map(|(txid, anchors)| anchors.into_iter().map(move |a| (a, txid)))
.collect();
tx_update.seen_ats = graph.last_seen.into_iter().collect();
tx_update
Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

but at the same time I wonder how useful are the seen-ats without the corresponding evictions?

@ValuedMammal what do you mean by this?

I think it makes sense to completely remove convertibility of TxGraph -> TxUpdate and TxUpdate -> TxGraph.

@LagginTimes LagginTimes force-pushed the missing_marker branch 3 times, most recently from 25b005c to 203c97d Compare February 7, 2025 09:32
LagginTimes and others added 7 commits February 11, 2025 23:25
This is for conveniently adding associations of txid <-> spk. We expect
that these txids exist in the spk history. Otherwise, it means the tx is
evicted from the mempool and we need to update the `missing_at` value in
the sync response.
This is a convenience method for adding unconfirmed txs alongside their
associated spks the the sync request. This way, we will be able to
detect evictions of these transactions from the mempool.

env.mine_blocks(101, None)?;

// Select a UTXO to use as an input for constructing our test transactions.
Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Which test transaction?

@LagginTimes LagginTimes marked this pull request as draft February 13, 2025 11:35
example-crates/example_bitcoind_rpc_polling/src/main.rs Outdated Show resolved Hide resolved
example-crates/example_wallet_rpc/src/main.rs Outdated Show resolved Hide resolved
crates/bitcoind_rpc/src/lib.rs Outdated Show resolved Hide resolved
crates/bitcoind_rpc/src/lib.rs Outdated Show resolved Hide resolved
@LagginTimes LagginTimes force-pushed the missing_marker branch 5 times, most recently from 98af6cd to 75adeee Compare February 16, 2025 05:41
let chain_tip = chain.get_chain_tip().unwrap();

self.list_canonical_txs(chain, chain_tip)
.filter(|c| !c.chain_position.is_confirmed() && indexer.is_tx_relevant(&c.tx_node))
Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Any confirmed tx that is missing from the Electrum's spk history should also be evicted. Since this means that there is a reorg and the tx is not re-introduced to the mempool (evicted).

ValuedMammal and others added 4 commits February 18, 2025 01:46
Make this method work when the indexer is `KeychainTxOutIndex`. We
reintroduce the ability to get the internal `SpkTxOutIndex` from
`KeychainTxOutIndex` so that `SpkTxOutIndex::relevant_spks_of_tx` is
callable from `KeychainTxOutIndex`.

This commit renames `iter_spks_with_expected_txids` to
`expected_unconfirmed_spk_txids` for `TxGraph`, `IndexedTxGraph` and
`SyncRequestBuilder`. Docs are also improved to explain how these
methods are useful.

Remove unused `SyncRequestBuilder` methods.
* Remove duplicate paragraphs about `ChangeSet`s.
* Add "Canonicalization" section which expands on methods that require
  canonicalization and the associated data types used in the
  canonicalization algorithm.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working
Projects
Status: Done
Development

Successfully merging this pull request may close these issues.

Undetected double-spent
6 participants