mpc_client/protocols/cggmp/
key_init.rs

//! Key init generation for CGGMP.
use crate::{
    protocols::{Bridge, Driver},
    NetworkTransport, Transport,
};
use async_trait::async_trait;
use mpc_protocol::{hex, Event, SessionState};

use super::{Error, Result};
use mpc_driver::{
    cggmp::KeyInitDriver as CggmpDriver,
    synedrion::{
        ecdsa::{SigningKey, VerifyingKey},
        KeyShare, SchemeParams, SessionId,
    },
};

/// CGGMP key initialization driver.
pub struct KeyInitDriver<P>
where
    P: SchemeParams + 'static,
{
    bridge: Bridge<CggmpDriver<P>>,
}

impl<P> KeyInitDriver<P>
where
    P: SchemeParams + 'static,
{
    /// Create a new CGGMP key generator.
    pub fn new(
        transport: Transport,
        session: SessionState,
        session_id: SessionId,
        signer: SigningKey,
        verifiers: Vec<VerifyingKey>,
    ) -> Result<Self> {
        let party_number = session
            .party_number(transport.public_key())
            .ok_or_else(|| {
                Error::NotSessionParticipant(hex::encode(
                    transport.public_key(),
                ))
            })?;

        let driver = CggmpDriver::new(session_id, signer, verifiers)?;

        let bridge = Bridge {
            transport,
            driver: Some(driver),
            session,
            party_number,
        };
        Ok(Self { bridge })
    }
}

#[async_trait]
impl<P> Driver for KeyInitDriver<P>
where
    P: SchemeParams + 'static,
{
    type Output = KeyShare<P, VerifyingKey>;

    async fn handle_event(
        &mut self,
        event: Event,
    ) -> Result<Option<Self::Output>> {
        Ok(self.bridge.handle_event(event).await?)
    }

    async fn execute(&mut self) -> Result<()> {
        Ok(self.bridge.execute().await?)
    }

    fn into_transport(self) -> Transport {
        self.bridge.transport
    }
}

impl<P> From<KeyInitDriver<P>> for Transport
where
    P: SchemeParams + 'static,
{
    fn from(value: KeyInitDriver<P>) -> Self {
        value.bridge.transport
    }
}
mpc_client/protocols/cggmp/key_init.rs

mpc_client/protocols/cggmp/
key_init.rs
