1 // Copyright (c) 2011 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_PLUGIN_DOWNLOAD_HELPER_H_ 6 #define CHROME_BROWSER_PLUGIN_DOWNLOAD_HELPER_H_ 7 #pragma once 8 9 #include <string> 10 #include "build/build_config.h" 11 12 #if defined(OS_WIN) 13 #include "base/file_path.h" 14 #include "net/base/file_stream.h" 15 #include "net/url_request/url_request.h" 16 #include "ui/gfx/native_widget_types.h" 17 18 // The PluginDownloadUrlHelper is used to handle one download URL request 19 // from the plugin. Each download request is handled by a new instance 20 // of this class. 21 class PluginDownloadUrlHelper : public net::URLRequest::Delegate { 22 static const int kDownloadFileBufferSize = 32768; 23 public: 24 // The delegate receives notification about the status of downloads 25 // initiated. 26 class DownloadDelegate { 27 public: ~DownloadDelegate()28 virtual ~DownloadDelegate() {} 29 OnDownloadCompleted(const FilePath & download_path,bool success)30 virtual void OnDownloadCompleted(const FilePath& download_path, 31 bool success) {} 32 }; 33 34 PluginDownloadUrlHelper(const std::string& download_url, 35 int source_pid, gfx::NativeWindow caller_window, 36 PluginDownloadUrlHelper::DownloadDelegate* delegate); 37 ~PluginDownloadUrlHelper(); 38 39 void InitiateDownload(net::URLRequestContext* request_context); 40 41 // net::URLRequest::Delegate 42 virtual void OnAuthRequired(net::URLRequest* request, 43 net::AuthChallengeInfo* auth_info); 44 virtual void OnSSLCertificateError(net::URLRequest* request, 45 int cert_error, 46 net::X509Certificate* cert); 47 virtual void OnResponseStarted(net::URLRequest* request); 48 virtual void OnReadCompleted(net::URLRequest* request, int bytes_read); 49 50 void OnDownloadCompleted(net::URLRequest* request); 51 52 protected: 53 void DownloadCompletedHelper(bool success); 54 55 // The download file request initiated by the plugin. 56 net::URLRequest* download_file_request_; 57 // Handle to the downloaded file. 58 scoped_ptr<net::FileStream> download_file_; 59 // The full path of the downloaded file. 60 FilePath download_file_path_; 61 // The buffer passed off to net::URLRequest::Read. 62 scoped_refptr<net::IOBuffer> download_file_buffer_; 63 // TODO(port): this comment doesn't describe the situation on Posix. 64 // The window handle for sending the WM_COPYDATA notification, 65 // indicating that the download completed. 66 gfx::NativeWindow download_file_caller_window_; 67 68 std::string download_url_; 69 int download_source_child_unique_id_; 70 71 PluginDownloadUrlHelper::DownloadDelegate* delegate_; 72 73 DISALLOW_COPY_AND_ASSIGN(PluginDownloadUrlHelper); 74 }; 75 76 #endif // OS_WIN 77 78 #endif // CHROME_BROWSER_PLUGIN_DOWNLOAD_HELPER_H_ 79 80 81