Skip to content

Commit

Permalink
* win32/win32.c (rb_w32_spawn, rb_w32_aspawn): get rid of too huge
Browse files Browse the repository at this point in the history
  alloca(). this is the real fix of [ruby-core:34833].


git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@30664 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
  • Loading branch information
unak committed Jan 27, 2011
1 parent fa57e46 commit 98929a2
Show file tree
Hide file tree
Showing 2 changed files with 23 additions and 10 deletions.
5 changes: 5 additions & 0 deletions ChangeLog
Original file line number Diff line number Diff line change
@@ -1,3 +1,8 @@
Thu Jan 27 15:11:52 2011 NAKAMURA Usaku <[email protected]>

* win32/win32.c (rb_w32_spawn, rb_w32_aspawn): get rid of too huge
alloca(). this is the real fix of [ruby-core:34833].

Thu Jan 27 12:46:25 2011 Nobuyoshi Nakada <[email protected]>

* process.c (ALLOC_ARGV_WITH_STR): fix void pointer arithmetic.
Expand Down
28 changes: 18 additions & 10 deletions win32/win32.c
Original file line number Diff line number Diff line change
Expand Up @@ -1113,6 +1113,7 @@ rb_w32_spawn(int mode, const char *cmd, const char *prog)
const char *shell = NULL;
WCHAR *wcmd, *wshell;
rb_pid_t ret;
VALUE tmp = Qnil;

if (check_spawn_mode(mode)) return -1;

Expand All @@ -1130,18 +1131,19 @@ rb_w32_spawn(int mode, const char *cmd, const char *prog)
int nt;
while (ISSPACE(*cmd)) cmd++;
if ((shell = getenv("RUBYSHELL")) && (redir = has_redirection(cmd))) {
char *tmp = ALLOCA_N(char, strlen(shell) + strlen(cmd) + sizeof(" -c ") + 2);
sprintf(tmp, "%s -c \"%s\"", shell, cmd);
cmd = tmp;
ALLOCV(tmp, strlen(shell) + strlen(cmd) + sizeof(" -c ") + 2);
sprintf(RSTRING_PTR(tmp), "%s -c \"%s\"", shell, cmd);
cmd = RSTRING_PTR(tmp);
}
else if ((shell = getenv("COMSPEC")) &&
(nt = !is_command_com(shell),
(redir < 0 ? has_redirection(cmd) : redir) ||
is_internal_cmd(cmd, nt))) {
char *tmp = ALLOCA_N(char, strlen(shell) + strlen(cmd) + sizeof(" /c ")
+ (nt ? 2 : 0));
sprintf(tmp, nt ? "%s /c \"%s\"" : "%s /c %s", shell, cmd);
cmd = tmp;
ALLOCV(tmp,
strlen(shell) + strlen(cmd) + sizeof(" /c ") + (nt ? 2 : 0));
sprintf(RSTRING_PTR(tmp),
nt ? "%s /c \"%s\"" : "%s /c %s", shell, cmd);
cmd = RSTRING_PTR(tmp);
}
else {
int len = 0, quote = (*cmd == '"') ? '"' : (*cmd == '\'') ? '\'' : 0;
Expand Down Expand Up @@ -1182,7 +1184,8 @@ rb_w32_spawn(int mode, const char *cmd, const char *prog)
if (p) translate_char(p, '/', '\\');
if (is_batch(shell)) {
int alen = strlen(prog);
cmd = p = ALLOCA_N(char, len + alen + (quote ? 2 : 0) + 1);
ALLOCV(tmp, len + alen + (quote ? 2 : 0) + 1);
cmd = p = RSTRING_PTR(tmp);
if (quote) *p++ = '"';
memcpy(p, shell, len);
p += len;
Expand All @@ -1196,6 +1199,7 @@ rb_w32_spawn(int mode, const char *cmd, const char *prog)

/* assume ACP */
wcmd = cmd ? acp_to_wstr(cmd, NULL) : NULL;
if (!NIL_P(tmp)) ALLOCV_END(tmp);
wshell = shell ? acp_to_wstr(shell, NULL) : NULL;

ret = child_result(CreateChild(wcmd, wshell, NULL, NULL, NULL, NULL), mode);
Expand All @@ -1214,6 +1218,7 @@ rb_w32_aspawn(int mode, const char *prog, char *const *argv)
char *cmd, fbuf[MAXPATHLEN];
WCHAR *wcmd, *wprog;
rb_pid_t ret;
VALUE tmp = Qnil;

if (check_spawn_mode(mode)) return -1;

Expand Down Expand Up @@ -1246,20 +1251,23 @@ rb_w32_aspawn(int mode, const char *prog, char *const *argv)
if (c_switch) len += 3;
else ++argv;
if (argv[0]) len += join_argv(NULL, argv, ntcmd);
cmd = ALLOCA_N(char, len);
ALLOCV(tmp, len);
cmd = RSTRING_PTR(tmp);
join_argv(cmd, progs, ntcmd);
if (c_switch) strlcat(cmd, " /c", len);
if (argv[0]) join_argv(cmd + strlcat(cmd, " ", len), argv, ntcmd);
prog = c_switch ? shell : 0;
}
else {
len = join_argv(NULL, argv, FALSE);
cmd = ALLOCA_N(char, len);
ALLOCV(tmp, len);
cmd = RSTRING_PTR(tmp);
join_argv(cmd, argv, FALSE);
}

/* assume ACP */
wcmd = cmd ? acp_to_wstr(cmd, NULL) : NULL;
if (!NIL_P(tmp)) ALLOCV_END(tmp);
wprog = prog ? acp_to_wstr(prog, NULL) : NULL;

ret = child_result(CreateChild(wcmd, wprog, NULL, NULL, NULL, NULL), mode);
Expand Down

0 comments on commit 98929a2

Please sign in to comment.