read/elf: impl Iterator for NoteIterator (#713)

gimli-rs · Aug 3, 2024 · 8a25b8e · 8a25b8e
1 parent b138dab
commit 8a25b8e
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/read/elf/note.rs b/src/read/elf/note.rs
@@ -87,6 +87,20 @@ where
     }
 }
 
+impl<'data, Elf: FileHeader> Iterator for NoteIterator<'data, Elf> {
+    type Item = read::Result<Note<'data, Elf>>;
+    fn next(&mut self) -> Option<Self::Item> {
+        match self.next() {
+            Err(e) => {
+                self.data = Bytes(&[]);
+                Some(Err(e))
+            }
+            Ok(Some(v)) => Some(Ok(v)),
+            Ok(None) => None,
+        }
+    }
+}
+
 /// A parsed [`NoteHeader`].
 #[derive(Debug)]
 pub struct Note<'data, Elf>