Fix prerender byte-count UMA.

chrome/browser/renderer_host/chrome_resource_dispatcher_host_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/renderer_host/chrome_resource_dispatcher_host_delegate.h"
 
 #include <string>
 
 #include "base/base64.h"
 #include "base/logging.h"
@@ skipping 110 matching lines @@
   if (!rvh)
     return NULL;
   content::WebContents* web_contents =
       content::WebContents::FromRenderViewHost(rvh);
   if (!web_contents)
     return NULL;
 
   return prerender::PrerenderContents::FromWebContents(web_contents);
 }
 
+prerender::PrerenderManager* GetPrerenderManager(int render_process_id,
+                                                 int render_view_id) {
+  DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
+
+  content::RenderViewHost* render_view_host =
+      content::RenderViewHost::FromID(render_process_id, render_view_id);
+  if (!render_view_host)

[jochen (gone - plz use gerrit), 2014/02/19 16:54:24]

when is that possible?

[jkarlin, 2014/02/19 17:31:58]

Nearly everything that calls RenderViewHost::FromID checks for NULL.

It seems feasible to me that the RVH might go away between the renderer issuing
a request and the RDH switching to the UI thread to get here.

+    return NULL;
+
+  content::WebContents* web_contents =
+      content::WebContents::FromRenderViewHost(render_view_host);
+  if (!web_contents)
+    return NULL;
+
+  content::BrowserContext* browser_context = web_contents->GetBrowserContext();
+  if (!browser_context)
+    return NULL;

[jochen (gone - plz use gerrit), 2014/02/19 16:54:24]

or this?

[jkarlin, 2014/02/19 17:31:58]

Done.

+
+  Profile* profile = Profile::FromBrowserContext(browser_context);
+  if (!profile)
+    return NULL;

[jochen (gone - plz use gerrit), 2014/02/19 16:54:24]

or this?

[jkarlin, 2014/02/19 17:31:58]

Done.

+
+  return prerender::PrerenderManagerFactory::GetForProfile(profile);
+}
+
 void UpdatePrerenderNetworkBytesCallback(int render_process_id,
                                          int render_view_id,
                                          int64 bytes) {
   DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
 
   prerender::PrerenderContents* prerender_contents =
       FindPrerenderContents(render_process_id, render_view_id);
 
-  if (!prerender_contents)
-    return;
-  prerender_contents->AddNetworkBytes(bytes);
-  prerender_contents->prerender_manager()->AddProfileNetworkBytesIfEnabled(
-      bytes);
+  if (prerender_contents)
+    prerender_contents->AddNetworkBytes(bytes);
+
+  prerender::PrerenderManager* prerender_manager =
+      GetPrerenderManager(render_process_id, render_view_id);
+  if (prerender_manager)
+    prerender_manager->AddProfileNetworkBytesIfEnabled(bytes);
 }
 
 #if !defined(OS_ANDROID)
 // Goes through the extension's file browser handlers and checks if there is one
 // that can handle the |mime_type|.
 // |extension| must not be NULL.
 bool ExtensionCanHandleMimeType(const Extension* extension,
                                 const std::string& mime_type) {
   MimeTypesHandler* handler = MimeTypesHandler::GetHandler(extension);
   if (!handler)
@@ skipping 512 matching lines @@
                                        url_request->GetTotalReceivedBytes()));
   }
 }
 
 // static
 void ChromeResourceDispatcherHostDelegate::
     SetExternalProtocolHandlerDelegateForTesting(
     ExternalProtocolHandler::Delegate* delegate) {
   g_external_protocol_handler_delegate = delegate;
 }