From eef3627c92c80db9237059daf2fea1dc2dc5238a Mon Sep 17 00:00:00 2001 From: Grant Wuerker Date: Sun, 16 Dec 2018 17:00:53 -0600 Subject: [PATCH] one test passing --- beacon_chain/types/src/beacon_state.rs | 2 +- beacon_chain/types/src/fork_data.rs | 2 +- .../validator_induction/src/inductor.old.rs | 281 ++++++++++++++++++ .../validator_induction/src/inductor.rs | 37 ++- 4 files changed, 301 insertions(+), 21 deletions(-) create mode 100644 beacon_chain/validator_induction/src/inductor.old.rs diff --git a/beacon_chain/types/src/beacon_state.rs b/beacon_chain/types/src/beacon_state.rs index 2f5581d57..d0e46f1b5 100644 --- a/beacon_chain/types/src/beacon_state.rs +++ b/beacon_chain/types/src/beacon_state.rs @@ -7,7 +7,7 @@ use super::shard_reassignment_record::ShardReassignmentRecord; use super::validator_record::ValidatorRecord; use super::Hash256; -#[derive(Debug, PartialEq)] +#[derive(Debug, PartialEq, Default)] pub struct BeaconState { pub validator_registry: Vec, pub validator_registry_latest_change_slot: u64, diff --git a/beacon_chain/types/src/fork_data.rs b/beacon_chain/types/src/fork_data.rs index 4c3541c3f..ce3cf77e2 100644 --- a/beacon_chain/types/src/fork_data.rs +++ b/beacon_chain/types/src/fork_data.rs @@ -1,4 +1,4 @@ -#[derive(Debug, Clone, PartialEq)] +#[derive(Debug, Clone, PartialEq, Default)] pub struct ForkData { pub pre_fork_version: u64, pub post_fork_version: u64, diff --git a/beacon_chain/validator_induction/src/inductor.old.rs b/beacon_chain/validator_induction/src/inductor.old.rs new file mode 100644 index 000000000..816d8e356 --- /dev/null +++ b/beacon_chain/validator_induction/src/inductor.old.rs @@ -0,0 +1,281 @@ +use bls::verify_proof_of_possession; +use types::{ValidatorRecord, ValidatorRegistration, ValidatorStatus}; + +/// The size of a validators deposit in GWei. +pub const DEPOSIT_GWEI: u64 = 32_000_000_000; + +/// Inducts validators into a `CrystallizedState`. +pub struct ValidatorInductor { + pub current_slot: u64, + pub shard_count: u16, + validators: Vec, + empty_validator_start: usize, +} + +#[derive(Debug, PartialEq, Clone)] +pub enum ValidatorInductionError { + InvalidShard, + InvaidProofOfPossession, +} + +impl ValidatorInductor { + pub fn new(current_slot: u64, shard_count: u16, validators: Vec) -> Self { + Self { + current_slot, + shard_count, + validators, + empty_validator_start: 0, + } + } + + /// Attempt to induct a validator into the CrystallizedState. + /// + /// Returns an error if the registration is invalid, otherwise returns the index of the + /// validator in `CrystallizedState.validators`. + pub fn induct( + &mut self, + rego: &ValidatorRegistration, + status: ValidatorStatus, + ) -> Result { + let v = self.process_registration(rego, status)?; + Ok(self.add_validator(v)) + } + + /// Verify a `ValidatorRegistration` and return a `ValidatorRecord` if valid. + fn process_registration( + &self, + r: &ValidatorRegistration, + status: ValidatorStatus, + ) -> Result { + /* + * Ensure withdrawal shard is not too high. + */ + if r.withdrawal_shard > self.shard_count { + return Err(ValidatorInductionError::InvalidShard); + } + + /* + * Prove validator has knowledge of their secret key. + */ + if !verify_proof_of_possession(&r.proof_of_possession, &r.pubkey) { + return Err(ValidatorInductionError::InvaidProofOfPossession); + } + + Ok(ValidatorRecord { + pubkey: r.pubkey.clone(), + withdrawal_shard: r.withdrawal_shard, + withdrawal_address: r.withdrawal_address, + randao_commitment: r.randao_commitment, + randao_last_change: self.current_slot, + balance: DEPOSIT_GWEI, + status: status, + exit_slot: 0, + }) + } + + /// Returns the index of the first `ValidatorRecord` in the `CrystallizedState` where + /// `validator.status == Withdrawn`. If no such record exists, `None` is returned. + fn first_withdrawn_validator(&mut self) -> Option { + for i in self.empty_validator_start..self.validators.len() { + if self.validators[i].status == ValidatorStatus::Withdrawn { + self.empty_validator_start = i + 1; + return Some(i); + } + } + None + } + + /// Adds a `ValidatorRecord` to the `CrystallizedState` by replacing first validator where + /// `validator.status == Withdraw`. If no such withdrawn validator exists, adds the new + /// validator to the end of the list. + fn add_validator(&mut self, v: ValidatorRecord) -> usize { + match self.first_withdrawn_validator() { + Some(i) => { + self.validators[i] = v; + i + } + None => { + self.validators.push(v); + self.validators.len() - 1 + } + } + } + + pub fn to_vec(self) -> Vec { + self.validators + } +} + +#[cfg(test)] +mod tests { + use super::*; + + use bls::{Keypair, Signature}; + use hashing::proof_of_possession_hash; + use types::{Address, Hash256}; + + fn registration_equals_record(reg: &ValidatorRegistration, rec: &ValidatorRecord) -> bool { + (reg.pubkey == rec.pubkey) + & (reg.withdrawal_shard == rec.withdrawal_shard) + & (reg.withdrawal_address == rec.withdrawal_address) + & (reg.randao_commitment == rec.randao_commitment) + & (verify_proof_of_possession(®.proof_of_possession, &rec.pubkey)) + } + + /// Generate a proof of possession for some keypair. + fn get_proof_of_possession(kp: &Keypair) -> Signature { + let pop_message = proof_of_possession_hash(&kp.pk.as_bytes()); + Signature::new_hashed(&pop_message, &kp.sk) + } + + /// Generate a basic working ValidatorRegistration for use in tests. + fn get_registration() -> ValidatorRegistration { + let kp = Keypair::random(); + ValidatorRegistration { + pubkey: kp.pk.clone(), + withdrawal_shard: 0, + withdrawal_address: Address::zero(), + randao_commitment: Hash256::zero(), + proof_of_possession: get_proof_of_possession(&kp), + } + } + + #[test] + fn test_validator_inductor_valid_empty_validators() { + let validators = vec![]; + + let r = get_registration(); + + let mut inductor = ValidatorInductor::new(0, 1024, validators); + let result = inductor.induct(&r, ValidatorStatus::PendingActivation); + let validators = inductor.to_vec(); + + assert_eq!(result.unwrap(), 0); + assert!(registration_equals_record(&r, &validators[0])); + assert_eq!(validators.len(), 1); + } + + #[test] + fn test_validator_inductor_status() { + let validators = vec![]; + + let r = get_registration(); + + let mut inductor = ValidatorInductor::new(0, 1024, validators); + let _ = inductor.induct(&r, ValidatorStatus::PendingActivation); + let _ = inductor.induct(&r, ValidatorStatus::Active); + let validators = inductor.to_vec(); + + assert!(validators[0].status == ValidatorStatus::PendingActivation); + assert!(validators[1].status == ValidatorStatus::Active); + assert_eq!(validators.len(), 2); + } + + #[test] + fn test_validator_inductor_valid_all_active_validators() { + let mut validators = vec![]; + for _ in 0..5 { + let (mut v, _) = ValidatorRecord::zero_with_thread_rand_keypair(); + v.status = ValidatorStatus::Active; + validators.push(v); + } + + let r = get_registration(); + + let mut inductor = ValidatorInductor::new(0, 1024, validators); + let result = inductor.induct(&r, ValidatorStatus::PendingActivation); + let validators = inductor.to_vec(); + + assert_eq!(result.unwrap(), 5); + assert!(registration_equals_record(&r, &validators[5])); + assert_eq!(validators.len(), 6); + } + + #[test] + fn test_validator_inductor_valid_all_second_validator_withdrawn() { + let mut validators = vec![]; + let (mut v, _) = ValidatorRecord::zero_with_thread_rand_keypair(); + v.status = ValidatorStatus::Active; + validators.push(v); + for _ in 0..4 { + let (mut v, _) = ValidatorRecord::zero_with_thread_rand_keypair(); + v.status = ValidatorStatus::Withdrawn; + validators.push(v); + } + + let r = get_registration(); + + let mut inductor = ValidatorInductor::new(0, 1024, validators); + let result = inductor.induct(&r, ValidatorStatus::PendingActivation); + let validators = inductor.to_vec(); + + assert_eq!(result.unwrap(), 1); + assert!(registration_equals_record(&r, &validators[1])); + assert_eq!(validators.len(), 5); + } + + #[test] + fn test_validator_inductor_valid_all_withdrawn_validators() { + let mut validators = vec![]; + for _ in 0..5 { + let (mut v, _) = ValidatorRecord::zero_with_thread_rand_keypair(); + v.status = ValidatorStatus::Withdrawn; + validators.push(v); + } + + /* + * Ensure the first validator gets the 0'th slot + */ + let r = get_registration(); + let mut inductor = ValidatorInductor::new(0, 1024, validators); + let result = inductor.induct(&r, ValidatorStatus::PendingActivation); + let validators = inductor.to_vec(); + assert_eq!(result.unwrap(), 0); + assert!(registration_equals_record(&r, &validators[0])); + + /* + * Ensure the second validator gets the 1'st slot + */ + let r_two = get_registration(); + let mut inductor = ValidatorInductor::new(0, 1024, validators); + let result = inductor.induct(&r_two, ValidatorStatus::PendingActivation); + let validators = inductor.to_vec(); + assert_eq!(result.unwrap(), 1); + assert!(registration_equals_record(&r_two, &validators[1])); + assert_eq!(validators.len(), 5); + } + + #[test] + fn test_validator_inductor_shard_too_high() { + let validators = vec![]; + + let mut r = get_registration(); + r.withdrawal_shard = 1025; + + let mut inductor = ValidatorInductor::new(0, 1024, validators); + let result = inductor.induct(&r, ValidatorStatus::PendingActivation); + let validators = inductor.to_vec(); + + assert_eq!(result, Err(ValidatorInductionError::InvalidShard)); + assert_eq!(validators.len(), 0); + } + + #[test] + fn test_validator_inductor_shard_proof_of_possession_failure() { + let validators = vec![]; + + let mut r = get_registration(); + let kp = Keypair::random(); + r.proof_of_possession = get_proof_of_possession(&kp); + + let mut inductor = ValidatorInductor::new(0, 1024, validators); + let result = inductor.induct(&r, ValidatorStatus::PendingActivation); + let validators = inductor.to_vec(); + + assert_eq!( + result, + Err(ValidatorInductionError::InvaidProofOfPossession) + ); + assert_eq!(validators.len(), 0); + } +} diff --git a/beacon_chain/validator_induction/src/inductor.rs b/beacon_chain/validator_induction/src/inductor.rs index 63022b6ba..6a455a74c 100644 --- a/beacon_chain/validator_induction/src/inductor.rs +++ b/beacon_chain/validator_induction/src/inductor.rs @@ -1,5 +1,5 @@ use bls::verify_proof_of_possession; -use types::{ValidatorRecord, Deposit, ValidatorStatus, BeaconState}; +use types::{ValidatorRecord, DepositInput, ValidatorStatus, BeaconState}; /// The size of a validators deposit in GWei. pub const DEPOSIT_GWEI: u64 = 32_000_000_000; @@ -34,20 +34,19 @@ impl ValidatorInductor { /// validator in `CrystallizedState.validators`. pub fn induct( &mut self, - deposit: &Deposit, + deposit_input: &DepositInput, status: ValidatorStatus, ) -> Result { - let v = self.process_registration(deposit, status)?; + let v = self.process_deposit(deposit_input, status)?; Ok(self.add_validator(v)) } /// Verify a `ValidatorRegistration` and return a `ValidatorRecord` if valid. fn process_deposit( &self, - deposit: &Deposit, + deposit_input: &DepositInput, status: ValidatorStatus, ) -> Result { - let deposit_input = &deposit.deposit_data.deposit_input; /* * Ensure withdrawal shard is not too high. */ @@ -80,9 +79,8 @@ impl ValidatorInductor { /// Returns the index of the first `ValidatorRecord` in the `CrystallizedState` where /// `validator.status == Withdrawn`. If no such record exists, `None` is returned. fn first_withdrawn_validator(&mut self) -> Option { - let validators = self.beacon_state.validator_registry; - for i in self.empty_validator_start..validators.len() { - if validators[i].status == ValidatorStatus::Withdrawn { + for i in self.empty_validator_start..self.beacon_state.validator_registry.len() { + if self.beacon_state.validator_registry[i].status == ValidatorStatus::Withdrawn { self.empty_validator_start = i + 1; return Some(i); } @@ -117,7 +115,7 @@ mod tests { use bls::{Keypair, Signature}; use hashing::proof_of_possession_hash; - use types::{Address, Hash256}; + use types::{Hash256}; /* fn registration_equals_record(reg: &ValidatorRegistration, rec: &ValidatorRecord) -> bool { @@ -135,26 +133,25 @@ mod tests { Signature::new_hashed(&pop_message, &kp.sk) } - /// Generate a basic working ValidatorRegistration for use in tests. - fn get_registration() -> ValidatorRegistration { + /// Generate a basic working Deposit for use in tests. + fn get_deposit_input() -> DepositInput { let kp = Keypair::random(); - ValidatorRegistration { + DepositInput { pubkey: kp.pk.clone(), - withdrawal_shard: 0, - withdrawal_address: Address::zero(), + withdrawal_credentials: Hash256::zero(), randao_commitment: Hash256::zero(), - proof_of_possession: get_proof_of_possession(&kp), + proof_of_possession: get_proof_of_possession(&kp) } } #[test] fn test_validator_inductor_valid_empty_validators() { - let validators = vec![]; + let state = BeaconState::default(); - let r = get_registration(); + let d = get_deposit_input(); - let mut inductor = ValidatorInductor::new(0, 1024, validators); - let result = inductor.induct(&r, ValidatorStatus::PendingActivation); + let mut inductor = ValidatorInductor::new(0, 1024, state); + let result = inductor.induct(&d, ValidatorStatus::PendingActivation); let validators = inductor.to_vec(); assert_eq!(result.unwrap(), 0); @@ -162,6 +159,7 @@ mod tests { assert_eq!(validators.len(), 1); } + /* #[test] fn test_validator_inductor_status() { let validators = vec![]; @@ -285,4 +283,5 @@ mod tests { ); assert_eq!(validators.len(), 0); } + */ }