mjit.c: prevent memory leak on realloc failure

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@64050 b2dd03c8-39d4-4d8f-98ff-823fe69b080e

1 files changed, 5 insertions, 2 deletions

diff --git a/mjit.c b/mjit.c
index eca995742b..3d5b74b849 100644
--- a/mjit.c
+++ b/mjit.c
@@ -327,15 +327,18 @@ form_args(int num, ...)
     va_list argp;
     size_t len, n;
     int i;
-    char **args, **res;
+    char **args, **res, **tmp;
 
     va_start(argp, num);
     res = NULL;
     for (i = len = 0; i < num; i++) {
         args = va_arg(argp, char **);
         n = args_len(args);
-        if ((res = (char **)realloc(res, sizeof(char *) * (len + n + 1))) == NULL)
+        if ((tmp = (char **)realloc(res, sizeof(char *) * (len + n + 1))) == NULL) {
+            free(res);
             return NULL;
+        }
+        res = tmp;
         MEMCPY(res + len, args, char *, n + 1);
         len += n;
     }