Index: chrome/browser/safe_browsing/download_feedback.h |
diff --git a/chrome/browser/safe_browsing/download_feedback.h b/chrome/browser/safe_browsing/download_feedback.h |
new file mode 100644 |
index 0000000000000000000000000000000000000000..2a9a6c7f2f245a5e892ff0b94f7eae844199cd25 |
--- /dev/null |
+++ b/chrome/browser/safe_browsing/download_feedback.h |
@@ -0,0 +1,79 @@ |
+// Copyright 2013 The Chromium Authors. All rights reserved. |
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+#ifndef CHROME_BROWSER_SAFE_BROWSING_DOWNLOAD_FEEDBACK_H_ |
+#define CHROME_BROWSER_SAFE_BROWSING_DOWNLOAD_FEEDBACK_H_ |
+ |
+#include <string> |
+ |
+#include "base/callback_forward.h" |
+#include "base/files/file_path.h" |
+#include "base/memory/scoped_ptr.h" |
+#include "base/threading/non_thread_safe.h" |
+#include "chrome/browser/safe_browsing/two_phase_uploader.h" |
+ |
+namespace content { |
+class DownloadItem; |
+} |
+ |
+namespace safe_browsing { |
+ |
+class DownloadFeedbackFactory; |
+ |
+// Handles the uploading of a single downloaded binary to the safebrowsing |
+// download feedback service. |
+class DownloadFeedback : public base::NonThreadSafe { |
+ public: |
+ // Takes ownership of the file pointed to be |file_path|, it will be deleted |
+ // when the DownloadFeedback is destructed. |
+ static DownloadFeedback* Create( |
+ net::URLRequestContextGetter* request_context_getter, |
+ base::TaskRunner* file_task_runner, |
+ const base::FilePath& file_path, |
+ const std::string& ping_request, |
+ const std::string& ping_response); |
+ |
+ // The largest file size we support uploading. |
+ // Note: changing this will affect the max size of |
+ // SBDownloadFeedback.SizeSuccess and SizeFailure histograms. |
+ static const int64 kMaxUploadSize; |
+ |
+ virtual ~DownloadFeedback() {} |
+ |
+ // Makes the passed |factory| the factory used to instantiate |
+ // a DownloadFeedback. Useful for tests. |
+ static void RegisterFactory(DownloadFeedbackFactory* factory) { |
+ factory_ = factory; |
+ } |
+ |
+ // Start uploading the file to the download feedback service. |
+ // |finish_callback| will be called when the upload completes or fails, but is |
+ // not called if the upload is cancelled by deleting the DownloadFeedback |
+ // while the upload is in progress. |
+ virtual void Start(const base::Closure& finish_callback) = 0; |
+ |
+ virtual const std::string& GetPingRequestForTesting() const = 0; |
+ virtual const std::string& GetPingResponseForTesting() const = 0; |
+ |
+ private: |
+ // The factory that controls the creation of DownloadFeedback objects. |
+ // This is used by tests. |
+ static DownloadFeedbackFactory* factory_; |
+}; |
+ |
+class DownloadFeedbackFactory { |
+ public: |
+ virtual ~DownloadFeedbackFactory() {} |
+ |
+ virtual DownloadFeedback* CreateDownloadFeedback( |
+ net::URLRequestContextGetter* request_context_getter, |
+ base::TaskRunner* file_task_runner, |
+ const base::FilePath& file_path, |
+ const std::string& ping_request, |
+ const std::string& ping_response) = 0; |
+}; |
+ |
+} // namespace safe_browsing |
+ |
+#endif // CHROME_BROWSER_SAFE_BROWSING_DOWNLOAD_FEEDBACK_H_ |