1 files changed, 94 insertions, 0 deletions

diff --git a/kernel/kapi/memory.cpp b/kernel/kapi/memory.cpp
new file mode 100644
index 0000000..7c9b1da
--- /dev/null
+++ b/kernel/kapi/memory.cpp
@@ -0,0 +1,94 @@
+#include "kapi/memory.hpp"
+
+#include "kapi/system.hpp"
+
+#include <cstddef>
+#include <optional>
+#include <utility>
+
+namespace kapi::memory
+{
+
+  namespace
+  {
+    struct bad_frame_allocator final : public frame_allocator
+    {
+      bad_frame_allocator static instance;
+
+      auto allocate_many(std::size_t) noexcept -> std::optional<std::pair<frame, std::size_t>> override
+      {
+        system::panic("Tried to allocate frames without an active allocator.");
+      }
+
+      auto release_many(std::pair<frame, std::size_t>) -> void override
+      {
+        system::panic("Tried to release frames without an active allocator.");
+      }
+    };
+
+    struct bad_page_mapper final : public page_mapper
+    {
+      bad_page_mapper static instance;
+
+      auto map(page, frame, flags) -> std::byte * override
+      {
+        system::panic("Tried to map a page without an active mapper.");
+      }
+
+      auto unmap(page) -> void override
+      {
+        system::panic("Tried to unmap a page without an active mapper.");
+      }
+
+      auto try_unmap(page) noexcept -> bool override
+      {
+        return false;
+      }
+    };
+
+    constinit bad_frame_allocator bad_frame_allocator::instance{};
+    constinit bad_page_mapper bad_page_mapper::instance{};
+  }  // namespace
+
+  constinit auto static active_frame_allocator = static_cast<frame_allocator *>(&bad_frame_allocator::instance);
+  constinit auto static active_page_mapper = static_cast<page_mapper *>(&bad_page_mapper::instance);
+
+  auto set_frame_allocator(frame_allocator & allocator) -> std::optional<frame_allocator *>
+  {
+    if (&allocator == active_frame_allocator)
+    {
+      return {};
+    }
+    return std::exchange(active_frame_allocator, &allocator);
+  }
+
+  auto set_page_mapper(page_mapper & mapper) -> std::optional<page_mapper *>
+  {
+    if (&mapper == active_page_mapper)
+    {
+      return {};
+    }
+    return std::exchange(active_page_mapper, &mapper);
+  }
+
+  auto allocate_frame() -> std::optional<frame>
+  {
+    return active_frame_allocator->allocate();
+  }
+
+  auto allocate_many_frames(std::size_t count) -> std::optional<std::pair<frame, std::size_t>>
+  {
+    return active_frame_allocator->allocate_many(count);
+  }
+
+  auto map(page page, frame frame) -> std::byte *
+  {
+    return active_page_mapper->map(page, frame, page_mapper::flags::empty);
+  }
+
+  auto unmap(page page) -> void
+  {
+    return active_page_mapper->unmap(page);
+  }
+
+}  // namespace kapi::memory
\ No newline at end of file