https://git.reactos.org/?p=reactos.git;a=commitdiff;h=88285670868e84b1459342f735ff7772e7d1bd4f
commit 88285670868e84b1459342f735ff7772e7d1bd4f
Author: Eric Kohl <eric.k...@reactos.org>
AuthorDate: Wed May 23 00:24:01 2018 +0200
Commit: Eric Kohl <eric.k...@reactos.org>
CommitDate: Wed May 23 00:24:53 2018 +0200
[MSGIINA] Make sure we do not leak bitmaps
---
dll/win32/msgina/gui.c | 143 +++++++++++++++++++++++++++-----------------
dll/win32/msgina/msgina.h | 3 -
dll/win32/msgina/shutdown.c | 9 +--
3 files changed, 92 insertions(+), 63 deletions(-)
diff --git a/dll/win32/msgina/gui.c b/dll/win32/msgina/gui.c
index 906998743d..83547425db 100644
--- a/dll/win32/msgina/gui.c
+++ b/dll/win32/msgina/gui.c
@@ -27,6 +27,11 @@ typedef struct _LEGALNOTICEDATA
LPWSTR pszText;
} LEGALNOTICEDATA, *PLEGALNOTICEDATA;
+typedef struct _DLG_DATA
+{
+ PGINA_CONTEXT pgContext;
+ HBITMAP hBitmap;
+} DLG_DATA, *PDLG_DATA;
static BOOL
GUIInitialize(
@@ -193,41 +198,48 @@ GUIRemoveStatusMessage(
}
static INT_PTR CALLBACK
-EmptyWindowProc(
+NoticeWindowProc(
IN HWND hwndDlg,
IN UINT uMsg,
IN WPARAM wParam,
IN LPARAM lParam)
{
- PGINA_CONTEXT pgContext;
-
- pgContext = (PGINA_CONTEXT)GetWindowLongPtrW(hwndDlg, GWLP_USERDATA);
+ PDLG_DATA pDlgData;
+
+ pDlgData = (PDLG_DATA)GetWindowLongPtrW(hwndDlg, GWLP_USERDATA);
switch (uMsg)
{
case WM_INITDIALOG:
{
- pgContext = (PGINA_CONTEXT)lParam;
- SetWindowLongPtrW(hwndDlg, GWLP_USERDATA, (LONG_PTR)pgContext);
+ pDlgData = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY,
sizeof(DLG_DATA));
+ if (pDlgData == NULL)
+ return FALSE;
- /* Draw the logo bitmap */
- pgContext->hBitmap = LoadImageW(pgContext->hDllInstance,
MAKEINTRESOURCEW(IDI_ROSLOGO), IMAGE_BITMAP, 0, 0, LR_DEFAULTCOLOR);
+ SetWindowLongPtrW(hwndDlg, GWLP_USERDATA, (LONG_PTR)pDlgData);
+
+ pDlgData->pgContext = (PGINA_CONTEXT)lParam;
+
+ /* Load the logo bitmap */
+ pDlgData->hBitmap = LoadImageW(pDlgData->pgContext->hDllInstance,
MAKEINTRESOURCEW(IDI_ROSLOGO), IMAGE_BITMAP, 0, 0, LR_DEFAULTCOLOR);
return TRUE;
}
+
case WM_PAINT:
{
PAINTSTRUCT ps;
- if (pgContext->hBitmap)
+ if (pDlgData->hBitmap)
{
BeginPaint(hwndDlg, &ps);
- DrawStateW(ps.hdc, NULL, NULL, (LPARAM)pgContext->hBitmap,
(WPARAM)0, 0, 0, 0, 0, DST_BITMAP);
+ DrawStateW(ps.hdc, NULL, NULL, (LPARAM)pDlgData->hBitmap,
(WPARAM)0, 0, 0, 0, 0, DST_BITMAP);
EndPaint(hwndDlg, &ps);
}
return TRUE;
}
case WM_DESTROY:
{
- DeleteObject(pgContext->hBitmap);
+ DeleteObject(pDlgData->hBitmap);
+ HeapFree(GetProcessHeap(), 0, pDlgData);
return TRUE;
}
}
@@ -245,7 +257,7 @@ GUIDisplaySASNotice(
pgContext->hDllInstance,
MAKEINTRESOURCEW(IDD_NOTICE_DLG),
GetDesktopWindow(),
- EmptyWindowProc,
+ NoticeWindowProc,
(LPARAM)pgContext);
}
@@ -943,39 +955,44 @@ LoggedOutWindowProc(
IN WPARAM wParam,
IN LPARAM lParam)
{
- PGINA_CONTEXT pgContext;
+ PDLG_DATA pDlgData;
- pgContext = (PGINA_CONTEXT)GetWindowLongPtrW(hwndDlg, GWLP_USERDATA);
+ pDlgData = (PDLG_DATA)GetWindowLongPtrW(hwndDlg, GWLP_USERDATA);
switch (uMsg)
{
case WM_INITDIALOG:
{
+ pDlgData = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY,
sizeof(DLG_DATA));
+ if (pDlgData == NULL)
+ return FALSE;
+
+ SetWindowLongPtrW(hwndDlg, GWLP_USERDATA, (LONG_PTR)pDlgData);
+
/* FIXME: take care of NoDomainUI */
- pgContext = (PGINA_CONTEXT)lParam;
- SetWindowLongPtrW(hwndDlg, GWLP_USERDATA, (LONG_PTR)pgContext);
+ pDlgData->pgContext = (PGINA_CONTEXT)lParam;
- if (pgContext->bAutoAdminLogon ||
- !pgContext->bDontDisplayLastUserName)
- SetDlgItemTextW(hwndDlg, IDC_USERNAME, pgContext->UserName);
+ if (pDlgData->pgContext->bAutoAdminLogon ||
+ !pDlgData->pgContext->bDontDisplayLastUserName)
+ SetDlgItemTextW(hwndDlg, IDC_USERNAME,
pDlgData->pgContext->UserName);
- if (pgContext->bAutoAdminLogon)
- SetDlgItemTextW(hwndDlg, IDC_PASSWORD, pgContext->Password);
+ if (pDlgData->pgContext->bAutoAdminLogon)
+ SetDlgItemTextW(hwndDlg, IDC_PASSWORD,
pDlgData->pgContext->Password);
- SetDomainComboBox(GetDlgItem(hwndDlg, IDC_LOGON_TO), pgContext);
+ SetDomainComboBox(GetDlgItem(hwndDlg, IDC_LOGON_TO),
pDlgData->pgContext);
- if (pgContext->bDisableCAD)
+ if (pDlgData->pgContext->bDisableCAD)
EnableWindow(GetDlgItem(hwndDlg, IDCANCEL), FALSE);
- if (!pgContext->bShutdownWithoutLogon)
+ if (!pDlgData->pgContext->bShutdownWithoutLogon)
EnableWindow(GetDlgItem(hwndDlg, IDC_SHUTDOWN), FALSE);
- SetFocus(GetDlgItem(hwndDlg, pgContext->bDontDisplayLastUserName ?
IDC_USERNAME : IDC_PASSWORD));
+ SetFocus(GetDlgItem(hwndDlg,
pDlgData->pgContext->bDontDisplayLastUserName ? IDC_USERNAME : IDC_PASSWORD));
/* Draw the logo bitmap */
- pgContext->hBitmap = LoadImageW(pgContext->hDllInstance,
MAKEINTRESOURCEW(IDI_ROSLOGO), IMAGE_BITMAP, 0, 0, LR_DEFAULTCOLOR);
+ pDlgData->hBitmap = LoadImageW(pDlgData->pgContext->hDllInstance,
MAKEINTRESOURCEW(IDI_ROSLOGO), IMAGE_BITMAP, 0, 0, LR_DEFAULTCOLOR);
- if (pgContext->bAutoAdminLogon)
+ if (pDlgData->pgContext->bAutoAdminLogon)
PostMessage(GetDlgItem(hwndDlg, IDOK), BM_CLICK, 0, 0);
return TRUE;
@@ -984,24 +1001,25 @@ LoggedOutWindowProc(
case WM_PAINT:
{
PAINTSTRUCT ps;
- if (pgContext->hBitmap)
+ if (pDlgData->hBitmap)
{
BeginPaint(hwndDlg, &ps);
- DrawStateW(ps.hdc, NULL, NULL, (LPARAM)pgContext->hBitmap,
(WPARAM)0, 0, 0, 0, 0, DST_BITMAP);
+ DrawStateW(ps.hdc, NULL, NULL, (LPARAM)pDlgData->hBitmap,
(WPARAM)0, 0, 0, 0, 0, DST_BITMAP);
EndPaint(hwndDlg, &ps);
}
return TRUE;
}
case WM_DESTROY:
- DeleteObject(pgContext->hBitmap);
+ DeleteObject(pDlgData->hBitmap);
+ HeapFree(GetProcessHeap(), 0, pDlgData);
return TRUE;
case WM_COMMAND:
switch (LOWORD(wParam))
{
case IDOK:
- if (DoLogon(hwndDlg, pgContext))
+ if (DoLogon(hwndDlg, pDlgData->pgContext))
EndDialog(hwndDlg, WLX_SAS_ACTION_LOGON);
return TRUE;
@@ -1010,8 +1028,8 @@ LoggedOutWindowProc(
return TRUE;
case IDC_SHUTDOWN:
- if (OnShutDown(hwndDlg, pgContext) == IDOK)
- EndDialog(hwndDlg, pgContext->nShutdownAction);
+ if (OnShutDown(hwndDlg, pDlgData->pgContext) == IDOK)
+ EndDialog(hwndDlg,
pDlgData->pgContext->nShutdownAction);
return TRUE;
}
break;
@@ -1214,51 +1232,57 @@ UnlockWindowProc(
IN WPARAM wParam,
IN LPARAM lParam)
{
- PGINA_CONTEXT pgContext;
+ PDLG_DATA pDlgData;
INT result = WLX_SAS_ACTION_NONE;
- pgContext = (PGINA_CONTEXT)GetWindowLongPtrW(hwndDlg, GWLP_USERDATA);
+ pDlgData = (PDLG_DATA)GetWindowLongPtrW(hwndDlg, GWLP_USERDATA);
switch (uMsg)
{
case WM_INITDIALOG:
{
- pgContext = (PGINA_CONTEXT)lParam;
- SetWindowLongPtrW(hwndDlg, GWLP_USERDATA, (LONG_PTR)pgContext);
+ pDlgData = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY,
sizeof(DLG_DATA));
+ if (pDlgData == NULL)
+ return FALSE;
+
+ SetWindowLongPtrW(hwndDlg, GWLP_USERDATA, (LONG_PTR)pDlgData);
- SetLockMessage(hwndDlg, IDC_LOCKMSG, pgContext);
+ pDlgData->pgContext = (PGINA_CONTEXT)lParam;
- SetDlgItemTextW(hwndDlg, IDC_USERNAME, pgContext->UserName);
+ SetLockMessage(hwndDlg, IDC_LOCKMSG, pDlgData->pgContext);
+
+ SetDlgItemTextW(hwndDlg, IDC_USERNAME,
pDlgData->pgContext->UserName);
SetFocus(GetDlgItem(hwndDlg, IDC_PASSWORD));
- if (pgContext->bDisableCAD)
+ if (pDlgData->pgContext->bDisableCAD)
EnableWindow(GetDlgItem(hwndDlg, IDCANCEL), FALSE);
- /* Draw the logo bitmap */
- pgContext->hBitmap = LoadImageW(pgContext->hDllInstance,
MAKEINTRESOURCEW(IDI_ROSLOGO), IMAGE_BITMAP, 0, 0, LR_DEFAULTCOLOR);
+ /* Load the logo bitmap */
+ pDlgData->hBitmap = LoadImageW(pDlgData->pgContext->hDllInstance,
MAKEINTRESOURCEW(IDI_ROSLOGO), IMAGE_BITMAP, 0, 0, LR_DEFAULTCOLOR);
return TRUE;
}
case WM_PAINT:
{
PAINTSTRUCT ps;
- if (pgContext->hBitmap)
+ if (pDlgData->hBitmap)
{
BeginPaint(hwndDlg, &ps);
- DrawStateW(ps.hdc, NULL, NULL, (LPARAM)pgContext->hBitmap,
(WPARAM)0, 0, 0, 0, 0, DST_BITMAP);
+ DrawStateW(ps.hdc, NULL, NULL, (LPARAM)pDlgData->hBitmap,
(WPARAM)0, 0, 0, 0, 0, DST_BITMAP);
EndPaint(hwndDlg, &ps);
}
return TRUE;
}
case WM_DESTROY:
- DeleteObject(pgContext->hBitmap);
+ DeleteObject(pDlgData->hBitmap);
+ HeapFree(GetProcessHeap(), 0, pDlgData);
return TRUE;
case WM_COMMAND:
switch (LOWORD(wParam))
{
case IDOK:
- if (DoUnlock(hwndDlg, pgContext, &result))
+ if (DoUnlock(hwndDlg, pDlgData->pgContext, &result))
EndDialog(hwndDlg, result);
return TRUE;
@@ -1307,36 +1331,43 @@ LockedWindowProc(
IN WPARAM wParam,
IN LPARAM lParam)
{
- PGINA_CONTEXT pgContext;
+ PDLG_DATA pDlgData;
- pgContext = (PGINA_CONTEXT)GetWindowLongPtrW(hwndDlg, GWLP_USERDATA);
+ pDlgData = (PDLG_DATA)GetWindowLongPtrW(hwndDlg, GWLP_USERDATA);
switch (uMsg)
{
case WM_INITDIALOG:
{
- pgContext = (PGINA_CONTEXT)lParam;
- SetWindowLongPtrW(hwndDlg, GWLP_USERDATA, (LONG_PTR)pgContext);
+ pDlgData = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY,
sizeof(DLG_DATA));
+ if (pDlgData == NULL)
+ return FALSE;
- /* Draw the logo bitmap */
- pgContext->hBitmap = LoadImageW(pgContext->hDllInstance,
MAKEINTRESOURCEW(IDI_ROSLOGO), IMAGE_BITMAP, 0, 0, LR_DEFAULTCOLOR);
- SetLockMessage(hwndDlg, IDC_LOCKMSG, pgContext);
+ SetWindowLongPtrW(hwndDlg, GWLP_USERDATA, (LONG_PTR)pDlgData);
+
+ pDlgData->pgContext = (PGINA_CONTEXT)lParam;
+
+ /* Load the logo bitmap */
+ pDlgData->hBitmap = LoadImageW(pDlgData->pgContext->hDllInstance,
MAKEINTRESOURCEW(IDI_ROSLOGO), IMAGE_BITMAP, 0, 0, LR_DEFAULTCOLOR);
+
+ SetLockMessage(hwndDlg, IDC_LOCKMSG, pDlgData->pgContext);
return TRUE;
}
case WM_PAINT:
{
PAINTSTRUCT ps;
- if (pgContext->hBitmap)
+ if (pDlgData->hBitmap)
{
BeginPaint(hwndDlg, &ps);
- DrawStateW(ps.hdc, NULL, NULL, (LPARAM)pgContext->hBitmap,
(WPARAM)0, 0, 0, 0, 0, DST_BITMAP);
+ DrawStateW(ps.hdc, NULL, NULL, (LPARAM)pDlgData->hBitmap,
(WPARAM)0, 0, 0, 0, 0, DST_BITMAP);
EndPaint(hwndDlg, &ps);
}
return TRUE;
}
case WM_DESTROY:
{
- DeleteObject(pgContext->hBitmap);
+ DeleteObject(pDlgData->hBitmap);
+ HeapFree(GetProcessHeap(), 0, pDlgData);
return TRUE;
}
}
diff --git a/dll/win32/msgina/msgina.h b/dll/win32/msgina/msgina.h
index e2b6b5bfb4..b644aa36ff 100644
--- a/dll/win32/msgina/msgina.h
+++ b/dll/win32/msgina/msgina.h
@@ -56,9 +56,6 @@ typedef struct
PDWORD pdwOptions;
PWLX_MPR_NOTIFY_INFO pMprNotifyInfo;
PVOID *pProfile;
-
- /* Current logo to display */
- HBITMAP hBitmap;
} GINA_CONTEXT, *PGINA_CONTEXT;
extern HINSTANCE hDllInstance;
diff --git a/dll/win32/msgina/shutdown.c b/dll/win32/msgina/shutdown.c
index 1351eabc14..e6f18e302f 100644
--- a/dll/win32/msgina/shutdown.c
+++ b/dll/win32/msgina/shutdown.c
@@ -24,6 +24,7 @@
typedef struct _SHUTDOWN_DLG_CONTEXT
{
PGINA_CONTEXT pgContext;
+ HBITMAP hBitmap;
DWORD ShutdownOptions;
BOOL bCloseDlg;
BOOL bReasonUI;
@@ -419,13 +420,13 @@ ShutdownDialogProc(
ShutdownOnInit(hDlg, pContext);
/* Draw the logo bitmap */
- pContext->pgContext->hBitmap =
+ pContext->hBitmap =
LoadImageW(pContext->pgContext->hDllInstance,
MAKEINTRESOURCEW(IDI_ROSLOGO), IMAGE_BITMAP, 0, 0, LR_DEFAULTCOLOR);
return TRUE;
}
case WM_DESTROY:
- DeleteObject(pContext->pgContext->hBitmap);
+ DeleteObject(pContext->hBitmap);
return TRUE;
case WM_ACTIVATE:
@@ -449,10 +450,10 @@ ShutdownDialogProc(
case WM_PAINT:
{
PAINTSTRUCT ps;
- if (pContext->pgContext->hBitmap)
+ if (pContext->hBitmap)
{
BeginPaint(hDlg, &ps);
- DrawStateW(ps.hdc, NULL, NULL,
(LPARAM)pContext->pgContext->hBitmap, (WPARAM)0, 0, 0, 0, 0, DST_BITMAP);
+ DrawStateW(ps.hdc, NULL, NULL, (LPARAM)pContext->hBitmap,
(WPARAM)0, 0, 0, 0, 0, DST_BITMAP);
EndPaint(hDlg, &ps);
}
return TRUE;
