uefi-test-runner: Add test for ShellParams protocol

Signed-off-by: Daniel Schaefer <[email protected]>

Author: JohnAZoidberg

uefi-test-runner/src/bin/shell_launcher.rs

@@ -69,7 +69,7 @@ fn efi_main(image: Handle, mut st: SystemTable<Boot>) -> Status {
     let mut shell_loaded_image = boot_services
         .open_protocol_exclusive::<LoadedImage>(shell_image_handle)
         .expect("failed to open LoadedImage protocol");
-    let load_options = cstr16!(r"shell.efi test_runner.efi");
+    let load_options = cstr16!(r"shell.efi test_runner.efi arg1 arg2");
     unsafe {
         shell_loaded_image.set_load_options(
             load_options.as_ptr().cast(),

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

@@ -20,6 +20,7 @@ pub fn test(image: Handle, st: &mut SystemTable<Boot>) {
     network::test(bt);
     pi::test(bt);
     rng::test(bt);
+    shell_params::test(bt);
     string::test(bt);
 
     #[cfg(any(
@@ -63,6 +64,7 @@ mod media;
 mod network;
 mod pi;
 mod rng;
+mod shell_params;
 #[cfg(any(
     target_arch = "x86",
     target_arch = "x86_64",

uefi-test-runner/src/proto/shell_params.rs

@@ -0,0 +1,28 @@
+use uefi::proto::shell_params::ShellParameters;
+use uefi::table::boot::BootServices;
+use uefi::CStr16;
+
+use alloc::vec::Vec;
+
+pub fn test(bt: &BootServices) {
+    info!("Running loaded image protocol test");
+
+    let image = bt
+        .get_handle_for_protocol::<ShellParameters>()
+        .expect("No ShellParameters handles");
+    let shell_params = bt
+        .open_protocol_exclusive::<ShellParameters>(image)
+        .expect("Failed to open ShellParameters protocol");
+
+    info!("Args:");
+    for arg in shell_params.get_args_slice() {
+        let arg_str = unsafe { CStr16::from_ptr(*arg) };
+        info!("  '{}'", arg_str);
+    }
+
+    assert_eq!(shell_params.get_args_slice().len(), 4);
+    assert_eq!(
+        shell_params.get_args().collect::<Vec<_>>(),
+        &["shell.efi", "test_runner.efi", "arg1", "arg2"]
+    );
+}