Fix wraparound when leader schedule close to 0

helius-labs · Aug 5, 2024 · f73a78d · f73a78d
1 parent cc4b526
commit f73a78d
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/validator-firewall/src/leader_tracker.rs b/validator-firewall/src/leader_tracker.rs
@@ -1,3 +1,4 @@
+use std::ops::Range;
 use aya::maps::{Array, Map};
 use log::{error, info, warn};
 use rangemap::RangeInclusiveSet;
@@ -167,6 +168,10 @@ impl RPCLeaderTracker {
             )
             .await
         {
+            Err(e) => {
+                error!("Failed to get leader schedule: {e}");
+                Err(())
+            }
             Ok(sched) => {
                 if sched.is_none() {
                     error!("Failed to get leader schedule.");
@@ -177,19 +182,20 @@ impl RPCLeaderTracker {
                     for slot in my_slots {
                         let end: u64 = *slot as u64;
 
-                        leader_ranges.insert(end - self.slot_buffer..=end);
+
+                        let range = end.saturating_sub(self.slot_buffer)..=end;
+                        leader_ranges.insert(range);
                     }
                     leader_ranges.insert(0..=10);
+                    // let rngs: Vec<Range<u64>> =leader_ranges.iter().collect();
+                    info!("Leader ranges: {leader_ranges:?}");
+
                     Ok(leader_ranges)
                 } else {
                     error!("No slots found for: {my_id}");
                     Err(())
                 }
             }
-            Err(e) => {
-                error!("Failed to get leader schedule: {e}");
-                Err(())
-            }
         };
     }
 }