OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 "content/browser/loader/throttling_resource_handler.h" | 5 #include "content/browser/loader/throttling_resource_handler.h" |
6 | 6 |
7 #include "content/public/browser/resource_request_info.h" | 7 #include "content/browser/loader/resource_request_info_impl.h" |
8 #include "content/public/browser/resource_throttle.h" | 8 #include "content/public/browser/resource_throttle.h" |
9 #include "content/public/common/resource_response.h" | 9 #include "content/public/common/resource_response.h" |
10 | 10 |
11 namespace content { | 11 namespace content { |
12 | 12 |
13 ThrottlingResourceHandler::ThrottlingResourceHandler( | 13 ThrottlingResourceHandler::ThrottlingResourceHandler( |
14 scoped_ptr<ResourceHandler> next_handler, | 14 scoped_ptr<ResourceHandler> next_handler, |
15 net::URLRequest* request, | 15 net::URLRequest* request, |
16 ScopedVector<ResourceThrottle> throttles) | 16 ScopedVector<ResourceThrottle> throttles) |
17 : LayeredResourceHandler(next_handler.Pass()), | 17 : LayeredResourceHandler(request, next_handler.Pass()), |
18 deferred_stage_(DEFERRED_NONE), | 18 deferred_stage_(DEFERRED_NONE), |
19 request_(request), | |
20 throttles_(throttles.Pass()), | 19 throttles_(throttles.Pass()), |
21 index_(0), | 20 index_(0), |
22 cancelled_by_resource_throttle_(false) { | 21 cancelled_by_resource_throttle_(false) { |
23 for (size_t i = 0; i < throttles_.size(); ++i) | 22 for (size_t i = 0; i < throttles_.size(); ++i) |
24 throttles_[i]->set_controller(this); | 23 throttles_[i]->set_controller(this); |
25 } | 24 } |
26 | 25 |
27 ThrottlingResourceHandler::~ThrottlingResourceHandler() { | 26 ThrottlingResourceHandler::~ThrottlingResourceHandler() { |
28 } | 27 } |
29 | 28 |
(...skipping 104 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
134 } | 133 } |
135 } | 134 } |
136 | 135 |
137 void ThrottlingResourceHandler::ResumeStart() { | 136 void ThrottlingResourceHandler::ResumeStart() { |
138 DCHECK(!cancelled_by_resource_throttle_); | 137 DCHECK(!cancelled_by_resource_throttle_); |
139 | 138 |
140 GURL url = deferred_url_; | 139 GURL url = deferred_url_; |
141 deferred_url_ = GURL(); | 140 deferred_url_ = GURL(); |
142 | 141 |
143 bool defer = false; | 142 bool defer = false; |
144 const ResourceRequestInfo* info = ResourceRequestInfo::ForRequest(request_); | 143 if (!OnWillStart(GetRequestID(), url, &defer)) { |
145 if (!OnWillStart(info->GetRequestID(), url, &defer)) { | |
146 controller()->Cancel(); | 144 controller()->Cancel(); |
147 } else if (!defer) { | 145 } else if (!defer) { |
148 controller()->Resume(); | 146 controller()->Resume(); |
149 } | 147 } |
150 } | 148 } |
151 | 149 |
152 void ThrottlingResourceHandler::ResumeRedirect() { | 150 void ThrottlingResourceHandler::ResumeRedirect() { |
153 DCHECK(!cancelled_by_resource_throttle_); | 151 DCHECK(!cancelled_by_resource_throttle_); |
154 | 152 |
155 GURL new_url = deferred_url_; | 153 GURL new_url = deferred_url_; |
156 deferred_url_ = GURL(); | 154 deferred_url_ = GURL(); |
157 scoped_refptr<ResourceResponse> response; | 155 scoped_refptr<ResourceResponse> response; |
158 deferred_response_.swap(response); | 156 deferred_response_.swap(response); |
159 | 157 |
160 bool defer = false; | 158 bool defer = false; |
161 const ResourceRequestInfo* info = ResourceRequestInfo::ForRequest(request_); | 159 if (!OnRequestRedirected(GetRequestID(), new_url, response.get(), &defer)) { |
162 if (!OnRequestRedirected(info->GetRequestID(), new_url, response.get(), | |
163 &defer)) { | |
164 controller()->Cancel(); | 160 controller()->Cancel(); |
165 } else if (!defer) { | 161 } else if (!defer) { |
166 controller()->Resume(); | 162 controller()->Resume(); |
167 } | 163 } |
168 } | 164 } |
169 | 165 |
170 void ThrottlingResourceHandler::ResumeResponse() { | 166 void ThrottlingResourceHandler::ResumeResponse() { |
171 DCHECK(!cancelled_by_resource_throttle_); | 167 DCHECK(!cancelled_by_resource_throttle_); |
172 | 168 |
173 scoped_refptr<ResourceResponse> response; | 169 scoped_refptr<ResourceResponse> response; |
174 deferred_response_.swap(response); | 170 deferred_response_.swap(response); |
175 | 171 |
176 bool defer = false; | 172 bool defer = false; |
177 const ResourceRequestInfo* info = ResourceRequestInfo::ForRequest(request_); | 173 if (!OnResponseStarted(GetRequestID(), response.get(), &defer)) { |
178 if (!OnResponseStarted(info->GetRequestID(), response.get(), &defer)) { | |
179 controller()->Cancel(); | 174 controller()->Cancel(); |
180 } else if (!defer) { | 175 } else if (!defer) { |
181 controller()->Resume(); | 176 controller()->Resume(); |
182 } | 177 } |
183 } | 178 } |
184 | 179 |
185 } // namespace content | 180 } // namespace content |
OLD | NEW |