Move `MRB_TRY` part of `mrb_close` to `src/error.c`; ref 6cc52b3

We don't want to increase number of files that should be compiled by C++
compiler when `enable_cxx_exception` is turned on.

2 files changed, 26 insertions, 18 deletions

diff --git a/src/error.c b/src/error.c
index d2f1c27ac..256f9169d 100644
--- a/src/error.c
+++ b/src/error.c
@@ -591,6 +591,28 @@ mrb_core_init_abort(mrb_state *mrb)
   exc_throw(mrb, mrb_nil_value());
 }
 
+void
+mrb_protect_atexit(mrb_state *mrb)
+{
+  if (mrb->atexit_stack_len > 0) {
+    struct mrb_jmpbuf *prev_jmp = mrb->jmp;
+    struct mrb_jmpbuf c_jmp;
+    for (int i = mrb->atexit_stack_len; i > 0; --i) {
+      MRB_TRY(&c_jmp) {
+        mrb->jmp = &c_jmp;
+        mrb->atexit_stack[i - 1](mrb);
+        mrb->jmp = prev_jmp;
+      } MRB_CATCH(&c_jmp) {
+        /* ignore atexit errors */
+      } MRB_END_EXC(&c_jmp);
+    }
+#ifndef MRB_FIXED_STATE_ATEXIT_STACK
+    mrb_free(mrb, mrb->atexit_stack);
+#endif
+    mrb->jmp = prev_jmp;
+  }
+}
+
 mrb_noreturn void
 mrb_raise_nomemory(mrb_state *mrb)
 {
diff --git a/src/state.c b/src/state.c
index 95c070deb..83cbf1629 100644
--- a/src/state.c
+++ b/src/state.c
@@ -12,7 +12,6 @@
 #include <mruby/debug.h>
 #include <mruby/string.h>
 #include <mruby/class.h>
-#include <mruby/throw.h>
 
 void mrb_init_core(mrb_state*);
 void mrb_init_mrbgems(mrb_state*);
@@ -177,26 +176,13 @@ mrb_free_context(mrb_state *mrb, struct mrb_context *c)
   mrb_free(mrb, c);
 }
 
-MRB_API void
+int mrb_protect_atexit(mrb_state *mrb);
+
+  MRB_API void
 mrb_close(mrb_state *mrb)
 {
   if (!mrb) return;
-  if (mrb->atexit_stack_len > 0) {
-    struct mrb_jmpbuf *prev_jmp = mrb->jmp;
-    struct mrb_jmpbuf c_jmp;
-    for (int i = mrb->atexit_stack_len; i > 0; --i) {
-      MRB_TRY(&c_jmp) {
-        mrb->jmp = &c_jmp;
-        mrb->atexit_stack[i - 1](mrb);
-        mrb->jmp = prev_jmp;
-      } MRB_CATCH(&c_jmp) {
-        /* ignore atexit errors */
-      } MRB_END_EXC(&c_jmp);
-    }
-#ifndef MRB_FIXED_STATE_ATEXIT_STACK
-    mrb_free(mrb, mrb->atexit_stack);
-#endif
-  }
+  mrb_protect_atexit(mrb);
 
   /* free */
   mrb_gc_destroy(mrb, &mrb->gc);