Merge pull request #8 from DenioD/master
change wallet.dat to silentdragonlite/ , merged new updates
This commit is contained in:
@@ -16,7 +16,7 @@ Run `silentdragonlite-cli help` to see a list of all commands.
|
|||||||
## Notes:
|
## Notes:
|
||||||
* If you want to run your own server, please see [SilentDragonLite-cli lightwalletd](https://github.com/MyHush/lightwalletd), and then run `./silentdragonlite-cli --server http://127.0.0.1:9069`. You might also need to pass `--dangerous` if you are using a self-signed TLS certificate.
|
* If you want to run your own server, please see [SilentDragonLite-cli lightwalletd](https://github.com/MyHush/lightwalletd), and then run `./silentdragonlite-cli --server http://127.0.0.1:9069`. You might also need to pass `--dangerous` if you are using a self-signed TLS certificate.
|
||||||
|
|
||||||
* The log file is in `~/.komodo/HUSH3/silentdragonlite-cli.debug.log`. Wallet is stored in `~/.komodo/HUSH3/silentdragonlite-cli.dat`
|
* The log file is in `~/.silentdragonlite/silentdragonlite-cli.debug.log`. Wallet is stored in `~/.silentdragonlite/silentdragonlite-cli.dat`
|
||||||
|
|
||||||
### Note Management
|
### Note Management
|
||||||
silentdragonlite does automatic note and utxo management, which means it doesn't allow you to manually select which address to send outgoing transactions from. It follows these principles:
|
silentdragonlite does automatic note and utxo management, which means it doesn't allow you to manually select which address to send outgoing transactions from. It follows these principles:
|
||||||
|
|||||||
295
cli/src/lib.rs
Normal file
295
cli/src/lib.rs
Normal file
@@ -0,0 +1,295 @@
|
|||||||
|
use std::io::{self, Error, ErrorKind};
|
||||||
|
use std::sync::Arc;
|
||||||
|
use std::sync::mpsc::{channel, Sender, Receiver};
|
||||||
|
|
||||||
|
use log::{info, error, LevelFilter};
|
||||||
|
use log4rs::append::rolling_file::RollingFileAppender;
|
||||||
|
use log4rs::encode::pattern::PatternEncoder;
|
||||||
|
use log4rs::config::{Appender, Config, Root};
|
||||||
|
use log4rs::filter::threshold::ThresholdFilter;
|
||||||
|
use log4rs::append::rolling_file::policy::compound::{
|
||||||
|
CompoundPolicy,
|
||||||
|
trigger::size::SizeTrigger,
|
||||||
|
roll::fixed_window::FixedWindowRoller,
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
|
use silentdragonlitelib::{commands,
|
||||||
|
lightclient::{LightClient, LightClientConfig},
|
||||||
|
};
|
||||||
|
|
||||||
|
#[macro_export]
|
||||||
|
macro_rules! configure_clapapp {
|
||||||
|
( $freshapp: expr ) => {
|
||||||
|
$freshapp.version("1.0.0")
|
||||||
|
.arg(Arg::with_name("dangerous")
|
||||||
|
.long("dangerous")
|
||||||
|
.help("Disable server TLS certificate verification. Use this if you're running a local lightwalletd with a self-signed certificate. WARNING: This is dangerous, don't use it with a server that is not your own.")
|
||||||
|
.takes_value(false))
|
||||||
|
.arg(Arg::with_name("nosync")
|
||||||
|
.help("By default, Silentdragonlite-cli will sync the wallet at startup. Pass --nosync to prevent the automatic sync at startup.")
|
||||||
|
.long("nosync")
|
||||||
|
.short("n")
|
||||||
|
.takes_value(false))
|
||||||
|
.arg(Arg::with_name("recover")
|
||||||
|
.long("recover")
|
||||||
|
.help("Attempt to recover the seed from the wallet")
|
||||||
|
.takes_value(false))
|
||||||
|
.arg(Arg::with_name("seed")
|
||||||
|
.short("s")
|
||||||
|
.long("seed")
|
||||||
|
.value_name("seed_phrase")
|
||||||
|
.help("Create a new wallet with the given 24-word seed phrase. Will fail if wallet already exists")
|
||||||
|
.takes_value(true))
|
||||||
|
.arg(Arg::with_name("birthday")
|
||||||
|
.long("birthday")
|
||||||
|
.value_name("birthday")
|
||||||
|
.help("Specify wallet birthday when restoring from seed. This is the earlist block height where the wallet has a transaction.")
|
||||||
|
.takes_value(true))
|
||||||
|
.arg(Arg::with_name("server")
|
||||||
|
.long("server")
|
||||||
|
.value_name("server")
|
||||||
|
.help("Lightwalletd server to connect to.")
|
||||||
|
.takes_value(true)
|
||||||
|
.default_value(lightclient::DEFAULT_SERVER))
|
||||||
|
.arg(Arg::with_name("COMMAND")
|
||||||
|
.help("Command to execute. If a command is not specified, Silentdragonlite-cli will start in interactive mode.")
|
||||||
|
.required(false)
|
||||||
|
.index(1))
|
||||||
|
.arg(Arg::with_name("PARAMS")
|
||||||
|
.help("Params to execute command with. Run the 'help' command to get usage help.")
|
||||||
|
.required(false)
|
||||||
|
.multiple(true))
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/// This function is only tested against Linux.
|
||||||
|
pub fn report_permission_error() {
|
||||||
|
let user = std::env::var("USER").expect(
|
||||||
|
"Unexpected error reading value of $USER!");
|
||||||
|
let home = std::env::var("HOME").expect(
|
||||||
|
"Unexpected error reading value of $HOME!");
|
||||||
|
let current_executable = std::env::current_exe()
|
||||||
|
.expect("Unexpected error reporting executable path!");
|
||||||
|
eprintln!("USER: {}", user);
|
||||||
|
eprintln!("HOME: {}", home);
|
||||||
|
eprintln!("Executable: {}", current_executable.display());
|
||||||
|
if home == "/" {
|
||||||
|
eprintln!("User {} must have permission to write to '{}silentdragonlite/' .",
|
||||||
|
user,
|
||||||
|
home);
|
||||||
|
} else {
|
||||||
|
eprintln!("User {} must have permission to write to '{}/silentdragonlite/' .",
|
||||||
|
user,
|
||||||
|
home);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Build the Logging config
|
||||||
|
pub fn get_log_config(config: &LightClientConfig) -> io::Result<Config> {
|
||||||
|
let window_size = 3; // log0, log1, log2
|
||||||
|
let fixed_window_roller =
|
||||||
|
FixedWindowRoller::builder().build("Silentdragonlite-light-wallet-log{}",window_size).unwrap();
|
||||||
|
let size_limit = 5 * 1024 * 1024; // 5MB as max log file size to roll
|
||||||
|
let size_trigger = SizeTrigger::new(size_limit);
|
||||||
|
let compound_policy = CompoundPolicy::new(Box::new(size_trigger),Box::new(fixed_window_roller));
|
||||||
|
|
||||||
|
Config::builder()
|
||||||
|
.appender(
|
||||||
|
Appender::builder()
|
||||||
|
.filter(Box::new(ThresholdFilter::new(LevelFilter::Info)))
|
||||||
|
.build(
|
||||||
|
"logfile",
|
||||||
|
Box::new(
|
||||||
|
RollingFileAppender::builder()
|
||||||
|
.encoder(Box::new(PatternEncoder::new("{d} {l}::{m}{n}")))
|
||||||
|
.build(config.get_log_path(), Box::new(compound_policy))?,
|
||||||
|
),
|
||||||
|
),
|
||||||
|
)
|
||||||
|
.build(
|
||||||
|
Root::builder()
|
||||||
|
.appender("logfile")
|
||||||
|
.build(LevelFilter::Debug),
|
||||||
|
)
|
||||||
|
.map_err(|e|Error::new(ErrorKind::Other, format!("{}", e)))
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
pub fn startup(server: http::Uri, dangerous: bool, seed: Option<String>, birthday: u64, first_sync: bool, print_updates: bool)
|
||||||
|
-> io::Result<(Sender<(String, Vec<String>)>, Receiver<String>)> {
|
||||||
|
// Try to get the configuration
|
||||||
|
let (config, latest_block_height) = LightClientConfig::create(server.clone(), dangerous)?;
|
||||||
|
|
||||||
|
// Configure logging first.
|
||||||
|
let log_config = get_log_config(&config)?;
|
||||||
|
log4rs::init_config(log_config).map_err(|e| {
|
||||||
|
std::io::Error::new(ErrorKind::Other, e)
|
||||||
|
})?;
|
||||||
|
|
||||||
|
let lightclient = match seed {
|
||||||
|
Some(phrase) => Arc::new(LightClient::new_from_phrase(phrase, &config, birthday)?),
|
||||||
|
None => {
|
||||||
|
if config.wallet_exists() {
|
||||||
|
Arc::new(LightClient::read_from_disk(&config)?)
|
||||||
|
} else {
|
||||||
|
println!("Creating a new wallet");
|
||||||
|
Arc::new(LightClient::new(&config, latest_block_height)?)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// Print startup Messages
|
||||||
|
info!(""); // Blank line
|
||||||
|
info!("Starting Silentdragonlite-CLI");
|
||||||
|
info!("Light Client config {:?}", config);
|
||||||
|
|
||||||
|
if print_updates {
|
||||||
|
println!("Lightclient connecting to {}", config.server);
|
||||||
|
}
|
||||||
|
|
||||||
|
// At startup, run a sync.
|
||||||
|
if first_sync {
|
||||||
|
let update = lightclient.do_sync(true);
|
||||||
|
if print_updates {
|
||||||
|
match update {
|
||||||
|
Ok(j) => {
|
||||||
|
println!("{}", j.pretty(2));
|
||||||
|
},
|
||||||
|
Err(e) => println!("{}", e)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Start the command loop
|
||||||
|
let (command_tx, resp_rx) = command_loop(lightclient.clone());
|
||||||
|
|
||||||
|
Ok((command_tx, resp_rx))
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn start_interactive(command_tx: Sender<(String, Vec<String>)>, resp_rx: Receiver<String>) {
|
||||||
|
// `()` can be used when no completer is required
|
||||||
|
let mut rl = rustyline::Editor::<()>::new();
|
||||||
|
|
||||||
|
println!("Ready!");
|
||||||
|
|
||||||
|
let send_command = |cmd: String, args: Vec<String>| -> String {
|
||||||
|
command_tx.send((cmd.clone(), args)).unwrap();
|
||||||
|
match resp_rx.recv() {
|
||||||
|
Ok(s) => s,
|
||||||
|
Err(e) => {
|
||||||
|
let e = format!("Error executing command {}: {}", cmd, e);
|
||||||
|
eprintln!("{}", e);
|
||||||
|
error!("{}", e);
|
||||||
|
return "".to_string()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let info = &send_command("info".to_string(), vec![]);
|
||||||
|
let chain_name = json::parse(info).unwrap()["chain_name"].as_str().unwrap().to_string();
|
||||||
|
|
||||||
|
loop {
|
||||||
|
// Read the height first
|
||||||
|
let height = json::parse(&send_command("height".to_string(), vec![])).unwrap()["height"].as_i64().unwrap();
|
||||||
|
|
||||||
|
let readline = rl.readline(&format!("({}) Block:{} (type 'help') >> ",
|
||||||
|
chain_name, height));
|
||||||
|
match readline {
|
||||||
|
Ok(line) => {
|
||||||
|
rl.add_history_entry(line.as_str());
|
||||||
|
// Parse command line arguments
|
||||||
|
let mut cmd_args = match shellwords::split(&line) {
|
||||||
|
Ok(args) => args,
|
||||||
|
Err(_) => {
|
||||||
|
println!("Mismatched Quotes");
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
if cmd_args.is_empty() {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
let cmd = cmd_args.remove(0);
|
||||||
|
let args: Vec<String> = cmd_args;
|
||||||
|
|
||||||
|
println!("{}", send_command(cmd, args));
|
||||||
|
|
||||||
|
// Special check for Quit command.
|
||||||
|
if line == "quit" {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
},
|
||||||
|
Err(rustyline::error::ReadlineError::Interrupted) => {
|
||||||
|
println!("CTRL-C");
|
||||||
|
info!("CTRL-C");
|
||||||
|
println!("{}", send_command("save".to_string(), vec![]));
|
||||||
|
break
|
||||||
|
},
|
||||||
|
Err(rustyline::error::ReadlineError::Eof) => {
|
||||||
|
println!("CTRL-D");
|
||||||
|
info!("CTRL-D");
|
||||||
|
println!("{}", send_command("save".to_string(), vec![]));
|
||||||
|
break
|
||||||
|
},
|
||||||
|
Err(err) => {
|
||||||
|
println!("Error: {:?}", err);
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
pub fn command_loop(lightclient: Arc<LightClient>) -> (Sender<(String, Vec<String>)>, Receiver<String>) {
|
||||||
|
let (command_tx, command_rx) = channel::<(String, Vec<String>)>();
|
||||||
|
let (resp_tx, resp_rx) = channel::<String>();
|
||||||
|
|
||||||
|
let lc = lightclient.clone();
|
||||||
|
std::thread::spawn(move || {
|
||||||
|
loop {
|
||||||
|
match command_rx.recv_timeout(std::time::Duration::from_secs(5 * 60)) {
|
||||||
|
Ok((cmd, args)) => {
|
||||||
|
let args = args.iter().map(|s| s.as_ref()).collect();
|
||||||
|
|
||||||
|
let cmd_response = commands::do_user_command(&cmd, &args, lc.as_ref());
|
||||||
|
resp_tx.send(cmd_response).unwrap();
|
||||||
|
|
||||||
|
if cmd == "quit" {
|
||||||
|
info!("Quit");
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
},
|
||||||
|
Err(_) => {
|
||||||
|
// Timeout. Do a sync to keep the wallet up-to-date. False to whether to print updates on the console
|
||||||
|
info!("Timeout, doing a sync");
|
||||||
|
match lc.do_sync(false) {
|
||||||
|
Ok(_) => {},
|
||||||
|
Err(e) => {error!("{}", e)}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
(command_tx, resp_rx)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn attempt_recover_seed() {
|
||||||
|
// Create a Light Client Config in an attempt to recover the file.
|
||||||
|
let config = LightClientConfig {
|
||||||
|
server: "0.0.0.0:0".parse().unwrap(),
|
||||||
|
chain_name: "main".to_string(),
|
||||||
|
sapling_activation_height: 0,
|
||||||
|
consensus_branch_id: "000000".to_string(),
|
||||||
|
anchor_offset: 0,
|
||||||
|
no_cert_verification: false,
|
||||||
|
data_dir: None,
|
||||||
|
};
|
||||||
|
|
||||||
|
match LightClient::attempt_recover_seed(&config) {
|
||||||
|
Ok(seed) => println!("Recovered seed: '{}'", seed),
|
||||||
|
Err(e) => eprintln!("Failed to recover seed. Error: {}", e)
|
||||||
|
};
|
||||||
|
}
|
||||||
285
cli/src/main.rs
285
cli/src/main.rs
@@ -1,116 +1,20 @@
|
|||||||
use std::io::{self, Error, ErrorKind};
|
use silentdragonlitelib::lightclient::{self, LightClientConfig};
|
||||||
use std::sync::Arc;
|
use silentdragonlite_cli::{configure_clapapp,
|
||||||
use std::sync::mpsc::{channel, Sender, Receiver};
|
report_permission_error,
|
||||||
|
startup,
|
||||||
use silentdragonlitelib::{commands, startup_helpers,
|
start_interactive,
|
||||||
lightclient::{self, LightClient, LightClientConfig},
|
attempt_recover_seed};
|
||||||
};
|
use log::error;
|
||||||
|
|
||||||
use log::{info, error, LevelFilter};
|
|
||||||
use log4rs::append::rolling_file::RollingFileAppender;
|
|
||||||
use log4rs::encode::pattern::PatternEncoder;
|
|
||||||
use log4rs::config::{Appender, Config, Root};
|
|
||||||
use log4rs::filter::threshold::ThresholdFilter;
|
|
||||||
use log4rs::append::rolling_file::policy::compound::{
|
|
||||||
CompoundPolicy,
|
|
||||||
trigger::size::SizeTrigger,
|
|
||||||
roll::fixed_window::FixedWindowRoller,
|
|
||||||
};
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
/// Build the Logging config
|
|
||||||
fn get_log_config(config: &LightClientConfig) -> io::Result<Config> {
|
|
||||||
let window_size = 3; // log0, log1, log2
|
|
||||||
let fixed_window_roller =
|
|
||||||
FixedWindowRoller::builder().build("SilentDragonLite-light-wallet-log{}",window_size).unwrap();
|
|
||||||
let size_limit = 5 * 1024 * 1024; // 5MB as max log file size to roll
|
|
||||||
let size_trigger = SizeTrigger::new(size_limit);
|
|
||||||
let compound_policy = CompoundPolicy::new(Box::new(size_trigger),Box::new(fixed_window_roller));
|
|
||||||
|
|
||||||
Config::builder()
|
|
||||||
.appender(
|
|
||||||
Appender::builder()
|
|
||||||
.filter(Box::new(ThresholdFilter::new(LevelFilter::Info)))
|
|
||||||
.build(
|
|
||||||
"logfile",
|
|
||||||
Box::new(
|
|
||||||
RollingFileAppender::builder()
|
|
||||||
.encoder(Box::new(PatternEncoder::new("{d} {l}::{m}{n}")))
|
|
||||||
.build(config.get_log_path(), Box::new(compound_policy))?,
|
|
||||||
),
|
|
||||||
),
|
|
||||||
)
|
|
||||||
.build(
|
|
||||||
Root::builder()
|
|
||||||
.appender("logfile")
|
|
||||||
.build(LevelFilter::Debug),
|
|
||||||
)
|
|
||||||
.map_err(|e|Error::new(ErrorKind::Other, format!("{}", e)))
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
pub fn main() {
|
pub fn main() {
|
||||||
// Get command line arguments
|
// Get command line arguments
|
||||||
use clap::{Arg, App};
|
use clap::{App, Arg};
|
||||||
let matches = App::new("SilentDragon CLI")
|
let fresh_app = App::new("SilentDragonLite CLI");
|
||||||
.version("1.1.0")
|
let configured_app = configure_clapapp!(fresh_app);
|
||||||
.arg(Arg::with_name("seed")
|
let matches = configured_app.get_matches();
|
||||||
.short("s")
|
|
||||||
.long("seed")
|
|
||||||
.value_name("seed_phrase")
|
|
||||||
.help("Create a new wallet with the given 24-word seed phrase. Will fail if wallet already exists")
|
|
||||||
.takes_value(true))
|
|
||||||
.arg(Arg::with_name("birthday")
|
|
||||||
.long("birthday")
|
|
||||||
.value_name("birthday")
|
|
||||||
.help("Specify wallet birthday when restoring from seed. This is the earlist block height where the wallet has a transaction.")
|
|
||||||
.takes_value(true))
|
|
||||||
.arg(Arg::with_name("server")
|
|
||||||
.long("server")
|
|
||||||
.value_name("server")
|
|
||||||
.help("Lightwalletd server to connect to.")
|
|
||||||
.takes_value(true)
|
|
||||||
.default_value(lightclient::DEFAULT_SERVER))
|
|
||||||
.arg(Arg::with_name("dangerous")
|
|
||||||
.long("dangerous")
|
|
||||||
.help("Disable server TLS certificate verification. Use this if you're running a local lightwalletd with a self-signed certificate. WARNING: This is dangerous, don't use it with a server that is not your own.")
|
|
||||||
.takes_value(false))
|
|
||||||
.arg(Arg::with_name("recover")
|
|
||||||
.long("recover")
|
|
||||||
.help("Attempt to recover the seed from the wallet")
|
|
||||||
.takes_value(false))
|
|
||||||
.arg(Arg::with_name("nosync")
|
|
||||||
.help("By default, silentdragonlite-cli will sync the wallet at startup. Pass --nosync to prevent the automatic sync at startup.")
|
|
||||||
.long("nosync")
|
|
||||||
.short("n")
|
|
||||||
.takes_value(false))
|
|
||||||
.arg(Arg::with_name("COMMAND")
|
|
||||||
.help("Command to execute. If a command is not specified, silentdragonlite-cli will start in interactive mode.")
|
|
||||||
.required(false)
|
|
||||||
.index(1))
|
|
||||||
.arg(Arg::with_name("PARAMS")
|
|
||||||
.help("Params to execute command with. Run the 'help' command to get usage help.")
|
|
||||||
.required(false)
|
|
||||||
.multiple(true))
|
|
||||||
.get_matches();
|
|
||||||
|
|
||||||
if matches.is_present("recover") {
|
if matches.is_present("recover") {
|
||||||
// Create a Light Client Config in an attempt to recover the file.
|
// Create a Light Client Config in an attempt to recover the file.
|
||||||
let config = LightClientConfig {
|
attempt_recover_seed();
|
||||||
server: "0.0.0.0:0".parse().unwrap(),
|
|
||||||
chain_name: "main".to_string(),
|
|
||||||
sapling_activation_height: 0,
|
|
||||||
consensus_branch_id: "000000".to_string(),
|
|
||||||
anchor_offset: 0,
|
|
||||||
no_cert_verification: false,
|
|
||||||
data_dir: None,
|
|
||||||
};
|
|
||||||
|
|
||||||
match LightClient::attempt_recover_seed(&config) {
|
|
||||||
Ok(seed) => println!("Recovered seed: '{}'", seed),
|
|
||||||
Err(e) => eprintln!("Failed to recover seed. Error: {}", e)
|
|
||||||
};
|
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -152,12 +56,12 @@ pub fn main() {
|
|||||||
Err(e) => {
|
Err(e) => {
|
||||||
eprintln!("Error during startup: {}", e);
|
eprintln!("Error during startup: {}", e);
|
||||||
error!("Error during startup: {}", e);
|
error!("Error during startup: {}", e);
|
||||||
match e.raw_os_error() {
|
if cfg!(target_os = "unix" ) {
|
||||||
Some(13) => {
|
match e.raw_os_error() {
|
||||||
startup_helpers::report_permission_error();
|
Some(13) => report_permission_error(),
|
||||||
},
|
_ => {},
|
||||||
_ => {}
|
}
|
||||||
}
|
};
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
@@ -184,156 +88,3 @@ pub fn main() {
|
|||||||
resp_rx.recv().unwrap();
|
resp_rx.recv().unwrap();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn startup(server: http::Uri, dangerous: bool, seed: Option<String>, birthday: u64, first_sync: bool, print_updates: bool)
|
|
||||||
-> io::Result<(Sender<(String, Vec<String>)>, Receiver<String>)> {
|
|
||||||
// Try to get the configuration
|
|
||||||
let (config, latest_block_height) = LightClientConfig::create(server.clone(), dangerous)?;
|
|
||||||
|
|
||||||
// Configure logging first.
|
|
||||||
let log_config = get_log_config(&config)?;
|
|
||||||
log4rs::init_config(log_config).map_err(|e| {
|
|
||||||
std::io::Error::new(ErrorKind::Other, e)
|
|
||||||
})?;
|
|
||||||
|
|
||||||
let lightclient = match seed {
|
|
||||||
Some(phrase) => Arc::new(LightClient::new_from_phrase(phrase, &config, birthday)?),
|
|
||||||
None => {
|
|
||||||
if config.wallet_exists() {
|
|
||||||
Arc::new(LightClient::read_from_disk(&config)?)
|
|
||||||
} else {
|
|
||||||
println!("Creating a new wallet");
|
|
||||||
Arc::new(LightClient::new(&config, latest_block_height)?)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
// Print startup Messages
|
|
||||||
info!(""); // Blank line
|
|
||||||
info!("Starting SilentDragonLite CLI");
|
|
||||||
info!("Light Client config {:?}", config);
|
|
||||||
|
|
||||||
if print_updates {
|
|
||||||
println!("Lightclient connecting to {}", config.server);
|
|
||||||
}
|
|
||||||
|
|
||||||
// At startup, run a sync.
|
|
||||||
if first_sync {
|
|
||||||
let update = lightclient.do_sync(true);
|
|
||||||
if print_updates {
|
|
||||||
println!("{}", update);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Start the command loop
|
|
||||||
let (command_tx, resp_rx) = command_loop(lightclient.clone());
|
|
||||||
|
|
||||||
Ok((command_tx, resp_rx))
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
fn start_interactive(command_tx: Sender<(String, Vec<String>)>, resp_rx: Receiver<String>) {
|
|
||||||
// `()` can be used when no completer is required
|
|
||||||
let mut rl = rustyline::Editor::<()>::new();
|
|
||||||
|
|
||||||
println!("Ready!");
|
|
||||||
|
|
||||||
let send_command = |cmd: String, args: Vec<String>| -> String {
|
|
||||||
command_tx.send((cmd.clone(), args)).unwrap();
|
|
||||||
match resp_rx.recv() {
|
|
||||||
Ok(s) => s,
|
|
||||||
Err(e) => {
|
|
||||||
let e = format!("Error executing command {}: {}", cmd, e);
|
|
||||||
eprintln!("{}", e);
|
|
||||||
error!("{}", e);
|
|
||||||
return "".to_string()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
let info = &send_command("info".to_string(), vec![]);
|
|
||||||
let chain_name = json::parse(info).unwrap()["chain_name"].as_str().unwrap().to_string();
|
|
||||||
|
|
||||||
loop {
|
|
||||||
// Read the height first
|
|
||||||
let height = json::parse(&send_command("height".to_string(), vec![])).unwrap()["height"].as_i64().unwrap();
|
|
||||||
|
|
||||||
let readline = rl.readline(&format!("({}) Block:{} (type 'help') >> ",
|
|
||||||
chain_name, height));
|
|
||||||
match readline {
|
|
||||||
Ok(line) => {
|
|
||||||
rl.add_history_entry(line.as_str());
|
|
||||||
// Parse command line arguments
|
|
||||||
let mut cmd_args = match shellwords::split(&line) {
|
|
||||||
Ok(args) => args,
|
|
||||||
Err(_) => {
|
|
||||||
println!("Mismatched Quotes");
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
if cmd_args.is_empty() {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
let cmd = cmd_args.remove(0);
|
|
||||||
let args: Vec<String> = cmd_args;
|
|
||||||
|
|
||||||
println!("{}", send_command(cmd, args));
|
|
||||||
|
|
||||||
// Special check for Quit command.
|
|
||||||
if line == "quit" {
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
},
|
|
||||||
Err(rustyline::error::ReadlineError::Interrupted) => {
|
|
||||||
println!("CTRL-C");
|
|
||||||
info!("CTRL-C");
|
|
||||||
println!("{}", send_command("save".to_string(), vec![]));
|
|
||||||
break
|
|
||||||
},
|
|
||||||
Err(rustyline::error::ReadlineError::Eof) => {
|
|
||||||
println!("CTRL-D");
|
|
||||||
info!("CTRL-D");
|
|
||||||
println!("{}", send_command("save".to_string(), vec![]));
|
|
||||||
break
|
|
||||||
},
|
|
||||||
Err(err) => {
|
|
||||||
println!("Error: {:?}", err);
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
fn command_loop(lightclient: Arc<LightClient>) -> (Sender<(String, Vec<String>)>, Receiver<String>) {
|
|
||||||
let (command_tx, command_rx) = channel::<(String, Vec<String>)>();
|
|
||||||
let (resp_tx, resp_rx) = channel::<String>();
|
|
||||||
|
|
||||||
let lc = lightclient.clone();
|
|
||||||
std::thread::spawn(move || {
|
|
||||||
loop {
|
|
||||||
match command_rx.recv_timeout(std::time::Duration::from_secs(5 * 60)) {
|
|
||||||
Ok((cmd, args)) => {
|
|
||||||
let args = args.iter().map(|s| s.as_ref()).collect();
|
|
||||||
|
|
||||||
let cmd_response = commands::do_user_command(&cmd, &args, lc.as_ref());
|
|
||||||
resp_tx.send(cmd_response).unwrap();
|
|
||||||
|
|
||||||
if cmd == "quit" {
|
|
||||||
info!("Quit");
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
},
|
|
||||||
Err(_) => {
|
|
||||||
// Timeout. Do a sync to keep the wallet up-to-date. False to whether to print updates on the console
|
|
||||||
info!("Timeout, doing a sync");
|
|
||||||
lc.do_sync(false);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
(command_tx, resp_rx)
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -28,7 +28,38 @@ impl Command for SyncCommand {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn exec(&self, _args: &[&str], lightclient: &LightClient) -> String {
|
fn exec(&self, _args: &[&str], lightclient: &LightClient) -> String {
|
||||||
lightclient.do_sync(true)
|
match lightclient.do_sync(true) {
|
||||||
|
Ok(j) => j.pretty(2),
|
||||||
|
Err(e) => e
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
struct SyncStatusCommand {}
|
||||||
|
impl Command for SyncStatusCommand {
|
||||||
|
fn help(&self) -> String {
|
||||||
|
let mut h = vec![];
|
||||||
|
h.push("Get the sync status of the wallet");
|
||||||
|
h.push("Usage:");
|
||||||
|
h.push("syncstatus");
|
||||||
|
h.push("");
|
||||||
|
|
||||||
|
h.join("\n")
|
||||||
|
}
|
||||||
|
|
||||||
|
fn short_help(&self) -> String {
|
||||||
|
"Get the sync status of the wallet".to_string()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn exec(&self, _args: &[&str], lightclient: &LightClient) -> String {
|
||||||
|
let status = lightclient.do_scan_status();
|
||||||
|
match status.is_syncing {
|
||||||
|
false => object!{ "syncing" => "false" },
|
||||||
|
true => object!{ "syncing" => "true",
|
||||||
|
"synced_blocks" => status.synced_blocks,
|
||||||
|
"total_blocks" => status.total_blocks }
|
||||||
|
}.pretty(2)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -51,7 +82,10 @@ impl Command for RescanCommand {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn exec(&self, _args: &[&str], lightclient: &LightClient) -> String {
|
fn exec(&self, _args: &[&str], lightclient: &LightClient) -> String {
|
||||||
lightclient.do_rescan()
|
match lightclient.do_rescan() {
|
||||||
|
Ok(j) => j.pretty(2),
|
||||||
|
Err(e) => e
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -116,9 +150,7 @@ impl Command for InfoCommand {
|
|||||||
"Get the lightwalletd server's info".to_string()
|
"Get the lightwalletd server's info".to_string()
|
||||||
}
|
}
|
||||||
|
|
||||||
fn exec(&self, _args: &[&str], lightclient: &LightClient) -> String {
|
fn exec(&self, _args: &[&str], lightclient: &LightClient) -> String {
|
||||||
lightclient.do_sync(true);
|
|
||||||
|
|
||||||
lightclient.do_info()
|
lightclient.do_info()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -141,9 +173,10 @@ impl Command for BalanceCommand {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn exec(&self, _args: &[&str], lightclient: &LightClient) -> String {
|
fn exec(&self, _args: &[&str], lightclient: &LightClient) -> String {
|
||||||
lightclient.do_sync(true);
|
match lightclient.do_sync(true) {
|
||||||
|
Ok(_) => format!("{}", lightclient.do_balance().pretty(2)),
|
||||||
format!("{}", lightclient.do_balance().pretty(2))
|
Err(e) => e
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -405,8 +438,8 @@ impl Command for SendCommand {
|
|||||||
return self.help();
|
return self.help();
|
||||||
}
|
}
|
||||||
|
|
||||||
// Check for a single argument that can be parsed as JSON
|
// Check for a single argument that can be parsed as JSON
|
||||||
if args.len() == 1 {
|
let send_args = if args.len() == 1 {
|
||||||
// Sometimes on the command line, people use "'" for the quotes, which json::parse doesn't
|
// Sometimes on the command line, people use "'" for the quotes, which json::parse doesn't
|
||||||
// understand. So replace it with double-quotes
|
// understand. So replace it with double-quotes
|
||||||
let arg_list = args[0].replace("'", "\"");
|
let arg_list = args[0].replace("'", "\"");
|
||||||
@@ -427,20 +460,14 @@ impl Command for SendCommand {
|
|||||||
if !j.has_key("address") || !j.has_key("amount") {
|
if !j.has_key("address") || !j.has_key("amount") {
|
||||||
Err(format!("Need 'address' and 'amount'\n"))
|
Err(format!("Need 'address' and 'amount'\n"))
|
||||||
} else {
|
} else {
|
||||||
Ok((j["address"].as_str().unwrap(), j["amount"].as_u64().unwrap(), j["memo"].as_str().map(|s| s.to_string())))
|
Ok((j["address"].as_str().unwrap().to_string().clone(), j["amount"].as_u64().unwrap(), j["memo"].as_str().map(|s| s.to_string().clone())))
|
||||||
}
|
}
|
||||||
}).collect::<Result<Vec<(&str, u64, Option<String>)>, String>>();
|
}).collect::<Result<Vec<(String, u64, Option<String>)>, String>>();
|
||||||
|
|
||||||
let send_args = match maybe_send_args {
|
match maybe_send_args {
|
||||||
Ok(a) => a,
|
Ok(a) => a.clone(),
|
||||||
Err(s) => { return format!("Error: {}\n{}", s, self.help()); }
|
Err(s) => { return format!("Error: {}\n{}", s, self.help()); }
|
||||||
};
|
}
|
||||||
|
|
||||||
lightclient.do_sync(true);
|
|
||||||
match lightclient.do_send(send_args) {
|
|
||||||
Ok(txid) => { object!{ "txid" => txid } },
|
|
||||||
Err(e) => { object!{ "error" => e } }
|
|
||||||
}.pretty(2)
|
|
||||||
} else if args.len() == 2 || args.len() == 3 {
|
} else if args.len() == 2 || args.len() == 3 {
|
||||||
// Make sure we can parse the amount
|
// Make sure we can parse the amount
|
||||||
let value = match args[1].parse::<u64>() {
|
let value = match args[1].parse::<u64>() {
|
||||||
@@ -451,13 +478,22 @@ impl Command for SendCommand {
|
|||||||
};
|
};
|
||||||
|
|
||||||
let memo = if args.len() == 3 { Some(args[2].to_string()) } else {None};
|
let memo = if args.len() == 3 { Some(args[2].to_string()) } else {None};
|
||||||
lightclient.do_sync(true);
|
|
||||||
match lightclient.do_send(vec!((args[0], value, memo))) {
|
vec![(args[0].to_string(), value, memo)]
|
||||||
Ok(txid) => { object!{ "txid" => txid } },
|
|
||||||
Err(e) => { object!{ "error" => e } }
|
|
||||||
}.pretty(2)
|
|
||||||
} else {
|
} else {
|
||||||
self.help()
|
return self.help()
|
||||||
|
};
|
||||||
|
|
||||||
|
match lightclient.do_sync(true) {
|
||||||
|
Ok(_) => {
|
||||||
|
// Convert to the right format. String -> &str.
|
||||||
|
let tos = send_args.iter().map(|(a, v, m)| (a.as_str(), *v, m.clone()) ).collect::<Vec<_>>();
|
||||||
|
match lightclient.do_send(tos) {
|
||||||
|
Ok(txid) => { object!{ "txid" => txid } },
|
||||||
|
Err(e) => { object!{ "error" => e } }
|
||||||
|
}.pretty(2)
|
||||||
|
},
|
||||||
|
Err(e) => e
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -539,9 +575,12 @@ impl Command for TransactionsCommand {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn exec(&self, _args: &[&str], lightclient: &LightClient) -> String {
|
fn exec(&self, _args: &[&str], lightclient: &LightClient) -> String {
|
||||||
lightclient.do_sync(true);
|
match lightclient.do_sync(true) {
|
||||||
|
Ok(_) => {
|
||||||
format!("{}", lightclient.do_list_transactions().pretty(2))
|
format!("{}", lightclient.do_list_transactions().pretty(2))
|
||||||
|
},
|
||||||
|
Err(e) => e
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -633,9 +672,12 @@ impl Command for NotesCommand {
|
|||||||
false
|
false
|
||||||
};
|
};
|
||||||
|
|
||||||
lightclient.do_sync(true);
|
match lightclient.do_sync(true) {
|
||||||
|
Ok(_) => {
|
||||||
format!("{}", lightclient.do_list_notes(all_notes).pretty(2))
|
format!("{}", lightclient.do_list_notes(all_notes).pretty(2))
|
||||||
|
},
|
||||||
|
Err(e) => e
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -690,6 +732,7 @@ pub fn get_commands() -> Box<HashMap<String, Box<dyn Command>>> {
|
|||||||
let mut map: HashMap<String, Box<dyn Command>> = HashMap::new();
|
let mut map: HashMap<String, Box<dyn Command>> = HashMap::new();
|
||||||
|
|
||||||
map.insert("sync".to_string(), Box::new(SyncCommand{}));
|
map.insert("sync".to_string(), Box::new(SyncCommand{}));
|
||||||
|
map.insert("syncstatus".to_string(), Box::new(SyncStatusCommand{}));
|
||||||
map.insert("rescan".to_string(), Box::new(RescanCommand{}));
|
map.insert("rescan".to_string(), Box::new(RescanCommand{}));
|
||||||
map.insert("help".to_string(), Box::new(HelpCommand{}));
|
map.insert("help".to_string(), Box::new(HelpCommand{}));
|
||||||
map.insert("balance".to_string(), Box::new(BalanceCommand{}));
|
map.insert("balance".to_string(), Box::new(BalanceCommand{}));
|
||||||
@@ -714,8 +757,37 @@ pub fn get_commands() -> Box<HashMap<String, Box<dyn Command>>> {
|
|||||||
}
|
}
|
||||||
|
|
||||||
pub fn do_user_command(cmd: &str, args: &Vec<&str>, lightclient: &LightClient) -> String {
|
pub fn do_user_command(cmd: &str, args: &Vec<&str>, lightclient: &LightClient) -> String {
|
||||||
match get_commands().get(cmd) {
|
match get_commands().get(&cmd.to_ascii_lowercase()) {
|
||||||
Some(cmd) => cmd.exec(args, lightclient),
|
Some(cmd) => cmd.exec(args, lightclient),
|
||||||
None => format!("Unknown command : {}. Type 'help' for a list of commands", cmd)
|
None => format!("Unknown command : {}. Type 'help' for a list of commands", cmd)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
pub mod tests {
|
||||||
|
use lazy_static::lazy_static;
|
||||||
|
use super::do_user_command;
|
||||||
|
use crate::lightclient::{LightClient};
|
||||||
|
|
||||||
|
lazy_static!{
|
||||||
|
static ref TEST_SEED: String = "youth strong sweet gorilla hammer unhappy congress stamp left stereo riot salute road tag clean toilet artefact fork certain leopard entire civil degree wonder".to_string();
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
pub fn test_command_caseinsensitive() {
|
||||||
|
let lc = LightClient::unconnected(TEST_SEED.to_string(), None).unwrap();
|
||||||
|
|
||||||
|
assert_eq!(do_user_command("addresses", &vec![], &lc),
|
||||||
|
do_user_command("AddReSSeS", &vec![], &lc));
|
||||||
|
assert_eq!(do_user_command("addresses", &vec![], &lc),
|
||||||
|
do_user_command("Addresses", &vec![], &lc));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
pub fn test_nosync_commands() {
|
||||||
|
// The following commands should run
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -1,7 +1,6 @@
|
|||||||
#[macro_use]
|
#[macro_use]
|
||||||
extern crate rust_embed;
|
extern crate rust_embed;
|
||||||
|
|
||||||
pub mod startup_helpers;
|
|
||||||
pub mod lightclient;
|
pub mod lightclient;
|
||||||
pub mod grpcconnector;
|
pub mod grpcconnector;
|
||||||
pub mod lightwallet;
|
pub mod lightwallet;
|
||||||
|
|||||||
@@ -28,9 +28,25 @@ use crate::ANCHOR_OFFSET;
|
|||||||
mod checkpoints;
|
mod checkpoints;
|
||||||
|
|
||||||
pub const DEFAULT_SERVER: &str = "https://";
|
pub const DEFAULT_SERVER: &str = "https://";
|
||||||
pub const WALLET_NAME: &str = "silentdragonlite-cli-wallet.dat";
|
pub const WALLET_NAME: &str = "silentdragonlite-wallet.dat";
|
||||||
pub const LOGFILE_NAME: &str = "silentdragonlite-cli-wallet.debug.log";
|
pub const LOGFILE_NAME: &str = "silentdragonlite-wallet.debug.log";
|
||||||
|
|
||||||
|
#[derive(Clone, Debug)]
|
||||||
|
pub struct WalletStatus {
|
||||||
|
pub is_syncing: bool,
|
||||||
|
pub total_blocks: u64,
|
||||||
|
pub synced_blocks: u64,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl WalletStatus {
|
||||||
|
pub fn new() -> Self {
|
||||||
|
WalletStatus {
|
||||||
|
is_syncing: false,
|
||||||
|
total_blocks: 0,
|
||||||
|
synced_blocks: 0
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
#[derive(Clone, Debug)]
|
#[derive(Clone, Debug)]
|
||||||
pub struct LightClientConfig {
|
pub struct LightClientConfig {
|
||||||
@@ -84,10 +100,10 @@ impl LightClientConfig {
|
|||||||
} else {
|
} else {
|
||||||
if cfg!(target_os="macos") || cfg!(target_os="windows") {
|
if cfg!(target_os="macos") || cfg!(target_os="windows") {
|
||||||
zcash_data_location = dirs::data_dir().expect("Couldn't determine app data directory!");
|
zcash_data_location = dirs::data_dir().expect("Couldn't determine app data directory!");
|
||||||
zcash_data_location.push("HUSH3");
|
zcash_data_location.push("silentdragonlite");
|
||||||
} else {
|
} else {
|
||||||
zcash_data_location = dirs::home_dir().expect("Couldn't determine home directory!");
|
zcash_data_location = dirs::home_dir().expect("Couldn't determine home directory!");
|
||||||
zcash_data_location.push(".komodo/HUSH3/");
|
zcash_data_location.push("silentdragonlite/");
|
||||||
};
|
};
|
||||||
|
|
||||||
match &self.chain_name[..] {
|
match &self.chain_name[..] {
|
||||||
@@ -203,6 +219,7 @@ pub struct LightClient {
|
|||||||
pub sapling_spend : Vec<u8>,
|
pub sapling_spend : Vec<u8>,
|
||||||
|
|
||||||
sync_lock : Mutex<()>,
|
sync_lock : Mutex<()>,
|
||||||
|
sync_status : Arc<RwLock<WalletStatus>>, // The current syncing status of the Wallet.
|
||||||
}
|
}
|
||||||
|
|
||||||
impl LightClient {
|
impl LightClient {
|
||||||
@@ -227,7 +244,7 @@ impl LightClient {
|
|||||||
|
|
||||||
/// Method to create a test-only version of the LightClient
|
/// Method to create a test-only version of the LightClient
|
||||||
#[allow(dead_code)]
|
#[allow(dead_code)]
|
||||||
fn unconnected(seed_phrase: String, dir: Option<String>) -> io::Result<Self> {
|
pub fn unconnected(seed_phrase: String, dir: Option<String>) -> io::Result<Self> {
|
||||||
let config = LightClientConfig::create_unconnected("test".to_string(), dir);
|
let config = LightClientConfig::create_unconnected("test".to_string(), dir);
|
||||||
let mut l = LightClient {
|
let mut l = LightClient {
|
||||||
wallet : Arc::new(RwLock::new(LightWallet::new(Some(seed_phrase), &config, 0)?)),
|
wallet : Arc::new(RwLock::new(LightWallet::new(Some(seed_phrase), &config, 0)?)),
|
||||||
@@ -235,6 +252,7 @@ impl LightClient {
|
|||||||
sapling_output : vec![],
|
sapling_output : vec![],
|
||||||
sapling_spend : vec![],
|
sapling_spend : vec![],
|
||||||
sync_lock : Mutex::new(()),
|
sync_lock : Mutex::new(()),
|
||||||
|
sync_status : Arc::new(RwLock::new(WalletStatus::new())),
|
||||||
};
|
};
|
||||||
|
|
||||||
l.set_wallet_initial_state(0);
|
l.set_wallet_initial_state(0);
|
||||||
@@ -260,6 +278,7 @@ impl LightClient {
|
|||||||
sapling_output : vec![],
|
sapling_output : vec![],
|
||||||
sapling_spend : vec![],
|
sapling_spend : vec![],
|
||||||
sync_lock : Mutex::new(()),
|
sync_lock : Mutex::new(()),
|
||||||
|
sync_status : Arc::new(RwLock::new(WalletStatus::new())),
|
||||||
};
|
};
|
||||||
|
|
||||||
l.set_wallet_initial_state(latest_block);
|
l.set_wallet_initial_state(latest_block);
|
||||||
@@ -283,6 +302,7 @@ impl LightClient {
|
|||||||
sapling_output : vec![],
|
sapling_output : vec![],
|
||||||
sapling_spend : vec![],
|
sapling_spend : vec![],
|
||||||
sync_lock : Mutex::new(()),
|
sync_lock : Mutex::new(()),
|
||||||
|
sync_status : Arc::new(RwLock::new(WalletStatus::new())),
|
||||||
};
|
};
|
||||||
|
|
||||||
println!("Setting birthday to {}", birthday);
|
println!("Setting birthday to {}", birthday);
|
||||||
@@ -310,6 +330,7 @@ impl LightClient {
|
|||||||
sapling_output : vec![],
|
sapling_output : vec![],
|
||||||
sapling_spend : vec![],
|
sapling_spend : vec![],
|
||||||
sync_lock : Mutex::new(()),
|
sync_lock : Mutex::new(()),
|
||||||
|
sync_status : Arc::new(RwLock::new(WalletStatus::new())),
|
||||||
};
|
};
|
||||||
|
|
||||||
lc.read_sapling_params();
|
lc.read_sapling_params();
|
||||||
@@ -720,7 +741,7 @@ impl LightClient {
|
|||||||
Ok(array![new_address])
|
Ok(array![new_address])
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn do_rescan(&self) -> String {
|
pub fn do_rescan(&self) -> Result<JsonValue, String> {
|
||||||
info!("Rescan starting");
|
info!("Rescan starting");
|
||||||
// First, clear the state from the wallet
|
// First, clear the state from the wallet
|
||||||
self.wallet.read().unwrap().clear_blocks();
|
self.wallet.read().unwrap().clear_blocks();
|
||||||
@@ -735,7 +756,12 @@ impl LightClient {
|
|||||||
response
|
response
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn do_sync(&self, print_updates: bool) -> String {
|
/// Return the syncing status of the wallet
|
||||||
|
pub fn do_scan_status(&self) -> WalletStatus {
|
||||||
|
self.sync_status.read().unwrap().clone()
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn do_sync(&self, print_updates: bool) -> Result<JsonValue, String> {
|
||||||
// We can only do one sync at a time because we sync blocks in serial order
|
// We can only do one sync at a time because we sync blocks in serial order
|
||||||
// If we allow multiple syncs, they'll all get jumbled up.
|
// If we allow multiple syncs, they'll all get jumbled up.
|
||||||
let _lock = self.sync_lock.lock().unwrap();
|
let _lock = self.sync_lock.lock().unwrap();
|
||||||
@@ -750,15 +776,17 @@ impl LightClient {
|
|||||||
// This will hold the latest block fetched from the RPC
|
// This will hold the latest block fetched from the RPC
|
||||||
let latest_block_height = Arc::new(AtomicU64::new(0));
|
let latest_block_height = Arc::new(AtomicU64::new(0));
|
||||||
let lbh = latest_block_height.clone();
|
let lbh = latest_block_height.clone();
|
||||||
fetch_latest_block(&self.get_server_uri(), self.config.no_cert_verification, move |block: BlockId| {
|
fetch_latest_block(&self.get_server_uri(), self.config.no_cert_verification,
|
||||||
|
move |block: BlockId| {
|
||||||
lbh.store(block.height, Ordering::SeqCst);
|
lbh.store(block.height, Ordering::SeqCst);
|
||||||
});
|
});
|
||||||
let latest_block = latest_block_height.load(Ordering::SeqCst);
|
let latest_block = latest_block_height.load(Ordering::SeqCst);
|
||||||
|
|
||||||
|
|
||||||
if latest_block < last_scanned_height {
|
if latest_block < last_scanned_height {
|
||||||
let w = format!("Server's latest block({}) is behind ours({})", latest_block, last_scanned_height);
|
let w = format!("Server's latest block({}) is behind ours({})", latest_block, last_scanned_height);
|
||||||
warn!("{}", w);
|
warn!("{}", w);
|
||||||
return w;
|
return Err(w);
|
||||||
}
|
}
|
||||||
|
|
||||||
info!("Latest block is {}", latest_block);
|
info!("Latest block is {}", latest_block);
|
||||||
@@ -769,7 +797,14 @@ impl LightClient {
|
|||||||
// If there's nothing to scan, just return
|
// If there's nothing to scan, just return
|
||||||
if last_scanned_height == latest_block {
|
if last_scanned_height == latest_block {
|
||||||
info!("Nothing to sync, returning");
|
info!("Nothing to sync, returning");
|
||||||
return "".to_string();
|
return Ok(object!{ "result" => "success" })
|
||||||
|
}
|
||||||
|
|
||||||
|
{
|
||||||
|
let mut status = self.sync_status.write().unwrap();
|
||||||
|
status.is_syncing = true;
|
||||||
|
status.synced_blocks = last_scanned_height;
|
||||||
|
status.total_blocks = latest_block;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Count how many bytes we've downloaded
|
// Count how many bytes we've downloaded
|
||||||
@@ -795,11 +830,18 @@ impl LightClient {
|
|||||||
info!("Start height is {}", start_height);
|
info!("Start height is {}", start_height);
|
||||||
|
|
||||||
// Show updates only if we're syncing a lot of blocks
|
// Show updates only if we're syncing a lot of blocks
|
||||||
if print_updates && end_height - start_height > 100 {
|
if print_updates && (latest_block - start_height) > 100 {
|
||||||
print!("Syncing {}/{}\r", start_height, latest_block);
|
print!("Syncing {}/{}\r", start_height, latest_block);
|
||||||
io::stdout().flush().ok().expect("Could not flush stdout");
|
io::stdout().flush().ok().expect("Could not flush stdout");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
{
|
||||||
|
let mut status = self.sync_status.write().unwrap();
|
||||||
|
status.is_syncing = true;
|
||||||
|
status.synced_blocks = start_height;
|
||||||
|
status.total_blocks = latest_block;
|
||||||
|
}
|
||||||
|
|
||||||
// Fetch compact blocks
|
// Fetch compact blocks
|
||||||
info!("Fetching blocks {}-{}", start_height, end_height);
|
info!("Fetching blocks {}-{}", start_height, end_height);
|
||||||
|
|
||||||
@@ -851,7 +893,7 @@ impl LightClient {
|
|||||||
// Make sure we're not re-orging too much!
|
// Make sure we're not re-orging too much!
|
||||||
if total_reorg > (crate::lightwallet::MAX_REORG - 1) as u64 {
|
if total_reorg > (crate::lightwallet::MAX_REORG - 1) as u64 {
|
||||||
error!("Reorg has now exceeded {} blocks!", crate::lightwallet::MAX_REORG);
|
error!("Reorg has now exceeded {} blocks!", crate::lightwallet::MAX_REORG);
|
||||||
return format!("Reorg has exceeded {} blocks. Aborting.", crate::lightwallet::MAX_REORG);
|
return Err(format!("Reorg has exceeded {} blocks. Aborting.", crate::lightwallet::MAX_REORG));
|
||||||
}
|
}
|
||||||
|
|
||||||
if invalid_height > 0 {
|
if invalid_height > 0 {
|
||||||
@@ -905,11 +947,14 @@ impl LightClient {
|
|||||||
println!(""); // New line to finish up the updates
|
println!(""); // New line to finish up the updates
|
||||||
}
|
}
|
||||||
|
|
||||||
let mut responses = vec![];
|
|
||||||
|
|
||||||
info!("Synced to {}, Downloaded {} kB", latest_block, bytes_downloaded.load(Ordering::SeqCst) / 1024);
|
info!("Synced to {}, Downloaded {} kB", latest_block, bytes_downloaded.load(Ordering::SeqCst) / 1024);
|
||||||
responses.push(format!("Synced to {}, Downloaded {} kB", latest_block, bytes_downloaded.load(Ordering::SeqCst) / 1024));
|
{
|
||||||
|
let mut status = self.sync_status.write().unwrap();
|
||||||
|
status.is_syncing = false;
|
||||||
|
status.synced_blocks = latest_block;
|
||||||
|
status.total_blocks = latest_block;
|
||||||
|
}
|
||||||
|
|
||||||
// Get the Raw transaction for all the wallet transactions
|
// Get the Raw transaction for all the wallet transactions
|
||||||
|
|
||||||
// We need to first copy over the Txids from the wallet struct, because
|
// We need to first copy over the Txids from the wallet struct, because
|
||||||
@@ -940,7 +985,11 @@ impl LightClient {
|
|||||||
});
|
});
|
||||||
};
|
};
|
||||||
|
|
||||||
responses.join("\n")
|
Ok(object!{
|
||||||
|
"result" => "success",
|
||||||
|
"latest_block" => latest_block,
|
||||||
|
"downloaded_bytes" => bytes_downloaded.load(Ordering::SeqCst)
|
||||||
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn do_send(&self, addrs: Vec<(&str, u64, Option<String>)>) -> Result<String, String> {
|
pub fn do_send(&self, addrs: Vec<(&str, u64, Option<String>)>) -> Result<String, String> {
|
||||||
|
|||||||
@@ -748,9 +748,9 @@ impl LightWallet {
|
|||||||
}
|
}
|
||||||
|
|
||||||
pub fn zbalance(&self, addr: Option<String>) -> u64 {
|
pub fn zbalance(&self, addr: Option<String>) -> u64 {
|
||||||
self.txs.read().unwrap()
|
self.txs.read().unwrap()
|
||||||
.values()
|
.values()
|
||||||
.map(|tx| {
|
.map (|tx| {
|
||||||
tx.notes.iter()
|
tx.notes.iter()
|
||||||
.filter(|nd| { // TODO, this whole section is shared with verified_balance. Refactor it.
|
.filter(|nd| { // TODO, this whole section is shared with verified_balance. Refactor it.
|
||||||
match addr.clone() {
|
match addr.clone() {
|
||||||
@@ -763,9 +763,11 @@ impl LightWallet {
|
|||||||
}
|
}
|
||||||
})
|
})
|
||||||
.map(|nd| if nd.spent.is_none() { nd.note.value } else { 0 })
|
.map(|nd| if nd.spent.is_none() { nd.note.value } else { 0 })
|
||||||
.sum::<u64>()
|
.sum::<u64>()
|
||||||
|
|
||||||
})
|
})
|
||||||
.sum::<u64>()
|
.sum::<u64>() as u64
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Get all (unspent) utxos. Unconfirmed spent utxos are included
|
// Get all (unspent) utxos. Unconfirmed spent utxos are included
|
||||||
@@ -789,7 +791,7 @@ impl LightWallet {
|
|||||||
}
|
}
|
||||||
})
|
})
|
||||||
.map(|utxo| utxo.value )
|
.map(|utxo| utxo.value )
|
||||||
.sum::<u64>()
|
.sum::<u64>() as u64
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn verified_zbalance(&self, addr: Option<String>) -> u64 {
|
pub fn verified_zbalance(&self, addr: Option<String>) -> u64 {
|
||||||
@@ -822,7 +824,7 @@ impl LightWallet {
|
|||||||
0
|
0
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
.sum::<u64>()
|
.sum::<u64>() as u64
|
||||||
}
|
}
|
||||||
|
|
||||||
fn add_toutput_to_wtx(&self, height: i32, timestamp: u64, txid: &TxId, vout: &TxOut, n: u64) {
|
fn add_toutput_to_wtx(&self, height: i32, timestamp: u64, txid: &TxId, vout: &TxOut, n: u64) {
|
||||||
@@ -1341,7 +1343,7 @@ impl LightWallet {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
let total_value = tos.iter().map(|to| to.1).sum::<u64>();
|
let total_value = tos.iter().map(|to| to.1).sum::<u64>() as u64;
|
||||||
println!(
|
println!(
|
||||||
"0: Creating transaction sending {} puposhis to {} addresses",
|
"0: Creating transaction sending {} puposhis to {} addresses",
|
||||||
total_value, tos.len()
|
total_value, tos.len()
|
||||||
|
|||||||
@@ -9,11 +9,11 @@ pub fn report_permission_error() {
|
|||||||
eprintln!("HOME: {}", home);
|
eprintln!("HOME: {}", home);
|
||||||
eprintln!("Executable: {}", current_executable.display());
|
eprintln!("Executable: {}", current_executable.display());
|
||||||
if home == "/" {
|
if home == "/" {
|
||||||
eprintln!("User {} must have permission to write to '{}.komodo/HUSH3/' .",
|
eprintln!("User {} must have permission to write to '{}silentdragonlite/' .",
|
||||||
user,
|
user,
|
||||||
home);
|
home);
|
||||||
} else {
|
} else {
|
||||||
eprintln!("User {} must have permission to write to '{}/.komodo/HUSH3/ .",
|
eprintln!("User {} must have permission to write to '{}/silentdragonlite/ .",
|
||||||
user,
|
user,
|
||||||
home);
|
home);
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -9,11 +9,11 @@ pub fn report_permission_error() {
|
|||||||
eprintln!("HOME: {}", home);
|
eprintln!("HOME: {}", home);
|
||||||
eprintln!("Executable: {}", current_executable.display());
|
eprintln!("Executable: {}", current_executable.display());
|
||||||
if home == "/" {
|
if home == "/" {
|
||||||
eprintln!("User {} must have permission to write to '{}.komodo/HUSH3/' .",
|
eprintln!("User {} must have permission to write to '{}silentdragonlite/' .",
|
||||||
user,
|
user,
|
||||||
home);
|
home);
|
||||||
} else {
|
} else {
|
||||||
eprintln!("User {} must have permission to write to '{}/.komodo/HUSH3/ .",
|
eprintln!("User {} must have permission to write to '{}/silentdragonlite/ .",
|
||||||
user,
|
user,
|
||||||
home);
|
home);
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user