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