Add (partial) safe protocol implementation for EFI_HII_DATABASE_PROTOCOL

This only grants access to the HII-database's raw buffer for now.

Author: seijikun

uefi/CHANGELOG.md

@@ -7,6 +7,7 @@
 - Added `proto::hii::config::ConfigKeywordHandler`.
 - Added `proto::hii::config::HiiConfigAccess`.
 - Added `proto::hii::config_str::ConfigurationString`.
+- Added `proto::hii::database::HiiDatabase`.
 
 ## Changed
 - **Breaking:** `boot::stall` now take `core::time::Duration` instead of `usize`.

uefi/src/proto/hii/database.rs

@@ -0,0 +1,38 @@
+// SPDX-License-Identifier: MIT OR Apache-2.0
+
+//! HII Database protocol.
+
+use alloc::boxed::Box;
+use uefi_macros::unsafe_protocol;
+use uefi_raw::protocol::hii::database::HiiDatabaseProtocol;
+
+use crate::{StatusExt, mem::make_boxed};
+
+/// The HII Configuration Access Protocol.
+///
+/// # UEFI Spec Description
+///
+/// Database manager for HII-related data structures.
+#[derive(Debug)]
+#[repr(transparent)]
+#[unsafe_protocol(HiiDatabaseProtocol::GUID)]
+pub struct HiiDatabase(HiiDatabaseProtocol);
+
+impl HiiDatabase {
+    /// Export all package lists as raw byte buffer.
+    pub fn export_all_raw<'a>(&self) -> crate::Result<Box<[u8]>> {
+        let fetch_data_fn = |buf: &'a mut [u8]| unsafe {
+            let mut size = buf.len();
+            let status = {
+                (self.0.export_package_lists)(
+                    &self.0,
+                    core::ptr::null_mut(),
+                    &mut size,
+                    buf.as_mut_ptr().cast(),
+                )
+            };
+            status.to_result_with_err(|_| Some(size)).map(|_| buf)
+        };
+        make_boxed::<[u8], _>(fetch_data_fn)
+    }
+}

uefi/src/proto/hii/mod.rs

@@ -5,3 +5,5 @@
 pub mod config;
 #[cfg(feature = "alloc")]
 pub mod config_str;
+#[cfg(feature = "alloc")]
+pub mod database;