Skip to main content

islet_rmm/realm/mm/
table_level.rs

1use crate::const_assert_size;
2use crate::realm::mm::address::GuestPhysAddr;
3use crate::realm::mm::entry::Entry;
4use vmsa::page_table::{HasSubtable, Level, PageTable};
5
6use crate::config::PAGE_SIZE;
7
8/// Root Table
9pub enum RootTable<const L: usize, const N: usize> {}
10impl<const L: usize, const N: usize> Level for RootTable<L, N> {
11    const THIS_LEVEL: usize = L;
12    const TABLE_SIZE: usize = (PAGE_SIZE * N);
13    const TABLE_ALIGN: usize = PAGE_SIZE;
14    const NUM_ENTRIES: usize = (Self::TABLE_SIZE / core::mem::size_of::<Entry>());
15}
16
17impl<const N: usize> HasSubtable for RootTable<0, N> {
18    type NextLevel = L1Table;
19}
20
21impl<const N: usize> HasSubtable for RootTable<1, N> {
22    type NextLevel = L2Table;
23}
24
25impl<const N: usize> HasSubtable for RootTable<2, N> {
26    type NextLevel = L3Table;
27}
28
29/// The Level 0 Table
30pub enum L0Table {}
31impl Level for L0Table {
32    const THIS_LEVEL: usize = 0;
33    const TABLE_SIZE: usize = PAGE_SIZE;
34    const TABLE_ALIGN: usize = PAGE_SIZE;
35    const NUM_ENTRIES: usize = (Self::TABLE_SIZE / core::mem::size_of::<Entry>());
36}
37impl HasSubtable for L0Table {
38    type NextLevel = L1Table;
39}
40
41/// The Level 1 Table
42pub enum L1Table {}
43impl Level for L1Table {
44    const THIS_LEVEL: usize = 1;
45    const TABLE_SIZE: usize = PAGE_SIZE;
46    const TABLE_ALIGN: usize = PAGE_SIZE;
47    const NUM_ENTRIES: usize = (Self::TABLE_SIZE / core::mem::size_of::<Entry>());
48}
49impl HasSubtable for L1Table {
50    type NextLevel = L2Table;
51}
52
53/// The Level 2 Table
54pub enum L2Table {}
55impl Level for L2Table {
56    const THIS_LEVEL: usize = 2;
57    const TABLE_SIZE: usize = PAGE_SIZE;
58    const TABLE_ALIGN: usize = PAGE_SIZE;
59    const NUM_ENTRIES: usize = (Self::TABLE_SIZE / core::mem::size_of::<Entry>());
60}
61impl HasSubtable for L2Table {
62    type NextLevel = L3Table;
63}
64
65/// The Level 3 Table (Doesn't have Subtable!)
66pub enum L3Table {}
67impl Level for L3Table {
68    const THIS_LEVEL: usize = 3;
69    const TABLE_SIZE: usize = PAGE_SIZE;
70    const TABLE_ALIGN: usize = PAGE_SIZE;
71    const NUM_ENTRIES: usize = (Self::TABLE_SIZE / core::mem::size_of::<Entry>());
72}
73
74const_assert_size!(PageTable<GuestPhysAddr, L0Table, Entry, { L0Table::NUM_ENTRIES }>, PAGE_SIZE);
75const_assert_size!(PageTable<GuestPhysAddr, L1Table, Entry, { L1Table::NUM_ENTRIES }>, PAGE_SIZE);
76const_assert_size!(PageTable<GuestPhysAddr, L2Table, Entry, { L2Table::NUM_ENTRIES }>, PAGE_SIZE);
77const_assert_size!(PageTable<GuestPhysAddr, L3Table, Entry, { L3Table::NUM_ENTRIES }>, PAGE_SIZE);