fix: store reshared key after generation (#615)

near · May 30, 2024 · df5c78c · df5c78c
1 parent 7a4b85b
commit df5c78c
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/node/src/protocol/cryptography.rs b/node/src/protocol/cryptography.rs
@@ -216,7 +216,7 @@ impl CryptographicProtocol for WaitingForConsensusState {
 impl CryptographicProtocol for ResharingState {
     async fn progress<C: CryptographicCtx + Send + Sync>(
         mut self,
-        ctx: C,
+        mut ctx: C,
     ) -> Result<NodeState, CryptographicError> {
         // TODO: we are not using active potential participants here, but we should in the future.
         // Currently resharing protocol does not timeout and restart with new set of participants.
@@ -300,6 +300,13 @@ impl CryptographicProtocol for ResharingState {
                 }
                 Action::Return(private_share) => {
                     tracing::debug!("resharing: successfully completed key reshare");
+                    ctx.secret_storage()
+                        .store(&PersistentNodeData {
+                            epoch: self.old_epoch + 1,
+                            private_share,
+                            public_key: self.public_key,
+                        })
+                        .await?;
 
                     // Send any leftover messages.
                     let failures = self