Have LazyBackgroundTaskQueue notify consumers when the transient page fails to

chrome/browser/extensions/lazy_background_task_queue.cc

 // Copyright (c) 2012 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/lazy_background_task_queue.h"
 
 #include "base/callback.h"
 #include "chrome/browser/extensions/extension_host.h"
 #include "chrome/browser/extensions/extension_process_manager.h"
 #include "chrome/browser/extensions/extension_service.h"
+#include "chrome/browser/extensions/extension_system.h"
 #include "chrome/browser/extensions/extension_tab_util.h"
 #include "chrome/browser/extensions/process_map.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/tab_contents/tab_util.h"
 #include "chrome/browser/ui/tab_contents/tab_contents_wrapper.h"
 #include "chrome/common/chrome_notification_types.h"
 #include "chrome/common/chrome_view_type.h"
 #include "chrome/common/extensions/extension.h"
 #include "chrome/common/extensions/extension_messages.h"
 #include "content/public/browser/notification_service.h"
@@ skipping 37 matching lines @@
     const PendingTask& task) {
   PendingTasksList* tasks_list = NULL;
   PendingTasksKey key(profile, extension_id);
   PendingTasksMap::iterator it = pending_tasks_.find(key);
   if (it == pending_tasks_.end()) {
     tasks_list = new PendingTasksList();
     pending_tasks_[key] = linked_ptr<PendingTasksList>(tasks_list);
 
     // If this is the first enqueued task, ensure the background page
     // is loaded.
-    const Extension* extension = profile->GetExtensionService()->
-        extensions()->GetByID(extension_id);
+    const Extension* extension =
+        ExtensionSystem::Get(profile)->extension_service()->
+            extensions()->GetByID(extension_id);
     DCHECK(extension->has_lazy_background_page());
-    ExtensionProcessManager* pm = profile->GetExtensionProcessManager();
+    ExtensionProcessManager* pm =
+        ExtensionSystem::Get(profile)->process_manager();
     pm->IncrementLazyKeepaliveCount(extension);
     pm->CreateBackgroundHost(extension, extension->GetBackgroundURL());
   } else {
     tasks_list = it->second.get();
   }
 
   tasks_list->push_back(task);
 }
 
-void LazyBackgroundTaskQueue::ProcessPendingTasks(ExtensionHost* host) {
-  PendingTasksKey key(host->profile(), host->extension()->id());
+void LazyBackgroundTaskQueue::ProcessPendingTasks(
+    ExtensionHost* host,
+    Profile* profile,
+    const Extension* extension) {
+  if (!profile->IsSameProfile(profile_) ||
+      !extension->has_lazy_background_page())
+    return;
+
+  PendingTasksKey key(profile, extension->id());
   PendingTasksMap::iterator map_it = pending_tasks_.find(key);
   if (map_it == pending_tasks_.end()) {
-    NOTREACHED();  // lazy page should not load without any pending tasks
+    CHECK(!host);  // lazy page should not load without any pending tasks
     return;
   }
 
   PendingTasksList* tasks = map_it->second.get();
   for (PendingTasksList::const_iterator it = tasks->begin();
        it != tasks->end(); ++it) {
     it->Run(host);
   }
 
   tasks->clear();
   pending_tasks_.erase(map_it);
 
-  // Balance the keepalive in AddPendingTask.
-  host->profile()->GetExtensionProcessManager()->
-      DecrementLazyKeepaliveCount(host->extension());
+  // Balance the keepalive in AddPendingTask. Note we don't do this on a
+  // failure to load, because the keepalive count is reset in that case.
+  if (host) {
+    ExtensionSystem::Get(profile)->process_manager()->
+        DecrementLazyKeepaliveCount(extension);
+  }
 }
 
 void LazyBackgroundTaskQueue::Observe(
     int type,
     const content::NotificationSource& source,
     const content::NotificationDetails& details) {
   switch (type) {
     case chrome::NOTIFICATION_EXTENSION_HOST_DID_STOP_LOADING: {
       // If an on-demand background page finished loading, dispatch queued up
       // events for it.
       ExtensionHost* host = content::Details<ExtensionHost>(details).ptr();
-      if (host->profile()->IsSameProfile(profile_) &&
-          host->extension_host_type() ==
-              chrome::VIEW_TYPE_EXTENSION_BACKGROUND_PAGE &&
-          host->extension()->has_lazy_background_page()) {
+      if (host->extension_host_type() ==
+              chrome::VIEW_TYPE_EXTENSION_BACKGROUND_PAGE) {
         CHECK(host->did_stop_loading());
-        ProcessPendingTasks(host);
+        ProcessPendingTasks(host, host->profile(), host->extension());
       }
       break;
     }
     case chrome::NOTIFICATION_EXTENSION_HOST_DESTROYED: {
-      // Clear pending tasks when the background host dies. This can happen
-      // if the extension crashes. This is not strictly necessary, since we
-      // also unload the extension in that case (which clears the tasks below),
-      // but is a good extra precaution.
+      // Notify consumers about the load failure when the background host dies.
+      // This can happen if the extension crashes. This is not strictly
+      // necessary, since we also unload the extension in that case (which
+      // dispatches the tasks below), but is a good extra precaution.
       Profile* profile = content::Source<Profile>(source).ptr();
       ExtensionHost* host = content::Details<ExtensionHost>(details).ptr();
       if (host->extension_host_type() ==
               chrome::VIEW_TYPE_EXTENSION_BACKGROUND_PAGE) {
-        pending_tasks_.erase(PendingTasksKey(profile, host->extension()->id()));
+        ProcessPendingTasks(NULL, profile, host->extension());
       }
       break;
     }
     case chrome::NOTIFICATION_EXTENSION_UNLOADED: {
-      // Clear pending tasks for this extension.
+      // Notify consumers that the page failed to load.
       Profile* profile = content::Source<Profile>(source).ptr();
       UnloadedExtensionInfo* unloaded =
           content::Details<UnloadedExtensionInfo>(details).ptr();
-      pending_tasks_.erase(PendingTasksKey(
-          profile, unloaded->extension->id()));
-      if (profile->HasOffTheRecordProfile())
-        pending_tasks_.erase(PendingTasksKey(
-            profile->GetOffTheRecordProfile(), unloaded->extension->id()));
+      ProcessPendingTasks(NULL, profile, unloaded->extension);
+      if (profile->HasOffTheRecordProfile()) {
+        ProcessPendingTasks(NULL, profile->GetOffTheRecordProfile(),
+                            unloaded->extension);
+      }
       break;
     }
     default:
       NOTREACHED();
       break;
   }
 }
 
 }  // namespace extensions