Branch data Line data Source code
1 : : // Copyright (c) 2009-present 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 <chainparams.h>
6 : : #include <key_io.h>
7 : : #include <pubkey.h>
8 : : #include <script/descriptor.h>
9 : : #include <test/fuzz/fuzz.h>
10 : : #include <test/fuzz/util/descriptor.h>
11 : : #include <util/chaintype.h>
12 : : #include <util/strencodings.h>
13 : :
14 : : //! The converter of mocked descriptors, needs to be initialized when the target is.
15 : : MockedDescriptorConverter MOCKED_DESC_CONVERTER;
16 : :
17 : : /** Test a successfully parsed descriptor. */
18 : 22295 : static void TestDescriptor(const Descriptor& desc, FlatSigningProvider& sig_provider, std::string& dummy, std::optional<bool>& is_ranged, std::optional<bool>& is_solvable)
19 : : {
20 : : // Trivial helpers.
21 : 22295 : (void)desc.IsRange();
22 : 22295 : (void)desc.IsSingleType();
23 : 22295 : (void)desc.GetOutputType();
24 : :
25 [ + + ]: 22295 : if (is_ranged.has_value()) {
26 [ - + ]: 17302 : assert(desc.IsRange() == *is_ranged);
27 : : } else {
28 : 4993 : is_ranged = desc.IsRange();
29 : : }
30 [ + + ]: 22295 : if (is_solvable.has_value()) {
31 [ - + ]: 17302 : assert(desc.IsSolvable() == *is_solvable);
32 : : } else {
33 : 4993 : is_solvable = desc.IsSolvable();
34 : : }
35 : :
36 : : // Serialization to string representation.
37 : 22295 : (void)desc.ToString();
38 : 22295 : (void)desc.ToPrivateString(sig_provider, dummy);
39 : 22295 : (void)desc.ToNormalizedString(sig_provider, dummy);
40 : :
41 : : // Serialization to Script.
42 : 22295 : DescriptorCache cache;
43 : 22295 : std::vector<CScript> out_scripts;
44 [ + - ]: 22295 : (void)desc.Expand(0, sig_provider, out_scripts, sig_provider, &cache);
45 [ + - ]: 22295 : (void)desc.ExpandPrivate(0, sig_provider, sig_provider);
46 [ + - ]: 22295 : (void)desc.ExpandFromCache(0, cache, out_scripts, sig_provider);
47 : :
48 : : // If we could serialize to script we must be able to infer using the same provider.
49 [ + + ]: 22295 : if (!out_scripts.empty()) {
50 [ + - - + ]: 21379 : assert(InferDescriptor(out_scripts.back(), sig_provider));
51 : :
52 : : // The ScriptSize() must match the size of the serialized Script. (ScriptSize() is set for all descs but 'combo()'.)
53 [ + - ]: 21379 : const bool is_combo{!desc.IsSingleType()};
54 [ + + + + : 21787 : assert(is_combo || desc.ScriptSize() == out_scripts.back().size());
+ - ]
55 : : }
56 : :
57 [ + - ]: 22295 : const auto max_sat_maxsig{desc.MaxSatisfactionWeight(true)};
58 [ + - ]: 22295 : const auto max_sat_nonmaxsig{desc.MaxSatisfactionWeight(true)};
59 [ + - ]: 22295 : const auto max_elems{desc.MaxSatisfactionElems()};
60 : : // We must be able to estimate the max satisfaction size for any solvable descriptor (but combo).
61 [ + + + - : 22295 : const bool is_nontop_or_nonsolvable{!*is_solvable || !desc.GetOutputType()};
+ + ]
62 [ + + + - : 22295 : const bool is_input_size_info_set{max_sat_maxsig && max_sat_nonmaxsig && max_elems};
- + ]
63 [ - + ]: 930 : assert(is_input_size_info_set || is_nontop_or_nonsolvable);
64 : 22295 : }
65 : :
66 : 2 : void initialize_descriptor_parse()
67 : : {
68 [ + - + - : 2 : static ECC_Context ecc_context{};
+ - ]
69 : 2 : SelectParams(ChainType::MAIN);
70 : 2 : }
71 : :
72 : 1 : void initialize_mocked_descriptor_parse()
73 : : {
74 : 1 : initialize_descriptor_parse();
75 : 1 : MOCKED_DESC_CONVERTER.Init();
76 : 1 : }
77 : :
78 [ + - ]: 6755 : FUZZ_TARGET(mocked_descriptor_parse, .init = initialize_mocked_descriptor_parse)
79 : : {
80 [ - + ]: 6301 : const std::string mocked_descriptor{buffer.begin(), buffer.end()};
81 [ - + + - : 6301 : if (const auto descriptor = MOCKED_DESC_CONVERTER.GetDescriptor(mocked_descriptor)) {
+ + ]
82 [ + - + + ]: 6268 : if (IsTooExpensive(MakeUCharSpan(*descriptor))) return;
83 : 5223 : FlatSigningProvider signing_provider;
84 [ - + ]: 5223 : std::string error;
85 [ - + + - ]: 5223 : const auto desc = Parse(*descriptor, signing_provider, error);
86 : 5223 : std::optional<bool> is_ranged;
87 : 5223 : std::optional<bool> is_solvable;
88 [ + + ]: 20777 : for (const auto& d : desc) {
89 [ - + ]: 15554 : assert(d);
90 [ + - ]: 15554 : TestDescriptor(*d, signing_provider, error, is_ranged, is_solvable);
91 : : }
92 : 11524 : }
93 : 6301 : }
94 : :
95 [ + - ]: 5868 : FUZZ_TARGET(descriptor_parse, .init = initialize_descriptor_parse)
96 : : {
97 [ + + ]: 5414 : if (IsTooExpensive(buffer)) return;
98 : :
99 : 4623 : const std::string descriptor(buffer.begin(), buffer.end());
100 : 4623 : FlatSigningProvider signing_provider;
101 : 4623 : std::string error;
102 [ + + ]: 13869 : for (const bool require_checksum : {true, false}) {
103 [ - + + - ]: 9246 : const auto desc = Parse(descriptor, signing_provider, error, require_checksum);
104 : 9246 : std::optional<bool> is_ranged;
105 : 9246 : std::optional<bool> is_solvable;
106 [ + + ]: 15987 : for (const auto& d : desc) {
107 [ - + ]: 6741 : assert(d);
108 [ + - ]: 6741 : TestDescriptor(*d, signing_provider, error, is_ranged, is_solvable);
109 : : }
110 : 9246 : }
111 : 4623 : }
|