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 { | 13 namespace { |
14 | 14 |
15 base::Time DefaultGetCurrentTime() { | 15 bool ShouldPrerender(const GURL& url, const ResourceResponse* response) { |
16 return base::Time::Now(); | |
17 } | |
18 | |
19 bool ShouldPrerender(const GURL& url, | |
20 const ResourceResponse* response, | |
21 PrerenderResourceHandler::GetCurrentTimeFunction get_time, | |
22 base::TimeDelta prerender_duration) { | |
23 if (!response) | 16 if (!response) |
24 return false; | 17 return false; |
25 const ResourceResponseHead& rrh = response->response_head; | 18 const ResourceResponseHead& rrh = response->response_head; |
26 if (!url.is_valid()) | 19 if (!url.is_valid()) |
27 return false; | 20 return false; |
28 if (!rrh.headers) | 21 if (!rrh.headers) |
29 return false; | 22 return false; |
30 if (!(url.SchemeIs("http") || url.SchemeIs("https"))) | 23 if (!(url.SchemeIs("http") || url.SchemeIs("https"))) |
31 return false; | 24 return false; |
32 if (rrh.mime_type != "text/html") | 25 if (rrh.mime_type != "text/html") |
33 return false; | 26 return false; |
34 if (rrh.headers->response_code() != 200) | 27 if (rrh.headers->response_code() != 200) |
35 return false; | 28 return false; |
36 if (rrh.headers->RequiresValidation( | |
37 rrh.request_time, | |
38 rrh.response_time, | |
39 get_time() + prerender_duration)) | |
40 return false; | |
41 return true; | 29 return true; |
42 } | 30 } |
43 | 31 |
44 } // namespace | 32 } // namespace |
45 | 33 |
46 PrerenderResourceHandler* PrerenderResourceHandler::MaybeCreate( | 34 PrerenderResourceHandler* PrerenderResourceHandler::MaybeCreate( |
47 const net::URLRequest& request, | 35 const net::URLRequest& request, |
48 ChromeURLRequestContext* context, | 36 ChromeURLRequestContext* context, |
49 ResourceHandler* next_handler) { | 37 ResourceHandler* next_handler) { |
50 if (!context || !context->prerender_manager()) | 38 if (!context || !context->prerender_manager()) |
51 return NULL; | 39 return NULL; |
52 if (!(request.load_flags() & net::LOAD_PREFETCH)) | 40 if (!(request.load_flags() & net::LOAD_PREFETCH)) |
53 return NULL; | 41 return NULL; |
54 if (request.method() != "GET") | 42 if (request.method() != "GET") |
55 return NULL; | 43 return NULL; |
56 return new PrerenderResourceHandler(next_handler, | 44 return new PrerenderResourceHandler(next_handler, |
57 context->prerender_manager()); | 45 context->prerender_manager()); |
58 } | 46 } |
59 | 47 |
60 PrerenderResourceHandler::PrerenderResourceHandler( | 48 PrerenderResourceHandler::PrerenderResourceHandler( |
61 ResourceHandler* next_handler, | 49 ResourceHandler* next_handler, |
62 PrerenderManager* prerender_manager) | 50 PrerenderManager* prerender_manager) |
63 : next_handler_(next_handler), | 51 : next_handler_(next_handler), |
64 prerender_manager_(prerender_manager), | 52 prerender_manager_(prerender_manager), |
65 ALLOW_THIS_IN_INITIALIZER_LIST( | 53 ALLOW_THIS_IN_INITIALIZER_LIST( |
66 prerender_callback_(NewCallback( | 54 prerender_callback_(NewCallback( |
67 this, &PrerenderResourceHandler::StartPrerender))), | 55 this, &PrerenderResourceHandler::StartPrerender))) { |
68 prerender_duration_( | |
69 base::TimeDelta::FromSeconds(kDefaultPrerenderDurationSeconds)), | |
70 get_current_time_(&DefaultGetCurrentTime) { | |
71 DCHECK(next_handler); | 56 DCHECK(next_handler); |
72 DCHECK(prerender_manager); | 57 DCHECK(prerender_manager); |
73 } | 58 } |
74 | 59 |
75 // This constructor is only used from unit tests. | 60 // This constructor is only used from unit tests. |
76 PrerenderResourceHandler::PrerenderResourceHandler( | 61 PrerenderResourceHandler::PrerenderResourceHandler( |
77 ResourceHandler* next_handler, | 62 ResourceHandler* next_handler, |
78 PrerenderCallback* callback) | 63 PrerenderCallback* callback) |
79 : next_handler_(next_handler), | 64 : next_handler_(next_handler), |
80 prerender_callback_(callback), | 65 prerender_callback_(callback) { |
81 prerender_duration_( | |
82 base::TimeDelta::FromSeconds(kDefaultPrerenderDurationSeconds)), | |
83 get_current_time_(&DefaultGetCurrentTime) { | |
84 DCHECK(next_handler); | 66 DCHECK(next_handler); |
85 DCHECK(callback); | 67 DCHECK(callback); |
86 } | 68 } |
87 | 69 |
88 PrerenderResourceHandler::~PrerenderResourceHandler() { | 70 PrerenderResourceHandler::~PrerenderResourceHandler() { |
89 } | 71 } |
90 | 72 |
91 bool PrerenderResourceHandler::OnUploadProgress(int request_id, | 73 bool PrerenderResourceHandler::OnUploadProgress(int request_id, |
92 uint64 position, | 74 uint64 position, |
93 uint64 size) { | 75 uint64 size) { |
94 return next_handler_->OnUploadProgress(request_id, position, size); | 76 return next_handler_->OnUploadProgress(request_id, position, size); |
95 } | 77 } |
96 | 78 |
97 bool PrerenderResourceHandler::OnRequestRedirected(int request_id, | 79 bool PrerenderResourceHandler::OnRequestRedirected(int request_id, |
98 const GURL& url, | 80 const GURL& url, |
99 ResourceResponse* response, | 81 ResourceResponse* response, |
100 bool* defer) { | 82 bool* defer) { |
101 bool will_redirect = next_handler_->OnRequestRedirected( | 83 bool will_redirect = next_handler_->OnRequestRedirected( |
102 request_id, url, response, defer); | 84 request_id, url, response, defer); |
103 if (will_redirect) { | 85 if (will_redirect) { |
104 alias_urls_.push_back(url); | 86 alias_urls_.push_back(url); |
105 url_ = url; | 87 url_ = url; |
106 } | 88 } |
107 return will_redirect; | 89 return will_redirect; |
108 } | 90 } |
109 | 91 |
110 bool PrerenderResourceHandler::OnResponseStarted(int request_id, | 92 bool PrerenderResourceHandler::OnResponseStarted(int request_id, |
111 ResourceResponse* response) { | 93 ResourceResponse* response) { |
112 if (ShouldPrerender(url_, | 94 if (ShouldPrerender(url_, response)) { |
113 response, | |
114 get_current_time_, | |
115 prerender_duration_)) { | |
116 BrowserThread::PostTask( | 95 BrowserThread::PostTask( |
117 BrowserThread::UI, | 96 BrowserThread::UI, |
118 FROM_HERE, | 97 FROM_HERE, |
119 NewRunnableMethod( | 98 NewRunnableMethod( |
120 this, | 99 this, |
121 &PrerenderResourceHandler::RunCallbackFromUIThread, | 100 &PrerenderResourceHandler::RunCallbackFromUIThread, |
122 url_, | 101 url_, |
123 alias_urls_)); | 102 alias_urls_)); |
124 } | 103 } |
125 return next_handler_->OnResponseStarted(request_id, response); | 104 return next_handler_->OnResponseStarted(request_id, response); |
(...skipping 39 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
165 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | 144 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
166 prerender_callback_->Run(url, alias_urls); | 145 prerender_callback_->Run(url, alias_urls); |
167 } | 146 } |
168 | 147 |
169 void PrerenderResourceHandler::StartPrerender(const GURL& url, | 148 void PrerenderResourceHandler::StartPrerender(const GURL& url, |
170 const std::vector<GURL>& | 149 const std::vector<GURL>& |
171 alias_urls) { | 150 alias_urls) { |
172 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | 151 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
173 prerender_manager_->AddPreload(url, alias_urls); | 152 prerender_manager_->AddPreload(url, alias_urls); |
174 } | 153 } |
175 | |
176 void PrerenderResourceHandler::set_prerender_duration(base::TimeDelta dt) { | |
177 prerender_duration_ = dt; | |
178 } | |
179 | |
180 void PrerenderResourceHandler::set_get_current_time_function( | |
181 GetCurrentTimeFunction get_current_time) { | |
182 DCHECK(get_current_time); | |
183 get_current_time_ = get_current_time; | |
184 } | |
185 | |
186 // Note: this should stay in line with prerendermanager | |
187 // static | |
188 const int PrerenderResourceHandler::kDefaultPrerenderDurationSeconds = 20; | |
OLD | NEW |