1005 lines
38 KiB
Rust
1005 lines
38 KiB
Rust
// Copyright 2015-2019 Parity Technologies (UK) Ltd.
|
|
// This file is part of Parity Ethereum.
|
|
|
|
// Parity Ethereum 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 Ethereum 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 Ethereum. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
//! Private transactions module.
|
|
|
|
// Recursion limit required because of
|
|
// error_chain foreign_links.
|
|
#![recursion_limit = "256"]
|
|
|
|
mod encryptor;
|
|
mod error;
|
|
mod key_server_keys;
|
|
mod messages;
|
|
mod private_transactions;
|
|
|
|
extern crate common_types as types;
|
|
extern crate ethabi;
|
|
extern crate ethcore;
|
|
extern crate ethcore_call_contract as call_contract;
|
|
extern crate ethcore_io as io;
|
|
extern crate ethcore_miner;
|
|
extern crate ethereum_types;
|
|
extern crate ethjson;
|
|
extern crate ethkey;
|
|
extern crate fetch;
|
|
extern crate futures;
|
|
extern crate heapsize;
|
|
extern crate keccak_hash as hash;
|
|
extern crate parity_bytes as bytes;
|
|
extern crate parity_crypto as crypto;
|
|
extern crate parking_lot;
|
|
extern crate patricia_trie_ethereum as ethtrie;
|
|
extern crate rlp;
|
|
extern crate rustc_hex;
|
|
extern crate transaction_pool as txpool;
|
|
extern crate trie_db as trie;
|
|
extern crate url;
|
|
#[macro_use]
|
|
extern crate log;
|
|
#[macro_use]
|
|
extern crate ethabi_derive;
|
|
#[macro_use]
|
|
extern crate ethabi_contract;
|
|
extern crate derive_more;
|
|
#[macro_use]
|
|
extern crate rlp_derive;
|
|
|
|
#[cfg(test)]
|
|
extern crate env_logger;
|
|
#[cfg(test)]
|
|
extern crate rand;
|
|
|
|
pub use encryptor::{Encryptor, EncryptorConfig, NoopEncryptor, SecretStoreEncryptor};
|
|
pub use error::Error;
|
|
pub use key_server_keys::{KeyProvider, SecretStoreKeys, StoringKeyProvider};
|
|
pub use messages::{PrivateTransaction, SignedPrivateTransaction};
|
|
pub use private_transactions::{
|
|
PrivateTransactionSigningDesc, SigningStore, VerificationStore, VerifiedPrivateTransaction,
|
|
};
|
|
|
|
use bytes::Bytes;
|
|
use call_contract::CallContract;
|
|
use ethabi::FunctionOutputDecoder;
|
|
use ethcore::{
|
|
client::{
|
|
BlockId, BlockInfo, Call, ChainMessageType, ChainNotify, Client, ClientIoMessage, NewBlocks,
|
|
},
|
|
contract_address as ethcore_contract_address,
|
|
executed::Executed,
|
|
executive::{Executive, TransactOptions},
|
|
miner::{self, pool_client::NonceCache, Miner, MinerService},
|
|
state, state_db,
|
|
trace::{Tracer, VMTracer},
|
|
};
|
|
use ethereum_types::{Address, H128, H256, U256};
|
|
use ethkey::{public_to_address, recover, Signature};
|
|
use hash::keccak;
|
|
use io::IoChannel;
|
|
use parking_lot::RwLock;
|
|
use rlp::*;
|
|
use rustc_hex::FromHex;
|
|
use std::{
|
|
collections::{BTreeMap, HashMap, HashSet},
|
|
sync::{Arc, Weak},
|
|
};
|
|
use types::transaction::{Action, SignedTransaction, Transaction, UnverifiedTransaction};
|
|
|
|
// Source avaiable at https://github.com/parity-contracts/private-tx/blob/master/contracts/PrivateContract.sol
|
|
const DEFAULT_STUB_CONTRACT: &'static str = include_str!("../res/private.evm");
|
|
|
|
use_contract!(private_contract, "res/private.json");
|
|
|
|
/// Initialization vector length.
|
|
const INIT_VEC_LEN: usize = 16;
|
|
|
|
/// Size of nonce cache
|
|
const NONCE_CACHE_SIZE: usize = 128;
|
|
|
|
/// Version for the initial private contract wrapper
|
|
const INITIAL_PRIVATE_CONTRACT_VER: usize = 1;
|
|
|
|
/// Version for the private contract notification about private state changes added
|
|
const PRIVATE_CONTRACT_WITH_NOTIFICATION_VER: usize = 2;
|
|
|
|
/// Configurtion for private transaction provider
|
|
#[derive(Default, PartialEq, Debug, Clone)]
|
|
pub struct ProviderConfig {
|
|
/// Accounts that can be used for validation
|
|
pub validator_accounts: Vec<Address>,
|
|
/// Account used for signing public transactions created from private transactions
|
|
pub signer_account: Option<Address>,
|
|
}
|
|
|
|
#[derive(Debug)]
|
|
/// Private transaction execution receipt.
|
|
pub struct Receipt {
|
|
/// Private transaction hash.
|
|
pub hash: H256,
|
|
/// Contract address.
|
|
pub contract_address: Address,
|
|
/// Execution status.
|
|
pub status_code: u8,
|
|
}
|
|
|
|
/// Payload signing and decrypting capabilities.
|
|
pub trait Signer: Send + Sync {
|
|
/// Decrypt payload using private key of given address.
|
|
fn decrypt(
|
|
&self,
|
|
account: Address,
|
|
shared_mac: &[u8],
|
|
payload: &[u8],
|
|
) -> Result<Vec<u8>, Error>;
|
|
/// Sign given hash using provided account.
|
|
fn sign(&self, account: Address, hash: ethkey::Message) -> Result<Signature, Error>;
|
|
}
|
|
|
|
/// Signer implementation that errors on any request.
|
|
pub struct DummySigner;
|
|
impl Signer for DummySigner {
|
|
fn decrypt(
|
|
&self,
|
|
_account: Address,
|
|
_shared_mac: &[u8],
|
|
_payload: &[u8],
|
|
) -> Result<Vec<u8>, Error> {
|
|
Err("Decrypting is not supported.".to_owned())?
|
|
}
|
|
|
|
fn sign(&self, _account: Address, _hash: ethkey::Message) -> Result<Signature, Error> {
|
|
Err("Signing is not supported.".to_owned())?
|
|
}
|
|
}
|
|
|
|
/// Signer implementation using multiple keypairs
|
|
pub struct KeyPairSigner(pub Vec<ethkey::KeyPair>);
|
|
impl Signer for KeyPairSigner {
|
|
fn decrypt(
|
|
&self,
|
|
account: Address,
|
|
shared_mac: &[u8],
|
|
payload: &[u8],
|
|
) -> Result<Vec<u8>, Error> {
|
|
let kp = self
|
|
.0
|
|
.iter()
|
|
.find(|k| k.address() == account)
|
|
.ok_or(ethkey::Error::InvalidAddress)?;
|
|
Ok(ethkey::crypto::ecies::decrypt(
|
|
kp.secret(),
|
|
shared_mac,
|
|
payload,
|
|
)?)
|
|
}
|
|
|
|
fn sign(&self, account: Address, hash: ethkey::Message) -> Result<Signature, Error> {
|
|
let kp = self
|
|
.0
|
|
.iter()
|
|
.find(|k| k.address() == account)
|
|
.ok_or(ethkey::Error::InvalidAddress)?;
|
|
Ok(ethkey::sign(kp.secret(), &hash)?)
|
|
}
|
|
}
|
|
|
|
/// Manager of private transactions
|
|
pub struct Provider {
|
|
encryptor: Box<Encryptor>,
|
|
validator_accounts: HashSet<Address>,
|
|
signer_account: Option<Address>,
|
|
notify: RwLock<Vec<Weak<ChainNotify>>>,
|
|
transactions_for_signing: RwLock<SigningStore>,
|
|
transactions_for_verification: VerificationStore,
|
|
client: Arc<Client>,
|
|
miner: Arc<Miner>,
|
|
accounts: Arc<Signer>,
|
|
channel: IoChannel<ClientIoMessage>,
|
|
keys_provider: Arc<KeyProvider>,
|
|
}
|
|
|
|
#[derive(Debug)]
|
|
pub struct PrivateExecutionResult<T, V>
|
|
where
|
|
T: Tracer,
|
|
V: VMTracer,
|
|
{
|
|
code: Option<Bytes>,
|
|
state: Bytes,
|
|
contract_address: Address,
|
|
result: Executed<T::Output, V::Output>,
|
|
}
|
|
|
|
impl Provider {
|
|
/// Create a new provider.
|
|
pub fn new(
|
|
client: Arc<Client>,
|
|
miner: Arc<Miner>,
|
|
accounts: Arc<Signer>,
|
|
encryptor: Box<Encryptor>,
|
|
config: ProviderConfig,
|
|
channel: IoChannel<ClientIoMessage>,
|
|
keys_provider: Arc<KeyProvider>,
|
|
) -> Self {
|
|
keys_provider.update_acl_contract();
|
|
Provider {
|
|
encryptor,
|
|
validator_accounts: config.validator_accounts.into_iter().collect(),
|
|
signer_account: config.signer_account,
|
|
notify: RwLock::default(),
|
|
transactions_for_signing: RwLock::default(),
|
|
transactions_for_verification: VerificationStore::default(),
|
|
client,
|
|
miner,
|
|
accounts,
|
|
channel,
|
|
keys_provider,
|
|
}
|
|
}
|
|
|
|
// TODO [ToDr] Don't use `ChainNotify` here!
|
|
// Better to create a separate notification type for this.
|
|
/// Adds an actor to be notified on certain events
|
|
pub fn add_notify(&self, target: Arc<ChainNotify>) {
|
|
self.notify.write().push(Arc::downgrade(&target));
|
|
}
|
|
|
|
fn notify<F>(&self, f: F)
|
|
where
|
|
F: Fn(&ChainNotify),
|
|
{
|
|
for np in self.notify.read().iter() {
|
|
if let Some(n) = np.upgrade() {
|
|
f(&*n);
|
|
}
|
|
}
|
|
}
|
|
|
|
/// 1. Create private transaction from the signed transaction
|
|
/// 2. Executes private transaction
|
|
/// 3. Save it with state returned on prev step to the queue for signing
|
|
/// 4. Broadcast corresponding message to the chain
|
|
pub fn create_private_transaction(
|
|
&self,
|
|
signed_transaction: SignedTransaction,
|
|
) -> Result<Receipt, Error> {
|
|
trace!(target: "privatetx", "Creating private transaction from regular transaction: {:?}", signed_transaction);
|
|
if self.signer_account.is_none() {
|
|
warn!(target: "privatetx", "Signing account not set");
|
|
return Err(Error::SignerAccountNotSet);
|
|
}
|
|
let tx_hash = signed_transaction.hash();
|
|
let contract = Self::contract_address_from_transaction(&signed_transaction)
|
|
.map_err(|_| Error::BadTransactionType)?;
|
|
let data = signed_transaction.rlp_bytes();
|
|
let encrypted_transaction = self.encrypt(
|
|
&contract,
|
|
&Self::iv_from_transaction(&signed_transaction),
|
|
&data,
|
|
)?;
|
|
let private = PrivateTransaction::new(encrypted_transaction, contract);
|
|
// TODO #9825 [ToDr] Using BlockId::Latest is bad here,
|
|
// the block may change in the middle of execution
|
|
// causing really weird stuff to happen.
|
|
// We should retrieve hash and stick to that. IMHO
|
|
// best would be to change the API and only allow H256 instead of BlockID
|
|
// in private-tx to avoid such mistakes.
|
|
let contract_nonce = self.get_contract_nonce(&contract, BlockId::Latest)?;
|
|
let private_state =
|
|
self.execute_private_transaction(BlockId::Latest, &signed_transaction)?;
|
|
trace!(target: "privatetx", "Private transaction created, encrypted transaction: {:?}, private state: {:?}", private, private_state);
|
|
let contract_validators = self.get_validators(BlockId::Latest, &contract)?;
|
|
trace!(target: "privatetx", "Required validators: {:?}", contract_validators);
|
|
let private_state_hash = self.calculate_state_hash(&private_state, contract_nonce);
|
|
trace!(target: "privatetx", "Hashed effective private state for sender: {:?}", private_state_hash);
|
|
self.transactions_for_signing.write().add_transaction(
|
|
private.hash(),
|
|
signed_transaction,
|
|
contract_validators,
|
|
private_state,
|
|
contract_nonce,
|
|
)?;
|
|
self.broadcast_private_transaction(private.hash(), private.rlp_bytes());
|
|
Ok(Receipt {
|
|
hash: tx_hash,
|
|
contract_address: contract,
|
|
status_code: 0,
|
|
})
|
|
}
|
|
|
|
/// Calculate hash from united private state and contract nonce
|
|
pub fn calculate_state_hash(&self, state: &Bytes, nonce: U256) -> H256 {
|
|
let state_hash = keccak(state);
|
|
let mut state_buf = [0u8; 64];
|
|
state_buf[..32].clone_from_slice(&state_hash);
|
|
state_buf[32..].clone_from_slice(&H256::from(nonce));
|
|
keccak(AsRef::<[u8]>::as_ref(&state_buf[..]))
|
|
}
|
|
|
|
fn pool_client<'a>(
|
|
&'a self,
|
|
nonce_cache: &'a NonceCache,
|
|
local_accounts: &'a HashSet<Address>,
|
|
) -> miner::pool_client::PoolClient<'a, Client> {
|
|
let engine = self.client.engine();
|
|
miner::pool_client::PoolClient::new(
|
|
&*self.client,
|
|
nonce_cache,
|
|
engine,
|
|
local_accounts,
|
|
None, // refuse_service_transactions = true
|
|
)
|
|
}
|
|
|
|
/// Retrieve and verify the first available private transaction for every sender
|
|
fn process_verification_queue(&self) -> Result<(), Error> {
|
|
let process_transaction = |transaction: &VerifiedPrivateTransaction| -> Result<_, String> {
|
|
let private_hash = transaction.private_transaction.hash();
|
|
match transaction.validator_account {
|
|
None => {
|
|
trace!(target: "privatetx", "Propagating transaction further");
|
|
self.broadcast_private_transaction(
|
|
private_hash,
|
|
transaction.private_transaction.rlp_bytes(),
|
|
);
|
|
return Ok(());
|
|
}
|
|
Some(validator_account) => {
|
|
if !self.validator_accounts.contains(&validator_account) {
|
|
trace!(target: "privatetx", "Propagating transaction further");
|
|
self.broadcast_private_transaction(
|
|
private_hash,
|
|
transaction.private_transaction.rlp_bytes(),
|
|
);
|
|
return Ok(());
|
|
}
|
|
let contract =
|
|
Self::contract_address_from_transaction(&transaction.transaction)
|
|
.map_err(|_| "Incorrect type of action for the transaction")?;
|
|
// TODO #9825 [ToDr] Usage of BlockId::Latest
|
|
let contract_nonce = self.get_contract_nonce(&contract, BlockId::Latest);
|
|
if let Err(e) = contract_nonce {
|
|
return Err(format!("Cannot retrieve contract nonce: {:?}", e).into());
|
|
}
|
|
let contract_nonce = contract_nonce.expect("Error was checked before");
|
|
let private_state =
|
|
self.execute_private_transaction(BlockId::Latest, &transaction.transaction);
|
|
if let Err(e) = private_state {
|
|
return Err(format!("Cannot retrieve private state: {:?}", e).into());
|
|
}
|
|
let private_state = private_state.expect("Error was checked before");
|
|
let private_state_hash =
|
|
self.calculate_state_hash(&private_state, contract_nonce);
|
|
trace!(target: "privatetx", "Hashed effective private state for validator: {:?}", private_state_hash);
|
|
let signed_state = self.accounts.sign(validator_account, private_state_hash);
|
|
if let Err(e) = signed_state {
|
|
return Err(format!("Cannot sign the state: {:?}", e).into());
|
|
}
|
|
let signed_state = signed_state.expect("Error was checked before");
|
|
let signed_private_transaction =
|
|
SignedPrivateTransaction::new(private_hash, signed_state, None);
|
|
trace!(target: "privatetx", "Sending signature for private transaction: {:?}", signed_private_transaction);
|
|
self.broadcast_signed_private_transaction(
|
|
signed_private_transaction.hash(),
|
|
signed_private_transaction.rlp_bytes(),
|
|
);
|
|
}
|
|
}
|
|
Ok(())
|
|
};
|
|
let nonce_cache = NonceCache::new(NONCE_CACHE_SIZE);
|
|
let local_accounts = HashSet::new();
|
|
let ready_transactions = self
|
|
.transactions_for_verification
|
|
.drain(self.pool_client(&nonce_cache, &local_accounts));
|
|
for transaction in ready_transactions {
|
|
if let Err(e) = process_transaction(&transaction) {
|
|
warn!(target: "privatetx", "Error: {:?}", e);
|
|
}
|
|
}
|
|
Ok(())
|
|
}
|
|
|
|
/// Add signed private transaction into the store
|
|
/// Creates corresponding public transaction if last required signature collected and sends it to the chain
|
|
pub fn process_signature(&self, signed_tx: &SignedPrivateTransaction) -> Result<(), Error> {
|
|
trace!(target: "privatetx", "Processing signed private transaction");
|
|
let private_hash = signed_tx.private_transaction_hash();
|
|
let desc = match self.transactions_for_signing.read().get(&private_hash) {
|
|
None => {
|
|
// Not our transaction, broadcast further to peers
|
|
self.broadcast_signed_private_transaction(signed_tx.hash(), signed_tx.rlp_bytes());
|
|
return Ok(());
|
|
}
|
|
Some(desc) => desc,
|
|
};
|
|
let last = self.last_required_signature(&desc, signed_tx.signature())?;
|
|
|
|
if last {
|
|
let mut signatures = desc.received_signatures.clone();
|
|
signatures.push(signed_tx.signature());
|
|
let rsv: Vec<Signature> = signatures
|
|
.into_iter()
|
|
.map(|sign| sign.into_electrum().into())
|
|
.collect();
|
|
// Create public transaction
|
|
let signer_account = self
|
|
.signer_account
|
|
.ok_or_else(|| Error::SignerAccountNotSet)?;
|
|
let state = self
|
|
.client
|
|
.state_at(BlockId::Latest)
|
|
.ok_or(Error::StatePruned)?;
|
|
let nonce = state.nonce(&signer_account)?;
|
|
let public_tx = self.public_transaction(
|
|
desc.state.clone(),
|
|
&desc.original_transaction,
|
|
&rsv,
|
|
nonce,
|
|
desc.original_transaction.gas_price,
|
|
)?;
|
|
trace!(target: "privatetx", "Last required signature received, public transaction created: {:?}", public_tx);
|
|
// Sign and add it to the queue
|
|
let chain_id = desc.original_transaction.chain_id();
|
|
let hash = public_tx.hash(chain_id);
|
|
let signature = self.accounts.sign(signer_account, hash)?;
|
|
let signed = SignedTransaction::new(public_tx.with_signature(signature, chain_id))?;
|
|
match self
|
|
.miner
|
|
.import_own_transaction(&*self.client, signed.into())
|
|
{
|
|
Ok(_) => trace!(target: "privatetx", "Public transaction added to queue"),
|
|
Err(err) => {
|
|
warn!(target: "privatetx", "Failed to add transaction to queue, error: {:?}", err);
|
|
return Err(err.into());
|
|
}
|
|
}
|
|
// Notify about state changes
|
|
let contract = Self::contract_address_from_transaction(&desc.original_transaction)?;
|
|
// TODO #9825 Usage of BlockId::Latest
|
|
if self.get_contract_version(BlockId::Latest, &contract)
|
|
>= PRIVATE_CONTRACT_WITH_NOTIFICATION_VER
|
|
{
|
|
match self.state_changes_notify(
|
|
BlockId::Latest,
|
|
&contract,
|
|
&desc.original_transaction.sender(),
|
|
desc.original_transaction.hash(),
|
|
) {
|
|
Ok(_) => {
|
|
trace!(target: "privatetx", "Notification about private state changes sent")
|
|
}
|
|
Err(err) => {
|
|
warn!(target: "privatetx", "Failed to send private state changed notification, error: {:?}", err)
|
|
}
|
|
}
|
|
}
|
|
// Remove from store for signing
|
|
if let Err(err) = self.transactions_for_signing.write().remove(&private_hash) {
|
|
warn!(target: "privatetx", "Failed to remove transaction from signing store, error: {:?}", err);
|
|
return Err(err);
|
|
}
|
|
} else {
|
|
// Add signature to the store
|
|
match self
|
|
.transactions_for_signing
|
|
.write()
|
|
.add_signature(&private_hash, signed_tx.signature())
|
|
{
|
|
Ok(_) => trace!(target: "privatetx", "Signature stored for private transaction"),
|
|
Err(err) => {
|
|
warn!(target: "privatetx", "Failed to add signature to signing store, error: {:?}", err);
|
|
return Err(err);
|
|
}
|
|
}
|
|
}
|
|
Ok(())
|
|
}
|
|
|
|
fn contract_address_from_transaction(
|
|
transaction: &SignedTransaction,
|
|
) -> Result<Address, Error> {
|
|
match transaction.action {
|
|
Action::Call(contract) => Ok(contract),
|
|
_ => {
|
|
warn!(target: "privatetx", "Incorrect type of action for the transaction");
|
|
return Err(Error::BadTransactionType);
|
|
}
|
|
}
|
|
}
|
|
|
|
fn last_required_signature(
|
|
&self,
|
|
desc: &PrivateTransactionSigningDesc,
|
|
sign: Signature,
|
|
) -> Result<bool, Error> {
|
|
if desc.received_signatures.contains(&sign) {
|
|
return Ok(false);
|
|
}
|
|
let state_hash = self.calculate_state_hash(&desc.state, desc.contract_nonce);
|
|
match recover(&sign, &state_hash) {
|
|
Ok(public) => {
|
|
let sender = public_to_address(&public);
|
|
match desc.validators.contains(&sender) {
|
|
true => Ok(desc.received_signatures.len() + 1 == desc.validators.len()),
|
|
false => {
|
|
warn!(target: "privatetx", "Sender's state doesn't correspond to validator's");
|
|
return Err(Error::StateIncorrect);
|
|
}
|
|
}
|
|
}
|
|
Err(err) => {
|
|
warn!(target: "privatetx", "Sender's state doesn't correspond to validator's, error {:?}", err);
|
|
return Err(err.into());
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Broadcast the private transaction message to the chain
|
|
fn broadcast_private_transaction(&self, transaction_hash: H256, message: Bytes) {
|
|
self.notify(|notify| {
|
|
notify.broadcast(ChainMessageType::PrivateTransaction(
|
|
transaction_hash,
|
|
message.clone(),
|
|
))
|
|
});
|
|
}
|
|
|
|
/// Broadcast signed private transaction message to the chain
|
|
fn broadcast_signed_private_transaction(&self, transaction_hash: H256, message: Bytes) {
|
|
self.notify(|notify| {
|
|
notify.broadcast(ChainMessageType::SignedPrivateTransaction(
|
|
transaction_hash,
|
|
message.clone(),
|
|
))
|
|
});
|
|
}
|
|
|
|
fn iv_from_transaction(transaction: &SignedTransaction) -> H128 {
|
|
let nonce = keccak(&transaction.nonce.rlp_bytes());
|
|
let (iv, _) = nonce.split_at(INIT_VEC_LEN);
|
|
H128::from_slice(iv)
|
|
}
|
|
|
|
fn iv_from_address(contract_address: &Address) -> H128 {
|
|
let address = keccak(&contract_address.rlp_bytes());
|
|
let (iv, _) = address.split_at(INIT_VEC_LEN);
|
|
H128::from_slice(iv)
|
|
}
|
|
|
|
fn encrypt(
|
|
&self,
|
|
contract_address: &Address,
|
|
initialisation_vector: &H128,
|
|
data: &[u8],
|
|
) -> Result<Bytes, Error> {
|
|
trace!(target: "privatetx", "Encrypt data using key(address): {:?}", contract_address);
|
|
Ok(self
|
|
.encryptor
|
|
.encrypt(contract_address, initialisation_vector, data)?)
|
|
}
|
|
|
|
fn decrypt(&self, contract_address: &Address, data: &[u8]) -> Result<Bytes, Error> {
|
|
trace!(target: "privatetx", "Decrypt data using key(address): {:?}", contract_address);
|
|
Ok(self.encryptor.decrypt(contract_address, data)?)
|
|
}
|
|
|
|
fn get_decrypted_state(&self, address: &Address, block: BlockId) -> Result<Bytes, Error> {
|
|
let (data, decoder) = private_contract::functions::state::call();
|
|
let value = self.client.call_contract(block, *address, data)?;
|
|
let state = decoder
|
|
.decode(&value)
|
|
.map_err(|e| Error::Call(format!("Contract call failed {:?}", e)))?;
|
|
self.decrypt(address, &state)
|
|
}
|
|
|
|
fn get_decrypted_code(&self, address: &Address, block: BlockId) -> Result<Bytes, Error> {
|
|
let (data, decoder) = private_contract::functions::code::call();
|
|
let value = self.client.call_contract(block, *address, data)?;
|
|
let state = decoder
|
|
.decode(&value)
|
|
.map_err(|e| Error::Call(format!("Contract call failed {:?}", e)))?;
|
|
self.decrypt(address, &state)
|
|
}
|
|
|
|
pub fn get_contract_nonce(&self, address: &Address, block: BlockId) -> Result<U256, Error> {
|
|
let (data, decoder) = private_contract::functions::nonce::call();
|
|
let value = self.client.call_contract(block, *address, data)?;
|
|
decoder
|
|
.decode(&value)
|
|
.map_err(|e| Error::Call(format!("Contract call failed {:?}", e)).into())
|
|
}
|
|
|
|
fn snapshot_to_storage(raw: Bytes) -> HashMap<H256, H256> {
|
|
let items = raw.len() / 64;
|
|
(0..items)
|
|
.map(|i| {
|
|
let offset = i * 64;
|
|
let key = H256::from_slice(&raw[offset..(offset + 32)]);
|
|
let value = H256::from_slice(&raw[(offset + 32)..(offset + 64)]);
|
|
(key, value)
|
|
})
|
|
.collect()
|
|
}
|
|
|
|
fn snapshot_from_storage(storage: &HashMap<H256, H256>) -> Bytes {
|
|
let mut raw = Vec::with_capacity(storage.len() * 64);
|
|
// Sort the storage to guarantee the order for all parties
|
|
let sorted_storage: BTreeMap<&H256, &H256> = storage.iter().collect();
|
|
for (key, value) in sorted_storage {
|
|
raw.extend_from_slice(key);
|
|
raw.extend_from_slice(value);
|
|
}
|
|
raw
|
|
}
|
|
|
|
fn patch_account_state(
|
|
&self,
|
|
contract_address: &Address,
|
|
block: BlockId,
|
|
state: &mut state::State<state_db::StateDB>,
|
|
) -> Result<(), Error> {
|
|
let contract_code = Arc::new(self.get_decrypted_code(contract_address, block)?);
|
|
let contract_state = self.get_decrypted_state(contract_address, block)?;
|
|
trace!(target: "privatetx", "Patching contract at {:?}, code: {:?}, state: {:?}", contract_address, contract_code, contract_state);
|
|
state.patch_account(
|
|
contract_address,
|
|
contract_code,
|
|
Self::snapshot_to_storage(contract_state),
|
|
)?;
|
|
Ok(())
|
|
}
|
|
|
|
pub fn execute_private<T, V>(
|
|
&self,
|
|
transaction: &SignedTransaction,
|
|
options: TransactOptions<T, V>,
|
|
block: BlockId,
|
|
) -> Result<PrivateExecutionResult<T, V>, Error>
|
|
where
|
|
T: Tracer,
|
|
V: VMTracer,
|
|
{
|
|
let mut env_info = self.client.env_info(block).ok_or(Error::StatePruned)?;
|
|
env_info.gas_limit = transaction.gas;
|
|
|
|
let mut state = self.client.state_at(block).ok_or(Error::StatePruned)?;
|
|
// TODO #9825 in case of BlockId::Latest these need to operate on the same state
|
|
let contract_address = match transaction.action {
|
|
Action::Call(ref contract_address) => {
|
|
// Patch current contract state
|
|
self.patch_account_state(contract_address, block, &mut state)?;
|
|
Some(*contract_address)
|
|
}
|
|
Action::Create => None,
|
|
};
|
|
|
|
let engine = self.client.engine();
|
|
let sender = transaction.sender();
|
|
let nonce = state.nonce(&sender)?;
|
|
let contract_address = contract_address.unwrap_or_else(|| {
|
|
let (new_address, _) = ethcore_contract_address(
|
|
engine.create_address_scheme(env_info.number),
|
|
&sender,
|
|
&nonce,
|
|
&transaction.data,
|
|
);
|
|
new_address
|
|
});
|
|
// Patch other available private contracts' states as well
|
|
// TODO: #10133 patch only required for the contract states
|
|
if let Some(key_server_account) = self.keys_provider.key_server_account() {
|
|
if let Some(available_contracts) = self
|
|
.keys_provider
|
|
.available_keys(block, &key_server_account)
|
|
{
|
|
for private_contract in available_contracts {
|
|
if private_contract == contract_address {
|
|
continue;
|
|
}
|
|
self.patch_account_state(&private_contract, block, &mut state)?;
|
|
}
|
|
}
|
|
}
|
|
let machine = engine.machine();
|
|
let schedule = machine.schedule(env_info.number);
|
|
let result = Executive::new(&mut state, &env_info, &machine, &schedule)
|
|
.transact_virtual(transaction, options)?;
|
|
let (encrypted_code, encrypted_storage) = {
|
|
let (code, storage) = state.into_account(&contract_address)?;
|
|
trace!(target: "privatetx", "Private contract executed. code: {:?}, state: {:?}, result: {:?}", code, storage, result.output);
|
|
let enc_code = match code {
|
|
Some(c) => Some(self.encrypt(
|
|
&contract_address,
|
|
&Self::iv_from_address(&contract_address),
|
|
&c,
|
|
)?),
|
|
None => None,
|
|
};
|
|
(
|
|
enc_code,
|
|
self.encrypt(
|
|
&contract_address,
|
|
&Self::iv_from_transaction(transaction),
|
|
&Self::snapshot_from_storage(&storage),
|
|
)?,
|
|
)
|
|
};
|
|
Ok(PrivateExecutionResult {
|
|
code: encrypted_code,
|
|
state: encrypted_storage,
|
|
contract_address: contract_address,
|
|
result,
|
|
})
|
|
}
|
|
|
|
fn generate_constructor(validators: &[Address], code: Bytes, storage: Bytes) -> Bytes {
|
|
let constructor_code = DEFAULT_STUB_CONTRACT
|
|
.from_hex()
|
|
.expect("Default contract code is valid");
|
|
private_contract::constructor(
|
|
constructor_code,
|
|
validators.iter().map(|a| *a).collect::<Vec<Address>>(),
|
|
code,
|
|
storage,
|
|
)
|
|
}
|
|
|
|
fn generate_set_state_call(signatures: &[Signature], storage: Bytes) -> Bytes {
|
|
private_contract::functions::set_state::encode_input(
|
|
storage,
|
|
signatures
|
|
.iter()
|
|
.map(|s| {
|
|
let mut v: [u8; 32] = [0; 32];
|
|
v[31] = s.v();
|
|
v
|
|
})
|
|
.collect::<Vec<[u8; 32]>>(),
|
|
signatures.iter().map(|s| s.r()).collect::<Vec<&[u8]>>(),
|
|
signatures.iter().map(|s| s.s()).collect::<Vec<&[u8]>>(),
|
|
)
|
|
}
|
|
|
|
/// Returns the key from the key server associated with the contract
|
|
pub fn contract_key_id(&self, contract_address: &Address) -> Result<H256, Error> {
|
|
Ok(key_server_keys::address_to_key(contract_address))
|
|
}
|
|
|
|
/// Create encrypted public contract deployment transaction.
|
|
pub fn public_creation_transaction(
|
|
&self,
|
|
block: BlockId,
|
|
source: &SignedTransaction,
|
|
validators: &[Address],
|
|
gas_price: U256,
|
|
) -> Result<(Transaction, Address), Error> {
|
|
if let Action::Call(_) = source.action {
|
|
return Err(Error::BadTransactionType);
|
|
}
|
|
let sender = source.sender();
|
|
let state = self.client.state_at(block).ok_or(Error::StatePruned)?;
|
|
let nonce = state.nonce(&sender)?;
|
|
let executed = self.execute_private(source, TransactOptions::with_no_tracing(), block)?;
|
|
let header = self
|
|
.client
|
|
.block_header(block)
|
|
.ok_or(Error::StatePruned)
|
|
.and_then(|h| h.decode().map_err(|_| Error::StateIncorrect).into())?;
|
|
let (executed_code, executed_state) = (executed.code.unwrap_or_default(), executed.state);
|
|
let tx_data =
|
|
Self::generate_constructor(validators, executed_code.clone(), executed_state.clone());
|
|
let mut tx = Transaction {
|
|
nonce: nonce,
|
|
action: Action::Create,
|
|
gas: u64::max_value().into(),
|
|
gas_price: gas_price,
|
|
value: source.value,
|
|
data: tx_data,
|
|
};
|
|
tx.gas = match self
|
|
.client
|
|
.estimate_gas(&tx.clone().fake_sign(sender), &state, &header)
|
|
{
|
|
Ok(estimated_gas) => estimated_gas,
|
|
Err(_) => self.estimate_tx_gas(validators, &executed_code, &executed_state, &[]),
|
|
};
|
|
|
|
Ok((tx, executed.contract_address))
|
|
}
|
|
|
|
fn estimate_tx_gas(
|
|
&self,
|
|
validators: &[Address],
|
|
code: &Bytes,
|
|
state: &Bytes,
|
|
signatures: &[Signature],
|
|
) -> U256 {
|
|
let default_gas = 650000
|
|
+ validators.len() as u64 * 30000
|
|
+ code.len() as u64 * 8000
|
|
+ signatures.len() as u64 * 50000
|
|
+ state.len() as u64 * 8000;
|
|
default_gas.into()
|
|
}
|
|
|
|
/// Create encrypted public contract deployment transaction. Returns updated encrypted state.
|
|
pub fn execute_private_transaction(
|
|
&self,
|
|
block: BlockId,
|
|
source: &SignedTransaction,
|
|
) -> Result<Bytes, Error> {
|
|
if let Action::Create = source.action {
|
|
return Err(Error::BadTransactionType);
|
|
}
|
|
let result = self.execute_private(source, TransactOptions::with_no_tracing(), block)?;
|
|
Ok(result.state)
|
|
}
|
|
|
|
/// Create encrypted public transaction from private transaction.
|
|
pub fn public_transaction(
|
|
&self,
|
|
state: Bytes,
|
|
source: &SignedTransaction,
|
|
signatures: &[Signature],
|
|
nonce: U256,
|
|
gas_price: U256,
|
|
) -> Result<Transaction, Error> {
|
|
let gas = self.estimate_tx_gas(&[], &Vec::new(), &state, signatures);
|
|
Ok(Transaction {
|
|
nonce: nonce,
|
|
action: source.action.clone(),
|
|
gas: gas.into(),
|
|
gas_price: gas_price,
|
|
value: 0.into(),
|
|
data: Self::generate_set_state_call(signatures, state),
|
|
})
|
|
}
|
|
|
|
/// Call into private contract.
|
|
pub fn private_call(
|
|
&self,
|
|
block: BlockId,
|
|
transaction: &SignedTransaction,
|
|
) -> Result<Executed, Error> {
|
|
let result =
|
|
self.execute_private(transaction, TransactOptions::with_no_tracing(), block)?;
|
|
Ok(result.result)
|
|
}
|
|
|
|
/// Returns private validators for a contract.
|
|
pub fn get_validators(&self, block: BlockId, address: &Address) -> Result<Vec<Address>, Error> {
|
|
let (data, decoder) = private_contract::functions::get_validators::call();
|
|
let value = self.client.call_contract(block, *address, data)?;
|
|
decoder
|
|
.decode(&value)
|
|
.map_err(|e| Error::Call(format!("Contract call failed {:?}", e)).into())
|
|
}
|
|
|
|
fn get_contract_version(&self, block: BlockId, address: &Address) -> usize {
|
|
let (data, decoder) = private_contract::functions::get_version::call();
|
|
match self
|
|
.client
|
|
.call_contract(block, *address, data)
|
|
.and_then(|value| decoder.decode(&value).map_err(|e| e.to_string()))
|
|
{
|
|
Ok(version) => version.low_u64() as usize,
|
|
Err(_) => INITIAL_PRIVATE_CONTRACT_VER,
|
|
}
|
|
}
|
|
|
|
fn state_changes_notify(
|
|
&self,
|
|
block: BlockId,
|
|
address: &Address,
|
|
originator: &Address,
|
|
transaction_hash: H256,
|
|
) -> Result<(), Error> {
|
|
let (data, _) = private_contract::functions::notify_changes::call(
|
|
*originator,
|
|
transaction_hash.0.to_vec(),
|
|
);
|
|
let _value = self.client.call_contract(block, *address, data)?;
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
pub trait Importer {
|
|
/// Process received private transaction
|
|
fn import_private_transaction(&self, _rlp: &[u8]) -> Result<H256, Error>;
|
|
|
|
/// Add signed private transaction into the store
|
|
///
|
|
/// Creates corresponding public transaction if last required signature collected and sends it to the chain
|
|
fn import_signed_private_transaction(&self, _rlp: &[u8]) -> Result<H256, Error>;
|
|
}
|
|
|
|
// TODO [ToDr] Offload more heavy stuff to the IoService thread.
|
|
// It seems that a lot of heavy work (verification) is done in this thread anyway
|
|
// it might actually make sense to decouple it from clientService and just use dedicated thread
|
|
// for both verification and execution.
|
|
|
|
impl Importer for Arc<Provider> {
|
|
fn import_private_transaction(&self, rlp: &[u8]) -> Result<H256, Error> {
|
|
trace!(target: "privatetx", "Private transaction received");
|
|
let private_tx: PrivateTransaction = Rlp::new(rlp).as_val()?;
|
|
let private_tx_hash = private_tx.hash();
|
|
let contract = private_tx.contract();
|
|
let contract_validators = self.get_validators(BlockId::Latest, &contract)?;
|
|
|
|
let validation_account = contract_validators
|
|
.iter()
|
|
.find(|address| self.validator_accounts.contains(address));
|
|
|
|
// Extract the original transaction
|
|
let encrypted_data = private_tx.encrypted();
|
|
let transaction_bytes = self.decrypt(&contract, &encrypted_data)?;
|
|
let original_tx: UnverifiedTransaction = Rlp::new(&transaction_bytes).as_val()?;
|
|
let nonce_cache = NonceCache::new(NONCE_CACHE_SIZE);
|
|
let local_accounts = HashSet::new();
|
|
// Add to the queue for further verification
|
|
self.transactions_for_verification.add_transaction(
|
|
original_tx,
|
|
validation_account.map(|&account| account),
|
|
private_tx,
|
|
self.pool_client(&nonce_cache, &local_accounts),
|
|
)?;
|
|
let provider = Arc::downgrade(self);
|
|
let result = self.channel.send(ClientIoMessage::execute(move |_| {
|
|
if let Some(provider) = provider.upgrade() {
|
|
if let Err(e) = provider.process_verification_queue() {
|
|
warn!(target: "privatetx", "Unable to process the queue: {}", e);
|
|
}
|
|
}
|
|
}));
|
|
if let Err(e) = result {
|
|
warn!(target: "privatetx", "Error sending NewPrivateTransaction message: {:?}", e);
|
|
}
|
|
Ok(private_tx_hash)
|
|
}
|
|
|
|
fn import_signed_private_transaction(&self, rlp: &[u8]) -> Result<H256, Error> {
|
|
let tx: SignedPrivateTransaction = Rlp::new(rlp).as_val()?;
|
|
trace!(target: "privatetx", "Signature for private transaction received: {:?}", tx);
|
|
let private_hash = tx.private_transaction_hash();
|
|
let provider = Arc::downgrade(self);
|
|
let result = self.channel.send(ClientIoMessage::execute(move |_| {
|
|
if let Some(provider) = provider.upgrade() {
|
|
if let Err(e) = provider.process_signature(&tx) {
|
|
warn!(target: "privatetx", "Unable to process the signature: {}", e);
|
|
}
|
|
}
|
|
}));
|
|
if let Err(e) = result {
|
|
warn!(target: "privatetx", "Error sending NewSignedPrivateTransaction message: {:?}", e);
|
|
}
|
|
Ok(private_hash)
|
|
}
|
|
}
|
|
|
|
impl ChainNotify for Provider {
|
|
fn new_blocks(&self, new_blocks: NewBlocks) {
|
|
if new_blocks.imported.is_empty() || new_blocks.has_more_blocks_to_import {
|
|
return;
|
|
}
|
|
trace!(target: "privatetx", "New blocks imported, try to prune the queue");
|
|
if let Err(err) = self.process_verification_queue() {
|
|
warn!(target: "privatetx", "Cannot prune private transactions queue. error: {:?}", err);
|
|
}
|
|
self.keys_provider.update_acl_contract();
|
|
}
|
|
}
|