feat: add dkg-stress tool

.github/workflows/dkg-runner.yml

@@ -40,11 +40,14 @@ jobs:
       fail-fast: false
       matrix:
         include:
-          - name: 4 Charon nodes
-            id: 4-charon
-            pluto_nodes: 0
-            charon_nodes: 4
-
+          - name: 2 Charon + 2 Pluto nodes
+            id: 2-charon-2-pluto
+            pluto_nodes: 2
+            charon_nodes: 2
+          - name: 4 Pluto nodes
+            id: 4-pluto
+            pluto_nodes: 4
+            charon_nodes: 0
     steps:
       - name: Checkout
         uses: actions/checkout@v6

Cargo.toml

@@ -20,6 +20,7 @@ members = [
     "crates/peerinfo",
     "crates/frost",
 ]
+exclude = ["tools/dkg-stress"]
 resolver = "3"
 
 [workspace.package]

crates/cli/src/main.rs

@@ -32,12 +32,31 @@ async fn run() -> std::result::Result<(), CliError> {
     let matches = cmd.get_matches();
     let cli = Cli::from_arg_matches(&matches)?;
 
-    // Top level cancellation token for graceful shutdown on Ctrl+C
+    // Top level cancellation token for graceful shutdown on Ctrl+C / SIGTERM.
     let ct = CancellationToken::new();
     tokio::spawn({
         let ct = ct.clone();
         async move {
-            let _ = tokio::signal::ctrl_c().await;
+            #[cfg(unix)]
+            {
+                use tokio::signal::unix::{SignalKind, signal};
+                let mut sigterm = match signal(SignalKind::terminate()) {
+                    Ok(s) => s,
+                    Err(_) => {
+                        let _ = tokio::signal::ctrl_c().await;
+                        ct.cancel();
+                        return;
+                    }
+                };
+                tokio::select! {
+                    _ = tokio::signal::ctrl_c() => {}
+                    _ = sigterm.recv() => {}
+                }
+            }
+            #[cfg(not(unix))]
+            {
+                let _ = tokio::signal::ctrl_c().await;
+            }
             ct.cancel();
         }
     });

crates/cluster/src/definition.rs

@@ -767,7 +767,7 @@ impl Definition {
 
     /// Returns true if the provided definition version supports partial
     /// deposits.
-    fn support_partial_deposits(version: &str) -> bool {
+    pub fn support_partial_deposits(version: &str) -> bool {
         !matches!(
             version,
             V1_0 | V1_1 | V1_2 | V1_3 | V1_4 | V1_5 | V1_6 | V1_7

crates/dkg/Cargo.toml

@@ -30,6 +30,7 @@ pluto-eth2api.workspace = true
 pluto-eth1wrap.workspace = true
 pluto-eth2util.workspace = true
 pluto-parsigex.workspace = true
+pluto-peerinfo.workspace = true
 pluto-frost.workspace = true
 async-trait.workspace = true
 pluto-tracing.workspace = true

crates/dkg/src/bcast/error.rs

@@ -155,6 +155,10 @@ pub enum Error {
     #[error("missing protobuf field: {0}")]
     MissingField(&'static str),
 
+    /// A typed broadcast message failed protocol-specific validation.
+    #[error("invalid message: {0}")]
+    InvalidMessage(&'static str),
+
     /// Protobuf encoding failed.
     #[error("protobuf encode failed: {0}")]
     Encode(#[from] prost::EncodeError),

crates/dkg/src/disk.rs

@@ -131,7 +131,7 @@ pub async fn load_definition(
         if conf.no_verify {
             warn!(
                 error = %error,
-                "Ignoring failed cluster definition signatures verification due to --no-verify flag"
+                "Ignoring failed cluster definition signature verification due to --no-verify flag"
             );
         } else {
             return Err(DiskError::ClusterDefinitionError(error));