Change API

Author: mjp41

src/redirect/generate.cc

@@ -24,14 +24,16 @@ int main(int argc, char* argv[])
       auto rsize = snmalloc::sizeclass_to_size(sizeclass);
       if (rsize == size && align == 0)
       {
-        outfile << "DEFINE_MALLOC_SIZE(malloc_size_" << size << "_" << align << ", " << size
+        outfile << "DEFINE_MALLOC_SIZE(__stack_alloc_small_" << size << "_" << align << ", " << size
                 << ");" << std::endl;
       }
       else
       {
-        outfile << "REDIRECT_MALLOC_SIZE(malloc_size_" << size << "_" << align << ", malloc_size_"
+        outfile << "REDIRECT_MALLOC_SIZE(__stack_alloc_small_" << size << "_" << align << ", __stack_alloc_small_"
                 << rsize << "_" << 0 << ");" << std::endl;
       }
+      outfile << "GENERATE_FREE_SIZE(__stack_free_small_" << size << "_" << align << ");" << std::endl;
+
     }
   }
 

src/redirect/redirect.cc

@@ -20,4 +20,14 @@
     return snmalloc::ThreadAlloc::get_noncachable()->template alloc<s>(); \
   }
 
+extern "C" void free_local_small(void* ptr)
+{
+  if (snmalloc::Alloc::small_local_dealloc(ptr))
+    return;
+  snmalloc::ThreadAlloc::get_noncachable()->small_local_dealloc_slow(ptr);
+}
+
+#  define GENERATE_FREE_SIZE(a) \
+    __attribute__((alias("free_local_small"))) extern "C" void* a()
+
 #include "generated.cc"