Release `mrb->atexit_stack` on `mrb_close`.

4c1fb65d · take_cheeze · b1dd57ef · 4c1fb65d
Commit 4c1fb65d authored Jun 06, 2014 by take_cheeze
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 4 deletions

src/state.c src/state.c +6 -4

No files found.
--- a/src/state.c
+++ b/src/state.c
@@ -221,10 +221,12 @@ mrb_free_context(mrb_state *mrb, struct mrb_context *c)
 void
 mrb_close(mrb_state *mrb)
 {
-  mrb_int i;
-
-  for (i = mrb->atexit_stack_len; i > 0; --i) {
-    mrb->atexit_stack[i - 1](mrb);
+  if (mrb->atexit_stack_len > 0) {
+    mrb_int i;
+    for (i = mrb->atexit_stack_len; i > 0; --i) {
+      mrb->atexit_stack[i - 1](mrb);
+    }
+    mrb_free(mrb, mrb->atexit_stack);
  }

  /* free */