Code-base clean-up and refactoring (#47)

[lunaix-os.git] / lunaix-os / includes / lunaix / mm / vmm.h

diff --git a/lunaix-os/includes/lunaix/mm/vmm.h b/lunaix-os/includes/lunaix/mm/vmm.h
index 2e9f72f67eb903be228ef7f05dd360980fa66b68..a5bde364761c53c1d5441ead8805a1b05fa156b7 100644 (file)
--- a/lunaix-os/includes/lunaix/mm/vmm.h
+++ b/lunaix-os/includes/lunaix/mm/vmm.h
@@ -1,140 +1,109 @@
 #ifndef __LUNAIX_VMM_H
 #define __LUNAIX_VMM_H
 #ifndef __LUNAIX_VMM_H
 #define __LUNAIX_VMM_H

-#include <lunaix/mm/page.h>
-#include <lunaix/process.h>
-#include <stddef.h>
-#include <stdint.h>
-#include <lunaix/mm/pmm.h>
-// Virtual memory manager

 
 

-/**
- * @brief 初始化虚拟内存管理器
- *
- */
-void
-vmm_init();
+#include <lunaix/mm/pagetable.h>
+#include <lunaix/mm/vmtlb.h>
+#include <lunaix/process.h>
+#include <lunaix/types.h>
+
+static inline void 
+vmm_set_ptes_contig(pte_t* ptep, pte_t pte, size_t lvl_size, size_t n)
+{
+    do {
+        set_pte(ptep, pte);
+        pte_val(pte) += lvl_size;
+        ptep++;
+    } while (--n > 0);
+}
+
+static inline void 
+vmm_set_ptes(pte_t* ptep, pte_t pte, size_t n)
+{
+    do {
+        set_pte(ptep, pte);
+        ptep++;
+    } while (--n > 0);
+}
+
+
+static inline void 
+vmm_unset_ptes(pte_t* ptep, size_t n)
+{
+    do {
+        set_pte(ptep, null_pte);
+        ptep++;
+    } while (--n > 0);
+}
+
+pte_t
+vmm_tryptep(pte_t* ptep, size_t lvl_size);

 
 /**
 
 /**

- * @brief å\88\9b建ä¸\80个页ç\9b®å½\95
+ * @brief å\9c¨æ\8c\87å®\9aç\9a\84è\99\9aæ\8b\9få\9c°å\9d\80空é\97´é\87\8cæ\9f¥æ\89¾ä¸\80个æ\98 å°\84

  *
  *

- * @return ptd_entry* 页目录的物理地址，随时可以加载进CR3
+ * @param mnt 地址空间锚定点
+ * @param va 虚拟地址
+ * @param mapping 映射相关属性
+ * @return int

  */
  */

-x86_page_table*
-vmm_init_pd();
+static inline bool
+vmm_lookupat(ptr_t mnt, ptr_t va, pte_t* pte_out)
+{
+    pte_t pte = vmm_tryptep(mkptep_va(mnt, va), LFT_SIZE);
+    *pte_out = pte;

 
 

-/**
- * @brief 尝试建立一个映射关系。映射指定的物理页地址至虚拟页地址，如果指定的虚拟页地址已被占用
- * 则尝试寻找新的可用地址（该地址总是大于指定的地址）。
- *
- * @param vpn 虚拟页地址
- * @param pa 物理页地址
- * @param dattr PDE 的属性
- * @param tattr PTE 的属性
- * @return 虚拟页地址，如不成功，则为 NULL
- */
-void*
-vmm_map_page(pid_t pid, void* va, void* pa, pt_attr tattr);
+    return !pte_isnull(pte);
+}

 
 /**
 
 /**

- * @brief 建立一个映射关系，映射指定的物理页地址至虚拟页地址。如果指定的虚拟页地址已被占用，
- * 则覆盖。
+ * @brief 挂载另一个虚拟地址空间至当前虚拟地址空间

  *
  *

- * @param va 虚拟页地址
- * @param pa 物理页地址
- * @param dattr PDE 的属性
- * @param tattr PTE 的属性
- * @return 虚拟页地址
+ * @param pde 页目录的物理地址
+ * @return ptr_t

  */
  */

-void*
-vmm_fmap_page(pid_t pid, void* va, void* pa, pt_attr tattr);
+ptr_t
+vms_mount(ptr_t mnt, ptr_t pde);

 
 /**
 
 /**

- * @brief å°\9dè¯\95为ä¸\80个è\99\9aæ\8b\9f页å\9c°å\9d\80å\88\9b建ä¸\80个å\8f¯ç\94¨ç\9a\84ç\89©ç\90\86页æ\98 å°\84
+ * @brief å\8d¸è½½å·²æ\8c\82è½½ç\9a\84è\99\9aæ\8b\9få\9c°å\9d\80空é\97´

  *
  *

- * @param va 虚拟页地址
- * @return 虚拟页地址，如不成功，则为 NULL

  */
  */

-void*
-vmm_alloc_page(pid_t pid, void* va, void** pa, pt_attr tattr, pp_attr_t pattr);
+ptr_t
+vms_unmount(ptr_t mnt);

 
 

+static inline ptr_t 
+mount_page(ptr_t mnt, ptr_t pa) {
+    assert(pa);
+    pte_t* ptep = mkptep_va(VMS_SELF, mnt);
+    set_pte(ptep, mkpte(pa, KERNEL_DATA));

 
 

-/**
- * @brief 尝试分配多个连续的虚拟页
- * 
- * @param va 起始虚拟地址
- * @param sz 大小（必须为4K对齐）
- * @param tattr 属性
- * @return int 是否成功
- */
-int
-vmm_alloc_pages(pid_t pid, void* va, size_t sz, pt_attr tattr, pp_attr_t pattr);
+    tlb_flush_kernel(mnt);
+    return mnt;
+}

 
 

-/**
- * @brief 设置一个映射，如果映射已存在，则忽略。
- * 
- * @param va 
- * @param pa 
- * @param attr 
- */
-int
-vmm_set_mapping(pid_t pid, void* va, void* pa, pt_attr attr);
+static inline ptr_t 
+unmount_page(ptr_t mnt) {
+    pte_t* ptep = mkptep_va(VMS_SELF, mnt);
+    set_pte(ptep, null_pte);

 
 

-/**
- * @brief 删除并释放一个映射
- *
- * @param vpn
- */
-void
-vmm_unmap_page(pid_t pid, void* va);
+    tlb_flush_kernel(mnt);
+    return mnt;
+}

 
 /**
 
 /**

- * @brief 删除一个映射
- *
- * @param vpn
- */
-void
-vmm_unset_mapping(void* va);
-
-/**
- * @brief 将虚拟地址翻译为其对应的物理映射
+ * @brief 将当前地址空间的虚拟地址转译为物理地址。

  *
  * @param va 虚拟地址
  *
  * @param va 虚拟地址

- * @return void* 物理地址，如映射不存在，则为NULL
+ * @return void*

  */
  */

-void*
-vmm_v2p(void* va);
+static inline ptr_t
+vmm_v2p(ptr_t va)
+{
+    pte_t* ptep = mkptep_va(VMS_SELF, va);
+    return pte_paddr(pte_at(ptep)) + va_offset(va);
+}

 
 

-/**
- * @brief 查找一个映射
- *
- * @param va 虚拟地址
- * @return v_mapping 映射相关属性
- */
-v_mapping
-vmm_lookup(void* va);
-
-/**
- * @brief (COW) 为虚拟页创建副本。
- * 
- * @return void* 包含虚拟页副本的物理页地址。
- * 
- */
-void* vmm_dup_page(pid_t pid, void* pa);
-
-/**
- * @brief 挂载另一个虚拟地址空间至当前虚拟地址空间
- * 
- * @param pde 页目录的物理地址
- * @return void* 
- */
-void*
-vmm_mount_pd(void* pde);
-
-/**
- * @brief 卸载已挂载的虚拟地址空间
- * 
- */
-void*
-vmm_unmount_pd();
+void
+vmap_set_start(ptr_t start_addr);

 
 #endif /* __LUNAIX_VMM_H */
 
 #endif /* __LUNAIX_VMM_H */