chrome/browser/download/download_resource_throttle.h

   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
   3 // found in the LICENSE file.
   4
   5 #ifndef CHROME_BROWSER_DOWNLOAD_DOWNLOAD_RESOURCE_THROTTLE_H_
   6 #define CHROME_BROWSER_DOWNLOAD_DOWNLOAD_RESOURCE_THROTTLE_H_
   7
   8 #include "base/memory/weak_ptr.h"
   9 #include "chrome/browser/download/download_request_limiter.h"
  10 #include "content/public/browser/resource_throttle.h"
  11
  12 class GURL;
  13
  14 // DownloadResourceThrottle is used to determine if a download should be
  15 // allowed. When a DownloadResourceThrottle is created it pauses the download
  16 // and asks the DownloadRequestLimiter if the download should be allowed. The
  17 // DownloadRequestLimiter notifies us asynchronously as to whether the download
  18 // is allowed or not. If the download is allowed the request is resumed.  If
  19 // the download is not allowed the request is canceled.
  20
  21 class DownloadResourceThrottle
  22     : public content::ResourceThrottle,
  23       public base::SupportsWeakPtr<DownloadResourceThrottle> {
  24  public:
  25   DownloadResourceThrottle(DownloadRequestLimiter* limiter,
  26                            int render_process_id,
  27                            int render_view_id,
  28                            const GURL& url,
  29                            const std::string& request_method);
  30
  31   // content::ResourceThrottle implementation:
  32   void WillStartRequest(bool* defer) override;
  33   void WillRedirectRequest(const net::RedirectInfo& redirect_info,
  34                            bool* defer) override;
  35   void WillProcessResponse(bool* defer) override;
  36   const char* GetNameForLogging() const override;
  37
  38  private:
  39   ~DownloadResourceThrottle() override;
  40
  41   void WillDownload(bool* defer);
  42   void ContinueDownload(bool allow);
  43
  44   // Set to true when we are querying the DownloadRequestLimiter.
  45   bool querying_limiter_;
  46
  47   // Set to true when we know that the request is allowed to start.
  48   bool request_allowed_;
  49
  50   // Set to true when we have deferred the request.
  51   bool request_deferred_;
  52
  53   DISALLOW_COPY_AND_ASSIGN(DownloadResourceThrottle);
  54 };
  55
  56 #endif  // CHROME_BROWSER_DOWNLOAD_DOWNLOAD_RESOURCE_THROTTLE_H_