| 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 // Information passed between callbacks to check whether download can proceed. | 25 // Information passed between callbacks to check whether download can proceed. |
| 26 struct DownloadRequestInfo { | 26 struct DownloadRequestInfo { |
| 27 DownloadRequestInfo( | 27 DownloadRequestInfo( |
| 28 scoped_refptr<DownloadRequestLimiter> limiter, | 28 scoped_refptr<DownloadRequestLimiter> limiter, |
| 29 int render_process_id, | 29 const base::Callback<content::WebContents*(void)>& web_contents_getter, |
| 30 int render_view_id, | |
| 31 const GURL& url, | 30 const GURL& url, |
| 32 const std::string& request_method, | 31 const std::string& request_method, |
| 33 const DownloadRequestLimiter::Callback& continue_callback); | 32 const DownloadRequestLimiter::Callback& continue_callback); |
| 34 ~DownloadRequestInfo(); | 33 ~DownloadRequestInfo(); |
| 35 | 34 |
| 36 scoped_refptr<DownloadRequestLimiter> limiter; | 35 scoped_refptr<DownloadRequestLimiter> limiter; |
| 37 int render_process_id; | 36 base::Callback<content::WebContents*(void)> web_contents_getter; |
| 38 int render_view_id; | |
| 39 GURL url; | 37 GURL url; |
| 40 std::string request_method; | 38 std::string request_method; |
| 41 DownloadRequestLimiter::Callback continue_callback; | 39 DownloadRequestLimiter::Callback continue_callback; |
| 42 private: | 40 private: |
| 43 DISALLOW_COPY_AND_ASSIGN(DownloadRequestInfo); | 41 DISALLOW_COPY_AND_ASSIGN(DownloadRequestInfo); |
| 44 }; | 42 }; |
| 45 | 43 |
| 46 DownloadResourceThrottle(scoped_refptr<DownloadRequestLimiter> limiter, | 44 DownloadResourceThrottle( |
| 47 int render_process_id, | 45 scoped_refptr<DownloadRequestLimiter> limiter, |
| 48 int render_view_id, | 46 const base::Callback<content::WebContents*(void)>& web_contents_getter, |
| 49 const GURL& url, | 47 const GURL& url, |
| 50 const std::string& request_method); | 48 const std::string& request_method); |
| 51 | 49 |
| 52 // content::ResourceThrottle implementation: | 50 // content::ResourceThrottle implementation: |
| 53 void WillStartRequest(bool* defer) override; | 51 void WillStartRequest(bool* defer) override; |
| 54 void WillRedirectRequest(const net::RedirectInfo& redirect_info, | 52 void WillRedirectRequest(const net::RedirectInfo& redirect_info, |
| 55 bool* defer) override; | 53 bool* defer) override; |
| 56 void WillProcessResponse(bool* defer) override; | 54 void WillProcessResponse(bool* defer) override; |
| 57 const char* GetNameForLogging() const override; | 55 const char* GetNameForLogging() const override; |
| 58 | 56 |
| 59 void ContinueDownload(bool allow); | 57 void ContinueDownload(bool allow); |
| 60 | 58 |
| 61 private: | 59 private: |
| 62 ~DownloadResourceThrottle() override; | 60 ~DownloadResourceThrottle() override; |
| 63 | 61 |
| 64 void WillDownload(bool* defer); | 62 void WillDownload(bool* defer); |
| 65 | 63 |
| 66 // Set to true when we are querying the DownloadRequestLimiter. | 64 // Set to true when we are querying the DownloadRequestLimiter. |
| 67 bool querying_limiter_; | 65 bool querying_limiter_; |
| 68 | 66 |
| 69 // Set to true when we know that the request is allowed to start. | 67 // Set to true when we know that the request is allowed to start. |
| 70 bool request_allowed_; | 68 bool request_allowed_; |
| 71 | 69 |
| 72 // Set to true when we have deferred the request. | 70 // Set to true when we have deferred the request. |
| 73 bool request_deferred_; | 71 bool request_deferred_; |
| 74 | 72 |
| 75 DISALLOW_COPY_AND_ASSIGN(DownloadResourceThrottle); | 73 DISALLOW_COPY_AND_ASSIGN(DownloadResourceThrottle); |
| 76 }; | 74 }; |
| 77 | 75 |
| 78 #endif // CHROME_BROWSER_DOWNLOAD_DOWNLOAD_RESOURCE_THROTTLE_H_ | 76 #endif // CHROME_BROWSER_DOWNLOAD_DOWNLOAD_RESOURCE_THROTTLE_H_ |
| OLD | NEW |