Keep lazy background page alive while there are pending network requests or

chrome/browser/net/chrome_network_delegate.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/net/chrome_network_delegate.h"
 
 #include "base/logging.h"
+#include "chrome/browser/browser_process.h"
 #include "chrome/browser/custom_handlers/protocol_handler_registry.h"
 #include "chrome/browser/extensions/api/webrequest/webrequest_api.h"
 #include "chrome/browser/extensions/extension_event_router_forwarder.h"
 #include "chrome/browser/extensions/extension_info_map.h"
+#include "chrome/browser/extensions/extension_process_manager.h"
 #include "chrome/browser/extensions/extension_proxy_api.h"
 #include "chrome/browser/prefs/pref_member.h"
+#include "chrome/browser/profiles/profile_manager.h"
 #include "chrome/browser/task_manager/task_manager.h"
 #include "chrome/common/pref_names.h"
+#include "content/browser/renderer_host/render_view_host.h"
+#include "content/browser/renderer_host/resource_dispatcher_host.h"
 #include "content/public/browser/browser_thread.h"
 #include "net/base/host_port_pair.h"
 #include "net/base/net_errors.h"
 #include "net/base/net_log.h"
 #include "net/http/http_request_headers.h"
 #include "net/http/http_response_headers.h"
 #include "net/url_request/url_request.h"
 
 #if defined(ENABLE_CONFIGURATION_POLICY)
 #include "chrome/browser/policy/url_blacklist_manager.h"
@@ skipping 12 matching lines @@
     switch (request->status().error()) {
       case net::ERR_PROXY_AUTH_UNSUPPORTED:
       case net::ERR_PROXY_CONNECTION_FAILED:
       case net::ERR_TUNNEL_CONNECTION_FAILED:
         ExtensionProxyEventRouter::GetInstance()->OnProxyError(
             event_router, profile, request->status().error());
     }
   }
 }
 
+enum RequestStatus { REQUEST_STARTED, REQUEST_DONE };
+
+// Notifies the ExtensionProcessManager that a request has started or stopped
+// for a particular RenderView.
+void NotifyEPMRequestStatus(RequestStatus status,
+                            void* profile_id,
+                            int process_id,
+                            int render_view_id) {
+  Profile* profile = reinterpret_cast<Profile*>(profile_id);

[Yoyo Zhou, 2012/03/02 03:10:46]

Might be a good idea to DCHECK we're on the UI thread.

[Matt Perry, 2012/03/02 20:25:21]

Done.

+  if (!g_browser_process->profile_manager()->IsValidProfile(profile))
+    return;
+
+  RenderViewHost* render_view_host =
+      RenderViewHost::FromID(process_id, render_view_id);
+  if (render_view_host) {
+    if (status == REQUEST_STARTED) {
+      profile->GetExtensionProcessManager()->OnNetworkRequestStarted(
+          render_view_host);
+    } else if (status == REQUEST_DONE) {
+      profile->GetExtensionProcessManager()->OnNetworkRequestDone(
+          render_view_host);
+    } else {
+      NOTREACHED();
+    }
+  }
+}
+
+void ForwardRequestStatus(
+    RequestStatus status, net::URLRequest* request, void* profile_id) {
+  int process_id, render_view_id;
+  if (ResourceDispatcherHost::RenderViewForRequest(
+          request, &process_id, &render_view_id)) {
+    BrowserThread::PostTask(BrowserThread::UI, FROM_HERE,
+        base::Bind(&NotifyEPMRequestStatus,
+                   status, profile_id, process_id, render_view_id));
+  }
+}
+
 }  // namespace
 
 ChromeNetworkDelegate::ChromeNetworkDelegate(
     ExtensionEventRouterForwarder* event_router,
     ExtensionInfoMap* extension_info_map,
     const policy::URLBlacklistManager* url_blacklist_manager,
     void* profile,
     BooleanPrefMember* enable_referrers)
     : event_router_(event_router),
       profile_(profile),
@@ skipping 27 matching lines @@
       url_blacklist_manager_->IsURLBlocked(request->url())) {
     // URL access blocked by policy.
     scoped_refptr<net::NetLog::EventParameters> params;
     params = new net::NetLogStringParameter("url", request->url().spec());
     request->net_log().AddEvent(
         net::NetLog::TYPE_CHROME_POLICY_ABORTED_REQUEST, params);
     return net::ERR_NETWORK_ACCESS_DENIED;
   }
 #endif
 
+  ForwardRequestStatus(REQUEST_STARTED, request, profile_);
+
   if (!enable_referrers_->GetValue())
     request->set_referrer(std::string());
   return ExtensionWebRequestEventRouter::GetInstance()->OnBeforeRequest(
       profile_, extension_info_map_.get(), request, callback, new_url);
 }
 
 int ChromeNetworkDelegate::OnBeforeSendHeaders(
     net::URLRequest* request,
     const net::CompletionCallback& callback,
     net::HttpRequestHeaders* headers) {
@@ skipping 48 matching lines @@
           profile_, extension_info_map_.get(), request);
     }
   } else if (request->status().status() == net::URLRequestStatus::FAILED ||
              request->status().status() == net::URLRequestStatus::CANCELED) {
     ExtensionWebRequestEventRouter::GetInstance()->OnErrorOccurred(
             profile_, extension_info_map_.get(), request, started);
   } else {
     NOTREACHED();
   }
   ForwardProxyErrors(request, event_router_.get(), profile_);
+
+  ForwardRequestStatus(REQUEST_DONE, request, profile_);
 }
 
 void ChromeNetworkDelegate::OnURLRequestDestroyed(net::URLRequest* request) {
   ExtensionWebRequestEventRouter::GetInstance()->OnURLRequestDestroyed(
       profile_, request);
 }
 
 void ChromeNetworkDelegate::OnPACScriptError(int line_number,
                                              const string16& error) {
   ExtensionProxyEventRouter::GetInstance()->OnPACScriptError(
       event_router_.get(), profile_, line_number, error);
 }
 
 net::NetworkDelegate::AuthRequiredResponse
 ChromeNetworkDelegate::OnAuthRequired(
     net::URLRequest* request,
     const net::AuthChallengeInfo& auth_info,
     const AuthCallback& callback,
     net::AuthCredentials* credentials) {
   return ExtensionWebRequestEventRouter::GetInstance()->OnAuthRequired(
       profile_, extension_info_map_.get(), request, auth_info,
       callback, credentials);
 }