src/chrome/browser/safe_browsing/download_feedback.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_H_
   6 #define CHROME_BROWSER_SAFE_BROWSING_DOWNLOAD_FEEDBACK_H_
   7
   8 #include <string>
   9
  10 #include "base/callback_forward.h"
  11 #include "base/files/file_path.h"
  12 #include "base/memory/scoped_ptr.h"
  13 #include "base/threading/non_thread_safe.h"
  14 #include "chrome/browser/safe_browsing/two_phase_uploader.h"
  15
  16 namespace content {
  17 class DownloadItem;
  18 }
  19
  20 namespace safe_browsing {
  21
  22 class DownloadFeedbackFactory;
  23
  24 // Handles the uploading of a single downloaded binary to the safebrowsing
  25 // download feedback service.
  26 class DownloadFeedback : public base::NonThreadSafe {
  27  public:
  28   // Takes ownership of the file pointed to be |file_path|, it will be deleted
  29   // when the DownloadFeedback is destructed.
  30   static DownloadFeedback* Create(
  31       net::URLRequestContextGetter* request_context_getter,
  32       base::TaskRunner* file_task_runner,
  33       const base::FilePath& file_path,
  34       const std::string& ping_request,
  35       const std::string& ping_response);
  36
  37   // The largest file size we support uploading.
  38   // Note: changing this will affect the max size of
  39   // SBDownloadFeedback.SizeSuccess and SizeFailure histograms.
  40   static const int64 kMaxUploadSize;
  41
  42   // The URL where the browser sends download feedback requests.
  43   static const char kSbFeedbackURL[];
  44
  45   virtual ~DownloadFeedback() {}
  46
  47   // Makes the passed |factory| the factory used to instantiate
  48   // a DownloadFeedback. Useful for tests.
  49   static void RegisterFactory(DownloadFeedbackFactory* factory) {
  50     factory_ = factory;
  51   }
  52
  53   // Start uploading the file to the download feedback service.
  54   // |finish_callback| will be called when the upload completes or fails, but is
  55   // not called if the upload is cancelled by deleting the DownloadFeedback
  56   // while the upload is in progress.
  57   virtual void Start(const base::Closure& finish_callback) = 0;
  58
  59   virtual const std::string& GetPingRequestForTesting() const = 0;
  60   virtual const std::string& GetPingResponseForTesting() const = 0;
  61
  62  private:
  63   // The factory that controls the creation of DownloadFeedback objects.
  64   // This is used by tests.
  65   static DownloadFeedbackFactory* factory_;
  66 };
  67
  68 class DownloadFeedbackFactory {
  69  public:
  70   virtual ~DownloadFeedbackFactory() {}
  71
  72   virtual DownloadFeedback* CreateDownloadFeedback(
  73       net::URLRequestContextGetter* request_context_getter,
  74       base::TaskRunner* file_task_runner,
  75       const base::FilePath& file_path,
  76       const std::string& ping_request,
  77       const std::string& ping_response) = 0;
  78 };
  79
  80 }  // namespace safe_browsing
  81
  82 #endif  // CHROME_BROWSER_SAFE_BROWSING_DOWNLOAD_FEEDBACK_H_