#ifndef __LUNAIX_VMM_H
#define __LUNAIX_VMM_H
-#include <stdint.h>
-#include <stddef.h>
-#include <lunaix/mm/page.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>
-/**
- * @brief 创建一个页目录
- *
- * @return ptd_entry* 页目录的物理地址,随时可以加载进CR3
- */
-ptd_t* vmm_init_pd();
+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 = pte_advance(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
- * 尝试建立一个映射关系。映射指定的物理页地址至虚拟页地址,如果指定的虚拟页地址已被占用
- * 则尝试寻找新的可用地址(改地址总是大于指定的地址)。
- *
- * @param vpn 虚拟页地址
- * @param ppn 物理页地址
- * @param dattr PDE 的属性
- * @param tattr PTE 的属性
- * @return 虚拟页地址,如不成功,则为 NULL
+ * @brief 在指定的虚拟地址空间里查找一个映射
+ *
+ * @param mnt 地址空间锚定点
+ * @param va 虚拟地址
+ * @param mapping 映射相关属性
+ * @return int
*/
-void* vmm_map_page(void* vpn, void* ppn, pt_attr dattr, pt_attr tattr);
+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;
+
+ return !pte_isnull(pte);
+}
/**
- * @brief 尝试为一个虚拟页地址创建一个可用的物理页映射
- *
- * @param vpn 虚拟页地址
- * @return 物理页地址,如不成功,则为 NULL
+ * @brief 挂载另一个虚拟地址空间至当前虚拟地址空间
+ *
+ * @param pde 页目录的物理地址
+ * @return ptr_t
*/
-void* vmm_alloc_page(void* vpn, pt_attr dattr, pt_attr tattr);
+ptr_t
+vms_mount(ptr_t mnt, ptr_t pde);
/**
- * @brief 删除一个映射
- *
- * @param vpn
+ * @brief 卸载已挂载的虚拟地址空间
+ *
*/
-void vmm_unmap_page(void* vpn);
+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));
-ptd_t* get_pd();
-void set_pd(ptd_t* pd);
+ tlb_flush_kernel(mnt);
+ return mnt;
+}
+
+static inline ptr_t
+unmount_page(ptr_t mnt) {
+ pte_t* ptep = mkptep_va(VMS_SELF, mnt);
+ set_pte(ptep, null_pte);
+
+ tlb_flush_kernel(mnt);
+ return mnt;
+}
/**
- * @brief 将虚拟地址翻译为其对应的物理映射
- *
+ * @brief 将当前地址空间的虚拟地址转译为物理地址。
+ *
* @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);
+}
+
+void
+vmap_set_start(ptr_t start_addr);
#endif /* __LUNAIX_VMM_H */