Chromium Code Reviews| 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 "chrome/browser/download/download_manager.h" | 5 #include "chrome/browser/download/download_manager.h" |
| 6 | 6 |
| 7 #include "base/callback.h" | 7 #include "base/callback.h" |
| 8 #include "base/file_util.h" | 8 #include "base/file_util.h" |
| 9 #include "base/logging.h" | 9 #include "base/logging.h" |
| 10 #include "base/path_service.h" | 10 #include "base/path_service.h" |
| 11 #include "base/rand_util.h" | 11 #include "base/rand_util.h" |
| 12 #include "base/stl_util-inl.h" | 12 #include "base/stl_util-inl.h" |
| 13 #include "base/stringprintf.h" | 13 #include "base/stringprintf.h" |
| 14 #include "base/sys_string_conversions.h" | 14 #include "base/sys_string_conversions.h" |
| 15 #include "base/task.h" | 15 #include "base/task.h" |
| 16 #include "base/utf_string_conversions.h" | 16 #include "base/utf_string_conversions.h" |
| 17 #include "build/build_config.h" | 17 #include "build/build_config.h" |
| 18 #include "chrome/browser/browser_process.h" | 18 #include "chrome/browser/browser_process.h" |
| 19 #include "chrome/browser/download/download_extensions.h" | 19 #include "chrome/browser/download/download_extensions.h" |
| 20 #include "chrome/browser/download/download_file_manager.h" | 20 #include "chrome/browser/download/download_file_manager.h" |
| 21 #include "chrome/browser/download/download_history.h" | 21 #include "chrome/browser/download/download_history.h" |
| 22 #include "chrome/browser/download/download_item.h" | 22 #include "chrome/browser/download/download_item.h" |
| 23 #include "chrome/browser/download/download_prefs.h" | 23 #include "chrome/browser/download/download_prefs.h" |
| 24 #include "chrome/browser/download/download_process_handle.h" | |
| 24 #include "chrome/browser/download/download_safe_browsing_client.h" | 25 #include "chrome/browser/download/download_safe_browsing_client.h" |
| 25 #include "chrome/browser/download/download_status_updater.h" | 26 #include "chrome/browser/download/download_status_updater.h" |
| 26 #include "chrome/browser/download/download_util.h" | 27 #include "chrome/browser/download/download_util.h" |
| 27 #include "chrome/browser/extensions/extension_service.h" | 28 #include "chrome/browser/extensions/extension_service.h" |
| 28 #include "chrome/browser/history/download_create_info.h" | 29 #include "chrome/browser/history/download_create_info.h" |
| 29 #include "chrome/browser/net/chrome_url_request_context.h" | 30 #include "chrome/browser/net/chrome_url_request_context.h" |
| 30 #include "chrome/browser/platform_util.h" | 31 #include "chrome/browser/platform_util.h" |
| 31 #include "chrome/browser/profiles/profile.h" | 32 #include "chrome/browser/profiles/profile.h" |
| 32 #include "chrome/browser/tab_contents/tab_util.h" | 33 #include "chrome/browser/tab_contents/tab_util.h" |
| 33 #include "chrome/browser/ui/browser.h" | 34 #include "chrome/browser/ui/browser.h" |
| (...skipping 391 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 425 void DownloadManager::OnPathExistenceAvailable(DownloadCreateInfo* info) { | 426 void DownloadManager::OnPathExistenceAvailable(DownloadCreateInfo* info) { |
| 426 VLOG(20) << __FUNCTION__ << "()" << " info = " << info->DebugString(); | 427 VLOG(20) << __FUNCTION__ << "()" << " info = " << info->DebugString(); |
| 427 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | 428 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
| 428 DCHECK(info); | 429 DCHECK(info); |
| 429 | 430 |
| 430 if (info->prompt_user_for_save_location) { | 431 if (info->prompt_user_for_save_location) { |
| 431 // We must ask the user for the place to put the download. | 432 // We must ask the user for the place to put the download. |
| 432 if (!select_file_dialog_.get()) | 433 if (!select_file_dialog_.get()) |
| 433 select_file_dialog_ = SelectFileDialog::Create(this); | 434 select_file_dialog_ = SelectFileDialog::Create(this); |
| 434 | 435 |
| 435 TabContents* contents = tab_util::GetTabContentsByID(info->child_id, | 436 TabContents* contents = info->process_handle.GetTabContents(); |
| 436 info->render_view_id); | |
| 437 SelectFileDialog::FileTypeInfo file_type_info; | 437 SelectFileDialog::FileTypeInfo file_type_info; |
| 438 FilePath::StringType extension = info->suggested_path.Extension(); | 438 FilePath::StringType extension = info->suggested_path.Extension(); |
| 439 if (!extension.empty()) { | 439 if (!extension.empty()) { |
| 440 extension.erase(extension.begin()); // drop the . | 440 extension.erase(extension.begin()); // drop the . |
| 441 file_type_info.extensions.resize(1); | 441 file_type_info.extensions.resize(1); |
| 442 file_type_info.extensions[0].push_back(extension); | 442 file_type_info.extensions[0].push_back(extension); |
| 443 } | 443 } |
| 444 file_type_info.include_all_files = true; | 444 file_type_info.include_all_files = true; |
| 445 gfx::NativeWindow owning_window = | 445 gfx::NativeWindow owning_window = |
| 446 contents ? platform_util::GetTopLevel(contents->GetNativeView()) : NULL; | 446 contents ? platform_util::GetTopLevel(contents->GetNativeView()) : NULL; |
| (...skipping 254 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 701 active_downloads_.erase(download_id); | 701 active_downloads_.erase(download_id); |
| 702 UpdateAppIcon(); // Reflect removal from in_progress_. | 702 UpdateAppIcon(); // Reflect removal from in_progress_. |
| 703 download_history_->UpdateEntry(download); | 703 download_history_->UpdateEntry(download); |
| 704 } | 704 } |
| 705 | 705 |
| 706 DownloadCancelledInternal(download_id, | 706 DownloadCancelledInternal(download_id, |
| 707 download->render_process_id(), | 707 download->render_process_id(), |
| 708 download->request_id()); | 708 download->request_id()); |
| 709 } | 709 } |
| 710 | 710 |
| 711 void DownloadManager::DownloadCancelledInternal(int download_id, | 711 void DownloadManager::DownloadCancelledInternal(int download_id, |
|
Randy Smith (Not in Mondays)
2011/05/05 20:25:34
The places from which this is called both have a D
ahendrickson
2011/05/06 16:16:50
Done.
| |
| 712 int render_process_id, | 712 int render_process_id, |
| 713 int request_id) { | 713 int request_id) { |
| 714 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | 714 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
| 715 // Cancel the network request. RDH is guaranteed to outlive the IO thread. | 715 // Cancel the network request. RDH is guaranteed to outlive the IO thread. |
| 716 BrowserThread::PostTask( | 716 DownloadProcessHandle process_handle(-1, render_process_id, request_id); |
| 717 BrowserThread::IO, FROM_HERE, | 717 process_handle.CancelDownload( |
| 718 NewRunnableFunction(&download_util::CancelDownloadRequest, | 718 g_browser_process->resource_dispatcher_host()); |
| 719 g_browser_process->resource_dispatcher_host(), | |
| 720 render_process_id, | |
| 721 request_id)); | |
| 722 | 719 |
| 723 BrowserThread::PostTask( | 720 BrowserThread::PostTask( |
| 724 BrowserThread::FILE, FROM_HERE, | 721 BrowserThread::FILE, FROM_HERE, |
| 725 NewRunnableMethod( | 722 NewRunnableMethod( |
| 726 file_manager_, &DownloadFileManager::CancelDownload, download_id)); | 723 file_manager_, &DownloadFileManager::CancelDownload, download_id)); |
| 727 } | 724 } |
| 728 | 725 |
| 729 void DownloadManager::OnDownloadError(int32 download_id, | 726 void DownloadManager::OnDownloadError(int32 download_id, |
| 730 int64 size, | 727 int64 size, |
| 731 int os_error) { | 728 int os_error) { |
| (...skipping 244 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 976 | 973 |
| 977 info->path = path; | 974 info->path = path; |
| 978 AttachDownloadItem(info); | 975 AttachDownloadItem(info); |
| 979 } | 976 } |
| 980 | 977 |
| 981 void DownloadManager::FileSelectionCanceled(void* params) { | 978 void DownloadManager::FileSelectionCanceled(void* params) { |
| 982 // The user didn't pick a place to save the file, so need to cancel the | 979 // The user didn't pick a place to save the file, so need to cancel the |
| 983 // download that's already in progress to the temporary location. | 980 // download that's already in progress to the temporary location. |
| 984 DownloadCreateInfo* info = reinterpret_cast<DownloadCreateInfo*>(params); | 981 DownloadCreateInfo* info = reinterpret_cast<DownloadCreateInfo*>(params); |
| 985 DownloadCancelledInternal(info->download_id, | 982 DownloadCancelledInternal(info->download_id, |
| 986 info->child_id, | 983 info->process_handle.child_id(), |
| 987 info->request_id); | 984 info->process_handle.child_id()); |
| 988 } | 985 } |
| 989 | 986 |
| 990 void DownloadManager::DangerousDownloadValidated(DownloadItem* download) { | 987 void DownloadManager::DangerousDownloadValidated(DownloadItem* download) { |
| 991 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | 988 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
| 992 DCHECK_EQ(DownloadItem::DANGEROUS, download->safety_state()); | 989 DCHECK_EQ(DownloadItem::DANGEROUS, download->safety_state()); |
| 993 download->set_safety_state(DownloadItem::DANGEROUS_BUT_VALIDATED); | 990 download->set_safety_state(DownloadItem::DANGEROUS_BUT_VALIDATED); |
| 994 download->UpdateObservers(); | 991 download->UpdateObservers(); |
| 995 | 992 |
| 996 MaybeCompleteDownload(download); | 993 MaybeCompleteDownload(download); |
| 997 } | 994 } |
| (...skipping 39 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 1037 db_handle = download_history_->GetNextFakeDbHandle(); | 1034 db_handle = download_history_->GetNextFakeDbHandle(); |
| 1038 | 1035 |
| 1039 DCHECK(download->db_handle() == DownloadHistory::kUninitializedHandle); | 1036 DCHECK(download->db_handle() == DownloadHistory::kUninitializedHandle); |
| 1040 download->set_db_handle(db_handle); | 1037 download->set_db_handle(db_handle); |
| 1041 | 1038 |
| 1042 DCHECK(!ContainsKey(history_downloads_, download->db_handle())); | 1039 DCHECK(!ContainsKey(history_downloads_, download->db_handle())); |
| 1043 history_downloads_[download->db_handle()] = download; | 1040 history_downloads_[download->db_handle()] = download; |
| 1044 | 1041 |
| 1045 // Show in the appropriate browser UI. | 1042 // Show in the appropriate browser UI. |
| 1046 // This includes buttons to save or cancel, for a dangerous download. | 1043 // This includes buttons to save or cancel, for a dangerous download. |
| 1047 ShowDownloadInBrowser(info, download); | 1044 ShowDownloadInBrowser(info.process_handle, download); |
| 1048 | 1045 |
| 1049 // Inform interested objects about the new download. | 1046 // Inform interested objects about the new download. |
| 1050 NotifyModelChanged(); | 1047 NotifyModelChanged(); |
| 1051 | 1048 |
| 1052 // If the download is still in progress, try to complete it. | 1049 // If the download is still in progress, try to complete it. |
| 1053 // | 1050 // |
| 1054 // Otherwise, download has been cancelled or interrupted before we've | 1051 // Otherwise, download has been cancelled or interrupted before we've |
| 1055 // received the DB handle. We post one final message to the history | 1052 // received the DB handle. We post one final message to the history |
| 1056 // service so that it can be properly in sync with the DownloadItem's | 1053 // service so that it can be properly in sync with the DownloadItem's |
| 1057 // completion status, and also inform any observers so that they get | 1054 // completion status, and also inform any observers so that they get |
| 1058 // more than just the start notification. | 1055 // more than just the start notification. |
| 1059 if (download->IsInProgress()) { | 1056 if (download->IsInProgress()) { |
| 1060 MaybeCompleteDownload(download); | 1057 MaybeCompleteDownload(download); |
| 1061 } else { | 1058 } else { |
| 1062 DCHECK(download->IsCancelled()) | 1059 DCHECK(download->IsCancelled()) |
| 1063 << " download = " << download->DebugString(true); | 1060 << " download = " << download->DebugString(true); |
| 1064 in_progress_.erase(it); | 1061 in_progress_.erase(it); |
| 1065 active_downloads_.erase(info.download_id); | 1062 active_downloads_.erase(info.download_id); |
| 1066 download_history_->UpdateEntry(download); | 1063 download_history_->UpdateEntry(download); |
| 1067 download->UpdateObservers(); | 1064 download->UpdateObservers(); |
| 1068 } | 1065 } |
| 1069 } | 1066 } |
| 1070 | 1067 |
| 1071 void DownloadManager::ShowDownloadInBrowser(const DownloadCreateInfo& info, | 1068 void DownloadManager::ShowDownloadInBrowser( |
| 1072 DownloadItem* download) { | 1069 DownloadProcessHandle process_handle, DownloadItem* download) { |
| 1073 // The 'contents' may no longer exist if the user closed the tab before we | 1070 // The 'contents' may no longer exist if the user closed the tab before we |
| 1074 // get this start completion event. If it does, tell the origin TabContents | 1071 // get this start completion event. If it does, tell the origin TabContents |
| 1075 // to display its download shelf. | 1072 // to display its download shelf. |
| 1076 TabContents* contents = tab_util::GetTabContentsByID(info.child_id, | 1073 TabContents* contents = process_handle.GetTabContents(); |
| 1077 info.render_view_id); | |
| 1078 | 1074 |
| 1079 // If the contents no longer exists, we start the download in the last active | 1075 // If the contents no longer exists, we start the download in the last active |
| 1080 // browser. This is not ideal but better than fully hiding the download from | 1076 // browser. This is not ideal but better than fully hiding the download from |
| 1081 // the user. | 1077 // the user. |
| 1082 if (!contents) { | 1078 if (!contents) { |
| 1083 Browser* last_active = BrowserList::GetLastActive(); | 1079 Browser* last_active = BrowserList::GetLastActive(); |
| 1084 if (last_active) | 1080 if (last_active) |
| 1085 contents = last_active->GetSelectedTabContents(); | 1081 contents = last_active->GetSelectedTabContents(); |
| 1086 } | 1082 } |
| 1087 | 1083 |
| (...skipping 86 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 1174 observed_download_manager_->RemoveObserver(this); | 1170 observed_download_manager_->RemoveObserver(this); |
| 1175 } | 1171 } |
| 1176 | 1172 |
| 1177 void DownloadManager::OtherDownloadManagerObserver::ModelChanged() { | 1173 void DownloadManager::OtherDownloadManagerObserver::ModelChanged() { |
| 1178 observing_download_manager_->NotifyModelChanged(); | 1174 observing_download_manager_->NotifyModelChanged(); |
| 1179 } | 1175 } |
| 1180 | 1176 |
| 1181 void DownloadManager::OtherDownloadManagerObserver::ManagerGoingDown() { | 1177 void DownloadManager::OtherDownloadManagerObserver::ManagerGoingDown() { |
| 1182 observed_download_manager_ = NULL; | 1178 observed_download_manager_ = NULL; |
| 1183 } | 1179 } |
| OLD | NEW |