feat(fdt): change memory method to return iterator for multiple memory nodes

The memory method now returns an iterator instead of a single option,
allowing iteration over multiple memory nodes in the device tree.
This change provides better support for systems with multiple memory
regions and improves the API flexibility.

Author: ZR233

fdt-raw/Cargo.toml

@@ -11,7 +11,7 @@ name = "fdt-raw"
 readme = "README.md"
 homepage = "https://github.com/drivercraft/fdt-parser"
 repository = "https://github.com/drivercraft/fdt-parser"
-version = "0.1.3"
+version = "0.1.4"
 
 [dependencies]
 heapless = "0.9"

fdt-raw/src/fdt.rs

@@ -243,13 +243,14 @@ impl<'a> Fdt<'a> {
         None
     }
 
-    pub fn memory(&self) -> Option<Memory<'a>> {
-        for node in self.all_nodes() {
-            if let Node::Memory(m) = node {
-                return Some(m);
+    pub fn memory(&self) -> impl Iterator<Item = Memory<'a>> + 'a {
+        self.all_nodes().filter_map(|node| {
+            if let Node::Memory(mem) = node {
+                Some(mem)
+            } else {
+                None
             }
-        }
-        None
+        })
     }
 }
 

fdt-raw/tests/classify.rs

@@ -27,6 +27,6 @@ fn test_memory() {
     init_logging();
     let raw = fdt_qemu();
     let fdt = Fdt::from_bytes(&raw).unwrap();
-    let memory = fdt.memory().unwrap();
+    let memory = fdt.memory().next().unwrap();
     println!("Memory node: {:#x?}", memory);
 }