Add ShellParams protocol

Useful to get the shell arguments for a commandline application.

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

Author: JohnAZoidberg

uefi/src/proto/mod.rs

@@ -76,6 +76,7 @@ pub mod network;
 pub mod pi;
 pub mod rng;
 pub mod security;
+pub mod shell_params;
 pub mod shim;
 pub mod string;
 pub mod tcg;

uefi/src/proto/shell_params.rs

@@ -0,0 +1,44 @@
+//! `ShellParams` protocol
+
+use crate::proto::unsafe_protocol;
+use crate::{CStr16, Char16};
+use core::ffi::c_void;
+
+#[cfg(feature = "alloc")]
+use alloc::string::String;
+use alloc::string::ToString;
+use alloc::vec;
+use alloc::vec::Vec;
+
+type ShellFileHandle = *const c_void;
+
+/// The ShellParameters protocol.
+#[repr(C)]
+#[unsafe_protocol("752f3136-4e16-4fdc-a22a-e5f46812f4ca")]
+pub struct ShellParameters {
+    /// Pointer to a list of arguments
+    pub argv: *const *const Char16,
+    /// Number of arguments
+    pub argc: usize,
+    /// Handle of the standard input
+    std_in: ShellFileHandle,
+    /// Handle of the standard output
+    std_out: ShellFileHandle,
+    /// Handle of the standard error output
+    std_err: ShellFileHandle,
+}
+
+impl ShellParameters {
+    /// Get a Vec of the shell parameter arguments
+    #[cfg(feature = "alloc")]
+    #[must_use]
+    pub fn get_args(&self) -> Vec<String> {
+        let mut args = vec![];
+        for i in 0..self.argc {
+            let str = unsafe { CStr16::from_ptr(*self.argv.add(i)) };
+            let string = str.to_string();
+            args.push(string);
+        }
+        args
+    }
+}