1
0
Fork 0
mirror of https://github.com/ruby/ruby.git synced 2022-11-09 12:17:21 -05:00
ruby--ruby/win32/stub.c
nobu ea9ff28f80 win32: realloc failures
* win32/file.c (code_page_i): handle realloc failure.
  reported by Denis Denisov <denji0k AT gmail.com>.
* win32/stub.c (stub_sysinit): ditto.

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@49023 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
2014-12-26 05:48:12 +00:00

47 lines
1 KiB
C

#include <ruby.h>
static void stub_sysinit(int *argc, char ***argv);
#define ruby_sysinit stub_sysinit
#include <main.c>
#undef ruby_sysinit
void
stub_sysinit(int *argc, char ***argv)
{
char exename[4096];
size_t lenexe, len0, lenall;
int i, ac;
char **av, *p;
lenexe = (size_t)GetModuleFileName(NULL, exename, sizeof exename);
ruby_sysinit(argc, argv);
ac = *argc;
av = *argv;
len0 = strlen(av[0]) + 1;
lenall = 0;
for (i = 1; i < ac; ++i) {
lenall += strlen(av[i]) + 1;
}
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);
memcpy(p, exename, lenexe);
p[lenexe] = '\0';
*av++ = p;
p += lenexe + 1;
memcpy(p, exename, lenexe);
p[lenexe] = '\0';
*av++ = p;
p += lenexe + 1;
while (--i) {
*av++ = p;
p += strlen(p) + 1;
}
*av = NULL;
}