[email protected] | 5626b089 | 2012-02-20 14:46:58 | [diff] [blame] | 1 | // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 2 | // Use of this source code is governed by a BSD-style license that can be |
| 3 | // found in the LICENSE file. |
| 4 | |
| 5 | #include "chrome/browser/file_select_helper.h" |
| 6 | |
avi | 6846aef | 2015-12-26 01:09:38 | [diff] [blame] | 7 | #include <stddef.h> |
| 8 | |
[email protected] | 5ac950b | 2010-12-09 21:34:25 | [diff] [blame] | 9 | #include <string> |
[email protected] | b7b4beb | 2013-07-09 14:06:50 | [diff] [blame] | 10 | #include <utility> |
[email protected] | 5ac950b | 2010-12-09 21:34:25 | [diff] [blame] | 11 | |
[email protected] | 9f054aa1 | 2011-09-29 19:13:45 | [diff] [blame] | 12 | #include "base/bind.h" |
[email protected] | 25a4c1c | 2013-06-08 04:53:36 | [diff] [blame] | 13 | #include "base/files/file_enumerator.h" |
thestig | 18dfb7a5 | 2014-08-26 10:44:04 | [diff] [blame] | 14 | #include "base/files/file_util.h" |
dcheng | 4af4858 | 2016-04-19 00:29:35 | [diff] [blame] | 15 | #include "base/memory/ptr_util.h" |
[email protected] | 1988e1c | 2013-02-28 20:27:42 | [diff] [blame] | 16 | #include "base/strings/string_split.h" |
[email protected] | d883056 | 2013-06-10 22:01:54 | [diff] [blame] | 17 | #include "base/strings/string_util.h" |
[email protected] | 112158af | 2013-06-07 23:46:18 | [diff] [blame] | 18 | #include "base/strings/utf_string_conversions.h" |
shahriar.rostami | 4d55d831 | 2016-02-25 22:44:28 | [diff] [blame] | 19 | #include "base/threading/worker_pool.h" |
avi | 6846aef | 2015-12-26 01:09:38 | [diff] [blame] | 20 | #include "build/build_config.h" |
hirono | a4a32af2 | 2014-11-07 03:17:29 | [diff] [blame] | 21 | #include "chrome/browser/browser_process.h" |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 22 | #include "chrome/browser/platform_util.h" |
[email protected] | 8ecad5e | 2010-12-02 21:18:33 | [diff] [blame] | 23 | #include "chrome/browser/profiles/profile.h" |
hirono | a4a32af2 | 2014-11-07 03:17:29 | [diff] [blame] | 24 | #include "chrome/browser/profiles/profile_manager.h" |
[email protected] | d989891 | 2011-04-15 21:10:00 | [diff] [blame] | 25 | #include "chrome/browser/ui/browser.h" |
| 26 | #include "chrome/browser/ui/browser_list.h" |
[email protected] | 6e1fcd1 | 2012-07-02 17:14:20 | [diff] [blame] | 27 | #include "chrome/browser/ui/chrome_select_file_policy.h" |
[email protected] | af39f00 | 2014-08-22 10:18:18 | [diff] [blame] | 28 | #include "chrome/grit/generated_resources.h" |
[email protected] | 6a1c98e0 | 2012-10-24 21:49:43 | [diff] [blame] | 29 | #include "content/public/browser/browser_thread.h" |
[email protected] | 6c2381d | 2011-10-19 02:52:53 | [diff] [blame] | 30 | #include "content/public/browser/notification_details.h" |
| 31 | #include "content/public/browser/notification_source.h" |
[email protected] | 0d6e9bd | 2011-10-18 04:29:16 | [diff] [blame] | 32 | #include "content/public/browser/notification_types.h" |
nasko | 35169af | 2016-06-10 18:47:13 | [diff] [blame] | 33 | #include "content/public/browser/render_frame_host.h" |
| 34 | #include "content/public/browser/render_process_host.h" |
[email protected] | 9c1662b | 2012-03-06 15:44:33 | [diff] [blame] | 35 | #include "content/public/browser/render_view_host.h" |
avi | f9ab5d94 | 2015-10-15 14:05:44 | [diff] [blame] | 36 | #include "content/public/browser/render_widget_host.h" |
[email protected] | 5626b089 | 2012-02-20 14:46:58 | [diff] [blame] | 37 | #include "content/public/browser/render_widget_host_view.h" |
rdevlin.cronin | bfbc11d9 | 2015-06-12 23:26:08 | [diff] [blame] | 38 | #include "content/public/browser/storage_partition.h" |
[email protected] | 33f8ad5 | 2012-05-22 18:10:13 | [diff] [blame] | 39 | #include "content/public/browser/web_contents.h" |
hirono | 570357bd | 2014-10-08 12:39:27 | [diff] [blame] | 40 | #include "content/public/common/file_chooser_file_info.h" |
[email protected] | 8caadeb | 2011-11-22 02:45:23 | [diff] [blame] | 41 | #include "content/public/common/file_chooser_params.h" |
asanka | ee261f3 | 2015-10-27 21:04:28 | [diff] [blame] | 42 | #include "net/base/filename_util.h" |
[email protected] | b3841c50 | 2011-03-09 01:21:31 | [diff] [blame] | 43 | #include "net/base/mime_util.h" |
[email protected] | c051a1b | 2011-01-21 23:30:17 | [diff] [blame] | 44 | #include "ui/base/l10n/l10n_util.h" |
[email protected] | 4344a3c | 2013-01-17 23:49:20 | [diff] [blame] | 45 | #include "ui/shell_dialogs/selected_file_info.h" |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 46 | |
hirono | 275bfbe | 2014-10-21 01:46:17 | [diff] [blame] | 47 | #if defined(OS_CHROMEOS) |
| 48 | #include "chrome/browser/chromeos/file_manager/fileapi_util.h" |
| 49 | #include "content/public/browser/site_instance.h" |
| 50 | #endif |
| 51 | |
asanka | ee261f3 | 2015-10-27 21:04:28 | [diff] [blame] | 52 | #if defined(FULL_SAFE_BROWSING) |
asanka | ef1f3872 | 2016-05-20 16:17:41 | [diff] [blame] | 53 | #include "chrome/browser/safe_browsing/download_protection_service.h" |
| 54 | #include "chrome/browser/safe_browsing/safe_browsing_service.h" |
asanka | ee261f3 | 2015-10-27 21:04:28 | [diff] [blame] | 55 | #endif |
| 56 | |
[email protected] | 631bb74 | 2011-11-02 11:29:39 | [diff] [blame] | 57 | using content::BrowserThread; |
[email protected] | 33f8ad5 | 2012-05-22 18:10:13 | [diff] [blame] | 58 | using content::FileChooserParams; |
[email protected] | eaabba2 | 2012-03-07 15:02:11 | [diff] [blame] | 59 | using content::RenderViewHost; |
| 60 | using content::RenderWidgetHost; |
[email protected] | ea049a0 | 2011-12-25 21:37:09 | [diff] [blame] | 61 | using content::WebContents; |
[email protected] | 631bb74 | 2011-11-02 11:29:39 | [diff] [blame] | 62 | |
[email protected] | 600ea40 | 2011-04-12 00:01:51 | [diff] [blame] | 63 | namespace { |
| 64 | |
| 65 | // There is only one file-selection happening at any given time, |
| 66 | // so we allocate an enumeration ID for that purpose. All IDs from |
| 67 | // the renderer must start at 0 and increase. |
[email protected] | 459fba8 | 2011-10-13 02:48:50 | [diff] [blame] | 68 | const int kFileSelectEnumerationId = -1; |
| 69 | |
[email protected] | 53f04c8 | 2012-07-26 02:31:09 | [diff] [blame] | 70 | // Converts a list of FilePaths to a list of ui::SelectedFileInfo. |
| 71 | std::vector<ui::SelectedFileInfo> FilePathListToSelectedFileInfoList( |
[email protected] | 650b2d5 | 2013-02-10 03:41:45 | [diff] [blame] | 72 | const std::vector<base::FilePath>& paths) { |
[email protected] | ddb034b | 2012-06-26 20:31:39 | [diff] [blame] | 73 | std::vector<ui::SelectedFileInfo> selected_files; |
[email protected] | fb11b6a4 | 2012-03-14 07:25:12 | [diff] [blame] | 74 | for (size_t i = 0; i < paths.size(); ++i) { |
| 75 | selected_files.push_back( |
[email protected] | 53f04c8 | 2012-07-26 02:31:09 | [diff] [blame] | 76 | ui::SelectedFileInfo(paths[i], paths[i])); |
[email protected] | fb11b6a4 | 2012-03-14 07:25:12 | [diff] [blame] | 77 | } |
| 78 | return selected_files; |
[email protected] | 600ea40 | 2011-04-12 00:01:51 | [diff] [blame] | 79 | } |
| 80 | |
erikchen | 22de64a3 | 2014-10-10 18:27:43 | [diff] [blame] | 81 | void DeleteFiles(const std::vector<base::FilePath>& paths) { |
| 82 | for (auto& file_path : paths) |
| 83 | base::DeleteFile(file_path, false); |
| 84 | } |
| 85 | |
hirono | a4a32af2 | 2014-11-07 03:17:29 | [diff] [blame] | 86 | bool IsValidProfile(Profile* profile) { |
| 87 | DCHECK_CURRENTLY_ON(content::BrowserThread::UI); |
| 88 | return g_browser_process->profile_manager()->IsValidProfile(profile); |
| 89 | } |
| 90 | |
asanka | ef1f3872 | 2016-05-20 16:17:41 | [diff] [blame] | 91 | #if defined(FULL_SAFE_BROWSING) |
| 92 | |
| 93 | bool IsDownloadAllowedBySafeBrowsing( |
| 94 | safe_browsing::DownloadProtectionService::DownloadCheckResult result) { |
| 95 | using Result = safe_browsing::DownloadProtectionService::DownloadCheckResult; |
| 96 | switch (result) { |
| 97 | // Only allow downloads that are marked as SAFE or UNKNOWN by SafeBrowsing. |
| 98 | // All other types are going to be blocked. UNKNOWN could be the result of a |
| 99 | // failed safe browsing ping. |
| 100 | case Result::UNKNOWN: |
| 101 | case Result::SAFE: |
| 102 | return true; |
| 103 | |
| 104 | case Result::DANGEROUS: |
| 105 | case Result::UNCOMMON: |
| 106 | case Result::DANGEROUS_HOST: |
| 107 | case Result::POTENTIALLY_UNWANTED: |
| 108 | return false; |
| 109 | } |
| 110 | NOTREACHED(); |
| 111 | return false; |
| 112 | } |
| 113 | |
| 114 | void InterpretSafeBrowsingVerdict( |
| 115 | const base::Callback<void(bool)>& recipient, |
| 116 | safe_browsing::DownloadProtectionService::DownloadCheckResult result) { |
| 117 | recipient.Run(IsDownloadAllowedBySafeBrowsing(result)); |
| 118 | } |
| 119 | |
| 120 | #endif |
| 121 | |
[email protected] | fb11b6a4 | 2012-03-14 07:25:12 | [diff] [blame] | 122 | } // namespace |
| 123 | |
[email protected] | 485a527 | 2011-04-12 00:49:29 | [diff] [blame] | 124 | struct FileSelectHelper::ActiveDirectoryEnumeration { |
[email protected] | d45f751 | 2011-06-21 21:18:27 | [diff] [blame] | 125 | ActiveDirectoryEnumeration() : rvh_(NULL) {} |
[email protected] | 485a527 | 2011-04-12 00:49:29 | [diff] [blame] | 126 | |
dcheng | 4af4858 | 2016-04-19 00:29:35 | [diff] [blame] | 127 | std::unique_ptr<DirectoryListerDispatchDelegate> delegate_; |
| 128 | std::unique_ptr<net::DirectoryLister> lister_; |
[email protected] | 485a527 | 2011-04-12 00:49:29 | [diff] [blame] | 129 | RenderViewHost* rvh_; |
[email protected] | 650b2d5 | 2013-02-10 03:41:45 | [diff] [blame] | 130 | std::vector<base::FilePath> results_; |
[email protected] | 485a527 | 2011-04-12 00:49:29 | [diff] [blame] | 131 | }; |
| 132 | |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 133 | FileSelectHelper::FileSelectHelper(Profile* profile) |
| 134 | : profile_(profile), |
nasko | 35169af | 2016-06-10 18:47:13 | [diff] [blame] | 135 | render_frame_host_(nullptr), |
| 136 | web_contents_(nullptr), |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 137 | select_file_dialog_(), |
[email protected] | 9f054aa1 | 2011-09-29 19:13:45 | [diff] [blame] | 138 | select_file_types_(), |
[email protected] | bfcf1e9 | 2013-07-11 04:37:25 | [diff] [blame] | 139 | dialog_type_(ui::SelectFileDialog::SELECT_OPEN_FILE), |
nasko | 35169af | 2016-06-10 18:47:13 | [diff] [blame] | 140 | dialog_mode_(FileChooserParams::Open) {} |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 141 | |
| 142 | FileSelectHelper::~FileSelectHelper() { |
| 143 | // There may be pending file dialogs, we need to tell them that we've gone |
| 144 | // away so they don't try and call back to us. |
| 145 | if (select_file_dialog_.get()) |
| 146 | select_file_dialog_->ListenerDestroyed(); |
| 147 | |
[email protected] | 600ea40 | 2011-04-12 00:01:51 | [diff] [blame] | 148 | // Stop any pending directory enumeration, prevent a callback, and free |
| 149 | // allocated memory. |
| 150 | std::map<int, ActiveDirectoryEnumeration*>::iterator iter; |
| 151 | for (iter = directory_enumerations_.begin(); |
| 152 | iter != directory_enumerations_.end(); |
| 153 | ++iter) { |
[email protected] | 05a81418 | 2011-04-27 19:50:34 | [diff] [blame] | 154 | iter->second->lister_.reset(); |
[email protected] | 600ea40 | 2011-04-12 00:01:51 | [diff] [blame] | 155 | delete iter->second; |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 156 | } |
| 157 | } |
| 158 | |
[email protected] | 23827ec | 2012-08-10 22:08:08 | [diff] [blame] | 159 | void FileSelectHelper::DirectoryListerDispatchDelegate::OnListFile( |
| 160 | const net::DirectoryLister::DirectoryListerData& data) { |
| 161 | parent_->OnListFile(id_, data); |
| 162 | } |
| 163 | |
| 164 | void FileSelectHelper::DirectoryListerDispatchDelegate::OnListDone(int error) { |
| 165 | parent_->OnListDone(id_, error); |
| 166 | } |
| 167 | |
[email protected] | 650b2d5 | 2013-02-10 03:41:45 | [diff] [blame] | 168 | void FileSelectHelper::FileSelected(const base::FilePath& path, |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 169 | int index, void* params) { |
[email protected] | 53f04c8 | 2012-07-26 02:31:09 | [diff] [blame] | 170 | FileSelectedWithExtraInfo(ui::SelectedFileInfo(path, path), index, params); |
[email protected] | fb11b6a4 | 2012-03-14 07:25:12 | [diff] [blame] | 171 | } |
| 172 | |
| 173 | void FileSelectHelper::FileSelectedWithExtraInfo( |
[email protected] | ddb034b | 2012-06-26 20:31:39 | [diff] [blame] | 174 | const ui::SelectedFileInfo& file, |
[email protected] | fb11b6a4 | 2012-03-14 07:25:12 | [diff] [blame] | 175 | int index, |
| 176 | void* params) { |
hirono | a4a32af2 | 2014-11-07 03:17:29 | [diff] [blame] | 177 | if (IsValidProfile(profile_)) |
| 178 | profile_->set_last_selected_directory(file.file_path.DirName()); |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 179 | |
nasko | 35169af | 2016-06-10 18:47:13 | [diff] [blame] | 180 | if (!render_frame_host_) { |
erikchen | 22de64a3 | 2014-10-10 18:27:43 | [diff] [blame] | 181 | RunFileChooserEnd(); |
| 182 | return; |
| 183 | } |
| 184 | |
[email protected] | 650b2d5 | 2013-02-10 03:41:45 | [diff] [blame] | 185 | const base::FilePath& path = file.local_path; |
[email protected] | 6bedbef | 2013-07-31 06:33:49 | [diff] [blame] | 186 | if (dialog_type_ == ui::SelectFileDialog::SELECT_UPLOAD_FOLDER) { |
nasko | 35169af | 2016-06-10 18:47:13 | [diff] [blame] | 187 | StartNewEnumeration(path, kFileSelectEnumerationId, |
| 188 | render_frame_host_->GetRenderViewHost()); |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 189 | return; |
| 190 | } |
| 191 | |
[email protected] | ddb034b | 2012-06-26 20:31:39 | [diff] [blame] | 192 | std::vector<ui::SelectedFileInfo> files; |
[email protected] | fb11b6a4 | 2012-03-14 07:25:12 | [diff] [blame] | 193 | files.push_back(file); |
[email protected] | 9f054aa1 | 2011-09-29 19:13:45 | [diff] [blame] | 194 | |
jam | 1c5a9149 | 2016-02-24 20:47:53 | [diff] [blame] | 195 | #if defined(OS_MACOSX) |
erikchen | 22de64a3 | 2014-10-10 18:27:43 | [diff] [blame] | 196 | content::BrowserThread::PostTask( |
| 197 | content::BrowserThread::FILE_USER_BLOCKING, |
| 198 | FROM_HERE, |
| 199 | base::Bind(&FileSelectHelper::ProcessSelectedFilesMac, this, files)); |
| 200 | #else |
nasko | f6a80ac | 2016-06-29 02:37:05 | [diff] [blame] | 201 | NotifyRenderFrameHostAndEnd(files); |
jam | 1c5a9149 | 2016-02-24 20:47:53 | [diff] [blame] | 202 | #endif // defined(OS_MACOSX) |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 203 | } |
| 204 | |
[email protected] | 650b2d5 | 2013-02-10 03:41:45 | [diff] [blame] | 205 | void FileSelectHelper::MultiFilesSelected( |
| 206 | const std::vector<base::FilePath>& files, |
| 207 | void* params) { |
[email protected] | ddb034b | 2012-06-26 20:31:39 | [diff] [blame] | 208 | std::vector<ui::SelectedFileInfo> selected_files = |
[email protected] | 53f04c8 | 2012-07-26 02:31:09 | [diff] [blame] | 209 | FilePathListToSelectedFileInfoList(files); |
| 210 | |
[email protected] | fb11b6a4 | 2012-03-14 07:25:12 | [diff] [blame] | 211 | MultiFilesSelectedWithExtraInfo(selected_files, params); |
| 212 | } |
| 213 | |
| 214 | void FileSelectHelper::MultiFilesSelectedWithExtraInfo( |
[email protected] | ddb034b | 2012-06-26 20:31:39 | [diff] [blame] | 215 | const std::vector<ui::SelectedFileInfo>& files, |
[email protected] | fb11b6a4 | 2012-03-14 07:25:12 | [diff] [blame] | 216 | void* params) { |
hirono | a4a32af2 | 2014-11-07 03:17:29 | [diff] [blame] | 217 | if (!files.empty() && IsValidProfile(profile_)) |
[email protected] | 53f04c8 | 2012-07-26 02:31:09 | [diff] [blame] | 218 | profile_->set_last_selected_directory(files[0].file_path.DirName()); |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 219 | |
jam | 1c5a9149 | 2016-02-24 20:47:53 | [diff] [blame] | 220 | #if defined(OS_MACOSX) |
erikchen | 22de64a3 | 2014-10-10 18:27:43 | [diff] [blame] | 221 | content::BrowserThread::PostTask( |
| 222 | content::BrowserThread::FILE_USER_BLOCKING, |
| 223 | FROM_HERE, |
| 224 | base::Bind(&FileSelectHelper::ProcessSelectedFilesMac, this, files)); |
| 225 | #else |
nasko | f6a80ac | 2016-06-29 02:37:05 | [diff] [blame] | 226 | NotifyRenderFrameHostAndEnd(files); |
jam | 1c5a9149 | 2016-02-24 20:47:53 | [diff] [blame] | 227 | #endif // defined(OS_MACOSX) |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 228 | } |
| 229 | |
| 230 | void FileSelectHelper::FileSelectionCanceled(void* params) { |
nasko | f6a80ac | 2016-06-29 02:37:05 | [diff] [blame] | 231 | NotifyRenderFrameHostAndEnd(std::vector<ui::SelectedFileInfo>()); |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 232 | } |
| 233 | |
[email protected] | 650b2d5 | 2013-02-10 03:41:45 | [diff] [blame] | 234 | void FileSelectHelper::StartNewEnumeration(const base::FilePath& path, |
[email protected] | 600ea40 | 2011-04-12 00:01:51 | [diff] [blame] | 235 | int request_id, |
| 236 | RenderViewHost* render_view_host) { |
dcheng | 4af4858 | 2016-04-19 00:29:35 | [diff] [blame] | 237 | std::unique_ptr<ActiveDirectoryEnumeration> entry( |
| 238 | new ActiveDirectoryEnumeration); |
[email protected] | 600ea40 | 2011-04-12 00:01:51 | [diff] [blame] | 239 | entry->rvh_ = render_view_host; |
| 240 | entry->delegate_.reset(new DirectoryListerDispatchDelegate(this, request_id)); |
olli.syrjala | 3396ad3 | 2015-04-24 07:56:41 | [diff] [blame] | 241 | entry->lister_.reset(new net::DirectoryLister( |
| 242 | path, net::DirectoryLister::NO_SORT_RECURSIVE, entry->delegate_.get())); |
shahriar.rostami | a8c06daf | 2016-02-12 00:07:04 | [diff] [blame] | 243 | if (!entry->lister_->Start(base::WorkerPool::GetTaskRunner(true).get())) { |
[email protected] | 600ea40 | 2011-04-12 00:01:51 | [diff] [blame] | 244 | if (request_id == kFileSelectEnumerationId) |
| 245 | FileSelectionCanceled(NULL); |
| 246 | else |
| 247 | render_view_host->DirectoryEnumerationFinished(request_id, |
| 248 | entry->results_); |
| 249 | } else { |
| 250 | directory_enumerations_[request_id] = entry.release(); |
| 251 | } |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 252 | } |
| 253 | |
| 254 | void FileSelectHelper::OnListFile( |
[email protected] | 600ea40 | 2011-04-12 00:01:51 | [diff] [blame] | 255 | int id, |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 256 | const net::DirectoryLister::DirectoryListerData& data) { |
[email protected] | 600ea40 | 2011-04-12 00:01:51 | [diff] [blame] | 257 | ActiveDirectoryEnumeration* entry = directory_enumerations_[id]; |
| 258 | |
[email protected] | 2c718a0d | 2014-03-24 02:42:22 | [diff] [blame] | 259 | // Directory upload only cares about files. |
[email protected] | 25a4c1c | 2013-06-08 04:53:36 | [diff] [blame] | 260 | if (data.info.IsDirectory()) |
[email protected] | 2c718a0d | 2014-03-24 02:42:22 | [diff] [blame] | 261 | return; |
| 262 | |
| 263 | entry->results_.push_back(data.path); |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 264 | } |
| 265 | |
[email protected] | 600ea40 | 2011-04-12 00:01:51 | [diff] [blame] | 266 | void FileSelectHelper::OnListDone(int id, int error) { |
| 267 | // This entry needs to be cleaned up when this function is done. |
dcheng | 4af4858 | 2016-04-19 00:29:35 | [diff] [blame] | 268 | std::unique_ptr<ActiveDirectoryEnumeration> entry( |
| 269 | directory_enumerations_[id]); |
[email protected] | 600ea40 | 2011-04-12 00:01:51 | [diff] [blame] | 270 | directory_enumerations_.erase(id); |
| 271 | if (!entry->rvh_) |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 272 | return; |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 273 | if (error) { |
| 274 | FileSelectionCanceled(NULL); |
| 275 | return; |
| 276 | } |
[email protected] | fb11b6a4 | 2012-03-14 07:25:12 | [diff] [blame] | 277 | |
[email protected] | ddb034b | 2012-06-26 20:31:39 | [diff] [blame] | 278 | std::vector<ui::SelectedFileInfo> selected_files = |
[email protected] | 53f04c8 | 2012-07-26 02:31:09 | [diff] [blame] | 279 | FilePathListToSelectedFileInfoList(entry->results_); |
[email protected] | fb11b6a4 | 2012-03-14 07:25:12 | [diff] [blame] | 280 | |
hirono | 275bfbe | 2014-10-21 01:46:17 | [diff] [blame] | 281 | if (id == kFileSelectEnumerationId) { |
nasko | f6a80ac | 2016-06-29 02:37:05 | [diff] [blame] | 282 | NotifyRenderFrameHostAndEnd(selected_files); |
hirono | 275bfbe | 2014-10-21 01:46:17 | [diff] [blame] | 283 | } else { |
[email protected] | 600ea40 | 2011-04-12 00:01:51 | [diff] [blame] | 284 | entry->rvh_->DirectoryEnumerationFinished(id, entry->results_); |
hirono | 275bfbe | 2014-10-21 01:46:17 | [diff] [blame] | 285 | EnumerateDirectoryEnd(); |
| 286 | } |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 287 | } |
| 288 | |
nasko | f6a80ac | 2016-06-29 02:37:05 | [diff] [blame] | 289 | void FileSelectHelper::NotifyRenderFrameHostAndEnd( |
erikchen | 22de64a3 | 2014-10-10 18:27:43 | [diff] [blame] | 290 | const std::vector<ui::SelectedFileInfo>& files) { |
nasko | 35169af | 2016-06-10 18:47:13 | [diff] [blame] | 291 | if (!render_frame_host_) { |
hirono | 275bfbe | 2014-10-21 01:46:17 | [diff] [blame] | 292 | RunFileChooserEnd(); |
| 293 | return; |
| 294 | } |
| 295 | |
| 296 | #if defined(OS_CHROMEOS) |
| 297 | if (!files.empty()) { |
hirono | a4a32af2 | 2014-11-07 03:17:29 | [diff] [blame] | 298 | if (!IsValidProfile(profile_)) { |
| 299 | RunFileChooserEnd(); |
| 300 | return; |
| 301 | } |
hirono | 275bfbe | 2014-10-21 01:46:17 | [diff] [blame] | 302 | // Converts |files| into FileChooserFileInfo with handling of non-native |
| 303 | // files. |
nasko | 35169af | 2016-06-10 18:47:13 | [diff] [blame] | 304 | content::SiteInstance* site_instance = |
| 305 | render_frame_host_->GetSiteInstance(); |
rdevlin.cronin | bfbc11d9 | 2015-06-12 23:26:08 | [diff] [blame] | 306 | storage::FileSystemContext* file_system_context = |
nasko | 35169af | 2016-06-10 18:47:13 | [diff] [blame] | 307 | content::BrowserContext::GetStoragePartition(profile_, site_instance) |
| 308 | ->GetFileSystemContext(); |
hirono | 275bfbe | 2014-10-21 01:46:17 | [diff] [blame] | 309 | file_manager::util::ConvertSelectedFileInfoListToFileChooserFileInfoList( |
nasko | 35169af | 2016-06-10 18:47:13 | [diff] [blame] | 310 | file_system_context, site_instance->GetSiteURL(), files, |
nasko | f6a80ac | 2016-06-29 02:37:05 | [diff] [blame] | 311 | base::Bind( |
| 312 | &FileSelectHelper::NotifyRenderFrameHostAndEndAfterConversion, |
| 313 | this)); |
hirono | 275bfbe | 2014-10-21 01:46:17 | [diff] [blame] | 314 | return; |
| 315 | } |
| 316 | #endif // defined(OS_CHROMEOS) |
| 317 | |
| 318 | std::vector<content::FileChooserFileInfo> chooser_files; |
| 319 | for (const auto& file : files) { |
| 320 | content::FileChooserFileInfo chooser_file; |
| 321 | chooser_file.file_path = file.local_path; |
| 322 | chooser_file.display_name = file.display_name; |
| 323 | chooser_files.push_back(chooser_file); |
| 324 | } |
hirono | cf350cb | 2014-10-23 08:37:18 | [diff] [blame] | 325 | |
nasko | f6a80ac | 2016-06-29 02:37:05 | [diff] [blame] | 326 | NotifyRenderFrameHostAndEndAfterConversion(chooser_files); |
hirono | cf350cb | 2014-10-23 08:37:18 | [diff] [blame] | 327 | } |
| 328 | |
nasko | f6a80ac | 2016-06-29 02:37:05 | [diff] [blame] | 329 | void FileSelectHelper::NotifyRenderFrameHostAndEndAfterConversion( |
hirono | cf350cb | 2014-10-23 08:37:18 | [diff] [blame] | 330 | const std::vector<content::FileChooserFileInfo>& list) { |
nasko | 35169af | 2016-06-10 18:47:13 | [diff] [blame] | 331 | if (render_frame_host_) |
| 332 | render_frame_host_->FilesSelectedInChooser(list, dialog_mode_); |
erikchen | 22de64a3 | 2014-10-10 18:27:43 | [diff] [blame] | 333 | |
| 334 | // No members should be accessed from here on. |
| 335 | RunFileChooserEnd(); |
| 336 | } |
| 337 | |
| 338 | void FileSelectHelper::DeleteTemporaryFiles() { |
| 339 | BrowserThread::PostTask(BrowserThread::FILE, |
| 340 | FROM_HERE, |
| 341 | base::Bind(&DeleteFiles, temporary_files_)); |
| 342 | temporary_files_.clear(); |
| 343 | } |
| 344 | |
nasko | f6a80ac | 2016-06-29 02:37:05 | [diff] [blame] | 345 | void FileSelectHelper::CleanUp() { |
lazyboy | 1cf60af | 2015-05-15 22:17:05 | [diff] [blame] | 346 | if (!temporary_files_.empty()) { |
| 347 | DeleteTemporaryFiles(); |
| 348 | |
| 349 | // Now that the temporary files have been scheduled for deletion, there |
| 350 | // is no longer any reason to keep this instance around. |
| 351 | Release(); |
| 352 | } |
| 353 | } |
| 354 | |
dcheng | 4af4858 | 2016-04-19 00:29:35 | [diff] [blame] | 355 | std::unique_ptr<ui::SelectFileDialog::FileTypeInfo> |
[email protected] | 92f5408 | 2012-07-31 01:43:14 | [diff] [blame] | 356 | FileSelectHelper::GetFileTypesFromAcceptType( |
[email protected] | d2065e06 | 2013-12-12 23:49:52 | [diff] [blame] | 357 | const std::vector<base::string16>& accept_types) { |
dcheng | 4af4858 | 2016-04-19 00:29:35 | [diff] [blame] | 358 | std::unique_ptr<ui::SelectFileDialog::FileTypeInfo> base_file_type( |
[email protected] | 479cce78 | 2012-09-15 20:15:53 | [diff] [blame] | 359 | new ui::SelectFileDialog::FileTypeInfo()); |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 360 | if (accept_types.empty()) |
dcheng | e73d8520c | 2015-12-27 01:19:09 | [diff] [blame] | 361 | return base_file_type; |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 362 | |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 363 | // Create FileTypeInfo and pre-allocate for the first extension list. |
dcheng | 4af4858 | 2016-04-19 00:29:35 | [diff] [blame] | 364 | std::unique_ptr<ui::SelectFileDialog::FileTypeInfo> file_type( |
[email protected] | 479cce78 | 2012-09-15 20:15:53 | [diff] [blame] | 365 | new ui::SelectFileDialog::FileTypeInfo(*base_file_type)); |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 366 | file_type->include_all_files = true; |
| 367 | file_type->extensions.resize(1); |
[email protected] | 650b2d5 | 2013-02-10 03:41:45 | [diff] [blame] | 368 | std::vector<base::FilePath::StringType>* extensions = |
| 369 | &file_type->extensions.back(); |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 370 | |
[email protected] | f9a4c41a | 2012-05-30 00:05:32 | [diff] [blame] | 371 | // Find the corresponding extensions. |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 372 | int valid_type_count = 0; |
| 373 | int description_id = 0; |
[email protected] | 3314c2b1 | 2011-11-02 08:05:46 | [diff] [blame] | 374 | for (size_t i = 0; i < accept_types.size(); ++i) { |
[email protected] | 74f778e | 2014-03-14 21:11:46 | [diff] [blame] | 375 | std::string ascii_type = base::UTF16ToASCII(accept_types[i]); |
[email protected] | f9a4c41a | 2012-05-30 00:05:32 | [diff] [blame] | 376 | if (!IsAcceptTypeValid(ascii_type)) |
| 377 | continue; |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 378 | |
| 379 | size_t old_extension_size = extensions->size(); |
[email protected] | f9a4c41a | 2012-05-30 00:05:32 | [diff] [blame] | 380 | if (ascii_type[0] == '.') { |
| 381 | // If the type starts with a period it is assumed to be a file extension |
| 382 | // so we just have to add it to the list. |
[email protected] | 650b2d5 | 2013-02-10 03:41:45 | [diff] [blame] | 383 | base::FilePath::StringType ext(ascii_type.begin(), ascii_type.end()); |
[email protected] | f9a4c41a | 2012-05-30 00:05:32 | [diff] [blame] | 384 | extensions->push_back(ext.substr(1)); |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 385 | } else { |
[email protected] | 4a66fa0e | 2012-09-10 06:45:20 | [diff] [blame] | 386 | if (ascii_type == "image/*") |
| 387 | description_id = IDS_IMAGE_FILES; |
| 388 | else if (ascii_type == "audio/*") |
| 389 | description_id = IDS_AUDIO_FILES; |
| 390 | else if (ascii_type == "video/*") |
| 391 | description_id = IDS_VIDEO_FILES; |
| 392 | |
[email protected] | f9a4c41a | 2012-05-30 00:05:32 | [diff] [blame] | 393 | net::GetExtensionsForMimeType(ascii_type, extensions); |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 394 | } |
| 395 | |
| 396 | if (extensions->size() > old_extension_size) |
| 397 | valid_type_count++; |
| 398 | } |
| 399 | |
[email protected] | cbcd12ed | 2010-12-16 23:42:57 | [diff] [blame] | 400 | // If no valid extension is added, bail out. |
| 401 | if (valid_type_count == 0) |
dcheng | e73d8520c | 2015-12-27 01:19:09 | [diff] [blame] | 402 | return base_file_type; |
[email protected] | cbcd12ed | 2010-12-16 23:42:57 | [diff] [blame] | 403 | |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 404 | // Use a generic description "Custom Files" if either of the following is |
| 405 | // true: |
| 406 | // 1) There're multiple types specified, like "audio/*,video/*" |
| 407 | // 2) There're multiple extensions for a MIME type without parameter, like |
| 408 | // "ehtml,shtml,htm,html" for "text/html". On Windows, the select file |
| 409 | // dialog uses the first extension in the list to form the description, |
| 410 | // like "EHTML Files". This is not what we want. |
| 411 | if (valid_type_count > 1 || |
| 412 | (valid_type_count == 1 && description_id == 0 && extensions->size() > 1)) |
| 413 | description_id = IDS_CUSTOM_FILES; |
| 414 | |
| 415 | if (description_id) { |
| 416 | file_type->extension_description_overrides.push_back( |
| 417 | l10n_util::GetStringUTF16(description_id)); |
| 418 | } |
| 419 | |
dcheng | e73d8520c | 2015-12-27 01:19:09 | [diff] [blame] | 420 | return file_type; |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 421 | } |
| 422 | |
[email protected] | 33f8ad5 | 2012-05-22 18:10:13 | [diff] [blame] | 423 | // static |
nasko | 35169af | 2016-06-10 18:47:13 | [diff] [blame] | 424 | void FileSelectHelper::RunFileChooser( |
| 425 | content::RenderFrameHost* render_frame_host, |
| 426 | const FileChooserParams& params) { |
| 427 | Profile* profile = Profile::FromBrowserContext( |
| 428 | render_frame_host->GetProcess()->GetBrowserContext()); |
[email protected] | 33f8ad5 | 2012-05-22 18:10:13 | [diff] [blame] | 429 | // FileSelectHelper will keep itself alive until it sends the result message. |
| 430 | scoped_refptr<FileSelectHelper> file_select_helper( |
| 431 | new FileSelectHelper(profile)); |
asanka | 1cbf4981 | 2015-11-03 18:28:15 | [diff] [blame] | 432 | file_select_helper->RunFileChooser( |
nasko | 35169af | 2016-06-10 18:47:13 | [diff] [blame] | 433 | render_frame_host, |
dcheng | 4af4858 | 2016-04-19 00:29:35 | [diff] [blame] | 434 | base::WrapUnique(new content::FileChooserParams(params))); |
[email protected] | 33f8ad5 | 2012-05-22 18:10:13 | [diff] [blame] | 435 | } |
| 436 | |
| 437 | // static |
| 438 | void FileSelectHelper::EnumerateDirectory(content::WebContents* tab, |
| 439 | int request_id, |
[email protected] | 650b2d5 | 2013-02-10 03:41:45 | [diff] [blame] | 440 | const base::FilePath& path) { |
[email protected] | 33f8ad5 | 2012-05-22 18:10:13 | [diff] [blame] | 441 | Profile* profile = Profile::FromBrowserContext(tab->GetBrowserContext()); |
| 442 | // FileSelectHelper will keep itself alive until it sends the result message. |
| 443 | scoped_refptr<FileSelectHelper> file_select_helper( |
| 444 | new FileSelectHelper(profile)); |
| 445 | file_select_helper->EnumerateDirectory( |
| 446 | request_id, tab->GetRenderViewHost(), path); |
| 447 | } |
| 448 | |
dcheng | 4af4858 | 2016-04-19 00:29:35 | [diff] [blame] | 449 | void FileSelectHelper::RunFileChooser( |
nasko | 35169af | 2016-06-10 18:47:13 | [diff] [blame] | 450 | content::RenderFrameHost* render_frame_host, |
dcheng | 4af4858 | 2016-04-19 00:29:35 | [diff] [blame] | 451 | std::unique_ptr<FileChooserParams> params) { |
nasko | 35169af | 2016-06-10 18:47:13 | [diff] [blame] | 452 | DCHECK(!render_frame_host_); |
[email protected] | ea049a0 | 2011-12-25 21:37:09 | [diff] [blame] | 453 | DCHECK(!web_contents_); |
asanka | 1cbf4981 | 2015-11-03 18:28:15 | [diff] [blame] | 454 | DCHECK(params->default_file_name.empty() || |
| 455 | params->mode == FileChooserParams::Save) |
asanka | ee261f3 | 2015-10-27 21:04:28 | [diff] [blame] | 456 | << "The default_file_name parameter should only be specified for Save " |
| 457 | "file choosers"; |
asanka | 1cbf4981 | 2015-11-03 18:28:15 | [diff] [blame] | 458 | DCHECK(params->default_file_name == params->default_file_name.BaseName()) |
asanka | ee261f3 | 2015-10-27 21:04:28 | [diff] [blame] | 459 | << "The default_file_name parameter should not contain path separators"; |
| 460 | |
nasko | 35169af | 2016-06-10 18:47:13 | [diff] [blame] | 461 | render_frame_host_ = render_frame_host; |
| 462 | web_contents_ = WebContents::FromRenderFrameHost(render_frame_host); |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 463 | notification_registrar_.RemoveAll(); |
lazyboy | 1cf60af | 2015-05-15 22:17:05 | [diff] [blame] | 464 | content::WebContentsObserver::Observe(web_contents_); |
[email protected] | 43211582 | 2011-07-10 15:52:27 | [diff] [blame] | 465 | notification_registrar_.Add( |
avi | f9ab5d94 | 2015-10-15 14:05:44 | [diff] [blame] | 466 | this, content::NOTIFICATION_RENDER_WIDGET_HOST_DESTROYED, |
nasko | 35169af | 2016-06-10 18:47:13 | [diff] [blame] | 467 | content::Source<RenderWidgetHost>( |
| 468 | render_frame_host_->GetRenderViewHost()->GetWidget())); |
[email protected] | 9f054aa1 | 2011-09-29 19:13:45 | [diff] [blame] | 469 | |
| 470 | BrowserThread::PostTask( |
| 471 | BrowserThread::FILE, FROM_HERE, |
asanka | 1cbf4981 | 2015-11-03 18:28:15 | [diff] [blame] | 472 | base::Bind(&FileSelectHelper::GetFileTypesOnFileThread, this, |
| 473 | base::Passed(¶ms))); |
[email protected] | 9f054aa1 | 2011-09-29 19:13:45 | [diff] [blame] | 474 | |
| 475 | // Because this class returns notifications to the RenderViewHost, it is |
| 476 | // difficult for callers to know how long to keep a reference to this |
| 477 | // instance. We AddRef() here to keep the instance alive after we return |
| 478 | // to the caller, until the last callback is received from the file dialog. |
| 479 | // At that point, we must call RunFileChooserEnd(). |
| 480 | AddRef(); |
| 481 | } |
| 482 | |
asanka | 1cbf4981 | 2015-11-03 18:28:15 | [diff] [blame] | 483 | void FileSelectHelper::GetFileTypesOnFileThread( |
dcheng | 4af4858 | 2016-04-19 00:29:35 | [diff] [blame] | 484 | std::unique_ptr<FileChooserParams> params) { |
asanka | 1cbf4981 | 2015-11-03 18:28:15 | [diff] [blame] | 485 | select_file_types_ = GetFileTypesFromAcceptType(params->accept_types); |
hirono | 4d4d339 | 2016-02-04 05:03:50 | [diff] [blame] | 486 | select_file_types_->allowed_paths = |
| 487 | params->need_local_path ? ui::SelectFileDialog::FileTypeInfo::NATIVE_PATH |
| 488 | : ui::SelectFileDialog::FileTypeInfo::ANY_PATH; |
[email protected] | 9f054aa1 | 2011-09-29 19:13:45 | [diff] [blame] | 489 | |
| 490 | BrowserThread::PostTask( |
| 491 | BrowserThread::UI, FROM_HERE, |
asanka | 1cbf4981 | 2015-11-03 18:28:15 | [diff] [blame] | 492 | base::Bind(&FileSelectHelper::GetSanitizedFilenameOnUIThread, this, |
| 493 | base::Passed(¶ms))); |
[email protected] | 9f054aa1 | 2011-09-29 19:13:45 | [diff] [blame] | 494 | } |
| 495 | |
asanka | 1cbf4981 | 2015-11-03 18:28:15 | [diff] [blame] | 496 | void FileSelectHelper::GetSanitizedFilenameOnUIThread( |
dcheng | 4af4858 | 2016-04-19 00:29:35 | [diff] [blame] | 497 | std::unique_ptr<FileChooserParams> params) { |
asanka | 1cbf4981 | 2015-11-03 18:28:15 | [diff] [blame] | 498 | base::FilePath default_file_path = profile_->last_selected_directory().Append( |
| 499 | GetSanitizedFileName(params->default_file_name)); |
asanka | 1cbf4981 | 2015-11-03 18:28:15 | [diff] [blame] | 500 | #if defined(FULL_SAFE_BROWSING) |
asanka | ef1f3872 | 2016-05-20 16:17:41 | [diff] [blame] | 501 | CheckDownloadRequestWithSafeBrowsing(default_file_path, std::move(params)); |
| 502 | #else |
dcheng | e73d8520c | 2015-12-27 01:19:09 | [diff] [blame] | 503 | RunFileChooserOnUIThread(default_file_path, std::move(params)); |
asanka | ef1f3872 | 2016-05-20 16:17:41 | [diff] [blame] | 504 | #endif |
asanka | 1cbf4981 | 2015-11-03 18:28:15 | [diff] [blame] | 505 | } |
| 506 | |
| 507 | #if defined(FULL_SAFE_BROWSING) |
asanka | ef1f3872 | 2016-05-20 16:17:41 | [diff] [blame] | 508 | void FileSelectHelper::CheckDownloadRequestWithSafeBrowsing( |
| 509 | const base::FilePath& default_file_path, |
| 510 | std::unique_ptr<FileChooserParams> params) { |
| 511 | safe_browsing::SafeBrowsingService* sb_service = |
| 512 | g_browser_process->safe_browsing_service(); |
| 513 | |
| 514 | if (!sb_service || !sb_service->download_protection_service() || |
| 515 | !sb_service->download_protection_service()->enabled()) { |
| 516 | RunFileChooserOnUIThread(default_file_path, std::move(params)); |
asanka | 1cbf4981 | 2015-11-03 18:28:15 | [diff] [blame] | 517 | return; |
| 518 | } |
| 519 | |
asanka | ef1f3872 | 2016-05-20 16:17:41 | [diff] [blame] | 520 | std::vector<base::FilePath::StringType> alternate_extensions; |
| 521 | if (select_file_types_) { |
| 522 | for (const auto& extensions_list : select_file_types_->extensions) { |
| 523 | for (const auto& extension_in_list : extensions_list) { |
| 524 | base::FilePath::StringType extension = |
| 525 | default_file_path.ReplaceExtension(extension_in_list) |
| 526 | .FinalExtension(); |
| 527 | alternate_extensions.push_back(extension); |
| 528 | } |
| 529 | } |
| 530 | } |
| 531 | |
| 532 | GURL requestor_url = params->requestor; |
| 533 | sb_service->download_protection_service()->CheckPPAPIDownloadRequest( |
| 534 | requestor_url, default_file_path, alternate_extensions, |
| 535 | base::Bind(&InterpretSafeBrowsingVerdict, |
| 536 | base::Bind(&FileSelectHelper::ProceedWithSafeBrowsingVerdict, |
| 537 | this, default_file_path, base::Passed(¶ms)))); |
| 538 | } |
| 539 | |
| 540 | void FileSelectHelper::ProceedWithSafeBrowsingVerdict( |
| 541 | const base::FilePath& default_file_path, |
| 542 | std::unique_ptr<content::FileChooserParams> params, |
| 543 | bool allowed_by_safe_browsing) { |
| 544 | if (!allowed_by_safe_browsing) { |
nasko | f6a80ac | 2016-06-29 02:37:05 | [diff] [blame] | 545 | NotifyRenderFrameHostAndEnd(std::vector<ui::SelectedFileInfo>()); |
asanka | ef1f3872 | 2016-05-20 16:17:41 | [diff] [blame] | 546 | return; |
| 547 | } |
| 548 | RunFileChooserOnUIThread(default_file_path, std::move(params)); |
asanka | 1cbf4981 | 2015-11-03 18:28:15 | [diff] [blame] | 549 | } |
| 550 | #endif |
| 551 | |
[email protected] | 9f054aa1 | 2011-09-29 19:13:45 | [diff] [blame] | 552 | void FileSelectHelper::RunFileChooserOnUIThread( |
asanka | 1cbf4981 | 2015-11-03 18:28:15 | [diff] [blame] | 553 | const base::FilePath& default_file_path, |
dcheng | 4af4858 | 2016-04-19 00:29:35 | [diff] [blame] | 554 | std::unique_ptr<FileChooserParams> params) { |
asanka | 1cbf4981 | 2015-11-03 18:28:15 | [diff] [blame] | 555 | DCHECK(params); |
nasko | 35169af | 2016-06-10 18:47:13 | [diff] [blame] | 556 | if (!render_frame_host_ || !web_contents_ || !IsValidProfile(profile_) || |
| 557 | !web_contents_->GetNativeView()) { |
[email protected] | b95b08d | 2011-12-15 20:23:16 | [diff] [blame] | 558 | // If the renderer was destroyed before we started, just cancel the |
| 559 | // operation. |
| 560 | RunFileChooserEnd(); |
[email protected] | 9f054aa1 | 2011-09-29 19:13:45 | [diff] [blame] | 561 | return; |
[email protected] | b95b08d | 2011-12-15 20:23:16 | [diff] [blame] | 562 | } |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 563 | |
[email protected] | 92f5408 | 2012-07-31 01:43:14 | [diff] [blame] | 564 | select_file_dialog_ = ui::SelectFileDialog::Create( |
[email protected] | 6e1fcd1 | 2012-07-02 17:14:20 | [diff] [blame] | 565 | this, new ChromeSelectFilePolicy(web_contents_)); |
dcheng | 319a2195 | 2014-08-26 22:52:40 | [diff] [blame] | 566 | if (!select_file_dialog_.get()) |
[email protected] | 6910cb24 | 2014-03-20 14:33:01 | [diff] [blame] | 567 | return; |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 568 | |
asanka | 1cbf4981 | 2015-11-03 18:28:15 | [diff] [blame] | 569 | dialog_mode_ = params->mode; |
| 570 | switch (params->mode) { |
[email protected] | 33f8ad5 | 2012-05-22 18:10:13 | [diff] [blame] | 571 | case FileChooserParams::Open: |
[email protected] | 92f5408 | 2012-07-31 01:43:14 | [diff] [blame] | 572 | dialog_type_ = ui::SelectFileDialog::SELECT_OPEN_FILE; |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 573 | break; |
[email protected] | 33f8ad5 | 2012-05-22 18:10:13 | [diff] [blame] | 574 | case FileChooserParams::OpenMultiple: |
[email protected] | 92f5408 | 2012-07-31 01:43:14 | [diff] [blame] | 575 | dialog_type_ = ui::SelectFileDialog::SELECT_OPEN_MULTI_FILE; |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 576 | break; |
[email protected] | 6bedbef | 2013-07-31 06:33:49 | [diff] [blame] | 577 | case FileChooserParams::UploadFolder: |
| 578 | dialog_type_ = ui::SelectFileDialog::SELECT_UPLOAD_FOLDER; |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 579 | break; |
[email protected] | 33f8ad5 | 2012-05-22 18:10:13 | [diff] [blame] | 580 | case FileChooserParams::Save: |
[email protected] | 92f5408 | 2012-07-31 01:43:14 | [diff] [blame] | 581 | dialog_type_ = ui::SelectFileDialog::SELECT_SAVEAS_FILE; |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 582 | break; |
| 583 | default: |
[email protected] | 92f5408 | 2012-07-31 01:43:14 | [diff] [blame] | 584 | // Prevent warning. |
| 585 | dialog_type_ = ui::SelectFileDialog::SELECT_OPEN_FILE; |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 586 | NOTREACHED(); |
| 587 | } |
[email protected] | 4e9149a | 2012-08-15 20:43:59 | [diff] [blame] | 588 | |
nasko | 35169af | 2016-06-10 18:47:13 | [diff] [blame] | 589 | gfx::NativeWindow owning_window = |
| 590 | platform_util::GetTopLevel(web_contents_->GetNativeView()); |
[email protected] | d989891 | 2011-04-15 21:10:00 | [diff] [blame] | 591 | |
[email protected] | 2d02a200 | 2012-09-18 21:47:56 | [diff] [blame] | 592 | #if defined(OS_ANDROID) |
| 593 | // Android needs the original MIME types and an additional capture value. |
[email protected] | d2065e06 | 2013-12-12 23:49:52 | [diff] [blame] | 594 | std::pair<std::vector<base::string16>, bool> accept_types = |
asanka | 1cbf4981 | 2015-11-03 18:28:15 | [diff] [blame] | 595 | std::make_pair(params->accept_types, params->capture); |
[email protected] | 2d02a200 | 2012-09-18 21:47:56 | [diff] [blame] | 596 | #endif |
| 597 | |
[email protected] | 9f054aa1 | 2011-09-29 19:13:45 | [diff] [blame] | 598 | select_file_dialog_->SelectFile( |
asanka | 1cbf4981 | 2015-11-03 18:28:15 | [diff] [blame] | 599 | dialog_type_, params->title, default_file_path, select_file_types_.get(), |
[email protected] | 007b3f8 | 2013-04-09 08:46:45 | [diff] [blame] | 600 | select_file_types_.get() && !select_file_types_->extensions.empty() |
| 601 | ? 1 |
| 602 | : 0, // 1-based index of default extension to show. |
| 603 | base::FilePath::StringType(), |
[email protected] | 9f054aa1 | 2011-09-29 19:13:45 | [diff] [blame] | 604 | owning_window, |
[email protected] | b8452fa | 2012-06-15 01:41:41 | [diff] [blame] | 605 | #if defined(OS_ANDROID) |
[email protected] | 2d02a200 | 2012-09-18 21:47:56 | [diff] [blame] | 606 | &accept_types); |
[email protected] | b8452fa | 2012-06-15 01:41:41 | [diff] [blame] | 607 | #else |
[email protected] | 9f054aa1 | 2011-09-29 19:13:45 | [diff] [blame] | 608 | NULL); |
[email protected] | b8452fa | 2012-06-15 01:41:41 | [diff] [blame] | 609 | #endif |
[email protected] | 9f054aa1 | 2011-09-29 19:13:45 | [diff] [blame] | 610 | |
| 611 | select_file_types_.reset(); |
| 612 | } |
| 613 | |
| 614 | // This method is called when we receive the last callback from the file |
| 615 | // chooser dialog. Perform any cleanup and release the reference we added |
| 616 | // in RunFileChooser(). |
| 617 | void FileSelectHelper::RunFileChooserEnd() { |
erikchen | 22de64a3 | 2014-10-10 18:27:43 | [diff] [blame] | 618 | // If there are temporary files, then this instance needs to stick around |
| 619 | // until web_contents_ is destroyed, so that this instance can delete the |
| 620 | // temporary files. |
| 621 | if (!temporary_files_.empty()) |
| 622 | return; |
| 623 | |
nasko | 35169af | 2016-06-10 18:47:13 | [diff] [blame] | 624 | render_frame_host_ = nullptr; |
| 625 | web_contents_ = nullptr; |
[email protected] | 9f054aa1 | 2011-09-29 19:13:45 | [diff] [blame] | 626 | Release(); |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 627 | } |
| 628 | |
[email protected] | 600ea40 | 2011-04-12 00:01:51 | [diff] [blame] | 629 | void FileSelectHelper::EnumerateDirectory(int request_id, |
| 630 | RenderViewHost* render_view_host, |
[email protected] | 650b2d5 | 2013-02-10 03:41:45 | [diff] [blame] | 631 | const base::FilePath& path) { |
[email protected] | 9f054aa1 | 2011-09-29 19:13:45 | [diff] [blame] | 632 | // Because this class returns notifications to the RenderViewHost, it is |
| 633 | // difficult for callers to know how long to keep a reference to this |
| 634 | // instance. We AddRef() here to keep the instance alive after we return |
| 635 | // to the caller, until the last callback is received from the enumeration |
| 636 | // code. At that point, we must call EnumerateDirectoryEnd(). |
| 637 | AddRef(); |
[email protected] | 600ea40 | 2011-04-12 00:01:51 | [diff] [blame] | 638 | StartNewEnumeration(path, request_id, render_view_host); |
| 639 | } |
| 640 | |
[email protected] | 9f054aa1 | 2011-09-29 19:13:45 | [diff] [blame] | 641 | // This method is called when we receive the last callback from the enumeration |
| 642 | // code. Perform any cleanup and release the reference we added in |
| 643 | // EnumerateDirectory(). |
| 644 | void FileSelectHelper::EnumerateDirectoryEnd() { |
| 645 | Release(); |
| 646 | } |
| 647 | |
[email protected] | 43211582 | 2011-07-10 15:52:27 | [diff] [blame] | 648 | void FileSelectHelper::Observe(int type, |
[email protected] | 6c2381d | 2011-10-19 02:52:53 | [diff] [blame] | 649 | const content::NotificationSource& source, |
| 650 | const content::NotificationDetails& details) { |
thestig | 5283579 | 2016-06-13 21:08:09 | [diff] [blame] | 651 | DCHECK_EQ(content::NOTIFICATION_RENDER_WIDGET_HOST_DESTROYED, type); |
thestig | 5283579 | 2016-06-13 21:08:09 | [diff] [blame] | 652 | render_frame_host_ = nullptr; |
[email protected] | ba70d08 | 2010-09-10 16:54:49 | [diff] [blame] | 653 | } |
[email protected] | f9a4c41a | 2012-05-30 00:05:32 | [diff] [blame] | 654 | |
nasko | f6a80ac | 2016-06-29 02:37:05 | [diff] [blame] | 655 | void FileSelectHelper::RenderFrameHostChanged( |
| 656 | content::RenderFrameHost* old_host, |
| 657 | content::RenderFrameHost* new_host) { |
nasko | 5e61b75f | 2016-07-01 22:35:05 | [diff] [blame^] | 658 | if (old_host == render_frame_host_) |
| 659 | render_frame_host_ = nullptr; |
nasko | f6a80ac | 2016-06-29 02:37:05 | [diff] [blame] | 660 | } |
| 661 | |
| 662 | void FileSelectHelper::RenderFrameDeleted( |
| 663 | content::RenderFrameHost* render_frame_host) { |
nasko | 5e61b75f | 2016-07-01 22:35:05 | [diff] [blame^] | 664 | if (render_frame_host == render_frame_host_) |
| 665 | render_frame_host_ = nullptr; |
lazyboy | 1cf60af | 2015-05-15 22:17:05 | [diff] [blame] | 666 | } |
| 667 | |
| 668 | void FileSelectHelper::WebContentsDestroyed() { |
nasko | f6a80ac | 2016-06-29 02:37:05 | [diff] [blame] | 669 | render_frame_host_ = nullptr; |
lazyboy | 1cf60af | 2015-05-15 22:17:05 | [diff] [blame] | 670 | web_contents_ = nullptr; |
nasko | f6a80ac | 2016-06-29 02:37:05 | [diff] [blame] | 671 | CleanUp(); |
lazyboy | 1cf60af | 2015-05-15 22:17:05 | [diff] [blame] | 672 | } |
| 673 | |
[email protected] | f9a4c41a | 2012-05-30 00:05:32 | [diff] [blame] | 674 | // static |
| 675 | bool FileSelectHelper::IsAcceptTypeValid(const std::string& accept_type) { |
| 676 | // TODO(raymes): This only does some basic checks, extend to test more cases. |
| 677 | // A 1 character accept type will always be invalid (either a "." in the case |
| 678 | // of an extension or a "/" in the case of a MIME type). |
| 679 | std::string unused; |
| 680 | if (accept_type.length() <= 1 || |
brettw | fce8d19 | 2015-08-10 19:07:51 | [diff] [blame] | 681 | base::ToLowerASCII(accept_type) != accept_type || |
[email protected] | 8af69c6c | 2014-03-03 19:05:31 | [diff] [blame] | 682 | base::TrimWhitespaceASCII(accept_type, base::TRIM_ALL, &unused) != |
| 683 | base::TRIM_NONE) { |
[email protected] | f9a4c41a | 2012-05-30 00:05:32 | [diff] [blame] | 684 | return false; |
| 685 | } |
| 686 | return true; |
| 687 | } |
asanka | ee261f3 | 2015-10-27 21:04:28 | [diff] [blame] | 688 | |
| 689 | // static |
| 690 | base::FilePath FileSelectHelper::GetSanitizedFileName( |
| 691 | const base::FilePath& suggested_filename) { |
| 692 | if (suggested_filename.empty()) |
| 693 | return base::FilePath(); |
| 694 | return net::GenerateFileName( |
| 695 | GURL(), std::string(), std::string(), suggested_filename.AsUTF8Unsafe(), |
| 696 | std::string(), l10n_util::GetStringUTF8(IDS_DEFAULT_DOWNLOAD_FILENAME)); |
| 697 | } |