For prerendering, keep track of all the intermediate redirects, and hook into...

chrome/browser/prerender/prerender_contents.cc

 // Copyright (c) 2010 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 "base/utf_string_conversions.h"
 #include "chrome/browser/background_contents_service.h"
 #include "chrome/browser/browsing_instance.h"
 #include "chrome/browser/prerender/prerender_manager.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/renderer_host/render_view_host.h"
 #include "chrome/browser/renderer_host/site_instance.h"
 #include "chrome/browser/renderer_preferences_util.h"
 #include "chrome/common/extensions/extension_constants.h"
 #include "chrome/common/notification_service.h"
 #include "chrome/common/url_constants.h"
 #include "chrome/common/view_types.h"
+#include "chrome/common/render_messages.h"
 #include "chrome/common/render_messages_params.h"
 #include "gfx/rect.h"
 
 PrerenderContents::PrerenderContents(PrerenderManager* prerender_manager,
                                      Profile* profile,
                                      const GURL& url)
     : prerender_manager_(prerender_manager),
       render_view_host_(NULL),
       prerender_url_(url),
       profile_(profile),
@@ skipping 13 matching lines @@
                  NotificationService::AllSources());
 
   // Register for our parent profile to shutdown, so we can shut ourselves down
   // as well (should only be called for OTR profiles, as we should receive
   // APP_TERMINATING before non-OTR profiles are destroyed).
   // TODO(tburkard): figure out if this is needed.
   registrar_.Add(this, NotificationType::PROFILE_DESTROYED,
                  Source<Profile>(profile_));
   render_view_host_->CreateRenderView(string16());
   render_view_host_->NavigateToURL(prerender_url_);
+  AddMatchingUrl(prerender_url_);
 }
 
 PrerenderContents::~PrerenderContents() {
   if (!render_view_host_)   // Will be null for unit tests.
     return;
 
   render_view_host_->Shutdown();  // deletes render_view_host
 }
 
 RenderViewHostDelegate::View* PrerenderContents::GetViewDelegate() {
@@ skipping 18 matching lines @@
   // We only care when the outer frame changes.
   if (!PageTransition::IsMainFrame(params.transition))
     return;
 
   // Store the navigation params.
   ViewHostMsg_FrameNavigate_Params* p = new ViewHostMsg_FrameNavigate_Params();
   *p = params;
   navigate_params_.reset(p);
 
   url_ = params.url;
+
+  AddMatchingUrl(url_);
 }
 
 void PrerenderContents::UpdateTitle(RenderViewHost* render_view_host,
                                     int32 page_id,
                                     const std::wstring& title) {
   if (title.empty()) {
     return;
   }
 
   title_ = WideToUTF16Hack(title);
@@ skipping 108 matching lines @@
 }
 
 void PrerenderContents::ShowCreatedWidget(int route_id,
                                           const gfx::Rect& initial_pos) {
   NOTIMPLEMENTED();
 }
 
 void PrerenderContents::ShowCreatedFullscreenWidget(int route_id) {
   NOTIMPLEMENTED();
 }
+
+bool PrerenderContents::OnMessageReceived(const IPC::Message& message) {
+  bool handled = true;
+  bool message_is_ok = true;
+  IPC_BEGIN_MESSAGE_MAP_EX(PrerenderContents, message, message_is_ok)
+    IPC_MESSAGE_HANDLER(ViewHostMsg_DidStartProvisionalLoadForFrame,
+                        OnDidStartProvisionalLoadForFrame)
+    IPC_MESSAGE_HANDLER(ViewHostMsg_DidRedirectProvisionalLoad,
+                        OnDidRedirectProvisionalLoad)
+    IPC_MESSAGE_UNHANDLED(handled = false)
+  IPC_END_MESSAGE_MAP_EX()
+
+  return handled;
+}
+
+void PrerenderContents::OnDidStartProvisionalLoadForFrame(int64 frame_id,
+                                                          bool is_main_frame,
+                                                          const GURL& url) {
+  if (is_main_frame)
+    AddMatchingUrl(url);
+}
+
+void PrerenderContents::OnDidRedirectProvisionalLoad(int32 page_id,
+                                                     const GURL& source_url,
+                                                     const GURL& target_url) {
+  AddMatchingUrl(target_url);
+}
+
+void PrerenderContents::AddMatchingUrl(const GURL& url) {
+  matching_urls_.push_back(url);
+}
+
+bool PrerenderContents::MatchesUrl(const GURL& url) {
+  for (std::vector<GURL>::iterator it = matching_urls_.begin();

[cbentzel, 2011/01/12 23:49:53]

i'd just do std::find here.

[cbentzel, 2011/01/12 23:49:53]

const_iterator to be safe [and this method should be const]

[tburkard, 2011/01/13 19:14:16]

Done.

[tburkard, 2011/01/13 19:14:16]

Done.

+       it != matching_urls_.end();
+       ++it) {
+    if (*it == url)
+      return true;
+  }
+  return false;
+}
+
+void PrerenderContents::AddAliasURLs(const std::vector<GURL>& alias_urls) {
+  for (std::vector<GURL>::const_iterator it = alias_urls.begin();

[cbentzel, 2011/01/12 23:49:53]

I think this can go away in favor of constructor (which would initialize
matching_urls_)

[tburkard, 2011/01/13 19:14:16]

Done.

+       it != alias_urls.end();
+       ++it) {
+    AddMatchingUrl(*it);
+  }
+}