sapling_key_tests.cpp

Go to the documentation of this file.

// Copyright (c) 2016-2020 The ZCash developers
// Copyright (c) 2020 The PIVX Core developers
// Distributed under the MIT software license, see the accompanying
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
 
#include "test/test_pivx.h"
 
#include "sapling/address.h"
#include "sapling/key_io_sapling.h"
 
#include <string>
#include <vector>
 
#include <boost/test/unit_test.hpp>
 
 
BOOST_FIXTURE_TEST_SUITE(sapling_key_tests, BasicTestingSetup)
 
BOOST_AUTO_TEST_CASE(ps_address_test)
{
    SelectParams(CBaseChainParams::REGTEST);
 
    std::vector<unsigned char, secure_allocator<unsigned char>> rawSeed(32);
    HDSeed seed(rawSeed);
    auto m = libzcash::SaplingExtendedSpendingKey::Master(seed);
 
    for (uint32_t i = 0; i < 1000; i++) {
        auto sk = m.Derive(i);
        {
            std::string sk_string = KeyIO::EncodeSpendingKey(sk);
            BOOST_CHECK(sk_string.compare(0, 26, Params().Bech32HRP(CChainParams::SAPLING_EXTENDED_SPEND_KEY)) == 0);
 
            libzcash::SpendingKey spendingkey2 = KeyIO::DecodeSpendingKey(sk_string);
            BOOST_CHECK(IsValidSpendingKey(spendingkey2));
 
            BOOST_ASSERT(boost::get<libzcash::SaplingExtendedSpendingKey>(&spendingkey2) != nullptr);
            auto sk2 = boost::get<libzcash::SaplingExtendedSpendingKey>(spendingkey2);
            BOOST_CHECK(sk == sk2);
        }
        {
            libzcash::SaplingPaymentAddress addr = sk.DefaultAddress();
 
            std::string addr_string = KeyIO::EncodePaymentAddress(addr);
            BOOST_CHECK(addr_string.compare(0, 12, Params().Bech32HRP(CChainParams::SAPLING_PAYMENT_ADDRESS)) == 0);
 
            auto paymentaddr2 = KeyIO::DecodePaymentAddress(addr_string);
            BOOST_CHECK(IsValidPaymentAddress(paymentaddr2));
 
            BOOST_ASSERT(boost::get<libzcash::SaplingPaymentAddress>(&paymentaddr2) != nullptr);
            auto addr2 = boost::get<libzcash::SaplingPaymentAddress>(paymentaddr2);
            BOOST_CHECK(addr == addr2);
        }
    }
}
 
BOOST_AUTO_TEST_CASE(EncodeAndDecodeSapling)
{
    SelectParams(CBaseChainParams::MAIN);
 
    std::vector<unsigned char, secure_allocator<unsigned char>> rawSeed(32);
    HDSeed seed(rawSeed);
    libzcash::SaplingExtendedSpendingKey m = libzcash::SaplingExtendedSpendingKey::Master(seed);
 
    for (uint32_t i = 0; i < 1000; i++) {
        auto sk = m.Derive(i);
        {
            std::string sk_string = KeyIO::EncodeSpendingKey(sk);
            BOOST_CHECK(
                    sk_string.substr(0, 26) ==
                    Params().Bech32HRP(CChainParams::SAPLING_EXTENDED_SPEND_KEY));
 
            auto spendingkey2 = KeyIO::DecodeSpendingKey(sk_string);
            BOOST_CHECK(IsValidSpendingKey(spendingkey2));
 
            BOOST_CHECK(boost::get<libzcash::SaplingExtendedSpendingKey>(&spendingkey2) != nullptr);
            auto sk2 = boost::get<libzcash::SaplingExtendedSpendingKey>(spendingkey2);
            BOOST_CHECK(sk == sk2);
        }
 
        {
            auto extfvk = sk.ToXFVK();
            std::string vk_string = KeyIO::EncodeViewingKey(extfvk);
            BOOST_CHECK(
                vk_string.substr(0, 7) ==
                Params().Bech32HRP(CChainParams::SAPLING_EXTENDED_FVK));
 
            auto viewingkey2 = KeyIO::DecodeViewingKey(vk_string);
            BOOST_CHECK(IsValidViewingKey(viewingkey2));
 
            BOOST_CHECK(boost::get<libzcash::SaplingExtendedFullViewingKey>(&viewingkey2) != nullptr);
            auto extfvk2 = boost::get<libzcash::SaplingExtendedFullViewingKey>(viewingkey2);
            BOOST_CHECK(extfvk == extfvk2);
        }
 
        {
            auto addr = sk.DefaultAddress();
 
            std::string addr_string = KeyIO::EncodePaymentAddress(addr);
            BOOST_CHECK(
                    addr_string.substr(0, 2) ==
                    Params().Bech32HRP(CChainParams::SAPLING_PAYMENT_ADDRESS));
 
            auto paymentaddr2 = KeyIO::DecodePaymentAddress(addr_string);
            BOOST_CHECK(IsValidPaymentAddress(paymentaddr2));
 
            BOOST_CHECK(boost::get<libzcash::SaplingPaymentAddress>(&paymentaddr2) != nullptr);
            auto addr2 = boost::get<libzcash::SaplingPaymentAddress>(paymentaddr2);
            BOOST_CHECK(addr == addr2);
        }
    }
}
 
BOOST_AUTO_TEST_SUITE_END()