1 files changed, 32 insertions, 0 deletions
diff --git a/dev/libDDK/src/ddk_alloc.c b/dev/libDDK/src/ddk_alloc.c
new file mode 100644
index 00000000..08527f4d
--- /dev/null
+++ b/dev/libDDK/src/ddk_alloc.c
@@ -0,0 +1,32 @@
+/* -------------------------------------------
+
+  Copyright Amlal El Mahrouss.
+
+  Purpose: DDK allocator.
+
+------------------------------------------- */
+
+#include <DriverKit/ddk.h>
+
+/**
+  \brief Allocates a new heap on the Kernel's side.
+  \param sz the size of the heap block.
+  \return the newly allocated pointer.
+*/
+DDK_EXTERN void* kalloc(size_t sz) {
+  if (!sz) ++sz;
+
+  void* ptr = ke_call_dispatch("mm_alloc_ptr", 1, &sz, sizeof(size_t));
+
+  return ptr;
+}
+
+/**
+  \brief Frees a pointer from the heap.
+  \param ptr the pointer to free.
+*/
+DDK_EXTERN void kfree(void* ptr) {
+  if (!ptr) return;
+
+  ke_call_dispatch("mm_free_ptr", 1, ptr, 0);
+}