#![no_std]
#![forbid(unsafe_code)]
#[cfg(feature = "std")]
extern crate std;
use core::{fmt, marker::PhantomData};
use eth_valkyoth_primitives::{BlockNumber, ChainId, UnixTimestamp};
#[non_exhaustive]
#[derive(Clone, Copy, Debug, Eq, PartialEq)]
pub enum ProtocolError {
Feature(FeatureError),
Fork(ForkError),
InvalidStateTransition,
}
impl ProtocolError {
#[must_use]
pub const fn code(self) -> &'static str {
match self {
Self::Feature(error) => error.code(),
Self::Fork(error) => error.code(),
Self::InvalidStateTransition => "ETH_PROTOCOL_INVALID_STATE_TRANSITION",
}
}
#[must_use]
pub const fn message(self) -> &'static str {
match self {
Self::Feature(error) => error.message(),
Self::Fork(error) => error.message(),
Self::InvalidStateTransition => {
"validation state transition is not allowed from the current state"
}
}
}
#[must_use]
pub const fn category(self) -> ProtocolErrorCategory {
match self {
Self::Feature(_) => ProtocolErrorCategory::Feature,
Self::Fork(_) => ProtocolErrorCategory::Fork,
Self::InvalidStateTransition => ProtocolErrorCategory::State,
}
}
}
impl fmt::Display for ProtocolError {
fn fmt(&self, formatter: &mut fmt::Formatter<'_>) -> fmt::Result {
formatter.write_str(self.message())
}
}
#[cfg(feature = "std")]
impl std::error::Error for ProtocolError {}
#[non_exhaustive]
#[derive(Clone, Copy, Debug, Eq, PartialEq)]
pub enum ProtocolErrorCategory {
Feature,
Fork,
State,
}
#[non_exhaustive]
#[derive(Clone, Copy, Debug, Eq, PartialEq)]
pub enum FeatureError {
Disabled,
Unsupported,
}
impl FeatureError {
#[must_use]
pub const fn code(self) -> &'static str {
match self {
Self::Disabled => "ETH_FEATURE_DISABLED",
Self::Unsupported => "ETH_FEATURE_UNSUPPORTED",
}
}
#[must_use]
pub const fn message(self) -> &'static str {
match self {
Self::Disabled => "feature is disabled for this operation",
Self::Unsupported => "feature is not supported by this crate version",
}
}
}
impl fmt::Display for FeatureError {
fn fmt(&self, formatter: &mut fmt::Formatter<'_>) -> fmt::Result {
formatter.write_str(self.message())
}
}
#[cfg(feature = "std")]
impl std::error::Error for FeatureError {}
#[non_exhaustive]
#[derive(Clone, Copy, Debug, Eq, PartialEq)]
pub enum ForkError {
Unsupported,
Inactive,
MissingActivation,
}
impl ForkError {
#[must_use]
pub const fn code(self) -> &'static str {
match self {
Self::Unsupported => "ETH_FORK_UNSUPPORTED",
Self::Inactive => "ETH_FORK_INACTIVE",
Self::MissingActivation => "ETH_FORK_MISSING_ACTIVATION",
}
}
#[must_use]
pub const fn message(self) -> &'static str {
match self {
Self::Unsupported => "fork is not supported by this crate version",
Self::Inactive => "fork is not active for the validation context",
Self::MissingActivation => "fork activation data is incomplete",
}
}
}
impl fmt::Display for ForkError {
fn fmt(&self, formatter: &mut fmt::Formatter<'_>) -> fmt::Result {
formatter.write_str(self.message())
}
}
#[cfg(feature = "std")]
impl std::error::Error for ForkError {}
#[derive(Clone, Copy, Debug, Eq, PartialEq)]
pub enum ForkActivation {
BlockOnly {
activation_block: BlockNumber,
},
BlockAndTimestamp {
activation_block: BlockNumber,
activation_timestamp: UnixTimestamp,
},
}
#[derive(Clone, Copy, Debug, Eq, PartialEq)]
pub struct ForkSpec {
pub chain_id: ChainId,
pub activation: ForkActivation,
}
#[derive(Clone, Copy, Debug, Eq, PartialEq)]
pub struct ValidationContext {
pub fork: ForkSpec,
pub block_number: BlockNumber,
pub timestamp: UnixTimestamp,
}
impl ValidationContext {
#[must_use]
pub fn fork_is_active(self) -> bool {
match self.fork.activation {
ForkActivation::BlockOnly { activation_block } => self.block_number >= activation_block,
ForkActivation::BlockAndTimestamp {
activation_block,
activation_timestamp,
} => self.block_number >= activation_block && self.timestamp >= activation_timestamp,
}
}
}
#[derive(Clone, Copy, Debug, Eq, PartialEq)]
pub struct Decoded;
#[derive(Clone, Copy, Debug, Eq, PartialEq)]
pub struct Canonical;
#[derive(Clone, Copy, Debug, Eq, PartialEq)]
pub struct ForkValidated;
#[derive(Clone, Copy, Debug, Eq, PartialEq)]
pub struct SenderRecovered;
#[derive(Clone, Copy, Debug, Eq, PartialEq)]
pub struct Transaction<State> {
_state: PhantomData<State>,
}
impl Transaction<Decoded> {
#[must_use]
#[cfg(test)]
pub(crate) const fn decoded() -> Self {
Self {
_state: PhantomData,
}
}
#[must_use]
pub const fn into_canonical(self) -> Transaction<Canonical> {
Transaction {
_state: PhantomData,
}
}
}
impl Transaction<Canonical> {
#[must_use]
pub const fn into_fork_validated(self) -> Transaction<ForkValidated> {
Transaction {
_state: PhantomData,
}
}
}
impl Transaction<ForkValidated> {
#[must_use]
pub const fn into_sender_recovered(self) -> Transaction<SenderRecovered> {
Transaction {
_state: PhantomData,
}
}
}
#[cfg(test)]
mod tests {
use super::*;
extern crate std;
use std::string::ToString;
#[test]
fn activation_requires_block_and_time() {
let context = ValidationContext {
fork: ForkSpec {
chain_id: ChainId::new(1),
activation: ForkActivation::BlockAndTimestamp {
activation_block: BlockNumber::new(10),
activation_timestamp: UnixTimestamp::new(20),
},
},
block_number: BlockNumber::new(10),
timestamp: UnixTimestamp::new(19),
};
assert!(!context.fork_is_active());
}
#[test]
fn block_only_activation_ignores_timestamp() {
let context = ValidationContext {
fork: ForkSpec {
chain_id: ChainId::new(1),
activation: ForkActivation::BlockOnly {
activation_block: BlockNumber::new(10),
},
},
block_number: BlockNumber::new(10),
timestamp: UnixTimestamp::new(0),
};
assert!(context.fork_is_active());
}
#[test]
fn transaction_typestate_advances_in_order() {
let transaction = Transaction::decoded()
.into_canonical()
.into_fork_validated()
.into_sender_recovered();
assert_eq!(
transaction,
Transaction::<SenderRecovered> {
_state: PhantomData
}
);
}
#[test]
fn protocol_errors_have_stable_codes_and_categories() {
let error = ProtocolError::Fork(ForkError::Inactive);
assert_eq!(error.code(), "ETH_FORK_INACTIVE");
assert_eq!(
error.message(),
"fork is not active for the validation context"
);
assert_eq!(error.category(), ProtocolErrorCategory::Fork);
assert_eq!(
error.to_string(),
"fork is not active for the validation context"
);
}
#[test]
fn feature_errors_format_without_payloads() {
let error = ProtocolError::Feature(FeatureError::Unsupported);
assert_eq!(error.code(), "ETH_FEATURE_UNSUPPORTED");
assert_eq!(
FeatureError::Unsupported.to_string(),
"feature is not supported by this crate version"
);
}
}