Branch data     Line data    Source code

       1                 :             : // Copyright (c) 2019-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/fuzz.h>
       6                 :             : 
       7                 :             : #include <base58.h>
       8                 :             : #include <psbt.h>
       9                 :             : #include <span.h>
      10                 :             : #include <test/fuzz/FuzzedDataProvider.h>
      11                 :             : #include <util/strencodings.h>
      12                 :             : #include <util/string.h>
      13                 :             : 
      14                 :             : #include <cassert>
      15                 :             : #include <string>
      16                 :             : #include <vector>
      17                 :             : #include <ranges>
      18                 :             : 
      19                 :             : using util::TrimStringView;
      20                 :             : 
      21         [ +  - ]:         530 : FUZZ_TARGET(base58_encode_decode)
      22                 :             : {
      23                 :          90 :     FuzzedDataProvider provider{buffer.data(), buffer.size()};
      24                 :          90 :     const auto random_string{provider.ConsumeRandomLengthString(100)};
      25                 :             : 
      26         [ +  - ]:          90 :     const auto encoded{EncodeBase58(MakeUCharSpan(random_string))};
      27   [ +  +  +  - ]:         129 :     const auto decode_input{provider.ConsumeBool() ? random_string : encoded};
      28                 :          90 :     const int max_ret_len{provider.ConsumeIntegralInRange<int>(-1, decode_input.size() + 1)};
      29   [ +  -  +  + ]:          90 :     if (std::vector<unsigned char> decoded; DecodeBase58(decode_input, decoded, max_ret_len)) {
      30         [ +  - ]:          61 :         const auto encoded_string{EncodeBase58(decoded)};
      31   [ +  -  -  + ]:          61 :         assert(encoded_string == TrimStringView(decode_input));
      32         [ +  + ]:          61 :         if (decoded.size() > 0) {
      33         [ -  + ]:          44 :             assert(max_ret_len > 0);
      34         [ -  + ]:          44 :             assert(decoded.size() <= static_cast<size_t>(max_ret_len));
      35   [ +  -  -  + ]:          44 :             assert(!DecodeBase58(encoded_string, decoded, provider.ConsumeIntegralInRange<int>(0, decoded.size() - 1)));
      36                 :             :         }
      37                 :         151 :     }
      38                 :          90 : }
      39                 :             : 
      40         [ +  - ]:         537 : FUZZ_TARGET(base58check_encode_decode)
      41                 :             : {
      42                 :          97 :     FuzzedDataProvider provider{buffer.data(), buffer.size()};
      43                 :          97 :     const auto random_string{provider.ConsumeRandomLengthString(100)};
      44                 :             : 
      45         [ +  - ]:          97 :     const auto encoded{EncodeBase58Check(MakeUCharSpan(random_string))};
      46   [ +  +  +  - ]:         140 :     const auto decode_input{provider.ConsumeBool() ? random_string : encoded};
      47                 :          97 :     const int max_ret_len{provider.ConsumeIntegralInRange<int>(-1, decode_input.size() + 1)};
      48   [ +  -  +  + ]:          97 :     if (std::vector<unsigned char> decoded; DecodeBase58Check(decode_input, decoded, max_ret_len)) {
      49         [ +  - ]:          25 :         const auto encoded_string{EncodeBase58Check(decoded)};
      50   [ +  -  -  + ]:          25 :         assert(encoded_string == TrimStringView(decode_input));
      51         [ +  + ]:          25 :         if (decoded.size() > 0) {
      52         [ -  + ]:          24 :             assert(max_ret_len > 0);
      53         [ -  + ]:          24 :             assert(decoded.size() <= static_cast<size_t>(max_ret_len));
      54   [ +  -  -  + ]:          24 :             assert(!DecodeBase58Check(encoded_string, decoded, provider.ConsumeIntegralInRange<int>(0, decoded.size() - 1)));
      55                 :             :         }
      56                 :         122 :     }
      57                 :          97 : }
      58                 :             : 
      59         [ +  - ]:         474 : FUZZ_TARGET(base32_encode_decode)
      60                 :             : {
      61         [ +  - ]:          34 :     const std::string random_string{buffer.begin(), buffer.end()};
      62                 :             : 
      63                 :             :     // Decode/Encode roundtrip
      64   [ +  -  +  + ]:          34 :     if (auto result{DecodeBase32(random_string)}) {
      65         [ +  - ]:          17 :         const auto encoded_string{EncodeBase32(*result)};
      66   [ +  -  +  -  :          34 :         assert(encoded_string == ToLower(TrimStringView(random_string)));
                   -  + ]
      67                 :          17 :     }
      68                 :             :     // Encode/Decode roundtrip
      69         [ +  - ]:          34 :     const auto encoded{EncodeBase32(buffer)};
      70         [ +  - ]:          34 :     const auto decoded{DecodeBase32(encoded)};
      71   [ +  -  -  + ]:          34 :     assert(decoded && std::ranges::equal(*decoded, buffer));
      72                 :          34 : }
      73                 :             : 
      74         [ +  - ]:         464 : FUZZ_TARGET(base64_encode_decode)
      75                 :             : {
      76         [ +  - ]:          24 :     const std::string random_string{buffer.begin(), buffer.end()};
      77                 :             : 
      78                 :             :     // Decode/Encode roundtrip
      79   [ +  -  +  + ]:          24 :     if (auto result{DecodeBase64(random_string)}) {
      80         [ +  - ]:          12 :         const auto encoded_string{EncodeBase64(*result)};
      81   [ +  -  -  + ]:          12 :         assert(encoded_string == TrimStringView(random_string));
      82                 :          12 :     }
      83                 :             :     // Encode/Decode roundtrip
      84         [ +  - ]:          24 :     const auto encoded{EncodeBase64(buffer)};
      85         [ +  - ]:          24 :     const auto decoded{DecodeBase64(encoded)};
      86   [ +  -  -  + ]:          24 :     assert(decoded && std::ranges::equal(*decoded, buffer));
      87                 :          24 : }
      88                 :             : 
      89         [ +  - ]:        1664 : FUZZ_TARGET(psbt_base64_decode)
      90                 :             : {
      91                 :        1224 :     const std::string random_string{buffer.begin(), buffer.end()};
      92                 :             : 
      93                 :        1224 :     PartiallySignedTransaction psbt;
      94         [ +  - ]:        1224 :     std::string error;
      95         [ +  - ]:        1224 :     const bool ok{DecodeBase64PSBT(psbt, random_string, error)};
      96         [ -  + ]:        1224 :     assert(ok == error.empty());
      97                 :        1224 : }