user.exe16: Fix memory leak in case when HeapReAlloc fails.

Signed-off-by: Dmitry Timoshkov <dmitry@baikal.ru> Signed-off-by: Alexandre Julliard <julliard@winehq.org>

user.exe16: Fix memory leak in case when HeapReAlloc fails.
8c3ed829 · Dmitry Timoshkov · Alexandre Julliard · f7fd2c2b · 8c3ed829
Commit 8c3ed829 authored Jun 22, 2017 by Dmitry Timoshkov Committed by Alexandre Julliard Jun 22, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

user.c dlls/user.exe16/user.c +4 -1

No files found.
--- a/dlls/user.exe16/user.c
+++ b/dlls/user.exe16/user.c
@@ -3245,8 +3245,11 @@ DWORD WINAPI FormatMessage16(

                        /* CMF - This makes a BIG assumption about va_list */
                        while ((ret = vsnprintf(b, sz, fmtstr, (va_list) argliststart)) < 0 || ret >= sz) {
+                            LPSTR new_b;
                            sz = (ret == -1 ? sz + 100 : ret + 1);
-                            b = HeapReAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, b, sz);
+                            new_b = HeapReAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, b, sz);
+                            if (!new_b) break;
+                            b = new_b;
                        }
                        for (x=b; *x; x++) ADD_TO_T(*x);
                        HeapFree(GetProcessHeap(), 0, b);