feat(dkg): implement validators

Author: iamquang95

Cargo.lock

@@ -14,6 +14,7 @@ bon.workspace = true
 libp2p.workspace = true
 futures.workspace = true
 tokio.workspace = true
+chrono.workspace = true
 sha2.workspace = true
 tracing.workspace = true
 either.workspace = true
@@ -22,8 +23,10 @@ tokio-util.workspace = true
 pluto-k1util.workspace = true
 pluto-p2p.workspace = true
 pluto-cluster.workspace = true
+pluto-core.workspace = true
 pluto-app.workspace = true
 pluto-crypto.workspace = true
+pluto-eth2api.workspace = true
 pluto-eth1wrap.workspace = true
 pluto-eth2util.workspace = true
 pluto-tracing.workspace = true
@@ -48,6 +51,7 @@ pluto-tracing.workspace = true
 serde_json.workspace = true
 tokio-util.workspace = true
 tempfile.workspace = true
+wiremock.workspace = true
 
 [lints]
 workspace = true

crates/dkg/Cargo.toml

@@ -2,7 +2,7 @@ use std::collections::HashMap;
 
 use pluto_core::{
     signeddata::{SignedDataError, VersionedSignedValidatorRegistration},
-    types::{ParSignedData, PubKey, Signature as CoreSignature},
+    types::{ParSignedData, PubKey},
 };
 use pluto_crypto::{
     blst_impl::BlstImpl,
@@ -213,7 +213,7 @@ pub fn agg_validator_registrations(
 
         res.push(set_registration_signature(
             msg,
-            CoreSignature::new(agg_sig),
+            pluto_core::types::Signature::new(agg_sig),
         )?);
     }
 
@@ -309,7 +309,7 @@ mod tests {
     }
 
     fn partial_signature(sig: Signature, share_idx: u64) -> ParSignedData {
-        ParSignedData::new(CoreSignature::new(sig), share_idx)
+        ParSignedData::new(pluto_core::types::Signature::new(sig), share_idx)
     }
 
     #[test]

crates/dkg/src/aggregate.rs

@@ -1,8 +1,19 @@
 use std::{path, time::Duration};
 
 use bon::Builder;
+use libp2p::PeerId;
 use tokio_util::sync::CancellationToken;
-use tracing::warn;
+use tracing::{info, warn};
+
+pub use crate::{
+    aggregate::{AggregateError, agg_deposit_data, agg_lock_hash_sig, agg_validator_registrations},
+    publish::{PublishError, write_lock_to_api},
+    signing::{SigningError, sign_deposit_msgs, sign_lock_hash, sign_validator_registrations},
+    validators::{
+        ValidatorsError, builder_registration_from_eth2, create_dist_validators,
+        set_registration_signature,
+    },
+};
 
 const DEFAULT_DATA_DIR: &str = ".charon";
 const DEFAULT_DEFINITION_FILE: &str = ".charon/cluster-definition.json";
@@ -212,6 +223,49 @@ fn validate_keymanager_flags(conf: &Config) -> Result<(), DkgError> {
     Ok(())
 }
 
+/// Logs peer summary with peer names and operator addresses.
+pub fn log_peer_summary(
+    current_peer: PeerId,
+    peers: &[pluto_p2p::peer::Peer],
+    operators: &[pluto_cluster::operator::Operator],
+) {
+    for (idx, peer) in peers.iter().enumerate() {
+        let address = operators
+            .get(idx)
+            .filter(|operator| !operator.address.is_empty())
+            .map(|operator| operator.address.as_str());
+        let is_current_peer = peer.id == current_peer;
+
+        if let Some(address) = address {
+            if is_current_peer {
+                info!(
+                    peer = peer.name,
+                    index = peer.index,
+                    address,
+                    you = "⭐️",
+                    "Peer summary"
+                );
+            } else {
+                info!(
+                    peer = peer.name,
+                    index = peer.index,
+                    address,
+                    "Peer summary"
+                );
+            }
+        } else if is_current_peer {
+            info!(
+                peer = peer.name,
+                index = peer.index,
+                you = "⭐️",
+                "Peer summary"
+            );
+        } else {
+            info!(peer = peer.name, index = peer.index, "Peer summary");
+        }
+    }
+}
+
 async fn verify_keymanager_connection(conf: &Config) -> Result<(), DkgError> {
     let addr = conf.keymanager.address.as_str();
 

crates/dkg/src/dkg.rs

@@ -11,11 +11,23 @@ pub mod dkgpb;
 /// Reliable broadcast protocol for DKG messages.
 pub mod bcast;
 
+/// Partial-signature verification and aggregation helpers.
+mod aggregate;
+
 /// General DKG IO operations.
 pub mod disk;
 
 /// Main DKG protocol implementation.
 pub mod dkg;
 
+/// Lock publishing helpers.
+mod publish;
+
 /// Shares distributed to each node in the cluster.
 pub mod share;
+
+/// Local DKG signing helpers.
+mod signing;
+
+/// Registration conversion and distributed-validator assembly helpers.
+mod validators;