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