Chromium Code Reviews| Index: chrome/common/terminate_on_heap_corruption_experiment_win.cc |
| diff --git a/chrome/common/terminate_on_heap_corruption_experiment_win.cc b/chrome/common/terminate_on_heap_corruption_experiment_win.cc |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..ee27975731a5511abd899c3b92cf8e86bf7da873 |
| --- /dev/null |
| +++ b/chrome/common/terminate_on_heap_corruption_experiment_win.cc |
| @@ -0,0 +1,66 @@ |
| +// Copyright 2014 The Chromium Authors. All rights reserved. |
| +// Use of this source code is governed by a BSD-style license that can be |
| +// found in the LICENSE file. |
| + |
| +#include "chrome/common/terminate_on_heap_corruption_experiment_win.h" |
| + |
| +#include "base/command_line.h" |
| +#include "base/metrics/field_trial.h" |
| +#include "base/win/registry.h" |
| +#include "chrome/common/chrome_constants.h" |
| +#include "chrome/common/chrome_version_info.h" |
| + |
| +#if defined(OS_WIN) |
| +#if defined(GOOGLE_CHROME_BUILD) |
| +#define PRODUCT_STRING_PATH L"Google\\Chrome" |
| +#elif defined(CHROMIUM_BUILD) |
| +#define PRODUCT_STRING_PATH L"Chromium" |
| +#else |
| +#error Unknown branding |
| +#endif |
| +#endif // defined(OS_WIN) |
| + |
| +namespace { |
| + |
| +wchar_t* GetBeaconKeyPath() { |
| + chrome::VersionInfo::Channel channel = chrome::VersionInfo::CHANNEL_UNKNOWN; |
| + |
| + // We are called quite early, before the CommandLine is initialized. We don't |
| + // want to permanently initialize it because ContentMainRunner::Initialize |
| + // sets some locale-related stuff to make sure it is parsed properly. But we |
| + // can temporarily initialize it for the purpose of determining if we are |
| + // Canary. |
| + if (!CommandLine::InitializedForCurrentProcess()) { |
| + CommandLine::Init(0, NULL); |
| + channel = chrome::VersionInfo::GetChannel(); |
| + CommandLine::Reset(); |
| + } else { |
| + channel = chrome::VersionInfo::GetChannel(); |
| + } |
| + |
| + if (channel == chrome::VersionInfo::CHANNEL_CANARY) { |
| + return L"SOFTWARE\\" PRODUCT_STRING_PATH |
|
Nico
2014/08/03 05:56:03
..\..\chrome\common\terminate_on_heap_corruption_e
|
| + L"\\DisableTerminateOnProcessHeapCorruptionSxs"; |
| + } |
| + return L"SOFTWARE\\" PRODUCT_STRING_PATH |
| + L"\\DisableTerminateOnProcessHeapCorruption"; |
| +} |
| + |
| +} // namespace |
| + |
| +bool ShouldExperimentallyDisableTerminateOnHeapCorruption() { |
| + base::win::RegKey regkey( |
| + HKEY_CURRENT_USER, GetBeaconKeyPath(), KEY_QUERY_VALUE); |
| + return regkey.Valid(); |
| +} |
| + |
| +void InitializeDisableTerminateOnHeapCorruptionExperiment() { |
| + base::win::RegKey regkey(HKEY_CURRENT_USER); |
| + |
| + if (base::FieldTrialList::FindFullName("TerminateOnProcessHeapCorruption") == |
| + "Disabled") { |
| + regkey.CreateKey(GetBeaconKeyPath(), KEY_SET_VALUE); |
| + } else { |
| + regkey.DeleteKey(GetBeaconKeyPath()); |
| + } |
| +} |