Add missing files.2

Author: mjp41

src/redirect/generate.cc

@@ -0,0 +1,35 @@
+#include "snmalloc.h"
+
+#include <fstream>
+#include <iostream>
+
+int main(int argc, char* argv[])
+{
+  if (argc != 2)
+  {
+    std::cerr << "Call with output file name" << std::endl;
+    return 1;
+  }
+
+  // open a file in write mode.
+  ofstream outfile;
+  outfile.open(argv[1]);
+
+  for (size_t size = 1024; size > 0; size -= 16)
+  {
+    auto sizeclass = snmalloc::size_to_sizeclass(size);
+    auto rsize = snmalloc::sizeclass_to_size(sizeclass);
+    if (rsize == size)
+    {
+      outfile << "DEFINE_MALLOC_SIZE(malloc_size_" << size << ", " << size
+              << ");" << std::endl;
+    }
+    else
+    {
+      outfile << "REDIRECT_MALLOC_SIZE(malloc_size_" << size << ", malloc_size_"
+              << rsize << ");" << std::endl;
+    }
+  }
+
+  outfile.close();
+}

src/redirect/redirect.cc

@@ -0,0 +1,23 @@
+
+#include "snmalloc.h"
+
+#define NAME(a) malloc_size_##a
+#define STRINGIFY(a) a
+#define NAME_STRING(a) NAME(a)
+
+#ifdef WIN32
+#  define REDIRECT_MALLOC_SIZE(a, b) \
+    extern "C" void* NAME(a)(); \
+    __pragma(comment(linker, "/alternatename:malloc_size_##a=malloc_size_##b"))
+#else
+#  define REDIRECT_MALLOC_SIZE(a, b) \
+    __attribute__((alias(#b))) extern "C" void* a()
+#endif
+
+#define DEFINE_MALLOC_SIZE(a, s) \
+  extern "C" void* a() \
+  { \
+    return snmalloc::ThreadAlloc::get_noncachable()->template alloc<s>(); \
+  }
+
+#include "generated.cc"