1// Copyright (c) 2021-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 <wallet/transaction.h>
6#include <wallet/walletdb.h>
7
8#include <consensus/validation.h>
9#include <interfaces/chain.h>
10
11using interfaces::FoundBlock;
12
13namespace wallet {
14bool CWalletTx::IsMalleation(const CWalletTx& _tx) const
15{
16    return GetTx()->Equals(*_tx.GetTx(), {.include_script_sig = false, .include_witness_data = false});
17}
18
19bool CWalletTx::InMempool() const
20{
21    return state<TxStateInMempool>();
22}
23
24int64_t CWalletTx::GetTxTime() const
25{
26    int64_t n = nTimeSmart;
27    return n ? n : nTimeReceived;
28}
29
30void CWalletTx::updateState(interfaces::Chain& chain)
31{
32    bool active;
33    auto lookup_block = [&](const uint256& hash, int& height, TxState& state) {
34        // If tx block (or conflicting block) was reorged out of chain
35        // while the wallet was shutdown, change tx status to UNCONFIRMED
36        // and reset block height, hash, and index. ABANDONED tx don't have
37        // associated blocks and don't need to be updated. The case where a
38        // transaction was reorged out while online and then reconfirmed
39        // while offline is covered by the rescan logic.
40        if (!chain.findBlock(hash, FoundBlock().inActiveChain(active).height(height)) || !active) {
41            state = TxStateInactive{};
42        }
43    };
44    if (auto* conf = state<TxStateConfirmed>()) {
45        lookup_block(conf->confirmed_block_hash, conf->confirmed_block_height, m_state);
46    } else if (auto* conf = state<TxStateBlockConflicted>()) {
47        lookup_block(conf->conflicting_block_hash, conf->conflicting_block_height, m_state);
48    }
49
50    // If the above downgraded a previously-confirmed witness variant back to unconfirmed,
51    // the canonical choice is no longer pinned by confirmation. Re-apply the least-weight rule.
52    if (!isConfirmed()) RecomputeCanonical();
53}
54
55bool CWalletTx::Update(CTransactionRef new_tx, const TxState& new_state, WalletBatch& batch, bool metadata_changed)
56{
57    Assert(new_tx);
58    if (!Assume(GetHash() == new_tx->GetHash())) {
59        return false;
60    }
61    const auto& [tx_pair, new_variant] = m_txs.emplace(new_tx->GetWitnessHash(), std::move(new_tx));
62    if (new_variant) {
63        if (!batch.WriteWtxVariant(GetHash(), tx_pair->second)) {
64            throw std::ios_base::failure("Unable to write wtxvariant record");
65        }
66    }
67    const auto& [wtxid, tx] = *tx_pair;
68
69    if (new_state.index() != m_state.index()) {
70        m_state = new_state;
71        if (state<TxStateConfirmed>()) {
72            m_canonical_wtxid = wtxid;
73        }
74        metadata_changed = true;
75    } else {
76        assert(TxStateSerializedIndex(m_state) == TxStateSerializedIndex(new_state));
77        assert(TxStateSerializedBlockHash(m_state) == TxStateSerializedBlockHash(new_state));
78    }
79
80    // While unconfirmed, derive the canonical variant from all known variants
81    if (!isConfirmed()) {
82        const Wtxid prev_canonical = m_canonical_wtxid;
83        RecomputeCanonical();
84        if (m_canonical_wtxid != prev_canonical) {
85            metadata_changed = true;
86        }
87    }
88
89    if (metadata_changed) {
90        if (!batch.WriteTxMetadata(*this)) {
91            throw std::ios_base::failure("Unable to write tx record");
92        }
93    }
94
95    return new_variant || metadata_changed;
96}
97
98void CWalletTx::RecomputeCanonical()
99{
100    // Recompute the canonical variant among the witness variants. They share
101    // the txid but differ in the wtxid. Prefer variant with witness data and
102    // the least weight.
103    Assert(!m_txs.empty());
104
105    m_canonical_wtxid = std::ranges::min_element(m_txs, std::less{}, [](const auto& entry) {
106                            return std::make_pair(!entry.second->HasWitness(), GetTransactionWeight(*entry.second));
107                        })->first;
108}
109} // namespace wallet
110