313 lines
8.5 KiB
Rust
313 lines
8.5 KiB
Rust
// Copyright 2015-2020 Parity Technologies (UK) Ltd.
|
|
// This file is part of OpenEthereum.
|
|
|
|
// OpenEthereum 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.
|
|
|
|
// OpenEthereum 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 OpenEthereum. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
//! DB backend wrapper for Account trie
|
|
use ethereum_types::H256;
|
|
use hash::{keccak, KECCAK_NULL_RLP};
|
|
use hash_db::{AsHashDB, HashDB};
|
|
use keccak_hasher::KeccakHasher;
|
|
use kvdb::DBValue;
|
|
use rlp::NULL_RLP;
|
|
|
|
#[cfg(test)]
|
|
use ethereum_types::Address;
|
|
|
|
// combines a key with an address hash to ensure uniqueness.
|
|
// leaves the first 96 bits untouched in order to support partial key lookup.
|
|
#[inline]
|
|
fn combine_key<'a>(address_hash: &'a H256, key: &'a H256) -> H256 {
|
|
let mut dst = key.clone();
|
|
{
|
|
let last_src: &[u8] = &*address_hash;
|
|
let last_dst: &mut [u8] = &mut *dst;
|
|
for (k, a) in last_dst[12..].iter_mut().zip(&last_src[12..]) {
|
|
*k ^= *a
|
|
}
|
|
}
|
|
|
|
dst
|
|
}
|
|
|
|
/// A factory for different kinds of account dbs.
|
|
#[derive(Debug, Clone)]
|
|
pub enum Factory {
|
|
/// Mangle hashes based on address. This is the default.
|
|
Mangled,
|
|
/// Don't mangle hashes.
|
|
Plain,
|
|
}
|
|
|
|
impl Default for Factory {
|
|
fn default() -> Self {
|
|
Factory::Mangled
|
|
}
|
|
}
|
|
|
|
impl Factory {
|
|
/// Create a read-only accountdb.
|
|
/// This will panic when write operations are called.
|
|
pub fn readonly<'db>(
|
|
&self,
|
|
db: &'db dyn HashDB<KeccakHasher, DBValue>,
|
|
address_hash: H256,
|
|
) -> Box<dyn HashDB<KeccakHasher, DBValue> + 'db> {
|
|
match *self {
|
|
Factory::Mangled => Box::new(AccountDB::from_hash(db, address_hash)),
|
|
Factory::Plain => Box::new(Wrapping(db)),
|
|
}
|
|
}
|
|
|
|
/// Create a new mutable hashdb.
|
|
pub fn create<'db>(
|
|
&self,
|
|
db: &'db mut dyn HashDB<KeccakHasher, DBValue>,
|
|
address_hash: H256,
|
|
) -> Box<dyn HashDB<KeccakHasher, DBValue> + 'db> {
|
|
match *self {
|
|
Factory::Mangled => Box::new(AccountDBMut::from_hash(db, address_hash)),
|
|
Factory::Plain => Box::new(WrappingMut(db)),
|
|
}
|
|
}
|
|
}
|
|
|
|
// TODO: introduce HashDBMut?
|
|
/// DB backend wrapper for Account trie
|
|
/// Transforms trie node keys for the database
|
|
pub struct AccountDB<'db> {
|
|
db: &'db dyn HashDB<KeccakHasher, DBValue>,
|
|
address_hash: H256,
|
|
}
|
|
|
|
impl<'db> AccountDB<'db> {
|
|
/// Create a new AccountDB from an address.
|
|
#[cfg(test)]
|
|
pub fn new(db: &'db dyn HashDB<KeccakHasher, DBValue>, address: &Address) -> Self {
|
|
Self::from_hash(db, keccak(address))
|
|
}
|
|
|
|
/// Create a new AcountDB from an address' hash.
|
|
pub fn from_hash(db: &'db dyn HashDB<KeccakHasher, DBValue>, address_hash: H256) -> Self {
|
|
AccountDB {
|
|
db: db,
|
|
address_hash: address_hash,
|
|
}
|
|
}
|
|
}
|
|
|
|
impl<'db> AsHashDB<KeccakHasher, DBValue> for AccountDB<'db> {
|
|
fn as_hash_db(&self) -> &dyn HashDB<KeccakHasher, DBValue> {
|
|
self
|
|
}
|
|
fn as_hash_db_mut(&mut self) -> &mut dyn HashDB<KeccakHasher, DBValue> {
|
|
self
|
|
}
|
|
}
|
|
|
|
impl<'db> HashDB<KeccakHasher, DBValue> for AccountDB<'db> {
|
|
fn get(&self, key: &H256) -> Option<DBValue> {
|
|
if key == &KECCAK_NULL_RLP {
|
|
return Some(DBValue::from_slice(&NULL_RLP));
|
|
}
|
|
self.db.get(&combine_key(&self.address_hash, key))
|
|
}
|
|
|
|
fn contains(&self, key: &H256) -> bool {
|
|
if key == &KECCAK_NULL_RLP {
|
|
return true;
|
|
}
|
|
self.db.contains(&combine_key(&self.address_hash, key))
|
|
}
|
|
|
|
fn insert(&mut self, _value: &[u8]) -> H256 {
|
|
unimplemented!()
|
|
}
|
|
|
|
fn emplace(&mut self, _key: H256, _value: DBValue) {
|
|
unimplemented!()
|
|
}
|
|
|
|
fn remove(&mut self, _key: &H256) {
|
|
unimplemented!()
|
|
}
|
|
}
|
|
|
|
/// DB backend wrapper for Account trie
|
|
pub struct AccountDBMut<'db> {
|
|
db: &'db mut dyn HashDB<KeccakHasher, DBValue>,
|
|
address_hash: H256,
|
|
}
|
|
|
|
impl<'db> AccountDBMut<'db> {
|
|
/// Create a new AccountDB from an address.
|
|
#[cfg(test)]
|
|
pub fn new(db: &'db mut dyn HashDB<KeccakHasher, DBValue>, address: &Address) -> Self {
|
|
Self::from_hash(db, keccak(address))
|
|
}
|
|
|
|
/// Create a new AcountDB from an address' hash.
|
|
pub fn from_hash(db: &'db mut dyn HashDB<KeccakHasher, DBValue>, address_hash: H256) -> Self {
|
|
AccountDBMut {
|
|
db: db,
|
|
address_hash: address_hash,
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
pub fn immutable(&'db self) -> AccountDB<'db> {
|
|
AccountDB {
|
|
db: self.db,
|
|
address_hash: self.address_hash.clone(),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl<'db> HashDB<KeccakHasher, DBValue> for AccountDBMut<'db> {
|
|
fn get(&self, key: &H256) -> Option<DBValue> {
|
|
if key == &KECCAK_NULL_RLP {
|
|
return Some(DBValue::from_slice(&NULL_RLP));
|
|
}
|
|
self.db.get(&combine_key(&self.address_hash, key))
|
|
}
|
|
|
|
fn contains(&self, key: &H256) -> bool {
|
|
if key == &KECCAK_NULL_RLP {
|
|
return true;
|
|
}
|
|
self.db.contains(&combine_key(&self.address_hash, key))
|
|
}
|
|
|
|
fn insert(&mut self, value: &[u8]) -> H256 {
|
|
if value == &NULL_RLP {
|
|
return KECCAK_NULL_RLP.clone();
|
|
}
|
|
let k = keccak(value);
|
|
let ak = combine_key(&self.address_hash, &k);
|
|
self.db.emplace(ak, DBValue::from_slice(value));
|
|
k
|
|
}
|
|
|
|
fn emplace(&mut self, key: H256, value: DBValue) {
|
|
if key == KECCAK_NULL_RLP {
|
|
return;
|
|
}
|
|
let key = combine_key(&self.address_hash, &key);
|
|
self.db.emplace(key, value)
|
|
}
|
|
|
|
fn remove(&mut self, key: &H256) {
|
|
if key == &KECCAK_NULL_RLP {
|
|
return;
|
|
}
|
|
let key = combine_key(&self.address_hash, key);
|
|
self.db.remove(&key)
|
|
}
|
|
}
|
|
|
|
impl<'db> AsHashDB<KeccakHasher, DBValue> for AccountDBMut<'db> {
|
|
fn as_hash_db(&self) -> &dyn HashDB<KeccakHasher, DBValue> {
|
|
self
|
|
}
|
|
fn as_hash_db_mut(&mut self) -> &mut dyn HashDB<KeccakHasher, DBValue> {
|
|
self
|
|
}
|
|
}
|
|
|
|
struct Wrapping<'db>(&'db dyn HashDB<KeccakHasher, DBValue>);
|
|
|
|
impl<'db> AsHashDB<KeccakHasher, DBValue> for Wrapping<'db> {
|
|
fn as_hash_db(&self) -> &dyn HashDB<KeccakHasher, DBValue> {
|
|
self
|
|
}
|
|
fn as_hash_db_mut(&mut self) -> &mut dyn HashDB<KeccakHasher, DBValue> {
|
|
self
|
|
}
|
|
}
|
|
|
|
impl<'db> HashDB<KeccakHasher, DBValue> for Wrapping<'db> {
|
|
fn get(&self, key: &H256) -> Option<DBValue> {
|
|
if key == &KECCAK_NULL_RLP {
|
|
return Some(DBValue::from_slice(&NULL_RLP));
|
|
}
|
|
self.0.get(key)
|
|
}
|
|
|
|
fn contains(&self, key: &H256) -> bool {
|
|
if key == &KECCAK_NULL_RLP {
|
|
return true;
|
|
}
|
|
self.0.contains(key)
|
|
}
|
|
|
|
fn insert(&mut self, _value: &[u8]) -> H256 {
|
|
unimplemented!()
|
|
}
|
|
|
|
fn emplace(&mut self, _key: H256, _value: DBValue) {
|
|
unimplemented!()
|
|
}
|
|
|
|
fn remove(&mut self, _key: &H256) {
|
|
unimplemented!()
|
|
}
|
|
}
|
|
|
|
struct WrappingMut<'db>(&'db mut dyn HashDB<KeccakHasher, DBValue>);
|
|
impl<'db> AsHashDB<KeccakHasher, DBValue> for WrappingMut<'db> {
|
|
fn as_hash_db(&self) -> &dyn HashDB<KeccakHasher, DBValue> {
|
|
self
|
|
}
|
|
fn as_hash_db_mut(&mut self) -> &mut dyn HashDB<KeccakHasher, DBValue> {
|
|
self
|
|
}
|
|
}
|
|
|
|
impl<'db> HashDB<KeccakHasher, DBValue> for WrappingMut<'db> {
|
|
fn get(&self, key: &H256) -> Option<DBValue> {
|
|
if key == &KECCAK_NULL_RLP {
|
|
return Some(DBValue::from_slice(&NULL_RLP));
|
|
}
|
|
self.0.get(key)
|
|
}
|
|
|
|
fn contains(&self, key: &H256) -> bool {
|
|
if key == &KECCAK_NULL_RLP {
|
|
return true;
|
|
}
|
|
self.0.contains(key)
|
|
}
|
|
|
|
fn insert(&mut self, value: &[u8]) -> H256 {
|
|
if value == &NULL_RLP {
|
|
return KECCAK_NULL_RLP.clone();
|
|
}
|
|
self.0.insert(value)
|
|
}
|
|
|
|
fn emplace(&mut self, key: H256, value: DBValue) {
|
|
if key == KECCAK_NULL_RLP {
|
|
return;
|
|
}
|
|
self.0.emplace(key, value)
|
|
}
|
|
|
|
fn remove(&mut self, key: &H256) {
|
|
if key == &KECCAK_NULL_RLP {
|
|
return;
|
|
}
|
|
self.0.remove(key)
|
|
}
|
|
}
|