1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
use clap::ArgMatches;
use storage::BlockRef;
use config::Config;
use primitives::hash::H256;
use util::init_db;
pub fn rollback(cfg: Config, matches: &ArgMatches) -> Result<(), String> {
try!(init_db(&cfg));
let block_ref = matches.value_of("BLOCK").expect("BLOCK is required in cli.yml; qed");
let block_ref = if block_ref.len() == 64 {
BlockRef::Hash({
let hash: H256 = block_ref.parse().map_err(|e| format!("Invalid block number: {}", e))?;
hash.reversed()
})
} else {
BlockRef::Number(block_ref.parse().map_err(|e| format!("Invalid block hash: {}", e))?)
};
let required_block_hash = cfg.db.block_header(block_ref.clone()).ok_or(format!("Block {:?} is unknown", block_ref))?.hash();
let genesis_hash = cfg.network.genesis_block().hash();
let mut best_block_hash = cfg.db.best_block().hash;
debug_assert!(best_block_hash != H256::default());
loop {
if best_block_hash == required_block_hash {
info!("Reverted to block {:?}", block_ref);
return Ok(());
}
if best_block_hash == genesis_hash {
return Err(format!("Failed to revert to block {:?}. Reverted to genesis", block_ref));
}
best_block_hash = cfg.db.rollback_best().map_err(|e| format!("{:?}", e))?;
}
}