Refactor into system service, add exe redirect.

ethcore/hash-fetch/src/client.rs

@@ -84,7 +84,7 @@ impl Client {
 
 impl HashFetch for Client {
 	fn fetch(&self, hash: H256, on_done: Box<Fn(Result<PathBuf, Error>) + Send>) -> Result<(), Error> {
-		debug!(target: "dapps", "Fetching: {:?}", hash);
+		debug!(target: "fetch", "Fetching: {:?}", hash);
 
 		let url = try!(
 			self.contract.resolve(hash.to_vec()).map(|content| match content {
@@ -97,7 +97,7 @@ impl HashFetch for Client {
 			}).ok_or_else(|| Error::NoResolution)
 		);
 
-		debug!(target: "dapps", "Resolved {:?} to {:?}. Fetching...", hash, url);
+		debug!(target: "fetch", "Resolved {:?} to {:?}. Fetching...", hash, url);
 
 		self.fetch.lock().request_async(&url, Default::default(), Box::new(move |result| {
 			fn validate_hash(hash: H256, result: Result<PathBuf, FetchError>) -> Result<PathBuf, Error> {
@@ -112,7 +112,7 @@ impl HashFetch for Client {
 				}
 			}
 
-			debug!(target: "dapps", "Content fetched, validating hash ({:?})", hash);
+			debug!(target: "fetch", "Content fetched, validating hash ({:?})", hash);
 			on_done(validate_hash(hash, result))
 		})).map_err(Into::into)
 	}

ethcore/src/client/client.rs

@@ -70,10 +70,7 @@ use factory::Factories;
 use rlp::{decode, View, UntrustedRlp};
 use state_db::StateDB;
 use rand::OsRng;
-use client::updater::Updater;
 use client::registry::Registry;
-use client::fetch::FetchHandler;
-use fetch::{self, HashFetch};
 
 // re-export
 pub use types::blockchain_info::BlockChainInfo;
@@ -144,7 +141,6 @@ pub struct Client {
 	panic_handler: Arc<PanicHandler>,
 	verifier: Box<Verifier>,
 	miner: Arc<Miner>,
-	updater: Mutex<Option<Updater>>,
 	sleep_state: Mutex<SleepState>,
 	liveness: AtomicBool,
 	io_channel: Mutex<IoChannel<ClientIoMessage>>,
@@ -156,7 +152,6 @@ pub struct Client {
 	rng: Mutex<OsRng>,
 	on_mode_change: Mutex<Option<Box<FnMut(&Mode) + 'static + Send>>>,
 	registrar: Mutex<Option<Registry>>,
-	fetch_service: Mutex<Option<Arc<HashFetch>>>,
 }
 
 impl Client {
@@ -247,7 +242,6 @@ impl Client {
 			import_lock: Mutex::new(()),
 			panic_handler: panic_handler,
 			miner: miner,
-			updater: Mutex::new(None),
 			io_channel: Mutex::new(message_channel),
 			notify: RwLock::new(Vec::new()),
 			queue_transactions: AtomicUsize::new(0),
@@ -257,23 +251,12 @@ impl Client {
 			rng: Mutex::new(try!(OsRng::new().map_err(::util::UtilError::StdIo))),
 			on_mode_change: Mutex::new(None),
 			registrar: Mutex::new(None),
-			fetch_service: Mutex::new(None),
 		});
 		if let Some(reg_addr) = client.additional_params().get("registrar").and_then(|s| Address::from_str(s).ok()) {
 			trace!(target: "client", "Found registrar at {}", reg_addr);
 			let weak = Arc::downgrade(&client);
-			let fetch = Arc::new(fetch::Client::new(Arc::new(FetchHandler::new(weak.clone()))));
 			let registrar = Registry::new(reg_addr, move |a, d| weak.upgrade().ok_or("No client!".into()).and_then(|c| c.call_contract(a, d)));
-			// TODO [ToDr] The address might not be available when client is starting (but may be available later).
-			// Shouldn't this be moved inside the `Updater`?
-			if let Ok(ops_addr) = registrar.get_address(&(&b"operations"[..]).sha3(), "A") {
-				if !ops_addr.is_zero() {
-					trace!(target: "client", "Found operations at {}", ops_addr);
-					*client.updater.lock() = Some(Updater::new(Arc::downgrade(&client), Arc::downgrade(&fetch), ops_addr, client.config.update_policy.clone()));
-				}
-			}
 			*client.registrar.lock() = Some(registrar);
-			*client.fetch_service.lock() = Some(fetch);
 		}
 		Ok(client)
 	}
@@ -686,12 +669,6 @@ impl Client {
 	pub fn tick(&self) {
 		self.check_garbage();
 		self.check_snooze();
-		if let Some(ref mut updater) = *self.updater.lock() {
-			updater.tick();
-			if updater.installed.is_some() {
-				info!("Client should restart now.");
-			}
-		}
 	}
 
 	fn check_garbage(&self) {
@@ -734,30 +711,6 @@ impl Client {
 		}
 	}
 
-	/// Like `call`, but with various defaults. Designed to be used for calling contracts.
-	pub fn call_contract(&self, address: Address, data: Bytes) -> Result<Bytes, String> {
-		let from = Address::default();
-		let transaction = Transaction {
-			nonce: self.latest_nonce(&from),
-			action: Action::Call(address),
-			gas: U256::from(50_000_000),
-			gas_price: U256::default(),
-			value: U256::default(),
-			data: data,
-		}.fake_sign(from);
-
-		self.call(&transaction, BlockId::Latest, Default::default())
-			.map_err(|e| format!("{:?}", e))
-			.map(|executed| {
-				executed.output
-			})
-	}
-
-	/// Get the updater object.
-	pub fn updater(&self) -> MutexGuard<Option<Updater>> {
-		self.updater.lock()
-	}
-
 	/// Look up the block number for the given block ID.
 	pub fn block_number(&self, id: BlockId) -> Option<BlockNumber> {
 		match id {
@@ -1377,6 +1330,34 @@ impl BlockChainClient for Client {
 			earliest_state: self.state_db.lock().journal_db().earliest_era().unwrap_or(0),
 		}
 	}
+
+	fn call_contract(&self, address: Address, data: Bytes) -> Result<Bytes, String> {
+		let from = Address::default();
+		let transaction = Transaction {
+			nonce: self.latest_nonce(&from),
+			action: Action::Call(address),
+			gas: U256::from(50_000_000),
+			gas_price: U256::default(),
+			value: U256::default(),
+			data: data,
+		}.fake_sign(from);
+
+		self.call(&transaction, BlockId::Latest, Default::default())
+			.map_err(|e| format!("{:?}", e))
+			.map(|executed| {
+				executed.output
+			})
+	}
+
+	fn registrar_address(&self) -> Option<Address> {
+		self.registrar.lock().as_ref().map(|r| r.address.clone())
+	}
+
+	fn registry_address(&self, name: String) -> Option<Address> {
+		self.registrar.lock().as_ref()
+			.and_then(|r| r.get_address(&(name.as_bytes().sha3()), "A").ok())
+			.and_then(|a| if a.is_zero() { None } else { Some(a) })
+	} 
 }
 
 impl MiningBlockChainClient for Client {

ethcore/src/client/config.rs

@@ -66,35 +66,6 @@ impl FromStr for DatabaseCompactionProfile {
 	}
 }
 
-/// Filter for releases.
-#[derive(Debug, Eq, PartialEq, Clone)]
-pub enum UpdateFilter {
-	/// All releases following the same track.
-	All,
-	/// As with `All`, but only those which are known to be critical. 
-	Critical,
-	/// None.
-	None,
-}
-
-/// The policy for auto-updating.
-#[derive(Debug, Eq, PartialEq, Clone)]
-pub struct UpdatePolicy {
-	/// Download potential updates.
-	pub enable_downloading: bool,
-	/// Which of those downloaded should be automatically installed.
-	pub filter: UpdateFilter,
-}
-
-impl Default for UpdatePolicy {
-	fn default() -> Self {
-		UpdatePolicy {
-			enable_downloading: false,
-			filter: UpdateFilter::None,
-		}
-	}
-}
-
 /// Operating mode for the client.
 #[derive(Debug, Eq, PartialEq, Clone)]
 pub enum Mode {
@@ -130,8 +101,6 @@ impl Display for Mode {
 /// Client configuration. Includes configs for all sub-systems.
 #[derive(Debug, PartialEq, Default)]
 pub struct ClientConfig {
-	/// Updater policy.
-	pub update_policy: UpdatePolicy,
 	/// Block queue configuration.
 	pub queue: QueueConfig,
 	/// Blockchain configuration.

ethcore/src/client/fetch.rs

@@ -1,49 +0,0 @@
-// 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 <http://www.gnu.org/licenses/>.
-
-use std::sync::Weak;
-use std::str::FromStr;
-use util::{Bytes, Address};
-
-use client::{Client, BlockChainClient};
-use fetch;
-
-/// Client wrapper implementing `fetch::urlhint::ContractClient`
-pub struct FetchHandler {
-	client: Weak<Client>,
-}
-
-impl FetchHandler {
-	/// Creates new wrapper
-	pub fn new(client: Weak<Client>) -> Self {
-		FetchHandler { client: client }
-	}
-}
-
-impl fetch::urlhint::ContractClient for FetchHandler {
-	fn registrar(&self) -> Result<Address, String> {
-		self.client.upgrade().ok_or_else(|| "Client not available".to_owned())?
-			.additional_params()
-			.get("registrar")
-			.and_then(|s| Address::from_str(s).ok())
-			.ok_or_else(|| "Registrar not available".into())
-	}
-
-	fn call(&self, address: Address, data: Bytes) -> Result<Bytes, String> {
-		self.client.upgrade().ok_or_else(|| "Client not available".to_owned())?
-			.call_contract(address, data)
-	}
-}

ethcore/src/client/mod.rs

@@ -16,18 +16,15 @@
 
 //! Blockchain database client.
 
-mod operations;
 mod registry;
 mod config;
 mod error;
 mod test_client;
 mod trace;
 mod client;
-mod updater;
-mod fetch;
 
 pub use self::client::*;
-pub use self::config::{Mode, ClientConfig, UpdatePolicy, UpdateFilter, DatabaseCompactionProfile, BlockChainConfig, VMType};
+pub use self::config::{Mode, ClientConfig, DatabaseCompactionProfile, BlockChainConfig, VMType};
 pub use self::error::Error;
 pub use self::test_client::{TestBlockChainClient, EachBlockWith};
 pub use self::chain_notify::ChainNotify;

ethcore/src/client/registry.rs

@@ -9,7 +9,7 @@ use util::Uint;
 
 pub struct Registry {
 	contract: ethabi::Contract,
-	address: util::Address,
+	pub address: util::Address,
 	do_call: Box<Fn(util::Address, Vec<u8>) -> Result<Vec<u8>, String> + Send + 'static>,
 }
 impl Registry {

ethcore/src/client/test_client.rs

@@ -679,4 +679,10 @@ impl BlockChainClient for TestBlockChainClient {
 			earliest_state: 1,
 		}
 	}
+
+	fn call_contract(&self, _address: Address, _data: Bytes) -> Result<Bytes, String> { Ok(vec![]) }
+
+	fn registrar_address(&self) -> Option<Address> { None }
+
+	fn registry_address(&self, _name: String) -> Option<Address> { None }
 }

ethcore/src/client/traits.rs

@@ -257,6 +257,15 @@ pub trait BlockChainClient : Sync + Send {
 
 	/// Returns information about pruning/data availability.
 	fn pruning_info(&self) -> PruningInfo;
+
+	/// Like `call`, but with various defaults. Designed to be used for calling contracts.
+	fn call_contract(&self, address: Address, data: Bytes) -> Result<Bytes, String>;
+
+	/// Get the address of the registry itself.
+	fn registrar_address(&self) -> Option<Address>;
+
+	/// Get the address of a particular blockchain service, if available. 
+	fn registry_address(&self, name: String) -> Option<Address>;
 }
 
 impl IpcConfig for BlockChainClient { }

parity/blockchain.rs

@@ -32,7 +32,6 @@ use ethcore::verification::queue::VerifierSettings;
 use cache::CacheConfig;
 use informant::{Informant, MillisecondDuration};
 use params::{SpecType, Pruning, Switch, tracing_switch_to_bool, fatdb_switch_to_bool};
-use io_handler::ImportIoHandler;
 use helpers::{to_client_config, execute_upgrades};
 use dir::Directories;
 use user_defaults::UserDefaults;
@@ -231,11 +230,8 @@ fn execute_import(cmd: ImportBlockchain) -> Result<String, String> {
 		}
 	};
 
-	let informant = Informant::new(client.clone(), None, None, None, cmd.with_color);
-
-	try!(service.register_io_handler(Arc::new(ImportIoHandler {
-		info: Arc::new(informant),
-	})).map_err(|_| "Unable to register informant handler".to_owned()));
+	let informant = Arc::new(Informant::new(client.clone(), None, None, None, cmd.with_color));
+	service.register_io_handler(informant).map_err(|_| "Unable to register informant handler".to_owned())?;
 
 	let do_import = |bytes| {
 		while client.queue_info().is_full() { sleep(Duration::from_secs(1)); }
@@ -251,7 +247,6 @@ fn execute_import(cmd: ImportBlockchain) -> Result<String, String> {
 		Ok(())
 	};
 
-
 	match format {
 		DataFormat::Binary => {
 			loop {

parity/cli/usage.txt

@@ -42,6 +42,8 @@ Operating Options:
   --no-consensus           Force the binary to run even if there are known
                            issues regarding consensus. Not recommended.
                            (default: {flag_no_consensus}).
+  --force-direct           Run the originally installed version of Parity,
+                           ignoring any updates that have since been installed.  
   --chain CHAIN            Specify the blockchain type. CHAIN may be either a
                            JSON chain specification file or olympic, frontier,
                            homestead, mainnet, morden, ropsten, classic, expanse,

parity/helpers.rs

@@ -209,7 +209,6 @@ pub fn default_network_config() -> ::ethsync::NetworkConfiguration {
 #[cfg_attr(feature = "dev", allow(too_many_arguments))]
 pub fn to_client_config(
 		cache_config: &CacheConfig,
-		update_policy: UpdatePolicy,
 		mode: Mode,
 		tracing: bool,
 		fat_db: bool,
@@ -243,7 +242,6 @@ pub fn to_client_config(
 	// in bytes
 	client_config.jump_table_size = cache_config.jump_tables() as usize * mb;
 
-	client_config.update_policy = update_policy;
 	client_config.mode = mode;
 	client_config.tracing.enabled = tracing;
 	client_config.fat_db = fat_db;

parity/informant.rs

@@ -222,3 +222,14 @@ impl ChainNotify for Informant {
 	}
 }
 
+impl IoHandler<ClientIoMessage> for Informant {
+	fn initialize(&self, io: &IoContext<ClientIoMessage>) {
+		io.register_timer(INFO_TIMER, 5000).expect("Error registering timer");
+	}
+
+	fn timeout(&self, _io: &IoContext<ClientIoMessage>, timer: TimerToken) {
+		if timer == INFO_TIMER && !self.shutdown.load(Ordering::SeqCst) {
+			self.info.tick();
+		}
+	}
+}

parity/io_handler.rs

@@ -1,64 +0,0 @@
-// 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 <http://www.gnu.org/licenses/>.
-
-use std::sync::Arc;
-use std::sync::atomic::{AtomicBool, Ordering};
-use ethcore::client::Client;
-use ethcore::service::ClientIoMessage;
-use ethsync::{SyncProvider, ManageNetwork};
-use ethcore::account_provider::AccountProvider;
-use io::{TimerToken, IoHandler, IoContext};
-
-use informant::Informant;
-
-const INFO_TIMER: TimerToken = 0;
-
-pub struct ClientIoHandler {
-	pub client: Arc<Client>,
-	pub sync: Arc<SyncProvider>,
-	pub net: Arc<ManageNetwork>,
-	pub accounts: Arc<AccountProvider>,
-	pub info: Arc<Informant>,
-	pub shutdown: Arc<AtomicBool>
-}
-
-impl IoHandler<ClientIoMessage> for ClientIoHandler {
-	fn initialize(&self, io: &IoContext<ClientIoMessage>) {
-		io.register_timer(INFO_TIMER, 5000).expect("Error registering timer");
-	}
-
-	fn timeout(&self, _io: &IoContext<ClientIoMessage>, timer: TimerToken) {
-		if timer == INFO_TIMER && !self.shutdown.load(Ordering::SeqCst) {
-			self.info.tick();
-		}
-	}
-}
-
-pub struct ImportIoHandler {
-	pub info: Arc<Informant>,
-}
-
-impl IoHandler<ClientIoMessage> for ImportIoHandler {
-	fn initialize(&self, io: &IoContext<ClientIoMessage>) {
-		io.register_timer(INFO_TIMER, 5000).expect("Error registering timer");
-	}
-
-	fn timeout(&self, _io: &IoContext<ClientIoMessage>, timer: TimerToken) {
-		if let INFO_TIMER = timer {
-			self.info.tick()
-		}
-	}
-}

parity/main.rs

@@ -109,13 +109,18 @@ mod sync;
 #[cfg(feature="ipc")]
 mod boot;
 mod user_defaults;
+mod updater;
+mod operations;
+mod fetch;
 
 #[cfg(feature="stratum")]
 mod stratum;
 
 use std::{process, env};
+use std::collections::HashMap;
 use std::io::{self as stdio, BufReader, Write};
 use std::fs::File;
+use std::path::PathBuf;
 use util::sha3::sha3;
 use cli::Args;
 use configuration::{Cmd, Execute, Configuration};
@@ -132,25 +137,31 @@ fn print_hash_of(maybe_file: Option<String>) -> Result<String, String> {
 	}
 }
 
-fn execute(command: Execute) -> Result<String, String> {
+enum PostExecutionAction {
+	Print(String),
+	Restart,
+	Quit,
+}
+
+fn execute(command: Execute) -> Result<PostExecutionAction, String> {
 	let logger = setup_log(&command.logger).expect("Logger is initialized only once; qed");
 
 	match command.cmd {
 		Cmd::Run(run_cmd) => {
-			try!(run::execute(run_cmd, logger));
-			Ok("".into())
+			let restart = run::execute(run_cmd, logger)?;
+			Ok(if restart { PostExecutionAction::Restart } else { PostExecutionAction::Quit })
 		},
-		Cmd::Version => Ok(Args::print_version()),
-		Cmd::Hash(maybe_file) => print_hash_of(maybe_file),
-		Cmd::Account(account_cmd) => account::execute(account_cmd),
-		Cmd::ImportPresaleWallet(presale_cmd) => presale::execute(presale_cmd),
-		Cmd::Blockchain(blockchain_cmd) => blockchain::execute(blockchain_cmd),
-		Cmd::SignerToken(signer_cmd) => signer::execute(signer_cmd),
-		Cmd::Snapshot(snapshot_cmd) => snapshot::execute(snapshot_cmd),
+		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(|s| PostExecutionAction::Print(s)),
+		Cmd::SignerToken(signer_cmd) => signer::execute(signer_cmd).map(|s| PostExecutionAction::Print(s)),
+		Cmd::Snapshot(snapshot_cmd) => snapshot::execute(snapshot_cmd).map(|s| PostExecutionAction::Print(s)),
 	}
 }
 
-fn start() -> Result<String, String> {
+fn start() -> Result<PostExecutionAction, String> {
 	let args: Vec<String> = env::args().collect();
 	let conf = Configuration::parse(&args).unwrap_or_else(|e| e.exit());
 
@@ -163,58 +174,108 @@ fn start() -> Result<String, String> {
 	execute(cmd)
 }
 
-#[cfg(feature="stratum")]
-mod stratum_optional {
-	pub fn probably_run() -> bool {
-		// just redirect to the stratum::main()
-		if ::std::env::args().nth(1).map_or(false, |arg| arg == "stratum") {
-			super::stratum::main();
-			true
-		}
-		else { false }
-	}
-}
-
 #[cfg(not(feature="stratum"))]
-mod stratum_optional {
-	pub fn probably_run() -> bool {
-		false
-	}
+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() -> bool {
-	false
-}
+fn sync_main(_: &mut HashMap<String, fn()>) {}
 
 #[cfg(feature="ipc")]
-fn sync_main() -> bool {
-	// just redirect to the sync::main()
-	if std::env::args().nth(1).map_or(false, |arg| arg == "sync") {
-		sync::main();
-		true
+fn sync_main(alt_mains: &mut HashMap<String, fn()>) {
+	alt_mains.insert("sync".to_owned(), sync::main);
+}
+
+// TODO: merge with version in Updater.
+fn updates_latest() -> PathBuf {
+	let mut dest = PathBuf::from(env::home_dir().unwrap().to_str().expect("env filesystem paths really should be valid; qed"));
+	dest.push(".parity-updates");
+	dest.push("parity");
+	dest
+}
+
+// Starts ~/.parity-updates/parity and returns the code it exits with.
+fn run_parity() -> Option<i32> {
+	let exe = updates_latest();
+	process::Command::new(exe)
+		.args(&env::args_os().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() -> 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 {
-		false
+		match start() {
+			Ok(result) => match result {
+				PostExecutionAction::Print(s) => { info!("{}", 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.
-	::std::env::set_var("RUST_BACKTRACE", "1");
+	env::set_var("RUST_BACKTRACE", "1");
 
-	if sync_main() {
-		return;
-	}
-
-	if stratum_optional::probably_run() { return; }
-
-	match start() {
-		Ok(result) => {
-			info!("{}", result);
-		},
-		Err(err) => {
-			writeln!(&mut stdio::stderr(), "{}", err).expect("StdErr available; qed");
-			process::exit(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.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.and_then(|p| p.file_stem().map_or(false, |s| s == "parity"));
+	let have_update = updates_latest().exists();
+	let is_non_updated_current = exe.map_or(false, p.canonicalize() != updates_latest().canonicalize());
+	trace_main!("Starting up {} (force-direct: {}, development: {}, have-update: {}, non-updated-current: {})", std::env::current_exe().map(|x| format!("{}", x.display())).unwrap_or("<unknown>".to_owned()), force_direct, development, have_update, is_non_updated_current);
+	if !force_direct && ! development && have_update && is_non_updated_current {
+		// 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. 
+			trace_main!("Attempting to run latest update...");
+			let exit_code = run_parity().unwrap_or_else(|| { trace_main!("Falling back to local..."); main_direct() });
+			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.
+		process::exit(main_direct());
 	}
 }

parity/run.rs

@@ -23,7 +23,7 @@ use ethsync::NetworkConfiguration;
 use util::{Colour, version, RotatingLogger};
 use io::{MayPanic, ForwardPanic, PanicHandler};
 use ethcore_logger::{Config as LogConfig};
-use ethcore::client::{Mode, UpdatePolicy, DatabaseCompactionProfile, VMType, ChainNotify, BlockChainClient};
+use ethcore::client::{Mode, UpdatePolicy, Updater, DatabaseCompactionProfile, VMType, ChainNotify, BlockChainClient};
 use ethcore::service::ClientService;
 use ethcore::account_provider::AccountProvider;
 use ethcore::miner::{Miner, MinerService, ExternalMiner, MinerOptions};
@@ -31,11 +31,11 @@ use ethcore::snapshot;
 use ethcore::verification::queue::VerifierSettings;
 use ethsync::SyncConfig;
 use informant::Informant;
+use updater::Updater;
 
 use rpc::{HttpServer, IpcServer, HttpConfiguration, IpcConfiguration};
 use signer::SignerServer;
 use dapps::WebappServer;
-use io_handler::ClientIoHandler;
 use params::{
 	SpecType, Pruning, AccountsConfig, GasPricerConfig, MinerExtras, Switch,
 	tracing_switch_to_bool, fatdb_switch_to_bool, mode_switch_to_bool
@@ -116,12 +116,12 @@ pub fn open_ui(dapps_conf: &dapps::Configuration, signer_conf: &signer::Configur
 	Ok(())
 }
 
-pub fn execute(cmd: RunCmd, logger: Arc<RotatingLogger>) -> Result<(), String> {
+pub fn execute(cmd: RunCmd, logger: Arc<RotatingLogger>) -> Result<bool, String> {
 	if cmd.ui && cmd.dapps_conf.enabled {
 		// Check if Parity is already running
 		let addr = format!("{}:{}", cmd.dapps_conf.interface, cmd.dapps_conf.port);
 		if !TcpListener::bind(&addr as &str).is_ok() {
-			return open_ui(&cmd.dapps_conf, &cmd.signer_conf);
+			return open_ui(&cmd.dapps_conf, &cmd.signer_conf).map(|_| false);
 		}
 	}
 
@@ -244,7 +244,6 @@ pub fn execute(cmd: RunCmd, logger: Arc<RotatingLogger>) -> Result<(), String> {
 	// create client config
 	let mut client_config = to_client_config(
 		&cmd.cache_config,
-		update_policy, 
 		mode.clone(),
 		tracing,
 		fat_db,
@@ -312,6 +311,12 @@ pub fn execute(cmd: RunCmd, logger: Arc<RotatingLogger>) -> Result<(), String> {
 		chain_notify.start();
 	}
 
+	// the updater service
+	let updater = Updater::new(service.client(), update_policy);
+	if let Some(ref u) = updater {
+		service.add_notify(u.clone());
+	}
+
 	// set up dependencies for rpc servers
 	let signer_path = cmd.signer_conf.signer_path.clone();
 	let deps_for_rpc_apis = Arc::new(rpc_apis::Dependencies {
@@ -348,24 +353,23 @@ pub fn execute(cmd: RunCmd, logger: Arc<RotatingLogger>) -> Result<(), String> {
 	let http_server = try!(rpc::new_http(cmd.http_conf, &dependencies));
 	let ipc_server = try!(rpc::new_ipc(cmd.ipc_conf, &dependencies));
 
+	// the dapps server
 	let dapps_deps = dapps::Dependencies {
 		panic_handler: panic_handler.clone(),
 		apis: deps_for_rpc_apis.clone(),
 		client: client.clone(),
 		sync: sync_provider.clone(),
 	};
-
-	// start dapps server
 	let dapps_server = try!(dapps::new(cmd.dapps_conf.clone(), dapps_deps));
 
+	// the signer server
 	let signer_deps = signer::Dependencies {
 		panic_handler: panic_handler.clone(),
 		apis: deps_for_rpc_apis.clone(),
 	};
-
-	// start signer server
 	let signer_server = try!(signer::start(cmd.signer_conf.clone(), signer_deps));
 
+	// the informant
 	let informant = Arc::new(Informant::new(
 		service.client(),
 		Some(sync_provider.clone()),
@@ -373,17 +377,8 @@ pub fn execute(cmd: RunCmd, logger: Arc<RotatingLogger>) -> Result<(), String> {
 		Some(snapshot_service.clone()),
 		cmd.logger_config.color
 	));
-	let info_notify: Arc<ChainNotify> = informant.clone();
-	service.add_notify(info_notify);
-	let io_handler = Arc::new(ClientIoHandler {
-		client: service.client(),
-		info: informant,
-		sync: sync_provider.clone(),
-		net: manage_network.clone(),
-		accounts: account_provider.clone(),
-		shutdown: Default::default(),
-	});
-	service.register_io_handler(io_handler.clone()).expect("Error registering IO handler");
+	service.add_notify(informant.clone());
+	service.register_io_handler(informant.clone()).map_err(|_| "Unable to register informant handler".to_owned())?;
 
 	// save user defaults
 	user_defaults.pruning = algorithm;
@@ -392,13 +387,11 @@ pub fn execute(cmd: RunCmd, logger: Arc<RotatingLogger>) -> Result<(), String> {
 	user_defaults.mode = mode;
 	try!(user_defaults.save(&user_defaults_path));
 
-	let on_mode_change = move |mode: &Mode| {
+	// tell client how to save the default mode if it gets changed.
+	client.on_mode_change(move |mode: &Mode| {
 		user_defaults.mode = mode.clone();
 		let _ = user_defaults.save(&user_defaults_path);	// discard failures - there's nothing we can do
-	};
-
-	// tell client how to save the default mode if it gets changed.
-	client.on_mode_change(on_mode_change);
+	});
 
 	// the watcher must be kept alive.
 	let _watcher = match cmd.no_periodic_snapshot {
@@ -424,7 +417,9 @@ pub fn execute(cmd: RunCmd, logger: Arc<RotatingLogger>) -> Result<(), String> {
 	}
 
 	// Handle exit
-	wait_for_exit(panic_handler, http_server, ipc_server, dapps_server, signer_server);
+	wait_for_exit(panic_handler, http_server, ipc_server, dapps_server, signer_server, updater);
+
+	info!("Finishing work, please wait...");
 
 	// to make sure timer does not spawn requests while shutdown is in progress
 	io_handler.shutdown.store(true, ::std::sync::atomic::Ordering::SeqCst);
@@ -435,7 +430,7 @@ pub fn execute(cmd: RunCmd, logger: Arc<RotatingLogger>) -> Result<(), String> {
 	// terminated gracefully
 	drop(hypervisor);
 
-	Ok(())
+	Ok(false)
 }
 
 #[cfg(not(windows))]
@@ -443,11 +438,11 @@ fn daemonize(pid_file: String) -> Result<(), String> {
 	extern crate daemonize;
 
 	daemonize::Daemonize::new()
-			.pid_file(pid_file)
-			.chown_pid_file(true)
-			.start()
-			.map(|_| ())
-			.map_err(|e| format!("Couldn't daemonize; {}", e))
+		.pid_file(pid_file)
+		.chown_pid_file(true)
+		.start()
+		.map(|_| ())
+		.map_err(|e| format!("Couldn't daemonize; {}", e))
 }
 
 #[cfg(windows)]
@@ -478,20 +473,26 @@ fn wait_for_exit(
 	_http_server: Option<HttpServer>,
 	_ipc_server: Option<IpcServer>,
 	_dapps_server: Option<WebappServer>,
-	_signer_server: Option<SignerServer>
-	) {
-	let exit = Arc::new(Condvar::new());
+	_signer_server: Option<SignerServer>,
+	updater: Option<Arc<Updater>>
+) -> bool {
+	let exit = Arc::new((Mutex::new(false), Condvar::new()));
 
 	// Handle possible exits
 	let e = exit.clone();
-	CtrlC::set_handler(move || { e.notify_all(); });
+	CtrlC::set_handler(move || { e.1.notify_all(); });
 
 	// Handle panics
 	let e = exit.clone();
-	panic_handler.on_panic(move |_reason| { e.notify_all(); });
+	panic_handler.on_panic(move |_reason| { e.1.notify_all(); });
+
+	// Handle updater wanting to restart us
+	if let Some(ref u) = updater {
+		let e = exit.clone();
+		u.set_exit_handler(move || { e.0.lock() = true; e.1.notify_all(); });
+	}
 
 	// Wait for signal
-	let mutex = Mutex::new(());
-	let _ = exit.wait(mutex.lock().unwrap());
-	info!("Finishing work, please wait...");
+	let _ = exit.1.wait(exit.0.lock().unwrap());
+	*exit.0.lock()
 }

parity/updater.rs

@@ -19,84 +19,140 @@ use std::{io, os, fs, env};
 use std::path::{Path, PathBuf};
 use util::misc::{VersionInfo, ReleaseTrack/*, platform*/};
 use util::{Address, H160, H256, FixedHash, Mutex};
-use client::operations::Operations;
-use client::{Client, UpdatePolicy, UpdateFilter, BlockId};
+use super::operations::Operations;
+use ethcore::client::{Client, BlockId};
 use fetch::HashFetch;
 use fetch;
 
+/// Filter for releases.
+#[derive(Debug, Eq, PartialEq, Clone)]
+pub enum UpdateFilter {
+	/// All releases following the same track.
+	All,
+	/// As with `All`, but only those which are known to be critical. 
+	Critical,
+	/// None.
+	None,
+}
+
+/// The policy for auto-updating.
+#[derive(Debug, Eq, PartialEq, Clone)]
+pub struct UpdatePolicy {
+	/// Download potential updates.
+	pub enable_downloading: bool,
+	/// Which of those downloaded should be automatically installed.
+	pub filter: UpdateFilter,
+}
+
+impl Default for UpdatePolicy {
+	fn default() -> Self {
+		UpdatePolicy {
+			enable_downloading: false,
+			filter: UpdateFilter::None,
+		}
+	}
+}
+
+/// Information regarding a particular release of Parity
 #[derive(Debug, Clone, PartialEq)]
 pub struct ReleaseInfo {
+	/// Information on the version.
 	pub version: VersionInfo,
+	/// Does this release contain critical security updates? 
 	pub is_critical: bool,
+	/// The latest fork that this release can handle.
 	pub fork: u64,
+	/// Our platform's binary, if known. 
 	pub binary: Option<H256>,
 }
 
+/// Information on our operations environment.
 #[derive(Debug, Clone, PartialEq)]
 pub struct OperationsInfo {
+	/// Our blockchain's latest fork.
 	pub fork: u64,
 
+	/// Last fork our client supports, if known. 
+	pub this_fork: Option<u64>,
+
+	/// Information on our track's latest release. 
 	pub track: ReleaseInfo,
+	/// Information on our minor version's latest release.
 	pub minor: Option<ReleaseInfo>,
 }
 
+#[derive(Debug, Default)]
+struct UpdaterState {
+	latest: Option<OperationsInfo>,
+
+	fetching: Option<ReleaseInfo>,
+	ready: Option<ReleaseInfo>,
+	installed: Option<ReleaseInfo>,
+}
+
+/// Service for checking for updates and determining whether we can achieve consensus.
 pub struct Updater {
-	client: Weak<Client>,
-	fetch: Weak<HashFetch>,
-	operations: Operations,
+	// Useful environmental stuff.
 	update_policy: UpdatePolicy,
-	fetching: Mutex<Option<ReleaseInfo>>,
+	weak_self: Weak<Updater>,
+	client: Weak<Client>,
+	fetcher: Option<fetch::Client>,
+	operations: Mutex<Option<Operations>>,
+	exit_handler: Mutex<Option<Fn()>>,
 
-	// These don't change
-	pub this: VersionInfo,
-	pub this_fork: Option<u64>,
+	// Our version info (static)
+	this: VersionInfo,
 
-	// This does change
-	pub latest: Option<OperationsInfo>,
-	pub ready: Option<ReleaseInfo>,
-	// If Some, client should restart itself.
-	pub installed: Option<ReleaseInfo>,
+	// All the other info - this changes so leave it behind a Mutex.
+	state: Mutex<UpdaterState>,
 }
 
 const CLIENT_ID: &'static str = "parity";
 
+// TODO!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! REMOVE!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
 fn platform() -> String {
 	"test".to_owned()
 }
 
-impl Updater {
-	pub fn new(client: Weak<Client>, fetch: Weak<fetch::Client>, operations: Address, update_policy: UpdatePolicy) -> Self {
-		let mut u = Updater {
-			client: client.clone(),
-			fetch: fetch.clone(),
-			operations: Operations::new(operations, move |a, d| client.upgrade().ok_or("No client!".into()).and_then(|c| c.call_contract(a, d))),
-			update_policy: update_policy,
-			fetching: Mutex::new(None),
-			this: VersionInfo::this(),
-			this_fork: None,
-			latest: None,
-			ready: None,
-			installed: None,
-		};
+#[cfg(windows)]
+fn symlink<P: AsRef<Path>, Q: AsRef<Path>>(src: P, dst: Q) -> io::Result<()> {
+	os::windows::fs::symlink_file(src, dst)
+}
 
-		u.this_fork = u.operations.release(CLIENT_ID, &u.this.hash.into()).ok()
-			.and_then(|(fork, track, _, _)| if track > 0 {Some(fork as u64)} else {None});
+#[cfg(not(windows))]
+fn symlink<P: AsRef<Path>, Q: AsRef<Path>>(src: P, dst: Q) -> io::Result<()> {
+	os::unix::fs::symlink(src, dst)
+}
+
+impl Updater {
+	pub fn new(client: Weak<BlockChainClient>, update_policy: UpdatePolicy) -> Arc<Self> {
+		let mut u = Updater {
+			update_policy: update_policy,
+			weak_self: Default::default(),
+			client: client.clone(),
+			fetcher: None,
+			operations: Mutex::new(None),
+			exit_handler: Mutex::new(None),
+			this: VersionInfo::this(),
+			state: Mutex::new(Default::default()),
+		};
 
 		// TODO!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! REMOVE!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
 		if u.this.track == ReleaseTrack::Unknown {
 			u.this.track = ReleaseTrack::Nightly;
 		}
 
-		u.latest = u.collect_latest().ok();
-
-		u
+		let r = Arc::new(u);
+		r.as_mut().weak_self = Arc::downgrade(&r);
+		r.as_mut().fetcher = Some(fetch::Client::new(r));
+		r
 	}
 
 	/// Is the currently running client capable of supporting the current chain?
 	/// `Some` answer or `None` if information on the running client is not available.
 	pub fn is_capable(&self) -> Option<bool> {
-		self.latest.as_ref().and_then(|latest| {
-			self.this_fork.map(|this_fork| {
+		self.state.lock().latest.as_ref().and_then(|latest| {
+			latest.this_fork.map(|this_fork| {
 				let current_number = self.client.upgrade().map_or(0, |c| c.block_number(BlockId::Latest).unwrap_or(0));
 				this_fork >= latest.fork || current_number < latest.fork
 			})
@@ -106,35 +162,29 @@ impl Updater {
 	/// The release which is ready to be upgraded to, if any. If this returns `Some`, then
 	/// `execute_upgrade` may be called.
 	pub fn upgrade_ready(&self) -> Option<ReleaseInfo> {
-		self.ready.clone()
-	}
-
-	#[cfg(windows)]
-	fn symlink<P: AsRef<Path>, Q: AsRef<Path>>(src: P, dst: Q) -> io::Result<()> {
-		os::windows::fs::symlink_file(src, dst)
-	}
-
-	#[cfg(not(windows))]
-	fn symlink<P: AsRef<Path>, Q: AsRef<Path>>(src: P, dst: Q) -> io::Result<()> {
-		os::unix::fs::symlink(src, dst)
+		self.state.lock().ready.clone()
 	}
 
 	/// Actually upgrades the client. Assumes that the binary has been downloaded.
 	/// @returns `true` on success.
 	pub fn execute_upgrade(&mut self) -> bool {
 		(|| -> Result<bool, String> {
-			if let Some(r) = self.ready.take() {
+			let s = state.lock();
+			if let Some(r) = s.ready.take() {
 				let p = Self::update_file_path(&r.version);
 				let n = Self::updates_latest();
 				let _ = fs::remove_file(&n);
-				match Self::symlink(p, n) {
+				match symlink(p, n) {
 					Ok(_) => {
 						info!("Completed upgrade to {}", &r.version);
-						self.installed = Some(r);
+						s.installed = Some(r);
+						if let Some(ref h) = self.exit_handler().lock() {
+							(*h)();
+						}
 						Ok(true)
 					}
 					Err(e) => {
-						self.ready = Some(r);
+						s.ready = Some(r);
 						Err(format!("Unable to create soft-link for update {:?}", e))
 					}
 				}
@@ -146,7 +196,7 @@ impl Updater {
 	}
 
 	/// Returns true iff the current version is capable of forming consensus.
-	pub fn consensus_capable(&self) -> bool {
+	pub fn is_consensus_capable(&self) -> bool {
 /*		if let Some(ref latest) = self.latest {
 			
 
@@ -157,7 +207,12 @@ impl Updater {
 	pub fn version_info(&self) -> &VersionInfo { &self.this }
 
 	/// Information gathered concerning the release.
-	pub fn info(&self) -> &Option<OperationsInfo> { &self.latest }
+	pub fn info(&self) -> Option<OperationsInfo> { self.state.lock().latest.clone() }
+
+	/// Set a closure to call when we want to restart the client
+	pub fn set_exit_handler(&self, f: Fn()) {
+		*self.exit_handler.lock() = f; 
+	}
 
 	fn collect_release_info(&self, release_id: &H256) -> Result<ReleaseInfo, String> {
 		let (fork, track, semver, is_critical) = self.operations.release(CLIENT_ID, release_id)?;
@@ -171,6 +226,9 @@ impl Updater {
 	}
 
 	fn collect_latest(&self) -> Result<OperationsInfo, String> {
+		let this_fork = u.operations.release(CLIENT_ID, &u.this.hash.into()).ok()
+			.and_then(|(fork, track, _, _)| if track > 0 {Some(fork as u64)} else {None});
+
 		if self.this.track == ReleaseTrack::Unknown {
 			return Err(format!("Current executable ({}) is unreleased.", H160::from(self.this.hash)));
 		}
@@ -190,6 +248,7 @@ impl Updater {
 
 		Ok(OperationsInfo {
 			fork: self.operations.latest_fork()? as u64,
+			this_fork: this_fork,
 			track: in_track,
 			minor: in_minor,
 		})
@@ -211,19 +270,23 @@ impl Updater {
 
 	fn fetch_done(&mut self, result: Result<PathBuf, fetch::Error>) {
 		(|| -> Result<(), String> {
-			let fetched = self.fetching.lock().take().unwrap();
-			let b = result.map_err(|e| format!("Unable to fetch update ({}): {:?}", fetched.version, e))?;
-			info!("Fetched latest version ({}) OK to {}", fetched.version, b.display());
-			let dest = Self::update_file_path(&fetched.version);
-			fs::create_dir_all(dest.parent().expect("at least one thing pushed; qed")).map_err(|e| format!("Unable to create updates path: {:?}", e))?;
-			fs::copy(&b, &dest).map_err(|e| format!("Unable to copy update: {:?}", e))?;
-			info!("Copied file to {}", dest.display());
-			let auto = match self.update_policy.filter {
-				UpdateFilter::All => true,
-				UpdateFilter::Critical if fetched.is_critical /* TODO: or is on a bad fork */ => true,
-				_ => false,
+			let auto = {
+				let mut s = state.lock();
+				let fetched = s.fetching.take().unwrap();
+				let b = result.map_err(|e| format!("Unable to fetch update ({}): {:?}", fetched.version, e))?;
+				info!("Fetched latest version ({}) OK to {}", fetched.version, b.display());
+				let dest = Self::update_file_path(&fetched.version);
+				fs::create_dir_all(dest.parent().expect("at least one thing pushed; qed")).map_err(|e| format!("Unable to create updates path: {:?}", e))?;
+				fs::copy(&b, &dest).map_err(|e| format!("Unable to copy update: {:?}", e))?;
+				info!("Copied file to {}", dest.display());
+				let auto = match self.update_policy.filter {
+					UpdateFilter::All => true,
+					UpdateFilter::Critical if fetched.is_critical /* TODO: or is on a bad fork */ => true,
+					_ => false,
+				};
+				s.ready = Some(fetched);
+				auto
 			};
-			self.ready = Some(fetched);
 			if auto {
 				self.execute_upgrade();
 			}
@@ -231,13 +294,26 @@ impl Updater {
 		})().unwrap_or_else(|e| warn!("{}", e));
 	}
 
-	pub fn tick(&mut self) {
+	fn poll(&mut self) {
 		info!(target: "updater", "Current release is {}", self.this);
 
-		self.latest = self.collect_latest().ok();
+		if *self.operations.lock().is_none() {
+			if let Some(ops_addr) = client.upgrade().registry_address("operations") {
+				trace!(target: "client", "Found operations at {}", ops_addr);
+				let client = self.client.clone();
+				*self.operations.lock() = Some(Operations::new(ops_addr, move |a, d| client.upgrade().ok_or("No client!".into()).and_then(|c| c.call_contract(a, d))));
+			} else {
+				// No Operations contract - bail.
+				return;
+			}
+		}
+
+		u.latest = u.collect_latest().ok();
+
 		let current_number = self.client.upgrade().map_or(0, |c| c.block_number(BlockId::Latest).unwrap_or(0));
 
-		if let Some(ref latest) = self.latest {
+		let latest = self.collect_latest().ok();
+		if let Some(ref latest) = latest {
 			info!(target: "updater", "Latest release in our track is v{} it is {}critical ({} binary is {})",
 				latest.track.version,
 				if latest.track.is_critical {""} else {"non-"},
@@ -248,27 +324,44 @@ impl Updater {
 					"unreleased".into()
 				}
 			);
-			if self.update_policy.enable_downloading && latest.track.version.hash != self.version_info().hash && self.installed.as_ref().or(self.ready.as_ref()).map_or(true, |t| *t != latest.track) {
+			let mut s = self.state.lock();
+			let running_latest = latest.track.version.hash == self.version_info().hash;
+			let already_have_latest = s.installed.as_ref().or(s.ready.as_ref()).map_or(false, |t| *t == latest.track);
+			if self.update_policy.enable_downloading && !running_latest && !already_have_latest {
 				if let Some(b) = latest.track.binary {
-					let mut fetching = self.fetching.lock();
 					if fetching.is_none() {
 						info!("Attempting to get parity binary {}", b);
-						let c = self.client.clone();
-						let f = move |r: Result<PathBuf, fetch::Error>| {
-							if let Some(client) = c.upgrade() {
-								if let Some(updater) = client.updater().as_mut() {
-									updater.fetch_done(r);
-								}
-							}
-						};
-						if let Some(fetch) = self.fetch.clone().upgrade() {
-							fetch.fetch(b, Box::new(f)).ok();
-							*fetching = Some(latest.track.clone());
-						}
+						s.fetching = Some(latest.track.clone());
+						let weak_self = self.weak_self.clone();
+						let f = move |r: Result<PathBuf, fetch::Error>| if let Some(this) = weak_self.upgrade().as_mut() { this.fetch_done(r) }};
+						fetcher.fetch(b, Box::new(f)).ok();
 					}
 				}
 			}
-			info!(target: "updater", "Fork: this/current/latest/latest-known: {}/#{}/#{}/#{}", match self.this_fork { Some(f) => format!("#{}", f), None => "unknown".into(), }, current_number, latest.track.fork, latest.fork);
+			info!(target: "updater", "Fork: this/current/latest/latest-known: {}/#{}/#{}/#{}", match s.latest.this_fork { Some(f) => format!("#{}", f), None => "unknown".into(), }, current_number, s.latest.track.fork, s.latest.fork);
 		}
+		(*self.state.lock()).latest = latest;
+	}
+}
+
+impl ChainNotify for Updater {
+	fn new_blocks(&self, _imported: Vec<H256>, _invalid: Vec<H256>, _enacted: Vec<H256>, _retracted: Vec<H256>, _sealed: Vec<H256>, duration: u64) {
+		// TODO: something like this
+//		if !self.client.upgrade().map_or(true, |c| c.is_major_syncing()) {
+			self.poll();
+//		}
+	}
+}
+
+impl fetch::urlhint::ContractClient for Updater {
+	fn registrar(&self) -> Result<Address, String> {
+		self.client.upgrade().ok_or_else(|| "Client not available".to_owned())?
+			.registrar_address()
+			.ok_or_else(|| "Registrar not available".into())
+	}
+
+	fn call(&self, address: Address, data: Bytes) -> Result<Bytes, String> {
+		self.client.upgrade().ok_or_else(|| "Client not available".to_owned())?
+			.call_contract(address, data)
 	}
 }

util/src/log.rs

@@ -25,7 +25,7 @@ pub use ansi_term::{Colour, Style};
 use parking_lot::{RwLock, RwLockReadGuard};
 
 lazy_static! {
-	static ref LOG_DUMMY: bool = {
+	static ref LOG_DUMMY: () = {
 		let mut builder = LogBuilder::new();
 		builder.filter(None, LogLevelFilter::Info);
 
@@ -36,13 +36,12 @@ lazy_static! {
 		if builder.init().is_ok() {
 			println!("logger initialized");
 		}
-		true
 	};
 }
 
 /// Intialize log with default settings
 pub fn init_log() {
-	let _ = *LOG_DUMMY;
+	*LOG_DUMMY
 }
 
 const LOG_SIZE : usize = 128;