feat(test): add alloc tests

Author: Mivik

Cargo.lock

@@ -30,3 +30,6 @@ riscv = { version = "0.14", default-features = false }
 
 [package.metadata.docs.rs]
 rustc-args = ["--cfg", "doc"]
+
+[dev-dependencies]
+rand = { version = "0.9.1", default-features = false, features = ["small_rng"] }

page_table_multiarch/Cargo.toml

@@ -0,0 +1,126 @@
+use std::{
+    alloc::{self, Layout},
+    cell::RefCell,
+    collections::HashSet,
+    marker::PhantomData,
+};
+
+use memory_addr::{PhysAddr, VirtAddr};
+use page_table_entry::{GenericPTE, MappingFlags};
+use page_table_multiarch::{PageSize, PageTable64, PagingHandler, PagingMetaData, PagingResult};
+use rand::{Rng, SeedableRng, rngs::SmallRng};
+
+const PAGE_LAYOUT: Layout = unsafe { Layout::from_size_align_unchecked(4096, 4096) };
+
+thread_local! {
+    static ALLOCATED: RefCell<HashSet<usize>> = RefCell::default();
+}
+struct TrackPagingHandler<M: PagingMetaData>(PhantomData<M>);
+impl<M: PagingMetaData> PagingHandler for TrackPagingHandler<M> {
+    fn alloc_frame() -> Option<PhysAddr> {
+        let ptr = unsafe { alloc::alloc(PAGE_LAYOUT) } as usize;
+        assert!(
+            ptr <= M::PA_MAX_ADDR,
+            "allocated frame address exceeds PA_MAX_ADDR"
+        );
+        ALLOCATED.with_borrow_mut(|it| it.insert(ptr));
+        Some(PhysAddr::from_usize(ptr))
+    }
+
+    fn dealloc_frame(paddr: PhysAddr) {
+        let ptr = paddr.as_usize();
+        ALLOCATED.with_borrow_mut(|it| {
+            assert!(it.remove(&ptr), "dealloc a frame that was not allocated");
+        });
+        unsafe {
+            alloc::dealloc(ptr as _, PAGE_LAYOUT);
+        }
+    }
+
+    fn phys_to_virt(paddr: PhysAddr) -> VirtAddr {
+        if paddr.as_usize() == 0 {
+            panic!();
+        }
+        VirtAddr::from_usize(paddr.as_usize())
+    }
+}
+
+fn run_test_for<M: PagingMetaData<VirtAddr = VirtAddr>, PTE: GenericPTE>() -> PagingResult<()> {
+    ALLOCATED.with_borrow_mut(|it| {
+        it.clear();
+    });
+
+    let vaddr_mask = ((1u64 << M::VA_MAX_BITS) - 1) & !0xfff;
+
+    let mut table = PageTable64::<M, PTE, TrackPagingHandler<M>>::try_new().unwrap();
+    let mut pages = HashSet::new();
+    let mut rng = SmallRng::seed_from_u64(1234);
+    for _ in 0..2048 {
+        if rng.random_ratio(3, 4) || pages.is_empty() {
+            // insert a mapping
+            let addr = loop {
+                let addr = rng.random::<u64>() & vaddr_mask;
+                if pages.insert(addr) {
+                    break addr;
+                }
+            };
+            table
+                .map(
+                    VirtAddr::from_usize(addr as usize),
+                    PhysAddr::from_usize((rng.random::<u64>() & vaddr_mask) as usize),
+                    PageSize::Size4K,
+                    MappingFlags::READ | MappingFlags::WRITE,
+                )?
+                .ignore();
+        } else {
+            // remove a mapping
+            let addr = *pages.iter().next().unwrap();
+            table.unmap(VirtAddr::from_usize(addr as usize))?.2.ignore();
+            pages.remove(&addr);
+        }
+    }
+
+    drop(table);
+    assert_eq!(
+        ALLOCATED.with_borrow(|it| it.len()),
+        0,
+        "Some frames were not deallocated"
+    );
+
+    Ok(())
+}
+
+#[test]
+fn test_dealloc() -> PagingResult<()> {
+    #[cfg(target_arch = "x86_64")]
+    run_test_for::<
+        page_table_multiarch::x86_64::X64PagingMetaData,
+        page_table_entry::x86_64::X64PTE,
+    >()?;
+
+    #[cfg(any(target_arch = "riscv32", target_arch = "riscv64"))]
+    {
+        run_test_for::<
+            page_table_multiarch::riscv::Sv39MetaData<VirtAddr>,
+            page_table_entry::riscv::Rv64PTE,
+        >()?;
+        run_test_for::<
+            page_table_multiarch::riscv::Sv48MetaData<VirtAddr>,
+            page_table_entry::riscv::Rv64PTE,
+        >()?;
+    }
+
+    #[cfg(target_arch = "aarch64")]
+    run_test_for::<
+        page_table_multiarch::aarch64::A64PagingMetaData,
+        page_table_entry::aarch64::A64PTE,
+    >()?;
+
+    #[cfg(target_arch = "loongarch64")]
+    run_test_for::<
+        page_table_multiarch::loongarch64::LA64MetaData,
+        page_table_entry::loongarch64::LA64PTE,
+    >()?;
+
+    Ok(())
+}