| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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_SAFE_BROWSING_DOWNLOAD_FEEDBACK_SERVICE_H_ | 5 #ifndef CHROME_BROWSER_SAFE_BROWSING_DOWNLOAD_FEEDBACK_SERVICE_H_ |
| 6 #define CHROME_BROWSER_SAFE_BROWSING_DOWNLOAD_FEEDBACK_SERVICE_H_ | 6 #define CHROME_BROWSER_SAFE_BROWSING_DOWNLOAD_FEEDBACK_SERVICE_H_ |
| 7 | 7 |
| 8 #include <memory> | 8 #include <memory> |
| 9 #include <queue> | 9 #include <queue> |
| 10 #include <string> | 10 #include <string> |
| 11 | 11 |
| 12 #include "base/macros.h" | 12 #include "base/macros.h" |
| 13 #include "base/memory/ref_counted.h" | 13 #include "base/memory/ref_counted.h" |
| 14 #include "chrome/browser/download/download_commands.h" |
| 14 #include "chrome/browser/safe_browsing/download_protection_service.h" | 15 #include "chrome/browser/safe_browsing/download_protection_service.h" |
| 15 #include "content/public/browser/download_danger_type.h" | 16 #include "content/public/browser/download_danger_type.h" |
| 16 | 17 |
| 17 namespace base { | 18 namespace base { |
| 18 class TaskRunner; | 19 class TaskRunner; |
| 19 } | 20 } |
| 20 | 21 |
| 21 namespace content { | 22 namespace content { |
| 22 class DownloadItem; | 23 class DownloadItem; |
| 23 } | 24 } |
| (...skipping 32 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 56 // are present. | 57 // are present. |
| 57 static bool GetPingsForDownloadForTesting( | 58 static bool GetPingsForDownloadForTesting( |
| 58 const content::DownloadItem& download, | 59 const content::DownloadItem& download, |
| 59 std::string* ping, | 60 std::string* ping, |
| 60 std::string* response); | 61 std::string* response); |
| 61 | 62 |
| 62 // Records histogram for download feedback option shown to user. | 63 // Records histogram for download feedback option shown to user. |
| 63 static void RecordEligibleDownloadShown( | 64 static void RecordEligibleDownloadShown( |
| 64 content::DownloadDangerType danger_type); | 65 content::DownloadDangerType danger_type); |
| 65 | 66 |
| 66 // Begin download feedback for |download|. The |download| will be deleted | 67 // Begin download feedback for |download|. Then delete download file if |
| 67 // when this function returns. This must only be called if | 68 // |download_command| is DISCARD, or run the KEEP command otherwise.This must |
| 68 // IsEnabledForDownload is true for |download|. | 69 // only be called if IsEnabledForDownload is true for |download|. |
| 69 void BeginFeedbackForDownload(content::DownloadItem* download); | 70 void BeginFeedbackForDownload(content::DownloadItem* download, |
| 71 DownloadCommands::Command download_command); |
| 70 | 72 |
| 71 private: | 73 private: |
| 72 static void BeginFeedbackOrDeleteFile( | 74 static void BeginFeedbackOrDeleteFile( |
| 73 const scoped_refptr<base::TaskRunner>& file_task_runner, | 75 const scoped_refptr<base::TaskRunner>& file_task_runner, |
| 74 const base::WeakPtr<DownloadFeedbackService>& service, | 76 const base::WeakPtr<DownloadFeedbackService>& service, |
| 75 const std::string& ping_request, | 77 const std::string& ping_request, |
| 76 const std::string& ping_response, | 78 const std::string& ping_response, |
| 77 const base::FilePath& path); | 79 const base::FilePath& path); |
| 78 void StartPendingFeedback(); | 80 void StartPendingFeedback(); |
| 79 void BeginFeedback(const std::string& ping_request, | 81 void BeginFeedback(const std::string& ping_request, |
| 80 const std::string& ping_response, | 82 const std::string& ping_response, |
| 81 const base::FilePath& path); | 83 const base::FilePath& path); |
| 82 void FeedbackComplete(); | 84 void FeedbackComplete(); |
| 83 | 85 |
| 84 scoped_refptr<net::URLRequestContextGetter> request_context_getter_; | 86 scoped_refptr<net::URLRequestContextGetter> request_context_getter_; |
| 85 scoped_refptr<base::TaskRunner> file_task_runner_; | 87 scoped_refptr<base::TaskRunner> file_task_runner_; |
| 86 | 88 |
| 87 // Currently active & pending uploads. The first item is active, remaining | 89 // Currently active & pending uploads. The first item is active, remaining |
| 88 // items are pending. | 90 // items are pending. |
| 89 std::queue<std::unique_ptr<DownloadFeedback>> active_feedback_; | 91 std::queue<std::unique_ptr<DownloadFeedback>> active_feedback_; |
| 90 | 92 |
| 91 base::WeakPtrFactory<DownloadFeedbackService> weak_ptr_factory_; | 93 base::WeakPtrFactory<DownloadFeedbackService> weak_ptr_factory_; |
| 92 | 94 |
| 93 DISALLOW_COPY_AND_ASSIGN(DownloadFeedbackService); | 95 DISALLOW_COPY_AND_ASSIGN(DownloadFeedbackService); |
| 94 }; | 96 }; |
| 95 } // namespace safe_browsing | 97 } // namespace safe_browsing |
| 96 | 98 |
| 97 #endif // CHROME_BROWSER_SAFE_BROWSING_DOWNLOAD_FEEDBACK_SERVICE_H_ | 99 #endif // CHROME_BROWSER_SAFE_BROWSING_DOWNLOAD_FEEDBACK_SERVICE_H_ |
| OLD | NEW |