uefi-raw: Add raw AllocateType enum

Author: seijikun

uefi-raw/CHANGELOG.md

@@ -18,6 +18,7 @@
 - Added `UsbIoProtocol`.
 - Added `Usb2HostControllerProtocol`.
 - Added  `DevicePathProtocol::length()` properly constructing the `u16` value
+- Added `AllocateType`.
 
 ## Changed
 - `DevicePathProtocol` now derives

uefi-raw/src/table/boot.rs

@@ -9,6 +9,15 @@ use bitflags::bitflags;
 use core::ffi::c_void;
 use core::ops::RangeInclusive;
 
+#[repr(u32)]
+#[derive(Debug, Clone, Copy, PartialEq, Eq)]
+pub enum AllocateType {
+    AnyPages = 0,
+    MaxAddress = 1,
+    Address = 2,
+    MaxAllocateType = 3,
+}
+
 /// Table of pointers to all the boot services.
 #[derive(Debug)]
 #[repr(C)]
@@ -21,7 +30,7 @@ pub struct BootServices {
 
     // Memory allocation functions
     pub allocate_pages: unsafe extern "efiapi" fn(
-        alloc_ty: u32,
+        alloc_ty: AllocateType,
         mem_ty: MemoryType,
         count: usize,
         addr: *mut PhysicalAddress,

uefi/src/boot.rs

@@ -45,7 +45,7 @@ use core::ptr::{self, NonNull};
 use core::sync::atomic::{AtomicPtr, Ordering};
 use core::time::Duration;
 use core::{mem, slice};
-use uefi_raw::table::boot::{InterfaceType, TimerDelay};
+use uefi_raw::table::boot::{AllocateType as RawAllocateType, InterfaceType, TimerDelay};
 #[cfg(feature = "alloc")]
 use {alloc::vec::Vec, uefi::ResultExt};
 
@@ -136,9 +136,9 @@ pub fn allocate_pages(ty: AllocateType, mem_ty: MemoryType, count: usize) -> Res
     let bt = unsafe { bt.as_ref() };
 
     let (ty, initial_addr) = match ty {
-        AllocateType::AnyPages => (0, 0),
-        AllocateType::MaxAddress(addr) => (1, addr),
-        AllocateType::Address(addr) => (2, addr),
+        AllocateType::AnyPages => (RawAllocateType::AnyPages, 0),
+        AllocateType::MaxAddress(addr) => (RawAllocateType::MaxAddress, addr),
+        AllocateType::Address(addr) => (RawAllocateType::Address, addr),
     };
 
     let mut addr1 = initial_addr;