From 67446040c856dd92a441bc66a7ca2d5fc7b16a0d Mon Sep 17 00:00:00 2001
From: mykola2312 <49044616+mykola2312@users.noreply.github.com>
Date: Tue, 24 May 2022 21:07:47 +0300
Subject: [PATCH] implement heap_free

---
 heap.c | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

diff --git a/heap.c b/heap.c
index de2b982..0e01677 100644
--- a/heap.c
+++ b/heap.c
@@ -71,6 +71,17 @@ mblock_t* heap_alloc(mheap_t* heap, unsigned int _size)
     return block;
 }
 
+void heap_free(mblock_t* block)
+{
+    block->size &= ~MBLOCK_ALLOCATED;
+    mblock_t* prev = block->prev;
+    mblock_t* next = (mblock_t*)((uint8_t*)block + block->size);
+    if (!(prev->size & MBLOCK_ALLOCATED))
+        heap_join(block, prev);
+    if (!(next->size & MBLOCK_ALLOCATED))
+        heap_join(block, next);
+}
+
 
 static void heap_debug(mheap_t* heap)
 {
@@ -84,7 +95,7 @@ static void heap_debug(mheap_t* heap)
     }
 }
 
-static uint8_t internal_heap[4096] = {0};
+static uint8_t internal_heap[4096];
 
 int main()
 {
@@ -93,6 +104,11 @@ int main()
     
     printf("heap_alloc\t%p\n", heap_alloc(&heap, 8));
     printf("heap_alloc\t%p\n", heap_alloc(&heap, 8));
+    mblock_t* m3 = heap_alloc(&heap, 12);
+    printf("heap_alloc\t%p\n", m3);
+    printf("heap_alloc\t%p\n", heap_alloc(&heap, 8));
+    heap_free(m3);
+    printf("heap_alloc\t%p\n", heap_alloc(&heap, 8));
 
     heap_debug(&heap);
     return 0;