POC - Mithril signature diffusion with a "fake Cardano network layer"

.github/workflows/actions/build-upload-mithril-artifact/action.yml

@@ -53,4 +53,5 @@ runs:
           target/release/mithril-signer.exe
           target/release/mithril-relay
           target/release/mithril-relay.exe
+          target/release/signature-network-node
         if-no-files-found: error

.github/workflows/ci.yml

@@ -38,7 +38,13 @@ jobs:
       - name: Build Mithril workspace & publish artifacts
         uses: ./.github/workflows/actions/build-upload-mithril-artifact
         with:
-          binaries-build-args: --bin mithril-aggregator --bin mithril-signer --bin mithril-client --bin mithril-relay --features bundle_openssl,full
+          binaries-build-args: |
+            --bin mithril-aggregator \
+            --bin mithril-signer \
+            --bin mithril-client \
+            --bin mithril-relay  \
+            --bin signature-network-node \
+            --features bundle_openssl,full
           libraries-build-args: --package mithril-stm --package mithril-client --features full,unstable
 
       - name: Build Debian packages
@@ -342,6 +348,7 @@ jobs:
           chmod +x ./bin/mithril-client
           chmod +x ./bin/mithril-signer
           chmod +x ./bin/mithril-relay
+          chmod +x ./bin/signature-network-node
           chmod +x ./mithril-end-to-end
           mkdir artifacts
 

Cargo.toml

@@ -23,6 +23,7 @@ members = [
   "mithril-stm",
   "mithril-test-lab/mithril-aggregator-fake",
   "mithril-test-lab/mithril-end-to-end",
+  "mithril-test-lab/signature-network-node",
 ]
 
 [workspace.package]

mithril-aggregator/src/commands/serve_command.rs

@@ -255,6 +255,17 @@ impl ServeCommand {
             });
         }
 
+        if let Some(signature_consumer) = dependencies_builder
+            .create_signature_consumer()
+            .await
+            .with_context(|| "Dependencies Builder can not create signature consumer")?
+        {
+            join_set.spawn(async move {
+                signature_consumer.listen().await;
+                Ok(())
+            });
+        }
+
         join_set.spawn(async { tokio::signal::ctrl_c().await.map_err(|e| e.to_string()) });
         dependencies_builder.vanish().await;
 

mithril-aggregator/src/configuration.rs

@@ -56,6 +56,12 @@ pub struct Configuration {
     #[example = "`/tmp/cardano.sock`"]
     pub cardano_node_socket_path: PathBuf,
 
+    /// Path of the socket used by a signature network node to publish signatures
+    ///
+    /// If not set, the signatures will be published to the aggregator using the HTTP API.
+    #[example = "`/tmp/signature-network-node_1/node.sock`"]
+    pub signature_network_node_socket_path: Option<PathBuf>,
+
     /// Cardano node version.
     ///
     /// **NOTE**: This cannot be verified for now (see [this
@@ -225,6 +231,7 @@ impl Configuration {
             environment: ExecutionEnvironment::Test,
             cardano_cli_path: PathBuf::new(),
             cardano_node_socket_path: PathBuf::new(),
+            signature_network_node_socket_path: None,
             cardano_node_version: "0.0.1".to_string(),
             network_magic: Some(42),
             network: "devnet".to_string(),

mithril-aggregator/src/dependency_injection/builder.rs

@@ -50,6 +50,7 @@ use mithril_persistence::{
 };
 
 use super::{DependenciesBuilderError, EpochServiceWrapper, Result};
+use crate::services::SignatureConsumer;
 use crate::{
     artifact_builder::{
         CardanoImmutableFilesFullArtifactBuilder, CardanoStakeDistributionArtifactBuilder,
@@ -1453,6 +1454,27 @@ impl DependenciesBuilder {
         Ok(runtime)
     }
 
+    /// Create a [SignatureConsumer] instance if `signature_network_node_socket_path`
+    /// is set in the [Configuration].
+    pub async fn create_signature_consumer(&mut self) -> Result<Option<SignatureConsumer>> {
+        let signature_network_node_socket_path = &self
+            .configuration
+            .signature_network_node_socket_path
+            .clone();
+        match signature_network_node_socket_path {
+            Some(socket_path) => {
+                let consumer = SignatureConsumer::new(
+                    socket_path,
+                    Duration::from_millis(50),
+                    self.get_certifier_service().await?,
+                    &self.root_logger(),
+                );
+                Ok(Some(consumer))
+            }
+            None => Ok(None),
+        }
+    }
+
     /// Create the HTTP route instance
     pub async fn create_http_routes(
         &mut self,

mithril-aggregator/src/runtime/state_machine.rs

@@ -173,7 +173,10 @@ impl AggregatorRuntime {
                     .with_context(|| "AggregatorRuntime can not get the current open message")?
                 {
                     // transition READY > SIGNING
-                    info!(self.logger, "→ Transitioning to SIGNING");
+                    info!(
+                        self.logger, "→ Transitioning to SIGNING";
+                        "signed_entity_type" => ?open_message.signed_entity_type
+                    );
                     let new_state = self
                         .transition_from_ready_to_signing(last_time_point.clone(), open_message.clone())
                         .await.with_context(|| format!("AggregatorRuntime can not perform a transition from READY state to SIGNING with entity_type: '{:?}'", open_message.signed_entity_type))?;
@@ -220,10 +223,14 @@ impl AggregatorRuntime {
                     self.state = AggregatorState::Ready(new_state);
                 } else {
                     // SIGNING > READY
+                    let signed_entity_type = state.open_message.signed_entity_type.clone();
                     let new_state = self
                         .transition_from_signing_to_ready_multisignature(state)
                         .await?;
-                    info!(self.logger, "→ A multi-signature has been created, build an artifact & a certificate and transitioning back to READY");
+                    info!(
+                        self.logger, "→ A multi-signature has been created, build an artifact & a certificate and transitioning back to READY";
+                        "signed_entity_type" => ?signed_entity_type
+                    );
                     self.state = AggregatorState::Ready(new_state);
                 }
             }

mithril-aggregator/src/services/mod.rs

@@ -15,6 +15,7 @@ mod epoch_service;
 mod message;
 mod prover;
 mod signable_builder;
+mod signature_consumer;
 mod signed_entity;
 mod stake_distribution;
 mod upkeep;
@@ -25,6 +26,7 @@ pub use epoch_service::*;
 pub use message::*;
 pub use prover::*;
 pub use signable_builder::*;
+pub use signature_consumer::*;
 pub use signed_entity::*;
 pub use stake_distribution::*;
 pub use upkeep::*;