forked from hush/silentdragonlite-cli
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.
2167 lines
82 KiB
2167 lines
82 KiB
use std::convert::TryInto;
|
|
use std::io::{Error};
|
|
use rand::{RngCore, rngs::OsRng};
|
|
|
|
use ff::{Field, PrimeField, PrimeFieldRepr};
|
|
use pairing::bls12_381::Bls12;
|
|
use protobuf::{Message, UnknownFields, CachedSize, RepeatedField};
|
|
use zcash_client_backend::{encoding::encode_payment_address,
|
|
proto::compact_formats::{
|
|
CompactBlock, CompactOutput, CompactSpend, CompactTx,
|
|
}
|
|
};
|
|
use zcash_primitives::{
|
|
block::BlockHash,
|
|
jubjub::fs::Fs,
|
|
note_encryption::{Memo, SaplingNoteEncryption},
|
|
primitives::{Note, PaymentAddress},
|
|
legacy::{Script, TransparentAddress,},
|
|
transaction::{
|
|
TxId, Transaction, TransactionData,
|
|
components::{TxOut, TxIn, OutPoint, Amount,},
|
|
components::amount::DEFAULT_FEE,
|
|
},
|
|
zip32::{ExtendedFullViewingKey, ExtendedSpendingKey},
|
|
JUBJUB,
|
|
};
|
|
|
|
use sha2::{Sha256, Digest};
|
|
|
|
use super::LightWallet;
|
|
use super::LightClientConfig;
|
|
use secp256k1::{Secp256k1, key::PublicKey, key::SecretKey};
|
|
use crate::SaplingParams;
|
|
|
|
fn get_sapling_params() -> Result<(Vec<u8>, Vec<u8>), Error> {
|
|
// Read Sapling Params
|
|
let mut sapling_output = vec![];
|
|
sapling_output.extend_from_slice(SaplingParams::get("sapling-output.params").unwrap().as_ref());
|
|
println!("Read output {}", sapling_output.len());
|
|
|
|
let mut sapling_spend = vec![];
|
|
sapling_spend.extend_from_slice(SaplingParams::get("sapling-spend.params").unwrap().as_ref());
|
|
println!("Read output {}", sapling_spend.len());
|
|
|
|
Ok((sapling_spend, sapling_output))
|
|
}
|
|
|
|
struct FakeCompactBlock {
|
|
block: CompactBlock,
|
|
}
|
|
|
|
impl FakeCompactBlock {
|
|
fn new(height: i32, prev_hash: BlockHash) -> Self {
|
|
// Create a fake Note for the account
|
|
let mut rng = OsRng;
|
|
|
|
let mut cb = CompactBlock::new();
|
|
|
|
cb.set_height(height as u64);
|
|
cb.hash.resize(32, 0);
|
|
rng.fill_bytes(&mut cb.hash);
|
|
|
|
cb.prevHash.extend_from_slice(&prev_hash.0);
|
|
|
|
FakeCompactBlock { block: cb }
|
|
}
|
|
|
|
fn as_bytes(&self) -> Vec<u8> {
|
|
self.block.write_to_bytes().unwrap()
|
|
}
|
|
|
|
fn hash(&self) -> BlockHash {
|
|
BlockHash(self.block.hash[..].try_into().unwrap())
|
|
}
|
|
|
|
fn tx_to_compact_tx(tx: &Transaction, index: u64) -> CompactTx {
|
|
let spends = tx.shielded_spends.iter().map(|s| {
|
|
let mut c_spend = CompactSpend::default();
|
|
c_spend.set_nf(s.nullifier.to_vec());
|
|
|
|
c_spend
|
|
}).collect::<Vec<CompactSpend>>();
|
|
|
|
let outputs = tx.shielded_outputs.iter().map(|o| {
|
|
let mut c_out = CompactOutput::default();
|
|
|
|
let mut cmu_bytes = vec![];
|
|
o.cmu.into_repr().write_le(&mut cmu_bytes).unwrap();
|
|
|
|
let mut epk_bytes = vec![];
|
|
o.ephemeral_key.write(&mut epk_bytes).unwrap();
|
|
|
|
c_out.set_cmu(cmu_bytes);
|
|
c_out.set_epk(epk_bytes);
|
|
c_out.set_ciphertext(o.enc_ciphertext[0..52].to_vec());
|
|
|
|
c_out
|
|
}).collect::<Vec<CompactOutput>>();
|
|
|
|
CompactTx {
|
|
index,
|
|
hash: tx.txid().0.to_vec(),
|
|
fee: 0, // TODO: Get Fee
|
|
spends: RepeatedField::from_vec(spends),
|
|
outputs: RepeatedField::from_vec(outputs),
|
|
unknown_fields: UnknownFields::default(),
|
|
cached_size: CachedSize::default(),
|
|
}
|
|
}
|
|
|
|
// Convert the transaction into a CompactTx and add it to this block
|
|
fn add_tx(&mut self, tx: &Transaction) {
|
|
let ctx = FakeCompactBlock::tx_to_compact_tx(&tx, self.block.vtx.len() as u64);
|
|
self.block.vtx.push(ctx);
|
|
}
|
|
|
|
// Add a new tx into the block, paying the given address the amount.
|
|
// Returns the nullifier of the new note.
|
|
fn add_tx_paying(&mut self, extfvk: ExtendedFullViewingKey, value: u64)
|
|
-> (Vec<u8>, TxId) {
|
|
let to = extfvk.default_address().unwrap().1;
|
|
let value = Amount::from_u64(value).unwrap();
|
|
|
|
// Create a fake Note for the account
|
|
let mut rng = OsRng;
|
|
let note = Note {
|
|
g_d: to.diversifier.g_d::<Bls12>(&JUBJUB).unwrap(),
|
|
pk_d: to.pk_d.clone(),
|
|
value: value.into(),
|
|
r: Fs::random(&mut rng),
|
|
};
|
|
let encryptor = SaplingNoteEncryption::new(
|
|
extfvk.fvk.ovk,
|
|
note.clone(),
|
|
to.clone(),
|
|
Memo::default(),
|
|
&mut rng,
|
|
);
|
|
let mut cmu = vec![];
|
|
note.cm(&JUBJUB).into_repr().write_le(&mut cmu).unwrap();
|
|
let mut epk = vec![];
|
|
encryptor.epk().write(&mut epk).unwrap();
|
|
let enc_ciphertext = encryptor.encrypt_note_plaintext();
|
|
|
|
// Create a fake CompactBlock containing the note
|
|
let mut cout = CompactOutput::new();
|
|
cout.set_cmu(cmu);
|
|
cout.set_epk(epk);
|
|
cout.set_ciphertext(enc_ciphertext[..52].to_vec());
|
|
let mut ctx = CompactTx::new();
|
|
let mut txid = vec![0; 32];
|
|
rng.fill_bytes(&mut txid);
|
|
ctx.set_hash(txid.clone());
|
|
ctx.outputs.push(cout);
|
|
|
|
self.block.vtx.push(ctx);
|
|
(note.nf(&extfvk.fvk.vk, 0, &JUBJUB), TxId(txid[..].try_into().unwrap()))
|
|
}
|
|
|
|
fn add_tx_spending(&mut self,
|
|
(nf, in_value): (Vec<u8>, u64),
|
|
extfvk: ExtendedFullViewingKey,
|
|
to: PaymentAddress<Bls12>,
|
|
value: u64) -> TxId {
|
|
let mut rng = OsRng;
|
|
|
|
let in_value = Amount::from_u64(in_value).unwrap();
|
|
let value = Amount::from_u64(value).unwrap();
|
|
|
|
// Create a fake CompactBlock containing the note
|
|
let mut cspend = CompactSpend::new();
|
|
cspend.set_nf(nf);
|
|
let mut ctx = CompactTx::new();
|
|
let mut txid = vec![0; 32];
|
|
rng.fill_bytes(&mut txid);
|
|
ctx.set_hash(txid.clone());
|
|
ctx.spends.push(cspend);
|
|
|
|
// Create a fake Note for the payment
|
|
ctx.outputs.push({
|
|
let note = Note {
|
|
g_d: to.diversifier.g_d::<Bls12>(&JUBJUB).unwrap(),
|
|
pk_d: to.pk_d.clone(),
|
|
value: value.into(),
|
|
r: Fs::random(&mut rng),
|
|
};
|
|
let encryptor = SaplingNoteEncryption::new(
|
|
extfvk.fvk.ovk,
|
|
note.clone(),
|
|
to,
|
|
Memo::default(),
|
|
&mut rng,
|
|
);
|
|
let mut cmu = vec![];
|
|
note.cm(&JUBJUB).into_repr().write_le(&mut cmu).unwrap();
|
|
let mut epk = vec![];
|
|
encryptor.epk().write(&mut epk).unwrap();
|
|
let enc_ciphertext = encryptor.encrypt_note_plaintext();
|
|
|
|
let mut cout = CompactOutput::new();
|
|
cout.set_cmu(cmu);
|
|
cout.set_epk(epk);
|
|
cout.set_ciphertext(enc_ciphertext[..52].to_vec());
|
|
cout
|
|
});
|
|
|
|
// Create a fake Note for the change
|
|
ctx.outputs.push({
|
|
let change_addr = extfvk.default_address().unwrap().1;
|
|
let note = Note {
|
|
g_d: change_addr.diversifier.g_d::<Bls12>(&JUBJUB).unwrap(),
|
|
pk_d: change_addr.pk_d.clone(),
|
|
value: (in_value - value).into(),
|
|
r: Fs::random(&mut rng),
|
|
};
|
|
let encryptor = SaplingNoteEncryption::new(
|
|
extfvk.fvk.ovk,
|
|
note.clone(),
|
|
change_addr,
|
|
Memo::default(),
|
|
&mut rng,
|
|
);
|
|
let mut cmu = vec![];
|
|
note.cm(&JUBJUB).into_repr().write_le(&mut cmu).unwrap();
|
|
let mut epk = vec![];
|
|
encryptor.epk().write(&mut epk).unwrap();
|
|
let enc_ciphertext = encryptor.encrypt_note_plaintext();
|
|
|
|
let mut cout = CompactOutput::new();
|
|
cout.set_cmu(cmu);
|
|
cout.set_epk(epk);
|
|
cout.set_ciphertext(enc_ciphertext[..52].to_vec());
|
|
cout
|
|
});
|
|
|
|
self.block.vtx.push(ctx);
|
|
|
|
TxId(txid[..].try_into().unwrap())
|
|
}
|
|
}
|
|
|
|
struct FakeTransaction {
|
|
tx: Transaction,
|
|
}
|
|
|
|
impl FakeTransaction {
|
|
// New FakeTransaction with random txid
|
|
fn new<R: RngCore>(rng: &mut R) -> Self {
|
|
let mut txid = [0u8; 32];
|
|
rng.fill_bytes(&mut txid);
|
|
FakeTransaction::new_with_txid(TxId(txid))
|
|
}
|
|
|
|
fn new_with_txid(txid: TxId) -> Self {
|
|
FakeTransaction {
|
|
tx: Transaction {
|
|
txid,
|
|
data: TransactionData::new()
|
|
}
|
|
}
|
|
}
|
|
|
|
fn get_tx(&self) -> &Transaction {
|
|
&self.tx
|
|
}
|
|
|
|
fn add_t_output(&mut self, pk: &PublicKey, value: u64) {
|
|
let mut hash160 = ripemd160::Ripemd160::new();
|
|
hash160.input(Sha256::digest(&pk.serialize()[..].to_vec()));
|
|
|
|
let taddr_bytes = hash160.result();
|
|
|
|
self.tx.data.vout.push(TxOut {
|
|
value: Amount::from_u64(value).unwrap(),
|
|
script_pubkey: TransparentAddress::PublicKey(taddr_bytes.try_into().unwrap()).script(),
|
|
});
|
|
}
|
|
|
|
fn add_t_input(&mut self, txid: TxId, n: u32) {
|
|
self.tx.data.vin.push(TxIn {
|
|
prevout: OutPoint{
|
|
hash: txid.0,
|
|
n
|
|
},
|
|
script_sig: Script{0: vec![]},
|
|
sequence: 0,
|
|
});
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_z_balances() {
|
|
let wallet = LightWallet::new(None, &get_test_config(), 0).unwrap();
|
|
|
|
const AMOUNT1:u64 = 5;
|
|
// Address is encoded in bech32
|
|
let address = Some(encode_payment_address(wallet.config.hrp_sapling_address(),
|
|
&wallet.extfvks.read().unwrap()[0].default_address().unwrap().1));
|
|
|
|
let mut cb1 = FakeCompactBlock::new(0, BlockHash([0; 32]));
|
|
cb1.add_tx_paying(wallet.extfvks.read().unwrap()[0].clone(), AMOUNT1);
|
|
|
|
// Make sure that the intial state is empty
|
|
assert_eq!(wallet.txs.read().unwrap().len(), 0);
|
|
assert_eq!(wallet.blocks.read().unwrap().len(), 0);
|
|
assert_eq!(wallet.zbalance(None), 0);
|
|
assert_eq!(wallet.zbalance(address.clone()), 0);
|
|
|
|
wallet.scan_block(&cb1.as_bytes()).unwrap();
|
|
|
|
assert_eq!(wallet.txs.read().unwrap().len(), 1);
|
|
assert_eq!(wallet.blocks.read().unwrap().len(), 1);
|
|
assert_eq!(wallet.zbalance(None), AMOUNT1);
|
|
assert_eq!(wallet.zbalance(address.clone()), AMOUNT1);
|
|
|
|
const AMOUNT2:u64 = 10;
|
|
|
|
// Add a second block
|
|
let mut cb2 = FakeCompactBlock::new(1, cb1.hash());
|
|
cb2.add_tx_paying(wallet.extfvks.read().unwrap()[0].clone(), AMOUNT2);
|
|
|
|
wallet.scan_block(&cb2.as_bytes()).unwrap();
|
|
|
|
assert_eq!(wallet.txs.read().unwrap().len(), 2);
|
|
assert_eq!(wallet.blocks.read().unwrap().len(), 2);
|
|
assert_eq!(wallet.zbalance(None), AMOUNT1 + AMOUNT2);
|
|
assert_eq!(wallet.zbalance(address.clone()), AMOUNT1 + AMOUNT2);
|
|
}
|
|
|
|
#[test]
|
|
fn test_z_change_balances() {
|
|
let wallet = LightWallet::new(None, &get_test_config(), 0).unwrap();
|
|
|
|
// First, add an incoming transaction
|
|
const AMOUNT1:u64 = 5;
|
|
|
|
let mut cb1 = FakeCompactBlock::new(0, BlockHash([0; 32]));
|
|
let (nf1, txid1) = cb1.add_tx_paying(wallet.extfvks.read().unwrap()[0].clone(), AMOUNT1);
|
|
|
|
wallet.scan_block(&cb1.as_bytes()).unwrap();
|
|
|
|
assert_eq!(wallet.txs.read().unwrap().len(), 1);
|
|
assert_eq!(wallet.blocks.read().unwrap().len(), 1);
|
|
assert_eq!(wallet.zbalance(None), AMOUNT1);
|
|
|
|
const AMOUNT2:u64 = 2;
|
|
|
|
// Add a second block, spending the first note
|
|
let addr2 = ExtendedFullViewingKey::from(&ExtendedSpendingKey::master(&[0u8; 32]))
|
|
.default_address().unwrap().1;
|
|
let mut cb2 = FakeCompactBlock::new(1, cb1.hash());
|
|
let txid2 = cb2.add_tx_spending((nf1, AMOUNT1), wallet.extfvks.read().unwrap()[0].clone(), addr2, AMOUNT2);
|
|
wallet.scan_block(&cb2.as_bytes()).unwrap();
|
|
|
|
// Now, the original note should be spent and there should be a change
|
|
assert_eq!(wallet.zbalance(None), AMOUNT1 - AMOUNT2);
|
|
|
|
let txs = wallet.txs.read().unwrap();
|
|
|
|
// Old note was spent
|
|
assert_eq!(txs[&txid1].txid, txid1);
|
|
assert_eq!(txs[&txid1].notes.len(), 1);
|
|
assert_eq!(txs[&txid1].notes[0].spent.unwrap(), txid2);
|
|
assert_eq!(txs[&txid1].notes[0].note.value, AMOUNT1);
|
|
assert_eq!(txs[&txid1].notes[0].is_change, false);
|
|
|
|
// new note is not spent
|
|
assert_eq!(txs[&txid2].txid, txid2);
|
|
assert_eq!(txs[&txid2].notes.len(), 1);
|
|
assert_eq!(txs[&txid2].notes[0].spent, None);
|
|
assert_eq!(txs[&txid2].notes[0].note.value, AMOUNT1 - AMOUNT2);
|
|
assert_eq!(txs[&txid2].notes[0].is_change, true);
|
|
assert_eq!(txs[&txid2].total_shielded_value_spent, AMOUNT1);
|
|
}
|
|
|
|
#[test]
|
|
fn test_t_receive_spend() {
|
|
let mut rng = OsRng;
|
|
let secp = Secp256k1::new();
|
|
|
|
let wallet = LightWallet::new(None, &get_test_config(), 0).unwrap();
|
|
|
|
let pk = PublicKey::from_secret_key(&secp, &wallet.tkeys.read().unwrap()[0]);
|
|
let taddr = wallet.address_from_sk(&wallet.tkeys.read().unwrap()[0]);
|
|
|
|
const AMOUNT1: u64 = 20;
|
|
|
|
let mut tx = FakeTransaction::new(&mut rng);
|
|
tx.add_t_output(&pk, AMOUNT1);
|
|
let txid1 = tx.get_tx().txid();
|
|
|
|
wallet.scan_full_tx(&tx.get_tx(), 100, 0); // Pretend it is at height 100
|
|
|
|
{
|
|
let txs = wallet.txs.read().unwrap();
|
|
|
|
// Now make sure the t addr was recieved
|
|
assert_eq!(txs.len(), 1);
|
|
assert_eq!(txs[&txid1].utxos.len(), 1);
|
|
assert_eq!(txs[&txid1].utxos[0].address, taddr);
|
|
assert_eq!(txs[&txid1].utxos[0].txid, txid1);
|
|
assert_eq!(txs[&txid1].utxos[0].output_index, 0);
|
|
assert_eq!(txs[&txid1].utxos[0].value, AMOUNT1);
|
|
assert_eq!(txs[&txid1].utxos[0].height, 100);
|
|
assert_eq!(txs[&txid1].utxos[0].spent, None);
|
|
assert_eq!(txs[&txid1].utxos[0].unconfirmed_spent, None);
|
|
|
|
assert_eq!(wallet.tbalance(None), AMOUNT1);
|
|
assert_eq!(wallet.tbalance(Some(taddr)), AMOUNT1);
|
|
}
|
|
|
|
// Create a new Tx, spending this taddr
|
|
let mut tx = FakeTransaction::new(&mut rng);
|
|
tx.add_t_input(txid1, 0);
|
|
let txid2 = tx.get_tx().txid();
|
|
|
|
wallet.scan_full_tx(&tx.get_tx(), 101, 0); // Pretent it is at height 101
|
|
|
|
{
|
|
// Make sure the txid was spent
|
|
let txs = wallet.txs.read().unwrap();
|
|
|
|
// Old utxo, that should be spent now
|
|
assert_eq!(txs.len(), 2);
|
|
assert_eq!(txs[&txid1].utxos.len(), 1);
|
|
assert_eq!(txs[&txid1].utxos[0].value, AMOUNT1);
|
|
assert_eq!(txs[&txid1].utxos[0].spent, Some(txid2));
|
|
assert_eq!(txs[&txid1].utxos[0].unconfirmed_spent, None);
|
|
|
|
assert_eq!(txs[&txid2].block, 101); // The second TxId is at block 101
|
|
assert_eq!(txs[&txid2].utxos.len(), 0); // The second TxId has no UTXOs
|
|
assert_eq!(txs[&txid2].total_transparent_value_spent, AMOUNT1);
|
|
|
|
// Make sure there is no t-ZEC left
|
|
assert_eq!(wallet.tbalance(None), 0);
|
|
}
|
|
}
|
|
|
|
|
|
#[test]
|
|
/// This test spends and receives t addresses among non-wallet t addresses to make sure that
|
|
/// we're detecting and spending only our t addrs.
|
|
fn test_t_receive_spend_among_tadds() {
|
|
let mut rng = OsRng;
|
|
let secp = Secp256k1::new();
|
|
|
|
let wallet = LightWallet::new(None, &get_test_config(), 0).unwrap();
|
|
|
|
let pk = PublicKey::from_secret_key(&secp, &wallet.tkeys.read().unwrap()[0]);
|
|
let taddr = wallet.address_from_sk(&wallet.tkeys.read().unwrap()[0]);
|
|
|
|
let non_wallet_sk = &SecretKey::from_slice(&[1u8; 32]).unwrap();
|
|
let non_wallet_pk = PublicKey::from_secret_key(&secp, &non_wallet_sk);
|
|
|
|
const AMOUNT1: u64 = 30;
|
|
|
|
let mut tx = FakeTransaction::new(&mut rng);
|
|
// Add a non-wallet output
|
|
tx.add_t_output(&non_wallet_pk, 20);
|
|
tx.add_t_output(&pk, AMOUNT1); // Our wallet t output
|
|
tx.add_t_output(&non_wallet_pk, 25);
|
|
let txid1 = tx.get_tx().txid();
|
|
|
|
wallet.scan_full_tx(&tx.get_tx(), 100, 0); // Pretend it is at height 100
|
|
|
|
{
|
|
let txs = wallet.txs.read().unwrap();
|
|
|
|
// Now make sure the t addr was received
|
|
assert_eq!(txs.len(), 1);
|
|
assert_eq!(txs[&txid1].utxos.len(), 1);
|
|
assert_eq!(txs[&txid1].utxos[0].address, taddr);
|
|
assert_eq!(txs[&txid1].utxos[0].txid, txid1);
|
|
assert_eq!(txs[&txid1].utxos[0].output_index, 1);
|
|
assert_eq!(txs[&txid1].utxos[0].value, AMOUNT1);
|
|
assert_eq!(txs[&txid1].utxos[0].height, 100);
|
|
assert_eq!(txs[&txid1].utxos[0].spent, None);
|
|
assert_eq!(txs[&txid1].utxos[0].unconfirmed_spent, None);
|
|
|
|
assert_eq!(wallet.tbalance(None), AMOUNT1);
|
|
assert_eq!(wallet.tbalance(Some(taddr)), AMOUNT1);
|
|
}
|
|
|
|
// Create a new Tx, spending this taddr
|
|
let mut tx = FakeTransaction::new(&mut rng);
|
|
tx.add_t_input(txid1, 1); // Ours was at position 1 in the input tx
|
|
let txid2 = tx.get_tx().txid();
|
|
|
|
wallet.scan_full_tx(&tx.get_tx(), 101, 0); // Pretent it is at height 101
|
|
|
|
{
|
|
// Make sure the txid was spent
|
|
let txs = wallet.txs.read().unwrap();
|
|
|
|
// Old utxo, that should be spent now
|
|
assert_eq!(txs.len(), 2);
|
|
assert_eq!(txs[&txid1].utxos.len(), 1);
|
|
assert_eq!(txs[&txid1].utxos[0].value, AMOUNT1);
|
|
assert_eq!(txs[&txid1].utxos[0].spent, Some(txid2));
|
|
assert_eq!(txs[&txid1].utxos[0].unconfirmed_spent, None);
|
|
|
|
assert_eq!(txs[&txid2].block, 101); // The second TxId is at block 101
|
|
assert_eq!(txs[&txid2].utxos.len(), 0); // The second TxId has no UTXOs
|
|
assert_eq!(txs[&txid2].total_transparent_value_spent, AMOUNT1);
|
|
|
|
// Make sure there is no t-ZEC left
|
|
assert_eq!(wallet.tbalance(None), 0);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_serialization() {
|
|
let secp = Secp256k1::new();
|
|
let config = get_test_config();
|
|
|
|
let wallet = LightWallet::new(None, &config, 0).unwrap();
|
|
|
|
// First, add an incoming transaction
|
|
const AMOUNT1:u64 = 5;
|
|
|
|
let mut cb1 = FakeCompactBlock::new(0, BlockHash([0; 32]));
|
|
let (nf1, txid1) = cb1.add_tx_paying(wallet.extfvks.read().unwrap()[0].clone(), AMOUNT1);
|
|
|
|
wallet.scan_block(&cb1.as_bytes()).unwrap();
|
|
|
|
assert_eq!(wallet.txs.read().unwrap().len(), 1);
|
|
assert_eq!(wallet.blocks.read().unwrap().len(), 1);
|
|
assert_eq!(wallet.zbalance(None), AMOUNT1);
|
|
|
|
// Add a t input at the Tx
|
|
let pk = PublicKey::from_secret_key(&secp, &wallet.tkeys.read().unwrap()[0]);
|
|
let taddr = wallet.address_from_sk(&wallet.tkeys.read().unwrap()[0]);
|
|
|
|
const TAMOUNT1: u64 = 20;
|
|
|
|
let mut tx = FakeTransaction::new_with_txid(txid1);
|
|
tx.add_t_output(&pk, TAMOUNT1);
|
|
wallet.scan_full_tx(&tx.get_tx(), 0, 0); // Height 0
|
|
|
|
const AMOUNT2:u64 = 2;
|
|
|
|
// Add a second block, spending the first note
|
|
let addr2 = ExtendedFullViewingKey::from(&ExtendedSpendingKey::master(&[0u8; 32]))
|
|
.default_address().unwrap().1;
|
|
let mut cb2 = FakeCompactBlock::new(1, cb1.hash());
|
|
let txid2 = cb2.add_tx_spending((nf1, AMOUNT1), wallet.extfvks.read().unwrap()[0].clone(), addr2, AMOUNT2);
|
|
wallet.scan_block(&cb2.as_bytes()).unwrap();
|
|
|
|
let mut tx = FakeTransaction::new_with_txid(txid2);
|
|
tx.add_t_input(txid1, 0);
|
|
wallet.scan_full_tx(&tx.get_tx(), 1, 0); // Height 1
|
|
|
|
// Now, the original note should be spent and there should be a change
|
|
assert_eq!(wallet.zbalance(None), AMOUNT1 - AMOUNT2 ); // The t addr amount is received + spent, so it cancels out
|
|
|
|
// Now, serialize the wallet and read it back again
|
|
let mut serialized_data = vec![];
|
|
wallet.write(&mut serialized_data).expect("Serialize wallet");
|
|
let wallet2 = LightWallet::read(&serialized_data[..], &config).unwrap();
|
|
|
|
assert_eq!(wallet2.zbalance(None), AMOUNT1 - AMOUNT2);
|
|
|
|
// Test the keys were serialized correctly
|
|
{
|
|
assert_eq!(wallet.seed, wallet2.seed);
|
|
|
|
assert_eq!(wallet.extsks.read().unwrap().len(), wallet2.extsks.read().unwrap().len());
|
|
assert_eq!(wallet.extsks.read().unwrap()[0], wallet2.extsks.read().unwrap()[0]);
|
|
assert_eq!(wallet.extfvks.read().unwrap()[0], wallet2.extfvks.read().unwrap()[0]);
|
|
assert_eq!(wallet.zaddress.read().unwrap()[0], wallet2.zaddress.read().unwrap()[0]);
|
|
|
|
assert_eq!(wallet.tkeys.read().unwrap().len(), wallet2.tkeys.read().unwrap().len());
|
|
assert_eq!(wallet.tkeys.read().unwrap()[0], wallet2.tkeys.read().unwrap()[0]);
|
|
}
|
|
|
|
// Test blocks were serialized properly
|
|
{
|
|
let blks = wallet2.blocks.read().unwrap();
|
|
|
|
assert_eq!(blks.len(), 2);
|
|
assert_eq!(blks[0].height, 0);
|
|
assert_eq!(blks[1].height, 1);
|
|
}
|
|
|
|
// Test txns were serialized properly.
|
|
{
|
|
let txs = wallet2.txs.read().unwrap();
|
|
|
|
// Old note was spent
|
|
assert_eq!(txs[&txid1].txid, txid1);
|
|
assert_eq!(txs[&txid1].notes.len(), 1);
|
|
assert_eq!(txs[&txid1].notes[0].spent.unwrap(), txid2);
|
|
assert_eq!(txs[&txid1].notes[0].note.value, AMOUNT1);
|
|
assert_eq!(txs[&txid1].notes[0].is_change, false);
|
|
|
|
// Old UTXO was spent
|
|
assert_eq!(txs[&txid1].utxos.len(), 1);
|
|
assert_eq!(txs[&txid1].utxos[0].address, taddr);
|
|
assert_eq!(txs[&txid1].utxos[0].txid, txid1);
|
|
assert_eq!(txs[&txid1].utxos[0].output_index, 0);
|
|
assert_eq!(txs[&txid1].utxos[0].value, TAMOUNT1);
|
|
assert_eq!(txs[&txid1].utxos[0].height, 0);
|
|
assert_eq!(txs[&txid1].utxos[0].spent, Some(txid2));
|
|
assert_eq!(txs[&txid1].utxos[0].unconfirmed_spent, None);
|
|
|
|
// new note is not spent
|
|
assert_eq!(txs[&txid2].txid, txid2);
|
|
assert_eq!(txs[&txid2].notes.len(), 1);
|
|
assert_eq!(txs[&txid2].notes[0].spent, None);
|
|
assert_eq!(txs[&txid2].notes[0].note.value, AMOUNT1 - AMOUNT2);
|
|
assert_eq!(txs[&txid2].notes[0].is_change, true);
|
|
assert_eq!(txs[&txid2].total_shielded_value_spent, AMOUNT1);
|
|
|
|
// The UTXO was spent in txid2
|
|
assert_eq!(txs[&txid2].utxos.len(), 0); // The second TxId has no UTXOs
|
|
assert_eq!(txs[&txid2].total_transparent_value_spent, TAMOUNT1);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_multi_serialization() {
|
|
let config = get_test_config();
|
|
|
|
let wallet = LightWallet::new(None, &config, 0).unwrap();
|
|
|
|
let taddr1 = wallet.address_from_sk(&wallet.tkeys.read().unwrap()[0]);
|
|
let taddr2 = wallet.add_taddr();
|
|
|
|
let (zaddr1, zpk1) = &wallet.get_z_private_keys()[0];
|
|
let zaddr2 = wallet.add_zaddr();
|
|
|
|
let mut serialized_data = vec![];
|
|
wallet.write(&mut serialized_data).expect("Serialize wallet");
|
|
let wallet2 = LightWallet::read(&serialized_data[..], &config).unwrap();
|
|
|
|
assert_eq!(wallet2.tkeys.read().unwrap().len(), 2);
|
|
assert_eq!(wallet2.extsks.read().unwrap().len(), 2);
|
|
assert_eq!(wallet2.extfvks.read().unwrap().len(), 2);
|
|
assert_eq!(wallet2.zaddress.read().unwrap().len(), 2);
|
|
|
|
assert_eq!(taddr1, wallet.address_from_sk(&wallet.tkeys.read().unwrap()[0]));
|
|
assert_eq!(taddr2, wallet.address_from_sk(&wallet.tkeys.read().unwrap()[1]));
|
|
|
|
let (w2_zaddr1, w2_zpk1) = &wallet.get_z_private_keys()[0];
|
|
let (w2_zaddr2, _) = &wallet.get_z_private_keys()[1];
|
|
assert_eq!(zaddr1, w2_zaddr1);
|
|
assert_eq!(zpk1, w2_zpk1);
|
|
assert_eq!(zaddr2, *w2_zaddr2);
|
|
|
|
}
|
|
|
|
fn get_test_config() -> LightClientConfig {
|
|
LightClientConfig {
|
|
server: "0.0.0.0:0".parse().unwrap(),
|
|
chain_name: "test".to_string(),
|
|
sapling_activation_height: 0,
|
|
consensus_branch_id: "000000".to_string(),
|
|
anchor_offset: 0,
|
|
no_cert_verification: false,
|
|
data_dir: None,
|
|
}
|
|
}
|
|
|
|
// Get a test wallet already setup with a single note
|
|
fn get_test_wallet(amount: u64) -> (LightWallet, TxId, BlockHash) {
|
|
let config = get_test_config();
|
|
|
|
let wallet = LightWallet::new(None, &config, 0).unwrap();
|
|
|
|
let mut cb1 = FakeCompactBlock::new(0, BlockHash([0; 32]));
|
|
let (_, txid1) = cb1.add_tx_paying(wallet.extfvks.read().unwrap()[0].clone(), amount);
|
|
wallet.scan_block(&cb1.as_bytes()).unwrap();
|
|
|
|
// We have one note
|
|
{
|
|
let txs = wallet.txs.read().unwrap();
|
|
assert_eq!(txs[&txid1].notes.len(), 1);
|
|
assert_eq!(txs[&txid1].notes[0].note.value, amount);
|
|
assert_eq!(txs[&txid1].notes[0].spent, None);
|
|
assert_eq!(txs[&txid1].notes[0].unconfirmed_spent, None);
|
|
}
|
|
|
|
assert_eq!(wallet.verified_zbalance(None), amount);
|
|
|
|
// Create a new block so that the note is now verified to be spent
|
|
let cb2 = FakeCompactBlock::new(1, cb1.hash());
|
|
wallet.scan_block(&cb2.as_bytes()).unwrap();
|
|
|
|
(wallet, txid1, cb2.hash())
|
|
}
|
|
|
|
#[test]
|
|
fn test_z_spend_to_z() {
|
|
const AMOUNT1: u64 = 50000;
|
|
let (wallet, txid1, block_hash) = get_test_wallet(AMOUNT1);
|
|
|
|
let fvk = ExtendedFullViewingKey::from(&ExtendedSpendingKey::master(&[1u8; 32]));
|
|
let ext_address = encode_payment_address(wallet.config.hrp_sapling_address(),
|
|
&fvk.default_address().unwrap().1);
|
|
|
|
const AMOUNT_SENT: u64 = 20;
|
|
|
|
let outgoing_memo = "Outgoing Memo".to_string();
|
|
let fee: u64 = DEFAULT_FEE.try_into().unwrap();
|
|
|
|
let branch_id = u32::from_str_radix("76b809bb", 16).unwrap();
|
|
let (ss, so) = get_sapling_params().unwrap();
|
|
|
|
// Make sure that the balance exists
|
|
{
|
|
assert_eq!(wallet.zbalance(None), AMOUNT1);
|
|
assert_eq!(wallet.verified_zbalance(None), AMOUNT1);
|
|
}
|
|
|
|
// Create a tx and send to address
|
|
let raw_tx = wallet.send_to_address(branch_id, &ss, &so,
|
|
vec![(&ext_address, AMOUNT_SENT, Some(outgoing_memo.clone()))]).unwrap();
|
|
|
|
let sent_tx = Transaction::read(&raw_tx[..]).unwrap();
|
|
let sent_txid = sent_tx.txid();
|
|
|
|
// Now, the note should be unconfirmed spent
|
|
{
|
|
let txs = wallet.txs.read().unwrap();
|
|
|
|
assert_eq!(txs[&txid1].notes[0].note.value, AMOUNT1);
|
|
assert_eq!(txs[&txid1].notes[0].is_change, false);
|
|
assert_eq!(txs[&txid1].notes[0].spent, None);
|
|
assert_eq!(txs[&txid1].notes[0].unconfirmed_spent, Some(sent_txid));
|
|
}
|
|
|
|
// It should also be in the mempool structure
|
|
{
|
|
let mem = wallet.mempool_txs.read().unwrap();
|
|
|
|
assert_eq!(mem[&sent_txid].block, 2); // block number is next block
|
|
assert! (mem[&sent_txid].datetime > 0);
|
|
assert_eq!(mem[&sent_txid].txid, sent_txid);
|
|
assert_eq!(mem[&sent_txid].outgoing_metadata.len(), 1);
|
|
assert_eq!(mem[&sent_txid].outgoing_metadata[0].address, ext_address);
|
|
assert_eq!(mem[&sent_txid].outgoing_metadata[0].value, AMOUNT_SENT);
|
|
assert_eq!(mem[&sent_txid].outgoing_metadata[0].memo.to_utf8().unwrap().unwrap(), outgoing_memo);
|
|
}
|
|
|
|
{
|
|
// The wallet should deduct this from the verified balance. The zbalance still includes it
|
|
assert_eq!(wallet.zbalance(None), AMOUNT1);
|
|
assert_eq!(wallet.verified_zbalance(None), 0);
|
|
}
|
|
|
|
let mut cb3 = FakeCompactBlock::new(2, block_hash);
|
|
cb3.add_tx(&sent_tx);
|
|
wallet.scan_block(&cb3.as_bytes()).unwrap();
|
|
|
|
// Now this new Spent tx should be in, so the note should be marked confirmed spent
|
|
{
|
|
let txs = wallet.txs.read().unwrap();
|
|
assert_eq!(txs[&txid1].notes.len(), 1);
|
|
assert_eq!(txs[&txid1].notes[0].note.value, AMOUNT1);
|
|
assert_eq!(txs[&txid1].notes[0].spent, Some(sent_txid));
|
|
assert_eq!(txs[&txid1].notes[0].unconfirmed_spent, None);
|
|
|
|
// The sent tx should generate change
|
|
assert_eq!(txs[&sent_txid].notes.len(), 1);
|
|
assert_eq!(txs[&sent_txid].notes[0].note.value, AMOUNT1 - AMOUNT_SENT - fee);
|
|
assert_eq!(wallet.zbalance(None), AMOUNT1 - AMOUNT_SENT - fee);
|
|
assert_eq!(txs[&sent_txid].notes[0].is_change, true);
|
|
assert_eq!(txs[&sent_txid].notes[0].spent, None);
|
|
assert_eq!(txs[&sent_txid].notes[0].unconfirmed_spent, None);
|
|
}
|
|
|
|
{
|
|
// And the mempool tx should disappear
|
|
let mem = wallet.mempool_txs.read().unwrap();
|
|
assert!(mem.get(&sent_txid).is_none());
|
|
}
|
|
|
|
// Now, full scan the Tx, which should populate the Outgoing Meta data
|
|
wallet.scan_full_tx(&sent_tx, 2, 0);
|
|
|
|
// Check Outgoing Metadata
|
|
{
|
|
let txs = wallet.txs.read().unwrap();
|
|
assert_eq!(txs[&sent_txid].total_shielded_value_spent, AMOUNT1);
|
|
|
|
assert_eq!(txs[&sent_txid].outgoing_metadata.len(), 1);
|
|
|
|
assert_eq!(txs[&sent_txid].outgoing_metadata[0].address, ext_address);
|
|
assert_eq!(txs[&sent_txid].outgoing_metadata[0].value, AMOUNT_SENT);
|
|
assert_eq!(txs[&sent_txid].outgoing_metadata[0].memo.to_utf8().unwrap().unwrap(), outgoing_memo);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_multi_z() {
|
|
const AMOUNT1: u64 = 50000;
|
|
let (wallet, txid1, block_hash) = get_test_wallet(AMOUNT1);
|
|
|
|
let zaddr2 = wallet.add_zaddr(); // This is acually address #6, since there are 5 initial addresses in the wallet
|
|
|
|
const AMOUNT_SENT: u64 = 20;
|
|
|
|
let outgoing_memo = "Outgoing Memo".to_string();
|
|
let fee: u64 = DEFAULT_FEE.try_into().unwrap();
|
|
|
|
let branch_id = u32::from_str_radix("76b809bb", 16).unwrap();
|
|
let (ss, so) =get_sapling_params().unwrap();
|
|
|
|
// Create a tx and send to address
|
|
let raw_tx = wallet.send_to_address(branch_id, &ss, &so,
|
|
vec![(&zaddr2, AMOUNT_SENT, Some(outgoing_memo.clone()))]).unwrap();
|
|
|
|
let sent_tx = Transaction::read(&raw_tx[..]).unwrap();
|
|
let sent_txid = sent_tx.txid();
|
|
|
|
let mut cb3 = FakeCompactBlock::new(2, block_hash);
|
|
cb3.add_tx(&sent_tx);
|
|
wallet.scan_block(&cb3.as_bytes()).unwrap();
|
|
wallet.scan_full_tx(&sent_tx, 2, 0);
|
|
|
|
// Because the builder will randomize notes outputted, we need to find
|
|
// which note number is the change and which is the output note (Because this tx
|
|
// had both outputs in the same Tx)
|
|
let (change_note_number, ext_note_number) = {
|
|
let txs = wallet.txs.read().unwrap();
|
|
if txs[&sent_txid].notes[0].is_change { (0,1) } else { (1,0) }
|
|
};
|
|
|
|
// Now this new Spent tx should be in, so the note should be marked confirmed spent
|
|
{
|
|
let txs = wallet.txs.read().unwrap();
|
|
assert_eq!(txs[&txid1].notes.len(), 1);
|
|
assert_eq!(txs[&txid1].notes[0].note.value, AMOUNT1);
|
|
assert_eq!(txs[&txid1].notes[0].spent, Some(sent_txid));
|
|
assert_eq!(txs[&txid1].notes[0].unconfirmed_spent, None);
|
|
|
|
// The sent tx should generate change + the new incoming note
|
|
assert_eq!(txs[&sent_txid].notes.len(), 2);
|
|
|
|
assert_eq!(txs[&sent_txid].notes[change_note_number].note.value, AMOUNT1 - AMOUNT_SENT - fee);
|
|
assert_eq!(txs[&sent_txid].notes[change_note_number].account, 0);
|
|
assert_eq!(txs[&sent_txid].notes[change_note_number].is_change, true);
|
|
assert_eq!(txs[&sent_txid].notes[change_note_number].spent, None);
|
|
assert_eq!(txs[&sent_txid].notes[change_note_number].unconfirmed_spent, None);
|
|
assert_eq!(LightWallet::memo_str(&txs[&sent_txid].notes[change_note_number].memo), None);
|
|
|
|
assert_eq!(txs[&sent_txid].notes[ext_note_number].note.value, AMOUNT_SENT);
|
|
assert_eq!(txs[&sent_txid].notes[ext_note_number].account, 6); // The new addr is added after the change addresses
|
|
assert_eq!(txs[&sent_txid].notes[ext_note_number].is_change, false);
|
|
assert_eq!(txs[&sent_txid].notes[ext_note_number].spent, None);
|
|
assert_eq!(txs[&sent_txid].notes[ext_note_number].unconfirmed_spent, None);
|
|
assert_eq!(LightWallet::memo_str(&txs[&sent_txid].notes[ext_note_number].memo), Some(outgoing_memo));
|
|
|
|
assert_eq!(txs[&sent_txid].total_shielded_value_spent, AMOUNT1);
|
|
|
|
// No Outgoing meta data, since this is a wallet -> wallet tx
|
|
assert_eq!(txs[&sent_txid].outgoing_metadata.len(), 0);
|
|
}
|
|
|
|
// Now spend the money, which should pick notes from both addresses
|
|
let amount_all:u64 = (AMOUNT1 - AMOUNT_SENT - fee) + (AMOUNT_SENT) - fee;
|
|
let taddr = wallet.address_from_sk(&SecretKey::from_slice(&[1u8; 32]).unwrap());
|
|
|
|
let raw_tx = wallet.send_to_address(branch_id, &ss, &so,
|
|
vec![(&taddr, amount_all, None)]).unwrap();
|
|
let sent_tx = Transaction::read(&raw_tx[..]).unwrap();
|
|
let sent_ext_txid = sent_tx.txid();
|
|
|
|
let mut cb4 = FakeCompactBlock::new(3, cb3.hash());
|
|
cb4.add_tx(&sent_tx);
|
|
wallet.scan_block(&cb4.as_bytes()).unwrap();
|
|
wallet.scan_full_tx(&sent_tx, 3, 0);
|
|
|
|
{
|
|
// Both notes should be spent now.
|
|
let txs = wallet.txs.read().unwrap();
|
|
|
|
assert_eq!(txs[&sent_txid].notes[change_note_number].is_change, true);
|
|
assert_eq!(txs[&sent_txid].notes[change_note_number].spent, Some(sent_ext_txid));
|
|
assert_eq!(txs[&sent_txid].notes[change_note_number].unconfirmed_spent, None);
|
|
|
|
assert_eq!(txs[&sent_txid].notes[ext_note_number].is_change, false);
|
|
assert_eq!(txs[&sent_txid].notes[ext_note_number].spent, Some(sent_ext_txid));
|
|
assert_eq!(txs[&sent_txid].notes[ext_note_number].unconfirmed_spent, None);
|
|
|
|
// Check outgoing metadata for the external sent tx
|
|
assert_eq!(txs[&sent_ext_txid].notes.len(), 0); // No change was generated
|
|
assert_eq!(txs[&sent_ext_txid].outgoing_metadata.len(), 1);
|
|
assert_eq!(txs[&sent_ext_txid].outgoing_metadata[0].address, taddr);
|
|
assert_eq!(txs[&sent_ext_txid].outgoing_metadata[0].value, amount_all);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_z_spend_to_taddr() {
|
|
const AMOUNT1: u64 = 50000;
|
|
let (wallet, txid1, block_hash) = get_test_wallet(AMOUNT1);
|
|
|
|
let branch_id = u32::from_str_radix("76b809bb", 16).unwrap();
|
|
let (ss, so) = get_sapling_params().unwrap();
|
|
|
|
let taddr = wallet.address_from_sk(&SecretKey::from_slice(&[1u8; 32]).unwrap());
|
|
const AMOUNT_SENT: u64 = 30;
|
|
let fee: u64 = DEFAULT_FEE.try_into().unwrap();
|
|
|
|
let raw_tx = wallet.send_to_address(branch_id, &ss, &so,
|
|
vec![(&taddr, AMOUNT_SENT, None)]).unwrap();
|
|
let sent_tx = Transaction::read(&raw_tx[..]).unwrap();
|
|
let sent_txid = sent_tx.txid();
|
|
|
|
// Now, the note should be unconfirmed spent
|
|
{
|
|
let txs = wallet.txs.read().unwrap();
|
|
|
|
assert_eq!(txs[&txid1].notes[0].note.value, AMOUNT1);
|
|
assert_eq!(txs[&txid1].notes[0].is_change, false);
|
|
assert_eq!(txs[&txid1].notes[0].spent, None);
|
|
assert_eq!(txs[&txid1].notes[0].unconfirmed_spent, Some(sent_txid));
|
|
}
|
|
|
|
let mut cb3 = FakeCompactBlock::new(2, block_hash);
|
|
cb3.add_tx(&sent_tx);
|
|
wallet.scan_block(&cb3.as_bytes()).unwrap();
|
|
|
|
// Now this new Spent tx should be in, so the note should be marked confirmed spent
|
|
{
|
|
let txs = wallet.txs.read().unwrap();
|
|
assert_eq!(txs[&txid1].notes.len(), 1);
|
|
assert_eq!(txs[&txid1].notes[0].note.value, AMOUNT1);
|
|
assert_eq!(txs[&txid1].notes[0].spent, Some(sent_txid));
|
|
assert_eq!(txs[&txid1].notes[0].unconfirmed_spent, None);
|
|
|
|
// The sent tx should generate change
|
|
assert_eq!(txs[&sent_txid].notes.len(), 1);
|
|
assert_eq!(txs[&sent_txid].notes[0].note.value, AMOUNT1 - AMOUNT_SENT - fee);
|
|
assert_eq!(txs[&sent_txid].notes[0].is_change, true);
|
|
assert_eq!(txs[&sent_txid].notes[0].spent, None);
|
|
assert_eq!(txs[&sent_txid].notes[0].unconfirmed_spent, None);
|
|
}
|
|
|
|
// Now, full scan the Tx, which should populate the Outgoing Meta data
|
|
wallet.scan_full_tx(&sent_tx, 2, 0);
|
|
|
|
// Check Outgoing Metadata for t address
|
|
{
|
|
let txs = wallet.txs.read().unwrap();
|
|
assert_eq!(txs[&sent_txid].outgoing_metadata.len(), 1);
|
|
assert_eq!(txs[&sent_txid].outgoing_metadata[0].address, taddr);
|
|
assert_eq!(txs[&sent_txid].outgoing_metadata[0].value, AMOUNT_SENT);
|
|
assert_eq!(txs[&sent_txid].total_shielded_value_spent, AMOUNT1);
|
|
}
|
|
|
|
// Create a new Tx, but this time with a memo.
|
|
let raw_tx = wallet.send_to_address(branch_id, &ss, &so,
|
|
vec![(&taddr, AMOUNT_SENT, Some("T address memo".to_string()))]).unwrap();
|
|
let sent_tx = Transaction::read(&raw_tx[..]).unwrap();
|
|
let sent_txid2 = sent_tx.txid();
|
|
|
|
// There should be a mempool Tx, but the memo should be dropped, because it was sent to a
|
|
// t address
|
|
{
|
|
let txs = wallet.mempool_txs.read().unwrap();
|
|
|
|
assert_eq!(txs[&sent_txid2].outgoing_metadata.len(), 1);
|
|
assert_eq!(txs[&sent_txid2].outgoing_metadata[0].address, taddr);
|
|
assert_eq!(txs[&sent_txid2].outgoing_metadata[0].value, AMOUNT_SENT);
|
|
assert_eq!(LightWallet::memo_str(&Some(txs[&sent_txid2].outgoing_metadata[0].memo.clone())), None);
|
|
}
|
|
|
|
// Now add the block
|
|
let mut cb4 = FakeCompactBlock::new(3, cb3.hash());
|
|
cb4.add_tx(&sent_tx);
|
|
wallet.scan_block(&cb4.as_bytes()).unwrap();
|
|
wallet.scan_full_tx(&sent_tx, 3, 0);
|
|
|
|
// Check Outgoing Metadata for t address, but once again there should be no memo
|
|
{
|
|
let txs = wallet.txs.read().unwrap();
|
|
assert_eq!(txs[&sent_txid2].outgoing_metadata.len(), 1);
|
|
assert_eq!(txs[&sent_txid2].outgoing_metadata[0].address, taddr);
|
|
assert_eq!(txs[&sent_txid2].outgoing_metadata[0].value, AMOUNT_SENT);
|
|
assert_eq!(LightWallet::memo_str(&Some(txs[&sent_txid2].outgoing_metadata[0].memo.clone())), None);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_t_spend_to_z() {
|
|
let mut rng = OsRng;
|
|
let secp = Secp256k1::new();
|
|
|
|
const AMOUNT_Z: u64 = 50000;
|
|
const AMOUNT_T: u64 = 40000;
|
|
let (wallet, txid1, block_hash) = get_test_wallet(AMOUNT_Z);
|
|
|
|
let pk = PublicKey::from_secret_key(&secp, &wallet.tkeys.read().unwrap()[0]);
|
|
let taddr = wallet.address_from_sk(&wallet.tkeys.read().unwrap()[0]);
|
|
|
|
let mut tx = FakeTransaction::new(&mut rng);
|
|
tx.add_t_output(&pk, AMOUNT_T);
|
|
let txid_t = tx.get_tx().txid();
|
|
|
|
wallet.scan_full_tx(&tx.get_tx(), 1, 0); // Pretend it is at height 1
|
|
|
|
{
|
|
let txs = wallet.txs.read().unwrap();
|
|
|
|
// Now make sure the t addr was recieved
|
|
assert_eq!(txs[&txid_t].utxos.len(), 1);
|
|
assert_eq!(txs[&txid_t].utxos[0].address, taddr);
|
|
assert_eq!(txs[&txid_t].utxos[0].spent, None);
|
|
assert_eq!(txs[&txid_t].utxos[0].unconfirmed_spent, None);
|
|
|
|
assert_eq!(wallet.tbalance(None), AMOUNT_T);
|
|
}
|
|
|
|
|
|
let fvk = ExtendedFullViewingKey::from(&ExtendedSpendingKey::master(&[1u8; 32]));
|
|
let ext_address = encode_payment_address(wallet.config.hrp_sapling_address(),
|
|
&fvk.default_address().unwrap().1);
|
|
const AMOUNT_SENT: u64 = 20;
|
|
|
|
let outgoing_memo = "Outgoing Memo".to_string();
|
|
let fee: u64 = DEFAULT_FEE.try_into().unwrap();
|
|
|
|
let branch_id = u32::from_str_radix("76b809bb", 16).unwrap();
|
|
let (ss, so) =get_sapling_params().unwrap();
|
|
|
|
// Create a tx and send to address. This should consume both the UTXO and the note
|
|
let raw_tx = wallet.send_to_address(branch_id, &ss, &so,
|
|
vec![(&ext_address, AMOUNT_SENT, Some(outgoing_memo.clone()))]).unwrap();
|
|
|
|
let sent_tx = Transaction::read(&raw_tx[..]).unwrap();
|
|
let sent_txid = sent_tx.txid();
|
|
|
|
// Verify the sent_tx for sanity
|
|
{
|
|
// The tx has 1 note spent, 1 utxo spent, and (1 note out, 1 note change)
|
|
assert_eq!(sent_tx.shielded_spends.len(), 1);
|
|
assert_eq!(sent_tx.vin.len(), 1);
|
|
assert_eq!(sent_tx.shielded_outputs.len(), 2);
|
|
}
|
|
|
|
// Now, the note and utxo should be unconfirmed spent
|
|
{
|
|
let txs = wallet.txs.read().unwrap();
|
|
|
|
// UTXO
|
|
assert_eq!(txs[&txid_t].utxos.len(), 1);
|
|
assert_eq!(txs[&txid_t].utxos[0].address, taddr);
|
|
assert_eq!(txs[&txid_t].utxos[0].spent, None);
|
|
assert_eq!(txs[&txid_t].utxos[0].unconfirmed_spent, Some(sent_txid));
|
|
|
|
// Note
|
|
assert_eq!(txs[&txid1].notes[0].note.value, AMOUNT_Z);
|
|
assert_eq!(txs[&txid1].notes[0].spent, None);
|
|
assert_eq!(txs[&txid1].notes[0].unconfirmed_spent, Some(sent_txid));
|
|
}
|
|
|
|
let mut cb3 = FakeCompactBlock::new(2, block_hash);
|
|
cb3.add_tx(&sent_tx);
|
|
|
|
// Scan the compact block and the full Tx
|
|
wallet.scan_block(&cb3.as_bytes()).unwrap();
|
|
wallet.scan_full_tx(&sent_tx, 2, 0);
|
|
|
|
// Now this new Spent tx should be in, so the note should be marked confirmed spent
|
|
{
|
|
let txs = wallet.txs.read().unwrap();
|
|
assert_eq!(txs[&txid1].notes.len(), 1);
|
|
assert_eq!(txs[&txid1].notes[0].note.value, AMOUNT_Z);
|
|
assert_eq!(txs[&txid1].notes[0].spent, Some(sent_txid));
|
|
assert_eq!(txs[&txid1].notes[0].unconfirmed_spent, None);
|
|
|
|
// The UTXO should also be spent
|
|
assert_eq!(txs[&txid_t].utxos[0].address, taddr);
|
|
assert_eq!(txs[&txid_t].utxos[0].spent, Some(sent_txid));
|
|
assert_eq!(txs[&txid_t].utxos[0].unconfirmed_spent, None);
|
|
|
|
// The sent tx should generate change
|
|
assert_eq!(txs[&sent_txid].notes.len(), 1);
|
|
assert_eq!(txs[&sent_txid].notes[0].note.value, AMOUNT_Z + AMOUNT_T - AMOUNT_SENT - fee);
|
|
assert_eq!(txs[&sent_txid].notes[0].is_change, true);
|
|
assert_eq!(txs[&sent_txid].notes[0].spent, None);
|
|
assert_eq!(txs[&sent_txid].notes[0].unconfirmed_spent, None);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_z_incoming_memo() {
|
|
const AMOUNT1: u64 = 50000;
|
|
let (wallet, _txid1, block_hash) = get_test_wallet(AMOUNT1);
|
|
|
|
let my_address = encode_payment_address(wallet.config.hrp_sapling_address(),
|
|
&wallet.extfvks.read().unwrap()[0].default_address().unwrap().1);
|
|
|
|
let memo = "Incoming Memo".to_string();
|
|
let fee: u64 = DEFAULT_FEE.try_into().unwrap();
|
|
|
|
let branch_id = u32::from_str_radix("76b809bb", 16).unwrap();
|
|
let (ss, so) = get_sapling_params().unwrap();
|
|
|
|
// Create a tx and send to address
|
|
let raw_tx = wallet.send_to_address(branch_id, &ss, &so,
|
|
vec![(&my_address, AMOUNT1 - fee, Some(memo.clone()))]).unwrap();
|
|
let sent_tx = Transaction::read(&raw_tx[..]).unwrap();
|
|
let sent_txid = sent_tx.txid();
|
|
|
|
// Add it to a block
|
|
let mut cb3 = FakeCompactBlock::new(2, block_hash);
|
|
cb3.add_tx(&sent_tx);
|
|
wallet.scan_block(&cb3.as_bytes()).unwrap();
|
|
|
|
// And scan the Full Tx to get the memo
|
|
wallet.scan_full_tx(&sent_tx, 2, 0);
|
|
|
|
{
|
|
let txs = wallet.txs.read().unwrap();
|
|
|
|
assert_eq!(txs[&sent_txid].notes.len(), 1);
|
|
|
|
assert_eq!(txs[&sent_txid].notes[0].extfvk, wallet.extfvks.read().unwrap()[0]);
|
|
assert_eq!(txs[&sent_txid].notes[0].note.value, AMOUNT1 - fee);
|
|
assert_eq!(LightWallet::note_address(wallet.config.hrp_sapling_address(), &txs[&sent_txid].notes[0]), Some(my_address));
|
|
assert_eq!(LightWallet::memo_str(&txs[&sent_txid].notes[0].memo), Some(memo));
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_add_new_zt_hd_after_incoming() {
|
|
// When an address recieves funds, a new, unused address should automatically get added
|
|
const AMOUNT1: u64 = 50000;
|
|
let (wallet, _txid1, block_hash) = get_test_wallet(AMOUNT1);
|
|
|
|
// Get the last address
|
|
let my_address = encode_payment_address(wallet.config.hrp_sapling_address(),
|
|
&wallet.extfvks.read().unwrap().last().unwrap().default_address().unwrap().1);
|
|
|
|
let fee: u64 = DEFAULT_FEE.try_into().unwrap();
|
|
|
|
let branch_id = u32::from_str_radix("2bb40e60", 16).unwrap();
|
|
let (ss, so) = get_sapling_params().unwrap();
|
|
|
|
assert_eq!(wallet.zaddress.read().unwrap().len(), 6); // Starts with 1+5 addresses
|
|
|
|
// Create a tx and send to the last address
|
|
let raw_tx = wallet.send_to_address(branch_id, &ss, &so,
|
|
vec![(&my_address, AMOUNT1 - fee, None)]).unwrap();
|
|
let sent_tx = Transaction::read(&raw_tx[..]).unwrap();
|
|
|
|
// Add it to a block
|
|
let mut cb3 = FakeCompactBlock::new(2, block_hash);
|
|
cb3.add_tx(&sent_tx);
|
|
wallet.scan_block(&cb3.as_bytes()).unwrap();
|
|
|
|
// NOw, 5 new addresses should be created
|
|
assert_eq!(wallet.zaddress.read().unwrap().len(), 6+5);
|
|
|
|
let mut rng = OsRng;
|
|
let secp = Secp256k1::new();
|
|
// Send a fake transaction to the last taddr
|
|
let pk = PublicKey::from_secret_key(&secp, &wallet.tkeys.read().unwrap().last().unwrap());
|
|
|
|
// Start with 1 taddr
|
|
assert_eq!(wallet.taddresses.read().unwrap().len(), 1);
|
|
|
|
// Send a Tx to the last address
|
|
let mut tx = FakeTransaction::new(&mut rng);
|
|
tx.add_t_output(&pk, AMOUNT1);
|
|
wallet.scan_full_tx(&tx.get_tx(), 3, 0);
|
|
|
|
// Now, 5 new addresses should be created.
|
|
assert_eq!(wallet.taddresses.read().unwrap().len(), 1+5);
|
|
}
|
|
|
|
#[test]
|
|
fn test_z_to_t_withinwallet() {
|
|
const AMOUNT: u64 = 500000;
|
|
const AMOUNT_SENT: u64 = 20000;
|
|
let (wallet, txid1, block_hash) = get_test_wallet(AMOUNT);
|
|
|
|
let taddr = wallet.address_from_sk(&wallet.tkeys.read().unwrap()[0]);
|
|
|
|
let fee: u64 = DEFAULT_FEE.try_into().unwrap();
|
|
|
|
let branch_id = u32::from_str_radix("76b809bb", 16).unwrap();
|
|
let (ss, so) = get_sapling_params().unwrap();
|
|
|
|
// Create a tx and send to address
|
|
let raw_tx = wallet.send_to_address(branch_id, &ss, &so,
|
|
vec![(&taddr, AMOUNT_SENT, None)]).unwrap();
|
|
let sent_tx = Transaction::read(&raw_tx[..]).unwrap();
|
|
let sent_txid = sent_tx.txid();
|
|
|
|
// Add it to a block
|
|
let mut cb3 = FakeCompactBlock::new(2, block_hash);
|
|
cb3.add_tx(&sent_tx);
|
|
wallet.scan_block(&cb3.as_bytes()).unwrap();
|
|
|
|
// And scan the Full Tx to get the memo
|
|
wallet.scan_full_tx(&sent_tx, 2, 0);
|
|
|
|
{
|
|
let txs = wallet.txs.read().unwrap();
|
|
|
|
// We have the original note
|
|
assert_eq!(txs[&txid1].notes.len(), 1);
|
|
assert_eq!(txs[&txid1].notes[0].note.value, AMOUNT);
|
|
|
|
// We have the spent tx
|
|
assert_eq!(txs[&sent_txid].notes.len(), 1);
|
|
assert_eq!(txs[&sent_txid].notes[0].note.value, AMOUNT - AMOUNT_SENT - fee);
|
|
assert_eq!(txs[&sent_txid].notes[0].is_change, true);
|
|
assert_eq!(txs[&sent_txid].notes[0].spent, None);
|
|
assert_eq!(txs[&sent_txid].notes[0].unconfirmed_spent, None);
|
|
|
|
// Since we sent the Tx to ourself, there should be no outgoing
|
|
// metadata
|
|
assert_eq!(txs[&sent_txid].total_shielded_value_spent, AMOUNT);
|
|
assert_eq!(txs[&sent_txid].outgoing_metadata.len(), 0);
|
|
|
|
|
|
// We have the taddr utxo in the same Tx
|
|
assert_eq!(txs[&sent_txid].utxos.len(), 1);
|
|
assert_eq!(txs[&sent_txid].utxos[0].address, taddr);
|
|
assert_eq!(txs[&sent_txid].utxos[0].value, AMOUNT_SENT);
|
|
assert_eq!(txs[&sent_txid].utxos[0].spent, None);
|
|
assert_eq!(txs[&sent_txid].utxos[0].unconfirmed_spent, None);
|
|
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_multi_t() {
|
|
const AMOUNT: u64 = 5000000;
|
|
const AMOUNT_SENT1: u64 = 20000;
|
|
const AMOUNT_SENT2: u64 = 10000;
|
|
|
|
let (wallet, txid1, block_hash) = get_test_wallet(AMOUNT);
|
|
|
|
// Add a new taddr
|
|
let taddr2 = wallet.add_taddr();
|
|
|
|
let fee: u64 = DEFAULT_FEE.try_into().unwrap();
|
|
|
|
let branch_id = u32::from_str_radix("76b809bb", 16).unwrap();
|
|
let (ss, so) = get_sapling_params().unwrap();
|
|
|
|
// Create a Tx and send to the second t address
|
|
let raw_tx = wallet.send_to_address(branch_id, &ss, &so,
|
|
vec![(&taddr2, AMOUNT_SENT1, None)]).unwrap();
|
|
let sent_tx = Transaction::read(&raw_tx[..]).unwrap();
|
|
let sent_txid1 = sent_tx.txid();
|
|
|
|
// Add it to a block
|
|
let mut cb3 = FakeCompactBlock::new(2, block_hash);
|
|
cb3.add_tx(&sent_tx);
|
|
wallet.scan_block(&cb3.as_bytes()).unwrap();
|
|
wallet.scan_full_tx(&sent_tx, 2, 0);
|
|
|
|
// Check that the send to the second taddr worked
|
|
{
|
|
let txs = wallet.txs.read().unwrap();
|
|
|
|
// We have the original note
|
|
assert_eq!(txs[&txid1].notes.len(), 1);
|
|
assert_eq!(txs[&txid1].notes[0].note.value, AMOUNT);
|
|
|
|
// We have the spent tx
|
|
assert_eq!(txs[&sent_txid1].notes.len(), 1);
|
|
assert_eq!(txs[&sent_txid1].notes[0].note.value, AMOUNT - AMOUNT_SENT1 - fee);
|
|
assert_eq!(txs[&sent_txid1].notes[0].is_change, true);
|
|
assert_eq!(txs[&sent_txid1].notes[0].spent, None);
|
|
assert_eq!(txs[&sent_txid1].notes[0].unconfirmed_spent, None);
|
|
|
|
// Since we sent the Tx to ourself, there should be no outgoing
|
|
// metadata
|
|
assert_eq!(txs[&sent_txid1].total_shielded_value_spent, AMOUNT);
|
|
assert_eq!(txs[&sent_txid1].outgoing_metadata.len(), 0);
|
|
|
|
|
|
// We have the taddr utxo in the same Tx
|
|
assert_eq!(txs[&sent_txid1].utxos.len(), 1);
|
|
assert_eq!(txs[&sent_txid1].utxos[0].address, taddr2);
|
|
assert_eq!(txs[&sent_txid1].utxos[0].value, AMOUNT_SENT1);
|
|
assert_eq!(txs[&sent_txid1].utxos[0].spent, None);
|
|
assert_eq!(txs[&sent_txid1].utxos[0].unconfirmed_spent, None);
|
|
}
|
|
|
|
// Send some money to the 3rd t addr
|
|
let taddr3 = wallet.add_taddr();
|
|
|
|
// Create a Tx and send to the second t address
|
|
let raw_tx = wallet.send_to_address(branch_id, &ss, &so,
|
|
vec![(&taddr3, AMOUNT_SENT2, None)]).unwrap();
|
|
let sent_tx = Transaction::read(&raw_tx[..]).unwrap();
|
|
let sent_txid2 = sent_tx.txid();
|
|
|
|
// Add it to a block
|
|
let mut cb4 = FakeCompactBlock::new(3, cb3.hash());
|
|
cb4.add_tx(&sent_tx);
|
|
wallet.scan_block(&cb4.as_bytes()).unwrap();
|
|
wallet.scan_full_tx(&sent_tx, 3, 0);
|
|
|
|
// Quickly check we have it
|
|
{
|
|
let txs = wallet.txs.read().unwrap();
|
|
|
|
// We have the taddr utxo in the same Tx
|
|
assert_eq!(txs[&sent_txid2].utxos.len(), 1);
|
|
assert_eq!(txs[&sent_txid2].utxos[0].address, taddr3);
|
|
assert_eq!(txs[&sent_txid2].utxos[0].value, AMOUNT_SENT2);
|
|
|
|
// Old UTXO was spent here
|
|
assert_eq!(txs[&sent_txid1].utxos.len(), 1);
|
|
assert_eq!(txs[&sent_txid1].utxos[0].value, AMOUNT_SENT1);
|
|
assert_eq!(txs[&sent_txid1].utxos[0].address, taddr2);
|
|
assert_eq!(txs[&sent_txid1].utxos[0].spent, Some(sent_txid2));
|
|
assert_eq!(txs[&sent_txid1].utxos[0].unconfirmed_spent, None);
|
|
}
|
|
|
|
// Now, spend to an external z address, which will select all the utxos
|
|
let fvk = ExtendedFullViewingKey::from(&ExtendedSpendingKey::master(&[1u8; 32]));
|
|
let ext_address = encode_payment_address(wallet.config.hrp_sapling_address(),
|
|
&fvk.default_address().unwrap().1);
|
|
|
|
const AMOUNT_SENT_EXT: u64 = 45;
|
|
let outgoing_memo = "Outgoing Memo".to_string();
|
|
|
|
// Create a tx and send to address
|
|
let raw_tx = wallet.send_to_address(branch_id, &ss, &so,
|
|
vec![(&ext_address, AMOUNT_SENT_EXT, Some(outgoing_memo.clone()))]).unwrap();
|
|
|
|
let sent_tx = Transaction::read(&raw_tx[..]).unwrap();
|
|
let sent_txid3 = sent_tx.txid();
|
|
|
|
let mut cb5 = FakeCompactBlock::new(4, cb4.hash());
|
|
cb5.add_tx(&sent_tx);
|
|
wallet.scan_block(&cb5.as_bytes()).unwrap();
|
|
wallet.scan_full_tx(&sent_tx, 4, 0);
|
|
|
|
{
|
|
let txs = wallet.txs.read().unwrap();
|
|
assert_eq!(txs[&sent_txid3].outgoing_metadata.len(), 1);
|
|
|
|
assert_eq!(txs[&sent_txid3].outgoing_metadata[0].address, ext_address);
|
|
assert_eq!(txs[&sent_txid3].outgoing_metadata[0].value, AMOUNT_SENT_EXT);
|
|
assert_eq!(txs[&sent_txid3].outgoing_metadata[0].memo.to_utf8().unwrap().unwrap(), outgoing_memo);
|
|
|
|
// Test to see that the UTXOs were spent.
|
|
|
|
// UTXO2
|
|
assert_eq!(txs[&sent_txid2].utxos[0].value, AMOUNT_SENT2);
|
|
assert_eq!(txs[&sent_txid2].utxos[0].address, taddr3);
|
|
assert_eq!(txs[&sent_txid2].utxos[0].spent, Some(sent_txid3));
|
|
assert_eq!(txs[&sent_txid2].utxos[0].unconfirmed_spent, None);
|
|
}
|
|
|
|
}
|
|
|
|
#[test]
|
|
fn test_multi_spends() {
|
|
const AMOUNT1: u64 = 50000;
|
|
let (wallet, txid1, block_hash) = get_test_wallet(AMOUNT1);
|
|
|
|
let zaddr2 = wallet.add_zaddr(); // Address number 6
|
|
const ZAMOUNT2:u64 = 30;
|
|
let outgoing_memo2 = "Outgoing Memo2".to_string();
|
|
|
|
let zaddr3 = wallet.add_zaddr(); // Address number 7
|
|
const ZAMOUNT3:u64 = 40;
|
|
let outgoing_memo3 = "Outgoing Memo3".to_string();
|
|
|
|
let taddr2 = wallet.add_taddr();
|
|
const TAMOUNT2:u64 = 50;
|
|
let taddr3 = wallet.add_taddr();
|
|
const TAMOUNT3:u64 = 60;
|
|
|
|
let fee: u64 = DEFAULT_FEE.try_into().unwrap();
|
|
|
|
let branch_id = u32::from_str_radix("76b809bb", 16).unwrap();
|
|
let (ss, so) = get_sapling_params().unwrap();
|
|
|
|
let tos = vec![ (zaddr2.as_str(), ZAMOUNT2, Some(outgoing_memo2.clone())),
|
|
(zaddr3.as_str(), ZAMOUNT3, Some(outgoing_memo3.clone())),
|
|
(taddr2.as_str(), TAMOUNT2, None),
|
|
(taddr3.as_str(), TAMOUNT3, None) ];
|
|
|
|
let raw_tx = wallet.send_to_address(branch_id, &ss, &so, tos).unwrap();
|
|
let sent_tx = Transaction::read(&raw_tx[..]).unwrap();
|
|
let sent_txid = sent_tx.txid();
|
|
|
|
let mut cb3 = FakeCompactBlock::new(2, block_hash);
|
|
cb3.add_tx(&sent_tx);
|
|
wallet.scan_block(&cb3.as_bytes()).unwrap();
|
|
wallet.scan_full_tx(&sent_tx, 2, 0);
|
|
|
|
// Make sure all the outputs are there!
|
|
{
|
|
let txs = wallet.txs.read().unwrap();
|
|
|
|
// The single note was spent
|
|
assert_eq!(txs[&txid1].notes[0].note.value, AMOUNT1);
|
|
assert_eq!(txs[&txid1].notes[0].spent, Some(sent_txid));
|
|
|
|
// The outputs are all sent to the wallet, so they should
|
|
// correspond to notes & utxos.
|
|
// 2 notes + 1 change
|
|
assert_eq!(txs[&sent_txid].notes.len(), 3);
|
|
|
|
// Find the change note
|
|
let change_note = txs[&sent_txid].notes.iter().find(|n| n.is_change).unwrap();
|
|
assert_eq!(change_note.note.value, AMOUNT1 - (ZAMOUNT2+ZAMOUNT3+TAMOUNT2+TAMOUNT3+fee));
|
|
assert_eq!(change_note.spent, None);
|
|
assert_eq!(change_note.unconfirmed_spent, None);
|
|
|
|
// Find zaddr2
|
|
let zaddr2_note = txs[&sent_txid].notes.iter().find(|n| n.note.value == ZAMOUNT2).unwrap();
|
|
assert_eq!(zaddr2_note.account, 6);
|
|
assert_eq!(zaddr2_note.is_change, false);
|
|
assert_eq!(zaddr2_note.spent, None);
|
|
assert_eq!(zaddr2_note.unconfirmed_spent, None);
|
|
assert_eq!(LightWallet::memo_str(&zaddr2_note.memo), Some(outgoing_memo2));
|
|
|
|
// Find zaddr3
|
|
let zaddr3_note = txs[&sent_txid].notes.iter().find(|n| n.note.value == ZAMOUNT3).unwrap();
|
|
assert_eq!(zaddr3_note.account, 7);
|
|
assert_eq!(zaddr3_note.is_change, false);
|
|
assert_eq!(zaddr3_note.spent, None);
|
|
assert_eq!(zaddr3_note.unconfirmed_spent, None);
|
|
assert_eq!(LightWallet::memo_str(&zaddr3_note.memo), Some(outgoing_memo3));
|
|
|
|
// Find taddr2
|
|
let utxo2 = txs[&sent_txid].utxos.iter().find(|u| u.value == TAMOUNT2).unwrap();
|
|
assert_eq!(utxo2.address, taddr2);
|
|
assert_eq!(utxo2.txid, sent_txid);
|
|
assert_eq!(utxo2.spent, None);
|
|
assert_eq!(utxo2.unconfirmed_spent, None);
|
|
|
|
// Find taddr3
|
|
let utxo3 = txs[&sent_txid].utxos.iter().find(|u| u.value == TAMOUNT3).unwrap();
|
|
assert_eq!(utxo3.address, taddr3);
|
|
assert_eq!(utxo3.txid, sent_txid);
|
|
assert_eq!(utxo3.spent, None);
|
|
assert_eq!(utxo3.unconfirmed_spent, None);
|
|
}
|
|
|
|
// Now send an outgoing tx to one ext taddr and one ext zaddr
|
|
let fvk = ExtendedFullViewingKey::from(&ExtendedSpendingKey::master(&[1u8; 32]));
|
|
let ext_address = encode_payment_address(wallet.config.hrp_sapling_address(),
|
|
&fvk.default_address().unwrap().1);
|
|
let ext_memo = "External memo".to_string();
|
|
let ext_taddr = wallet.address_from_sk(&SecretKey::from_slice(&[1u8; 32]).unwrap());
|
|
|
|
const EXT_ZADDR_AMOUNT: u64 = 3000;
|
|
let ext_taddr_amount = AMOUNT1 - fee - EXT_ZADDR_AMOUNT - fee; // Spend everything
|
|
println!("taddr amount {}", ext_taddr_amount);
|
|
|
|
let tos = vec![ (ext_address.as_str(), EXT_ZADDR_AMOUNT, Some(ext_memo.clone())),
|
|
(ext_taddr.as_str(), ext_taddr_amount, None)];
|
|
let raw_tx = wallet.send_to_address(branch_id, &ss, &so, tos).unwrap();
|
|
let sent_tx = Transaction::read(&raw_tx[..]).unwrap();
|
|
let sent_txid2 = sent_tx.txid();
|
|
|
|
let mut cb4 = FakeCompactBlock::new(3, cb3.hash());
|
|
cb4.add_tx(&sent_tx);
|
|
wallet.scan_block(&cb4.as_bytes()).unwrap();
|
|
wallet.scan_full_tx(&sent_tx, 3, 0);
|
|
|
|
// Make sure all the outputs are there!
|
|
{
|
|
let txs = wallet.txs.read().unwrap();
|
|
|
|
// All notes were spent
|
|
assert_eq!(txs[&sent_txid].notes[0].spent, Some(sent_txid2));
|
|
assert_eq!(txs[&sent_txid].notes[1].spent, Some(sent_txid2));
|
|
assert_eq!(txs[&sent_txid].notes[2].spent, Some(sent_txid2));
|
|
|
|
// All utxos were spent
|
|
assert_eq!(txs[&sent_txid].utxos[0].spent, Some(sent_txid2));
|
|
assert_eq!(txs[&sent_txid].utxos[1].spent, Some(sent_txid2));
|
|
|
|
// The new tx has no change
|
|
assert_eq!(txs[&sent_txid2].notes.len(), 0);
|
|
assert_eq!(txs[&sent_txid2].utxos.len(), 0);
|
|
|
|
// Test the outgoing metadata
|
|
// Find the znote
|
|
let zoutgoing = txs[&sent_txid2].outgoing_metadata.iter().find(|o| o.address == ext_address).unwrap();
|
|
assert_eq!(zoutgoing.value, EXT_ZADDR_AMOUNT);
|
|
assert_eq!(LightWallet::memo_str(&Some(zoutgoing.memo.clone())), Some(ext_memo));
|
|
|
|
// Find the taddr
|
|
let toutgoing = txs[&sent_txid2].outgoing_metadata.iter().find(|o| o.address == ext_taddr).unwrap();
|
|
assert_eq!(toutgoing.value, ext_taddr_amount);
|
|
assert_eq!(LightWallet::memo_str(&Some(toutgoing.memo.clone())), None);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_bad_send() {
|
|
// Test all the ways in which a send should fail
|
|
const AMOUNT1: u64 = 50000;
|
|
let _fee: u64 = DEFAULT_FEE.try_into().unwrap();
|
|
|
|
let (wallet, _txid1, _block_hash) = get_test_wallet(AMOUNT1);
|
|
|
|
let branch_id = u32::from_str_radix("76b809bb", 16).unwrap();
|
|
let (ss, so) = get_sapling_params().unwrap();
|
|
let ext_taddr = wallet.address_from_sk(&SecretKey::from_slice(&[1u8; 32]).unwrap());
|
|
|
|
// Bad address
|
|
let raw_tx = wallet.send_to_address(branch_id, &ss, &so,
|
|
vec![(&"badaddress", 10, None)]);
|
|
assert!(raw_tx.err().unwrap().contains("Invalid recipient address"));
|
|
|
|
// Insufficient funds
|
|
let raw_tx = wallet.send_to_address(branch_id, &ss, &so,
|
|
vec![(&ext_taddr, AMOUNT1 + 10, None)]);
|
|
assert!(raw_tx.err().unwrap().contains("Insufficient verified funds"));
|
|
|
|
// No addresses
|
|
let raw_tx = wallet.send_to_address(branch_id, &ss, &so, vec![]);
|
|
assert!(raw_tx.err().unwrap().contains("at least one"));
|
|
|
|
}
|
|
|
|
#[test]
|
|
fn test_duplicate_outputs() {
|
|
// Test all the ways in which a send should fail
|
|
const AMOUNT1: u64 = 50000;
|
|
let _fee: u64 = DEFAULT_FEE.try_into().unwrap();
|
|
|
|
let (wallet, _txid1, _block_hash) = get_test_wallet(AMOUNT1);
|
|
|
|
let branch_id = u32::from_str_radix("2bb40e60", 16).unwrap();
|
|
let (ss, so) = get_sapling_params().unwrap();
|
|
let ext_taddr = wallet.address_from_sk(&SecretKey::from_slice(&[1u8; 32]).unwrap());
|
|
|
|
// Duplicated addresses with memos are fine too
|
|
let raw_tx = wallet.send_to_address(branch_id, &ss, &so,
|
|
vec![(&ext_taddr, 100, Some("First memo".to_string())),
|
|
(&ext_taddr, 0, Some("Second memo".to_string())),
|
|
(&ext_taddr, 0, Some("Third memo".to_string()))]);
|
|
assert!(raw_tx.is_ok());
|
|
}
|
|
|
|
#[test]
|
|
#[should_panic]
|
|
fn test_bad_params() {
|
|
let (wallet, _, _) = get_test_wallet(100000);
|
|
let ext_taddr = wallet.address_from_sk(&SecretKey::from_slice(&[1u8; 32]).unwrap());
|
|
|
|
let branch_id = u32::from_str_radix("76b809bb", 16).unwrap();
|
|
// Bad params
|
|
let _ = wallet.send_to_address(branch_id, &[], &[],
|
|
vec![(&ext_taddr, 10, None)]);
|
|
}
|
|
|
|
/// Test helper to add blocks
|
|
fn add_blocks(wallet: &LightWallet, start: i32, num: i32, mut prev_hash: BlockHash) -> Result<BlockHash, i32>{
|
|
// Add it to a block
|
|
let mut new_blk = FakeCompactBlock::new(start, prev_hash);
|
|
for i in 0..num {
|
|
new_blk = FakeCompactBlock::new(start+i, prev_hash);
|
|
prev_hash = new_blk.hash();
|
|
match wallet.scan_block(&new_blk.as_bytes()) {
|
|
Ok(_) => {}, // continue
|
|
Err(e) => return Err(e)
|
|
};
|
|
}
|
|
|
|
Ok(new_blk.hash())
|
|
}
|
|
|
|
#[test]
|
|
fn test_z_mempool_expiry() {
|
|
const AMOUNT1: u64 = 50000;
|
|
let (wallet, _, block_hash) = get_test_wallet(AMOUNT1);
|
|
|
|
let fvk = ExtendedFullViewingKey::from(&ExtendedSpendingKey::master(&[1u8; 32]));
|
|
let ext_address = encode_payment_address(wallet.config.hrp_sapling_address(),
|
|
&fvk.default_address().unwrap().1);
|
|
|
|
const AMOUNT_SENT: u64 = 20;
|
|
|
|
let outgoing_memo = "Outgoing Memo".to_string();
|
|
|
|
let branch_id = u32::from_str_radix("76b809bb", 16).unwrap();
|
|
let (ss, so) = get_sapling_params().unwrap();
|
|
|
|
// Create a tx and send to address
|
|
let raw_tx = wallet.send_to_address(branch_id, &ss, &so,
|
|
vec![(&ext_address, AMOUNT_SENT, Some(outgoing_memo.clone()))]).unwrap();
|
|
|
|
let sent_tx = Transaction::read(&raw_tx[..]).unwrap();
|
|
let sent_txid = sent_tx.txid();
|
|
|
|
// It should also be in the mempool structure
|
|
{
|
|
let mem = wallet.mempool_txs.read().unwrap();
|
|
|
|
assert_eq!(mem[&sent_txid].block, 2); // block number is next block
|
|
assert! (mem[&sent_txid].datetime > 0);
|
|
assert_eq!(mem[&sent_txid].txid, sent_txid);
|
|
assert_eq!(mem[&sent_txid].outgoing_metadata.len(), 1);
|
|
assert_eq!(mem[&sent_txid].outgoing_metadata[0].address, ext_address);
|
|
assert_eq!(mem[&sent_txid].outgoing_metadata[0].value, AMOUNT_SENT);
|
|
assert_eq!(mem[&sent_txid].outgoing_metadata[0].memo.to_utf8().unwrap().unwrap(), outgoing_memo);
|
|
}
|
|
|
|
// Don't mine the Tx, but just add several blocks
|
|
add_blocks(&wallet, 2, 21, block_hash).unwrap();
|
|
|
|
// After 21 blocks, it should disappear (expiry is 20 blocks) since it was not mined
|
|
{
|
|
let mem = wallet.mempool_txs.read().unwrap();
|
|
|
|
assert!(mem.get(&sent_txid).is_none());
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_block_limit() {
|
|
const AMOUNT: u64 = 500000;
|
|
let (wallet, _txid1, block_hash) = get_test_wallet(AMOUNT);
|
|
|
|
let prev_hash = add_blocks(&wallet, 2, 1, block_hash).unwrap();
|
|
assert_eq!(wallet.blocks.read().unwrap().len(), 3);
|
|
|
|
let prev_hash = add_blocks(&wallet, 3, 47, prev_hash).unwrap();
|
|
assert_eq!(wallet.blocks.read().unwrap().len(), 50);
|
|
|
|
let prev_hash = add_blocks(&wallet, 50, 51, prev_hash).unwrap();
|
|
assert_eq!(wallet.blocks.read().unwrap().len(), 101);
|
|
|
|
// Subsequent blocks should start to trim
|
|
let prev_hash = add_blocks(&wallet, 101, 1, prev_hash).unwrap();
|
|
assert_eq!(wallet.blocks.read().unwrap().len(), 101);
|
|
|
|
// Add lots
|
|
let _ = add_blocks(&wallet, 102, 10, prev_hash).unwrap();
|
|
assert_eq!(wallet.blocks.read().unwrap().len(), 101);
|
|
|
|
// Now clear the blocks
|
|
wallet.clear_blocks();
|
|
assert_eq!(wallet.blocks.read().unwrap().len(), 0);
|
|
|
|
let prev_hash = add_blocks(&wallet, 0, 1, BlockHash([0;32])).unwrap();
|
|
assert_eq!(wallet.blocks.read().unwrap().len(), 1);
|
|
|
|
let _ = add_blocks(&wallet, 1, 10, prev_hash).unwrap();
|
|
assert_eq!(wallet.blocks.read().unwrap().len(), 11);
|
|
}
|
|
|
|
#[test]
|
|
fn test_rollback() {
|
|
const AMOUNT: u64 = 500000;
|
|
|
|
let (wallet, txid1, block_hash) = get_test_wallet(AMOUNT);
|
|
|
|
add_blocks(&wallet, 2, 5, block_hash).unwrap();
|
|
|
|
// Make sure the note exists with the witnesses
|
|
{
|
|
let txs = wallet.txs.read().unwrap();
|
|
assert_eq!(txs[&txid1].notes[0].witnesses.len(), 7);
|
|
}
|
|
|
|
// Invalidate 2 blocks
|
|
assert_eq!(wallet.last_scanned_height(), 6);
|
|
assert_eq!(wallet.invalidate_block(5), 2);
|
|
|
|
// THe witnesses should be rolledback
|
|
{
|
|
let txs = wallet.txs.read().unwrap();
|
|
assert_eq!(txs[&txid1].notes[0].witnesses.len(), 5);
|
|
}
|
|
|
|
let blk3_hash;
|
|
let blk4_hash;
|
|
{
|
|
let blks = wallet.blocks.read().unwrap();
|
|
blk3_hash = blks[3].hash.clone();
|
|
blk4_hash = blks[4].hash.clone();
|
|
}
|
|
|
|
// This should result in an exception, because the "prevhash" is wrong
|
|
assert!(add_blocks(&wallet, 5, 2, blk3_hash).is_err(),
|
|
"Shouldn't be able to add because of invalid prev hash");
|
|
|
|
// Add with the proper prev hash
|
|
add_blocks(&wallet, 5, 2, blk4_hash).unwrap();
|
|
|
|
let blk6_hash;
|
|
{
|
|
let blks = wallet.blocks.read().unwrap();
|
|
blk6_hash = blks[6].hash.clone();
|
|
}
|
|
|
|
// Now do a Tx
|
|
let taddr = wallet.address_from_sk(&SecretKey::from_slice(&[1u8; 32]).unwrap());
|
|
|
|
let branch_id = u32::from_str_radix("76b809bb", 16).unwrap();
|
|
let (ss, so) = get_sapling_params().unwrap();
|
|
|
|
// Create a tx and send to address
|
|
const AMOUNT_SENT: u64 = 30000;
|
|
let fee: u64 = DEFAULT_FEE.try_into().unwrap();
|
|
let raw_tx = wallet.send_to_address(branch_id, &ss, &so,
|
|
vec![(&taddr, AMOUNT_SENT, None)]).unwrap();
|
|
|
|
let sent_tx = Transaction::read(&raw_tx[..]).unwrap();
|
|
let sent_txid = sent_tx.txid();
|
|
let mut cb3 = FakeCompactBlock::new(7, blk6_hash);
|
|
cb3.add_tx(&sent_tx);
|
|
wallet.scan_block(&cb3.as_bytes()).unwrap();
|
|
wallet.scan_full_tx(&sent_tx, 7, 0);
|
|
|
|
// Make sure the Tx is in.
|
|
{
|
|
let txs = wallet.txs.read().unwrap();
|
|
assert_eq!(txs[&txid1].notes.len(), 1);
|
|
assert_eq!(txs[&txid1].notes[0].note.value, AMOUNT);
|
|
assert_eq!(txs[&txid1].notes[0].spent, Some(sent_txid));
|
|
assert_eq!(txs[&txid1].notes[0].unconfirmed_spent, None);
|
|
|
|
// The sent tx should generate change
|
|
assert_eq!(txs[&sent_txid].notes.len(), 1);
|
|
assert_eq!(txs[&sent_txid].notes[0].note.value, AMOUNT - AMOUNT_SENT - fee);
|
|
assert_eq!(txs[&sent_txid].notes[0].is_change, true);
|
|
assert_eq!(txs[&sent_txid].notes[0].spent, None);
|
|
assert_eq!(txs[&sent_txid].notes[0].unconfirmed_spent, None);
|
|
assert_eq!(txs[&sent_txid].notes[0].witnesses.len(), 1);
|
|
}
|
|
|
|
// Invalidate 3 blocks
|
|
assert_eq!(wallet.last_scanned_height(), 7);
|
|
assert_eq!(wallet.invalidate_block(5), 3);
|
|
assert_eq!(wallet.last_scanned_height(), 4);
|
|
|
|
// Make sure the orig Tx is there, but new Tx has disappeared
|
|
{
|
|
let txs = wallet.txs.read().unwrap();
|
|
|
|
// Orig Tx is still there, since this is in block 0
|
|
// But now the spent tx is gone
|
|
assert_eq!(txs[&txid1].notes.len(), 1);
|
|
assert_eq!(txs[&txid1].notes[0].note.value, AMOUNT);
|
|
assert_eq!(txs[&txid1].notes[0].spent, None);
|
|
assert_eq!(txs[&txid1].notes[0].unconfirmed_spent, None);
|
|
|
|
// The sent tx is missing
|
|
assert!(txs.get(&sent_txid).is_none());
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_t_derivation() {
|
|
let lc = LightClientConfig {
|
|
server: "0.0.0.0:0".parse().unwrap(),
|
|
chain_name: "main".to_string(),
|
|
sapling_activation_height: 0,
|
|
consensus_branch_id: "000000".to_string(),
|
|
anchor_offset: 1,
|
|
no_cert_verification: false,
|
|
data_dir: None,
|
|
};
|
|
|
|
let seed_phrase = Some("chimney better bulb horror rebuild whisper improve intact letter giraffe brave rib appear bulk aim burst snap salt hill sad merge tennis phrase raise".to_string());
|
|
|
|
let wallet = LightWallet::new(seed_phrase.clone(), &lc, 0).unwrap();
|
|
|
|
// Test the addresses against https://iancoleman.io/bip39/
|
|
let (taddr, pk) = &wallet.get_t_secret_keys()[0];
|
|
assert_eq!(taddr, "RXBxhYDg8vSsHVmAGadniQKh3NvzAtzjRe");
|
|
assert_eq!(pk, "UvGDnY7bpsyz9GnRPvMRQTMKHPyCK5k6c2FBiGQcgRSe9xVNuGGs");
|
|
|
|
// Test a couple more
|
|
wallet.add_taddr();
|
|
let (taddr, pk) = &wallet.get_t_secret_keys()[1];
|
|
assert_eq!(taddr, "RKDGjDoFHf9BfTFuL27voFdqdV7LhWw9rG");
|
|
assert_eq!(pk, "UqGi6D8rFfdoEtbqhjz1utu1hKLmagY5TBVHWau54pput9HRfYbG");
|
|
|
|
let (zaddr, sk) = &wallet.get_z_private_keys()[0];
|
|
assert_eq!(zaddr, "zs1wp96063hjs496d28e05uz5gavg7mwshhsgglchtan57el88uwa5jfdgk4nd68kda62vgwucfe6z");
|
|
assert_eq!(sk, "secret-extended-key-main1qvkk0dn2qqqqpqrk6fl6c6fzzrmkhlj59d2c6kkz37hmal6d4dm69ne75xf0exuvnyk6qrjgp6crvdtaehkda2edg0llv488u25vjh5jtldnp53nrphqeexel57a0dn4t2kkcr6uj6y832yg8wsx3wx6t6rk470dynzdx3cp37xdwl9mpe59vj6yqh67x09vea9khzk5wdqkt65c6x9qkuht7nxyetthu0pr7jrwpthzq2ncgm0dvczadqxuhhk5ekua5v5zzw2kydcudu965");
|
|
|
|
assert_eq!(seed_phrase, Some(wallet.get_seed_phrase()));
|
|
}
|
|
|
|
#[test]
|
|
fn test_lock_unlock() {
|
|
const AMOUNT: u64 = 500000;
|
|
|
|
let (mut wallet, _, _) = get_test_wallet(AMOUNT);
|
|
let config = wallet.config.clone();
|
|
|
|
// Add some addresses
|
|
let zaddr0 = encode_payment_address(config.hrp_sapling_address(),
|
|
&wallet.extfvks.read().unwrap()[0].default_address().unwrap().1);
|
|
let zaddr1 = wallet.add_zaddr(); // This is actually address at index 6
|
|
let zaddr2 = wallet.add_zaddr(); // This is actually address at index 7
|
|
|
|
let taddr0 = wallet.address_from_sk(&wallet.tkeys.read().unwrap()[0]);
|
|
let taddr1 = wallet.add_taddr();
|
|
let taddr2 = wallet.add_taddr();
|
|
|
|
let seed = wallet.seed;
|
|
|
|
// Trying to lock a wallet that's not encrpyted is an error
|
|
assert!(wallet.lock().is_err());
|
|
|
|
// Encrypt the wallet
|
|
wallet.encrypt("somepassword".to_string()).unwrap();
|
|
|
|
// Encrypting an already encrypted wallet should fail
|
|
assert!(wallet.encrypt("somepassword".to_string()).is_err());
|
|
|
|
// Serialize a locked wallet
|
|
let mut serialized_data = vec![];
|
|
wallet.write(&mut serialized_data).expect("Serialize wallet");
|
|
|
|
// Should fail when there's a wrong password
|
|
assert!(wallet.unlock("differentpassword".to_string()).is_err());
|
|
|
|
// Properly unlock
|
|
wallet.unlock("somepassword".to_string()).unwrap();
|
|
|
|
assert_eq!(seed, wallet.seed);
|
|
{
|
|
let extsks = wallet.extsks.read().unwrap();
|
|
let tkeys = wallet.tkeys.read().unwrap();
|
|
assert_eq!(extsks.len(), 8); // 3 zaddrs + 1 original + 4 extra HD addreses
|
|
assert_eq!(tkeys.len(), 3);
|
|
|
|
assert_eq!(zaddr0, encode_payment_address(config.hrp_sapling_address(),
|
|
&ExtendedFullViewingKey::from(&extsks[0]).default_address().unwrap().1));
|
|
assert_eq!(zaddr1, encode_payment_address(config.hrp_sapling_address(),
|
|
&ExtendedFullViewingKey::from(&extsks[6]).default_address().unwrap().1));
|
|
assert_eq!(zaddr2, encode_payment_address(config.hrp_sapling_address(),
|
|
&ExtendedFullViewingKey::from(&extsks[7]).default_address().unwrap().1));
|
|
|
|
assert_eq!(taddr0, wallet.address_from_sk(&tkeys[0]));
|
|
assert_eq!(taddr1, wallet.address_from_sk(&tkeys[1]));
|
|
assert_eq!(taddr2, wallet.address_from_sk(&tkeys[2]));
|
|
}
|
|
|
|
// Unlocking an already unlocked wallet should fail
|
|
assert!(wallet.unlock("somepassword".to_string()).is_err());
|
|
|
|
// Trying to serialize a encrypted but unlocked wallet should fail
|
|
assert!(wallet.write(&mut vec![]).is_err());
|
|
|
|
// ...but if we lock it again, it should serialize
|
|
wallet.lock().unwrap();
|
|
wallet.write(&mut vec![]).expect("Serialize wallet");
|
|
|
|
// Locking an already locked wallet is an error
|
|
assert!(wallet.lock().is_err());
|
|
|
|
// Try from a deserialized, locked wallet
|
|
let mut wallet2 = LightWallet::read(&serialized_data[..], &config).unwrap();
|
|
wallet2.unlock("somepassword".to_string()).unwrap();
|
|
|
|
assert_eq!(seed, wallet2.seed);
|
|
{
|
|
let extsks = wallet2.extsks.read().unwrap();
|
|
let tkeys = wallet2.tkeys.read().unwrap();
|
|
assert_eq!(extsks.len(), 8);
|
|
assert_eq!(tkeys.len(), 3);
|
|
|
|
assert_eq!(zaddr0, encode_payment_address(wallet2.config.hrp_sapling_address(),
|
|
&ExtendedFullViewingKey::from(&extsks[0]).default_address().unwrap().1));
|
|
assert_eq!(zaddr1, encode_payment_address(wallet2.config.hrp_sapling_address(),
|
|
&ExtendedFullViewingKey::from(&extsks[6]).default_address().unwrap().1));
|
|
assert_eq!(zaddr2, encode_payment_address(wallet2.config.hrp_sapling_address(),
|
|
&ExtendedFullViewingKey::from(&extsks[7]).default_address().unwrap().1));
|
|
|
|
assert_eq!(taddr0, wallet2.address_from_sk(&tkeys[0]));
|
|
assert_eq!(taddr1, wallet2.address_from_sk(&tkeys[1]));
|
|
assert_eq!(taddr2, wallet2.address_from_sk(&tkeys[2]));
|
|
}
|
|
|
|
// Remove encryption from a unlocked wallet should succeed
|
|
wallet2.remove_encryption("somepassword".to_string()).unwrap();
|
|
assert_eq!(seed, wallet2.seed);
|
|
|
|
// Now encrypt with a different password
|
|
wallet2.encrypt("newpassword".to_string()).unwrap();
|
|
assert_eq!([0u8; 32], wallet2.seed); // Seed is cleared out
|
|
|
|
// Locking should fail because it is already locked
|
|
assert!(wallet2.lock().is_err());
|
|
|
|
// The old password shouldn't work
|
|
assert!(wallet2.remove_encryption("somepassword".to_string()).is_err());
|
|
|
|
// Remove encryption with the right password
|
|
wallet2.remove_encryption("newpassword".to_string()).unwrap();
|
|
assert_eq!(seed, wallet2.seed);
|
|
|
|
// Unlocking a wallet without encryption is an error
|
|
assert!(wallet2.remove_encryption("newpassword".to_string()).is_err());
|
|
// Can't lock/unlock a wallet that's not encrypted
|
|
assert!(wallet2.lock().is_err());
|
|
assert!(wallet2.unlock("newpassword".to_string()).is_err());
|
|
}
|
|
|
|
#[test]
|
|
#[should_panic]
|
|
fn test_invalid_bip39_t() {
|
|
// Passing a 32-byte seed to bip32 should fail.
|
|
let config = get_test_config();
|
|
LightWallet::get_taddr_from_bip39seed(&config, &[0u8; 32], 0);
|
|
}
|
|
|
|
#[test]
|
|
#[should_panic]
|
|
fn test_invalid_bip39_z() {
|
|
// Passing a 32-byte seed to bip32 should fail.
|
|
let config = get_test_config();
|
|
LightWallet::get_zaddr_from_bip39seed(&config, &[0u8; 32], 0);
|
|
}
|
|
|
|
#[test]
|
|
fn test_invalid_scan_blocks() {
|
|
const AMOUNT: u64 = 500000;
|
|
let (wallet, _txid1, block_hash) = get_test_wallet(AMOUNT);
|
|
|
|
let prev_hash = add_blocks(&wallet, 2, 1, block_hash).unwrap();
|
|
assert_eq!(wallet.blocks.read().unwrap().len(), 3);
|
|
|
|
// Block fails to scan for bad encoding
|
|
assert_eq!(wallet.scan_block(&[0; 32]), Err(-1));
|
|
|
|
// Block is invalid height
|
|
let new_blk = FakeCompactBlock::new(4, prev_hash);
|
|
assert_eq!(wallet.scan_block(&new_blk.as_bytes()), Err(2));
|
|
|
|
// Block is right height, but invalid prev height (for reorgs)
|
|
let new_blk = FakeCompactBlock::new(2, BlockHash([0; 32]));
|
|
assert_eq!(wallet.scan_block(&new_blk.as_bytes()), Err(2));
|
|
|
|
// Block is right height, but invalid prev height (for reorgs)
|
|
let new_blk = FakeCompactBlock::new(3, BlockHash([0; 32]));
|
|
assert_eq!(wallet.scan_block(&new_blk.as_bytes()), Err(2));
|
|
|
|
// Then the rest add properly
|
|
let _ = add_blocks(&wallet, 3, 2, prev_hash).unwrap();
|
|
assert_eq!(wallet.blocks.read().unwrap().len(), 5);
|
|
}
|
|
|
|
#[test]
|
|
fn test_encrypted_zreceive() {
|
|
const AMOUNT1: u64 = 50000;
|
|
let password: String = "password".to_string();
|
|
|
|
let (mut wallet, txid1, block_hash) = get_test_wallet(AMOUNT1);
|
|
|
|
let fvk = ExtendedFullViewingKey::from(&ExtendedSpendingKey::master(&[1u8; 32]));
|
|
let ext_address = encode_payment_address(wallet.config.hrp_sapling_address(),
|
|
&fvk.default_address().unwrap().1);
|
|
|
|
const AMOUNT_SENT: u64 = 20;
|
|
|
|
let outgoing_memo = "Outgoing Memo".to_string();
|
|
let fee: u64 = DEFAULT_FEE.try_into().unwrap();
|
|
|
|
let branch_id = u32::from_str_radix("76b809bb", 16).unwrap();
|
|
let (ss, so) = get_sapling_params().unwrap();
|
|
|
|
// Create a tx and send to address
|
|
let raw_tx = wallet.send_to_address(branch_id, &ss, &so,
|
|
vec![(&ext_address, AMOUNT_SENT, Some(outgoing_memo.clone()))]).unwrap();
|
|
|
|
// Now that we have the transaction, we'll encrypt the wallet
|
|
wallet.encrypt(password.clone()).unwrap();
|
|
|
|
// Scan the tx and make sure it gets added
|
|
let sent_tx = Transaction::read(&raw_tx[..]).unwrap();
|
|
let sent_txid = sent_tx.txid();
|
|
|
|
let mut cb3 = FakeCompactBlock::new(2, block_hash);
|
|
cb3.add_tx(&sent_tx);
|
|
wallet.scan_block(&cb3.as_bytes()).unwrap();
|
|
|
|
// Now, full scan the Tx, which should populate the Outgoing Meta data
|
|
wallet.scan_full_tx(&sent_tx, 2, 0);
|
|
|
|
// Now this new Spent tx should be in, so the note should be marked confirmed spent
|
|
{
|
|
let txs = wallet.txs.read().unwrap();
|
|
assert_eq!(txs[&txid1].notes.len(), 1);
|
|
assert_eq!(txs[&txid1].notes[0].note.value, AMOUNT1);
|
|
assert_eq!(txs[&txid1].notes[0].spent, Some(sent_txid));
|
|
assert_eq!(txs[&txid1].notes[0].unconfirmed_spent, None);
|
|
|
|
// The sent tx should generate change
|
|
assert_eq!(txs[&sent_txid].notes.len(), 1);
|
|
assert_eq!(txs[&sent_txid].notes[0].note.value, AMOUNT1 - AMOUNT_SENT - fee);
|
|
assert_eq!(txs[&sent_txid].notes[0].is_change, true);
|
|
assert_eq!(txs[&sent_txid].notes[0].spent, None);
|
|
assert_eq!(txs[&sent_txid].notes[0].unconfirmed_spent, None);
|
|
|
|
// Outgoing Metadata
|
|
assert_eq!(txs[&sent_txid].total_shielded_value_spent, AMOUNT1);
|
|
|
|
assert_eq!(txs[&sent_txid].outgoing_metadata.len(), 1);
|
|
|
|
assert_eq!(txs[&sent_txid].outgoing_metadata[0].address, ext_address);
|
|
assert_eq!(txs[&sent_txid].outgoing_metadata[0].value, AMOUNT_SENT);
|
|
assert_eq!(txs[&sent_txid].outgoing_metadata[0].memo.to_utf8().unwrap().unwrap(), outgoing_memo);
|
|
}
|
|
|
|
// Trying to spend from a locked wallet is an error
|
|
assert!(wallet.send_to_address(branch_id, &ss, &so,
|
|
vec![(&ext_address, AMOUNT_SENT, None)]).is_err());
|
|
|
|
// unlock the wallet so we can spend to the second z address
|
|
wallet.unlock(password.clone()).unwrap();
|
|
|
|
// Second z address
|
|
let zaddr2 = wallet.add_zaddr(); // This is address number 6
|
|
const ZAMOUNT2:u64 = 30;
|
|
let outgoing_memo2 = "Outgoing Memo2".to_string();
|
|
|
|
let raw_tx = wallet.send_to_address(branch_id, &ss, &so,
|
|
vec![(&zaddr2, ZAMOUNT2, Some(outgoing_memo2.clone()))]).unwrap();
|
|
|
|
// Now lock the wallet again
|
|
wallet.lock().unwrap();
|
|
|
|
let sent_tx2 = Transaction::read(&raw_tx[..]).unwrap();
|
|
let txid2 = sent_tx2.txid();
|
|
|
|
let mut cb4 = FakeCompactBlock::new(3, cb3.hash());
|
|
cb4.add_tx(&sent_tx2);
|
|
wallet.scan_block(&cb4.as_bytes()).unwrap();
|
|
wallet.scan_full_tx(&sent_tx2, 3, 0);
|
|
|
|
{
|
|
let txs = wallet.txs.read().unwrap();
|
|
let prev_change_value = AMOUNT1 - AMOUNT_SENT - fee;
|
|
|
|
// Change note from prev transaction is spent
|
|
assert_eq!(txs[&sent_txid].notes[0].note.value, prev_change_value);
|
|
assert_eq!(txs[&sent_txid].notes[0].is_change, true);
|
|
assert_eq!(txs[&sent_txid].notes[0].spent, Some(txid2));
|
|
|
|
// New change note. So find it.
|
|
let change_note = txs[&txid2].notes.iter().find(|n| n.is_change).unwrap();
|
|
|
|
// New incoming tx is present
|
|
assert_eq!(change_note.note.value, prev_change_value - (ZAMOUNT2+fee));
|
|
assert_eq!(change_note.spent, None);
|
|
assert_eq!(change_note.unconfirmed_spent, None);
|
|
|
|
// Find zaddr2
|
|
let zaddr2_note = txs[&txid2].notes.iter().find(|n| n.note.value == ZAMOUNT2).unwrap();
|
|
assert_eq!(zaddr2_note.account, 6);
|
|
assert_eq!(zaddr2_note.is_change, false);
|
|
assert_eq!(zaddr2_note.spent, None);
|
|
assert_eq!(zaddr2_note.unconfirmed_spent, None);
|
|
assert_eq!(LightWallet::memo_str(&zaddr2_note.memo), Some(outgoing_memo2));
|
|
}
|
|
}
|
|
|
|
|
|
#[test]
|
|
fn test_encrypted_treceive() {
|
|
const AMOUNT1: u64 = 50000;
|
|
let password: String = "password".to_string();
|
|
let (mut wallet, txid1, block_hash) = get_test_wallet(AMOUNT1);
|
|
|
|
let branch_id = u32::from_str_radix("76b809bb", 16).unwrap();
|
|
let (ss, so) = get_sapling_params().unwrap();
|
|
|
|
let taddr = wallet.address_from_sk(&SecretKey::from_slice(&[1u8; 32]).unwrap());
|
|
const AMOUNT_SENT: u64 = 30;
|
|
let fee: u64 = DEFAULT_FEE.try_into().unwrap();
|
|
|
|
let raw_tx = wallet.send_to_address(branch_id, &ss, &so,
|
|
vec![(&taddr, AMOUNT_SENT, None)]).unwrap();
|
|
|
|
// Now that we have the transaction, we'll encrypt the wallet
|
|
wallet.encrypt(password.clone()).unwrap();
|
|
|
|
let sent_tx = Transaction::read(&raw_tx[..]).unwrap();
|
|
let sent_txid = sent_tx.txid();
|
|
let mut cb3 = FakeCompactBlock::new(2, block_hash);
|
|
cb3.add_tx(&sent_tx);
|
|
wallet.scan_block(&cb3.as_bytes()).unwrap();
|
|
wallet.scan_full_tx(&sent_tx, 2, 0);
|
|
|
|
// Now this new Spent tx should be in, so the note should be marked confirmed spent
|
|
{
|
|
let txs = wallet.txs.read().unwrap();
|
|
assert_eq!(txs[&txid1].notes.len(), 1);
|
|
assert_eq!(txs[&txid1].notes[0].note.value, AMOUNT1);
|
|
assert_eq!(txs[&txid1].notes[0].spent, Some(sent_txid));
|
|
assert_eq!(txs[&txid1].notes[0].unconfirmed_spent, None);
|
|
|
|
// The sent tx should generate change
|
|
assert_eq!(txs[&sent_txid].notes.len(), 1);
|
|
assert_eq!(txs[&sent_txid].notes[0].note.value, AMOUNT1 - AMOUNT_SENT - fee);
|
|
assert_eq!(txs[&sent_txid].notes[0].is_change, true);
|
|
assert_eq!(txs[&sent_txid].notes[0].spent, None);
|
|
assert_eq!(txs[&sent_txid].notes[0].unconfirmed_spent, None);
|
|
|
|
// Outgoing Metadata
|
|
assert_eq!(txs[&sent_txid].outgoing_metadata.len(), 1);
|
|
assert_eq!(txs[&sent_txid].outgoing_metadata[0].address, taddr);
|
|
assert_eq!(txs[&sent_txid].outgoing_metadata[0].value, AMOUNT_SENT);
|
|
assert_eq!(txs[&sent_txid].total_shielded_value_spent, AMOUNT1);
|
|
}
|
|
|
|
// Trying to spend from a locked wallet is an error
|
|
assert!(wallet.send_to_address(branch_id, &ss, &so,
|
|
vec![(&taddr, AMOUNT_SENT, None)]).is_err());
|
|
|
|
// unlock the wallet so we can spend to the second z address
|
|
wallet.unlock(password.clone()).unwrap();
|
|
|
|
// Second z address
|
|
let taddr2 = wallet.add_taddr();
|
|
const TAMOUNT2:u64 = 50;
|
|
|
|
let raw_tx = wallet.send_to_address(branch_id, &ss, &so,
|
|
vec![(&taddr2, TAMOUNT2, None)]).unwrap();
|
|
|
|
// Now lock the wallet again
|
|
wallet.lock().unwrap();
|
|
|
|
let sent_tx2 = Transaction::read(&raw_tx[..]).unwrap();
|
|
let txid2 = sent_tx2.txid();
|
|
|
|
let mut cb4 = FakeCompactBlock::new(3, cb3.hash());
|
|
cb4.add_tx(&sent_tx2);
|
|
wallet.scan_block(&cb4.as_bytes()).unwrap();
|
|
wallet.scan_full_tx(&sent_tx2, 3, 0);
|
|
|
|
{
|
|
let txs = wallet.txs.read().unwrap();
|
|
let prev_change_value = AMOUNT1 - AMOUNT_SENT - fee;
|
|
|
|
// Change note from prev transaction is spent
|
|
assert_eq!(txs[&sent_txid].notes[0].note.value, prev_change_value);
|
|
assert_eq!(txs[&sent_txid].notes[0].is_change, true);
|
|
assert_eq!(txs[&sent_txid].notes[0].spent, Some(txid2));
|
|
|
|
// New change note. So find it.
|
|
let change_note = txs[&txid2].notes.iter().find(|n| n.is_change).unwrap();
|
|
|
|
// New incoming tx is present
|
|
assert_eq!(change_note.note.value, prev_change_value - (TAMOUNT2+fee));
|
|
assert_eq!(change_note.spent, None);
|
|
assert_eq!(change_note.unconfirmed_spent, None);
|
|
|
|
// Find taddr2
|
|
let utxo2 = txs[&txid2].utxos.iter().find(|u| u.value == TAMOUNT2).unwrap();
|
|
assert_eq!(txs[&txid2].utxos.len(), 1);
|
|
assert_eq!(utxo2.address, taddr2);
|
|
assert_eq!(utxo2.txid, txid2);
|
|
assert_eq!(utxo2.spent, None);
|
|
assert_eq!(utxo2.unconfirmed_spent, None);
|
|
}
|
|
}
|
|
|