uefi: Add set_main macro

Author: nicholasbishop

uefi/src/lib.rs

@@ -136,6 +136,7 @@ pub(crate) mod polyfill;
 
 pub mod helpers;
 
+mod macros;
 mod util;
 
 #[cfg(test)]

uefi/src/macros.rs

@@ -0,0 +1,32 @@
+/// Set the entry point for a UEFI executable.
+///
+/// This macro takes one argument, the function to run when the executable is
+/// launched. That function must take no arguments and return a [`uefi::Result`].
+///
+/// # Example
+///
+/// ```
+/// uefi::set_main!(main);
+///
+/// fn main() -> uefi::Result {
+///     Ok(())
+/// }
+/// ```
+#[macro_export]
+macro_rules! set_main {
+    ($main:ident) => {
+        #[export_name = "efi_main"]
+        fn efi_main(
+            image: ::uefi::Handle,
+            system_table: *mut ::core::ffi::c_void,
+        ) -> ::uefi::Status {
+            unsafe { ::uefi::boot::set_image_handle(image) };
+            unsafe { ::uefi::system::set_system_table(system_table.cast()) };
+            let result = $main();
+            match result {
+                Ok(()) => ::uefi::Status::SUCCESS,
+                Err(err) => err.status(),
+            }
+        }
+    };
+}