extensions/browser/extension_request_limiting_throttle.h

   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
   3 // found in the LICENSE file.
   4
   5 #ifndef EXTENSIONS_BROWSER_EXTENSION_REQUEST_LIMITING_THROTTLE_H_
   6 #define EXTENSIONS_BROWSER_EXTENSION_REQUEST_LIMITING_THROTTLE_H_
   7
   8 #include "base/macros.h"
   9 #include "base/memory/ref_counted.h"
  10 #include "content/public/browser/resource_throttle.h"
  11
  12 namespace net {
  13 struct RedirectInfo;
  14 class URLRequest;
  15 }
  16
  17 namespace extensions {
  18
  19 class ExtensionThrottleEntryInterface;
  20 class ExtensionThrottleManager;
  21
  22 // This class monitors requests issued by extensions and throttles the request
  23 // if there are too many requests made within a short time to urls with the same
  24 // scheme, host, port and path. For the exact criteria for throttling, please
  25 // also see extension_throttle_manager.cc.
  26 class ExtensionRequestLimitingThrottle : public content::ResourceThrottle {
  27  public:
  28   ExtensionRequestLimitingThrottle(const net::URLRequest* request,
  29                                    ExtensionThrottleManager* manager);
  30   ~ExtensionRequestLimitingThrottle() override;
  31
  32   // content::ResourceThrottle implementation (called on IO thread):
  33   void WillStartRequest(bool* defer) override;
  34   void WillRedirectRequest(const net::RedirectInfo& redirect_info,
  35                            bool* defer) override;
  36   void WillProcessResponse(bool* defer) override;
  37
  38   const char* GetNameForLogging() const override;
  39
  40  private:
  41   const net::URLRequest* request_;
  42   ExtensionThrottleManager* manager_;
  43
  44   // This is used to supervise traffic and enforce exponential back-off.
  45   scoped_refptr<ExtensionThrottleEntryInterface> throttling_entry_;
  46
  47   DISALLOW_COPY_AND_ASSIGN(ExtensionRequestLimitingThrottle);
  48 };
  49
  50 }  // namespace extensions
  51
  52 #endif  // EXTENSIONS_BROWSER_EXTENSION_REQUEST_LIMITING_THROTTLE_H_