fdt: enumerate all memory nodes present

It is possible for the device tree to have multiple memory nodes. Iterate
through all regions defined by all memory nodes.

Signed-off-by: Thomas Barrett <tbarrett@crusoe.ai>

Author: Thomas Barrett

src/fdt.rs

@@ -78,11 +78,16 @@ impl Info for StartInfo<'_> {
     }
 
     fn num_entries(&self) -> usize {
-        self.fdt.memory().regions().count()
+        let nodes = self.fdt.find_all_nodes("/memory");
+        let regions = nodes.flat_map(|n| n.reg().unwrap());
+        regions.count()
+        // self.fdt.memory().regions().count()
     }
 
     fn entry(&self, idx: usize) -> MemoryEntry {
-        for (i, region) in self.fdt.memory().regions().enumerate() {
+        let nodes = self.fdt.find_all_nodes("/memory");
+        let regions = nodes.flat_map(|n| n.reg().unwrap());
+        for (i, region) in regions.enumerate() {
             if i == idx {
                 return MemoryEntry {
                     addr: region.starting_address as u64,

src/main.rs

@@ -202,6 +202,8 @@ pub extern "C" fn rust64_start(#[cfg(not(feature = "coreboot"))] pvh_info: &pvh:
 #[cfg(target_arch = "aarch64")]
 #[no_mangle]
 pub extern "C" fn rust64_start(x0: *const u8) -> ! {
+    use bootinfo::Info;
+
     arch::aarch64::simd::setup_simd();
     arch::aarch64::paging::setup();
 
@@ -217,6 +219,15 @@ pub extern "C" fn rust64_start(x0: *const u8) -> ! {
         None,
     );
 
+    for i in 0..info.num_entries() {
+        let region = info.entry(i);
+        info!(
+            "Memory region {}MiB@0x{:x}",
+            region.size / 1024 / 1024,
+            region.addr
+        );
+    }
+
     if let Some((base, length)) = info.find_compatible_region(&["pci-host-ecam-generic"]) {
         pci::init(base as u64, length as u64);
     }