uefi-test-runner: Add test for ShellParams protocol

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

Author: JohnAZoidberg

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,26 @@
+use uefi::proto::shell_params::ShellParameters;
+use uefi::table::boot::BootServices;
+use uefi::CStr16;
+
+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!("Argc: {}", shell_params.argc);
+    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.argc, shell_params.get_args_slice().len());
+
+    // Was run as: shell.efi test_runner.efi
+    assert_eq!(shell_params.argc, 2);
+}