| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "content/browser/download/download_file_impl.h" | 5 #include "content/browser/download/download_file_impl.h" |
| 6 | 6 |
| 7 #include <string> | 7 #include <string> |
| 8 | 8 |
| 9 #include "base/bind.h" | 9 #include "base/bind.h" |
| 10 #include "base/file_util.h" | 10 #include "base/file_util.h" |
| 11 #include "base/message_loop_proxy.h" | 11 #include "base/message_loop_proxy.h" |
| 12 #include "base/time.h" | 12 #include "base/time.h" |
| 13 #include "content/browser/download/byte_stream.h" | 13 #include "content/browser/download/byte_stream.h" |
| 14 #include "content/browser/download/download_create_info.h" | 14 #include "content/browser/download/download_create_info.h" |
| 15 #include "content/browser/download/download_interrupt_reasons_impl.h" | 15 #include "content/browser/download/download_interrupt_reasons_impl.h" |
| 16 #include "content/browser/download/download_net_log_parameters.h" | 16 #include "content/browser/download/download_net_log_parameters.h" |
| 17 #include "content/browser/power_save_blocker.h" | 17 #include "content/browser/power_save_blocker.h" |
| 18 #include "content/public/browser/browser_thread.h" | 18 #include "content/public/browser/browser_thread.h" |
| 19 #include "content/public/browser/download_manager.h" | 19 #include "content/public/browser/download_destination_observer.h" |
| 20 #include "content/browser/download/download_stats.h" | 20 #include "content/browser/download/download_stats.h" |
| 21 #include "net/base/io_buffer.h" | 21 #include "net/base/io_buffer.h" |
| 22 | 22 |
| 23 using content::BrowserThread; | 23 using content::BrowserThread; |
| 24 using content::DownloadId; | 24 using content::DownloadId; |
| 25 using content::DownloadManager; | 25 using content::DownloadManager; |
| 26 | 26 |
| 27 const int kUpdatePeriodMs = 500; | 27 const int kUpdatePeriodMs = 500; |
| 28 const int kMaxTimeBlockingFileThreadMs = 1000; | 28 const int kMaxTimeBlockingFileThreadMs = 1000; |
| 29 | 29 |
| 30 int content::DownloadFile::number_active_objects_ = 0; |
| 31 |
| 30 DownloadFileImpl::DownloadFileImpl( | 32 DownloadFileImpl::DownloadFileImpl( |
| 31 const DownloadCreateInfo* info, | 33 const content::DownloadSaveInfo& save_info, |
| 34 const FilePath& default_download_directory, |
| 35 const GURL& url, |
| 36 const GURL& referrer_url, |
| 37 int64 received_bytes, |
| 38 bool calculate_hash, |
| 32 scoped_ptr<content::ByteStreamReader> stream, | 39 scoped_ptr<content::ByteStreamReader> stream, |
| 33 DownloadRequestHandleInterface* request_handle, | 40 const net::BoundNetLog& bound_net_log, |
| 34 scoped_refptr<DownloadManager> download_manager, | |
| 35 bool calculate_hash, | |
| 36 scoped_ptr<content::PowerSaveBlocker> power_save_blocker, | 41 scoped_ptr<content::PowerSaveBlocker> power_save_blocker, |
| 37 const net::BoundNetLog& bound_net_log) | 42 base::WeakPtr<content::DownloadDestinationObserver> observer) |
| 38 : file_(info->save_info.file_path, | 43 : file_(save_info.file_path, |
| 39 info->url(), | 44 url, |
| 40 info->referrer_url, | 45 referrer_url, |
| 41 info->received_bytes, | 46 received_bytes, |
| 42 calculate_hash, | 47 calculate_hash, |
| 43 info->save_info.hash_state, | 48 save_info.hash_state, |
| 44 info->save_info.file_stream, | 49 save_info.file_stream, |
| 45 bound_net_log), | 50 bound_net_log), |
| 51 default_download_directory_(default_download_directory), |
| 46 stream_reader_(stream.Pass()), | 52 stream_reader_(stream.Pass()), |
| 47 id_(info->download_id), | |
| 48 default_download_directory_(info->default_download_directory), | |
| 49 request_handle_(request_handle), | |
| 50 download_manager_(download_manager), | |
| 51 bytes_seen_(0), | 53 bytes_seen_(0), |
| 52 bound_net_log_(bound_net_log), | 54 bound_net_log_(bound_net_log), |
| 55 observer_(observer), |
| 53 weak_factory_(ALLOW_THIS_IN_INITIALIZER_LIST(this)), | 56 weak_factory_(ALLOW_THIS_IN_INITIALIZER_LIST(this)), |
| 54 power_save_blocker_(power_save_blocker.Pass()) { | 57 power_save_blocker_(power_save_blocker.Pass()) { |
| 55 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); | |
| 56 DCHECK(download_manager.get()); | |
| 57 } | 58 } |
| 58 | 59 |
| 59 DownloadFileImpl::~DownloadFileImpl() { | 60 DownloadFileImpl::~DownloadFileImpl() { |
| 61 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); |
| 62 --number_active_objects_; |
| 60 } | 63 } |
| 61 | 64 |
| 62 content::DownloadInterruptReason DownloadFileImpl::Initialize() { | 65 void DownloadFileImpl::Initialize(const InitializeCallback& callback) { |
| 63 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); | 66 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); |
| 64 | 67 |
| 65 update_timer_.reset(new base::RepeatingTimer<DownloadFileImpl>()); | 68 update_timer_.reset(new base::RepeatingTimer<DownloadFileImpl>()); |
| 66 net::Error net_result = file_.Initialize(default_download_directory_); | 69 net::Error net_result = file_.Initialize(default_download_directory_); |
| 67 if (net_result != net::OK) { | 70 if (net_result != net::OK) { |
| 68 return content::ConvertNetErrorToInterruptReason( | 71 BrowserThread::PostTask( |
| 69 net_result, content::DOWNLOAD_INTERRUPT_FROM_DISK); | 72 BrowserThread::UI, FROM_HERE, base::Bind( |
| 73 callback, content::ConvertNetErrorToInterruptReason( |
| 74 net_result, content::DOWNLOAD_INTERRUPT_FROM_DISK))); |
| 75 return; |
| 70 } | 76 } |
| 71 | 77 |
| 72 stream_reader_->RegisterCallback( | 78 stream_reader_->RegisterCallback( |
| 73 base::Bind(&DownloadFileImpl::StreamActive, weak_factory_.GetWeakPtr())); | 79 base::Bind(&DownloadFileImpl::StreamActive, weak_factory_.GetWeakPtr())); |
| 74 | 80 |
| 75 download_start_ = base::TimeTicks::Now(); | 81 download_start_ = base::TimeTicks::Now(); |
| 76 | 82 |
| 77 // Initial pull from the straw. | 83 // Initial pull from the straw. |
| 78 StreamActive(); | 84 StreamActive(); |
| 79 | 85 |
| 80 return content::DOWNLOAD_INTERRUPT_REASON_NONE; | 86 BrowserThread::PostTask( |
| 87 BrowserThread::UI, FROM_HERE, base::Bind( |
| 88 callback, content::DOWNLOAD_INTERRUPT_REASON_NONE)); |
| 89 |
| 90 ++number_active_objects_; |
| 81 } | 91 } |
| 82 | 92 |
| 83 content::DownloadInterruptReason DownloadFileImpl::AppendDataToFile( | 93 content::DownloadInterruptReason DownloadFileImpl::AppendDataToFile( |
| 84 const char* data, size_t data_len) { | 94 const char* data, size_t data_len) { |
| 95 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); |
| 96 |
| 85 if (!update_timer_->IsRunning()) { | 97 if (!update_timer_->IsRunning()) { |
| 86 update_timer_->Start(FROM_HERE, | 98 update_timer_->Start(FROM_HERE, |
| 87 base::TimeDelta::FromMilliseconds(kUpdatePeriodMs), | 99 base::TimeDelta::FromMilliseconds(kUpdatePeriodMs), |
| 88 this, &DownloadFileImpl::SendUpdate); | 100 this, &DownloadFileImpl::SendUpdate); |
| 89 } | 101 } |
| 90 return content::ConvertNetErrorToInterruptReason( | 102 return content::ConvertNetErrorToInterruptReason( |
| 91 file_.AppendDataToFile(data, data_len), | 103 file_.AppendDataToFile(data, data_len), |
| 92 content::DOWNLOAD_INTERRUPT_FROM_DISK); | 104 content::DOWNLOAD_INTERRUPT_FROM_DISK); |
| 93 } | 105 } |
| 94 | 106 |
| 95 void DownloadFileImpl::Rename(const FilePath& full_path, | 107 void DownloadFileImpl::Rename(const FilePath& full_path, |
| 96 bool overwrite_existing_file, | 108 bool overwrite_existing_file, |
| 97 const RenameCompletionCallback& callback) { | 109 const RenameCompletionCallback& callback) { |
| 110 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); |
| 111 |
| 98 FilePath new_path(full_path); | 112 FilePath new_path(full_path); |
| 99 if (!overwrite_existing_file) { | 113 if (!overwrite_existing_file) { |
| 100 // Make the file unique if requested. | 114 // Make the file unique if requested. |
| 101 int uniquifier = | 115 int uniquifier = |
| 102 file_util::GetUniquePathNumber(new_path, FILE_PATH_LITERAL("")); | 116 file_util::GetUniquePathNumber(new_path, FILE_PATH_LITERAL("")); |
| 103 if (uniquifier > 0) { | 117 if (uniquifier > 0) { |
| 104 new_path = new_path.InsertBeforeExtensionASCII( | 118 new_path = new_path.InsertBeforeExtensionASCII( |
| 105 StringPrintf(" (%d)", uniquifier)); | 119 StringPrintf(" (%d)", uniquifier)); |
| 106 } | 120 } |
| 107 } | 121 } |
| (...skipping 61 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 169 } | 183 } |
| 170 | 184 |
| 171 bool DownloadFileImpl::GetHash(std::string* hash) { | 185 bool DownloadFileImpl::GetHash(std::string* hash) { |
| 172 return file_.GetHash(hash); | 186 return file_.GetHash(hash); |
| 173 } | 187 } |
| 174 | 188 |
| 175 std::string DownloadFileImpl::GetHashState() { | 189 std::string DownloadFileImpl::GetHashState() { |
| 176 return file_.GetHashState(); | 190 return file_.GetHashState(); |
| 177 } | 191 } |
| 178 | 192 |
| 179 // DownloadFileInterface implementation. | |
| 180 void DownloadFileImpl::CancelDownloadRequest() { | |
| 181 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); | |
| 182 request_handle_->CancelRequest(); | |
| 183 } | |
| 184 | |
| 185 int DownloadFileImpl::Id() const { | |
| 186 return id_.local(); | |
| 187 } | |
| 188 | |
| 189 DownloadManager* DownloadFileImpl::GetDownloadManager() { | |
| 190 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); | |
| 191 return download_manager_.get(); | |
| 192 } | |
| 193 | |
| 194 const DownloadId& DownloadFileImpl::GlobalId() const { | |
| 195 return id_; | |
| 196 } | |
| 197 | |
| 198 std::string DownloadFileImpl::DebugString() const { | |
| 199 return base::StringPrintf("{" | |
| 200 " id_ = " "%d" | |
| 201 " request_handle = %s" | |
| 202 " Base File = %s" | |
| 203 " }", | |
| 204 id_.local(), | |
| 205 request_handle_->DebugString().c_str(), | |
| 206 file_.DebugString().c_str()); | |
| 207 } | |
| 208 | |
| 209 void DownloadFileImpl::StreamActive() { | 193 void DownloadFileImpl::StreamActive() { |
| 210 base::TimeTicks start(base::TimeTicks::Now()); | 194 base::TimeTicks start(base::TimeTicks::Now()); |
| 211 base::TimeTicks now; | 195 base::TimeTicks now; |
| 212 scoped_refptr<net::IOBuffer> incoming_data; | 196 scoped_refptr<net::IOBuffer> incoming_data; |
| 213 size_t incoming_data_size = 0; | 197 size_t incoming_data_size = 0; |
| 214 size_t total_incoming_data_size = 0; | 198 size_t total_incoming_data_size = 0; |
| 215 size_t num_buffers = 0; | 199 size_t num_buffers = 0; |
| 216 content::ByteStreamReader::StreamState state( | 200 content::ByteStreamReader::StreamState state( |
| 217 content::ByteStreamReader::STREAM_EMPTY); | 201 content::ByteStreamReader::STREAM_EMPTY); |
| 218 content::DownloadInterruptReason reason = | 202 content::DownloadInterruptReason reason = |
| (...skipping 48 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 267 BrowserThread::FILE, FROM_HERE, | 251 BrowserThread::FILE, FROM_HERE, |
| 268 base::Bind(&DownloadFileImpl::StreamActive, | 252 base::Bind(&DownloadFileImpl::StreamActive, |
| 269 weak_factory_.GetWeakPtr())); | 253 weak_factory_.GetWeakPtr())); |
| 270 } | 254 } |
| 271 | 255 |
| 272 if (total_incoming_data_size) | 256 if (total_incoming_data_size) |
| 273 download_stats::RecordFileThreadReceiveBuffers(num_buffers); | 257 download_stats::RecordFileThreadReceiveBuffers(num_buffers); |
| 274 | 258 |
| 275 download_stats::RecordContiguousWriteTime(now - start); | 259 download_stats::RecordContiguousWriteTime(now - start); |
| 276 | 260 |
| 277 // Take care of communication with our controller. | 261 // Take care of communication with our observer. |
| 278 if (reason != content::DOWNLOAD_INTERRUPT_REASON_NONE) { | 262 if (reason != content::DOWNLOAD_INTERRUPT_REASON_NONE) { |
| 279 // Error case for both upstream source and file write. | 263 // Error case for both upstream source and file write. |
| 280 // Shut down processing and signal an error to our controller. | 264 // Shut down processing and signal an error to our observer. |
| 281 // Our controller will clean us up. | 265 // Our observer will clean us up. |
| 282 stream_reader_->RegisterCallback(base::Closure()); | 266 stream_reader_->RegisterCallback(base::Closure()); |
| 283 weak_factory_.InvalidateWeakPtrs(); | 267 weak_factory_.InvalidateWeakPtrs(); |
| 284 SendUpdate(); // Make info up to date before error. | 268 SendUpdate(); // Make info up to date before error. |
| 285 BrowserThread::PostTask( | 269 BrowserThread::PostTask( |
| 286 BrowserThread::UI, FROM_HERE, | 270 BrowserThread::UI, FROM_HERE, |
| 287 base::Bind(&DownloadManager::OnDownloadInterrupted, | 271 base::Bind(&content::DownloadDestinationObserver::DestinationError, |
| 288 download_manager_, id_.local(), reason)); | 272 observer_, reason)); |
| 289 } else if (state == content::ByteStreamReader::STREAM_COMPLETE) { | 273 } else if (state == content::ByteStreamReader::STREAM_COMPLETE) { |
| 290 // Signal successful completion and shut down processing. | 274 // Signal successful completion and shut down processing. |
| 291 stream_reader_->RegisterCallback(base::Closure()); | 275 stream_reader_->RegisterCallback(base::Closure()); |
| 292 weak_factory_.InvalidateWeakPtrs(); | 276 weak_factory_.InvalidateWeakPtrs(); |
| 293 std::string hash; | 277 std::string hash; |
| 294 if (!GetHash(&hash) || file_.IsEmptyHash(hash)) | 278 if (!GetHash(&hash) || file_.IsEmptyHash(hash)) |
| 295 hash.clear(); | 279 hash.clear(); |
| 280 SendUpdate(); |
| 296 BrowserThread::PostTask( | 281 BrowserThread::PostTask( |
| 297 BrowserThread::UI, FROM_HERE, | 282 BrowserThread::UI, FROM_HERE, |
| 298 base::Bind(&DownloadManager::OnResponseCompleted, | 283 base::Bind( |
| 299 download_manager_, id_.local(), | 284 &content::DownloadDestinationObserver::DestinationCompleted, |
| 300 BytesSoFar(), hash)); | 285 observer_, hash)); |
| 301 } | 286 } |
| 302 if (bound_net_log_.IsLoggingAllEvents()) { | 287 if (bound_net_log_.IsLoggingAllEvents()) { |
| 303 bound_net_log_.AddEvent( | 288 bound_net_log_.AddEvent( |
| 304 net::NetLog::TYPE_DOWNLOAD_STREAM_DRAINED, | 289 net::NetLog::TYPE_DOWNLOAD_STREAM_DRAINED, |
| 305 base::Bind(&download_net_logs::FileStreamDrainedCallback, | 290 base::Bind(&download_net_logs::FileStreamDrainedCallback, |
| 306 total_incoming_data_size, num_buffers)); | 291 total_incoming_data_size, num_buffers)); |
| 307 } | 292 } |
| 308 } | 293 } |
| 309 | 294 |
| 310 void DownloadFileImpl::SendUpdate() { | 295 void DownloadFileImpl::SendUpdate() { |
| 311 BrowserThread::PostTask( | 296 BrowserThread::PostTask( |
| 312 BrowserThread::UI, FROM_HERE, | 297 BrowserThread::UI, FROM_HERE, |
| 313 base::Bind(&DownloadManager::UpdateDownload, | 298 base::Bind(&content::DownloadDestinationObserver::DestinationUpdate, |
| 314 download_manager_, id_.local(), | 299 observer_, BytesSoFar(), CurrentSpeed(), GetHashState())); |
| 315 BytesSoFar(), CurrentSpeed(), GetHashState())); | |
| 316 } | 300 } |
| 301 |
| 302 // static |
| 303 int content::DownloadFile::GetNumberOfDownloadFiles() { |
| 304 return number_active_objects_; |
| 305 } |
| 306 |
| OLD | NEW |