diff --git a/src/node/coinstats.cpp b/src/node/coinstats.cpp index e6f275011..f6f946145 100644 --- a/src/node/coinstats.cpp +++ b/src/node/coinstats.cpp @@ -1,173 +1,171 @@ // Copyright (c) 2010 Satoshi Nakamoto // Copyright (c) 2009-2019 The Bitcoin Core developers // Distributed under the MIT software license, see the accompanying // file COPYING or http://www.opensource.org/licenses/mit-license.php. #include #include #include #include #include #include #include #include #include #include static uint64_t GetBogoSize(const CScript &scriptPubKey) { return 32 /* txid */ + 4 /* vout index */ + 4 /* height + coinbase */ + 8 /* amount */ + 2 /* scriptPubKey len */ + scriptPubKey.size() /* scriptPubKey */; } static void ApplyHash(CCoinsStats &stats, CHashWriter &ss, const TxId &txid, const std::map &outputs, std::map::const_iterator it) { if (it == outputs.begin()) { ss << txid; ss << VARINT(it->second.GetHeight() * 2 + it->second.IsCoinBase()); } ss << VARINT(it->first + 1); ss << it->second.GetTxOut().scriptPubKey; ss << VARINT_MODE(it->second.GetTxOut().nValue / SATOSHI, VarIntMode::NONNEGATIVE_SIGNED); if (it == std::prev(outputs.end())) { ss << VARINT(0u); } } static void ApplyHash(CCoinsStats &stats, std::nullptr_t, const TxId &txid, const std::map &outputs, std::map::const_iterator it) {} static void ApplyHash(CCoinsStats &stats, MuHash3072 &muhash, const TxId &txid, const std::map &outputs, std::map::const_iterator it) { COutPoint outpoint = COutPoint(txid, it->first); Coin coin = it->second; CDataStream ss(SER_DISK, PROTOCOL_VERSION); ss << outpoint; ss << static_cast(coin.GetHeight() * 2 + coin.IsCoinBase()); ss << coin.GetTxOut(); muhash.Insert(MakeUCharSpan(ss)); } //! Warning: be very careful when changing this! assumeutxo and UTXO snapshot //! validation commitments are reliant on the hash constructed by this //! function. //! //! If the construction of this hash is changed, it will invalidate //! existing UTXO snapshots. This will not result in any kind of consensus //! failure, but it will force clients that were expecting to make use of //! assumeutxo to do traditional IBD instead. //! //! It is also possible, though very unlikely, that a change in this //! construction could cause a previously invalid (and potentially malicious) //! UTXO snapshot to be considered valid. template static void ApplyStats(CCoinsStats &stats, T &hash_obj, const TxId &txid, const std::map &outputs) { assert(!outputs.empty()); stats.nTransactions++; for (auto it = outputs.begin(); it != outputs.end(); ++it) { ApplyHash(stats, hash_obj, txid, outputs, it); stats.nTransactionOutputs++; stats.nTotalAmount += it->second.GetTxOut().nValue; stats.nBogoSize += GetBogoSize(it->second.GetTxOut().scriptPubKey); } } //! Calculate statistics about the unspent transaction output set template static bool GetUTXOStats(CCoinsView *view, BlockManager &blockman, CCoinsStats &stats, T hash_obj, const std::function &interruption_point) { - stats = CCoinsStats(); std::unique_ptr pcursor(view->Cursor()); assert(pcursor); stats.hashBlock = pcursor->GetBestBlock(); { LOCK(cs_main); assert(std::addressof(g_chainman.m_blockman) == std::addressof(blockman)); const CBlockIndex *block = blockman.LookupBlockIndex(stats.hashBlock); stats.nHeight = Assert(block)->nHeight; } PrepareHash(hash_obj, stats); TxId prevkey; std::map outputs; while (pcursor->Valid()) { interruption_point(); COutPoint key; Coin coin; if (pcursor->GetKey(key) && pcursor->GetValue(coin)) { if (!outputs.empty() && key.GetTxId() != prevkey) { ApplyStats(stats, hash_obj, prevkey, outputs); outputs.clear(); } prevkey = key.GetTxId(); outputs[key.GetN()] = std::move(coin); stats.coins_count++; } else { return error("%s: unable to read value", __func__); } pcursor->Next(); } if (!outputs.empty()) { ApplyStats(stats, hash_obj, prevkey, outputs); } FinalizeHash(hash_obj, stats); stats.nDiskSize = view->EstimateSize(); return true; } bool GetUTXOStats(CCoinsView *view, BlockManager &blockman, CCoinsStats &stats, - CoinStatsHashType hash_type, const std::function &interruption_point) { - switch (hash_type) { + switch (stats.m_hash_type) { case (CoinStatsHashType::HASH_SERIALIZED): { CHashWriter ss(SER_GETHASH, PROTOCOL_VERSION); return GetUTXOStats(view, blockman, stats, ss, interruption_point); } case (CoinStatsHashType::MUHASH): { MuHash3072 muhash; return GetUTXOStats(view, blockman, stats, muhash, interruption_point); } case (CoinStatsHashType::NONE): { return GetUTXOStats(view, blockman, stats, nullptr, interruption_point); } } // no default case, so the compiler can warn about missing cases assert(false); } // The legacy hash serializes the hashBlock static void PrepareHash(CHashWriter &ss, const CCoinsStats &stats) { ss << stats.hashBlock; } // MuHash does not need the prepare step static void PrepareHash(MuHash3072 &muhash, CCoinsStats &stats) {} static void PrepareHash(std::nullptr_t, CCoinsStats &stats) {} static void FinalizeHash(CHashWriter &ss, CCoinsStats &stats) { stats.hashSerialized = ss.GetHash(); } static void FinalizeHash(MuHash3072 &muhash, CCoinsStats &stats) { uint256 out; muhash.Finalize(out); stats.hashSerialized = out; } static void FinalizeHash(std::nullptr_t, CCoinsStats &stats) {} diff --git a/src/node/coinstats.h b/src/node/coinstats.h index 665d8d70f..40cb11466 100644 --- a/src/node/coinstats.h +++ b/src/node/coinstats.h @@ -1,44 +1,46 @@ // Copyright (c) 2010 Satoshi Nakamoto // Copyright (c) 2009-2019 The Bitcoin Core developers // Distributed under the MIT software license, see the accompanying // file COPYING or http://www.opensource.org/licenses/mit-license.php. #ifndef BITCOIN_NODE_COINSTATS_H #define BITCOIN_NODE_COINSTATS_H #include #include #include #include #include class BlockManager; class CCoinsView; enum class CoinStatsHashType { HASH_SERIALIZED, MUHASH, NONE, }; struct CCoinsStats { + CoinStatsHashType m_hash_type; int nHeight{0}; BlockHash hashBlock{}; uint64_t nTransactions{0}; uint64_t nTransactionOutputs{0}; uint64_t nBogoSize{0}; uint256 hashSerialized{}; uint64_t nDiskSize{0}; Amount nTotalAmount{Amount::zero()}; //! The number of coins contained. uint64_t coins_count{0}; + + CCoinsStats(CoinStatsHashType hash_type) : m_hash_type(hash_type) {} }; //! Calculate statistics about the unspent transaction output set bool GetUTXOStats(CCoinsView *view, BlockManager &blockman, CCoinsStats &stats, - const CoinStatsHashType hash_type, const std::function &interruption_point = {}); #endif // BITCOIN_NODE_COINSTATS_H diff --git a/src/rpc/blockchain.cpp b/src/rpc/blockchain.cpp index 2e5a82619..696486b14 100644 --- a/src/rpc/blockchain.cpp +++ b/src/rpc/blockchain.cpp @@ -1,3191 +1,3191 @@ // Copyright (c) 2010 Satoshi Nakamoto // Copyright (c) 2009-2019 The Bitcoin Core developers // Distributed under the MIT software license, see the accompanying // file COPYING or http://www.opensource.org/licenses/mit-license.php. #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include