| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/browser/prerender/prerender_resource_handler.h" | 5 #include "chrome/browser/prerender/prerender_resource_handler.h" |
| 6 | 6 |
| 7 #include "chrome/browser/net/chrome_url_request_context.h" | 7 #include "chrome/browser/net/chrome_url_request_context.h" |
| 8 #include "chrome/common/resource_response.h" | 8 #include "chrome/common/resource_response.h" |
| 9 #include "net/base/load_flags.h" | 9 #include "net/base/load_flags.h" |
| 10 #include "net/http/http_response_headers.h" | 10 #include "net/http/http_response_headers.h" |
| 11 #include "net/url_request/url_request.h" | 11 #include "net/url_request/url_request.h" |
| 12 | 12 |
| 13 namespace prerender { | 13 namespace prerender { |
| 14 | 14 |
| 15 namespace { | 15 namespace { |
| 16 | 16 |
| 17 bool ShouldPrerenderURL(const GURL& url) { | 17 bool ShouldPrerender(const GURL& url, const ResourceResponse* response) { |
| 18 if (!url.is_valid()) | |
| 19 return false; | |
| 20 if (!url.SchemeIs("http")) { | |
| 21 RecordFinalStatus(FINAL_STATUS_HTTPS); | |
| 22 return false; | |
| 23 } | |
| 24 return true; | |
| 25 } | |
| 26 | |
| 27 bool ValidateAliasURLs(const std::vector<GURL>& urls) { | |
| 28 for (std::vector<GURL>::const_iterator it = urls.begin(); | |
| 29 it != urls.end(); | |
| 30 ++it) { | |
| 31 if (!ShouldPrerenderURL(*it)) | |
| 32 return false; | |
| 33 } | |
| 34 return true; | |
| 35 } | |
| 36 | |
| 37 bool ShouldPrerender(const ResourceResponse* response) { | |
| 38 if (!response) | 18 if (!response) |
| 39 return false; | 19 return false; |
| 40 const ResourceResponseHead& rrh = response->response_head; | 20 const ResourceResponseHead& rrh = response->response_head; |
| 21 if (!url.is_valid()) |
| 22 return false; |
| 41 if (!rrh.headers) | 23 if (!rrh.headers) |
| 42 return false; | 24 return false; |
| 25 if (!(url.SchemeIs("http") || url.SchemeIs("https"))) |
| 26 return false; |
| 43 if (rrh.mime_type != "text/html") | 27 if (rrh.mime_type != "text/html") |
| 44 return false; | 28 return false; |
| 45 if (rrh.headers->response_code() != 200) | 29 if (rrh.headers->response_code() != 200) |
| 46 return false; | 30 return false; |
| 47 return true; | 31 return true; |
| 48 } | 32 } |
| 49 | 33 |
| 50 } // namespace | 34 } // namespace |
| 51 | 35 |
| 52 PrerenderResourceHandler* PrerenderResourceHandler::MaybeCreate( | 36 PrerenderResourceHandler* PrerenderResourceHandler::MaybeCreate( |
| 53 const net::URLRequest& request, | 37 const net::URLRequest& request, |
| 54 ChromeURLRequestContext* context, | 38 ChromeURLRequestContext* context, |
| 55 ResourceHandler* next_handler) { | 39 ResourceHandler* next_handler) { |
| 56 if (!context || !context->prerender_manager()) | 40 if (!context || !context->prerender_manager()) |
| 57 return NULL; | 41 return NULL; |
| 58 if (!(request.load_flags() & net::LOAD_PREFETCH)) | 42 if (!(request.load_flags() & net::LOAD_PREFETCH)) |
| 59 return NULL; | 43 return NULL; |
| 60 if (!ShouldPrerenderURL(request.url())) | |
| 61 return NULL; | |
| 62 if (request.method() != "GET") | 44 if (request.method() != "GET") |
| 63 return NULL; | 45 return NULL; |
| 64 return new PrerenderResourceHandler(request, | 46 return new PrerenderResourceHandler(request, |
| 65 next_handler, | 47 next_handler, |
| 66 context->prerender_manager()); | 48 context->prerender_manager()); |
| 67 } | 49 } |
| 68 | 50 |
| 69 PrerenderResourceHandler::PrerenderResourceHandler( | 51 PrerenderResourceHandler::PrerenderResourceHandler( |
| 70 const net::URLRequest& request, | 52 const net::URLRequest& request, |
| 71 ResourceHandler* next_handler, | 53 ResourceHandler* next_handler, |
| (...skipping 28 matching lines...) Expand all Loading... |
| 100 return next_handler_->OnUploadProgress(request_id, position, size); | 82 return next_handler_->OnUploadProgress(request_id, position, size); |
| 101 } | 83 } |
| 102 | 84 |
| 103 bool PrerenderResourceHandler::OnRequestRedirected(int request_id, | 85 bool PrerenderResourceHandler::OnRequestRedirected(int request_id, |
| 104 const GURL& url, | 86 const GURL& url, |
| 105 ResourceResponse* response, | 87 ResourceResponse* response, |
| 106 bool* defer) { | 88 bool* defer) { |
| 107 bool will_redirect = next_handler_->OnRequestRedirected( | 89 bool will_redirect = next_handler_->OnRequestRedirected( |
| 108 request_id, url, response, defer); | 90 request_id, url, response, defer); |
| 109 if (will_redirect) { | 91 if (will_redirect) { |
| 110 if (!ShouldPrerenderURL(url)) | |
| 111 return false; | |
| 112 alias_urls_.push_back(url); | 92 alias_urls_.push_back(url); |
| 113 url_ = url; | 93 url_ = url; |
| 114 } | 94 } |
| 115 return will_redirect; | 95 return will_redirect; |
| 116 } | 96 } |
| 117 | 97 |
| 118 bool PrerenderResourceHandler::OnResponseStarted(int request_id, | 98 bool PrerenderResourceHandler::OnResponseStarted(int request_id, |
| 119 ResourceResponse* response) { | 99 ResourceResponse* response) { |
| 120 if (ShouldPrerender(response)) { | 100 if (ShouldPrerender(url_, response)) { |
| 121 DCHECK(ValidateAliasURLs(alias_urls_)); | |
| 122 BrowserThread::PostTask( | 101 BrowserThread::PostTask( |
| 123 BrowserThread::UI, | 102 BrowserThread::UI, |
| 124 FROM_HERE, | 103 FROM_HERE, |
| 125 NewRunnableMethod( | 104 NewRunnableMethod( |
| 126 this, | 105 this, |
| 127 &PrerenderResourceHandler::RunCallbackFromUIThread, | 106 &PrerenderResourceHandler::RunCallbackFromUIThread, |
| 128 url_, | 107 url_, |
| 129 alias_urls_, | 108 alias_urls_, |
| 130 GURL(request_.referrer()))); | 109 GURL(request_.referrer()))); |
| 131 } | 110 } |
| 132 return next_handler_->OnResponseStarted(request_id, response); | 111 return next_handler_->OnResponseStarted(request_id, response); |
| 133 } | 112 } |
| 134 | 113 |
| 135 bool PrerenderResourceHandler::OnWillStart(int request_id, | 114 bool PrerenderResourceHandler::OnWillStart(int request_id, |
| 136 const GURL& url, | 115 const GURL& url, |
| 137 bool* defer) { | 116 bool* defer) { |
| 138 bool will_start = next_handler_->OnWillStart(request_id, url, defer); | 117 bool will_start = next_handler_->OnWillStart(request_id, url, defer); |
| 139 if (will_start) { | 118 if (will_start) { |
| 140 if (!ShouldPrerenderURL(url)) | |
| 141 return false; | |
| 142 alias_urls_.push_back(url); | 119 alias_urls_.push_back(url); |
| 143 url_ = url; | 120 url_ = url; |
| 144 } | 121 } |
| 145 return will_start; | 122 return will_start; |
| 146 } | 123 } |
| 147 | 124 |
| 148 bool PrerenderResourceHandler::OnWillRead(int request_id, | 125 bool PrerenderResourceHandler::OnWillRead(int request_id, |
| 149 net::IOBuffer** buf, | 126 net::IOBuffer** buf, |
| 150 int* buf_size, | 127 int* buf_size, |
| 151 int min_size) { | 128 int min_size) { |
| (...skipping 26 matching lines...) Expand all Loading... |
| 178 | 155 |
| 179 void PrerenderResourceHandler::StartPrerender( | 156 void PrerenderResourceHandler::StartPrerender( |
| 180 const GURL& url, | 157 const GURL& url, |
| 181 const std::vector<GURL>& alias_urls, | 158 const std::vector<GURL>& alias_urls, |
| 182 const GURL& referrer) { | 159 const GURL& referrer) { |
| 183 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | 160 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
| 184 prerender_manager_->AddPreload(url, alias_urls, referrer); | 161 prerender_manager_->AddPreload(url, alias_urls, referrer); |
| 185 } | 162 } |
| 186 | 163 |
| 187 } // namespace prerender | 164 } // namespace prerender |
| OLD | NEW |