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                 :       13725 : 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                 :       13725 :     (void)desc.IsRange();
      22                 :       13725 :     (void)desc.IsSingleType();
      23                 :       13725 :     (void)desc.GetOutputType();
      24                 :             : 
      25         [ +  + ]:       13725 :     if (is_ranged.has_value()) {
      26         [ -  + ]:        9782 :         assert(desc.IsRange() == *is_ranged);
      27                 :             :     } else {
      28                 :        3943 :         is_ranged = desc.IsRange();
      29                 :             :     }
      30         [ +  + ]:       13725 :     if (is_solvable.has_value()) {
      31         [ -  + ]:        9782 :         assert(desc.IsSolvable() == *is_solvable);
      32                 :             :     } else {
      33                 :        3943 :         is_solvable = desc.IsSolvable();
      34                 :             :     }
      35                 :             : 
      36                 :             :     // Serialization to string representation.
      37                 :       13725 :     (void)desc.ToString();
      38                 :       13725 :     (void)desc.ToPrivateString(sig_provider, dummy);
      39                 :       13725 :     (void)desc.ToNormalizedString(sig_provider, dummy);
      40                 :             : 
      41                 :             :     // Serialization to Script.
      42                 :       13725 :     DescriptorCache cache;
      43                 :       13725 :     std::vector<CScript> out_scripts;
      44         [ +  - ]:       13725 :     (void)desc.Expand(0, sig_provider, out_scripts, sig_provider, &cache);
      45         [ +  - ]:       13725 :     (void)desc.ExpandPrivate(0, sig_provider, sig_provider);
      46         [ +  - ]:       13725 :     (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         [ +  + ]:       13725 :     if (!out_scripts.empty()) {
      50   [ +  -  -  + ]:       13146 :         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         [ +  - ]:       13146 :         const bool is_combo{!desc.IsSingleType()};
      54   [ +  +  +  +  :       13481 :         assert(is_combo || desc.ScriptSize() == out_scripts.back().size());
                   +  - ]
      55                 :             :     }
      56                 :             : 
      57         [ +  - ]:       13725 :     const auto max_sat_maxsig{desc.MaxSatisfactionWeight(true)};
      58         [ +  - ]:       13725 :     const auto max_sat_nonmaxsig{desc.MaxSatisfactionWeight(true)};
      59         [ +  - ]:       13725 :     const auto max_elems{desc.MaxSatisfactionElems()};
      60                 :             :     // We must be able to estimate the max satisfaction size for any solvable descriptor (but combo).
      61   [ +  +  +  -  :       13725 :     const bool is_nontop_or_nonsolvable{!*is_solvable || !desc.GetOutputType()};
                   +  + ]
      62   [ +  +  +  -  :       13725 :     const bool is_input_size_info_set{max_sat_maxsig && max_sat_nonmaxsig && max_elems};
                   -  + ]
      63         [ -  + ]:         705 :     assert(is_input_size_info_set || is_nontop_or_nonsolvable);
      64                 :       13725 : }
      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         [ +  - ]:        5707 : FUZZ_TARGET(mocked_descriptor_parse, .init = initialize_mocked_descriptor_parse)
      79                 :             : {
      80         [ -  + ]:        5257 :     const std::string mocked_descriptor{buffer.begin(), buffer.end()};
      81   [ -  +  +  -  :        5257 :     if (const auto descriptor = MOCKED_DESC_CONVERTER.GetDescriptor(mocked_descriptor)) {
                   +  + ]
      82   [ +  -  +  + ]:        5225 :         if (IsTooExpensive(MakeUCharSpan(*descriptor))) return;
      83                 :        4184 :         FlatSigningProvider signing_provider;
      84         [ -  + ]:        4184 :         std::string error;
      85   [ -  +  +  - ]:        4184 :         const auto desc = Parse(*descriptor, signing_provider, error);
      86                 :        4184 :         std::optional<bool> is_ranged;
      87                 :        4184 :         std::optional<bool> is_solvable;
      88         [ +  + ]:       14746 :         for (const auto& d : desc) {
      89         [ -  + ]:       10562 :             assert(d);
      90         [ +  - ]:       10562 :             TestDescriptor(*d, signing_provider, error, is_ranged, is_solvable);
      91                 :             :         }
      92                 :        9441 :     }
      93                 :        5257 : }
      94                 :             : 
      95         [ +  - ]:        4988 : FUZZ_TARGET(descriptor_parse, .init = initialize_descriptor_parse)
      96                 :             : {
      97         [ +  + ]:        4538 :     if (IsTooExpensive(buffer)) return;
      98                 :             : 
      99                 :        3750 :     const std::string descriptor(buffer.begin(), buffer.end());
     100                 :        3750 :     FlatSigningProvider signing_provider;
     101                 :        3750 :     std::string error;
     102         [ +  + ]:       11250 :     for (const bool require_checksum : {true, false}) {
     103   [ -  +  +  - ]:        7500 :         const auto desc = Parse(descriptor, signing_provider, error, require_checksum);
     104                 :        7500 :         std::optional<bool> is_ranged;
     105                 :        7500 :         std::optional<bool> is_solvable;
     106         [ +  + ]:       10663 :         for (const auto& d : desc) {
     107         [ -  + ]:        3163 :             assert(d);
     108         [ +  - ]:        3163 :             TestDescriptor(*d, signing_provider, error, is_ranged, is_solvable);
     109                 :             :         }
     110                 :        7500 :     }
     111                 :        3750 : }