Implemented set_env()

Author: RenTrieu

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

@@ -19,16 +19,31 @@ pub fn test() {
     let cur_env_str = shell.get_env(None).expect("Could not get environment variable");
     info!("cur_env_str size: {}", cur_env_str.num_chars());
     info!("cur_env_str: {}", cur_env_str);
+
+    /* Testing setting and getting a specific environment variable */
+    let mut test_env_buf = [0u16; 32];
+    let test_var = CStr16::from_str_with_buf("test_var", &mut test_env_buf).unwrap();
+    let mut test_val_buf = [0u16; 32];
+    let test_val = CStr16::from_str_with_buf("test_val", &mut test_val_buf).unwrap();
+    assert!(shell.get_env(Some(test_var)).is_none());
+    shell.set_env(test_var, test_val, false);
+    let cur_env_str = shell.get_env(Some(test_var)).expect("Could not get environment variable");
+    assert_eq!(cur_env_str, test_val);
+
     // for (i, c) in cur_env_str.iter().enumerate() {
     //     info!("cur_env_str: i: {}, c: {}", i, c);
     // }
 
     // let mut cur_fs_buf = [0u16; 32];
     // let cur_fs_str = CStr16::from_str_with_buf("", &mut cur_fs_buf).unwrap();
     // info!("cur_fs_str size 1: {}", cur_fs_str.num_chars());
-    let cur_fs_str = shell.get_cur_dir(None).expect("Could not get the current file system mapping");
-    info!("cur_fs_str size: {}", cur_fs_str.num_chars());
-    info!("cur_fs_str: {}", cur_fs_str);
+
+
+    // let cur_fs_str = shell.get_cur_dir(None).expect("Could not get the current file system mapping");
+    // info!("cur_fs_str size: {}", cur_fs_str.num_chars());
+    // info!("cur_fs_str: {}", cur_fs_str);
+
+
     // for (i, c) in cur_fs_str.iter().enumerate() {
     //     info!("cur_fs_str: i: {}, c: {}", i, c);
     // }

uefi/src/proto/shell/mod.rs

@@ -21,7 +21,11 @@ pub struct Shell {
     get_env: extern "efiapi" fn(
         name: *const Char16,
     ) -> *const Char16,
-    set_env: usize,
+    set_env: extern "efiapi" fn(
+        name: *const Char16,
+        value: *const Char16,
+        volatile: bool,
+    ) -> Status,
     get_alias: usize,
     set_alias: usize,
     get_help_text: usize,
@@ -137,6 +141,24 @@ impl Shell {
         }
     }
 
+    /// Sets the environment variable
+    ///
+    /// # Arguments
+    ///
+    /// * `name` - The environment variable for which to set the value
+    /// * `value` - The new value of the environment variable
+    /// * `volatile` - Indicates whether or not the variable is volatile or
+    ///                not
+    ///
+    /// # Returns
+    ///
+    /// * `Status::SUCCESS` The variable was successfully set
+    pub fn set_env(&self, name: &CStr16, value: &CStr16, volatile: bool) -> Status {
+        let name_ptr: *const Char16 = (name as *const CStr16).cast();
+        let value_ptr: *const Char16 = (value as *const CStr16).cast();
+        (self.set_env)(name_ptr, value_ptr, volatile)
+    }
+
     /// TODO
     #[must_use]
     pub fn get_cur_dir<'a>(&'a self, file_system_mapping: Option<&CStr16>) -> Option<&'a CStr16> {