124 lines
4.7 KiB
Rust
124 lines
4.7 KiB
Rust
// 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 util::*;
|
|
use ethcore::snapshot::{SnapshotService, ManifestData, RestorationStatus};
|
|
use ethcore::header::BlockNumber;
|
|
use ethcore::client::{EachBlockWith};
|
|
use super::helpers::*;
|
|
|
|
pub struct TestSnapshotService {
|
|
manifest: Option<ManifestData>,
|
|
chunks: HashMap<H256, Bytes>,
|
|
|
|
restoration_manifest: Mutex<Option<ManifestData>>,
|
|
state_restoration_chunks: Mutex<HashMap<H256, Bytes>>,
|
|
block_restoration_chunks: Mutex<HashMap<H256, Bytes>>,
|
|
}
|
|
|
|
impl TestSnapshotService {
|
|
pub fn new() -> TestSnapshotService {
|
|
TestSnapshotService {
|
|
manifest: None,
|
|
chunks: HashMap::new(),
|
|
restoration_manifest: Mutex::new(None),
|
|
state_restoration_chunks: Mutex::new(HashMap::new()),
|
|
block_restoration_chunks: Mutex::new(HashMap::new()),
|
|
}
|
|
}
|
|
|
|
pub fn new_with_snapshot(num_chunks: usize, block_hash: H256, block_number: BlockNumber) -> TestSnapshotService {
|
|
let num_state_chunks = num_chunks / 2;
|
|
let num_block_chunks = num_chunks - num_state_chunks;
|
|
let state_chunks: Vec<Bytes> = (0..num_state_chunks).map(|_| H256::random().to_vec()).collect();
|
|
let block_chunks: Vec<Bytes> = (0..num_block_chunks).map(|_| H256::random().to_vec()).collect();
|
|
let manifest = ManifestData {
|
|
state_hashes: state_chunks.iter().map(|data| data.sha3()).collect(),
|
|
block_hashes: block_chunks.iter().map(|data| data.sha3()).collect(),
|
|
state_root: H256::new(),
|
|
block_number: block_number,
|
|
block_hash: block_hash,
|
|
};
|
|
let mut chunks: HashMap<H256, Bytes> = state_chunks.into_iter().map(|data| (data.sha3(), data)).collect();
|
|
chunks.extend(block_chunks.into_iter().map(|data| (data.sha3(), data)));
|
|
TestSnapshotService {
|
|
manifest: Some(manifest),
|
|
chunks: chunks,
|
|
restoration_manifest: Mutex::new(None),
|
|
state_restoration_chunks: Mutex::new(HashMap::new()),
|
|
block_restoration_chunks: Mutex::new(HashMap::new()),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl SnapshotService for TestSnapshotService {
|
|
fn manifest(&self) -> Option<ManifestData> {
|
|
self.manifest.as_ref().cloned()
|
|
}
|
|
|
|
fn chunk(&self, hash: H256) -> Option<Bytes> {
|
|
self.chunks.get(&hash).cloned()
|
|
}
|
|
|
|
fn status(&self) -> RestorationStatus {
|
|
match *self.restoration_manifest.lock() {
|
|
Some(ref manifest) if self.state_restoration_chunks.lock().len() == manifest.state_hashes.len() &&
|
|
self.block_restoration_chunks.lock().len() == manifest.block_hashes.len() => RestorationStatus::Inactive,
|
|
Some(_) => RestorationStatus::Ongoing {
|
|
state_chunks_done: self.state_restoration_chunks.lock().len() as u32,
|
|
block_chunks_done: self.block_restoration_chunks.lock().len() as u32,
|
|
},
|
|
None => RestorationStatus::Inactive,
|
|
}
|
|
}
|
|
|
|
fn begin_restore(&self, manifest: ManifestData) {
|
|
*self.restoration_manifest.lock() = Some(manifest);
|
|
self.state_restoration_chunks.lock().clear();
|
|
self.block_restoration_chunks.lock().clear();
|
|
}
|
|
|
|
fn abort_restore(&self) {
|
|
*self.restoration_manifest.lock() = None;
|
|
self.state_restoration_chunks.lock().clear();
|
|
self.block_restoration_chunks.lock().clear();
|
|
}
|
|
|
|
fn restore_state_chunk(&self, hash: H256, chunk: Bytes) {
|
|
if self.restoration_manifest.lock().as_ref().map_or(false, |m| m.state_hashes.iter().any(|h| h == &hash)) {
|
|
self.state_restoration_chunks.lock().insert(hash, chunk);
|
|
}
|
|
}
|
|
|
|
fn restore_block_chunk(&self, hash: H256, chunk: Bytes) {
|
|
if self.restoration_manifest.lock().as_ref().map_or(false, |m| m.block_hashes.iter().any(|h| h == &hash)) {
|
|
self.block_restoration_chunks.lock().insert(hash, chunk);
|
|
}
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn snapshot_sync() {
|
|
::env_logger::init().ok();
|
|
let mut net = TestNet::new(2);
|
|
net.peer_mut(0).snapshot_service = Arc::new(TestSnapshotService::new_with_snapshot(16, H256::new(), 1));
|
|
net.peer_mut(0).chain.add_blocks(1, EachBlockWith::Nothing);
|
|
net.sync_steps(19); // status + manifest + chunks
|
|
assert_eq!(net.peer(1).snapshot_service.state_restoration_chunks.lock().len(), net.peer(0).snapshot_service.manifest.as_ref().unwrap().state_hashes.len());
|
|
assert_eq!(net.peer(1).snapshot_service.block_restoration_chunks.lock().len(), net.peer(0).snapshot_service.manifest.as_ref().unwrap().block_hashes.len());
|
|
}
|
|
|