Temporarily disable extensions and sync while a profile is locked - Profiles Approach

extensions/browser/extension_registry.cc

 // Copyright 2013 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 "extensions/browser/extension_registry.h"
 
 #include "base/strings/string_util.h"
 #include "extensions/browser/extension_registry_factory.h"
 #include "extensions/browser/extension_registry_observer.h"
 
 namespace extensions {
 
 ExtensionRegistry::ExtensionRegistry(content::BrowserContext* browser_context)
     : browser_context_(browser_context) {}
 ExtensionRegistry::~ExtensionRegistry() {}
 
 // static
 ExtensionRegistry* ExtensionRegistry::Get(content::BrowserContext* context) {
   return ExtensionRegistryFactory::GetForBrowserContext(context);
 }
 
 scoped_ptr<ExtensionSet> ExtensionRegistry::GenerateInstalledExtensionsSet()
     const {
+  return GenerateInstalledExtensionsSet(EVERYTHING).Pass();
+}
+
+scoped_ptr<ExtensionSet> ExtensionRegistry::GenerateInstalledExtensionsSet(
+    int include_mask) const {
   scoped_ptr<ExtensionSet> installed_extensions(new ExtensionSet);
-  installed_extensions->InsertAll(enabled_extensions_);
-  installed_extensions->InsertAll(disabled_extensions_);
-  installed_extensions->InsertAll(terminated_extensions_);
-  installed_extensions->InsertAll(blacklisted_extensions_);
+  if (include_mask & IncludeFlag::ENABLED)
+    installed_extensions->InsertAll(enabled_extensions_);
+  if (include_mask & IncludeFlag::DISABLED)
+    installed_extensions->InsertAll(disabled_extensions_);
+  if (include_mask & IncludeFlag::TERMINATED)
+    installed_extensions->InsertAll(terminated_extensions_);
+  if (include_mask & IncludeFlag::BLACKLISTED)
+    installed_extensions->InsertAll(blacklisted_extensions_);
+  if (include_mask & IncludeFlag::BLOCKED)
+    installed_extensions->InsertAll(blocked_extensions_);
   return installed_extensions.Pass();
 }
 
 void ExtensionRegistry::AddObserver(ExtensionRegistryObserver* observer) {
   observers_.AddObserver(observer);
 }
 
 void ExtensionRegistry::RemoveObserver(ExtensionRegistryObserver* observer) {
   observers_.RemoveObserver(observer);
 }
@@ skipping 62 matching lines @@
   if (include_mask & TERMINATED) {
     const Extension* extension = terminated_extensions_.GetByID(lowercase_id);
     if (extension)
       return extension;
   }
   if (include_mask & BLACKLISTED) {
     const Extension* extension = blacklisted_extensions_.GetByID(lowercase_id);
     if (extension)
       return extension;
   }
+  if (include_mask & BLOCKED) {
+    const Extension* extension = blocked_extensions_.GetByID(lowercase_id);
+    if (extension)
+      return extension;
+  }
   return NULL;
 }
 
 bool ExtensionRegistry::AddEnabled(
     const scoped_refptr<const Extension>& extension) {
   return enabled_extensions_.Insert(extension);
 }
 
 bool ExtensionRegistry::RemoveEnabled(const std::string& id) {
   return enabled_extensions_.Remove(id);
@@ skipping 19 matching lines @@
 
 bool ExtensionRegistry::AddBlacklisted(
     const scoped_refptr<const Extension>& extension) {
   return blacklisted_extensions_.Insert(extension);
 }
 
 bool ExtensionRegistry::RemoveBlacklisted(const std::string& id) {
   return blacklisted_extensions_.Remove(id);
 }
 
+bool ExtensionRegistry::AddBlocked(
+    const scoped_refptr<const Extension>& extension) {
+  return blocked_extensions_.Insert(extension);
+}
+
+bool ExtensionRegistry::RemoveBlocked(const std::string& id) {
+  return blocked_extensions_.Remove(id);
+}
+
 void ExtensionRegistry::ClearAll() {
   enabled_extensions_.Clear();
   disabled_extensions_.Clear();
   terminated_extensions_.Clear();
   blacklisted_extensions_.Clear();
+  blocked_extensions_.Clear();
 }
 
 void ExtensionRegistry::SetDisabledModificationCallback(
     const ExtensionSet::ModificationCallback& callback) {
   disabled_extensions_.set_modification_callback(callback);
 }
 
 void ExtensionRegistry::Shutdown() {
   // Release references to all Extension objects in the sets.
   ClearAll();
   FOR_EACH_OBSERVER(ExtensionRegistryObserver, observers_, OnShutdown(this));
 }
 
 }  // namespace extensions