From 6816e8efcff3be75f8020cd1b0ea57d3cd664bbc Mon Sep 17 00:00:00 2001 From: Adam Hess Date: Thu, 12 Oct 2023 11:15:53 -0700 Subject: Free everything at shutdown when the RUBY_FREE_ON_SHUTDOWN environment variable is set, manually free memory at shutdown. Co-authored-by: Nobuyoshi Nakada Co-authored-by: Peter Zhu --- variable.c | 27 +++++++++++++++++++++++++++ 1 file changed, 27 insertions(+) (limited to 'variable.c') diff --git a/variable.c b/variable.c index 4c2bc5c97e..de3a3561f8 100644 --- a/variable.c +++ b/variable.c @@ -438,6 +438,33 @@ struct rb_global_entry { bool ractor_local; }; +static enum rb_id_table_iterator_result +free_global_entry_i(ID key, VALUE val, void *arg) +{ + struct rb_global_entry *entry = (struct rb_global_entry *)val; + if (entry->var->counter == 1) { + ruby_xfree(entry->var); + } + else { + entry->var->counter--; + } + ruby_xfree(entry); + return ID_TABLE_DELETE; +} + +void +rb_free_rb_global_tbl(void) +{ + rb_id_table_foreach(rb_global_tbl, free_global_entry_i, 0); + rb_id_table_free(rb_global_tbl); +} + +void +rb_free_generic_iv_tbl_(void) +{ + st_free_table(generic_iv_tbl_); +} + static struct rb_global_entry* rb_find_global_entry(ID id) { -- cgit v1.2.3