Free given pointer if realloc() fails; fix #3658

mruby · May 25, 2017 · 4566c80 · 4566c80
1 parent ff46957
commit 4566c80
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/state.c b/src/state.c
@@ -52,13 +52,13 @@ mrb_open_core(mrb_allocf f, void *ud)
 void*
 mrb_default_allocf(mrb_state *mrb, void *p, size_t size, void *ud)
 {
-  if (size == 0) {
+  void *p2;
+
+  if (size == 0 || (p2 = realloc(p, size)) == NULL) {
     free(p);
     return NULL;
   }
-  else {
-    return realloc(p, size);
-  }
+  return p2;
 }
 
 struct alloca_header {