实现页面反向映射 (#670)

* 实现页面反向映射

* 完善PAGE_MANAGER初始化时机 && 封装lock函数 && 删掉过时注释

Author: Jomocool

kernel/src/driver/net/dma.rs

@@ -3,7 +3,7 @@ use crate::arch::mm::kernel_page_flags;
 use crate::arch::MMArch;
 
 use crate::mm::kernel_mapper::KernelMapper;
-use crate::mm::page::PageFlags;
+use crate::mm::page::{page_manager_lock_irasave, PageFlags};
 use crate::mm::{
     allocator::page_frame::{
         allocate_page_frames, deallocate_page_frames, PageFrameCount, PhysPageFrame,
@@ -57,7 +57,11 @@ pub unsafe fn dma_dealloc(paddr: usize, vaddr: NonNull<u8>, pages: usize) -> i32
     flusher.flush();
 
     unsafe {
-        deallocate_page_frames(PhysPageFrame::new(PhysAddr::new(paddr)), page_count);
+        deallocate_page_frames(
+            PhysPageFrame::new(PhysAddr::new(paddr)),
+            page_count,
+            &mut page_manager_lock_irasave(),
+        );
     }
     return 0;
 }

kernel/src/driver/virtio/virtio_impl.rs

@@ -3,7 +3,7 @@ use crate::arch::mm::kernel_page_flags;
 use crate::arch::MMArch;
 
 use crate::mm::kernel_mapper::KernelMapper;
-use crate::mm::page::PageFlags;
+use crate::mm::page::{page_manager_lock_irasave, PageFlags};
 use crate::mm::{
     allocator::page_frame::{
         allocate_page_frames, deallocate_page_frames, PageFrameCount, PhysPageFrame,
@@ -68,7 +68,11 @@ unsafe impl Hal for HalImpl {
         flusher.flush();
 
         unsafe {
-            deallocate_page_frames(PhysPageFrame::new(PhysAddr::new(paddr)), page_count);
+            deallocate_page_frames(
+                PhysPageFrame::new(PhysAddr::new(paddr)),
+                page_count,
+                &mut page_manager_lock_irasave(),
+            );
         }
         return 0;
     }

kernel/src/mm/allocator/page_frame.rs

@@ -5,6 +5,7 @@ use core::{
 
 use crate::{
     arch::{mm::LockedFrameAllocator, MMArch},
+    libs::spinlock::SpinLockGuard,
     mm::{MemoryManagementArch, PhysAddr, VirtAddr},
 };
 
@@ -348,8 +349,20 @@ pub unsafe fn allocate_page_frames(count: PageFrameCount) -> Option<(PhysAddr, P
 ///
 /// @param frame 要释放的第一个页帧
 /// @param count 要释放的页帧数量 (必须是2的n次幂)
-pub unsafe fn deallocate_page_frames(frame: PhysPageFrame, count: PageFrameCount) {
+pub unsafe fn deallocate_page_frames(
+    frame: PhysPageFrame,
+    count: PageFrameCount,
+    page_manager: &mut SpinLockGuard<'_, crate::mm::page::PageManager>,
+) {
     unsafe {
         LockedFrameAllocator.free(frame.phys_address(), count);
     }
+
+    // 将已回收的物理页面对应的Page从PAGE_MANAGER中删去
+    let mut frame = frame;
+    for _ in 0..count.data() {
+        let paddr = frame.phys_address();
+        page_manager.remove_page(&paddr);
+        frame = frame.next();
+    }
 }

kernel/src/mm/init.rs

@@ -1,8 +1,11 @@
 use core::{fmt::Write, sync::atomic::Ordering};
 
 use crate::{
-    arch::MMArch, driver::serial::serial8250::send_to_default_serial8250_port,
-    filesystem::procfs::kmsg::kmsg_init, libs::printk::PrintkWriter, mm::mmio_buddy::mmio_init,
+    arch::MMArch,
+    driver::serial::serial8250::send_to_default_serial8250_port,
+    filesystem::procfs::kmsg::kmsg_init,
+    libs::printk::PrintkWriter,
+    mm::{mmio_buddy::mmio_init, page::page_manager_init},
 };
 
 use super::MemoryManagementArch;
@@ -44,6 +47,8 @@ pub unsafe fn mm_init() {
     mmio_init();
     // enable KMSG
     kmsg_init();
+    // enable PAGE_MANAGER
+    page_manager_init();
 
     MM_INIT
         .compare_exchange(

kernel/src/mm/page.rs

@@ -6,17 +6,102 @@ use core::{
     sync::atomic::{compiler_fence, Ordering},
 };
 
+use alloc::sync::Arc;
+use hashbrown::{HashMap, HashSet};
+
 use crate::{
     arch::{interrupt::ipi::send_ipi, MMArch},
     exception::ipi::{IpiKind, IpiTarget},
     kerror, kwarn,
+    libs::spinlock::{SpinLock, SpinLockGuard},
 };
 
 use super::{
-    allocator::page_frame::FrameAllocator, syscall::ProtFlags, MemoryManagementArch, PageTableKind,
-    PhysAddr, VirtAddr,
+    allocator::page_frame::FrameAllocator, syscall::ProtFlags, ucontext::LockedVMA,
+    MemoryManagementArch, PageTableKind, PhysAddr, VirtAddr,
 };
 
+/// 全局物理页信息管理器
+pub static mut PAGE_MANAGER: Option<SpinLock<PageManager>> = None;
+
+/// 初始化PAGE_MANAGER
+pub fn page_manager_init() {
+    kinfo!("page_manager_init");
+    let page_manager = SpinLock::new(PageManager::new());
+
+    compiler_fence(Ordering::SeqCst);
+    unsafe { PAGE_MANAGER = Some(page_manager) };
+    compiler_fence(Ordering::SeqCst);
+
+    kinfo!("page_manager_init done");
+}
+
+pub fn page_manager_lock_irasave() -> SpinLockGuard<'static, PageManager> {
+    unsafe { PAGE_MANAGER.as_ref().unwrap().lock_irqsave() }
+}
+
+// 物理页管理器
+pub struct PageManager {
+    phys2page: HashMap<PhysAddr, Page>,
+}
+
+impl PageManager {
+    pub fn new() -> Self {
+        Self {
+            phys2page: HashMap::new(),
+        }
+    }
+
+    pub fn get_mut(&mut self, paddr: &PhysAddr) -> &mut Page {
+        self.phys2page.get_mut(paddr).unwrap()
+    }
+
+    pub fn insert(&mut self, paddr: PhysAddr, page: Page) {
+        self.phys2page.insert(paddr, page);
+    }
+
+    pub fn remove_page(&mut self, paddr: &PhysAddr) {
+        self.phys2page.remove(paddr);
+    }
+}
+
+/// 物理页面信息
+pub struct Page {
+    /// 映射计数
+    map_count: usize,
+    /// 是否为共享页
+    shared: bool,
+    /// 映射到当前page的VMA
+    anon_vma: HashSet<Arc<LockedVMA>>,
+}
+
+impl Page {
+    pub fn new(shared: bool) -> Self {
+        Self {
+            map_count: 0,
+            shared,
+            anon_vma: HashSet::new(),
+        }
+    }
+
+    /// 将vma加入anon_vma
+    pub fn insert_vma(&mut self, vma: Arc<LockedVMA>) {
+        self.anon_vma.insert(vma);
+        self.map_count += 1;
+    }
+
+    /// 将vma从anon_vma中删去
+    pub fn remove_vma(&mut self, vma: &LockedVMA) {
+        self.anon_vma.remove(vma);
+        self.map_count -= 1;
+    }
+
+    /// 判断当前物理页是否能被回
+    pub fn can_deallocate(&self) -> bool {
+        self.map_count == 0 && !self.shared
+    }
+}
+
 #[derive(Debug)]
 pub struct PageTable<Arch> {
     /// 当前页表表示的虚拟地址空间的起始地址
@@ -591,6 +676,8 @@ impl<Arch: MemoryManagementArch, F: FrameAllocator> PageMapper<Arch, F> {
         compiler_fence(Ordering::SeqCst);
         let phys: PhysAddr = self.frame_allocator.allocate_one()?;
         compiler_fence(Ordering::SeqCst);
+
+        page_manager_lock_irasave().insert(phys, Page::new(false));
         return self.map_phys(virt, phys, flags);
     }