List iterators: hold lock when calling find_true_head/tail

sweet-security · Aug 17, 2024 · b54d503 · b54d503
1 parent 351c909
commit b54d503
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/lists.rs b/src/lists.rs
@@ -138,6 +138,7 @@ impl<'a> Iterator for LinkedListIterator<'a> {
     type Item = Result<Option<KVPair>>;
     fn next(&mut self) -> Option<Self::Item> {
         if self.fph.is_none() {
+            let _guard = self.store._list_lock(self.list_ph);
             let buf = match self.store.get_raw(&self.list_key) {
                 Ok(buf) => buf,
                 Err(e) => return Some(Err(e)),
@@ -193,6 +194,7 @@ impl<'a> Iterator for RevLinkedListIterator<'a> {
     type Item = Result<Option<KVPair>>;
     fn next(&mut self) -> Option<Self::Item> {
         if self.fph.is_none() {
+            let _guard = self.store._list_lock(self.list_ph);
             let buf = match self.store.get_raw(&self.list_key) {
                 Ok(buf) => buf,
                 Err(e) => return Some(Err(e)),