In get_buffer_space() function, HeapReAlloc() will not allocate memory

if passed *buffer value is NULL.

In get_buffer_space() function, HeapReAlloc() will not allocate memory
f4a535d6 · Andrey Panov · Alexandre Julliard · b41466b3 · f4a535d6
Commit f4a535d6 authored May 06, 2004 by Andrey Panov Committed by Alexandre Julliard May 06, 2004
Hide whitespace changes
Inline Side-by-side

Showing with 8 additions and 2 deletions

message.c dlls/user/message.c +8 -2

No files found.
--- a/dlls/user/message.c
+++ b/dlls/user/message.c
@@ -254,8 +254,14 @@ inline static BOOL check_string( LPCWSTR str, size_t size )
 inline static void *get_buffer_space( void **buffer, size_t size )
 {
    void *ret;
-    if (!(ret = HeapReAlloc( GetProcessHeap(), 0, *buffer, size )))
-        HeapFree( GetProcessHeap(), 0, *buffer );
+
+    if (*buffer)
+    {
+        if (!(ret = HeapReAlloc( GetProcessHeap(), 0, *buffer, size )))
+            HeapFree( GetProcessHeap(), 0, *buffer );
+    }
+    else ret = HeapAlloc( GetProcessHeap(), 0, size );
+
    *buffer = ret;
    return ret;
 }