Warn user in case extension delays network traffic too much.

chrome/browser/extensions/extension_service.cc

 // Copyright (c) 2011 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/browser/extensions/extension_service.h"
 
 #include <algorithm>
 #include <set>
 
 #include "base/basictypes.h"
@@ skipping 17 matching lines @@
 #include "chrome/browser/extensions/app_notification_manager.h"
 #include "chrome/browser/extensions/apps_promo.h"
 #include "chrome/browser/extensions/crx_installer.h"
 #include "chrome/browser/extensions/extension_accessibility_api.h"
 #include "chrome/browser/extensions/extension_bookmarks_module.h"
 #include "chrome/browser/extensions/extension_browser_event_router.h"
 #include "chrome/browser/extensions/extension_cookies_api.h"
 #include "chrome/browser/extensions/extension_data_deleter.h"
 #include "chrome/browser/extensions/extension_downloads_api.h"
 #include "chrome/browser/extensions/extension_error_reporter.h"
+#include "chrome/browser/extensions/extension_global_error.h"
 #include "chrome/browser/extensions/extension_history_api.h"
 #include "chrome/browser/extensions/extension_host.h"
 #include "chrome/browser/extensions/extension_input_ime_api.h"
 #include "chrome/browser/extensions/extension_install_ui.h"
 #include "chrome/browser/extensions/extension_management_api.h"
 #include "chrome/browser/extensions/extension_preference_api.h"
 #include "chrome/browser/extensions/extension_process_manager.h"
 #include "chrome/browser/extensions/extension_processes_api.h"
 #include "chrome/browser/extensions/extension_special_storage_policy.h"
 #include "chrome/browser/extensions/extension_sync_data.h"
 #include "chrome/browser/extensions/extension_updater.h"
 #include "chrome/browser/extensions/extension_web_ui.h"
 #include "chrome/browser/extensions/extension_webnavigation_api.h"
 #include "chrome/browser/extensions/external_extension_provider_impl.h"
 #include "chrome/browser/extensions/external_extension_provider_interface.h"
 #include "chrome/browser/extensions/pending_extension_manager.h"
 #include "chrome/browser/net/chrome_url_request_context.h"
 #include "chrome/browser/prefs/pref_service.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/search_engines/template_url_service.h"
 #include "chrome/browser/search_engines/template_url_service_factory.h"
 #include "chrome/browser/sync/api/sync_change.h"
 #include "chrome/browser/themes/theme_service.h"
 #include "chrome/browser/themes/theme_service_factory.h"
+#include "chrome/browser/ui/global_error_service.h"
+#include "chrome/browser/ui/global_error_service_factory.h"
 #include "chrome/browser/ui/webui/chrome_url_data_manager.h"
 #include "chrome/browser/ui/webui/favicon_source.h"
 #include "chrome/browser/ui/webui/ntp/shown_sections_handler.h"
 #include "chrome/common/child_process_logging.h"
 #include "chrome/common/chrome_notification_types.h"
 #include "chrome/common/chrome_paths.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/extensions/extension.h"
 #include "chrome/common/extensions/extension_constants.h"
 #include "chrome/common/extensions/extension_error_utils.h"
@@ skipping 134 matching lines @@
         extension_, false, -1);  // Not from web store.
   delete this;
 }
 
 void SimpleExtensionLoadPrompt::InstallUIAbort(bool user_initiated) {
   delete this;
 }
 
 }  // namespace
 
+
+ExtensionServiceWarning::ExtensionServiceWarning()
+    : type_(ExtensionServiceWarning::kInvalid) {
+}
+
+ExtensionServiceWarning::ExtensionServiceWarning(
+    WarningType type,
+    const std::set<std::string>& extension_ids)
+    : type_(type), extension_ids_(extension_ids) {
+}
+
+ExtensionServiceWarning::~ExtensionServiceWarning() {
+}
+
+ExtensionServiceWarning::WarningType
+ExtensionServiceWarning::warning_type() const {
+  return type_;
+}
+
+const std::set<std::string> ExtensionServiceWarning::extension_ids() const {
+  return extension_ids_;
+}
+
+bool ExtensionServiceWarning::AffectsExtension(const std::string& id) const {
+  return extension_ids_.find(id) != extension_ids_.end();
+}
+
 bool ExtensionService::ComponentExtensionInfo::Equals(
     const ComponentExtensionInfo& other) const {
   return other.manifest == manifest && other.root_directory == root_directory;
 }
 
 ExtensionService::ExtensionRuntimeData::ExtensionRuntimeData()
     : background_page_ready(false),
       being_upgraded(false) {
 }
 
@@ skipping 365 matching lines @@
           profile->GetPath().AppendASCII(kSettingsDirectoryName)),
       pending_extension_manager_(*ALLOW_THIS_IN_INITIALIZER_LIST(this)),
       install_directory_(install_directory),
       extensions_enabled_(extensions_enabled),
       show_extensions_prompts_(true),
       ready_(false),
       toolbar_model_(ALLOW_THIS_IN_INITIALIZER_LIST(this)),
       app_notification_manager_(new AppNotificationManager(profile)),
       permissions_manager_(ALLOW_THIS_IN_INITIALIZER_LIST(this)),
       apps_promo_(profile->GetPrefs()),
-      event_routers_initialized_(false) {
+      event_routers_initialized_(false),
+      extension_global_error_(NULL) {
   CHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
 
   // Figure out if extension installation should be enabled.
   if (command_line->HasSwitch(switches::kDisableExtensions)) {
     extensions_enabled_ = false;
   } else if (profile->GetPrefs()->GetBoolean(prefs::kDisableExtensions)) {
     extensions_enabled_ = false;
   }
 
   registrar_.Add(this, chrome::NOTIFICATION_EXTENSION_PROCESS_TERMINATED,
@@ skipping 349 matching lines @@
   if (sync_bundle && sync_bundle->HasExtensionId(extension_id)) {
     sync_bundle->sync_processor->ProcessSyncChanges(
         FROM_HERE, SyncChangeList(1, sync_change));
     sync_bundle->synced_extensions.erase(extension_id);
   }
 
   // Track the uninstallation.
   UserMetrics::RecordAction(
       UserMetricsAction("Extensions.ExtensionUninstalled"));
 
+  ClearWarningsAffectingExtension(extension_id);
+
   return true;
 }
 
 void ExtensionService::ClearExtensionData(const std::string& extension_id,
                                           const GURL& storage_url,
                                           bool is_storage_isolated) {
   scoped_refptr<ExtensionDataDeleter> deleter(new ExtensionDataDeleter(
       profile_, extension_id, storage_url, is_storage_isolated));
   deleter->StartDeleting();
 }
@@ skipping 80 matching lines @@
   } else {
     iter = std::find(terminated_extensions_.begin(),
                      terminated_extensions_.end(),
                      extension);
     terminated_extensions_.erase(iter);
   }
 
   NotifyExtensionUnloaded(extension, extension_misc::UNLOAD_REASON_DISABLE);
 
   SyncExtensionChangeIfNeeded(*extension);
+
+  ClearWarningsAffectingExtension(extension_id);
 }
 
 void ExtensionService::GrantPermissions(const Extension* extension) {
   CHECK(extension);
 
   // We only maintain the granted permissions prefs for extensions that can't
   // silently increase their permissions.
   if (extension->CanSilentlyIncreasePermissions())
     return;
 
@@ skipping 1824 matching lines @@
 
 ExtensionService::NaClModuleInfoList::iterator
     ExtensionService::FindNaClModule(const GURL& url) {
   for (NaClModuleInfoList::iterator iter = nacl_module_list_.begin();
        iter != nacl_module_list_.end(); ++iter) {
     if (iter->url == url)
       return iter;
   }
   return nacl_module_list_.end();
 }
+
+void ExtensionService::SetWarning(const ExtensionServiceWarning& warning) {
+  extension_warnings_[warning.warning_type()] = warning;
+  NotificationService::current()->Notify(
+        chrome::NOTIFICATION_EXTENSION_WARNING,
+        Source<Profile>(profile_),
+        NotificationService::NoDetails());
+  UpdateWarningBadge();
+}
+
+void ExtensionService::ClearWarning(ExtensionServiceWarning::WarningType type) {
+  if (extension_warnings_.erase(type)) {
+    NotificationService::current()->Notify(
+          chrome::NOTIFICATION_EXTENSION_WARNING,
+          Source<Profile>(profile_),
+          NotificationService::NoDetails());
+  }
+  UpdateWarningBadge();
+}
+
+void ExtensionService::ClearWarningsAffectingExtension(
+    const std::string& extension_id) {
+  typedef std::set<ExtensionServiceWarning::WarningType> WarningTypeSet;
+  WarningTypeSet to_clear;
+
+  for (ExtensionWarningsType::const_iterator i = extension_warnings_.begin();
+       i != extension_warnings_.end(); ++i) {
+    if (i->second.AffectsExtension(extension_id))
+      to_clear.insert(i->first);
+  }
+
+  for (WarningTypeSet::const_iterator i = to_clear.begin();
+       i != to_clear.end(); ++i) {
+    ClearWarning(*i);
+  }
+}
+
+ExtensionService::ExtensionWarningsType ExtensionService::GetWarnings() const {
+  return extension_warnings_;
+}
+
+bool ExtensionService::HasWarning(
+    ExtensionServiceWarning::WarningType type) const {
+  return extension_warnings_.find(type) != extension_warnings_.end();
+}
+
+void ExtensionService::SuppressWarning(
+    ExtensionServiceWarning::WarningType type) {
+  suppressed_warnings_.insert(type);
+  UpdateWarningBadge();
+}
+
+void ExtensionService::UpdateWarningBadge() {
+  // We need a badge if a warning exists that has not been suppressed.
+  bool need_warning_badge = false;
+  for (ExtensionWarningsType::const_iterator i = extension_warnings_.begin();
+       i != extension_warnings_.end(); ++i) {
+    if (suppressed_warnings_.find(i->first) == suppressed_warnings_.end()) {
+      need_warning_badge = true;
+      break;
+    }
+  }
+
+  // Activate or hide the warning bade in case the current state is incorrect.
+  if (extension_global_error_ && !need_warning_badge) {
+    GlobalErrorServiceFactory::GetForProfile(profile_)->RemoveGlobalError(
+        extension_global_error_);
+    delete extension_global_error_;
+    extension_global_error_ = NULL;
+  } else if (!extension_global_error_ && need_warning_badge) {
+    extension_global_error_ = new ExtensionGlobalError;
+    GlobalErrorServiceFactory::GetForProfile(profile_)->AddGlobalError(
+        extension_global_error_);
+  }
+}