fix bugs found in brk & add simple security checks on lx_free

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

diff --git a/lunaix-os/includes/lunaix/mm/dmm.h b/lunaix-os/includes/lunaix/mm/dmm.h
index 6ea878b57cfdd12e016807c50f4c05d2586019f3..cfe15b3bfdce3d5578624c34d8ce4484238eba6e 100644 (file)
--- a/lunaix-os/includes/lunaix/mm/dmm.h
+++ b/lunaix-os/includes/lunaix/mm/dmm.h
@@ -4,13 +4,27 @@
 
 #include <stddef.h>
 
-void
-lxsbrk(void* current, void* next);
+#define HEAP_INIT_SIZE 4096
 
-void
-lxmalloc(size_t size);
+typedef struct 
+{
+    void* start;
+    void* brk;
+} heap_context_t;
+
+
+int
+dmm_init(heap_context_t* heap);
+
+int
+lxsbrk(heap_context_t* heap, void* addr);
+void*
+lxbrk(heap_context_t* heap, size_t size);
+
+void*
+lx_malloc(heap_context_t* heap, size_t size);
 
 void
-lxfree(size_t size);
+lx_free(void* ptr);
 
 #endif /* __LUNAIX_DMM_H */