Samuka007/asterinas
1
0
Fork 0
mirror of https://github.com/asterinas/asterinas.git synced 2025-06-23 01:13:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

Refactor alloc_single API for page and frame

Browse Source
This commit is contained in:
nishirong
2024-06-14 16:21:40 +08:00
committed by Tate, Hongliang Tian
parent bd1d076e8c
commit 2ceba78d47
2 changed files with 11 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
framework/aster-frame/src/mm/frame/options.rs
View File

@ -54,7 +54,9 @@ impl FrameAllocOptions {
} else {
let mut frame_list = Vec::new();
for _ in 0..self.nframes {
frame_list.push(allocator::alloc_single().ok_or(Error::NoMemory)?);
let page = allocator::alloc_single().ok_or(Error::NoMemory)?;
let frame = Frame { page };
frame_list.push(frame);
}
FrameVec(frame_list)
};
@ -73,7 +75,8 @@ impl FrameAllocOptions {
return Err(Error::InvalidArgs);
}
let frame = allocator::alloc_single().ok_or(Error::NoMemory)?;
let page = allocator::alloc_single().ok_or(Error::NoMemory)?;
let frame = Frame { page };
if !self.uninit {
frame.writer().fill(0);
}

11
framework/aster-frame/src/mm/page/allocator.rs
View File

@ -12,7 +12,10 @@ use buddy_system_allocator::FrameAllocator;
use log::info;
use spin::Once;
use super::{meta::FrameMeta, Page};
use super::{
meta::{FrameMeta, PageMeta},
Page,
};
use crate::{
boot::memory_region::MemoryRegionType,
mm::{Frame, FrameVec, Segment, PAGE_SIZE},
@ -40,12 +43,10 @@ pub(crate) fn alloc(nframes: usize) -> Option<FrameVec> {
})
}
pub(crate) fn alloc_single() -> Option<Frame> {
pub(crate) fn alloc_single<T: PageMeta>() -> Option<Page<T>> {
FRAME_ALLOCATOR.get().unwrap().lock().alloc(1).map(|idx| {
let paddr = idx * PAGE_SIZE;
Frame {
page: Page::<FrameMeta>::from_unused(paddr),
}
Page::<T>::from_unused(paddr)
})
}