xref: /reactos/dll/win32/msgina/gui.c (revision cc439606) 
1 /*
2  * PROJECT:         ReactOS msgina.dll
3  * FILE:            dll/win32/msgina/gui.c
4  * PURPOSE:         ReactOS Logon GINA DLL
5  * PROGRAMMER:      Herv� Poussineau (hpoussin@reactos.org)
6  */
7 
8 #include "msgina.h"
9 
10 #include <wingdi.h>
11 #include <winnls.h>
12 #include <winreg.h>
13 
14 typedef struct _DISPLAYSTATUSMSG
15 {
16     PGINA_CONTEXT Context;
17     HDESK hDesktop;
18     DWORD dwOptions;
19     PWSTR pTitle;
20     PWSTR pMessage;
21     HANDLE StartupEvent;
22 } DISPLAYSTATUSMSG, *PDISPLAYSTATUSMSG;
23 
24 typedef struct _LEGALNOTICEDATA
25 {
26     LPWSTR pszCaption;
27     LPWSTR pszText;
28 } LEGALNOTICEDATA, *PLEGALNOTICEDATA;
29 
30 typedef struct _DLG_DATA
31 {
32     PGINA_CONTEXT pgContext;
33     HBITMAP hBitmap;
34 } DLG_DATA, *PDLG_DATA;
35 
36 static BOOL
37 GUIInitialize(
38     IN OUT PGINA_CONTEXT pgContext)
39 {
40     TRACE("GUIInitialize(%p)\n", pgContext);
41     return TRUE;
42 }
43 
44 
45 static
46 VOID
47 SetWelcomeText(HWND hWnd)
48 {
49     PWCHAR pBuffer = NULL, p;
50     HKEY hKey;
51     DWORD BufSize, dwType, dwWelcomeSize, dwTitleLength;
52     LONG rc;
53 
54     TRACE("SetWelcomeText(%p)\n", hWnd);
55 
56     /* Open the Winlogon key */
57     rc = RegOpenKeyExW(HKEY_LOCAL_MACHINE,
58                        L"SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion\\Winlogon",
59                        0,
60                        KEY_QUERY_VALUE,
61                        &hKey);
62     if (rc != ERROR_SUCCESS)
63     {
64         WARN("RegOpenKeyExW() failed with error %lu\n", rc);
65         return;
66     }
67 
68     /* Get the size of the Welcome value */
69     dwWelcomeSize = 0;
70     rc = RegQueryValueExW(hKey,
71                           L"Welcome",
72                           NULL,
73                           &dwType,
74                           NULL,
75                           &dwWelcomeSize);
76     if (rc == ERROR_FILE_NOT_FOUND || dwWelcomeSize == 0 || dwType != REG_SZ)
77         goto done;
78 
79     dwTitleLength = GetWindowTextLengthW(hWnd);
80     BufSize = dwWelcomeSize + ((dwTitleLength + 1) * sizeof(WCHAR));
81 
82     pBuffer = HeapAlloc(GetProcessHeap(), 0, BufSize);
83     if (pBuffer == NULL)
84         goto done;
85 
86     GetWindowTextW(hWnd, pBuffer, BufSize / sizeof(WCHAR));
87     wcscat(pBuffer, L" ");
88     p = &pBuffer[dwTitleLength + 1];
89 
90     RegQueryValueExW(hKey,
91                      L"Welcome",
92                      NULL,
93                      &dwType,
94                      (PBYTE)p,
95                      &dwWelcomeSize);
96 
97     SetWindowText(hWnd, pBuffer);
98 
99 done:
100     if (pBuffer != NULL)
101         HeapFree(GetProcessHeap(), 0, pBuffer);
102 
103     RegCloseKey(hKey);
104 }
105 
106 
107 static INT_PTR CALLBACK
108 StatusDialogProc(
109     IN HWND hwndDlg,
110     IN UINT uMsg,
111     IN WPARAM wParam,
112     IN LPARAM lParam)
113 {
114     UNREFERENCED_PARAMETER(wParam);
115 
116     switch (uMsg)
117     {
118         case WM_INITDIALOG:
119         {
120             PDISPLAYSTATUSMSG msg = (PDISPLAYSTATUSMSG)lParam;
121             if (!msg)
122                 return FALSE;
123 
124             msg->Context->hStatusWindow = hwndDlg;
125 
126             if (msg->pTitle)
127                 SetWindowTextW(hwndDlg, msg->pTitle);
128             SetDlgItemTextW(hwndDlg, IDC_STATUS_MESSAGE, msg->pMessage);
129             SetEvent(msg->StartupEvent);
130             return TRUE;
131         }
132     }
133     return FALSE;
134 }
135 
136 static DWORD WINAPI
137 StartupWindowThread(LPVOID lpParam)
138 {
139     HDESK hDesk;
140     PDISPLAYSTATUSMSG msg = (PDISPLAYSTATUSMSG)lpParam;
141 
142     /* When SetThreadDesktop is called the system closes the desktop handle when needed
143        so we have to create a new handle because this handle may still be in use by winlogon  */
144     if (!DuplicateHandle (  GetCurrentProcess(),
145                             msg->hDesktop,
146                             GetCurrentProcess(),
147                             (HANDLE*)&hDesk,
148                             0,
149                             FALSE,
150                             DUPLICATE_SAME_ACCESS))
151     {
152         ERR("Duplicating handle failed!\n");
153         HeapFree(GetProcessHeap(), 0, lpParam);
154         return FALSE;
155     }
156 
157     if(!SetThreadDesktop(hDesk))
158     {
159         ERR("Setting thread desktop failed!\n");
160         HeapFree(GetProcessHeap(), 0, lpParam);
161         return FALSE;
162     }
163 
164     DialogBoxParamW(
165         hDllInstance,
166         MAKEINTRESOURCEW(IDD_STATUS),
167         GetDesktopWindow(),
168         StatusDialogProc,
169         (LPARAM)lpParam);
170 
171     HeapFree(GetProcessHeap(), 0, lpParam);
172     return TRUE;
173 }
174 
175 static BOOL
176 GUIDisplayStatusMessage(
177     IN PGINA_CONTEXT pgContext,
178     IN HDESK hDesktop,
179     IN DWORD dwOptions,
180     IN PWSTR pTitle,
181     IN PWSTR pMessage)
182 {
183     PDISPLAYSTATUSMSG msg;
184     HANDLE Thread;
185     DWORD ThreadId;
186 
187     TRACE("GUIDisplayStatusMessage(%ws)\n", pMessage);
188 
189     if (!pgContext->hStatusWindow)
190     {
191         /*
192          * If everything goes correctly, 'msg' is freed
193          * by the 'StartupWindowThread' thread.
194          */
195         msg = (PDISPLAYSTATUSMSG)HeapAlloc(GetProcessHeap(),
196                                            HEAP_ZERO_MEMORY,
197                                            sizeof(*msg));
198         if(!msg)
199             return FALSE;
200 
201         msg->Context = pgContext;
202         msg->dwOptions = dwOptions;
203         msg->pTitle = pTitle;
204         msg->pMessage = pMessage;
205         msg->hDesktop = hDesktop;
206 
207         msg->StartupEvent = CreateEventW(NULL, TRUE, FALSE, NULL);
208 
209         if (!msg->StartupEvent)
210         {
211             HeapFree(GetProcessHeap(), 0, msg);
212             return FALSE;
213         }
214 
215         Thread = CreateThread(NULL,
216                               0,
217                               StartupWindowThread,
218                               (PVOID)msg,
219                               0,
220                               &ThreadId);
221         if (Thread)
222         {
223             /* 'msg' will be freed by 'StartupWindowThread' */
224 
225             CloseHandle(Thread);
226             WaitForSingleObject(msg->StartupEvent, INFINITE);
227             CloseHandle(msg->StartupEvent);
228             return TRUE;
229         }
230         else
231         {
232             /*
233              * The 'StartupWindowThread' thread couldn't be created,
234              * so we need to free the allocated 'msg'.
235              */
236             HeapFree(GetProcessHeap(), 0, msg);
237         }
238 
239         return FALSE;
240     }
241 
242     if (pTitle)
243         SetWindowTextW(pgContext->hStatusWindow, pTitle);
244 
245     SetDlgItemTextW(pgContext->hStatusWindow, IDC_STATUS_MESSAGE, pMessage);
246 
247     return TRUE;
248 }
249 
250 static BOOL
251 GUIRemoveStatusMessage(
252     IN PGINA_CONTEXT pgContext)
253 {
254     if (pgContext->hStatusWindow)
255     {
256         EndDialog(pgContext->hStatusWindow, 0);
257         pgContext->hStatusWindow = NULL;
258     }
259 
260     return TRUE;
261 }
262 
263 static INT_PTR CALLBACK
264 WelcomeDialogProc(
265     IN HWND hwndDlg,
266     IN UINT uMsg,
267     IN WPARAM wParam,
268     IN LPARAM lParam)
269 {
270     PDLG_DATA pDlgData;
271 
272     pDlgData = (PDLG_DATA)GetWindowLongPtrW(hwndDlg, GWLP_USERDATA);
273 
274     switch (uMsg)
275     {
276         case WM_INITDIALOG:
277         {
278             pDlgData = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(DLG_DATA));
279             if (pDlgData == NULL)
280                 return FALSE;
281 
282             SetWindowLongPtrW(hwndDlg, GWLP_USERDATA, (LONG_PTR)pDlgData);
283 
284             pDlgData->pgContext = (PGINA_CONTEXT)lParam;
285 
286             /* Load the logo bitmap */
287             pDlgData->hBitmap = LoadImageW(pDlgData->pgContext->hDllInstance, MAKEINTRESOURCEW(IDI_ROSLOGO), IMAGE_BITMAP, 0, 0, LR_DEFAULTCOLOR);
288             return TRUE;
289         }
290 
291         case WM_PAINT:
292         {
293             PAINTSTRUCT ps;
294             if (pDlgData->hBitmap)
295             {
296                 BeginPaint(hwndDlg, &ps);
297                 DrawStateW(ps.hdc, NULL, NULL, (LPARAM)pDlgData->hBitmap, (WPARAM)0, 0, 0, 0, 0, DST_BITMAP);
298                 EndPaint(hwndDlg, &ps);
299             }
300             return TRUE;
301         }
302         case WM_DESTROY:
303         {
304             DeleteObject(pDlgData->hBitmap);
305             HeapFree(GetProcessHeap(), 0, pDlgData);
306             return TRUE;
307         }
308     }
309     return FALSE;
310 }
311 
312 static VOID
313 GUIDisplaySASNotice(
314     IN OUT PGINA_CONTEXT pgContext)
315 {
316     TRACE("GUIDisplaySASNotice()\n");
317 
318     /* Display the notice window */
319     pgContext->pWlxFuncs->WlxDialogBoxParam(pgContext->hWlx,
320                                             pgContext->hDllInstance,
321                                             MAKEINTRESOURCEW(IDD_WELCOME),
322                                             GetDesktopWindow(),
323                                             WelcomeDialogProc,
324                                             (LPARAM)pgContext);
325 }
326 
327 /* Get the text contained in a textbox. Allocates memory in pText
328  * to contain the text. Returns TRUE in case of success */
329 static BOOL
330 GetTextboxText(
331     IN HWND hwndDlg,
332     IN INT TextboxId,
333     OUT LPWSTR *pText)
334 {
335     LPWSTR Text;
336     int Count;
337 
338     Count = GetWindowTextLength(GetDlgItem(hwndDlg, TextboxId));
339     Text = HeapAlloc(GetProcessHeap(), 0, (Count + 1) * sizeof(WCHAR));
340     if (!Text)
341         return FALSE;
342     if (Count != GetWindowTextW(GetDlgItem(hwndDlg, TextboxId), Text, Count + 1))
343     {
344         HeapFree(GetProcessHeap(), 0, Text);
345         return FALSE;
346     }
347     *pText = Text;
348     return TRUE;
349 }
350 
351 
352 static
353 INT
354 ResourceMessageBox(
355     IN PGINA_CONTEXT pgContext,
356     IN HWND hwnd,
357     IN UINT uType,
358     IN UINT uCaption,
359     IN UINT uText)
360 {
361     WCHAR szCaption[256];
362     WCHAR szText[256];
363 
364     LoadStringW(pgContext->hDllInstance, uCaption, szCaption, _countof(szCaption));
365     LoadStringW(pgContext->hDllInstance, uText, szText, _countof(szText));
366 
367     return pgContext->pWlxFuncs->WlxMessageBox(pgContext->hWlx,
368                                                hwnd,
369                                                szText,
370                                                szCaption,
371                                                uType);
372 }
373 
374 
375 static
376 BOOL
377 DoChangePassword(
378     IN PGINA_CONTEXT pgContext,
379     IN HWND hwndDlg)
380 {
381     WCHAR UserName[256];
382     WCHAR Domain[256];
383     WCHAR OldPassword[256];
384     WCHAR NewPassword1[256];
385     WCHAR NewPassword2[256];
386     PMSV1_0_CHANGEPASSWORD_REQUEST RequestBuffer = NULL;
387     PMSV1_0_CHANGEPASSWORD_RESPONSE ResponseBuffer = NULL;
388     ULONG RequestBufferSize;
389     ULONG ResponseBufferSize = 0;
390     LPWSTR Ptr;
391     BOOL res = FALSE;
392     NTSTATUS ProtocolStatus;
393     NTSTATUS Status;
394 
395     GetDlgItemTextW(hwndDlg, IDC_CHANGEPWD_USERNAME, UserName, _countof(UserName));
396     GetDlgItemTextW(hwndDlg, IDC_CHANGEPWD_DOMAIN, Domain, _countof(Domain));
397     GetDlgItemTextW(hwndDlg, IDC_CHANGEPWD_OLDPWD, OldPassword, _countof(OldPassword));
398     GetDlgItemTextW(hwndDlg, IDC_CHANGEPWD_NEWPWD1, NewPassword1, _countof(NewPassword1));
399     GetDlgItemTextW(hwndDlg, IDC_CHANGEPWD_NEWPWD2, NewPassword2, _countof(NewPassword2));
400 
401     /* Compare the two passwords and fail if they do not match */
402     if (wcscmp(NewPassword1, NewPassword2) != 0)
403     {
404         ResourceMessageBox(pgContext,
405                            hwndDlg,
406                            MB_OK | MB_ICONEXCLAMATION,
407                            IDS_CHANGEPWDTITLE,
408                            IDS_NONMATCHINGPASSWORDS);
409         return FALSE;
410     }
411 
412     /* Calculate the request buffer size */
413     RequestBufferSize = sizeof(MSV1_0_CHANGEPASSWORD_REQUEST) +
414                         ((wcslen(Domain) + 1) * sizeof(WCHAR)) +
415                         ((wcslen(UserName) + 1) * sizeof(WCHAR)) +
416                         ((wcslen(OldPassword) + 1) * sizeof(WCHAR)) +
417                         ((wcslen(NewPassword1) + 1) * sizeof(WCHAR));
418 
419     /* Allocate the request buffer */
420     RequestBuffer = HeapAlloc(GetProcessHeap(),
421                               HEAP_ZERO_MEMORY,
422                               RequestBufferSize);
423     if (RequestBuffer == NULL)
424     {
425         ERR("HeapAlloc failed\n");
426         return FALSE;
427     }
428 
429     /* Initialize the request buffer */
430     RequestBuffer->MessageType = MsV1_0ChangePassword;
431     RequestBuffer->Impersonating = TRUE;
432 
433     Ptr = (LPWSTR)((ULONG_PTR)RequestBuffer + sizeof(MSV1_0_CHANGEPASSWORD_REQUEST));
434 
435     /* Pack the domain name */
436     RequestBuffer->DomainName.Length = wcslen(Domain) * sizeof(WCHAR);
437     RequestBuffer->DomainName.MaximumLength = RequestBuffer->DomainName.Length + sizeof(WCHAR);
438     RequestBuffer->DomainName.Buffer = Ptr;
439 
440     RtlCopyMemory(RequestBuffer->DomainName.Buffer,
441                   Domain,
442                   RequestBuffer->DomainName.MaximumLength);
443 
444     Ptr = (LPWSTR)((ULONG_PTR)Ptr + RequestBuffer->DomainName.MaximumLength);
445 
446     /* Pack the user name */
447     RequestBuffer->AccountName.Length = wcslen(UserName) * sizeof(WCHAR);
448     RequestBuffer->AccountName.MaximumLength = RequestBuffer->AccountName.Length + sizeof(WCHAR);
449     RequestBuffer->AccountName.Buffer = Ptr;
450 
451     RtlCopyMemory(RequestBuffer->AccountName.Buffer,
452                   UserName,
453                   RequestBuffer->AccountName.MaximumLength);
454 
455     Ptr = (LPWSTR)((ULONG_PTR)Ptr + RequestBuffer->AccountName.MaximumLength);
456 
457     /* Pack the old password */
458     RequestBuffer->OldPassword.Length = wcslen(OldPassword) * sizeof(WCHAR);
459     RequestBuffer->OldPassword.MaximumLength = RequestBuffer->OldPassword.Length + sizeof(WCHAR);
460     RequestBuffer->OldPassword.Buffer = Ptr;
461 
462     RtlCopyMemory(RequestBuffer->OldPassword.Buffer,
463                   OldPassword,
464                   RequestBuffer->OldPassword.MaximumLength);
465 
466     Ptr = (LPWSTR)((ULONG_PTR)Ptr + RequestBuffer->OldPassword.MaximumLength);
467 
468     /* Pack the new password */
469     RequestBuffer->NewPassword.Length = wcslen(NewPassword1) * sizeof(WCHAR);
470     RequestBuffer->NewPassword.MaximumLength = RequestBuffer->NewPassword.Length + sizeof(WCHAR);
471     RequestBuffer->NewPassword.Buffer = Ptr;
472 
473     RtlCopyMemory(RequestBuffer->NewPassword.Buffer,
474                   NewPassword1,
475                   RequestBuffer->NewPassword.MaximumLength);
476 
477     /* Connect to the LSA server */
478     if (ConnectToLsa(pgContext) != ERROR_SUCCESS)
479     {
480         ERR("ConnectToLsa() failed\n");
481         goto done;
482     }
483 
484     /* Call the authentication package */
485     Status = LsaCallAuthenticationPackage(pgContext->LsaHandle,
486                                           pgContext->AuthenticationPackage,
487                                           RequestBuffer,
488                                           RequestBufferSize,
489                                           (PVOID*)&ResponseBuffer,
490                                           &ResponseBufferSize,
491                                           &ProtocolStatus);
492     if (!NT_SUCCESS(Status))
493     {
494         ERR("LsaCallAuthenticationPackage failed (Status 0x%08lx)\n", Status);
495         goto done;
496     }
497 
498     if (!NT_SUCCESS(ProtocolStatus))
499     {
500         TRACE("LsaCallAuthenticationPackage failed (ProtocolStatus 0x%08lx)\n", ProtocolStatus);
501         goto done;
502     }
503 
504     res = TRUE;
505 
506     ResourceMessageBox(pgContext,
507                        hwndDlg,
508                        MB_OK | MB_ICONINFORMATION,
509                        IDS_CHANGEPWDTITLE,
510                        IDS_PASSWORDCHANGED);
511 
512     if ((wcscmp(UserName, pgContext->UserName) == 0) &&
513         (wcscmp(Domain, pgContext->DomainName) == 0) &&
514         (wcscmp(OldPassword, pgContext->Password) == 0))
515     {
516         ZeroMemory(pgContext->Password, sizeof(pgContext->Password));
517         wcscpy(pgContext->Password, NewPassword1);
518     }
519 
520 done:
521     if (RequestBuffer != NULL)
522         HeapFree(GetProcessHeap(), 0, RequestBuffer);
523 
524     if (ResponseBuffer != NULL)
525         LsaFreeReturnBuffer(ResponseBuffer);
526 
527     return res;
528 }
529 
530 
531 static INT_PTR CALLBACK
532 ChangePasswordDialogProc(
533     IN HWND hwndDlg,
534     IN UINT uMsg,
535     IN WPARAM wParam,
536     IN LPARAM lParam)
537 {
538     PGINA_CONTEXT pgContext;
539 
540     pgContext = (PGINA_CONTEXT)GetWindowLongPtrW(hwndDlg, GWLP_USERDATA);
541 
542     switch (uMsg)
543     {
544         case WM_INITDIALOG:
545         {
546             pgContext = (PGINA_CONTEXT)lParam;
547             SetWindowLongPtrW(hwndDlg, GWLP_USERDATA, (LONG_PTR)pgContext);
548 
549             SetDlgItemTextW(hwndDlg, IDC_CHANGEPWD_USERNAME, pgContext->UserName);
550             SendDlgItemMessageW(hwndDlg, IDC_CHANGEPWD_DOMAIN, CB_ADDSTRING, 0, (LPARAM)pgContext->DomainName);
551             SendDlgItemMessageW(hwndDlg, IDC_CHANGEPWD_DOMAIN, CB_SETCURSEL, 0, 0);
552             SetFocus(GetDlgItem(hwndDlg, IDC_CHANGEPWD_OLDPWD));
553             return TRUE;
554         }
555 
556         case WM_COMMAND:
557             switch (LOWORD(wParam))
558             {
559                 case IDOK:
560                     if (DoChangePassword(pgContext, hwndDlg))
561                     {
562                         EndDialog(hwndDlg, TRUE);
563                     }
564                     else
565                     {
566                         SetDlgItemTextW(hwndDlg, IDC_CHANGEPWD_NEWPWD1, NULL);
567                         SetDlgItemTextW(hwndDlg, IDC_CHANGEPWD_NEWPWD2, NULL);
568                         SetFocus(GetDlgItem(hwndDlg, IDC_CHANGEPWD_OLDPWD));
569                     }
570                     return TRUE;
571 
572                 case IDCANCEL:
573                     EndDialog(hwndDlg, FALSE);
574                     return TRUE;
575             }
576             break;
577 
578         case WM_CLOSE:
579             EndDialog(hwndDlg, FALSE);
580             return TRUE;
581     }
582 
583     return FALSE;
584 }
585 
586 
587 static VOID
588 OnInitSecurityDlg(HWND hwnd,
589                   PGINA_CONTEXT pgContext)
590 {
591     WCHAR Buffer1[256];
592     WCHAR Buffer2[256];
593     WCHAR Buffer3[256];
594     WCHAR Buffer4[512];
595 
596     LoadStringW(pgContext->hDllInstance, IDS_LOGONMSG, Buffer1, _countof(Buffer1));
597 
598     wsprintfW(Buffer2, L"%s\\%s", pgContext->DomainName, pgContext->UserName);
599     wsprintfW(Buffer4, Buffer1, Buffer2);
600 
601     SetDlgItemTextW(hwnd, IDC_SECURITY_MESSAGE, Buffer4);
602 
603     LoadStringW(pgContext->hDllInstance, IDS_LOGONDATE, Buffer1, _countof(Buffer1));
604 
605     GetDateFormatW(LOCALE_USER_DEFAULT, DATE_SHORTDATE,
606                    (SYSTEMTIME*)&pgContext->LogonTime, NULL, Buffer2, _countof(Buffer2));
607 
608     GetTimeFormatW(LOCALE_USER_DEFAULT, 0,
609                    (SYSTEMTIME*)&pgContext->LogonTime, NULL, Buffer3, _countof(Buffer3));
610 
611     wsprintfW(Buffer4, Buffer1, Buffer2, Buffer3);
612 
613     SetDlgItemTextW(hwnd, IDC_SECURITY_LOGONDATE, Buffer4);
614 
615     if (pgContext->bAutoAdminLogon)
616         EnableWindow(GetDlgItem(hwnd, IDC_SECURITY_LOGOFF), FALSE);
617 }
618 
619 
620 static BOOL
621 OnChangePassword(
622     IN HWND hwnd,
623     IN PGINA_CONTEXT pgContext)
624 {
625     INT res;
626 
627     TRACE("OnChangePassword()\n");
628 
629     res = pgContext->pWlxFuncs->WlxDialogBoxParam(
630         pgContext->hWlx,
631         pgContext->hDllInstance,
632         MAKEINTRESOURCEW(IDD_CHANGEPWD),
633         hwnd,
634         ChangePasswordDialogProc,
635         (LPARAM)pgContext);
636 
637     TRACE("Result: %x\n", res);
638 
639     return FALSE;
640 }
641 
642 
643 static INT_PTR CALLBACK
644 LogOffDialogProc(
645     IN HWND hwndDlg,
646     IN UINT uMsg,
647     IN WPARAM wParam,
648     IN LPARAM lParam)
649 {
650     switch (uMsg)
651     {
652         case WM_INITDIALOG:
653             return TRUE;
654 
655         case WM_COMMAND:
656             switch (LOWORD(wParam))
657             {
658                 case IDYES:
659                     EndDialog(hwndDlg, IDYES);
660                     return TRUE;
661 
662                 case IDNO:
663                     EndDialog(hwndDlg, IDNO);
664                     return TRUE;
665             }
666             break;
667 
668         case WM_CLOSE:
669             EndDialog(hwndDlg, IDNO);
670             return TRUE;
671     }
672 
673     return FALSE;
674 }
675 
676 
677 static
678 INT
679 OnLogOff(
680     IN HWND hwndDlg,
681     IN PGINA_CONTEXT pgContext)
682 {
683     return pgContext->pWlxFuncs->WlxDialogBoxParam(
684         pgContext->hWlx,
685         pgContext->hDllInstance,
686         MAKEINTRESOURCEW(IDD_LOGOFF),
687         hwndDlg,
688         LogOffDialogProc,
689         (LPARAM)pgContext);
690 }
691 
692 
693 static
694 INT
695 OnShutDown(
696     IN HWND hwndDlg,
697     IN PGINA_CONTEXT pgContext)
698 {
699     INT ret;
700     DWORD ShutdownOptions;
701 
702     TRACE("OnShutDown(%p %p)\n", hwndDlg, pgContext);
703 
704     pgContext->nShutdownAction = GetDefaultShutdownSelState();
705     ShutdownOptions = GetDefaultShutdownOptions();
706 
707     if (pgContext->UserToken != NULL)
708     {
709         if (ImpersonateLoggedOnUser(pgContext->UserToken))
710         {
711             pgContext->nShutdownAction = LoadShutdownSelState();
712             ShutdownOptions = GetAllowedShutdownOptions();
713             RevertToSelf();
714         }
715         else
716         {
717             ERR("WL: ImpersonateLoggedOnUser() failed with error %lu\n", GetLastError());
718         }
719     }
720 
721     ret = ShutdownDialog(hwndDlg, ShutdownOptions, pgContext);
722 
723     if (ret == IDOK)
724     {
725         if (pgContext->UserToken != NULL)
726         {
727             if (ImpersonateLoggedOnUser(pgContext->UserToken))
728             {
729                 SaveShutdownSelState(pgContext->nShutdownAction);
730                 RevertToSelf();
731             }
732             else
733             {
734                 ERR("WL: ImpersonateLoggedOnUser() failed with error %lu\n", GetLastError());
735             }
736         }
737     }
738 
739     return ret;
740 }
741 
742 
743 static INT_PTR CALLBACK
744 SecurityDialogProc(
745     IN HWND hwndDlg,
746     IN UINT uMsg,
747     IN WPARAM wParam,
748     IN LPARAM lParam)
749 {
750     PGINA_CONTEXT pgContext;
751 
752     pgContext = (PGINA_CONTEXT)GetWindowLongPtrW(hwndDlg, GWLP_USERDATA);
753 
754     switch (uMsg)
755     {
756         case WM_INITDIALOG:
757         {
758             pgContext = (PGINA_CONTEXT)lParam;
759             SetWindowLongPtrW(hwndDlg, GWLP_USERDATA, (LONG_PTR)pgContext);
760 
761             SetWelcomeText(hwndDlg);
762 
763             OnInitSecurityDlg(hwndDlg, (PGINA_CONTEXT)lParam);
764             SetFocus(GetDlgItem(hwndDlg, IDNO));
765             return TRUE;
766         }
767 
768         case WM_COMMAND:
769         {
770             switch (LOWORD(wParam))
771             {
772                 case IDC_SECURITY_LOCK:
773                     EndDialog(hwndDlg, WLX_SAS_ACTION_LOCK_WKSTA);
774                     return TRUE;
775                 case IDC_SECURITY_LOGOFF:
776                     if (OnLogOff(hwndDlg, pgContext) == IDYES)
777                         EndDialog(hwndDlg, WLX_SAS_ACTION_LOGOFF);
778                     return TRUE;
779                 case IDC_SECURITY_SHUTDOWN:
780                     if (OnShutDown(hwndDlg, pgContext) == IDOK)
781                         EndDialog(hwndDlg, pgContext->nShutdownAction);
782                     return TRUE;
783                 case IDC_SECURITY_CHANGEPWD:
784                     if (OnChangePassword(hwndDlg, pgContext))
785                         EndDialog(hwndDlg, WLX_SAS_ACTION_PWD_CHANGED);
786                     return TRUE;
787                 case IDC_SECURITY_TASKMGR:
788                     EndDialog(hwndDlg, WLX_SAS_ACTION_TASKLIST);
789                     return TRUE;
790                 case IDCANCEL:
791                     EndDialog(hwndDlg, WLX_SAS_ACTION_NONE);
792                     return TRUE;
793             }
794             break;
795         }
796         case WM_CLOSE:
797         {
798             EndDialog(hwndDlg, WLX_SAS_ACTION_NONE);
799             return TRUE;
800         }
801     }
802 
803     return FALSE;
804 }
805 
806 static INT
807 GUILoggedOnSAS(
808     IN OUT PGINA_CONTEXT pgContext,
809     IN DWORD dwSasType)
810 {
811     INT result;
812 
813     TRACE("GUILoggedOnSAS()\n");
814 
815     if (dwSasType != WLX_SAS_TYPE_CTRL_ALT_DEL)
816     {
817         /* Nothing to do for WLX_SAS_TYPE_TIMEOUT ; the dialog will
818          * close itself thanks to the use of WlxDialogBoxParam */
819         return WLX_SAS_ACTION_NONE;
820     }
821 
822     pgContext->pWlxFuncs->WlxSwitchDesktopToWinlogon(
823         pgContext->hWlx);
824 
825     result = pgContext->pWlxFuncs->WlxDialogBoxParam(
826         pgContext->hWlx,
827         pgContext->hDllInstance,
828         MAKEINTRESOURCEW(IDD_SECURITY),
829         GetDesktopWindow(),
830         SecurityDialogProc,
831         (LPARAM)pgContext);
832 
833     if (result < WLX_SAS_ACTION_LOGON ||
834         result > WLX_SAS_ACTION_SWITCH_CONSOLE)
835     {
836         result = WLX_SAS_ACTION_NONE;
837     }
838 
839     if (result == WLX_SAS_ACTION_NONE)
840     {
841         pgContext->pWlxFuncs->WlxSwitchDesktopToUser(
842             pgContext->hWlx);
843     }
844 
845     return result;
846 }
847 
848 
849 static
850 BOOL
851 DoLogon(
852     IN HWND hwndDlg,
853     IN OUT PGINA_CONTEXT pgContext)
854 {
855     LPWSTR UserName = NULL;
856     LPWSTR Password = NULL;
857     LPWSTR Domain = NULL;
858     BOOL result = FALSE;
859     NTSTATUS Status, SubStatus = STATUS_SUCCESS;
860 
861     if (GetTextboxText(hwndDlg, IDC_LOGON_USERNAME, &UserName) && *UserName == '\0')
862         goto done;
863 
864     if (GetTextboxText(hwndDlg, IDC_LOGON_DOMAIN, &Domain) && *Domain == '\0')
865         goto done;
866 
867     if (!GetTextboxText(hwndDlg, IDC_LOGON_PASSWORD, &Password))
868         goto done;
869 
870     Status = DoLoginTasks(pgContext, UserName, Domain, Password, &SubStatus);
871     if (Status == STATUS_LOGON_FAILURE)
872     {
873         ResourceMessageBox(pgContext,
874                            hwndDlg,
875                            MB_OK | MB_ICONEXCLAMATION,
876                            IDS_LOGONTITLE,
877                            IDS_LOGONWRONGUSERORPWD);
878         goto done;
879     }
880     else if (Status == STATUS_ACCOUNT_RESTRICTION)
881     {
882         TRACE("DoLoginTasks failed! Status 0x%08lx  SubStatus 0x%08lx\n", Status, SubStatus);
883 
884         if (SubStatus == STATUS_ACCOUNT_DISABLED)
885         {
886             ResourceMessageBox(pgContext,
887                                hwndDlg,
888                                MB_OK | MB_ICONEXCLAMATION,
889                                IDS_LOGONTITLE,
890                                IDS_LOGONUSERDISABLED);
891             goto done;
892         }
893         else if (SubStatus == STATUS_ACCOUNT_LOCKED_OUT)
894         {
895             TRACE("Account locked!\n");
896             pgContext->pWlxFuncs->WlxMessageBox(pgContext->hWlx,
897                                                 hwndDlg,
898                                                 L"Account locked!",
899                                                 L"Logon error",
900                                                 MB_OK | MB_ICONERROR);
901             goto done;
902         }
903         else if ((SubStatus == STATUS_PASSWORD_MUST_CHANGE) ||
904                  (SubStatus == STATUS_PASSWORD_EXPIRED))
905         {
906             if (SubStatus == STATUS_PASSWORD_MUST_CHANGE)
907                 ResourceMessageBox(pgContext,
908                                    hwndDlg,
909                                    MB_OK | MB_ICONSTOP,
910                                    IDS_LOGONTITLE,
911                                    IDS_PASSWORDMUSTCHANGE);
912             else
913                 ResourceMessageBox(pgContext,
914                                    hwndDlg,
915                                    MB_OK | MB_ICONSTOP,
916                                    IDS_LOGONTITLE,
917                                    IDS_PASSWORDEXPIRED);
918 
919             if (!OnChangePassword(hwndDlg,
920                                   pgContext))
921                 goto done;
922 
923             Status = DoLoginTasks(pgContext,
924                                   pgContext->UserName,
925                                   pgContext->DomainName,
926                                   pgContext->Password,
927                                   &SubStatus);
928             if (!NT_SUCCESS(Status))
929             {
930                 TRACE("Login after password change failed! (Status 0x%08lx)\n", Status);
931 
932                 goto done;
933             }
934         }
935         else if (SubStatus == STATUS_ACCOUNT_EXPIRED)
936         {
937             ResourceMessageBox(pgContext,
938                                hwndDlg,
939                                MB_OK | MB_ICONEXCLAMATION,
940                                IDS_LOGONTITLE,
941                                IDS_ACCOUNTEXPIRED);
942         }
943         else
944         {
945             TRACE("Other error!\n");
946             pgContext->pWlxFuncs->WlxMessageBox(pgContext->hWlx,
947                                                 hwndDlg,
948                                                 L"Other error!",
949                                                 L"Logon error",
950                                                 MB_OK | MB_ICONERROR);
951             goto done;
952         }
953     }
954     else if (!NT_SUCCESS(Status))
955     {
956         TRACE("DoLoginTasks failed! Status 0x%08lx\n", Status);
957         goto done;
958     }
959 
960 
961     if (!CreateProfile(pgContext, UserName, Domain, Password))
962     {
963         ERR("Failed to create the profile!\n");
964         goto done;
965     }
966 
967     ZeroMemory(pgContext->Password, sizeof(pgContext->Password));
968     wcscpy(pgContext->Password, Password);
969 
970     result = TRUE;
971 
972 done:
973     pgContext->bAutoAdminLogon = FALSE;
974 
975     if (UserName != NULL)
976         HeapFree(GetProcessHeap(), 0, UserName);
977 
978     if (Password != NULL)
979         HeapFree(GetProcessHeap(), 0, Password);
980 
981     if (Domain != NULL)
982         HeapFree(GetProcessHeap(), 0, Domain);
983 
984     return result;
985 }
986 
987 
988 static
989 VOID
990 SetDomainComboBox(
991     HWND hwndDomainComboBox,
992     PGINA_CONTEXT pgContext)
993 {
994     WCHAR szComputerName[MAX_COMPUTERNAME_LENGTH + 1];
995     DWORD dwComputerNameLength;
996     LONG lIndex = 0;
997     LONG lFindIndex;
998 
999     SendMessageW(hwndDomainComboBox, CB_RESETCONTENT, 0, 0);
1000 
1001     dwComputerNameLength = _countof(szComputerName);
1002     if (GetComputerNameW(szComputerName, &dwComputerNameLength))
1003     {
1004         lIndex = SendMessageW(hwndDomainComboBox, CB_ADDSTRING, 0, (LPARAM)szComputerName);
1005     }
1006 
1007     if (wcslen(pgContext->DomainName) != 0)
1008     {
1009         lFindIndex = SendMessageW(hwndDomainComboBox, CB_FINDSTRINGEXACT, (WPARAM)-1, (LPARAM)pgContext->DomainName);
1010         if (lFindIndex == CB_ERR)
1011         {
1012             lIndex = SendMessageW(hwndDomainComboBox, CB_ADDSTRING, 0, (LPARAM)pgContext->DomainName);
1013         }
1014         else
1015         {
1016             lIndex = lFindIndex;
1017         }
1018     }
1019 
1020     SendMessageW(hwndDomainComboBox, CB_SETCURSEL, lIndex, 0);
1021 }
1022 
1023 
1024 static INT_PTR CALLBACK
1025 LogonDialogProc(
1026     IN HWND hwndDlg,
1027     IN UINT uMsg,
1028     IN WPARAM wParam,
1029     IN LPARAM lParam)
1030 {
1031     PDLG_DATA pDlgData;
1032 
1033     pDlgData = (PDLG_DATA)GetWindowLongPtrW(hwndDlg, GWLP_USERDATA);
1034 
1035     switch (uMsg)
1036     {
1037         case WM_INITDIALOG:
1038         {
1039             pDlgData = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(DLG_DATA));
1040             if (pDlgData == NULL)
1041                 return FALSE;
1042 
1043             SetWindowLongPtrW(hwndDlg, GWLP_USERDATA, (LONG_PTR)pDlgData);
1044 
1045             /* FIXME: take care of NoDomainUI */
1046             pDlgData->pgContext = (PGINA_CONTEXT)lParam;
1047 
1048             /* Draw the logo bitmap */
1049             pDlgData->hBitmap = LoadImageW(pDlgData->pgContext->hDllInstance, MAKEINTRESOURCEW(IDI_ROSLOGO), IMAGE_BITMAP, 0, 0, LR_DEFAULTCOLOR);
1050 
1051             SetWelcomeText(hwndDlg);
1052 
1053             if (pDlgData->pgContext->bAutoAdminLogon ||
1054                 !pDlgData->pgContext->bDontDisplayLastUserName)
1055                 SetDlgItemTextW(hwndDlg, IDC_LOGON_USERNAME, pDlgData->pgContext->UserName);
1056 
1057             if (pDlgData->pgContext->bAutoAdminLogon)
1058                 SetDlgItemTextW(hwndDlg, IDC_LOGON_PASSWORD, pDlgData->pgContext->Password);
1059 
1060             SetDomainComboBox(GetDlgItem(hwndDlg, IDC_LOGON_DOMAIN), pDlgData->pgContext);
1061 
1062             if (pDlgData->pgContext->bDisableCAD)
1063                 EnableWindow(GetDlgItem(hwndDlg, IDCANCEL), FALSE);
1064 
1065             if (!pDlgData->pgContext->bShutdownWithoutLogon)
1066                 EnableWindow(GetDlgItem(hwndDlg, IDC_LOGON_SHUTDOWN), FALSE);
1067 
1068             SetFocus(GetDlgItem(hwndDlg, pDlgData->pgContext->bDontDisplayLastUserName ? IDC_LOGON_USERNAME : IDC_LOGON_PASSWORD));
1069 
1070             if (pDlgData->pgContext->bAutoAdminLogon)
1071                 PostMessage(GetDlgItem(hwndDlg, IDOK), BM_CLICK, 0, 0);
1072 
1073             return TRUE;
1074         }
1075 
1076         case WM_PAINT:
1077         {
1078             PAINTSTRUCT ps;
1079             if (pDlgData->hBitmap)
1080             {
1081                 BeginPaint(hwndDlg, &ps);
1082                 DrawStateW(ps.hdc, NULL, NULL, (LPARAM)pDlgData->hBitmap, (WPARAM)0, 0, 0, 0, 0, DST_BITMAP);
1083                 EndPaint(hwndDlg, &ps);
1084             }
1085             return TRUE;
1086         }
1087 
1088         case WM_DESTROY:
1089             DeleteObject(pDlgData->hBitmap);
1090             HeapFree(GetProcessHeap(), 0, pDlgData);
1091             return TRUE;
1092 
1093         case WM_COMMAND:
1094             switch (LOWORD(wParam))
1095             {
1096                 case IDOK:
1097                     if (DoLogon(hwndDlg, pDlgData->pgContext))
1098                         EndDialog(hwndDlg, WLX_SAS_ACTION_LOGON);
1099                     return TRUE;
1100 
1101                 case IDCANCEL:
1102                     EndDialog(hwndDlg, WLX_SAS_ACTION_NONE);
1103                     return TRUE;
1104 
1105                 case IDC_LOGON_SHUTDOWN:
1106                     if (OnShutDown(hwndDlg, pDlgData->pgContext) == IDOK)
1107                         EndDialog(hwndDlg, pDlgData->pgContext->nShutdownAction);
1108                     return TRUE;
1109             }
1110             break;
1111     }
1112 
1113     return FALSE;
1114 }
1115 
1116 
1117 static
1118 INT_PTR
1119 CALLBACK
1120 LegalNoticeDialogProc(
1121     IN HWND hwndDlg,
1122     IN UINT uMsg,
1123     IN WPARAM wParam,
1124     IN LPARAM lParam)
1125 {
1126     PLEGALNOTICEDATA pLegalNotice;
1127 
1128     switch (uMsg)
1129     {
1130         case WM_INITDIALOG:
1131             pLegalNotice = (PLEGALNOTICEDATA)lParam;
1132             SetWindowTextW(hwndDlg, pLegalNotice->pszCaption);
1133             SetDlgItemTextW(hwndDlg, IDC_LEGALNOTICE_TEXT, pLegalNotice->pszText);
1134             return TRUE;
1135 
1136         case WM_COMMAND:
1137             switch (LOWORD(wParam))
1138             {
1139                 case IDOK:
1140                     EndDialog(hwndDlg, 0);
1141                     return TRUE;
1142 
1143                 case IDCANCEL:
1144                     EndDialog(hwndDlg, 0);
1145                     return TRUE;
1146             }
1147             break;
1148     }
1149 
1150     return FALSE;
1151 }
1152 
1153 
1154 static INT
1155 GUILoggedOutSAS(
1156     IN OUT PGINA_CONTEXT pgContext)
1157 {
1158     LEGALNOTICEDATA LegalNotice = {NULL, NULL};
1159     HKEY hKey = NULL;
1160     LONG rc;
1161     int result;
1162 
1163     TRACE("GUILoggedOutSAS()\n");
1164 
1165     rc = RegOpenKeyExW(HKEY_LOCAL_MACHINE,
1166                        L"SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion\\Winlogon",
1167                        0,
1168                        KEY_QUERY_VALUE,
1169                        &hKey);
1170     if (rc == ERROR_SUCCESS)
1171     {
1172         ReadRegSzValue(hKey,
1173                        L"LegalNoticeCaption",
1174                        &LegalNotice.pszCaption);
1175 
1176         ReadRegSzValue(hKey,
1177                        L"LegalNoticeText",
1178                        &LegalNotice.pszText);
1179 
1180         RegCloseKey(hKey);
1181     }
1182 
1183     if (LegalNotice.pszCaption != NULL && wcslen(LegalNotice.pszCaption) != 0 &&
1184         LegalNotice.pszText != NULL && wcslen(LegalNotice.pszText) != 0)
1185     {
1186         pgContext->pWlxFuncs->WlxDialogBoxParam(pgContext->hWlx,
1187                                                 pgContext->hDllInstance,
1188                                                 MAKEINTRESOURCEW(IDD_LEGALNOTICE),
1189                                                 GetDesktopWindow(),
1190                                                 LegalNoticeDialogProc,
1191                                                 (LPARAM)&LegalNotice);
1192     }
1193 
1194     if (LegalNotice.pszCaption != NULL)
1195         HeapFree(GetProcessHeap(), 0, LegalNotice.pszCaption);
1196 
1197     if (LegalNotice.pszText != NULL)
1198         HeapFree(GetProcessHeap(), 0, LegalNotice.pszText);
1199 
1200     result = pgContext->pWlxFuncs->WlxDialogBoxParam(
1201         pgContext->hWlx,
1202         pgContext->hDllInstance,
1203         MAKEINTRESOURCEW(IDD_LOGON),
1204         GetDesktopWindow(),
1205         LogonDialogProc,
1206         (LPARAM)pgContext);
1207     if (result >= WLX_SAS_ACTION_LOGON &&
1208         result <= WLX_SAS_ACTION_SWITCH_CONSOLE)
1209     {
1210         WARN("WlxLoggedOutSAS() returns 0x%x\n", result);
1211         return result;
1212     }
1213 
1214     WARN("WlxDialogBoxParam() failed (0x%x)\n", result);
1215     return WLX_SAS_ACTION_NONE;
1216 }
1217 
1218 
1219 static VOID
1220 SetLockMessage(HWND hwnd,
1221                INT nDlgItem,
1222                PGINA_CONTEXT pgContext)
1223 {
1224     WCHAR Buffer1[256];
1225     WCHAR Buffer2[256];
1226     WCHAR Buffer3[512];
1227 
1228     LoadStringW(pgContext->hDllInstance, IDS_LOCKMSG, Buffer1, _countof(Buffer1));
1229 
1230     wsprintfW(Buffer2, L"%s\\%s", pgContext->DomainName, pgContext->UserName);
1231     wsprintfW(Buffer3, Buffer1, Buffer2);
1232 
1233     SetDlgItemTextW(hwnd, nDlgItem, Buffer3);
1234 }
1235 
1236 
1237 static
1238 BOOL
1239 DoUnlock(
1240     IN HWND hwndDlg,
1241     IN PGINA_CONTEXT pgContext,
1242     OUT LPINT Action)
1243 {
1244     WCHAR Buffer1[256];
1245     WCHAR Buffer2[256];
1246     LPWSTR UserName = NULL;
1247     LPWSTR Password = NULL;
1248     BOOL res = FALSE;
1249 
1250     if (GetTextboxText(hwndDlg, IDC_UNLOCK_USERNAME, &UserName) && *UserName == '\0')
1251     {
1252         HeapFree(GetProcessHeap(), 0, UserName);
1253         return FALSE;
1254     }
1255 
1256     if (GetTextboxText(hwndDlg, IDC_UNLOCK_PASSWORD, &Password))
1257     {
1258         if (UserName != NULL && Password != NULL &&
1259             wcscmp(UserName, pgContext->UserName) == 0 &&
1260             wcscmp(Password, pgContext->Password) == 0)
1261         {
1262             *Action = WLX_SAS_ACTION_UNLOCK_WKSTA;
1263             res = TRUE;
1264         }
1265         else if (wcscmp(UserName, pgContext->UserName) == 0 &&
1266                  wcscmp(Password, pgContext->Password) != 0)
1267         {
1268             /* Wrong Password */
1269             LoadStringW(pgContext->hDllInstance, IDS_LOCKEDWRONGPASSWORD, Buffer2, _countof(Buffer2));
1270             LoadStringW(pgContext->hDllInstance, IDS_COMPUTERLOCKED, Buffer1, _countof(Buffer1));
1271             MessageBoxW(hwndDlg, Buffer2, Buffer1, MB_OK | MB_ICONERROR);
1272         }
1273         else
1274         {
1275             /* Wrong user name */
1276             if (DoAdminUnlock(pgContext, UserName, NULL, Password))
1277             {
1278                 *Action = WLX_SAS_ACTION_UNLOCK_WKSTA;
1279                 res = TRUE;
1280             }
1281             else
1282             {
1283                 LoadStringW(pgContext->hDllInstance, IDS_LOCKEDWRONGUSER, Buffer1, _countof(Buffer1));
1284                 wsprintfW(Buffer2, Buffer1, pgContext->DomainName, pgContext->UserName);
1285                 LoadStringW(pgContext->hDllInstance, IDS_COMPUTERLOCKED, Buffer1, _countof(Buffer1));
1286                 MessageBoxW(hwndDlg, Buffer2, Buffer1, MB_OK | MB_ICONERROR);
1287             }
1288         }
1289     }
1290 
1291     if (UserName != NULL)
1292         HeapFree(GetProcessHeap(), 0, UserName);
1293 
1294     if (Password != NULL)
1295         HeapFree(GetProcessHeap(), 0, Password);
1296 
1297     return res;
1298 }
1299 
1300 
1301 static
1302 INT_PTR
1303 CALLBACK
1304 UnlockDialogProc(
1305     IN HWND hwndDlg,
1306     IN UINT uMsg,
1307     IN WPARAM wParam,
1308     IN LPARAM lParam)
1309 {
1310     PDLG_DATA pDlgData;
1311     INT result = WLX_SAS_ACTION_NONE;
1312 
1313     pDlgData = (PDLG_DATA)GetWindowLongPtrW(hwndDlg, GWLP_USERDATA);
1314 
1315     switch (uMsg)
1316     {
1317         case WM_INITDIALOG:
1318         {
1319             pDlgData = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(DLG_DATA));
1320             if (pDlgData == NULL)
1321                 return FALSE;
1322 
1323             SetWindowLongPtrW(hwndDlg, GWLP_USERDATA, (LONG_PTR)pDlgData);
1324 
1325             pDlgData->pgContext = (PGINA_CONTEXT)lParam;
1326 
1327             SetWelcomeText(hwndDlg);
1328 
1329             SetLockMessage(hwndDlg, IDC_UNLOCK_MESSAGE, pDlgData->pgContext);
1330 
1331             SetDlgItemTextW(hwndDlg, IDC_UNLOCK_USERNAME, pDlgData->pgContext->UserName);
1332             SetFocus(GetDlgItem(hwndDlg, IDC_UNLOCK_PASSWORD));
1333 
1334             if (pDlgData->pgContext->bDisableCAD)
1335                 EnableWindow(GetDlgItem(hwndDlg, IDCANCEL), FALSE);
1336 
1337             /* Load the logo bitmap */
1338             pDlgData->hBitmap = LoadImageW(pDlgData->pgContext->hDllInstance, MAKEINTRESOURCEW(IDI_ROSLOGO), IMAGE_BITMAP, 0, 0, LR_DEFAULTCOLOR);
1339             return TRUE;
1340         }
1341 
1342         case WM_PAINT:
1343         {
1344             PAINTSTRUCT ps;
1345             if (pDlgData->hBitmap)
1346             {
1347                 BeginPaint(hwndDlg, &ps);
1348                 DrawStateW(ps.hdc, NULL, NULL, (LPARAM)pDlgData->hBitmap, (WPARAM)0, 0, 0, 0, 0, DST_BITMAP);
1349                 EndPaint(hwndDlg, &ps);
1350             }
1351             return TRUE;
1352         }
1353         case WM_DESTROY:
1354             DeleteObject(pDlgData->hBitmap);
1355             HeapFree(GetProcessHeap(), 0, pDlgData);
1356             return TRUE;
1357 
1358         case WM_COMMAND:
1359             switch (LOWORD(wParam))
1360             {
1361                 case IDOK:
1362                     if (DoUnlock(hwndDlg, pDlgData->pgContext, &result))
1363                         EndDialog(hwndDlg, result);
1364                     return TRUE;
1365 
1366                 case IDCANCEL:
1367                     EndDialog(hwndDlg, WLX_SAS_ACTION_NONE);
1368                     return TRUE;
1369             }
1370             break;
1371     }
1372 
1373     return FALSE;
1374 }
1375 
1376 
1377 static INT
1378 GUILockedSAS(
1379     IN OUT PGINA_CONTEXT pgContext)
1380 {
1381     int result;
1382 
1383     TRACE("GUILockedSAS()\n");
1384 
1385     result = pgContext->pWlxFuncs->WlxDialogBoxParam(
1386         pgContext->hWlx,
1387         pgContext->hDllInstance,
1388         MAKEINTRESOURCEW(IDD_UNLOCK),
1389         GetDesktopWindow(),
1390         UnlockDialogProc,
1391         (LPARAM)pgContext);
1392     if (result >= WLX_SAS_ACTION_LOGON &&
1393         result <= WLX_SAS_ACTION_SWITCH_CONSOLE)
1394     {
1395         WARN("GUILockedSAS() returns 0x%x\n", result);
1396         return result;
1397     }
1398 
1399     WARN("GUILockedSAS() failed (0x%x)\n", result);
1400     return WLX_SAS_ACTION_NONE;
1401 }
1402 
1403 
1404 static INT_PTR CALLBACK
1405 LockedDialogProc(
1406     IN HWND hwndDlg,
1407     IN UINT uMsg,
1408     IN WPARAM wParam,
1409     IN LPARAM lParam)
1410 {
1411     PDLG_DATA pDlgData;
1412 
1413     pDlgData = (PDLG_DATA)GetWindowLongPtrW(hwndDlg, GWLP_USERDATA);
1414 
1415     switch (uMsg)
1416     {
1417         case WM_INITDIALOG:
1418         {
1419             pDlgData = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(DLG_DATA));
1420             if (pDlgData == NULL)
1421                 return FALSE;
1422 
1423             SetWindowLongPtrW(hwndDlg, GWLP_USERDATA, (LONG_PTR)pDlgData);
1424 
1425             pDlgData->pgContext = (PGINA_CONTEXT)lParam;
1426 
1427             /* Load the logo bitmap */
1428             pDlgData->hBitmap = LoadImageW(pDlgData->pgContext->hDllInstance, MAKEINTRESOURCEW(IDI_ROSLOGO), IMAGE_BITMAP, 0, 0, LR_DEFAULTCOLOR);
1429 
1430             SetWelcomeText(hwndDlg);
1431 
1432             SetLockMessage(hwndDlg, IDC_LOCKED_MESSAGE, pDlgData->pgContext);
1433             return TRUE;
1434         }
1435         case WM_PAINT:
1436         {
1437             PAINTSTRUCT ps;
1438             if (pDlgData->hBitmap)
1439             {
1440                 BeginPaint(hwndDlg, &ps);
1441                 DrawStateW(ps.hdc, NULL, NULL, (LPARAM)pDlgData->hBitmap, (WPARAM)0, 0, 0, 0, 0, DST_BITMAP);
1442                 EndPaint(hwndDlg, &ps);
1443             }
1444             return TRUE;
1445         }
1446         case WM_DESTROY:
1447         {
1448             DeleteObject(pDlgData->hBitmap);
1449             HeapFree(GetProcessHeap(), 0, pDlgData);
1450             return TRUE;
1451         }
1452     }
1453 
1454     return FALSE;
1455 }
1456 
1457 
1458 static VOID
1459 GUIDisplayLockedNotice(
1460     IN OUT PGINA_CONTEXT pgContext)
1461 {
1462     TRACE("GUIdisplayLockedNotice()\n");
1463 
1464     pgContext->pWlxFuncs->WlxDialogBoxParam(
1465         pgContext->hWlx,
1466         pgContext->hDllInstance,
1467         MAKEINTRESOURCEW(IDD_LOCKED),
1468         GetDesktopWindow(),
1469         LockedDialogProc,
1470         (LPARAM)pgContext);
1471 }
1472 
1473 GINA_UI GinaGraphicalUI = {
1474     GUIInitialize,
1475     GUIDisplayStatusMessage,
1476     GUIRemoveStatusMessage,
1477     GUIDisplaySASNotice,
1478     GUILoggedOnSAS,
1479     GUILoggedOutSAS,
1480     GUILockedSAS,
1481     GUIDisplayLockedNotice,
1482 };
1483