clear nonce after signing with the state machine

Trust-Machines · Dec 17, 2024 · 89b850f · 89b850f
1 parent c4c6e69
commit 89b850f
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/state_machine/signer/mod.rs b/src/state_machine/signer/mod.rs
@@ -326,7 +326,7 @@ impl<SignerType: SignerTrait> Signer<SignerType> {
                 self.dkg_private_shares(dkg_private_shares, rng)
             }
             Message::SignatureShareRequest(sign_share_request) => {
-                self.sign_share_request(sign_share_request)
+                self.sign_share_request(sign_share_request, rng)
             }
             Message::NonceRequest(nonce_request) => self.nonce_request(nonce_request, rng),
             _ => Ok(vec![]), // TODO
@@ -565,9 +565,10 @@ impl<SignerType: SignerTrait> Signer<SignerType> {
         Ok(msgs)
     }
 
-    fn sign_share_request(
+    fn sign_share_request<R: RngCore + CryptoRng>(
         &mut self,
         sign_request: &SignatureShareRequest,
+        rng: &mut R,
     ) -> Result<Vec<Message>, Error> {
         let mut msgs = vec![];
 
@@ -611,6 +612,8 @@ impl<SignerType: SignerTrait> Signer<SignerType> {
                     }
                 };
 
+                self.signer.gen_nonces(rng);
+
                 let response = SignatureShareResponse {
                     dkg_id: sign_request.dkg_id,
                     sign_id: sign_request.sign_id,