DKG level method for validators to create refresh transcripts

nucypher · May 20, 2024 · 3339680 · 3339680
1 parent 0819612
commit 3339680
Showing 1 changed file with 16 additions and 2 deletions.
diff --git a/ferveo/src/dkg.rs b/ferveo/src/dkg.rs
@@ -9,8 +9,9 @@ use rand::RngCore;
 use serde::{Deserialize, Serialize};
 
 use crate::{
-    assert_no_share_duplicates, AggregatedTranscript, Error, EthereumAddress,
-    PubliclyVerifiableParams, PubliclyVerifiableSS, Result, Validator,
+    assert_no_share_duplicates, refresh, AggregatedTranscript, Error,
+    EthereumAddress, PubliclyVerifiableParams, PubliclyVerifiableSS, Result,
+    UpdateTranscript, Validator,
 };
 
 pub type DomainPoint<E> = <E as Pairing>::ScalarField;
@@ -226,6 +227,19 @@ impl<E: Pairing> PubliclyVerifiableDkg<E> {
 
         Ok(())
     }
+
+    // Returns a new refresh transcript for current validators in DKG
+    // TODO: Allow to pass a parameter to restrict target validators
+    pub fn generate_refresh_transcript<R: RngCore>(
+        &self,
+        rng: &mut R,
+    ) -> Result<refresh::UpdateTranscript<E>> {
+        Ok(UpdateTranscript::create_refresh_updates(
+            &self.domain_and_key_map(),
+            self.dkg_params.security_threshold(),
+            rng,
+        ))
+    }
 }
 
 /// Test initializing DKG