Implement EFI Shell Protocol

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

@@ -29,6 +29,7 @@ pub fn test(image: Handle, st: &mut SystemTable<Boot>) {
         target_arch = "aarch64"
     ))]
     shim::test(bt);
+    shell::test(bt);
     tcg::test(bt);
 }
 
@@ -69,6 +70,7 @@ mod rng;
     target_arch = "arm",
     target_arch = "aarch64"
 ))]
+mod shell;
 mod shim;
 mod string;
 mod tcg;

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

@@ -0,0 +1,28 @@
+use uefi::CStr16;
+use uefi::prelude::BootServices;
+use uefi::proto::shell::Shell;
+
+pub fn test(bt: &BootServices) {
+    info!("Running shell protocol tests");
+
+    let handle = bt.get_handle_for_protocol::<Shell>().expect("No Shell handles");
+
+    let mut shell = bt
+        .open_protocol_exclusive::<Shell>(handle)
+        .expect("Failed to open Shell protocol");
+
+    // create some files
+    let mut test_buf = [0u16; 12];
+    let test_str = CStr16::from_str_with_buf("test", &mut test_buf).unwrap();
+    shell.create_file(test_str, 0);
+
+    // get file tree
+    let mut str_buf = [0u16; 12];
+    let str_str = CStr16::from_str_with_buf(r"fs0:\*", &mut str_buf).unwrap();
+    let res = shell.find_files(str_str);
+    let list = res.unwrap();
+    let list = list.unwrap();
+    let first = list.first();
+
+    info!("filetree test successful")
+}

uefi/src/proto/mod.rs

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