/root/bitcoin/src/index/txindex.cpp
Line | Count | Source |
1 | | // Copyright (c) 2017-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 <index/txindex.h> |
6 | | |
7 | | #include <common/args.h> |
8 | | #include <dbwrapper.h> |
9 | | #include <flatfile.h> |
10 | | #include <index/base.h> |
11 | | #include <index/disktxpos.h> |
12 | | #include <interfaces/chain.h> |
13 | | #include <node/blockstorage.h> |
14 | | #include <primitives/block.h> |
15 | | #include <primitives/transaction.h> |
16 | | #include <serialize.h> |
17 | | #include <streams.h> |
18 | | #include <uint256.h> |
19 | | #include <util/fs.h> |
20 | | #include <util/log.h> |
21 | | #include <validation.h> |
22 | | |
23 | | #include <cassert> |
24 | | #include <cstdint> |
25 | | #include <cstdio> |
26 | | #include <exception> |
27 | | #include <string> |
28 | | #include <utility> |
29 | | #include <vector> |
30 | | |
31 | | constexpr uint8_t DB_TXINDEX{'t'}; |
32 | | |
33 | | std::unique_ptr<TxIndex> g_txindex; |
34 | | |
35 | | |
36 | | /** Access to the txindex database (indexes/txindex/) */ |
37 | | class TxIndex::DB : public BaseIndex::DB |
38 | | { |
39 | | public: |
40 | | explicit DB(size_t n_cache_size, bool f_memory = false, bool f_wipe = false); |
41 | | |
42 | | /// Read the disk location of the transaction data with the given hash. Returns false if the |
43 | | /// transaction hash is not indexed. |
44 | | bool ReadTxPos(const Txid& txid, CDiskTxPos& pos) const; |
45 | | |
46 | | /// Write a batch of transaction positions to the DB. |
47 | | void WriteTxs(const std::vector<std::pair<Txid, CDiskTxPos>>& v_pos); |
48 | | }; |
49 | | |
50 | | TxIndex::DB::DB(size_t n_cache_size, bool f_memory, bool f_wipe) : |
51 | 0 | BaseIndex::DB(gArgs.GetDataDirNet() / "indexes" / "txindex", n_cache_size, f_memory, f_wipe) |
52 | 0 | {} |
53 | | |
54 | | bool TxIndex::DB::ReadTxPos(const Txid& txid, CDiskTxPos& pos) const |
55 | 0 | { |
56 | 0 | return Read(std::make_pair(DB_TXINDEX, txid.ToUint256()), pos); |
57 | 0 | } |
58 | | |
59 | | void TxIndex::DB::WriteTxs(const std::vector<std::pair<Txid, CDiskTxPos>>& v_pos) |
60 | 0 | { |
61 | 0 | CDBBatch batch(*this); |
62 | 0 | for (const auto& [txid, pos] : v_pos) { |
63 | 0 | batch.Write(std::make_pair(DB_TXINDEX, txid.ToUint256()), pos); |
64 | 0 | } |
65 | 0 | WriteBatch(batch); |
66 | 0 | } |
67 | | |
68 | | TxIndex::TxIndex(std::unique_ptr<interfaces::Chain> chain, size_t n_cache_size, bool f_memory, bool f_wipe) |
69 | 0 | : BaseIndex(std::move(chain), "txindex"), m_db(std::make_unique<TxIndex::DB>(n_cache_size, f_memory, f_wipe)) |
70 | 0 | {} |
71 | | |
72 | 0 | TxIndex::~TxIndex() = default; |
73 | | |
74 | | bool TxIndex::CustomAppend(const interfaces::BlockInfo& block) |
75 | 0 | { |
76 | | // Exclude genesis block transaction because outputs are not spendable. |
77 | 0 | if (block.height == 0) return true; |
78 | | |
79 | 0 | assert(block.data); |
80 | 0 | CDiskTxPos pos({block.file_number, block.data_pos}, GetSizeOfCompactSize(block.data->vtx.size())); |
81 | 0 | std::vector<std::pair<Txid, CDiskTxPos>> vPos; |
82 | 0 | vPos.reserve(block.data->vtx.size()); |
83 | 0 | for (const auto& tx : block.data->vtx) { |
84 | 0 | vPos.emplace_back(tx->GetHash(), pos); |
85 | 0 | pos.nTxOffset += ::GetSerializeSize(TX_WITH_WITNESS(*tx)); |
86 | 0 | } |
87 | 0 | m_db->WriteTxs(vPos); |
88 | 0 | return true; |
89 | 0 | } |
90 | | |
91 | 0 | BaseIndex::DB& TxIndex::GetDB() const { return *m_db; } |
92 | | |
93 | | bool TxIndex::FindTx(const Txid& tx_hash, uint256& block_hash, CTransactionRef& tx) const |
94 | 0 | { |
95 | 0 | CDiskTxPos postx; |
96 | 0 | if (!m_db->ReadTxPos(tx_hash, postx)) { |
97 | 0 | return false; |
98 | 0 | } |
99 | | |
100 | 0 | AutoFile file{m_chainstate->m_blockman.OpenBlockFile(postx, true)}; |
101 | 0 | if (file.IsNull()) { |
102 | 0 | LogError("OpenBlockFile failed");Line | Count | Source | 99 | 0 | #define LogError(...) LogPrintLevel_(BCLog::LogFlags::ALL, BCLog::Level::Error, /*should_ratelimit=*/true, __VA_ARGS__) Line | Count | Source | 91 | 0 | #define LogPrintLevel_(category, level, should_ratelimit, ...) LogPrintFormatInternal(SourceLocation{__func__}, category, level, should_ratelimit, __VA_ARGS__) |
|
|
103 | 0 | return false; |
104 | 0 | } |
105 | 0 | CBlockHeader header; |
106 | 0 | try { |
107 | 0 | file >> header; |
108 | 0 | file.seek(postx.nTxOffset, SEEK_CUR); |
109 | 0 | file >> TX_WITH_WITNESS(tx); |
110 | 0 | } catch (const std::exception& e) { |
111 | 0 | LogError("Deserialize or I/O error - %s", e.what());Line | Count | Source | 99 | 0 | #define LogError(...) LogPrintLevel_(BCLog::LogFlags::ALL, BCLog::Level::Error, /*should_ratelimit=*/true, __VA_ARGS__) Line | Count | Source | 91 | 0 | #define LogPrintLevel_(category, level, should_ratelimit, ...) LogPrintFormatInternal(SourceLocation{__func__}, category, level, should_ratelimit, __VA_ARGS__) |
|
|
112 | 0 | return false; |
113 | 0 | } |
114 | 0 | if (tx->GetHash() != tx_hash) { |
115 | 0 | LogError("txid mismatch");Line | Count | Source | 99 | 0 | #define LogError(...) LogPrintLevel_(BCLog::LogFlags::ALL, BCLog::Level::Error, /*should_ratelimit=*/true, __VA_ARGS__) Line | Count | Source | 91 | 0 | #define LogPrintLevel_(category, level, should_ratelimit, ...) LogPrintFormatInternal(SourceLocation{__func__}, category, level, should_ratelimit, __VA_ARGS__) |
|
|
116 | 0 | return false; |
117 | 0 | } |
118 | 0 | block_hash = header.GetHash(); |
119 | 0 | return true; |
120 | 0 | } |