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