Remove a chrome dependency by removing Prerender from ResourceDispatcherHost.

chrome/browser/prerender/prerender_tracker.cc

 // Copyright (c) 2011 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 "base/logging.h"
+#include "chrome/browser/browser_process.h"
 #include "chrome/browser/prerender/prerender_manager.h"
 #include "chrome/browser/prerender/prerender_tracker.h"
+#include "chrome/browser/ui/browser.h"

[mmenke, 2011/05/23 21:58:41]

Is this needed?

[dominich, 2011/05/23 22:22:55]

Done.

 #include "content/browser/browser_thread.h"
+#include "content/browser/resource_context.h"
+#include "content/common/resource_messages.h"
+#include "net/base/load_flags.h"
 
 namespace prerender {
 
 namespace {
 
 void DestroyPreloadForRenderView(
     const base::WeakPtr<PrerenderManager>& prerender_manager_weak_ptr,
     int child_id,
     int route_id,
     FinalStatus final_status) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
   PrerenderManager* prerender_manager = prerender_manager_weak_ptr.get();
   if (!prerender_manager)
     return;
 
   prerender_manager->DestroyPreloadForChildRouteIdPair(
       std::make_pair(child_id, route_id),
       final_status);
 }
 
 }  // namespace
 
-struct RenderViewInfo {
-  explicit RenderViewInfo(PrerenderManager* prerender_manager)
+PrerenderTracker::RenderViewInfo::RenderViewInfo(
+    PrerenderManager* prerender_manager)
       : final_status(FINAL_STATUS_MAX),
         prerender_manager(prerender_manager->AsWeakPtr()) {
-  }
-
-  FinalStatus final_status;
-  base::WeakPtr<PrerenderManager> prerender_manager;
-};
-
-// static
-PrerenderTracker* PrerenderTracker::GetInstance() {
-  return Singleton<PrerenderTracker>::get();
 }
 
 bool PrerenderTracker::TryUse(int child_id, int route_id) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
 
   FinalStatus final_status = SetFinalStatus(child_id, route_id,
                                             FINAL_STATUS_USED);
   return final_status == FINAL_STATUS_USED;
 }
 
@@ skipping 73 matching lines @@
 
   BrowserThread::PostTask(
       BrowserThread::IO, FROM_HERE,
       NewRunnableFunction(&RemovePrerenderOnIOThreadTask, child_route_id_pair));
 
   base::AutoLock lock(final_status_map_lock_);
   size_t num_erased = final_status_map_.erase(child_route_id_pair);
   DCHECK_EQ(1u, num_erased);
 }
 
-PrerenderTracker::PrerenderTracker() {
-}
-
-PrerenderTracker::~PrerenderTracker() {
-}
-
 FinalStatus PrerenderTracker::SetFinalStatus(int child_id, int route_id,
                                              FinalStatus final_status) {
   DCHECK(final_status >= FINAL_STATUS_USED && final_status < FINAL_STATUS_MAX);
 
   ChildRouteIdPair child_route_id_pair(child_id, route_id);
 
   base::AutoLock lock(final_status_map_lock_);
   FinalStatusMap::iterator final_status_it =
       final_status_map_.find(child_route_id_pair);
   if (final_status_it == final_status_map_.end()) {
     // The RenderView has already been either used or destroyed.
     return FINAL_STATUS_MAX;
   }
 
   if (final_status_it->second.final_status == FINAL_STATUS_MAX) {
     final_status_it->second.final_status = final_status;
     if (final_status != FINAL_STATUS_USED) {
       BrowserThread::PostTask(
           BrowserThread::UI, FROM_HERE,
           NewRunnableFunction(&DestroyPreloadForRenderView,
                               final_status_it->second.prerender_manager,
                               child_id,
                               route_id,
                               final_status));
     }
   }
   return final_status_it->second.final_status;
 }
 
+bool PrerenderTracker::ShouldBeginRequest(
+    int child_id, int route_id,
+    const ResourceHostMsg_Request& request_data,
+    const content::ResourceContext& resource_context,
+    const GURL& referrer) {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
+  bool is_prerendering = IsPrerenderingOnIOThread(child_id, route_id);
+
+  // Handle a PREFETCH resource type.
+  if (request_data.resource_type == ResourceType::PREFETCH) {
+    // All PREFETCH requests should be GETs, but be defensive about it.
+    if (request_data.method != "GET")
+      return false;
+
+    // If prefetch is disabled, kill the request.
+    if (!ResourceDispatcherHost::is_prefetch_enabled())
+      return false;
+
+    // If prerendering is enabled, trigger a prerender for the URL and abort the
+    // request to prevent double-gets.
+    if (PrerenderManager::IsPrerenderingPossible()) {
+      BrowserThread::PostTask(BrowserThread::UI, FROM_HERE,
+          NewRunnableFunction(HandlePrefetchTag,
+              resource_context.prerender_manager(),
+              child_id,
+              route_id,
+              request_data.url,
+              referrer,
+              is_prerendering));
+      return false;
+    }
+  }
+
+  // Abort any prerenders that spawn requests that use invalid HTTP methods.
+  if (is_prerendering &&
+      !PrerenderManager::IsValidHttpMethod(request_data.method)) {
+    BrowserThread::PostTask(BrowserThread::UI, FROM_HERE,
+                            NewRunnableFunction(
+                                DestroyPreloadForRenderView,
+                                resource_context.prerender_manager(),
+                                child_id,
+                                route_id,
+                                FINAL_STATUS_INVALID_HTTP_METHOD));
+    return false;
+  }
+
+  return true;
+}
+
+void PrerenderTracker::MutateLoadFlags(int child_id, int route_id,
+                                       int* load_flags) {
+  DCHECK(load_flags);
+  if (IsPrerenderingOnIOThread(child_id, route_id))
+    *load_flags |= net::LOAD_PRERENDER;
+}
+
 void PrerenderTracker::AddPrerenderOnIOThread(
     const ChildRouteIdPair& child_route_id_pair) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
   DCHECK(!IsPrerenderingOnIOThread(child_route_id_pair.first,
                                    child_route_id_pair.second));
 
   possibly_prerendering_io_thread_set_.insert(child_route_id_pair);
 }
 
 void PrerenderTracker::RemovePrerenderOnIOThread(
     const ChildRouteIdPair& child_route_id_pair) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
   DCHECK(IsPrerenderingOnIOThread(child_route_id_pair.first,
                                   child_route_id_pair.second));
 
   possibly_prerendering_io_thread_set_.erase(child_route_id_pair);
 }
 
 // static
 void PrerenderTracker::AddPrerenderOnIOThreadTask(
     const ChildRouteIdPair& child_route_id_pair) {
-  GetInstance()->AddPrerenderOnIOThread(child_route_id_pair);
+  g_browser_process->prerender_tracker()->AddPrerenderOnIOThread(
+      child_route_id_pair);
 }
 
 // static
 void PrerenderTracker::RemovePrerenderOnIOThreadTask(
     const ChildRouteIdPair& child_route_id_pair) {
-  GetInstance()->RemovePrerenderOnIOThread(child_route_id_pair);
+  g_browser_process->prerender_tracker()->RemovePrerenderOnIOThread(
+      child_route_id_pair);
 }
 
 }  // namespace prerender