diff --git a/cont.c b/cont.c index e2f01c4..314340e 100644 --- a/cont.c +++ b/cont.c @@ -330,6 +330,17 @@ fiber_memsize(const void *ptr) return size; } +VALUE +rb_obj_is_fiber(VALUE obj) +{ + if (rb_typeddata_is_kind_of(obj, &fiber_data_type)) { + return Qtrue; + } + else { + return Qfalse; + } +} + static void cont_save_machine_stack(rb_thread_t *th, rb_context_t *cont) { @@ -1347,6 +1358,19 @@ rb_fiber_yield(int argc, VALUE *argv) return rb_fiber_transfer(return_fiber(), argc, argv); } +void +rb_fiber_reset_root_local_storage(VALUE thval) +{ + rb_thread_t *th; + rb_fiber_t *fib; + + GetThreadPtr(thval, th); + if (th->root_fiber && th->root_fiber != th->fiber) { + GetFiberPtr(th->root_fiber, fib); + th->local_storage = fib->cont.saved_thread.local_storage; + } +} + /* * call-seq: * fiber.alive? -> true or false diff --git a/gc.c b/gc.c index 2c49f96..422b766 100644 --- a/gc.c +++ b/gc.c @@ -3248,7 +3248,8 @@ rb_objspace_call_finalizer(rb_objspace_t *objspace) while (p < pend) { if (BUILTIN_TYPE(p) == T_DATA && DATA_PTR(p) && RANY(p)->as.data.dfree && - !rb_obj_is_thread((VALUE)p) && !rb_obj_is_mutex((VALUE)p) ) { + !rb_obj_is_thread((VALUE)p) && !rb_obj_is_mutex((VALUE)p) && + !rb_obj_is_fiber((VALUE)p)) { p->as.free.flags = 0; if (RTYPEDDATA_P(p)) { RDATA(p)->dfree = RANY(p)->as.typeddata.type->function.dfree; diff --git a/internal.h b/internal.h index 4c476ec..78875af 100644 --- a/internal.h +++ b/internal.h @@ -62,6 +62,10 @@ int rb_parse_in_eval(void); int rb_parse_in_main(void); VALUE rb_insns_name_array(void); +/* cont.c */ +VALUE rb_obj_is_fiber(VALUE); +void rb_fiber_reset_root_local_storage(VALUE); + /* debug.c */ PRINTF_ARGS(void ruby_debug_printf(const char*, ...), 1, 2); diff --git a/vm.c b/vm.c index 8621709..1d343ce 100644 --- a/vm.c +++ b/vm.c @@ -1611,6 +1611,7 @@ ruby_vm_destruct(rb_vm_t *vm) rb_gc_force_recycle(vm->self); vm->main_thread = 0; if (th) { + rb_fiber_reset_root_local_storage(th->self); thread_free(th); } if (vm->living_threads) {