1 files changed, 118 insertions, 0 deletions
diff --git a/NewKernel/Source/PageManager.cxx b/NewKernel/Source/PageManager.cxx
new file mode 100644
index 00000000..7225d4ad
--- /dev/null
+++ b/NewKernel/Source/PageManager.cxx
@@ -0,0 +1,118 @@
+/* -------------------------------------------
+
+    Copyright Mahrouss Logic
+
+------------------------------------------- */
+
+#include <KernelKit/DebugOutput.hpp>
+#include <NewKit/PageManager.hpp>
+
+#ifdef __x86_64__
+#include <HALKit/AMD64/HalPageAlloc.hpp>
+#endif // ifdef __x86_64__
+
+//! null deref will throw (Page Zero detected, aborting app!)
+#define kProtectedRegionEnd (512)
+
+namespace NewOS
+{
+	PTEWrapper::PTEWrapper(Boolean Rw, Boolean User, Boolean ExecDisable, UIntPtr VirtAddr)
+		: fRw(Rw),
+		  fUser(User),
+		  fExecDisable(ExecDisable),
+		  fVirtAddr(VirtAddr),
+		  fCache(false),
+		  fShareable(false),
+		  fWt(false),
+		  fPresent(true),
+		  fAccessed(false)
+	{
+	}
+
+	PTEWrapper::~PTEWrapper()
+	{
+	}
+
+	/// @brief Flush virtual address.
+	/// @param VirtAddr
+	void PageManager::FlushTLB(UIntPtr VirtAddr)
+	{
+		if (VirtAddr == kBadAddress)
+			return;
+
+		hal_flush_tlb();
+	}
+
+	/// @brief Reclaim freed page.
+	/// @return
+	bool PTEWrapper::Reclaim()
+	{
+		if (!this->fPresent)
+		{
+			this->fPresent = true;
+			return true;
+		}
+
+		return false;
+	}
+
+	/// @brief Request a PTE.
+	/// @param Rw r/w?
+	/// @param User user mode?
+	/// @param ExecDisable disable execution on page?
+	/// @return
+	PTEWrapper PageManager::Request(Boolean Rw, Boolean User, Boolean ExecDisable, SizeT Sz)
+	{
+		// Store PTE wrapper right after PTE.
+		VoidPtr ptr = NewOS::HAL::hal_alloc_page(Rw, User, Sz);
+
+		return PTEWrapper{Rw, User, ExecDisable, reinterpret_cast<UIntPtr>(ptr)};
+	}
+
+	/// @brief Disable PTE.
+	/// @param wrapper the wrapper.
+	/// @return
+	bool PageManager::Free(Ref<PTEWrapper*>& wrapper)
+	{
+		if (wrapper)
+		{
+			if (!Detail::page_disable(wrapper->VirtualAddress()))
+				return false;
+			return true;
+		}
+
+		return false;
+	}
+
+	/// @brief Virtual PTE address.
+	/// @return The virtual address of the page.
+	const UIntPtr PTEWrapper::VirtualAddress()
+	{
+		return (fVirtAddr);
+	}
+
+	bool PTEWrapper::Shareable()
+	{
+		return fShareable;
+	}
+
+	bool PTEWrapper::Present()
+	{
+		return fPresent;
+	}
+
+	bool PTEWrapper::Access()
+	{
+		return fAccessed;
+	}
+
+	void PTEWrapper::NoExecute(const bool enable)
+	{
+		this->fExecDisable = enable;
+	}
+
+	const bool& PTEWrapper::NoExecute()
+	{
+		return this->fExecDisable;
+	}
+} // namespace NewOS