// Copyright 2015, 2016 Ethcore (UK) Ltd. // This file is part of Parity. // Parity is free software: you can redistribute it and/or modify // it under the terms of the GNU General Public License as published by // the Free Software Foundation, either version 3 of the License, or // (at your option) any later version. // Parity is distributed in the hope that it will be useful, // but WITHOUT ANY WARRANTY; without even the implied warranty of // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the // GNU General Public License for more details. // You should have received a copy of the GNU General Public License // along with Parity. If not, see . use std::{io, env}; use std::io::{Write, Read, BufReader, BufRead}; use std::time::Duration; use std::path::Path; use std::fs::File; use util::{clean_0x, U256, Uint, Address, path, is_valid_node_url, H256, CompactionProfile}; use util::journaldb::Algorithm; use ethcore::client::{Mode, BlockID, Switch, VMType, DatabaseCompactionProfile, ClientConfig}; use ethcore::miner::PendingSet; use cache::CacheConfig; use dir::Directories; use params::Pruning; use upgrade::upgrade; use migration::migrate; pub fn to_duration(s: &str) -> Result { to_seconds(s).map(Duration::from_secs) } fn to_seconds(s: &str) -> Result { let bad = |_| { format!("{}: Invalid duration given. See parity --help for more information.", s) }; match s { "twice-daily" => Ok(12 * 60 * 60), "half-hourly" => Ok(30 * 60), "1second" | "1 second" | "second" => Ok(1), "1minute" | "1 minute" | "minute" => Ok(60), "hourly" | "1hour" | "1 hour" | "hour" => Ok(60 * 60), "daily" | "1day" | "1 day" | "day" => Ok(24 * 60 * 60), x if x.ends_with("seconds") => x[0..x.len() - 7].parse().map_err(bad), x if x.ends_with("minutes") => x[0..x.len() -7].parse::().map_err(bad).map(|x| x * 60), x if x.ends_with("hours") => x[0..x.len() - 5].parse::().map_err(bad).map(|x| x * 60 * 60), x if x.ends_with("days") => x[0..x.len() - 4].parse::().map_err(bad).map(|x| x * 24 * 60 * 60), x => x.parse().map_err(bad), } } pub fn to_mode(s: &str, timeout: u64, alarm: u64) -> Result { match s { "active" => Ok(Mode::Active), "passive" => Ok(Mode::Passive(Duration::from_secs(timeout), Duration::from_secs(alarm))), "dark" => Ok(Mode::Dark(Duration::from_secs(timeout))), _ => Err(format!("{}: Invalid address for --mode. Must be one of active, passive or dark.", s)), } } pub fn to_block_id(s: &str) -> Result { if s == "latest" { Ok(BlockID::Latest) } else if let Ok(num) = s.parse() { Ok(BlockID::Number(num)) } else if let Ok(hash) = s.parse() { Ok(BlockID::Hash(hash)) } else { Err("Invalid block.".into()) } } pub fn to_u256(s: &str) -> Result { if let Ok(decimal) = U256::from_dec_str(s) { Ok(decimal) } else if let Ok(hex) = clean_0x(s).parse() { Ok(hex) } else { Err(format!("Invalid numeric value: {}", s)) } } pub fn to_pending_set(s: &str) -> Result { match s { "cheap" => Ok(PendingSet::AlwaysQueue), "strict" => Ok(PendingSet::AlwaysSealing), "lenient" => Ok(PendingSet::SealingOrElseQueue), other => Err(format!("Invalid pending set value: {:?}", other)), } } pub fn to_address(s: Option) -> Result { match s { Some(ref a) => clean_0x(a).parse().map_err(|_| format!("Invalid address: {:?}", a)), None => Ok(Address::default()) } } pub fn to_addresses(s: &Option) -> Result, String> { match *s { Some(ref adds) if adds.is_empty() => adds.split(',') .map(|a| clean_0x(a).parse().map_err(|_| format!("Invalid address: {:?}", a))) .collect(), _ => Ok(Vec::new()), } } /// Tries to parse string as a price. pub fn to_price(s: &str) -> Result { s.parse::().map_err(|_| format!("Invalid transaciton price 's' given. Must be a decimal number.")) } /// Replaces `$HOME` str with home directory path. pub fn replace_home(arg: &str) -> String { // the $HOME directory on mac os should be `~/Library` or `~/Library/Application Support` let r = arg.replace("$HOME", env::home_dir().unwrap().to_str().unwrap()); r.replace("/", &::std::path::MAIN_SEPARATOR.to_string() ) } /// Flush output buffer. pub fn flush_stdout() { io::stdout().flush().expect("stdout is flushable; qed"); } /// Returns default geth ipc path. pub fn geth_ipc_path(testnet: bool) -> String { // Windows path should not be hardcoded here. // Instead it should be a part of path::ethereum if cfg!(windows) { return r"\\.\pipe\geth.ipc".to_owned(); } if testnet { path::ethereum::with_testnet("geth.ipc").to_str().unwrap().to_owned() } else { path::ethereum::with_default("geth.ipc").to_str().unwrap().to_owned() } } /// Formats and returns parity ipc path. pub fn parity_ipc_path(s: &str) -> String { // Windows path should not be hardcoded here. if cfg!(windows) { return r"\\.\pipe\parity.jsonrpc".to_owned(); } replace_home(s) } /// Validates and formats bootnodes option. pub fn to_bootnodes(bootnodes: &Option) -> Result, String> { match *bootnodes { Some(ref x) if !x.is_empty() => x.split(',').map(|s| { if is_valid_node_url(s) { Ok(s.to_owned()) } else { Err(format!("Invalid node address format given for a boot node: {}", s)) } }).collect(), Some(_) => Ok(vec![]), None => Ok(vec![]) } } #[cfg(test)] pub fn default_network_config() -> ::util::NetworkConfiguration { use util::{NetworkConfiguration, NonReservedPeerMode}; NetworkConfiguration { config_path: Some(replace_home("$HOME/.parity/network")), listen_address: Some("0.0.0.0:30303".parse().unwrap()), public_address: None, udp_port: None, nat_enabled: true, discovery_enabled: true, boot_nodes: Vec::new(), use_secret: None, ideal_peers: 25, reserved_nodes: Vec::new(), non_reserved_mode: NonReservedPeerMode::Accept, } } #[cfg_attr(feature = "dev", allow(too_many_arguments))] pub fn to_client_config( cache_config: &CacheConfig, dirs: &Directories, genesis_hash: H256, mode: Mode, tracing: Switch, pruning: Pruning, compaction: DatabaseCompactionProfile, vm_type: VMType, name: String, fork_name: Option<&String>, ) -> ClientConfig { let mut client_config = ClientConfig::default(); let mb = 1024 * 1024; // in bytes client_config.blockchain.max_cache_size = cache_config.blockchain() as usize * mb; // in bytes client_config.blockchain.pref_cache_size = cache_config.blockchain() as usize * 3 / 4 * mb; // db blockchain cache size, in megabytes client_config.blockchain.db_cache_size = Some(cache_config.db_blockchain_cache_size() as usize); // db state cache size, in megabytes client_config.db_cache_size = Some(cache_config.db_state_cache_size() as usize); // db queue cache size, in bytes client_config.queue.max_mem_use = cache_config.queue() as usize * mb; client_config.mode = mode; client_config.tracing.enabled = tracing; client_config.pruning = pruning.to_algorithm(dirs, genesis_hash, fork_name); client_config.db_compaction = compaction; client_config.vm_type = vm_type; client_config.name = name; client_config } pub fn execute_upgrades( dirs: &Directories, genesis_hash: H256, fork_name: Option<&String>, pruning: Algorithm, compaction_profile: CompactionProfile ) -> Result<(), String> { match upgrade(Some(&dirs.db)) { Ok(upgrades_applied) if upgrades_applied > 0 => { debug!("Executed {} upgrade scripts - ok", upgrades_applied); }, Err(e) => { return Err(format!("Error upgrading parity data: {:?}", e)); }, _ => {}, } let client_path = dirs.db_version_path(genesis_hash, fork_name, pruning); migrate(&client_path, pruning, compaction_profile).map_err(|e| format!("{}", e)) } /// Prompts user asking for password. pub fn password_prompt() -> Result { use rpassword::read_password; println!("Please note that password is NOT RECOVERABLE."); print!("Type password: "); flush_stdout(); let password = read_password().unwrap(); print!("Repeat password: "); flush_stdout(); let password_repeat = read_password().unwrap(); if password != password_repeat { return Err("Passwords do not match!".into()); } Ok(password) } /// Read a password from password file. pub fn password_from_file

(path: P) -> Result where P: AsRef { let mut file = try!(File::open(path).map_err(|_| "Unable to open password file.")); let mut file_content = String::new(); try!(file.read_to_string(&mut file_content).map_err(|_| "Unable to read password file.")); // remove eof Ok((&file_content[..file_content.len() - 1]).to_owned()) } /// Reads passwords from files. Treats each line as a separate password. pub fn passwords_from_files(files: Vec) -> Result, String> { let passwords = files.iter().map(|filename| { let file = try!(File::open(filename).map_err(|_| format!("{} Unable to read password file. Ensure it exists and permissions are correct.", filename))); let reader = BufReader::new(&file); let lines = reader.lines() .map(|l| l.unwrap()) .collect::>(); Ok(lines) }).collect::>, String>>(); Ok(try!(passwords).into_iter().flat_map(|x| x).collect()) } #[cfg(test)] mod tests { use std::time::Duration; use util::{U256}; use ethcore::client::{Mode, BlockID}; use ethcore::miner::PendingSet; use super::{to_duration, to_mode, to_block_id, to_u256, to_pending_set, to_address, to_price, geth_ipc_path, to_bootnodes}; #[test] fn test_to_duration() { assert_eq!(to_duration("twice-daily").unwrap(), Duration::from_secs(12 * 60 * 60)); assert_eq!(to_duration("half-hourly").unwrap(), Duration::from_secs(30 * 60)); assert_eq!(to_duration("1second").unwrap(), Duration::from_secs(1)); assert_eq!(to_duration("2seconds").unwrap(), Duration::from_secs(2)); assert_eq!(to_duration("15seconds").unwrap(), Duration::from_secs(15)); assert_eq!(to_duration("1minute").unwrap(), Duration::from_secs(1 * 60)); assert_eq!(to_duration("2minutes").unwrap(), Duration::from_secs(2 * 60)); assert_eq!(to_duration("15minutes").unwrap(), Duration::from_secs(15 * 60)); assert_eq!(to_duration("hourly").unwrap(), Duration::from_secs(60 * 60)); assert_eq!(to_duration("daily").unwrap(), Duration::from_secs(24 * 60 * 60)); assert_eq!(to_duration("1hour").unwrap(), Duration::from_secs(1 * 60 * 60)); assert_eq!(to_duration("2hours").unwrap(), Duration::from_secs(2 * 60 * 60)); assert_eq!(to_duration("15hours").unwrap(), Duration::from_secs(15 * 60 * 60)); assert_eq!(to_duration("1day").unwrap(), Duration::from_secs(1 * 24 * 60 * 60)); assert_eq!(to_duration("2days").unwrap(), Duration::from_secs(2 * 24 *60 * 60)); assert_eq!(to_duration("15days").unwrap(), Duration::from_secs(15 * 24 * 60 * 60)); } #[test] fn test_to_mode() { assert_eq!(to_mode("active", 0, 0).unwrap(), Mode::Active); assert_eq!(to_mode("passive", 10, 20).unwrap(), Mode::Passive(Duration::from_secs(10), Duration::from_secs(20))); assert_eq!(to_mode("dark", 20, 30).unwrap(), Mode::Dark(Duration::from_secs(20))); assert!(to_mode("other", 20, 30).is_err()); } #[test] fn test_to_block_id() { assert_eq!(to_block_id("latest").unwrap(), BlockID::Latest); assert_eq!(to_block_id("0").unwrap(), BlockID::Number(0)); assert_eq!(to_block_id("2").unwrap(), BlockID::Number(2)); assert_eq!(to_block_id("15").unwrap(), BlockID::Number(15)); assert_eq!( to_block_id("9fc84d84f6a785dc1bd5abacfcf9cbdd3b6afb80c0f799bfb2fd42c44a0c224e").unwrap(), BlockID::Hash("9fc84d84f6a785dc1bd5abacfcf9cbdd3b6afb80c0f799bfb2fd42c44a0c224e".parse().unwrap()) ); } #[test] fn test_to_u256() { assert_eq!(to_u256("0").unwrap(), U256::from(0)); assert_eq!(to_u256("11").unwrap(), U256::from(11)); assert_eq!(to_u256("0x11").unwrap(), U256::from(17)); assert!(to_u256("u").is_err()) } #[test] fn test_pending_set() { assert_eq!(to_pending_set("cheap").unwrap(), PendingSet::AlwaysQueue); assert_eq!(to_pending_set("strict").unwrap(), PendingSet::AlwaysSealing); assert_eq!(to_pending_set("lenient").unwrap(), PendingSet::SealingOrElseQueue); assert!(to_pending_set("othe").is_err()); } #[test] fn test_to_address() { assert_eq!( to_address(Some("0xD9A111feda3f362f55Ef1744347CDC8Dd9964a41".into())).unwrap(), "D9A111feda3f362f55Ef1744347CDC8Dd9964a41".parse().unwrap() ); assert_eq!( to_address(Some("D9A111feda3f362f55Ef1744347CDC8Dd9964a41".into())).unwrap(), "D9A111feda3f362f55Ef1744347CDC8Dd9964a41".parse().unwrap() ); assert_eq!(to_address(None).unwrap(), Default::default()); } #[test] #[cfg_attr(feature = "dev", allow(float_cmp))] fn test_to_price() { assert_eq!(to_price("1").unwrap(), 1.0); assert_eq!(to_price("2.3").unwrap(), 2.3); assert_eq!(to_price("2.33").unwrap(), 2.33); } #[test] #[cfg(windows)] fn test_geth_ipc_path() { assert_eq!(geth_ipc_path(true), r"\\.\pipe\geth.ipc".to_owned()); assert_eq!(geth_ipc_path(false), r"\\.\pipe\geth.ipc".to_owned()); } #[test] #[cfg(not(windows))] fn test_geth_ipc_path() { use util::path; assert_eq!(geth_ipc_path(true), path::ethereum::with_testnet("geth.ipc").to_str().unwrap().to_owned()); assert_eq!(geth_ipc_path(false), path::ethereum::with_default("geth.ipc").to_str().unwrap().to_owned()); } #[test] fn test_to_bootnodes() { let one_bootnode = "enode://e731347db0521f3476e6bbbb83375dcd7133a1601425ebd15fd10f3835fd4c304fba6282087ca5a0deeafadf0aa0d4fd56c3323331901c1f38bd181c283e3e35@128.199.55.137:30303"; let two_bootnodes = "enode://e731347db0521f3476e6bbbb83375dcd7133a1601425ebd15fd10f3835fd4c304fba6282087ca5a0deeafadf0aa0d4fd56c3323331901c1f38bd181c283e3e35@128.199.55.137:30303,enode://e731347db0521f3476e6bbbb83375dcd7133a1601425ebd15fd10f3835fd4c304fba6282087ca5a0deeafadf0aa0d4fd56c3323331901c1f38bd181c283e3e35@128.199.55.137:30303"; assert_eq!(to_bootnodes(&Some("".into())), Ok(vec![])); assert_eq!(to_bootnodes(&None), Ok(vec![])); assert_eq!(to_bootnodes(&Some(one_bootnode.into())), Ok(vec![one_bootnode.into()])); assert_eq!(to_bootnodes(&Some(two_bootnodes.into())), Ok(vec![one_bootnode.into(), one_bootnode.into()])); } }