show total heap sizeHEADmaster

1 files changed, 8 insertions, 7 deletions

diff --git a/stage3/heap.h b/stage3/heap.h
index 0b7057b..15bf071 100644
--- a/stage3/heap.h
+++ b/stage3/heap.h
@@ -3,21 +3,22 @@
 
 #include "def.h"
 
+typedef struct __attribute__((packed)) heap_header {
+	struct heap_header *next;
+	usize size;
+} heap_header;
+
 void heap_init();
+
 void heap_add(void *ptr, usize size);
+heap_header *heap_get_free_ptr();
+usize heap_total_size();
 
 void *try_kmalloc(usize size);
 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