Handle interface to prerenders.

chrome/browser/prerender/prerender_contents.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/prerender/prerender_contents.h"
 
 #include <algorithm>
 #include <utility>
 
+#include "base/memory/linked_ptr.h"
 #include "base/process_util.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/browser/history/history_tab_helper.h"
 #include "chrome/browser/history/history_types.h"
 #include "chrome/browser/prerender/prerender_final_status.h"
+#include "chrome/browser/prerender/prerender_handle.h"
 #include "chrome/browser/prerender/prerender_manager.h"
 #include "chrome/browser/prerender/prerender_render_view_host_observer.h"
 #include "chrome/browser/prerender/prerender_tracker.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/browser_finder.h"
 #include "chrome/browser/ui/tab_contents/tab_contents.h"
 #include "chrome/common/chrome_notification_types.h"
 #include "chrome/common/icon_messages.h"
 #include "chrome/common/prerender_messages.h"
 #include "chrome/common/url_constants.h"
 #include "content/public/browser/resource_request_details.h"
 #include "content/public/browser/browser_child_process_host.h"
 #include "content/public/browser/notification_service.h"
 #include "content/public/browser/render_process_host.h"
 #include "content/public/browser/render_view_host.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/browser/web_contents_delegate.h"
 #include "content/public/browser/web_contents_view.h"
 #include "ui/gfx/rect.h"
 
 using content::DownloadItem;
 using content::OpenURLParams;
 using content::RenderViewHost;
 using content::ResourceRedirectDetails;
+using content::SessionStorageNamespace;
 using content::WebContents;
 
 namespace prerender {
 
 namespace {
 
-// Compares URLs ignoring any ref for the purposes of matching URLs when
-// prerendering.
-struct PrerenderURLPredicate {
-  explicit PrerenderURLPredicate(const GURL& url)
-      : url_(url) {
-  }
-
-  bool operator()(const GURL& url) const {
-    return url.scheme() == url_.scheme() &&
-           url.host() == url_.host() &&
-           url.port() == url_.port() &&
-           url.path() == url_.path() &&
-           url.query() == url_.query();
-  }
-  GURL url_;
-};
-
 // Tells the render process at |child_id| whether |url| is a new prerendered
 // page, or whether |url| is being removed as a prerendered page. Currently
 // this will only inform the render process that created the prerendered page
 // with <link rel="prerender"> tags about it. This means that if the user
 // clicks on a link for a prerendered URL in a different page, the prerender
 // will not be swapped in.
 void InformRenderProcessAboutPrerender(const GURL& url,
                                        bool is_add,
                                        int child_id) {
   if (child_id < 0)
@@ skipping 16 matching lines @@
  public:
   virtual PrerenderContents* CreatePrerenderContents(
       PrerenderManager* prerender_manager, PrerenderTracker* prerender_tracker,
       Profile* profile, const GURL& url, const content::Referrer& referrer,
       Origin origin, uint8 experiment_id) OVERRIDE {
     return new PrerenderContents(prerender_manager, prerender_tracker, profile,
                                  url, referrer, origin, experiment_id);
   }
 };
 
+bool PrerenderContents::LessThanByLoadStartTime::operator ()(
+    const PrerenderContents* lhs,
+    const PrerenderContents* rhs) const {
+  DCHECK(!lhs->load_start_time().is_null());
+  DCHECK(!rhs->load_start_time().is_null());
+  return lhs->load_start_time() < rhs->load_start_time();
+}
+
 struct PrerenderContents::PendingPrerenderInfo {
-  PendingPrerenderInfo(const GURL& url,
+  PendingPrerenderInfo(linked_ptr<PrerenderHandle> prerender_handle,

[mmenke, 2012/06/22 16:12:45]

I assume this is to allow copying.  linked_ptr is very rarely used.  The much
more common thing to do in this case is to just use a vector or list of (gasp)
pointers.

+                       const GURL& url,
                        const content::Referrer& referrer,
                        const gfx::Size& size);
+
+  linked_ptr<PrerenderHandle> prerender_handle;

[dominich, 2012/06/22 15:36:16]

You don't see those very often. I would have expected a scoped_refptr I think,
or at least for the PrerenderHandle to be refcounted and passed raw.

   const GURL url;
   const content::Referrer referrer;
   const gfx::Size size;
 };
 
 PrerenderContents::PendingPrerenderInfo::PendingPrerenderInfo(
+    linked_ptr<PrerenderHandle> prerender_handle,
     const GURL& url,
     const content::Referrer& referrer,
     const gfx::Size& size)
-    : url(url),
+    : prerender_handle(prerender_handle),
+      url(url),
       referrer(referrer),
       size(size) {
 }
 
 // TabContentsDelegateImpl -----------------------------------------------------
 
 class PrerenderContents::TabContentsDelegateImpl
     : public content::WebContentsDelegate {
  public:
   explicit TabContentsDelegateImpl(PrerenderContents* prerender_contents) :
@@ skipping 76 matching lines @@
  private:
   typedef std::vector<scoped_refptr<history::HistoryAddPageArgs> >
       AddPageVector;
 
   // Caches pages to be added to the history.
   AddPageVector add_page_vector_;
 
   PrerenderContents* prerender_contents_;
 };
 
-void PrerenderContents::AddPendingPrerender(const GURL& url,
-                                            const content::Referrer& referrer,
-                                            const gfx::Size& size) {
-  pending_prerender_list_.push_back(PendingPrerenderInfo(url, referrer, size));
+void PrerenderContents::AddPendingPrerender(
+    scoped_ptr<PrerenderHandle> prerender_handle,
+    const GURL& url,
+    const content::Referrer& referrer,
+    const gfx::Size& size) {
+  pending_prerender_list_.push_back(
+      PendingPrerenderInfo(
+          make_linked_ptr(prerender_handle.release()),

[dominich, 2012/06/22 15:36:16]

what is this voodoo?

So you have a prerender handle passed as a scoped ptr, which has an explicit
lifetime, then you release that and make it into a linked ptr which is
ref-counted.

Why not pass in the prerender handle directly and have it be refcounted
internally instead of relying on this external code (to PrerenderHandle) to do
the refcounting?

+          url, referrer, size));
 }
 
-bool PrerenderContents::IsPendingEntry(const GURL& url) const {
+bool PrerenderContents::IsPendingEntry(
+    const PrerenderHandle* prerender_handle) const {
   for (PendingPrerenderList::const_iterator it =

[dominich, 2012/06/22 15:36:16]

what no std::count_if?

            pending_prerender_list_.begin();
        it != pending_prerender_list_.end();
        ++it) {
-    if (it->url == url)
+    if (it->prerender_handle == prerender_handle)
       return true;
   }
   return false;
 }
 
 void PrerenderContents::StartPendingPrerenders() {
   PendingPrerenderList pending_prerender_list;
   pending_prerender_list.swap(pending_prerender_list_);
   for (PendingPrerenderList::iterator it = pending_prerender_list.begin();
        it != pending_prerender_list.end();
@@ skipping 11 matching lines @@
     const content::Referrer& referrer,
     Origin origin,
     uint8 experiment_id)
     : prerendering_has_started_(false),
       prerender_manager_(prerender_manager),
       prerender_tracker_(prerender_tracker),
       prerender_url_(url),
       referrer_(referrer),
       profile_(profile),
       page_id_(0),
+      client_count_(1),
       has_stopped_loading_(false),
       has_finished_loading_(false),
       final_status_(FINAL_STATUS_MAX),
       match_complete_status_(MATCH_COMPLETE_DEFAULT),
       prerendering_has_been_cancelled_(false),
       child_id_(-1),
       route_id_(-1),
       origin_(origin),
       experiment_id_(experiment_id),
       creator_child_id_(-1) {
   DCHECK(prerender_manager != NULL);
 }
 
+PrerenderContents* PrerenderContents::CreateDummyReplacement() const {
+  PrerenderContents* dummy =
+      new PrerenderContents(prerender_manager_, prerender_tracker_, profile_,
+                            prerender_url_, referrer_, origin_, experiment_id_);
+  dummy->load_start_time_ = load_start_time_;
+  dummy->client_count_ = client_count_;

[dominich, 2012/06/22 15:36:16]

do we not need to copy the child_id, route_id, final_status_, etc?

+  return dummy;
+}
+
 bool PrerenderContents::Init() {
   return AddAliasURL(prerender_url_);
 }
 
 // static
 PrerenderContents::Factory* PrerenderContents::CreateFactory() {
   return new PrerenderContentsFactoryImpl();
 }
 
 void PrerenderContents::StartPrerendering(
     int creator_child_id,
     const gfx::Size& size,
-    content::SessionStorageNamespace* session_storage_namespace,
+    SessionStorageNamespace* session_storage_namespace,
     bool is_control_group) {
   DCHECK(profile_ != NULL);
   DCHECK(!prerendering_has_started_);
   DCHECK(prerender_contents_.get() == NULL);
   DCHECK_EQ(-1, creator_child_id_);
   DCHECK(size_.IsEmpty());
   DCHECK_EQ(1U, alias_urls_.size());
 
   creator_child_id_ = creator_child_id;
 
@@ skipping 207 matching lines @@
       NOTREACHED() << "Unexpected notification sent.";
       break;
   }
 }
 
 void PrerenderContents::OnRenderViewHostCreated(
     RenderViewHost* new_render_view_host) {
 }
 
 WebContents* PrerenderContents::CreateWebContents(
-    content::SessionStorageNamespace* session_storage_namespace) {
+    SessionStorageNamespace* session_storage_namespace) {
   return  WebContents::Create(profile_, NULL, MSG_ROUTING_NONE, NULL,
                               session_storage_namespace);
 }
 
 void PrerenderContents::OnUpdateFaviconURL(
     int32 page_id,
     const std::vector<FaviconURL>& urls) {
   VLOG(1) << "PrerenderContents::OnUpdateFaviconURL" << icon_url_;
   for (std::vector<FaviconURL>::const_iterator it = urls.begin();
        it != urls.end(); ++it) {
@@ skipping 31 matching lines @@
 
 void PrerenderContents::AddAliasURLsFromOtherPrerenderContents(
     PrerenderContents* other_pc) {
   for (std::vector<GURL>::const_iterator it = other_pc->alias_urls_.begin();
        it != other_pc->alias_urls_.end();
        ++it) {
     alias_urls_.push_back(*it);
   }
 }
 
-bool PrerenderContents::MatchesURL(const GURL& url, GURL* matching_url) const {
-  std::vector<GURL>::const_iterator matching_url_iterator =
-      std::find_if(alias_urls_.begin(),
-                   alias_urls_.end(),
-                   PrerenderURLPredicate(url));
-  if (matching_url_iterator != alias_urls_.end()) {
-    if (matching_url)
-      *matching_url = *matching_url_iterator;
-    return true;
-  }
-  return false;
+void PrerenderContents::IncrementClientCount() {
+  DCHECK_GT(0, client_count_);
+  ++client_count_;
+}
+
+void PrerenderContents::DecrementClientCount() {

[dominich, 2012/06/22 15:36:16]

DCHECK_GT(0, client_count_);

+  if (--client_count_ == 0)
+    Destroy(FINAL_STATUS_CANCELLED);
+}
+
+bool PrerenderContents::Matches(
+    const GURL& url, 
+    const SessionStorageNamespace* session_storage_namespace) {
+  SessionStorageNamespace* my_session_storage_namespace =
+      child_id_ == -1 ? NULL
+                      : GetRenderViewHostMutable()->GetSessionStorageNamespace();
+
+  if (session_storage_namespace != my_session_storage_namespace)
+    return false;
+      
+  return implicit_cast<bool>(
+      std::count_if(alias_urls_.begin(), alias_urls_.end(),
+                    std::bind2nd(std::equal_to<GURL>(), url)));

[dominich, 2012/06/22 15:36:16]

This will break fragment mismatches. There's a very specific reason to the prior
matching only working on a subset of URL features.

Also, it's unreadable, and an explicit ' > 0' would be better than an
implicit_cast.

 }
 
 void PrerenderContents::OnJSOutOfMemory() {
   Destroy(FINAL_STATUS_JS_OUT_OF_MEMORY);
 }
 
 void PrerenderContents::RenderViewGone(base::TerminationStatus status) {
   Destroy(FINAL_STATUS_RENDERER_CRASHED);
 }
 
@@ skipping 145 matching lines @@
 bool PrerenderContents::IsCrossSiteNavigationPending() const {
   if (!prerender_contents_.get() || !prerender_contents_->web_contents())
     return false;
   const WebContents* web_contents = prerender_contents_->web_contents();
   return (web_contents->GetSiteInstance() !=
           web_contents->GetPendingSiteInstance());
 }
 
 
 }  // namespace prerender