don't loop forever in DHT poll

EspressoSystems · Mar 22, 2024 · 3508c95 · 3508c95
1 parent d64446b
commit 3508c95
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/crates/libp2p-networking/src/network/behaviours/dht/mod.rs b/crates/libp2p-networking/src/network/behaviours/dht/mod.rs
@@ -589,7 +589,10 @@ impl NetworkBehaviour for DHTBehaviour {
         }
 
         // retry put/gets if they are ready
-        while let Some(req) = self.queued_get_record_queries.pop_front() {
+        for _i in 0..self.queued_get_record_queries.len() {
+            let Some(req) = self.queued_get_record_queries.pop_front() else {
+                continue;
+            };
             if req.backoff.is_expired() {
                 self.get_record(
                     req.key,
@@ -603,7 +606,10 @@ impl NetworkBehaviour for DHTBehaviour {
             }
         }
 
-        while let Some(req) = self.queued_put_record_queries.pop_front() {
+        for _i in 0..self.queued_put_record_queries.len() {
+            let Some(req) = self.queued_put_record_queries.pop_front() else {
+                continue;
+            };
             if req.backoff.is_expired() {
                 self.put_record(req);
             } else {