Skip to content

Commit fa4926c

Browse files
author
MarcoFalke
committed
fuzz: [refactor] Use IsValidFlagCombination in signature_checker fuzz target
Can be reviewed with --color-moved=dimmed-zebra
1 parent eeee8f5 commit fa4926c

File tree

5 files changed

+20
-14
lines changed

5 files changed

+20
-14
lines changed

src/Makefile.test_util.include

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@ libtest_util_a_SOURCES = \
2626
test/util/logging.cpp \
2727
test/util/mining.cpp \
2828
test/util/net.cpp \
29+
test/util/script.cpp \
2930
test/util/setup_common.cpp \
3031
test/util/str.cpp \
3132
test/util/transaction_utils.cpp \

src/test/fuzz/script_flags.cpp

Lines changed: 1 addition & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -5,13 +5,11 @@
55
#include <pubkey.h>
66
#include <script/interpreter.h>
77
#include <streams.h>
8+
#include <test/util/script.h>
89
#include <version.h>
910

1011
#include <test/fuzz/fuzz.h>
1112

12-
/** Flags that are not forbidden by an assert */
13-
static bool IsValidFlagCombination(unsigned flags);
14-
1513
void initialize_script_flags()
1614
{
1715
static const ECCVerifyHandle verify_handle;
@@ -74,10 +72,3 @@ FUZZ_TARGET_INIT(script_flags, initialize_script_flags)
7472
return;
7573
}
7674
}
77-
78-
static bool IsValidFlagCombination(unsigned flags)
79-
{
80-
if (flags & SCRIPT_VERIFY_CLEANSTACK && ~flags & (SCRIPT_VERIFY_P2SH | SCRIPT_VERIFY_WITNESS)) return false;
81-
if (flags & SCRIPT_VERIFY_WITNESS && ~flags & SCRIPT_VERIFY_P2SH) return false;
82-
return true;
83-
}

src/test/fuzz/signature_checker.cpp

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
#include <test/fuzz/FuzzedDataProvider.h>
88
#include <test/fuzz/fuzz.h>
99
#include <test/fuzz/util.h>
10+
#include <test/util/script.h>
1011

1112
#include <cstdint>
1213
#include <limits>
@@ -61,10 +62,7 @@ FUZZ_TARGET_INIT(signature_checker, initialize_signature_checker)
6162
const auto script_2 = ConsumeScript(fuzzed_data_provider, 65536);
6263
std::vector<std::vector<unsigned char>> stack;
6364
(void)EvalScript(stack, script_1, flags, FuzzedSignatureChecker(fuzzed_data_provider), sig_version, nullptr);
64-
if ((flags & SCRIPT_VERIFY_CLEANSTACK) != 0 && ((flags & SCRIPT_VERIFY_P2SH) == 0 || (flags & SCRIPT_VERIFY_WITNESS) == 0)) {
65-
return;
66-
}
67-
if ((flags & SCRIPT_VERIFY_WITNESS) != 0 && (flags & SCRIPT_VERIFY_P2SH) == 0) {
65+
if (!IsValidFlagCombination(flags)) {
6866
return;
6967
}
7068
(void)VerifyScript(script_1, script_2, nullptr, flags, FuzzedSignatureChecker(fuzzed_data_provider), nullptr);

src/test/util/script.cpp

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
// Copyright (c) 2021 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 <script/interpreter.h>
6+
#include <test/util/script.h>
7+
8+
bool IsValidFlagCombination(unsigned flags)
9+
{
10+
if (flags & SCRIPT_VERIFY_CLEANSTACK && ~flags & (SCRIPT_VERIFY_P2SH | SCRIPT_VERIFY_WITNESS)) return false;
11+
if (flags & SCRIPT_VERIFY_WITNESS && ~flags & SCRIPT_VERIFY_P2SH) return false;
12+
return true;
13+
}

src/test/util/script.h

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,4 +18,7 @@ static const CScript P2WSH_OP_TRUE{
1818
return hash;
1919
}())};
2020

21+
/** Flags that are not forbidden by an assert in script validation */
22+
bool IsValidFlagCombination(unsigned flags);
23+
2124
#endif // BITCOIN_TEST_UTIL_SCRIPT_H

0 commit comments

Comments
 (0)