| // Copyright 2015 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "extensions/browser/extension_request_limiting_throttle.h" |
| |
| #include "base/logging.h" |
| #include "content/public/browser/resource_controller.h" |
| #include "extensions/browser/extension_throttle_entry.h" |
| #include "extensions/browser/extension_throttle_manager.h" |
| #include "net/base/net_errors.h" |
| #include "net/url_request/redirect_info.h" |
| #include "net/url_request/url_request.h" |
| |
| namespace extensions { |
| |
| ExtensionRequestLimitingThrottle::ExtensionRequestLimitingThrottle( |
| const net::URLRequest* request, |
| ExtensionThrottleManager* manager) |
| : request_(request), manager_(manager) { |
| DCHECK(manager_); |
| } |
| |
| ExtensionRequestLimitingThrottle::~ExtensionRequestLimitingThrottle() { |
| } |
| |
| void ExtensionRequestLimitingThrottle::WillStartRequest(bool* defer) { |
| throttling_entry_ = manager_->RegisterRequestUrl(request_->url()); |
| if (throttling_entry_->ShouldRejectRequest(*request_)) |
| controller()->CancelWithError(net::ERR_TEMPORARILY_THROTTLED); |
| } |
| |
| void ExtensionRequestLimitingThrottle::WillRedirectRequest( |
| const net::RedirectInfo& redirect_info, |
| bool* defer) { |
| DCHECK_EQ(manager_->GetIdFromUrl(request_->url()), |
| throttling_entry_->GetURLIdForDebugging()); |
| |
| throttling_entry_->UpdateWithResponse(redirect_info.status_code); |
| |
| throttling_entry_ = manager_->RegisterRequestUrl(redirect_info.new_url); |
| if (throttling_entry_->ShouldRejectRequest(*request_)) |
| controller()->CancelWithError(net::ERR_TEMPORARILY_THROTTLED); |
| } |
| |
| void ExtensionRequestLimitingThrottle::WillProcessResponse(bool* defer) { |
| DCHECK_EQ(manager_->GetIdFromUrl(request_->url()), |
| throttling_entry_->GetURLIdForDebugging()); |
| |
| if (!request_->was_cached()) |
| throttling_entry_->UpdateWithResponse(request_->GetResponseCode()); |
| } |
| |
| const char* ExtensionRequestLimitingThrottle::GetNameForLogging() const { |
| return "ExtensionRequestLimitingThrottle"; |
| } |
| |
| } // namespace extensions |