ab_client_block_builder/

beacon_chain.rs

//! Block building for the beacon chain

use crate::{BlockBuilder, BlockBuilderError, BlockBuilderResult};
use ab_client_api::{BeaconChainInfo, BlockDetails, ContractSlotState};
use ab_client_consensus_common::ConsensusConstants;
use ab_client_consensus_common::consensus_parameters::{
    DeriveConsensusParametersError, DeriveSuperSegmentForBlockError, derive_consensus_parameters,
    derive_super_segments_for_block,
};
use ab_client_consensus_common::state::GlobalState;
use ab_core_primitives::block::body::owned::OwnedBeaconChainBodyError;
use ab_core_primitives::block::header::owned::{GenericOwnedBlockHeader, OwnedBeaconChainHeader};
use ab_core_primitives::block::header::{
    BeaconChainHeader, BlockHeaderConsensusInfo, BlockHeaderConsistencyError, BlockHeaderPrefix,
    OwnedBlockHeaderConsensusParameters, OwnedBlockHeaderSeal,
};
use ab_core_primitives::block::owned::OwnedBeaconChainBlock;
use ab_core_primitives::block::{BlockNumber, BlockRoot, BlockTimestamp};
use ab_core_primitives::hashes::Blake3Hash;
use ab_core_primitives::pot::{PotCheckpoints, SlotNumber};
use ab_core_primitives::segments::SuperSegmentRoot;
use ab_core_primitives::shard::ShardIndex;
use ab_core_primitives::trees::BlockMerkleMountainRange;
use std::iter;
use std::time::SystemTime;
use triomphe::Arc;

/// Error for [`BeaconChainBlockBuilder`]
#[derive(Debug, thiserror::Error)]
pub enum BeaconChainBlockBuilderError {
    /// Consensus parameters derivation error
    #[error("Consensus parameters derivation error: {error}")]
    ConsensusParametersDerivation {
        /// Consensus parameters derivation error
        #[from]
        error: DeriveConsensusParametersError,
    },
    /// Super segment derivation error
    #[error("Super segment derivation error: {error}")]
    SuperSegmentDerivation {
        /// Super segment derivation error
        #[from]
        error: DeriveSuperSegmentForBlockError,
    },
    /// Failed to create body
    #[error("Failed to create body: {error}")]
    FailedToCreateBody {
        // Body creation error
        #[from]
        error: OwnedBeaconChainBodyError,
    },
}

impl From<BeaconChainBlockBuilderError> for BlockBuilderError {
    #[inline(always)]
    fn from(error: BeaconChainBlockBuilderError) -> Self {
        Self::Custom {
            error: error.into(),
        }
    }
}

// TODO: Another domain-specific abstraction over `ChainInfo`, which will be implemented for
//  `ChainInfo`, but could also be implemented in simpler way directly for tests without dealing
//  with complete headers, etc.
/// Beacon chain block builder
#[derive(Debug)]
pub struct BeaconChainBlockBuilder<BCI> {
    consensus_constants: ConsensusConstants,
    chain_info: BCI,
}

impl<CI> BlockBuilder<OwnedBeaconChainBlock> for BeaconChainBlockBuilder<CI>
where
    CI: BeaconChainInfo,
{
    async fn build<SealBlock>(
        &mut self,
        parent_block_root: &BlockRoot,
        parent_header: &<OwnedBeaconChainHeader as GenericOwnedBlockHeader>::Header<'_>,
        parent_block_details: &BlockDetails,
        consensus_info: &BlockHeaderConsensusInfo,
        checkpoints: &[PotCheckpoints],
        seal_block: SealBlock,
    ) -> Result<BlockBuilderResult<OwnedBeaconChainBlock>, BlockBuilderError>
    where
        SealBlock: AsyncFnOnce<(Blake3Hash,), Output = Option<OwnedBlockHeaderSeal>, CallOnceFuture: Send>
            + Send,
    {
        let block_number = parent_header.prefix.number + BlockNumber::ONE;

        let header_prefix = self.create_header_prefix(
            parent_block_root,
            parent_header.prefix.timestamp,
            &parent_block_details.mmr_with_block,
            block_number,
        )?;
        // TODO: Handle super segment error caused by too many segments included in shard blocks.
        //  While not a concern any time soon, but it is theoretically possible and should be
        //  handled in a general case.
        let maybe_super_segment = derive_super_segments_for_block(
            &self.chain_info,
            parent_header.prefix.number,
            self.consensus_constants.block_confirmation_depth,
            self.consensus_constants.shard_confirmation_depth,
        )
        .map_err(BeaconChainBlockBuilderError::from)?;

        let consensus_parameters = self.derive_consensus_parameters(
            parent_block_root,
            parent_header,
            block_number,
            consensus_info.slot,
            maybe_super_segment
                .as_ref()
                .map(|super_segment| super_segment.header.root),
        )?;

        let (state_root, system_contract_states) = self.execute_block(parent_block_details);

        let block_builder = OwnedBeaconChainBlock::init(
            self.chain_info
                .segment_headers_for_block(block_number)
                .into_iter()
                .map(|segment_header| (segment_header.index.as_inner(), segment_header.root)),
            // TODO: Real intermediate shard blocks
            iter::empty(),
            checkpoints,
        )
        .map_err(BeaconChainBlockBuilderError::from)?;

        let block_unsealed = block_builder.with_header(
            &header_prefix,
            state_root,
            consensus_info,
            &consensus_parameters.as_ref(),
        )?;

        let pre_seal_hash = block_unsealed.pre_seal_hash();
        let seal = seal_block(pre_seal_hash)
            .await
            .ok_or(BlockBuilderError::FailedToSeal)?;
        let block = block_unsealed
            .with_seal(seal.as_ref())
            .map_err(|error| match error {
                BlockHeaderConsistencyError::SealPublicKeyHashMismatch {
                    solution_public_key_hash: _,
                    seal_public_key_hash,
                } => BlockBuilderError::InvalidSeal {
                    public_key_hash: seal_public_key_hash,
                    pre_seal_hash,
                },
                error => BlockBuilderError::InvalidHeaderParts { error },
            })?;

        let mut block_mmr = *parent_block_details.mmr_with_block;

        if !block_mmr.add_leaf(&block.header.header().root()) {
            return Err(BlockBuilderError::CantExtendMmr);
        }

        Ok(BlockBuilderResult {
            block,
            block_details: BlockDetails {
                mmr_with_block: Arc::new(block_mmr),
                system_contract_states,
            },
            extra: (),
        })
    }
}

impl<BCI> BeaconChainBlockBuilder<BCI>
where
    BCI: BeaconChainInfo,
{
    /// Create a new instance
    pub fn new(consensus_constants: ConsensusConstants, chain_info: BCI) -> Self {
        Self {
            consensus_constants,
            chain_info,
        }
    }

    fn create_header_prefix(
        &self,
        parent_block_root: &BlockRoot,
        parent_timestamp: BlockTimestamp,
        mmr_with_block: &BlockMerkleMountainRange,
        block_number: BlockNumber,
    ) -> Result<BlockHeaderPrefix, BlockBuilderError> {
        let timestamp = SystemTime::now()
            .duration_since(SystemTime::UNIX_EPOCH)
            .unwrap_or_default()
            .as_millis();
        let mut timestamp =
            BlockTimestamp::from_millis(u64::try_from(timestamp).unwrap_or(u64::MAX));

        if timestamp <= parent_timestamp {
            timestamp = BlockTimestamp::from_millis(parent_timestamp.as_millis().saturating_add(1));
        }

        Ok(BlockHeaderPrefix {
            number: block_number,
            shard_index: ShardIndex::BEACON_CHAIN,
            padding_0: [0; _],
            timestamp,
            parent_root: *parent_block_root,
            mmr_root: Blake3Hash::new(
                mmr_with_block
                    .root()
                    .ok_or(BlockBuilderError::InvalidParentMmr)?,
            ),
        })
    }

    fn derive_consensus_parameters(
        &self,
        parent_block_root: &BlockRoot,
        parent_header: &BeaconChainHeader<'_>,
        block_number: BlockNumber,
        slot: SlotNumber,
        super_segment_root: Option<SuperSegmentRoot>,
    ) -> Result<OwnedBlockHeaderConsensusParameters, BeaconChainBlockBuilderError> {
        let derived_consensus_parameters = derive_consensus_parameters(
            &self.consensus_constants,
            &self.chain_info,
            parent_block_root,
            parent_header.consensus_parameters(),
            parent_header.consensus_info.slot,
            block_number,
            slot,
        )?;

        Ok(OwnedBlockHeaderConsensusParameters {
            fixed_parameters: derived_consensus_parameters.fixed_parameters,
            super_segment_root,
            next_solution_range: derived_consensus_parameters.next_solution_range,
            pot_parameters_change: derived_consensus_parameters.pot_parameters_change,
        })
    }

    fn execute_block(
        &self,
        parent_block_details: &BlockDetails,
    ) -> (Blake3Hash, Arc<[ContractSlotState]>) {
        let global_state = GlobalState::new(&parent_block_details.system_contract_states);

        // TODO: Execute block

        let state_root = global_state.root();
        let system_contract_states = global_state.to_system_contract_states();

        (state_root, system_contract_states)
    }
}