17d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)// Copyright 2013 The Chromium Authors. All rights reserved. 27d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 37d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)// found in the LICENSE file. 47d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 57d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)#ifndef CHROME_BROWSER_SAFE_BROWSING_DOWNLOAD_FEEDBACK_H_ 67d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)#define CHROME_BROWSER_SAFE_BROWSING_DOWNLOAD_FEEDBACK_H_ 77d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 87d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)#include <string> 97d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 107d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)#include "base/callback_forward.h" 117d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)#include "base/files/file_path.h" 127d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)#include "base/memory/scoped_ptr.h" 137d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)#include "base/threading/non_thread_safe.h" 147d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)#include "chrome/browser/safe_browsing/two_phase_uploader.h" 157d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 167d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)namespace content { 177d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)class DownloadItem; 187d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)} 197d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 207d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)namespace safe_browsing { 217d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 227d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)class DownloadFeedbackFactory; 237d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 247d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)// Handles the uploading of a single downloaded binary to the safebrowsing 257d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)// download feedback service. 267d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)class DownloadFeedback : public base::NonThreadSafe { 277d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) public: 287d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) // Takes ownership of the file pointed to be |file_path|, it will be deleted 297d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) // when the DownloadFeedback is destructed. 307d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) static DownloadFeedback* Create( 317d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) net::URLRequestContextGetter* request_context_getter, 327d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) base::TaskRunner* file_task_runner, 337d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) const base::FilePath& file_path, 347d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) const std::string& ping_request, 357d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) const std::string& ping_response); 367d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 377d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) // The largest file size we support uploading. 387d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) // Note: changing this will affect the max size of 397d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) // SBDownloadFeedback.SizeSuccess and SizeFailure histograms. 407d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) static const int64 kMaxUploadSize; 417d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 420529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch // The URL where the browser sends download feedback requests. 430529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch static const char kSbFeedbackURL[]; 440529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch 457d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) virtual ~DownloadFeedback() {} 467d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 477d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) // Makes the passed |factory| the factory used to instantiate 487d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) // a DownloadFeedback. Useful for tests. 497d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) static void RegisterFactory(DownloadFeedbackFactory* factory) { 507d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) factory_ = factory; 517d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) } 527d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 537d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) // Start uploading the file to the download feedback service. 547d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) // |finish_callback| will be called when the upload completes or fails, but is 557d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) // not called if the upload is cancelled by deleting the DownloadFeedback 567d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) // while the upload is in progress. 577d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) virtual void Start(const base::Closure& finish_callback) = 0; 587d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 597d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) virtual const std::string& GetPingRequestForTesting() const = 0; 607d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) virtual const std::string& GetPingResponseForTesting() const = 0; 617d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 627d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) private: 637d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) // The factory that controls the creation of DownloadFeedback objects. 647d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) // This is used by tests. 657d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) static DownloadFeedbackFactory* factory_; 667d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)}; 677d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 687d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)class DownloadFeedbackFactory { 697d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) public: 707d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) virtual ~DownloadFeedbackFactory() {} 717d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 727d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) virtual DownloadFeedback* CreateDownloadFeedback( 737d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) net::URLRequestContextGetter* request_context_getter, 747d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) base::TaskRunner* file_task_runner, 757d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) const base::FilePath& file_path, 767d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) const std::string& ping_request, 777d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) const std::string& ping_response) = 0; 787d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)}; 797d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 807d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)} // namespace safe_browsing 817d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 827d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)#endif // CHROME_BROWSER_SAFE_BROWSING_DOWNLOAD_FEEDBACK_H_ 83