1use crate::{Pte, VmMeta, PPN};
2use core::{
3 marker::PhantomData,
4 ops::{BitAnd, BitAndAssign, BitOr, BitOrAssign, BitXor, BitXorAssign},
5};
6
7#[derive(Clone, Copy, PartialEq, Eq, Hash, Debug)]
11#[repr(transparent)]
12pub struct VmFlags<Meta: VmMeta>(usize, PhantomData<Meta>);
13
14impl<Meta: VmMeta> VmFlags<Meta> {
15 pub const ZERO: Self = Self(0, PhantomData);
17
18 pub const VALID: Self = Self(Meta::VALID_FLAG, PhantomData);
20
21 #[inline]
27 pub const unsafe fn from_raw(raw: usize) -> Self {
28 Self(raw, PhantomData)
29 }
30
31 #[inline]
33 pub const fn val(self) -> usize {
34 self.0
35 }
36
37 #[inline]
39 pub const fn contains(self, flags: VmFlags<Meta>) -> bool {
40 self.0 & flags.0 == flags.0
41 }
42
43 #[inline]
45 pub fn is_leaf(self) -> bool {
46 Meta::is_leaf(self.0)
47 }
48
49 #[inline]
51 pub fn is_huge(self, level: usize) -> bool {
52 Meta::is_huge(self.0, level)
53 }
54
55 #[inline]
57 pub fn valid(self) -> bool {
58 Meta::is_valid(self.0)
59 }
60
61 #[inline]
63 pub fn build_pte(mut self, ppn: PPN<Meta>) -> Pte<Meta> {
64 Meta::set_ppn(&mut self.0, ppn);
65 Pte(self.0, PhantomData)
66 }
67}
68
69impl<Meta: VmMeta> BitAnd for VmFlags<Meta> {
70 type Output = Self;
71
72 #[inline]
73 fn bitand(self, rhs: Self) -> Self::Output {
74 Self(self.0 & rhs.0, PhantomData)
75 }
76}
77
78impl<Meta: VmMeta> BitOr for VmFlags<Meta> {
79 type Output = Self;
80
81 #[inline]
82 fn bitor(self, rhs: Self) -> Self::Output {
83 Self(self.0 | rhs.0, PhantomData)
84 }
85}
86
87impl<Meta: VmMeta> BitXor for VmFlags<Meta> {
88 type Output = Self;
89
90 #[inline]
91 fn bitxor(self, rhs: Self) -> Self::Output {
92 Self(self.0 ^ rhs.0, PhantomData)
93 }
94}
95
96impl<Meta: VmMeta> BitAndAssign for VmFlags<Meta> {
97 #[inline]
98 fn bitand_assign(&mut self, rhs: Self) {
99 self.0 &= rhs.0;
100 }
101}
102
103impl<Meta: VmMeta> BitOrAssign for VmFlags<Meta> {
104 #[inline]
105 fn bitor_assign(&mut self, rhs: Self) {
106 self.0 |= rhs.0;
107 }
108}
109
110impl<Meta: VmMeta> BitXorAssign for VmFlags<Meta> {
111 #[inline]
112 fn bitxor_assign(&mut self, rhs: Self) {
113 self.0 ^= rhs.0;
114 }
115}