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.
7 #pragma comment(lib, "wtsapi32.lib")
9 #include "chrome/browser/policy/policy_path_parser.h"
11 #include "base/command_line.h"
12 #include "base/memory/scoped_ptr.h"
13 #include "base/strings/utf_string_conversions.h"
14 #include "base/win/registry.h"
15 #include "chrome/common/chrome_switches.h"
16 #include "policy/policy_constants.h"
20 // Checks if the registry key exists in the given hive and expands any
21 // variables in the string.
22 bool LoadUserDataDirPolicyFromRegistry(HKEY hive,
23 const std::wstring& key_name,
24 base::FilePath* user_data_dir) {
27 base::win::RegKey policy_key(hive,
28 policy::kRegistryChromePolicyKey,
30 if (policy_key.ReadValue(key_name.c_str(), &value) == ERROR_SUCCESS) {
32 base::FilePath(policy::path_parser::ExpandPathVariables(value));
38 const WCHAR* kMachineNamePolicyVarName = L"${machine_name}";
39 const WCHAR* kUserNamePolicyVarName = L"${user_name}";
40 const WCHAR* kWinDocumentsFolderVarName = L"${documents}";
41 const WCHAR* kWinLocalAppDataFolderVarName = L"${local_app_data}";
42 const WCHAR* kWinRoamingAppDataFolderVarName = L"${roaming_app_data}";
43 const WCHAR* kWinProfileFolderVarName = L"${profile}";
44 const WCHAR* kWinProgramDataFolderVarName = L"${global_app_data}";
45 const WCHAR* kWinProgramFilesFolderVarName = L"${program_files}";
46 const WCHAR* kWinWindowsFolderVarName = L"${windows}";
47 const WCHAR* kWinClientName = L"${client_name}";
49 struct WinFolderNamesToCSIDLMapping {
54 // Mapping from variable names to Windows CSIDL ids.
55 const WinFolderNamesToCSIDLMapping win_folder_mapping[] = {
56 { kWinWindowsFolderVarName, CSIDL_WINDOWS},
57 { kWinProgramFilesFolderVarName, CSIDL_PROGRAM_FILES},
58 { kWinProgramDataFolderVarName, CSIDL_COMMON_APPDATA},
59 { kWinProfileFolderVarName, CSIDL_PROFILE},
60 { kWinLocalAppDataFolderVarName, CSIDL_LOCAL_APPDATA},
61 { kWinRoamingAppDataFolderVarName, CSIDL_APPDATA},
62 { kWinDocumentsFolderVarName, CSIDL_PERSONAL}
69 namespace path_parser {
71 // Replaces all variable occurances in the policy string with the respective
72 // system settings values.
73 base::FilePath::StringType ExpandPathVariables(
74 const base::FilePath::StringType& untranslated_string) {
75 base::FilePath::StringType result(untranslated_string);
76 if (result.length() == 0)
78 // Sanitize quotes in case of any around the whole string.
79 if (result.length() > 1 &&
80 ((result[0] == L'"' && result[result.length() - 1] == L'"') ||
81 (result[0] == L'\'' && result[result.length() - 1] == L'\''))) {
82 // Strip first and last char which should be matching quotes now.
83 result = result.substr(1, result.length() - 2);
85 // First translate all path variables we recognize.
86 for (int i = 0; i < arraysize(win_folder_mapping); ++i) {
87 size_t position = result.find(win_folder_mapping[i].name);
88 if (position != std::wstring::npos) {
90 ::SHGetSpecialFolderPath(0, path, win_folder_mapping[i].id, false);
91 std::wstring path_string(path);
92 result.replace(position, wcslen(win_folder_mapping[i].name), path_string);
95 // Next translate other windows specific variables.
96 size_t position = result.find(kUserNamePolicyVarName);
97 if (position != std::wstring::npos) {
98 DWORD return_length = 0;
99 ::GetUserName(NULL, &return_length);
100 if (return_length != 0) {
101 scoped_ptr<WCHAR[]> username(new WCHAR[return_length]);
102 ::GetUserName(username.get(), &return_length);
103 std::wstring username_string(username.get());
104 result.replace(position, wcslen(kUserNamePolicyVarName), username_string);
107 position = result.find(kMachineNamePolicyVarName);
108 if (position != std::wstring::npos) {
109 DWORD return_length = 0;
110 ::GetComputerNameEx(ComputerNamePhysicalDnsHostname, NULL, &return_length);
111 if (return_length != 0) {
112 scoped_ptr<WCHAR[]> machinename(new WCHAR[return_length]);
113 ::GetComputerNameEx(ComputerNamePhysicalDnsHostname,
114 machinename.get(), &return_length);
115 std::wstring machinename_string(machinename.get());
117 position, wcslen(kMachineNamePolicyVarName), machinename_string);
120 position = result.find(kWinClientName);
121 if (position != std::wstring::npos) {
122 LPWSTR buffer = NULL;
123 DWORD buffer_length = 0;
124 if (::WTSQuerySessionInformation(WTS_CURRENT_SERVER, WTS_CURRENT_SESSION,
126 &buffer, &buffer_length)) {
127 std::wstring clientname_string(buffer);
128 result.replace(position, wcslen(kWinClientName), clientname_string);
129 ::WTSFreeMemory(buffer);
136 void CheckUserDataDirPolicy(base::FilePath* user_data_dir) {
137 DCHECK(user_data_dir);
138 // We are running as Chrome Frame if we were invoked with user-data-dir,
139 // chrome-frame, and automation-channel switches.
140 CommandLine* command_line = CommandLine::ForCurrentProcess();
141 const bool is_chrome_frame =
142 !user_data_dir->empty() &&
143 command_line->HasSwitch(switches::kChromeFrame) &&
144 command_line->HasSwitch(switches::kAutomationClientChannelID);
146 // In the case of Chrome Frame, the last path component of the user-data-dir
147 // provided on the command line must be preserved since it is specific to
149 base::FilePath cf_host_dir;
151 cf_host_dir = user_data_dir->BaseName();
153 // Policy from the HKLM hive has precedence over HKCU so if we have one here
154 // we don't have to try to load HKCU.
155 const char* key_name_ascii = (is_chrome_frame ? policy::key::kGCFUserDataDir :
156 policy::key::kUserDataDir);
157 std::wstring key_name(ASCIIToWide(key_name_ascii));
158 if (LoadUserDataDirPolicyFromRegistry(HKEY_LOCAL_MACHINE, key_name,
160 LoadUserDataDirPolicyFromRegistry(HKEY_CURRENT_USER, key_name,
162 // A Group Policy value was loaded. Append the Chrome Frame host directory
165 *user_data_dir = user_data_dir->Append(cf_host_dir);
169 } // namespace path_parser
171 } // namespace policy