ruby_xfree: reject memory allocated by ruby_mimalloc

1 files changed, 7 insertions, 1 deletions

diff --git a/vm.c b/vm.c
index 2cae6779d9..f2e2ba1b89 100644
--- a/vm.c
+++ b/vm.c
@@ -3821,7 +3821,12 @@ thread_free(void *ptr)
     else {
         // ruby_xfree(th->nt);
         // TODO: MN system collect nt, but without MN system it should be freed here.
-        ruby_xfree(th);
+        if (th->main_thread) {
+            ruby_mimfree(th);
+        }
+        else {
+            ruby_xfree(th);
+        }
     }
 
     RUBY_FREE_LEAVE("thread");
@@ -4594,6 +4599,7 @@ Init_BareVM(void)
     vm->global_hooks.type = hook_list_type_global;
 
     // setup main thread
+    th->main_thread = 1;
     th->nt = ZALLOC(struct rb_native_thread);
     th->vm = vm;
     th->ractor = vm->ractor.main_ractor = rb_ractor_main_alloc();