[on hold] Remove dependency on RenderViewHost from LoadState notifications

content/browser/loader/resource_dispatcher_host_impl.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.
 
 // See http://dev.chromium.org/developers/design-documents/multi-process-resource-loading
 
 #include "content/browser/loader/resource_dispatcher_host_impl.h"
 
 #include <stddef.h>
 
@@ skipping 38 matching lines @@
 #include "content/browser/loader/async_revalidation_manager.h"
 #include "content/browser/loader/cross_site_resource_handler.h"
 #include "content/browser/loader/detachable_resource_handler.h"
 #include "content/browser/loader/mime_type_resource_handler.h"
 #include "content/browser/loader/navigation_resource_handler.h"
 #include "content/browser/loader/navigation_resource_throttle.h"
 #include "content/browser/loader/navigation_url_loader_impl_core.h"
 #include "content/browser/loader/power_save_block_resource_throttle.h"
 #include "content/browser/loader/redirect_to_file_resource_handler.h"
 #include "content/browser/loader/resource_message_filter.h"
-#include "content/browser/loader/resource_request_info_impl.h"
 #include "content/browser/loader/stream_resource_handler.h"
 #include "content/browser/loader/sync_resource_handler.h"
 #include "content/browser/loader/throttling_resource_handler.h"
 #include "content/browser/loader/upload_data_stream_builder.h"
 #include "content/browser/renderer_host/render_view_host_delegate.h"
 #include "content/browser/renderer_host/render_view_host_impl.h"
 #include "content/browser/resource_context_impl.h"
 #include "content/browser/service_worker/foreign_fetch_request_handler.h"
 #include "content/browser/service_worker/link_header_support.h"
 #include "content/browser/service_worker/service_worker_request_handler.h"
@@ skipping 2369 matching lines @@
 
 net::URLRequest* ResourceDispatcherHostImpl::GetURLRequest(
     const GlobalRequestID& id) {
   ResourceLoader* loader = GetLoader(id);
   if (!loader)
     return NULL;
 
   return loader->request();
 }
 
-// static
-bool ResourceDispatcherHostImpl::LoadInfoIsMoreInteresting(const LoadInfo& a,
-                                                           const LoadInfo& b) {
-  // Set |*_uploading_size| to be the size of the corresponding upload body if
-  // it's currently being uploaded.
+ResourceDispatcherHostImpl::LoadInfo::LoadInfo() {}
 
-  uint64_t a_uploading_size = 0;
-  if (a.load_state.state == net::LOAD_STATE_SENDING_REQUEST)
-    a_uploading_size = a.upload_size;
+ResourceDispatcherHostImpl::LoadInfo::~LoadInfo() {}
 
-  uint64_t b_uploading_size = 0;
-  if (b.load_state.state == net::LOAD_STATE_SENDING_REQUEST)
-    b_uploading_size = b.upload_size;
+ResourceDispatcherHostImpl::LoadInfo::LoadInfo(const LoadInfo& info) = default;
 
-  if (a_uploading_size != b_uploading_size)
-    return a_uploading_size > b_uploading_size;
-
-  return a.load_state.state > b.load_state.state;
+bool ResourceDispatcherHostImpl::LoadInfo::IsUploading() const {
+  return load_state.state == net::LOAD_STATE_SENDING_REQUEST && upload_size > 0;
 }
 
-// static
+bool ResourceDispatcherHostImpl::LoadInfo::IsMoreInteresting(
+    const LoadInfo& other) const {
+  if (IsUploading())
+    return !other.IsUploading() || upload_size > other.upload_size;
+  return !other.IsUploading() && load_state.state > other.load_state.state;
+}
+
 void ResourceDispatcherHostImpl::UpdateLoadInfoOnUIThread(
     std::unique_ptr<LoadInfoMap> info_map) {
   // TODO(erikchen): Remove ScopedTracker below once http://crbug.com/466285
   // is fixed.
   tracked_objects::ScopedTracker tracking_profile(
       FROM_HERE_WITH_EXPLICIT_FUNCTION(
           "466285 ResourceDispatcherHostImpl::UpdateLoadInfoOnUIThread"));
   DCHECK_CURRENTLY_ON(BrowserThread::UI);
+  std::map<WebContents*, const LoadInfo*> contents_map;
   for (const auto& load_info : *info_map) {
-    RenderViewHostImpl* view = RenderViewHostImpl::FromID(
-        load_info.first.child_id, load_info.first.route_id);
-    // The view could be gone at this point.
-    if (view) {
-      view->LoadStateChanged(load_info.second.url, load_info.second.load_state,
-                             load_info.second.upload_position,
-                             load_info.second.upload_size);
+    WebContents* web_contents = load_info.second.web_contents_getter.Run();
+    if (!web_contents)
+      continue;
+    auto existing = contents_map.find(web_contents);
+    if (existing == contents_map.end() ||
+        load_info.second.IsMoreInteresting(*existing->second)) {
+      contents_map[web_contents] = &load_info.second;
+    }
+  }
+  for (const auto& it : contents_map) {
+    // Re-run the getter to prevent use after free errors if calling
+    // LoadStateChanged on one WebContents deletes another one.
+    WebContentsImpl* web_contents =
+        static_cast<WebContentsImpl*>(it.second->web_contents_getter.Run());
+    if (web_contents) {
+      web_contents->LoadStateChanged(it.second->url, it.second->load_state,
+                                     it.second->upload_position,
+                                     it.second->upload_size);
     }
   }
 }
 
 std::unique_ptr<ResourceDispatcherHostImpl::LoadInfoMap>
 ResourceDispatcherHostImpl::GetLoadInfoForAllRoutes() {
   // Populate this map with load state changes, and then send them on to the UI
-  // thread where they can be passed along to the respective RVHs.
+  // thread where they can be passed along to the respective WebContents.
   std::unique_ptr<LoadInfoMap> info_map(new LoadInfoMap());
 
   for (const auto& loader : pending_loaders_) {
     net::URLRequest* request = loader.second->request();
     net::UploadProgress upload_progress = request->GetUploadProgress();
 
     LoadInfo load_info;
     load_info.url = request->url();
     load_info.load_state = request->GetLoadState();
     load_info.upload_size = upload_progress.size();
     load_info.upload_position = upload_progress.position();
+    load_info.web_contents_getter =
+        loader.second->GetRequestInfo()->GetWebContentsGetterForRequest();
 
-    GlobalRoutingID id(loader.second->GetRequestInfo()->GetGlobalRoutingID());
+    ResourceType type = loader.second->GetRequestInfo()->GetResourceType();
+    // The WebContents is only really interested in navigation updates. Send
+    // those along with requests that are sending data.
+    // TODO(csharrison): Consider updating this logic to send separate updates
+    // for uploads and navigation requests.
+    if (!load_info.IsUploading() &&
+        type != ResourceType::RESOURCE_TYPE_MAIN_FRAME &&
+        type != ResourceType::RESOURCE_TYPE_SUB_FRAME) {
+      continue;
+    }
+
+    GlobalFrameRoutingId id(
+        loader.second->GetRequestInfo()->GetGlobalFrameRoutingId());
     LoadInfoMap::iterator existing = info_map->find(id);
 
     if (existing == info_map->end() ||
-        LoadInfoIsMoreInteresting(load_info, existing->second)) {
+        load_info.IsMoreInteresting(existing->second)) {
       (*info_map)[id] = load_info;
     }
   }
   return info_map;
 }
 
 void ResourceDispatcherHostImpl::UpdateLoadInfo() {
+  // Group LoadInfo by frame, then pass to the UI thread where each WebContents
+  // picks the most interesting one.
   std::unique_ptr<LoadInfoMap> info_map(GetLoadInfoForAllRoutes());
 
   // Stop the timer if there are no more pending requests. Future new requests
   // will restart it as necessary.
   // Also stop the timer if there are no loading clients, to avoid waking up
   // unnecessarily when there is a long running (hanging get) request.
   if (info_map->empty() || !scheduler_->HasLoadingClients()) {
     update_load_states_timer_->Stop();
     return;
   }
@@ skipping 162 matching lines @@
   ssl.cert_id = GetCertStore()->StoreCert(ssl_info.cert.get(), child_id);
   response->head.security_info = SerializeSecurityInfo(ssl);
 }
 
 CertStore* ResourceDispatcherHostImpl::GetCertStore() {
   return cert_store_for_testing_ ? cert_store_for_testing_
                                  : CertStore::GetInstance();
 }
 
 }  // namespace content