From 77d443e691cc1a9de7e0cde3e06e4847da327150 Mon Sep 17 00:00:00 2001
From: "Zone.N" <zone.niuzh@hotmail.com>
Date: Wed, 11 Sep 2024 05:18:49 +0000
Subject: [PATCH] feat: working on vmm

Signed-off-by: Zone.N <zone.niuzh@hotmail.com>
---
 .../include/memory/virtual_memory_manager.h   | 17 ++++---------
 src/kernel/virtual_memory_manager.cpp         | 25 ++++++++-----------
 2 files changed, 16 insertions(+), 26 deletions(-)

diff --git a/src/kernel/include/memory/virtual_memory_manager.h b/src/kernel/include/memory/virtual_memory_manager.h
index 7de5158c..f92c7d11 100644
--- a/src/kernel/include/memory/virtual_memory_manager.h
+++ b/src/kernel/include/memory/virtual_memory_manager.h
@@ -70,24 +70,17 @@ class VirtualMemoryManager {
   ~VirtualMemoryManager() = default;
   /// @}
 
-  /**
-   * @brief 初始化
-   * @return true            成功
-   * @return false           失败
-   */
-  bool init(void);
-
   /**
    * @brief 获取当前页目录
    * @return pt_t            当前页目录
    */
-  pt_t get_pgd(void);
+  pt_t GetPageDirectory();
 
   /**
    * @brief 设置当前页目录
    * @param  _pgd            要设置的页目录
    */
-  void set_pgd(const pt_t _pgd);
+  void SetPageDirectory(const pt_t _pgd);
 
   /**
    * @brief 映射物理地址到虚拟地址
@@ -96,14 +89,14 @@ class VirtualMemoryManager {
    * @param  _pa             物理地址
    * @param  _flag           属性
    */
-  void mmap(const pt_t _pgd, uintptr_t _va, uintptr_t _pa, uint32_t _flag);
+  void Mmap(const pt_t _pgd, uintptr_t _va, uintptr_t _pa, uint32_t _flag);
 
   /**
    * @brief 取消映射
    * @param  _pgd            要操作的页目录
    * @param  _va             要取消映射的虚拟地址
    */
-  void unmmap(const pt_t _pgd, uintptr_t _va);
+  void Unmmap(const pt_t _pgd, uintptr_t _va);
 
   /**
    * @brief 获取映射的物理地址
@@ -113,7 +106,7 @@ class VirtualMemoryManager {
    * @return true            已映射
    * @return false           未映射
    */
-  bool get_mmap(const pt_t _pgd, uintptr_t _va, const void *_pa);
+  bool GetMmap(const pt_t _pgd, uintptr_t _va, const void *_pa);
 
  private:
   /// @todo
diff --git a/src/kernel/virtual_memory_manager.cpp b/src/kernel/virtual_memory_manager.cpp
index 73e8ad6f..cc082ba2 100644
--- a/src/kernel/virtual_memory_manager.cpp
+++ b/src/kernel/virtual_memory_manager.cpp
@@ -32,30 +32,29 @@ VirtualMemoryManager::VirtualMemoryManager(uint32_t, uint8_t*) {
        addr < kBasicInfo.GetInstance().kernel_addr + kKernelSpaceSize;
        addr += PhysicalMemoryManager::kPageSize) {
     // TODO: 区分代码/数据等段分别映射
-    mmap(pgd_kernel, addr, addr,
+    Mmap(pgd_kernel, addr, addr,
          cpu::vmm::VMM_PAGE_READABLE | cpu::vmm::VMM_PAGE_WRITABLE |
              cpu::vmm::VMM_PAGE_EXECUTABLE);
   }
   // 设置页目录
-  klog::Debug("set_pgd: 0x%X\n", pgd_kernel);
-  set_pgd(pgd_kernel);
+  klog::Debug("SetPageDirectory: 0x%X\n", pgd_kernel);
+  SetPageDirectory(pgd_kernel);
   // 开启分页
   cpu::vmm::EnablePage();
 }
 
-pt_t VirtualMemoryManager::get_pgd(void) {
+pt_t VirtualMemoryManager::GetPageDirectory() {
   return (pt_t)cpu::vmm::GetPageDirectory();
 }
 
-void VirtualMemoryManager::set_pgd(const pt_t _pgd) {
+void VirtualMemoryManager::SetPageDirectory(const pt_t _pgd) {
   // 设置页目录
   cpu::vmm::SetPageDirectory((uint64_t)_pgd);
   // 刷新缓存
   cpu::vmm::FlushPage(0);
-  return;
 }
 
-void VirtualMemoryManager::mmap(const pt_t _pgd, uintptr_t _va, uintptr_t _pa,
+void VirtualMemoryManager::Mmap(const pt_t _pgd, uintptr_t _va, uintptr_t _pa,
                                 uint32_t _flag) {
   pte_t* pte = find(_pgd, _va, true);
   // 一般情况下不应该为空
@@ -79,31 +78,29 @@ void VirtualMemoryManager::mmap(const pt_t _pgd, uintptr_t _va, uintptr_t _pa,
     // 刷新缓存
     cpu::vmm::FlushPage(0);
   }
-  return;
 }
 
-void VirtualMemoryManager::unmmap(const pt_t _pgd, uintptr_t _va) {
+void VirtualMemoryManager::Unmmap(const pt_t _pgd, uintptr_t _va) {
   pte_t* pte = find(_pgd, _va, false);
   // 找到页表项
   // 未找到
   if (pte == nullptr) {
-    klog::Warn("VirtualMemoryManager::unmmap: find.\n");
+    klog::Warn("VirtualMemoryManager::Unmmap: find.\n");
     return;
   }
   // 找到了，但是并没有被映射
   if ((*pte & cpu::vmm::VMM_PAGE_VALID) == 0) {
-    klog::Warn("VirtualMemoryManager::unmmap: not mapped.\n");
+    klog::Warn("VirtualMemoryManager::Unmmap: not mapped.\n");
   }
   // 置零
   *pte = 0x00;
   // 刷新缓存
   cpu::vmm::FlushPage(0);
   // TODO: 如果一页表都被 unmap，释放占用的物理内存
-  return;
 }
 
-bool VirtualMemoryManager::get_mmap(const pt_t _pgd, uintptr_t _va,
-                                    const void* _pa) {
+bool VirtualMemoryManager::GetMmap(const pt_t _pgd, uintptr_t _va,
+                                   const void* _pa) {
   pte_t* pte = find(_pgd, _va, false);
   bool res = false;
   // pte 不为空且有效，说明映射了