1 files changed, 61 insertions, 0 deletions
diff --git a/LauncherInjector/memalloc.cpp b/LauncherInjector/memalloc.cpp
new file mode 100644
index 00000000..64bc7b76
--- /dev/null
+++ b/LauncherInjector/memalloc.cpp
@@ -0,0 +1,61 @@
+#define WIN32_LEAN_AND_MEAN
+#include <Windows.h>
+#include "memalloc.h"
+
+HMODULE hTier0Module;
+IMemAlloc** g_ppMemAllocSingleton;
+
+void LoadTier0Handle()
+{
+    hTier0Module = GetModuleHandleA("tier0.dll");
+    if (!hTier0Module) return;
+
+    g_ppMemAllocSingleton = (IMemAlloc**)GetProcAddress(hTier0Module, "g_pMemAllocSingleton");
+}
+
+const int STATIC_ALLOC_SIZE = 16384;
+
+size_t g_iStaticAllocated = 0;
+char pStaticAllocBuf[STATIC_ALLOC_SIZE];
+
+// they should never be used here, except in LibraryLoadError
+
+void* malloc(size_t n)
+{
+    // allocate into static buffer
+    if (g_iStaticAllocated + n <= STATIC_ALLOC_SIZE)
+    {
+        void* ret = pStaticAllocBuf + g_iStaticAllocated;
+        g_iStaticAllocated += n;
+        return ret;
+    }
+    else
+    {
+        // try to fallback to g_pMemAllocSingleton
+        if (!hTier0Module) LoadTier0Handle();
+        if (g_ppMemAllocSingleton && *g_ppMemAllocSingleton)
+            return (*g_ppMemAllocSingleton)->m_vtable->Alloc(*g_ppMemAllocSingleton, n);
+        else
+            throw "Cannot allocate";
+    }
+}
+
+void free(void* p)
+{
+    // if it was allocated into the static buffer, just do nothing, safest way to deal with it
+    if (p >= pStaticAllocBuf && p <= pStaticAllocBuf + STATIC_ALLOC_SIZE)
+        return;
+
+    if (g_ppMemAllocSingleton && *g_ppMemAllocSingleton)
+        (*g_ppMemAllocSingleton)->m_vtable->Free(*g_ppMemAllocSingleton, p);
+}
+
+void* operator new(size_t n)
+{
+    return malloc(n);
+}
+
+void operator delete(void* p)
+{
+    free(p);
+}