1 files changed, 47 insertions, 0 deletions
diff --git a/src/kernel/KernelKit/KernelTaskScheduler.h b/src/kernel/KernelKit/KernelTaskScheduler.h
new file mode 100644
index 00000000..9bfce1d6
--- /dev/null
+++ b/src/kernel/KernelKit/KernelTaskScheduler.h
@@ -0,0 +1,47 @@
+/* ========================================
+
+  Copyright (C) 2025, Amlal El Mahrouss, licensed under the Apache 2.0 license.
+
+======================================== */
+
+#pragma once
+
+/// @file KernelTaskScheduler.h
+/// @brief Kernel Task Scheduler header file.
+/// @author Amlal El Mahrouss (amlal@nekernel.org)
+
+#include <ArchKit/ArchKit.h>
+#include <KernelKit/CoreProcessScheduler.h>
+#include <KernelKit/LockDelegate.h>
+
+namespace Kernel {
+class KernelTaskHelper;
+
+typedef ProcessID KID;
+
+/// @brief Equivalent of USER_PROCESS, but for kernel tasks.
+/// @author Amlal
+class KERNEL_TASK final {
+ public:
+  Char               Name[kSchedNameLen] = {"KERNEL_TASK"};
+  ProcessSubsystem   SubSystem{ProcessSubsystem::kProcessSubsystemKernel};
+  HAL::StackFramePtr StackFrame{nullptr};
+  UInt8*             StackReserve{nullptr};
+  SizeT              StackSize{kSchedMaxStackSz};
+  ProcessImage       Image{};
+  /// @brief a KID is a Kernel ID, it is used to find a task running within
+  /// the kernel.
+  KID Kid{0};
+};
+
+/// @brief Equivalent of UserProcessHelper, but for kernel tasks.
+/// @author Amlal
+class KernelTaskHelper final {
+ public:
+  STATIC Bool Add(HAL::StackFramePtr frame_ptr, ProcessID new_kid);
+  STATIC Bool Remove(const KID kid);
+  STATIC Bool CanBeScheduled(const KERNEL_TASK& process);
+  STATIC ErrorOr<KID> TheCurrentKID();
+  STATIC SizeT        StartScheduling();
+};
+}  // namespace Kernel
+\ No newline at end of file