feat: spec-compliant AHCI HBA initialization

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

diff --git a/lunaix-os/includes/lunaix/mm/pmm.h b/lunaix-os/includes/lunaix/mm/pmm.h
index 36396dfcaed049055ad4ee6a2ca71d5060c4c3eb..44aa99d7b0f2e8120b970ecc1343aedd45031320 100644 (file)
--- a/lunaix-os/includes/lunaix/mm/pmm.h
+++ b/lunaix-os/includes/lunaix/mm/pmm.h
@@ -2,64 +2,111 @@
 #define __LUNAIX_PMM_H
 // Physical memory manager
 
 #define __LUNAIX_PMM_H
 // Physical memory manager
 

-#include <stdint.h>
+#include <lunaix/process.h>

 #include <stddef.h>
 #include <stddef.h>

+#include <stdint.h>

 
 

-#define PM_PAGE_SIZE            4096
-#define PM_BMP_MAX_SIZE        (128 * 1024)
+#define PM_PAGE_SIZE 4096
+#define PM_BMP_MAX_SIZE (1024 * 1024)
+
+/**
+ * @brief 长久页：不会被缓存，但允许释放
+ *
+ */
+#define PP_FGPERSIST 0x1
+
+/**
+ * @brief 锁定页：不会被缓存，不能被释放
+ *
+ */
+#define PP_FGLOCKED 0x2
+
+typedef uint32_t pp_attr_t;
+
+struct pp_struct
+{
+    pid_t owner;
+    uint32_t ref_counts;
+    pp_attr_t attr;
+};

 
 /**
  * @brief 标注物理页为可使用
 
 /**
  * @brief 标注物理页为可使用

- * 
+ *

  * @param ppn page number
  */
  * @param ppn page number
  */

-void pmm_mark_page_free(uintptr_t ppn);
+void
+pmm_mark_page_free(uintptr_t ppn);

 
 /**
  * @brief 标注物理页为已占用
 
 /**
  * @brief 标注物理页为已占用

- * 
- * @param ppn 
+ *
+ * @param ppn

  */
  */

-void pmm_mark_page_occupied(uintptr_t ppn);
+void
+pmm_mark_page_occupied(pid_t owner, uintptr_t ppn, pp_attr_t attr);

 
 /**
  * @brief 标注多个连续的物理页为可用
 
 /**
  * @brief 标注多个连续的物理页为可用

- * 
+ *

  * @param start_ppn 起始PPN
  * @param page_count 数量
  */
  * @param start_ppn 起始PPN
  * @param page_count 数量
  */

-void pmm_mark_chunk_free(uintptr_t start_ppn, size_t page_count);
+void
+pmm_mark_chunk_free(uintptr_t start_ppn, size_t page_count);

 
 /**
  * @brief 标注多个连续的物理页为已占用
 
 /**
  * @brief 标注多个连续的物理页为已占用

- * 
+ *

  * @param start_ppn 起始PPN
  * @param page_count 数量
  */
  * @param start_ppn 起始PPN
  * @param page_count 数量
  */

-void pmm_mark_chunk_occupied(uintptr_t start_ppn, size_t page_count);
+void
+pmm_mark_chunk_occupied(pid_t owner,
+                        uintptr_t start_ppn,
+                        size_t page_count,
+                        pp_attr_t attr);

 
 /**
  * @brief 分配一个可用的物理页
 
 /**
  * @brief 分配一个可用的物理页

- * 
+ *

  * @return void* 可用的页地址，否则为 NULL
  */
  * @return void* 可用的页地址，否则为 NULL
  */

-void* pmm_alloc_page();
+void*
+pmm_alloc_page(pid_t owner, pp_attr_t attr);
+
+/**
+ * @brief 分配一个连续的物理内存区域
+ *
+ * @param owner
+ * @param num_pages 区域大小，单位为页
+ * @param attr
+ * @return void*
+ */
+void*
+pmm_alloc_cpage(pid_t owner, size_t num_pages, pp_attr_t attr);

 
 /**
  * @brief 初始化物理内存管理器
 
 /**
  * @brief 初始化物理内存管理器

- * 
+ *

  * @param mem_upper_lim 最大可用内存地址
  */
  * @param mem_upper_lim 最大可用内存地址
  */

-void pmm_init(uintptr_t mem_upper_lim);
+void
+pmm_init(uintptr_t mem_upper_lim);

 
 

+struct pp_struct*
+pmm_query(void* pa);

 
 /**
  * @brief 释放一个已分配的物理页，假若页地址不存在，则无操作。
 
 /**
  * @brief 释放一个已分配的物理页，假若页地址不存在，则无操作。

- * 
+ *

  * @param page 页地址
  * @return 是否成功
  */
  * @param page 页地址
  * @return 是否成功
  */

-int pmm_free_page(void* page);
+int
+pmm_free_page(pid_t owner, void* page);

 
 

+int
+pmm_ref_page(pid_t owner, void* page);

 
 #endif /* __LUNAIX_PMM_H */
 
 #endif /* __LUNAIX_PMM_H */