Add struct Rng and test instantiate it

Author: gelven4sec

src/proto/mod.rs

@@ -35,3 +35,4 @@ pub mod loaded_image;
 pub mod media;
 pub mod pi;
 pub mod shim;
+pub mod rng;

src/proto/rng.rs

@@ -0,0 +1,23 @@
+//! `Rng` protocol.
+
+use crate::{data_types::Guid, proto::Protocol, Status, unsafe_guid};
+
+/// Contain a Rng algorithm Guid
+#[repr(C)]
+struct RngAlgorithm(Guid);
+
+/// Rng protocol
+#[repr(C)]
+#[unsafe_guid("3152bca5-eade-433d-862e-c01cdc291f44")]
+#[derive(Protocol)]
+pub struct Rng {
+ get_info: extern "efiapi" fn(this: &Rng, algorithm_list_size: *mut usize, algorithm_list: *mut RngAlgorithm) -> Status,
+ get_rng: extern "efiapi" fn(this: &Rng, algorithm: Option<RngAlgorithm>, value_length: usize, value: *mut u8) -> Status,
+}
+
+/*impl Rng {
+ fn get_info(&mut self) -> Result<[RngAlgorithm]> {
+ let mut algorithm_list_size: usize;
+ let mut algorithm_list: []
+ }
+}*/

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

@@ -17,6 +17,7 @@ pub fn test(image: Handle, st: &mut SystemTable<Boot>) {
  loaded_image::test(image, bt);
  media::test(image, bt);
  pi::test(bt);
+ rng::test(image, bt);
 
  #[cfg(any(
  target_arch = "i386",
@@ -67,3 +68,4 @@ mod pi;
  target_arch = "aarch64"
 ))]
 mod shim;
+mod rng;

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

@@ -0,0 +1,21 @@
+use uefi::prelude::*;
+use uefi::proto::rng::Rng;
+use uefi::table::boot::{BootServices, OpenProtocolAttributes, OpenProtocolParams};
+
+pub fn test(image: Handle, bt: &BootServices) {
+ info!("Running loaded image protocol test");
+
+ let rng = bt
+ .open_protocol::<Rng>(
+ OpenProtocolParams {
+ handle: image,
+ agent: image,
+ controller: None,
+ },
+ OpenProtocolAttributes::Exclusive,
+ )
+ .expect_success("Failed to open LoadedImage protocol");
+ let _rng = unsafe { &*rng.interface.get() };
+
+ info!("Rng loaded !");
+}