base/system/winlogon/environment.c

   1 /*
   2  * COPYRIGHT:       See COPYING in the top level directory
   3  * PROJECT:         ReactOS Winlogon
   4  * FILE:            base/system/winlogon/environment.c
   5  * PURPOSE:         User environment routines
   6  * PROGRAMMERS:     Thomas Weidenmueller (w3seek@users.sourceforge.net)
   7  *                  Hervé Poussineau (hpoussin@reactos.org)
   8  *                  Eric Kohl
   9  */
  10
  11 /* INCLUDES *****************************************************************/
  12
  13 #include "winlogon.h"
  14
  15 /* FUNCTIONS ****************************************************************/
  16
  17 static
  18 VOID
  19 BuildVolatileEnvironment(
  20     IN PWLSESSION Session,
  21     IN HKEY hKeyCurrentUser)
  22 {
  23     WCHAR szPath[MAX_PATH + 1];
  24     LPCWSTR wstr;
  25     SIZE_T size;
  26     WCHAR szEnvKey[MAX_PATH];
  27     WCHAR szEnvValue[1024];
  28     SIZE_T length;
  29     LPWSTR eqptr, endptr;
  30     DWORD dwDisp;
  31     LONG lError;
  32     HKEY hKeyVolatileEnv;
  33     HKEY hKeyShellFolders;
  34     DWORD dwType;
  35     DWORD dwSize;
  36
  37     /* Create the 'Volatile Environment' key */
  38     lError = RegCreateKeyExW(hKeyCurrentUser,
  39                              L"Volatile Environment",
  40                              0,
  41                              NULL,
  42                              REG_OPTION_VOLATILE,
  43                              KEY_WRITE,
  44                              NULL,
  45                              &hKeyVolatileEnv,
  46                              &dwDisp);
  47     if (lError != ERROR_SUCCESS)
  48     {
  49         WARN("WL: RegCreateKeyExW() failed to create the volatile environment key (Error: %ld)\n", lError);
  50         return;
  51     }
  52
  53     /* Parse the environment variables and add them to the volatile environment key */
  54     if (Session->Profile->dwType == WLX_PROFILE_TYPE_V2_0 &&
  55         Session->Profile->pszEnvironment != NULL)
  56     {
  57         wstr = Session->Profile->pszEnvironment;
  58         while (*wstr != UNICODE_NULL)
  59         {
  60             size = wcslen(wstr) + 1;
  61
  62             eqptr = wcschr(wstr, L'=');
  63
  64             if (eqptr != NULL)
  65             {
  66                 endptr = eqptr;
  67
  68                 endptr--;
  69                 while (iswspace(*endptr))
  70                     endptr--;
  71
  72                 length = (SIZE_T)(endptr - wstr + 1);
  73
  74                 wcsncpy(szEnvKey, wstr, length);
  75                 szEnvKey[length] = 0;
  76
  77                 eqptr++;
  78                 while (iswspace(*eqptr))
  79                     eqptr++;
  80                 wcscpy(szEnvValue, eqptr);
  81
  82                 RegSetValueExW(hKeyVolatileEnv,
  83                                szEnvKey,
  84                                0,
  85                                REG_SZ,
  86                                (LPBYTE)szEnvValue,
  87                                (wcslen(szEnvValue) + 1) * sizeof(WCHAR));
  88             }
  89
  90             wstr += size;
  91         }
  92     }
  93
  94     /* Set the 'APPDATA' environment variable */
  95     lError = RegOpenKeyExW(hKeyCurrentUser,
  96                            L"Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\Shell Folders",
  97                            0,
  98                            KEY_READ,
  99                            &hKeyShellFolders);
 100     if (lError == ERROR_SUCCESS)
 101     {
 102         dwSize = (MAX_PATH + 1) * sizeof(WCHAR);
 103         lError = RegQueryValueExW(hKeyShellFolders,
 104                                   L"AppData",
 105                                   NULL,
 106                                   &dwType,
 107                                   (LPBYTE)szPath,
 108                                   &dwSize);
 109         if (lError == ERROR_SUCCESS)
 110         {
 111             TRACE("APPDATA path: %S\n", szPath);
 112             RegSetValueExW(hKeyVolatileEnv,
 113                            L"APPDATA",
 114                            0,
 115                            REG_SZ,
 116                            (LPBYTE)szPath,
 117                            (wcslen(szPath) + 1) * sizeof(WCHAR));
 118         }
 119
 120         RegCloseKey(hKeyShellFolders);
 121     }
 122
 123     RegCloseKey(hKeyVolatileEnv);
 124 }
 125
 126
 127 BOOL
 128 CreateUserEnvironment(
 129     IN PWLSESSION Session)
 130 {
 131     HKEY hKeyCurrentUser;
 132     LONG lError;
 133
 134     TRACE("WL: CreateUserEnvironment called\n");
 135
 136     /* Impersonate the new user */
 137     if (!ImpersonateLoggedOnUser(Session->UserToken))
 138     {
 139         ERR("ImpersonateLoggedOnUser() failed with error %lu\n", GetLastError());
 140         return FALSE;
 141     }
 142
 143     /* Open the new user HKCU key */
 144     lError = RegOpenCurrentUser(KEY_CREATE_SUB_KEY,
 145                                 &hKeyCurrentUser);
 146     if (lError == ERROR_SUCCESS)
 147     {
 148         BuildVolatileEnvironment(Session,
 149                                  hKeyCurrentUser);
 150         RegCloseKey(hKeyCurrentUser);
 151     }
 152
 153     /* Revert the impersonation */
 154     RevertToSelf();
 155
 156     TRACE("WL: CreateUserEnvironment done\n");
 157
 158     return TRUE;
 159 }