Index: chrome/install_static/install_util.cc |
diff --git a/chrome/install_static/install_util.cc b/chrome/install_static/install_util.cc |
index 6fbef47569c23475978e43550443272fd66e21a5..6853780cd3b82f71880ea1d2bcd74b4de75d387f 100644 |
--- a/chrome/install_static/install_util.cc |
+++ b/chrome/install_static/install_util.cc |
@@ -5,14 +5,31 @@ |
#include "chrome/install_static/install_util.h" |
#include <assert.h> |
+#include <memory> |
#include <windows.h> |
-#include <stddef.h> |
#include "base/macros.h" |
-#include "base/strings/string16.h" |
+#include "base/memory/scoped_ptr.h" |
+#include "base/strings/pattern.h" |
+#include "base/strings/string_util.h" |
scottmg
2016/04/27 23:29:24
Is this used?
ananta
2016/04/28 00:04:25
Removed
|
+#include "build/build_config.h" |
+ |
+namespace install_static { |
ProcessType g_process_type = ProcessType::UNINITIALIZED; |
+// Chrome channel display names. |
+// TODO(ananta) |
+// http://crbug.com/604923 |
+// These constants are defined in the chrome/installer directory as well. We |
+// need to unify them. |
+const wchar_t kChromeChannelUnknown[] = L"unknown"; |
+const wchar_t kChromeChannelCanary[] = L"canary"; |
+const wchar_t kChromeChannelDev[] = L"dev"; |
+const wchar_t kChromeChannelBeta[] = L"beta"; |
+const wchar_t kChromeChannelStable[] = L""; |
+const wchar_t kChromeChannelStableExplicit[] = L"stable"; |
+ |
namespace { |
const wchar_t kRegPathClientState[] = L"Software\\Google\\Update\\ClientState"; |
@@ -28,6 +45,9 @@ const wchar_t kRegValueUsageStats[] = L"usagestats"; |
const wchar_t kUninstallArgumentsField[] = L"UninstallArguments"; |
const wchar_t kMetricsReportingEnabled[] = L"MetricsReportingEnabled"; |
+const wchar_t kRegPathGoogleUpdate[] = L"Software\\Google\\Update"; |
+const wchar_t kRegGoogleUpdateVersion[] = L"version"; |
+ |
const wchar_t kAppGuidCanary[] = |
L"{4ea16ac7-fd5a-47c3-875b-dbf4a2008c20}"; |
const wchar_t kAppGuidGoogleChrome[] = |
@@ -35,6 +55,21 @@ const wchar_t kAppGuidGoogleChrome[] = |
const wchar_t kAppGuidGoogleBinaries[] = |
L"{4DC8B4CA-1BDA-483e-B5FA-D3C12E15B62D}"; |
+// TODO(ananta) |
+// http://crbug.com/604923 |
+// These constants are defined in the chrome/installer directory as well. We |
+// need to unify them. |
+#if defined(GOOGLE_CHROME_BUILD) |
+const wchar_t kSxSSuffix[] = L" SxS"; |
+const wchar_t kGoogleChromeInstallSubDir1[] = L"Google"; |
+const wchar_t kGoogleChromeInstallSubDir2[] = L"Chrome"; |
+const wchar_t kRegApField[] = L"ap"; |
+#else |
+const wchar_t kChromiumInstallSubDir[] = L"Chromium"; |
+#endif |
+const wchar_t kUserDataDirname[] = L"User Data"; |
+const wchar_t kBrowserCrashDumpMetricsSubKey[] = L"\\BrowserCrashDumpAttempts"; |
+ |
bool ReadKeyValueString(bool system_install, const wchar_t* key_path, |
const wchar_t* guid, const wchar_t* value_to_read, |
base::string16* value_out) { |
@@ -42,8 +77,10 @@ bool ReadKeyValueString(bool system_install, const wchar_t* key_path, |
value_out->clear(); |
base::string16 full_key_path(key_path); |
- full_key_path.append(1, L'\\'); |
- full_key_path.append(guid); |
+ if (wcslen(guid) > 0) { |
+ full_key_path.append(1, L'\\'); |
+ full_key_path.append(guid); |
+ } |
if (::RegOpenKeyEx(system_install ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER, |
full_key_path.c_str(), 0, |
@@ -107,6 +144,168 @@ bool ReadKeyValueDW(bool system_install, const wchar_t* key_path, |
return result == ERROR_SUCCESS && size == sizeof(*value_out); |
} |
+struct LanguageAndCodePage { |
+ WORD language; |
+ WORD code_page; |
+}; |
+ |
+bool GetLanguageAndCodePageFromVersionResource(const char* version_resource, |
+ WORD* language, |
+ WORD* code_page) { |
+ if (!version_resource) |
+ return false; |
+ |
+ LanguageAndCodePage* translation_info = nullptr; |
+ uint32_t page_count = 0; |
+ BOOL query_result = VerQueryValue(version_resource, |
+ L"\\VarFileInfo\\Translation", |
+ reinterpret_cast<void**>(&translation_info), |
+ &page_count); |
+ if (!query_result) |
+ return false; |
+ |
+ *language = translation_info->language; |
+ *code_page = translation_info->code_page; |
+ return true; |
+} |
+ |
+bool GetValueFromVersionResource(const char* version_resource, |
+ const base::string16& name, |
+ base::string16* value_str) { |
+ DCHECK(value_str); |
+ value_str->clear(); |
+ |
+ WORD language = 0; |
+ WORD code_page = 0; |
+ if (!GetLanguageAndCodePageFromVersionResource(version_resource, |
+ &language, |
+ &code_page)) { |
+ return false; |
+ } |
+ |
+ WORD lang_codepage[8] = {}; |
+ size_t i = 0; |
+ // Use the language and codepage |
+ lang_codepage[i++] = language; |
+ lang_codepage[i++] = code_page; |
+ // Use the default language and codepage from the resource. |
+ lang_codepage[i++] = ::GetUserDefaultLangID(); |
+ lang_codepage[i++] = code_page; |
+ // Use the language from the resource and Latin codepage (most common). |
+ lang_codepage[i++] = language; |
+ lang_codepage[i++] = 1252; |
+ // Use the default language and Latin codepage (most common). |
+ lang_codepage[i++] = ::GetUserDefaultLangID(); |
+ lang_codepage[i++] = 1252; |
+ |
+ static_assert((arraysize(lang_codepage) % 2) == 0, |
+ "Language code page size should be a multiple of 2"); |
+ DCHECK_EQ(arraysize(lang_codepage), i); |
+ |
+ i = 0; |
+ while (i < arraysize(lang_codepage)) { |
+ wchar_t sub_block[MAX_PATH] = {}; |
+ WORD language = lang_codepage[i++]; |
+ WORD code_page = lang_codepage[i++]; |
+ _snwprintf_s(sub_block, MAX_PATH, MAX_PATH, |
+ L"\\StringFileInfo\\%04x%04x\\%ls", language, code_page, name.c_str()); |
+ void* value = NULL; |
+ uint32_t size = 0; |
+ BOOL r = ::VerQueryValue(version_resource, sub_block, &value, &size); |
+ if (r && value) { |
+ value_str->assign(static_cast<wchar_t*>(value)); |
+ return true; |
+ } |
+ } |
+ return false; |
+} |
+ |
+// Returns the executable path for the current process. |
+base::string16 GetCurrentProcessExePath() { |
+ wchar_t exe_path[MAX_PATH] = {}; |
+ if (GetModuleFileName(NULL, exe_path, arraysize(exe_path)) == 0) |
+ return base::string16(); |
+ return exe_path; |
+} |
+ |
+// UTF8 to UTF16 and vice versa conversion helpers. We cannot use the base |
+// string conversion utilities here as they bring about a dependency on |
+// user32.dll which is not allowed in this file. |
+ |
+// Convert a UTF16 string to an UTF8 string. |
+std::string utf16_to_utf8(const base::string16 &source) { |
+ if (source.empty()) |
+ return std::string(); |
+ int size = ::WideCharToMultiByte(CP_UTF8, 0, &source[0], |
+ static_cast<int>(source.size()), NULL, 0, NULL, NULL); |
scottmg
2016/04/27 23:29:24
NULL -> nullptr and elsewhere.
ananta
2016/04/28 00:04:24
Done.
|
+ std::string result(size, 0); |
+ ::WideCharToMultiByte(CP_UTF8, 0, &source[0], |
+ static_cast<int>(source.size()), &result[0], size, NULL, NULL); |
scottmg
2016/04/27 23:29:24
Check that the return value is == size.
ananta
2016/04/28 00:04:24
Done.
|
+ return result; |
+} |
+ |
+// Convert a UTF8 string to a UTF16 string. |
+base::string16 utf8_to_string16(const std::string &source) { |
+ if (source.empty()) |
+ return base::string16(); |
+ int size = ::MultiByteToWideChar(CP_UTF8, 0, &source[0], |
+ static_cast<int>(source.size()), NULL, 0); |
+ base::string16 result(size, 0); |
+ ::MultiByteToWideChar(CP_UTF8, 0, &source[0], |
+ static_cast<int>(source.size()), &result[0], size); |
scottmg
2016/04/27 23:29:24
Check that the return value is == size.
ananta
2016/04/28 00:04:24
Done.
|
+ return result; |
+} |
+ |
+bool RecursiveDirectoryCreate(const base::string16& full_path) { |
+ // If the path exists, we've succeeded if it's a directory, failed otherwise. |
+ const wchar_t* full_path_str = full_path.c_str(); |
+ DWORD file_attributes = ::GetFileAttributes(full_path_str); |
+ if (file_attributes != INVALID_FILE_ATTRIBUTES) { |
+ if ((file_attributes & FILE_ATTRIBUTE_DIRECTORY) != 0) { |
+ DVLOG(1) << "RecursiveDirectoryCreate(" << full_path_str << "), " |
+ << "directory already exists."; |
+ return true; |
+ } |
+ DLOG(WARNING) << "RecursiveDirectoryCreate(" << full_path_str << "), " |
+ << "conflicts with existing file."; |
+ return false; |
+ } |
+ |
+ // Invariant: Path does not exist as file or directory. |
+ |
+ // Attempt to create the parent recursively. This will immediately return |
+ // true if it already exists, otherwise will create all required parent |
+ // directories starting with the highest-level missing parent. |
+ base::string16 parent_path; |
+ std::size_t pos = full_path.find_last_of(L"/\\"); |
+ if (pos != base::string16::npos) { |
+ parent_path = full_path.substr(0, pos); |
+ if (!RecursiveDirectoryCreate(parent_path)) { |
+ DLOG(WARNING) << "Failed to create one of the parent directories."; |
+ return false; |
+ } |
+ } |
+ if (!::CreateDirectory(full_path_str, NULL)) { |
+ DWORD error_code = ::GetLastError(); |
+ if (error_code == ERROR_ALREADY_EXISTS) { |
+ DWORD file_attributes = GetFileAttributes(full_path_str); |
+ if ((file_attributes != INVALID_FILE_ATTRIBUTES) && |
+ ((file_attributes & FILE_ATTRIBUTE_DIRECTORY) != 0)) { |
+ // This error code ERROR_ALREADY_EXISTS doesn't indicate whether we |
+ // were racing with someone creating the same directory, or a file |
+ // with the same path. If the directory exists, we lost the |
+ // race to create the same directory. |
+ return true; |
+ } else { |
+ DLOG(WARNING) << "Failed to create directory " << full_path_str |
+ << ", last error is " << error_code << "."; |
+ return false; |
+ } |
+ } |
+ } |
+ return true; |
+} |
+ |
} // namespace |
bool IsCanary(const wchar_t* exe_path) { |
@@ -138,27 +337,31 @@ bool IsMultiInstall(bool is_system_install) { |
return args.find(L"--multi-install") != base::string16::npos; |
} |
-bool AreUsageStatsEnabled(const wchar_t* exe_path) { |
+bool GetCollectStatsConsent() { |
+ return GetCollectStatsConsentForExe(GetCurrentProcessExePath()); |
+} |
+ |
+bool GetCollectStatsConsentForExe(const base::string16& exe_path) { |
bool enabled = true; |
bool controlled_by_policy = ReportingIsEnforcedByPolicy(&enabled); |
if (controlled_by_policy && !enabled) |
return false; |
- bool system_install = IsSystemInstall(exe_path); |
+ bool system_install = IsSystemInstall(exe_path.c_str()); |
base::string16 app_guid; |
- if (IsCanary(exe_path)) { |
+ if (IsCanary(exe_path.c_str())) { |
app_guid = kAppGuidCanary; |
} else { |
app_guid = IsMultiInstall(system_install) ? kAppGuidGoogleBinaries : |
- kAppGuidGoogleChrome; |
+ kAppGuidGoogleChrome; |
} |
DWORD out_value = 0; |
if (system_install && |
- ReadKeyValueDW(system_install, kRegPathClientStateMedium, app_guid, |
- kRegValueUsageStats, &out_value)) { |
+ ReadKeyValueDW(system_install, kRegPathClientStateMedium, app_guid, |
+ kRegValueUsageStats, &out_value)) { |
return out_value == 1; |
} |
@@ -222,3 +425,214 @@ bool IsNonBrowserProcess() { |
assert(g_process_type != ProcessType::UNINITIALIZED); |
return g_process_type == ProcessType::NON_BROWSER_PROCESS; |
} |
+ |
+bool GetDefaultUserDataDirectory(base::string16* result) { |
+ scoped_ptr<wchar_t> user_data_dir_path; |
+ |
+ // This environment variable should be set on Windows 7 and up. |
+ // If we fail to find this variable then we default to the temporary files |
+ // path. |
+ DWORD size = ::ExpandEnvironmentStrings(L"%LOCALAPPDATA%", nullptr, 0); |
scottmg
2016/04/27 23:29:25
static const wchar_t kLocalAppData[] = L"%LOCALAPP
ananta
2016/04/28 00:04:24
Done.
|
+ if (size) { |
+ user_data_dir_path.reset(new wchar_t[size]); |
+ if (::ExpandEnvironmentStrings(L"%LOCALAPPDATA%", |
+ user_data_dir_path.get(), |
+ size) != size) { |
+ user_data_dir_path.reset(nullptr); |
+ } |
+ } |
+ // We failed to find the %LOCALAPPDATA% folder. Fallback to the temporary |
+ // files path. If we fail to find this we bail. |
+ if (!user_data_dir_path.get()) { |
+ size = ::GetTempPath(0, nullptr); |
+ if (!size) |
+ return false; |
+ user_data_dir_path.reset(new wchar_t[size + 1]); |
+ if (::GetTempPath(size + 1, user_data_dir_path.get()) != size) |
+ return false; |
+ } |
+ |
+ base::string16 install_sub_directory = GetChromeInstallSubDirectory(); |
+ |
+ result->append(user_data_dir_path.get()); |
scottmg
2016/04/27 23:29:24
result = user_data_dir_path here to be more clear
ananta
2016/04/28 00:04:25
Done.
|
+ result->append(L"\\"); |
scottmg
2016/04/27 23:29:24
GetTempPath() says it returns with a trailing \, w
ananta
2016/04/28 00:04:25
Yeah. Added a check for that.
|
+ result->append(install_sub_directory); |
+ result->append(L"\\"); |
+ result->append(kUserDataDirname); |
+ return true; |
+} |
+ |
+bool GetDefaultCrashDumpLocation(base::string16* crash_dir) { |
+ // In order to be able to start crash handling very early, we do not rely on |
+ // chrome's PathService entries (for DIR_CRASH_DUMPS) being available on |
+ // Windows. See https://crbug.com/564398. |
+ if (!GetDefaultUserDataDirectory(crash_dir)) |
+ return false; |
+ |
+ // We have to make sure the user data dir exists on first run. See |
+ // http://crbug.com/591504. |
+ if ((::GetFileAttributes(crash_dir->c_str()) == INVALID_FILE_ATTRIBUTES) && |
scottmg
2016/04/27 23:29:24
Probably no need to check that it exists first alr
ananta
2016/04/28 00:04:24
Done.
|
+ (!RecursiveDirectoryCreate(crash_dir->c_str()))) { |
+ return false; |
+ } |
+ crash_dir->append(L"\\CrashPad"); |
scottmg
2016/04/27 23:29:24
"Crashpad", not "CrashPad".
ananta
2016/04/28 00:04:24
Done.
|
+ return true; |
+} |
+ |
+ |
+std::string GetEnvironmentString(const std::string& variable_name) { |
+ DWORD value_length = ::GetEnvironmentVariable( |
+ utf8_to_string16(variable_name).c_str(), NULL, 0); |
+ if (value_length == 0) |
+ return std::string(); |
+ std::unique_ptr<wchar_t[]> value(new wchar_t[value_length]); |
+ ::GetEnvironmentVariable(utf8_to_string16(variable_name).c_str(), |
+ value.get(), value_length); |
+ return utf16_to_utf8(value.get()); |
+} |
+ |
+bool SetEnvironmentString(const std::string& variable_name, |
+ const std::string& new_value) { |
+ return !!SetEnvironmentVariable(utf8_to_string16(variable_name).c_str(), |
+ utf8_to_string16(new_value).c_str()); |
+} |
+ |
+bool HasEnvironmentVariable(const std::string& variable_name) { |
+ return !!::GetEnvironmentVariable(utf8_to_string16(variable_name).c_str(), |
+ NULL, 0); |
+} |
+ |
+bool GetExecutableVersionDetails(const base::string16& exe_path, |
+ base::string16* product_name, |
+ base::string16* version, |
+ base::string16* special_build, |
+ base::string16* channel_name) { |
+ DCHECK(product_name); |
+ DCHECK(version); |
+ DCHECK(special_build); |
+ DCHECK(channel_name); |
+ |
+ // Default values in case we don't find a version resource. |
+ *product_name = L"Chrome"; |
+ *version = L"0.0.0.0-devel"; |
+ *channel_name = kChromeChannelUnknown; |
+ special_build->clear(); |
+ |
+ DWORD dummy = 0; |
+ DWORD length = ::GetFileVersionInfoSize(exe_path.c_str(), &dummy); |
+ if (length) { |
+ scoped_ptr<char> data(new char[length]); |
+ if (::GetFileVersionInfo(exe_path.c_str(), dummy, length, |
+ data.get())) { |
+ GetValueFromVersionResource(data.get(), L"ProductVersion", version); |
+ |
+ base::string16 official_build; |
+ GetValueFromVersionResource(data.get(), L"Official Build", |
+ &official_build); |
+ if (official_build != L"1") |
+ version->append(L"-devel"); |
+ GetValueFromVersionResource(data.get(), L"ProductShortName", |
+ product_name); |
+ GetValueFromVersionResource(data.get(), L"SpecialBuild", special_build); |
+ } |
+ } |
+ GetChromeChannelName(!IsSystemInstall(exe_path.c_str()), channel_name); |
+ return true; |
+} |
+ |
+void GetChromeChannelName(bool is_per_user_install, |
+ base::string16* channel_name) { |
+#if !defined(GOOGLE_CHROME_BUILD) |
+ *channel_name = kChromeChannelUnknown; |
+ return; |
+#else |
+ channel_name->clear(); |
+ // TODO(ananta) |
+ // http://crbug.com/604923 |
+ // Unify this with the chrome/installer/util/channel_info.h/.cc. |
+ if (IsCanary(GetCurrentProcessExePath().c_str())) { |
+ *channel_name = L"canary"; |
+ } else { |
+ base::string16 value; |
+ if (ReadKeyValueString(!is_per_user_install, |
+ kRegPathClientState, |
+ kAppGuidGoogleChrome, |
+ kRegApField, |
+ &value)) { |
+ static const wchar_t kChromeChannelBetaPattern[] = L"1?1-"; |
+ static const wchar_t kChromeChannelBetaX64Pattern[] = L"*x64-beta*"; |
+ static const wchar_t kChromeChannelDevPattern[] = L"2?0-d"; |
+ static const wchar_t kChromeChannelDevX64Pattern[] = L"x64-dev"; |
+ static const wchar_t kChromeStableMultiInstallPattern[] = L"*multi*"; |
+ |
+ std::transform(value.begin(), value.end(), value.begin(), ::tolower); |
+ |
+ // Channel names containing stable should be reported as an empty string. |
+ if (value.find(kChromeChannelStableExplicit) != |
+ base::string16::npos) { |
+ return; |
+ } |
+ |
+ if (base::MatchPattern(value, kChromeChannelDevPattern) || |
+ base::MatchPattern(value, kChromeChannelDevX64Pattern)) { |
+ channel_name->assign(kChromeChannelDev); |
+ } |
+ |
+ if (base::MatchPattern(value, kChromeChannelBetaPattern) || |
+ base::MatchPattern(value, kChromeChannelBetaPattern)) { |
+ channel_name->assign(kChromeChannelBeta); |
+ } |
+ |
+ if (value.find(kChromeStableMultiInstallPattern) != |
+ base::string16::npos) { |
+ *channel_name = L"-m"; |
+ } |
+ |
+ // If we fail to find any matching pattern in the channel name then we |
+ // default to empty which means stable. Not sure if this is ok. |
+ // TODO(ananta) |
+ // http://crbug.com/604923 |
+ // Check if this is ok. |
+ } else { |
+ *channel_name = kChromeChannelUnknown; |
+ } |
+ } |
+#endif // GOOGLE_CHROME_BUILD |
+} |
+ |
+std::string GetGoogleUpdateVersion() { |
+ base::string16 update_version; |
+ if (ReadKeyValueString(IsSystemInstall(GetCurrentProcessExePath().c_str()), |
+ kRegPathGoogleUpdate, |
+ L"", |
+ kRegGoogleUpdateVersion, |
+ &update_version)) { |
+ return utf16_to_utf8(update_version); |
+ } |
+ return std::string(); |
+} |
+ |
+base::string16 GetChromeInstallSubDirectory() { |
+ base::string16 result; |
+#if defined(GOOGLE_CHROME_BUILD) |
+ base::string16 sub_directory = kGoogleChromeInstallSubDir1; |
+ sub_directory += L"\\"; |
+ sub_directory += kGoogleChromeInstallSubDir2; |
+ if (IsCanary(GetCurrentProcessExePath().c_str())) |
+ sub_directory += kSxSSuffix; |
+ result.append(sub_directory); |
+#else |
+ result = result.append(kChromiumInstallSubDir); |
+#endif |
+ return result; |
+} |
+ |
+base::string16 GetBrowserCrashDumpAttemptsRegistryPath() { |
+ base::string16 registry_path = L"Software"; |
+ base::string16 install_sub_directory = GetChromeInstallSubDirectory(); |
+ registry_path += install_sub_directory; |
+ registry_path += kBrowserCrashDumpMetricsSubKey; |
+ return registry_path; |
+} |
+ |
+} // namespace install_static |