| 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_destination_observer.h" | 19 #include "content/public/browser/download_manager.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 | |
| 32 DownloadFileImpl::DownloadFileImpl( | 30 DownloadFileImpl::DownloadFileImpl( |
| 33 const content::DownloadSaveInfo& save_info, | 31 const DownloadCreateInfo* info, |
| 34 const GURL& url, | 32 scoped_ptr<content::ByteStreamReader> stream, |
| 35 const GURL& referrer_url, | 33 DownloadRequestHandleInterface* request_handle, |
| 36 int64 received_bytes, | 34 scoped_refptr<DownloadManager> download_manager, |
| 37 bool calculate_hash, | 35 bool calculate_hash, |
| 38 scoped_ptr<content::ByteStreamReader> stream, | |
| 39 const net::BoundNetLog& bound_net_log, | |
| 40 scoped_ptr<content::PowerSaveBlocker> power_save_blocker, | 36 scoped_ptr<content::PowerSaveBlocker> power_save_blocker, |
| 41 base::WeakPtr<content::DownloadDestinationObserver> observer) | 37 const net::BoundNetLog& bound_net_log) |
| 42 : file_(save_info.file_path, | 38 : file_(info->save_info.file_path, |
| 43 url, | 39 info->url(), |
| 44 referrer_url, | 40 info->referrer_url, |
| 45 received_bytes, | 41 info->received_bytes, |
| 46 calculate_hash, | 42 calculate_hash, |
| 47 save_info.hash_state, | 43 info->save_info.hash_state, |
| 48 save_info.file_stream, | 44 info->save_info.file_stream, |
| 49 bound_net_log), | 45 bound_net_log), |
| 50 stream_reader_(stream.Pass()), | 46 stream_reader_(stream.Pass()), |
| 47 id_(info->download_id), |
| 48 request_handle_(request_handle), |
| 49 download_manager_(download_manager), |
| 51 bytes_seen_(0), | 50 bytes_seen_(0), |
| 52 bound_net_log_(bound_net_log), | 51 bound_net_log_(bound_net_log), |
| 53 observer_(observer), | |
| 54 weak_factory_(ALLOW_THIS_IN_INITIALIZER_LIST(this)), | 52 weak_factory_(ALLOW_THIS_IN_INITIALIZER_LIST(this)), |
| 55 power_save_blocker_(power_save_blocker.Pass()) { | 53 power_save_blocker_(power_save_blocker.Pass()) { |
| 54 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); |
| 55 DCHECK(download_manager.get()); |
| 56 } | 56 } |
| 57 | 57 |
| 58 DownloadFileImpl::~DownloadFileImpl() { | 58 DownloadFileImpl::~DownloadFileImpl() { |
| 59 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); | |
| 60 --number_active_objects_; | |
| 61 } | 59 } |
| 62 | 60 |
| 63 void DownloadFileImpl::Initialize(const InitializeCallback& callback) { | 61 content::DownloadInterruptReason DownloadFileImpl::Initialize() { |
| 64 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); | |
| 65 | |
| 66 update_timer_.reset(new base::RepeatingTimer<DownloadFileImpl>()); | 62 update_timer_.reset(new base::RepeatingTimer<DownloadFileImpl>()); |
| 67 | 63 net::Error result = file_.Initialize(); |
| 68 net::Error net_result = file_.Initialize(); | 64 if (result != net::OK) { |
| 69 if (net_result != net::OK) { | 65 return content::ConvertNetErrorToInterruptReason( |
| 70 BrowserThread::PostTask( | 66 result, content::DOWNLOAD_INTERRUPT_FROM_DISK); |
| 71 BrowserThread::UI, FROM_HERE, base::Bind( | |
| 72 callback, content::ConvertNetErrorToInterruptReason( | |
| 73 net_result, content::DOWNLOAD_INTERRUPT_FROM_DISK))); | |
| 74 return; | |
| 75 } | 67 } |
| 76 | 68 |
| 77 stream_reader_->RegisterCallback( | 69 stream_reader_->RegisterCallback( |
| 78 base::Bind(&DownloadFileImpl::StreamActive, weak_factory_.GetWeakPtr())); | 70 base::Bind(&DownloadFileImpl::StreamActive, weak_factory_.GetWeakPtr())); |
| 79 | 71 |
| 80 download_start_ = base::TimeTicks::Now(); | 72 download_start_ = base::TimeTicks::Now(); |
| 81 | |
| 82 // Initial pull from the straw. | 73 // Initial pull from the straw. |
| 83 StreamActive(); | 74 StreamActive(); |
| 84 | 75 |
| 85 BrowserThread::PostTask( | 76 return content::DOWNLOAD_INTERRUPT_REASON_NONE; |
| 86 BrowserThread::UI, FROM_HERE, base::Bind( | |
| 87 callback, content::DOWNLOAD_INTERRUPT_REASON_NONE)); | |
| 88 | |
| 89 ++number_active_objects_; | |
| 90 } | 77 } |
| 91 | 78 |
| 92 content::DownloadInterruptReason DownloadFileImpl::AppendDataToFile( | 79 content::DownloadInterruptReason DownloadFileImpl::AppendDataToFile( |
| 93 const char* data, size_t data_len) { | 80 const char* data, size_t data_len) { |
| 94 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); | |
| 95 | |
| 96 if (!update_timer_->IsRunning()) { | 81 if (!update_timer_->IsRunning()) { |
| 97 update_timer_->Start(FROM_HERE, | 82 update_timer_->Start(FROM_HERE, |
| 98 base::TimeDelta::FromMilliseconds(kUpdatePeriodMs), | 83 base::TimeDelta::FromMilliseconds(kUpdatePeriodMs), |
| 99 this, &DownloadFileImpl::SendUpdate); | 84 this, &DownloadFileImpl::SendUpdate); |
| 100 } | 85 } |
| 101 return content::ConvertNetErrorToInterruptReason( | 86 return content::ConvertNetErrorToInterruptReason( |
| 102 file_.AppendDataToFile(data, data_len), | 87 file_.AppendDataToFile(data, data_len), |
| 103 content::DOWNLOAD_INTERRUPT_FROM_DISK); | 88 content::DOWNLOAD_INTERRUPT_FROM_DISK); |
| 104 } | 89 } |
| 105 | 90 |
| 106 void DownloadFileImpl::Rename(const FilePath& full_path, | 91 void DownloadFileImpl::Rename(const FilePath& full_path, |
| 107 bool overwrite_existing_file, | 92 bool overwrite_existing_file, |
| 108 const RenameCompletionCallback& callback) { | 93 const RenameCompletionCallback& callback) { |
| 109 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); | |
| 110 | |
| 111 FilePath new_path(full_path); | 94 FilePath new_path(full_path); |
| 112 if (!overwrite_existing_file) { | 95 if (!overwrite_existing_file) { |
| 113 // Make the file unique if requested. | 96 // Make the file unique if requested. |
| 114 int uniquifier = | 97 int uniquifier = |
| 115 file_util::GetUniquePathNumber(new_path, FILE_PATH_LITERAL("")); | 98 file_util::GetUniquePathNumber(new_path, FILE_PATH_LITERAL("")); |
| 116 if (uniquifier > 0) { | 99 if (uniquifier > 0) { |
| 117 new_path = new_path.InsertBeforeExtensionASCII( | 100 new_path = new_path.InsertBeforeExtensionASCII( |
| 118 StringPrintf(" (%d)", uniquifier)); | 101 StringPrintf(" (%d)", uniquifier)); |
| 119 } | 102 } |
| 120 } | 103 } |
| (...skipping 15 matching lines...) Expand all Loading... |
| 136 content::DOWNLOAD_INTERRUPT_FROM_DISK); | 119 content::DOWNLOAD_INTERRUPT_FROM_DISK); |
| 137 new_path.clear(); | 120 new_path.clear(); |
| 138 } | 121 } |
| 139 | 122 |
| 140 BrowserThread::PostTask( | 123 BrowserThread::PostTask( |
| 141 BrowserThread::UI, FROM_HERE, | 124 BrowserThread::UI, FROM_HERE, |
| 142 base::Bind(callback, reason, new_path)); | 125 base::Bind(callback, reason, new_path)); |
| 143 } | 126 } |
| 144 | 127 |
| 145 void DownloadFileImpl::Detach() { | 128 void DownloadFileImpl::Detach() { |
| 146 // Done here on Windows so that anti-virus scanners invoked by | |
| 147 // the attachment service actually see the data; see | |
| 148 // http://crbug.com/127999. | |
| 149 // Done here for mac because we only want to do this once; see | |
| 150 // http://crbug.com/13120 for details. | |
| 151 // Other platforms don't currently do source annotation. | |
| 152 AnnotateWithSourceInformation(); | |
| 153 | |
| 154 file_.Detach(); | 129 file_.Detach(); |
| 155 } | 130 } |
| 156 | 131 |
| 157 void DownloadFileImpl::Cancel() { | 132 void DownloadFileImpl::Cancel() { |
| 158 file_.Cancel(); | 133 file_.Cancel(); |
| 159 } | 134 } |
| 160 | 135 |
| 161 void DownloadFileImpl::AnnotateWithSourceInformation() { | 136 void DownloadFileImpl::AnnotateWithSourceInformation() { |
| 162 bound_net_log_.BeginEvent(net::NetLog::TYPE_DOWNLOAD_FILE_ANNOTATED); | 137 bound_net_log_.BeginEvent(net::NetLog::TYPE_DOWNLOAD_FILE_ANNOTATED); |
| 163 file_.AnnotateWithSourceInformation(); | 138 file_.AnnotateWithSourceInformation(); |
| (...skipping 17 matching lines...) Expand all Loading... |
| 181 } | 156 } |
| 182 | 157 |
| 183 bool DownloadFileImpl::GetHash(std::string* hash) { | 158 bool DownloadFileImpl::GetHash(std::string* hash) { |
| 184 return file_.GetHash(hash); | 159 return file_.GetHash(hash); |
| 185 } | 160 } |
| 186 | 161 |
| 187 std::string DownloadFileImpl::GetHashState() { | 162 std::string DownloadFileImpl::GetHashState() { |
| 188 return file_.GetHashState(); | 163 return file_.GetHashState(); |
| 189 } | 164 } |
| 190 | 165 |
| 166 // DownloadFileInterface implementation. |
| 167 void DownloadFileImpl::CancelDownloadRequest() { |
| 168 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); |
| 169 request_handle_->CancelRequest(); |
| 170 } |
| 171 |
| 172 int DownloadFileImpl::Id() const { |
| 173 return id_.local(); |
| 174 } |
| 175 |
| 176 DownloadManager* DownloadFileImpl::GetDownloadManager() { |
| 177 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); |
| 178 return download_manager_.get(); |
| 179 } |
| 180 |
| 181 const DownloadId& DownloadFileImpl::GlobalId() const { |
| 182 return id_; |
| 183 } |
| 184 |
| 185 std::string DownloadFileImpl::DebugString() const { |
| 186 return base::StringPrintf("{" |
| 187 " id_ = " "%d" |
| 188 " request_handle = %s" |
| 189 " Base File = %s" |
| 190 " }", |
| 191 id_.local(), |
| 192 request_handle_->DebugString().c_str(), |
| 193 file_.DebugString().c_str()); |
| 194 } |
| 195 |
| 191 void DownloadFileImpl::StreamActive() { | 196 void DownloadFileImpl::StreamActive() { |
| 192 base::TimeTicks start(base::TimeTicks::Now()); | 197 base::TimeTicks start(base::TimeTicks::Now()); |
| 193 base::TimeTicks now; | 198 base::TimeTicks now; |
| 194 scoped_refptr<net::IOBuffer> incoming_data; | 199 scoped_refptr<net::IOBuffer> incoming_data; |
| 195 size_t incoming_data_size = 0; | 200 size_t incoming_data_size = 0; |
| 196 size_t total_incoming_data_size = 0; | 201 size_t total_incoming_data_size = 0; |
| 197 size_t num_buffers = 0; | 202 size_t num_buffers = 0; |
| 198 content::ByteStreamReader::StreamState state( | 203 content::ByteStreamReader::StreamState state( |
| 199 content::ByteStreamReader::STREAM_EMPTY); | 204 content::ByteStreamReader::STREAM_EMPTY); |
| 200 content::DownloadInterruptReason reason = | 205 content::DownloadInterruptReason reason = |
| (...skipping 48 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 249 BrowserThread::FILE, FROM_HERE, | 254 BrowserThread::FILE, FROM_HERE, |
| 250 base::Bind(&DownloadFileImpl::StreamActive, | 255 base::Bind(&DownloadFileImpl::StreamActive, |
| 251 weak_factory_.GetWeakPtr())); | 256 weak_factory_.GetWeakPtr())); |
| 252 } | 257 } |
| 253 | 258 |
| 254 if (total_incoming_data_size) | 259 if (total_incoming_data_size) |
| 255 download_stats::RecordFileThreadReceiveBuffers(num_buffers); | 260 download_stats::RecordFileThreadReceiveBuffers(num_buffers); |
| 256 | 261 |
| 257 download_stats::RecordContiguousWriteTime(now - start); | 262 download_stats::RecordContiguousWriteTime(now - start); |
| 258 | 263 |
| 259 // Take care of communication with our observer. | 264 // Take care of communication with our controller. |
| 260 if (reason != content::DOWNLOAD_INTERRUPT_REASON_NONE) { | 265 if (reason != content::DOWNLOAD_INTERRUPT_REASON_NONE) { |
| 261 // Error case for both upstream source and file write. | 266 // Error case for both upstream source and file write. |
| 262 // Shut down processing and signal an error to our observer. | 267 // Shut down processing and signal an error to our controller. |
| 263 // Our observer will clean us up. | 268 // Our controller will clean us up. |
| 264 stream_reader_->RegisterCallback(base::Closure()); | 269 stream_reader_->RegisterCallback(base::Closure()); |
| 265 weak_factory_.InvalidateWeakPtrs(); | 270 weak_factory_.InvalidateWeakPtrs(); |
| 266 SendUpdate(); // Make info up to date before error. | 271 SendUpdate(); // Make info up to date before error. |
| 267 BrowserThread::PostTask( | 272 BrowserThread::PostTask( |
| 268 BrowserThread::UI, FROM_HERE, | 273 BrowserThread::UI, FROM_HERE, |
| 269 base::Bind(&content::DownloadDestinationObserver::DestinationError, | 274 base::Bind(&DownloadManager::OnDownloadInterrupted, |
| 270 observer_, reason)); | 275 download_manager_, id_.local(), reason)); |
| 271 } else if (state == content::ByteStreamReader::STREAM_COMPLETE) { | 276 } else if (state == content::ByteStreamReader::STREAM_COMPLETE) { |
| 272 // Signal successful completion and shut down processing. | 277 // Signal successful completion and shut down processing. |
| 273 stream_reader_->RegisterCallback(base::Closure()); | 278 stream_reader_->RegisterCallback(base::Closure()); |
| 274 weak_factory_.InvalidateWeakPtrs(); | 279 weak_factory_.InvalidateWeakPtrs(); |
| 275 std::string hash; | 280 std::string hash; |
| 276 if (!GetHash(&hash) || file_.IsEmptyHash(hash)) | 281 if (!GetHash(&hash) || file_.IsEmptyHash(hash)) |
| 277 hash.clear(); | 282 hash.clear(); |
| 278 SendUpdate(); | |
| 279 BrowserThread::PostTask( | 283 BrowserThread::PostTask( |
| 280 BrowserThread::UI, FROM_HERE, | 284 BrowserThread::UI, FROM_HERE, |
| 281 base::Bind( | 285 base::Bind(&DownloadManager::OnResponseCompleted, |
| 282 &content::DownloadDestinationObserver::DestinationCompleted, | 286 download_manager_, id_.local(), |
| 283 observer_, hash)); | 287 BytesSoFar(), hash)); |
| 284 } | 288 } |
| 285 if (bound_net_log_.IsLoggingAllEvents()) { | 289 if (bound_net_log_.IsLoggingAllEvents()) { |
| 286 bound_net_log_.AddEvent( | 290 bound_net_log_.AddEvent( |
| 287 net::NetLog::TYPE_DOWNLOAD_STREAM_DRAINED, | 291 net::NetLog::TYPE_DOWNLOAD_STREAM_DRAINED, |
| 288 base::Bind(&download_net_logs::FileStreamDrainedCallback, | 292 base::Bind(&download_net_logs::FileStreamDrainedCallback, |
| 289 total_incoming_data_size, num_buffers)); | 293 total_incoming_data_size, num_buffers)); |
| 290 } | 294 } |
| 291 } | 295 } |
| 292 | 296 |
| 293 void DownloadFileImpl::SendUpdate() { | 297 void DownloadFileImpl::SendUpdate() { |
| 294 BrowserThread::PostTask( | 298 BrowserThread::PostTask( |
| 295 BrowserThread::UI, FROM_HERE, | 299 BrowserThread::UI, FROM_HERE, |
| 296 base::Bind(&content::DownloadDestinationObserver::DestinationUpdate, | 300 base::Bind(&DownloadManager::UpdateDownload, |
| 297 observer_, BytesSoFar(), CurrentSpeed(), GetHashState())); | 301 download_manager_, id_.local(), |
| 302 BytesSoFar(), CurrentSpeed(), GetHashState())); |
| 298 } | 303 } |
| 299 | |
| 300 // static | |
| 301 int content::DownloadFile::GetNumberOfDownloadFiles() { | |
| 302 return number_active_objects_; | |
| 303 } | |
| 304 | |
| OLD | NEW |