check `mrb_open` return value

8df63b4d · cremno · 835da83c · 8df63b4d
Commit 8df63b4d authored Jul 09, 2014 by cremno
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

mrbgems/mruby-bin-strip/tools/mruby-strip/mruby-strip.c mrbgems/mruby-bin-strip/tools/mruby-strip/mruby-strip.c +4 -0

No files found.
--- a/mrbgems/mruby-bin-strip/tools/mruby-strip/mruby-strip.c
+++ b/mrbgems/mruby-bin-strip/tools/mruby-strip/mruby-strip.c
@@ -96,6 +96,10 @@ main(int argc, char **argv)
  }

  mrb = mrb_open();
+  if (mrb == NULL) {
+    fputs("Invalid mrb_state, exiting mruby-strip\n", stderr);
+    return EXIT_FAILURE;
+  }

  ireps = (mrb_irep**)malloc(sizeof(mrb_irep*) * argc);
  for (i = args_result; i < argc; ++i) {