| 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 |