1// Copyright (c) 2011 The Chromium Authors. All rights reserved. 2// Use of this source code is governed by a BSD-style license that can be 3// found in the LICENSE file. 4 5#include <shlobj.h> 6#include <wtsapi32.h> 7#pragma comment(lib, "wtsapi32.lib") 8 9#include "chrome/browser/policy/policy_path_parser.h" 10 11#include "base/memory/scoped_ptr.h" 12#include "base/strings/utf_string_conversions.h" 13#include "base/win/registry.h" 14#include "chrome/common/chrome_switches.h" 15#include "policy/policy_constants.h" 16 17namespace { 18 19// Checks if the key exists in the given hive and expands any string variables. 20bool LoadUserDataDirPolicyFromRegistry(HKEY hive, base::FilePath* dir) { 21 std::wstring value; 22 std::wstring key_name(base::ASCIIToWide(policy::key::kUserDataDir)); 23 base::win::RegKey key(hive, policy::kRegistryChromePolicyKey, KEY_READ); 24 if (key.ReadValue(key_name.c_str(), &value) == ERROR_SUCCESS) { 25 *dir = base::FilePath(policy::path_parser::ExpandPathVariables(value)); 26 return true; 27 } 28 return false; 29} 30 31const WCHAR* kMachineNamePolicyVarName = L"${machine_name}"; 32const WCHAR* kUserNamePolicyVarName = L"${user_name}"; 33const WCHAR* kWinDocumentsFolderVarName = L"${documents}"; 34const WCHAR* kWinLocalAppDataFolderVarName = L"${local_app_data}"; 35const WCHAR* kWinRoamingAppDataFolderVarName = L"${roaming_app_data}"; 36const WCHAR* kWinProfileFolderVarName = L"${profile}"; 37const WCHAR* kWinProgramDataFolderVarName = L"${global_app_data}"; 38const WCHAR* kWinProgramFilesFolderVarName = L"${program_files}"; 39const WCHAR* kWinWindowsFolderVarName = L"${windows}"; 40const WCHAR* kWinClientName = L"${client_name}"; 41 42struct WinFolderNamesToCSIDLMapping { 43 const WCHAR* name; 44 int id; 45}; 46 47// Mapping from variable names to Windows CSIDL ids. 48const WinFolderNamesToCSIDLMapping win_folder_mapping[] = { 49 { kWinWindowsFolderVarName, CSIDL_WINDOWS}, 50 { kWinProgramFilesFolderVarName, CSIDL_PROGRAM_FILES}, 51 { kWinProgramDataFolderVarName, CSIDL_COMMON_APPDATA}, 52 { kWinProfileFolderVarName, CSIDL_PROFILE}, 53 { kWinLocalAppDataFolderVarName, CSIDL_LOCAL_APPDATA}, 54 { kWinRoamingAppDataFolderVarName, CSIDL_APPDATA}, 55 { kWinDocumentsFolderVarName, CSIDL_PERSONAL} 56}; 57 58} // namespace 59 60namespace policy { 61 62namespace path_parser { 63 64// Replaces all variable occurances in the policy string with the respective 65// system settings values. 66base::FilePath::StringType ExpandPathVariables( 67 const base::FilePath::StringType& untranslated_string) { 68 base::FilePath::StringType result(untranslated_string); 69 if (result.length() == 0) 70 return result; 71 // Sanitize quotes in case of any around the whole string. 72 if (result.length() > 1 && 73 ((result[0] == L'"' && result[result.length() - 1] == L'"') || 74 (result[0] == L'\'' && result[result.length() - 1] == L'\''))) { 75 // Strip first and last char which should be matching quotes now. 76 result = result.substr(1, result.length() - 2); 77 } 78 // First translate all path variables we recognize. 79 for (int i = 0; i < arraysize(win_folder_mapping); ++i) { 80 size_t position = result.find(win_folder_mapping[i].name); 81 if (position != std::wstring::npos) { 82 WCHAR path[MAX_PATH]; 83 ::SHGetSpecialFolderPath(0, path, win_folder_mapping[i].id, false); 84 std::wstring path_string(path); 85 result.replace(position, wcslen(win_folder_mapping[i].name), path_string); 86 } 87 } 88 // Next translate other windows specific variables. 89 size_t position = result.find(kUserNamePolicyVarName); 90 if (position != std::wstring::npos) { 91 DWORD return_length = 0; 92 ::GetUserName(NULL, &return_length); 93 if (return_length != 0) { 94 scoped_ptr<WCHAR[]> username(new WCHAR[return_length]); 95 ::GetUserName(username.get(), &return_length); 96 std::wstring username_string(username.get()); 97 result.replace(position, wcslen(kUserNamePolicyVarName), username_string); 98 } 99 } 100 position = result.find(kMachineNamePolicyVarName); 101 if (position != std::wstring::npos) { 102 DWORD return_length = 0; 103 ::GetComputerNameEx(ComputerNamePhysicalDnsHostname, NULL, &return_length); 104 if (return_length != 0) { 105 scoped_ptr<WCHAR[]> machinename(new WCHAR[return_length]); 106 ::GetComputerNameEx(ComputerNamePhysicalDnsHostname, 107 machinename.get(), &return_length); 108 std::wstring machinename_string(machinename.get()); 109 result.replace( 110 position, wcslen(kMachineNamePolicyVarName), machinename_string); 111 } 112 } 113 position = result.find(kWinClientName); 114 if (position != std::wstring::npos) { 115 LPWSTR buffer = NULL; 116 DWORD buffer_length = 0; 117 if (::WTSQuerySessionInformation(WTS_CURRENT_SERVER, WTS_CURRENT_SESSION, 118 WTSClientName, 119 &buffer, &buffer_length)) { 120 std::wstring clientname_string(buffer); 121 result.replace(position, wcslen(kWinClientName), clientname_string); 122 ::WTSFreeMemory(buffer); 123 } 124 } 125 126 return result; 127} 128 129void CheckUserDataDirPolicy(base::FilePath* user_data_dir) { 130 DCHECK(user_data_dir); 131 // Policy from the HKLM hive has precedence over HKCU. 132 if (!LoadUserDataDirPolicyFromRegistry(HKEY_LOCAL_MACHINE, user_data_dir)) 133 LoadUserDataDirPolicyFromRegistry(HKEY_CURRENT_USER, user_data_dir); 134} 135 136} // namespace path_parser 137 138} // namespace policy 139