uefi-raw: Add raw AllocateType enum

Author: seijikun

uefi-raw/CHANGELOG.md

@@ -1,5 +1,8 @@
 # uefi-raw - [Unreleased]
 
+## Added
+- Added `AllocateType`.
+
 
 # uefi-raw - 0.11.0 (2025-05-04)
 

uefi-raw/src/table/boot.rs

@@ -9,6 +9,15 @@ use bitflags::bitflags;
 use core::ffi::c_void;
 use core::ops::RangeInclusive;
 
+newtype_enum! {
+    pub enum AllocateType: u32 => {
+        ANY_PAGES = 0,
+        MAX_ADDRESS = 1,
+        ADDRESS = 2,
+        MAX_ALLOCATE_TYPE = 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};
 
@@ -154,9 +154,9 @@ pub fn allocate_pages(
     let bt = unsafe { bt.as_ref() };
 
     let (ty, initial_addr) = match allocation_type {
-        AllocateType::AnyPages => (0, 0),
-        AllocateType::MaxAddress(addr) => (1, addr),
-        AllocateType::Address(addr) => (2, addr),
+        AllocateType::AnyPages => (RawAllocateType::ANY_PAGES, 0),
+        AllocateType::MaxAddress(addr) => (RawAllocateType::MAX_ADDRESS, addr),
+        AllocateType::Address(addr) => (RawAllocateType::ADDRESS, addr),
     };
 
     let mut addr1 = initial_addr;