Merge bitcoin/bitcoin#32243: test: added fuzz coverage for consensus/merkle.cpp

95969bc58a test: added fuzz coverage to consensus/merkle.cpp (kevkevinpal)

Pull request description:

  ### Summary
  This adds a new fuzz target "merkle" which adds fuzz coverage to `consensus/merkle.cpp`

  I can also add this to an existing fuzz target if that is preferable

  Before:
  ![Screenshot 2025-04-09 at 10 12 54 PM](https://github.com/user-attachments/assets/e5f8da56-4583-441d-b08f-dfcc255ff248)

  After:
  ![Screenshot 2025-04-11 at 4 20 41 PM](https://github.com/user-attachments/assets/849ee079-b715-4089-9e36-d156233236c6)

ACKs for top commit:
  marcofleon:
    ReACK 95969bc58a
  Prabhat1308:
    ACK [`95969bc`](95969bc58a)
  maflcko:
    lgtm ACK 95969bc58a
  achow101:
    ACK 95969bc58a

Tree-SHA512: e1fe8b69444733516bfa6cf2adaa199fde4c7c5582b7b908408f9313ed0f2e8cb803d27d707a1716d49606d5eaef8c1e722990bbc3cffc30fa91fe73d2233e9d
This commit is contained in:
Ava Chow
2025-06-27 13:34:30 -07:00
2 changed files with 93 additions and 0 deletions

View File

@ -59,6 +59,7 @@ add_executable(fuzz
kitchen_sink.cpp
load_external_block_file.cpp
locale.cpp
merkle.cpp
merkleblock.cpp
message.cpp
miniscript.cpp

92
src/test/fuzz/merkle.cpp Normal file
View File

@ -0,0 +1,92 @@
// Copyright (c) 2025 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 <consensus/merkle.h>
#include <test/fuzz/fuzz.h>
#include <test/fuzz/FuzzedDataProvider.h>
#include <test/fuzz/util.h>
#include <test/util/str.h>
#include <util/strencodings.h>
#include <hash.h>
#include <cassert>
#include <cstdint>
#include <vector>
uint256 ComputeMerkleRootFromPath(const CBlock& block, uint32_t position, const std::vector<uint256>& merkle_path) {
if (position >= block.vtx.size()) {
throw std::out_of_range("Position out of range");
}
uint256 current_hash = block.vtx[position]->GetHash();
for (const uint256& sibling : merkle_path) {
if (position % 2 == 0) {
current_hash = Hash(current_hash, sibling);
} else {
current_hash = Hash(sibling, current_hash);
}
position = position / 2;
}
return current_hash;
}
FUZZ_TARGET(merkle)
{
FuzzedDataProvider fuzzed_data_provider(buffer.data(), buffer.size());
const bool with_witness = fuzzed_data_provider.ConsumeBool();
std::optional<CBlock> block {ConsumeDeserializable<CBlock>(fuzzed_data_provider, with_witness ? TX_WITH_WITNESS : TX_NO_WITNESS)};
if (!block){
return;
}
const size_t num_txs = block->vtx.size();
std::vector<uint256> tx_hashes;
tx_hashes.reserve(num_txs);
for (size_t i = 0; i < num_txs; ++i) {
tx_hashes.push_back(block->vtx[i]->GetHash());
}
// Test ComputeMerkleRoot
bool mutated = fuzzed_data_provider.ConsumeBool();
const uint256 merkle_root = ComputeMerkleRoot(tx_hashes, &mutated);
// Basic sanity checks for ComputeMerkleRoot
if (tx_hashes.size() == 1) {
assert(merkle_root == tx_hashes[0]);
}
const uint256 block_merkle_root = BlockMerkleRoot(*block, &mutated);
if (tx_hashes.size() == 1) {
assert(block_merkle_root == tx_hashes[0]);
}
if (!block->vtx.empty()){
const uint256 block_witness_merkle_root = BlockWitnessMerkleRoot(*block, &mutated);
if (tx_hashes.size() == 1) {
assert(block_witness_merkle_root == uint256());
}
}
// Test TransactionMerklePath
const uint32_t position = fuzzed_data_provider.ConsumeIntegralInRange<uint32_t>(0, num_txs > 0 ? num_txs - 1 : 0);
std::vector<uint256> merkle_path = TransactionMerklePath(*block, position);
// Check that the root we compute from TransactionMerklePath equals the same merkle_root and block_merkle_root
if (tx_hashes.size() > 1) {
uint256 merkle_root_from_merkle_path = ComputeMerkleRootFromPath(*block, position, merkle_path);
assert(merkle_root_from_merkle_path == merkle_root);
assert(merkle_root_from_merkle_path == block_merkle_root);
}
// Basic sanity checks for TransactionMerklePath
assert(merkle_path.size() <= 32); // Maximum depth of a Merkle tree with 2^32 leaves
if (num_txs == 1 || num_txs == 0) {
assert(merkle_path.empty()); // Single transaction has no path
}
}