coreboot_table/
mapper.rs

1#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd)]
2pub struct PhysicalAddress(pub usize);
3
4#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd)]
5pub struct VirtualAddress(pub usize);
6
7pub trait Mapper {
8    unsafe fn map_aligned(&mut self, address: PhysicalAddress, size: usize) -> Result<VirtualAddress, &'static str>;
9    unsafe fn unmap_aligned(&mut self, address: VirtualAddress) -> Result<(), &'static str>;
10    fn page_size(&self) -> usize;
11
12    unsafe fn map(&mut self, address: PhysicalAddress, size: usize) -> Result<VirtualAddress, &'static str> {
13        let page_size = self.page_size();
14        let page = address.0/page_size;
15        let aligned_address = PhysicalAddress(page * page_size);
16        let offset = address.0 - aligned_address.0;
17        let pages = (offset + size + page_size - 1) / page_size;
18        let aligned_size = pages * page_size;
19        let virtual_address = self.map_aligned(aligned_address, aligned_size)?;
20        Ok(VirtualAddress(virtual_address.0 + offset))
21    }
22
23    unsafe fn unmap(&mut self, address: VirtualAddress) -> Result<(), &'static str> {
24        let page_size = self.page_size();
25        let page = address.0/page_size;
26        let aligned_address = VirtualAddress(page * page_size);
27        self.unmap_aligned(aligned_address)
28    }
29}
coreboot_table/mapper.rs

coreboot_table/
mapper.rs
