1 files changed, 22 insertions, 0 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)
 {