1 files changed, 85 insertions, 0 deletions
diff --git a/Kernel/Source/Pmm.cxx b/Kernel/Source/Pmm.cxx
new file mode 100644
index 00000000..f60f5b23
--- /dev/null
+++ b/Kernel/Source/Pmm.cxx
@@ -0,0 +1,85 @@
+/* -------------------------------------------
+
+    Copyright SoftwareLabs
+
+------------------------------------------- */
+
+#include <KernelKit/DebugOutput.hpp>
+#include <NewKit/Pmm.hpp>
+
+namespace NewOS
+{
+	Pmm::Pmm()
+		: fPageManager()
+	{
+		kcout << "[PMM] Allocate PageMemoryManager";
+	}
+
+	Pmm::~Pmm() = default;
+
+	/* If this returns Null pointer, enter emergency mode */
+	Ref<PTEWrapper> Pmm::RequestPage(Boolean user, Boolean readWrite)
+	{
+		PTEWrapper pt = fPageManager.Leak().Request(user, readWrite, false, kPTESize);
+
+		if (pt.fPresent)
+		{
+			kcout << "[PMM]: Allocation was successful.";
+			return Ref<PTEWrapper>(pt);
+		}
+
+		kcout << "[PMM]: Allocation failure.";
+
+		return {};
+	}
+
+	Boolean Pmm::FreePage(Ref<PTEWrapper> PageRef)
+	{
+		if (!PageRef)
+			return false;
+
+		PageRef.Leak().fPresent = false;
+
+		return true;
+	}
+
+	Boolean Pmm::TogglePresent(Ref<PTEWrapper> PageRef, Boolean Enable)
+	{
+		if (!PageRef)
+			return false;
+
+		PageRef.Leak().fPresent = Enable;
+
+		return true;
+	}
+
+	Boolean Pmm::ToggleUser(Ref<PTEWrapper> PageRef, Boolean Enable)
+	{
+		if (!PageRef)
+			return false;
+
+		PageRef.Leak().fRw = Enable;
+
+		return true;
+	}
+
+	Boolean Pmm::ToggleRw(Ref<PTEWrapper> PageRef, Boolean Enable)
+	{
+		if (!PageRef)
+			return false;
+
+		PageRef.Leak().fRw = Enable;
+
+		return true;
+	}
+
+	Boolean Pmm::ToggleShare(Ref<PTEWrapper> PageRef, Boolean Enable)
+	{
+		if (!PageRef)
+			return false;
+
+		PageRef.Leak().fShareable = Enable;
+
+		return true;
+	}
+} // namespace NewOS