highperfocused/bitcoin
1
0
Fork 0
mirror of https://github.com/bitcoin/bitcoin.git synced 2025-07-28 20:22:44 +02:00
Code Issues Packages Projects Releases Wiki Activity

util: Add ConsumeArithUInt256InRange fuzzing helper

Browse Source
This commit is contained in:
marcofleon
2024-11-04 11:32:23 +00:00
parent f1bcf3edc5
commit fa327c77e3
1 changed files with 16 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/test/fuzz/util.h
View File

@@ -180,6 +180,22 @@ template <typename WeakEnumType, size_t size>
return UintToArith256(ConsumeUInt256(fuzzed_data_provider));
}
[[nodiscard]] inline arith_uint256 ConsumeArithUInt256InRange(FuzzedDataProvider& fuzzed_data_provider, const arith_uint256& min, const arith_uint256& max) noexcept
{
assert(min <= max);
const arith_uint256 range = max - min;
const arith_uint256 value = ConsumeArithUInt256(fuzzed_data_provider);
arith_uint256 result = value;
// Avoid division by 0, in case range + 1 results in overflow.
if (range != ~arith_uint256(0)) {
const arith_uint256 quotient = value / (range + 1);
result = value - (quotient * (range + 1));
}
result += min;
assert(result >= min && result <= max);
return result;
}
[[nodiscard]] std::map<COutPoint, Coin> ConsumeCoins(FuzzedDataProvider& fuzzed_data_provider) noexcept;
[[nodiscard]] CTxDestination ConsumeTxDestination(FuzzedDataProvider& fuzzed_data_provider) noexcept;