loaded_image: Provide getters for memory types

Author: medhefgo

CHANGELOG.md

@@ -27,6 +27,7 @@
 - Added `LoadedImageDevicePath` protocol.
 - Added `FileAttribute::is_directory(&self)` and
   `FileAttribute::is_regular_file(&self)`
+- Added `LoadedImage::code_type()` and `LoadedImage::data_type()`
 
 ### Changed
 

uefi/src/proto/loaded_image.rs

@@ -170,4 +170,26 @@ impl LoadedImage {
     pub const fn info(&self) -> (*const c_void, u64) {
         (self.image_base, self.image_size)
     }
+
+    /// Get the memory type of the image's code sections.
+    ///
+    /// Normally the returned value is one of:
+    ///  - `MemoryType::LOADER_CODE` for UEFI applications
+    ///  - `MemoryType::BOOT_SERVICES_CODE` for UEFI boot drivers
+    ///  - `MemoryType::RUNTIME_SERVICES_CODE` for UEFI runtime drivers
+    #[must_use]
+    pub fn code_type(&self) -> MemoryType {
+        self.image_code_type
+    }
+
+    /// Get the memory type of the image's data sections.
+    ///
+    /// Normally the returned value is one of:
+    ///  - `MemoryType::LOADER_DATA` for UEFI applications
+    ///  - `MemoryType::BOOT_SERVICES_DATA` for UEFI boot drivers
+    ///  - `MemoryType::RUNTIME_SERVICES_DATA` for UEFI runtime drivers
+    #[must_use]
+    pub fn data_type(&self) -> MemoryType {
+        self.image_data_type
+    }
 }