use crate::{
Application, Automaton, Block, CertifiableAutomaton, CertifiableBlock, Epochable, Handoff,
Heightable, Relay, Reporter,
marshal::{
ancestry::Ancestry,
application::{
build::{self, Resolved},
gates::{self, GateOutcome, Gates},
relay,
validation::{Stage, is_inferred_reproposal_at_certify},
},
core::{CommitmentFallback, DigestFallback, Mailbox},
standard::{
Standard,
validation::{
Decision, ParentCheck, await_and_validate_parent, precheck_epoch_and_reproposal,
run_app_verify,
},
},
},
simplex::{
Plan,
types::{Activity, Context},
},
types::{Epocher, Round, View},
};
use commonware_actor::Feedback;
use commonware_codec::Codec;
use commonware_cryptography::{Digested, Digestible, certificate::Scheme};
use commonware_macros::select;
use commonware_runtime::{
Clock, Metrics, Spawner,
telemetry::{
metrics::{
MetricsExt as _,
histogram::{Buckets, Timed},
},
traces::TracedExt as _,
},
};
use commonware_utils::channel::{fallible::OneshotExt, oneshot};
use std::{
future::{Future, ready},
sync::Arc,
};
use tracing::{Instrument as _, debug, info_span};
pub struct Deferred<E, S, A, B, ES>
where
S: Scheme,
A: Application<E>,
B: CertifiableBlock + Codec,
ES: Epocher,
{
context: Arc<E>,
application: A,
marshal: Mailbox<S, Standard<B>>,
epocher: ES,
gates: Gates<<B as Digestible>::Digest, Digested<B>>,
build_duration: Timed,
proposal_parent_fetch_duration: Timed,
ancestor_fetch_duration: Timed,
}
impl<E, S, A, B, ES> Clone for Deferred<E, S, A, B, ES>
where
S: Scheme,
A: Application<E>,
B: CertifiableBlock + Codec,
ES: Epocher,
{
fn clone(&self) -> Self {
Self {
context: self.context.clone(),
application: self.application.clone(),
marshal: self.marshal.clone(),
epocher: self.epocher.clone(),
gates: self.gates.clone(),
build_duration: self.build_duration.clone(),
proposal_parent_fetch_duration: self.proposal_parent_fetch_duration.clone(),
ancestor_fetch_duration: self.ancestor_fetch_duration.clone(),
}
}
}
impl<E, S, A, B, ES> Deferred<E, S, A, B, ES>
where
E: Spawner + Metrics + Clock,
S: Scheme,
A: Application<
E,
Block = B,
SigningScheme = S,
Context = Context<B::Digest, S::PublicKey>,
Input = (),
>,
B: CertifiableBlock<Context = <A as Application<E>>::Context> + Codec,
ES: Epocher,
{
pub fn new(context: E, application: A, marshal: Mailbox<S, Standard<B>>, epocher: ES) -> Self {
let build_histogram = context.histogram(
"build_duration",
"Histogram of time taken for the application to build a new block, in seconds",
Buckets::LOCAL,
);
let build_duration = Timed::new(build_histogram);
let parent_fetch_histogram = context.histogram(
"parent_fetch_duration",
"Histogram of time taken to fetch a parent block in proposal, in seconds",
Buckets::LOCAL,
);
let proposal_parent_fetch_duration = Timed::new(parent_fetch_histogram);
let ancestor_fetch_histogram = context.histogram(
"ancestor_fetch_duration",
"Histogram of time taken to fetch a block via the ancestry stream, in seconds",
Buckets::LOCAL,
);
let ancestor_fetch_duration = Timed::new(ancestor_fetch_histogram);
Self {
context: Arc::new(context),
application,
marshal,
epocher,
gates: Gates::new(),
build_duration,
proposal_parent_fetch_duration,
ancestor_fetch_duration,
}
}
#[inline]
fn deferred_verify(
&self,
context: <Self as Automaton>::Context,
block: Digested<B>,
parent_request: oneshot::Receiver<Digested<B>>,
stage: Stage,
) -> oneshot::Receiver<GateOutcome> {
let marshal = self.marshal.clone();
let mut application = self.application.clone();
let (mut tx, rx) = oneshot::channel();
let ancestor_fetch_duration = self.ancestor_fetch_duration.clone();
let runtime_context = self
.context
.child("deferred_verify")
.with_attribute("round", context.round);
let span = info_span!(
"marshal.deferred.verify.deferred",
round = %context.round
);
runtime_context.spawn(move |runtime_context| {
async move {
let round = context.round;
let store = stage.store(&marshal, round, block.clone());
let verify = async {
let parent = match await_and_validate_parent(
context.parent.1,
&block,
parent_request,
&mut tx,
)
.await
{
Some(ParentCheck::Valid(parent)) => parent,
Some(ParentCheck::Invalid) => return Some(false),
None => return None,
};
run_app_verify(
runtime_context,
context,
block.clone(),
parent,
&mut application,
&marshal,
&mut tx,
ancestor_fetch_duration,
)
.await
};
let (verdict, durable) = futures::join!(verify, store);
if let Some(application_valid) = gates::resolve(verdict, durable) {
tx.send_lossy(GateOutcome::Ready(application_valid));
}
}
.instrument(span)
});
rx
}
fn certify_from_embedded_context(
&self,
round: Round,
digest: B::Digest,
) -> oneshot::Receiver<bool> {
debug!(
?round,
?digest,
"subscribing to block for certification using embedded context"
);
let block_rx = self
.marshal
.subscribe_by_digest(digest, DigestFallback::FetchByRound { round });
let marshaled = self.clone();
let epocher = self.epocher.clone();
let (mut tx, rx) = oneshot::channel();
let context = self.context.child("certify").with_attribute("round", round);
context.spawn(move |_| {
async move {
let block = select! {
_ = tx.closed() => {
debug!(
reason = "consensus dropped receiver",
"skipping certification"
);
return;
},
result = block_rx => match result {
Ok(block) => block,
Err(_) => {
debug!(
?digest,
reason = "failed to fetch block for certification",
"skipping certification"
);
return;
}
},
};
let embedded_context = block.context();
let is_reproposal = is_inferred_reproposal_at_certify(
&epocher,
block.height(),
embedded_context.round,
round,
);
if is_reproposal {
if !marshaled.marshal.certified(round, block).await {
return;
}
tx.send_lossy(true);
return;
}
let (parent_view, parent_commitment) = embedded_context.parent;
let parent_request = marshaled.marshal.subscribe_by_commitment(
parent_commitment,
CommitmentFallback::FetchByRound {
round: Round::new(embedded_context.epoch(), parent_view),
},
);
let verify_rx = marshaled.deferred_verify(
embedded_context,
block,
parent_request,
Stage::Certified,
);
gates::forward(tx, verify_rx, |result| match result {
GateOutcome::Ready(result) => Some(result),
GateOutcome::Recover => None,
})
.await;
}
.instrument(info_span!(
"marshal.deferred.certify.embedded",
round = %round,
digest = %digest
))
});
rx
}
fn certify_from_existing_task(
&self,
round: Round,
digest: B::Digest,
task: oneshot::Receiver<GateOutcome>,
) -> oneshot::Receiver<bool> {
self.marshal.hint_notarized(round, digest);
let marshaled = self.clone();
let (tx, rx) = oneshot::channel();
let context = self
.context
.child("certify_existing")
.with_attribute("round", round);
context.spawn(move |_| {
gates::drive(tx, task, round, digest, move || {
marshaled.certify_from_embedded_context(round, digest)
})
.instrument(info_span!(
"marshal.deferred.certify.existing",
round = %round,
digest = %digest
))
});
rx
}
fn checks(
&self,
consensus_context: Context<B::Digest, S::PublicKey>,
) -> impl Future<Output = Resolved<B::Digest, Digested<B>, impl Ancestry<B>>> + Send + 'static
{
let context = self.context.clone();
let marshal = self.marshal.clone();
let epocher = self.epocher.clone();
let proposal_parent_fetch_duration = self.proposal_parent_fetch_duration.clone();
let ancestor_fetch_duration = self.ancestor_fetch_duration.clone();
let build_duration = self.build_duration.clone();
async move {
let (parent_view, parent_commitment) = consensus_context.parent;
let parent_request = marshal.subscribe_by_commitment(
parent_commitment,
CommitmentFallback::FetchByRound {
round: Round::new(consensus_context.epoch(), parent_view),
},
);
let parent_timer = proposal_parent_fetch_duration.timer(&*context);
let Ok(parent) = parent_request.await else {
debug!(
?parent_commitment,
reason = "failed to fetch parent block",
"skipping proposal"
);
return Resolved::Skip;
};
parent_timer.observe(&*context);
if parent_view != View::zero() && parent.parent() != parent.context().parent.1 {
debug!(
?parent_commitment,
reason = "parent does not link to its context parent",
"skipping proposal"
);
return Resolved::Skip;
}
let last_in_epoch = epocher
.last(consensus_context.epoch())
.expect("current epoch should exist");
if parent.height() == last_in_epoch {
return Resolved::Reuse(parent.digest(), parent);
}
let ancestor_stream =
marshal.ancestor_stream(context.clone(), [parent], ancestor_fetch_duration);
Resolved::Build(ancestor_stream, build_duration.timer(&*context), ())
}
}
}
impl<E, S, A, B, ES> Automaton for Deferred<E, S, A, B, ES>
where
E: Spawner + Metrics + Clock,
S: Scheme,
A: Application<
E,
Block = B,
SigningScheme = S,
Context = Context<B::Digest, S::PublicKey>,
Input = (),
>,
B: CertifiableBlock<Context = <A as Application<E>>::Context> + Codec,
ES: Epocher,
{
type Digest = B::Digest;
type Context = Context<Self::Digest, S::PublicKey>;
#[allow(clippy::async_yields_async)]
#[tracing::instrument(name = "marshal.deferred.propose", level = "info", skip_all, fields(round = %consensus_context.round))]
async fn propose(
&mut self,
consensus_context: Context<Self::Digest, S::PublicKey>,
) -> oneshot::Receiver<Self::Digest> {
let checks = self.checks(consensus_context.clone());
let span = info_span!(
"marshal.deferred.propose.task",
round = %consensus_context.round
);
let round = consensus_context.round;
let (parent_view, parent_commitment) = consensus_context.parent;
build::propose(
self.context.as_ref(),
&self.application,
self.gates.clone(),
consensus_context,
checks,
span,
move || {
info_span!(
"marshal.deferred.application.propose",
round = %round,
parent_view = parent_view.traced(),
parent = %parent_commitment
)
},
|block, ()| ready((block.digest(), block)),
)
}
#[allow(clippy::async_yields_async)]
#[tracing::instrument(name = "marshal.deferred.verify", level = "info", skip_all, fields(round = %context.round, digest = %digest))]
async fn verify(
&mut self,
context: Context<Self::Digest, S::PublicKey>,
digest: Self::Digest,
) -> oneshot::Receiver<bool> {
let marshal = self.marshal.clone();
let marshaled = self.clone();
let round = context.round;
let block_request = marshal.subscribe_by_digest(digest, DigestFallback::Wait);
let (task_tx, task_rx) = oneshot::channel();
self.gates.insert(round, digest, task_rx);
let (mut tx, rx) = oneshot::channel();
let runtime_context = self
.context
.child("optimistic_verify")
.with_attribute("round", round);
runtime_context.spawn(move |_| {
async move {
let parent_request = (digest != context.parent.1).then(|| {
let (parent_view, parent_commitment) = context.parent;
marshal.subscribe_by_commitment(
parent_commitment,
CommitmentFallback::FetchByRound {
round: Round::new(context.epoch(), parent_view),
},
)
});
let block = select! {
_ = tx.closed() => {
debug!(
reason = "consensus dropped receiver",
"skipping optimistic verification"
);
return;
},
result = block_request => match result {
Ok(block) => block,
Err(_) => {
debug!(
?digest,
reason = "failed to fetch block for optimistic verification",
"skipping optimistic verification"
);
return;
}
},
};
let Some(decision) = precheck_epoch_and_reproposal(
&marshaled.epocher,
&marshal,
&context,
digest,
block,
)
.await
else {
return;
};
let block = match decision {
Decision::Complete(valid) => {
task_tx.send_lossy(GateOutcome::Ready(valid));
tx.send_lossy(valid);
return;
}
Decision::Continue(block) => block,
};
let parent_request =
parent_request.expect("non-reproposal has a parent subscription");
if block.context() != context {
debug!(
?context,
block_context = ?block.context(),
"block-embedded context does not match consensus context during optimistic verification"
);
task_tx.send_lossy(GateOutcome::Recover);
tx.send_lossy(false);
return;
}
let deferred_rx = marshaled
.deferred_verify(context, block, parent_request, Stage::Verified);
tx.send_lossy(true);
gates::forward(task_tx, deferred_rx, Some).await;
}
.instrument(info_span!(
"marshal.deferred.verify.optimistic",
round = %round,
digest = %digest
))
});
rx
}
}
impl<E, S, A, B, ES> CertifiableAutomaton for Deferred<E, S, A, B, ES>
where
E: Spawner + Metrics + Clock,
S: Scheme,
A: Application<
E,
Block = B,
SigningScheme = S,
Context = Context<B::Digest, S::PublicKey>,
Input = (),
>,
B: CertifiableBlock<Context = <A as Application<E>>::Context> + Codec,
ES: Epocher,
{
#[allow(clippy::async_yields_async)]
#[tracing::instrument(name = "marshal.deferred.prepare", level = "info", skip_all, fields(round = %consensus_context.round))]
async fn prepare(
&mut self,
consensus_context: Context<Self::Digest, S::PublicKey>,
) -> oneshot::Receiver<Handoff<Self::Digest>> {
let checks = self.checks(consensus_context.clone());
let span = info_span!(
"marshal.deferred.prepare.task",
round = %consensus_context.round
);
let round = consensus_context.round;
let (parent_view, parent_commitment) = consensus_context.parent;
build::prepare(
self.context.as_ref(),
&self.application,
self.gates.clone(),
consensus_context,
checks,
span,
move || {
info_span!(
"marshal.deferred.application.prepare",
round = %round,
parent_view = parent_view.traced(),
parent = %parent_commitment
)
},
|block, ()| ready((block.digest(), block)),
)
}
#[allow(clippy::async_yields_async)]
#[tracing::instrument(name = "marshal.deferred.certify", level = "info", skip_all, fields(round = %round, digest = %digest))]
async fn certify(&mut self, round: Round, digest: Self::Digest) -> oneshot::Receiver<bool> {
let task = self.gates.claim(round, digest, |block, ack| {
self.marshal.verified_deferred(round, block, ack)
});
if let Some(task) = task {
return self.certify_from_existing_task(round, digest, task);
}
self.certify_from_embedded_context(round, digest)
}
}
impl<E, S, A, B, ES> Relay for Deferred<E, S, A, B, ES>
where
E: Send + Sync + 'static,
S: Scheme,
A: Application<E, Block = B, Context = Context<B::Digest, S::PublicKey>>,
B: CertifiableBlock<Context = <A as Application<E>>::Context> + Codec,
ES: Epocher,
{
type Digest = B::Digest;
type PublicKey = S::PublicKey;
type Plan = Plan<S::PublicKey>;
fn broadcast(&mut self, commitment: Self::Digest, plan: Plan<S::PublicKey>) -> Feedback {
relay::broadcast(&self.gates, &self.marshal, commitment, plan)
}
}
impl<E, S, A, B, ES> Reporter for Deferred<E, S, A, B, ES>
where
E: Send + Sync + 'static,
S: Scheme,
A: Application<E, Block = B, Context = Context<B::Digest, S::PublicKey>>,
B: CertifiableBlock<Context = <A as Application<E>>::Context> + Codec,
ES: Epocher,
{
type Activity = Activity<S, B::Digest>;
fn report(&mut self, activity: Self::Activity) -> Feedback {
let retired = match &activity {
Activity::Finalization(finalization) => finalization.round(),
activity => Round::new(activity.epoch(), View::zero()),
};
self.gates.retain_after(retired);
self.marshal.report(activity)
}
}
#[cfg(test)]
mod tests {
use super::Deferred;
use crate::{
Automaton, CertifiableAutomaton, Relay,
marshal::{
application::gates::Gates,
mocks::{
harness::{
B, BLOCKS_PER_EPOCH, Ctx, D, NAMESPACE, NUM_VALIDATORS, S, StandardHarness,
TestHarness, V, default_leader, make_raw_block,
setup_network_with_participants,
},
verifying::{GatedVerifyingApp, MockVerifyingApp},
},
standard::tests::ready,
},
simplex::{Plan, scheme::bls12381_threshold::vrf as bls12381_threshold_vrf},
types::{Epoch, Epocher, FixedEpocher, Height, Round, View},
};
use commonware_broadcast::Broadcaster;
use commonware_cryptography::{
Digested, Digestible, Hasher as _,
certificate::{ConstantProvider, mocks::Fixture},
sha256::Sha256,
};
use commonware_macros::{select, test_traced};
use commonware_runtime::{Clock, Runner, Supervisor as _, deterministic};
use commonware_utils::{NZUsize, channel::fallible::OneshotExt};
use std::time::Duration;
#[test_traced("INFO")]
fn test_certify_lower_view_after_higher_view() {
let runner = deterministic::Runner::timed(Duration::from_secs(60));
runner.start(|mut context| async move {
let Fixture {
participants,
schemes,
..
} = bls12381_threshold_vrf::fixture::<V, _>(&mut context, NAMESPACE, NUM_VALIDATORS);
let mut oracle = setup_network_with_participants(
context.child("network"),
NZUsize!(1),
participants.clone(),
1024 * 1024,
)
.await;
let me = participants[0].clone();
let setup = StandardHarness::setup_validator(
context.child("validator").with_attribute("index", 0),
&mut oracle,
me.clone(),
ConstantProvider::new(schemes[0].clone()),
)
.await;
let marshal = setup.mailbox;
let genesis = make_raw_block(Sha256::hash(&[b""]), Height::zero(), 0);
let mock_app: MockVerifyingApp<B, S> = MockVerifyingApp::new();
let mut marshaled = Deferred::new(
context.child("deferred"),
mock_app,
marshal.clone(),
FixedEpocher::new(BLOCKS_PER_EPOCH),
);
let parent = make_raw_block(genesis.digest(), Height::new(1), 100);
let parent_digest = parent.digest();
assert!(
marshal
.verified(
Round::new(Epoch::new(0), View::new(1)),
ready(parent.clone())
)
.await
);
let round_a = Round::new(Epoch::new(0), View::new(5));
let context_a = Ctx {
round: round_a,
leader: me.clone(),
parent: (View::new(1), parent_digest),
};
let block_a = B::new::<Sha256>(context_a.clone(), parent_digest, Height::new(2), 200);
let commitment_a = block_a.digest();
assert!(marshal.verified(round_a, ready(block_a.clone())).await);
let round_b = Round::new(Epoch::new(0), View::new(10));
let context_b = Ctx {
round: round_b,
leader: me.clone(),
parent: (View::new(1), parent_digest),
};
let block_b = B::new::<Sha256>(context_b.clone(), parent_digest, Height::new(2), 300);
let commitment_b = block_b.digest();
assert!(marshal.verified(round_b, ready(block_b.clone())).await);
context.sleep(Duration::from_millis(10)).await;
let _ = marshaled.verify(context_a, commitment_a).await.await;
let _ = marshaled.verify(context_b, commitment_b).await.await;
let certify_b = marshaled.certify(round_b, commitment_b).await;
assert!(
certify_b.await.unwrap(),
"Block B certification should succeed"
);
let certify_a = marshaled.certify(round_a, commitment_a).await;
select! {
result = certify_a => {
assert!(result.unwrap(), "Block A certification should succeed");
},
_ = context.sleep(Duration::from_secs(5)) => {
panic!("Block A certification timed out");
},
}
})
}
#[test_traced("WARN")]
fn test_marshaled_rejects_unsupported_epoch() {
#[derive(Clone)]
struct LimitedEpocher {
inner: FixedEpocher,
max_epoch: u64,
}
impl Epocher for LimitedEpocher {
fn containing(&self, height: Height) -> Option<crate::types::EpochInfo> {
let bounds = self.inner.containing(height)?;
if bounds.epoch().get() > self.max_epoch {
None
} else {
Some(bounds)
}
}
fn first(&self, epoch: Epoch) -> Option<Height> {
if epoch.get() > self.max_epoch {
None
} else {
self.inner.first(epoch)
}
}
fn last(&self, epoch: Epoch) -> Option<Height> {
if epoch.get() > self.max_epoch {
None
} else {
self.inner.last(epoch)
}
}
}
let runner = deterministic::Runner::timed(Duration::from_secs(60));
runner.start(|mut context| async move {
let Fixture {
participants,
schemes,
..
} = bls12381_threshold_vrf::fixture::<V, _>(&mut context, NAMESPACE, NUM_VALIDATORS);
let mut oracle = setup_network_with_participants(
context.child("network"),
NZUsize!(1),
participants.clone(),
1024 * 1024,
)
.await;
let me = participants[0].clone();
let setup = StandardHarness::setup_validator(
context.child("validator").with_attribute("index", 0),
&mut oracle,
me.clone(),
ConstantProvider::new(schemes[0].clone()),
)
.await;
let marshal = setup.mailbox;
let genesis = make_raw_block(Sha256::hash(&[b""]), Height::zero(), 0);
let mock_app: MockVerifyingApp<B, S> = MockVerifyingApp::new();
let limited_epocher = LimitedEpocher {
inner: FixedEpocher::new(BLOCKS_PER_EPOCH),
max_epoch: 0,
};
let mut marshaled = Deferred::new(
context.child("deferred"),
mock_app,
marshal.clone(),
limited_epocher,
);
let parent_ctx = Ctx {
round: Round::new(Epoch::zero(), View::new(19)),
leader: default_leader(),
parent: (View::zero(), genesis.digest()),
};
let parent =
B::new::<Sha256>(parent_ctx.clone(), genesis.digest(), Height::new(19), 1000);
let parent_digest = parent.digest();
assert!(
marshal
.clone()
.verified(
Round::new(Epoch::zero(), View::new(19)),
ready(parent.clone())
)
.await
);
let unsupported_round = Round::new(Epoch::new(1), View::new(20));
let unsupported_context = Ctx {
round: unsupported_round,
leader: me.clone(),
parent: (View::new(19), parent_digest),
};
let block = B::new::<Sha256>(
unsupported_context.clone(),
parent_digest,
Height::new(20),
2000,
);
let block_commitment = block.digest();
assert!(
marshal
.clone()
.verified(unsupported_round, ready(block.clone()))
.await
);
context.sleep(Duration::from_millis(10)).await;
let verify_result = marshaled
.verify(unsupported_context, block_commitment)
.await;
let optimistic_result = verify_result.await;
assert!(
!optimistic_result.unwrap(),
"Optimistic verify should reject block in unsupported epoch"
);
})
}
#[test_traced("WARN")]
fn test_marshaled_rejects_mismatched_context() {
let runner = deterministic::Runner::timed(Duration::from_secs(30));
runner.start(|mut context| async move {
let Fixture {
participants,
schemes,
..
} = bls12381_threshold_vrf::fixture::<V, _>(&mut context, NAMESPACE, NUM_VALIDATORS);
let mut oracle = setup_network_with_participants(
context.child("network"),
NZUsize!(1),
participants.clone(),
1024 * 1024,
)
.await;
let me = participants[0].clone();
let setup = StandardHarness::setup_validator(
context.child("validator").with_attribute("index", 0),
&mut oracle,
me.clone(),
ConstantProvider::new(schemes[0].clone()),
)
.await;
let marshal = setup.mailbox;
let genesis = make_raw_block(Sha256::hash(&[b""]), Height::zero(), 0);
let mock_app: MockVerifyingApp<B, S> = MockVerifyingApp::new();
let mut marshaled = Deferred::new(
context.child("deferred"),
mock_app,
marshal.clone(),
FixedEpocher::new(BLOCKS_PER_EPOCH),
);
let parent_ctx = Ctx {
round: Round::new(Epoch::zero(), View::new(1)),
leader: default_leader(),
parent: (View::zero(), genesis.digest()),
};
let parent = B::new::<Sha256>(parent_ctx, genesis.digest(), Height::new(1), 100);
let parent_commitment = parent.digest();
assert!(
marshal
.clone()
.verified(
Round::new(Epoch::zero(), View::new(1)),
ready(parent.clone())
)
.await
);
let round_a = Round::new(Epoch::zero(), View::new(2));
let context_a = Ctx {
round: round_a,
leader: me.clone(),
parent: (View::new(1), parent_commitment),
};
let block_a = B::new::<Sha256>(context_a, parent.digest(), Height::new(2), 200);
let commitment_a = block_a.digest();
assert!(marshal.verified(round_a, ready(block_a)).await);
context.sleep(Duration::from_millis(10)).await;
let round_b = Round::new(Epoch::zero(), View::new(3));
let context_b = Ctx {
round: round_b,
leader: participants[1].clone(),
parent: (View::new(1), parent_commitment),
};
let verify_rx = marshaled.verify(context_b, commitment_a).await;
select! {
result = verify_rx => {
assert!(
!result.unwrap(),
"mismatched context hash should be rejected"
);
},
_ = context.sleep(Duration::from_secs(5)) => {
panic!("verify should reject mismatched context hash promptly");
},
}
})
}
#[test_traced("WARN")]
fn test_deferred_certify_recovers_after_verify_receiver_drop() {
let runner = deterministic::Runner::timed(Duration::from_secs(30));
runner.start(|mut context| async move {
let Fixture {
participants,
schemes,
..
} = bls12381_threshold_vrf::fixture::<V, _>(&mut context, NAMESPACE, NUM_VALIDATORS);
let mut oracle = setup_network_with_participants(
context.child("network"),
NZUsize!(1),
participants.clone(),
1024 * 1024,
)
.await;
let me = participants[0].clone();
let setup = StandardHarness::setup_validator(
context.child("validator").with_attribute("index", 0),
&mut oracle,
me.clone(),
ConstantProvider::new(schemes[0].clone()),
)
.await;
let marshal = setup.mailbox;
let genesis = make_raw_block(Sha256::hash(&[b""]), Height::zero(), 0);
let mock_app: MockVerifyingApp<B, S> = MockVerifyingApp::new();
let mut marshaled = Deferred::new(
context.child("deferred"),
mock_app,
marshal.clone(),
FixedEpocher::new(BLOCKS_PER_EPOCH),
);
let round = Round::new(Epoch::zero(), View::new(1));
let block_context = Ctx {
round,
leader: me,
parent: (View::zero(), genesis.digest()),
};
let block =
B::new::<Sha256>(block_context.clone(), genesis.digest(), Height::new(1), 100);
let digest = block.digest();
let verify_rx = marshaled.verify(block_context, digest).await;
drop(verify_rx);
context.sleep(Duration::from_millis(10)).await;
assert!(marshal.verified(round, ready(block)).await);
let certify_rx = marshaled.certify(round, digest).await;
select! {
result = certify_rx => {
assert!(
result.expect("certify result missing"),
"certify should recover after verify receiver drop"
);
},
_ = context.sleep(Duration::from_secs(5)) => {
panic!("certify should recover promptly after verify drop");
},
}
});
}
#[test_traced("WARN")]
fn test_deferred_store_overlaps_app_verify() {
let runner = deterministic::Runner::timed(Duration::from_secs(30));
runner.start(|mut context| async move {
let Fixture {
participants,
schemes,
..
} = bls12381_threshold_vrf::fixture::<V, _>(&mut context, NAMESPACE, NUM_VALIDATORS);
let mut oracle = setup_network_with_participants(
context.child("network"),
NZUsize!(1),
participants.clone(),
1024 * 1024,
)
.await;
let me = participants[0].clone();
let setup = StandardHarness::setup_validator(
context.child("validator").with_attribute("index", 0),
&mut oracle,
me.clone(),
ConstantProvider::new(schemes[0].clone()),
)
.await;
let marshal = setup.mailbox;
let buffer = setup.extra;
let genesis = make_raw_block(Sha256::hash(&[b""]), Height::zero(), 0);
let (mock_app, verify_started, release_verify): (GatedVerifyingApp<B, S>, _, _) =
GatedVerifyingApp::new();
let mut marshaled = Deferred::new(
context.child("deferred"),
mock_app,
marshal.clone(),
FixedEpocher::new(BLOCKS_PER_EPOCH),
);
let parent = make_raw_block(genesis.digest(), Height::new(1), 100);
let parent_digest = parent.digest();
let child_round = Round::new(Epoch::zero(), View::new(2));
let child_ctx = Ctx {
round: child_round,
leader: me,
parent: (View::new(1), parent_digest),
};
let child = B::new::<Sha256>(child_ctx.clone(), parent_digest, Height::new(2), 200);
let child_digest = child.digest();
assert!(
buffer
.broadcast(commonware_p2p::Recipients::Some(vec![]), ready(parent))
.accepted(),
"buffer broadcast for parent should be accepted"
);
assert!(
buffer
.broadcast(commonware_p2p::Recipients::Some(vec![]), ready(child))
.accepted(),
"buffer broadcast for child should be accepted"
);
let optimistic_rx = marshaled.verify(child_ctx, child_digest).await;
assert!(
optimistic_rx
.await
.expect("optimistic verify should resolve"),
"optimistic verify should accept the available block"
);
verify_started
.await
.expect("verify should reach the gated application");
assert!(
marshal.get_block(&child_digest).await.is_some(),
"the store request runs concurrently with app.verify, so the block is locally queryable while verification is still gated"
);
release_verify.send_lossy(());
let certify_rx = marshaled.certify(child_round, child_digest).await;
select! {
result = certify_rx => {
assert!(
result.expect("certify result missing"),
"certify should succeed once verification passes"
);
},
_ = context.sleep(Duration::from_secs(5)) => {
panic!("certify should resolve after verification is released");
},
}
});
}
#[test_traced("WARN")]
fn test_propose_reuses_reproposed_boundary_block_on_restart() {
let runner = deterministic::Runner::timed(Duration::from_secs(30));
runner.start(|mut context| async move {
let Fixture {
participants,
schemes,
..
} = bls12381_threshold_vrf::fixture::<V, _>(&mut context, NAMESPACE, NUM_VALIDATORS);
let mut oracle = setup_network_with_participants(
context.child("network"),
NZUsize!(1),
participants.clone(),
1024 * 1024,
)
.await;
let me = participants[0].clone();
let setup = StandardHarness::setup_validator(
context.child("validator").with_attribute("index", 0),
&mut oracle,
me.clone(),
ConstantProvider::new(schemes[0].clone()),
)
.await;
let marshal = setup.mailbox;
let genesis = make_raw_block(Sha256::hash(&[b""]), Height::zero(), 0);
let boundary_height = Height::new(BLOCKS_PER_EPOCH.get() - 1);
let boundary_round = Round::new(Epoch::zero(), View::new(boundary_height.get()));
let boundary_block = B::new::<Sha256>(
Ctx {
round: boundary_round,
leader: default_leader(),
parent: (View::zero(), genesis.digest()),
},
genesis.digest(),
boundary_height,
1900,
);
let boundary_digest = boundary_block.digest();
let round = Round::new(Epoch::zero(), View::new(boundary_height.get() + 1));
assert!(marshal.verified(round, ready(boundary_block)).await);
let ctx = Ctx {
round,
leader: me.clone(),
parent: (View::new(boundary_height.get()), boundary_digest),
};
let (mock_app, verify_started, _release_verify): (GatedVerifyingApp<B, S>, _, _) =
GatedVerifyingApp::new();
let mut marshaled = Deferred::new(
context.child("deferred"),
mock_app,
marshal.clone(),
FixedEpocher::new(BLOCKS_PER_EPOCH),
);
let digest_rx = marshaled.propose(ctx).await;
let digest = digest_rx.await.expect("propose must return a digest");
assert_eq!(
digest, boundary_digest,
"propose must re-propose the boundary block marshal already persisted for this round"
);
let _ = marshaled.broadcast(digest, Plan::Propose { round });
let certify_rx = marshaled.certify(round, digest).await;
select! {
result = certify_rx => {
assert!(
result.expect("certify result missing"),
"re-proposed boundary block must certify through the relay handshake"
);
},
_ = verify_started => {
panic!("certifying a re-proposed boundary block must not run app verification");
},
}
});
}
#[test_traced("WARN")]
fn test_deferred_propose_then_certify_persists_block() {
let runner = deterministic::Runner::timed(Duration::from_secs(30));
runner.start(|mut context| async move {
let Fixture {
participants,
schemes,
..
} = bls12381_threshold_vrf::fixture::<V, _>(&mut context, NAMESPACE, NUM_VALIDATORS);
let mut oracle = setup_network_with_participants(
context.child("network"),
NZUsize!(1),
participants.clone(),
1024 * 1024,
)
.await;
let me = participants[0].clone();
let setup = StandardHarness::setup_validator(
context.child("validator").with_attribute("index", 0),
&mut oracle,
me.clone(),
ConstantProvider::new(schemes[0].clone()),
)
.await;
let marshal = setup.mailbox;
let actor_handle = setup.actor_handle;
let genesis = make_raw_block(Sha256::hash(&[b""]), Height::zero(), 0);
let parent_round = Round::new(Epoch::zero(), View::new(1));
let parent_ctx = Ctx {
round: parent_round,
leader: default_leader(),
parent: (View::zero(), genesis.digest()),
};
let parent = B::new::<Sha256>(parent_ctx, genesis.digest(), Height::new(1), 100);
let parent_digest = parent.digest();
assert!(marshal.verified(parent_round, ready(parent)).await);
let round = Round::new(Epoch::zero(), View::new(2));
let ctx = Ctx {
round,
leader: me.clone(),
parent: (View::new(1), parent_digest),
};
let child = B::new::<Sha256>(ctx.clone(), parent_digest, Height::new(2), 200);
let child_digest = child.digest();
let mock_app: MockVerifyingApp<B, S> =
MockVerifyingApp::new().with_propose_result(child);
let mut marshaled = Deferred::new(
context.child("deferred"),
mock_app,
marshal.clone(),
FixedEpocher::new(BLOCKS_PER_EPOCH),
);
let digest = marshaled
.propose(ctx)
.await
.await
.expect("propose must return a digest");
assert_eq!(
digest, child_digest,
"propose must return the built block's digest"
);
assert!(
marshaled
.certify(round, child_digest)
.await
.await
.expect("certify result missing"),
"certify must succeed for the leader's own proposal"
);
actor_handle.abort();
let _ = actor_handle.await;
drop(marshaled);
drop(marshal);
let setup2 = StandardHarness::setup_validator(
context
.child("validator_restart")
.with_attribute("index", 0),
&mut oracle,
me,
ConstantProvider::new(schemes[0].clone()),
)
.await;
let marshal2 = setup2.mailbox;
assert!(
marshal2.get_block(&child_digest).await.is_some(),
"certify resolved true for the leader's own proposal so the block must be durable"
);
});
}
struct EquivocationFixture {
marshaled: Deferred<deterministic::Context, S, MockVerifyingApp<B, S>, B, FixedEpocher>,
round: Round,
digest: <B as Digestible>::Digest,
embedded_ctx: Ctx,
equivocating_ctx: Ctx,
_extra: <StandardHarness as TestHarness>::ValidatorExtra,
}
async fn equivocation_fixture(
context: &mut deterministic::Context,
app: MockVerifyingApp<B, S>,
) -> EquivocationFixture {
let Fixture {
participants,
schemes,
..
} = bls12381_threshold_vrf::fixture::<V, _>(context, NAMESPACE, NUM_VALIDATORS);
let mut oracle = setup_network_with_participants(
context.child("network"),
NZUsize!(1),
participants.clone(),
1024 * 1024,
)
.await;
let me = participants[0].clone();
let setup = StandardHarness::setup_validator(
context.child("validator").with_attribute("index", 0),
&mut oracle,
me,
ConstantProvider::new(schemes[0].clone()),
)
.await;
let marshal = setup.mailbox;
let buffer = setup.extra;
let genesis = make_raw_block(Sha256::hash(&[b""]), Height::zero(), 0);
let leader = participants[1].clone();
let certified_round = Round::new(Epoch::zero(), View::new(1));
let certified_ctx = Ctx {
round: certified_round,
leader: default_leader(),
parent: (View::zero(), genesis.digest()),
};
let certified = B::new::<Sha256>(certified_ctx, genesis.digest(), Height::new(1), 100);
let certified_digest = certified.digest();
assert!(marshal.verified(certified_round, ready(certified)).await);
let notarized_round = Round::new(Epoch::zero(), View::new(2));
let notarized_ctx = Ctx {
round: notarized_round,
leader: leader.clone(),
parent: (View::new(1), certified_digest),
};
let notarized = B::new::<Sha256>(notarized_ctx, certified_digest, Height::new(2), 200);
let notarized_digest = notarized.digest();
assert!(marshal.verified(notarized_round, ready(notarized)).await);
let round = Round::new(Epoch::zero(), View::new(3));
let embedded_ctx = Ctx {
round,
leader: leader.clone(),
parent: (View::new(2), notarized_digest),
};
let block = B::new::<Sha256>(embedded_ctx.clone(), notarized_digest, Height::new(3), 300);
let digest = block.digest();
assert!(
buffer
.broadcast(commonware_p2p::Recipients::Some(vec![]), ready(block))
.accepted(),
"buffer broadcast for the candidate should be accepted"
);
let equivocating_ctx = Ctx {
round,
leader,
parent: (View::new(1), certified_digest),
};
let marshaled = Deferred::new(
context.child("deferred"),
app,
marshal,
FixedEpocher::new(BLOCKS_PER_EPOCH),
);
context.sleep(Duration::from_millis(10)).await;
EquivocationFixture {
marshaled,
round,
digest,
embedded_ctx,
equivocating_ctx,
_extra: buffer,
}
}
#[test_traced("WARN")]
fn test_certify_not_poisoned_by_equivocating_parent_verify() {
let runner = deterministic::Runner::timed(Duration::from_secs(30));
runner.start(|mut context| async move {
let mut fixture = equivocation_fixture(&mut context, MockVerifyingApp::new()).await;
let verify_rx = fixture
.marshaled
.verify(fixture.equivocating_ctx.clone(), fixture.digest)
.await;
assert!(
!verify_rx.await.expect("verify result missing"),
"the equivocating proposal must not be notarized"
);
let certify_rx = fixture
.marshaled
.certify(fixture.round, fixture.digest)
.await;
select! {
result = certify_rx => {
assert!(
result.expect("certify result missing"),
"certify of the notarized digest must not adopt the verdict computed under the equivocating context"
);
},
_ = context.sleep(Duration::from_secs(5)) => {
panic!("certify should resolve promptly");
},
}
});
}
#[test_traced("WARN")]
fn test_certify_honors_application_rejection() {
let runner = deterministic::Runner::timed(Duration::from_secs(30));
runner.start(|mut context| async move {
let mut fixture =
equivocation_fixture(&mut context, MockVerifyingApp::with_verify_result(false))
.await;
let verify_rx = fixture
.marshaled
.verify(fixture.embedded_ctx.clone(), fixture.digest)
.await;
assert!(
verify_rx.await.expect("verify result missing"),
"optimistic verify accepts an available block with a matching context"
);
let certify_rx = fixture
.marshaled
.certify(fixture.round, fixture.digest)
.await;
select! {
result = certify_rx => {
assert!(
!result.expect("certify result missing"),
"certify must propagate the application rejection"
);
},
_ = context.sleep(Duration::from_secs(5)) => {
panic!("certify should resolve promptly");
},
}
});
}
#[test_traced("WARN")]
fn test_certify_without_prior_verify_honors_application_rejection() {
let runner = deterministic::Runner::timed(Duration::from_secs(30));
runner.start(|mut context| async move {
let mut fixture =
equivocation_fixture(&mut context, MockVerifyingApp::with_verify_result(false))
.await;
let certify_rx = fixture
.marshaled
.certify(fixture.round, fixture.digest)
.await;
select! {
result = certify_rx => {
assert!(
!result.expect("certify result missing"),
"certify must propagate the application rejection"
);
},
_ = context.sleep(Duration::from_secs(5)) => {
panic!("certify should resolve promptly");
},
}
});
}
impl Deferred<deterministic::Context, S, MockVerifyingApp<B, S>, B, FixedEpocher> {
pub(in crate::marshal::standard) const fn gates(&self) -> &Gates<D, Digested<B>> {
&self.gates
}
}
}