Hush Full Node software. We were censored from Github, this is where all development happens now.
https://hush.is
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
204 lines
7.1 KiB
204 lines
7.1 KiB
// Copyright (c) 2016-2020 The Hush developers
|
|
// Distributed under the GPLv3 software license, see the accompanying
|
|
// file COPYING or https://www.gnu.org/licenses/gpl-3.0.en.html
|
|
#include <gtest/gtest.h>
|
|
#include "zcash/Address.hpp"
|
|
#include "wallet/wallet.h"
|
|
#include "wallet/walletdb.h"
|
|
#include "util.h"
|
|
#include <boost/filesystem.hpp>
|
|
|
|
/**
|
|
* This test covers Sapling methods on CWallet
|
|
* GenerateNewSaplingZKey()
|
|
* AddSaplingZKey()
|
|
* AddSaplingIncomingViewingKey()
|
|
* LoadSaplingZKey()
|
|
* LoadSaplingIncomingViewingKey()
|
|
* LoadSaplingZKeyMetadata()
|
|
*/
|
|
TEST(wallet_zkeys_tests, StoreAndLoadSaplingZkeys) {
|
|
SelectParams(CBaseChainParams::MAIN);
|
|
|
|
CWallet wallet;
|
|
|
|
// wallet should be empty
|
|
std::set<libzcash::SaplingPaymentAddress> addrs;
|
|
wallet.GetSaplingPaymentAddresses(addrs);
|
|
ASSERT_EQ(0, addrs.size());
|
|
|
|
// No HD seed in the wallet
|
|
EXPECT_ANY_THROW(wallet.GenerateNewSaplingZKey());
|
|
|
|
// Load the all-zeroes seed
|
|
CKeyingMaterial rawSeed(32, 0);
|
|
HDSeed seed(rawSeed);
|
|
wallet.LoadHDSeed(seed);
|
|
|
|
// Now this call succeeds
|
|
auto address = wallet.GenerateNewSaplingZKey();
|
|
|
|
// wallet should have one key
|
|
wallet.GetSaplingPaymentAddresses(addrs);
|
|
ASSERT_EQ(1, addrs.size());
|
|
|
|
// verify wallet has incoming viewing key for the address
|
|
ASSERT_TRUE(wallet.HaveSaplingIncomingViewingKey(address));
|
|
|
|
// manually add new spending key to wallet
|
|
auto m = libzcash::SaplingExtendedSpendingKey::Master(seed);
|
|
auto sk = m.Derive(0);
|
|
ASSERT_TRUE(wallet.AddSaplingZKey(sk, sk.DefaultAddress()));
|
|
|
|
// verify wallet did add it
|
|
auto fvk = sk.expsk.full_viewing_key();
|
|
ASSERT_TRUE(wallet.HaveSaplingSpendingKey(fvk));
|
|
|
|
// verify spending key stored correctly
|
|
libzcash::SaplingExtendedSpendingKey keyOut;
|
|
wallet.GetSaplingSpendingKey(fvk, keyOut);
|
|
ASSERT_EQ(sk, keyOut);
|
|
|
|
// verify there are two keys
|
|
wallet.GetSaplingPaymentAddresses(addrs);
|
|
EXPECT_EQ(2, addrs.size());
|
|
EXPECT_EQ(1, addrs.count(address));
|
|
EXPECT_EQ(1, addrs.count(sk.DefaultAddress()));
|
|
|
|
// Generate a diversified address different to the default
|
|
// If we can't get an early diversified address, we are very unlucky
|
|
blob88 diversifier;
|
|
diversifier.begin()[0] = 10;
|
|
auto dpa = sk.ToXFVK().Address(diversifier).get().second;
|
|
|
|
// verify wallet only has the default address
|
|
EXPECT_TRUE(wallet.HaveSaplingIncomingViewingKey(sk.DefaultAddress()));
|
|
EXPECT_FALSE(wallet.HaveSaplingIncomingViewingKey(dpa));
|
|
|
|
// manually add a diversified address
|
|
auto ivk = fvk.in_viewing_key();
|
|
EXPECT_TRUE(wallet.AddSaplingIncomingViewingKey(ivk, dpa));
|
|
|
|
// verify wallet did add it
|
|
EXPECT_TRUE(wallet.HaveSaplingIncomingViewingKey(sk.DefaultAddress()));
|
|
EXPECT_TRUE(wallet.HaveSaplingIncomingViewingKey(dpa));
|
|
|
|
// verify that resets nTimeFirstKey, since there is no birthday info for watch-only keys
|
|
EXPECT_EQ(wallet.nTimeFirstKey, 1);
|
|
|
|
// Load a third key into the wallet
|
|
auto sk2 = m.Derive(1);
|
|
ASSERT_TRUE(wallet.LoadSaplingZKey(sk2));
|
|
|
|
// attach metadata to this third key
|
|
auto ivk2 = sk2.expsk.full_viewing_key().in_viewing_key();
|
|
int64_t now = GetTime();
|
|
CKeyMetadata meta(now);
|
|
ASSERT_TRUE(wallet.LoadSaplingZKeyMetadata(ivk2, meta));
|
|
|
|
// check metadata is the same
|
|
ASSERT_EQ(wallet.mapSaplingZKeyMetadata[ivk2].nCreateTime, now);
|
|
|
|
// Load a diversified address for the third key into the wallet
|
|
auto dpa2 = sk2.ToXFVK().Address(diversifier).get().second;
|
|
EXPECT_TRUE(wallet.HaveSaplingIncomingViewingKey(sk2.DefaultAddress()));
|
|
EXPECT_FALSE(wallet.HaveSaplingIncomingViewingKey(dpa2));
|
|
EXPECT_TRUE(wallet.LoadSaplingPaymentAddress(dpa2, ivk2));
|
|
EXPECT_TRUE(wallet.HaveSaplingIncomingViewingKey(dpa2));
|
|
}
|
|
|
|
/**
|
|
* This test covers methods on CWalletDB to load/save crypted sapling z keys.
|
|
*/
|
|
TEST(wallet_zkeys_tests, WriteCryptedSaplingZkeyDirectToDb) {
|
|
SelectParams(CBaseChainParams::TESTNET);
|
|
|
|
// Get temporary and unique path for file.
|
|
// Note: / operator to append paths
|
|
boost::filesystem::path pathTemp = boost::filesystem::temp_directory_path() / boost::filesystem::unique_path();
|
|
boost::filesystem::create_directories(pathTemp);
|
|
mapArgs["-datadir"] = pathTemp.string();
|
|
|
|
bool fFirstRun;
|
|
CWallet wallet("wallet_crypted_sapling.dat");
|
|
ASSERT_EQ(DB_LOAD_OK, wallet.LoadWallet(fFirstRun));
|
|
|
|
// No default CPubKey set
|
|
ASSERT_TRUE(fFirstRun);
|
|
|
|
ASSERT_FALSE(wallet.HaveHDSeed());
|
|
wallet.GenerateNewSeed();
|
|
|
|
// wallet should be empty
|
|
std::set<libzcash::SaplingPaymentAddress> addrs;
|
|
wallet.GetSaplingPaymentAddresses(addrs);
|
|
ASSERT_EQ(0, addrs.size());
|
|
|
|
// Add random key to the wallet
|
|
auto address = wallet.GenerateNewSaplingZKey();
|
|
|
|
// wallet should have one key
|
|
wallet.GetSaplingPaymentAddresses(addrs);
|
|
ASSERT_EQ(1, addrs.size());
|
|
|
|
// Generate a diversified address different to the default
|
|
// If we can't get an early diversified address, we are very unlucky
|
|
libzcash::SaplingExtendedSpendingKey extsk;
|
|
EXPECT_TRUE(wallet.GetSaplingExtendedSpendingKey(address, extsk));
|
|
blob88 diversifier;
|
|
diversifier.begin()[0] = 10;
|
|
auto dpa = extsk.ToXFVK().Address(diversifier).get().second;
|
|
|
|
// Add diversified address to the wallet
|
|
auto ivk = extsk.expsk.full_viewing_key().in_viewing_key();
|
|
EXPECT_TRUE(wallet.AddSaplingIncomingViewingKey(ivk, dpa));
|
|
|
|
// encrypt wallet
|
|
SecureString strWalletPass;
|
|
strWalletPass.reserve(100);
|
|
strWalletPass = "hello";
|
|
ASSERT_TRUE(wallet.EncryptWallet(strWalletPass));
|
|
|
|
// adding a new key will fail as the wallet is locked
|
|
EXPECT_ANY_THROW(wallet.GenerateNewSaplingZKey());
|
|
|
|
// unlock wallet and then add
|
|
wallet.Unlock(strWalletPass);
|
|
auto address2 = wallet.GenerateNewSaplingZKey();
|
|
|
|
// Create a new wallet from the existing wallet path
|
|
CWallet wallet2("wallet_crypted_sapling.dat");
|
|
ASSERT_EQ(DB_LOAD_OK, wallet2.LoadWallet(fFirstRun));
|
|
|
|
// Confirm it's not the same as the other wallet
|
|
ASSERT_TRUE(&wallet != &wallet2);
|
|
ASSERT_TRUE(wallet2.HaveHDSeed());
|
|
|
|
// wallet should have three addresses
|
|
wallet2.GetSaplingPaymentAddresses(addrs);
|
|
ASSERT_EQ(3, addrs.size());
|
|
|
|
//check we have entries for our payment addresses
|
|
ASSERT_TRUE(addrs.count(address));
|
|
ASSERT_TRUE(addrs.count(address2));
|
|
ASSERT_TRUE(addrs.count(dpa));
|
|
|
|
// spending key is crypted, so we can't extract valid payment address
|
|
libzcash::SaplingExtendedSpendingKey keyOut;
|
|
EXPECT_FALSE(wallet2.GetSaplingExtendedSpendingKey(address, keyOut));
|
|
ASSERT_FALSE(address == keyOut.DefaultAddress());
|
|
|
|
// address -> ivk mapping is not crypted
|
|
libzcash::SaplingIncomingViewingKey ivkOut;
|
|
EXPECT_TRUE(wallet2.GetSaplingIncomingViewingKey(dpa, ivkOut));
|
|
EXPECT_EQ(ivk, ivkOut);
|
|
|
|
// unlock wallet to get spending keys and verify payment addresses
|
|
wallet2.Unlock(strWalletPass);
|
|
|
|
EXPECT_TRUE(wallet2.GetSaplingExtendedSpendingKey(address, keyOut));
|
|
ASSERT_EQ(address, keyOut.DefaultAddress());
|
|
|
|
EXPECT_TRUE(wallet2.GetSaplingExtendedSpendingKey(address2, keyOut));
|
|
ASSERT_EQ(address2, keyOut.DefaultAddress());
|
|
}
|
|
|