kernel32: Handle memory allocation failure in ReadConsoleA.

2451d263 · Dmitry Timoshkov · Alexandre Julliard · 00ff2ed8 · 2451d263
Commit 2451d263 authored Apr 15, 2014 by Dmitry Timoshkov Committed by Alexandre Julliard Apr 15, 2014
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 0 deletions

console.c dlls/kernel32/console.c +6 -0

No files found.
--- a/dlls/kernel32/console.c
+++ b/dlls/kernel32/console.c
@@ -1607,6 +1607,12 @@ BOOL WINAPI ReadConsoleA(HANDLE hConsoleInput, LPVOID lpBuffer, DWORD nNumberOfC
    DWORD	ncr = 0;
    BOOL	ret;

+    if (!ptr)
+    {
+        SetLastError(ERROR_NOT_ENOUGH_MEMORY);
+        return FALSE;
+    }
+
    if ((ret = ReadConsoleW(hConsoleInput, ptr, nNumberOfCharsToRead, &ncr, NULL)))
        ncr = WideCharToMultiByte(GetConsoleCP(), 0, ptr, ncr, lpBuffer, nNumberOfCharsToRead, NULL, NULL);