MHR-23: Improve project structure, make it better.

Signed-off-by: Amlal El Mahrouss <amlal@el-mahrouss-logic.com>

1 files changed, 24 insertions, 0 deletions

diff --git a/DriverKit/KernelStdCxx.cc b/DriverKit/KernelStdCxx.cc
new file mode 100644
index 00000000..633437dc
--- /dev/null
+++ b/DriverKit/KernelStdCxx.cc
@@ -0,0 +1,24 @@
+/* -------------------------------------------
+
+    Copyright SoftwareLabs
+
+    Purpose: Driver C++ Definitions.
+
+------------------------------------------- */
+
+#include <DriverKit/KernelStd.h>
+
+void* operator new(size_t sz) {
+    if (!sz) ++sz;
+
+    auto ptr = kernelCall("NewKernelHeap", 1, sz);
+    kernelCall("ProtectKernelHeap", 1, ptr);
+
+    return ptr;
+}
+
+void operator delete(void* ptr) noexcept {
+    if (!ptr) return;
+
+    kernelCall("DeleteKernelHeap", 1,ptr);
+}