mirror of
https://github.com/jwinterm/grin-wallet.git
synced 2026-10-19 08:08:45 +02:00
957 lines
26 KiB
Rust
957 lines
26 KiB
Rust
// Copyright 2021 The Grin Developers
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
//! Test a wallet file send/recieve
|
|
#[macro_use]
|
|
extern crate log;
|
|
extern crate grin_wallet_controller as wallet;
|
|
extern crate grin_wallet_impls as impls;
|
|
|
|
use grin_core as core;
|
|
use grin_wallet_config::GlobalWalletConfig;
|
|
use grin_wallet_libwallet as libwallet;
|
|
use impls::test_framework::{self, LocalWalletClient};
|
|
use impls::DefaultLCProvider;
|
|
use std::fs::File;
|
|
use std::io::Write;
|
|
use std::path::PathBuf;
|
|
use std::sync::atomic::Ordering;
|
|
use std::time::Duration;
|
|
use std::{fs, thread};
|
|
|
|
use grin_wallet_libwallet::{
|
|
slatepack, Error, InitTxArgs, InitTxSendArgs, IssueInvoiceTxArgs, Slate, Slatepack,
|
|
SlatepackAddress, SlatepackBin, WalletHandle,
|
|
};
|
|
|
|
use ed25519_dalek::VerifyingKey as edDalekPublicKey;
|
|
|
|
#[macro_use]
|
|
mod common;
|
|
use common::{clean_output_dir, create_wallet_proxy, setup};
|
|
use grin_keychain::ExtKeychain;
|
|
use grin_util::secp::SecretKey;
|
|
use grin_wallet_libwallet::api_impl::owner::{decrypt_slatepack, deserialize_slatepack};
|
|
use grin_wallet_libwallet::slatepack::{get_slatepack_file_contents, SlatepackAddressIndex};
|
|
use grin_wallet_util::byte_ser;
|
|
|
|
fn output_slatepack(
|
|
slate: &Slate,
|
|
file: &str,
|
|
armored: bool,
|
|
use_bin: bool,
|
|
sender: Option<SlatepackAddress>,
|
|
recipients: Vec<SlatepackAddress>,
|
|
) -> Result<(), Error> {
|
|
let mut file = file.into();
|
|
if armored {
|
|
file = format!("{}.armored", file);
|
|
}
|
|
let slatepack = slate.create_slatepack(sender, recipients)?;
|
|
let mut pub_tx = File::create(&file)?;
|
|
if use_bin {
|
|
if armored {
|
|
let armored = slatepack.armor()?;
|
|
pub_tx.write_all(armored.as_bytes())?;
|
|
} else {
|
|
pub_tx.write_all(
|
|
&byte_ser::to_bytes(&SlatepackBin(slatepack)).map_err(|_| Error::SlatepackSer)?,
|
|
)?;
|
|
}
|
|
} else {
|
|
pub_tx.write_all(
|
|
serde_json::to_string_pretty(&slatepack)
|
|
.map_err(|_| Error::SlateSer)?
|
|
.as_bytes(),
|
|
)?;
|
|
}
|
|
pub_tx.sync_all()?;
|
|
Ok(())
|
|
}
|
|
|
|
fn slate_from_packed(
|
|
wallet_inst: WalletHandle<
|
|
DefaultLCProvider<LocalWalletClient, ExtKeychain>,
|
|
LocalWalletClient,
|
|
ExtKeychain,
|
|
>,
|
|
keychain_mask: Option<&SecretKey>,
|
|
file: &str,
|
|
armored: bool,
|
|
) -> Result<(Slatepack, Slate), Error> {
|
|
let mut file = file.into();
|
|
if armored {
|
|
file = format!("{}.armored", file);
|
|
}
|
|
let data = get_slatepack_file_contents(&file.into())?;
|
|
let slatepack = deserialize_slatepack(
|
|
&data,
|
|
wallet_inst,
|
|
keychain_mask,
|
|
vec![SlatepackAddressIndex(0)],
|
|
)?;
|
|
Ok((slatepack.clone(), slatepack.get_slate()?))
|
|
}
|
|
|
|
/// self send impl
|
|
fn slatepack_exchange_test_impl(
|
|
test_dir: &'static str,
|
|
use_bin: bool,
|
|
use_armored: bool,
|
|
use_encryption: bool,
|
|
target_slate_version: Option<u16>,
|
|
) -> Result<(), Error> {
|
|
// Create a new proxy to simulate server and wallet responses
|
|
let mut wallet_proxy = create_wallet_proxy(test_dir);
|
|
let chain = wallet_proxy.chain.clone();
|
|
let stopper = wallet_proxy.running.clone();
|
|
|
|
// Create a new wallet test client, and set its queues to communicate with the
|
|
// proxy
|
|
create_wallet_and_add!(
|
|
client1,
|
|
wallet1,
|
|
mask1_i,
|
|
test_dir,
|
|
"wallet1",
|
|
None,
|
|
&mut wallet_proxy,
|
|
false,
|
|
api1
|
|
);
|
|
let mask1 = mask1_i.as_ref();
|
|
create_wallet_and_add!(
|
|
client2,
|
|
wallet2,
|
|
mask2_i,
|
|
test_dir,
|
|
"wallet2",
|
|
None,
|
|
&mut wallet_proxy,
|
|
false,
|
|
api2
|
|
);
|
|
let mask2 = mask2_i.as_ref();
|
|
|
|
// Set the wallet proxy listener running
|
|
thread::spawn(move || {
|
|
if let Err(e) = wallet_proxy.run() {
|
|
error!("Wallet Proxy error: {}", e);
|
|
}
|
|
});
|
|
|
|
// few values to keep things shorter
|
|
let reward = core::consensus::REWARD;
|
|
|
|
// add some accounts
|
|
api1.create_account_path(mask1, "mining")?;
|
|
api1.create_account_path(mask1, "listener")?;
|
|
|
|
api2.create_account_path(mask2, "account1")?;
|
|
api2.create_account_path(mask2, "account2")?;
|
|
|
|
{
|
|
wallet_inst!(wallet1, w);
|
|
w.set_account_by_name("mining")?;
|
|
}
|
|
|
|
{
|
|
wallet_inst!(wallet2, w);
|
|
w.set_account_by_name("account1")?;
|
|
}
|
|
|
|
// Get some mining done
|
|
let mut bh = 10u64;
|
|
let _ =
|
|
test_framework::award_blocks_to_wallet(&chain, wallet1.clone(), mask1, bh as usize, false);
|
|
|
|
let (recipients_1, sender_1) = match use_encryption {
|
|
true => {
|
|
let sec_key = api1.get_slatepack_secret_key(mask1, SlatepackAddressIndex(0))?;
|
|
let pub_key = edDalekPublicKey::from(&sec_key);
|
|
let rec_address = SlatepackAddress::new(&pub_key);
|
|
|
|
let sen_index = SlatepackAddressIndex(7);
|
|
let sec_key = api1.get_slatepack_secret_key(mask1, sen_index)?;
|
|
let pub_key = edDalekPublicKey::from(&sec_key);
|
|
let sen_address = SlatepackAddress::new(&pub_key);
|
|
(vec![rec_address.clone()], Some(sen_address.clone()))
|
|
}
|
|
false => (vec![], None),
|
|
};
|
|
|
|
let (recipients_2, sender_2) = match use_encryption {
|
|
true => {
|
|
let sec_key = api2.get_slatepack_secret_key(mask2, SlatepackAddressIndex(0))?;
|
|
let pub_key = edDalekPublicKey::from(&sec_key);
|
|
let rec_address = SlatepackAddress::new(&pub_key);
|
|
|
|
let sen_index = SlatepackAddressIndex(11);
|
|
let sec_key = api2.get_slatepack_secret_key(mask2, sen_index)?;
|
|
let pub_key = edDalekPublicKey::from(&sec_key);
|
|
let sen_address = SlatepackAddress::new(&pub_key);
|
|
(vec![rec_address.clone()], Some(sen_address.clone()))
|
|
}
|
|
false => (vec![], None),
|
|
};
|
|
|
|
let (send_file, receive_file, final_file) = match use_bin {
|
|
false => (
|
|
format!("{}/standard_S1.slatepack", test_dir),
|
|
format!("{}/standard_S2.slatepack", test_dir),
|
|
format!("{}/standard_S3.slatepack", test_dir),
|
|
),
|
|
true => (
|
|
format!("{}/standard_S1.slatepackbin", test_dir),
|
|
format!("{}/standard_S2.slatepackbin", test_dir),
|
|
format!("{}/standard_S3.slatepackbin", test_dir),
|
|
),
|
|
};
|
|
|
|
let (wallet1_refreshed, wallet1_info) = api1.retrieve_summary_info(mask1, true, 1)?;
|
|
assert!(wallet1_refreshed);
|
|
assert_eq!(wallet1_info.last_confirmed_height, bh);
|
|
assert_eq!(wallet1_info.total, bh * reward);
|
|
// send to send
|
|
let args = InitTxArgs {
|
|
target_slate_version,
|
|
address_index: Some(SlatepackAddressIndex(7)),
|
|
amount: reward * 2,
|
|
minimum_confirmations: 2,
|
|
max_outputs: 500,
|
|
num_change_outputs: 1,
|
|
selection_strategy_is_use_all: true,
|
|
..Default::default()
|
|
};
|
|
let slate = api1.init_send_tx(mask1, args)?;
|
|
// output tx file
|
|
output_slatepack(
|
|
&slate,
|
|
&send_file,
|
|
use_armored,
|
|
use_bin,
|
|
sender_1.clone(),
|
|
recipients_2.clone(),
|
|
)?;
|
|
api1.tx_lock_outputs(mask1, &slate)?;
|
|
|
|
let (slatepack, mut slate) =
|
|
slate_from_packed(wallet2.clone(), mask2, &send_file, use_armored)?;
|
|
assert_eq!(
|
|
slate.version_info.version,
|
|
target_slate_version.unwrap_or(4)
|
|
);
|
|
|
|
// wallet 2 receives file, completes, sends file back
|
|
wallet::controller::foreign_single_use(
|
|
wallet2.clone(),
|
|
PathBuf::from(test_dir),
|
|
mask2_i.clone(),
|
|
|api| {
|
|
slate = api.receive_tx(&slate, None, None)?;
|
|
output_slatepack(
|
|
&slate,
|
|
&receive_file,
|
|
use_armored,
|
|
use_bin,
|
|
sender_2.clone(),
|
|
// re-encrypt for sender!
|
|
match slatepack.sender.clone() {
|
|
Some(s) => vec![s.clone()],
|
|
None => vec![],
|
|
},
|
|
)?;
|
|
Ok(())
|
|
},
|
|
)?;
|
|
|
|
// wallet 1 finalizes and posts
|
|
let (slatepack, mut slate) =
|
|
slate_from_packed(wallet1.clone(), mask1, &receive_file, use_armored)?;
|
|
assert_eq!(
|
|
slate.version_info.version,
|
|
target_slate_version.unwrap_or(4)
|
|
);
|
|
slate = api1.finalize_tx(mask1, &slate)?;
|
|
// Output final file for reference
|
|
output_slatepack(
|
|
&slate,
|
|
&final_file,
|
|
use_armored,
|
|
use_bin,
|
|
sender_1.clone(),
|
|
match slatepack.sender {
|
|
Some(s) => vec![s.clone()],
|
|
None => vec![],
|
|
},
|
|
)?;
|
|
api1.post_tx(mask1, &slate, false)?;
|
|
bh += 1;
|
|
|
|
let _ = test_framework::award_blocks_to_wallet(&chain, wallet1.clone(), mask1, 3, false);
|
|
bh += 3;
|
|
|
|
// Check total in wallet 1 account
|
|
let (wallet1_refreshed, wallet1_info) = api1.retrieve_summary_info(mask1, true, 1)?;
|
|
assert!(wallet1_refreshed);
|
|
assert_eq!(wallet1_info.last_confirmed_height, bh);
|
|
assert_eq!(wallet1_info.total, bh * reward - reward * 2);
|
|
|
|
// Check total in wallet 2 account
|
|
let (wallet2_refreshed, wallet2_info) = api2.retrieve_summary_info(mask2, true, 1)?;
|
|
assert!(wallet2_refreshed);
|
|
assert_eq!(wallet2_info.last_confirmed_height, bh);
|
|
assert_eq!(wallet2_info.total, 2 * reward);
|
|
|
|
// Now other types of exchange, for reference
|
|
// Invoice transaction
|
|
let (send_file, receive_file, final_file) = match use_bin {
|
|
false => (
|
|
format!("{}/invoice_I1.slatepack", test_dir),
|
|
format!("{}/invoice_I2.slatepack", test_dir),
|
|
format!("{}/invoice_I3.slatepack", test_dir),
|
|
),
|
|
true => (
|
|
format!("{}/invoice_I1.slatepackbin", test_dir),
|
|
format!("{}/invoice_I2.slatepackbin", test_dir),
|
|
format!("{}/invoice_I3.slatepackbin", test_dir),
|
|
),
|
|
};
|
|
|
|
let args = IssueInvoiceTxArgs {
|
|
target_slate_version,
|
|
address_index: Some(SlatepackAddressIndex(11)),
|
|
amount: 1000000000,
|
|
..Default::default()
|
|
};
|
|
let slate = api2.issue_invoice_tx(mask2, args)?;
|
|
output_slatepack(
|
|
&slate,
|
|
&send_file,
|
|
use_armored,
|
|
use_bin,
|
|
sender_2.clone(),
|
|
recipients_1.clone(),
|
|
)?;
|
|
|
|
let args = InitTxArgs {
|
|
target_slate_version,
|
|
src_acct_name: None,
|
|
amount: slate.amount,
|
|
minimum_confirmations: 2,
|
|
max_outputs: 500,
|
|
num_change_outputs: 1,
|
|
selection_strategy_is_use_all: true,
|
|
..Default::default()
|
|
};
|
|
let (slatepack, mut slate) =
|
|
slate_from_packed(wallet1.clone(), mask1, &send_file, use_armored)?;
|
|
assert_eq!(
|
|
slate.version_info.version,
|
|
target_slate_version.unwrap_or(4)
|
|
);
|
|
slate = api1.process_invoice_tx(mask1, &slate, args)?;
|
|
api1.tx_lock_outputs(mask1, &slate)?;
|
|
output_slatepack(
|
|
&slate,
|
|
&receive_file,
|
|
use_armored,
|
|
use_bin,
|
|
sender_1.clone(),
|
|
match slatepack.sender {
|
|
Some(s) => vec![s.clone()],
|
|
None => vec![],
|
|
},
|
|
)?;
|
|
// Wallet 2 receives the invoice transaction
|
|
let (_, mut slate) = slate_from_packed(wallet2.clone(), mask2, &receive_file, use_armored)?;
|
|
assert_eq!(
|
|
slate.version_info.version,
|
|
target_slate_version.unwrap_or(4)
|
|
);
|
|
slate = api2.finalize_tx(mask2, &slate)?;
|
|
output_slatepack(&slate, &final_file, use_armored, use_bin, None, vec![])?;
|
|
api2.post_tx(mask2, &slate, false)?;
|
|
|
|
// Standard, with payment proof
|
|
let _ = test_framework::award_blocks_to_wallet(&chain, wallet1.clone(), mask1, 3, false);
|
|
let (send_file, receive_file, final_file) = match use_bin {
|
|
false => (
|
|
format!("{}/standard_pp_S1.slatepack", test_dir),
|
|
format!("{}/standard_pp_S2.slatepack", test_dir),
|
|
format!("{}/standard_pp_S3.slatepack", test_dir),
|
|
),
|
|
true => (
|
|
format!("{}/standard_pp_S1.slatepackbin", test_dir),
|
|
format!("{}/standard_pp_S2.slatepackbin", test_dir),
|
|
format!("{}/standard_pp_S3.slatepackbin", test_dir),
|
|
),
|
|
};
|
|
|
|
// send from 1st wallet
|
|
let args = InitTxArgs {
|
|
target_slate_version,
|
|
address_index: Some(SlatepackAddressIndex(7)),
|
|
amount: reward,
|
|
minimum_confirmations: 2,
|
|
max_outputs: 500,
|
|
num_change_outputs: 1,
|
|
selection_strategy_is_use_all: true,
|
|
payment_proof_recipient_address: Some(
|
|
api2.get_slatepack_address(mask2, SlatepackAddressIndex(0))?,
|
|
),
|
|
..Default::default()
|
|
};
|
|
let mut slate = api1.init_send_tx(mask1, args)?;
|
|
output_slatepack(
|
|
&slate,
|
|
&send_file,
|
|
use_armored,
|
|
use_bin,
|
|
sender_1,
|
|
recipients_2.clone(),
|
|
)?;
|
|
api1.tx_lock_outputs(mask1, &slate)?;
|
|
|
|
// receive at 2nd wallet
|
|
wallet::controller::foreign_single_use(
|
|
wallet2.clone(),
|
|
PathBuf::from(test_dir),
|
|
mask2_i.clone(),
|
|
|api| {
|
|
let (slatepack, mut slate) =
|
|
slate_from_packed(wallet2.clone(), mask2, &send_file, use_armored)?;
|
|
assert_eq!(
|
|
slate.version_info.version,
|
|
target_slate_version.unwrap_or(4)
|
|
);
|
|
slate = api.receive_tx(&slate, None, None)?;
|
|
output_slatepack(
|
|
&slate,
|
|
&receive_file,
|
|
use_armored,
|
|
use_bin,
|
|
sender_2.clone(),
|
|
match slatepack.sender {
|
|
Some(s) => vec![s.clone()],
|
|
None => vec![],
|
|
},
|
|
)?;
|
|
Ok(())
|
|
},
|
|
)?;
|
|
|
|
// wallet 1 finalizes and posts
|
|
let res = slate_from_packed(wallet1.clone(), mask1, &receive_file, use_armored)?;
|
|
assert_eq!(
|
|
res.1.version_info.version,
|
|
target_slate_version.unwrap_or(4)
|
|
);
|
|
slate = res.1;
|
|
slate = api1.finalize_tx(mask1, &slate)?;
|
|
// Output final file for reference
|
|
output_slatepack(&slate, &final_file, use_armored, use_bin, None, vec![])?;
|
|
api1.post_tx(mask1, &slate, false)?;
|
|
|
|
// let logging finish
|
|
stopper.store(false, Ordering::Relaxed);
|
|
thread::sleep(Duration::from_millis(200));
|
|
Ok(())
|
|
}
|
|
|
|
/// Exercise slate encryption/decryption via the API,
|
|
/// Since doctests don't cover encryption
|
|
fn slatepack_api_impl(test_dir: &'static str) -> Result<(), libwallet::Error> {
|
|
// Create a new proxy to simulate server and wallet responses
|
|
let mut wallet_proxy = create_wallet_proxy(test_dir);
|
|
let chain = wallet_proxy.chain.clone();
|
|
let stopper = wallet_proxy.running.clone();
|
|
|
|
// Create a new wallet test client, and set its queues to communicate with the
|
|
// proxy
|
|
create_wallet_and_add!(
|
|
client1,
|
|
wallet1,
|
|
mask1_i,
|
|
test_dir,
|
|
"wallet1",
|
|
None,
|
|
&mut wallet_proxy,
|
|
false,
|
|
api1
|
|
);
|
|
let mask1 = mask1_i.as_ref();
|
|
|
|
// Set the wallet proxy listener running
|
|
thread::spawn(move || {
|
|
if let Err(e) = wallet_proxy.run() {
|
|
error!("Wallet Proxy error: {}", e);
|
|
}
|
|
});
|
|
|
|
// few values to keep things shorter
|
|
let reward = core::consensus::REWARD;
|
|
|
|
// Get some mining done
|
|
let bh = 6u64;
|
|
let _ =
|
|
test_framework::award_blocks_to_wallet(&chain, wallet1.clone(), mask1, bh as usize, false);
|
|
|
|
let args = InitTxArgs {
|
|
amount: reward * 2,
|
|
minimum_confirmations: 2,
|
|
max_outputs: 500,
|
|
num_change_outputs: 1,
|
|
selection_strategy_is_use_all: true,
|
|
..Default::default()
|
|
};
|
|
for index in [SlatepackAddressIndex::MAX + 1, u32::MAX] {
|
|
assert!(matches!(
|
|
api1.init_send_tx(
|
|
mask1,
|
|
InitTxArgs {
|
|
address_index: Some(SlatepackAddressIndex(index)),
|
|
..args.clone()
|
|
}
|
|
),
|
|
Err(Error::SlatepackAddress(_))
|
|
));
|
|
assert!(matches!(
|
|
api1.issue_invoice_tx(
|
|
mask1,
|
|
IssueInvoiceTxArgs {
|
|
amount: reward,
|
|
address_index: Some(SlatepackAddressIndex(index)),
|
|
..Default::default()
|
|
}
|
|
),
|
|
Err(Error::SlatepackAddress(_))
|
|
));
|
|
}
|
|
api1.init_send_tx(
|
|
mask1,
|
|
InitTxArgs {
|
|
estimate_only: Some(true),
|
|
..args.clone()
|
|
},
|
|
)?;
|
|
let slate = api1.init_send_tx(mask1, args)?;
|
|
// Both the recipient and sender can decrypt the contract's reply.
|
|
let recipient_key = ed25519_dalek::SigningKey::from_bytes(&[1; 32]);
|
|
let recipient = SlatepackAddress::new(&edDalekPublicKey::from(&recipient_key));
|
|
let message = api1.create_slatepack_message(
|
|
mask1,
|
|
&slate,
|
|
Some(SlatepackAddressIndex(0)),
|
|
vec![recipient],
|
|
)?;
|
|
let mut recipient_copy = libwallet::api_impl::owner::parse_slatepack(message.as_bytes())?;
|
|
recipient_copy.try_decrypt_payload(Some(&recipient_key))?;
|
|
recipient_copy.get_slate()?;
|
|
api1.slate_from_slatepack_message(mask1, message.clone(), vec![SlatepackAddressIndex(0)])?;
|
|
api1.create_account_path(mask1, "second")?;
|
|
api1.set_active_account(mask1, "second")?;
|
|
let err = api1
|
|
.slate_from_slatepack_message(mask1, message, vec![SlatepackAddressIndex(0)])
|
|
.unwrap_err();
|
|
assert!(err.to_string().contains("account 'default'"), "{}", err);
|
|
api1.set_active_account(mask1, "default")?;
|
|
// create an encrypted slatepack (just encrypted for self)
|
|
let address_index = SlatepackAddressIndex(0);
|
|
let enc_addr = api1.get_slatepack_address(mask1, address_index)?;
|
|
let slatepack =
|
|
api1.create_slatepack_message(mask1, &slate, Some(address_index), vec![enc_addr])?;
|
|
let mut encrypted = api1.decode_slatepack_message(mask1, slatepack.clone(), vec![])?;
|
|
assert!(encrypted.is_encrypted());
|
|
decrypt_slatepack(&mut encrypted, wallet1.clone(), mask1, vec![address_index])?;
|
|
assert_eq!(encrypted.get_slate()?.id, slate.id);
|
|
let slatepack_raw =
|
|
api1.decode_slatepack_message(mask1, slatepack.clone(), vec![address_index])?;
|
|
assert!(!slatepack_raw.is_encrypted());
|
|
assert_eq!(slatepack_raw.get_slate()?.id, slate.id);
|
|
let decoded_slate = api1.slate_from_slatepack_message(mask1, slatepack, vec![address_index])?;
|
|
assert_eq!(decoded_slate.id, slate.id);
|
|
|
|
let unknown_index = SlatepackAddressIndex(77);
|
|
let unknown_address = api1.get_slatepack_address(mask1, unknown_index)?;
|
|
let message = api1.create_slatepack_message(mask1, &slate, None, vec![unknown_address])?;
|
|
let encrypted = api1.decode_slatepack_message(mask1, message.clone(), vec![])?;
|
|
let decoded = api1.decode_slatepack_message(mask1, message.clone(), vec![address_index])?;
|
|
assert!(decoded.is_encrypted());
|
|
assert_eq!(decoded.payload, encrypted.payload);
|
|
assert!(matches!(
|
|
api1.slate_from_slatepack_message(mask1, message.clone(), vec![address_index]),
|
|
Err(Error::SlatepackDecryption(_))
|
|
));
|
|
let mut api1 = api1;
|
|
assert!(matches!(
|
|
wallet::command::parse_slatepack(&mut api1, mask1, None, Some(message)),
|
|
Err(wallet::Error::LibWallet(Error::SlatepackDecryption(_)))
|
|
));
|
|
|
|
let index = {
|
|
wallet_inst!(wallet1, w);
|
|
w.get_private_context(mask1, slate.id.as_bytes())?
|
|
.payment_proof_derivation_index
|
|
.unwrap()
|
|
};
|
|
assert_eq!(index, SlatepackAddressIndex(1));
|
|
let address = api1.get_slatepack_address(mask1, index)?;
|
|
let message = api1.create_slatepack_message(mask1, &slate, None, vec![address])?;
|
|
let decoded = api1.decode_slatepack_message(mask1, message.clone(), vec![address_index])?;
|
|
assert!(!decoded.is_encrypted());
|
|
assert_eq!(decoded.get_slate()?.id, slate.id);
|
|
|
|
api1.create_account_path(mask1, "other")?;
|
|
api1.set_active_account(mask1, "other")?;
|
|
test_framework::award_blocks_to_wallet(&chain, wallet1.clone(), mask1, 6, false)?;
|
|
api1.set_active_account(mask1, "default")?;
|
|
let invoice = api1.issue_invoice_tx(
|
|
mask1,
|
|
IssueInvoiceTxArgs {
|
|
amount: reward,
|
|
dest_acct_name: Some("other".into()),
|
|
..Default::default()
|
|
},
|
|
)?;
|
|
let send = api1.init_send_tx(
|
|
mask1,
|
|
InitTxArgs {
|
|
amount: reward,
|
|
src_acct_name: Some("other".into()),
|
|
minimum_confirmations: 2,
|
|
..Default::default()
|
|
},
|
|
)?;
|
|
api1.set_active_account(mask1, "other")?;
|
|
let next = api1.issue_invoice_tx(
|
|
mask1,
|
|
IssueInvoiceTxArgs {
|
|
amount: reward,
|
|
..Default::default()
|
|
},
|
|
)?;
|
|
for (slate, expected) in [(&invoice, 1), (&send, 2), (&next, 3)] {
|
|
wallet_inst!(wallet1, w);
|
|
let context = w.get_private_context(mask1, slate.id.as_bytes())?;
|
|
assert_eq!(context.parent_key_id, w.parent_key_id());
|
|
assert_eq!(
|
|
context.payment_proof_derivation_index,
|
|
Some(SlatepackAddressIndex(expected))
|
|
);
|
|
}
|
|
api1.set_active_account(mask1, "default")?;
|
|
let invoice = api1.issue_invoice_tx(
|
|
mask1,
|
|
IssueInvoiceTxArgs {
|
|
amount: reward,
|
|
..Default::default()
|
|
},
|
|
)?;
|
|
{
|
|
wallet_inst!(wallet1, w);
|
|
assert_eq!(
|
|
w.get_private_context(mask1, invoice.id.as_bytes())?
|
|
.payment_proof_derivation_index,
|
|
Some(SlatepackAddressIndex(2))
|
|
);
|
|
}
|
|
|
|
api1.close_wallet(None)?;
|
|
let encrypted = api1.decode_slatepack_message(None, message, vec![])?;
|
|
assert!(encrypted.is_encrypted());
|
|
|
|
// let logging finish
|
|
stopper.store(false, Ordering::Relaxed);
|
|
thread::sleep(Duration::from_millis(200));
|
|
Ok(())
|
|
}
|
|
|
|
/// Do not create transaction for invalid or wrong network Slatepack address.
|
|
fn slatepack_address_validation(test_dir: &'static str) -> Result<(), libwallet::Error> {
|
|
let mut wallet_proxy = create_wallet_proxy(test_dir);
|
|
let chain = wallet_proxy.chain.clone();
|
|
let stopper = wallet_proxy.running.clone();
|
|
let config_path = PathBuf::from(test_dir).join("grin-wallet.toml");
|
|
GlobalWalletConfig::for_chain(&core::global::ChainTypes::AutomatedTesting, &config_path)
|
|
.write_to_file(config_path.to_str().unwrap(), false, None, None)?;
|
|
|
|
create_wallet_and_add!(
|
|
client1,
|
|
wallet1,
|
|
mask1_i,
|
|
test_dir,
|
|
"wallet1",
|
|
None,
|
|
&mut wallet_proxy,
|
|
false,
|
|
api1
|
|
);
|
|
let mask1 = mask1_i.as_ref();
|
|
|
|
create_wallet_and_add!(
|
|
client2,
|
|
wallet2,
|
|
mask2_i,
|
|
test_dir,
|
|
"wallet2",
|
|
None,
|
|
&mut wallet_proxy,
|
|
false,
|
|
api2
|
|
);
|
|
let mask2 = mask2_i.as_ref();
|
|
|
|
let proxy_thread = thread::spawn(move || {
|
|
if let Err(e) = wallet_proxy.run() {
|
|
error!("Wallet Proxy error: {}", e);
|
|
}
|
|
});
|
|
|
|
let reward = core::consensus::REWARD;
|
|
let _ = test_framework::award_blocks_to_wallet(&chain, wallet1.clone(), mask1, 6, false);
|
|
let slate = api2.issue_invoice_tx(
|
|
mask2,
|
|
IssueInvoiceTxArgs {
|
|
amount: reward,
|
|
..Default::default()
|
|
},
|
|
)?;
|
|
|
|
let args = InitTxArgs {
|
|
amount: reward,
|
|
minimum_confirmations: 2,
|
|
max_outputs: 500,
|
|
num_change_outputs: 1,
|
|
selection_strategy_is_use_all: true,
|
|
..Default::default()
|
|
};
|
|
|
|
let mut wrong_net_args = args.clone();
|
|
wrong_net_args.send_args = Some(InitTxSendArgs {
|
|
dest: "grin1dvge9z4uqgqlpspmljrd7smh3grrw9xu2r9lkz3u67s3emj3ud2sd5gk9p".to_string(),
|
|
post_tx: false,
|
|
fluff: false,
|
|
skip_tor: Some(true),
|
|
});
|
|
assert!(api1.init_send_tx(mask1, wrong_net_args.clone()).is_err());
|
|
assert!(api1
|
|
.process_invoice_tx(mask1, &slate, wrong_net_args)
|
|
.is_err());
|
|
|
|
let mut invalid_args = args.clone();
|
|
invalid_args.send_args = Some(InitTxSendArgs {
|
|
dest: "tgrinaddr10qlk22rxjap2ny8qltc2tl996kenxr3hhwuu6hrzs6tdq08yaqgqnlumr7".to_string(),
|
|
post_tx: false,
|
|
fluff: false,
|
|
skip_tor: Some(true),
|
|
});
|
|
assert!(api1.init_send_tx(mask1, invalid_args.clone()).is_err());
|
|
assert!(api1
|
|
.process_invoice_tx(mask1, &slate, invalid_args)
|
|
.is_err());
|
|
|
|
let mut valid_args = args;
|
|
valid_args.send_args = Some(InitTxSendArgs {
|
|
dest: "tgrin1xtxavwfgs48ckf3gk8wwgcndmn0nt4tvkl8a7ltyejjcy2mc6nfs9gm2lp".to_string(),
|
|
post_tx: false,
|
|
fluff: false,
|
|
skip_tor: Some(true),
|
|
});
|
|
let missing_config = PathBuf::from(test_dir).join("missing.toml");
|
|
let api = grin_wallet_api::Owner::new(wallet1.clone(), None, missing_config);
|
|
assert!(api
|
|
.process_invoice_tx(mask1, &slate, valid_args.clone())
|
|
.is_err());
|
|
{
|
|
wallet_inst!(wallet1, w);
|
|
assert!(w.get_private_context(mask1, slate.id.as_bytes()).is_err());
|
|
}
|
|
api1.process_invoice_tx(mask1, &slate, valid_args.clone())?;
|
|
api1.init_send_tx(mask1, valid_args)?;
|
|
|
|
stopper.store(false, Ordering::Relaxed);
|
|
proxy_thread.join().expect("wallet proxy thread panicked");
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
fn slatepack_exchange_json() {
|
|
let test_dir = "test_output/slatepack_exchange_json";
|
|
setup(test_dir);
|
|
// JSON output
|
|
if let Err(e) = slatepack_exchange_test_impl(test_dir, false, false, false, None) {
|
|
panic!("Libwallet Error: {}", e);
|
|
}
|
|
clean_output_dir(test_dir);
|
|
}
|
|
|
|
#[test]
|
|
fn slatepack_exchange_bin() {
|
|
let test_dir = "test_output/slatepack_exchange_bin";
|
|
setup(test_dir);
|
|
// Bin output
|
|
if let Err(e) = slatepack_exchange_test_impl(test_dir, true, false, false, None) {
|
|
panic!("Libwallet Error: {}", e);
|
|
}
|
|
clean_output_dir(test_dir);
|
|
}
|
|
|
|
#[test]
|
|
fn slatepack_exchange_armored() {
|
|
let test_dir = "test_output/slatepack_exchange_armored";
|
|
setup(test_dir);
|
|
// Bin output
|
|
if let Err(e) = slatepack_exchange_test_impl(test_dir, true, true, true, None) {
|
|
panic!("Libwallet Error: {}", e);
|
|
}
|
|
clean_output_dir(test_dir);
|
|
}
|
|
|
|
#[test]
|
|
fn slatepack_exchange_json_enc() {
|
|
let test_dir = "test_output/slatepack_exchange_json_enc";
|
|
setup(test_dir);
|
|
// JSON output
|
|
if let Err(e) = slatepack_exchange_test_impl(test_dir, false, false, true, None) {
|
|
panic!("Libwallet Error: {}", e);
|
|
}
|
|
clean_output_dir(test_dir);
|
|
}
|
|
|
|
#[test]
|
|
fn slatepack_exchange_bin_enc() {
|
|
let test_dir = "test_output/slatepack_exchange_bin_enc";
|
|
setup(test_dir);
|
|
// Bin output
|
|
if let Err(e) = slatepack_exchange_test_impl(test_dir, true, false, true, None) {
|
|
panic!("Libwallet Error: {}", e);
|
|
}
|
|
clean_output_dir(test_dir);
|
|
}
|
|
|
|
#[test]
|
|
fn slatepack_exchange_armored_enc() {
|
|
let test_dir = "test_output/slatepack_exchange_armored_enc";
|
|
setup(test_dir);
|
|
// Bin output
|
|
if let Err(e) = slatepack_exchange_test_impl(test_dir, true, true, true, None) {
|
|
panic!("Libwallet Error: {}", e);
|
|
}
|
|
clean_output_dir(test_dir);
|
|
}
|
|
|
|
#[test]
|
|
fn slatepack_api() {
|
|
let test_dir = "test_output/slatepack_api";
|
|
setup(test_dir);
|
|
// JSON output
|
|
if let Err(e) = slatepack_api_impl(test_dir) {
|
|
panic!("Libwallet Error: {}", e);
|
|
}
|
|
clean_output_dir(test_dir);
|
|
}
|
|
|
|
#[test]
|
|
fn slatepack_address() {
|
|
let test_dir = "test_output/slatepack_address";
|
|
setup(test_dir);
|
|
if let Err(e) = slatepack_address_validation(test_dir) {
|
|
panic!("Libwallet Error: {}", e);
|
|
}
|
|
clean_output_dir(test_dir);
|
|
}
|
|
|
|
#[test]
|
|
fn slatepack_file_contents() {
|
|
let test_dir = "test_output/slatepack_file_contents";
|
|
setup(test_dir);
|
|
|
|
fs::create_dir_all(test_dir).unwrap();
|
|
let sp_path = PathBuf::from(test_dir).join("pack_file");
|
|
|
|
// set Slatepack file to minimum allowable size
|
|
{
|
|
let f = File::create(sp_path.clone()).unwrap();
|
|
f.set_len(slatepack::min_size()).unwrap();
|
|
}
|
|
|
|
let mut pack_path = get_slatepack_file_contents(&sp_path);
|
|
assert!(pack_path.is_ok());
|
|
|
|
// set Slatepack file to maximum allowable size
|
|
{
|
|
let f = File::create(sp_path.clone()).unwrap();
|
|
f.set_len(slatepack::max_size()).unwrap();
|
|
}
|
|
pack_path = get_slatepack_file_contents(&sp_path);
|
|
assert!(pack_path.is_ok());
|
|
|
|
// set Slatepack file below minimum allowable size
|
|
{
|
|
let f = File::create(sp_path.clone()).unwrap();
|
|
f.set_len(slatepack::min_size() - 1).unwrap();
|
|
}
|
|
pack_path = get_slatepack_file_contents(&sp_path);
|
|
assert!(pack_path.is_err());
|
|
|
|
// set Slatepack file above maximum allowable size
|
|
{
|
|
let f = File::create(sp_path.clone()).unwrap();
|
|
f.set_len(slatepack::max_size() + 1).unwrap();
|
|
}
|
|
pack_path = get_slatepack_file_contents(&sp_path);
|
|
assert!(pack_path.is_err());
|
|
clean_output_dir(test_dir);
|
|
}
|
|
|
|
#[test]
|
|
fn slatepack_exchange_json_v5() {
|
|
let test_dir = "test_output/slatepack_exchange_json_v5";
|
|
setup(test_dir);
|
|
// Json output
|
|
if let Err(e) = slatepack_exchange_test_impl(test_dir, false, false, false, Some(5)) {
|
|
panic!("Libwallet Error: {}", e);
|
|
}
|
|
clean_output_dir(test_dir);
|
|
}
|
|
|
|
#[test]
|
|
fn slatepack_v4() {
|
|
const V4_SLATEPACK: &str = include_str!("slates/v4.slatepack");
|
|
common::setup_global_chain_type();
|
|
|
|
let packed = libwallet::api_impl::owner::parse_slatepack(V4_SLATEPACK.as_bytes()).unwrap();
|
|
let slate = packed.get_slate().unwrap();
|
|
|
|
assert_eq!(slate.version_info.version, 4);
|
|
assert_eq!(slate.state, libwallet::SlateState::Standard2);
|
|
assert_eq!(slate.participant_data.len(), 1);
|
|
assert_eq!(
|
|
slate
|
|
.participant_data
|
|
.iter()
|
|
.filter(|participant| participant.part_sig.is_some())
|
|
.count(),
|
|
1
|
|
);
|
|
let tx = slate.tx_or_err().unwrap();
|
|
assert_eq!(tx.inputs().len(), 0);
|
|
assert_eq!(tx.outputs().len(), 1);
|
|
assert_eq!(tx.kernels().len(), 1);
|
|
assert_eq!(
|
|
slate.create_slatepack(None, vec![]).unwrap().payload,
|
|
packed.payload
|
|
);
|
|
}
|