Chromium Code Reviews| Index: chrome/browser/policy/policy_path_parser_win.cc | 
| diff --git a/chrome/browser/policy/policy_path_parser_win.cc b/chrome/browser/policy/policy_path_parser_win.cc | 
| index 9de7ed9da96a9f77ac5b1848d2bce64c45010ced..c66417b58b83f79172037e197c8a73f46c878335 100644 | 
| --- a/chrome/browser/policy/policy_path_parser_win.cc | 
| +++ b/chrome/browser/policy/policy_path_parser_win.cc | 
| @@ -14,6 +14,7 @@ | 
| #include "base/strings/utf_string_conversions.h" | 
| #include "base/win/registry.h" | 
| #include "chrome/common/chrome_switches.h" | 
| +#include "chrome/install_static/policy_path_parser.h" | 
| #include "components/policy/policy_constants.h" | 
| namespace { | 
| @@ -32,34 +33,6 @@ bool LoadUserDataDirPolicyFromRegistry(HKEY hive, | 
| return false; | 
| } | 
| -const WCHAR* kMachineNamePolicyVarName = L"${machine_name}"; | 
| -const WCHAR* kUserNamePolicyVarName = L"${user_name}"; | 
| -const WCHAR* kWinDocumentsFolderVarName = L"${documents}"; | 
| -const WCHAR* kWinLocalAppDataFolderVarName = L"${local_app_data}"; | 
| -const WCHAR* kWinRoamingAppDataFolderVarName = L"${roaming_app_data}"; | 
| -const WCHAR* kWinProfileFolderVarName = L"${profile}"; | 
| -const WCHAR* kWinProgramDataFolderVarName = L"${global_app_data}"; | 
| -const WCHAR* kWinProgramFilesFolderVarName = L"${program_files}"; | 
| -const WCHAR* kWinWindowsFolderVarName = L"${windows}"; | 
| -const WCHAR* kWinClientName = L"${client_name}"; | 
| -const WCHAR* kWinSessionName = L"${session_name}"; | 
| - | 
| -struct WinFolderNamesToCSIDLMapping { | 
| - const WCHAR* name; | 
| - int id; | 
| -}; | 
| - | 
| -// Mapping from variable names to Windows CSIDL ids. | 
| -const WinFolderNamesToCSIDLMapping win_folder_mapping[] = { | 
| - { kWinWindowsFolderVarName, CSIDL_WINDOWS}, | 
| - { kWinProgramFilesFolderVarName, CSIDL_PROGRAM_FILES}, | 
| - { kWinProgramDataFolderVarName, CSIDL_COMMON_APPDATA}, | 
| - { kWinProfileFolderVarName, CSIDL_PROFILE}, | 
| - { kWinLocalAppDataFolderVarName, CSIDL_LOCAL_APPDATA}, | 
| - { kWinRoamingAppDataFolderVarName, CSIDL_APPDATA}, | 
| - { kWinDocumentsFolderVarName, CSIDL_PERSONAL} | 
| -}; | 
| - | 
| } // namespace | 
| namespace policy { | 
| @@ -70,80 +43,7 @@ namespace path_parser { | 
| // system settings values. | 
| base::FilePath::StringType ExpandPathVariables( | 
| const base::FilePath::StringType& untranslated_string) { | 
| - base::FilePath::StringType result(untranslated_string); | 
| - if (result.length() == 0) | 
| - return result; | 
| - // Sanitize quotes in case of any around the whole string. | 
| - if (result.length() > 1 && | 
| - ((result.front() == L'"' && result.back() == L'"') || | 
| - (result.front() == L'\'' && result.back() == L'\''))) { | 
| - // Strip first and last char which should be matching quotes now. | 
| - result = result.substr(1, result.length() - 2); | 
| - } | 
| - // First translate all path variables we recognize. | 
| - for (size_t i = 0; i < arraysize(win_folder_mapping); ++i) { | 
| - size_t position = result.find(win_folder_mapping[i].name); | 
| - if (position != std::wstring::npos) { | 
| - WCHAR path[MAX_PATH]; | 
| - ::SHGetSpecialFolderPath(0, path, win_folder_mapping[i].id, false); | 
| - std::wstring path_string(path); | 
| - result.replace(position, wcslen(win_folder_mapping[i].name), path_string); | 
| - } | 
| - } | 
| - // Next translate other windows specific variables. | 
| - size_t position = result.find(kUserNamePolicyVarName); | 
| - if (position != std::wstring::npos) { | 
| - DWORD return_length = 0; | 
| - ::GetUserName(NULL, &return_length); | 
| - if (return_length != 0) { | 
| - std::unique_ptr<WCHAR[]> username(new WCHAR[return_length]); | 
| - ::GetUserName(username.get(), &return_length); | 
| - std::wstring username_string(username.get()); | 
| - result.replace(position, wcslen(kUserNamePolicyVarName), username_string); | 
| - } | 
| - } | 
| - position = result.find(kMachineNamePolicyVarName); | 
| - if (position != std::wstring::npos) { | 
| - DWORD return_length = 0; | 
| - ::GetComputerNameEx(ComputerNamePhysicalDnsHostname, NULL, &return_length); | 
| - if (return_length != 0) { | 
| - std::unique_ptr<WCHAR[]> machinename(new WCHAR[return_length]); | 
| - ::GetComputerNameEx(ComputerNamePhysicalDnsHostname, | 
| - machinename.get(), &return_length); | 
| - std::wstring machinename_string(machinename.get()); | 
| - result.replace( | 
| - position, wcslen(kMachineNamePolicyVarName), machinename_string); | 
| - } | 
| - } | 
| - position = result.find(kWinClientName); | 
| - if (position != std::wstring::npos) { | 
| - LPWSTR buffer = NULL; | 
| - DWORD buffer_length = 0; | 
| - if (::WTSQuerySessionInformation(WTS_CURRENT_SERVER, WTS_CURRENT_SESSION, | 
| - WTSClientName, | 
| - &buffer, &buffer_length)) { | 
| - std::wstring clientname_string(buffer); | 
| - result.replace(position, wcslen(kWinClientName), clientname_string); | 
| - ::WTSFreeMemory(buffer); | 
| - } | 
| - } | 
| - position = result.find(kWinSessionName); | 
| - if (position != std::wstring::npos) { | 
| - LPWSTR buffer = NULL; | 
| - DWORD buffer_length = 0; | 
| - if (::WTSQuerySessionInformation(WTS_CURRENT_SERVER, WTS_CURRENT_SESSION, | 
| - WTSWinStationName, | 
| - &buffer, &buffer_length)) { | 
| - std::wstring sessionname_string(buffer); | 
| - result.replace(position, wcslen(kWinSessionName), sessionname_string); | 
| - ::WTSFreeMemory(buffer); | 
| - } | 
| - } | 
| - // TODO(pastarmovj): Consider reorganizing this code once there are even more | 
| - // variables to be supported. The search for the var and its replacement can | 
| - // be extracted as common functionality. | 
| - | 
| - return result; | 
| + return install_static::ExpandPathVariables(untranslated_string); | 
| 
 
pastarmovj
2016/11/14 09:42:50
nit: Please put a small comment here why the code
 
scottmg
2016/11/15 23:23:14
Done.
(Agreed that the "install_static" library i
 
 | 
| } | 
| void CheckUserDataDirPolicy(base::FilePath* user_data_dir) { |