use crate::simplex::{
scheme,
types::{Finalize, Notarize, Proposal, Vote},
};
use commonware_codec::{DecodeExt, Encode};
use commonware_cryptography::{certificate::Scheme, Hasher};
use commonware_math::algebra::Random;
use commonware_p2p::{Receiver, Recipients, Sender};
use commonware_runtime::{spawn_cell, Clock, ContextCell, Handle, Spawner};
use rand_core::CryptoRngCore;
use std::marker::PhantomData;
use tracing::debug;
pub struct Config<S: Scheme> {
pub scheme: S,
}
pub struct Conflicter<E: Clock + CryptoRngCore + Spawner, S: Scheme, H: Hasher> {
context: ContextCell<E>,
scheme: S,
_hasher: PhantomData<H>,
}
impl<E, S, H> Conflicter<E, S, H>
where
E: Clock + CryptoRngCore + Spawner,
S: scheme::Scheme<H::Digest>,
H: Hasher,
{
pub fn new(context: E, cfg: Config<S>) -> Self {
Self {
context: ContextCell::new(context),
scheme: cfg.scheme,
_hasher: PhantomData,
}
}
pub fn start(mut self, vote_network: (impl Sender, impl Receiver)) -> Handle<()> {
spawn_cell!(self.context, self.run(vote_network).await)
}
async fn run(mut self, vote_network: (impl Sender, impl Receiver)) {
let (mut sender, mut receiver) = vote_network;
while let Ok((s, msg)) = receiver.recv().await {
let msg = match Vote::<S, H::Digest>::decode(msg) {
Ok(msg) => msg,
Err(err) => {
debug!(?err, sender = ?s, "failed to decode message");
continue;
}
};
match msg {
Vote::Notarize(notarize) => {
let payload = H::Digest::random(&mut self.context);
let proposal =
Proposal::new(notarize.round(), notarize.proposal.parent, payload);
let n = Notarize::<S, _>::sign(&self.scheme, proposal).unwrap();
let msg = Vote::Notarize(n).encode();
sender.send(Recipients::All, msg, true).await.unwrap();
let n = Notarize::<S, _>::sign(&self.scheme, notarize.proposal).unwrap();
let msg = Vote::Notarize(n).encode();
sender.send(Recipients::All, msg, true).await.unwrap();
}
Vote::Finalize(finalize) => {
let payload = H::Digest::random(&mut self.context);
let proposal =
Proposal::new(finalize.round(), finalize.proposal.parent, payload);
let f = Finalize::<S, _>::sign(&self.scheme, proposal).unwrap();
let msg = Vote::Finalize(f).encode();
sender.send(Recipients::All, msg, true).await.unwrap();
let f = Finalize::<S, _>::sign(&self.scheme, finalize.proposal).unwrap();
let msg = Vote::Finalize(f).encode();
sender.send(Recipients::All, msg, true).await.unwrap();
}
_ => continue,
}
}
}
}