use active signers to get active key_ids, then pass them in DkgPrivat…

…eBegin
Trust-Machines · Jan 3, 2024 · 89fc17f · 89fc17f
1 parent 919ad5c
commit 89fc17f
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/state_machine/coordinator/fire.rs b/src/state_machine/coordinator/fire.rs
@@ -316,9 +316,15 @@ impl<Aggregator: AggregatorTrait> Coordinator<Aggregator> {
             "DKG Round {}: Starting Private Share Distribution",
             self.current_dkg_id
         );
+        let active_key_ids = self
+            .dkg_public_shares
+            .keys()
+            .flat_map(|signer_id| self.config.signer_key_ids[signer_id].clone())
+            .collect::<Vec<u32>>();
+
         let dkg_begin = DkgPrivateBegin {
             dkg_id: self.current_dkg_id,
-            key_ids: (0..self.config.num_keys).collect(),
+            key_ids: active_key_ids,
         };
         let dkg_private_begin_msg = Packet {
             sig: dkg_begin