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 WINE_DEFAULT_DEBUG_CHANNEL(winlogon);
  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(IN PWLSESSION Session,
 217                      IN HDESK hDesktop,
 218                      IN UINT ResourceId)
 219 {
 220     WCHAR StatusMsg[MAX_PATH];
 221
 222     if (Session->Gina.Version < WLX_VERSION_1_3)
 223         return TRUE;
 224
 225     if (Session->SuppressStatus)
 226         return TRUE;
 227
 228     if (LoadStringW(hAppInstance, ResourceId, StatusMsg, MAX_PATH) == 0)
 229         return FALSE;
 230
 231     return Session->Gina.Functions.WlxDisplayStatusMessage(Session->Gina.Context, hDesktop, 0, NULL, StatusMsg);
 232 }
 233
 234
 235 BOOL
 236 RemoveStatusMessage(IN PWLSESSION Session)
 237 {
 238     if (Session->Gina.Version < WLX_VERSION_1_3)
 239         return TRUE;
 240
 241     return Session->Gina.Functions.WlxRemoveStatusMessage(Session->Gina.Context);
 242 }
 243
 244
 245 static
 246 INT_PTR
 247 CALLBACK
 248 GinaLoadFailedWindowProc(IN HWND hwndDlg,
 249                          IN UINT uMsg,
 250                          IN WPARAM wParam,
 251                          IN LPARAM lParam)
 252 {
 253     switch (uMsg)
 254     {
 255         case WM_COMMAND:
 256         {
 257             switch (LOWORD(wParam))
 258             {
 259                 case IDOK:
 260                     EndDialog(hwndDlg, IDOK);
 261                     return TRUE;
 262             }
 263             break;
 264         }
 265
 266         case WM_INITDIALOG:
 267         {
 268             int len;
 269             WCHAR templateText[MAX_PATH], text[MAX_PATH];
 270
 271             len = GetDlgItemTextW(hwndDlg, IDC_GINALOADFAILED, templateText, MAX_PATH);
 272             if (len)
 273             {
 274                 wsprintfW(text, templateText, (LPWSTR)lParam);
 275                 SetDlgItemTextW(hwndDlg, IDC_GINALOADFAILED, text);
 276             }
 277
 278             SetFocus(GetDlgItem(hwndDlg, IDOK));
 279             return TRUE;
 280         }
 281
 282         case WM_CLOSE:
 283         {
 284             EndDialog(hwndDlg, IDCANCEL);
 285             return TRUE;
 286         }
 287     }
 288
 289     return FALSE;
 290 }
 291
 292
 293 int
 294 WINAPI
 295 WinMain(IN HINSTANCE hInstance,
 296         IN HINSTANCE hPrevInstance,
 297         IN LPSTR lpCmdLine,
 298         IN int nShowCmd)
 299 {
 300 #if 0
 301     LSA_STRING ProcessName, PackageName;
 302     HANDLE LsaHandle;
 303     LSA_OPERATIONAL_MODE Mode;
 304     BOOLEAN Old;
 305     ULONG AuthenticationPackage;
 306     NTSTATUS Status;
 307 #endif
 308     ULONG HardErrorResponse;
 309     MSG Msg;
 310
 311     UNREFERENCED_PARAMETER(hPrevInstance);
 312     UNREFERENCED_PARAMETER(lpCmdLine);
 313     UNREFERENCED_PARAMETER(nShowCmd);
 314
 315     hAppInstance = hInstance;
 316
 317     if (!RegisterLogonProcess(GetCurrentProcessId(), TRUE))
 318     {
 319         ERR("WL: Could not register logon process\n");
 320         NtShutdownSystem(ShutdownNoReboot);
 321         ExitProcess(0);
 322     }
 323
 324     WLSession = (PWLSESSION)HeapAlloc(GetProcessHeap(), 0, sizeof(WLSESSION));
 325     if (!WLSession)
 326     {
 327         ERR("WL: Could not allocate memory for winlogon instance\n");
 328         NtRaiseHardError(STATUS_SYSTEM_PROCESS_TERMINATED, 0, 0, NULL, OptionOk, &HardErrorResponse);
 329         ExitProcess(1);
 330     }
 331
 332     ZeroMemory(WLSession, sizeof(WLSESSION));
 333     WLSession->DialogTimeout = 120; /* 2 minutes */
 334
 335     if (!CreateWindowStationAndDesktops(WLSession))
 336     {
 337         ERR("WL: Could not create window station and desktops\n");
 338         NtRaiseHardError(STATUS_SYSTEM_PROCESS_TERMINATED, 0, 0, NULL, OptionOk, &HardErrorResponse);
 339         ExitProcess(1);
 340     }
 341
 342     LockWorkstation(WLSession);
 343
 344     /* Load default keyboard layouts */
 345     if (!InitKeyboardLayouts())
 346     {
 347         ERR("WL: Could not preload keyboard layouts\n");
 348         NtRaiseHardError(STATUS_SYSTEM_PROCESS_TERMINATED, 0, 0, NULL, OptionOk, &HardErrorResponse);
 349         ExitProcess(1);
 350     }
 351
 352     if (!StartServicesManager())
 353     {
 354         ERR("WL: Could not start services.exe\n");
 355         NtRaiseHardError(STATUS_SYSTEM_PROCESS_TERMINATED, 0, 0, NULL, OptionOk, &HardErrorResponse);
 356         ExitProcess(1);
 357     }
 358
 359     if (!StartLsass())
 360     {
 361         ERR("WL: Failed to start lsass.exe service (error %lu)\n", GetLastError());
 362         NtRaiseHardError(STATUS_SYSTEM_PROCESS_TERMINATED, 0, 0, 0, OptionOk, &HardErrorResponse);
 363         ExitProcess(1);
 364     }
 365
 366     /* Load and initialize gina */
 367     if (!GinaInit(WLSession))
 368     {
 369         ERR("WL: Failed to initialize Gina\n");
 370         DialogBoxParam(hAppInstance, MAKEINTRESOURCE(IDD_GINALOADFAILED), GetDesktopWindow(), GinaLoadFailedWindowProc, (LPARAM)L"");
 371         HandleShutdown(WLSession, WLX_SAS_ACTION_SHUTDOWN_REBOOT);
 372         ExitProcess(1);
 373     }
 374
 375     DisplayStatusMessage(WLSession, WLSession->WinlogonDesktop, IDS_REACTOSISSTARTINGUP);
 376
 377     /* Wait for the LSA server */
 378     WaitForLsass();
 379
 380 #if 0
 381     /* Connect to NetLogon service (lsass.exe) */
 382     /* Real winlogon uses "Winlogon" */
 383     RtlInitUnicodeString((PUNICODE_STRING)&ProcessName, L"Winlogon");
 384     Status = LsaRegisterLogonProcess(&ProcessName, &LsaHandle, &Mode);
 385     if (Status == STATUS_PORT_CONNECTION_REFUSED)
 386     {
 387         /* Add the 'SeTcbPrivilege' privilege and try again */
 388         Status = RtlAdjustPrivilege(SE_TCB_PRIVILEGE, TRUE, TRUE, &Old);
 389         if (!NT_SUCCESS(Status))
 390         {
 391             ERR("RtlAdjustPrivilege() failed with error %lu\n", LsaNtStatusToWinError(Status));
 392             return 1;
 393         }
 394
 395         Status = LsaRegisterLogonProcess(&ProcessName, &LsaHandle, &Mode);
 396     }
 397
 398     if (!NT_SUCCESS(Status))
 399     {
 400         ERR("LsaRegisterLogonProcess() failed with error %lu\n", LsaNtStatusToWinError(Status));
 401         return 1;
 402     }
 403
 404     RtlInitUnicodeString((PUNICODE_STRING)&PackageName, MICROSOFT_KERBEROS_NAME_W);
 405     Status = LsaLookupAuthenticationPackage(LsaHandle, &PackageName, &AuthenticationPackage);
 406     if (!NT_SUCCESS(Status))
 407     {
 408         ERR("LsaLookupAuthenticationPackage() failed with error %lu\n", LsaNtStatusToWinError(Status));
 409         LsaDeregisterLogonProcess(LsaHandle);
 410         return 1;
 411     }
 412 #endif
 413
 414     /* Create a hidden window to get SAS notifications */
 415     if (!InitializeSAS(WLSession))
 416     {
 417         ERR("WL: Failed to initialize SAS\n");
 418         ExitProcess(2);
 419     }
 420
 421     //DisplayStatusMessage(Session, Session->WinlogonDesktop, IDS_PREPARENETWORKCONNECTIONS);
 422     //DisplayStatusMessage(Session, Session->WinlogonDesktop, IDS_APPLYINGCOMPUTERSETTINGS);
 423
 424     /* Display logged out screen */
 425     WLSession->LogonState = STATE_LOGGED_OFF;
 426     RemoveStatusMessage(WLSession);
 427
 428     /* Check for pending setup */
 429     if (GetSetupType() != 0)
 430     {
 431         TRACE("WL: Setup mode detected\n");
 432
 433         /* Run setup and reboot when done */
 434         SwitchDesktop(WLSession->ApplicationDesktop);
 435         RunSetup();
 436     }
 437     else
 438         PostMessageW(WLSession->SASWindow, WLX_WM_SAS, WLX_SAS_TYPE_TIMEOUT, 0);
 439
 440     /* Tell kernel that CurrentControlSet is good (needed
 441      * to support Last good known configuration boot) */
 442     NtInitializeRegistry(CM_BOOT_FLAG_ACCEPTED | 1);
 443
 444     /* Message loop for the SAS window */
 445     while (GetMessageW(&Msg, WLSession->SASWindow, 0, 0))
 446     {
 447         TranslateMessage(&Msg);
 448         DispatchMessageW(&Msg);
 449     }
 450
 451     /* We never go there */
 452
 453     return 0;
 454 }