OLD | NEW |
(Empty) | |
| 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 |
| 3 // found in the LICENSE file. |
| 4 |
| 5 #ifndef CHROME_BROWSER_SAFE_BROWSING_DOWNLOAD_FEEDBACK_SERVICE_H_ |
| 6 #define CHROME_BROWSER_SAFE_BROWSING_DOWNLOAD_FEEDBACK_SERVICE_H_ |
| 7 |
| 8 #include <string> |
| 9 |
| 10 #include "base/basictypes.h" |
| 11 #include "base/memory/ref_counted.h" |
| 12 #include "base/memory/scoped_vector.h" |
| 13 #include "base/threading/non_thread_safe.h" |
| 14 #include "chrome/browser/safe_browsing/download_protection_service.h" |
| 15 #include "content/public/browser/download_danger_type.h" |
| 16 |
| 17 namespace base { |
| 18 class TaskRunner; |
| 19 } |
| 20 |
| 21 namespace content { |
| 22 class DownloadItem; |
| 23 } |
| 24 |
| 25 namespace net { |
| 26 class URLRequestContextGetter; |
| 27 } |
| 28 |
| 29 namespace safe_browsing { |
| 30 |
| 31 class DownloadFeedback; |
| 32 |
| 33 // Tracks active DownloadFeedback objects, provides interface for storing ping |
| 34 // data for malicious downloads. |
| 35 class DownloadFeedbackService : public base::NonThreadSafe { |
| 36 public: |
| 37 DownloadFeedbackService(net::URLRequestContextGetter* request_context_getter, |
| 38 base::TaskRunner* file_task_runner); |
| 39 ~DownloadFeedbackService(); |
| 40 |
| 41 // Stores the request and response ping data from the download check, if the |
| 42 // check result and file size are eligible. This must be called after a |
| 43 // download has been flagged as malicious in order for the download to be |
| 44 // enabled for uploading. |
| 45 static void MaybeStorePingsForDownload( |
| 46 DownloadProtectionService::DownloadCheckResult result, |
| 47 content::DownloadItem* download, |
| 48 const std::string& ping, |
| 49 const std::string& response); |
| 50 |
| 51 // Test if pings have been stored for |download|. |
| 52 static bool IsEnabledForDownload(const content::DownloadItem& download); |
| 53 |
| 54 // Get the ping values stored in |download|. Returns false if no ping values |
| 55 // are present. |
| 56 static bool GetPingsForDownloadForTesting( |
| 57 const content::DownloadItem& download, |
| 58 std::string* ping, |
| 59 std::string* response); |
| 60 |
| 61 // Records histogram for download feedback option shown to user. |
| 62 static void RecordFeedbackButtonShown( |
| 63 content::DownloadDangerType danger_type); |
| 64 |
| 65 // Begin download feedback for |download|. The |download| will be deleted |
| 66 // when this function returns. This must only be called if |
| 67 // IsEnabledForDownload is true for |download|. |
| 68 void BeginFeedbackForDownload(content::DownloadItem* download); |
| 69 |
| 70 private: |
| 71 static void BeginFeedbackOrDeleteFile( |
| 72 const scoped_refptr<base::TaskRunner>& file_task_runner, |
| 73 const base::WeakPtr<DownloadFeedbackService>& service, |
| 74 const std::string& ping_request, |
| 75 const std::string& ping_response, |
| 76 const base::FilePath& path); |
| 77 void StartPendingFeedback(); |
| 78 void BeginFeedback(const std::string& ping_request, |
| 79 const std::string& ping_response, |
| 80 const base::FilePath& path); |
| 81 void FeedbackComplete(); |
| 82 |
| 83 scoped_refptr<net::URLRequestContextGetter> request_context_getter_; |
| 84 scoped_refptr<base::TaskRunner> file_task_runner_; |
| 85 |
| 86 // Currently active & pending uploads. The first item is active, remaining |
| 87 // items are pending. |
| 88 ScopedVector<DownloadFeedback> active_feedback_; |
| 89 |
| 90 base::WeakPtrFactory<DownloadFeedbackService> weak_ptr_factory_; |
| 91 |
| 92 DISALLOW_COPY_AND_ASSIGN(DownloadFeedbackService); |
| 93 }; |
| 94 } // namespace safe_browsing |
| 95 |
| 96 #endif // CHROME_BROWSER_SAFE_BROWSING_DOWNLOAD_FEEDBACK_SERVICE_H_ |
OLD | NEW |