Fix potential leak and crash

hdlx · Aug 30, 2024 · 9d95787 · 9d95787
1 parent fdd3be5
commit 9d95787
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/Sources/App.c b/Sources/App.c
@@ -1366,8 +1366,13 @@ LRESULT CALLBACK WindowProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
             x += iconContainerSize;
         }
         BitBlt(ps.hdc, clientRect.left, clientRect.top, clientRect.right - clientRect.left, clientRect.bottom - clientRect.top, pGraphRes->_DC, 0, 0, SRCCOPY);
+
+        // Always restore old bitmap (see fn doc)
+        SelectObject(pGraphRes->_DC, oldBitmap);
+
         GdipDeleteGraphics(pGraphics);
         EndPaint(hwnd, &ps);
+
         return 0;
     }
     case WM_ERASEBKGND: