Branch data Line data Source code
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>
6 : : #include <test/util/setup_common.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 : 0 : BOOST_FIXTURE_TEST_SUITE(wallet_crypto_tests, BasicTestingSetup)
16 : :
17 : : class TestCrypter
18 : : {
19 : : public:
20 : 0 : 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 : 0 : CCrypter crypt;
25 : 0 : crypt.SetKeyFromPassphrase(passphrase, vchSalt, rounds, 0);
26 : :
27 : 0 : if(!correctKey.empty())
28 : 0 : BOOST_CHECK_MESSAGE(memcmp(crypt.vchKey.data(), correctKey.data(), crypt.vchKey.size()) == 0, \
29 : : HexStr(crypt.vchKey) + std::string(" != ") + HexStr(correctKey));
30 : 0 : if(!correctIV.empty())
31 : 0 : BOOST_CHECK_MESSAGE(memcmp(crypt.vchIV.data(), correctIV.data(), crypt.vchIV.size()) == 0,
32 : : HexStr(crypt.vchIV) + std::string(" != ") + HexStr(correctIV));
33 : 0 : }
34 : :
35 : 0 : 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 : 0 : TestPassphraseSingle(vchSalt, passphrase, rounds, correctKey, correctIV);
40 : 0 : for(SecureString::const_iterator i(passphrase.begin()); i != passphrase.end(); ++i)
41 : 0 : TestPassphraseSingle(vchSalt, SecureString(i, passphrase.end()), rounds);
42 : 0 : }
43 : :
44 : 0 : 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 : 0 : CKeyingMaterial vchDecrypted;
48 : 0 : crypt.Decrypt(vchCiphertext, vchDecrypted);
49 : 0 : if (vchPlaintext.size())
50 : 0 : BOOST_CHECK(CKeyingMaterial(vchPlaintext.begin(), vchPlaintext.end()) == vchDecrypted);
51 : 0 : }
52 : :
53 : 0 : static void TestEncryptSingle(const CCrypter& crypt, const CKeyingMaterial& vchPlaintext,
54 : : const std::vector<unsigned char>& vchCiphertextCorrect = std::vector<unsigned char>())
55 : : {
56 : 0 : std::vector<unsigned char> vchCiphertext;
57 : 0 : crypt.Encrypt(vchPlaintext, vchCiphertext);
58 : :
59 : 0 : if (!vchCiphertextCorrect.empty())
60 : 0 : BOOST_CHECK(vchCiphertext == vchCiphertextCorrect);
61 : :
62 : 0 : const std::vector<unsigned char> vchPlaintext2(vchPlaintext.begin(), vchPlaintext.end());
63 : 0 : TestDecrypt(crypt, vchCiphertext, vchPlaintext2);
64 : 0 : }
65 : :
66 : 0 : 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 : 0 : TestEncryptSingle(crypt, CKeyingMaterial(vchPlaintextIn.begin(), vchPlaintextIn.end()), vchCiphertextCorrect);
70 : 0 : for(std::vector<unsigned char>::const_iterator i(vchPlaintextIn.begin()); i != vchPlaintextIn.end(); ++i)
71 : 0 : TestEncryptSingle(crypt, CKeyingMaterial(i, vchPlaintextIn.end()));
72 : 0 : }
73 : :
74 : 0 : };
75 : :
76 : 0 : BOOST_AUTO_TEST_CASE(passphrase) {
77 : : // These are expensive.
78 : :
79 : 0 : TestCrypter::TestPassphrase(ParseHex("0000deadbeef0000"), "test", 25000, \
80 : 0 : ParseHex("fc7aba077ad5f4c3a0988d8daa4810d0d4a0e3bcb53af662998898f33df0556a"), \
81 : 0 : ParseHex("cf2f2691526dd1aa220896fb8bf7c369"));
82 : :
83 : 0 : std::string hash(GetRandHash().ToString());
84 : 0 : std::vector<unsigned char> vchSalt(8);
85 : 0 : GetRandBytes(vchSalt);
86 : 0 : uint32_t rounds = InsecureRand32();
87 : 0 : if (rounds > 30000)
88 : 0 : rounds = 30000;
89 : 0 : TestCrypter::TestPassphrase(vchSalt, SecureString(hash.begin(), hash.end()), rounds);
90 : 0 : }
91 : :
92 : 0 : BOOST_AUTO_TEST_CASE(encrypt) {
93 : 0 : std::vector<unsigned char> vchSalt = ParseHex("0000deadbeef0000");
94 : 0 : BOOST_CHECK(vchSalt.size() == WALLET_CRYPTO_SALT_SIZE);
95 : 0 : CCrypter crypt;
96 : 0 : crypt.SetKeyFromPassphrase("passphrase", vchSalt, 25000, 0);
97 : 0 : TestCrypter::TestEncrypt(crypt, ParseHex("22bcade09ac03ff6386914359cfe885cfeb5f77ff0d670f102f619687453b29d"));
98 : :
99 : 0 : for (int i = 0; i != 100; i++)
100 : : {
101 : 0 : uint256 hash(GetRandHash());
102 : 0 : TestCrypter::TestEncrypt(crypt, std::vector<unsigned char>(hash.begin(), hash.end()));
103 : 0 : }
104 : :
105 : 0 : }
106 : :
107 : 0 : BOOST_AUTO_TEST_CASE(decrypt) {
108 : 0 : std::vector<unsigned char> vchSalt = ParseHex("0000deadbeef0000");
109 : 0 : BOOST_CHECK(vchSalt.size() == WALLET_CRYPTO_SALT_SIZE);
110 : 0 : CCrypter crypt;
111 : 0 : crypt.SetKeyFromPassphrase("passphrase", vchSalt, 25000, 0);
112 : :
113 : : // Some corner cases the came up while testing
114 : 0 : TestCrypter::TestDecrypt(crypt,ParseHex("795643ce39d736088367822cdc50535ec6f103715e3e48f4f3b1a60a08ef59ca"));
115 : 0 : TestCrypter::TestDecrypt(crypt,ParseHex("de096f4a8f9bd97db012aa9d90d74de8cdea779c3ee8bc7633d8b5d6da703486"));
116 : 0 : TestCrypter::TestDecrypt(crypt,ParseHex("32d0a8974e3afd9c6c3ebf4d66aa4e6419f8c173de25947f98cf8b7ace49449c"));
117 : 0 : TestCrypter::TestDecrypt(crypt,ParseHex("e7c055cca2faa78cb9ac22c9357a90b4778ded9b2cc220a14cea49f931e596ea"));
118 : 0 : TestCrypter::TestDecrypt(crypt,ParseHex("b88efddd668a6801d19516d6830da4ae9811988ccbaf40df8fbb72f3f4d335fd"));
119 : 0 : TestCrypter::TestDecrypt(crypt,ParseHex("8cae76aa6a43694e961ebcb28c8ca8f8540b84153d72865e8561ddd93fa7bfa9"));
120 : :
121 : 0 : for (int i = 0; i != 100; i++)
122 : : {
123 : 0 : uint256 hash(GetRandHash());
124 : 0 : TestCrypter::TestDecrypt(crypt, std::vector<unsigned char>(hash.begin(), hash.end()));
125 : 0 : }
126 : 0 : }
127 : :
128 : 0 : BOOST_AUTO_TEST_SUITE_END()
129 : : } // namespace wallet
|
