message.cpp raw

   1  // Copyright (c) 2020-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 <common/signmessage.h>
   7  #include <key_io.h>
   8  #include <test/fuzz/FuzzedDataProvider.h>
   9  #include <test/fuzz/fuzz.h>
  10  #include <test/fuzz/util.h>
  11  #include <util/chaintype.h>
  12  #include <util/strencodings.h>
  13  
  14  #include <cassert>
  15  #include <cstdint>
  16  #include <iostream>
  17  #include <string>
  18  #include <vector>
  19  
  20  void initialize_message()
  21  {
  22      static ECC_Context ecc_context{};
  23      SelectParams(ChainType::REGTEST);
  24  }
  25  
  26  FUZZ_TARGET(message, .init = initialize_message)
  27  {
  28      FuzzedDataProvider fuzzed_data_provider(buffer.data(), buffer.size());
  29      const std::string random_message = fuzzed_data_provider.ConsumeRandomLengthString(1024);
  30      {
  31          CKey private_key = ConsumePrivateKey(fuzzed_data_provider);
  32          std::string signature;
  33          const bool message_signed = MessageSign(private_key, random_message, signature);
  34          if (private_key.IsValid()) {
  35              assert(message_signed);
  36              const MessageVerificationResult verification_result = MessageVerify(EncodeDestination(PKHash(private_key.GetPubKey().GetID())), signature, random_message);
  37              assert(verification_result == MessageVerificationResult::OK);
  38          }
  39      }
  40      {
  41          (void)MessageHash(random_message);
  42          auto address = fuzzed_data_provider.ConsumeRandomLengthString(1024);
  43          auto signature = fuzzed_data_provider.ConsumeRandomLengthString(1024);
  44          (void)MessageVerify(address, signature, random_message);
  45          (void)SigningResultString(fuzzed_data_provider.PickValueInArray({SigningResult::OK, SigningResult::PRIVATE_KEY_NOT_AVAILABLE, SigningResult::SIGNING_FAILED}));
  46      }
  47  }
  48