chrome/browser/safe_browsing/download_feedback_service.h

   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 RecordEligibleDownloadShown(
  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_