src/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   virtual void WillStartRequest(bool* defer) OVERRIDE;
  33   virtual void WillRedirectRequest(const GURL& new_url, bool* defer) OVERRIDE;
  34   virtual void WillProcessResponse(bool* defer) OVERRIDE;
  35   virtual const char* GetNameForLogging() const OVERRIDE;
  36
  37  private:
  38   virtual ~DownloadResourceThrottle();
  39
  40   void WillDownload(bool* defer);
  41   void ContinueDownload(bool allow);
  42
  43   // Set to true when we are querying the DownloadRequestLimiter.
  44   bool querying_limiter_;
  45
  46   // Set to true when we know that the request is allowed to start.
  47   bool request_allowed_;
  48
  49   // Set to true when we have deferred the request.
  50   bool request_deferred_;
  51
  52   DISALLOW_COPY_AND_ASSIGN(DownloadResourceThrottle);
  53 };
  54
  55 #endif  // CHROME_BROWSER_DOWNLOAD_DOWNLOAD_RESOURCE_THROTTLE_H_