chrome/common/terminate_on_heap_corruption_experiment_win.cc

   1 // Copyright 2014 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 "chrome/common/terminate_on_heap_corruption_experiment_win.h"
   6
   7 #include "base/command_line.h"
   8 #include "base/metrics/field_trial.h"
   9 #include "base/win/registry.h"
  10 #include "chrome/common/chrome_constants.h"
  11 #include "chrome/common/chrome_version_info.h"
  12
  13 #if defined(OS_WIN)
  14 #if defined(GOOGLE_CHROME_BUILD)
  15 #define PRODUCT_STRING_PATH L"Google\\Chrome"
  16 #elif defined(CHROMIUM_BUILD)
  17 #define PRODUCT_STRING_PATH L"Chromium"
  18 #else
  19 #error Unknown branding
  20 #endif
  21 #endif  // defined(OS_WIN)
  22
  23 namespace {
  24
  25 const wchar_t* GetBeaconKeyPath() {
  26   chrome::VersionInfo::Channel channel = chrome::VersionInfo::CHANNEL_UNKNOWN;
  27
  28   // We are called quite early, before the CommandLine is initialized. We don't
  29   // want to permanently initialize it because ContentMainRunner::Initialize
  30   // sets some locale-related stuff to make sure it is parsed properly. But we
  31   // can temporarily initialize it for the purpose of determining if we are
  32   // Canary.
  33   if (!CommandLine::InitializedForCurrentProcess()) {
  34     CommandLine::Init(0, NULL);
  35     channel = chrome::VersionInfo::GetChannel();
  36     CommandLine::Reset();
  37   } else {
  38     channel = chrome::VersionInfo::GetChannel();
  39   }
  40
  41   if (channel == chrome::VersionInfo::CHANNEL_CANARY) {
  42     return L"SOFTWARE\\" PRODUCT_STRING_PATH
  43         L"\\DisableTerminateOnProcessHeapCorruptionSxs";
  44   }
  45   return L"SOFTWARE\\" PRODUCT_STRING_PATH
  46       L"\\DisableTerminateOnProcessHeapCorruption";
  47 }
  48
  49 }  // namespace
  50
  51 bool ShouldExperimentallyDisableTerminateOnHeapCorruption() {
  52   base::win::RegKey regkey(
  53       HKEY_CURRENT_USER, GetBeaconKeyPath(), KEY_QUERY_VALUE);
  54   return regkey.Valid();
  55 }
  56
  57 void InitializeDisableTerminateOnHeapCorruptionExperiment() {
  58   base::win::RegKey regkey(HKEY_CURRENT_USER);
  59
  60   if (base::FieldTrialList::FindFullName("TerminateOnProcessHeapCorruption") ==
  61       "Disabled") {
  62     regkey.CreateKey(GetBeaconKeyPath(), KEY_SET_VALUE);
  63   } else {
  64     regkey.DeleteKey(GetBeaconKeyPath());
  65   }
  66 }