1 files changed, 95 insertions, 0 deletions
diff --git a/src/kernel/src/PageMgr.cc b/src/kernel/src/PageMgr.cc
new file mode 100644
index 00000000..81833ee2
--- /dev/null
+++ b/src/kernel/src/PageMgr.cc
@@ -0,0 +1,95 @@
+/* ========================================
+
+  Copyright (C) 2024-2025, Amlal El Mahrouss, licensed under the Apache 2.0 license.
+
+======================================== */
+
+#include <KernelKit/DebugOutput.h>
+#include <NeKit/PageMgr.h>
+
+#ifdef __NE_AMD64__
+#include <HALKit/AMD64/Paging.h>
+#elif defined(__NE_ARM64__)
+#include <HALKit/ARM64/Paging.h>
+#endif  // ifdef __NE_AMD64__ || defined(__NE_ARM64__)
+
+namespace Kernel {
+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() = default;
+
+/// @brief Flush virtual address.
+/// @param VirtAddr
+Void PageMgr::FlushTLB() {
+#ifdef __NE_VIRTUAL_MEMORY_SUPPORT__
+  hal_flush_tlb();
+#endif  // !__NE_VIRTUAL_MEMORY_SUPPORT__
+}
+
+/// @brief Reclaim freed page.
+/// @return
+Bool PTEWrapper::Reclaim() {
+  if (!this->fPresent) {
+    this->fPresent = true;
+    return true;
+  }
+
+  return false;
+}
+
+/// @brief Request a PTE.
+/// @param Rw Is it read write? or is it read only?
+/// @param User user mode?
+/// @param ExecDisable disable execution on page?
+/// @return
+PTEWrapper PageMgr::Request(Boolean Rw, Boolean User, Boolean ExecDisable, SizeT Sz, SizeT Pad) {
+  // Store PTE wrapper right after PTE.
+  VoidPtr ptr = Kernel::HAL::mm_alloc_bitmap(Rw, User, Sz, NO, Pad);
+
+  return PTEWrapper{Rw, User, ExecDisable, reinterpret_cast<UIntPtr>(ptr)};
+}
+
+/// @brief Disable BitMap.
+/// @param wrapper the wrapper.
+/// @return If the page bitmap was cleared or not.
+Bool PageMgr::Free(Ref<PTEWrapper>& wrapper) {
+  if (!Kernel::HAL::mm_free_bitmap((VoidPtr) wrapper.Leak().VirtualAddress())) return false;
+
+  return true;
+}
+
+/// @brief Virtual PTE address.
+/// @return The virtual address of the page.
+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) {
+  fExecDisable = enable;
+}
+
+Bool PTEWrapper::NoExecute() {
+  return fExecDisable;
+}
+}  // namespace Kernel