OLD | NEW |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 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 "extensions/browser/extension_request_limiting_throttle.h" | 5 #include "extensions/browser/extension_request_limiting_throttle.h" |
6 | 6 |
7 #include "base/logging.h" | 7 #include "base/logging.h" |
8 #include "content/public/browser/resource_controller.h" | |
9 #include "extensions/browser/extension_throttle_entry.h" | 8 #include "extensions/browser/extension_throttle_entry.h" |
10 #include "extensions/browser/extension_throttle_manager.h" | 9 #include "extensions/browser/extension_throttle_manager.h" |
11 #include "net/base/net_errors.h" | 10 #include "net/base/net_errors.h" |
12 #include "net/url_request/redirect_info.h" | 11 #include "net/url_request/redirect_info.h" |
13 #include "net/url_request/url_request.h" | 12 #include "net/url_request/url_request.h" |
14 | 13 |
15 namespace extensions { | 14 namespace extensions { |
16 | 15 |
17 ExtensionRequestLimitingThrottle::ExtensionRequestLimitingThrottle( | 16 ExtensionRequestLimitingThrottle::ExtensionRequestLimitingThrottle( |
18 const net::URLRequest* request, | 17 const net::URLRequest* request, |
19 ExtensionThrottleManager* manager) | 18 ExtensionThrottleManager* manager) |
20 : request_(request), manager_(manager) { | 19 : request_(request), manager_(manager) { |
21 DCHECK(manager_); | 20 DCHECK(manager_); |
22 } | 21 } |
23 | 22 |
24 ExtensionRequestLimitingThrottle::~ExtensionRequestLimitingThrottle() { | 23 ExtensionRequestLimitingThrottle::~ExtensionRequestLimitingThrottle() { |
25 } | 24 } |
26 | 25 |
27 void ExtensionRequestLimitingThrottle::WillStartRequest(bool* defer) { | 26 void ExtensionRequestLimitingThrottle::WillStartRequest(bool* defer) { |
28 throttling_entry_ = manager_->RegisterRequestUrl(request_->url()); | 27 throttling_entry_ = manager_->RegisterRequestUrl(request_->url()); |
29 if (throttling_entry_->ShouldRejectRequest(*request_)) | 28 if (throttling_entry_->ShouldRejectRequest(*request_)) |
30 controller()->CancelWithError(net::ERR_TEMPORARILY_THROTTLED); | 29 CancelWithError(net::ERR_TEMPORARILY_THROTTLED); |
31 } | 30 } |
32 | 31 |
33 void ExtensionRequestLimitingThrottle::WillRedirectRequest( | 32 void ExtensionRequestLimitingThrottle::WillRedirectRequest( |
34 const net::RedirectInfo& redirect_info, | 33 const net::RedirectInfo& redirect_info, |
35 bool* defer) { | 34 bool* defer) { |
36 DCHECK_EQ(manager_->GetIdFromUrl(request_->url()), | 35 DCHECK_EQ(manager_->GetIdFromUrl(request_->url()), |
37 throttling_entry_->GetURLIdForDebugging()); | 36 throttling_entry_->GetURLIdForDebugging()); |
38 | 37 |
39 throttling_entry_->UpdateWithResponse(redirect_info.status_code); | 38 throttling_entry_->UpdateWithResponse(redirect_info.status_code); |
40 | 39 |
41 throttling_entry_ = manager_->RegisterRequestUrl(redirect_info.new_url); | 40 throttling_entry_ = manager_->RegisterRequestUrl(redirect_info.new_url); |
42 if (throttling_entry_->ShouldRejectRequest(*request_)) | 41 if (throttling_entry_->ShouldRejectRequest(*request_)) |
43 controller()->CancelWithError(net::ERR_TEMPORARILY_THROTTLED); | 42 CancelWithError(net::ERR_TEMPORARILY_THROTTLED); |
44 } | 43 } |
45 | 44 |
46 void ExtensionRequestLimitingThrottle::WillProcessResponse(bool* defer) { | 45 void ExtensionRequestLimitingThrottle::WillProcessResponse(bool* defer) { |
47 DCHECK_EQ(manager_->GetIdFromUrl(request_->url()), | 46 DCHECK_EQ(manager_->GetIdFromUrl(request_->url()), |
48 throttling_entry_->GetURLIdForDebugging()); | 47 throttling_entry_->GetURLIdForDebugging()); |
49 | 48 |
50 if (!request_->was_cached()) | 49 if (!request_->was_cached()) |
51 throttling_entry_->UpdateWithResponse(request_->GetResponseCode()); | 50 throttling_entry_->UpdateWithResponse(request_->GetResponseCode()); |
52 } | 51 } |
53 | 52 |
54 const char* ExtensionRequestLimitingThrottle::GetNameForLogging() const { | 53 const char* ExtensionRequestLimitingThrottle::GetNameForLogging() const { |
55 return "ExtensionRequestLimitingThrottle"; | 54 return "ExtensionRequestLimitingThrottle"; |
56 } | 55 } |
57 | 56 |
58 } // namespace extensions | 57 } // namespace extensions |
OLD | NEW |