export snmalloc::alloc_size to rust wrappers. (#676)

2024-09-19 16:54:56 +01:00
parent 8b95b9a916
commit d537d35268
1 changed files with 7 additions and 1 deletions
--- 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);
+}