casper-node 2.0.3

use std::{collections::BTreeSet, fmt::Debug};

use datasize::DataSize;
use serde::{Deserialize, Deserializer, Serialize, Serializer};

use casper_types::Timestamp;

use crate::components::consensus::{
    highway_core::{
        endorsement::SignedEndorsement,
        highway::{PingError, VertexError},
        state::Panorama,
    },
    traits::{Context, ValidatorSecret},
    utils::{ValidatorIndex, Validators},
};

#[allow(clippy::arithmetic_side_effects)]
mod relaxed {
    // This module exists solely to exempt the `EnumDiscriminants` macro generated code from the
    // module-wide `clippy::arithmetic_side_effects` lint.

    use casper_types::Timestamp;
    use datasize::DataSize;
    use serde::{Deserialize, Serialize};
    use strum::EnumDiscriminants;

    use crate::components::consensus::{
        highway_core::evidence::Evidence, traits::Context, utils::ValidatorIndex,
    };

    use super::{Endorsements, Ping, SignedWireUnit};

    /// A dependency of a `Vertex` that can be satisfied by one or more other vertices.
    #[derive(
        DataSize,
        Clone,
        Debug,
        Eq,
        PartialEq,
        PartialOrd,
        Ord,
        Hash,
        Serialize,
        Deserialize,
        EnumDiscriminants,
    )]
    #[serde(bound(
        serialize = "C::Hash: Serialize",
        deserialize = "C::Hash: Deserialize<'de>",
    ))]
    #[strum_discriminants(derive(strum::EnumIter))]
    pub enum Dependency<C>
    where
        C: Context,
    {
        /// The hash of a unit.
        Unit(C::Hash),
        /// The index of the validator against which evidence is needed.
        Evidence(ValidatorIndex),
        /// The hash of the unit to be endorsed.
        Endorsement(C::Hash),
        /// The ping by a particular validator for a particular timestamp.
        Ping(ValidatorIndex, Timestamp),
    }

    /// An element of the protocol state, that might depend on other elements.
    ///
    /// It is the vertex in a directed acyclic graph, whose edges are dependencies.
    #[derive(
        DataSize, Clone, Debug, Eq, PartialEq, Serialize, Deserialize, Hash, EnumDiscriminants,
    )]
    #[serde(bound(
        serialize = "C::Hash: Serialize",
        deserialize = "C::Hash: Deserialize<'de>",
    ))]
    #[strum_discriminants(derive(strum::EnumIter))]
    pub enum Vertex<C>
    where
        C: Context,
    {
        /// A signed unit of the consensus DAG.
        Unit(SignedWireUnit<C>),
        /// Evidence of a validator's transgression.
        Evidence(Evidence<C>),
        /// Endorsements for a unit.
        Endorsements(Endorsements<C>),
        /// A ping conveying the activity of its creator.
        Ping(Ping<C>),
    }
}
pub use relaxed::{Dependency, DependencyDiscriminants, Vertex, VertexDiscriminants};

impl<C: Context> Dependency<C> {
    /// Returns whether this identifies a unit, as opposed to other types of vertices.
    pub fn is_unit(&self) -> bool {
        matches!(self, Dependency::Unit(_))
    }
}

impl<C: Context> Vertex<C> {
    /// Returns the consensus value mentioned in this vertex, if any.
    ///
    /// These need to be validated before passing the vertex into the protocol state. E.g. if
    /// `C::ConsensusValue` is a transaction, it should be validated first (correct signature,
    /// structure, gas limit, etc.). If it is a hash of a transaction, the transaction should be
    /// obtained _and_ validated. Only after that, the vertex can be considered valid.
    pub fn value(&self) -> Option<&C::ConsensusValue> {
        match self {
            Vertex::Unit(swunit) => swunit.wire_unit().value.as_ref(),
            Vertex::Evidence(_) | Vertex::Endorsements(_) | Vertex::Ping(_) => None,
        }
    }

    /// Returns the unit hash of this vertex (if it is a unit).
    pub fn unit_hash(&self) -> Option<C::Hash> {
        match self {
            Vertex::Unit(swunit) => Some(swunit.hash()),
            Vertex::Evidence(_) | Vertex::Endorsements(_) | Vertex::Ping(_) => None,
        }
    }

    /// Returns the seq number of this vertex (if it is a unit).
    pub fn unit_seq_number(&self) -> Option<u64> {
        match self {
            Vertex::Unit(swunit) => Some(swunit.wire_unit().seq_number),
            _ => None,
        }
    }

    /// Returns whether this is evidence, as opposed to other types of vertices.
    pub fn is_evidence(&self) -> bool {
        matches!(self, Vertex::Evidence(_))
    }

    /// Returns a `Timestamp` provided the vertex is a `Vertex::Unit` or `Vertex::Ping`.
    pub fn timestamp(&self) -> Option<Timestamp> {
        match self {
            Vertex::Unit(signed_wire_unit) => Some(signed_wire_unit.wire_unit().timestamp),
            Vertex::Ping(ping) => Some(ping.timestamp()),
            Vertex::Evidence(_) | Vertex::Endorsements(_) => None,
        }
    }

    /// Returns the creator of this vertex, if one is defined.
    pub fn creator(&self) -> Option<ValidatorIndex> {
        match self {
            Vertex::Unit(signed_wire_unit) => Some(signed_wire_unit.wire_unit().creator),
            Vertex::Ping(ping) => Some(ping.creator),
            Vertex::Evidence(_) | Vertex::Endorsements(_) => None,
        }
    }

    /// Returns the ID of this vertex.
    pub fn id(&self) -> Dependency<C> {
        match self {
            Vertex::Unit(signed_wire_unit) => Dependency::Unit(signed_wire_unit.hash()),
            Vertex::Evidence(evidence) => Dependency::Evidence(evidence.perpetrator()),
            Vertex::Endorsements(endorsement) => Dependency::Endorsement(endorsement.unit),
            Vertex::Ping(ping) => Dependency::Ping(ping.creator(), ping.timestamp()),
        }
    }

    /// Returns a reference to the unit, or `None` if this is not a unit.
    pub fn unit(&self) -> Option<&SignedWireUnit<C>> {
        match self {
            Vertex::Unit(signed_wire_unit) => Some(signed_wire_unit),
            _ => None,
        }
    }

    /// Returns true whether unit is a proposal.
    pub fn is_proposal(&self) -> bool {
        self.value().is_some()
    }
}

mod specimen_support {
    use super::{
        Dependency, DependencyDiscriminants, Endorsements, HashedWireUnit, Ping, SignedEndorsement,
        SignedWireUnit, Vertex, VertexDiscriminants, WireUnit,
    };
    use crate::{
        components::consensus::ClContext,
        utils::specimen::{
            btree_set_distinct_from_prop, largest_variant, vec_prop_specimen, Cache,
            LargestSpecimen, SizeEstimator,
        },
    };

    impl LargestSpecimen for Vertex<ClContext> {
        fn largest_specimen<E: SizeEstimator>(estimator: &E, cache: &mut Cache) -> Self {
            largest_variant::<Self, VertexDiscriminants, _, _>(estimator, |variant| match variant {
                VertexDiscriminants::Unit => {
                    Vertex::Unit(LargestSpecimen::largest_specimen(estimator, cache))
                }
                VertexDiscriminants::Evidence => {
                    Vertex::Evidence(LargestSpecimen::largest_specimen(estimator, cache))
                }
                VertexDiscriminants::Endorsements => {
                    if estimator.parameter_bool("endorsements_enabled") {
                        Vertex::Endorsements(LargestSpecimen::largest_specimen(estimator, cache))
                    } else {
                        Vertex::Ping(LargestSpecimen::largest_specimen(estimator, cache))
                    }
                }
                VertexDiscriminants::Ping => {
                    Vertex::Ping(LargestSpecimen::largest_specimen(estimator, cache))
                }
            })
        }
    }

    impl LargestSpecimen for Dependency<ClContext> {
        fn largest_specimen<E: SizeEstimator>(estimator: &E, cache: &mut Cache) -> Self {
            largest_variant::<Self, DependencyDiscriminants, _, _>(estimator, |variant| {
                match variant {
                    DependencyDiscriminants::Unit => {
                        Dependency::Unit(LargestSpecimen::largest_specimen(estimator, cache))
                    }
                    DependencyDiscriminants::Evidence => {
                        Dependency::Evidence(LargestSpecimen::largest_specimen(estimator, cache))
                    }
                    DependencyDiscriminants::Endorsement => {
                        Dependency::Endorsement(LargestSpecimen::largest_specimen(estimator, cache))
                    }
                    DependencyDiscriminants::Ping => Dependency::Ping(
                        LargestSpecimen::largest_specimen(estimator, cache),
                        LargestSpecimen::largest_specimen(estimator, cache),
                    ),
                }
            })
        }
    }

    impl LargestSpecimen for SignedWireUnit<ClContext> {
        fn largest_specimen<E: SizeEstimator>(estimator: &E, cache: &mut Cache) -> Self {
            SignedWireUnit {
                hashed_wire_unit: LargestSpecimen::largest_specimen(estimator, cache),
                signature: LargestSpecimen::largest_specimen(estimator, cache),
            }
        }
    }

    impl LargestSpecimen for Endorsements<ClContext> {
        fn largest_specimen<E: SizeEstimator>(estimator: &E, cache: &mut Cache) -> Self {
            Endorsements {
                unit: LargestSpecimen::largest_specimen(estimator, cache),
                endorsers: if estimator.parameter_bool("endorsements_enabled") {
                    vec_prop_specimen(estimator, "validator_count", cache)
                } else {
                    Vec::new()
                },
            }
        }
    }

    impl LargestSpecimen for SignedEndorsement<ClContext> {
        fn largest_specimen<E: SizeEstimator>(estimator: &E, cache: &mut Cache) -> Self {
            SignedEndorsement::new(
                LargestSpecimen::largest_specimen(estimator, cache),
                LargestSpecimen::largest_specimen(estimator, cache),
            )
        }
    }

    impl LargestSpecimen for Ping<ClContext> {
        fn largest_specimen<E: SizeEstimator>(estimator: &E, cache: &mut Cache) -> Self {
            Ping {
                creator: LargestSpecimen::largest_specimen(estimator, cache),
                timestamp: LargestSpecimen::largest_specimen(estimator, cache),
                instance_id: LargestSpecimen::largest_specimen(estimator, cache),
                signature: LargestSpecimen::largest_specimen(estimator, cache),
            }
        }
    }

    impl LargestSpecimen for HashedWireUnit<ClContext> {
        fn largest_specimen<E: SizeEstimator>(estimator: &E, cache: &mut Cache) -> Self {
            if let Some(item) = cache.get::<Self>() {
                return item.clone();
            }

            let hash = LargestSpecimen::largest_specimen(estimator, cache);
            let wire_unit = LargestSpecimen::largest_specimen(estimator, cache);
            cache.set(HashedWireUnit { hash, wire_unit }).clone()
        }
    }

    impl LargestSpecimen for WireUnit<ClContext> {
        fn largest_specimen<E: SizeEstimator>(estimator: &E, cache: &mut Cache) -> Self {
            WireUnit {
                panorama: LargestSpecimen::largest_specimen(estimator, cache),
                creator: LargestSpecimen::largest_specimen(estimator, cache),
                instance_id: LargestSpecimen::largest_specimen(estimator, cache),
                value: LargestSpecimen::largest_specimen(estimator, cache),
                seq_number: LargestSpecimen::largest_specimen(estimator, cache),
                timestamp: LargestSpecimen::largest_specimen(estimator, cache),
                round_exp: LargestSpecimen::largest_specimen(estimator, cache),
                endorsed: btree_set_distinct_from_prop(estimator, "validator_count", cache),
            }
        }
    }
}

/// A `WireUnit` together with its hash and a cryptographic signature by its creator.
#[derive(DataSize, Clone, Debug, Eq, PartialEq, Serialize, Deserialize, Hash)]
#[serde(bound(
    serialize = "C::Hash: Serialize",
    deserialize = "C::Hash: Deserialize<'de>",
))]
pub struct SignedWireUnit<C>
where
    C: Context,
{
    pub(crate) hashed_wire_unit: HashedWireUnit<C>,
    pub(crate) signature: C::Signature,
}

impl<C: Context> SignedWireUnit<C> {
    pub(crate) fn new(
        hashed_wire_unit: HashedWireUnit<C>,
        secret_key: &C::ValidatorSecret,
    ) -> Self {
        let signature = secret_key.sign(&hashed_wire_unit.hash);
        SignedWireUnit {
            hashed_wire_unit,
            signature,
        }
    }

    /// Returns the inner `WireUnit`.
    pub fn wire_unit(&self) -> &WireUnit<C> {
        self.hashed_wire_unit.wire_unit()
    }

    /// Returns this unit's hash.
    pub fn hash(&self) -> C::Hash {
        self.hashed_wire_unit.hash()
    }
}

/// A `WireUnit` together with its hash.
#[derive(Clone, DataSize, Debug, Eq, PartialEq, Hash)]
pub struct HashedWireUnit<C>
where
    C: Context,
{
    hash: C::Hash,
    wire_unit: WireUnit<C>,
}

impl<C> HashedWireUnit<C>
where
    C: Context,
{
    /// Computes the unit's hash and creates a new `HashedWireUnit`.
    pub(crate) fn new(wire_unit: WireUnit<C>) -> Self {
        let hash = wire_unit.compute_hash();
        Self::new_with_hash(wire_unit, hash)
    }

    /// Returns the inner `WireUnit`.
    pub fn into_inner(self) -> WireUnit<C> {
        self.wire_unit
    }

    /// Returns a reference to the inner `WireUnit`.
    pub fn wire_unit(&self) -> &WireUnit<C> {
        &self.wire_unit
    }

    /// Returns this unit's hash.
    pub fn hash(&self) -> C::Hash {
        self.hash
    }

    /// Creates a new `HashedWireUnit`. Make sure the `hash` is correct, and identical with the
    /// result of `wire_unit.compute_hash`.
    pub(crate) fn new_with_hash(wire_unit: WireUnit<C>, hash: C::Hash) -> Self {
        HashedWireUnit { hash, wire_unit }
    }
}

impl<C: Context> Serialize for HashedWireUnit<C> {
    fn serialize<S: Serializer>(&self, serializer: S) -> Result<S::Ok, S::Error> {
        self.wire_unit.serialize(serializer)
    }
}

impl<'de, C: Context> Deserialize<'de> for HashedWireUnit<C> {
    fn deserialize<D: Deserializer<'de>>(deserializer: D) -> Result<Self, D::Error> {
        Ok(HashedWireUnit::new(<_>::deserialize(deserializer)?))
    }
}

/// A unit as it is sent over the wire, possibly containing a new block.
#[derive(DataSize, Clone, Eq, PartialEq, Serialize, Deserialize, Hash)]
#[serde(bound(
    serialize = "C::Hash: Serialize",
    deserialize = "C::Hash: Deserialize<'de>",
))]
pub struct WireUnit<C>
where
    C: Context,
{
    /// The panorama of cited units.
    pub panorama: Panorama<C>,
    /// The index of the creator of this unit.
    pub creator: ValidatorIndex,
    /// The consensus instance ID for which this unit was created.
    pub instance_id: C::InstanceId,
    /// The consensus value included in the unit, if any.
    pub value: Option<C::ConsensusValue>,
    /// The sequence number of this unit in the creator's swimlane.
    pub seq_number: u64,
    /// Timestamp of when the unit was created.
    pub timestamp: Timestamp,
    /// The current round exponent of the unit's creator.
    pub round_exp: u8,
    /// The units this unit endorses.
    pub endorsed: BTreeSet<C::Hash>,
}

impl<C: Context> Debug for WireUnit<C> {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        /// A type whose debug implementation prints ".." (without the quotes).
        struct Ellipsis;

        impl Debug for Ellipsis {
            fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
                write!(f, "..")
            }
        }

        f.debug_struct("WireUnit")
            .field("value", &self.value.as_ref().map(|_| Ellipsis))
            .field("creator.0", &self.creator.0)
            .field("instance_id", &self.instance_id)
            .field("seq_number", &self.seq_number)
            .field("timestamp", &self.timestamp.millis())
            .field("panorama", self.panorama.as_ref())
            .field("round_exp", &self.round_exp)
            .field("endorsed", &self.endorsed)
            .finish()
    }
}

impl<C: Context> WireUnit<C> {
    pub(crate) fn into_hashed(self) -> HashedWireUnit<C> {
        HashedWireUnit::new(self)
    }

    /// Returns the creator's previous unit.
    pub fn previous(&self) -> Option<&C::Hash> {
        self.panorama[self.creator].correct()
    }

    /// Returns the unit's hash, which is used as a unit identifier.
    fn compute_hash(&self) -> C::Hash {
        // TODO: Use serialize_into to avoid allocation?
        <C as Context>::hash(&bincode::serialize(self).expect("serialize WireUnit"))
    }
}

/// A set of endorsements for a unit.
#[derive(Clone, DataSize, Debug, Eq, PartialEq, Serialize, Deserialize, Hash)]
#[serde(bound(
    serialize = "C::Hash: Serialize",
    deserialize = "C::Hash: Deserialize<'de>",
))]
pub struct Endorsements<C>
where
    C: Context,
{
    /// The endorsed unit.
    pub unit: C::Hash,
    /// The endorsements for the unit.
    pub endorsers: Vec<(ValidatorIndex, C::Signature)>,
}

impl<C: Context> Endorsements<C> {
    /// Returns hash of the endorsed vote.
    pub fn unit(&self) -> &C::Hash {
        &self.unit
    }

    /// Returns an iterator over validator indexes that endorsed the `unit`.
    pub fn validator_ids(&self) -> impl Iterator<Item = ValidatorIndex> + '_ {
        self.endorsers.iter().map(|(v, _)| *v)
    }
}

impl<C: Context> From<SignedEndorsement<C>> for Endorsements<C> {
    fn from(signed_e: SignedEndorsement<C>) -> Self {
        Endorsements {
            unit: *signed_e.unit(),
            endorsers: vec![(signed_e.validator_idx(), *signed_e.signature())],
        }
    }
}

/// A ping sent by a validator to signal that it is online but has not created new units in a
/// while.
#[derive(Clone, DataSize, Debug, Eq, PartialEq, Serialize, Deserialize, Hash)]
#[serde(bound(
    serialize = "C::Hash: Serialize",
    deserialize = "C::Hash: Deserialize<'de>",
))]
pub struct Ping<C>
where
    C: Context,
{
    creator: ValidatorIndex,
    timestamp: Timestamp,
    instance_id: C::InstanceId,
    signature: C::Signature,
}

impl<C: Context> Ping<C> {
    /// Creates a new signed ping.
    pub(crate) fn new(
        creator: ValidatorIndex,
        timestamp: Timestamp,
        instance_id: C::InstanceId,
        sk: &C::ValidatorSecret,
    ) -> Self {
        let signature = sk.sign(&Self::hash(creator, timestamp, instance_id));
        Ping {
            creator,
            timestamp,
            instance_id,
            signature,
        }
    }

    /// The creator who signals that it is online.
    pub fn creator(&self) -> ValidatorIndex {
        self.creator
    }

    /// The timestamp when the ping was created.
    pub fn timestamp(&self) -> Timestamp {
        self.timestamp
    }

    /// Validates the ping and returns an error if it is not signed by the creator.
    pub(crate) fn validate(
        &self,
        validators: &Validators<C::ValidatorId>,
        our_instance_id: &C::InstanceId,
    ) -> Result<(), VertexError> {
        let Ping {
            creator,
            timestamp,
            instance_id,
            signature,
        } = self;
        if instance_id != our_instance_id {
            return Err(PingError::InstanceId.into());
        }
        let v_id = validators.id(self.creator).ok_or(PingError::Creator)?;
        let hash = Self::hash(*creator, *timestamp, *instance_id);
        if !C::verify_signature(&hash, v_id, signature) {
            return Err(PingError::Signature.into());
        }
        Ok(())
    }

    /// Computes the hash of a ping, i.e. of the creator and timestamp.
    fn hash(creator: ValidatorIndex, timestamp: Timestamp, instance_id: C::InstanceId) -> C::Hash {
        let bytes = bincode::serialize(&(creator, timestamp, instance_id)).expect("serialize Ping");
        <C as Context>::hash(&bytes)
    }
}