Omit pubkeys from hot states

consensus/types/src/beacon_state.rs

@@ -1,6 +1,7 @@
 use self::committee_cache::get_active_validator_indices;
 use self::exit_cache::ExitCache;
 use crate::test_utils::TestRandom;
+use crate::validator::ValidatorTrait;
 use crate::*;
 use compare_fields::CompareFields;
 use compare_fields_derive::CompareFields;
@@ -31,6 +32,7 @@ pub use milhouse::{interface::Interface, List as VList, List, Vector as FixedVec
 
 #[macro_use]
 mod committee_cache;
+pub mod compact_state;
 mod diff;
 mod exit_cache;
 mod iter;
@@ -138,6 +140,7 @@ pub enum Error {
         current_fork: ForkName,
     },
     TotalActiveBalanceDiffUninitialized,
+    MissingImmutableValidator(usize),
 }
 
 /// Control whether an epoch-indexed field can be indexed at the next epoch or not.
@@ -212,7 +215,7 @@ impl From<BeaconStateHash> for Hash256 {
 #[cfg_attr(feature = "arbitrary-fuzz", derive(arbitrary::Arbitrary))]
 #[tree_hash(enum_behaviour = "transparent")]
 #[ssz(enum_behaviour = "transparent")]
-pub struct BeaconState<T>
+pub struct BeaconState<T, GenericValidator: ValidatorTrait = Validator>
 where
     T: EthSpec,
 {
@@ -246,7 +249,7 @@ where
 
     // Registry
     #[test_random(default)]
-    pub validators: VList<Validator, T::ValidatorRegistryLimit>,
+    pub validators: VList<GenericValidator, T::ValidatorRegistryLimit>,
     // FIXME(sproul): serde quoting
     // #[serde(with = "ssz_types::serde_utils::quoted_u64_var_list")]
     #[test_random(default)]
@@ -418,30 +421,6 @@ impl<T: EthSpec> BeaconState<T> {
         }
     }
 
-    /// Specialised deserialisation method that uses the `ChainSpec` as context.
-    #[allow(clippy::integer_arithmetic)]
-    pub fn from_ssz_bytes(bytes: &[u8], spec: &ChainSpec) -> Result<Self, ssz::DecodeError> {
-        // Slot is after genesis_time (u64) and genesis_validators_root (Hash256).
-        let slot_start = <u64 as Decode>::ssz_fixed_len() + <Hash256 as Decode>::ssz_fixed_len();
-        let slot_end = slot_start + <Slot as Decode>::ssz_fixed_len();
-
-        let slot_bytes = bytes
-            .get(slot_start..slot_end)
-            .ok_or(DecodeError::InvalidByteLength {
-                len: bytes.len(),
-                expected: slot_end,
-            })?;
-
-        let slot = Slot::from_ssz_bytes(slot_bytes)?;
-        let fork_at_slot = spec.fork_name_at_slot::<T>(slot);
-
-        Ok(map_fork_name!(
-            fork_at_slot,
-            Self,
-            <_>::from_ssz_bytes(bytes)?
-        ))
-    }
-
     /// Returns the `tree_hash_root` of the state.
     ///
     /// Spec v0.12.1
@@ -1668,6 +1647,32 @@ impl<T: EthSpec> BeaconState<T> {
     }
 }
 
+impl<T: EthSpec, V: ValidatorTrait> BeaconState<T, V> {
+    /// Specialised deserialisation method that uses the `ChainSpec` as context.
+    #[allow(clippy::integer_arithmetic)]
+    pub fn from_ssz_bytes(bytes: &[u8], spec: &ChainSpec) -> Result<Self, ssz::DecodeError> {
+        // Slot is after genesis_time (u64) and genesis_validators_root (Hash256).
+        let slot_start = <u64 as Decode>::ssz_fixed_len() + <Hash256 as Decode>::ssz_fixed_len();
+        let slot_end = slot_start + <Slot as Decode>::ssz_fixed_len();
+
+        let slot_bytes = bytes
+            .get(slot_start..slot_end)
+            .ok_or(DecodeError::InvalidByteLength {
+                len: bytes.len(),
+                expected: slot_end,
+            })?;
+
+        let slot = Slot::from_ssz_bytes(slot_bytes)?;
+        let fork_at_slot = spec.fork_name_at_slot::<T>(slot);
+
+        Ok(map_fork_name!(
+            fork_at_slot,
+            Self,
+            <_>::from_ssz_bytes(bytes)?
+        ))
+    }
+}
+
 impl From<RelativeEpochError> for Error {
     fn from(e: RelativeEpochError) -> Error {
         Error::RelativeEpochError(e)

consensus/types/src/beacon_state/compact_state.rs

@@ -0,0 +1,207 @@
+use crate::{
+    BeaconState, BeaconStateAltair, BeaconStateBase, BeaconStateError as Error, BeaconStateMerge,
+    EthSpec, VList, Validator, ValidatorImmutable, ValidatorMutable,
+};
+use itertools::process_results;
+use std::sync::Arc;
+
+pub type CompactBeaconState<E> = BeaconState<E, ValidatorMutable>;
+
+/// Implement the conversion function from BeaconState -> CompactBeaconState.
+macro_rules! full_to_compact {
+    ($s:ident, $outer:ident, $variant_name:ident, $struct_name:ident, [$($extra_fields:ident),*]) => {
+        BeaconState::$variant_name($struct_name {
+            // Versioning
+            genesis_time: $s.genesis_time,
+            genesis_validators_root: $s.genesis_validators_root,
+            slot: $s.slot,
+            fork: $s.fork,
+
+            // History
+            latest_block_header: $s.latest_block_header.clone(),
+            block_roots: $s.block_roots.clone(),
+            state_roots: $s.state_roots.clone(),
+            historical_roots: $s.historical_roots.clone(),
+
+            // Eth1
+            eth1_data: $s.eth1_data.clone(),
+            eth1_data_votes: $s.eth1_data_votes.clone(),
+            eth1_deposit_index: $s.eth1_deposit_index,
+
+            // Validator registry
+            validators: VList::try_from_iter(
+                $s.validators.into_iter().map(|validator| validator.mutable.clone())
+            ).expect("fix this"),
+            balances: $s.balances.clone(),
+
+            // Shuffling
+            randao_mixes: $s.randao_mixes.clone(),
+
+            // Slashings
+            slashings: $s.slashings.clone(),
+
+            // Finality
+            justification_bits: $s.justification_bits.clone(),
+            previous_justified_checkpoint: $s.previous_justified_checkpoint,
+            current_justified_checkpoint: $s.current_justified_checkpoint,
+            finalized_checkpoint: $s.finalized_checkpoint,
+
+            // Caches.
+            total_active_balance: $s.total_active_balance.clone(),
+            committee_caches: $s.committee_caches.clone(),
+            pubkey_cache: $s.pubkey_cache.clone(),
+            exit_cache: $s.exit_cache.clone(),
+
+            // Variant-specific fields
+            $(
+                $extra_fields: $s.$extra_fields.clone()
+            ),*
+        })
+    }
+}
+
+/// Implement the conversion from CompactBeaconState -> BeaconState.
+macro_rules! compact_to_full {
+    ($inner:ident, $variant_name:ident, $struct_name:ident, $immutable_validators:ident, [$($extra_fields:ident),*]) => {
+        BeaconState::$variant_name($struct_name {
+            // Versioning
+            genesis_time: $inner.genesis_time,
+            genesis_validators_root: $inner.genesis_validators_root,
+            slot: $inner.slot,
+            fork: $inner.fork,
+
+            // History
+            latest_block_header: $inner.latest_block_header,
+            block_roots: $inner.block_roots,
+            state_roots: $inner.state_roots,
+            historical_roots: $inner.historical_roots,
+
+            // Eth1
+            eth1_data: $inner.eth1_data,
+            eth1_data_votes: $inner.eth1_data_votes,
+            eth1_deposit_index: $inner.eth1_deposit_index,
+
+            // Validator registry
+            validators: process_results($inner.validators.into_iter().enumerate().map(|(i, mutable)| {
+                $immutable_validators(i)
+                    .ok_or(Error::MissingImmutableValidator(i))
+                    .map(move |immutable| {
+                        Validator {
+                            immutable,
+                            mutable: mutable.clone(),
+                        }
+                    })
+            }), |iter| VList::try_from_iter(iter))??,
+            balances: $inner.balances,
+
+            // Shuffling
+            randao_mixes: $inner.randao_mixes,
+
+            // Slashings
+            slashings: $inner.slashings,
+
+            // Finality
+            justification_bits: $inner.justification_bits,
+            previous_justified_checkpoint: $inner.previous_justified_checkpoint,
+            current_justified_checkpoint: $inner.current_justified_checkpoint,
+            finalized_checkpoint: $inner.finalized_checkpoint,
+
+            // Caching
+            total_active_balance: $inner.total_active_balance,
+            committee_caches: $inner.committee_caches,
+            pubkey_cache: $inner.pubkey_cache,
+            exit_cache: $inner.exit_cache,
+
+            // Variant-specific fields
+            $(
+                $extra_fields: $inner.$extra_fields
+            ),*
+        })
+    }
+}
+
+impl<E: EthSpec> BeaconState<E> {
+    pub fn into_compact_state(self) -> CompactBeaconState<E> {
+        match self {
+            BeaconState::Base(s) => full_to_compact!(
+                s,
+                self,
+                Base,
+                BeaconStateBase,
+                [previous_epoch_attestations, current_epoch_attestations]
+            ),
+            BeaconState::Altair(s) => full_to_compact!(
+                s,
+                self,
+                Altair,
+                BeaconStateAltair,
+                [
+                    previous_epoch_participation,
+                    current_epoch_participation,
+                    current_sync_committee,
+                    next_sync_committee,
+                    inactivity_scores
+                ]
+            ),
+            BeaconState::Merge(s) => full_to_compact!(
+                s,
+                self,
+                Merge,
+                BeaconStateMerge,
+                [
+                    previous_epoch_participation,
+                    current_epoch_participation,
+                    current_sync_committee,
+                    next_sync_committee,
+                    inactivity_scores,
+                    latest_execution_payload_header
+                ]
+            ),
+        }
+    }
+}
+
+impl<E: EthSpec> CompactBeaconState<E> {
+    pub fn try_into_full_state<F>(self, immutable_validators: F) -> Result<BeaconState<E>, Error>
+    where
+        F: Fn(usize) -> Option<Arc<ValidatorImmutable>>,
+    {
+        let state = match self {
+            BeaconState::Base(inner) => compact_to_full!(
+                inner,
+                Base,
+                BeaconStateBase,
+                immutable_validators,
+                [previous_epoch_attestations, current_epoch_attestations]
+            ),
+            BeaconState::Altair(inner) => compact_to_full!(
+                inner,
+                Altair,
+                BeaconStateAltair,
+                immutable_validators,
+                [
+                    previous_epoch_participation,
+                    current_epoch_participation,
+                    current_sync_committee,
+                    next_sync_committee,
+                    inactivity_scores
+                ]
+            ),
+            BeaconState::Merge(inner) => compact_to_full!(
+                inner,
+                Merge,
+                BeaconStateMerge,
+                immutable_validators,
+                [
+                    previous_epoch_participation,
+                    current_epoch_participation,
+                    current_sync_committee,
+                    next_sync_committee,
+                    inactivity_scores,
+                    latest_execution_payload_header
+                ]
+            ),
+        };
+        Ok(state)
+    }
+}

consensus/types/src/lib.rs

@@ -106,7 +106,7 @@ pub use crate::beacon_block_body::{
 };
 pub use crate::beacon_block_header::BeaconBlockHeader;
 pub use crate::beacon_committee::{BeaconCommittee, OwnedBeaconCommittee};
-pub use crate::beacon_state::{Error as BeaconStateError, *};
+pub use crate::beacon_state::{compact_state::CompactBeaconState, Error as BeaconStateError, *};
 pub use crate::chain_spec::{ChainSpec, Config, Domain};
 pub use crate::checkpoint::Checkpoint;
 pub use crate::config_and_preset::{

consensus/types/src/validator.rs

@@ -40,6 +40,24 @@ pub struct ValidatorImmutable {
     pub withdrawal_credentials: Hash256,
 }
 
+pub trait ValidatorTrait:
+    std::fmt::Debug
+    + PartialEq
+    + Clone
+    + serde::Serialize
+    + Send
+    + Sync
+    + serde::de::DeserializeOwned
+    + ssz::Encode
+    + ssz::Decode
+    + TreeHash
+    + TestRandom
+{
+}
+
+impl ValidatorTrait for Validator {}
+impl ValidatorTrait for ValidatorMutable {}
+
 impl Validator {
     pub fn pubkey(&self) -> &PublicKeyBytes {
         &self.immutable.pubkey