feat: precheck triple and presigs

chain-signatures/contract/src/config/impls.rs

@@ -46,6 +46,7 @@ impl Default for TripleConfig {
             min_triples: 1024,
             max_triples: 1024 * MAX_EXPECTED_PARTICIPANTS * NETWORK_MULTIPLIER,
             generation_timeout: min_to_ms(10),
+            preview_limit: 128,
 
             other: Default::default(),
         }
@@ -58,6 +59,7 @@ impl Default for PresignatureConfig {
             min_presignatures: 512,
             max_presignatures: 512 * MAX_EXPECTED_PARTICIPANTS * NETWORK_MULTIPLIER,
             generation_timeout: secs_to_ms(45),
+            preview_limit: 128,
 
             other: Default::default(),
         }

chain-signatures/contract/src/config/mod.rs

@@ -59,6 +59,8 @@ pub struct TripleConfig {
     pub max_triples: u32,
     /// Timeout for triple generation in milliseconds.
     pub generation_timeout: u64,
+    /// Max amount of Triple IDs allowed to be previewed in state.
+    pub preview_limit: u32,
 
     /// The remaining entries that can be present in future forms of the configuration.
     #[serde(flatten)]
@@ -73,6 +75,8 @@ pub struct PresignatureConfig {
     pub max_presignatures: u32,
     /// Timeout for presignature generation in milliseconds.
     pub generation_timeout: u64,
+    /// Max amount of Presignature IDs allowed to be previewed in state.
+    pub preview_limit: u32,
 
     /// The remaining entries that can be present in future forms of the configuration.
     #[serde(flatten)]
@@ -110,12 +114,14 @@ mod tests {
                 "triple": {
                     "min_triples": 10,
                     "max_triples": 100,
-                    "generation_timeout": 10000
+                    "generation_timeout": 10000,
+                    "preview_limit": 128,
                 },
                 "presignature": {
                     "min_presignatures": 10,
                     "max_presignatures": 100,
-                    "generation_timeout": 10000
+                    "generation_timeout": 10000,
+                    "preview_limit": 256,
                 },
                 "signature": {
                     "generation_timeout": 10000,

chain-signatures/node/src/cli.rs

@@ -211,6 +211,14 @@ pub fn run(cmd: Cli) -> anyhow::Result<()> {
                 client_headers.insert(http::header::REFERER, referer_param.parse().unwrap());
             }
 
+            let config = Arc::new(RwLock::new(Config::new(LocalConfig {
+                over: override_config.unwrap_or_else(Default::default),
+                network: NetworkConfig {
+                    cipher_pk: hpke::PublicKey::try_from_bytes(&hex::decode(cipher_pk)?)?,
+                    sign_sk,
+                },
+            })));
+
             tracing::debug!(rpc_addr = rpc_client.rpc_addr(), "rpc client initialized");
             let signer = InMemorySigner::from_secret_key(account_id.clone(), account_sk);
             let (protocol, protocol_state) = MpcSignProtocol::init(
@@ -223,13 +231,7 @@ pub fn run(cmd: Cli) -> anyhow::Result<()> {
                 sign_queue,
                 key_storage,
                 triple_storage,
-                Config::new(LocalConfig {
-                    over: override_config.unwrap_or_else(Default::default),
-                    network: NetworkConfig {
-                        cipher_pk: hpke::PublicKey::try_from_bytes(&hex::decode(cipher_pk)?)?,
-                        sign_sk,
-                    },
-                }),
+                config.clone(),
             );
 
             rt.block_on(async {
@@ -238,7 +240,7 @@ pub fn run(cmd: Cli) -> anyhow::Result<()> {
                 tracing::debug!("protocol thread spawned");
                 let cipher_sk = hpke::SecretKey::try_from_bytes(&hex::decode(cipher_sk)?)?;
                 let web_handle = tokio::spawn(async move {
-                    web::run(web_port, sender, cipher_sk, protocol_state, indexer).await
+                    web::run(web_port, sender, cipher_sk, protocol_state, indexer, config).await
                 });
                 tracing::debug!("protocol http server spawned");
 

chain-signatures/node/src/mesh/connection.rs

@@ -1,11 +1,13 @@
-use std::collections::HashMap;
+use std::collections::{HashMap, HashSet};
 use std::time::{Duration, Instant};
 
 use cait_sith::protocol::Participant;
 use tokio::sync::RwLock;
 use url::Url;
 
 use crate::protocol::contract::primitives::Participants;
+use crate::protocol::presignature::PresignatureId;
+use crate::protocol::triple::TripleId;
 use crate::protocol::ProtocolState;
 use crate::web::StateView;
 
@@ -19,16 +21,19 @@ pub struct Pool {
     http: reqwest::Client,
     connections: RwLock<Participants>,
     potential_connections: RwLock<Participants>,
-    status: RwLock<HashMap<Participant, StateView>>,
-
     /// The currently active participants for this epoch.
     current_active: RwLock<Option<(Participants, Instant)>>,
     // Potentially active participants that we can use to establish a connection in the next epoch.
     potential_active: RwLock<Option<(Participants, Instant)>>,
+
+    pub status: RwLock<HashMap<Participant, StateView>>,
 }
 
 impl Pool {
-    pub async fn ping(&self) -> Participants {
+    pub async fn ping(
+        &mut self,
+        previews: Option<(HashSet<TripleId>, HashSet<PresignatureId>)>,
+    ) -> Participants {
         if let Some((ref active, timestamp)) = *self.current_active.read().await {
             if timestamp.elapsed() < DEFAULT_TIMEOUT {
                 return active.clone();
@@ -37,6 +42,16 @@ impl Pool {
 
         let connections = self.connections.read().await;
 
+        let mut params = HashMap::new();
+        if let Some((triples, presignatures)) = previews {
+            if !triples.is_empty() {
+                params.insert("triple_preview", triples);
+            }
+            if !presignatures.is_empty() {
+                params.insert("presignature_preview", presignatures);
+            }
+        }
+
         let mut status = self.status.write().await;
         let mut participants = Participants::default();
         for (participant, info) in connections.iter() {
@@ -49,13 +64,24 @@ impl Pool {
                 continue;
             };
 
-            let Ok(resp) = self.http.get(url.clone()).send().await else {
-                tracing::warn!(
-                    "Pool.ping resp err participant {:?} url {}",
-                    participant,
-                    url
-                );
-                continue;
+            let resp = match self
+                .http
+                .get(url.clone())
+                .header("content-type", "application/json")
+                .json(&params)
+                .send()
+                .await
+            {
+                Ok(resp) => resp,
+                Err(err) => {
+                    tracing::warn!(
+                        ?err,
+                        "Pool.ping resp err participant {:?} url {}",
+                        participant,
+                        url
+                    );
+                    continue;
+                }
             };
 
             let Ok(state): Result<StateView, _> = resp.json().await else {
@@ -68,6 +94,8 @@ impl Pool {
             };
 
             status.insert(*participant, state);
+            // self.status.insert(*participant, state);
+
             participants.insert(participant, info.clone());
         }
         drop(status);
@@ -77,7 +105,7 @@ impl Pool {
         participants
     }
 
-    pub async fn ping_potential(&self) -> Participants {
+    pub async fn ping_potential(&mut self) -> Participants {
         if let Some((ref active, timestamp)) = *self.potential_active.read().await {
             if timestamp.elapsed() < DEFAULT_TIMEOUT {
                 return active.clone();
@@ -104,7 +132,6 @@ impl Pool {
             status.insert(*participant, state);
             participants.insert(participant, info.clone());
         }
-        drop(status);
 
         let mut potential_active = self.potential_active.write().await;
         *potential_active = Some((participants.clone(), Instant::now()));
@@ -155,4 +182,9 @@ impl Pool {
                 _ => false,
             })
     }
+
+    pub async fn clear_status(&self) {
+        let mut status = self.status.write().await;
+        status.clear();
+    }
 }

chain-signatures/node/src/mesh/mod.rs

@@ -1,5 +1,12 @@
+use std::collections::{HashMap, HashSet};
+
+use cait_sith::protocol::Participant;
+
 use crate::protocol::contract::primitives::Participants;
+use crate::protocol::presignature::PresignatureId;
+use crate::protocol::triple::TripleId;
 use crate::protocol::ProtocolState;
+use crate::web::StateView;
 
 pub mod connection;
 
@@ -64,7 +71,13 @@ impl Mesh {
         self.connections
             .establish_participants(contract_state)
             .await;
-        self.ping().await;
+    }
+
+    /// Ping the active participants such that we can see who is alive.
+    pub async fn ping(&mut self, previews: Option<(HashSet<TripleId>, HashSet<PresignatureId>)>) {
+        self.connections.clear_status().await;
+        self.active_participants = self.connections.ping(previews).await;
+        self.active_potential_participants = self.connections.ping_potential().await;
 
         tracing::debug!(
             active = ?self.active_participants.keys_vec(),
@@ -73,9 +86,7 @@ impl Mesh {
         );
     }
 
-    /// Ping the active participants such that we can see who is alive.
-    pub async fn ping(&mut self) {
-        self.active_participants = self.connections.ping().await;
-        self.active_potential_participants = self.connections.ping_potential().await;
+    pub async fn state_views(&self) -> HashMap<Participant, StateView> {
+        self.connections.status.read().await.clone()
     }
 }

chain-signatures/node/src/protocol/consensus.rs

@@ -31,6 +31,7 @@ use url::Url;
 use near_account_id::AccountId;
 use near_crypto::InMemorySigner;
 
+#[async_trait::async_trait]
 pub trait ConsensusCtx {
     fn my_account_id(&self) -> &AccountId;
     fn http_client(&self) -> &reqwest::Client;
@@ -41,7 +42,7 @@ pub trait ConsensusCtx {
     fn sign_queue(&self) -> Arc<RwLock<SignQueue>>;
     fn secret_storage(&self) -> &SecretNodeStorageBox;
     fn triple_storage(&self) -> LockTripleNodeStorageBox;
-    fn cfg(&self) -> &Config;
+    async fn cfg(&self) -> Config;
 }
 
 #[derive(thiserror::Error, Debug)]
@@ -668,12 +669,13 @@ impl ConsensusProtocol for JoiningState {
                         tracing::info!(
                             "joining(running): sending a transaction to join the participant set"
                         );
+                        let cfg = ctx.cfg().await;
                         ctx.rpc_client()
                             .call(ctx.signer(), ctx.mpc_contract_id(), "join")
                             .args_json(json!({
                                 "url": ctx.my_address(),
-                                "cipher_pk": ctx.cfg().local.network.cipher_pk.to_bytes(),
-                                "sign_pk": ctx.cfg().local.network.sign_sk.public_key(),
+                                "cipher_pk": cfg.local.network.cipher_pk.to_bytes(),
+                                "sign_pk": cfg.local.network.sign_sk.public_key(),
                             }))
                             .max_gas()
                             .retry_exponential(10, 3)