reactos/base/system/winlogon/winlogon.c

   1 /*
   2  * COPYRIGHT:       See COPYING in the top level directory
   3  * PROJECT:         ReactOS Winlogon
   4  * FILE:            base/system/winlogon/winlogon.c
   5  * PURPOSE:         Logon
   6  * PROGRAMMERS:     Thomas Weidenmueller (w3seek@users.sourceforge.net)
   7  *                  Filip Navara
   8  *                  Hervé Poussineau (hpoussin@reactos.org)
   9  */
  10
  11 /* INCLUDES *****************************************************************/
  12
  13 #include "winlogon.h"
  14
  15 #include <ndk/cmfuncs.h>
  16
  17 /* GLOBALS ******************************************************************/
  18
  19 HINSTANCE hAppInstance;
  20 PWLSESSION WLSession = NULL;
  21
  22 /* FUNCTIONS *****************************************************************/
  23
  24 static
  25 BOOL
  26 StartServicesManager(VOID)
  27 {
  28     STARTUPINFOW StartupInfo;
  29     PROCESS_INFORMATION ProcessInformation;
  30     LPCWSTR ServiceString = L"services.exe";
  31     BOOL res;
  32
  33     /* Start the service control manager (services.exe) */
  34     ZeroMemory(&StartupInfo, sizeof(STARTUPINFOW));
  35     StartupInfo.cb = sizeof(StartupInfo);
  36     StartupInfo.lpReserved = NULL;
  37     StartupInfo.lpDesktop = NULL;
  38     StartupInfo.lpTitle = NULL;
  39     StartupInfo.dwFlags = 0;
  40     StartupInfo.cbReserved2 = 0;
  41     StartupInfo.lpReserved2 = 0;
  42
  43     TRACE("WL: Creating new process - %S\n", ServiceString);
  44
  45     res = CreateProcessW(ServiceString,
  46                          NULL,
  47                          NULL,
  48                          NULL,
  49                          FALSE,
  50                          DETACHED_PROCESS,
  51                          NULL,
  52                          NULL,
  53                          &StartupInfo,
  54                          &ProcessInformation);
  55     if (!res)
  56     {
  57         ERR("WL: Failed to execute services (error %lu)\n", GetLastError());
  58         return FALSE;
  59     }
  60
  61     TRACE("WL: Created new process - %S\n", ServiceString);
  62
  63     CloseHandle(ProcessInformation.hThread);
  64     CloseHandle(ProcessInformation.hProcess);
  65
  66     TRACE("WL: StartServicesManager() done.\n");
  67
  68     return TRUE;
  69 }
  70
  71
  72 static
  73 BOOL
  74 StartLsass(VOID)
  75 {
  76     STARTUPINFOW StartupInfo;
  77     PROCESS_INFORMATION ProcessInformation;
  78     LPCWSTR ServiceString = L"lsass.exe";
  79     BOOL res;
  80
  81     /* Start the local security authority subsystem (lsass.exe) */
  82     ZeroMemory(&StartupInfo, sizeof(STARTUPINFOW));
  83     StartupInfo.cb = sizeof(StartupInfo);
  84     StartupInfo.lpReserved = NULL;
  85     StartupInfo.lpDesktop = NULL;
  86     StartupInfo.lpTitle = NULL;
  87     StartupInfo.dwFlags = 0;
  88     StartupInfo.cbReserved2 = 0;
  89     StartupInfo.lpReserved2 = 0;
  90
  91     TRACE("WL: Creating new process - %S\n", ServiceString);
  92
  93     res = CreateProcessW(ServiceString,
  94                          NULL,
  95                          NULL,
  96                          NULL,
  97                          FALSE,
  98                          DETACHED_PROCESS,
  99                          NULL,
 100                          NULL,
 101                          &StartupInfo,
 102                          &ProcessInformation);
 103
 104     TRACE("WL: Created new process - %S\n", ServiceString);
 105
 106     CloseHandle(ProcessInformation.hThread);
 107     CloseHandle(ProcessInformation.hProcess);
 108
 109     return res;
 110 }
 111
 112
 113 static
 114 VOID
 115 WaitForLsass(VOID)
 116 {
 117     HANDLE hEvent;
 118     DWORD dwError;
 119
 120     hEvent = CreateEventW(NULL,
 121                           TRUE,
 122                           FALSE,
 123                           L"LSA_RPC_SERVER_ACTIVE");
 124     if (hEvent == NULL)
 125     {
 126         dwError = GetLastError();
 127         TRACE("WL: Failed to create the notication event (Error %lu)\n", dwError);
 128
 129         if (dwError == ERROR_ALREADY_EXISTS)
 130         {
 131             hEvent = OpenEventW(SYNCHRONIZE,
 132                                 FALSE,
 133                                 L"LSA_RPC_SERVER_ACTIVE");
 134             if (hEvent == NULL)
 135             {
 136                ERR("WL: Could not open the notification event (Error %lu)\n", GetLastError());
 137                return;
 138             }
 139         }
 140     }
 141
 142     TRACE("WL: Wait for the LSA server!\n");
 143     WaitForSingleObject(hEvent, INFINITE);
 144     TRACE("WL: LSA server running!\n");
 145
 146     CloseHandle(hEvent);
 147 }
 148
 149
 150 static
 151 BOOL
 152 InitKeyboardLayouts(VOID)
 153 {
 154     WCHAR wszKeyName[12], wszKLID[10];
 155     DWORD dwSize = sizeof(wszKLID), dwType, i = 1;
 156     HKEY hKey;
 157     UINT Flags;
 158     BOOL bRet = FALSE;
 159
 160     /* Open registry key with preloaded layouts */
 161     if (RegOpenKeyExW(HKEY_CURRENT_USER, L"Keyboard Layout\\Preload", 0, KEY_READ, &hKey) == ERROR_SUCCESS)
 162     {
 163         while(TRUE)
 164         {
 165             /* Read values with integer names only */
 166             swprintf(wszKeyName, L"%d", i++);
 167             if (RegQueryValueExW(hKey, wszKeyName, NULL, &dwType, (LPBYTE)wszKLID, &dwSize) != ERROR_SUCCESS)
 168             {
 169                 /* There is no more entries */
 170                 break;
 171             }
 172
 173             /* Only REG_SZ values are valid */
 174             if (dwType != REG_SZ)
 175             {
 176                 ERR("Wrong type: %ws!\n", wszKLID);
 177                 continue;
 178             }
 179
 180             /* Load keyboard layout with given locale id */
 181             Flags = KLF_SUBSTITUTE_OK;
 182             if (i > 1)
 183                 Flags |= KLF_NOTELLSHELL|KLF_REPLACELANG;
 184             else // First layout
 185                 Flags |= KLF_ACTIVATE; // |0x40000000
 186             if (!LoadKeyboardLayoutW(wszKLID, Flags))
 187             {
 188                 ERR("LoadKeyboardLayoutW(%ws) failed!\n", wszKLID);
 189                 continue;
 190             }
 191             else
 192             {
 193                 /* We loaded at least one layout - success */
 194                 bRet = TRUE;
 195             }
 196         }
 197
 198         /* Close the key now */
 199         RegCloseKey(hKey);
 200     }
 201     else
 202         WARN("RegOpenKeyExW(Keyboard Layout\\Preload) failed!\n");
 203
 204     if (!bRet)
 205     {
 206         /* If we failed, load US keyboard layout */
 207         if (LoadKeyboardLayoutW(L"00000409", 0x04090409))
 208             bRet = TRUE;
 209     }
 210
 211     return bRet;
 212 }
 213
 214
 215 BOOL
 216 DisplayStatusMessage(
 217      IN PWLSESSION Session,
 218      IN HDESK hDesktop,
 219      IN UINT ResourceId)
 220 {
 221     WCHAR StatusMsg[MAX_PATH];
 222
 223     if (Session->Gina.Version < WLX_VERSION_1_3)
 224         return TRUE;
 225
 226     if (Session->SuppressStatus)
 227         return TRUE;
 228
 229     if (LoadStringW(hAppInstance, ResourceId, StatusMsg, MAX_PATH) == 0)
 230         return FALSE;
 231
 232     return Session->Gina.Functions.WlxDisplayStatusMessage(Session->Gina.Context, hDesktop, 0, NULL, StatusMsg);
 233 }
 234
 235
 236 BOOL
 237 RemoveStatusMessage(
 238     IN PWLSESSION Session)
 239 {
 240     if (Session->Gina.Version < WLX_VERSION_1_3)
 241         return TRUE;
 242
 243     return Session->Gina.Functions.WlxRemoveStatusMessage(Session->Gina.Context);
 244 }
 245
 246
 247 static
 248 INT_PTR
 249 CALLBACK
 250 GinaLoadFailedWindowProc(
 251     IN HWND hwndDlg,
 252     IN UINT uMsg,
 253     IN WPARAM wParam,
 254     IN LPARAM lParam)
 255 {
 256     switch (uMsg)
 257     {
 258         case WM_COMMAND:
 259         {
 260             switch (LOWORD(wParam))
 261             {
 262                 case IDOK:
 263                     EndDialog(hwndDlg, IDOK);
 264                     return TRUE;
 265             }
 266             break;
 267         }
 268
 269         case WM_INITDIALOG:
 270         {
 271             int len;
 272             WCHAR templateText[MAX_PATH], text[MAX_PATH];
 273
 274             len = GetDlgItemTextW(hwndDlg, IDC_GINALOADFAILED, templateText, MAX_PATH);
 275             if (len)
 276             {
 277                 wsprintfW(text, templateText, (LPWSTR)lParam);
 278                 SetDlgItemTextW(hwndDlg, IDC_GINALOADFAILED, text);
 279             }
 280
 281             SetFocus(GetDlgItem(hwndDlg, IDOK));
 282             return TRUE;
 283         }
 284
 285         case WM_CLOSE:
 286         {
 287             EndDialog(hwndDlg, IDCANCEL);
 288             return TRUE;
 289         }
 290     }
 291
 292     return FALSE;
 293 }
 294
 295
 296 int
 297 WINAPI
 298 WinMain(
 299     IN HINSTANCE hInstance,
 300     IN HINSTANCE hPrevInstance,
 301     IN LPSTR lpCmdLine,
 302     IN int nShowCmd)
 303 {
 304 #if 0
 305     LSA_STRING ProcessName, PackageName;
 306     HANDLE LsaHandle;
 307     LSA_OPERATIONAL_MODE Mode;
 308     BOOLEAN Old;
 309     ULONG AuthenticationPackage;
 310     NTSTATUS Status;
 311 #endif
 312     ULONG HardErrorResponse;
 313     MSG Msg;
 314
 315     UNREFERENCED_PARAMETER(hPrevInstance);
 316     UNREFERENCED_PARAMETER(lpCmdLine);
 317     UNREFERENCED_PARAMETER(nShowCmd);
 318
 319     hAppInstance = hInstance;
 320
 321     if (!RegisterLogonProcess(GetCurrentProcessId(), TRUE))
 322     {
 323         ERR("WL: Could not register logon process\n");
 324         NtShutdownSystem(ShutdownNoReboot);
 325         ExitProcess(0);
 326     }
 327
 328     WLSession = (PWLSESSION)HeapAlloc(GetProcessHeap(), 0, sizeof(WLSESSION));
 329     if (!WLSession)
 330     {
 331         ERR("WL: Could not allocate memory for winlogon instance\n");
 332         NtRaiseHardError(STATUS_SYSTEM_PROCESS_TERMINATED, 0, 0, NULL, OptionOk, &HardErrorResponse);
 333         ExitProcess(1);
 334     }
 335
 336     ZeroMemory(WLSession, sizeof(WLSESSION));
 337     WLSession->DialogTimeout = 120; /* 2 minutes */
 338
 339     /* Initialize the dialog tracking list */
 340     InitDialogListHead();
 341
 342     if (!CreateWindowStationAndDesktops(WLSession))
 343     {
 344         ERR("WL: Could not create window station and desktops\n");
 345         NtRaiseHardError(STATUS_SYSTEM_PROCESS_TERMINATED, 0, 0, NULL, OptionOk, &HardErrorResponse);
 346         ExitProcess(1);
 347     }
 348
 349     LockWorkstation(WLSession);
 350
 351     /* Load default keyboard layouts */
 352     if (!InitKeyboardLayouts())
 353     {
 354         ERR("WL: Could not preload keyboard layouts\n");
 355         NtRaiseHardError(STATUS_SYSTEM_PROCESS_TERMINATED, 0, 0, NULL, OptionOk, &HardErrorResponse);
 356         ExitProcess(1);
 357     }
 358
 359     if (!StartServicesManager())
 360     {
 361         ERR("WL: Could not start services.exe\n");
 362         NtRaiseHardError(STATUS_SYSTEM_PROCESS_TERMINATED, 0, 0, NULL, OptionOk, &HardErrorResponse);
 363         ExitProcess(1);
 364     }
 365
 366     if (!StartLsass())
 367     {
 368         ERR("WL: Failed to start lsass.exe service (error %lu)\n", GetLastError());
 369         NtRaiseHardError(STATUS_SYSTEM_PROCESS_TERMINATED, 0, 0, 0, OptionOk, &HardErrorResponse);
 370         ExitProcess(1);
 371     }
 372
 373     /* Wait for the LSA server */
 374     WaitForLsass();
 375
 376     /* Load and initialize gina */
 377     if (!GinaInit(WLSession))
 378     {
 379         ERR("WL: Failed to initialize Gina\n");
 380         DialogBoxParam(hAppInstance, MAKEINTRESOURCE(IDD_GINALOADFAILED), GetDesktopWindow(), GinaLoadFailedWindowProc, (LPARAM)L"");
 381         HandleShutdown(WLSession, WLX_SAS_ACTION_SHUTDOWN_REBOOT);
 382         ExitProcess(1);
 383     }
 384
 385     DisplayStatusMessage(WLSession, WLSession->WinlogonDesktop, IDS_REACTOSISSTARTINGUP);
 386
 387 #if 0
 388     /* Connect to NetLogon service (lsass.exe) */
 389     /* Real winlogon uses "Winlogon" */
 390     RtlInitUnicodeString((PUNICODE_STRING)&ProcessName, L"Winlogon");
 391     Status = LsaRegisterLogonProcess(&ProcessName, &LsaHandle, &Mode);
 392     if (Status == STATUS_PORT_CONNECTION_REFUSED)
 393     {
 394         /* Add the 'SeTcbPrivilege' privilege and try again */
 395         Status = RtlAdjustPrivilege(SE_TCB_PRIVILEGE, TRUE, TRUE, &Old);
 396         if (!NT_SUCCESS(Status))
 397         {
 398             ERR("RtlAdjustPrivilege() failed with error %lu\n", LsaNtStatusToWinError(Status));
 399             return 1;
 400         }
 401
 402         Status = LsaRegisterLogonProcess(&ProcessName, &LsaHandle, &Mode);
 403     }
 404
 405     if (!NT_SUCCESS(Status))
 406     {
 407         ERR("LsaRegisterLogonProcess() failed with error %lu\n", LsaNtStatusToWinError(Status));
 408         return 1;
 409     }
 410
 411     RtlInitUnicodeString((PUNICODE_STRING)&PackageName, MICROSOFT_KERBEROS_NAME_W);
 412     Status = LsaLookupAuthenticationPackage(LsaHandle, &PackageName, &AuthenticationPackage);
 413     if (!NT_SUCCESS(Status))
 414     {
 415         ERR("LsaLookupAuthenticationPackage() failed with error %lu\n", LsaNtStatusToWinError(Status));
 416         LsaDeregisterLogonProcess(LsaHandle);
 417         return 1;
 418     }
 419 #endif
 420
 421     /* Create a hidden window to get SAS notifications */
 422     if (!InitializeSAS(WLSession))
 423     {
 424         ERR("WL: Failed to initialize SAS\n");
 425         ExitProcess(2);
 426     }
 427
 428     //DisplayStatusMessage(Session, Session->WinlogonDesktop, IDS_PREPARENETWORKCONNECTIONS);
 429     //DisplayStatusMessage(Session, Session->WinlogonDesktop, IDS_APPLYINGCOMPUTERSETTINGS);
 430
 431     /* Display logged out screen */
 432     WLSession->LogonState = STATE_INIT;
 433     RemoveStatusMessage(WLSession);
 434
 435     /* Check for pending setup */
 436     if (GetSetupType() != 0)
 437     {
 438         TRACE("WL: Setup mode detected\n");
 439
 440         /* Run setup and reboot when done */
 441         SwitchDesktop(WLSession->ApplicationDesktop);
 442         RunSetup();
 443     }
 444     else
 445         PostMessageW(WLSession->SASWindow, WLX_WM_SAS, WLX_SAS_TYPE_CTRL_ALT_DEL, 0);
 446
 447     /* Tell kernel that CurrentControlSet is good (needed
 448      * to support Last good known configuration boot) */
 449     NtInitializeRegistry(CM_BOOT_FLAG_ACCEPTED | 1);
 450
 451     /* Message loop for the SAS window */
 452     while (GetMessageW(&Msg, WLSession->SASWindow, 0, 0))
 453     {
 454         TranslateMessage(&Msg);
 455         DispatchMessageW(&Msg);
 456     }
 457
 458     /* We never go there */
 459
 460     return 0;
 461 }