1 files changed, 126 insertions, 0 deletions
diff --git a/Kernel/KernelKit/MPManager.hpp b/Kernel/KernelKit/MPManager.hpp
new file mode 100644
index 00000000..6f23dd66
--- /dev/null
+++ b/Kernel/KernelKit/MPManager.hpp
@@ -0,0 +1,126 @@
+/* -------------------------------------------
+
+	Copyright ZKA Technologies
+
+------------------------------------------- */
+
+#ifndef __INC_MP_MANAGER_HPP__
+#define __INC_MP_MANAGER_HPP__
+
+#include <ArchKit/ArchKit.hpp>
+#include <CompilerKit/CompilerKit.hxx>
+#include <NewKit/Ref.hxx>
+
+/// @note Last Rev Sun 28 Jul CET 2024
+
+#define kMaxHarts 8
+
+namespace Kernel
+{
+	using ThreadID = UInt32;
+
+	enum ThreadKind
+	{
+		kHartSystemReserved, // System reserved thread, well user can't use it
+		kHartStandard,		 // user thread, cannot be used by kernel
+		kHartFallback,		 // fallback thread, cannot be used by user if not clear or
+							 // used by kernel.
+		kHartBoot,			 // The core we booted from, the mama.
+		kInvalidHart,
+		kHartCount,
+	};
+
+	typedef enum ThreadKind SmThreadKind;
+	typedef ThreadID		SmThreadID;
+
+	///
+	/// \name HardwareThread
+	/// @brief CPU Hardware Thread (POWER, x64, or 64x0)
+	///
+
+	class HardwareThread final
+	{
+	public:
+		explicit HardwareThread();
+		~HardwareThread();
+
+	public:
+		NEWOS_COPY_DEFAULT(HardwareThread)
+
+	public:
+		operator bool();
+
+	public:
+		void Wake(const bool wakeup = false) noexcept;
+		void Busy(const bool busy = false) noexcept;
+
+	public:
+		bool Switch(HAL::StackFrame* stack);
+		bool IsWakeup() noexcept;
+
+	public:
+		HAL::StackFrame*  StackFrame() noexcept;
+		const ThreadKind& Kind() noexcept;
+		bool			  IsBusy() noexcept;
+		const ThreadID&	  ID() noexcept;
+
+	private:
+		HAL::StackFrame* fStack{nullptr};
+		ThreadKind		 fKind;
+		ThreadID		 fID;
+		ProcessID		 fSourcePID;
+		bool			 fWakeup;
+		bool			 fBusy;
+
+	private:
+		friend class MPManager;
+	};
+
+	///
+	/// \name MPManager
+	/// @brief Multi processor manager to manage other cores and dispatch tasks.
+	///
+
+	class MPManager final
+	{
+	private:
+		explicit MPManager();
+
+	public:
+		~MPManager();
+		NEWOS_COPY_DEFAULT(MPManager);
+
+	public:
+		bool			   Switch(HAL::StackFramePtr the);
+		HAL::StackFramePtr Leak() noexcept;
+
+	public:
+		Ref<HardwareThread*> operator[](const SizeT& idx);
+		bool				 operator!() noexcept;
+		operator bool() noexcept;
+
+	public:
+		/// @brief Shared instance of the SMP Manager.
+		/// @return the reference to the smp manager.
+		static Ref<MPManager> The();
+
+	public:
+		/// @brief Returns the amount of threads present in the system.
+		/// @returns SizeT the amount of cores present.
+		SizeT Count() noexcept;
+
+	private:
+		Array<HardwareThread, kMaxHarts> fThreadList;
+		ThreadID						 fCurrentThread{0};
+	};
+
+	/// @brief wakes up thread.
+	/// wakes up thread from hang.
+	Void rt_wakeup_thread(HAL::StackFramePtr stack);
+
+	/// @brief makes thread sleep.
+	/// hooks and hangs thread to prevent code from executing.
+	Void rt_hang_thread(HAL::StackFramePtr stack);
+} // namespace Kernel
+
+#endif // !__INC_MP_MANAGER_HPP__