From ea9ff28f80165ec054106702ee38f5995bc4a491 Mon Sep 17 00:00:00 2001 From: nobu Date: Fri, 26 Dec 2014 05:48:12 +0000 Subject: win32: realloc failures * win32/file.c (code_page_i): handle realloc failure. reported by Denis Denisov . * win32/stub.c (stub_sysinit): ditto. git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@49023 b2dd03c8-39d4-4d8f-98ff-823fe69b080e --- win32/file.c | 7 +++++-- win32/stub.c | 7 ++++++- 2 files changed, 11 insertions(+), 3 deletions(-) (limited to 'win32') diff --git a/win32/file.c b/win32/file.c index a50d6d8638..e2b7f93aa4 100644 --- a/win32/file.c +++ b/win32/file.c @@ -160,8 +160,11 @@ code_page_i(st_data_t name, st_data_t idx, st_data_t arg) USHORT *table = cp->table; if (count <= idx) { unsigned int i = count; - cp->count = count = (((idx + 4) & ~31) | 28); - cp->table = table = realloc(table, count * sizeof(*table)); + count = (((idx + 4) & ~31) | 28); + table = realloc(table, count * sizeof(*table)); + if (!table) return ST_CONTINUE; + cp->count = count; + cp->table = table; while (i < count) table[i++] = INVALID_CODE_PAGE; } table[idx] = (USHORT)code_page; diff --git a/win32/stub.c b/win32/stub.c index fc82d73e2c..3960024fc2 100644 --- a/win32/stub.c +++ b/win32/stub.c @@ -21,7 +21,12 @@ stub_sysinit(int *argc, char ***argv) for (i = 1; i < ac; ++i) { lenall += strlen(av[i]) + 1; } - *argv = av = realloc(av, lenall + (lenexe + 1) * 2 + sizeof(char *) * (i + 2)); + av = realloc(av, lenall + (lenexe + 1) * 2 + sizeof(char *) * (i + 2)); + if (!av) { + perror("realloc command line"); + exit(-1); + } + *argv = av; *argc = ++ac; p = (char *)(av + i + 2); memmove(p + (lenexe + 1) * 2, (char *)(av + ac) + len0, lenall); -- cgit v1.2.3