test: add load_image integration test

Author: phip1611

uefi-test-runner/src/boot/mod.rs

@@ -1,17 +1,21 @@
+use alloc::string::ToString;
 use uefi::proto::console::text::Output;
-use uefi::table::boot::{BootServices, SearchType};
-use uefi::Identify;
+use uefi::proto::device_path::text::{AllowShortcuts, DevicePathToText, DisplayOnly};
+use uefi::proto::loaded_image::LoadedImage;
+use uefi::table::boot::{BootServices, LoadImageSource, SearchType};
+use uefi::{CString16, Identify};
+
+mod memory;
+mod misc;
 
 pub fn test(bt: &BootServices) {
     info!("Testing boot services");
     memory::test(bt);
     misc::test(bt);
     test_locate_handle_buffer(bt);
+    test_load_image(bt);
 }
 
-mod memory;
-mod misc;
-
 fn test_locate_handle_buffer(bt: &BootServices) {
     info!("Testing the `locate_handle_buffer` function");
 
@@ -34,3 +38,53 @@ fn test_locate_handle_buffer(bt: &BootServices) {
         );
     }
 }
+
+/// This test loads the "self image" again into memory using the `load_image`
+/// boot service function. The image is not started but just loaded into memory.
+///
+/// It transitively tests the protocols
+/// * `DevicePathToText`
+/// * `LoadedImage`
+/// which are required as helpers.
+fn test_load_image(bt: &BootServices) {
+    info!("Testing the `load_image` function");
+
+    let loaded_image = bt
+        .open_protocol_exclusive::<LoadedImage>(bt.image_handle())
+        .expect("should open LoadedImage protocol");
+
+    let image_device_path = loaded_image.file_path().expect("should have file path");
+
+    let &device_path_to_text_handle = bt
+        .locate_handle_buffer(SearchType::ByProtocol(&DevicePathToText::GUID))
+        .expect("Should be able to loace handle for DevicePathToText")
+        .first()
+        .expect("should locate DevicePathToText protocol handle");
+
+    let device_path_to_text = bt
+        .open_protocol_exclusive::<DevicePathToText>(device_path_to_text_handle)
+        .expect("should open DevicePathToText protocol");
+
+    let device_path_text = device_path_to_text
+        .convert_device_path_to_text(
+            bt,
+            image_device_path,
+            DisplayOnly(true),
+            AllowShortcuts(false),
+        )
+        .expect("should convert to human-readable path");
+    let device_path_text = CString16::from(&*device_path_text);
+    let device_path_text = device_path_text.to_string().to_uppercase();
+
+    // On x86_64, this will be `\EFI\BOOT\BOOTX64.EFI` for example.
+    assert!(device_path_text.as_str().starts_with(r"\EFI\BOOT\BOOT"));
+    assert!(device_path_text.as_str().ends_with(".EFI"));
+
+    let load_source = LoadImageSource::FromFilePath {
+        file_path: image_device_path,
+        from_boot_manager: false,
+    };
+    let _ = bt
+        .load_image(bt.image_handle(), load_source)
+        .expect("should load image");
+}