b4689e20c6
## Proposed Changes Implement the consensus changes necessary for the upcoming Altair hard fork. ## Additional Info This is quite a heavy refactor, with pivotal types like the `BeaconState` and `BeaconBlock` changing from structs to enums. This ripples through the whole codebase with field accesses changing to methods, e.g. `state.slot` => `state.slot()`. Co-authored-by: realbigsean <seananderson33@gmail.com>
61 lines
2.1 KiB
Rust
61 lines
2.1 KiB
Rust
use account_utils::{eth2_keystore::keypair_from_secret, mnemonic_from_phrase};
|
|
use clap::ArgMatches;
|
|
use eth2_network_config::Eth2NetworkConfig;
|
|
use eth2_wallet::bip39::Seed;
|
|
use eth2_wallet::{recover_validator_secret_from_mnemonic, KeyType};
|
|
use ssz::Encode;
|
|
use std::fs::File;
|
|
use std::io::{Read, Write};
|
|
use std::path::PathBuf;
|
|
use types::{BeaconState, EthSpec};
|
|
|
|
pub fn run<T: EthSpec>(testnet_dir: PathBuf, matches: &ArgMatches) -> Result<(), String> {
|
|
let path = matches
|
|
.value_of("ssz-state")
|
|
.ok_or("ssz-state not specified")?
|
|
.parse::<PathBuf>()
|
|
.map_err(|e| format!("Unable to parse ssz-state: {}", e))?;
|
|
|
|
let mnemonic_phrase = matches
|
|
.value_of("mnemonic")
|
|
.ok_or("mnemonic not specified")?;
|
|
|
|
let eth2_network_config = Eth2NetworkConfig::load(testnet_dir)?;
|
|
let spec = ð2_network_config.chain_spec::<T>()?;
|
|
|
|
let mut state: BeaconState<T> = {
|
|
let mut file = File::open(&path).map_err(|e| format!("Unable to open file: {}", e))?;
|
|
|
|
let mut ssz = vec![];
|
|
|
|
file.read_to_end(&mut ssz)
|
|
.map_err(|e| format!("Unable to read file: {}", e))?;
|
|
|
|
BeaconState::from_ssz_bytes(&ssz, spec)
|
|
.map_err(|e| format!("Unable to decode SSZ: {:?}", e))?
|
|
};
|
|
|
|
let mnemonic = mnemonic_from_phrase(mnemonic_phrase)?;
|
|
let seed = Seed::new(&mnemonic, "");
|
|
|
|
for (index, validator) in state.validators_mut().iter_mut().enumerate() {
|
|
let (secret, _) =
|
|
recover_validator_secret_from_mnemonic(seed.as_bytes(), index as u32, KeyType::Voting)
|
|
.map_err(|e| format!("Unable to generate validator key: {:?}", e))?;
|
|
|
|
let keypair = keypair_from_secret(secret.as_bytes())
|
|
.map_err(|e| format!("Unable build keystore: {:?}", e))?;
|
|
|
|
eprintln!("{}: {}", index, keypair.pk);
|
|
|
|
validator.pubkey = keypair.pk.into();
|
|
}
|
|
|
|
let mut file = File::create(path).map_err(|e| format!("Unable to create file: {}", e))?;
|
|
|
|
file.write_all(&state.as_ssz_bytes())
|
|
.map_err(|e| format!("Unable to write to file: {}", e))?;
|
|
|
|
Ok(())
|
|
}
|