blob: ea740edef2e593388f6be492938b9f8ca3c9e11b [file] [log] [blame]
[email protected]21ca982c2010-01-26 22:49:551// Copyright (c) 2010 The Chromium Authors. All rights reserved.
license.botbf09a502008-08-24 00:55:552// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
initial.commit09911bf2008-07-26 23:55:294
[email protected]cdaa8652008-09-13 02:48:595#include "chrome/browser/download/download_manager.h"
initial.commit09911bf2008-07-26 23:55:296
[email protected]a92b8642009-05-05 23:38:567#include "app/l10n_util.h"
[email protected]21ca982c2010-01-26 22:49:558#include "app/resource_bundle.h"
[email protected]2041cf342010-02-19 03:15:599#include "base/callback.h"
initial.commit09911bf2008-07-26 23:55:2910#include "base/file_util.h"
11#include "base/logging.h"
initial.commit09911bf2008-07-26 23:55:2912#include "base/path_service.h"
[email protected]1b5044d2009-02-24 00:04:1413#include "base/rand_util.h"
[email protected]1b5044d2009-02-24 00:04:1414#include "base/sys_string_conversions.h"
initial.commit09911bf2008-07-26 23:55:2915#include "base/task.h"
[email protected]d2a8fb72010-01-21 05:31:4216#include "build/build_config.h"
[email protected]6c69796d2010-07-16 21:41:1617#include "chrome/browser/browser.h"
initial.commit09911bf2008-07-26 23:55:2918#include "chrome/browser/browser_list.h"
19#include "chrome/browser/browser_process.h"
[email protected]d83d03aa2009-11-02 21:44:3720#include "chrome/browser/chrome_thread.h"
[email protected]6c69796d2010-07-16 21:41:1621#include "chrome/browser/download/download_file_manager.h"
22#include "chrome/browser/download/download_item.h"
[email protected]e9ef0a62009-08-11 22:50:1323#include "chrome/browser/download/download_util.h"
[email protected]866930682009-08-18 22:53:4724#include "chrome/browser/extensions/crx_installer.h"
[email protected]2a464a92009-08-01 17:58:3525#include "chrome/browser/extensions/extension_install_ui.h"
[email protected]8f783752009-04-01 23:33:4526#include "chrome/browser/extensions/extensions_service.h"
[email protected]6c69796d2010-07-16 21:41:1627#include "chrome/browser/history/download_types.h"
[email protected]be180c802009-10-23 06:33:3128#include "chrome/browser/net/chrome_url_request_context.h"
[email protected]14a000d2010-04-29 21:44:2429#include "chrome/browser/platform_util.h"
initial.commit09911bf2008-07-26 23:55:2930#include "chrome/browser/profile.h"
[email protected]8c8657d62009-01-16 18:31:2631#include "chrome/browser/renderer_host/render_process_host.h"
[email protected]6524b5f92009-01-22 17:48:2532#include "chrome/browser/renderer_host/render_view_host.h"
[email protected]21ca982c2010-01-26 22:49:5533#include "chrome/browser/tab_contents/infobar_delegate.h"
[email protected]57c6a652009-05-04 07:58:3434#include "chrome/browser/tab_contents/tab_contents.h"
[email protected]d2a8fb72010-01-21 05:31:4235#include "chrome/browser/tab_contents/tab_util.h"
initial.commit09911bf2008-07-26 23:55:2936#include "chrome/common/chrome_paths.h"
[email protected]91e1bd82009-09-03 22:04:4037#include "chrome/common/notification_service.h"
38#include "chrome/common/notification_type.h"
initial.commit09911bf2008-07-26 23:55:2939#include "chrome/common/pref_names.h"
[email protected]46072d42008-07-28 14:49:3540#include "googleurl/src/gurl.h"
[email protected]34ac8f32009-02-22 23:03:2741#include "grit/generated_resources.h"
[email protected]21ca982c2010-01-26 22:49:5542#include "grit/theme_resources.h"
initial.commit09911bf2008-07-26 23:55:2943#include "net/base/mime_util.h"
44#include "net/base/net_util.h"
initial.commit09911bf2008-07-26 23:55:2945
[email protected]b7f05882009-02-22 01:21:5646#if defined(OS_WIN)
[email protected]4a0765a2009-05-08 23:12:2547#include "app/win_util.h"
[email protected]b7f05882009-02-22 01:21:5648#include "base/registry.h"
49#include "base/win_util.h"
[email protected]a0a9577b2009-05-27 23:52:3250#endif
51
[email protected]21ca982c2010-01-26 22:49:5552namespace {
53
[email protected]b0ab1d42010-02-24 19:29:2854// Used to sort download items based on descending start time.
55bool CompareStartTime(DownloadItem* first, DownloadItem* second) {
56 return first->start_time() > second->start_time();
57}
58
[email protected]da74e5ae22010-07-14 16:12:3759void DeleteDownloadedFile(const FilePath& path) {
60 DCHECK(ChromeThread::CurrentlyOn(ChromeThread::FILE));
61
62 // Make sure we only delete files.
63 if (!file_util::DirectoryExists(path))
64 file_util::Delete(path, false);
65}
66
[email protected]21ca982c2010-01-26 22:49:5567} // namespace
68
[email protected]6c69796d2010-07-16 21:41:1669// Our download table ID starts at 1, so we use 0 to represent a download that
70// has started, but has not yet had its data persisted in the table. We use fake
71// database handles in incognito mode starting at -1 and progressively getting
72// more negative.
73// static
74const int DownloadManager::kUninitializedHandle = 0;
initial.commit09911bf2008-07-26 23:55:2975
76// static
77void DownloadManager::RegisterUserPrefs(PrefService* prefs) {
78 prefs->RegisterBooleanPref(prefs::kPromptForDownload, false);
[email protected]20ce516d2010-06-18 02:20:0479 prefs->RegisterStringPref(prefs::kDownloadExtensionsToOpen, "");
[email protected]f052118e2008-09-05 02:25:3280 prefs->RegisterBooleanPref(prefs::kDownloadDirUpgraded, false);
81
82 // The default download path is userprofile\download.
[email protected]290c9702010-03-09 04:01:3683 const FilePath& default_download_path =
84 download_util::GetDefaultDownloadDirectory();
[email protected]b9636002009-03-04 00:05:2585 prefs->RegisterFilePathPref(prefs::kDownloadDefaultDirectory,
86 default_download_path);
[email protected]f052118e2008-09-05 02:25:3287
88 // If the download path is dangerous we forcefully reset it. But if we do
89 // so we set a flag to make sure we only do it once, to avoid fighting
90 // the user if he really wants it on an unsafe place such as the desktop.
91
92 if (!prefs->GetBoolean(prefs::kDownloadDirUpgraded)) {
[email protected]38233e62010-04-26 04:43:3693 FilePath current_download_dir = prefs->GetFilePath(
94 prefs::kDownloadDefaultDirectory);
[email protected]290c9702010-03-09 04:01:3695 if (download_util::DownloadPathIsDangerous(current_download_dir)) {
[email protected]38233e62010-04-26 04:43:3696 prefs->SetFilePath(prefs::kDownloadDefaultDirectory,
97 default_download_path);
[email protected]f052118e2008-09-05 02:25:3298 }
99 prefs->SetBoolean(prefs::kDownloadDirUpgraded, true);
100 }
initial.commit09911bf2008-07-26 23:55:29101}
102
103DownloadManager::DownloadManager()
104 : shutdown_needed_(false),
105 profile_(NULL),
[email protected]d2a8fb72010-01-21 05:31:42106 file_manager_(NULL),
107 fake_db_handle_(kUninitializedHandle - 1) {
initial.commit09911bf2008-07-26 23:55:29108}
109
110DownloadManager::~DownloadManager() {
[email protected]b0ab1d42010-02-24 19:29:28111 FOR_EACH_OBSERVER(Observer, observers_, ManagerGoingDown());
112
initial.commit09911bf2008-07-26 23:55:29113 if (shutdown_needed_)
114 Shutdown();
115}
116
117void DownloadManager::Shutdown() {
118 DCHECK(shutdown_needed_) << "Shutdown called when not needed.";
119
120 // Stop receiving download updates
[email protected]2941c2392010-07-15 22:54:30121 if (file_manager_)
122 file_manager_->RemoveDownloadManager(this);
initial.commit09911bf2008-07-26 23:55:29123
124 // Stop making history service requests
125 cancelable_consumer_.CancelAllRequests();
126
127 // 'in_progress_' may contain DownloadItems that have not finished the start
128 // complete (from the history service) and thus aren't in downloads_.
129 DownloadMap::iterator it = in_progress_.begin();
[email protected]9ccbb372008-10-10 18:50:32130 std::set<DownloadItem*> to_remove;
initial.commit09911bf2008-07-26 23:55:29131 for (; it != in_progress_.end(); ++it) {
132 DownloadItem* download = it->second;
[email protected]9ccbb372008-10-10 18:50:32133 if (download->safety_state() == DownloadItem::DANGEROUS) {
134 // Forget about any download that the user did not approve.
135 // Note that we cannot call download->Remove() this would invalidate our
136 // iterator.
137 to_remove.insert(download);
138 continue;
initial.commit09911bf2008-07-26 23:55:29139 }
[email protected]9ccbb372008-10-10 18:50:32140 DCHECK_EQ(DownloadItem::IN_PROGRESS, download->state());
141 download->Cancel(false);
142 UpdateHistoryForDownload(download);
initial.commit09911bf2008-07-26 23:55:29143 if (download->db_handle() == kUninitializedHandle) {
144 // An invalid handle means that 'download' does not yet exist in
145 // 'downloads_', so we have to delete it here.
146 delete download;
147 }
148 }
149
[email protected]9ccbb372008-10-10 18:50:32150 // 'dangerous_finished_' contains all complete downloads that have not been
151 // approved. They should be removed.
152 it = dangerous_finished_.begin();
153 for (; it != dangerous_finished_.end(); ++it)
154 to_remove.insert(it->second);
155
156 // Remove the dangerous download that are not approved.
157 for (std::set<DownloadItem*>::const_iterator rm_it = to_remove.begin();
158 rm_it != to_remove.end(); ++rm_it) {
159 DownloadItem* download = *rm_it;
[email protected]e10e17c72008-10-15 17:48:32160 int64 handle = download->db_handle();
[email protected]9ccbb372008-10-10 18:50:32161 download->Remove(true);
[email protected]e10e17c72008-10-15 17:48:32162 // Same as above, delete the download if it is not in 'downloads_' (as the
163 // Remove() call above won't have deleted it).
164 if (handle == kUninitializedHandle)
[email protected]9ccbb372008-10-10 18:50:32165 delete download;
166 }
167 to_remove.clear();
168
initial.commit09911bf2008-07-26 23:55:29169 in_progress_.clear();
[email protected]9ccbb372008-10-10 18:50:32170 dangerous_finished_.clear();
initial.commit09911bf2008-07-26 23:55:29171 STLDeleteValues(&downloads_);
172
173 file_manager_ = NULL;
174
175 // Save our file extensions to auto open.
176 SaveAutoOpens();
177
178 // Make sure the save as dialog doesn't notify us back if we're gone before
179 // it returns.
180 if (select_file_dialog_.get())
181 select_file_dialog_->ListenerDestroyed();
182
183 shutdown_needed_ = false;
184}
185
186// Issue a history query for downloads matching 'search_text'. If 'search_text'
187// is empty, return all downloads that we know about.
188void DownloadManager::GetDownloads(Observer* observer,
189 const std::wstring& search_text) {
[email protected]b0ab1d42010-02-24 19:29:28190 std::vector<DownloadItem*> otr_downloads;
191
192 if (profile_->IsOffTheRecord() && search_text.empty()) {
193 // List all incognito downloads and add that to the downloads the parent
194 // profile lists.
195 otr_downloads.reserve(downloads_.size());
196 for (DownloadMap::iterator it = downloads_.begin();
197 it != downloads_.end(); ++it) {
198 DownloadItem* download = it->second;
199 if (download->is_otr() && !download->is_extension_install() &&
200 !download->is_temporary()) {
201 otr_downloads.push_back(download);
202 }
203 }
204 }
205
206 profile_->GetOriginalProfile()->GetDownloadManager()->
207 DoGetDownloads(observer, search_text, otr_downloads);
208}
209
210void DownloadManager::DoGetDownloads(
211 Observer* observer,
212 const std::wstring& search_text,
213 std::vector<DownloadItem*>& otr_downloads) {
initial.commit09911bf2008-07-26 23:55:29214 DCHECK(observer);
215
216 // Return a empty list if we've not yet received the set of downloads from the
217 // history system (we'll update all observers once we get that list in
218 // OnQueryDownloadEntriesComplete), or if there are no downloads at all.
initial.commit09911bf2008-07-26 23:55:29219 if (downloads_.empty()) {
[email protected]b0ab1d42010-02-24 19:29:28220 observer->SetDownloads(otr_downloads);
initial.commit09911bf2008-07-26 23:55:29221 return;
222 }
223
[email protected]b0ab1d42010-02-24 19:29:28224 std::vector<DownloadItem*> download_copy;
initial.commit09911bf2008-07-26 23:55:29225 // We already know all the downloads and there is no filter, so just return a
226 // copy to the observer.
227 if (search_text.empty()) {
228 download_copy.reserve(downloads_.size());
229 for (DownloadMap::iterator it = downloads_.begin();
230 it != downloads_.end(); ++it) {
[email protected]67f373a2009-09-22 02:44:51231 if (it->second->db_handle() > kUninitializedHandle)
232 download_copy.push_back(it->second);
initial.commit09911bf2008-07-26 23:55:29233 }
234
[email protected]b0ab1d42010-02-24 19:29:28235 // Merge sort based on start time.
236 std::vector<DownloadItem*> merged_downloads;
237 std::merge(otr_downloads.begin(), otr_downloads.end(),
238 download_copy.begin(), download_copy.end(),
239 std::back_inserter(merged_downloads),
240 CompareStartTime);
241
initial.commit09911bf2008-07-26 23:55:29242 // We retain ownership of the DownloadItems.
[email protected]b0ab1d42010-02-24 19:29:28243 observer->SetDownloads(merged_downloads);
initial.commit09911bf2008-07-26 23:55:29244 return;
245 }
246
[email protected]b0ab1d42010-02-24 19:29:28247 DCHECK(otr_downloads.empty());
248
initial.commit09911bf2008-07-26 23:55:29249 // Issue a request to the history service for a list of downloads matching
250 // our search text.
251 HistoryService* hs =
252 profile_->GetHistoryService(Profile::EXPLICIT_ACCESS);
253 if (hs) {
254 HistoryService::Handle h =
[email protected]e53668962010-06-23 15:35:25255 hs->SearchDownloads(WideToUTF16(search_text),
initial.commit09911bf2008-07-26 23:55:29256 &cancelable_consumer_,
257 NewCallback(this,
258 &DownloadManager::OnSearchComplete));
259 cancelable_consumer_.SetClientData(hs, h, observer);
260 }
261}
262
[email protected]6aa4a1c02010-01-15 18:49:58263void DownloadManager::GetTemporaryDownloads(Observer* observer,
264 const FilePath& dir_path) {
265 DCHECK(observer);
266
267 std::vector<DownloadItem*> download_copy;
268
269 for (DownloadMap::iterator it = downloads_.begin();
270 it != downloads_.end(); ++it) {
271 if (it->second->is_temporary() &&
272 it->second->full_path().DirName() == dir_path)
273 download_copy.push_back(it->second);
274 }
275
276 observer->SetDownloads(download_copy);
277}
278
[email protected]8ddbd66a2010-05-21 16:38:34279void DownloadManager::GetAllDownloads(Observer* observer,
280 const FilePath& dir_path) {
281 DCHECK(observer);
282
283 std::vector<DownloadItem*> download_copy;
284
285 for (DownloadMap::iterator it = downloads_.begin();
286 it != downloads_.end(); ++it) {
287 if (!it->second->is_temporary() &&
288 (dir_path.empty() || it->second->full_path().DirName() == dir_path))
289 download_copy.push_back(it->second);
290 }
291
292 observer->SetDownloads(download_copy);
293}
294
[email protected]c4a530b2010-03-08 17:33:03295void DownloadManager::GetCurrentDownloads(Observer* observer,
296 const FilePath& dir_path) {
297 DCHECK(observer);
298
299 std::vector<DownloadItem*> download_copy;
300
301 for (DownloadMap::iterator it = downloads_.begin();
302 it != downloads_.end(); ++it) {
303 if (!it->second->is_temporary() &&
304 (it->second->state() == DownloadItem::IN_PROGRESS ||
305 it->second->safety_state() == DownloadItem::DANGEROUS) &&
306 (dir_path.empty() || it->second->full_path().DirName() == dir_path))
307 download_copy.push_back(it->second);
308 }
309
310 observer->SetDownloads(download_copy);
311}
312
initial.commit09911bf2008-07-26 23:55:29313// Query the history service for information about all persisted downloads.
314bool DownloadManager::Init(Profile* profile) {
315 DCHECK(profile);
316 DCHECK(!shutdown_needed_) << "DownloadManager already initialized.";
317 shutdown_needed_ = true;
318
319 profile_ = profile;
[email protected]be180c802009-10-23 06:33:31320 request_context_getter_ = profile_->GetRequestContext();
initial.commit09911bf2008-07-26 23:55:29321
322 // 'incognito mode' will have access to past downloads, but we won't store
323 // information about new downloads while in that mode.
324 QueryHistoryForDownloads();
325
[email protected]024f2f02010-04-30 22:51:46326 // Cleans up entries only when called for the first time. Subsequent calls are
327 // a no op.
328 CleanUpInProgressHistoryEntries();
329
[email protected]2941c2392010-07-15 22:54:30330 // In test mode, there may be no ResourceDispatcherHost. In this case it's
331 // safe to avoid setting |file_manager_| because we only call a small set of
332 // functions, none of which need it.
initial.commit09911bf2008-07-26 23:55:29333 ResourceDispatcherHost* rdh = g_browser_process->resource_dispatcher_host();
[email protected]2941c2392010-07-15 22:54:30334 if (rdh) {
335 file_manager_ = rdh->download_file_manager();
336 DCHECK(file_manager_);
initial.commit09911bf2008-07-26 23:55:29337 }
338
initial.commit09911bf2008-07-26 23:55:29339 // Get our user preference state.
340 PrefService* prefs = profile_->GetPrefs();
341 DCHECK(prefs);
342 prompt_for_download_.Init(prefs::kPromptForDownload, prefs, NULL);
343
initial.commit09911bf2008-07-26 23:55:29344 download_path_.Init(prefs::kDownloadDefaultDirectory, prefs, NULL);
345
[email protected]bb69e9b32008-08-14 23:08:14346 // Ensure that the download directory specified in the preferences exists.
[email protected]d83d03aa2009-11-02 21:44:37347 ChromeThread::PostTask(
348 ChromeThread::FILE, FROM_HERE,
[email protected]309b7642009-12-09 03:08:50349 NewRunnableFunction(&file_util::CreateDirectory, download_path()));
initial.commit09911bf2008-07-26 23:55:29350
[email protected]2b2f8f72009-02-24 22:42:05351 // We store any file extension that should be opened automatically at
352 // download completion in this pref.
[email protected]ddd231e2010-06-29 20:35:19353 std::string extensions_to_open =
initial.commit09911bf2008-07-26 23:55:29354 prefs->GetString(prefs::kDownloadExtensionsToOpen);
[email protected]ddd231e2010-06-29 20:35:19355 std::vector<std::string> extensions;
356 SplitString(extensions_to_open, ':', &extensions);
357
initial.commit09911bf2008-07-26 23:55:29358 for (size_t i = 0; i < extensions.size(); ++i) {
[email protected]ddd231e2010-06-29 20:35:19359#if defined(OS_POSIX)
360 FilePath path(extensions[i]);
361#elif defined(OS_WIN)
362 FilePath path(UTF8ToWide(extensions[i]));
363#endif
364 if (!extensions[i].empty() && !IsExecutableFile(path))
365 auto_open_.insert(path.value());
initial.commit09911bf2008-07-26 23:55:29366 }
367
[email protected]b0ab1d42010-02-24 19:29:28368 other_download_manager_observer_.reset(
369 new OtherDownloadManagerObserver(this));
370
initial.commit09911bf2008-07-26 23:55:29371 return true;
372}
373
374void DownloadManager::QueryHistoryForDownloads() {
375 HistoryService* hs = profile_->GetHistoryService(Profile::EXPLICIT_ACCESS);
376 if (hs) {
377 hs->QueryDownloads(
378 &cancelable_consumer_,
379 NewCallback(this, &DownloadManager::OnQueryDownloadEntriesComplete));
380 }
381}
382
[email protected]024f2f02010-04-30 22:51:46383void DownloadManager::CleanUpInProgressHistoryEntries() {
384 static bool already_cleaned_up = false;
385
386 if (!already_cleaned_up) {
387 HistoryService* hs = profile_->GetHistoryService(Profile::EXPLICIT_ACCESS);
388 if (hs) {
389 hs->CleanUpInProgressEntries();
390 already_cleaned_up = true;
391 }
392 }
393}
394
initial.commit09911bf2008-07-26 23:55:29395// We have received a message from DownloadFileManager about a new download. We
396// create a download item and store it in our download map, and inform the
397// history system of a new download. Since this method can be called while the
398// history service thread is still reading the persistent state, we do not
399// insert the new DownloadItem into 'downloads_' or inform our observers at this
400// point. OnCreateDatabaseEntryComplete() handles that finalization of the the
401// download creation as a callback from the history thread.
402void DownloadManager::StartDownload(DownloadCreateInfo* info) {
[email protected]d83d03aa2009-11-02 21:44:37403 DCHECK(ChromeThread::CurrentlyOn(ChromeThread::UI));
initial.commit09911bf2008-07-26 23:55:29404 DCHECK(info);
405
[email protected]a60c8ae2009-12-25 06:50:57406 // Check whether this download is for an extension install or not.
[email protected]e6875c12010-07-18 11:14:13407 // Allow extensions to be explicitly saved.
408 if (!info->prompt_user_for_save_location) {
[email protected]a60c8ae2009-12-25 06:50:57409 if (UserScript::HasUserScriptFileExtension(info->url) ||
410 info->mime_type == Extension::kMimeType)
411 info->is_extension_install = true;
412 }
413
[email protected]8af9d032010-02-10 00:00:32414 if (info->save_info.file_path.empty()) {
[email protected]2941c2392010-07-15 22:54:30415 FilePath generated_name;
416 GenerateFileNameFromInfo(info, &generated_name);
417
418 // Freeze the user's preference for showing a Save As dialog. We're going
419 // to bounce around a bunch of threads and we don't want to worry about race
420 // conditions where the user changes this pref out from under us.
421 if (*prompt_for_download_) {
422 // But ignore the user's preference for the following scenarios:
423 // 1) Extension installation. Note that we only care here about the case
424 // where an extension is installed, not when one is downloaded with
425 // "save as...".
426 // 2) Filetypes marked "always open." If the user just wants this file
427 // opened, don't bother asking where to keep it.
428 if (!info->is_extension_install &&
429 !ShouldOpenFileBasedOnExtension(generated_name))
[email protected]e6875c12010-07-18 11:14:13430 info->prompt_user_for_save_location = true;
[email protected]2941c2392010-07-15 22:54:30431 }
432
[email protected]8af9d032010-02-10 00:00:32433 // Determine the proper path for a download, by either one of the following:
434 // 1) using the default download directory.
435 // 2) prompting the user.
[email protected]e6875c12010-07-18 11:14:13436 if (info->prompt_user_for_save_location && !last_download_path_.empty())
[email protected]8af9d032010-02-10 00:00:32437 info->suggested_path = last_download_path_;
438 else
439 info->suggested_path = download_path();
440 info->suggested_path = info->suggested_path.Append(generated_name);
441 } else {
442 info->suggested_path = info->save_info.file_path;
443 }
initial.commit09911bf2008-07-26 23:55:29444
[email protected]e6875c12010-07-18 11:14:13445 if (!info->prompt_user_for_save_location &&
446 info->save_info.file_path.empty()) {
[email protected]4289d9b2009-07-25 21:17:34447 // Downloads can be marked as dangerous for two reasons:
448 // a) They have a dangerous-looking filename
449 // b) They are an extension that is not from the gallery
450 if (IsDangerous(info->suggested_path.BaseName()))
451 info->is_dangerous = true;
[email protected]a60c8ae2009-12-25 06:50:57452 else if (info->is_extension_install &&
[email protected]b7c2f252009-12-08 00:47:23453 !ExtensionsService::IsDownloadFromGallery(info->url,
454 info->referrer_url)) {
[email protected]4289d9b2009-07-25 21:17:34455 info->is_dangerous = true;
456 }
[email protected]e9ebf3fc2008-10-17 22:06:58457 }
458
initial.commit09911bf2008-07-26 23:55:29459 // We need to move over to the download thread because we don't want to stat
460 // the suggested path on the UI thread.
[email protected]d83d03aa2009-11-02 21:44:37461 ChromeThread::PostTask(
462 ChromeThread::FILE, FROM_HERE,
463 NewRunnableMethod(
464 this, &DownloadManager::CheckIfSuggestedPathExists, info));
initial.commit09911bf2008-07-26 23:55:29465}
466
467void DownloadManager::CheckIfSuggestedPathExists(DownloadCreateInfo* info) {
468 DCHECK(info);
469
470 // Check writability of the suggested path. If we can't write to it, default
471 // to the user's "My Documents" directory. We'll prompt them in this case.
[email protected]7ae7c2cb2009-01-06 23:31:41472 FilePath dir = info->suggested_path.DirName();
473 FilePath filename = info->suggested_path.BaseName();
[email protected]9ccbb372008-10-10 18:50:32474 if (!file_util::PathIsWritable(dir)) {
[email protected]e6875c12010-07-18 11:14:13475 info->prompt_user_for_save_location = true;
initial.commit09911bf2008-07-26 23:55:29476 PathService::Get(chrome::DIR_USER_DOCUMENTS, &info->suggested_path);
[email protected]7ae7c2cb2009-01-06 23:31:41477 info->suggested_path = info->suggested_path.Append(filename);
initial.commit09911bf2008-07-26 23:55:29478 }
479
[email protected]6cade212008-12-03 00:32:22480 // If the download is deemed dangerous, we'll use a temporary name for it.
[email protected]e9ebf3fc2008-10-17 22:06:58481 if (info->is_dangerous) {
[email protected]7ae7c2cb2009-01-06 23:31:41482 info->original_name = FilePath(info->suggested_path).BaseName();
[email protected]9ccbb372008-10-10 18:50:32483 // Create a temporary file to hold the file until the user approves its
484 // download.
[email protected]7ae7c2cb2009-01-06 23:31:41485 FilePath::StringType file_name;
486 FilePath path;
[email protected]9ccbb372008-10-10 18:50:32487 while (path.empty()) {
[email protected]594cd7d2010-07-21 03:23:56488 SStringPrintf(&file_name, FILE_PATH_LITERAL("unconfirmed %d.crdownload"),
[email protected]9ccbb372008-10-10 18:50:32489 base::RandInt(0, 100000));
[email protected]7ae7c2cb2009-01-06 23:31:41490 path = dir.Append(file_name);
[email protected]7d3851d82008-12-12 03:26:07491 if (file_util::PathExists(path))
[email protected]7ae7c2cb2009-01-06 23:31:41492 path = FilePath();
[email protected]9ccbb372008-10-10 18:50:32493 }
494 info->suggested_path = path;
[email protected]7a256ea2008-10-17 17:34:16495 } else {
[email protected]594cd7d2010-07-21 03:23:56496 // Do not add the path uniquifier if we are saving to a specific path as in
497 // the drag-out case.
498 if (info->save_info.file_path.empty()) {
499 info->path_uniquifier = download_util::GetUniquePathNumberWithCrDownload(
500 info->suggested_path);
501 }
[email protected]7a256ea2008-10-17 17:34:16502 // We know the final path, build it if necessary.
503 if (info->path_uniquifier > 0) {
[email protected]5a2388a2010-03-26 16:13:39504 download_util::AppendNumberToPath(&(info->suggested_path),
505 info->path_uniquifier);
[email protected]7a256ea2008-10-17 17:34:16506 // Setting path_uniquifier to 0 to make sure we don't try to unique it
507 // later on.
508 info->path_uniquifier = 0;
[email protected]7d3851d82008-12-12 03:26:07509 } else if (info->path_uniquifier == -1) {
510 // We failed to find a unique path. We have to prompt the user.
[email protected]e6875c12010-07-18 11:14:13511 info->prompt_user_for_save_location = true;
[email protected]7a256ea2008-10-17 17:34:16512 }
[email protected]9ccbb372008-10-10 18:50:32513 }
514
[email protected]594cd7d2010-07-21 03:23:56515 // Create an empty file at the suggested path so that we don't allocate the
516 // same "non-existant" path to multiple downloads.
517 // See: http://code.google.com/p/chromium/issues/detail?id=3662
[email protected]e6875c12010-07-18 11:14:13518 if (!info->prompt_user_for_save_location &&
519 info->save_info.file_path.empty()) {
[email protected]594cd7d2010-07-21 03:23:56520 if (info->is_dangerous)
521 file_util::WriteFile(info->suggested_path, "", 0);
522 else
523 file_util::WriteFile(download_util::GetCrDownloadPath(
524 info->suggested_path), "", 0);
[email protected]7d3851d82008-12-12 03:26:07525 }
526
initial.commit09911bf2008-07-26 23:55:29527 // Now we return to the UI thread.
[email protected]d83d03aa2009-11-02 21:44:37528 ChromeThread::PostTask(
529 ChromeThread::UI, FROM_HERE,
initial.commit09911bf2008-07-26 23:55:29530 NewRunnableMethod(this,
531 &DownloadManager::OnPathExistenceAvailable,
532 info));
533}
534
535void DownloadManager::OnPathExistenceAvailable(DownloadCreateInfo* info) {
[email protected]d83d03aa2009-11-02 21:44:37536 DCHECK(ChromeThread::CurrentlyOn(ChromeThread::UI));
initial.commit09911bf2008-07-26 23:55:29537 DCHECK(info);
538
[email protected]e6875c12010-07-18 11:14:13539 if (info->prompt_user_for_save_location) {
initial.commit09911bf2008-07-26 23:55:29540 // We must ask the user for the place to put the download.
541 if (!select_file_dialog_.get())
542 select_file_dialog_ = SelectFileDialog::Create(this);
543
[email protected]76543b92009-08-31 17:27:45544 TabContents* contents = tab_util::GetTabContentsByID(info->child_id,
545 info->render_view_id);
[email protected]b949f1112009-04-12 20:03:08546 SelectFileDialog::FileTypeInfo file_type_info;
547 file_type_info.extensions.resize(1);
548 file_type_info.extensions[0].push_back(info->suggested_path.Extension());
[email protected]15bc8052009-04-17 19:57:24549 if (!file_type_info.extensions[0][0].empty())
550 file_type_info.extensions[0][0].erase(0, 1); // drop the .
[email protected]b949f1112009-04-12 20:03:08551 file_type_info.include_all_files = true;
[email protected]076700e62009-04-01 18:41:23552 gfx::NativeWindow owning_window =
553 contents ? platform_util::GetTopLevel(contents->GetNativeView()) : NULL;
initial.commit09911bf2008-07-26 23:55:29554 select_file_dialog_->SelectFile(SelectFileDialog::SELECT_SAVEAS_FILE,
[email protected]561abe62009-04-06 18:08:34555 string16(),
556 info->suggested_path,
[email protected]b949f1112009-04-12 20:03:08557 &file_type_info, 0, FILE_PATH_LITERAL(""),
[email protected]0f44d3e2009-03-12 23:36:30558 owning_window, info);
initial.commit09911bf2008-07-26 23:55:29559 } else {
560 // No prompting for download, just continue with the suggested name.
561 ContinueStartDownload(info, info->suggested_path);
562 }
563}
564
565void DownloadManager::ContinueStartDownload(DownloadCreateInfo* info,
[email protected]7ae7c2cb2009-01-06 23:31:41566 const FilePath& target_path) {
initial.commit09911bf2008-07-26 23:55:29567 scoped_ptr<DownloadCreateInfo> infop(info);
568 info->path = target_path;
569
570 DownloadItem* download = NULL;
571 DownloadMap::iterator it = in_progress_.find(info->download_id);
572 if (it == in_progress_.end()) {
573 download = new DownloadItem(info->download_id,
574 info->path,
[email protected]7a256ea2008-10-17 17:34:16575 info->path_uniquifier,
initial.commit09911bf2008-07-26 23:55:29576 info->url,
[email protected]494c06e2009-07-25 01:06:42577 info->referrer_url,
[email protected]e435d6b72009-07-25 03:15:58578 info->mime_type,
[email protected]d4a71882010-06-25 16:36:41579 info->original_mime_type,
[email protected]9ccbb372008-10-10 18:50:32580 info->original_name,
initial.commit09911bf2008-07-26 23:55:29581 info->start_time,
582 info->total_bytes,
[email protected]76543b92009-08-31 17:27:45583 info->child_id,
[email protected]9ccbb372008-10-10 18:50:32584 info->request_id,
[email protected]67f373a2009-09-22 02:44:51585 info->is_dangerous,
[email protected]e6875c12010-07-18 11:14:13586 info->prompt_user_for_save_location,
[email protected]b0ab1d42010-02-24 19:29:28587 profile_->IsOffTheRecord(),
[email protected]6aa4a1c02010-01-15 18:49:58588 info->is_extension_install,
[email protected]8af9d032010-02-10 00:00:32589 !info->save_info.file_path.empty());
initial.commit09911bf2008-07-26 23:55:29590 download->set_manager(this);
591 in_progress_[info->download_id] = download;
592 } else {
593 NOTREACHED(); // Should not exist!
594 return;
595 }
596
initial.commit09911bf2008-07-26 23:55:29597 PendingFinishedMap::iterator pending_it =
598 pending_finished_downloads_.find(info->download_id);
[email protected]594cd7d2010-07-21 03:23:56599 bool download_finished = (pending_it != pending_finished_downloads_.end());
600
601 if (download_finished || info->is_dangerous) {
602 // The download has already finished or the download is not safe.
603 // We can now rename the file to its final name (or its tentative name
604 // in dangerous download cases).
605 ChromeThread::PostTask(
606 ChromeThread::FILE, FROM_HERE,
607 NewRunnableMethod(
608 file_manager_, &DownloadFileManager::OnFinalDownloadName,
609 download->id(), target_path, !info->is_dangerous, this));
610 } else {
611 // The download hasn't finished and it is a safe download. We need to
612 // rename it to its intermediate '.crdownload' path.
613 FilePath download_path = download_util::GetCrDownloadPath(target_path);
614 ChromeThread::PostTask(
615 ChromeThread::FILE, FROM_HERE,
616 NewRunnableMethod(
617 file_manager_, &DownloadFileManager::OnIntermediateDownloadName,
618 download->id(), download_path, this));
619 download->set_need_final_rename(true);
620 }
621
622 if (download_finished) {
623 // If the download already completed by the time we reached this point, then
624 // notify observers that it did.
initial.commit09911bf2008-07-26 23:55:29625 DownloadFinished(pending_it->first, pending_it->second);
[email protected]594cd7d2010-07-21 03:23:56626 }
initial.commit09911bf2008-07-26 23:55:29627
628 download->Rename(target_path);
629
[email protected]67f373a2009-09-22 02:44:51630 // Do not store the download in the history database for a few special cases:
631 // - incognito mode (that is the point of this mode)
632 // - extensions (users don't think of extension installation as 'downloading')
[email protected]6aa4a1c02010-01-15 18:49:58633 // - temporary download, like in drag-and-drop
[email protected]1b2f8da2010-07-22 08:29:29634 // - history service is not available (e.g. in tests)
[email protected]67f373a2009-09-22 02:44:51635 // We have to make sure that these handles don't collide with normal db
636 // handles, so we use a negative value. Eventually, they could overlap, but
637 // you'd have to do enough downloading that your ISP would likely stab you in
638 // the neck first. YMMV.
[email protected]1b2f8da2010-07-22 08:29:29639 // FIXME(paulg) see bug 958058. EXPLICIT_ACCESS below is wrong.
640 HistoryService* hs = profile_->GetHistoryService(Profile::EXPLICIT_ACCESS);
[email protected]b0ab1d42010-02-24 19:29:28641 if (download->is_otr() || download->is_extension_install() ||
[email protected]1b2f8da2010-07-22 08:29:29642 download->is_temporary() || !hs) {
[email protected]d2a8fb72010-01-21 05:31:42643 OnCreateDownloadEntryComplete(*info, fake_db_handle_.GetNext());
initial.commit09911bf2008-07-26 23:55:29644 } else {
645 // Update the history system with the new download.
[email protected]1b2f8da2010-07-22 08:29:29646 hs->CreateDownload(
647 *info, &cancelable_consumer_,
648 NewCallback(this, &DownloadManager::OnCreateDownloadEntryComplete));
initial.commit09911bf2008-07-26 23:55:29649 }
[email protected]6a7fb042010-02-01 16:30:47650
651 UpdateAppIcon();
initial.commit09911bf2008-07-26 23:55:29652}
653
654// Convenience function for updating the history service for a download.
655void DownloadManager::UpdateHistoryForDownload(DownloadItem* download) {
656 DCHECK(download);
657
658 // Don't store info in the database if the download was initiated while in
659 // incognito mode or if it hasn't been initialized in our database table.
660 if (download->db_handle() <= kUninitializedHandle)
661 return;
662
[email protected]6cade212008-12-03 00:32:22663 // FIXME(paulg) see bug 958058. EXPLICIT_ACCESS below is wrong.
initial.commit09911bf2008-07-26 23:55:29664 HistoryService* hs = profile_->GetHistoryService(Profile::EXPLICIT_ACCESS);
665 if (hs) {
666 hs->UpdateDownload(download->received_bytes(),
667 download->state(),
668 download->db_handle());
669 }
670}
671
672void DownloadManager::RemoveDownloadFromHistory(DownloadItem* download) {
673 DCHECK(download);
[email protected]6cade212008-12-03 00:32:22674 // FIXME(paulg) see bug 958058. EXPLICIT_ACCESS below is wrong.
initial.commit09911bf2008-07-26 23:55:29675 HistoryService* hs = profile_->GetHistoryService(Profile::EXPLICIT_ACCESS);
676 if (download->db_handle() > kUninitializedHandle && hs)
677 hs->RemoveDownload(download->db_handle());
678}
679
[email protected]e93d2822009-01-30 05:59:59680void DownloadManager::RemoveDownloadsFromHistoryBetween(
681 const base::Time remove_begin,
682 const base::Time remove_end) {
[email protected]6cade212008-12-03 00:32:22683 // FIXME(paulg) see bug 958058. EXPLICIT_ACCESS below is wrong.
initial.commit09911bf2008-07-26 23:55:29684 HistoryService* hs = profile_->GetHistoryService(Profile::EXPLICIT_ACCESS);
685 if (hs)
686 hs->RemoveDownloadsBetween(remove_begin, remove_end);
687}
688
689void DownloadManager::UpdateDownload(int32 download_id, int64 size) {
690 DownloadMap::iterator it = in_progress_.find(download_id);
691 if (it != in_progress_.end()) {
692 DownloadItem* download = it->second;
693 download->Update(size);
694 UpdateHistoryForDownload(download);
695 }
[email protected]6a7fb042010-02-01 16:30:47696 UpdateAppIcon();
initial.commit09911bf2008-07-26 23:55:29697}
698
699void DownloadManager::DownloadFinished(int32 download_id, int64 size) {
700 DownloadMap::iterator it = in_progress_.find(download_id);
[email protected]9ccbb372008-10-10 18:50:32701 if (it == in_progress_.end()) {
initial.commit09911bf2008-07-26 23:55:29702 // The download is done, but the user hasn't selected a final location for
703 // it yet (the Save As dialog box is probably still showing), so just keep
704 // track of the fact that this download id is complete, when the
705 // DownloadItem is constructed later we'll notify its completion then.
706 PendingFinishedMap::iterator erase_it =
707 pending_finished_downloads_.find(download_id);
708 DCHECK(erase_it == pending_finished_downloads_.end());
709 pending_finished_downloads_[download_id] = size;
[email protected]9ccbb372008-10-10 18:50:32710 return;
initial.commit09911bf2008-07-26 23:55:29711 }
[email protected]9ccbb372008-10-10 18:50:32712
713 // Remove the id from the list of pending ids.
714 PendingFinishedMap::iterator erase_it =
715 pending_finished_downloads_.find(download_id);
716 if (erase_it != pending_finished_downloads_.end())
717 pending_finished_downloads_.erase(erase_it);
718
719 DownloadItem* download = it->second;
720 download->Finished(size);
721
722 // Clean up will happen when the history system create callback runs if we
723 // don't have a valid db_handle yet.
724 if (download->db_handle() != kUninitializedHandle) {
725 in_progress_.erase(it);
[email protected]9ccbb372008-10-10 18:50:32726 UpdateHistoryForDownload(download);
727 }
728
[email protected]6a7fb042010-02-01 16:30:47729 UpdateAppIcon();
730
[email protected]9ccbb372008-10-10 18:50:32731 // If this a dangerous download not yet validated by the user, don't do
732 // anything. When the user notifies us, it will trigger a call to
733 // ProceedWithFinishedDangerousDownload.
734 if (download->safety_state() == DownloadItem::DANGEROUS) {
735 dangerous_finished_[download_id] = download;
736 return;
737 }
738
739 if (download->safety_state() == DownloadItem::DANGEROUS_BUT_VALIDATED) {
[email protected]6cade212008-12-03 00:32:22740 // We first need to rename the downloaded file from its temporary name to
[email protected]9ccbb372008-10-10 18:50:32741 // its final name before we can continue.
[email protected]d83d03aa2009-11-02 21:44:37742 ChromeThread::PostTask(
743 ChromeThread::FILE, FROM_HERE,
[email protected]9ccbb372008-10-10 18:50:32744 NewRunnableMethod(
745 this, &DownloadManager::ProceedWithFinishedDangerousDownload,
746 download->db_handle(),
747 download->full_path(), download->original_name()));
748 return;
749 }
[email protected]594cd7d2010-07-21 03:23:56750
751 if (download->need_final_rename()) {
752 ChromeThread::PostTask(
753 ChromeThread::FILE, FROM_HERE,
754 NewRunnableMethod(
755 file_manager_, &DownloadFileManager::OnFinalDownloadName,
756 download->id(), download->full_path(), false, this));
757 return;
758 }
759
[email protected]9ccbb372008-10-10 18:50:32760 ContinueDownloadFinished(download);
761}
762
[email protected]8f783752009-04-01 23:33:45763void DownloadManager::DownloadRenamedToFinalName(int download_id,
764 const FilePath& full_path) {
[email protected]2e030682010-07-23 19:45:36765 DownloadItem* item = GetDownloadItem(download_id);
766 if (!item)
767 return;
768 item->OnNameFinalized();
[email protected]594cd7d2010-07-21 03:23:56769
[email protected]2e030682010-07-23 19:45:36770 // This was called from DownloadFinished; continue to call
771 // ContinueDownloadFinished.
772 if (item->need_final_rename()) {
773 item->set_need_final_rename(false);
774 ContinueDownloadFinished(item);
[email protected]6aa4a1c02010-01-15 18:49:58775 }
[email protected]8f783752009-04-01 23:33:45776}
777
[email protected]9ccbb372008-10-10 18:50:32778void DownloadManager::ContinueDownloadFinished(DownloadItem* download) {
779 // If this was a dangerous download, it has now been approved and must be
780 // removed from dangerous_finished_ so it does not get deleted on shutdown.
781 DownloadMap::iterator it = dangerous_finished_.find(download->id());
782 if (it != dangerous_finished_.end())
783 dangerous_finished_.erase(it);
784
[email protected]0aad67b2009-07-15 20:34:28785 // Handle chrome extensions explicitly and skip the shell execute.
[email protected]a60c8ae2009-12-25 06:50:57786 if (download->is_extension_install()) {
[email protected]d4a71882010-06-25 16:36:41787 OpenChromeExtension(download->full_path(),
788 download->url(),
789 download->referrer_url(),
790 download->original_mime_type());
[email protected]0aad67b2009-07-15 20:34:28791 download->set_auto_opened(true);
792 } else if (download->open_when_complete() ||
[email protected]6aa4a1c02010-01-15 18:49:58793 ShouldOpenFileBasedOnExtension(download->full_path()) ||
794 download->is_temporary()) {
795 // If the download is temporary, like in drag-and-drop, do not open it but
796 // we still need to set it auto-opened so that it can be removed from the
797 // download shelf.
798 if (!download->is_temporary())
799 OpenDownloadInShell(download, NULL);
[email protected]0aad67b2009-07-15 20:34:28800 download->set_auto_opened(true);
801 }
[email protected]9ccbb372008-10-10 18:50:32802
[email protected]0aad67b2009-07-15 20:34:28803 // Notify our observers that we are complete (the call to Finished() set the
804 // state to complete but did not notify).
805 download->UpdateObservers();
[email protected]6aa4a1c02010-01-15 18:49:58806
807 // The download file is meant to be completed if both the filename is
808 // finalized and the file data is downloaded. The ordering of these two
809 // actions is indeterministic. Thus, if the filename is not finalized yet,
810 // delay the notification.
811 if (download->name_finalized())
812 download->NotifyObserversDownloadFileCompleted();
[email protected]0aad67b2009-07-15 20:34:28813}
[email protected]eccb9d12009-10-28 05:40:09814
[email protected]9ccbb372008-10-10 18:50:32815// Called on the file thread. Renames the downloaded file to its original name.
816void DownloadManager::ProceedWithFinishedDangerousDownload(
817 int64 download_handle,
[email protected]7ae7c2cb2009-01-06 23:31:41818 const FilePath& path,
819 const FilePath& original_name) {
[email protected]9ccbb372008-10-10 18:50:32820 bool success = false;
[email protected]7ae7c2cb2009-01-06 23:31:41821 FilePath new_path;
[email protected]7a256ea2008-10-17 17:34:16822 int uniquifier = 0;
[email protected]9ccbb372008-10-10 18:50:32823 if (file_util::PathExists(path)) {
[email protected]889ed35c2009-01-21 00:07:24824 new_path = path.DirName().Append(original_name);
[email protected]7a256ea2008-10-17 17:34:16825 // Make our name unique at this point, as if a dangerous file is downloading
826 // and a 2nd download is started for a file with the same name, they would
827 // have the same path. This is because we uniquify the name on download
828 // start, and at that time the first file does not exists yet, so the second
829 // file gets the same name.
[email protected]5a2388a2010-03-26 16:13:39830 uniquifier = download_util::GetUniquePathNumber(new_path);
[email protected]7a256ea2008-10-17 17:34:16831 if (uniquifier > 0)
[email protected]5a2388a2010-03-26 16:13:39832 download_util::AppendNumberToPath(&new_path, uniquifier);
[email protected]9ccbb372008-10-10 18:50:32833 success = file_util::Move(path, new_path);
834 } else {
835 NOTREACHED();
836 }
[email protected]6cade212008-12-03 00:32:22837
[email protected]d83d03aa2009-11-02 21:44:37838 ChromeThread::PostTask(
839 ChromeThread::UI, FROM_HERE,
[email protected]9ccbb372008-10-10 18:50:32840 NewRunnableMethod(this, &DownloadManager::DangerousDownloadRenamed,
[email protected]7a256ea2008-10-17 17:34:16841 download_handle, success, new_path, uniquifier));
[email protected]9ccbb372008-10-10 18:50:32842}
843
844// Call from the file thread when the finished dangerous download was renamed.
845void DownloadManager::DangerousDownloadRenamed(int64 download_handle,
846 bool success,
[email protected]7ae7c2cb2009-01-06 23:31:41847 const FilePath& new_path,
[email protected]7a256ea2008-10-17 17:34:16848 int new_path_uniquifier) {
[email protected]9ccbb372008-10-10 18:50:32849 DownloadMap::iterator it = downloads_.find(download_handle);
850 if (it == downloads_.end()) {
851 NOTREACHED();
852 return;
853 }
854
855 DownloadItem* download = it->second;
856 // If we failed to rename the file, we'll just keep the name as is.
[email protected]7a256ea2008-10-17 17:34:16857 if (success) {
858 // We need to update the path uniquifier so that the UI shows the right
859 // name when calling GetFileName().
860 download->set_path_uniquifier(new_path_uniquifier);
[email protected]9ccbb372008-10-10 18:50:32861 RenameDownload(download, new_path);
[email protected]7a256ea2008-10-17 17:34:16862 }
[email protected]9ccbb372008-10-10 18:50:32863
864 // Continue the download finished sequence.
865 ContinueDownloadFinished(download);
initial.commit09911bf2008-07-26 23:55:29866}
867
initial.commit09911bf2008-07-26 23:55:29868void DownloadManager::DownloadCancelled(int32 download_id) {
869 DownloadMap::iterator it = in_progress_.find(download_id);
870 if (it == in_progress_.end())
871 return;
872 DownloadItem* download = it->second;
873
initial.commit09911bf2008-07-26 23:55:29874 // Clean up will happen when the history system create callback runs if we
875 // don't have a valid db_handle yet.
876 if (download->db_handle() != kUninitializedHandle) {
877 in_progress_.erase(it);
initial.commit09911bf2008-07-26 23:55:29878 UpdateHistoryForDownload(download);
879 }
880
[email protected]d7d1c5c2009-08-05 23:52:50881 DownloadCancelledInternal(download_id,
882 download->render_process_id(),
883 download->request_id());
[email protected]6a7fb042010-02-01 16:30:47884 UpdateAppIcon();
[email protected]d7d1c5c2009-08-05 23:52:50885}
886
887void DownloadManager::DownloadCancelledInternal(int download_id,
888 int render_process_id,
889 int request_id) {
[email protected]d85cf072009-10-27 03:59:31890 // Cancel the network request. RDH is guaranteed to outlive the IO thread.
891 ChromeThread::PostTask(
892 ChromeThread::IO, FROM_HERE,
[email protected]ae8945192010-07-20 16:56:26893 NewRunnableFunction(&download_util::CancelDownloadRequest,
[email protected]d85cf072009-10-27 03:59:31894 g_browser_process->resource_dispatcher_host(),
895 render_process_id,
896 request_id));
[email protected]d7d1c5c2009-08-05 23:52:50897
initial.commit09911bf2008-07-26 23:55:29898 // Tell the file manager to cancel the download.
[email protected]d7d1c5c2009-08-05 23:52:50899 file_manager_->RemoveDownload(download_id, this); // On the UI thread
[email protected]d83d03aa2009-11-02 21:44:37900 ChromeThread::PostTask(
901 ChromeThread::FILE, FROM_HERE,
902 NewRunnableMethod(
903 file_manager_, &DownloadFileManager::CancelDownload, download_id));
initial.commit09911bf2008-07-26 23:55:29904}
905
906void DownloadManager::PauseDownload(int32 download_id, bool pause) {
907 DownloadMap::iterator it = in_progress_.find(download_id);
[email protected]d85cf072009-10-27 03:59:31908 if (it == in_progress_.end())
909 return;
initial.commit09911bf2008-07-26 23:55:29910
[email protected]d85cf072009-10-27 03:59:31911 DownloadItem* download = it->second;
912 if (pause == download->is_paused())
913 return;
initial.commit09911bf2008-07-26 23:55:29914
[email protected]d85cf072009-10-27 03:59:31915 // Inform the ResourceDispatcherHost of the new pause state.
916 ChromeThread::PostTask(
917 ChromeThread::IO, FROM_HERE,
918 NewRunnableFunction(&DownloadManager::OnPauseDownloadRequest,
919 g_browser_process->resource_dispatcher_host(),
920 download->render_process_id(),
921 download->request_id(),
922 pause));
initial.commit09911bf2008-07-26 23:55:29923}
924
925// static
926void DownloadManager::OnPauseDownloadRequest(ResourceDispatcherHost* rdh,
927 int render_process_id,
928 int request_id,
929 bool pause) {
930 rdh->PauseRequest(render_process_id, request_id, pause);
931}
932
[email protected]7ae7c2cb2009-01-06 23:31:41933bool DownloadManager::IsDangerous(const FilePath& file_name) {
[email protected]9ccbb372008-10-10 18:50:32934 // TODO(jcampan): Improve me.
[email protected]eccb9d12009-10-28 05:40:09935 return IsExecutableFile(file_name);
[email protected]9ccbb372008-10-10 18:50:32936}
937
[email protected]6a7fb042010-02-01 16:30:47938void DownloadManager::UpdateAppIcon() {
939 int64 total_bytes = 0;
940 int64 received_bytes = 0;
941 int download_count = 0;
942 bool progress_known = true;
943
944 for (DownloadMap::iterator i = in_progress_.begin();
945 i != in_progress_.end();
946 ++i) {
947 ++download_count;
948 const DownloadItem* item = i->second;
949 if (item->total_bytes() > 0) {
950 total_bytes += item->total_bytes();
951 received_bytes += item->received_bytes();
952 } else {
953 // This download didn't specify a Content-Length, so the combined progress
954 // bar neeeds to be indeterminate.
955 progress_known = false;
956 }
957 }
958
959 float progress = 0;
960 if (progress_known && download_count)
961 progress = (float)received_bytes / total_bytes;
962
963 download_util::UpdateAppIconDownloadProgress(download_count,
964 progress_known,
965 progress);
966}
967
[email protected]9ccbb372008-10-10 18:50:32968void DownloadManager::RenameDownload(DownloadItem* download,
[email protected]7ae7c2cb2009-01-06 23:31:41969 const FilePath& new_path) {
[email protected]9ccbb372008-10-10 18:50:32970 download->Rename(new_path);
971
972 // Update the history.
973
974 // No update necessary if the download was initiated while in incognito mode.
975 if (download->db_handle() <= kUninitializedHandle)
976 return;
977
[email protected]6cade212008-12-03 00:32:22978 // FIXME(paulg) see bug 958058. EXPLICIT_ACCESS below is wrong.
[email protected]9ccbb372008-10-10 18:50:32979 HistoryService* hs = profile_->GetHistoryService(Profile::EXPLICIT_ACCESS);
980 if (hs)
[email protected]e53668962010-06-23 15:35:25981 hs->UpdateDownloadPath(new_path, download->db_handle());
[email protected]9ccbb372008-10-10 18:50:32982}
983
initial.commit09911bf2008-07-26 23:55:29984void DownloadManager::RemoveDownload(int64 download_handle) {
985 DownloadMap::iterator it = downloads_.find(download_handle);
986 if (it == downloads_.end())
987 return;
988
989 // Make history update.
990 DownloadItem* download = it->second;
991 RemoveDownloadFromHistory(download);
992
993 // Remove from our tables and delete.
994 downloads_.erase(it);
[email protected]9ccbb372008-10-10 18:50:32995 it = dangerous_finished_.find(download->id());
996 if (it != dangerous_finished_.end())
997 dangerous_finished_.erase(it);
initial.commit09911bf2008-07-26 23:55:29998
999 // Tell observers to refresh their views.
[email protected]b0ab1d42010-02-24 19:29:281000 NotifyModelChanged();
[email protected]6f712872008-11-07 00:35:361001
1002 delete download;
initial.commit09911bf2008-07-26 23:55:291003}
1004
[email protected]e93d2822009-01-30 05:59:591005int DownloadManager::RemoveDownloadsBetween(const base::Time remove_begin,
1006 const base::Time remove_end) {
initial.commit09911bf2008-07-26 23:55:291007 RemoveDownloadsFromHistoryBetween(remove_begin, remove_end);
1008
initial.commit09911bf2008-07-26 23:55:291009 DownloadMap::iterator it = downloads_.begin();
[email protected]78b8fcc92009-03-31 17:36:281010 std::vector<DownloadItem*> pending_deletes;
initial.commit09911bf2008-07-26 23:55:291011 while (it != downloads_.end()) {
1012 DownloadItem* download = it->second;
1013 DownloadItem::DownloadState state = download->state();
1014 if (download->start_time() >= remove_begin &&
1015 (remove_end.is_null() || download->start_time() < remove_end) &&
1016 (state == DownloadItem::COMPLETE ||
1017 state == DownloadItem::CANCELLED)) {
1018 // Remove from the map and move to the next in the list.
[email protected]b7f05882009-02-22 01:21:561019 downloads_.erase(it++);
[email protected]a6604d92008-10-30 00:58:581020
1021 // Also remove it from any completed dangerous downloads.
1022 DownloadMap::iterator dit = dangerous_finished_.find(download->id());
1023 if (dit != dangerous_finished_.end())
1024 dangerous_finished_.erase(dit);
1025
[email protected]78b8fcc92009-03-31 17:36:281026 pending_deletes.push_back(download);
[email protected]b0ab1d42010-02-24 19:29:281027 // Observer interface.
initial.commit09911bf2008-07-26 23:55:291028
initial.commit09911bf2008-07-26 23:55:291029 continue;
1030 }
1031
1032 ++it;
1033 }
1034
1035 // Tell observers to refresh their views.
[email protected]78b8fcc92009-03-31 17:36:281036 int num_deleted = static_cast<int>(pending_deletes.size());
initial.commit09911bf2008-07-26 23:55:291037 if (num_deleted > 0)
[email protected]b0ab1d42010-02-24 19:29:281038 NotifyModelChanged();
initial.commit09911bf2008-07-26 23:55:291039
[email protected]78b8fcc92009-03-31 17:36:281040 // Delete the download items after updating the observers.
1041 STLDeleteContainerPointers(pending_deletes.begin(), pending_deletes.end());
1042 pending_deletes.clear();
1043
initial.commit09911bf2008-07-26 23:55:291044 return num_deleted;
1045}
1046
[email protected]e93d2822009-01-30 05:59:591047int DownloadManager::RemoveDownloads(const base::Time remove_begin) {
1048 return RemoveDownloadsBetween(remove_begin, base::Time());
initial.commit09911bf2008-07-26 23:55:291049}
1050
[email protected]d41355e6f2009-04-07 21:21:121051int DownloadManager::RemoveAllDownloads() {
[email protected]024f2f02010-04-30 22:51:461052 if (this != profile_->GetOriginalProfile()->GetDownloadManager()) {
1053 // This is an incognito downloader. Clear All should clear main download
1054 // manager as well.
1055 profile_->GetOriginalProfile()->GetDownloadManager()->RemoveAllDownloads();
1056 }
[email protected]d41355e6f2009-04-07 21:21:121057 // The null times make the date range unbounded.
1058 return RemoveDownloadsBetween(base::Time(), base::Time());
1059}
1060
initial.commit09911bf2008-07-26 23:55:291061// Initiate a download of a specific URL. We send the request to the
1062// ResourceDispatcherHost, and let it send us responses like a regular
1063// download.
1064void DownloadManager::DownloadUrl(const GURL& url,
1065 const GURL& referrer,
[email protected]c9825a42009-05-01 22:51:501066 const std::string& referrer_charset,
[email protected]57c6a652009-05-04 07:58:341067 TabContents* tab_contents) {
[email protected]ae8945192010-07-20 16:56:261068 DownloadUrlToFile(url, referrer, referrer_charset, DownloadSaveInfo(),
1069 tab_contents);
[email protected]6aa4a1c02010-01-15 18:49:581070}
1071
1072void DownloadManager::DownloadUrlToFile(const GURL& url,
1073 const GURL& referrer,
1074 const std::string& referrer_charset,
[email protected]8af9d032010-02-10 00:00:321075 const DownloadSaveInfo& save_info,
[email protected]6aa4a1c02010-01-15 18:49:581076 TabContents* tab_contents) {
[email protected]57c6a652009-05-04 07:58:341077 DCHECK(tab_contents);
[email protected]ae8945192010-07-20 16:56:261078 ChromeThread::PostTask(ChromeThread::IO, FROM_HERE,
1079 NewRunnableFunction(&download_util::DownloadUrl,
1080 url,
1081 referrer,
1082 referrer_charset,
1083 save_info,
1084 g_browser_process->resource_dispatcher_host(),
1085 tab_contents->GetRenderProcessHost()->id(),
1086 tab_contents->render_view_host()->routing_id(),
1087 request_context_getter_));
initial.commit09911bf2008-07-26 23:55:291088}
1089
[email protected]7ae7c2cb2009-01-06 23:31:411090void DownloadManager::GenerateExtension(
1091 const FilePath& file_name,
1092 const std::string& mime_type,
1093 FilePath::StringType* generated_extension) {
initial.commit09911bf2008-07-26 23:55:291094 // We're worried about three things here:
1095 //
1096 // 1) Security. Many sites let users upload content, such as buddy icons, to
1097 // their web sites. We want to mitigate the case where an attacker
1098 // supplies a malicious executable with an executable file extension but an
1099 // honest site serves the content with a benign content type, such as
1100 // image/jpeg.
1101 //
1102 // 2) Usability. If the site fails to provide a file extension, we want to
1103 // guess a reasonable file extension based on the content type.
1104 //
1105 // 3) Shell integration. Some file extensions automatically integrate with
1106 // the shell. We block these extensions to prevent a malicious web site
1107 // from integrating with the user's shell.
1108
[email protected]7ae7c2cb2009-01-06 23:31:411109 static const FilePath::CharType default_extension[] =
1110 FILE_PATH_LITERAL("download");
initial.commit09911bf2008-07-26 23:55:291111
1112 // See if our file name already contains an extension.
[email protected]63597e4e2010-07-08 17:49:051113 FilePath::StringType extension = file_name.Extension();
1114 if (!extension.empty())
1115 extension.erase(extension.begin()); // Erase preceding '.'.
initial.commit09911bf2008-07-26 23:55:291116
[email protected]b7f05882009-02-22 01:21:561117#if defined(OS_WIN)
initial.commit09911bf2008-07-26 23:55:291118 // Rename shell-integrated extensions.
1119 if (win_util::IsShellIntegratedExtension(extension))
1120 extension.assign(default_extension);
[email protected]b7f05882009-02-22 01:21:561121#endif
initial.commit09911bf2008-07-26 23:55:291122
1123 std::string mime_type_from_extension;
[email protected]bae0ea12009-02-14 01:20:411124 net::GetMimeTypeFromFile(file_name,
[email protected]7ae7c2cb2009-01-06 23:31:411125 &mime_type_from_extension);
initial.commit09911bf2008-07-26 23:55:291126 if (mime_type == mime_type_from_extension) {
1127 // The hinted extension matches the mime type. It looks like a winner.
1128 generated_extension->swap(extension);
1129 return;
1130 }
1131
[email protected]eccb9d12009-10-28 05:40:091132 if (IsExecutableExtension(extension) && !IsExecutableMimeType(mime_type)) {
initial.commit09911bf2008-07-26 23:55:291133 // We want to be careful about executable extensions. The worry here is
1134 // that a trusted web site could be tricked into dropping an executable file
1135 // on the user's filesystem.
[email protected]a9bb6f692008-07-30 16:40:101136 if (!net::GetPreferredExtensionForMimeType(mime_type, &extension)) {
initial.commit09911bf2008-07-26 23:55:291137 // We couldn't find a good extension for this content type. Use a dummy
1138 // extension instead.
1139 extension.assign(default_extension);
1140 }
1141 }
1142
1143 if (extension.empty()) {
[email protected]a9bb6f692008-07-30 16:40:101144 net::GetPreferredExtensionForMimeType(mime_type, &extension);
initial.commit09911bf2008-07-26 23:55:291145 } else {
[email protected]6cade212008-12-03 00:32:221146 // Append extension generated from the mime type if:
initial.commit09911bf2008-07-26 23:55:291147 // 1. New extension is not ".txt"
1148 // 2. New extension is not the same as the already existing extension.
1149 // 3. New extension is not executable. This action mitigates the case when
[email protected]7ae7c2cb2009-01-06 23:31:411150 // an executable is hidden in a benign file extension;
initial.commit09911bf2008-07-26 23:55:291151 // E.g. my-cat.jpg becomes my-cat.jpg.js if content type is
1152 // application/x-javascript.
[email protected]e106457b2009-03-25 22:43:371153 // 4. New extension is not ".tar" for .gz files. For misconfigured web
1154 // servers, i.e. bug 5772.
[email protected]e32642f62009-09-11 21:58:291155 // 5. The original extension is not ".tgz" & the new extension is not "gz".
[email protected]7ae7c2cb2009-01-06 23:31:411156 FilePath::StringType append_extension;
[email protected]a9bb6f692008-07-30 16:40:101157 if (net::GetPreferredExtensionForMimeType(mime_type, &append_extension)) {
[email protected]3f156552009-02-09 19:44:171158 if (append_extension != FILE_PATH_LITERAL("txt") &&
[email protected]7ae7c2cb2009-01-06 23:31:411159 append_extension != extension &&
[email protected]eccb9d12009-10-28 05:40:091160 !IsExecutableExtension(append_extension) &&
[email protected]e32642f62009-09-11 21:58:291161 !(append_extension == FILE_PATH_LITERAL("gz") &&
1162 extension == FILE_PATH_LITERAL("tgz")) &&
[email protected]e106457b2009-03-25 22:43:371163 (append_extension != FILE_PATH_LITERAL("tar") ||
1164 extension != FILE_PATH_LITERAL("gz"))) {
[email protected]3f156552009-02-09 19:44:171165 extension += FILE_PATH_LITERAL(".");
initial.commit09911bf2008-07-26 23:55:291166 extension += append_extension;
[email protected]3f156552009-02-09 19:44:171167 }
initial.commit09911bf2008-07-26 23:55:291168 }
1169 }
1170
1171 generated_extension->swap(extension);
1172}
1173
[email protected]6aa4a1c02010-01-15 18:49:581174void DownloadManager::GenerateFileNameFromInfo(DownloadCreateInfo* info,
1175 FilePath* generated_name) {
1176 GenerateFileName(GURL(info->url),
1177 info->content_disposition,
1178 info->referrer_charset,
1179 info->mime_type,
1180 generated_name);
1181}
1182
1183void DownloadManager::GenerateFileName(const GURL& url,
1184 const std::string& content_disposition,
1185 const std::string& referrer_charset,
1186 const std::string& mime_type,
[email protected]7ae7c2cb2009-01-06 23:31:411187 FilePath* generated_name) {
[email protected]630947c2009-11-04 18:37:311188 std::wstring default_name =
1189 l10n_util::GetString(IDS_DEFAULT_DOWNLOAD_FILENAME);
1190#if defined(OS_WIN)
1191 FilePath default_file_path(default_name);
1192#elif defined(OS_POSIX)
1193 FilePath default_file_path(base::SysWideToNativeMB(default_name));
1194#endif
1195
[email protected]6aa4a1c02010-01-15 18:49:581196 *generated_name = net::GetSuggestedFilename(GURL(url),
1197 content_disposition,
1198 referrer_charset,
[email protected]630947c2009-11-04 18:37:311199 default_file_path);
1200
[email protected]7ae7c2cb2009-01-06 23:31:411201 DCHECK(!generated_name->empty());
initial.commit09911bf2008-07-26 23:55:291202
[email protected]6aa4a1c02010-01-15 18:49:581203 GenerateSafeFileName(mime_type, generated_name);
initial.commit09911bf2008-07-26 23:55:291204}
1205
1206void DownloadManager::AddObserver(Observer* observer) {
1207 observers_.AddObserver(observer);
1208 observer->ModelChanged();
1209}
1210
1211void DownloadManager::RemoveObserver(Observer* observer) {
1212 observers_.RemoveObserver(observer);
1213}
1214
1215// Post Windows Shell operations to the Download thread, to avoid blocking the
1216// user interface.
1217void DownloadManager::ShowDownloadInShell(const DownloadItem* download) {
1218 DCHECK(file_manager_);
[email protected]d83d03aa2009-11-02 21:44:371219 DCHECK(ChromeThread::CurrentlyOn(ChromeThread::UI));
[email protected]8b6ff012009-08-18 22:29:581220#if defined(OS_MACOSX)
1221 // Mac needs to run this operation on the UI thread.
1222 platform_util::ShowItemInFolder(download->full_path());
1223#else
[email protected]d83d03aa2009-11-02 21:44:371224 ChromeThread::PostTask(
1225 ChromeThread::FILE, FROM_HERE,
1226 NewRunnableMethod(
1227 file_manager_, &DownloadFileManager::OnShowDownloadInShell,
1228 FilePath(download->full_path())));
[email protected]8b6ff012009-08-18 22:29:581229#endif
initial.commit09911bf2008-07-26 23:55:291230}
1231
[email protected]8f783752009-04-01 23:33:451232void DownloadManager::OpenDownload(const DownloadItem* download,
1233 gfx::NativeView parent_window) {
[email protected]0e34d7892009-06-05 19:17:401234 // Open Chrome extensions with ExtensionsService. For everything else do shell
[email protected]8f783752009-04-01 23:33:451235 // execute.
[email protected]a60c8ae2009-12-25 06:50:571236 if (download->is_extension_install()) {
[email protected]d4a71882010-06-25 16:36:411237 OpenChromeExtension(download->full_path(),
1238 download->url(),
1239 download->referrer_url(),
1240 download->original_mime_type());
[email protected]8f783752009-04-01 23:33:451241 } else {
1242 OpenDownloadInShell(download, parent_window);
1243 }
1244}
1245
[email protected]d4a71882010-06-25 16:36:411246void DownloadManager::OpenChromeExtension(
1247 const FilePath& full_path,
1248 const GURL& download_url,
1249 const GURL& referrer_url,
1250 const std::string& original_mime_type) {
[email protected]6ef635e42009-07-26 06:16:121251 // We don't support extensions in OTR mode.
[email protected]2a464a92009-08-01 17:58:351252 ExtensionsService* service = profile_->GetExtensionsService();
1253 if (service) {
[email protected]91e1bd82009-09-03 22:04:401254 NotificationService* nservice = NotificationService::current();
[email protected]6dfbbf82010-03-12 23:09:161255 GURL nonconst_download_url = download_url;
1256 nservice->Notify(NotificationType::EXTENSION_READY_FOR_INSTALL,
1257 Source<DownloadManager>(this),
1258 Details<GURL>(&nonconst_download_url));
1259
1260 scoped_refptr<CrxInstaller> installer(
1261 new CrxInstaller(service->install_directory(),
1262 service,
1263 new ExtensionInstallUI(profile_)));
1264 installer->set_delete_source(true);
1265
[email protected]a050d712009-11-06 00:40:101266 if (UserScript::HasUserScriptFileExtension(download_url)) {
[email protected]6dfbbf82010-03-12 23:09:161267 installer->InstallUserScript(full_path, download_url);
[email protected]6657afa62009-11-04 02:15:201268 } else {
[email protected]be18e1582010-06-03 21:50:421269 bool is_gallery_download =
1270 ExtensionsService::IsDownloadFromGallery(download_url, referrer_url);
[email protected]d4a71882010-06-25 16:36:411271 installer->set_original_mime_type(original_mime_type);
1272 installer->set_apps_require_extension_mime_type(true);
[email protected]6dfbbf82010-03-12 23:09:161273 installer->set_allow_privilege_increase(true);
1274 installer->set_original_url(download_url);
[email protected]9f72aa02010-06-25 10:01:051275 installer->set_limit_web_extent_to_download_host(!is_gallery_download);
[email protected]6dfbbf82010-03-12 23:09:161276 installer->InstallCrx(full_path);
[email protected]6657afa62009-11-04 02:15:201277 }
[email protected]21ca982c2010-01-26 22:49:551278 } else {
1279 TabContents* contents = NULL;
[email protected]a65bce4e2010-06-09 16:00:151280 // Get last active normal browser of profile.
1281 Browser* last_active = BrowserList::FindBrowserWithType(profile_,
[email protected]074e4f42010-06-23 16:12:061282 Browser::TYPE_NORMAL, true);
[email protected]21ca982c2010-01-26 22:49:551283 if (last_active)
1284 contents = last_active->GetSelectedTabContents();
1285 if (contents) {
1286 contents->AddInfoBar(
1287 new SimpleAlertInfoBarDelegate(contents,
1288 l10n_util::GetString(
1289 IDS_EXTENSION_INCOGNITO_INSTALL_INFOBAR_LABEL),
1290 ResourceBundle::GetSharedInstance().GetBitmapNamed(
[email protected]938e1f92010-04-01 18:09:421291 IDR_INFOBAR_PLUGIN_INSTALL),
1292 true));
[email protected]21ca982c2010-01-26 22:49:551293 }
[email protected]2a464a92009-08-01 17:58:351294 }
[email protected]8f783752009-04-01 23:33:451295}
1296
initial.commit09911bf2008-07-26 23:55:291297void DownloadManager::OpenDownloadInShell(const DownloadItem* download,
[email protected]e93d2822009-01-30 05:59:591298 gfx::NativeView parent_window) {
initial.commit09911bf2008-07-26 23:55:291299 DCHECK(file_manager_);
[email protected]d83d03aa2009-11-02 21:44:371300 DCHECK(ChromeThread::CurrentlyOn(ChromeThread::UI));
[email protected]8b6ff012009-08-18 22:29:581301#if defined(OS_MACOSX)
1302 // Mac OS X requires opening downloads on the UI thread.
1303 platform_util::OpenItem(download->full_path());
1304#else
[email protected]d83d03aa2009-11-02 21:44:371305 ChromeThread::PostTask(
1306 ChromeThread::FILE, FROM_HERE,
1307 NewRunnableMethod(
1308 file_manager_, &DownloadFileManager::OnOpenDownloadInShell,
1309 download->full_path(), download->url(), parent_window));
[email protected]8b6ff012009-08-18 22:29:581310#endif
initial.commit09911bf2008-07-26 23:55:291311}
1312
[email protected]eccb9d12009-10-28 05:40:091313void DownloadManager::OpenFilesBasedOnExtension(
1314 const FilePath& path, bool open) {
1315 FilePath::StringType extension = path.Extension();
1316 if (extension.empty())
1317 return;
1318 DCHECK(extension[0] == FilePath::kExtensionSeparator);
1319 extension.erase(0, 1);
1320 if (open && !IsExecutableExtension(extension))
initial.commit09911bf2008-07-26 23:55:291321 auto_open_.insert(extension);
1322 else
1323 auto_open_.erase(extension);
1324 SaveAutoOpens();
1325}
1326
[email protected]eccb9d12009-10-28 05:40:091327bool DownloadManager::ShouldOpenFileBasedOnExtension(
1328 const FilePath& path) const {
[email protected]eccb9d12009-10-28 05:40:091329 FilePath::StringType extension = path.Extension();
1330 if (extension.empty())
1331 return false;
1332 if (IsExecutableExtension(extension))
1333 return false;
[email protected]92e11c82010-01-13 06:39:561334 if (Extension::IsExtension(path))
1335 return false;
[email protected]eccb9d12009-10-28 05:40:091336 DCHECK(extension[0] == FilePath::kExtensionSeparator);
1337 extension.erase(0, 1);
[email protected]92e11c82010-01-13 06:39:561338 if (auto_open_.find(extension) != auto_open_.end())
[email protected]8c756ac2009-01-30 23:36:411339 return true;
initial.commit09911bf2008-07-26 23:55:291340 return false;
1341}
1342
[email protected]7b73d992008-12-15 20:56:461343static const char* kExecutableWhiteList[] = {
initial.commit09911bf2008-07-26 23:55:291344 // JavaScript is just as powerful as EXE.
[email protected]7b73d992008-12-15 20:56:461345 "text/javascript",
1346 "text/javascript;version=*",
[email protected]54d8d452009-04-08 17:29:241347 // Registry files can cause critical changes to the MS OS behavior.
1348 // Addition of this mimetype also addresses bug 7337.
1349 "text/x-registry",
[email protected]60ff8f912008-12-05 07:58:391350 // Some sites use binary/octet-stream to mean application/octet-stream.
1351 // See http://code.google.com/p/chromium/issues/detail?id=1573
[email protected]7b73d992008-12-15 20:56:461352 "binary/octet-stream"
1353};
initial.commit09911bf2008-07-26 23:55:291354
[email protected]7b73d992008-12-15 20:56:461355static const char* kExecutableBlackList[] = {
initial.commit09911bf2008-07-26 23:55:291356 // These application types are not executable.
[email protected]7b73d992008-12-15 20:56:461357 "application/*+xml",
1358 "application/xml"
1359};
initial.commit09911bf2008-07-26 23:55:291360
[email protected]7b73d992008-12-15 20:56:461361// static
1362bool DownloadManager::IsExecutableMimeType(const std::string& mime_type) {
[email protected]bae0ea12009-02-14 01:20:411363 for (size_t i = 0; i < arraysize(kExecutableWhiteList); ++i) {
[email protected]7b73d992008-12-15 20:56:461364 if (net::MatchesMimeType(kExecutableWhiteList[i], mime_type))
1365 return true;
1366 }
[email protected]bae0ea12009-02-14 01:20:411367 for (size_t i = 0; i < arraysize(kExecutableBlackList); ++i) {
[email protected]7b73d992008-12-15 20:56:461368 if (net::MatchesMimeType(kExecutableBlackList[i], mime_type))
1369 return false;
1370 }
1371 // We consider only other application types to be executable.
1372 return net::MatchesMimeType("application/*", mime_type);
initial.commit09911bf2008-07-26 23:55:291373}
1374
[email protected]eccb9d12009-10-28 05:40:091375bool DownloadManager::IsExecutableFile(const FilePath& path) const {
1376 return IsExecutableExtension(path.Extension());
1377}
1378
1379bool DownloadManager::IsExecutableExtension(
[email protected]6aa4a1c02010-01-15 18:49:581380 const FilePath::StringType& extension) {
[email protected]eccb9d12009-10-28 05:40:091381 if (extension.empty())
1382 return false;
[email protected]64da0b932009-02-24 02:30:041383 if (!IsStringASCII(extension))
1384 return false;
[email protected]a0a9577b2009-05-27 23:52:321385#if defined(OS_WIN)
[email protected]64da0b932009-02-24 02:30:041386 std::string ascii_extension = WideToASCII(extension);
[email protected]e9ef0a62009-08-11 22:50:131387#elif defined(OS_POSIX)
[email protected]a0a9577b2009-05-27 23:52:321388 std::string ascii_extension = extension;
1389#endif
[email protected]64da0b932009-02-24 02:30:041390
[email protected]eccb9d12009-10-28 05:40:091391 // Strip out leading dot if it's still there
1392 if (ascii_extension[0] == FilePath::kExtensionSeparator)
1393 ascii_extension.erase(0, 1);
1394
[email protected]6aa4a1c02010-01-15 18:49:581395 return download_util::IsExecutableExtension(ascii_extension);
initial.commit09911bf2008-07-26 23:55:291396}
1397
1398void DownloadManager::ResetAutoOpenFiles() {
1399 auto_open_.clear();
1400 SaveAutoOpens();
1401}
1402
1403bool DownloadManager::HasAutoOpenFileTypesRegistered() const {
1404 return !auto_open_.empty();
1405}
1406
1407void DownloadManager::SaveAutoOpens() {
1408 PrefService* prefs = profile_->GetPrefs();
1409 if (prefs) {
[email protected]ddd231e2010-06-29 20:35:191410 std::string extensions;
[email protected]eccb9d12009-10-28 05:40:091411 for (AutoOpenSet::iterator it = auto_open_.begin();
initial.commit09911bf2008-07-26 23:55:291412 it != auto_open_.end(); ++it) {
[email protected]ddd231e2010-06-29 20:35:191413#if defined(OS_POSIX)
1414 std::string this_extension = *it;
1415#elif defined(OS_WIN)
1416 std::string this_extension = base::SysWideToUTF8(*it);
1417#endif
1418 extensions += this_extension + ":";
initial.commit09911bf2008-07-26 23:55:291419 }
1420 if (!extensions.empty())
1421 extensions.erase(extensions.size() - 1);
[email protected]b7f05882009-02-22 01:21:561422
[email protected]ddd231e2010-06-29 20:35:191423 prefs->SetString(prefs::kDownloadExtensionsToOpen, extensions);
initial.commit09911bf2008-07-26 23:55:291424 }
1425}
1426
[email protected]561abe62009-04-06 18:08:341427void DownloadManager::FileSelected(const FilePath& path,
[email protected]23b357b2009-03-30 20:02:361428 int index, void* params) {
initial.commit09911bf2008-07-26 23:55:291429 DownloadCreateInfo* info = reinterpret_cast<DownloadCreateInfo*>(params);
[email protected]e6875c12010-07-18 11:14:131430 if (info->prompt_user_for_save_location)
[email protected]7ae7c2cb2009-01-06 23:31:411431 last_download_path_ = path.DirName();
initial.commit09911bf2008-07-26 23:55:291432 ContinueStartDownload(info, path);
1433}
1434
1435void DownloadManager::FileSelectionCanceled(void* params) {
1436 // The user didn't pick a place to save the file, so need to cancel the
1437 // download that's already in progress to the temporary location.
1438 DownloadCreateInfo* info = reinterpret_cast<DownloadCreateInfo*>(params);
[email protected]d7d1c5c2009-08-05 23:52:501439 DownloadCancelledInternal(info->download_id,
[email protected]76543b92009-08-31 17:27:451440 info->child_id,
[email protected]d7d1c5c2009-08-05 23:52:501441 info->request_id);
initial.commit09911bf2008-07-26 23:55:291442}
1443
[email protected]7ae7c2cb2009-01-06 23:31:411444void DownloadManager::DeleteDownload(const FilePath& path) {
[email protected]d83d03aa2009-11-02 21:44:371445 ChromeThread::PostTask(
1446 ChromeThread::FILE, FROM_HERE,
[email protected]da74e5ae22010-07-14 16:12:371447 NewRunnableFunction(&DeleteDownloadedFile, path));
[email protected]9ccbb372008-10-10 18:50:321448}
1449
[email protected]9ccbb372008-10-10 18:50:321450void DownloadManager::DangerousDownloadValidated(DownloadItem* download) {
1451 DCHECK_EQ(DownloadItem::DANGEROUS, download->safety_state());
1452 download->set_safety_state(DownloadItem::DANGEROUS_BUT_VALIDATED);
1453 download->UpdateObservers();
1454
1455 // If the download is not complete, nothing to do. The required
1456 // post-processing will be performed when it does complete.
1457 if (download->state() != DownloadItem::COMPLETE)
1458 return;
1459
[email protected]d83d03aa2009-11-02 21:44:371460 ChromeThread::PostTask(
1461 ChromeThread::FILE, FROM_HERE,
1462 NewRunnableMethod(
1463 this, &DownloadManager::ProceedWithFinishedDangerousDownload,
1464 download->db_handle(), download->full_path(),
1465 download->original_name()));
[email protected]9ccbb372008-10-10 18:50:321466}
1467
[email protected]6aa4a1c02010-01-15 18:49:581468void DownloadManager::GenerateSafeFileName(const std::string& mime_type,
[email protected]7ae7c2cb2009-01-06 23:31:411469 FilePath* file_name) {
1470 // Make sure we get the right file extension
1471 FilePath::StringType extension;
[email protected]763f946a2009-01-06 19:04:391472 GenerateExtension(*file_name, mime_type, &extension);
[email protected]1bdaa60c2010-06-01 01:57:431473 *file_name = file_name->ReplaceExtension(extension);
[email protected]763f946a2009-01-06 19:04:391474
[email protected]2b2f8f72009-02-24 22:42:051475#if defined(OS_WIN)
[email protected]763f946a2009-01-06 19:04:391476 // Prepend "_" to the file name if it's a reserved name
[email protected]7ae7c2cb2009-01-06 23:31:411477 FilePath::StringType leaf_name = file_name->BaseName().value();
[email protected]763f946a2009-01-06 19:04:391478 DCHECK(!leaf_name.empty());
1479 if (win_util::IsReservedName(leaf_name)) {
[email protected]7ae7c2cb2009-01-06 23:31:411480 leaf_name = FilePath::StringType(FILE_PATH_LITERAL("_")) + leaf_name;
1481 *file_name = file_name->DirName();
1482 if (file_name->value() == FilePath::kCurrentDirectory) {
1483 *file_name = FilePath(leaf_name);
[email protected]763f946a2009-01-06 19:04:391484 } else {
[email protected]7ae7c2cb2009-01-06 23:31:411485 *file_name = file_name->Append(leaf_name);
[email protected]763f946a2009-01-06 19:04:391486 }
1487 }
[email protected]b7f05882009-02-22 01:21:561488#endif
[email protected]763f946a2009-01-06 19:04:391489}
1490
initial.commit09911bf2008-07-26 23:55:291491// Operations posted to us from the history service ----------------------------
1492
1493// The history service has retrieved all download entries. 'entries' contains
1494// 'DownloadCreateInfo's in sorted order (by ascending start_time).
1495void DownloadManager::OnQueryDownloadEntriesComplete(
1496 std::vector<DownloadCreateInfo>* entries) {
1497 for (size_t i = 0; i < entries->size(); ++i) {
1498 DownloadItem* download = new DownloadItem(entries->at(i));
1499 DCHECK(downloads_.find(download->db_handle()) == downloads_.end());
1500 downloads_[download->db_handle()] = download;
1501 download->set_manager(this);
1502 }
[email protected]b0ab1d42010-02-24 19:29:281503 NotifyModelChanged();
initial.commit09911bf2008-07-26 23:55:291504}
1505
initial.commit09911bf2008-07-26 23:55:291506// Once the new DownloadItem's creation info has been committed to the history
1507// service, we associate the DownloadItem with the db handle, update our
1508// 'downloads_' map and inform observers.
1509void DownloadManager::OnCreateDownloadEntryComplete(DownloadCreateInfo info,
1510 int64 db_handle) {
1511 DownloadMap::iterator it = in_progress_.find(info.download_id);
1512 DCHECK(it != in_progress_.end());
1513
1514 DownloadItem* download = it->second;
[email protected]d2a8fb72010-01-21 05:31:421515
1516 // It's not immediately obvious, but HistoryBackend::CreateDownload() can
1517 // call this function with an invalid |db_handle|. For instance, this can
1518 // happen when the history database is offline. We cannot have multiple
1519 // DownloadItems with the same invalid db_handle, so we need to assign a
1520 // unique |db_handle| here.
1521 if (db_handle == kUninitializedHandle)
1522 db_handle = fake_db_handle_.GetNext();
1523
initial.commit09911bf2008-07-26 23:55:291524 DCHECK(download->db_handle() == kUninitializedHandle);
1525 download->set_db_handle(db_handle);
1526
1527 // Insert into our full map.
1528 DCHECK(downloads_.find(download->db_handle()) == downloads_.end());
1529 downloads_[download->db_handle()] = download;
1530
[email protected]5e595482009-05-06 20:16:531531 // Show in the appropropriate browser UI.
1532 ShowDownloadInBrowser(info, download);
initial.commit09911bf2008-07-26 23:55:291533
1534 // Inform interested objects about the new download.
[email protected]b0ab1d42010-02-24 19:29:281535 NotifyModelChanged();
initial.commit09911bf2008-07-26 23:55:291536
1537 // If this download has been completed before we've received the db handle,
1538 // post one final message to the history service so that it can be properly
1539 // in sync with the DownloadItem's completion status, and also inform any
1540 // observers so that they get more than just the start notification.
1541 if (download->state() != DownloadItem::IN_PROGRESS) {
1542 in_progress_.erase(it);
initial.commit09911bf2008-07-26 23:55:291543 UpdateHistoryForDownload(download);
1544 download->UpdateObservers();
1545 }
[email protected]6a7fb042010-02-01 16:30:471546
1547 UpdateAppIcon();
initial.commit09911bf2008-07-26 23:55:291548}
1549
1550// Called when the history service has retrieved the list of downloads that
1551// match the search text.
1552void DownloadManager::OnSearchComplete(HistoryService::Handle handle,
1553 std::vector<int64>* results) {
1554 HistoryService* hs = profile_->GetHistoryService(Profile::EXPLICIT_ACCESS);
1555 Observer* requestor = cancelable_consumer_.GetClientData(hs, handle);
1556 if (!requestor)
1557 return;
1558
1559 std::vector<DownloadItem*> searched_downloads;
1560 for (std::vector<int64>::iterator it = results->begin();
1561 it != results->end(); ++it) {
1562 DownloadMap::iterator dit = downloads_.find(*it);
1563 if (dit != downloads_.end())
1564 searched_downloads.push_back(dit->second);
1565 }
1566
1567 requestor->SetDownloads(searched_downloads);
1568}
[email protected]905a08d2008-11-19 07:24:121569
[email protected]5e595482009-05-06 20:16:531570void DownloadManager::ShowDownloadInBrowser(const DownloadCreateInfo& info,
1571 DownloadItem* download) {
[email protected]8ddbd66a2010-05-21 16:38:341572 // The 'contents' may no longer exist if the user closed the tab before we
1573 // get this start completion event. If it does, tell the origin TabContents
1574 // to display its download shelf.
[email protected]76543b92009-08-31 17:27:451575 TabContents* contents = tab_util::GetTabContentsByID(info.child_id,
1576 info.render_view_id);
[email protected]5e595482009-05-06 20:16:531577
1578 // If the contents no longer exists, we start the download in the last active
1579 // browser. This is not ideal but better than fully hiding the download from
1580 // the user.
1581 if (!contents) {
1582 Browser* last_active = BrowserList::GetLastActive();
1583 if (last_active)
1584 contents = last_active->GetSelectedTabContents();
1585 }
1586
1587 if (contents)
1588 contents->OnStartDownload(download);
1589}
1590
[email protected]6cade212008-12-03 00:32:221591// Clears the last download path, used to initialize "save as" dialogs.
[email protected]905a08d2008-11-19 07:24:121592void DownloadManager::ClearLastDownloadPath() {
[email protected]7ae7c2cb2009-01-06 23:31:411593 last_download_path_ = FilePath();
[email protected]eea46622009-07-15 20:49:381594}
[email protected]b0ab1d42010-02-24 19:29:281595
1596void DownloadManager::NotifyModelChanged() {
1597 FOR_EACH_OBSERVER(Observer, observers_, ModelChanged());
1598}
1599
[email protected]2e030682010-07-23 19:45:361600DownloadItem* DownloadManager::GetDownloadItem(int id) {
1601 for (DownloadMap::iterator it = downloads_.begin();
1602 it != downloads_.end(); ++it) {
1603 DownloadItem* item = it->second;
1604 if (item->id() == id)
1605 return item;
1606 }
1607 return NULL;
1608}
1609
[email protected]b0ab1d42010-02-24 19:29:281610// DownloadManager::OtherDownloadManagerObserver implementation ----------------
1611
1612DownloadManager::OtherDownloadManagerObserver::OtherDownloadManagerObserver(
1613 DownloadManager* observing_download_manager)
1614 : observing_download_manager_(observing_download_manager),
1615 observed_download_manager_(NULL) {
1616 if (observing_download_manager->profile_->GetOriginalProfile() ==
1617 observing_download_manager->profile_) {
1618 return;
1619 }
1620
1621 observed_download_manager_ = observing_download_manager_->
1622 profile_->GetOriginalProfile()->GetDownloadManager();
1623 observed_download_manager_->AddObserver(this);
1624}
1625
1626DownloadManager::OtherDownloadManagerObserver::~OtherDownloadManagerObserver() {
1627 if (observed_download_manager_)
1628 observed_download_manager_->RemoveObserver(this);
1629}
1630
1631void DownloadManager::OtherDownloadManagerObserver::ModelChanged() {
1632 observing_download_manager_->NotifyModelChanged();
1633}
1634
1635void DownloadManager::OtherDownloadManagerObserver::SetDownloads(
1636 std::vector<DownloadItem*>& downloads) {
1637}
1638
1639void DownloadManager::OtherDownloadManagerObserver::ManagerGoingDown() {
1640 observed_download_manager_ = NULL;
1641}