From ca082f686fd2dc7ee6f0284421f6212d6d4acee8 Mon Sep 17 00:00:00 2001
From: Anton Kling <anton@kling.gg>
Date: Thu, 11 Apr 2024 17:22:00 +0200
Subject: bug fixes

---
 kernel/includes/mmu.h | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

(limited to 'kernel/includes')

diff --git a/kernel/includes/mmu.h b/kernel/includes/mmu.h
index 9a27c5b..43814d5 100644
--- a/kernel/includes/mmu.h
+++ b/kernel/includes/mmu.h
@@ -9,8 +9,11 @@ typedef u8 mmu_flags;
 #define MMU_FLAG_RW (1 << 0)
 #define MMU_FLAG_KERNEL (1 << 1)
 
-void *next_page(void *a);
-void *align_page(void *a);
+#define PAGE_SIZE ((uintptr_t)0x1000)
+#define next_page(_ptr)                                                        \
+  ((_ptr) + (PAGE_SIZE - (((uintptr_t)_ptr) & (PAGE_SIZE - 1))))
+#define align_page(_ptr)                                                       \
+  (((((uintptr_t)_ptr) & (PAGE_SIZE - 1)) > 0) ? next_page((_ptr)) : (_ptr))
 
 typedef struct Page {
   u32 present : 1;
-- 
cgit v1.2.3