Line data    Source code

       1             : // Copyright (c) 2022 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 <test/fuzz/FuzzedDataProvider.h>
       6             : #include <test/fuzz/fuzz.h>
       7             : #include <test/fuzz/util.h>
       8             : #include <test/util/setup_common.h>
       9             : #include <wallet/coincontrol.h>
      10             : #include <wallet/test/util.h>
      11             : 
      12             : namespace wallet {
      13             : namespace {
      14             : 
      15             : const TestingSetup* g_setup;
      16             : 
      17         174 : void initialize_coincontrol()
      18         173 : {
      19           1 :     static const auto testing_setup = MakeNoLogFileContext<const TestingSetup>();
      20           1 :     g_setup = testing_setup.get();
      21           1 : }
      22             : 
      23        3185 : FUZZ_TARGET(coincontrol, .init = initialize_coincontrol)
      24             : {
      25        2839 :     FuzzedDataProvider fuzzed_data_provider(buffer.data(), buffer.size());
      26        2839 :     const auto& node = g_setup->m_node;
      27        2839 :     ArgsManager& args = *node.args;
      28             : 
      29             :     // for GetBoolArg to return true sometimes
      30        2839 :     args.ForceSetArg("-avoidpartialspends", fuzzed_data_provider.ConsumeBool()?"1":"0");
      31             : 
      32        2839 :     CCoinControl coin_control;
      33        2839 :     COutPoint out_point;
      34             : 
      35      191785 :     LIMITED_WHILE(fuzzed_data_provider.ConsumeBool(), 10000)
      36             :     {
      37      188773 :         CallOneOf(
      38             :             fuzzed_data_provider,
      39      206956 :             [&] {
      40       18183 :                 std::optional<COutPoint> optional_out_point = ConsumeDeserializable<COutPoint>(fuzzed_data_provider);
      41       18183 :                 if (!optional_out_point) {
      42        6395 :                     return;
      43             :                 }
      44       11788 :                 out_point = *optional_out_point;
      45       18183 :             },
      46      196703 :             [&] {
      47        7930 :                 (void)coin_control.HasSelected();
      48        7930 :             },
      49      211144 :             [&] {
      50       22371 :                 (void)coin_control.IsSelected(out_point);
      51       22371 :             },
      52      207937 :             [&] {
      53       19164 :                 (void)coin_control.IsExternalSelected(out_point);
      54       19164 :             },
      55      197779 :             [&] {
      56        9006 :                 (void)coin_control.GetExternalOutput(out_point);
      57        9006 :             },
      58      214924 :             [&] {
      59       26151 :                 (void)coin_control.Select(out_point);
      60       26151 :             },
      61      202649 :             [&] {
      62       13876 :                 const CTxOut tx_out{ConsumeMoney(fuzzed_data_provider), ConsumeScript(fuzzed_data_provider)};
      63       13876 :                 (void)coin_control.SelectExternal(out_point, tx_out);
      64       13876 :             },
      65      227786 :             [&] {
      66       39013 :                 (void)coin_control.UnSelect(out_point);
      67       39013 :             },
      68      192432 :             [&] {
      69        3659 :                 (void)coin_control.UnSelectAll();
      70        3659 :             },
      71      194670 :             [&] {
      72        5897 :                 (void)coin_control.ListSelected();
      73        5897 :             },
      74      201372 :             [&] {
      75       12426 :                 int64_t weight{fuzzed_data_provider.ConsumeIntegral<int64_t>()};
      76       12426 :                 (void)coin_control.SetInputWeight(out_point, weight);
      77       12426 :             },
      78      199870 :             [&] {
      79             :                 // Condition to avoid the assertion in GetInputWeight
      80       11097 :                 if (coin_control.HasInputWeight(out_point)) {
      81        5798 :                     (void)coin_control.GetInputWeight(out_point);
      82        5798 :                 }
      83       11097 :             });
      84      188773 :     }
      85        2839 : }
      86             : } // namespace
      87             : } // namespace wallet