openethereum/ethcore/src/engines/signer.rs

// 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/>.

//! A signer used by Engines which need to sign messages.

use util::{Arc, Mutex, RwLock, H256, Address};
use ethkey::Signature;
use account_provider::{self, AccountProvider};

/// Everything that an Engine needs to sign messages.
pub struct EngineSigner {
	account_provider: Mutex<Arc<AccountProvider>>,
	address: RwLock<Address>,
	password: RwLock<Option<String>>,
}

impl Default for EngineSigner {
	fn default() -> Self {
		EngineSigner {
			account_provider: Mutex::new(Arc::new(AccountProvider::transient_provider())),
			address: Default::default(),
			password: Default::default(),
		}
	}
}

impl EngineSigner {
	/// Set up the signer to sign with given address and password.
	pub fn set(&self, ap: Arc<AccountProvider>, address: Address, password: String) {
		*self.account_provider.lock() = ap;
		*self.address.write()	= address;
		*self.password.write() = Some(password);
		debug!(target: "poa", "Setting Engine signer to {}", address);
	}

	/// Sign a consensus message hash.
	pub fn sign(&self, hash: H256) -> Result<Signature, account_provider::Error> {
		self.account_provider.lock().sign(*self.address.read(), self.password.read().clone(), hash)
	}

	/// Signing address.
	pub fn address(&self) -> Address {
		self.address.read().clone()
	}

	/// Check if the given address is the signing address.
	pub fn is_address(&self, address: &Address) -> bool {
		*self.address.read() == *address
	}
}