mirror of
git://source.winehq.org/git/wine.git
synced 2024-10-31 11:43:31 +00:00
cmd: Check the return value from HeapAlloc.
This commit is contained in:
parent
76e02581a9
commit
e6bce797d3
1 changed files with 16 additions and 0 deletions
|
@ -601,10 +601,21 @@ void WCMD_execute (WCHAR *command, WCHAR *redirects,
|
||||||
|
|
||||||
/* Move copy of the command onto the heap so it can be expanded */
|
/* Move copy of the command onto the heap so it can be expanded */
|
||||||
new_cmd = HeapAlloc( GetProcessHeap(), 0, MAXSTRING * sizeof(WCHAR));
|
new_cmd = HeapAlloc( GetProcessHeap(), 0, MAXSTRING * sizeof(WCHAR));
|
||||||
|
if (!new_cmd)
|
||||||
|
{
|
||||||
|
WINE_ERR("Could not allocate memory for new_cmd\n");
|
||||||
|
return;
|
||||||
|
}
|
||||||
strcpyW(new_cmd, command);
|
strcpyW(new_cmd, command);
|
||||||
|
|
||||||
/* Move copy of the redirects onto the heap so it can be expanded */
|
/* Move copy of the redirects onto the heap so it can be expanded */
|
||||||
new_redir = HeapAlloc( GetProcessHeap(), 0, MAXSTRING * sizeof(WCHAR));
|
new_redir = HeapAlloc( GetProcessHeap(), 0, MAXSTRING * sizeof(WCHAR));
|
||||||
|
if (!new_redir)
|
||||||
|
{
|
||||||
|
WINE_ERR("Could not allocate memory for new_redir\n");
|
||||||
|
HeapFree( GetProcessHeap(), 0, new_cmd );
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
/* If piped output, send stdout to the pipe by appending >filename to redirects */
|
/* If piped output, send stdout to the pipe by appending >filename to redirects */
|
||||||
if (piped) {
|
if (piped) {
|
||||||
|
@ -2086,6 +2097,11 @@ WCHAR *WCMD_ReadAndParseLine(WCHAR *optionalcmd, CMD_LIST **output, HANDLE readF
|
||||||
/* Allocate working space for a command read from keyboard, file etc */
|
/* Allocate working space for a command read from keyboard, file etc */
|
||||||
if (!extraSpace)
|
if (!extraSpace)
|
||||||
extraSpace = HeapAlloc(GetProcessHeap(), 0, (MAXSTRING+1) * sizeof(WCHAR));
|
extraSpace = HeapAlloc(GetProcessHeap(), 0, (MAXSTRING+1) * sizeof(WCHAR));
|
||||||
|
if (!extraSpace)
|
||||||
|
{
|
||||||
|
WINE_ERR("Could not allocate memory for extraSpace\n");
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
/* If initial command read in, use that, otherwise get input from handle */
|
/* If initial command read in, use that, otherwise get input from handle */
|
||||||
if (optionalcmd != NULL) {
|
if (optionalcmd != NULL) {
|
||||||
|
|
Loading…
Reference in a new issue