Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "base/bind.h" | 5 #include "base/bind.h" |
| 6 #include "base/metrics/field_trial.h" | 6 #include "base/metrics/field_trial.h" |
| 7 #include "base/metrics/histogram.h" | 7 #include "base/metrics/histogram.h" |
| 8 #include "base/metrics/sparse_histogram.h" | 8 #include "base/metrics/sparse_histogram.h" |
| 9 #include "base/strings/utf_string_conversions.h" | 9 #include "base/strings/utf_string_conversions.h" |
| 10 #include "base/win/registry.h" | 10 #include "base/win/registry.h" |
| 11 #include "chrome/browser/chrome_elf_init_win.h" | 11 #include "chrome/browser/chrome_elf_init_win.h" |
| 12 #include "chrome_elf/blacklist/blacklist.h" | 12 #include "chrome_elf/blacklist/blacklist.h" |
| 13 #include "chrome_elf/chrome_elf_constants.h" | 13 #include "chrome_elf/chrome_elf_constants.h" |
| 14 #include "chrome_elf/dll_hash/dll_hash.h" | 14 #include "chrome_elf/dll_hash/dll_hash.h" |
| 15 #include "components/variations/variations_associated_data.h" | |
| 15 #include "content/public/browser/browser_thread.h" | 16 #include "content/public/browser/browser_thread.h" |
| 16 #include "version.h" // NOLINT | 17 #include "version.h" // NOLINT |
| 17 | 18 |
| 18 const char kBrowserBlacklistTrialName[] = "BrowserBlacklist"; | 19 const char kBrowserBlacklistTrialName[] = "BrowserBlacklist"; |
| 19 const char kBrowserBlacklistTrialDisabledGroupName[] = "NoBlacklist"; | 20 const char kBrowserBlacklistTrialDisabledGroupName[] = "NoBlacklist"; |
| 20 | 21 |
| 21 namespace { | 22 namespace { |
| 22 | 23 |
| 23 // How long to wait, in seconds, before reporting for the second (and last | 24 // How long to wait, in seconds, before reporting for the second (and last |
| 24 // time), what dlls were blocked from the browser process. | 25 // time), what dlls were blocked from the browser process. |
| (...skipping 55 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 80 | 81 |
| 81 } // namespace | 82 } // namespace |
| 82 | 83 |
| 83 void InitializeChromeElf() { | 84 void InitializeChromeElf() { |
| 84 if (base::FieldTrialList::FindFullName(kBrowserBlacklistTrialName) == | 85 if (base::FieldTrialList::FindFullName(kBrowserBlacklistTrialName) == |
| 85 kBrowserBlacklistTrialDisabledGroupName) { | 86 kBrowserBlacklistTrialDisabledGroupName) { |
| 86 // Disable the blacklist for all future runs by removing the beacon. | 87 // Disable the blacklist for all future runs by removing the beacon. |
| 87 base::win::RegKey blacklist_registry_key(HKEY_CURRENT_USER); | 88 base::win::RegKey blacklist_registry_key(HKEY_CURRENT_USER); |
| 88 blacklist_registry_key.DeleteKey(blacklist::kRegistryBeaconPath); | 89 blacklist_registry_key.DeleteKey(blacklist::kRegistryBeaconPath); |
| 89 } else { | 90 } else { |
| 91 AddFinchBlacklistToRegistry(); | |
| 90 BrowserBlacklistBeaconSetup(); | 92 BrowserBlacklistBeaconSetup(); |
| 91 } | 93 } |
| 92 | 94 |
| 93 // Report all successful blacklist interceptions. | 95 // Report all successful blacklist interceptions. |
| 94 ReportSuccessfulBlocks(); | 96 ReportSuccessfulBlocks(); |
| 95 | 97 |
| 96 // Schedule another task to report all sucessful interceptions later. | 98 // Schedule another task to report all sucessful interceptions later. |
| 97 // This time delay should be long enough to catch any dlls that attempt to | 99 // This time delay should be long enough to catch any dlls that attempt to |
| 98 // inject after Chrome has started up. | 100 // inject after Chrome has started up. |
| 99 content::BrowserThread::PostDelayedTask( | 101 content::BrowserThread::PostDelayedTask( |
| 100 content::BrowserThread::UI, | 102 content::BrowserThread::UI, |
| 101 FROM_HERE, | 103 FROM_HERE, |
| 102 base::Bind(&ReportSuccessfulBlocks), | 104 base::Bind(&ReportSuccessfulBlocks), |
| 103 base::TimeDelta::FromSeconds(kBlacklistReportingDelaySec)); | 105 base::TimeDelta::FromSeconds(kBlacklistReportingDelaySec)); |
| 104 } | 106 } |
| 105 | 107 |
| 108 void AddFinchBlacklistToRegistry() { | |
| 109 std::map<std::string, std::string> params; | |
| 110 if (chrome_variations::GetVariationParams(kBrowserBlacklistTrialName, | |
| 111 ¶ms)) { | |
| 112 base::win::RegKey finch_blacklist_registry_key( | |
| 113 HKEY_CURRENT_USER, blacklist::kRegistryFinchListPath, KEY_SET_VALUE); | |
| 114 | |
| 115 // No point in trying to continue if the registry key isn't valid. | |
| 116 if (!finch_blacklist_registry_key.Valid()) | |
| 117 return; | |
| 118 | |
| 119 std::map<std::string, std::string>::iterator it = params.begin(); | |
| 120 while (it != params.end()) { | |
|
robertshield
2014/06/03 03:03:00
prefer for (; it != params.end(); ++it) { ... } fo
krstnmnlsn
2014/06/03 19:30:16
Done.
| |
| 121 std::wstring name = base::UTF8ToWide(it->first); | |
| 122 std::wstring val = base::UTF8ToWide(it->second); | |
| 123 | |
| 124 finch_blacklist_registry_key.WriteValue(name.c_str(), val.c_str()); | |
| 125 | |
| 126 ++it; | |
| 127 } | |
| 128 } | |
| 129 } | |
| 130 | |
| 106 void BrowserBlacklistBeaconSetup() { | 131 void BrowserBlacklistBeaconSetup() { |
| 107 base::win::RegKey blacklist_registry_key(HKEY_CURRENT_USER, | 132 base::win::RegKey blacklist_registry_key(HKEY_CURRENT_USER, |
| 108 blacklist::kRegistryBeaconPath, | 133 blacklist::kRegistryBeaconPath, |
| 109 KEY_QUERY_VALUE | KEY_SET_VALUE); | 134 KEY_QUERY_VALUE | KEY_SET_VALUE); |
| 110 | 135 |
| 111 // No point in trying to continue if the registry key isn't valid. | 136 // No point in trying to continue if the registry key isn't valid. |
| 112 if (!blacklist_registry_key.Valid()) | 137 if (!blacklist_registry_key.Valid()) |
| 113 return; | 138 return; |
| 114 | 139 |
| 115 // Record the results of the last blacklist setup. | 140 // Record the results of the last blacklist setup. |
| (...skipping 37 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 153 LONG set_state = blacklist_registry_key.WriteValue( | 178 LONG set_state = blacklist_registry_key.WriteValue( |
| 154 blacklist::kBeaconState, | 179 blacklist::kBeaconState, |
| 155 blacklist::BLACKLIST_ENABLED); | 180 blacklist::BLACKLIST_ENABLED); |
| 156 | 181 |
| 157 // Only report the blacklist as getting setup when both registry writes | 182 // Only report the blacklist as getting setup when both registry writes |
| 158 // succeed, since otherwise the blacklist wasn't properly setup. | 183 // succeed, since otherwise the blacklist wasn't properly setup. |
| 159 if (set_version == ERROR_SUCCESS && set_state == ERROR_SUCCESS) | 184 if (set_version == ERROR_SUCCESS && set_state == ERROR_SUCCESS) |
| 160 RecordBlacklistSetupEvent(BLACKLIST_SETUP_ENABLED); | 185 RecordBlacklistSetupEvent(BLACKLIST_SETUP_ENABLED); |
| 161 } | 186 } |
| 162 } | 187 } |
| OLD | NEW |