Skip to content

Commit fa98f3f

Browse files
author
MarcoFalke
committed
fuzz: [refactor] Use ConsumeScript in signature_checker fuzz target
1 parent 1c7be9a commit fa98f3f

File tree

2 files changed

+6
-7
lines changed

2 files changed

+6
-7
lines changed

src/test/fuzz/signature_checker.cpp

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
#include <script/interpreter.h>
77
#include <test/fuzz/FuzzedDataProvider.h>
88
#include <test/fuzz/fuzz.h>
9+
#include <test/fuzz/util.h>
910

1011
#include <cstdint>
1112
#include <limits>
@@ -56,17 +57,15 @@ FUZZ_TARGET_INIT(signature_checker, initialize_signature_checker)
5657
FuzzedDataProvider fuzzed_data_provider(buffer.data(), buffer.size());
5758
const unsigned int flags = fuzzed_data_provider.ConsumeIntegral<unsigned int>();
5859
const SigVersion sig_version = fuzzed_data_provider.PickValueInArray({SigVersion::BASE, SigVersion::WITNESS_V0});
59-
const std::string script_string_1 = fuzzed_data_provider.ConsumeRandomLengthString(65536);
60-
const std::vector<uint8_t> script_bytes_1{script_string_1.begin(), script_string_1.end()};
61-
const std::string script_string_2 = fuzzed_data_provider.ConsumeRandomLengthString(65536);
62-
const std::vector<uint8_t> script_bytes_2{script_string_2.begin(), script_string_2.end()};
60+
const auto script_1 = ConsumeScript(fuzzed_data_provider, 65536);
61+
const auto script_2 = ConsumeScript(fuzzed_data_provider, 65536);
6362
std::vector<std::vector<unsigned char>> stack;
64-
(void)EvalScript(stack, {script_bytes_1.begin(), script_bytes_1.end()}, flags, FuzzedSignatureChecker(fuzzed_data_provider), sig_version, nullptr);
63+
(void)EvalScript(stack, script_1, flags, FuzzedSignatureChecker(fuzzed_data_provider), sig_version, nullptr);
6564
if ((flags & SCRIPT_VERIFY_CLEANSTACK) != 0 && ((flags & SCRIPT_VERIFY_P2SH) == 0 || (flags & SCRIPT_VERIFY_WITNESS) == 0)) {
6665
return;
6766
}
6867
if ((flags & SCRIPT_VERIFY_WITNESS) != 0 && (flags & SCRIPT_VERIFY_P2SH) == 0) {
6968
return;
7069
}
71-
(void)VerifyScript({script_bytes_1.begin(), script_bytes_1.end()}, {script_bytes_2.begin(), script_bytes_2.end()}, nullptr, flags, FuzzedSignatureChecker(fuzzed_data_provider), nullptr);
70+
(void)VerifyScript(script_1, script_2, nullptr, flags, FuzzedSignatureChecker(fuzzed_data_provider), nullptr);
7271
}

src/test/fuzz/util.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,7 @@ CScriptWitness ConsumeScriptWitness(FuzzedDataProvider& fuzzed_data_provider, co
8080

8181
CScript ConsumeScript(FuzzedDataProvider& fuzzed_data_provider, const size_t max_length, const bool maybe_p2wsh) noexcept
8282
{
83-
const std::vector<uint8_t> b = ConsumeRandomLengthByteVector(fuzzed_data_provider);
83+
const std::vector<uint8_t> b = ConsumeRandomLengthByteVector(fuzzed_data_provider, max_length);
8484
CScript r_script{b.begin(), b.end()};
8585
if (maybe_p2wsh && fuzzed_data_provider.ConsumeBool()) {
8686
uint256 script_hash;

0 commit comments

Comments
 (0)