refactor heap code and add heap debugging

1 files changed, 13 insertions, 0 deletions

diff --git a/stage3/heap.h b/stage3/heap.h
index 50a3fcf..10fe261 100644
--- a/stage3/heap.h
+++ b/stage3/heap.h
@@ -14,4 +14,17 @@ void *kmalloc(usize siz);
 void kfree(void *ptr);
 void *krealloc(void *ptr, usize size);
 
+typedef struct __attribute__((packed)) heap_header {
+	struct heap_header *next;
+	usize size;
+} heap_header;
+
+heap_header *heap_get_free_ptr();
+
+#ifdef DEBUG
+void heap_check();
+#else
+#define heap_check()
+#endif
+
 #endif