phantom/bitcoin
1
0
Fork 0
mirror of https://github.com/bitcoin/bitcoin.git synced 2025-04-29 23:09:44 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

fuzz: wallet: fix crypter target

Browse source
This commit is contained in:
brunoerg 2025-03-20 19:12:32 -03:00
parent 8046759305
commit 28dc118001
1 changed files with 13 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
src/wallet/test/fuzz/crypter.cpp
View file

@ -20,6 +20,7 @@ void initialize_crypter()
FUZZ_TARGET(crypter, .init = initialize_crypter) FUZZ_TARGET(crypter, .init = initialize_crypter)
{ {
SeedRandomStateForTest(SeedRand::ZEROS);
FuzzedDataProvider fuzzed_data_provider(buffer.data(), buffer.size()); FuzzedDataProvider fuzzed_data_provider(buffer.data(), buffer.size());
bool good_data{true}; bool good_data{true};
@ -42,6 +43,11 @@ FUZZ_TARGET(crypter, .init = initialize_crypter)
/*derivation_method=*/derivation_method); /*derivation_method=*/derivation_method);
} }
CKey random_ckey;
random_ckey.Set(random_key.begin(), random_key.end(), /*fCompressedIn=*/fuzzed_data_provider.ConsumeBool());
if (!random_ckey.IsValid()) return;
CPubKey pubkey{random_ckey.GetPubKey()};
LIMITED_WHILE(good_data && fuzzed_data_provider.ConsumeBool(), 100) LIMITED_WHILE(good_data && fuzzed_data_provider.ConsumeBool(), 100)
{ {
CallOneOf( CallOneOf(
@ -60,26 +66,21 @@ FUZZ_TARGET(crypter, .init = initialize_crypter)
(void)crypt.Decrypt(cipher_text_ed, plain_text_ed); (void)crypt.Decrypt(cipher_text_ed, plain_text_ed);
}, },
[&] { [&] {
const CKeyingMaterial master_key(random_key.begin(), random_key.end()); const CKeyingMaterial master_key(random_key.begin(), random_key.end());;
const uint256 iv = ConsumeUInt256(fuzzed_data_provider); (void)EncryptSecret(master_key, plain_text_ed, pubkey.GetHash(), cipher_text_ed);
(void)EncryptSecret(master_key, plain_text_ed, iv, cipher_text_ed);
}, },
[&] { [&] {
const CKeyingMaterial master_key(random_key.begin(), random_key.end()); std::optional<CPubKey> random_pub_key{ConsumeDeserializable<CPubKey>(fuzzed_data_provider)};
const uint256 iv = ConsumeUInt256(fuzzed_data_provider);
(void)DecryptSecret(master_key, cipher_text_ed, iv, plain_text_ed);
},
[&] {
std::optional<CPubKey> random_pub_key = ConsumeDeserializable<CPubKey>(fuzzed_data_provider);
if (!random_pub_key) { if (!random_pub_key) {
good_data = false; good_data = false;
return; return;
} }
const CPubKey pub_key = *random_pub_key; pubkey = *random_pub_key;
},
[&] {
const CKeyingMaterial master_key(random_key.begin(), random_key.end()); const CKeyingMaterial master_key(random_key.begin(), random_key.end());
const std::vector<unsigned char> crypted_secret = ConsumeRandomLengthByteVector(fuzzed_data_provider, 64);
CKey key; CKey key;
(void)DecryptKey(master_key, crypted_secret, pub_key, key); (void)DecryptKey(master_key, cipher_text_ed, pubkey, key);
}); });
} }
} }