Protect exceptions within main() function

mrbgems/mruby-bin-mruby/tools/mruby/mruby.c

@@ -6,6 +6,7 @@
 #include <mruby/compile.h>
 #include <mruby/dump.h>
 #include <mruby/variable.h>
+#include <mruby/throw.h>
 
 #ifdef MRB_DISABLE_STDIO
 static void
@@ -176,6 +177,8 @@ main(int argc, char **argv)
   mrbc_context *c;
   mrb_value v;
   mrb_sym zero_sym;
+  struct mrb_jmpbuf c_jmp;
+  int ai;
 
   if (mrb == NULL) {
     fputs("Invalid mrb_state, exiting mruby\n", stderr);
@@ -189,6 +192,9 @@ main(int argc, char **argv)
     return n;
   }
 
+  ai = mrb_gc_arena_save(mrb);
+  MRB_TRY(&c_jmp) {
+    mrb->jmp = &c_jmp;
     ARGV = mrb_ary_new_capa(mrb, args.argc);
     for (i = 0; i < args.argc; i++) {
       char* utf8 = mrb_utf8_from_locale(args.argv[i], -1);
@@ -232,6 +238,7 @@ main(int argc, char **argv)
       mrb_utf8_free(utf8);
     }
 
+    mrb_gc_arena_restore(mrb, ai);
     mrbc_context_free(mrb, c);
     if (mrb->exc) {
       if (!mrb_undef_p(v)) {
@@ -242,6 +249,10 @@ main(int argc, char **argv)
     else if (args.check_syntax) {
       printf("Syntax OK\n");
     }
+  }
+  MRB_CATCH(&c_jmp) {           /* error */
+  }
+  MRB_END_EXC(&c_jmp);
   cleanup(mrb, &args);
 
   return n == 0 ? EXIT_SUCCESS : EXIT_FAILURE;