From bc9ea585bee075480b4f12f84c8ab99315766595 Mon Sep 17 00:00:00 2001
From: Samuel Williams <samuel.williams@shopify.com>
Date: Mon, 1 Dec 2025 17:49:31 +1300
Subject: Add `rb_ec_close` function to manage execution context cleanup.
 (#15253)

---
 cont.c    | 1 +
 vm.c      | 7 +++++++
 vm_core.h | 4 ++++
 3 files changed, 12 insertions(+)

diff --git a/cont.c b/cont.c
index c49256c977..50e8ffb349 100644
--- a/cont.c
+++ b/cont.c
@@ -2907,6 +2907,7 @@ void
 rb_fiber_close(rb_fiber_t *fiber)
 {
     fiber_status_set(fiber, FIBER_TERMINATED);
+    rb_ec_close(&fiber->cont.saved_ec);
 }
 
 static void
diff --git a/vm.c b/vm.c
index f9c615c3e2..1a8a6d059b 100644
--- a/vm.c
+++ b/vm.c
@@ -3890,6 +3890,13 @@ rb_ec_clear_vm_stack(rb_execution_context_t *ec)
     rb_ec_set_vm_stack(ec, NULL, 0);
 }
 
+void
+rb_ec_close(rb_execution_context_t *ec)
+{
+    // Fiber storage is not accessible from outside the running fiber, so it is safe to clear it here.
+    ec->storage = Qnil;
+}
+
 static void
 th_init(rb_thread_t *th, VALUE self, rb_vm_t *vm)
 {
diff --git a/vm_core.h b/vm_core.h
index 28d585deb2..9d11457966 100644
--- a/vm_core.h
+++ b/vm_core.h
@@ -1104,6 +1104,10 @@ void rb_ec_initialize_vm_stack(rb_execution_context_t *ec, VALUE *stack, size_t
 // @param ec the execution context to update.
 void rb_ec_clear_vm_stack(rb_execution_context_t *ec);
 
+// Close an execution context and free related resources that are no longer needed.
+// @param ec the execution context to close.
+void rb_ec_close(rb_execution_context_t *ec);
+
 struct rb_ext_config {
     bool ractor_safe;
 };
-- 
cgit v1.2.3