feat(heap): Create new HeapAllocator implementation

kernel/src/Memory/Heap.cpp

@@ -14,38 +14,122 @@
 #include <Platform/Util.hpp>
 
 #include <Common/Panic.hpp>
+#include <CppLib/Vector.hpp>
+
+#include "PageFrameAllocator.hpp"
 
 namespace Memory
 {
-    constexpr size_t metadata_size = 8; // 8 bytes
-    
-    HeapAllocator::HeapAllocator(LargestSection section)
+    HeapAllocator::Header* HeapAllocator::GetHeader(void* block) {
+        uintptr_t ptr = (uintptr_t)block;
+        return (Header*)(block - sizeof(Header));
+    }
+
+    size_t HeapAllocator::GetAllocatedBlockSize(void* ptr) {
+        Header* header = GetHeader(ptr);
+        return header->size;
+    }
+
+    void HeapAllocator::InsertToFreelist(void* ptr, std::size_t size) {
+        auto prev_next = head.next;
+
+        head.next = (Node*)ptr;
+        head.next->next = prev_next;
+        head.next->size = size;
+    }
+
+    void HeapAllocator::InsertPageToFreelist() {
+        InsertToFreelist(Memory::g_pfa->Allocate(), 0x1000);
+    }
+
+    void HeapAllocator::InsertPagesToFreelist(std::size_t n) {
+        auto ptr = Memory::g_pfa->ReallocConsecutive(nullptr, n);
+        size_t size = 0x1000 * n;
+
+        InsertToFreelist(ptr, size);
+
+        kout << "HeapAllocator: expanded heap by " << base::dec << n << " pages" << Kt::newline;
+    }    
+
+    HeapAllocator::HeapAllocator()
     {
         kout << "HeapAllocator: constructor called" << Kt::newline;
+        InsertPagesToFreelist(8);
     }
 
     void* HeapAllocator::Request(size_t size) {
-        return nullptr;
+        Node* current = head.next;
+        Node* prev = &head;
+
+        size_t sizeNeeded = size + sizeof(Header);
+
+        while (current != nullptr) {
+            if (current->size >= sizeNeeded) {
+                // Unlink the node
+                auto locatedBlockSize = current->size;
+
+                prev->next = current->next;
+                Header* header = (Header*)current;
+                header->size = size;
+
+                void* block = (void*)((uintptr_t)header + sizeof(Header));
+
+                if (locatedBlockSize > sizeNeeded) {
+                    void* rest = (void*)((uintptr_t)header + sizeNeeded);
+                    auto newBlockSize = locatedBlockSize - sizeNeeded;
+
+                    InsertToFreelist(rest, newBlockSize);
+                }
+
+                return block;
+            }
+
+            prev = current;
+            current = current->next;
+        }
+
+        // First pass allocation failed
+        size_t pagesNeeded = size / 0x1000;
+        InsertPagesToFreelist(pagesNeeded);
+
+        return Request(size);
     }
 
     void* HeapAllocator::Realloc(void* ptr, size_t size) {
-        return nullptr;
+        auto new_block = Request(size);
+
+        if (ptr != nullptr && new_block != nullptr) {
+            memcpy(new_block, ptr, size);
+            Free(ptr);
+        }
+
+        return new_block;
     }
 
     void HeapAllocator::Free(void* ptr) {
+        Header* header = GetHeader(ptr);
+        auto size = header->size;
+        auto actualSize = size + sizeof(Header);
+        void* actualBlock = (void*)header;
 
+        auto prev_next = head.next;
+
+        // Relink the full node back into the list
+        head.next = (Node*)actualBlock;
+        head.next->size = actualSize;
+        head.next->next = prev_next;
     }
     
     // Traverses the Allocator's linked list for debugging
     void HeapAllocator::Walk() {
-        // Node* current = {head.next};
-        // size_t i{0};
+        Node* current = {head.next};
+        size_t i{0};
 
-        // while (current != nullptr) {
-        //     kout << "Block " << base::dec << i << " {" << current->size << " bytes & address 0x" << base::hex << (uint64_t)current << "}" << Kt::newline;
-        //     current = current->next;
-        //     i++;
-        // }
+        while (current != nullptr) {
+            kout << "HeapAllocator: " << base::dec << i << " " << current->size << " bytes & address 0x" << base::hex << (uint64_t)current << Kt::newline;
+            current = current->next;
+            i++;
+        }
     }
 
 };

kernel/src/Memory/Heap.hpp

@@ -3,25 +3,28 @@
 
 namespace Memory {
     class HeapAllocator {
-        LargestSection g_section;
-
         struct Node {
             size_t size;
             Node* next;
         };
 
-        struct Metadata {
-            size_t size;
-        };
+        struct Header {
+            std::size_t size;
+        }__attribute__((packed));
 
-        Node head{0, 0};
+        Node head{};
 
+        Header* GetHeader(void* block);
+        void InsertToFreelist(void* ptr, std::size_t size);
+        void InsertPageToFreelist();
+        void InsertPagesToFreelist(std::size_t n);
     public:
-        HeapAllocator(LargestSection section);
+        HeapAllocator();
         void* Request(size_t size);
         void* Realloc(void* ptr, size_t size);
         void Free(void *pagePtr);
         void Walk();
+        size_t GetAllocatedBlockSize(void* ptr);
     };
 
     extern HeapAllocator* g_heap;

kernel/src/Memory/PageFrameAllocator.cpp

@@ -6,11 +6,10 @@
 
 #include "PageFrameAllocator.hpp"
 #include "HHDM.hpp"
-
 #include <Libraries/Memory.hpp>
 #include <Terminal/Terminal.hpp>
-
 #include <CppLib/Spinlock.hpp>
+#include <Common/Panic.hpp>
 
 namespace Memory {
     PageFrameAllocator::PageFrameAllocator(LargestSection section) {
@@ -26,7 +25,7 @@ namespace Memory {
     }
 
     void* PageFrameAllocator::Allocate() {
-        Lock.Aquire();
+        Lock.Acquire();
 
         Page* current = head.next;
         Page* prev = &head;
@@ -64,7 +63,9 @@ namespace Memory {
         auto first = Allocate();
 
         for (int i = 0; i < n - 1; i++) {
-            Allocate();
+            if (Allocate() == nullptr) {
+                Panic("PageFrameAllocator Reallocation failed", nullptr);
+            };
         }
 
         if (ptr != nullptr) {
@@ -76,7 +77,7 @@ namespace Memory {
     }
 
     void PageFrameAllocator::Free(void* ptr) {
-        Lock.Aquire();
+        Lock.Acquire();
         auto prev_next = head.next;
         head.next = (Page*)ptr;