userinit.c File Reference

#include "userinit.h"
#include <userenv.h>

Include dependency graph for userinit.c:

Go to the source code of this file.

Macros
#define	CMP_MAGIC   0x01234567

Typedefs
typedef DWORD(WINAPI *	PCMP_REPORT_LOGON) (DWORD, DWORD)

Functions
LONG	ReadRegSzKey (IN HKEY hKey, IN LPCWSTR pszKey, OUT LPWSTR *pValue)
static BOOL	IsConsoleShell (VOID)
static BOOL	GetShell (OUT WCHAR *CommandLine, IN HKEY hRootKey)
static BOOL	StartProcess (_In_ PCWSTR CommandLine, _In_opt_ PVOID pEnvironment)
static BOOL	StartShell (_In_opt_ PVOID pEnvironment)
static COLORREF	StrToColorref (IN LPWSTR lpszCol)
static VOID	SetUserSysColors (VOID)
static VOID	SetUserWallpaper (VOID)
static VOID	SetUserSettings (VOID)
static VOID	NotifyLogon (VOID)
static BOOL	StartInstaller (VOID)
static BOOL	EnablePrivilege (LPCWSTR lpszPrivilegeName, BOOL bEnablePrivilege)
int WINAPI	wWinMain (IN HINSTANCE hInst, IN HINSTANCE hPrevInstance, IN LPWSTR lpszCmdLine, IN int nCmdShow)

Variables
HINSTANCE	hInstance
const WCHAR	g_RegColorNames [][32]

Macro Definition Documentation

CMP_MAGIC

#define CMP_MAGIC   0x01234567

Definition at line 30 of file userinit.c.

Typedef Documentation

PCMP_REPORT_LOGON

typedef DWORD(WINAPI * PCMP_REPORT_LOGON) (DWORD, DWORD)

Definition at line 471 of file userinit.c.

Function Documentation

EnablePrivilege()

static BOOL EnablePrivilege ( LPCWSTR  lpszPrivilegeName, BOOL  bEnablePrivilege  )

static

Definition at line 514 of file userinit.c.

{
    BOOL   Success;
    HANDLE hToken;
    TOKEN_PRIVILEGES tp;
 
    Success = OpenProcessToken(GetCurrentProcess(),
                               TOKEN_ADJUST_PRIVILEGES,
                               &hToken);
    if (!Success) return Success;
 
    Success = LookupPrivilegeValueW(NULL,
                                    lpszPrivilegeName,
                                    &tp.Privileges[0].Luid);
    if (!Success) goto Quit;
 
    tp.PrivilegeCount = 1;
    tp.Privileges[0].Attributes = (bEnablePrivilege ? SE_PRIVILEGE_ENABLED : 0);
 
    Success = AdjustTokenPrivileges(hToken, FALSE, &tp, 0, NULL, NULL);
 
Quit:
    CloseHandle(hToken);
    return Success;
}

Referenced by wWinMain().

GetShell()

static BOOL GetShell ( OUT WCHAR *  CommandLine, IN HKEY  hRootKey  )

static

Definition at line 133 of file userinit.c.

{
    HKEY hKey;
    DWORD Type, Size;
    WCHAR Shell[MAX_PATH];
    BOOL ConsoleShell = IsConsoleShell();
    LONG rc;
 
    rc = RegOpenKeyExW(hRootKey, L"Software\\Microsoft\\Windows NT\\CurrentVersion\\Winlogon",
                       0, KEY_QUERY_VALUE, &hKey);
    if (rc != ERROR_SUCCESS)
    {
        WARN("RegOpenKeyEx() failed with error %lu\n", rc);
        return FALSE;
    }
 
    Size = sizeof(Shell);
    rc = RegQueryValueExW(hKey,
                          ConsoleShell ? L"ConsoleShell" : L"Shell",
                          NULL,
                          &Type,
                          (LPBYTE)Shell,
                          &Size);
    RegCloseKey(hKey);
 
    if (rc != ERROR_SUCCESS)
    {
        WARN("RegQueryValueEx() failed with error %lu\n", rc);
        return FALSE;
    }
 
    if ((Type == REG_SZ) || (Type == REG_EXPAND_SZ))
    {
        TRACE("Found command line %s\n", debugstr_w(Shell));
        wcscpy(CommandLine, Shell);
        return TRUE;
    }
    else
    {
        WARN("Wrong type %lu (expected %u or %u)\n", Type, REG_SZ, REG_EXPAND_SZ);
        return FALSE;
    }
}

Referenced by StartShell().

IsConsoleShell()

static BOOL IsConsoleShell ( VOID  )

static

Definition at line 81 of file userinit.c.

{
    HKEY ControlKey = NULL;
    LPWSTR SystemStartOptions = NULL;
    LPWSTR CurrentOption, NextOption; /* Pointers into SystemStartOptions */
    LONG rc;
    BOOL ret = FALSE;
 
    rc = RegOpenKeyExW(HKEY_LOCAL_MACHINE,
                       REGSTR_PATH_CURRENT_CONTROL_SET,
                       0,
                       KEY_QUERY_VALUE,
                       &ControlKey);
    if (rc != ERROR_SUCCESS)
    {
        WARN("RegOpenKeyEx() failed with error %lu\n", rc);
        goto cleanup;
    }
 
    rc = ReadRegSzKey(ControlKey, L"SystemStartOptions", &SystemStartOptions);
    if (rc != ERROR_SUCCESS)
    {
        WARN("ReadRegSzKey() failed with error %lu\n", rc);
        goto cleanup;
    }
 
    /* Check for CONSOLE switch in SystemStartOptions */
    CurrentOption = SystemStartOptions;
    while (CurrentOption)
    {
        NextOption = wcschr(CurrentOption, L' ');
        if (NextOption)
            *NextOption = L'\0';
        if (_wcsicmp(CurrentOption, L"CONSOLE") == 0)
        {
            TRACE("Found 'CONSOLE' boot option\n");
            ret = TRUE;
            goto cleanup;
        }
        CurrentOption = NextOption ? NextOption + 1 : NULL;
    }
 
cleanup:
    if (ControlKey != NULL)
        RegCloseKey(ControlKey);
    HeapFree(GetProcessHeap(), 0, SystemStartOptions);
 
    TRACE("IsConsoleShell() returning %u\n", ret);
    return ret;
}

Referenced by GetShell(), and StartShell().

NotifyLogon()

static VOID NotifyLogon ( VOID  )

static

Definition at line 474 of file userinit.c.

{
    HINSTANCE hModule;
    PCMP_REPORT_LOGON CMP_Report_LogOn;
 
    hModule = LoadLibraryW(L"setupapi.dll");
    if (!hModule)
    {
        WARN("LoadLibrary() failed with error %lu\n", GetLastError());
        return;
    }
 
    CMP_Report_LogOn = (PCMP_REPORT_LOGON)GetProcAddress(hModule, "CMP_Report_LogOn");
    if (CMP_Report_LogOn)
        CMP_Report_LogOn(CMP_MAGIC, GetCurrentProcessId());
    else
        WARN("GetProcAddress() failed\n");
 
    FreeLibrary(hModule);
}

Referenced by wWinMain().

ReadRegSzKey()

LONG ReadRegSzKey	(	IN HKEY	hKey,
		IN LPCWSTR	pszKey,
		OUT LPWSTR *	pValue
	)

Definition at line 39 of file userinit.c.

{
    LONG rc;
    DWORD dwType;
    DWORD cbData = 0;
    LPWSTR Value;
 
    rc = RegQueryValueExW(hKey, pszKey, NULL, &dwType, NULL, &cbData);
    if (rc != ERROR_SUCCESS)
    {
        WARN("RegQueryValueEx(%s) failed with error %lu\n", debugstr_w(pszKey), rc);
        return rc;
    }
    if (dwType != REG_SZ)
    {
        WARN("Wrong registry data type (%u vs %u)\n", dwType, REG_SZ);
        return ERROR_FILE_NOT_FOUND;
    }
    Value = (WCHAR*) HeapAlloc(GetProcessHeap(), 0, cbData + sizeof(WCHAR));
    if (!Value)
    {
        WARN("No memory\n");
        return ERROR_NOT_ENOUGH_MEMORY;
    }
    rc = RegQueryValueExW(hKey, pszKey, NULL, NULL, (LPBYTE)Value, &cbData);
    if (rc != ERROR_SUCCESS)
    {
        WARN("RegQueryValueEx(%s) failed with error %lu\n", debugstr_w(pszKey), rc);
        HeapFree(GetProcessHeap(), 0, Value);
        return rc;
    }
    /* NULL-terminate the string */
    Value[cbData / sizeof(WCHAR)] = L'\0';
 
    *pValue = Value;
    return ERROR_SUCCESS;
}

Referenced by IsConsoleShell().

SetUserSettings()

static VOID SetUserSettings ( VOID  )

static

Definition at line 464 of file userinit.c.

{
    UpdatePerUserSystemParameters(1, TRUE);
    SetUserSysColors();
    SetUserWallpaper();
}

Referenced by wWinMain().

SetUserSysColors()

static VOID SetUserSysColors ( VOID  )

static

Definition at line 385 of file userinit.c.

{
    HKEY hKey;
    INT i;
    WCHAR szColor[25];
    DWORD Type, Size;
    COLORREF crColor;
    LONG rc;
 
    rc = RegOpenKeyExW(HKEY_CURRENT_USER, REGSTR_PATH_COLORS,
                       0, KEY_QUERY_VALUE, &hKey);
    if (rc != ERROR_SUCCESS)
    {
        WARN("RegOpenKeyEx() failed with error %lu\n", rc);
        return;
    }
 
    for (i = 0; i < ARRAYSIZE(g_RegColorNames); i++)
    {
        Size = sizeof(szColor);
        rc = RegQueryValueExW(hKey, g_RegColorNames[i], NULL, &Type,
                              (LPBYTE)szColor, &Size);
        if (rc == ERROR_SUCCESS && Type == REG_SZ)
        {
            crColor = StrToColorref(szColor);
            SetSysColors(1, &i, &crColor);
        }
        else
        {
            WARN("RegQueryValueEx(%s) failed with error %lu\n",
                debugstr_w(g_RegColorNames[i]), rc);
        }
    }
 
    RegCloseKey(hKey);
}

Referenced by SetUserSettings().

SetUserWallpaper()

static VOID SetUserWallpaper ( VOID  )

static

Definition at line 423 of file userinit.c.

{
    HKEY hKey;
    DWORD Type, Size;
    WCHAR szWallpaper[MAX_PATH + 1];
    LONG rc;
 
    rc = RegOpenKeyExW(HKEY_CURRENT_USER, REGSTR_PATH_DESKTOP,
                       0, KEY_QUERY_VALUE, &hKey);
    if (rc != ERROR_SUCCESS)
    {
        WARN("RegOpenKeyEx() failed with error %lu\n", rc);
        return;
    }
 
    Size = sizeof(szWallpaper);
    rc = RegQueryValueExW(hKey,
                          L"Wallpaper",
                          NULL,
                          &Type,
                          (LPBYTE)szWallpaper,
                          &Size);
    RegCloseKey(hKey);
 
    if (rc == ERROR_SUCCESS && Type == REG_SZ)
    {
        ExpandEnvironmentStringsW(szWallpaper, szWallpaper, ARRAYSIZE(szWallpaper));
        TRACE("Using wallpaper %s\n", debugstr_w(szWallpaper));
 
        /* Load and change the wallpaper */
        SystemParametersInfoW(SPI_SETDESKWALLPAPER, 0, szWallpaper, SPIF_SENDCHANGE);
    }
    else
    {
        /* Remove the wallpaper */
        TRACE("No wallpaper set in registry (error %lu)\n", rc);
        SystemParametersInfoW(SPI_SETDESKWALLPAPER, 0, NULL, SPIF_SENDCHANGE);
    }
}

Referenced by SetUserSettings().

StartInstaller()

static BOOL StartInstaller ( VOID  )

static

Definition at line 496 of file userinit.c.

{
    extern WCHAR Installer[MAX_PATH]; // In livecd.c
    WCHAR szMsg[RC_STRING_MAX_SIZE];
 
    /* If we have found the ReactOS Installer, start it */
    if (*Installer && StartProcess(Installer, NULL))
        return TRUE;
 
    /* We failed, display an error message and quit */
    ERR("Failed to start the installer '%s'\n", debugstr_w(Installer));
    LoadStringW(GetModuleHandle(NULL), IDS_INSTALLER_FAIL, szMsg, ARRAYSIZE(szMsg));
    MessageBoxW(NULL, szMsg, NULL, MB_OK);
    return FALSE;
}

Referenced by wWinMain().

StartProcess()

static BOOL StartProcess ( _In_ PCWSTR  CommandLine, _In_opt_ PVOID  pEnvironment  )

static

Definition at line 180 of file userinit.c.

{
    STARTUPINFO si;
    PROCESS_INFORMATION pi;
    WCHAR ExpandedCmdLine[MAX_PATH], ProfilePath[MAX_PATH];
    HANDLE hToken;
    PWCHAR WorkingDir = NULL;
 
    ExpandEnvironmentStringsW(CommandLine, ExpandedCmdLine, ARRAYSIZE(ExpandedCmdLine));
 
    ZeroMemory(&si, sizeof(si));
    si.cb = sizeof(si);
    si.dwFlags = STARTF_USESHOWWINDOW;
    si.wShowWindow = SW_SHOWNORMAL;
    ZeroMemory(&pi, sizeof(pi));
 
    if (OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, &hToken))
    {
        DWORD PathSize = _countof(ProfilePath);
        if (GetUserProfileDirectoryW(hToken, ProfilePath, &PathSize))
            WorkingDir = ProfilePath;
        CloseHandle(hToken);
    }
 
    if (!CreateProcessW(NULL,
                        ExpandedCmdLine,
                        NULL,
                        NULL,
                        FALSE,
                        NORMAL_PRIORITY_CLASS | CREATE_UNICODE_ENVIRONMENT,
                        pEnvironment,
                        WorkingDir,
                        &si,
                        &pi))
    {
        WARN("CreateProcessW() failed with error %lu\n", GetLastError());
        return FALSE;
    }
 
    CloseHandle(pi.hProcess);
    CloseHandle(pi.hThread);
    return TRUE;
}

Referenced by StartInstaller(), and StartShell().

StartShell()

static BOOL StartShell ( _In_opt_ PVOID  pEnvironment )

static

Definition at line 227 of file userinit.c.

{
    DWORD Type, Size;
    DWORD Value = 0;
    LONG rc;
    HKEY hKey;
    WCHAR Shell[MAX_PATH];
    WCHAR szMsg[RC_STRING_MAX_SIZE];
 
    /* Safe Mode shell run */
    rc = RegOpenKeyExW(HKEY_LOCAL_MACHINE,
                       L"SYSTEM\\CurrentControlSet\\Control\\SafeBoot\\Option",
                       0, KEY_QUERY_VALUE, &hKey);
    if (rc == ERROR_SUCCESS)
    {
        Size = sizeof(Value);
        rc = RegQueryValueExW(hKey, L"UseAlternateShell", NULL,
                              &Type, (LPBYTE)&Value, &Size);
        RegCloseKey(hKey);
 
        if (rc == ERROR_SUCCESS)
        {
            if (Type == REG_DWORD)
            {
                if (Value)
                {
                    /* Safe Mode Alternate Shell required */
                    rc = RegOpenKeyExW(HKEY_LOCAL_MACHINE,
                                       L"SYSTEM\\CurrentControlSet\\Control\\SafeBoot",
                                       0, KEY_READ, &hKey);
                    if (rc == ERROR_SUCCESS)
                    {
                        Size = sizeof(Shell);
                        rc = RegQueryValueExW(hKey, L"AlternateShell", NULL,
                                              &Type, (LPBYTE)Shell, &Size);
                        RegCloseKey(hKey);
 
                        if (rc == ERROR_SUCCESS)
                        {
                            if ((Type == REG_SZ) || (Type == REG_EXPAND_SZ))
                            {
                                TRACE("Key located - %s\n", debugstr_w(Shell));
 
                                /* Try to run alternate shell */
                                if (StartProcess(Shell, pEnvironment))
                                {
                                    TRACE("Alternate shell started (Safe Mode)\n");
                                    return TRUE;
                                }
                            }
                            else
                            {
                                WARN("Wrong type %lu (expected %u or %u)\n",
                                     Type, REG_SZ, REG_EXPAND_SZ);
                            }
                        }
                        else
                        {
                            WARN("Alternate shell in Safe Mode required but not specified.\n");
                        }
                    }
                }
            }
            else
            {
                WARN("Wrong type %lu (expected %u)\n", Type, REG_DWORD);
            }
        }
    }
 
    /* Try to run shell in user key */
    if (GetShell(Shell, HKEY_CURRENT_USER) && StartProcess(Shell, pEnvironment))
    {
        TRACE("Started shell from HKEY_CURRENT_USER\n");
        return TRUE;
    }
 
    /* Try to run shell in local machine key */
    if (GetShell(Shell, HKEY_LOCAL_MACHINE) && StartProcess(Shell, pEnvironment))
    {
        TRACE("Started shell from HKEY_LOCAL_MACHINE\n");
        return TRUE;
    }
 
    /* Try default shell */
    if (IsConsoleShell())
    {
        *Shell = UNICODE_NULL;
        if (GetSystemDirectoryW(Shell, ARRAYSIZE(Shell) - 8))
            StringCchCatW(Shell, ARRAYSIZE(Shell), L"\\");
        StringCchCatW(Shell, ARRAYSIZE(Shell), L"cmd.exe");
    }
    else
    {
        *Shell = UNICODE_NULL;
        if (GetSystemWindowsDirectoryW(Shell, ARRAYSIZE(Shell) - 13))
            StringCchCatW(Shell, ARRAYSIZE(Shell), L"\\");
        StringCchCatW(Shell, ARRAYSIZE(Shell), L"explorer.exe");
    }
 
    if (StartProcess(Shell, pEnvironment))
        return TRUE;
 
    /* We failed, display an error message and quit */
    ERR("Failed to start default shell '%s'\n", debugstr_w(Shell));
    LoadStringW(GetModuleHandle(NULL), IDS_SHELL_FAIL, szMsg, ARRAYSIZE(szMsg));
    MessageBoxW(NULL, szMsg, NULL, MB_OK);
    return FALSE;
}

Referenced by wWinMain().

StrToColorref()

static COLORREF StrToColorref ( IN LPWSTR  lpszCol )

static

Definition at line 373 of file userinit.c.

{
    BYTE rgb[3];
 
    rgb[0] = (BYTE)wcstoul(lpszCol, &lpszCol, 10);
    rgb[1] = (BYTE)wcstoul(lpszCol, &lpszCol, 10);
    rgb[2] = (BYTE)wcstoul(lpszCol, &lpszCol, 10);
    return RGB(rgb[0], rgb[1], rgb[2]);
}

Referenced by SetUserSysColors().

wWinMain()

int WINAPI wWinMain	(	IN HINSTANCE	hInst,
		IN HINSTANCE	hPrevInstance,
		IN LPWSTR	lpszCmdLine,
		IN int	nCmdShow
	)

Definition at line 542 of file userinit.c.

{
    BOOL bIsLiveCD, Success = TRUE;
    STATE State;
 
    hInstance = hInst;
 
    bIsLiveCD = IsMiniNT();
 
Restart:
    SetUserSettings();
 
    if (bIsLiveCD)
    {
        State.NextPage = LOCALEPAGE;
        State.Run = SHELL;
    }
    else
    {
        State.NextPage = DONE;
        State.Run = SHELL;
    }
 
    if (State.NextPage != DONE) // && bIsLiveCD
    {
        RunLiveCD(&State);
    }
 
    switch (State.Run)
    {
        case SHELL:
        {
            /* In LiveCD mode, create a suitable environment block for the
             * shell; otherwise, use the current one (built by WinLogon) */
            PVOID pEnvironment = NULL;
            if (bIsLiveCD && /* In LiveCD mode we run under the LocalSystem account */
                !CreateEnvironmentBlock(&pEnvironment, NULL, TRUE))
            {
                WARN("CreateEnvironmentBlock() failed, fall back to default (error %lu)\n",
                     GetLastError());
            }
            Success = StartShell(pEnvironment);
            if (pEnvironment)
                DestroyEnvironmentBlock(pEnvironment);
            if (Success)
                NotifyLogon();
            break;
        }
 
        case INSTALLER:
            Success = StartInstaller();
            break;
 
        case REBOOT:
        {
            EnablePrivilege(SE_SHUTDOWN_NAME, TRUE);
            ExitWindowsEx(EWX_REBOOT, 0);
            EnablePrivilege(SE_SHUTDOWN_NAME, FALSE);
            Success = TRUE;
            break;
        }
 
        default:
            Success = FALSE;
            break;
    }
 
    /*
     * In LiveCD mode, go back to the main menu if we failed
     * to either start the shell or the installer.
     */
    if (bIsLiveCD && !Success)
        goto Restart;
 
    return 0;
}

Variable Documentation

g_RegColorNames

const WCHAR g_RegColorNames[][32]

Definition at line 338 of file userinit.c.

Referenced by SetUserSysColors().

hInstance

HINSTANCE hInstance

Definition at line 34 of file userinit.c.

Referenced by wWinMain().