latticex-foundation / opentss Goto Github PK

pub mod communication;
pub mod protocols;
pub mod utilities;

use crate::communication::sending_messages::*;
use curv::elliptic::curves::{Point, Scalar, Secp256k1};
use protocols::multi_party::dmz21::common::*;
use protocols::multi_party::dmz21::keygen::*;
pub type CU = Secp256k1;
pub type FE = Scalar<Secp256k1>;
pub type GE = Point<Secp256k1>;

fn main() {
    // step 1：新建一个Phase对象。
    println!("第 1 步：新建一个Phase对象。");
    let partyid = "1".to_string();
    let params = Parameters {
        threshold: 1,
        share_count: 3,
    };
    let party_ids = vec!["1".to_string(), "2".to_string(), "3".to_string()];
    let mut keygen = KeyGenPhase::new(partyid, params, &Some(party_ids)).unwrap();
    // step 2：首先调用process_begin，返回下一轮要发送的消息。
    println!("第 2 步：首先调用process_begin，返回下一轮要发送的消息。");
    let sending_msg: SendingMessages = keygen.process_begin().unwrap();
    match sending_msg {
        SendingMessages::BroadcastMessage(msg) => {
            // broadcast the msg to all(including self).
            println!("1");
        }
        SendingMessages::P2pMessage(msg) => {
            // send according to the k,v in the msg. k is the index which v will to be sent to.
            println!("2");
        }
        SendingMessages::SubsetMessage(msg) => {
            // send according to the k in the party_ids or subset(used in sign phase). k is the index which msg will to be sent to.
            println!("3");
        }
        _ => {}
    }

    // step 3：通过 处理消息msg_handler。
    println!("第 3 步：通过 处理消息msg_handler。");
    loop {
        // let (recv_from, recv_msg) = According to the last round of SendingMessages
        let recv_from = "".to_string();
        let recv_msg = vec![0u8];
        let sending_msg = keygen.msg_handler(recv_from, &recv_msg).unwrap();
        match sending_msg {
            SendingMessages::KeyGenSuccessWithResult(msg) => {
                println!("{}", msg);
                // got the keygen result
                break;
            }
            _ => {
                // other sending messages, ref Step 2.
            }
        }
    }
}