Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(1567)

Unified Diff: chrome/browser/extensions/external_registry_loader_win.cc

Issue 1495403002: Observe adding external extensions via windows registry. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: guard updater_->CheckNow() call, rework Provider->ExtensionService interaction Created 4 years, 11 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: chrome/browser/extensions/external_registry_loader_win.cc
diff --git a/chrome/browser/extensions/external_registry_loader_win.cc b/chrome/browser/extensions/external_registry_loader_win.cc
index 0b5a67f83687c838b7723790af6f0953dc435e9f..2acc6c08373583212131edf88060e3f51cd661b4 100644
--- a/chrome/browser/extensions/external_registry_loader_win.cc
+++ b/chrome/browser/extensions/external_registry_loader_win.cc
@@ -61,9 +61,8 @@ void ExternalRegistryLoader::StartLoading() {
base::Bind(&ExternalRegistryLoader::LoadOnFileThread, this));
}
-void ExternalRegistryLoader::LoadOnFileThread() {
+void ExternalRegistryLoader::LoadPrefsOnFileThread() {
CHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE));
- base::TimeTicks start_time = base::TimeTicks::Now();
scoped_ptr<base::DictionaryValue> prefs(new base::DictionaryValue);
// A map of IDs, to weed out duplicates between HKCU and HKLM.
@@ -183,11 +182,67 @@ void ExternalRegistryLoader::LoadOnFileThread() {
}
prefs_.reset(prefs.release());
+}
+
+void ExternalRegistryLoader::LoadOnFileThread() {
+ base::TimeTicks start_time = base::TimeTicks::Now();
+ LoadPrefsOnFileThread();
LOCAL_HISTOGRAM_TIMES("Extensions.ExternalRegistryLoaderWin",
base::TimeTicks::Now() - start_time);
BrowserThread::PostTask(
BrowserThread::UI, FROM_HERE,
- base::Bind(&ExternalRegistryLoader::LoadFinished, this));
+ base::Bind(&ExternalRegistryLoader::CompleteLoadAndStartWatchingRegistry,
+ this));
+}
+
+void ExternalRegistryLoader::CompleteLoadAndStartWatchingRegistry() {
+ CHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
+ LoadFinished();
+
+ // Start watching registry.
+ // TODO(lazyboy): Is using KEY_WOW64_32KEY correct?
+ if (hklm_key_.Create(HKEY_LOCAL_MACHINE, kRegistryExtensions,
+ KEY_NOTIFY | KEY_WOW64_32KEY) == ERROR_SUCCESS) {
+ base::win::RegKey::ChangeCallback callback =
+ base::Bind(&ExternalRegistryLoader::OnRegistryKeyChanged,
+ base::Unretained(this), base::Unretained(&hklm_key_));
+ hklm_key_.StartWatching(callback);
+ } else {
+ // TODO(lazyboy): Error handling.
+ LOG(ERROR) << "Error observing HKLM.";
+ }
+
+ if (hkcu_key_.Create(HKEY_CURRENT_USER, kRegistryExtensions, KEY_NOTIFY) ==
+ ERROR_SUCCESS) {
+ base::win::RegKey::ChangeCallback callback =
+ base::Bind(&ExternalRegistryLoader::OnRegistryKeyChanged,
+ base::Unretained(this), base::Unretained(&hkcu_key_));
+ hkcu_key_.StartWatching(callback);
+ } else {
+ // TODO(lazyboy): Error handling.
+ LOG(ERROR) << "Error observing HKCU.";
+ }
+}
+
+void ExternalRegistryLoader::OnRegistryKeyChanged(base::win::RegKey* key) {
+ // |OnRegistryKeyChanged| is removed as an observer when the ChangeCallback is
+ // called, so we need to re-register.
+ key->StartWatching(base::Bind(&ExternalRegistryLoader::OnRegistryKeyChanged,
+ base::Unretained(this), base::Unretained(key)));
+
+ BrowserThread::PostTask(
+ BrowserThread::FILE, FROM_HERE,
+ base::Bind(&ExternalRegistryLoader::UpdatePrefsOnFileThread, this));
+}
+
+void ExternalRegistryLoader::UpdatePrefsOnFileThread() {
+ CHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE));
+ base::TimeTicks start_time = base::TimeTicks::Now();
+ LoadPrefsOnFileThread();
+ LOCAL_HISTOGRAM_TIMES("Extensions.ExternalRegistryLoaderWinUpdate",
+ base::TimeTicks::Now() - start_time);
+ BrowserThread::PostTask(BrowserThread::UI, FROM_HERE,
+ base::Bind(&ExternalRegistryLoader::OnUpdated, this));
}
} // namespace extensions

Powered by Google App Engine
This is Rietveld 408576698