| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "chrome/browser/prerender/prerender_interceptor.h" | |
| 6 | |
| 7 #include <string> | |
| 8 | |
| 9 #include "base/logging.h" | |
| 10 #include "chrome/browser/browser_thread.h" | |
| 11 #include "chrome/browser/browser_process.h" | |
| 12 #include "chrome/browser/io_thread.h" | |
| 13 #include "googleurl/src/gurl.h" | |
| 14 #include "net/base/load_flags.h" | |
| 15 | |
| 16 DISABLE_RUNNABLE_METHOD_REFCOUNT(PrerenderInterceptor); | |
| 17 | |
| 18 PrerenderInterceptor::PrerenderInterceptor() | |
| 19 : ALLOW_THIS_IN_INITIALIZER_LIST( | |
| 20 callback_(NewCallback(this, | |
| 21 &PrerenderInterceptor::PrerenderDispatch))) { | |
| 22 net::URLRequest::RegisterRequestInterceptor(this); | |
| 23 } | |
| 24 | |
| 25 PrerenderInterceptor::PrerenderInterceptor( | |
| 26 PrerenderInterceptorCallback* callback) | |
| 27 : callback_(callback) { | |
| 28 net::URLRequest::RegisterRequestInterceptor(this); | |
| 29 } | |
| 30 | |
| 31 PrerenderInterceptor::~PrerenderInterceptor() { | |
| 32 net::URLRequest::UnregisterRequestInterceptor(this); | |
| 33 } | |
| 34 | |
| 35 net::URLRequestJob* PrerenderInterceptor::MaybeIntercept( | |
| 36 net::URLRequest* request) { | |
| 37 return NULL; | |
| 38 } | |
| 39 | |
| 40 net::URLRequestJob* PrerenderInterceptor::MaybeInterceptResponse( | |
| 41 net::URLRequest* request) { | |
| 42 // TODO(gavinp): unfortunately, we can't figure out the origin | |
| 43 // of this request here on systems where the referrer is blocked by | |
| 44 // configuration. | |
| 45 | |
| 46 // TODO(gavinp): note that the response still might be intercepted | |
| 47 // by a later interceptor. Should we write an interposing delegate | |
| 48 // and only prerender dispatch on requests that aren't intercepted? | |
| 49 // Or is this a slippery slope? | |
| 50 | |
| 51 if (request->load_flags() & net::LOAD_PREFETCH) { | |
| 52 std::string mime_type; | |
| 53 request->GetMimeType(&mime_type); | |
| 54 if (mime_type == "text/html") | |
| 55 BrowserThread::PostTask( | |
| 56 BrowserThread::UI, | |
| 57 FROM_HERE, | |
| 58 NewRunnableMethod(this, | |
| 59 &PrerenderInterceptor::RunCallbackFromUIThread, | |
| 60 request->url())); | |
| 61 } | |
| 62 return NULL; | |
| 63 } | |
| 64 | |
| 65 void PrerenderInterceptor::RunCallbackFromUIThread(const GURL& url) { | |
| 66 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | |
| 67 callback_->Run(url); | |
| 68 } | |
| 69 | |
| 70 void PrerenderInterceptor::PrerenderDispatch( | |
| 71 const GURL& url) { | |
| 72 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | |
| 73 DVLOG(2) << "PrerenderDispatchOnUIThread: url=" << url; | |
| 74 } | |
| 75 | |
| OLD | NEW |