uefi: remove code duplication

Author: phip1611

uefi/src/boot.rs

@@ -5,20 +5,18 @@
 use crate::data_types::PhysicalAddress;
 use crate::proto::device_path::DevicePath;
 use crate::proto::{Protocol, ProtocolPointer};
+pub use crate::table::boot::{
+    AllocateType, EventNotifyFn, LoadImageSource, OpenProtocolAttributes, OpenProtocolParams,
+    SearchType, TimerTrigger,
+};
 use crate::util::opt_nonnull_to_ptr;
+use crate::{table, Char16, Event, Guid, Handle, Result, Status, StatusExt};
 use core::ffi::c_void;
 use core::ops::{Deref, DerefMut};
 use core::ptr::{self, NonNull};
 use core::sync::atomic::{AtomicPtr, Ordering};
 use core::{mem, slice};
-use uefi::{table, Char16, Event, Guid, Handle, Result, Status, StatusExt};
 use uefi_raw::table::boot::InterfaceType;
-
-use crate::proto::BootPolicy;
-pub use crate::table::boot::{
-    AllocateType, EventNotifyFn, LoadImageSource, OpenProtocolAttributes, OpenProtocolParams,
-    SearchType, TimerTrigger,
-};
 pub use uefi_raw::table::boot::{EventType, MemoryType, Tpl};
 #[cfg(doc)]
 use {
@@ -642,29 +640,7 @@ pub fn load_image(parent_image_handle: Handle, source: LoadImageSource) -> Resul
     let bt = boot_services_raw_panicking();
     let bt = unsafe { bt.as_ref() };
 
-    let boot_policy;
-    let device_path;
-    let source_buffer;
-    let source_size;
-    match source {
-        LoadImageSource::FromBuffer { buffer, file_path } => {
-            // Boot policy is ignored when loading from source buffer.
-            boot_policy = BootPolicy::FALSE;
-
-            device_path = file_path.map(|p| p.as_ffi_ptr()).unwrap_or(ptr::null());
-            source_buffer = buffer.as_ptr();
-            source_size = buffer.len();
-        }
-        LoadImageSource::FromDevicePath {
-            device_path: file_path,
-            bool_policy,
-        } => {
-            boot_policy = bool_policy;
-            device_path = file_path.as_ffi_ptr();
-            source_buffer = ptr::null();
-            source_size = 0;
-        }
-    };
+    let (boot_policy, device_path, source_buffer, source_size) = source.to_ffi_params();
 
     let mut image_handle = ptr::null_mut();
     unsafe {

uefi/src/table/boot.rs

@@ -23,6 +23,7 @@ use core::mem::{self, MaybeUninit};
 use core::ops::{Deref, DerefMut};
 use core::ptr::NonNull;
 use core::{ptr, slice};
+use uefi::proto::device_path::FfiDevicePath;
 
 /// Size in bytes of a UEFI page.
 ///
@@ -842,29 +843,7 @@ impl BootServices {
         parent_image_handle: Handle,
         source: LoadImageSource,
     ) -> uefi::Result<Handle> {
-        let boot_policy;
-        let device_path;
-        let source_buffer;
-        let source_size;
-        match source {
-            LoadImageSource::FromBuffer { buffer, file_path } => {
-                // Boot policy is ignored when loading from source buffer.
-                boot_policy = BootPolicy::FALSE;
-
-                device_path = file_path.map(|p| p.as_ffi_ptr()).unwrap_or(ptr::null());
-                source_buffer = buffer.as_ptr();
-                source_size = buffer.len();
-            }
-            LoadImageSource::FromDevicePath {
-                device_path: file_path,
-                bool_policy,
-            } => {
-                boot_policy = bool_policy;
-                device_path = file_path.as_ffi_ptr();
-                source_buffer = ptr::null();
-                source_size = 0;
-            }
-        };
+        let (boot_policy, device_path, source_buffer, source_size) = source.to_ffi_params();
 
         let mut image_handle = ptr::null_mut();
         unsafe {
@@ -1422,6 +1401,43 @@ pub enum LoadImageSource<'a> {
     },
 }
 
+impl<'a> LoadImageSource<'a> {
+    /// Returns the raw FFI parameters for `load_image`.
+    pub fn to_ffi_params(
+        self,
+    ) -> (
+        BootPolicy,
+        *const FfiDevicePath,
+        *const u8, /* buffer */
+        usize,     /* buffer length */
+    ) {
+        let boot_policy;
+        let device_path;
+        let source_buffer;
+        let source_size;
+        match self {
+            LoadImageSource::FromBuffer { buffer, file_path } => {
+                // Boot policy is ignored when loading from source buffer.
+                boot_policy = BootPolicy::FALSE;
+
+                device_path = file_path.map(|p| p.as_ffi_ptr()).unwrap_or(ptr::null());
+                source_buffer = buffer.as_ptr();
+                source_size = buffer.len();
+            }
+            LoadImageSource::FromDevicePath {
+                device_path: file_path,
+                bool_policy,
+            } => {
+                boot_policy = bool_policy;
+                device_path = file_path.as_ffi_ptr();
+                source_buffer = ptr::null();
+                source_size = 0;
+            }
+        };
+        (boot_policy, device_path, source_buffer, source_size)
+    }
+}
+
 /// RAII guard for task priority level changes
 ///
 /// Will automatically restore the former task priority level when dropped.