1 files changed, 41 insertions, 4 deletions
diff --git a/kc/core/memory.h b/kc/core/memory.h
index 526d8ca..ea86912 100644
--- a/kc/core/memory.h
+++ b/kc/core/memory.h
@@ -1,11 +1,15 @@
 #pragma once
 
-#include <lib.h>
-
 #include <kernel/entry.h>
 #include <kernel/memory/paging.h>
 #include <kernel/memory/range.h>
 
+#include <lib.h>
+
+#include <core/memory.h>
+
+#include "vm_tree.h"
+
 typedef int (*memory_space_handler_func)(uint32_t code, void *vaddr);
 
 enum page_map_flags
@@ -21,20 +25,53 @@ enum page_map_flags
     SIZE_MASK = 0xc,
 };
 
+enum page_alloc_flags
+{
+    PAGE_ALLOC_NONE,
+    PAGE_ALLOC_LOW,
+    PAGE_ALLOC_CONV,
+    PAGE_ALLOC_HIGH,
+    PAGE_ALLOC_ANY
+};
+
+enum vm_alloc_flags
+{
+    VM_ALLOC_ANY = 0,
+    VM_ALLOC_CACHE = 1,
+    VM_ALLOC_CORE = 2,
+    VM_ALLOC_TYPE_MASK = 3,
+
+    VM_ALLOC_IMMEDIATE = 4,
+    VM_ALLOC_ACTION_MASK = 4,
+
+    VM_ALLOC_ANONYMOUS = 8,
+    VM_ALLOC_DIRECT = 16,
+    VM_ALLOC_TRANSLATE = 24,
+    VM_ALLOC_MECHANISM_MASK = 24,
+};
+
 void memory_init(void);
 
+void page_set_present(kc_phys_addr page);
+void page_set_allocated(kc_phys_addr page);
+void page_set_free(kc_phys_addr page);
+
 void *page_map(phys_addr_t paddr, enum page_map_flags flags);
 void page_unmap(void *vaddr);
 
-phys_addr_t page_alloc(void);
+phys_addr_t page_alloc(enum page_alloc_flags flags);
 void page_free(phys_addr_t paddr);
 
 void *heap_alloc(size_t size);
 void heap_free(void *block);
 
-void *vm_alloc(size_t size);
+void *vm_alloc(size_t size, enum vm_alloc_flags flags);
 void vm_free(void *block);
 
 void *memory_alloc(size_t size);
 void memory_free(void *block);
 
+struct vm_tree *vm_get_tree(void);
+
+int anonymous_page_handler(struct vm_tree_node *, uint32_t, void *);
+