fix load reshare params for new committee

RockX-SG · Mar 1, 2023 · 75bdf2a · 75bdf2a
1 parent 0321b07
commit 75bdf2a
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/dkg/messages.go b/dkg/messages.go
@@ -219,6 +219,15 @@ func (reshareMsg *ReshareParams) LoadFromStorage(pk types.ValidatorPK, storage S
 	return nil
 }
 
+func inOldCommittee(op types.OperatorID, l []types.OperatorID) bool {
+	for _, opID := range l {
+		if opID == op {
+			return true
+		}
+	}
+	return false
+}
+
 // Output is the last message in every DKG which marks a specific node's end of process
 type Output struct {
 	// RequestID for the DKG instance (not used for signing)

diff --git a/dkg/node.go b/dkg/node.go
@@ -65,8 +65,10 @@ func (n *Node) newRunner(id RequestID, initMsg *Init) (Runner, error) {
 
 func (n *Node) newResharingRunner(id RequestID, reshareMsg *Reshare) (Runner, error) {
 	reshareParams := &ReshareParams{}
-	if err := reshareParams.LoadFromStorage(reshareMsg.ValidatorPK, n.config.Storage); err != nil {
-		return nil, err
+	if inOldCommittee(n.operator.OperatorID, reshareMsg.OldOperatorIDs) {
+		if err := reshareParams.LoadFromStorage(reshareMsg.ValidatorPK, n.config.Storage); err != nil {
+			return nil, err
+		}
 	}
 
 	r := &runner{