312 lines
10 KiB
Rust
312 lines
10 KiB
Rust
// Copyright 2015-2017 Parity Technologies (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 <http://www.gnu.org/licenses/>.
|
|
|
|
//! Ethcore client application.
|
|
|
|
#![warn(missing_docs)]
|
|
#![cfg_attr(feature="dev", feature(plugin))]
|
|
#![cfg_attr(feature="dev", plugin(clippy))]
|
|
#![cfg_attr(feature="dev", allow(useless_format))]
|
|
#![cfg_attr(feature="dev", allow(match_bool))]
|
|
|
|
extern crate ansi_term;
|
|
extern crate app_dirs;
|
|
extern crate ctrlc;
|
|
extern crate docopt;
|
|
extern crate env_logger;
|
|
extern crate fdlimit;
|
|
extern crate hyper; // for price_info.rs
|
|
extern crate isatty;
|
|
extern crate jsonrpc_core;
|
|
extern crate num_cpus;
|
|
extern crate number_prefix;
|
|
extern crate regex;
|
|
extern crate rlp;
|
|
extern crate rpassword;
|
|
extern crate rustc_serialize;
|
|
extern crate semver;
|
|
extern crate serde;
|
|
extern crate serde_json;
|
|
extern crate time;
|
|
extern crate toml;
|
|
|
|
extern crate ethcore;
|
|
extern crate ethcore_devtools as devtools;
|
|
extern crate ethcore_io as io;
|
|
extern crate ethcore_ipc as ipc;
|
|
extern crate ethcore_ipc_hypervisor as hypervisor;
|
|
extern crate ethcore_ipc_nano as nanoipc;
|
|
extern crate ethcore_light as light;
|
|
extern crate ethcore_logger;
|
|
extern crate ethcore_rpc;
|
|
extern crate ethcore_signer;
|
|
extern crate ethcore_util as util;
|
|
extern crate ethsync;
|
|
extern crate parity_hash_fetch as hash_fetch;
|
|
extern crate parity_ipfs_api;
|
|
extern crate parity_reactor;
|
|
extern crate parity_updater as updater;
|
|
extern crate rpc_cli;
|
|
|
|
#[cfg(feature="stratum")]
|
|
extern crate ethcore_stratum;
|
|
#[cfg(feature = "dapps")]
|
|
extern crate ethcore_dapps;
|
|
|
|
|
|
#[macro_use]
|
|
extern crate log as rlog;
|
|
|
|
macro_rules! dependency {
|
|
($dep_ty:ident, $url:expr) => {
|
|
{
|
|
let dep = boot::dependency::<$dep_ty<_>>($url)
|
|
.unwrap_or_else(|e| panic!("Fatal: error connecting service ({:?})", e));
|
|
dep.handshake()
|
|
.unwrap_or_else(|e| panic!("Fatal: error in connected service ({:?})", e));
|
|
dep
|
|
}
|
|
}
|
|
}
|
|
|
|
mod account;
|
|
mod blockchain;
|
|
mod cache;
|
|
mod cli;
|
|
mod configuration;
|
|
mod dapps;
|
|
mod ipfs;
|
|
mod deprecated;
|
|
mod dir;
|
|
mod helpers;
|
|
mod informant;
|
|
mod migration;
|
|
mod modules;
|
|
mod params;
|
|
mod presale;
|
|
mod rpc;
|
|
mod rpc_apis;
|
|
mod run;
|
|
mod signer;
|
|
mod snapshot;
|
|
mod upgrade;
|
|
mod url;
|
|
mod user_defaults;
|
|
|
|
#[cfg(feature="ipc")]
|
|
mod boot;
|
|
#[cfg(feature="ipc")]
|
|
mod sync;
|
|
|
|
#[cfg(feature="stratum")]
|
|
mod stratum;
|
|
|
|
use std::{process, env};
|
|
use std::collections::HashMap;
|
|
use std::io::{self as stdio, BufReader, Read, Write};
|
|
use std::fs::File;
|
|
use std::path::PathBuf;
|
|
use util::sha3::sha3;
|
|
use cli::Args;
|
|
use configuration::{Cmd, Execute, Configuration};
|
|
use deprecated::find_deprecated;
|
|
use ethcore_logger::setup_log;
|
|
use dir::default_hypervisor_path;
|
|
|
|
fn print_hash_of(maybe_file: Option<String>) -> Result<String, String> {
|
|
if let Some(file) = maybe_file {
|
|
let mut f = BufReader::new(File::open(&file).map_err(|_| "Unable to open file".to_owned())?);
|
|
let hash = sha3(&mut f).map_err(|_| "Unable to read from file".to_owned())?;
|
|
Ok(hash.hex())
|
|
} else {
|
|
Err("Streaming from standard input not yet supported. Specify a file.".to_owned())
|
|
}
|
|
}
|
|
|
|
enum PostExecutionAction {
|
|
Print(String),
|
|
Restart,
|
|
Quit,
|
|
}
|
|
|
|
fn execute(command: Execute, can_restart: bool) -> Result<PostExecutionAction, String> {
|
|
let logger = setup_log(&command.logger).expect("Logger is initialized only once; qed");
|
|
|
|
match command.cmd {
|
|
Cmd::Run(run_cmd) => {
|
|
let restart = run::execute(run_cmd, can_restart, logger)?;
|
|
Ok(if restart { PostExecutionAction::Restart } else { PostExecutionAction::Quit })
|
|
},
|
|
Cmd::Version => Ok(PostExecutionAction::Print(Args::print_version())),
|
|
Cmd::Hash(maybe_file) => print_hash_of(maybe_file).map(|s| PostExecutionAction::Print(s)),
|
|
Cmd::Account(account_cmd) => account::execute(account_cmd).map(|s| PostExecutionAction::Print(s)),
|
|
Cmd::ImportPresaleWallet(presale_cmd) => presale::execute(presale_cmd).map(|s| PostExecutionAction::Print(s)),
|
|
Cmd::Blockchain(blockchain_cmd) => blockchain::execute(blockchain_cmd).map(|_| PostExecutionAction::Quit),
|
|
Cmd::SignerToken(signer_cmd) => signer::execute(signer_cmd).map(|s| PostExecutionAction::Print(s)),
|
|
Cmd::SignerSign { id, pwfile, port, authfile } => rpc_cli::signer_sign(id, pwfile, port, authfile).map(|s| PostExecutionAction::Print(s)),
|
|
Cmd::SignerList { port, authfile } => rpc_cli::signer_list(port, authfile).map(|s| PostExecutionAction::Print(s)),
|
|
Cmd::SignerReject { id, port, authfile } => rpc_cli::signer_reject(id, port, authfile).map(|s| PostExecutionAction::Print(s)),
|
|
Cmd::Snapshot(snapshot_cmd) => snapshot::execute(snapshot_cmd).map(|s| PostExecutionAction::Print(s)),
|
|
}
|
|
}
|
|
|
|
fn start(can_restart: bool) -> Result<PostExecutionAction, String> {
|
|
let args: Vec<String> = env::args().collect();
|
|
let conf = Configuration::parse(&args).unwrap_or_else(|e| e.exit());
|
|
|
|
let deprecated = find_deprecated(&conf.args);
|
|
for d in deprecated {
|
|
println!("{}", d);
|
|
}
|
|
|
|
let cmd = conf.into_command()?;
|
|
execute(cmd, can_restart)
|
|
}
|
|
|
|
#[cfg(not(feature="stratum"))]
|
|
fn stratum_main(_: &mut HashMap<String, fn()>) {}
|
|
|
|
#[cfg(feature="stratum")]
|
|
fn stratum_main(alt_mains: &mut HashMap<String, fn()>) {
|
|
alt_mains.insert("stratum".to_owned(), stratum::main);
|
|
}
|
|
|
|
#[cfg(not(feature="ipc"))]
|
|
fn sync_main(_: &mut HashMap<String, fn()>) {}
|
|
|
|
#[cfg(feature="ipc")]
|
|
fn sync_main(alt_mains: &mut HashMap<String, fn()>) {
|
|
alt_mains.insert("sync".to_owned(), sync::main);
|
|
}
|
|
|
|
fn updates_path(name: &str) -> PathBuf {
|
|
let mut dest = PathBuf::from(default_hypervisor_path());
|
|
dest.push(name);
|
|
dest
|
|
}
|
|
|
|
fn latest_exe_path() -> Option<PathBuf> {
|
|
File::open(updates_path("latest")).ok()
|
|
.and_then(|mut f| { let mut exe = String::new(); f.read_to_string(&mut exe).ok().map(|_| updates_path(&exe)) })
|
|
}
|
|
|
|
#[cfg(windows)]
|
|
fn global_cleanup() {
|
|
extern "system" { pub fn WSACleanup() -> i32; }
|
|
// We need to cleanup all sockets before spawning another Parity process. This makes shure everything is cleaned up.
|
|
// The loop is required because of internal refernce counter for winsock dll. We don't know how many crates we use do
|
|
// initialize it. There's at least 2 now.
|
|
for _ in 0.. 10 {
|
|
unsafe { WSACleanup(); }
|
|
}
|
|
}
|
|
|
|
#[cfg(not(windows))]
|
|
fn global_cleanup() {}
|
|
|
|
// Starts ~/.parity-updates/parity and returns the code it exits with.
|
|
fn run_parity() -> Option<i32> {
|
|
global_cleanup();
|
|
use ::std::ffi::OsString;
|
|
let prefix = vec![OsString::from("--can-restart"), OsString::from("--force-direct")];
|
|
latest_exe_path().and_then(|exe| process::Command::new(exe)
|
|
.args(&(env::args_os().skip(1).chain(prefix.into_iter()).collect::<Vec<_>>()))
|
|
.status()
|
|
.map(|es| es.code().unwrap_or(128))
|
|
.ok()
|
|
)
|
|
}
|
|
|
|
const PLEASE_RESTART_EXIT_CODE: i32 = 69;
|
|
|
|
// Run our version of parity.
|
|
// Returns the exit error code.
|
|
fn main_direct(can_restart: bool) -> i32 {
|
|
let mut alt_mains = HashMap::new();
|
|
sync_main(&mut alt_mains);
|
|
stratum_main(&mut alt_mains);
|
|
if let Some(f) = std::env::args().nth(1).and_then(|arg| alt_mains.get(&arg.to_string())) {
|
|
f();
|
|
0
|
|
} else {
|
|
match start(can_restart) {
|
|
Ok(result) => match result {
|
|
PostExecutionAction::Print(s) => { println!("{}", s); 0 },
|
|
PostExecutionAction::Restart => PLEASE_RESTART_EXIT_CODE,
|
|
PostExecutionAction::Quit => 0,
|
|
},
|
|
Err(err) => {
|
|
writeln!(&mut stdio::stderr(), "{}", err).expect("StdErr available; qed");
|
|
1
|
|
},
|
|
}
|
|
}
|
|
}
|
|
|
|
fn println_trace_main(s: String) {
|
|
if env::var("RUST_LOG").ok().and_then(|s| s.find("main=trace")).is_some() {
|
|
println!("{}", s);
|
|
}
|
|
}
|
|
|
|
#[macro_export]
|
|
macro_rules! trace_main {
|
|
($arg:expr) => (println_trace_main($arg.into()));
|
|
($($arg:tt)*) => (println_trace_main(format!("{}", format_args!($($arg)*))));
|
|
}
|
|
|
|
fn main() {
|
|
// Always print backtrace on panic.
|
|
env::set_var("RUST_BACKTRACE", "1");
|
|
|
|
// assuming the user is not running with `--force-direct`, then:
|
|
// if argv[0] == "parity" and this executable != ~/.parity-updates/parity, run that instead.
|
|
let force_direct = std::env::args().any(|arg| arg == "--force-direct");
|
|
let exe = std::env::current_exe().ok();
|
|
let development = exe.as_ref().and_then(|p| p.parent().and_then(|p| p.parent()).and_then(|p| p.file_name()).map(|n| n == "target")).unwrap_or(false);
|
|
let same_name = exe.as_ref().map(|p| p.file_stem().map_or(false, |s| s == "parity") && p.extension().map_or(true, |x| x == "exe")).unwrap_or(false);
|
|
trace_main!("Starting up {} (force-direct: {}, development: {}, same-name: {})", std::env::current_exe().map(|x| format!("{}", x.display())).unwrap_or("<unknown>".to_owned()), force_direct, development, same_name);
|
|
if !force_direct && !development && same_name {
|
|
// looks like we're not running ~/.parity-updates/parity when the user is expecting otherwise.
|
|
// Everything run inside a loop, so we'll be able to restart from the child into a new version seamlessly.
|
|
loop {
|
|
// If we fail to run the updated parity then fallback to local version.
|
|
let latest_exe = latest_exe_path();
|
|
let have_update = latest_exe.as_ref().map_or(false, |p| p.exists());
|
|
let is_non_updated_current = exe.as_ref().map_or(false, |exe| latest_exe.as_ref().map_or(false, |lexe| exe.canonicalize().ok() != lexe.canonicalize().ok()));
|
|
trace_main!("Starting... (have-update: {}, non-updated-current: {})", have_update, is_non_updated_current);
|
|
let exit_code = if have_update && is_non_updated_current {
|
|
trace_main!("Attempting to run latest update ({})...", latest_exe.as_ref().expect("guarded by have_update; latest_exe must exist for have_update; qed").display());
|
|
run_parity().unwrap_or_else(|| { trace_main!("Falling back to local..."); main_direct(true) })
|
|
} else {
|
|
trace_main!("No latest update. Attempting to direct...");
|
|
main_direct(true)
|
|
};
|
|
trace_main!("Latest exited with {}", exit_code);
|
|
if exit_code != PLEASE_RESTART_EXIT_CODE {
|
|
trace_main!("Quitting...");
|
|
process::exit(exit_code);
|
|
}
|
|
trace_main!("Rerunning...");
|
|
}
|
|
} else {
|
|
trace_main!("Running direct");
|
|
// Otherwise, we're presumably running the version we want. Just run and fall-through.
|
|
let can_restart = std::env::args().any(|arg| arg == "--can-restart");
|
|
process::exit(main_direct(can_restart));
|
|
}
|
|
}
|