Bounds check split_at

2023-01-12 13:05:28 +01:00
parent e9144fb1bd
commit c7cb4ae0d5
1 changed files with 7 additions and 1 deletions
@@ -265,8 +265,14 @@ impl RewardedSetUpdater {
        let new_rewarded_set =
            self.determine_rewarded_set(all_mixnodes, rewarding_parameters.rewarded_set_size)?;

+        let empty = vec![];
+
        let (active_set, reserve_set) =
-            new_rewarded_set.split_at(rewarding_parameters.active_set_size as usize);
+            if new_rewarded_set.len() <= rewarding_parameters.active_set_size as usize {
+                (new_rewarded_set.as_slice(), empty.as_slice())
+            } else {
+                new_rewarded_set.split_at(rewarding_parameters.active_set_size as usize)
+            };

        let mut active_set_layer_assignments = self.determine_layers(active_set).await?;
        let reserve_set_layer_assignments = self.determine_layers(reserve_set).await?;