export snmalloc::alloc_size to rust wrappers.

microsoft · Sep 15, 2024 · 380e338 · 380e338
1 parent 8b95b9a
commit 380e338
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/snmalloc/override/rust.cc b/src/snmalloc/override/rust.cc
@@ -50,4 +50,10 @@ extern "C" SNMALLOC_EXPORT void SNMALLOC_NAME_MANGLE(rust_statistics)(
 {
   *current_memory_usage = Alloc::Config::Backend::get_current_usage();
   *peak_memory_usage = Alloc::Config::Backend::get_peak_usage();
-}
+}
+
+extern "C" SNMALLOC_EXPORT size_t SNMALLOC_NAME_MANGLE(rust_usable_size)(
+  const void *ptr)
+{
+  return ThreadAlloc::get().alloc_size(ptr);
+}