VM (new pixeltower): 原始内存分配器

Author: MetsukiMio

include/openminecraft/mem/om_mem_prealloc.hpp

@@ -0,0 +1,31 @@
+#ifndef OM_MEM_PREALLOC_HPP
+#define OM_MEM_PREALLOC_HPP
+
+#include "openminecraft/log/om_log_common.hpp"
+#include <cstdint>
+namespace openminecraft::mem
+{
+class OMHeap
+{
+  public:
+    OMHeap(uint64_t minSize, uint64_t maxSize);
+    ~OMHeap();
+
+    void init();
+    void expand(void *target);
+    void shrink(void *target);
+
+    void *block;
+
+  private:
+    log::OMLogger logger;
+    uint64_t minSize;
+    uint64_t maxSize;
+
+    void activate(void *p, uint64_t length);
+    void deactivate(void *p, uint64_t length);
+    void *heapTop;
+};
+} // namespace openminecraft::mem
+
+#endif

plat/unix/om_mem_plat.cpp

@@ -1,10 +1,59 @@
+#include "openminecraft/log/om_log_common.hpp"
+#include "openminecraft/mem/om_mem_prealloc.hpp"
 #include "openminecraft/mem/om_mem_record.hpp"
+#include <cerrno>
+#include <cstring>
+#include <new>
 #if defined(OM_PLATFORM_IOS) || defined(OM_PLATFORM_MACOS)
 #include <malloc/malloc.h>
 #else
 #include <malloc.h>
 #endif
 
+#include <sys/mman.h>
+
+namespace openminecraft::mem
+{
+OMHeap::OMHeap(uint64_t minSize, uint64_t maxSize) : logger("OMHeap", this), maxSize(maxSize), minSize(minSize)
+{
+    block = mmap(nullptr, maxSize, PROT_NONE, MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);
+    if (block == nullptr)
+    {
+        logger.error("[unix-like] unable to map memory ({})", strerror(errno));
+        throw std::bad_alloc();
+    }
+}
+
+OMHeap::~OMHeap()
+{
+    munmap(block, maxSize);
+}
+
+void OMHeap::activate(void *p, uint64_t length)
+{
+    if (mprotect(p, length, PROT_READ | PROT_WRITE) == -1)
+    {
+        logger.error("[unix-like] mprotect fail ({})", strerror(errno));
+        throw std::bad_alloc();
+    }
+}
+
+void OMHeap::deactivate(void *p, uint64_t length)
+{
+    if (madvise(p, length, MADV_DONTNEED) == -1)
+    {
+        logger.error("[unix-like] madvise fail ({})", strerror(errno));
+        throw std::bad_alloc();
+    }
+
+    if (mprotect(p, length, PROT_NONE) == -1)
+    {
+        logger.error("[unix-like] mprotect fail ({})", strerror(errno));
+        throw std::bad_alloc();
+    }
+}
+} // namespace openminecraft::mem
+
 namespace openminecraft::mem::castorice
 {
 size_t heapSize(void *p)

plat/windows/om_mem_plat.cpp

@@ -1,6 +1,26 @@
+#include "openminecraft/mem/om_mem_prealloc.hpp"
 #include "openminecraft/mem/om_mem_record.hpp"
 #include <malloc.h>
 
+namespace openminecraft::mem
+{
+OMHeap::OMHeap(uint64_t minSize, uint64_t maxSize) : logger("OMHeap", this), maxSize(maxSize), minSize(minSize)
+{
+}
+
+OMHeap::~OMHeap()
+{
+}
+
+void OMHeap::activate(void *p, uint64_t length)
+{
+}
+
+void OMHeap::deactivate(void *p, uint64_t length)
+{
+}
+} // namespace openminecraft::mem
+
 namespace openminecraft::mem::castorice
 {
 size_t heapSize(void *p)

src/boot/entrypoint.cpp

@@ -28,6 +28,7 @@
 #include "openminecraft/log/om_log_common.hpp"
 #include "openminecraft/log/om_log_threadname.hpp"
 #include "openminecraft/mem/om_mem_allocator.hpp"
+#include "openminecraft/mem/om_mem_prealloc.hpp"
 #include "openminecraft/mem/om_mem_record.hpp"
 #include "openminecraft/renderer/om_renderer_layer.hpp"
 #include "openminecraft/renderer/vk/om_renderer_layer_vk.hpp"
@@ -116,8 +117,9 @@ int boot(std::vector<std::string> args)
 
     SDL_Quit();
 
-    std::shared_ptr<OMClassFileParser> parser;
     bool recovermode = false;
+    mem::OMHeap heap(0ul, 8ull * 1024 * 1024 * 1024);
+    heap.init();
 
     if (setjmp(recoverBuffer) == 0)
     {
@@ -148,6 +150,17 @@ int boot(std::vector<std::string> args)
                 commandBuffer.clear();
                 mem::castorice::printres();
                 break;
+            case "memtest"_hash: {
+                heap.expand(static_cast<uint8_t *>(heap.block) + 255);
+                auto i = static_cast<int *>(heap.block);
+                i[0] = 33550336;
+                logger->debug("heap write : {}", i[0]);
+                heap.shrink(heap.block);
+
+                commandBuffer.clear();
+
+                break;
+            }
             case "crash"_hash: {
                 logger->info("{}", *((int *)nullptr));
             }

src/mem/om_mem_prealloc.cpp

@@ -0,0 +1,34 @@
+#include "openminecraft/mem/om_mem_prealloc.hpp"
+
+namespace openminecraft::mem
+{
+void OMHeap::init()
+{
+    heapTop = block;
+    if (minSize == 0)
+    {
+        return;
+    }
+    expand(static_cast<uint8_t *>(block) + minSize);
+}
+void OMHeap::expand(void *target)
+{
+    if (target < block || target <= heapTop)
+    {
+        logger.error("target is not valid!");
+    }
+
+    activate(heapTop, (size_t)target - (size_t)heapTop);
+    heapTop = target;
+}
+void OMHeap::shrink(void *target)
+{
+    if (target < block || target >= heapTop)
+    {
+        logger.error("target is not valid!");
+    }
+
+    deactivate(target, (size_t)heapTop - (size_t)target);
+    heapTop = target;
+}
+} // namespace openminecraft::mem