use crate::{Pte, VmMeta, PPN};
use core::{
marker::PhantomData,
ops::{BitAnd, BitAndAssign, BitOr, BitOrAssign, BitXor, BitXorAssign},
};
#[derive(Clone, Copy, PartialEq, Eq, Hash, Debug)]
#[repr(transparent)]
pub struct VmFlags<Meta: VmMeta>(usize, PhantomData<Meta>);
impl<Meta: VmMeta> VmFlags<Meta> {
pub const ZERO: Self = Self(0, PhantomData);
pub const VALID: Self = Self(Meta::VALID_FLAG, PhantomData);
#[inline]
pub const unsafe fn from_raw(raw: usize) -> Self {
Self(raw, PhantomData)
}
#[inline]
pub const fn val(self) -> usize {
self.0
}
#[inline]
pub const fn contains(self, flags: VmFlags<Meta>) -> bool {
self.0 & flags.0 == flags.0
}
#[inline]
pub fn is_leaf(self) -> bool {
Meta::is_leaf(self.0)
}
#[inline]
pub fn is_huge(self, level: usize) -> bool {
Meta::is_huge(self.0, level)
}
#[inline]
pub fn valid(self) -> bool {
Meta::is_valid(self.0)
}
#[inline]
pub fn build_pte(mut self, ppn: PPN<Meta>) -> Pte<Meta> {
Meta::set_ppn(&mut self.0, ppn);
Pte(self.0, PhantomData)
}
}
impl<Meta: VmMeta> BitAnd for VmFlags<Meta> {
type Output = Self;
#[inline]
fn bitand(self, rhs: Self) -> Self::Output {
Self(self.0 & rhs.0, PhantomData)
}
}
impl<Meta: VmMeta> BitOr for VmFlags<Meta> {
type Output = Self;
#[inline]
fn bitor(self, rhs: Self) -> Self::Output {
Self(self.0 | rhs.0, PhantomData)
}
}
impl<Meta: VmMeta> BitXor for VmFlags<Meta> {
type Output = Self;
#[inline]
fn bitxor(self, rhs: Self) -> Self::Output {
Self(self.0 ^ rhs.0, PhantomData)
}
}
impl<Meta: VmMeta> BitAndAssign for VmFlags<Meta> {
#[inline]
fn bitand_assign(&mut self, rhs: Self) {
self.0 &= rhs.0;
}
}
impl<Meta: VmMeta> BitOrAssign for VmFlags<Meta> {
#[inline]
fn bitor_assign(&mut self, rhs: Self) {
self.0 |= rhs.0;
}
}
impl<Meta: VmMeta> BitXorAssign for VmFlags<Meta> {
#[inline]
fn bitxor_assign(&mut self, rhs: Self) {
self.0 ^= rhs.0;
}
}