Bitcoin Core  27.99.0
P2P Digital Currency
wallet_crypto_tests.cpp
Go to the documentation of this file.
1 // Copyright (c) 2014-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/util/random.h>
7 #include <util/strencodings.h>
8 #include <wallet/crypter.h>
9 
10 #include <vector>
11 
12 #include <boost/test/unit_test.hpp>
13 
14 namespace wallet {
15 BOOST_FIXTURE_TEST_SUITE(wallet_crypto_tests, BasicTestingSetup)
16 
18 {
19 public:
20 static void TestPassphraseSingle(const std::vector<unsigned char>& vchSalt, const SecureString& passphrase, uint32_t rounds,
21  const std::vector<unsigned char>& correctKey = std::vector<unsigned char>(),
22  const std::vector<unsigned char>& correctIV=std::vector<unsigned char>())
23 {
24  CCrypter crypt;
25  crypt.SetKeyFromPassphrase(passphrase, vchSalt, rounds, 0);
26 
27  if(!correctKey.empty())
28  BOOST_CHECK_MESSAGE(memcmp(crypt.vchKey.data(), correctKey.data(), crypt.vchKey.size()) == 0, \
29  HexStr(crypt.vchKey) + std::string(" != ") + HexStr(correctKey));
30  if(!correctIV.empty())
31  BOOST_CHECK_MESSAGE(memcmp(crypt.vchIV.data(), correctIV.data(), crypt.vchIV.size()) == 0,
32  HexStr(crypt.vchIV) + std::string(" != ") + HexStr(correctIV));
33 }
34 
35 static void TestPassphrase(const std::vector<unsigned char>& vchSalt, const SecureString& passphrase, uint32_t rounds,
36  const std::vector<unsigned char>& correctKey = std::vector<unsigned char>(),
37  const std::vector<unsigned char>& correctIV=std::vector<unsigned char>())
38 {
39  TestPassphraseSingle(vchSalt, passphrase, rounds, correctKey, correctIV);
40  for(SecureString::const_iterator i(passphrase.begin()); i != passphrase.end(); ++i)
41  TestPassphraseSingle(vchSalt, SecureString(i, passphrase.end()), rounds);
42 }
43 
44 static void TestDecrypt(const CCrypter& crypt, const std::vector<unsigned char>& vchCiphertext, \
45  const std::vector<unsigned char>& vchPlaintext = std::vector<unsigned char>())
46 {
47  CKeyingMaterial vchDecrypted;
48  crypt.Decrypt(vchCiphertext, vchDecrypted);
49  if (vchPlaintext.size())
50  BOOST_CHECK(CKeyingMaterial(vchPlaintext.begin(), vchPlaintext.end()) == vchDecrypted);
51 }
52 
53 static void TestEncryptSingle(const CCrypter& crypt, const CKeyingMaterial& vchPlaintext,
54  const std::vector<unsigned char>& vchCiphertextCorrect = std::vector<unsigned char>())
55 {
56  std::vector<unsigned char> vchCiphertext;
57  crypt.Encrypt(vchPlaintext, vchCiphertext);
58 
59  if (!vchCiphertextCorrect.empty())
60  BOOST_CHECK(vchCiphertext == vchCiphertextCorrect);
61 
62  const std::vector<unsigned char> vchPlaintext2(vchPlaintext.begin(), vchPlaintext.end());
63  TestDecrypt(crypt, vchCiphertext, vchPlaintext2);
64 }
65 
66 static void TestEncrypt(const CCrypter& crypt, const std::vector<unsigned char>& vchPlaintextIn, \
67  const std::vector<unsigned char>& vchCiphertextCorrect = std::vector<unsigned char>())
68 {
69  TestEncryptSingle(crypt, CKeyingMaterial(vchPlaintextIn.begin(), vchPlaintextIn.end()), vchCiphertextCorrect);
70  for(std::vector<unsigned char>::const_iterator i(vchPlaintextIn.begin()); i != vchPlaintextIn.end(); ++i)
71  TestEncryptSingle(crypt, CKeyingMaterial(i, vchPlaintextIn.end()));
72 }
73 
74 };
75 
76 BOOST_AUTO_TEST_CASE(passphrase) {
77  // These are expensive.
78 
79  TestCrypter::TestPassphrase(ParseHex("0000deadbeef0000"), "test", 25000, \
80  ParseHex("fc7aba077ad5f4c3a0988d8daa4810d0d4a0e3bcb53af662998898f33df0556a"), \
81  ParseHex("cf2f2691526dd1aa220896fb8bf7c369"));
82 
83  std::string hash(GetRandHash().ToString());
84  std::vector<unsigned char> vchSalt(8);
85  GetRandBytes(vchSalt);
86  uint32_t rounds = InsecureRand32();
87  if (rounds > 30000)
88  rounds = 30000;
89  TestCrypter::TestPassphrase(vchSalt, SecureString(hash.begin(), hash.end()), rounds);
90 }
91 
93  std::vector<unsigned char> vchSalt = ParseHex("0000deadbeef0000");
94  BOOST_CHECK(vchSalt.size() == WALLET_CRYPTO_SALT_SIZE);
95  CCrypter crypt;
96  crypt.SetKeyFromPassphrase("passphrase", vchSalt, 25000, 0);
97  TestCrypter::TestEncrypt(crypt, ParseHex("22bcade09ac03ff6386914359cfe885cfeb5f77ff0d670f102f619687453b29d"));
98 
99  for (int i = 0; i != 100; i++)
100  {
101  uint256 hash(GetRandHash());
102  TestCrypter::TestEncrypt(crypt, std::vector<unsigned char>(hash.begin(), hash.end()));
103  }
104 
105 }
106 
108  std::vector<unsigned char> vchSalt = ParseHex("0000deadbeef0000");
109  BOOST_CHECK(vchSalt.size() == WALLET_CRYPTO_SALT_SIZE);
110  CCrypter crypt;
111  crypt.SetKeyFromPassphrase("passphrase", vchSalt, 25000, 0);
112 
113  // Some corner cases the came up while testing
114  TestCrypter::TestDecrypt(crypt,ParseHex("795643ce39d736088367822cdc50535ec6f103715e3e48f4f3b1a60a08ef59ca"));
115  TestCrypter::TestDecrypt(crypt,ParseHex("de096f4a8f9bd97db012aa9d90d74de8cdea779c3ee8bc7633d8b5d6da703486"));
116  TestCrypter::TestDecrypt(crypt,ParseHex("32d0a8974e3afd9c6c3ebf4d66aa4e6419f8c173de25947f98cf8b7ace49449c"));
117  TestCrypter::TestDecrypt(crypt,ParseHex("e7c055cca2faa78cb9ac22c9357a90b4778ded9b2cc220a14cea49f931e596ea"));
118  TestCrypter::TestDecrypt(crypt,ParseHex("b88efddd668a6801d19516d6830da4ae9811988ccbaf40df8fbb72f3f4d335fd"));
119  TestCrypter::TestDecrypt(crypt,ParseHex("8cae76aa6a43694e961ebcb28c8ca8f8540b84153d72865e8561ddd93fa7bfa9"));
120 
121  for (int i = 0; i != 100; i++)
122  {
123  uint256 hash(GetRandHash());
124  TestCrypter::TestDecrypt(crypt, std::vector<unsigned char>(hash.begin(), hash.end()));
125  }
126 }
127 
129 } // namespace wallet
constexpr unsigned char * end()
Definition: uint256.h:69
constexpr unsigned char * begin()
Definition: uint256.h:68
256-bit opaque blob.
Definition: uint256.h:106
Encryption/decryption context with key information.
Definition: crypter.h:71
bool Decrypt(const std::vector< unsigned char > &vchCiphertext, CKeyingMaterial &vchPlaintext) const
Definition: crypter.cpp:90
std::vector< unsigned char, secure_allocator< unsigned char > > vchKey
Definition: crypter.h:74
bool Encrypt(const CKeyingMaterial &vchPlaintext, std::vector< unsigned char > &vchCiphertext) const
Definition: crypter.cpp:72
std::vector< unsigned char, secure_allocator< unsigned char > > vchIV
Definition: crypter.h:75
bool SetKeyFromPassphrase(const SecureString &strKeyData, const std::vector< unsigned char > &chSalt, const unsigned int nRounds, const unsigned int nDerivationMethod)
Definition: crypter.cpp:40
static void TestDecrypt(const CCrypter &crypt, const std::vector< unsigned char > &vchCiphertext, const std::vector< unsigned char > &vchPlaintext=std::vector< unsigned char >())
static void TestPassphrase(const std::vector< unsigned char > &vchSalt, const SecureString &passphrase, uint32_t rounds, const std::vector< unsigned char > &correctKey=std::vector< unsigned char >(), const std::vector< unsigned char > &correctIV=std::vector< unsigned char >())
static void TestEncryptSingle(const CCrypter &crypt, const CKeyingMaterial &vchPlaintext, const std::vector< unsigned char > &vchCiphertextCorrect=std::vector< unsigned char >())
static void TestEncrypt(const CCrypter &crypt, const std::vector< unsigned char > &vchPlaintextIn, const std::vector< unsigned char > &vchCiphertextCorrect=std::vector< unsigned char >())
static void TestPassphraseSingle(const std::vector< unsigned char > &vchSalt, const SecureString &passphrase, uint32_t rounds, const std::vector< unsigned char > &correctKey=std::vector< unsigned char >(), const std::vector< unsigned char > &correctIV=std::vector< unsigned char >())
BOOST_AUTO_TEST_SUITE_END()
std::vector< unsigned char, secure_allocator< unsigned char > > CKeyingMaterial
Definition: crypter.h:62
const unsigned int WALLET_CRYPTO_SALT_SIZE
Definition: crypter.h:15
BOOST_AUTO_TEST_CASE(bnb_search_test)
#define BOOST_CHECK(expr)
Definition: object.cpp:17
void GetRandBytes(Span< unsigned char > bytes) noexcept
Overall design of the RNG and entropy sources.
Definition: random.cpp:638
uint256 GetRandHash() noexcept
Definition: random.cpp:650
std::basic_string< char, std::char_traits< char >, secure_allocator< char > > SecureString
Definition: secure.h:58
std::vector< Byte > ParseHex(std::string_view hex_str)
Like TryParseHex, but returns an empty vector on invalid input.
Definition: strencodings.h:65
std::string ToString(const T &t)
Locale-independent version of std::to_string.
Definition: string.h:109
Basic testing setup.
Definition: setup_common.h:52
static uint32_t InsecureRand32()
Definition: random.h:45
std::string HexStr(const Span< const uint8_t > s)
Convert a span of bytes to a lower-case hexadecimal string.