FIX: Fixing ARM64 execution when booting from bootloader.

FIX: Now works both for ARM64 and AMD64 EFI backends.
TODO: The ACB backend needs to be done too.

Signed-off-by: Amlal El Mahrouss <amlal.elmahrouss@icloud.com>

1 files changed, 40 insertions, 0 deletions

diff --git a/dev/ZKAKit/KernelKit/UserProcessScheduler.inl b/dev/ZKAKit/KernelKit/UserProcessScheduler.inl
new file mode 100644
index 00000000..b99fafbf
--- /dev/null
+++ b/dev/ZKAKit/KernelKit/UserProcessScheduler.inl
@@ -0,0 +1,40 @@
+namespace Kernel
+{
+	/***********************************************************************************/
+	/** @brief Free pointer from usage. */
+	/***********************************************************************************/
+
+    template <typename T>
+	Boolean UserProcess::Delete(ErrorOr<T*> ptr, const SizeT& sz)
+	{
+		if (!ptr ||
+			sz == 0)
+			return No;
+
+		UserProcessHeapList* entry = this->MemoryHeap;
+
+		while (entry != nullptr)
+		{
+			if (entry->MemoryEntry == ptr.Leak().Leak())
+			{
+#ifdef __ZKA_AMD64__
+				auto pd = hal_read_cr3();
+				hal_write_cr3(this->VMRegister);
+
+				auto ret = mm_delete_heap(entry->MemoryEntry);
+
+				hal_write_cr3(pd);
+
+				return ret;
+#else
+				Bool ret = mm_delete_heap(ptr.Leak().Leak());
+				return ret;
+#endif
+			}
+
+			entry = entry->MemoryNext;
+		}
+
+		return No;
+	}
+} // namespace Kernel
\ No newline at end of file