From 4566c80d7b8b9b9117128f6b11f7420aae530a86 Mon Sep 17 00:00:00 2001
From: "Yukihiro \"Matz\" Matsumoto" <matz@ruby-lang.org>
Date: Thu, 25 May 2017 12:36:38 +0900
Subject: Free given pointer if `realloc()` fails; fix #3658

---
 src/state.c | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

(limited to 'src')

diff --git a/src/state.c b/src/state.c
index 0be5a184a..a02e73f54 100644
--- 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 {
-- 
cgit v1.2.3