bjorn3
84 lines
1.8 KiB
Rust
84 lines
1.8 KiB
Rust
use crate::PhysicalAddress;
|
|
|
|
pub use self::{buddy::*, bump::*};
|
|
|
|
mod buddy;
|
|
mod bump;
|
|
|
|
#[derive(Clone, Copy, Debug)]
|
|
#[repr(transparent)]
|
|
pub struct FrameCount(usize);
|
|
|
|
impl FrameCount {
|
|
pub fn new(count: usize) -> Self {
|
|
Self(count)
|
|
}
|
|
|
|
pub fn data(&self) -> usize {
|
|
self.0
|
|
}
|
|
}
|
|
|
|
#[derive(Debug)]
|
|
pub struct FrameUsage {
|
|
used: FrameCount,
|
|
total: FrameCount,
|
|
}
|
|
|
|
impl FrameUsage {
|
|
pub fn new(used: FrameCount, total: FrameCount) -> Self {
|
|
Self { used, total }
|
|
}
|
|
|
|
pub fn used(&self) -> FrameCount {
|
|
self.used
|
|
}
|
|
|
|
pub fn free(&self) -> FrameCount {
|
|
FrameCount(self.total.0 - self.used.0)
|
|
}
|
|
|
|
pub fn total(&self) -> FrameCount {
|
|
self.total
|
|
}
|
|
}
|
|
|
|
pub unsafe trait FrameAllocator {
|
|
fn allocate(&mut self, count: FrameCount) -> Option<PhysicalAddress>;
|
|
|
|
unsafe fn free(&mut self, address: PhysicalAddress, count: FrameCount);
|
|
|
|
fn allocate_one(&mut self) -> Option<PhysicalAddress> {
|
|
self.allocate(FrameCount::new(1))
|
|
}
|
|
|
|
unsafe fn free_one(&mut self, address: PhysicalAddress) {
|
|
unsafe {
|
|
self.free(address, FrameCount::new(1));
|
|
}
|
|
}
|
|
|
|
fn usage(&self) -> FrameUsage;
|
|
}
|
|
|
|
unsafe impl<T> FrameAllocator for &mut T
|
|
where
|
|
T: FrameAllocator,
|
|
{
|
|
fn allocate(&mut self, count: FrameCount) -> Option<PhysicalAddress> {
|
|
T::allocate(self, count)
|
|
}
|
|
unsafe fn free(&mut self, address: PhysicalAddress, count: FrameCount) {
|
|
unsafe { T::free(self, address, count) }
|
|
}
|
|
fn allocate_one(&mut self) -> Option<PhysicalAddress> {
|
|
T::allocate_one(self)
|
|
}
|
|
unsafe fn free_one(&mut self, address: PhysicalAddress) {
|
|
unsafe { T::free_one(self, address) }
|
|
}
|
|
fn usage(&self) -> FrameUsage {
|
|
T::usage(self)
|
|
}
|
|
}
|