environment.c

Go to the documentation of this file.

/*
 * COPYRIGHT:       See COPYING in the top level directory
 * PROJECT:         ReactOS Winlogon
 * FILE:            base/system/winlogon/environment.c
 * PURPOSE:         User environment routines
 * PROGRAMMERS:     Thomas Weidenmueller (w3seek@users.sourceforge.net)
 *                  Hervé Poussineau (hpoussin@reactos.org)
 *                  Eric Kohl
 */
 
/* INCLUDES *****************************************************************/
 
#include "winlogon.h"
 
/* GLOBALS ******************************************************************/
 
static const struct
{
    PCWSTR pszVariable;
    PCWSTR pszShellFolder;
} AppDataVariables[] =
{
    {L"APPDATA",      L"AppData"},
    {L"LOCALAPPDATA", L"Local AppData"},
};
 
 
/* FUNCTIONS ****************************************************************/
 
static
VOID
BuildVolatileEnvironment(
    IN PWLSESSION Session,
    IN HKEY hKeyCurrentUser)
{
    WCHAR szPath[MAX_PATH + 1];
    LPCWSTR wstr;
    SIZE_T size;
    WCHAR szEnvKey[MAX_PATH];
    WCHAR szEnvValue[1024];
    SIZE_T length;
    LPWSTR eqptr, endptr;
    DWORD dwDisp;
    LONG lError;
    HKEY hKeyVolatileEnv;
    HKEY hKeyShellFolders;
    DWORD dwType;
    DWORD dwSize;
 
    /* Create the 'Volatile Environment' key */
    lError = RegCreateKeyExW(hKeyCurrentUser,
                             L"Volatile Environment",
                             0,
                             NULL,
                             REG_OPTION_VOLATILE,
                             KEY_WRITE,
                             NULL,
                             &hKeyVolatileEnv,
                             &dwDisp);
    if (lError != ERROR_SUCCESS)
    {
        WARN("WL: RegCreateKeyExW() failed to create the volatile environment key (Error: %ld)\n", lError);
        return;
    }
 
    /* Parse the environment variables and add them to the volatile environment key */
    if (Session->Profile->dwType == WLX_PROFILE_TYPE_V2_0 &&
        Session->Profile->pszEnvironment != NULL)
    {
        wstr = Session->Profile->pszEnvironment;
        while (*wstr != UNICODE_NULL)
        {
            size = wcslen(wstr) + 1;
 
            eqptr = wcschr(wstr, L'=');
 
            if (eqptr != NULL)
            {
                endptr = eqptr;
 
                endptr--;
                while (iswspace(*endptr))
                    endptr--;
 
                length = (SIZE_T)(endptr - wstr + 1);
 
                wcsncpy(szEnvKey, wstr, length);
                szEnvKey[length] = 0;
 
                eqptr++;
                while (iswspace(*eqptr))
                    eqptr++;
                wcscpy(szEnvValue, eqptr);
 
                RegSetValueExW(hKeyVolatileEnv,
                               szEnvKey,
                               0,
                               REG_SZ,
                               (LPBYTE)szEnvValue,
                               (wcslen(szEnvValue) + 1) * sizeof(WCHAR));
            }
 
            wstr += size;
        }
    }
 
    /* Set the 'APPDATA' and 'LOCALAPPDATA' environment variables */
    lError = RegOpenKeyExW(hKeyCurrentUser,
                           L"Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\Shell Folders",
                           0,
                           KEY_READ,
                           &hKeyShellFolders);
    if (lError == ERROR_SUCCESS)
    {
        ULONG i;
 
        for (i = 0; i < ARRAYSIZE(AppDataVariables); i++)
        {
            dwSize = (MAX_PATH + 1) * sizeof(WCHAR);
            lError = RegQueryValueExW(hKeyShellFolders,
                                      AppDataVariables[i].pszShellFolder,
                                      NULL,
                                      &dwType,
                                      (LPBYTE)szPath,
                                      &dwSize);
            if (lError != ERROR_SUCCESS)
            {
                WARN("WL: No '%S' shell folder (Error: %ld)\n",
                     AppDataVariables[i].pszShellFolder, lError);
                continue;
            }
 
            TRACE("%S path: %S\n", AppDataVariables[i].pszVariable, szPath);
            RegSetValueExW(hKeyVolatileEnv,
                           AppDataVariables[i].pszVariable,
                           0,
                           REG_SZ,
                           (LPBYTE)szPath,
                           (wcslen(szPath) + 1) * sizeof(WCHAR));
        }
 
        RegCloseKey(hKeyShellFolders);
    }
 
    RegCloseKey(hKeyVolatileEnv);
}
 
 
BOOL
CreateUserEnvironment(
    IN PWLSESSION Session)
{
    HKEY hKeyCurrentUser;
    LONG lError;
 
    TRACE("WL: CreateUserEnvironment called\n");
 
    /* Impersonate the new user */
    if (!ImpersonateLoggedOnUser(Session->UserToken))
    {
        ERR("ImpersonateLoggedOnUser() failed with error %lu\n", GetLastError());
        return FALSE;
    }
 
    /* Open the new user HKCU key */
    lError = RegOpenCurrentUser(KEY_CREATE_SUB_KEY,
                                &hKeyCurrentUser);
    if (lError == ERROR_SUCCESS)
    {
        BuildVolatileEnvironment(Session,
                                 hKeyCurrentUser);
        RegCloseKey(hKeyCurrentUser);
    }
 
    /* Revert the impersonation */
    RevertToSelf();
 
    TRACE("WL: CreateUserEnvironment done\n");
 
    return TRUE;
}