| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 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 | 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/save_package.h" | 5 #include "content/browser/download/save_package.h" |
| 6 | 6 |
| 7 #include <algorithm> | 7 #include <algorithm> |
| 8 | 8 |
| 9 #include "base/bind.h" | 9 #include "base/bind.h" |
| 10 #include "base/file_path.h" | 10 #include "base/file_path.h" |
| 11 #include "base/file_util.h" | 11 #include "base/file_util.h" |
| 12 #include "base/i18n/file_util_icu.h" | 12 #include "base/i18n/file_util_icu.h" |
| 13 #include "base/logging.h" | 13 #include "base/logging.h" |
| 14 #include "base/message_loop.h" | 14 #include "base/message_loop.h" |
| 15 #include "base/stl_util.h" | 15 #include "base/stl_util.h" |
| 16 #include "base/string_piece.h" | 16 #include "base/string_piece.h" |
| 17 #include "base/string_split.h" | 17 #include "base/string_split.h" |
| 18 #include "base/sys_string_conversions.h" | 18 #include "base/sys_string_conversions.h" |
| 19 #include "base/threading/thread.h" | 19 #include "base/threading/thread.h" |
| 20 #include "base/utf_string_conversions.h" | 20 #include "base/utf_string_conversions.h" |
| 21 #include "content/browser/browser_context.h" | 21 #include "content/browser/browser_context.h" |
| 22 #include "content/browser/download/download_file_manager.h" | 22 #include "content/browser/download/download_file_manager.h" |
| 23 #include "content/browser/download/download_item.h" | 23 #include "content/browser/download/download_item_impl.h" |
| 24 #include "content/browser/download/download_manager.h" | 24 #include "content/browser/download/download_manager.h" |
| 25 #include "content/browser/download/save_file.h" | 25 #include "content/browser/download/save_file.h" |
| 26 #include "content/browser/download/save_file_manager.h" | 26 #include "content/browser/download/save_file_manager.h" |
| 27 #include "content/browser/download/save_item.h" | 27 #include "content/browser/download/save_item.h" |
| 28 #include "content/browser/renderer_host/render_process_host_impl.h" | 28 #include "content/browser/renderer_host/render_process_host_impl.h" |
| 29 #include "content/browser/renderer_host/render_view_host.h" | 29 #include "content/browser/renderer_host/render_view_host.h" |
| 30 #include "content/browser/renderer_host/render_view_host_delegate.h" | 30 #include "content/browser/renderer_host/render_view_host_delegate.h" |
| 31 #include "content/browser/renderer_host/resource_dispatcher_host.h" | 31 #include "content/browser/renderer_host/resource_dispatcher_host.h" |
| 32 #include "content/browser/resource_context.h" | 32 #include "content/browser/resource_context.h" |
| 33 #include "content/browser/tab_contents/tab_contents.h" | 33 #include "content/browser/tab_contents/tab_contents.h" |
| (...skipping 225 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 259 wait_state_ = START_PROCESS; | 259 wait_state_ = START_PROCESS; |
| 260 | 260 |
| 261 // Initialize the request context and resource dispatcher. | 261 // Initialize the request context and resource dispatcher. |
| 262 content::BrowserContext* browser_context = tab_contents()->browser_context(); | 262 content::BrowserContext* browser_context = tab_contents()->browser_context(); |
| 263 if (!browser_context) { | 263 if (!browser_context) { |
| 264 NOTREACHED(); | 264 NOTREACHED(); |
| 265 return false; | 265 return false; |
| 266 } | 266 } |
| 267 | 267 |
| 268 // Create the download item, and add ourself as an observer. | 268 // Create the download item, and add ourself as an observer. |
| 269 download_ = new DownloadItem(download_manager_, | 269 download_ = new DownloadItemImpl(download_manager_, |
| 270 saved_main_file_path_, | 270 saved_main_file_path_, |
| 271 page_url_, | 271 page_url_, |
| 272 browser_context->IsOffTheRecord(), | 272 browser_context->IsOffTheRecord(), |
| 273 download_manager_->GetNextId()); | 273 download_manager_->GetNextId()); |
| 274 download_->AddObserver(this); | 274 download_->AddObserver(this); |
| 275 | 275 |
| 276 // Transfer ownership to the download manager. | 276 // Transfer ownership to the download manager. |
| 277 download_manager_->SavePageDownloadStarted(download_); | 277 download_manager_->SavePageDownloadStarted(download_); |
| 278 | 278 |
| 279 // Check save type and process the save page job. | 279 // Check save type and process the save page job. |
| 280 if (save_type_ == SAVE_AS_COMPLETE_HTML) { | 280 if (save_type_ == SAVE_AS_COMPLETE_HTML) { |
| 281 // Get directory | 281 // Get directory |
| 282 DCHECK(!saved_main_directory_path_.empty()); | 282 DCHECK(!saved_main_directory_path_.empty()); |
| 283 GetAllSavableResourceLinksForCurrentPage(); | 283 GetAllSavableResourceLinksForCurrentPage(); |
| 284 } else { | 284 } else { |
| 285 wait_state_ = NET_FILES; | 285 wait_state_ = NET_FILES; |
| 286 SaveFileCreateInfo::SaveFileSource save_source = page_url_.SchemeIsFile() ? | 286 SaveFileCreateInfo::SaveFileSource save_source = page_url_.SchemeIsFile() ? |
| 287 SaveFileCreateInfo::SAVE_FILE_FROM_FILE : | 287 SaveFileCreateInfo::SAVE_FILE_FROM_FILE : |
| 288 SaveFileCreateInfo::SAVE_FILE_FROM_NET; | 288 SaveFileCreateInfo::SAVE_FILE_FROM_NET; |
| 289 SaveItem* save_item = new SaveItem(page_url_, | 289 SaveItem* save_item = new SaveItem(page_url_, |
| 290 GURL(), | 290 GURL(), |
| 291 this, | 291 this, |
| 292 save_source); | 292 save_source); |
| 293 // Add this item to waiting list. | 293 // Add this item to waiting list. |
| 294 waiting_item_queue_.push(save_item); | 294 waiting_item_queue_.push(save_item); |
| 295 all_save_items_count_ = 1; | 295 all_save_items_count_ = 1; |
| 296 download_->set_total_bytes(1); | 296 download_->SetTotalBytes(1); |
| 297 | 297 |
| 298 DoSavingProcess(); | 298 DoSavingProcess(); |
| 299 } | 299 } |
| 300 | 300 |
| 301 return true; | 301 return true; |
| 302 } | 302 } |
| 303 | 303 |
| 304 // On POSIX, the length of |pure_file_name| + |file_name_ext| is further | 304 // On POSIX, the length of |pure_file_name| + |file_name_ext| is further |
| 305 // restricted by NAME_MAX. The maximum allowed path looks like: | 305 // restricted by NAME_MAX. The maximum allowed path looks like: |
| 306 // '/path/to/save_dir' + '/' + NAME_MAX. | 306 // '/path/to/save_dir' + '/' + NAME_MAX. |
| (...skipping 700 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 1007 const std::vector<GURL>& frames_list) { | 1007 const std::vector<GURL>& frames_list) { |
| 1008 if (wait_state_ != RESOURCES_LIST) | 1008 if (wait_state_ != RESOURCES_LIST) |
| 1009 return; | 1009 return; |
| 1010 | 1010 |
| 1011 DCHECK(resources_list.size() == referrers_list.size()); | 1011 DCHECK(resources_list.size() == referrers_list.size()); |
| 1012 all_save_items_count_ = static_cast<int>(resources_list.size()) + | 1012 all_save_items_count_ = static_cast<int>(resources_list.size()) + |
| 1013 static_cast<int>(frames_list.size()); | 1013 static_cast<int>(frames_list.size()); |
| 1014 | 1014 |
| 1015 // We use total bytes as the total number of files we want to save. | 1015 // We use total bytes as the total number of files we want to save. |
| 1016 if (download_) | 1016 if (download_) |
| 1017 download_->set_total_bytes(all_save_items_count_); | 1017 download_->SetTotalBytes(all_save_items_count_); |
| 1018 | 1018 |
| 1019 if (all_save_items_count_) { | 1019 if (all_save_items_count_) { |
| 1020 // Put all sub-resources to wait list. | 1020 // Put all sub-resources to wait list. |
| 1021 for (int i = 0; i < static_cast<int>(resources_list.size()); ++i) { | 1021 for (int i = 0; i < static_cast<int>(resources_list.size()); ++i) { |
| 1022 const GURL& u = resources_list[i]; | 1022 const GURL& u = resources_list[i]; |
| 1023 DCHECK(u.is_valid()); | 1023 DCHECK(u.is_valid()); |
| 1024 SaveFileCreateInfo::SaveFileSource save_source = u.SchemeIsFile() ? | 1024 SaveFileCreateInfo::SaveFileSource save_source = u.SchemeIsFile() ? |
| 1025 SaveFileCreateInfo::SAVE_FILE_FROM_FILE : | 1025 SaveFileCreateInfo::SAVE_FILE_FROM_FILE : |
| 1026 SaveFileCreateInfo::SAVE_FILE_FROM_NET; | 1026 SaveFileCreateInfo::SAVE_FILE_FROM_NET; |
| 1027 SaveItem* save_item = new SaveItem(u, referrers_list[i], | 1027 SaveItem* save_item = new SaveItem(u, referrers_list[i], |
| (...skipping 242 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 1270 download_ = NULL; | 1270 download_ = NULL; |
| 1271 download_manager_ = NULL; | 1271 download_manager_ = NULL; |
| 1272 } | 1272 } |
| 1273 | 1273 |
| 1274 void SavePackage::OnDownloadUpdated(DownloadItem* download) { | 1274 void SavePackage::OnDownloadUpdated(DownloadItem* download) { |
| 1275 DCHECK(download_); | 1275 DCHECK(download_); |
| 1276 DCHECK(download_ == download); | 1276 DCHECK(download_ == download); |
| 1277 DCHECK(download_manager_); | 1277 DCHECK(download_manager_); |
| 1278 | 1278 |
| 1279 // Check for removal. | 1279 // Check for removal. |
| 1280 if (download->state() == DownloadItem::REMOVING) | 1280 if (download->GetState() == DownloadItem::REMOVING) |
| 1281 StopObservation(); | 1281 StopObservation(); |
| 1282 } | 1282 } |
| 1283 | 1283 |
| 1284 void SavePackage::FinalizeDownloadEntry() { | 1284 void SavePackage::FinalizeDownloadEntry() { |
| 1285 DCHECK(download_); | 1285 DCHECK(download_); |
| 1286 DCHECK(download_manager_); | 1286 DCHECK(download_manager_); |
| 1287 | 1287 |
| 1288 download_manager_->SavePageDownloadFinished(download_); | 1288 download_manager_->SavePageDownloadFinished(download_); |
| 1289 StopObservation(); | 1289 StopObservation(); |
| 1290 } | 1290 } |
| OLD | NEW |