2018-09-14 01:58:39 -07:00
|
|
|
use bs58;
|
2018-09-19 12:44:16 -07:00
|
|
|
use chrono::prelude::*;
|
2018-09-14 01:58:39 -07:00
|
|
|
use clap::ArgMatches;
|
2019-02-27 11:17:32 -08:00
|
|
|
use log::*;
|
2018-10-08 11:10:47 -07:00
|
|
|
use serde_json;
|
2018-12-14 20:39:10 -08:00
|
|
|
use serde_json::json;
|
2019-03-02 13:23:22 -08:00
|
|
|
use solana_budget_api;
|
2019-03-23 05:14:30 -07:00
|
|
|
use solana_budget_api::budget_instruction::BudgetInstruction;
|
2019-03-16 22:37:20 -07:00
|
|
|
use solana_client::rpc_client::{get_rpc_request_str, RpcClient};
|
2019-01-13 23:10:03 -08:00
|
|
|
#[cfg(not(test))]
|
|
|
|
use solana_drone::drone::request_airdrop_transaction;
|
|
|
|
use solana_drone::drone::DRONE_PORT;
|
2019-03-16 21:47:16 -07:00
|
|
|
#[cfg(test)]
|
|
|
|
use solana_drone::drone_mock::request_airdrop_transaction;
|
2018-12-03 13:32:31 -08:00
|
|
|
use solana_sdk::bpf_loader;
|
2018-11-16 08:04:46 -08:00
|
|
|
use solana_sdk::hash::Hash;
|
2019-03-21 07:14:14 -07:00
|
|
|
use solana_sdk::loader_instruction::LoaderInstruction;
|
2018-10-25 11:13:08 -07:00
|
|
|
use solana_sdk::pubkey::Pubkey;
|
2019-03-16 17:43:39 -07:00
|
|
|
use solana_sdk::rpc_port::DEFAULT_RPC_PORT;
|
2018-12-03 10:26:28 -08:00
|
|
|
use solana_sdk::signature::{Keypair, KeypairUtil, Signature};
|
2018-12-04 20:19:48 -08:00
|
|
|
use solana_sdk::system_transaction::SystemTransaction;
|
2018-11-29 16:18:47 -08:00
|
|
|
use solana_sdk::transaction::Transaction;
|
2019-03-06 17:02:20 -08:00
|
|
|
use solana_vote_api::vote_instruction::VoteInstruction;
|
2018-12-12 13:13:18 -08:00
|
|
|
use std::fs::File;
|
|
|
|
use std::io::Read;
|
2019-01-16 20:43:00 -08:00
|
|
|
use std::net::{IpAddr, Ipv4Addr, SocketAddr};
|
2018-09-14 01:58:39 -07:00
|
|
|
use std::{error, fmt, mem};
|
|
|
|
|
2018-10-22 21:21:33 -07:00
|
|
|
const USERDATA_CHUNK_SIZE: usize = 256;
|
|
|
|
|
2018-09-14 01:59:09 -07:00
|
|
|
#[derive(Debug, PartialEq)]
|
2018-09-14 01:58:39 -07:00
|
|
|
pub enum WalletCommand {
|
|
|
|
Address,
|
2019-01-13 23:10:03 -08:00
|
|
|
Airdrop(u64),
|
2019-03-20 09:44:16 -07:00
|
|
|
Balance(Pubkey),
|
2018-09-18 14:42:32 -07:00
|
|
|
Cancel(Pubkey),
|
2018-09-14 01:58:39 -07:00
|
|
|
Confirm(Signature),
|
2019-03-06 17:02:20 -08:00
|
|
|
// ConfigureStakingAccount(delegate_id, authorized_voter_id)
|
|
|
|
ConfigureStakingAccount(Option<Pubkey>, Option<Pubkey>),
|
|
|
|
CreateStakingAccount(Pubkey, u64),
|
2018-10-22 21:21:33 -07:00
|
|
|
Deploy(String),
|
2018-10-25 10:20:17 -07:00
|
|
|
GetTransactionCount,
|
2019-03-05 17:22:46 -08:00
|
|
|
// Pay(lamports, to, timestamp, timestamp_pubkey, witness(es), cancelable)
|
2018-09-18 14:42:32 -07:00
|
|
|
Pay(
|
2018-11-05 08:36:22 -08:00
|
|
|
u64,
|
2018-09-18 14:42:32 -07:00
|
|
|
Pubkey,
|
|
|
|
Option<DateTime<Utc>>,
|
|
|
|
Option<Pubkey>,
|
2018-09-19 12:44:16 -07:00
|
|
|
Option<Vec<Pubkey>>,
|
2018-09-22 16:51:21 -07:00
|
|
|
Option<Pubkey>,
|
2018-09-18 14:42:32 -07:00
|
|
|
),
|
2018-09-22 16:51:21 -07:00
|
|
|
// TimeElapsed(to, process_id, timestamp)
|
|
|
|
TimeElapsed(Pubkey, Pubkey, DateTime<Utc>),
|
2018-09-24 09:23:16 -07:00
|
|
|
// Witness(to, process_id)
|
|
|
|
Witness(Pubkey, Pubkey),
|
2018-09-14 01:58:39 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Debug, Clone)]
|
|
|
|
pub enum WalletError {
|
|
|
|
CommandNotRecognized(String),
|
|
|
|
BadParameter(String),
|
2018-10-22 21:21:33 -07:00
|
|
|
DynamicProgramError(String),
|
2018-09-20 22:27:06 -07:00
|
|
|
RpcRequestError(String),
|
2018-09-14 01:58:39 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
impl fmt::Display for WalletError {
|
2018-12-08 21:44:20 -08:00
|
|
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
2018-09-14 01:58:39 -07:00
|
|
|
write!(f, "invalid")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl error::Error for WalletError {
|
|
|
|
fn description(&self) -> &str {
|
|
|
|
"invalid"
|
|
|
|
}
|
|
|
|
|
2018-12-08 21:44:20 -08:00
|
|
|
fn cause(&self) -> Option<&dyn error::Error> {
|
2018-09-14 01:58:39 -07:00
|
|
|
// Generic error, underlying cause isn't tracked.
|
|
|
|
None
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
pub struct WalletConfig {
|
|
|
|
pub id: Keypair,
|
|
|
|
pub command: WalletCommand,
|
2019-01-16 20:43:00 -08:00
|
|
|
pub drone_host: Option<IpAddr>,
|
|
|
|
pub drone_port: u16,
|
|
|
|
pub host: IpAddr,
|
2019-01-13 23:10:03 -08:00
|
|
|
pub rpc_client: Option<RpcClient>,
|
2019-01-16 20:43:00 -08:00
|
|
|
pub rpc_host: Option<IpAddr>,
|
|
|
|
pub rpc_port: u16,
|
|
|
|
pub rpc_tls: bool,
|
2018-09-14 01:58:39 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
impl Default for WalletConfig {
|
|
|
|
fn default() -> WalletConfig {
|
|
|
|
WalletConfig {
|
2019-03-20 09:44:16 -07:00
|
|
|
command: WalletCommand::Balance(Pubkey::default()),
|
2019-01-16 20:43:00 -08:00
|
|
|
drone_host: None,
|
|
|
|
drone_port: DRONE_PORT,
|
|
|
|
host: IpAddr::V4(Ipv4Addr::new(127, 0, 0, 1)),
|
|
|
|
id: Keypair::new(),
|
2019-01-13 23:10:03 -08:00
|
|
|
rpc_client: None,
|
2019-01-16 20:43:00 -08:00
|
|
|
rpc_host: None,
|
2019-03-16 17:43:39 -07:00
|
|
|
rpc_port: DEFAULT_RPC_PORT,
|
2019-01-16 20:43:00 -08:00
|
|
|
rpc_tls: false,
|
2018-09-14 01:58:39 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-11-09 14:10:44 -08:00
|
|
|
impl WalletConfig {
|
2019-01-04 13:25:47 -08:00
|
|
|
pub fn drone_addr(&self) -> SocketAddr {
|
2019-01-16 20:43:00 -08:00
|
|
|
SocketAddr::new(self.drone_host.unwrap_or(self.host), self.drone_port)
|
2018-11-09 14:36:08 -08:00
|
|
|
}
|
2018-11-09 16:27:36 -08:00
|
|
|
|
2019-01-11 14:37:52 -08:00
|
|
|
pub fn rpc_addr(&self) -> String {
|
2019-01-16 20:43:00 -08:00
|
|
|
get_rpc_request_str(
|
|
|
|
SocketAddr::new(self.rpc_host.unwrap_or(self.host), self.rpc_port),
|
|
|
|
self.rpc_tls,
|
|
|
|
)
|
2018-11-09 16:27:36 -08:00
|
|
|
}
|
2018-11-09 14:10:44 -08:00
|
|
|
}
|
|
|
|
|
2019-03-20 15:20:31 -07:00
|
|
|
// Return the pubkey for an argument with `name` or None if not present.
|
|
|
|
fn pubkey_of(matches: &ArgMatches<'_>, name: &str) -> Option<Pubkey> {
|
|
|
|
matches.value_of(name).map(|x| x.parse::<Pubkey>().unwrap())
|
|
|
|
}
|
|
|
|
|
|
|
|
// Return the pubkeys for arguments with `name` or None if none present.
|
|
|
|
fn pubkeys_of(matches: &ArgMatches<'_>, name: &str) -> Option<Vec<Pubkey>> {
|
|
|
|
matches
|
|
|
|
.values_of(name)
|
|
|
|
.map(|xs| xs.map(|x| x.parse::<Pubkey>().unwrap()).collect())
|
|
|
|
}
|
|
|
|
|
2018-09-14 01:58:39 -07:00
|
|
|
pub fn parse_command(
|
2019-03-09 19:28:43 -08:00
|
|
|
pubkey: &Pubkey,
|
2018-12-08 21:44:20 -08:00
|
|
|
matches: &ArgMatches<'_>,
|
|
|
|
) -> Result<WalletCommand, Box<dyn error::Error>> {
|
2018-09-14 01:58:39 -07:00
|
|
|
let response = match matches.subcommand() {
|
2018-09-20 22:27:06 -07:00
|
|
|
("address", Some(_address_matches)) => Ok(WalletCommand::Address),
|
2018-09-14 01:58:39 -07:00
|
|
|
("airdrop", Some(airdrop_matches)) => {
|
2019-03-05 17:22:46 -08:00
|
|
|
let lamports = airdrop_matches.value_of("lamports").unwrap().parse()?;
|
|
|
|
Ok(WalletCommand::Airdrop(lamports))
|
2018-09-14 01:58:39 -07:00
|
|
|
}
|
2019-03-20 09:44:16 -07:00
|
|
|
("balance", Some(balance_matches)) => {
|
|
|
|
let pubkey = pubkey_of(&balance_matches, "pubkey").unwrap_or(*pubkey);
|
|
|
|
Ok(WalletCommand::Balance(pubkey))
|
|
|
|
}
|
2018-09-19 12:44:16 -07:00
|
|
|
("cancel", Some(cancel_matches)) => {
|
2019-03-20 15:20:31 -07:00
|
|
|
let process_id = pubkey_of(cancel_matches, "process_id").unwrap();
|
2018-09-19 12:44:16 -07:00
|
|
|
Ok(WalletCommand::Cancel(process_id))
|
|
|
|
}
|
2018-09-20 22:27:06 -07:00
|
|
|
("confirm", Some(confirm_matches)) => {
|
|
|
|
let signatures = bs58::decode(confirm_matches.value_of("signature").unwrap())
|
|
|
|
.into_vec()
|
|
|
|
.expect("base58-encoded signature");
|
|
|
|
|
|
|
|
if signatures.len() == mem::size_of::<Signature>() {
|
|
|
|
let signature = Signature::new(&signatures);
|
|
|
|
Ok(WalletCommand::Confirm(signature))
|
|
|
|
} else {
|
|
|
|
eprintln!("{}", confirm_matches.usage());
|
|
|
|
Err(WalletError::BadParameter("Invalid signature".to_string()))
|
|
|
|
}
|
|
|
|
}
|
2019-03-06 17:02:20 -08:00
|
|
|
("configure-staking-account", Some(staking_config_matches)) => {
|
2019-03-20 15:20:31 -07:00
|
|
|
let delegate_id = pubkey_of(staking_config_matches, "delegate");
|
|
|
|
let authorized_voter_id = pubkey_of(staking_config_matches, "authorize");
|
2019-03-06 17:02:20 -08:00
|
|
|
Ok(WalletCommand::ConfigureStakingAccount(
|
|
|
|
delegate_id,
|
|
|
|
authorized_voter_id,
|
|
|
|
))
|
|
|
|
}
|
|
|
|
("create-staking-account", Some(staking_matches)) => {
|
2019-03-20 15:20:31 -07:00
|
|
|
let voting_account_id = pubkey_of(staking_matches, "voting_account_id").unwrap();
|
2019-03-06 17:02:20 -08:00
|
|
|
let lamports = staking_matches.value_of("lamports").unwrap().parse()?;
|
|
|
|
Ok(WalletCommand::CreateStakingAccount(
|
|
|
|
voting_account_id,
|
|
|
|
lamports,
|
|
|
|
))
|
|
|
|
}
|
2018-10-22 21:21:33 -07:00
|
|
|
("deploy", Some(deploy_matches)) => Ok(WalletCommand::Deploy(
|
|
|
|
deploy_matches
|
2019-03-07 16:00:12 -08:00
|
|
|
.value_of("program_location")
|
2018-10-22 21:21:33 -07:00
|
|
|
.unwrap()
|
|
|
|
.to_string(),
|
|
|
|
)),
|
2018-10-25 10:20:17 -07:00
|
|
|
("get-transaction-count", Some(_matches)) => Ok(WalletCommand::GetTransactionCount),
|
2018-09-14 01:58:39 -07:00
|
|
|
("pay", Some(pay_matches)) => {
|
2019-03-05 17:22:46 -08:00
|
|
|
let lamports = pay_matches.value_of("lamports").unwrap().parse()?;
|
2019-03-20 15:20:31 -07:00
|
|
|
let to = pubkey_of(&pay_matches, "to").unwrap_or(*pubkey);
|
2018-09-19 12:44:16 -07:00
|
|
|
let timestamp = if pay_matches.is_present("timestamp") {
|
|
|
|
// Parse input for serde_json
|
2018-09-21 20:48:46 -07:00
|
|
|
let date_string = if !pay_matches.value_of("timestamp").unwrap().contains('Z') {
|
2018-09-19 12:44:16 -07:00
|
|
|
format!("\"{}Z\"", pay_matches.value_of("timestamp").unwrap())
|
|
|
|
} else {
|
|
|
|
format!("\"{}\"", pay_matches.value_of("timestamp").unwrap())
|
|
|
|
};
|
|
|
|
Some(serde_json::from_str(&date_string)?)
|
|
|
|
} else {
|
|
|
|
None
|
|
|
|
};
|
2019-03-20 15:20:31 -07:00
|
|
|
let timestamp_pubkey = pubkey_of(&pay_matches, "timestamp_pubkey");
|
|
|
|
let witness_vec = pubkeys_of(&pay_matches, "witness");
|
2018-09-19 12:44:16 -07:00
|
|
|
let cancelable = if pay_matches.is_present("cancelable") {
|
2019-03-09 19:28:43 -08:00
|
|
|
Some(*pubkey)
|
2018-09-19 12:44:16 -07:00
|
|
|
} else {
|
|
|
|
None
|
|
|
|
};
|
|
|
|
|
|
|
|
Ok(WalletCommand::Pay(
|
2019-03-05 17:22:46 -08:00
|
|
|
lamports,
|
2018-09-19 12:44:16 -07:00
|
|
|
to,
|
|
|
|
timestamp,
|
|
|
|
timestamp_pubkey,
|
|
|
|
witness_vec,
|
|
|
|
cancelable,
|
|
|
|
))
|
|
|
|
}
|
|
|
|
("send-signature", Some(sig_matches)) => {
|
2019-03-20 15:20:31 -07:00
|
|
|
let to = pubkey_of(&sig_matches, "to").unwrap();
|
|
|
|
let process_id = pubkey_of(&sig_matches, "process_id").unwrap();
|
2018-09-24 09:23:16 -07:00
|
|
|
Ok(WalletCommand::Witness(to, process_id))
|
2018-09-19 12:44:16 -07:00
|
|
|
}
|
|
|
|
("send-timestamp", Some(timestamp_matches)) => {
|
2019-03-20 15:20:31 -07:00
|
|
|
let to = pubkey_of(×tamp_matches, "to").unwrap();
|
|
|
|
let process_id = pubkey_of(×tamp_matches, "process_id").unwrap();
|
2018-09-19 16:44:03 -07:00
|
|
|
let dt = if timestamp_matches.is_present("datetime") {
|
2018-09-19 12:44:16 -07:00
|
|
|
// Parse input for serde_json
|
|
|
|
let date_string = if !timestamp_matches
|
|
|
|
.value_of("datetime")
|
|
|
|
.unwrap()
|
2018-09-21 20:48:46 -07:00
|
|
|
.contains('Z')
|
2018-09-19 12:44:16 -07:00
|
|
|
{
|
|
|
|
format!("\"{}Z\"", timestamp_matches.value_of("datetime").unwrap())
|
|
|
|
} else {
|
|
|
|
format!("\"{}\"", timestamp_matches.value_of("datetime").unwrap())
|
|
|
|
};
|
|
|
|
serde_json::from_str(&date_string)?
|
|
|
|
} else {
|
|
|
|
Utc::now()
|
|
|
|
};
|
2018-09-22 16:51:21 -07:00
|
|
|
Ok(WalletCommand::TimeElapsed(to, process_id, dt))
|
2018-09-14 01:58:39 -07:00
|
|
|
}
|
|
|
|
("", None) => {
|
2018-09-24 09:23:16 -07:00
|
|
|
eprintln!("{}", matches.usage());
|
2018-09-14 01:58:39 -07:00
|
|
|
Err(WalletError::CommandNotRecognized(
|
|
|
|
"no subcommand given".to_string(),
|
|
|
|
))
|
|
|
|
}
|
|
|
|
_ => unreachable!(),
|
|
|
|
}?;
|
|
|
|
Ok(response)
|
|
|
|
}
|
|
|
|
|
2019-02-09 13:15:44 -08:00
|
|
|
type ProcessResult = Result<String, Box<dyn error::Error>>;
|
|
|
|
|
|
|
|
fn process_airdrop(
|
|
|
|
rpc_client: &RpcClient,
|
|
|
|
config: &WalletConfig,
|
|
|
|
drone_addr: SocketAddr,
|
2019-03-05 17:22:46 -08:00
|
|
|
lamports: u64,
|
2019-02-09 13:15:44 -08:00
|
|
|
) -> ProcessResult {
|
|
|
|
println!(
|
2019-03-05 17:22:46 -08:00
|
|
|
"Requesting airdrop of {:?} lamports from {}",
|
|
|
|
lamports, drone_addr
|
2019-02-09 13:15:44 -08:00
|
|
|
);
|
2019-03-15 23:49:28 -07:00
|
|
|
let previous_balance = match rpc_client.retry_get_balance(&config.id.pubkey(), 5)? {
|
2019-03-05 17:22:46 -08:00
|
|
|
Some(lamports) => lamports,
|
2019-02-09 13:15:44 -08:00
|
|
|
None => Err(WalletError::RpcRequestError(
|
|
|
|
"Received result of an unexpected type".to_string(),
|
|
|
|
))?,
|
|
|
|
};
|
|
|
|
|
2019-03-14 19:18:28 -07:00
|
|
|
request_and_confirm_airdrop(&rpc_client, &drone_addr, &config.id.pubkey(), lamports)?;
|
2019-02-09 13:15:44 -08:00
|
|
|
|
|
|
|
let current_balance = rpc_client
|
2019-03-15 23:49:28 -07:00
|
|
|
.retry_get_balance(&config.id.pubkey(), 5)?
|
2019-02-09 13:15:44 -08:00
|
|
|
.unwrap_or(previous_balance);
|
|
|
|
|
|
|
|
if current_balance < previous_balance {
|
|
|
|
Err(format!(
|
|
|
|
"Airdrop failed: current_balance({}) < previous_balance({})",
|
|
|
|
current_balance, previous_balance
|
|
|
|
))?;
|
|
|
|
}
|
2019-03-05 17:22:46 -08:00
|
|
|
if current_balance - previous_balance < lamports {
|
2019-02-09 13:15:44 -08:00
|
|
|
Err(format!(
|
|
|
|
"Airdrop failed: Account balance increased by {} instead of {}",
|
|
|
|
current_balance - previous_balance,
|
2019-03-05 17:22:46 -08:00
|
|
|
lamports
|
2019-02-09 13:15:44 -08:00
|
|
|
))?;
|
|
|
|
}
|
|
|
|
Ok(format!("Your balance is: {:?}", current_balance))
|
|
|
|
}
|
|
|
|
|
2019-03-20 09:44:16 -07:00
|
|
|
fn process_balance(pubkey: &Pubkey, rpc_client: &RpcClient) -> ProcessResult {
|
|
|
|
let balance = rpc_client.retry_get_balance(pubkey, 5)?;
|
2019-02-09 13:15:44 -08:00
|
|
|
match balance {
|
2019-03-20 09:44:16 -07:00
|
|
|
Some(lamports) => {
|
|
|
|
let ess = if lamports == 1 { "" } else { "s" };
|
|
|
|
Ok(format!("{:?} lamport{}", lamports, ess))
|
|
|
|
}
|
2019-02-09 13:15:44 -08:00
|
|
|
None => Err(WalletError::RpcRequestError(
|
|
|
|
"Received result of an unexpected type".to_string(),
|
|
|
|
))?,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
fn process_confirm(rpc_client: &RpcClient, signature: Signature) -> ProcessResult {
|
2019-03-17 00:40:45 -07:00
|
|
|
match rpc_client.get_signature_status(&signature.to_string()) {
|
|
|
|
Ok(status) => {
|
|
|
|
if status == solana_client::rpc_signature_status::RpcSignatureStatus::Confirmed {
|
2019-02-09 13:15:44 -08:00
|
|
|
Ok("Confirmed".to_string())
|
|
|
|
} else {
|
|
|
|
Ok("Not found".to_string())
|
|
|
|
}
|
|
|
|
}
|
2019-03-17 00:40:45 -07:00
|
|
|
Err(err) => Err(WalletError::RpcRequestError(format!(
|
|
|
|
"Unable to confirm: {:?}",
|
|
|
|
err
|
|
|
|
)))?,
|
2019-02-09 13:15:44 -08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-03-06 17:02:20 -08:00
|
|
|
fn process_configure_staking(
|
|
|
|
rpc_client: &RpcClient,
|
|
|
|
config: &WalletConfig,
|
|
|
|
delegate_option: Option<Pubkey>,
|
|
|
|
authorized_voter_option: Option<Pubkey>,
|
|
|
|
) -> ProcessResult {
|
2019-03-16 17:17:44 -07:00
|
|
|
let recent_blockhash = rpc_client.get_recent_blockhash()?;
|
2019-03-15 09:54:56 -07:00
|
|
|
let mut ixs = vec![];
|
2019-03-06 17:02:20 -08:00
|
|
|
if let Some(delegate_id) = delegate_option {
|
2019-03-15 09:54:56 -07:00
|
|
|
ixs.push(VoteInstruction::new_delegate_stake(
|
2019-03-09 19:28:43 -08:00
|
|
|
&config.id.pubkey(),
|
|
|
|
&delegate_id,
|
2019-03-06 17:02:20 -08:00
|
|
|
));
|
|
|
|
}
|
|
|
|
if let Some(authorized_voter_id) = authorized_voter_option {
|
2019-03-15 09:54:56 -07:00
|
|
|
ixs.push(VoteInstruction::new_authorize_voter(
|
2019-03-09 19:28:43 -08:00
|
|
|
&config.id.pubkey(),
|
|
|
|
&authorized_voter_id,
|
2019-03-06 17:02:20 -08:00
|
|
|
));
|
|
|
|
}
|
2019-03-25 15:08:22 -07:00
|
|
|
let mut tx = Transaction::new_signed_instructions(&[&config.id], ixs, recent_blockhash, 0);
|
2019-03-16 17:17:44 -07:00
|
|
|
let signature_str = rpc_client.send_and_confirm_transaction(&mut tx, &config.id)?;
|
2019-03-06 17:02:20 -08:00
|
|
|
Ok(signature_str.to_string())
|
|
|
|
}
|
|
|
|
|
|
|
|
fn process_create_staking(
|
|
|
|
rpc_client: &RpcClient,
|
|
|
|
config: &WalletConfig,
|
2019-03-09 19:28:43 -08:00
|
|
|
voting_account_id: &Pubkey,
|
2019-03-06 17:02:20 -08:00
|
|
|
lamports: u64,
|
|
|
|
) -> ProcessResult {
|
2019-03-16 17:17:44 -07:00
|
|
|
let recent_blockhash = rpc_client.get_recent_blockhash()?;
|
2019-03-25 15:08:22 -07:00
|
|
|
let ixs = VoteInstruction::new_account(&config.id.pubkey(), voting_account_id, lamports);
|
|
|
|
let mut tx = Transaction::new_signed_instructions(&[&config.id], ixs, recent_blockhash, 0);
|
2019-03-16 17:17:44 -07:00
|
|
|
let signature_str = rpc_client.send_and_confirm_transaction(&mut tx, &config.id)?;
|
2019-03-06 17:02:20 -08:00
|
|
|
Ok(signature_str.to_string())
|
|
|
|
}
|
|
|
|
|
2019-02-09 13:15:44 -08:00
|
|
|
fn process_deploy(
|
|
|
|
rpc_client: &RpcClient,
|
|
|
|
config: &WalletConfig,
|
|
|
|
program_location: &str,
|
|
|
|
) -> ProcessResult {
|
2019-03-15 23:49:28 -07:00
|
|
|
let balance = rpc_client.retry_get_balance(&config.id.pubkey(), 5)?;
|
2019-03-05 17:22:46 -08:00
|
|
|
if let Some(lamports) = balance {
|
|
|
|
if lamports < 1 {
|
2019-02-09 13:15:44 -08:00
|
|
|
Err(WalletError::DynamicProgramError(
|
|
|
|
"Insufficient funds".to_string(),
|
|
|
|
))?
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-03-16 17:17:44 -07:00
|
|
|
let blockhash = rpc_client.get_recent_blockhash()?;
|
2019-02-09 13:15:44 -08:00
|
|
|
let program_id = Keypair::new();
|
|
|
|
let mut file = File::open(program_location).map_err(|err| {
|
|
|
|
WalletError::DynamicProgramError(
|
|
|
|
format!("Unable to open program file: {}", err).to_string(),
|
|
|
|
)
|
|
|
|
})?;
|
2019-03-14 09:48:27 -07:00
|
|
|
let mut program_data = Vec::new();
|
|
|
|
file.read_to_end(&mut program_data).map_err(|err| {
|
2019-02-09 13:15:44 -08:00
|
|
|
WalletError::DynamicProgramError(
|
|
|
|
format!("Unable to read program file: {}", err).to_string(),
|
|
|
|
)
|
|
|
|
})?;
|
|
|
|
|
|
|
|
let mut tx = SystemTransaction::new_program_account(
|
|
|
|
&config.id,
|
2019-03-09 19:28:43 -08:00
|
|
|
&program_id.pubkey(),
|
2019-03-02 10:25:16 -08:00
|
|
|
blockhash,
|
2019-02-09 13:15:44 -08:00
|
|
|
1,
|
2019-03-14 09:48:27 -07:00
|
|
|
program_data.len() as u64,
|
2019-03-09 19:28:43 -08:00
|
|
|
&bpf_loader::id(),
|
2019-02-09 13:15:44 -08:00
|
|
|
0,
|
|
|
|
);
|
2019-02-27 11:17:32 -08:00
|
|
|
trace!("Creating program account");
|
2019-03-16 17:17:44 -07:00
|
|
|
rpc_client
|
|
|
|
.send_and_confirm_transaction(&mut tx, &config.id)
|
|
|
|
.map_err(|_| {
|
|
|
|
WalletError::DynamicProgramError("Program allocate space failed".to_string())
|
|
|
|
})?;
|
2019-02-09 13:15:44 -08:00
|
|
|
|
2019-02-27 11:17:32 -08:00
|
|
|
trace!("Writing program data");
|
2019-03-14 09:48:27 -07:00
|
|
|
let write_transactions: Vec<_> = program_data
|
2019-02-27 11:17:32 -08:00
|
|
|
.chunks(USERDATA_CHUNK_SIZE)
|
|
|
|
.zip(0..)
|
|
|
|
.map(|(chunk, i)| {
|
2019-03-21 07:14:14 -07:00
|
|
|
let instruction = LoaderInstruction::new_write(
|
|
|
|
&program_id.pubkey(),
|
2019-03-09 19:28:43 -08:00
|
|
|
&bpf_loader::id(),
|
2019-02-27 11:17:32 -08:00
|
|
|
(i * USERDATA_CHUNK_SIZE) as u32,
|
|
|
|
chunk.to_vec(),
|
2019-03-21 07:14:14 -07:00
|
|
|
);
|
|
|
|
Transaction::new_signed_instructions(&[&program_id], vec![instruction], blockhash, 0)
|
2019-02-27 11:17:32 -08:00
|
|
|
})
|
|
|
|
.collect();
|
2019-03-16 17:17:44 -07:00
|
|
|
rpc_client.send_and_confirm_transactions(write_transactions, &program_id)?;
|
2019-02-09 13:15:44 -08:00
|
|
|
|
2019-02-27 11:17:32 -08:00
|
|
|
trace!("Finalizing program account");
|
2019-03-21 07:14:14 -07:00
|
|
|
let instruction = LoaderInstruction::new_finalize(&program_id.pubkey(), &bpf_loader::id());
|
|
|
|
let mut tx =
|
|
|
|
Transaction::new_signed_instructions(&[&program_id], vec![instruction], blockhash, 0);
|
2019-03-16 17:17:44 -07:00
|
|
|
rpc_client
|
|
|
|
.send_and_confirm_transaction(&mut tx, &program_id)
|
|
|
|
.map_err(|_| {
|
|
|
|
WalletError::DynamicProgramError("Program finalize transaction failed".to_string())
|
|
|
|
})?;
|
2019-02-09 13:15:44 -08:00
|
|
|
|
|
|
|
Ok(json!({
|
|
|
|
"programId": format!("{}", program_id.pubkey()),
|
|
|
|
})
|
|
|
|
.to_string())
|
|
|
|
}
|
|
|
|
|
|
|
|
fn process_pay(
|
|
|
|
rpc_client: &RpcClient,
|
|
|
|
config: &WalletConfig,
|
2019-03-05 17:22:46 -08:00
|
|
|
lamports: u64,
|
2019-03-09 19:28:43 -08:00
|
|
|
to: &Pubkey,
|
2019-02-09 13:15:44 -08:00
|
|
|
timestamp: Option<DateTime<Utc>>,
|
|
|
|
timestamp_pubkey: Option<Pubkey>,
|
|
|
|
witnesses: &Option<Vec<Pubkey>>,
|
|
|
|
cancelable: Option<Pubkey>,
|
|
|
|
) -> ProcessResult {
|
2019-03-16 17:17:44 -07:00
|
|
|
let blockhash = rpc_client.get_recent_blockhash()?;
|
2019-02-09 13:15:44 -08:00
|
|
|
|
|
|
|
if timestamp == None && *witnesses == None {
|
2019-03-05 17:22:46 -08:00
|
|
|
let mut tx = SystemTransaction::new_move(&config.id, to, lamports, blockhash, 0);
|
2019-03-16 17:17:44 -07:00
|
|
|
let signature_str = rpc_client.send_and_confirm_transaction(&mut tx, &config.id)?;
|
2019-02-09 13:15:44 -08:00
|
|
|
Ok(signature_str.to_string())
|
|
|
|
} else if *witnesses == None {
|
|
|
|
let dt = timestamp.unwrap();
|
|
|
|
let dt_pubkey = match timestamp_pubkey {
|
|
|
|
Some(pubkey) => pubkey,
|
|
|
|
None => config.id.pubkey(),
|
|
|
|
};
|
|
|
|
|
|
|
|
let contract_state = Keypair::new();
|
|
|
|
|
|
|
|
// Initializing contract
|
2019-03-23 05:14:30 -07:00
|
|
|
let ixs = BudgetInstruction::new_on_date(
|
|
|
|
&config.id.pubkey(),
|
2019-02-09 13:15:44 -08:00
|
|
|
to,
|
2019-03-09 19:28:43 -08:00
|
|
|
&contract_state.pubkey(),
|
2019-02-09 13:15:44 -08:00
|
|
|
dt,
|
2019-03-09 19:28:43 -08:00
|
|
|
&dt_pubkey,
|
2019-02-09 13:15:44 -08:00
|
|
|
cancelable,
|
2019-03-05 17:22:46 -08:00
|
|
|
lamports,
|
2019-02-09 13:15:44 -08:00
|
|
|
);
|
2019-03-23 05:14:30 -07:00
|
|
|
let mut tx = Transaction::new_signed_instructions(&[&config.id], ixs, blockhash, 0);
|
2019-03-16 17:17:44 -07:00
|
|
|
let signature_str = rpc_client.send_and_confirm_transaction(&mut tx, &config.id)?;
|
2019-02-09 13:15:44 -08:00
|
|
|
|
|
|
|
Ok(json!({
|
|
|
|
"signature": signature_str,
|
|
|
|
"processId": format!("{}", contract_state.pubkey()),
|
|
|
|
})
|
|
|
|
.to_string())
|
|
|
|
} else if timestamp == None {
|
2019-03-16 17:17:44 -07:00
|
|
|
let blockhash = rpc_client.get_recent_blockhash()?;
|
2019-02-09 13:15:44 -08:00
|
|
|
|
|
|
|
let witness = if let Some(ref witness_vec) = *witnesses {
|
|
|
|
witness_vec[0]
|
|
|
|
} else {
|
|
|
|
Err(WalletError::BadParameter(
|
|
|
|
"Could not parse required signature pubkey(s)".to_string(),
|
|
|
|
))?
|
|
|
|
};
|
|
|
|
|
|
|
|
let contract_state = Keypair::new();
|
|
|
|
|
|
|
|
// Initializing contract
|
2019-03-23 05:14:30 -07:00
|
|
|
let ixs = BudgetInstruction::new_when_signed(
|
|
|
|
&config.id.pubkey(),
|
2019-02-09 13:15:44 -08:00
|
|
|
to,
|
2019-03-09 19:28:43 -08:00
|
|
|
&contract_state.pubkey(),
|
|
|
|
&witness,
|
2019-02-09 13:15:44 -08:00
|
|
|
cancelable,
|
2019-03-05 17:22:46 -08:00
|
|
|
lamports,
|
2019-02-09 13:15:44 -08:00
|
|
|
);
|
2019-03-23 05:14:30 -07:00
|
|
|
let mut tx = Transaction::new_signed_instructions(&[&config.id], ixs, blockhash, 0);
|
2019-03-16 17:17:44 -07:00
|
|
|
let signature_str = rpc_client.send_and_confirm_transaction(&mut tx, &config.id)?;
|
2019-02-09 13:15:44 -08:00
|
|
|
|
|
|
|
Ok(json!({
|
|
|
|
"signature": signature_str,
|
|
|
|
"processId": format!("{}", contract_state.pubkey()),
|
|
|
|
})
|
|
|
|
.to_string())
|
|
|
|
} else {
|
|
|
|
Ok("Combo transactions not yet handled".to_string())
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-03-09 19:28:43 -08:00
|
|
|
fn process_cancel(rpc_client: &RpcClient, config: &WalletConfig, pubkey: &Pubkey) -> ProcessResult {
|
2019-03-16 17:17:44 -07:00
|
|
|
let blockhash = rpc_client.get_recent_blockhash()?;
|
2019-03-23 05:14:30 -07:00
|
|
|
let ix =
|
|
|
|
BudgetInstruction::new_apply_signature(&config.id.pubkey(), pubkey, &config.id.pubkey());
|
|
|
|
let mut tx = Transaction::new_signed_instructions(&[&config.id], vec![ix], blockhash, 0);
|
2019-03-16 17:17:44 -07:00
|
|
|
let signature_str = rpc_client.send_and_confirm_transaction(&mut tx, &config.id)?;
|
2019-02-09 13:15:44 -08:00
|
|
|
Ok(signature_str.to_string())
|
|
|
|
}
|
|
|
|
|
|
|
|
fn process_get_transaction_count(rpc_client: &RpcClient) -> ProcessResult {
|
2019-03-17 00:40:45 -07:00
|
|
|
let transaction_count = rpc_client.get_transaction_count()?;
|
|
|
|
Ok(transaction_count.to_string())
|
2019-02-09 13:15:44 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
fn process_time_elapsed(
|
|
|
|
rpc_client: &RpcClient,
|
|
|
|
config: &WalletConfig,
|
|
|
|
drone_addr: SocketAddr,
|
2019-03-09 19:28:43 -08:00
|
|
|
to: &Pubkey,
|
|
|
|
pubkey: &Pubkey,
|
2019-02-09 13:15:44 -08:00
|
|
|
dt: DateTime<Utc>,
|
|
|
|
) -> ProcessResult {
|
2019-03-15 23:49:28 -07:00
|
|
|
let balance = rpc_client.retry_get_balance(&config.id.pubkey(), 5)?;
|
2019-02-09 13:15:44 -08:00
|
|
|
|
|
|
|
if let Some(0) = balance {
|
2019-03-14 19:18:28 -07:00
|
|
|
request_and_confirm_airdrop(&rpc_client, &drone_addr, &config.id.pubkey(), 1)?;
|
2019-02-09 13:15:44 -08:00
|
|
|
}
|
|
|
|
|
2019-03-16 17:17:44 -07:00
|
|
|
let blockhash = rpc_client.get_recent_blockhash()?;
|
2019-02-09 13:15:44 -08:00
|
|
|
|
2019-03-23 05:14:30 -07:00
|
|
|
let ix = BudgetInstruction::new_apply_timestamp(&config.id.pubkey(), pubkey, to, dt);
|
|
|
|
let mut tx = Transaction::new_signed_instructions(&[&config.id], vec![ix], blockhash, 0);
|
2019-03-16 17:17:44 -07:00
|
|
|
let signature_str = rpc_client.send_and_confirm_transaction(&mut tx, &config.id)?;
|
2019-02-09 13:15:44 -08:00
|
|
|
|
|
|
|
Ok(signature_str.to_string())
|
|
|
|
}
|
|
|
|
|
|
|
|
fn process_witness(
|
|
|
|
rpc_client: &RpcClient,
|
|
|
|
config: &WalletConfig,
|
|
|
|
drone_addr: SocketAddr,
|
2019-03-09 19:28:43 -08:00
|
|
|
to: &Pubkey,
|
|
|
|
pubkey: &Pubkey,
|
2019-02-09 13:15:44 -08:00
|
|
|
) -> ProcessResult {
|
2019-03-15 23:49:28 -07:00
|
|
|
let balance = rpc_client.retry_get_balance(&config.id.pubkey(), 5)?;
|
2019-02-09 13:15:44 -08:00
|
|
|
|
|
|
|
if let Some(0) = balance {
|
2019-03-14 19:18:28 -07:00
|
|
|
request_and_confirm_airdrop(&rpc_client, &drone_addr, &config.id.pubkey(), 1)?;
|
2019-02-09 13:15:44 -08:00
|
|
|
}
|
|
|
|
|
2019-03-16 17:17:44 -07:00
|
|
|
let blockhash = rpc_client.get_recent_blockhash()?;
|
2019-03-23 05:14:30 -07:00
|
|
|
let ix = BudgetInstruction::new_apply_signature(&config.id.pubkey(), pubkey, to);
|
|
|
|
let mut tx = Transaction::new_signed_instructions(&[&config.id], vec![ix], blockhash, 0);
|
2019-03-16 17:17:44 -07:00
|
|
|
let signature_str = rpc_client.send_and_confirm_transaction(&mut tx, &config.id)?;
|
2019-02-09 13:15:44 -08:00
|
|
|
|
|
|
|
Ok(signature_str.to_string())
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn process_command(config: &WalletConfig) -> ProcessResult {
|
2018-11-09 15:37:34 -08:00
|
|
|
if let WalletCommand::Address = config.command {
|
2018-11-09 14:42:11 -08:00
|
|
|
// Get address of this client
|
2018-11-09 15:37:34 -08:00
|
|
|
return Ok(format!("{}", config.id.pubkey()));
|
2018-11-09 14:42:11 -08:00
|
|
|
}
|
|
|
|
|
2019-01-04 13:25:47 -08:00
|
|
|
let drone_addr = config.drone_addr();
|
2019-03-16 21:51:41 -07:00
|
|
|
|
|
|
|
let mut _rpc_client;
|
2019-01-13 23:10:03 -08:00
|
|
|
let rpc_client = if config.rpc_client.is_none() {
|
2019-01-11 14:37:52 -08:00
|
|
|
let rpc_addr = config.rpc_addr();
|
2019-03-16 21:51:41 -07:00
|
|
|
_rpc_client = RpcClient::new(rpc_addr);
|
|
|
|
&_rpc_client
|
2019-01-13 23:10:03 -08:00
|
|
|
} else {
|
|
|
|
// Primarily for testing
|
2019-03-16 21:51:41 -07:00
|
|
|
config.rpc_client.as_ref().unwrap()
|
2019-01-13 23:10:03 -08:00
|
|
|
};
|
2018-11-09 14:36:08 -08:00
|
|
|
|
2018-09-14 01:58:39 -07:00
|
|
|
match config.command {
|
2018-09-20 22:27:06 -07:00
|
|
|
// Get address of this client
|
2018-11-09 14:42:11 -08:00
|
|
|
WalletCommand::Address => unreachable!(),
|
2018-11-14 18:57:34 -08:00
|
|
|
|
2019-02-09 13:15:44 -08:00
|
|
|
// Request an airdrop from Solana Drone;
|
2019-03-05 17:22:46 -08:00
|
|
|
WalletCommand::Airdrop(lamports) => {
|
|
|
|
process_airdrop(&rpc_client, config, drone_addr, lamports)
|
|
|
|
}
|
2018-11-14 18:57:34 -08:00
|
|
|
|
2018-09-18 14:42:32 -07:00
|
|
|
// Check client balance
|
2019-03-20 09:44:16 -07:00
|
|
|
WalletCommand::Balance(pubkey) => process_balance(&pubkey, &rpc_client),
|
2019-02-09 13:15:44 -08:00
|
|
|
|
2018-09-18 14:42:32 -07:00
|
|
|
// Cancel a contract by contract Pubkey
|
2019-03-09 19:28:43 -08:00
|
|
|
WalletCommand::Cancel(pubkey) => process_cancel(&rpc_client, config, &pubkey),
|
2019-02-09 13:15:44 -08:00
|
|
|
|
2018-09-14 01:58:39 -07:00
|
|
|
// Confirm the last client transaction by signature
|
2019-02-09 13:15:44 -08:00
|
|
|
WalletCommand::Confirm(signature) => process_confirm(&rpc_client, signature),
|
|
|
|
|
2019-03-06 17:02:20 -08:00
|
|
|
// Configure staking account already created
|
|
|
|
WalletCommand::ConfigureStakingAccount(delegate_option, authorized_voter_option) => {
|
|
|
|
process_configure_staking(
|
|
|
|
&rpc_client,
|
|
|
|
config,
|
|
|
|
delegate_option,
|
|
|
|
authorized_voter_option,
|
|
|
|
)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Create staking account
|
|
|
|
WalletCommand::CreateStakingAccount(voting_account_id, lamports) => {
|
2019-03-09 19:28:43 -08:00
|
|
|
process_create_staking(&rpc_client, config, &voting_account_id, lamports)
|
2019-03-06 17:02:20 -08:00
|
|
|
}
|
|
|
|
|
2018-10-22 21:21:33 -07:00
|
|
|
// Deploy a custom program to the chain
|
|
|
|
WalletCommand::Deploy(ref program_location) => {
|
2019-02-09 13:15:44 -08:00
|
|
|
process_deploy(&rpc_client, config, program_location)
|
|
|
|
}
|
2018-10-22 21:21:33 -07:00
|
|
|
|
2019-02-09 13:15:44 -08:00
|
|
|
WalletCommand::GetTransactionCount => process_get_transaction_count(&rpc_client),
|
2018-10-24 09:01:19 -07:00
|
|
|
|
2019-03-05 17:22:46 -08:00
|
|
|
// If client has positive balance, pay lamports to another address
|
|
|
|
WalletCommand::Pay(
|
|
|
|
lamports,
|
|
|
|
to,
|
|
|
|
timestamp,
|
|
|
|
timestamp_pubkey,
|
|
|
|
ref witnesses,
|
|
|
|
cancelable,
|
|
|
|
) => process_pay(
|
|
|
|
&rpc_client,
|
|
|
|
config,
|
|
|
|
lamports,
|
2019-03-09 19:28:43 -08:00
|
|
|
&to,
|
2019-03-05 17:22:46 -08:00
|
|
|
timestamp,
|
|
|
|
timestamp_pubkey,
|
|
|
|
witnesses,
|
|
|
|
cancelable,
|
|
|
|
),
|
2019-02-09 13:15:44 -08:00
|
|
|
|
2018-09-19 16:44:03 -07:00
|
|
|
// Apply time elapsed to contract
|
2018-09-22 16:51:21 -07:00
|
|
|
WalletCommand::TimeElapsed(to, pubkey, dt) => {
|
2019-03-09 19:28:43 -08:00
|
|
|
process_time_elapsed(&rpc_client, config, drone_addr, &to, &pubkey, dt)
|
2018-09-14 01:58:39 -07:00
|
|
|
}
|
2019-02-09 13:15:44 -08:00
|
|
|
|
2018-09-18 14:42:32 -07:00
|
|
|
// Apply witness signature to contract
|
2018-09-24 09:23:16 -07:00
|
|
|
WalletCommand::Witness(to, pubkey) => {
|
2019-03-09 19:28:43 -08:00
|
|
|
process_witness(&rpc_client, config, drone_addr, &to, &pubkey)
|
2018-09-19 16:44:03 -07:00
|
|
|
}
|
2018-09-14 01:58:39 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-03-15 06:39:42 -07:00
|
|
|
// Quick and dirty Keypair that assumes the client will do retries but not update the
|
|
|
|
// blockhash. If the client updates the blockhash, the signature will be invalid.
|
|
|
|
// TODO: Parse `msg` and use that data to make a new airdrop request.
|
|
|
|
struct DroneKeypair {
|
|
|
|
transaction: Transaction,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl DroneKeypair {
|
|
|
|
fn new_keypair(
|
|
|
|
drone_addr: &SocketAddr,
|
|
|
|
to_pubkey: &Pubkey,
|
|
|
|
lamports: u64,
|
|
|
|
blockhash: Hash,
|
|
|
|
) -> Result<Self, Box<dyn error::Error>> {
|
|
|
|
let transaction = request_airdrop_transaction(drone_addr, to_pubkey, lamports, blockhash)?;
|
|
|
|
Ok(Self { transaction })
|
|
|
|
}
|
|
|
|
|
|
|
|
fn airdrop_transaction(&self) -> Transaction {
|
|
|
|
self.transaction.clone()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl KeypairUtil for DroneKeypair {
|
|
|
|
fn new() -> Self {
|
|
|
|
unimplemented!();
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Return the public key of the keypair used to sign votes
|
|
|
|
fn pubkey(&self) -> Pubkey {
|
|
|
|
self.transaction.account_keys[0]
|
|
|
|
}
|
|
|
|
|
|
|
|
fn sign_message(&self, _msg: &[u8]) -> Signature {
|
|
|
|
self.transaction.signatures[0]
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-01-13 23:10:03 -08:00
|
|
|
pub fn request_and_confirm_airdrop(
|
2018-12-10 11:20:55 -08:00
|
|
|
rpc_client: &RpcClient,
|
|
|
|
drone_addr: &SocketAddr,
|
2019-03-14 19:18:28 -07:00
|
|
|
to_pubkey: &Pubkey,
|
2019-03-05 17:22:46 -08:00
|
|
|
lamports: u64,
|
2018-12-10 11:20:55 -08:00
|
|
|
) -> Result<(), Box<dyn error::Error>> {
|
2019-03-16 17:17:44 -07:00
|
|
|
let blockhash = rpc_client.get_recent_blockhash()?;
|
2019-03-15 06:39:42 -07:00
|
|
|
let keypair = DroneKeypair::new_keypair(drone_addr, to_pubkey, lamports, blockhash)?;
|
|
|
|
let mut tx = keypair.airdrop_transaction();
|
2019-03-16 17:17:44 -07:00
|
|
|
rpc_client.send_and_confirm_transaction(&mut tx, &keypair)?;
|
2019-01-15 16:24:09 -08:00
|
|
|
Ok(())
|
2018-10-22 21:21:33 -07:00
|
|
|
}
|
|
|
|
|
2018-09-14 01:58:39 -07:00
|
|
|
#[cfg(test)]
|
2018-09-14 01:59:09 -07:00
|
|
|
mod tests {
|
|
|
|
use super::*;
|
2019-03-06 20:29:08 -08:00
|
|
|
use clap::{App, Arg, ArgGroup, SubCommand};
|
2018-10-08 11:10:47 -07:00
|
|
|
use serde_json::Value;
|
2019-03-16 17:17:44 -07:00
|
|
|
use solana_client::mock_rpc_client_request::SIGNATURE;
|
2018-12-12 13:13:18 -08:00
|
|
|
use solana_sdk::signature::{gen_keypair_file, read_keypair, read_pkcs8, Keypair, KeypairUtil};
|
|
|
|
use std::fs;
|
2019-01-13 23:10:03 -08:00
|
|
|
use std::net::{Ipv4Addr, SocketAddr};
|
2019-01-17 08:49:16 -08:00
|
|
|
use std::path::{Path, PathBuf};
|
2018-09-14 01:59:09 -07:00
|
|
|
|
2018-12-10 11:20:55 -08:00
|
|
|
#[test]
|
2019-01-13 23:10:03 -08:00
|
|
|
fn test_wallet_config_drone_addr() {
|
|
|
|
let mut config = WalletConfig::default();
|
2019-01-16 20:43:00 -08:00
|
|
|
assert_eq!(
|
|
|
|
config.drone_addr(),
|
|
|
|
SocketAddr::new(config.host, config.drone_port)
|
|
|
|
);
|
|
|
|
|
|
|
|
config.drone_port = 1234;
|
|
|
|
assert_eq!(config.drone_addr(), SocketAddr::new(config.host, 1234));
|
|
|
|
|
|
|
|
config.drone_host = Some(IpAddr::V4(Ipv4Addr::new(127, 0, 0, 2)));
|
|
|
|
assert_eq!(
|
|
|
|
config.drone_addr(),
|
|
|
|
SocketAddr::new(config.drone_host.unwrap(), 1234)
|
|
|
|
);
|
|
|
|
}
|
2019-01-11 14:37:52 -08:00
|
|
|
|
2019-01-16 20:43:00 -08:00
|
|
|
#[test]
|
|
|
|
fn test_wallet_config_rpc_addr() {
|
|
|
|
let mut config = WalletConfig::default();
|
|
|
|
assert_eq!(config.rpc_addr(), "http://127.0.0.1:8899");
|
|
|
|
config.rpc_port = 1234;
|
|
|
|
assert_eq!(config.rpc_addr(), "http://127.0.0.1:1234");
|
|
|
|
config.rpc_host = Some(IpAddr::V4(Ipv4Addr::new(127, 0, 0, 2)));
|
|
|
|
assert_eq!(config.rpc_addr(), "http://127.0.0.2:1234");
|
2018-12-10 11:20:55 -08:00
|
|
|
}
|
|
|
|
|
2018-09-14 01:59:09 -07:00
|
|
|
#[test]
|
2018-09-26 10:17:45 -07:00
|
|
|
fn test_wallet_parse_command() {
|
2018-09-14 01:59:09 -07:00
|
|
|
let test_commands = App::new("test")
|
2018-09-19 12:44:16 -07:00
|
|
|
.subcommand(SubCommand::with_name("address").about("Get your public key"))
|
2018-09-14 01:59:09 -07:00
|
|
|
.subcommand(
|
|
|
|
SubCommand::with_name("airdrop")
|
2019-03-05 17:22:46 -08:00
|
|
|
.about("Request a batch of lamports")
|
2018-09-14 01:59:09 -07:00
|
|
|
.arg(
|
2019-03-05 17:22:46 -08:00
|
|
|
Arg::with_name("lamports")
|
2018-09-19 12:44:16 -07:00
|
|
|
.index(1)
|
|
|
|
.value_name("NUM")
|
2018-09-14 01:59:09 -07:00
|
|
|
.takes_value(true)
|
|
|
|
.required(true)
|
2019-03-05 17:22:46 -08:00
|
|
|
.help("The number of lamports to request"),
|
2018-09-14 01:59:09 -07:00
|
|
|
),
|
2018-12-07 19:01:28 -08:00
|
|
|
)
|
|
|
|
.subcommand(SubCommand::with_name("balance").about("Get your balance"))
|
2018-09-19 12:44:16 -07:00
|
|
|
.subcommand(
|
|
|
|
SubCommand::with_name("cancel")
|
|
|
|
.about("Cancel a transfer")
|
|
|
|
.arg(
|
2019-03-07 16:00:12 -08:00
|
|
|
Arg::with_name("process_id")
|
2018-09-19 12:44:16 -07:00
|
|
|
.index(1)
|
|
|
|
.value_name("PROCESS_ID")
|
|
|
|
.takes_value(true)
|
|
|
|
.required(true)
|
|
|
|
.help("The process id of the transfer to cancel"),
|
|
|
|
),
|
2018-12-07 19:01:28 -08:00
|
|
|
)
|
|
|
|
.subcommand(
|
2018-09-19 12:44:16 -07:00
|
|
|
SubCommand::with_name("confirm")
|
|
|
|
.about("Confirm transaction by signature")
|
|
|
|
.arg(
|
|
|
|
Arg::with_name("signature")
|
|
|
|
.index(1)
|
|
|
|
.value_name("SIGNATURE")
|
|
|
|
.takes_value(true)
|
|
|
|
.required(true)
|
|
|
|
.help("The transaction signature to confirm"),
|
|
|
|
),
|
2018-12-07 19:01:28 -08:00
|
|
|
)
|
2019-03-06 20:29:08 -08:00
|
|
|
.subcommand(
|
|
|
|
SubCommand::with_name("configure-staking-account")
|
|
|
|
.about("Configure staking account for node")
|
|
|
|
.group(
|
|
|
|
ArgGroup::with_name("options")
|
|
|
|
.args(&["delegate", "authorize"])
|
|
|
|
.multiple(true)
|
|
|
|
.required(true),
|
|
|
|
)
|
|
|
|
.arg(
|
|
|
|
Arg::with_name("delegate")
|
|
|
|
.long("delegate-account")
|
|
|
|
.value_name("PUBKEY")
|
|
|
|
.takes_value(true)
|
|
|
|
.help("Address to delegate this vote account to"),
|
|
|
|
)
|
|
|
|
.arg(
|
|
|
|
Arg::with_name("authorize")
|
|
|
|
.long("authorize-voter")
|
|
|
|
.value_name("PUBKEY")
|
|
|
|
.takes_value(true)
|
|
|
|
.help("Vote signer to authorize"),
|
|
|
|
),
|
|
|
|
)
|
|
|
|
.subcommand(
|
|
|
|
SubCommand::with_name("create-staking-account")
|
|
|
|
.about("Create staking account for node")
|
|
|
|
.arg(
|
2019-03-07 16:00:12 -08:00
|
|
|
Arg::with_name("voting_account_id")
|
2019-03-06 20:29:08 -08:00
|
|
|
.index(1)
|
|
|
|
.value_name("PUBKEY")
|
|
|
|
.takes_value(true)
|
|
|
|
.required(true)
|
|
|
|
.help("Staking account address to fund"),
|
|
|
|
)
|
|
|
|
.arg(
|
|
|
|
Arg::with_name("lamports")
|
|
|
|
.index(2)
|
|
|
|
.value_name("NUM")
|
|
|
|
.takes_value(true)
|
|
|
|
.required(true)
|
|
|
|
.help("The number of lamports to send to staking account"),
|
|
|
|
),
|
|
|
|
)
|
2018-12-07 19:01:28 -08:00
|
|
|
.subcommand(
|
2018-11-12 12:48:59 -08:00
|
|
|
SubCommand::with_name("deploy")
|
|
|
|
.about("Deploy a program")
|
|
|
|
.arg(
|
2019-03-07 16:00:12 -08:00
|
|
|
Arg::with_name("program_location")
|
2018-11-12 12:48:59 -08:00
|
|
|
.index(1)
|
|
|
|
.value_name("PATH")
|
|
|
|
.takes_value(true)
|
|
|
|
.required(true)
|
|
|
|
.help("/path/to/program.o"),
|
2018-12-03 13:32:31 -08:00
|
|
|
), // TODO: Add "loader" argument; current default is bpf_loader
|
2018-12-07 19:01:28 -08:00
|
|
|
)
|
|
|
|
.subcommand(
|
2018-11-12 12:48:59 -08:00
|
|
|
SubCommand::with_name("get-transaction-count")
|
|
|
|
.about("Get current transaction count"),
|
2018-12-07 19:01:28 -08:00
|
|
|
)
|
|
|
|
.subcommand(
|
2018-09-14 01:59:09 -07:00
|
|
|
SubCommand::with_name("pay")
|
|
|
|
.about("Send a payment")
|
|
|
|
.arg(
|
2018-09-19 12:44:16 -07:00
|
|
|
Arg::with_name("to")
|
|
|
|
.index(1)
|
|
|
|
.value_name("PUBKEY")
|
|
|
|
.takes_value(true)
|
|
|
|
.required(true)
|
|
|
|
.help("The pubkey of recipient"),
|
2018-12-07 19:01:28 -08:00
|
|
|
)
|
|
|
|
.arg(
|
2019-03-05 17:22:46 -08:00
|
|
|
Arg::with_name("lamports")
|
2018-09-19 12:44:16 -07:00
|
|
|
.index(2)
|
|
|
|
.value_name("NUM")
|
2018-09-14 01:59:09 -07:00
|
|
|
.takes_value(true)
|
|
|
|
.required(true)
|
2019-03-05 17:22:46 -08:00
|
|
|
.help("The number of lamports to send"),
|
2018-12-07 19:01:28 -08:00
|
|
|
)
|
|
|
|
.arg(
|
2018-09-19 12:44:16 -07:00
|
|
|
Arg::with_name("timestamp")
|
|
|
|
.long("after")
|
|
|
|
.value_name("DATETIME")
|
|
|
|
.takes_value(true)
|
|
|
|
.help("A timestamp after which transaction will execute"),
|
2018-12-07 19:01:28 -08:00
|
|
|
)
|
|
|
|
.arg(
|
2019-03-07 16:00:12 -08:00
|
|
|
Arg::with_name("timestamp_pubkey")
|
2018-09-19 12:44:16 -07:00
|
|
|
.long("require-timestamp-from")
|
2018-09-14 01:59:09 -07:00
|
|
|
.value_name("PUBKEY")
|
|
|
|
.takes_value(true)
|
2018-09-19 12:44:16 -07:00
|
|
|
.requires("timestamp")
|
|
|
|
.help("Require timestamp from this third party"),
|
2018-12-07 19:01:28 -08:00
|
|
|
)
|
|
|
|
.arg(
|
2018-09-19 12:44:16 -07:00
|
|
|
Arg::with_name("witness")
|
|
|
|
.long("require-signature-from")
|
|
|
|
.value_name("PUBKEY")
|
|
|
|
.takes_value(true)
|
|
|
|
.multiple(true)
|
|
|
|
.use_delimiter(true)
|
2019-03-05 17:22:46 -08:00
|
|
|
.help("Any third party signatures required to unlock the lamports"),
|
2018-12-07 19:01:28 -08:00
|
|
|
)
|
|
|
|
.arg(
|
2018-09-25 11:45:25 -07:00
|
|
|
Arg::with_name("cancelable")
|
|
|
|
.long("cancelable")
|
|
|
|
.takes_value(false),
|
2018-09-14 01:59:09 -07:00
|
|
|
),
|
2018-12-07 19:01:28 -08:00
|
|
|
)
|
|
|
|
.subcommand(
|
2018-09-19 12:44:16 -07:00
|
|
|
SubCommand::with_name("send-signature")
|
|
|
|
.about("Send a signature to authorize a transfer")
|
2018-09-14 01:59:09 -07:00
|
|
|
.arg(
|
2018-09-24 09:23:16 -07:00
|
|
|
Arg::with_name("to")
|
2018-09-14 01:59:09 -07:00
|
|
|
.index(1)
|
2018-09-24 09:23:16 -07:00
|
|
|
.value_name("PUBKEY")
|
|
|
|
.takes_value(true)
|
|
|
|
.required(true)
|
|
|
|
.help("The pubkey of recipient"),
|
2018-12-07 19:01:28 -08:00
|
|
|
)
|
|
|
|
.arg(
|
2019-03-07 16:00:12 -08:00
|
|
|
Arg::with_name("process_id")
|
2018-09-24 09:23:16 -07:00
|
|
|
.index(2)
|
2018-09-19 12:44:16 -07:00
|
|
|
.value_name("PROCESS_ID")
|
|
|
|
.takes_value(true)
|
2018-09-14 01:59:09 -07:00
|
|
|
.required(true)
|
2018-09-19 12:44:16 -07:00
|
|
|
.help("The process id of the transfer to authorize"),
|
2018-09-14 01:59:09 -07:00
|
|
|
),
|
2018-12-07 19:01:28 -08:00
|
|
|
)
|
|
|
|
.subcommand(
|
2018-09-19 12:44:16 -07:00
|
|
|
SubCommand::with_name("send-timestamp")
|
|
|
|
.about("Send a timestamp to unlock a transfer")
|
|
|
|
.arg(
|
2018-09-24 09:23:16 -07:00
|
|
|
Arg::with_name("to")
|
2018-09-19 12:44:16 -07:00
|
|
|
.index(1)
|
2018-09-24 09:23:16 -07:00
|
|
|
.value_name("PUBKEY")
|
|
|
|
.takes_value(true)
|
|
|
|
.required(true)
|
|
|
|
.help("The pubkey of recipient"),
|
2018-12-07 19:01:28 -08:00
|
|
|
)
|
|
|
|
.arg(
|
2019-03-07 16:00:12 -08:00
|
|
|
Arg::with_name("process_id")
|
2018-09-24 09:23:16 -07:00
|
|
|
.index(2)
|
2018-09-19 12:44:16 -07:00
|
|
|
.value_name("PROCESS_ID")
|
|
|
|
.takes_value(true)
|
|
|
|
.required(true)
|
|
|
|
.help("The process id of the transfer to unlock"),
|
2018-12-07 19:01:28 -08:00
|
|
|
)
|
|
|
|
.arg(
|
2018-09-19 12:44:16 -07:00
|
|
|
Arg::with_name("datetime")
|
|
|
|
.long("date")
|
|
|
|
.value_name("DATETIME")
|
|
|
|
.takes_value(true)
|
|
|
|
.help("Optional arbitrary timestamp to apply"),
|
|
|
|
),
|
|
|
|
);
|
|
|
|
let pubkey = Keypair::new().pubkey();
|
|
|
|
let pubkey_string = format!("{}", pubkey);
|
|
|
|
let witness0 = Keypair::new().pubkey();
|
|
|
|
let witness0_string = format!("{}", witness0);
|
|
|
|
let witness1 = Keypair::new().pubkey();
|
|
|
|
let witness1_string = format!("{}", witness1);
|
|
|
|
let dt = Utc.ymd(2018, 9, 19).and_hms(17, 30, 59);
|
2018-09-14 01:59:09 -07:00
|
|
|
|
2018-09-19 12:44:16 -07:00
|
|
|
// Test Airdrop Subcommand
|
2018-09-14 01:59:09 -07:00
|
|
|
let test_airdrop = test_commands
|
|
|
|
.clone()
|
2018-09-19 12:44:16 -07:00
|
|
|
.get_matches_from(vec!["test", "airdrop", "50"]);
|
2018-09-14 01:59:09 -07:00
|
|
|
assert_eq!(
|
2019-03-09 19:28:43 -08:00
|
|
|
parse_command(&pubkey, &test_airdrop).unwrap(),
|
2019-01-13 23:10:03 -08:00
|
|
|
WalletCommand::Airdrop(50)
|
2018-09-14 01:59:09 -07:00
|
|
|
);
|
|
|
|
let test_bad_airdrop = test_commands
|
|
|
|
.clone()
|
2018-09-19 12:44:16 -07:00
|
|
|
.get_matches_from(vec!["test", "airdrop", "notint"]);
|
2019-03-09 19:28:43 -08:00
|
|
|
assert!(parse_command(&pubkey, &test_bad_airdrop).is_err());
|
2018-09-14 01:59:09 -07:00
|
|
|
|
2018-09-19 12:44:16 -07:00
|
|
|
// Test Cancel Subcommand
|
|
|
|
let test_cancel =
|
|
|
|
test_commands
|
|
|
|
.clone()
|
|
|
|
.get_matches_from(vec!["test", "cancel", &pubkey_string]);
|
2018-09-14 01:59:09 -07:00
|
|
|
assert_eq!(
|
2019-03-09 19:28:43 -08:00
|
|
|
parse_command(&pubkey, &test_cancel).unwrap(),
|
2018-09-19 12:44:16 -07:00
|
|
|
WalletCommand::Cancel(pubkey)
|
2018-09-14 01:59:09 -07:00
|
|
|
);
|
|
|
|
|
2018-09-19 12:44:16 -07:00
|
|
|
// Test Confirm Subcommand
|
2018-09-14 01:59:09 -07:00
|
|
|
let signature = Signature::new(&vec![1; 64]);
|
|
|
|
let signature_string = format!("{:?}", signature);
|
|
|
|
let test_confirm =
|
|
|
|
test_commands
|
|
|
|
.clone()
|
|
|
|
.get_matches_from(vec!["test", "confirm", &signature_string]);
|
|
|
|
assert_eq!(
|
2019-03-09 19:28:43 -08:00
|
|
|
parse_command(&pubkey, &test_confirm).unwrap(),
|
2018-09-14 01:59:09 -07:00
|
|
|
WalletCommand::Confirm(signature)
|
|
|
|
);
|
2018-09-19 12:44:16 -07:00
|
|
|
let test_bad_signature = test_commands
|
|
|
|
.clone()
|
|
|
|
.get_matches_from(vec!["test", "confirm", "deadbeef"]);
|
2019-03-09 19:28:43 -08:00
|
|
|
assert!(parse_command(&pubkey, &test_bad_signature).is_err());
|
2018-09-19 12:44:16 -07:00
|
|
|
|
2019-03-06 20:29:08 -08:00
|
|
|
// Test ConfigureStakingAccount Subcommand
|
|
|
|
let second_pubkey = Keypair::new().pubkey();
|
|
|
|
let second_pubkey_string = format!("{}", second_pubkey);
|
|
|
|
let test_configure_staking_account = test_commands.clone().get_matches_from(vec![
|
|
|
|
"test",
|
|
|
|
"configure-staking-account",
|
|
|
|
"--delegate-account",
|
|
|
|
&pubkey_string,
|
|
|
|
"--authorize-voter",
|
|
|
|
&second_pubkey_string,
|
|
|
|
]);
|
|
|
|
assert_eq!(
|
2019-03-09 19:28:43 -08:00
|
|
|
parse_command(&pubkey, &test_configure_staking_account).unwrap(),
|
2019-03-06 20:29:08 -08:00
|
|
|
WalletCommand::ConfigureStakingAccount(Some(pubkey), Some(second_pubkey))
|
|
|
|
);
|
|
|
|
let test_configure_staking_account = test_commands.clone().get_matches_from(vec![
|
|
|
|
"test",
|
|
|
|
"configure-staking-account",
|
|
|
|
"--delegate-account",
|
|
|
|
&pubkey_string,
|
|
|
|
]);
|
|
|
|
assert_eq!(
|
2019-03-09 19:28:43 -08:00
|
|
|
parse_command(&pubkey, &test_configure_staking_account).unwrap(),
|
2019-03-06 20:29:08 -08:00
|
|
|
WalletCommand::ConfigureStakingAccount(Some(pubkey), None)
|
|
|
|
);
|
|
|
|
|
|
|
|
// Test CreateStakingAccount SubCommand
|
|
|
|
let test_create_staking_account = test_commands.clone().get_matches_from(vec![
|
|
|
|
"test",
|
|
|
|
"create-staking-account",
|
|
|
|
&pubkey_string,
|
|
|
|
"50",
|
|
|
|
]);
|
|
|
|
assert_eq!(
|
2019-03-09 19:28:43 -08:00
|
|
|
parse_command(&pubkey, &test_create_staking_account).unwrap(),
|
2019-03-06 20:29:08 -08:00
|
|
|
WalletCommand::CreateStakingAccount(pubkey, 50)
|
|
|
|
);
|
|
|
|
|
2018-11-12 12:48:59 -08:00
|
|
|
// Test Deploy Subcommand
|
|
|
|
let test_deploy =
|
|
|
|
test_commands
|
|
|
|
.clone()
|
|
|
|
.get_matches_from(vec!["test", "deploy", "/Users/test/program.o"]);
|
|
|
|
assert_eq!(
|
2019-03-09 19:28:43 -08:00
|
|
|
parse_command(&pubkey, &test_deploy).unwrap(),
|
2018-11-12 12:48:59 -08:00
|
|
|
WalletCommand::Deploy("/Users/test/program.o".to_string())
|
|
|
|
);
|
|
|
|
|
2018-09-19 12:44:16 -07:00
|
|
|
// Test Simple Pay Subcommand
|
|
|
|
let test_pay =
|
|
|
|
test_commands
|
|
|
|
.clone()
|
|
|
|
.get_matches_from(vec!["test", "pay", &pubkey_string, "50"]);
|
|
|
|
assert_eq!(
|
2019-03-09 19:28:43 -08:00
|
|
|
parse_command(&pubkey, &test_pay).unwrap(),
|
2018-09-19 12:44:16 -07:00
|
|
|
WalletCommand::Pay(50, pubkey, None, None, None, None)
|
|
|
|
);
|
|
|
|
|
|
|
|
// Test Pay Subcommand w/ Witness
|
|
|
|
let test_pay_multiple_witnesses = test_commands.clone().get_matches_from(vec![
|
|
|
|
"test",
|
|
|
|
"pay",
|
|
|
|
&pubkey_string,
|
|
|
|
"50",
|
|
|
|
"--require-signature-from",
|
|
|
|
&witness0_string,
|
|
|
|
"--require-signature-from",
|
|
|
|
&witness1_string,
|
|
|
|
]);
|
|
|
|
assert_eq!(
|
2019-03-09 19:28:43 -08:00
|
|
|
parse_command(&pubkey, &test_pay_multiple_witnesses).unwrap(),
|
2018-09-19 12:44:16 -07:00
|
|
|
WalletCommand::Pay(50, pubkey, None, None, Some(vec![witness0, witness1]), None)
|
|
|
|
);
|
|
|
|
let test_pay_single_witness = test_commands.clone().get_matches_from(vec![
|
|
|
|
"test",
|
|
|
|
"pay",
|
|
|
|
&pubkey_string,
|
|
|
|
"50",
|
|
|
|
"--require-signature-from",
|
|
|
|
&witness0_string,
|
|
|
|
]);
|
|
|
|
assert_eq!(
|
2019-03-09 19:28:43 -08:00
|
|
|
parse_command(&pubkey, &test_pay_single_witness).unwrap(),
|
2018-09-19 12:44:16 -07:00
|
|
|
WalletCommand::Pay(50, pubkey, None, None, Some(vec![witness0]), None)
|
|
|
|
);
|
|
|
|
|
|
|
|
// Test Pay Subcommand w/ Timestamp
|
|
|
|
let test_pay_timestamp = test_commands.clone().get_matches_from(vec![
|
|
|
|
"test",
|
|
|
|
"pay",
|
|
|
|
&pubkey_string,
|
|
|
|
"50",
|
|
|
|
"--after",
|
|
|
|
"2018-09-19T17:30:59",
|
|
|
|
"--require-timestamp-from",
|
|
|
|
&witness0_string,
|
|
|
|
]);
|
|
|
|
assert_eq!(
|
2019-03-09 19:28:43 -08:00
|
|
|
parse_command(&pubkey, &test_pay_timestamp).unwrap(),
|
2018-09-19 12:44:16 -07:00
|
|
|
WalletCommand::Pay(50, pubkey, Some(dt), Some(witness0), None, None)
|
|
|
|
);
|
|
|
|
|
|
|
|
// Test Send-Signature Subcommand
|
2018-09-24 09:23:16 -07:00
|
|
|
let test_send_signature = test_commands.clone().get_matches_from(vec![
|
|
|
|
"test",
|
|
|
|
"send-signature",
|
|
|
|
&pubkey_string,
|
|
|
|
&pubkey_string,
|
|
|
|
]);
|
2018-09-19 12:44:16 -07:00
|
|
|
assert_eq!(
|
2019-03-09 19:28:43 -08:00
|
|
|
parse_command(&pubkey, &test_send_signature).unwrap(),
|
2018-09-24 09:23:16 -07:00
|
|
|
WalletCommand::Witness(pubkey, pubkey)
|
2018-09-19 12:44:16 -07:00
|
|
|
);
|
|
|
|
let test_pay_multiple_witnesses = test_commands.clone().get_matches_from(vec![
|
|
|
|
"test",
|
|
|
|
"pay",
|
|
|
|
&pubkey_string,
|
|
|
|
"50",
|
|
|
|
"--after",
|
|
|
|
"2018-09-19T17:30:59",
|
|
|
|
"--require-signature-from",
|
|
|
|
&witness0_string,
|
|
|
|
"--require-timestamp-from",
|
|
|
|
&witness0_string,
|
|
|
|
"--require-signature-from",
|
|
|
|
&witness1_string,
|
|
|
|
]);
|
|
|
|
assert_eq!(
|
2019-03-09 19:28:43 -08:00
|
|
|
parse_command(&pubkey, &test_pay_multiple_witnesses).unwrap(),
|
2018-09-19 12:44:16 -07:00
|
|
|
WalletCommand::Pay(
|
|
|
|
50,
|
|
|
|
pubkey,
|
|
|
|
Some(dt),
|
|
|
|
Some(witness0),
|
|
|
|
Some(vec![witness0, witness1]),
|
|
|
|
None
|
|
|
|
)
|
|
|
|
);
|
|
|
|
|
|
|
|
// Test Send-Timestamp Subcommand
|
|
|
|
let test_send_timestamp = test_commands.clone().get_matches_from(vec![
|
|
|
|
"test",
|
|
|
|
"send-timestamp",
|
|
|
|
&pubkey_string,
|
2018-09-24 09:23:16 -07:00
|
|
|
&pubkey_string,
|
2018-09-19 12:44:16 -07:00
|
|
|
"--date",
|
|
|
|
"2018-09-19T17:30:59",
|
|
|
|
]);
|
|
|
|
assert_eq!(
|
2019-03-09 19:28:43 -08:00
|
|
|
parse_command(&pubkey, &test_send_timestamp).unwrap(),
|
2018-09-24 09:23:16 -07:00
|
|
|
WalletCommand::TimeElapsed(pubkey, pubkey, dt)
|
2018-09-19 12:44:16 -07:00
|
|
|
);
|
|
|
|
let test_bad_timestamp = test_commands.clone().get_matches_from(vec![
|
|
|
|
"test",
|
|
|
|
"send-timestamp",
|
|
|
|
&pubkey_string,
|
2018-09-24 09:23:16 -07:00
|
|
|
&pubkey_string,
|
2018-09-19 12:44:16 -07:00
|
|
|
"--date",
|
|
|
|
"20180919T17:30:59",
|
|
|
|
]);
|
2019-03-09 19:28:43 -08:00
|
|
|
assert!(parse_command(&pubkey, &test_bad_timestamp).is_err());
|
2018-09-14 01:59:09 -07:00
|
|
|
}
|
2019-01-13 23:10:03 -08:00
|
|
|
|
2018-09-14 01:59:09 -07:00
|
|
|
#[test]
|
2018-09-25 11:45:25 -07:00
|
|
|
fn test_wallet_process_command() {
|
2019-01-13 23:10:03 -08:00
|
|
|
// Success cases
|
2018-11-12 12:48:59 -08:00
|
|
|
let mut config = WalletConfig::default();
|
2019-03-16 21:51:41 -07:00
|
|
|
config.rpc_client = Some(RpcClient::new_mock("succeeds".to_string()));
|
2018-11-12 12:48:59 -08:00
|
|
|
|
2019-01-13 23:10:03 -08:00
|
|
|
let keypair = Keypair::new();
|
|
|
|
let pubkey = keypair.pubkey().to_string();
|
|
|
|
config.id = keypair;
|
|
|
|
config.command = WalletCommand::Address;
|
|
|
|
assert_eq!(process_command(&config).unwrap(), pubkey);
|
2018-09-14 01:59:09 -07:00
|
|
|
|
2019-03-20 09:44:16 -07:00
|
|
|
config.command = WalletCommand::Balance(config.id.pubkey());
|
|
|
|
assert_eq!(process_command(&config).unwrap(), "50 lamports");
|
2018-09-14 01:59:09 -07:00
|
|
|
|
2019-01-13 23:10:03 -08:00
|
|
|
let process_id = Keypair::new().pubkey();
|
|
|
|
config.command = WalletCommand::Cancel(process_id);
|
|
|
|
assert_eq!(process_command(&config).unwrap(), SIGNATURE);
|
2018-09-17 12:15:26 -07:00
|
|
|
|
2019-01-13 23:10:03 -08:00
|
|
|
let good_signature = Signature::new(&bs58::decode(SIGNATURE).into_vec().unwrap());
|
|
|
|
config.command = WalletCommand::Confirm(good_signature);
|
2018-09-20 22:27:06 -07:00
|
|
|
assert_eq!(process_command(&config).unwrap(), "Confirmed");
|
2018-09-14 01:59:09 -07:00
|
|
|
|
2019-03-06 20:29:08 -08:00
|
|
|
let bob_pubkey = Keypair::new().pubkey();
|
|
|
|
config.command = WalletCommand::ConfigureStakingAccount(None, Some(bob_pubkey));
|
|
|
|
let signature = process_command(&config);
|
|
|
|
assert_eq!(signature.unwrap(), SIGNATURE.to_string());
|
|
|
|
|
|
|
|
config.command = WalletCommand::CreateStakingAccount(bob_pubkey, 10);
|
|
|
|
let signature = process_command(&config);
|
|
|
|
assert_eq!(signature.unwrap(), SIGNATURE.to_string());
|
|
|
|
|
2019-01-13 23:10:03 -08:00
|
|
|
config.command = WalletCommand::GetTransactionCount;
|
|
|
|
assert_eq!(process_command(&config).unwrap(), "1234");
|
|
|
|
|
|
|
|
config.command = WalletCommand::Pay(10, bob_pubkey, None, None, None, None);
|
|
|
|
let signature = process_command(&config);
|
|
|
|
assert_eq!(signature.unwrap(), SIGNATURE.to_string());
|
|
|
|
|
|
|
|
let date_string = "\"2018-09-19T17:30:59Z\"";
|
|
|
|
let dt: DateTime<Utc> = serde_json::from_str(&date_string).unwrap();
|
|
|
|
config.command = WalletCommand::Pay(
|
|
|
|
10,
|
|
|
|
bob_pubkey,
|
|
|
|
Some(dt),
|
|
|
|
Some(config.id.pubkey()),
|
|
|
|
None,
|
|
|
|
None,
|
|
|
|
);
|
|
|
|
let result = process_command(&config);
|
|
|
|
let json: Value = serde_json::from_str(&result.unwrap()).unwrap();
|
2018-09-17 12:15:26 -07:00
|
|
|
assert_eq!(
|
2019-01-13 23:10:03 -08:00
|
|
|
json.as_object()
|
|
|
|
.unwrap()
|
|
|
|
.get("signature")
|
|
|
|
.unwrap()
|
|
|
|
.as_str()
|
|
|
|
.unwrap(),
|
|
|
|
SIGNATURE.to_string()
|
2018-09-17 12:15:26 -07:00
|
|
|
);
|
2018-09-25 11:45:25 -07:00
|
|
|
|
2019-01-13 23:10:03 -08:00
|
|
|
let witness = Keypair::new().pubkey();
|
|
|
|
config.command = WalletCommand::Pay(
|
|
|
|
10,
|
|
|
|
bob_pubkey,
|
2019-01-03 21:29:21 -08:00
|
|
|
None,
|
2018-11-05 09:50:58 -08:00
|
|
|
None,
|
2019-01-13 23:10:03 -08:00
|
|
|
Some(vec![witness]),
|
|
|
|
Some(config.id.pubkey()),
|
|
|
|
);
|
|
|
|
let result = process_command(&config);
|
|
|
|
let json: Value = serde_json::from_str(&result.unwrap()).unwrap();
|
|
|
|
assert_eq!(
|
|
|
|
json.as_object()
|
|
|
|
.unwrap()
|
|
|
|
.get("signature")
|
|
|
|
.unwrap()
|
|
|
|
.as_str()
|
|
|
|
.unwrap(),
|
|
|
|
SIGNATURE.to_string()
|
2018-09-14 01:59:09 -07:00
|
|
|
);
|
|
|
|
|
2019-01-13 23:10:03 -08:00
|
|
|
let process_id = Keypair::new().pubkey();
|
|
|
|
config.command = WalletCommand::TimeElapsed(bob_pubkey, process_id, dt);
|
|
|
|
let signature = process_command(&config);
|
|
|
|
assert_eq!(signature.unwrap(), SIGNATURE.to_string());
|
2018-09-14 01:59:09 -07:00
|
|
|
|
2019-01-13 23:10:03 -08:00
|
|
|
let witness = Keypair::new().pubkey();
|
|
|
|
config.command = WalletCommand::Witness(bob_pubkey, witness);
|
|
|
|
let signature = process_command(&config);
|
|
|
|
assert_eq!(signature.unwrap(), SIGNATURE.to_string());
|
2018-09-20 22:27:06 -07:00
|
|
|
|
2019-01-13 23:10:03 -08:00
|
|
|
// Need airdrop cases
|
|
|
|
config.command = WalletCommand::Airdrop(50);
|
|
|
|
assert!(process_command(&config).is_err());
|
2018-11-14 18:57:34 -08:00
|
|
|
|
2019-03-16 21:51:41 -07:00
|
|
|
config.rpc_client = Some(RpcClient::new_mock("airdrop".to_string()));
|
2019-01-13 23:10:03 -08:00
|
|
|
config.command = WalletCommand::TimeElapsed(bob_pubkey, process_id, dt);
|
|
|
|
let signature = process_command(&config);
|
|
|
|
assert_eq!(signature.unwrap(), SIGNATURE.to_string());
|
2018-11-27 21:08:14 -08:00
|
|
|
|
2019-01-13 23:10:03 -08:00
|
|
|
let witness = Keypair::new().pubkey();
|
|
|
|
config.command = WalletCommand::Witness(bob_pubkey, witness);
|
|
|
|
let signature = process_command(&config);
|
|
|
|
assert_eq!(signature.unwrap(), SIGNATURE.to_string());
|
|
|
|
|
2019-03-17 00:40:45 -07:00
|
|
|
// bad_sig_status cases
|
|
|
|
config.rpc_client = Some(RpcClient::new_mock("bad_sig_status".to_string()));
|
|
|
|
let missing_signature = Signature::new(&bs58::decode("5VERv8NMvzbJMEkV8xnrLkEaWRtSz9CosKDYjCJjBRnbJLgp8uirBgmQpjKhoR4tjF3ZpRzrFmBV6UjKdiSZkQUW").into_vec().unwrap());
|
|
|
|
config.command = WalletCommand::Confirm(missing_signature);
|
|
|
|
assert_eq!(process_command(&config).unwrap(), "Not found");
|
|
|
|
|
|
|
|
// Failure cases
|
2019-03-16 21:51:41 -07:00
|
|
|
config.rpc_client = Some(RpcClient::new_mock("fails".to_string()));
|
2019-01-13 23:10:03 -08:00
|
|
|
|
|
|
|
config.command = WalletCommand::Airdrop(50);
|
|
|
|
assert!(process_command(&config).is_err());
|
2018-09-25 11:45:25 -07:00
|
|
|
|
2019-03-20 09:44:16 -07:00
|
|
|
config.command = WalletCommand::Balance(config.id.pubkey());
|
2019-01-13 23:10:03 -08:00
|
|
|
assert!(process_command(&config).is_err());
|
|
|
|
|
|
|
|
let any_signature = Signature::new(&bs58::decode(SIGNATURE).into_vec().unwrap());
|
|
|
|
config.command = WalletCommand::Confirm(any_signature);
|
|
|
|
assert!(process_command(&config).is_err());
|
|
|
|
|
2019-03-06 20:29:08 -08:00
|
|
|
config.command = WalletCommand::ConfigureStakingAccount(None, Some(bob_pubkey));
|
|
|
|
assert!(process_command(&config).is_err());
|
|
|
|
|
|
|
|
config.command = WalletCommand::CreateStakingAccount(bob_pubkey, 10);
|
|
|
|
assert!(process_command(&config).is_err());
|
|
|
|
|
2019-01-13 23:10:03 -08:00
|
|
|
config.command = WalletCommand::GetTransactionCount;
|
|
|
|
assert!(process_command(&config).is_err());
|
|
|
|
|
|
|
|
config.command = WalletCommand::Pay(10, bob_pubkey, None, None, None, None);
|
|
|
|
assert!(process_command(&config).is_err());
|
|
|
|
|
|
|
|
config.command = WalletCommand::Pay(
|
|
|
|
10,
|
|
|
|
bob_pubkey,
|
|
|
|
Some(dt),
|
|
|
|
Some(config.id.pubkey()),
|
|
|
|
None,
|
|
|
|
None,
|
|
|
|
);
|
|
|
|
assert!(process_command(&config).is_err());
|
|
|
|
|
|
|
|
config.command = WalletCommand::Pay(
|
|
|
|
10,
|
|
|
|
bob_pubkey,
|
|
|
|
None,
|
|
|
|
None,
|
|
|
|
Some(vec![witness]),
|
|
|
|
Some(config.id.pubkey()),
|
|
|
|
);
|
|
|
|
assert!(process_command(&config).is_err());
|
|
|
|
|
|
|
|
config.command = WalletCommand::TimeElapsed(bob_pubkey, process_id, dt);
|
|
|
|
assert!(process_command(&config).is_err());
|
2018-09-14 01:59:09 -07:00
|
|
|
}
|
2019-01-17 08:49:16 -08:00
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_wallet_deploy() {
|
2019-02-27 11:17:32 -08:00
|
|
|
solana_logger::setup();
|
2019-01-17 08:49:16 -08:00
|
|
|
let mut pathbuf = PathBuf::from(env!("CARGO_MANIFEST_DIR"));
|
|
|
|
pathbuf.push("tests");
|
|
|
|
pathbuf.push("fixtures");
|
|
|
|
pathbuf.push("noop");
|
|
|
|
pathbuf.set_extension("so");
|
|
|
|
|
|
|
|
// Success case
|
|
|
|
let mut config = WalletConfig::default();
|
2019-03-16 21:51:41 -07:00
|
|
|
config.rpc_client = Some(RpcClient::new_mock("succeeds".to_string()));
|
2019-01-17 08:49:16 -08:00
|
|
|
|
|
|
|
config.command = WalletCommand::Deploy(pathbuf.to_str().unwrap().to_string());
|
|
|
|
let result = process_command(&config);
|
|
|
|
let json: Value = serde_json::from_str(&result.unwrap()).unwrap();
|
|
|
|
let program_id = json
|
|
|
|
.as_object()
|
|
|
|
.unwrap()
|
|
|
|
.get("programId")
|
|
|
|
.unwrap()
|
|
|
|
.as_str()
|
|
|
|
.unwrap();
|
|
|
|
let program_id_vec = bs58::decode(program_id).into_vec().unwrap();
|
|
|
|
assert_eq!(program_id_vec.len(), mem::size_of::<Pubkey>());
|
|
|
|
|
|
|
|
// Failure cases
|
2019-03-16 21:51:41 -07:00
|
|
|
config.rpc_client = Some(RpcClient::new_mock("airdrop".to_string()));
|
2019-01-17 08:49:16 -08:00
|
|
|
assert!(process_command(&config).is_err());
|
|
|
|
|
|
|
|
config.command = WalletCommand::Deploy("bad/file/location.so".to_string());
|
|
|
|
assert!(process_command(&config).is_err());
|
|
|
|
}
|
2018-10-17 13:42:54 -07:00
|
|
|
|
2018-10-08 16:15:17 -07:00
|
|
|
fn tmp_file_path(name: &str) -> String {
|
|
|
|
use std::env;
|
|
|
|
let out_dir = env::var("OUT_DIR").unwrap_or_else(|_| "target".to_string());
|
|
|
|
let keypair = Keypair::new();
|
|
|
|
|
|
|
|
format!("{}/tmp/{}-{}", out_dir, name, keypair.pubkey()).to_string()
|
|
|
|
}
|
|
|
|
|
2018-09-17 09:48:04 -07:00
|
|
|
#[test]
|
2018-09-25 11:45:25 -07:00
|
|
|
fn test_wallet_gen_keypair_file() {
|
2018-10-08 16:15:17 -07:00
|
|
|
let outfile = tmp_file_path("test_gen_keypair_file.json");
|
2018-09-17 09:48:04 -07:00
|
|
|
let serialized_keypair = gen_keypair_file(outfile.to_string()).unwrap();
|
|
|
|
let keypair_vec: Vec<u8> = serde_json::from_str(&serialized_keypair).unwrap();
|
2018-10-08 16:15:17 -07:00
|
|
|
assert!(Path::new(&outfile).exists());
|
2018-11-12 22:34:43 -08:00
|
|
|
assert_eq!(keypair_vec, read_pkcs8(&outfile).unwrap());
|
2019-01-28 14:52:35 -08:00
|
|
|
read_keypair(&outfile).unwrap();
|
2018-09-17 09:48:04 -07:00
|
|
|
assert_eq!(
|
|
|
|
read_keypair(&outfile).unwrap().pubkey().as_ref().len(),
|
|
|
|
mem::size_of::<Pubkey>()
|
|
|
|
);
|
2018-10-08 16:15:17 -07:00
|
|
|
fs::remove_file(&outfile).unwrap();
|
|
|
|
assert!(!Path::new(&outfile).exists());
|
2018-09-17 09:48:04 -07:00
|
|
|
}
|
2018-09-14 01:59:09 -07:00
|
|
|
}
|