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.
5 #ifndef CHROME_BROWSER_DOWNLOAD_DOWNLOAD_RESOURCE_THROTTLE_H_
6 #define CHROME_BROWSER_DOWNLOAD_DOWNLOAD_RESOURCE_THROTTLE_H_
8 #include "base/memory/weak_ptr.h"
9 #include "chrome/browser/download/download_request_limiter.h"
10 #include "content/public/browser/resource_throttle.h"
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.
21 class DownloadResourceThrottle
22 : public content::ResourceThrottle,
23 public base::SupportsWeakPtr<DownloadResourceThrottle> {
25 DownloadResourceThrottle(DownloadRequestLimiter* limiter,
26 int render_process_id,
29 const std::string& request_method);
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;
37 virtual ~DownloadResourceThrottle();
39 void WillDownload(bool* defer);
40 void ContinueDownload(bool allow);
42 // Set to true when we are querying the DownloadRequestLimiter.
43 bool querying_limiter_;
45 // Set to true when we know that the request is allowed to start.
46 bool request_allowed_;
48 // Set to true when we have deferred the request.
49 bool request_deferred_;
51 DISALLOW_COPY_AND_ASSIGN(DownloadResourceThrottle);
54 #endif // CHROME_BROWSER_DOWNLOAD_DOWNLOAD_RESOURCE_THROTTLE_H_