Nikolay Sivov : uxtheme: Fix buffer bitmap leak.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: e015928a8583024ba4a69c3c9c37659c5226c10c
URL:    http://source.winehq.org/git/wine.git/?a=commit;h=e015928a8583024ba4a69c3c9c37659c5226c10c

Author: Nikolay Sivov <nsivov at codeweavers.com>
Date:   Fri Apr 21 23:09:16 2017 +0300

uxtheme: Fix buffer bitmap leak.

Signed-off-by: Nikolay Sivov <nsivov at codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard at winehq.org>

---

 dlls/uxtheme/buffer.c | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/dlls/uxtheme/buffer.c b/dlls/uxtheme/buffer.c
index 17bdf3d..bbf4c9d 100644
--- a/dlls/uxtheme/buffer.c
+++ b/dlls/uxtheme/buffer.c
@@ -39,12 +39,14 @@ struct paintbuffer
 {
     HDC targetdc;
     HDC memorydc;
+    HBITMAP bitmap;
     RECT rect;
     void *bits;
 };
 
 static void free_paintbuffer(struct paintbuffer *buffer)
 {
+    DeleteObject(buffer->bitmap);
     DeleteDC(buffer->memorydc);
     HeapFree(GetProcessHeap(), 0, buffer);
 }
@@ -83,7 +85,6 @@ HPAINTBUFFER WINAPI BeginBufferedPaint(HDC targetdc, const RECT *rect,
     char bmibuf[FIELD_OFFSET(BITMAPINFO, bmiColors[256])];
     BITMAPINFO *bmi = (BITMAPINFO *)bmibuf;
     struct paintbuffer *buffer;
-    HBITMAP hbm;
 
     TRACE("(%p %s %d %p %p)\n", targetdc, wine_dbgstr_rect(rect), format,
           params, retdc);
@@ -105,7 +106,7 @@ HPAINTBUFFER WINAPI BeginBufferedPaint(HDC targetdc, const RECT *rect,
     switch (format)
     {
     case BPBF_COMPATIBLEBITMAP:
-        hbm = CreateCompatibleBitmap(targetdc, rect->right - rect->left, rect->bottom - rect->top);
+        buffer->bitmap = CreateCompatibleBitmap(targetdc, rect->right - rect->left, rect->bottom - rect->top);
         buffer->bits = NULL;
         break;
     case BPBF_DIB:
@@ -120,22 +121,23 @@ HPAINTBUFFER WINAPI BeginBufferedPaint(HDC targetdc, const RECT *rect,
         bmi->bmiHeader.biBitCount = format == BPBF_TOPDOWNMONODIB ? 1 : 32;
         bmi->bmiHeader.biPlanes = 1;
         bmi->bmiHeader.biCompression = BI_RGB;
-        hbm = CreateDIBSection(buffer->memorydc, bmi, DIB_RGB_COLORS, &buffer->bits, NULL, 0);
+        buffer->bitmap = CreateDIBSection(buffer->memorydc, bmi, DIB_RGB_COLORS, &buffer->bits, NULL, 0);
         break;
     default:
         WARN("Unknown buffer format %d\n", format);
+        buffer->bitmap = NULL;
         free_paintbuffer(buffer);
         return NULL;
     }
 
-    if (!hbm)
+    if (!buffer->bitmap)
     {
         WARN("Failed to create buffer bitmap\n");
         free_paintbuffer(buffer);
         return NULL;
     }
 
-    DeleteObject(SelectObject(buffer->memorydc, hbm));
+    DeleteObject(SelectObject(buffer->memorydc, buffer->bitmap));
 
     *retdc = buffer->memorydc;