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.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.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" |
34 #include "content/common/view_messages.h" | 34 #include "content/common/view_messages.h" |
35 #include "content/public/browser/browser_thread.h" | 35 #include "content/public/browser/browser_thread.h" |
36 #include "content/public/browser/content_browser_client.h" | 36 #include "content/public/browser/content_browser_client.h" |
37 #include "content/public/browser/download_manager_delegate.h" | 37 #include "content/public/browser/download_manager_delegate.h" |
38 #include "content/public/common/url_constants.h" | 38 #include "content/public/common/url_constants.h" |
(...skipping 83 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
122 page_url_(GetUrlToBeSaved()), | 122 page_url_(GetUrlToBeSaved()), |
123 saved_main_file_path_(file_full_path), | 123 saved_main_file_path_(file_full_path), |
124 saved_main_directory_path_(directory_full_path), | 124 saved_main_directory_path_(directory_full_path), |
125 title_(tab_contents->GetTitle()), | 125 title_(tab_contents->GetTitle()), |
126 finished_(false), | 126 finished_(false), |
127 user_canceled_(false), | 127 user_canceled_(false), |
128 disk_error_occurred_(false), | 128 disk_error_occurred_(false), |
129 save_type_(save_type), | 129 save_type_(save_type), |
130 all_save_items_count_(0), | 130 all_save_items_count_(0), |
131 wait_state_(INITIALIZE), | 131 wait_state_(INITIALIZE), |
132 tab_id_(tab_contents->GetRenderProcessHost()->id()), | 132 tab_id_(tab_contents->GetRenderProcessHost()->GetID()), |
133 unique_id_(g_save_package_id++), | 133 unique_id_(g_save_package_id++), |
134 ALLOW_THIS_IN_INITIALIZER_LIST(method_factory_(this)) { | 134 ALLOW_THIS_IN_INITIALIZER_LIST(method_factory_(this)) { |
135 DCHECK(page_url_.is_valid()); | 135 DCHECK(page_url_.is_valid()); |
136 DCHECK(save_type_ == SAVE_AS_ONLY_HTML || | 136 DCHECK(save_type_ == SAVE_AS_ONLY_HTML || |
137 save_type_ == SAVE_AS_COMPLETE_HTML); | 137 save_type_ == SAVE_AS_COMPLETE_HTML); |
138 DCHECK(!saved_main_file_path_.empty() && | 138 DCHECK(!saved_main_file_path_.empty() && |
139 saved_main_file_path_.value().length() <= kMaxFilePathLength); | 139 saved_main_file_path_.value().length() <= kMaxFilePathLength); |
140 DCHECK(!saved_main_directory_path_.empty() && | 140 DCHECK(!saved_main_directory_path_.empty() && |
141 saved_main_directory_path_.value().length() < kMaxFilePathLength); | 141 saved_main_directory_path_.value().length() < kMaxFilePathLength); |
142 InternalInit(); | 142 InternalInit(); |
143 } | 143 } |
144 | 144 |
145 SavePackage::SavePackage(TabContents* tab_contents) | 145 SavePackage::SavePackage(TabContents* tab_contents) |
146 : TabContentsObserver(tab_contents), | 146 : TabContentsObserver(tab_contents), |
147 file_manager_(NULL), | 147 file_manager_(NULL), |
148 download_manager_(NULL), | 148 download_manager_(NULL), |
149 download_(NULL), | 149 download_(NULL), |
150 page_url_(GetUrlToBeSaved()), | 150 page_url_(GetUrlToBeSaved()), |
151 title_(tab_contents->GetTitle()), | 151 title_(tab_contents->GetTitle()), |
152 finished_(false), | 152 finished_(false), |
153 user_canceled_(false), | 153 user_canceled_(false), |
154 disk_error_occurred_(false), | 154 disk_error_occurred_(false), |
155 save_type_(SAVE_TYPE_UNKNOWN), | 155 save_type_(SAVE_TYPE_UNKNOWN), |
156 all_save_items_count_(0), | 156 all_save_items_count_(0), |
157 wait_state_(INITIALIZE), | 157 wait_state_(INITIALIZE), |
158 tab_id_(tab_contents->GetRenderProcessHost()->id()), | 158 tab_id_(tab_contents->GetRenderProcessHost()->GetID()), |
159 unique_id_(g_save_package_id++), | 159 unique_id_(g_save_package_id++), |
160 ALLOW_THIS_IN_INITIALIZER_LIST(method_factory_(this)) { | 160 ALLOW_THIS_IN_INITIALIZER_LIST(method_factory_(this)) { |
161 DCHECK(page_url_.is_valid()); | 161 DCHECK(page_url_.is_valid()); |
162 InternalInit(); | 162 InternalInit(); |
163 } | 163 } |
164 | 164 |
165 // This is for testing use. Set |finished_| as true because we don't want | 165 // This is for testing use. Set |finished_| as true because we don't want |
166 // method Cancel to be be called in destructor in test mode. | 166 // method Cancel to be be called in destructor in test mode. |
167 // We also don't call InternalInit(). | 167 // We also don't call InternalInit(). |
168 SavePackage::SavePackage(TabContents* tab_contents, | 168 SavePackage::SavePackage(TabContents* tab_contents, |
(...skipping 477 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
646 it != saved_success_items_.end(); ++it) | 646 it != saved_success_items_.end(); ++it) |
647 final_names.push_back(std::make_pair(it->first, | 647 final_names.push_back(std::make_pair(it->first, |
648 it->second->full_path())); | 648 it->second->full_path())); |
649 | 649 |
650 BrowserThread::PostTask( | 650 BrowserThread::PostTask( |
651 BrowserThread::FILE, FROM_HERE, | 651 BrowserThread::FILE, FROM_HERE, |
652 base::Bind(&SaveFileManager::RenameAllFiles, | 652 base::Bind(&SaveFileManager::RenameAllFiles, |
653 file_manager_, | 653 file_manager_, |
654 final_names, | 654 final_names, |
655 dir, | 655 dir, |
656 tab_contents()->GetRenderProcessHost()->id(), | 656 tab_contents()->GetRenderProcessHost()->GetID(), |
657 tab_contents()->render_view_host()->routing_id(), | 657 tab_contents()->render_view_host()->routing_id(), |
658 id())); | 658 id())); |
659 } | 659 } |
660 | 660 |
661 // Successfully finished all items of this SavePackage. | 661 // Successfully finished all items of this SavePackage. |
662 void SavePackage::Finish() { | 662 void SavePackage::Finish() { |
663 // User may cancel the job when we're moving files to the final directory. | 663 // User may cancel the job when we're moving files to the final directory. |
664 if (canceled()) | 664 if (canceled()) |
665 return; | 665 return; |
666 | 666 |
(...skipping 127 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
794 waiting_item_queue_.pop(); | 794 waiting_item_queue_.pop(); |
795 | 795 |
796 // Add the item to in_progress_items_. | 796 // Add the item to in_progress_items_. |
797 SaveUrlItemMap::iterator it = in_progress_items_.find( | 797 SaveUrlItemMap::iterator it = in_progress_items_.find( |
798 save_item->url().spec()); | 798 save_item->url().spec()); |
799 DCHECK(it == in_progress_items_.end()); | 799 DCHECK(it == in_progress_items_.end()); |
800 in_progress_items_[save_item->url().spec()] = save_item; | 800 in_progress_items_[save_item->url().spec()] = save_item; |
801 save_item->Start(); | 801 save_item->Start(); |
802 file_manager_->SaveURL(save_item->url(), | 802 file_manager_->SaveURL(save_item->url(), |
803 save_item->referrer(), | 803 save_item->referrer(), |
804 tab_contents()->GetRenderProcessHost()->id(), | 804 tab_contents()->GetRenderProcessHost()->GetID(), |
805 routing_id(), | 805 routing_id(), |
806 save_item->save_source(), | 806 save_item->save_source(), |
807 save_item->full_path(), | 807 save_item->full_path(), |
808 tab_contents()-> | 808 tab_contents()-> |
809 browser_context()->GetResourceContext(), | 809 browser_context()->GetResourceContext(), |
810 this); | 810 this); |
811 } while (process_all_remaining_items && waiting_item_queue_.size()); | 811 } while (process_all_remaining_items && waiting_item_queue_.size()); |
812 } | 812 } |
813 | 813 |
814 // Calculate the percentage of whole save page job. | 814 // Calculate the percentage of whole save page job. |
(...skipping 465 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
1280 StopObservation(); | 1280 StopObservation(); |
1281 } | 1281 } |
1282 | 1282 |
1283 void SavePackage::FinalizeDownloadEntry() { | 1283 void SavePackage::FinalizeDownloadEntry() { |
1284 DCHECK(download_); | 1284 DCHECK(download_); |
1285 DCHECK(download_manager_); | 1285 DCHECK(download_manager_); |
1286 | 1286 |
1287 download_manager_->SavePageDownloadFinished(download_); | 1287 download_manager_->SavePageDownloadFinished(download_); |
1288 StopObservation(); | 1288 StopObservation(); |
1289 } | 1289 } |
OLD | NEW |