62 lines
2.4 KiB
Rust
62 lines
2.4 KiB
Rust
|
use super::per_block_processing::{errors::BlockProcessingError, process_deposits};
|
||
|
|
use crate::common::get_compact_committees_root;
|
||
|
|
use tree_hash::TreeHash;
|
||
|
|
use types::typenum::U4294967296;
|
||
|
|
use types::*;
|
||
|
|
|
||
|
|
/// Initialize a `BeaconState` from genesis data.
|
||
|
|
///
|
||
|
|
/// Spec v0.8.0
|
||
|
|
// TODO: this is quite inefficient and we probably want to rethink how we do this
|
||
|
|
pub fn initialize_beacon_state_from_eth1<T: EthSpec>(
|
||
|
|
eth1_block_hash: Hash256,
|
||
|
|
eth1_timestamp: u64,
|
||
|
|
deposits: Vec<Deposit>,
|
||
|
|
spec: &ChainSpec,
|
||
|
|
) -> Result<BeaconState<T>, BlockProcessingError> {
|
||
|
|
let genesis_time =
|
||
|
|
eth1_timestamp - eth1_timestamp % spec.seconds_per_day + 2 * spec.seconds_per_day;
|
||
|
|
let eth1_data = Eth1Data {
|
||
|
|
// Temporary deposit root
|
||
|
|
deposit_root: Hash256::zero(),
|
||
|
|
deposit_count: deposits.len() as u64,
|
||
|
|
block_hash: eth1_block_hash,
|
||
|
|
};
|
||
|
|
let mut state = BeaconState::new(genesis_time, eth1_data, spec);
|
||
|
|
|
||
|
|
// Process deposits
|
||
|
|
let leaves: Vec<_> = deposits
|
||
|
|
.iter()
|
||
|
|
.map(|deposit| deposit.data.clone())
|
||
|
|
.collect();
|
||
|
|
for (index, deposit) in deposits.into_iter().enumerate() {
|
||
|
|
let deposit_data_list = VariableList::<_, U4294967296>::from(leaves[..=index].to_vec());
|
||
|
|
state.eth1_data.deposit_root = Hash256::from_slice(&deposit_data_list.tree_hash_root());
|
||
|
|
process_deposits(&mut state, &[deposit], spec)?;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Process activations
|
||
|
|
for (index, validator) in state.validators.iter_mut().enumerate() {
|
||
|
|
let balance = state.balances[index];
|
||
|
|
validator.effective_balance = std::cmp::min(
|
||
|
|
balance - balance % spec.effective_balance_increment,
|
||
|
|
spec.max_effective_balance,
|
||
|
|
);
|
||
|
|
if validator.effective_balance == spec.max_effective_balance {
|
||
|
|
validator.activation_eligibility_epoch = T::genesis_epoch();
|
||
|
|
validator.activation_epoch = T::genesis_epoch();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Populate active_index_roots and compact_committees_roots
|
||
|
|
let indices_list = VariableList::<usize, T::ValidatorRegistryLimit>::from(
|
||
|
|
state.get_active_validator_indices(T::genesis_epoch()),
|
||
|
|
);
|
||
|
|
let active_index_root = Hash256::from_slice(&indices_list.tree_hash_root());
|
||
|
|
let committee_root = get_compact_committees_root(&state, RelativeEpoch::Current, spec)?;
|
||
|
|
state.fill_active_index_roots_with(active_index_root);
|
||
|
|
state.fill_compact_committees_roots_with(committee_root);
|
||
|
|
|
||
|
|
Ok(state)
|
||
|
|
}
|