2018-06-27 14:41:49 -07:00
|
|
|
extern crate atty;
|
2018-06-26 22:52:34 -07:00
|
|
|
extern crate bincode;
|
2018-06-29 16:06:05 -07:00
|
|
|
extern crate bs58;
|
2018-06-26 22:52:34 -07:00
|
|
|
extern crate env_logger;
|
|
|
|
extern crate getopts;
|
|
|
|
extern crate serde_json;
|
|
|
|
extern crate solana;
|
|
|
|
|
|
|
|
use bincode::serialize;
|
2018-06-29 16:06:05 -07:00
|
|
|
use getopts::{Matches, Options};
|
2018-06-26 22:52:34 -07:00
|
|
|
use solana::crdt::{get_ip_addr, ReplicatedData};
|
|
|
|
use solana::drone::DroneRequest;
|
2018-06-27 14:41:49 -07:00
|
|
|
use solana::mint::Mint;
|
2018-06-29 16:06:05 -07:00
|
|
|
use solana::signature::{PublicKey, Signature};
|
2018-06-26 22:52:34 -07:00
|
|
|
use solana::thin_client::ThinClient;
|
|
|
|
use std::env;
|
2018-06-29 10:38:00 -07:00
|
|
|
use std::error;
|
|
|
|
use std::fmt;
|
2018-06-26 22:52:34 -07:00
|
|
|
use std::fs::File;
|
2018-06-27 14:41:49 -07:00
|
|
|
use std::io;
|
2018-06-26 22:52:34 -07:00
|
|
|
use std::io::prelude::*;
|
|
|
|
use std::net::{IpAddr, Ipv4Addr, SocketAddr, TcpStream, UdpSocket};
|
|
|
|
use std::process::exit;
|
|
|
|
use std::thread::sleep;
|
2018-06-27 14:21:30 -07:00
|
|
|
use std::time::Duration;
|
2018-06-26 22:52:34 -07:00
|
|
|
|
2018-06-29 10:38:00 -07:00
|
|
|
enum WalletCommand {
|
|
|
|
Balance,
|
|
|
|
AirDrop,
|
2018-06-29 16:06:05 -07:00
|
|
|
Address,
|
|
|
|
Pay(i64, PublicKey),
|
|
|
|
Confirm(Signature),
|
2018-06-29 10:38:00 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Debug, Clone)]
|
|
|
|
enum WalletError {
|
|
|
|
CommandNotRecognized(String),
|
|
|
|
}
|
|
|
|
|
|
|
|
impl fmt::Display for WalletError {
|
|
|
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
|
|
|
write!(f, "invalid")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl error::Error for WalletError {
|
|
|
|
fn description(&self) -> &str {
|
|
|
|
"invalid"
|
|
|
|
}
|
|
|
|
|
|
|
|
fn cause(&self) -> Option<&error::Error> {
|
|
|
|
// Generic error, underlying cause isn't tracked.
|
|
|
|
None
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
struct WalletConfig {
|
|
|
|
leader: ReplicatedData,
|
|
|
|
id: Mint,
|
|
|
|
client_addr: SocketAddr,
|
|
|
|
drone_addr: SocketAddr,
|
|
|
|
command: WalletCommand,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl Default for WalletConfig {
|
|
|
|
fn default() -> WalletConfig {
|
|
|
|
let default_addr = SocketAddr::new(IpAddr::V4(Ipv4Addr::new(0, 0, 0, 0)), 8000);
|
|
|
|
WalletConfig {
|
|
|
|
leader: ReplicatedData::new_leader(&default_addr.clone()),
|
|
|
|
id: Mint::new(0),
|
|
|
|
client_addr: default_addr.clone(),
|
|
|
|
drone_addr: default_addr.clone(),
|
|
|
|
command: WalletCommand::Balance,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-06-26 22:52:34 -07:00
|
|
|
fn print_usage(program: &str, opts: Options) {
|
|
|
|
let mut brief = format!("Usage: {} [options]\n\n", program);
|
2018-06-29 15:04:35 -07:00
|
|
|
brief += " solana-wallet allows you to perform basic actions, including";
|
2018-06-26 22:52:34 -07:00
|
|
|
brief += " requesting an airdrop, checking your balance, and spending tokens.";
|
|
|
|
brief += " Takes json formatted mint file to stdin.";
|
|
|
|
|
|
|
|
print!("{}", opts.usage(&brief));
|
2018-06-29 16:06:05 -07:00
|
|
|
display_actions();
|
2018-06-26 22:52:34 -07:00
|
|
|
}
|
|
|
|
|
2018-06-29 16:06:05 -07:00
|
|
|
fn parse_command(matches: &Matches) -> Result<WalletCommand, WalletError> {
|
|
|
|
let input = &matches.free[0];
|
|
|
|
match input.as_ref() {
|
|
|
|
"address" => Ok(WalletCommand::Address),
|
2018-06-29 10:38:00 -07:00
|
|
|
"balance" => Ok(WalletCommand::Balance),
|
|
|
|
"airdrop" => Ok(WalletCommand::AirDrop),
|
2018-06-29 16:06:05 -07:00
|
|
|
"pay" => {
|
|
|
|
if matches.free.len() < 3 {
|
|
|
|
eprintln!("No tokens and public key provided");
|
|
|
|
exit(1);
|
|
|
|
}
|
|
|
|
let tokens = matches.free[1].parse().expect("parse integer");
|
|
|
|
let pubkey_vec = bs58::decode(&matches.free[2])
|
|
|
|
.into_vec()
|
|
|
|
.expect("base58-encoded public key");
|
|
|
|
let to = PublicKey::clone_from_slice(&pubkey_vec);
|
|
|
|
Ok(WalletCommand::Pay(tokens, to))
|
|
|
|
}
|
|
|
|
"confirm" => {
|
|
|
|
if matches.free.len() < 2 {
|
|
|
|
eprintln!("No signature provided");
|
|
|
|
exit(1);
|
|
|
|
}
|
|
|
|
let sig_vec = bs58::decode(&matches.free[1])
|
|
|
|
.into_vec()
|
|
|
|
.expect("base58-encoded signature");
|
|
|
|
let sig = Signature::clone_from_slice(&sig_vec);
|
|
|
|
Ok(WalletCommand::Confirm(sig))
|
|
|
|
}
|
2018-06-29 10:38:00 -07:00
|
|
|
_ => Err(WalletError::CommandNotRecognized(input.to_string())),
|
|
|
|
}
|
|
|
|
}
|
2018-06-27 14:41:49 -07:00
|
|
|
|
2018-06-29 10:38:00 -07:00
|
|
|
fn parse_args(args: Vec<String>) -> Result<WalletConfig, Box<error::Error>> {
|
2018-06-26 22:52:34 -07:00
|
|
|
let mut opts = Options::new();
|
|
|
|
opts.optopt("l", "", "leader", "leader.json");
|
2018-06-28 11:58:33 -07:00
|
|
|
opts.optopt("m", "", "mint", "mint.json");
|
2018-06-26 22:52:34 -07:00
|
|
|
opts.optopt("c", "", "client port", "port");
|
|
|
|
opts.optflag("d", "dyn", "detect network address dynamically");
|
|
|
|
opts.optflag("h", "help", "print help");
|
2018-06-29 10:38:00 -07:00
|
|
|
|
2018-06-26 22:52:34 -07:00
|
|
|
let matches = match opts.parse(&args[1..]) {
|
|
|
|
Ok(m) => m,
|
|
|
|
Err(e) => {
|
|
|
|
eprintln!("{}", e);
|
|
|
|
exit(1);
|
|
|
|
}
|
|
|
|
};
|
2018-06-29 10:38:00 -07:00
|
|
|
|
2018-06-29 15:04:35 -07:00
|
|
|
if matches.opt_present("h") || matches.free.len() < 1 {
|
2018-06-29 16:06:05 -07:00
|
|
|
print_usage(&args[0], opts);
|
2018-06-29 10:38:00 -07:00
|
|
|
return Ok(WalletConfig::default());
|
2018-06-26 22:52:34 -07:00
|
|
|
}
|
2018-06-29 10:38:00 -07:00
|
|
|
|
2018-06-26 22:52:34 -07:00
|
|
|
let mut client_addr: SocketAddr = "0.0.0.0:8100".parse().unwrap();
|
|
|
|
if matches.opt_present("c") {
|
|
|
|
let port = matches.opt_str("c").unwrap().parse().unwrap();
|
|
|
|
client_addr.set_port(port);
|
|
|
|
}
|
2018-06-29 10:38:00 -07:00
|
|
|
|
2018-06-26 22:52:34 -07:00
|
|
|
if matches.opt_present("d") {
|
|
|
|
client_addr.set_ip(get_ip_addr().unwrap());
|
|
|
|
}
|
2018-06-29 10:38:00 -07:00
|
|
|
|
2018-06-28 12:09:10 -07:00
|
|
|
let leader = if matches.opt_present("l") {
|
2018-06-26 22:52:34 -07:00
|
|
|
read_leader(matches.opt_str("l").unwrap())
|
|
|
|
} else {
|
|
|
|
let server_addr = SocketAddr::new(IpAddr::V4(Ipv4Addr::new(0, 0, 0, 0)), 8000);
|
|
|
|
ReplicatedData::new_leader(&server_addr)
|
|
|
|
};
|
2018-06-29 10:38:00 -07:00
|
|
|
|
2018-06-28 12:09:10 -07:00
|
|
|
let id = if matches.opt_present("m") {
|
2018-06-29 10:38:00 -07:00
|
|
|
read_mint(matches.opt_str("m").unwrap())?
|
2018-06-28 11:58:33 -07:00
|
|
|
} else {
|
2018-06-29 09:30:09 -07:00
|
|
|
eprintln!("No mint found!");
|
2018-06-28 12:09:10 -07:00
|
|
|
exit(1);
|
2018-06-28 11:58:33 -07:00
|
|
|
};
|
2018-06-26 22:52:34 -07:00
|
|
|
|
|
|
|
let mut drone_addr = leader.transactions_addr.clone();
|
|
|
|
drone_addr.set_port(9900);
|
|
|
|
|
2018-06-29 16:06:05 -07:00
|
|
|
let command = parse_command(&matches)?;
|
2018-06-29 10:38:00 -07:00
|
|
|
Ok(WalletConfig {
|
|
|
|
leader,
|
|
|
|
id,
|
|
|
|
client_addr,
|
|
|
|
drone_addr, // TODO: Add an option for this.
|
|
|
|
command,
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
|
|
|
fn process_command(
|
|
|
|
config: &WalletConfig,
|
|
|
|
client: &mut ThinClient,
|
|
|
|
) -> Result<(), Box<error::Error>> {
|
|
|
|
match config.command {
|
|
|
|
// Check client balance
|
2018-06-29 16:06:05 -07:00
|
|
|
WalletCommand::Address => {
|
|
|
|
println!("{}", bs58::encode(config.id.pubkey()).into_string());
|
|
|
|
}
|
2018-06-29 10:38:00 -07:00
|
|
|
WalletCommand::Balance => {
|
|
|
|
println!("Balance requested...");
|
|
|
|
let balance = client.poll_get_balance(&config.id.pubkey());
|
|
|
|
match balance {
|
|
|
|
Ok(balance) => {
|
|
|
|
println!("Your balance is: {:?}", balance);
|
|
|
|
}
|
|
|
|
Err(ref e) if e.kind() == std::io::ErrorKind::Other => {
|
|
|
|
println!("No account found! Request an airdrop to get started.");
|
|
|
|
}
|
|
|
|
Err(error) => {
|
|
|
|
println!("An error occurred: {:?}", error);
|
2018-06-26 22:52:34 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2018-06-29 10:38:00 -07:00
|
|
|
// Request an airdrop from Solana Drone;
|
|
|
|
// Request amount is set in request_airdrop function
|
|
|
|
WalletCommand::AirDrop => {
|
|
|
|
println!("Airdrop requested...");
|
|
|
|
let _airdrop = request_airdrop(&config.drone_addr, &config.id);
|
|
|
|
// TODO: return airdrop Result from Drone
|
|
|
|
sleep(Duration::from_millis(100));
|
|
|
|
println!(
|
|
|
|
"Your balance is: {:?}",
|
|
|
|
client.poll_get_balance(&config.id.pubkey()).unwrap()
|
|
|
|
);
|
|
|
|
}
|
|
|
|
// If client has positive balance, spend tokens in {balance} number of transactions
|
2018-06-29 16:06:05 -07:00
|
|
|
WalletCommand::Pay(tokens, to) => {
|
2018-06-29 10:38:00 -07:00
|
|
|
let last_id = client.get_last_id();
|
2018-06-29 16:06:05 -07:00
|
|
|
let sig = client.transfer(tokens, &config.id.keypair(), to, &last_id)?;
|
|
|
|
println!("{}", bs58::encode(sig).into_string());
|
2018-06-29 10:38:00 -07:00
|
|
|
}
|
|
|
|
// Confirm the last client transaction by signature
|
2018-06-29 16:06:05 -07:00
|
|
|
WalletCommand::Confirm(sig) => {
|
|
|
|
if client.check_signature(&sig) {
|
|
|
|
println!("Confirmed");
|
|
|
|
} else {
|
|
|
|
println!("Not found");
|
2018-06-29 10:38:00 -07:00
|
|
|
}
|
2018-06-29 16:06:05 -07:00
|
|
|
}
|
2018-06-26 22:52:34 -07:00
|
|
|
}
|
2018-06-29 10:38:00 -07:00
|
|
|
Ok(())
|
2018-06-26 22:52:34 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
fn display_actions() {
|
|
|
|
println!("");
|
2018-06-29 15:04:35 -07:00
|
|
|
println!("Commands:");
|
|
|
|
println!(" balance Get your account balance");
|
|
|
|
println!(" airdrop Request a batch of tokens");
|
|
|
|
println!(" pay Spend your tokens as fast as possible");
|
|
|
|
println!(" confirm Confirm your last payment by signature");
|
2018-06-26 22:52:34 -07:00
|
|
|
println!("");
|
|
|
|
}
|
|
|
|
|
|
|
|
fn read_leader(path: String) -> ReplicatedData {
|
|
|
|
let file = File::open(path.clone()).expect(&format!("file not found: {}", path));
|
|
|
|
serde_json::from_reader(file).expect(&format!("failed to parse {}", path))
|
2018-06-28 11:58:33 -07:00
|
|
|
}
|
|
|
|
|
2018-06-29 10:38:00 -07:00
|
|
|
fn read_mint(path: String) -> Result<Mint, Box<error::Error>> {
|
|
|
|
let file = File::open(path.clone())?;
|
|
|
|
let mint = serde_json::from_reader(file)?;
|
|
|
|
Ok(mint)
|
2018-06-26 22:52:34 -07:00
|
|
|
}
|
|
|
|
|
2018-06-27 14:41:49 -07:00
|
|
|
fn mk_client(client_addr: &SocketAddr, r: &ReplicatedData) -> io::Result<ThinClient> {
|
2018-06-26 22:52:34 -07:00
|
|
|
let mut addr = client_addr.clone();
|
|
|
|
let port = addr.port();
|
2018-06-27 14:41:49 -07:00
|
|
|
let transactions_socket = UdpSocket::bind(addr.clone())?;
|
2018-06-26 22:52:34 -07:00
|
|
|
addr.set_port(port + 1);
|
2018-06-27 14:41:49 -07:00
|
|
|
let requests_socket = UdpSocket::bind(addr.clone())?;
|
|
|
|
requests_socket.set_read_timeout(Some(Duration::new(1, 0)))?;
|
2018-06-26 22:52:34 -07:00
|
|
|
|
|
|
|
addr.set_port(port + 2);
|
2018-06-27 14:41:49 -07:00
|
|
|
Ok(ThinClient::new(
|
2018-06-26 22:52:34 -07:00
|
|
|
r.requests_addr,
|
|
|
|
requests_socket,
|
|
|
|
r.transactions_addr,
|
|
|
|
transactions_socket,
|
2018-06-27 14:41:49 -07:00
|
|
|
))
|
2018-06-26 22:52:34 -07:00
|
|
|
}
|
|
|
|
|
2018-06-27 14:41:49 -07:00
|
|
|
fn request_airdrop(drone_addr: &SocketAddr, id: &Mint) {
|
2018-06-26 22:52:34 -07:00
|
|
|
let mut stream = TcpStream::connect(drone_addr).unwrap();
|
|
|
|
let req = DroneRequest::GetAirdrop {
|
2018-06-27 14:41:49 -07:00
|
|
|
airdrop_request_amount: id.tokens as u64,
|
|
|
|
client_public_key: id.pubkey(),
|
2018-06-26 22:52:34 -07:00
|
|
|
};
|
|
|
|
let tx = serialize(&req).expect("serialize drone request");
|
|
|
|
stream.write_all(&tx).unwrap();
|
|
|
|
// TODO: add timeout to this function, in case of unresponsive drone
|
|
|
|
}
|
2018-06-29 10:38:00 -07:00
|
|
|
|
|
|
|
fn main() -> Result<(), Box<error::Error>> {
|
|
|
|
env_logger::init();
|
|
|
|
let config = parse_args(env::args().collect())?;
|
|
|
|
let mut client = mk_client(&config.client_addr, &config.leader)?;
|
|
|
|
process_command(&config, &mut client)
|
|
|
|
}
|