/* * Implementation of userenv.dll * * Copyright 2006 Mike McCormack for CodeWeavers * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * This library is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this library; if not, write to the Free Software * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA */ #include #include "ntstatus.h" #define WIN32_NO_STATUS #include "windef.h" #include "winbase.h" #include "winreg.h" #include "winternl.h" #include "winnls.h" #include "sddl.h" #include "objbase.h" #include "userenv.h" #include "wine/debug.h" WINE_DEFAULT_DEBUG_CHANNEL( userenv ); BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved) { TRACE("%p %d %p\n", hinstDLL, fdwReason, lpvReserved); switch (fdwReason) { case DLL_WINE_PREATTACH: return FALSE; /* prefer native version */ case DLL_PROCESS_ATTACH: DisableThreadLibraryCalls(hinstDLL); break; } return TRUE; } static BOOL get_reg_value(WCHAR *env, HKEY hkey, const WCHAR *name, WCHAR *val, DWORD size) { DWORD type, res_size=0; if (RegQueryValueExW(hkey, name, 0, &type, NULL, &res_size) != ERROR_SUCCESS) return FALSE; if (type == REG_SZ) { if (res_size > size) return FALSE; return RegQueryValueExW(hkey, name, 0, NULL, (BYTE*)val, &size) == ERROR_SUCCESS; } else if (type == REG_EXPAND_SZ) { UNICODE_STRING us_buf, us_expanded; WCHAR *buf = HeapAlloc(GetProcessHeap(), 0, res_size); if (!buf) return FALSE; if (RegQueryValueExW(hkey, name, 0, NULL, (BYTE*)buf, &res_size) != ERROR_SUCCESS) { HeapFree(GetProcessHeap(), 0, buf); return FALSE; } RtlInitUnicodeString(&us_buf, buf); us_expanded.Buffer = val; us_expanded.MaximumLength = size; if (RtlExpandEnvironmentStrings_U(env, &us_buf, &us_expanded, &size) != STATUS_SUCCESS) { HeapFree(GetProcessHeap(), 0, buf); return FALSE; } HeapFree(GetProcessHeap(), 0, buf); return TRUE; } return FALSE; } static void set_env_var( WCHAR **env, const WCHAR *name, const WCHAR *val ) { UNICODE_STRING nameW, valW; RtlInitUnicodeString( &nameW, name ); RtlInitUnicodeString( &valW, val ); RtlSetEnvironmentVariable( env, &nameW, &valW ); } static void set_registry_variables(WCHAR **env, HKEY hkey, DWORD type, BOOL set_path) { UNICODE_STRING us_name, us_value; WCHAR name[1024], value[1024]; DWORD ret, index, size; for (index = 0; ; index++) { size = ARRAY_SIZE(name); ret = RegEnumValueW(hkey, index, name, &size, NULL, NULL, NULL, NULL); if (ret != ERROR_SUCCESS) break; if (!wcsicmp(name, L"SystemRoot")) continue; if (!wcsicmp(name, L"SystemDrive")) continue; RtlInitUnicodeString(&us_name, name); us_value.Buffer = value; us_value.MaximumLength = sizeof(value); if (!wcsicmp(name, L"PATH") && !RtlQueryEnvironmentVariable_U(*env, &us_name, &us_value)) { if (!set_path) continue; size = lstrlenW(value)+1; if (!get_reg_value(*env, hkey, name, value+size, sizeof(value)-size*sizeof(WCHAR))) continue; value[size] = ';'; set_env_var(env, name, value); continue; } if (get_reg_value(*env, hkey, name, value, sizeof(value)) && value[0]) set_env_var(env, name, value); } } static void set_wow64_environment(WCHAR **env) { WCHAR buf[64]; HKEY hkey; BOOL is_win64 = (sizeof(void *) > sizeof(int)); BOOL is_wow64; IsWow64Process( GetCurrentProcess(), &is_wow64 ); if (RegOpenKeyExW(HKEY_LOCAL_MACHINE, L"Software\\Microsoft\\Windows\\CurrentVersion", 0, KEY_READ|KEY_WOW64_64KEY, &hkey)) return; /* set the ProgramFiles variables */ if (get_reg_value(*env, hkey, L"ProgramFilesDir", buf, sizeof(buf))) { if (is_win64 || is_wow64) set_env_var(env, L"ProgramW6432", buf); if (is_win64 || !is_wow64) set_env_var(env, L"ProgramFiles", buf); } if (get_reg_value(*env, hkey, L"ProgramFilesDir (x86)", buf, sizeof(buf))) { if (is_win64 || is_wow64) set_env_var(env, L"ProgramFiles(x86)", buf); if (is_wow64) set_env_var(env, L"ProgramFiles", buf); } /* set the CommonProgramFiles variables */ if (get_reg_value(*env, hkey, L"CommonFilesDir", buf, sizeof(buf))) { if (is_win64 || is_wow64) set_env_var(env, L"CommonProgramW6432", buf); if (is_win64 || !is_wow64) set_env_var(env, L"CommonProgramFiles", buf); } if (get_reg_value(*env, hkey, L"CommonFilesDir (x86)", buf, sizeof(buf))) { if (is_win64 || is_wow64) set_env_var(env, L"CommonProgramFiles(x86)", buf); if (is_wow64) set_env_var(env, L"CommonProgramFiles", buf); } RegCloseKey(hkey); } BOOL WINAPI CreateEnvironmentBlock( LPVOID* lpEnvironment, HANDLE hToken, BOOL bInherit ) { static const WCHAR env_keyW[] = L"System\\CurrentControlSet\\Control\\Session Manager\\Environment"; static const WCHAR profile_keyW[] = L"Software\\Microsoft\\Windows NT\\CurrentVersion\\ProfileList"; WCHAR *env, buf[UNICODE_STRING_MAX_CHARS], profiles_dir[MAX_PATH]; DWORD len; HKEY hkey, hsubkey; TRACE("%p %p %d\n", lpEnvironment, hToken, bInherit ); if (!lpEnvironment) return FALSE; if (RegOpenKeyExW(HKEY_LOCAL_MACHINE, env_keyW, 0, KEY_READ, &hkey) != ERROR_SUCCESS) return FALSE; if (RtlCreateEnvironment(bInherit, &env) != STATUS_SUCCESS) { RegCloseKey(hkey); return FALSE; } if (!GetEnvironmentVariableW(L"SystemRoot", buf, UNICODE_STRING_MAX_CHARS)) { if (!get_reg_value(env, hkey, L"SystemRoot", buf, UNICODE_STRING_MAX_CHARS)) { buf[0] = 0; WARN("SystemRoot variable not set\n"); } } set_env_var(&env, L"SystemRoot", buf); if (!GetEnvironmentVariableW(L"SystemDrive", buf, UNICODE_STRING_MAX_CHARS)) { if (!get_reg_value(env, hkey, L"SystemDrive", buf, UNICODE_STRING_MAX_CHARS)) { buf[0] = 0; WARN("SystemDrive variable not set\n"); } } set_env_var(&env, L"SystemDrive", buf); set_registry_variables(&env, hkey, REG_SZ, !bInherit); set_registry_variables(&env, hkey, REG_EXPAND_SZ, !bInherit); if (RegOpenKeyExW(hkey, L"Environment", 0, KEY_READ, &hsubkey) == ERROR_SUCCESS) { set_registry_variables(&env, hsubkey, REG_SZ, !bInherit); set_registry_variables(&env, hsubkey, REG_EXPAND_SZ, !bInherit); RegCloseKey(hsubkey); } if (RegOpenKeyExW(hkey, L"Volatile Environment", 0, KEY_READ, &hsubkey) == ERROR_SUCCESS) { set_registry_variables(&env, hsubkey, REG_SZ, !bInherit); set_registry_variables(&env, hsubkey, REG_EXPAND_SZ, !bInherit); RegCloseKey(hsubkey); } RegCloseKey(hkey); if (RegOpenKeyExW(HKEY_LOCAL_MACHINE, profile_keyW, 0, KEY_READ, &hkey) == ERROR_SUCCESS) { if (get_reg_value(env, hkey, L"ProfilesDirectory", profiles_dir, MAX_PATH - sizeof(WCHAR))) { len = lstrlenW(profiles_dir); if (profiles_dir[len-1] != '\\') { profiles_dir[len++] = '\\'; profiles_dir[len] = '\0'; } if (get_reg_value(env, hkey, L"Public", buf, UNICODE_STRING_MAX_CHARS)) set_env_var(&env, L"ALLUSERSPROFILE", buf); } else { profiles_dir[0] = 0; } RegCloseKey(hkey); } len = ARRAY_SIZE(buf); if (GetComputerNameW(buf, &len)) set_env_var(&env, L"COMPUTERNAME", buf); set_wow64_environment(&env); if (!hToken) { if (profiles_dir[0]) { len = lstrlenW(profiles_dir); if (len * sizeof(WCHAR) + sizeof(L"Default") < sizeof(buf)) { wcscpy(buf, profiles_dir); wcscat(buf, L"Default"); set_env_var(&env, L"USERPROFILE", buf); } } wcscpy(buf, L".Default"); } else { TOKEN_USER *token_user = NULL; SID_NAME_USE use; WCHAR *sidW; DWORD size, tmp=0; if (GetTokenInformation(hToken, TokenUser, NULL, 0, &len) || GetLastError()!=ERROR_INSUFFICIENT_BUFFER || !(token_user = HeapAlloc(GetProcessHeap(), 0, len)) || !GetTokenInformation(hToken, TokenUser, token_user, len, &len) || !ConvertSidToStringSidW(token_user->User.Sid, &sidW)) { HeapFree(GetProcessHeap(), 0, token_user); RtlDestroyEnvironment(env); return FALSE; } len = lstrlenW(profiles_dir); memcpy(buf, profiles_dir, len*sizeof(WCHAR)); size = UNICODE_STRING_MAX_CHARS-len; if (LookupAccountSidW(NULL, token_user->User.Sid, buf+len, &size, NULL, &tmp, &use)) { set_env_var(&env, L"USERNAME", buf+len); if (len) set_env_var(&env, L"USERPROFILE", buf); } HeapFree(GetProcessHeap(), 0, token_user); lstrcpyW(buf, sidW); LocalFree(sidW); } if (RegOpenKeyExW(HKEY_USERS, buf, 0, KEY_READ, &hkey) == ERROR_SUCCESS) { if (RegOpenKeyExW(hkey, L"Environment", 0, KEY_READ, &hsubkey) == ERROR_SUCCESS) { set_registry_variables(&env, hsubkey, REG_SZ, !bInherit); set_registry_variables(&env, hsubkey, REG_EXPAND_SZ, !bInherit); RegCloseKey(hsubkey); } if (RegOpenKeyExW(hkey, L"Volatile Environment", 0, KEY_READ, &hsubkey) == ERROR_SUCCESS) { set_registry_variables(&env, hsubkey, REG_SZ, !bInherit); set_registry_variables(&env, hsubkey, REG_EXPAND_SZ, !bInherit); RegCloseKey(hsubkey); } RegCloseKey(hkey); } *lpEnvironment = env; return TRUE; } BOOL WINAPI DestroyEnvironmentBlock(LPVOID lpEnvironment) { NTSTATUS r; TRACE("%p\n", lpEnvironment); r = RtlDestroyEnvironment(lpEnvironment); if (r == STATUS_SUCCESS) return TRUE; return FALSE; } BOOL WINAPI ExpandEnvironmentStringsForUserA( HANDLE hToken, LPCSTR lpSrc, LPSTR lpDest, DWORD dwSize ) { BOOL ret; TRACE("%p %s %p %d\n", hToken, debugstr_a(lpSrc), lpDest, dwSize); ret = ExpandEnvironmentStringsA( lpSrc, lpDest, dwSize ); TRACE("<- %s\n", debugstr_a(lpDest)); return ret; } BOOL WINAPI ExpandEnvironmentStringsForUserW( HANDLE hToken, LPCWSTR lpSrc, LPWSTR lpDest, DWORD dwSize ) { BOOL ret; TRACE("%p %s %p %d\n", hToken, debugstr_w(lpSrc), lpDest, dwSize); ret = ExpandEnvironmentStringsW( lpSrc, lpDest, dwSize ); TRACE("<- %s\n", debugstr_w(lpDest)); return ret; } BOOL WINAPI GetDefaultUserProfileDirectoryA( LPSTR lpProfileDir, LPDWORD lpcchSize ) { FIXME("%p %p\n", lpProfileDir, lpcchSize ); return FALSE; } BOOL WINAPI GetDefaultUserProfileDirectoryW( LPWSTR lpProfileDir, LPDWORD lpcchSize ) { FIXME("%p %p\n", lpProfileDir, lpcchSize ); return FALSE; } BOOL WINAPI GetUserProfileDirectoryA( HANDLE hToken, LPSTR lpProfileDir, LPDWORD lpcchSize ) { BOOL ret; WCHAR *dirW = NULL; TRACE( "%p %p %p\n", hToken, lpProfileDir, lpcchSize ); if (!lpProfileDir || !lpcchSize) { SetLastError( ERROR_INVALID_PARAMETER ); return FALSE; } if (!(dirW = HeapAlloc( GetProcessHeap(), 0, *lpcchSize * sizeof(WCHAR) ))) return FALSE; if ((ret = GetUserProfileDirectoryW( hToken, dirW, lpcchSize ))) WideCharToMultiByte( CP_ACP, 0, dirW, *lpcchSize, lpProfileDir, *lpcchSize, NULL, NULL ); HeapFree( GetProcessHeap(), 0, dirW ); return ret; } BOOL WINAPI GetUserProfileDirectoryW( HANDLE hToken, LPWSTR lpProfileDir, LPDWORD lpcchSize ) { static const WCHAR slashW[] = {'\\',0}; TOKEN_USER *t; WCHAR *userW = NULL, *dirW = NULL; DWORD len, dir_len, domain_len; SID_NAME_USE use; BOOL ret = FALSE; TRACE( "%p %p %p\n", hToken, lpProfileDir, lpcchSize ); if (!lpcchSize) { SetLastError( ERROR_INVALID_PARAMETER ); return FALSE; } len = 0; GetTokenInformation( hToken, TokenUser, NULL, 0, &len ); if (GetLastError() != ERROR_INSUFFICIENT_BUFFER) return FALSE; if (!(t = HeapAlloc( GetProcessHeap(), 0, len ))) return FALSE; if (!GetTokenInformation( hToken, TokenUser, t, len, &len )) goto done; len = domain_len = 0; LookupAccountSidW( NULL, t->User.Sid, NULL, &len, NULL, &domain_len, NULL ); if (GetLastError() != ERROR_INSUFFICIENT_BUFFER) goto done; if (!(userW = HeapAlloc( GetProcessHeap(), 0, len * sizeof(WCHAR) ))) goto done; if (!LookupAccountSidW( NULL, t->User.Sid, userW, &len, NULL, &domain_len, &use )) goto done; dir_len = 0; GetProfilesDirectoryW( NULL, &dir_len ); if (GetLastError() != ERROR_INSUFFICIENT_BUFFER) goto done; if (!(dirW = HeapAlloc( GetProcessHeap(), 0, (dir_len + 1) * sizeof(WCHAR) ))) goto done; if (!GetProfilesDirectoryW( dirW, &dir_len )) goto done; len += dir_len + 2; if (*lpcchSize < len) { SetLastError( ERROR_INSUFFICIENT_BUFFER ); *lpcchSize = len; goto done; } lstrcpyW( lpProfileDir, dirW ); lstrcatW( lpProfileDir, slashW ); lstrcatW( lpProfileDir, userW ); *lpcchSize = len; ret = TRUE; done: HeapFree( GetProcessHeap(), 0, t ); HeapFree( GetProcessHeap(), 0, userW ); HeapFree( GetProcessHeap(), 0, dirW ); return ret; } static const char ProfileListA[] = "Software\\Microsoft\\Windows NT\\CurrentVersion\\ProfileList"; BOOL WINAPI GetProfilesDirectoryA( LPSTR lpProfilesDir, LPDWORD lpcchSize ) { static const char ProfilesDirectory[] = "ProfilesDirectory"; LONG l; HKEY key; BOOL ret = FALSE; DWORD len = 0, expanded_len; LPSTR unexpanded_profiles_dir = NULL; TRACE("%p %p\n", lpProfilesDir, lpcchSize ); if (!lpProfilesDir || !lpcchSize) { SetLastError(ERROR_INVALID_PARAMETER); return FALSE; } l = RegOpenKeyExA(HKEY_LOCAL_MACHINE, ProfileListA, 0, KEY_READ, &key); if (l) { SetLastError(l); return FALSE; } l = RegQueryValueExA(key, ProfilesDirectory, NULL, NULL, NULL, &len); if (l) { SetLastError(l); goto end; } unexpanded_profiles_dir = HeapAlloc(GetProcessHeap(), 0, len); if (!unexpanded_profiles_dir) { SetLastError(ERROR_OUTOFMEMORY); goto end; } l = RegQueryValueExA(key, ProfilesDirectory, NULL, NULL, (BYTE *)unexpanded_profiles_dir, &len); if (l) { SetLastError(l); goto end; } expanded_len = ExpandEnvironmentStringsA(unexpanded_profiles_dir, NULL, 0); /* The returned length doesn't include the NULL terminator. */ if (*lpcchSize < expanded_len - 1) { *lpcchSize = expanded_len - 1; SetLastError(ERROR_INSUFFICIENT_BUFFER); goto end; } *lpcchSize = expanded_len - 1; /* The return value is also the expected length. */ ret = ExpandEnvironmentStringsA(unexpanded_profiles_dir, lpProfilesDir, expanded_len) - 1; end: HeapFree(GetProcessHeap(), 0, unexpanded_profiles_dir); RegCloseKey(key); return ret; } static const WCHAR ProfileListW[] = {'S','o','f','t','w','a','r','e','\\','M','i','c','r','o','s','o','f','t','\\','W','i','n','d','o','w','s',' ','N','T','\\','C','u','r','r','e','n','t','V','e','r','s','i','o','n','\\','P','r','o','f','i','l','e','L','i','s','t',0}; BOOL WINAPI GetProfilesDirectoryW( LPWSTR lpProfilesDir, LPDWORD lpcchSize ) { static const WCHAR ProfilesDirectory[] = {'P','r','o','f','i','l','e','s','D','i','r','e','c','t','o','r','y',0}; LONG l; HKEY key; BOOL ret = FALSE; DWORD len = 0, expanded_len; LPWSTR unexpanded_profiles_dir = NULL; TRACE("%p %p\n", lpProfilesDir, lpcchSize ); if (!lpcchSize) { SetLastError(ERROR_INVALID_PARAMETER); return FALSE; } l = RegOpenKeyExW(HKEY_LOCAL_MACHINE, ProfileListW, 0, KEY_READ, &key); if (l) { SetLastError(l); return FALSE; } l = RegQueryValueExW(key, ProfilesDirectory, NULL, NULL, NULL, &len); if (l) { SetLastError(l); goto end; } unexpanded_profiles_dir = HeapAlloc(GetProcessHeap(), 0, len); if (!unexpanded_profiles_dir) { SetLastError(ERROR_OUTOFMEMORY); goto end; } l = RegQueryValueExW(key, ProfilesDirectory, NULL, NULL, (BYTE *)unexpanded_profiles_dir, &len); if (l) { SetLastError(l); goto end; } expanded_len = ExpandEnvironmentStringsW(unexpanded_profiles_dir, NULL, 0); /* The returned length doesn't include the NULL terminator. */ if (*lpcchSize < expanded_len - 1 || !lpProfilesDir) { *lpcchSize = expanded_len - 1; SetLastError(ERROR_INSUFFICIENT_BUFFER); goto end; } *lpcchSize = expanded_len - 1; /* The return value is also the expected length. */ ret = ExpandEnvironmentStringsW(unexpanded_profiles_dir, lpProfilesDir, expanded_len) - 1; end: HeapFree(GetProcessHeap(), 0, unexpanded_profiles_dir); RegCloseKey(key); return ret; } BOOL WINAPI GetAllUsersProfileDirectoryA( LPSTR lpProfileDir, LPDWORD lpcchSize ) { FIXME("%p %p\n", lpProfileDir, lpcchSize); return FALSE; } BOOL WINAPI GetAllUsersProfileDirectoryW( LPWSTR lpProfileDir, LPDWORD lpcchSize ) { FIXME("%p %p\n", lpProfileDir, lpcchSize); return FALSE; } BOOL WINAPI GetProfileType( DWORD *pdwFlags ) { FIXME("%p\n", pdwFlags ); *pdwFlags = 0; return TRUE; } BOOL WINAPI LoadUserProfileA( HANDLE hToken, LPPROFILEINFOA lpProfileInfo ) { FIXME("%p %p\n", hToken, lpProfileInfo ); lpProfileInfo->hProfile = HKEY_CURRENT_USER; return TRUE; } BOOL WINAPI LoadUserProfileW( HANDLE hToken, LPPROFILEINFOW lpProfileInfo ) { FIXME("%p %p\n", hToken, lpProfileInfo ); lpProfileInfo->hProfile = HKEY_CURRENT_USER; return TRUE; } BOOL WINAPI RegisterGPNotification( HANDLE event, BOOL machine ) { FIXME("%p %d\n", event, machine ); return TRUE; } BOOL WINAPI UnregisterGPNotification( HANDLE event ) { FIXME("%p\n", event ); return TRUE; } BOOL WINAPI UnloadUserProfile( HANDLE hToken, HANDLE hProfile ) { FIXME("(%p, %p): stub\n", hToken, hProfile); return FALSE; } HANDLE WINAPI EnterCriticalPolicySection(BOOL bMachine) { FIXME("(%x)\n", bMachine); SetLastError(ERROR_ACCESS_DENIED); return NULL; } BOOL WINAPI LeaveCriticalPolicySection(HANDLE hSection) { FIXME("(%p)\n", hSection); return TRUE; } DWORD WINAPI GetAppliedGPOListW(DWORD dwFlags, LPCWSTR pMachineName, PSID pSidUser, GUID *pGuidExtension, PGROUP_POLICY_OBJECTW *ppGPOList) { FIXME("(%x %s %p %s %p)\n", dwFlags, debugstr_w(pMachineName), pSidUser, debugstr_guid(pGuidExtension), ppGPOList); return ERROR_ACCESS_DENIED; } /****************************************************************************** * USERENV.138 * * Create .lnk file * * PARAMETERS * int csidl [in] well-known directory location to create link in * LPCSTR lnk_dir [in] directory (relative to directory specified by csidl) to create link in * LPCSTR lnk_filename [in] filename of the link file without .lnk extension * LPCSTR lnk_target [in] file/directory pointed to by link * LPCSTR lnk_iconfile [in] link icon resource filename * DWORD lnk_iconid [in] link icon resource id in file referred by lnk_iconfile * LPCSTR work_directory [in] link target's work directory * WORD hotkey [in] link hotkey (virtual key id) * DWORD win_state [in] initial window size (SW_SHOWMAXIMIZED to start maximized, * SW_SHOWMINNOACTIVE to start minimized, everything else is default state) * LPCSTR comment [in] comment - link's comment * LPCSTR loc_filename_resfile [in] resource file which holds localized filename for this link file * DWORD loc_filename_resid [in] resource id for this link file's localized filename * * RETURNS * TRUE: Link file was successfully created * FALSE: Link file was not created */ BOOL WINAPI USERENV_138( int csidl, LPCSTR lnk_dir, LPCSTR lnk_filename, LPCSTR lnk_target, LPCSTR lnk_iconfile, DWORD lnk_iconid, LPCSTR work_directory, WORD hotkey, DWORD win_state, LPCSTR comment, LPCSTR loc_filename_resfile, DWORD loc_filename_resid) { FIXME("(%d,%s,%s,%s,%s,%d,%s,0x%x,%d,%s,%s,%d) - stub\n", csidl, debugstr_a(lnk_dir), debugstr_a(lnk_filename), debugstr_a(lnk_target), debugstr_a(lnk_iconfile), lnk_iconid, debugstr_a(work_directory), hotkey, win_state, debugstr_a(comment), debugstr_a(loc_filename_resfile), loc_filename_resid ); return FALSE; }