someboot 0.4.0

use kernutil::StaticCell;
use page_table_generic::PageTable;
pub use page_table_generic::{PagingError, PagingResult};

use crate::mem::ram::Ram;

pub type ArchPageTable<A> = PageTable<<crate::arch::Arch as crate::ArchTrait>::P, A>;

pub type ArchPte = <<crate::arch::Arch as crate::ArchTrait>::P as page_table_generic::TableMeta>::P;

/// Cacheability used while constructing boot-time leaf descriptors.
#[derive(Clone, Copy, Debug, Default, Eq, PartialEq)]
pub enum MemAttributes {
    /// Ordinary coherent memory.
    #[default]
    Normal,
    /// Ordinary coherent memory with a dedicated CPU-local alias.
    PerCpu,
    /// Device memory.
    Device,
    /// Uncached ordinary memory.
    Uncached,
}

/// Boot-time leaf descriptor attributes interpreted by each architecture.
#[derive(Clone, Copy, Debug, Default, Eq, PartialEq)]
pub struct PteConfig {
    pub read: bool,
    pub writable: bool,
    pub executable: bool,
    pub lower: bool,
    pub dirty: bool,
    pub global: bool,
    pub mem_attr: MemAttributes,
}

static BOOT_TABLE: StaticCell<ArchPageTable<Ram>> = StaticCell::uninit();
pub static mut BOOT_TABLE_ADDR: usize = 0;

pub(crate) fn new_boot_table() -> ArchPageTable<Ram> {
    ArchPageTable::<Ram>::new(Ram).unwrap()
}

pub fn new_page_table<A: page_table_generic::FrameAllocator>(
    allocator: A,
) -> Result<ArchPageTable<A>, PagingError> {
    ArchPageTable::<A>::new(allocator)
}

pub(crate) fn set_boot_table(table: ArchPageTable<Ram>) {
    let root_addr: usize = table.root_paddr().into();
    // aarch64 `LDXR` `LDAXR` not work here before MMU is enabled
    unsafe {
        BOOT_TABLE_ADDR = root_addr;
        BOOT_TABLE.init_single_core(table)
    };
}

pub(crate) fn boot_table_addr() -> usize {
    unsafe { BOOT_TABLE_ADDR }
}

pub fn is_mmu_enabled() -> bool {
    <crate::arch::Arch as crate::ArchTrait>::is_mmu_enabled()
}

pub(crate) fn is_kernel_relocated() -> bool {
    if crate::consts::VM_LOAD_ADDRESS == crate::consts::KERNEL_LOAD_ADDRESS {
        return true;
    }

    <crate::arch::Arch as crate::ArchTrait>::is_kernel_relocated_at(
        is_kernel_relocated as *const () as usize,
    )
}

pub trait PageTableOp {
    /// 映射虚拟地址范围到物理地址范围
    fn map(&mut self, config: &page_table_generic::MapConfig<PteConfig>) -> PagingResult;

    fn unmap(&mut self, start_vaddr: page_table_generic::VirtAddr, size: usize)
    -> PagingResult<()>;
}