Make translate and map_to safe by making them RecursivePageTable methods

This commit is contained in:
Philipp Oppermann
2015-12-09 01:01:32 +01:00
parent fb7d2d22b6
commit 86d8e99271

View File

@@ -57,17 +57,15 @@ impl RecursivePageTable {
fn p4_mut(&mut self) -> &mut Table<Level4> {
unsafe { self.p4.get_mut() }
}
}
pub unsafe fn translate(virtual_address: VirtualAddress) -> Option<PhysicalAddress> {
pub fn translate(&self, virtual_address: VirtualAddress) -> Option<PhysicalAddress> {
let offset = virtual_address % PAGE_SIZE;
translate_page(Page::containing_address(virtual_address))
self.translate_page(Page::containing_address(virtual_address))
.map(|frame| frame.number * PAGE_SIZE + offset)
}
unsafe fn translate_page(page: Page) -> Option<Frame> {
let p3 = unsafe { &*table::P4 }.next_table(page.p4_index());
fn translate_page(&self, page: Page) -> Option<Frame> {
let p3 = self.p4().next_table(page.p4_index());
let huge_page = || {
p3.and_then(|p3| {
@@ -104,14 +102,14 @@ unsafe fn translate_page(page: Page) -> Option<Frame> {
.or_else(huge_page)
}
pub unsafe fn map_to<A>(page: Page, frame: Frame, flags: EntryFlags, allocator: &mut A)
pub fn map_to<A>(&mut self, page: Page, frame: Frame, flags: EntryFlags, allocator: &mut A)
where A: FrameAllocator
{
let p4 = unsafe { &mut *table::P4 };
let mut p3 = p4.next_table_create(page.p4_index(), allocator);
let mut p3 = self.p4_mut().next_table_create(page.p4_index(), allocator);
let mut p2 = p3.next_table_create(page.p3_index(), allocator);
let mut p1 = p2.next_table_create(page.p2_index(), allocator);
assert!(p1[page.p1_index()].is_unused());
p1[page.p1_index()].set(frame, flags | PRESENT);
}
}