multiboot2: Get a mutable reference to the basic memory information tag

Author: YtvwlD

multiboot2/src/lib.rs

@@ -249,6 +249,12 @@ impl BootInformation {
         self.get_tag::<BasicMemoryInfoTag, _>(TagType::BasicMeminfo)
     }
 
+    /// Search for the basic memory info tag, return a mutable reference.
+    pub fn basic_memory_info_tag_mut(&mut self) -> Option<&mut BasicMemoryInfoTag> {
+        self.get_tag_mut(TagType::BasicMeminfo)
+            .map(|tag| unsafe { &mut *(tag as *mut Tag as *mut BasicMemoryInfoTag) })
+    }
+
     /// Search for the ELF Sections.
     ///
     /// # Examples