dll/win32/msgina/gui.c

   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
 936         {
 937             TRACE("Other error!\n");
 938             pgContext->pWlxFuncs->WlxMessageBox(pgContext->hWlx,
 939                                                 hwndDlg,
 940                                                 L"Other error!",
 941                                                 L"Logon error",
 942                                                 MB_OK | MB_ICONERROR);
 943             goto done;
 944         }
 945     }
 946     else if (!NT_SUCCESS(Status))
 947     {
 948         TRACE("DoLoginTasks failed! Status 0x%08lx\n", Status);
 949         goto done;
 950     }
 951
 952
 953     if (!CreateProfile(pgContext, UserName, Domain, Password))
 954     {
 955         ERR("Failed to create the profile!\n");
 956         goto done;
 957     }
 958
 959     ZeroMemory(pgContext->Password, sizeof(pgContext->Password));
 960     wcscpy(pgContext->Password, Password);
 961
 962     result = TRUE;
 963
 964 done:
 965     pgContext->bAutoAdminLogon = FALSE;
 966
 967     if (UserName != NULL)
 968         HeapFree(GetProcessHeap(), 0, UserName);
 969
 970     if (Password != NULL)
 971         HeapFree(GetProcessHeap(), 0, Password);
 972
 973     if (Domain != NULL)
 974         HeapFree(GetProcessHeap(), 0, Domain);
 975
 976     return result;
 977 }
 978
 979
 980 static
 981 VOID
 982 SetDomainComboBox(
 983     HWND hwndDomainComboBox,
 984     PGINA_CONTEXT pgContext)
 985 {
 986     WCHAR szComputerName[MAX_COMPUTERNAME_LENGTH + 1];
 987     DWORD dwComputerNameLength;
 988     LONG lIndex = 0;
 989     LONG lFindIndex;
 990
 991     SendMessageW(hwndDomainComboBox, CB_RESETCONTENT, 0, 0);
 992
 993     dwComputerNameLength = _countof(szComputerName);
 994     if (GetComputerNameW(szComputerName, &dwComputerNameLength))
 995     {
 996         lIndex = SendMessageW(hwndDomainComboBox, CB_ADDSTRING, 0, (LPARAM)szComputerName);
 997     }
 998
 999     if (wcslen(pgContext->DomainName) != 0)
1000     {
1001         lFindIndex = SendMessageW(hwndDomainComboBox, CB_FINDSTRINGEXACT, (WPARAM)-1, (LPARAM)pgContext->DomainName);
1002         if (lFindIndex == CB_ERR)
1003         {
1004             lIndex = SendMessageW(hwndDomainComboBox, CB_ADDSTRING, 0, (LPARAM)pgContext->DomainName);
1005         }
1006         else
1007         {
1008             lIndex = lFindIndex;
1009         }
1010     }
1011
1012     SendMessageW(hwndDomainComboBox, CB_SETCURSEL, lIndex, 0);
1013 }
1014
1015
1016 static INT_PTR CALLBACK
1017 LogonDialogProc(
1018     IN HWND hwndDlg,
1019     IN UINT uMsg,
1020     IN WPARAM wParam,
1021     IN LPARAM lParam)
1022 {
1023     PDLG_DATA pDlgData;
1024
1025     pDlgData = (PDLG_DATA)GetWindowLongPtrW(hwndDlg, GWLP_USERDATA);
1026
1027     switch (uMsg)
1028     {
1029         case WM_INITDIALOG:
1030         {
1031             pDlgData = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(DLG_DATA));
1032             if (pDlgData == NULL)
1033                 return FALSE;
1034
1035             SetWindowLongPtrW(hwndDlg, GWLP_USERDATA, (LONG_PTR)pDlgData);
1036
1037             /* FIXME: take care of NoDomainUI */
1038             pDlgData->pgContext = (PGINA_CONTEXT)lParam;
1039
1040             /* Draw the logo bitmap */
1041             pDlgData->hBitmap = LoadImageW(pDlgData->pgContext->hDllInstance, MAKEINTRESOURCEW(IDI_ROSLOGO), IMAGE_BITMAP, 0, 0, LR_DEFAULTCOLOR);
1042
1043             SetWelcomeText(hwndDlg);
1044
1045             if (pDlgData->pgContext->bAutoAdminLogon ||
1046                 !pDlgData->pgContext->bDontDisplayLastUserName)
1047                 SetDlgItemTextW(hwndDlg, IDC_LOGON_USERNAME, pDlgData->pgContext->UserName);
1048
1049             if (pDlgData->pgContext->bAutoAdminLogon)
1050                 SetDlgItemTextW(hwndDlg, IDC_LOGON_PASSWORD, pDlgData->pgContext->Password);
1051
1052             SetDomainComboBox(GetDlgItem(hwndDlg, IDC_LOGON_DOMAIN), pDlgData->pgContext);
1053
1054             if (pDlgData->pgContext->bDisableCAD)
1055                 EnableWindow(GetDlgItem(hwndDlg, IDCANCEL), FALSE);
1056
1057             if (!pDlgData->pgContext->bShutdownWithoutLogon)
1058                 EnableWindow(GetDlgItem(hwndDlg, IDC_LOGON_SHUTDOWN), FALSE);
1059
1060             SetFocus(GetDlgItem(hwndDlg, pDlgData->pgContext->bDontDisplayLastUserName ? IDC_LOGON_USERNAME : IDC_LOGON_PASSWORD));
1061
1062             if (pDlgData->pgContext->bAutoAdminLogon)
1063                 PostMessage(GetDlgItem(hwndDlg, IDOK), BM_CLICK, 0, 0);
1064
1065             return TRUE;
1066         }
1067
1068         case WM_PAINT:
1069         {
1070             PAINTSTRUCT ps;
1071             if (pDlgData->hBitmap)
1072             {
1073                 BeginPaint(hwndDlg, &ps);
1074                 DrawStateW(ps.hdc, NULL, NULL, (LPARAM)pDlgData->hBitmap, (WPARAM)0, 0, 0, 0, 0, DST_BITMAP);
1075                 EndPaint(hwndDlg, &ps);
1076             }
1077             return TRUE;
1078         }
1079
1080         case WM_DESTROY:
1081             DeleteObject(pDlgData->hBitmap);
1082             HeapFree(GetProcessHeap(), 0, pDlgData);
1083             return TRUE;
1084
1085         case WM_COMMAND:
1086             switch (LOWORD(wParam))
1087             {
1088                 case IDOK:
1089                     if (DoLogon(hwndDlg, pDlgData->pgContext))
1090                         EndDialog(hwndDlg, WLX_SAS_ACTION_LOGON);
1091                     return TRUE;
1092
1093                 case IDCANCEL:
1094                     EndDialog(hwndDlg, WLX_SAS_ACTION_NONE);
1095                     return TRUE;
1096
1097                 case IDC_LOGON_SHUTDOWN:
1098                     if (OnShutDown(hwndDlg, pDlgData->pgContext) == IDOK)
1099                         EndDialog(hwndDlg, pDlgData->pgContext->nShutdownAction);
1100                     return TRUE;
1101             }
1102             break;
1103     }
1104
1105     return FALSE;
1106 }
1107
1108
1109 static
1110 INT_PTR
1111 CALLBACK
1112 LegalNoticeDialogProc(
1113     IN HWND hwndDlg,
1114     IN UINT uMsg,
1115     IN WPARAM wParam,
1116     IN LPARAM lParam)
1117 {
1118     PLEGALNOTICEDATA pLegalNotice;
1119
1120     switch (uMsg)
1121     {
1122         case WM_INITDIALOG:
1123             pLegalNotice = (PLEGALNOTICEDATA)lParam;
1124             SetWindowTextW(hwndDlg, pLegalNotice->pszCaption);
1125             SetDlgItemTextW(hwndDlg, IDC_LEGALNOTICE_TEXT, pLegalNotice->pszText);
1126             return TRUE;
1127
1128         case WM_COMMAND:
1129             switch (LOWORD(wParam))
1130             {
1131                 case IDOK:
1132                     EndDialog(hwndDlg, 0);
1133                     return TRUE;
1134
1135                 case IDCANCEL:
1136                     EndDialog(hwndDlg, 0);
1137                     return TRUE;
1138             }
1139             break;
1140     }
1141
1142     return FALSE;
1143 }
1144
1145
1146 static INT
1147 GUILoggedOutSAS(
1148     IN OUT PGINA_CONTEXT pgContext)
1149 {
1150     LEGALNOTICEDATA LegalNotice = {NULL, NULL};
1151     HKEY hKey = NULL;
1152     LONG rc;
1153     int result;
1154
1155     TRACE("GUILoggedOutSAS()\n");
1156
1157     rc = RegOpenKeyExW(HKEY_LOCAL_MACHINE,
1158                        L"SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion\\Winlogon",
1159                        0,
1160                        KEY_QUERY_VALUE,
1161                        &hKey);
1162     if (rc == ERROR_SUCCESS)
1163     {
1164         ReadRegSzValue(hKey,
1165                        L"LegalNoticeCaption",
1166                        &LegalNotice.pszCaption);
1167
1168         ReadRegSzValue(hKey,
1169                        L"LegalNoticeText",
1170                        &LegalNotice.pszText);
1171
1172         RegCloseKey(hKey);
1173     }
1174
1175     if (LegalNotice.pszCaption != NULL && wcslen(LegalNotice.pszCaption) != 0 &&
1176         LegalNotice.pszText != NULL && wcslen(LegalNotice.pszText) != 0)
1177     {
1178         pgContext->pWlxFuncs->WlxDialogBoxParam(pgContext->hWlx,
1179                                                 pgContext->hDllInstance,
1180                                                 MAKEINTRESOURCEW(IDD_LEGALNOTICE),
1181                                                 GetDesktopWindow(),
1182                                                 LegalNoticeDialogProc,
1183                                                 (LPARAM)&LegalNotice);
1184     }
1185
1186     if (LegalNotice.pszCaption != NULL)
1187         HeapFree(GetProcessHeap(), 0, LegalNotice.pszCaption);
1188
1189     if (LegalNotice.pszText != NULL)
1190         HeapFree(GetProcessHeap(), 0, LegalNotice.pszText);
1191
1192     result = pgContext->pWlxFuncs->WlxDialogBoxParam(
1193         pgContext->hWlx,
1194         pgContext->hDllInstance,
1195         MAKEINTRESOURCEW(IDD_LOGON),
1196         GetDesktopWindow(),
1197         LogonDialogProc,
1198         (LPARAM)pgContext);
1199     if (result >= WLX_SAS_ACTION_LOGON &&
1200         result <= WLX_SAS_ACTION_SWITCH_CONSOLE)
1201     {
1202         WARN("WlxLoggedOutSAS() returns 0x%x\n", result);
1203         return result;
1204     }
1205
1206     WARN("WlxDialogBoxParam() failed (0x%x)\n", result);
1207     return WLX_SAS_ACTION_NONE;
1208 }
1209
1210
1211 static VOID
1212 SetLockMessage(HWND hwnd,
1213                INT nDlgItem,
1214                PGINA_CONTEXT pgContext)
1215 {
1216     WCHAR Buffer1[256];
1217     WCHAR Buffer2[256];
1218     WCHAR Buffer3[512];
1219
1220     LoadStringW(pgContext->hDllInstance, IDS_LOCKMSG, Buffer1, _countof(Buffer1));
1221
1222     wsprintfW(Buffer2, L"%s\\%s", pgContext->DomainName, pgContext->UserName);
1223     wsprintfW(Buffer3, Buffer1, Buffer2);
1224
1225     SetDlgItemTextW(hwnd, nDlgItem, Buffer3);
1226 }
1227
1228
1229 static
1230 BOOL
1231 DoUnlock(
1232     IN HWND hwndDlg,
1233     IN PGINA_CONTEXT pgContext,
1234     OUT LPINT Action)
1235 {
1236     WCHAR Buffer1[256];
1237     WCHAR Buffer2[256];
1238     LPWSTR UserName = NULL;
1239     LPWSTR Password = NULL;
1240     BOOL res = FALSE;
1241
1242     if (GetTextboxText(hwndDlg, IDC_UNLOCK_USERNAME, &UserName) && *UserName == '\0')
1243     {
1244         HeapFree(GetProcessHeap(), 0, UserName);
1245         return FALSE;
1246     }
1247
1248     if (GetTextboxText(hwndDlg, IDC_UNLOCK_PASSWORD, &Password))
1249     {
1250         if (UserName != NULL && Password != NULL &&
1251             wcscmp(UserName, pgContext->UserName) == 0 &&
1252             wcscmp(Password, pgContext->Password) == 0)
1253         {
1254             *Action = WLX_SAS_ACTION_UNLOCK_WKSTA;
1255             res = TRUE;
1256         }
1257         else if (wcscmp(UserName, pgContext->UserName) == 0 &&
1258                  wcscmp(Password, pgContext->Password) != 0)
1259         {
1260             /* Wrong Password */
1261             LoadStringW(pgContext->hDllInstance, IDS_LOCKEDWRONGPASSWORD, Buffer2, _countof(Buffer2));
1262             LoadStringW(pgContext->hDllInstance, IDS_COMPUTERLOCKED, Buffer1, _countof(Buffer1));
1263             MessageBoxW(hwndDlg, Buffer2, Buffer1, MB_OK | MB_ICONERROR);
1264         }
1265         else
1266         {
1267             /* Wrong user name */
1268             if (DoAdminUnlock(pgContext, UserName, NULL, Password))
1269             {
1270                 *Action = WLX_SAS_ACTION_UNLOCK_WKSTA;
1271                 res = TRUE;
1272             }
1273             else
1274             {
1275                 LoadStringW(pgContext->hDllInstance, IDS_LOCKEDWRONGUSER, Buffer1, _countof(Buffer1));
1276                 wsprintfW(Buffer2, Buffer1, pgContext->DomainName, pgContext->UserName);
1277                 LoadStringW(pgContext->hDllInstance, IDS_COMPUTERLOCKED, Buffer1, _countof(Buffer1));
1278                 MessageBoxW(hwndDlg, Buffer2, Buffer1, MB_OK | MB_ICONERROR);
1279             }
1280         }
1281     }
1282
1283     if (UserName != NULL)
1284         HeapFree(GetProcessHeap(), 0, UserName);
1285
1286     if (Password != NULL)
1287         HeapFree(GetProcessHeap(), 0, Password);
1288
1289     return res;
1290 }
1291
1292
1293 static
1294 INT_PTR
1295 CALLBACK
1296 UnlockDialogProc(
1297     IN HWND hwndDlg,
1298     IN UINT uMsg,
1299     IN WPARAM wParam,
1300     IN LPARAM lParam)
1301 {
1302     PDLG_DATA pDlgData;
1303     INT result = WLX_SAS_ACTION_NONE;
1304
1305     pDlgData = (PDLG_DATA)GetWindowLongPtrW(hwndDlg, GWLP_USERDATA);
1306
1307     switch (uMsg)
1308     {
1309         case WM_INITDIALOG:
1310         {
1311             pDlgData = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(DLG_DATA));
1312             if (pDlgData == NULL)
1313                 return FALSE;
1314
1315             SetWindowLongPtrW(hwndDlg, GWLP_USERDATA, (LONG_PTR)pDlgData);
1316
1317             pDlgData->pgContext = (PGINA_CONTEXT)lParam;
1318
1319             SetWelcomeText(hwndDlg);
1320
1321             SetLockMessage(hwndDlg, IDC_UNLOCK_MESSAGE, pDlgData->pgContext);
1322
1323             SetDlgItemTextW(hwndDlg, IDC_UNLOCK_USERNAME, pDlgData->pgContext->UserName);
1324             SetFocus(GetDlgItem(hwndDlg, IDC_UNLOCK_PASSWORD));
1325
1326             if (pDlgData->pgContext->bDisableCAD)
1327                 EnableWindow(GetDlgItem(hwndDlg, IDCANCEL), FALSE);
1328
1329             /* Load the logo bitmap */
1330             pDlgData->hBitmap = LoadImageW(pDlgData->pgContext->hDllInstance, MAKEINTRESOURCEW(IDI_ROSLOGO), IMAGE_BITMAP, 0, 0, LR_DEFAULTCOLOR);
1331             return TRUE;
1332         }
1333
1334         case WM_PAINT:
1335         {
1336             PAINTSTRUCT ps;
1337             if (pDlgData->hBitmap)
1338             {
1339                 BeginPaint(hwndDlg, &ps);
1340                 DrawStateW(ps.hdc, NULL, NULL, (LPARAM)pDlgData->hBitmap, (WPARAM)0, 0, 0, 0, 0, DST_BITMAP);
1341                 EndPaint(hwndDlg, &ps);
1342             }
1343             return TRUE;
1344         }
1345         case WM_DESTROY:
1346             DeleteObject(pDlgData->hBitmap);
1347             HeapFree(GetProcessHeap(), 0, pDlgData);
1348             return TRUE;
1349
1350         case WM_COMMAND:
1351             switch (LOWORD(wParam))
1352             {
1353                 case IDOK:
1354                     if (DoUnlock(hwndDlg, pDlgData->pgContext, &result))
1355                         EndDialog(hwndDlg, result);
1356                     return TRUE;
1357
1358                 case IDCANCEL:
1359                     EndDialog(hwndDlg, WLX_SAS_ACTION_NONE);
1360                     return TRUE;
1361             }
1362             break;
1363     }
1364
1365     return FALSE;
1366 }
1367
1368
1369 static INT
1370 GUILockedSAS(
1371     IN OUT PGINA_CONTEXT pgContext)
1372 {
1373     int result;
1374
1375     TRACE("GUILockedSAS()\n");
1376
1377     result = pgContext->pWlxFuncs->WlxDialogBoxParam(
1378         pgContext->hWlx,
1379         pgContext->hDllInstance,
1380         MAKEINTRESOURCEW(IDD_UNLOCK),
1381         GetDesktopWindow(),
1382         UnlockDialogProc,
1383         (LPARAM)pgContext);
1384     if (result >= WLX_SAS_ACTION_LOGON &&
1385         result <= WLX_SAS_ACTION_SWITCH_CONSOLE)
1386     {
1387         WARN("GUILockedSAS() returns 0x%x\n", result);
1388         return result;
1389     }
1390
1391     WARN("GUILockedSAS() failed (0x%x)\n", result);
1392     return WLX_SAS_ACTION_NONE;
1393 }
1394
1395
1396 static INT_PTR CALLBACK
1397 LockedDialogProc(
1398     IN HWND hwndDlg,
1399     IN UINT uMsg,
1400     IN WPARAM wParam,
1401     IN LPARAM lParam)
1402 {
1403     PDLG_DATA pDlgData;
1404
1405     pDlgData = (PDLG_DATA)GetWindowLongPtrW(hwndDlg, GWLP_USERDATA);
1406
1407     switch (uMsg)
1408     {
1409         case WM_INITDIALOG:
1410         {
1411             pDlgData = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(DLG_DATA));
1412             if (pDlgData == NULL)
1413                 return FALSE;
1414
1415             SetWindowLongPtrW(hwndDlg, GWLP_USERDATA, (LONG_PTR)pDlgData);
1416
1417             pDlgData->pgContext = (PGINA_CONTEXT)lParam;
1418
1419             /* Load the logo bitmap */
1420             pDlgData->hBitmap = LoadImageW(pDlgData->pgContext->hDllInstance, MAKEINTRESOURCEW(IDI_ROSLOGO), IMAGE_BITMAP, 0, 0, LR_DEFAULTCOLOR);
1421
1422             SetWelcomeText(hwndDlg);
1423
1424             SetLockMessage(hwndDlg, IDC_LOCKED_MESSAGE, pDlgData->pgContext);
1425             return TRUE;
1426         }
1427         case WM_PAINT:
1428         {
1429             PAINTSTRUCT ps;
1430             if (pDlgData->hBitmap)
1431             {
1432                 BeginPaint(hwndDlg, &ps);
1433                 DrawStateW(ps.hdc, NULL, NULL, (LPARAM)pDlgData->hBitmap, (WPARAM)0, 0, 0, 0, 0, DST_BITMAP);
1434                 EndPaint(hwndDlg, &ps);
1435             }
1436             return TRUE;
1437         }
1438         case WM_DESTROY:
1439         {
1440             DeleteObject(pDlgData->hBitmap);
1441             HeapFree(GetProcessHeap(), 0, pDlgData);
1442             return TRUE;
1443         }
1444     }
1445
1446     return FALSE;
1447 }
1448
1449
1450 static VOID
1451 GUIDisplayLockedNotice(
1452     IN OUT PGINA_CONTEXT pgContext)
1453 {
1454     TRACE("GUIdisplayLockedNotice()\n");
1455
1456     pgContext->pWlxFuncs->WlxDialogBoxParam(
1457         pgContext->hWlx,
1458         pgContext->hDllInstance,
1459         MAKEINTRESOURCEW(IDD_LOCKED),
1460         GetDesktopWindow(),
1461         LockedDialogProc,
1462         (LPARAM)pgContext);
1463 }
1464
1465 GINA_UI GinaGraphicalUI = {
1466     GUIInitialize,
1467     GUIDisplayStatusMessage,
1468     GUIRemoveStatusMessage,
1469     GUIDisplaySASNotice,
1470     GUILoggedOnSAS,
1471     GUILoggedOutSAS,
1472     GUILockedSAS,
1473     GUIDisplayLockedNotice,
1474 };