feat: KID, UPS improvements, and future refactors regarding the

FileTree system inside the UPS.

Signed-off-by: Amlal El Mahrouss <amlal@nekernel.org>

1 files changed, 18 insertions, 0 deletions

diff --git a/dev/kernel/KernelKit/KernelTaskScheduler.h b/dev/kernel/KernelKit/KernelTaskScheduler.h
index 57b83ccb..222041e8 100644
--- a/dev/kernel/KernelKit/KernelTaskScheduler.h
+++ b/dev/kernel/KernelKit/KernelTaskScheduler.h
@@ -15,6 +15,12 @@
 #include <KernelKit/LockDelegate.h>
 
 namespace Kernel {
+class KernelTaskHelper;
+
+typedef PID KID;
+
+/// @brief Equivalent of USER_PROCESS, but for kernel tasks.
+/// @author Amlal
 class KERNEL_TASK final {
  public:
   Char               Name[kSchedNameLen] = {"KERNEL_TASK"};
@@ -23,5 +29,17 @@ class KERNEL_TASK final {
   UInt8*             StackReserve{nullptr};
   SizeT              StackSize{kSchedMaxStackSz};
   PROCESS_IMAGE      Image{};
+  /// @brief a KID is a Kernel Identification Descriptor, 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 Switch(HAL::StackFramePtr frame_ptr, PID new_kid);
+  STATIC Bool CanBeScheduled(const KERNEL_TASK& process);
+  STATIC ErrorOr<PID> TheCurrentKID();
+  STATIC SizeT        StartScheduling();
 };
 }  // namespace Kernel
\ No newline at end of file