Move libzerocash (dd5db5815be70f0e4895784cc905df6f1c73cb17) into the src tree.
This commit is contained in:
Sean Bowe
634
src/zerocash/tests/zerocashTest.cpp
Normal file
634
src/zerocash/tests/zerocashTest.cpp
Normal file
@@ -0,0 +1,634 @@
|
||||
/** @file
|
||||
*****************************************************************************
|
||||
|
||||
A test for Zerocash.
|
||||
|
||||
*****************************************************************************
|
||||
* @author This file is part of libzerocash, developed by the Zerocash
|
||||
* project and contributors (see AUTHORS).
|
||||
* @copyright MIT license (see LICENSE file)
|
||||
*****************************************************************************/
|
||||
|
||||
#include <stdlib.h>
|
||||
#include <iostream>
|
||||
|
||||
#define BOOST_TEST_MODULE zerocashTest
|
||||
#include <boost/test/included/unit_test.hpp>
|
||||
|
||||
#include "timer.h"
|
||||
|
||||
#include "libzerocash/Zerocash.h"
|
||||
#include "libzerocash/ZerocashParams.h"
|
||||
#include "libzerocash/Address.h"
|
||||
#include "libzerocash/CoinCommitment.h"
|
||||
#include "libzerocash/Coin.h"
|
||||
#include "libzerocash/IncrementalMerkleTree.h"
|
||||
#include "libzerocash/MintTransaction.h"
|
||||
#include "libzerocash/PourTransaction.h"
|
||||
#include "libzerocash/PourInput.h"
|
||||
#include "libzerocash/PourOutput.h"
|
||||
#include "libzerocash/utils/util.h"
|
||||
|
||||
using namespace std;
|
||||
using namespace libsnark;
|
||||
|
||||
#define TEST_TREE_DEPTH 4
|
||||
|
||||
BOOST_AUTO_TEST_CASE( SaveAndLoadKeysFromFiles ) {
|
||||
cout << "\nSaveAndLoadKeysFromFiles TEST\n" << endl;
|
||||
|
||||
cout << "Creating Params...\n" << endl;
|
||||
|
||||
libzerocash::timer_start("Param Generation");
|
||||
auto keypair = libzerocash::ZerocashParams::GenerateNewKeyPair(TEST_TREE_DEPTH);
|
||||
libzerocash::ZerocashParams p(
|
||||
TEST_TREE_DEPTH,
|
||||
&keypair
|
||||
);
|
||||
libzerocash::timer_stop("Param Generation");
|
||||
print_mem("after param generation");
|
||||
|
||||
cout << "Successfully created Params.\n" << endl;
|
||||
|
||||
std::string vk_path = "./zerocashTest-verification-key";
|
||||
std::string pk_path = "./zerocashTest-proving-key";
|
||||
|
||||
libzerocash::timer_start("Saving Proving Key");
|
||||
|
||||
libzerocash::ZerocashParams::SaveProvingKeyToFile(
|
||||
&p.getProvingKey(),
|
||||
pk_path
|
||||
);
|
||||
|
||||
libzerocash::timer_stop("Saving Proving Key");
|
||||
|
||||
libzerocash::timer_start("Saving Verification Key");
|
||||
|
||||
libzerocash::ZerocashParams::SaveVerificationKeyToFile(
|
||||
&p.getVerificationKey(),
|
||||
vk_path
|
||||
);
|
||||
|
||||
libzerocash::timer_stop("Saving Verification Key");
|
||||
|
||||
libzerocash::timer_start("Loading Proving Key");
|
||||
auto pk_loaded = libzerocash::ZerocashParams::LoadProvingKeyFromFile(pk_path, TEST_TREE_DEPTH);
|
||||
libzerocash::timer_stop("Loading Proving Key");
|
||||
|
||||
libzerocash::timer_start("Loading Verification Key");
|
||||
auto vk_loaded = libzerocash::ZerocashParams::LoadVerificationKeyFromFile(vk_path, TEST_TREE_DEPTH);
|
||||
libzerocash::timer_stop("Loading Verification Key");
|
||||
|
||||
cout << "Comparing Proving and Verification key.\n" << endl;
|
||||
|
||||
if ( !( p.getProvingKey() == pk_loaded && p.getVerificationKey() == vk_loaded) ) {
|
||||
BOOST_ERROR("Proving and verification key are not equal.");
|
||||
}
|
||||
|
||||
vector<libzerocash::Coin> coins;
|
||||
vector<libzerocash::Address> addrs;
|
||||
|
||||
cout << "Creating Addresses and Coins...\n" << endl;
|
||||
for(size_t i = 0; i < 5; i++) {
|
||||
addrs.push_back(libzerocash::Address::CreateNewRandomAddress());
|
||||
coins.push_back(libzerocash::Coin(addrs.at(i).getPublicAddress(), i));
|
||||
}
|
||||
cout << "Successfully created address and coins.\n" << endl;
|
||||
|
||||
cout << "Creating a Mint Transaction...\n" << endl;
|
||||
libzerocash::MintTransaction minttx(coins.at(0));
|
||||
cout << "Successfully created a Mint Transaction.\n" << endl;
|
||||
|
||||
vector<std::vector<bool>> coinValues(5);
|
||||
vector<bool> temp_comVal(ZC_CM_SIZE * 8);
|
||||
for(size_t i = 0; i < coinValues.size(); i++) {
|
||||
libzerocash::convertBytesVectorToVector(coins.at(i).getCoinCommitment().getCommitmentValue(), temp_comVal);
|
||||
coinValues.at(i) = temp_comVal;
|
||||
}
|
||||
|
||||
cout << "Creating Merkle Tree...\n" << endl;
|
||||
libzerocash::IncrementalMerkleTree merkleTree(coinValues, TEST_TREE_DEPTH);
|
||||
cout << "Successfully created Merkle Tree.\n" << endl;
|
||||
|
||||
std::vector<bool> index;
|
||||
|
||||
cout << "Creating Witness 1...\n" << endl;
|
||||
merkle_authentication_path witness_1(TEST_TREE_DEPTH);
|
||||
libzerocash::convertIntToVector(1, index);
|
||||
merkleTree.getWitness(index, witness_1);
|
||||
cout << "Successfully created Witness 1.\n" << endl;
|
||||
|
||||
cout << "Creating Witness 2...\n" << endl;
|
||||
merkle_authentication_path witness_2(TEST_TREE_DEPTH);
|
||||
libzerocash::convertIntToVector(3, index);
|
||||
merkleTree.getWitness(index, witness_2);
|
||||
cout << "Successfully created Witness 2.\n" << endl;
|
||||
|
||||
cout << "Creating coins to spend...\n" << endl;
|
||||
libzerocash::Address newAddress3 = libzerocash::Address::CreateNewRandomAddress();
|
||||
libzerocash::PublicAddress pubAddress3 = newAddress3.getPublicAddress();
|
||||
|
||||
libzerocash::Address newAddress4 = libzerocash::Address::CreateNewRandomAddress();
|
||||
libzerocash::PublicAddress pubAddress4 = newAddress4.getPublicAddress();
|
||||
|
||||
libzerocash::Coin c_1_new(pubAddress3, 2);
|
||||
libzerocash::Coin c_2_new(pubAddress4, 2);
|
||||
cout << "Successfully created coins to spend.\n" << endl;
|
||||
|
||||
vector<unsigned char> rt(ZC_ROOT_SIZE);
|
||||
merkleTree.getRootValue(rt);
|
||||
|
||||
// XXX: debugging
|
||||
std::cout << "Root: " << rt.size() << endl;
|
||||
std::cout << "wit1: " << witness_1.size() << endl;
|
||||
std::cout << "wit2: " << witness_1.size() << endl;
|
||||
|
||||
vector<unsigned char> as(ZC_SIG_PK_SIZE, 'a');
|
||||
|
||||
cout << "Creating a pour transaction...\n" << endl;
|
||||
libzerocash::PourTransaction pourtx(1, p,
|
||||
rt,
|
||||
coins.at(1), coins.at(3),
|
||||
addrs.at(1), addrs.at(3),
|
||||
1, 3,
|
||||
witness_1, witness_2,
|
||||
pubAddress3, pubAddress4,
|
||||
0,
|
||||
0,
|
||||
as,
|
||||
c_1_new, c_2_new);
|
||||
cout << "Successfully created a pour transaction.\n" << endl;
|
||||
|
||||
std::vector<unsigned char> pubkeyHash(ZC_SIG_PK_SIZE, 'a');
|
||||
|
||||
cout << "Verifying a pour transaction...\n" << endl;
|
||||
bool pourtx_res = pourtx.verify(p, pubkeyHash, rt);
|
||||
|
||||
BOOST_CHECK(pourtx_res);
|
||||
}
|
||||
|
||||
BOOST_AUTO_TEST_CASE( PourInputOutputTest ) {
|
||||
// dummy input
|
||||
{
|
||||
libzerocash::PourInput input(TEST_TREE_DEPTH);
|
||||
|
||||
BOOST_CHECK(input.old_coin.getValue() == 0);
|
||||
BOOST_CHECK(input.old_address.getPublicAddress() == input.old_coin.getPublicAddress());
|
||||
}
|
||||
|
||||
// dummy output
|
||||
{
|
||||
libzerocash::PourOutput output(0);
|
||||
|
||||
BOOST_CHECK(output.new_coin.getValue() == 0);
|
||||
BOOST_CHECK(output.to_address == output.new_coin.getPublicAddress());
|
||||
}
|
||||
}
|
||||
|
||||
// testing with general situational setup
|
||||
bool test_pour(libzerocash::ZerocashParams& p,
|
||||
uint64_t vpub_in,
|
||||
uint64_t vpub_out,
|
||||
std::vector<uint64_t> inputs, // values of the inputs (max 2)
|
||||
std::vector<uint64_t> outputs) // values of the outputs (max 2)
|
||||
{
|
||||
using pour_input_state = std::tuple<libzerocash::Address, libzerocash::Coin, std::vector<bool>>;
|
||||
|
||||
// Construct incremental merkle tree
|
||||
libzerocash::IncrementalMerkleTree merkleTree(TEST_TREE_DEPTH);
|
||||
|
||||
// Dummy sig_pk
|
||||
vector<unsigned char> as(ZC_SIG_PK_SIZE, 'a');
|
||||
|
||||
vector<libzerocash::PourInput> pour_inputs;
|
||||
vector<libzerocash::PourOutput> pour_outputs;
|
||||
|
||||
vector<pour_input_state> input_state;
|
||||
|
||||
for(std::vector<uint64_t>::iterator it = inputs.begin(); it != inputs.end(); ++it) {
|
||||
libzerocash::Address addr = libzerocash::Address::CreateNewRandomAddress();
|
||||
libzerocash::Coin coin(addr.getPublicAddress(), *it);
|
||||
|
||||
// commitment from coin
|
||||
std::vector<bool> commitment(ZC_CM_SIZE * 8);
|
||||
libzerocash::convertBytesVectorToVector(coin.getCoinCommitment().getCommitmentValue(), commitment);
|
||||
|
||||
// insert commitment into the merkle tree
|
||||
std::vector<bool> index;
|
||||
merkleTree.insertElement(commitment, index);
|
||||
|
||||
// store the state temporarily
|
||||
input_state.push_back(std::make_tuple(addr, coin, index));
|
||||
}
|
||||
|
||||
// compute the merkle root we will be working with
|
||||
vector<unsigned char> rt(ZC_ROOT_SIZE);
|
||||
{
|
||||
vector<bool> root_bv(ZC_ROOT_SIZE * 8);
|
||||
merkleTree.getRootValue(root_bv);
|
||||
libzerocash::convertVectorToBytesVector(root_bv, rt);
|
||||
}
|
||||
|
||||
// get witnesses for all the input coins and construct the pours
|
||||
for(vector<pour_input_state>::iterator it = input_state.begin(); it != input_state.end(); ++it) {
|
||||
merkle_authentication_path path(TEST_TREE_DEPTH);
|
||||
|
||||
auto index = std::get<2>(*it);
|
||||
merkleTree.getWitness(index, path);
|
||||
|
||||
pour_inputs.push_back(libzerocash::PourInput(std::get<1>(*it), std::get<0>(*it), libzerocash::convertVectorToInt(index), path));
|
||||
}
|
||||
|
||||
// construct dummy outputs with the given values
|
||||
for(vector<uint64_t>::iterator it = outputs.begin(); it != outputs.end(); ++it) {
|
||||
pour_outputs.push_back(libzerocash::PourOutput(*it));
|
||||
}
|
||||
|
||||
libzerocash::PourTransaction pourtx(p, as, rt, pour_inputs, pour_outputs, vpub_in, vpub_out);
|
||||
|
||||
BOOST_CHECK(pourtx.verify(p, as, rt));
|
||||
}
|
||||
|
||||
BOOST_AUTO_TEST_CASE( PourVpubInTest ) {
|
||||
auto keypair = libzerocash::ZerocashParams::GenerateNewKeyPair(TEST_TREE_DEPTH);
|
||||
libzerocash::ZerocashParams p(
|
||||
TEST_TREE_DEPTH,
|
||||
&keypair
|
||||
);
|
||||
|
||||
// Things that should work..
|
||||
BOOST_CHECK(test_pour(p, 0, 0, {1}, {1}));
|
||||
BOOST_CHECK(test_pour(p, 0, 0, {2}, {1, 1}));
|
||||
BOOST_CHECK(test_pour(p, 0, 0, {2, 2}, {3, 1}));
|
||||
BOOST_CHECK(test_pour(p, 0, 1, {1}, {}));
|
||||
BOOST_CHECK(test_pour(p, 0, 1, {2}, {1}));
|
||||
BOOST_CHECK(test_pour(p, 0, 1, {2, 2}, {2, 1}));
|
||||
BOOST_CHECK(test_pour(p, 1, 0, {}, {1}));
|
||||
BOOST_CHECK(test_pour(p, 1, 0, {1}, {1, 1}));
|
||||
BOOST_CHECK(test_pour(p, 1, 0, {2, 2}, {2, 3}));
|
||||
|
||||
// Things that should not work...
|
||||
BOOST_CHECK_THROW(test_pour(p, 0, 1, {1}, {1}), std::invalid_argument);
|
||||
BOOST_CHECK_THROW(test_pour(p, 0, 1, {2}, {1, 1}), std::invalid_argument);
|
||||
BOOST_CHECK_THROW(test_pour(p, 0, 1, {2, 2}, {3, 1}), std::invalid_argument);
|
||||
BOOST_CHECK_THROW(test_pour(p, 0, 2, {1}, {}), std::invalid_argument);
|
||||
BOOST_CHECK_THROW(test_pour(p, 0, 2, {2}, {1}), std::invalid_argument);
|
||||
BOOST_CHECK_THROW(test_pour(p, 0, 2, {2, 2}, {2, 1}), std::invalid_argument);
|
||||
BOOST_CHECK_THROW(test_pour(p, 1, 1, {}, {1}), std::invalid_argument);
|
||||
BOOST_CHECK_THROW(test_pour(p, 1, 1, {1}, {1, 1}), std::invalid_argument);
|
||||
BOOST_CHECK_THROW(test_pour(p, 1, 1, {2, 2}, {2, 3}), std::invalid_argument);
|
||||
|
||||
BOOST_CHECK_THROW(test_pour(p, 0, 0, {2, 2}, {2, 3}), std::invalid_argument);
|
||||
}
|
||||
|
||||
BOOST_AUTO_TEST_CASE( CoinTest ) {
|
||||
cout << "\nCOIN TEST\n" << endl;
|
||||
|
||||
libzerocash::Address newAddress = libzerocash::Address::CreateNewRandomAddress();
|
||||
libzerocash::PublicAddress pubAddress = newAddress.getPublicAddress();
|
||||
|
||||
libzerocash::Coin coin(pubAddress, 0);
|
||||
|
||||
cout << "Successfully created a coin.\n" << endl;
|
||||
|
||||
///////////////////////////////////////////////////////////////////////////
|
||||
|
||||
libzerocash::timer_start("Coin");
|
||||
libzerocash::Coin coin2(pubAddress, 0);
|
||||
libzerocash::timer_stop("Coin");
|
||||
|
||||
cout << "Successfully created a coin.\n" << endl;
|
||||
}
|
||||
|
||||
BOOST_AUTO_TEST_CASE( MintTxTest ) {
|
||||
cout << "\nMINT TRANSACTION TEST\n" << endl;
|
||||
|
||||
libzerocash::Address newAddress = libzerocash::Address::CreateNewRandomAddress();
|
||||
libzerocash::PublicAddress pubAddress = newAddress.getPublicAddress();
|
||||
|
||||
vector<unsigned char> value(ZC_V_SIZE, 0);
|
||||
|
||||
libzerocash::timer_start("Coin");
|
||||
const libzerocash::Coin coin(pubAddress, 0);
|
||||
libzerocash::timer_stop("Coin");
|
||||
|
||||
libzerocash::timer_start("Mint Transaction");
|
||||
libzerocash::MintTransaction minttx(coin);
|
||||
libzerocash::timer_stop("Mint Transaction");
|
||||
|
||||
cout << "Successfully created a mint transaction.\n" << endl;
|
||||
|
||||
libzerocash::timer_start("Mint Transaction Verify");
|
||||
bool minttx_res = minttx.verify();
|
||||
libzerocash::timer_stop("Mint Transaction Verify");
|
||||
|
||||
BOOST_CHECK(minttx_res);
|
||||
}
|
||||
|
||||
BOOST_AUTO_TEST_CASE( PourTxTest ) {
|
||||
cout << "\nPOUR TRANSACTION TEST\n" << endl;
|
||||
|
||||
cout << "Creating Params...\n" << endl;
|
||||
|
||||
libzerocash::timer_start("Param Generation");
|
||||
auto keypair = libzerocash::ZerocashParams::GenerateNewKeyPair(TEST_TREE_DEPTH);
|
||||
libzerocash::ZerocashParams p(
|
||||
TEST_TREE_DEPTH,
|
||||
&keypair
|
||||
);
|
||||
libzerocash::timer_stop("Param Generation");
|
||||
print_mem("after param generation");
|
||||
|
||||
cout << "Successfully created Params.\n" << endl;
|
||||
|
||||
vector<libzerocash::Coin> coins;
|
||||
vector<libzerocash::Address> addrs;
|
||||
|
||||
for(size_t i = 0; i < 5; i++) {
|
||||
addrs.push_back(libzerocash::Address::CreateNewRandomAddress());
|
||||
coins.push_back(libzerocash::Coin(addrs.at(i).getPublicAddress(), i));
|
||||
}
|
||||
|
||||
cout << "Successfully created coins.\n" << endl;
|
||||
|
||||
vector<std::vector<bool>> coinValues(5);
|
||||
|
||||
vector<bool> temp_comVal(ZC_CM_SIZE * 8);
|
||||
for(size_t i = 0; i < coinValues.size(); i++) {
|
||||
libzerocash::convertBytesVectorToVector(coins.at(i).getCoinCommitment().getCommitmentValue(), temp_comVal);
|
||||
coinValues.at(i) = temp_comVal;
|
||||
libzerocash::printVectorAsHex("Coin => ", coinValues.at(i));
|
||||
}
|
||||
|
||||
cout << "Creating Merkle Tree...\n" << endl;
|
||||
|
||||
libzerocash::timer_start("Merkle Tree");
|
||||
libzerocash::IncrementalMerkleTree merkleTree(coinValues, TEST_TREE_DEPTH);
|
||||
libzerocash::timer_stop("Merkle Tree");
|
||||
|
||||
cout << "Successfully created Merkle Tree.\n" << endl;
|
||||
|
||||
merkle_authentication_path witness_1(TEST_TREE_DEPTH);
|
||||
|
||||
libzerocash::timer_start("Witness");
|
||||
std::vector<bool> index;
|
||||
libzerocash::convertIntToVector(1, index);
|
||||
if (merkleTree.getWitness(index, witness_1) == false) {
|
||||
BOOST_ERROR("Could not get witness");
|
||||
}
|
||||
libzerocash::timer_stop("Witness");
|
||||
|
||||
cout << "Witness 1: " << endl;
|
||||
for(size_t i = 0; i < witness_1.size(); i++) {
|
||||
libzerocash::printVectorAsHex(witness_1.at(i));
|
||||
}
|
||||
cout << "\n" << endl;
|
||||
|
||||
merkle_authentication_path witness_2(TEST_TREE_DEPTH);
|
||||
libzerocash::convertIntToVector(3, index);
|
||||
if (merkleTree.getWitness(index, witness_2) == false) {
|
||||
cout << "Could not get witness" << endl;
|
||||
}
|
||||
|
||||
cout << "Witness 2: " << endl;
|
||||
for(size_t i = 0; i < witness_2.size(); i++) {
|
||||
libzerocash::printVectorAsHex(witness_2.at(i));
|
||||
}
|
||||
cout << "\n" << endl;
|
||||
|
||||
libzerocash::Address newAddress3 = libzerocash::Address::CreateNewRandomAddress();
|
||||
libzerocash::PublicAddress pubAddress3 = newAddress3.getPublicAddress();
|
||||
|
||||
libzerocash::Address newAddress4 = libzerocash::Address::CreateNewRandomAddress();
|
||||
libzerocash::PublicAddress pubAddress4 = newAddress4.getPublicAddress();
|
||||
|
||||
libzerocash::Coin c_1_new(pubAddress3, 2);
|
||||
libzerocash::Coin c_2_new(pubAddress4, 2);
|
||||
|
||||
vector<bool> root_bv(ZC_ROOT_SIZE * 8);
|
||||
merkleTree.getRootValue(root_bv);
|
||||
vector<unsigned char> rt(ZC_ROOT_SIZE);
|
||||
libzerocash::convertVectorToBytesVector(root_bv, rt);
|
||||
|
||||
vector<unsigned char> ones(ZC_V_SIZE, 1);
|
||||
vector<unsigned char> twos(ZC_V_SIZE, 2);
|
||||
vector<unsigned char> as(ZC_SIG_PK_SIZE, 'a');
|
||||
|
||||
cout << "Creating a pour transaction...\n" << endl;
|
||||
|
||||
libzerocash::timer_start("Pour Transaction");
|
||||
libzerocash::PourTransaction pourtx(1, p, rt, coins.at(1), coins.at(3), addrs.at(1), addrs.at(3), 1, 3, witness_1, witness_2, pubAddress3, pubAddress4, 0, 0, as, c_1_new, c_2_new);
|
||||
libzerocash::timer_stop("Pour Transaction");
|
||||
print_mem("after pour transaction");
|
||||
|
||||
cout << "Successfully created a pour transaction.\n" << endl;
|
||||
|
||||
std::vector<unsigned char> pubkeyHash(ZC_SIG_PK_SIZE, 'a');
|
||||
|
||||
libzerocash::timer_start("Pour Transaction Verify");
|
||||
bool pourtx_res = pourtx.verify(p, pubkeyHash, rt);
|
||||
libzerocash::timer_stop("Pour Transaction Verify");
|
||||
|
||||
BOOST_CHECK(pourtx_res);
|
||||
}
|
||||
|
||||
BOOST_AUTO_TEST_CASE( MerkleTreeSimpleTest ) {
|
||||
cout << "\nMERKLE TREE SIMPLE TEST\n" << endl;
|
||||
|
||||
vector<libzerocash::Coin> coins;
|
||||
vector<libzerocash::Address> addrs;
|
||||
|
||||
cout << "Creating coins...\n" << endl;
|
||||
|
||||
for(size_t i = 0; i < 5; i++) {
|
||||
addrs.push_back(libzerocash::Address::CreateNewRandomAddress());
|
||||
coins.push_back(libzerocash::Coin(addrs.at(i).getPublicAddress(), i));
|
||||
}
|
||||
|
||||
cout << "Successfully created coins.\n" << endl;
|
||||
|
||||
vector<std::vector<bool>> coinValues(coins.size());
|
||||
|
||||
vector<bool> temp_comVal(ZC_CM_SIZE * 8);
|
||||
for(size_t i = 0; i < coinValues.size(); i++) {
|
||||
libzerocash::convertBytesVectorToVector(coins.at(i).getCoinCommitment().getCommitmentValue(), temp_comVal);
|
||||
coinValues.at(i) = temp_comVal;
|
||||
libzerocash::printVectorAsHex(coinValues.at(i));
|
||||
}
|
||||
|
||||
cout << "Creating Merkle Tree...\n" << endl;
|
||||
|
||||
libzerocash::IncrementalMerkleTree merkleTree(64);
|
||||
vector<bool> root;
|
||||
merkleTree.getRootValue(root);
|
||||
cout << "Root: ";
|
||||
libzerocash::printVectorAsHex(root);
|
||||
cout << endl;
|
||||
|
||||
cout << "Successfully created Merkle Tree.\n" << endl;
|
||||
|
||||
cout << "Copying and pruning Merkle Tree...\n" << endl;
|
||||
libzerocash::IncrementalMerkleTree copyTree = merkleTree;
|
||||
copyTree.prune();
|
||||
|
||||
cout << "Obtaining compact representation and reconstituting tree...\n" << endl;
|
||||
libzerocash::IncrementalMerkleTreeCompact compactTree = merkleTree.getCompactRepresentation();
|
||||
|
||||
cout << "Compact representation vector: ";
|
||||
libzerocash::printVector(compactTree.getHashList());
|
||||
|
||||
libzerocash::IncrementalMerkleTree reconstitutedTree(compactTree);
|
||||
reconstitutedTree.getRootValue(root);
|
||||
cout << "New root: ";
|
||||
libzerocash::printVectorAsHex(root);
|
||||
cout << endl;
|
||||
|
||||
reconstitutedTree.insertVector(coinValues);
|
||||
merkleTree.insertVector(coinValues);
|
||||
|
||||
reconstitutedTree.getRootValue(root);
|
||||
cout << "New root (added a bunch more): ";
|
||||
libzerocash::printVectorAsHex(root);
|
||||
cout << endl;
|
||||
|
||||
merkleTree.getRootValue(root);
|
||||
cout << "Old root (added a bunch more): ";
|
||||
libzerocash::printVectorAsHex(root);
|
||||
cout << endl;
|
||||
|
||||
merkle_authentication_path witness(16);
|
||||
std::vector<bool> index;
|
||||
libzerocash::convertIntToVector(3, index);
|
||||
if (merkleTree.getWitness(index, witness) == false) {
|
||||
BOOST_ERROR("Witness generation failed.");
|
||||
}
|
||||
|
||||
cout << "Successfully created witness.\n" << endl;
|
||||
|
||||
cout << "Witness: " << endl;
|
||||
for(size_t i = 0; i < witness.size(); i++) {
|
||||
libzerocash::printVectorAsHex(witness.at(i));
|
||||
}
|
||||
cout << "\n" << endl;
|
||||
|
||||
vector<bool> wit1(SHA256_BLOCK_SIZE * 8);
|
||||
vector<bool> wit2(SHA256_BLOCK_SIZE * 8);
|
||||
vector<bool> wit3(SHA256_BLOCK_SIZE * 8);
|
||||
vector<bool> inter_1(SHA256_BLOCK_SIZE * 8);
|
||||
vector<bool> inter_2(SHA256_BLOCK_SIZE * 8);
|
||||
std::vector<bool> zeros(SHA256_BLOCK_SIZE * 8, 0);
|
||||
|
||||
wit1 = coinValues.at(2);
|
||||
libzerocash::hashVectors(coinValues.at(0), coinValues.at(1), wit2);
|
||||
libzerocash::hashVectors(coinValues.at(4), zeros, inter_1);
|
||||
inter_2 = zeros;
|
||||
libzerocash::hashVectors(inter_1, inter_2, wit3);
|
||||
|
||||
BOOST_CHECK(witness.size() == 64);
|
||||
for (size_t i = 0; i < 61; i++) {
|
||||
BOOST_CHECK(witness.at(i) == zeros);
|
||||
}
|
||||
BOOST_CHECK(
|
||||
(witness.at(61) == wit3) &&
|
||||
(witness.at(62) == wit2) &&
|
||||
(witness.at(63) == wit1)
|
||||
);
|
||||
}
|
||||
|
||||
BOOST_AUTO_TEST_CASE( SimpleTxTest ) {
|
||||
cout << "\nSIMPLE TRANSACTION TEST\n" << endl;
|
||||
|
||||
libzerocash::timer_start("Param Generation");
|
||||
auto keypair = libzerocash::ZerocashParams::GenerateNewKeyPair(TEST_TREE_DEPTH);
|
||||
libzerocash::ZerocashParams p(
|
||||
TEST_TREE_DEPTH,
|
||||
&keypair
|
||||
);
|
||||
libzerocash::timer_stop("Param Generation");
|
||||
|
||||
vector<libzerocash::Coin> coins;
|
||||
vector<libzerocash::Address> addrs;
|
||||
|
||||
cout << "Creating Addresses and Coins...\n" << endl;
|
||||
for(size_t i = 0; i < 5; i++) {
|
||||
addrs.push_back(libzerocash::Address::CreateNewRandomAddress());
|
||||
coins.push_back(libzerocash::Coin(addrs.at(i).getPublicAddress(), i));
|
||||
}
|
||||
cout << "Successfully created address and coins.\n" << endl;
|
||||
|
||||
cout << "Creating a Mint Transaction...\n" << endl;
|
||||
libzerocash::MintTransaction minttx(coins.at(0));
|
||||
cout << "Successfully created a Mint Transaction.\n" << endl;
|
||||
|
||||
cout << "Verifying a Mint Transaction...\n" << endl;
|
||||
bool minttx_res = minttx.verify();
|
||||
|
||||
vector<std::vector<bool>> coinValues(5);
|
||||
vector<bool> temp_comVal(ZC_CM_SIZE * 8);
|
||||
for(size_t i = 0; i < coinValues.size(); i++) {
|
||||
libzerocash::convertBytesVectorToVector(coins.at(i).getCoinCommitment().getCommitmentValue(), temp_comVal);
|
||||
coinValues.at(i) = temp_comVal;
|
||||
}
|
||||
|
||||
cout << "Creating Merkle Tree...\n" << endl;
|
||||
libzerocash::IncrementalMerkleTree merkleTree(coinValues, TEST_TREE_DEPTH);
|
||||
cout << "Successfully created Merkle Tree.\n" << endl;
|
||||
|
||||
std::vector<bool> index;
|
||||
|
||||
cout << "Creating Witness 1...\n" << endl;
|
||||
merkle_authentication_path witness_1(TEST_TREE_DEPTH);
|
||||
libzerocash::convertIntToVector(1, index);
|
||||
if (merkleTree.getWitness(index, witness_1) == false) {
|
||||
BOOST_ERROR("Could not get witness");
|
||||
}
|
||||
cout << "Successfully created Witness 1.\n" << endl;
|
||||
|
||||
cout << "Creating Witness 2...\n" << endl;
|
||||
merkle_authentication_path witness_2(TEST_TREE_DEPTH);
|
||||
libzerocash::convertIntToVector(3, index);
|
||||
if (merkleTree.getWitness(index, witness_2) == false) {
|
||||
cout << "Could not get witness" << endl;
|
||||
}
|
||||
cout << "Successfully created Witness 2.\n" << endl;
|
||||
|
||||
cout << "Creating coins to spend...\n" << endl;
|
||||
libzerocash::Address newAddress3 = libzerocash::Address::CreateNewRandomAddress();
|
||||
libzerocash::PublicAddress pubAddress3 = newAddress3.getPublicAddress();
|
||||
|
||||
libzerocash::Address newAddress4 = libzerocash::Address::CreateNewRandomAddress();
|
||||
libzerocash::PublicAddress pubAddress4 = newAddress4.getPublicAddress();
|
||||
|
||||
libzerocash::Coin c_1_new(pubAddress3, 2);
|
||||
libzerocash::Coin c_2_new(pubAddress4, 2);
|
||||
cout << "Successfully created coins to spend.\n" << endl;
|
||||
|
||||
vector<bool> root_bv(ZC_ROOT_SIZE * 8);
|
||||
merkleTree.getRootValue(root_bv);
|
||||
vector<unsigned char> rt(ZC_ROOT_SIZE);
|
||||
libzerocash::convertVectorToBytesVector(root_bv, rt);
|
||||
|
||||
|
||||
vector<unsigned char> as(ZC_SIG_PK_SIZE, 'a');
|
||||
|
||||
cout << "Creating a pour transaction...\n" << endl;
|
||||
libzerocash::PourTransaction pourtx(1, p,
|
||||
rt,
|
||||
coins.at(1), coins.at(3),
|
||||
addrs.at(1), addrs.at(3),
|
||||
1, 3,
|
||||
witness_1, witness_2,
|
||||
pubAddress3, pubAddress4,
|
||||
0,
|
||||
0,
|
||||
as,
|
||||
c_1_new, c_2_new);
|
||||
cout << "Successfully created a pour transaction.\n" << endl;
|
||||
|
||||
std::vector<unsigned char> pubkeyHash(ZC_SIG_PK_SIZE, 'a');
|
||||
|
||||
cout << "Verifying a pour transaction...\n" << endl;
|
||||
bool pourtx_res = pourtx.verify(p, pubkeyHash, rt);
|
||||
|
||||
BOOST_CHECK(minttx_res && pourtx_res);
|
||||
}
|
||||
Reference in New Issue
Block a user