From 3b3b36dcc6542e203475fe1d50ed89799e3f3fc6 Mon Sep 17 00:00:00 2001
From: Amlal El Mahrouss <amlal@el-mahrouss-logic.com>
Date: Mon, 25 Mar 2024 20:17:53 +0100
Subject: Kernel: implement some tickets, improved stuff.

Signed-off-by: Amlal El Mahrouss <amlal@el-mahrouss-logic.com>
---
 Public/Developer/System.Core/Sources/Heap.cxx      | 83 ++++++++++++++++++++++
 .../System.Core/Sources/LibEntrypoint.cxx          | 14 ++++
 .../Developer/System.Core/Sources/New+Delete.cxx   | 55 ++++++++++++++
 3 files changed, 152 insertions(+)
 create mode 100644 Public/Developer/System.Core/Sources/Heap.cxx
 create mode 100644 Public/Developer/System.Core/Sources/LibEntrypoint.cxx
 create mode 100644 Public/Developer/System.Core/Sources/New+Delete.cxx

(limited to 'Public/Developer/System.Core/Sources')

diff --git a/Public/Developer/System.Core/Sources/Heap.cxx b/Public/Developer/System.Core/Sources/Heap.cxx
new file mode 100644
index 00000000..18106f41
--- /dev/null
+++ b/Public/Developer/System.Core/Sources/Heap.cxx
@@ -0,0 +1,83 @@
+/* -------------------------------------------
+
+    Copyright Mahrouss Logic
+
+------------------------------------------- */
+
+#include <System.Core/Headers/Heap.hxx>
+
+/// @brief Allocate from the user's heap.
+/// @param refObj Process object.
+/// @param sz size of object.
+/// @param flags flags.
+/// @return
+CA_EXTERN_C PtrVoidType HcAllocateProcessHeap(ObjectPtr refObj, QWordType sz,
+                                              DWordType flags) {
+  CA_MUST_PASS(sz);
+  CA_MUST_PASS(flags);
+
+  return (PtrVoidType)refObj->Invoke(refObj, kProcessCallAllocPtr, sz, flags);
+}
+
+/// @brief Free pointer from the user's heap.
+/// @param refObj Process object.
+/// @param ptr the pointer to free.
+CA_EXTERN_C VoidType HcFreeProcessHeap(ObjectPtr refObj, PtrVoidType ptr) {
+  CA_MUST_PASS(ptr);
+  CA_UNREFERENCED_PARAMETER(refObj->Invoke(refObj, kProcessCallFreePtr, ptr));
+}
+
+/// @brief Get pointer size.
+/// @param refObj Process object.
+/// @param ptr the pointer to find.
+/// @return the size.
+CA_EXTERN_C QWordType HcProcessHeapSize(ObjectPtr refObj, PtrVoidType ptr) {
+  CA_MUST_PASS(ptr);
+  return refObj->Invoke(refObj, kProcessCallSizePtr, ptr);
+}
+
+/// @brief Check if the pointer exists.
+/// @param refObj Process object.
+/// @param ptr the pointer to check.
+/// @return if it exists
+CA_EXTERN_C BooleanType HcProcessHeapExists(ObjectPtr refObj, PtrVoidType ptr) {
+  CA_MUST_PASS(ptr);
+  return refObj->Invoke(refObj, kProcessCallCheckPtr, ptr);
+}
+
+using namespace System;
+
+/// @brief Shared instance of the heap.
+/// @return
+HeapInterface* HeapInterface::Shared() noexcept {
+  static HeapInterface* heap = nullptr;
+
+  if (!heap) {
+    heap = new HeapInterface();
+  }
+
+  return heap;
+}
+
+HeapInterface::HeapInterface() {
+  CA_MUST_PASS(HcProcessHeapExists(kApplicationObject, (PtrVoidType)this));
+}
+
+HeapInterface::~HeapInterface() { delete this; }
+
+void HeapInterface::Delete(PtrHeapType me) noexcept {
+  CA_MUST_PASS(me);
+  HcFreeProcessHeap(kApplicationObject, me);
+}
+
+SizeType HeapInterface::Size(PtrHeapType me) noexcept {
+  CA_MUST_PASS(me);
+  return HcProcessHeapSize(kApplicationObject, me);
+}
+
+PtrHeapType HeapInterface::New(const SizeType& sz, const DWordType flags) {
+  SizeType _sz = sz;
+  if (!_sz) ++_sz;
+
+  return HcAllocateProcessHeap(kApplicationObject, _sz, flags);
+}
diff --git a/Public/Developer/System.Core/Sources/LibEntrypoint.cxx b/Public/Developer/System.Core/Sources/LibEntrypoint.cxx
new file mode 100644
index 00000000..b874ef55
--- /dev/null
+++ b/Public/Developer/System.Core/Sources/LibEntrypoint.cxx
@@ -0,0 +1,14 @@
+/** ===========================================
+ (C) Mahrouss Logic
+    ===========================================*/
+
+#include <System.Core/Headers/Heap.hxx>
+
+/// @brief Inits the DLL.
+/// @return if it was succesful or not.
+DWordType __DllMain(VoidType) {
+  kApplicationObject = HcGetInstanceObject();
+  CA_MUST_PASS(kApplicationObject);
+
+  return 0;
+}
\ No newline at end of file
diff --git a/Public/Developer/System.Core/Sources/New+Delete.cxx b/Public/Developer/System.Core/Sources/New+Delete.cxx
new file mode 100644
index 00000000..98674a73
--- /dev/null
+++ b/Public/Developer/System.Core/Sources/New+Delete.cxx
@@ -0,0 +1,55 @@
+/* -------------------------------------------
+
+    Copyright Mahrouss Logic
+
+------------------------------------------- */
+
+#include <System.Core/Headers/Heap.hxx>
+#include <System.Core/Headers/Heap.hxx>
+
+#define kAllocationTypes 2
+
+enum HcAllocationKind {
+  kStandardAllocation = 0xC,
+  kArrayAllocation = 0xD,
+};
+
+CA_EXTERN_C PtrVoidType HcAllocateProcessHeap(ObjectPtr refObj, QWordType sz,
+                                              DWordType flags);
+CA_EXTERN_C BooleanType HcProcessHeapExists(ObjectPtr refObj, PtrVoidType ptr);
+CA_EXTERN_C QWordType HcProcessHeapSize(ObjectPtr refObj, PtrVoidType ptr);
+CA_EXTERN_C VoidType HcFreeProcessHeap(ObjectPtr refObj, PtrVoidType ptr);
+
+typedef SizeType size_t;
+
+void* operator new[](size_t sz) {
+  if (sz == 0) ++sz;
+
+  return HcAllocateProcessHeap(kApplicationObject, sz, kStandardAllocation);
+}
+
+void* operator new(size_t sz) {
+  if (sz == 0) ++sz;
+
+  return HcAllocateProcessHeap(kApplicationObject, sz, kArrayAllocation);
+}
+
+void operator delete[](void* ptr) {
+  if (ptr == nullptr) return;
+
+  HcFreeProcessHeap(kApplicationObject, ptr);
+}
+
+void operator delete(void* ptr) {
+  if (ptr == nullptr) return;
+
+  HcFreeProcessHeap(kApplicationObject, ptr);
+}
+
+void operator delete(void* ptr, size_t sz) {
+  if (ptr == nullptr) return;
+
+  (void)sz;
+
+  HcFreeProcessHeap(kApplicationObject, ptr);
+}
-- 
cgit v1.2.3