| // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "chrome/browser/file_select_helper.h" |
| |
| #include <string> |
| |
| #include "base/file_util.h" |
| #include "base/string_split.h" |
| #include "base/string_util.h" |
| #include "base/utf_string_conversions.h" |
| #include "chrome/browser/platform_util.h" |
| #include "chrome/browser/profiles/profile.h" |
| #include "content/browser/child_process_security_policy.h" |
| #include "content/browser/renderer_host/render_process_host.h" |
| #include "content/browser/renderer_host/render_view_host.h" |
| #include "content/browser/renderer_host/render_widget_host_view.h" |
| #include "content/browser/tab_contents/tab_contents.h" |
| #include "chrome/browser/ui/browser.h" |
| #include "chrome/browser/ui/browser_list.h" |
| #include "content/common/notification_details.h" |
| #include "content/common/notification_source.h" |
| #include "content/common/view_messages.h" |
| #include "grit/generated_resources.h" |
| #include "net/base/mime_util.h" |
| #include "ui/base/l10n/l10n_util.h" |
| |
| namespace { |
| |
| // There is only one file-selection happening at any given time, |
| // so we allocate an enumeration ID for that purpose. All IDs from |
| // the renderer must start at 0 and increase. |
| static const int kFileSelectEnumerationId = -1; |
| } |
| |
| struct FileSelectHelper::ActiveDirectoryEnumeration { |
| ActiveDirectoryEnumeration() {} |
| |
| scoped_ptr<DirectoryListerDispatchDelegate> delegate_; |
| scoped_refptr<net::DirectoryLister> lister_; |
| RenderViewHost* rvh_; |
| std::vector<FilePath> results_; |
| }; |
| |
| FileSelectHelper::FileSelectHelper(Profile* profile) |
| : profile_(profile), |
| render_view_host_(NULL), |
| select_file_dialog_(), |
| dialog_type_(SelectFileDialog::SELECT_OPEN_FILE) { |
| } |
| |
| FileSelectHelper::~FileSelectHelper() { |
| // There may be pending file dialogs, we need to tell them that we've gone |
| // away so they don't try and call back to us. |
| if (select_file_dialog_.get()) |
| select_file_dialog_->ListenerDestroyed(); |
| |
| // Stop any pending directory enumeration, prevent a callback, and free |
| // allocated memory. |
| std::map<int, ActiveDirectoryEnumeration*>::iterator iter; |
| for (iter = directory_enumerations_.begin(); |
| iter != directory_enumerations_.end(); |
| ++iter) { |
| if (iter->second->lister_.get()) { |
| iter->second->lister_->set_delegate(NULL); |
| iter->second->lister_->Cancel(); |
| } |
| delete iter->second; |
| } |
| } |
| |
| void FileSelectHelper::FileSelected(const FilePath& path, |
| int index, void* params) { |
| if (!render_view_host_) |
| return; |
| |
| profile_->set_last_selected_directory(path.DirName()); |
| |
| if (dialog_type_ == SelectFileDialog::SELECT_FOLDER) { |
| StartNewEnumeration(path, kFileSelectEnumerationId, render_view_host_); |
| return; |
| } |
| |
| std::vector<FilePath> files; |
| files.push_back(path); |
| render_view_host_->FilesSelectedInChooser(files); |
| // We are done with this showing of the dialog. |
| render_view_host_ = NULL; |
| } |
| |
| void FileSelectHelper::MultiFilesSelected(const std::vector<FilePath>& files, |
| void* params) { |
| if (!files.empty()) |
| profile_->set_last_selected_directory(files[0].DirName()); |
| if (!render_view_host_) |
| return; |
| |
| render_view_host_->FilesSelectedInChooser(files); |
| // We are done with this showing of the dialog. |
| render_view_host_ = NULL; |
| } |
| |
| void FileSelectHelper::FileSelectionCanceled(void* params) { |
| if (!render_view_host_) |
| return; |
| |
| // If the user cancels choosing a file to upload we pass back an |
| // empty vector. |
| render_view_host_->FilesSelectedInChooser(std::vector<FilePath>()); |
| |
| // We are done with this showing of the dialog. |
| render_view_host_ = NULL; |
| } |
| |
| void FileSelectHelper::StartNewEnumeration(const FilePath& path, |
| int request_id, |
| RenderViewHost* render_view_host) { |
| scoped_ptr<ActiveDirectoryEnumeration> entry(new ActiveDirectoryEnumeration); |
| entry->rvh_ = render_view_host; |
| entry->delegate_.reset(new DirectoryListerDispatchDelegate(this, request_id)); |
| entry->lister_ = new net::DirectoryLister(path, |
| true, |
| net::DirectoryLister::NO_SORT, |
| entry->delegate_.get()); |
| if (!entry->lister_->Start()) { |
| if (request_id == kFileSelectEnumerationId) |
| FileSelectionCanceled(NULL); |
| else |
| render_view_host->DirectoryEnumerationFinished(request_id, |
| entry->results_); |
| } else { |
| directory_enumerations_[request_id] = entry.release(); |
| } |
| } |
| |
| void FileSelectHelper::OnListFile( |
| int id, |
| const net::DirectoryLister::DirectoryListerData& data) { |
| ActiveDirectoryEnumeration* entry = directory_enumerations_[id]; |
| |
| // Directory upload returns directories via a "." file, so that |
| // empty directories are included. This util call just checks |
| // the flags in the structure; there's no file I/O going on. |
| if (file_util::FileEnumerator::IsDirectory(data.info)) |
| entry->results_.push_back(data.path.Append(FILE_PATH_LITERAL("."))); |
| else |
| entry->results_.push_back(data.path); |
| } |
| |
| void FileSelectHelper::OnListDone(int id, int error) { |
| // This entry needs to be cleaned up when this function is done. |
| scoped_ptr<ActiveDirectoryEnumeration> entry(directory_enumerations_[id]); |
| directory_enumerations_.erase(id); |
| if (!entry->rvh_) |
| return; |
| if (error) { |
| FileSelectionCanceled(NULL); |
| return; |
| } |
| if (id == kFileSelectEnumerationId) |
| entry->rvh_->FilesSelectedInChooser(entry->results_); |
| else |
| entry->rvh_->DirectoryEnumerationFinished(id, entry->results_); |
| } |
| |
| SelectFileDialog::FileTypeInfo* FileSelectHelper::GetFileTypesFromAcceptType( |
| const string16& accept_types) { |
| if (accept_types.empty()) |
| return NULL; |
| |
| // Split the accept-type string on commas. |
| std::vector<string16> mime_types; |
| base::SplitStringUsingSubstr(accept_types, ASCIIToUTF16(","), &mime_types); |
| if (mime_types.empty()) |
| return NULL; |
| |
| // Create FileTypeInfo and pre-allocate for the first extension list. |
| scoped_ptr<SelectFileDialog::FileTypeInfo> file_type( |
| new SelectFileDialog::FileTypeInfo()); |
| file_type->include_all_files = true; |
| file_type->extensions.resize(1); |
| std::vector<FilePath::StringType>* extensions = &file_type->extensions.back(); |
| |
| // Find the correspondinge extensions. |
| int valid_type_count = 0; |
| int description_id = 0; |
| for (size_t i = 0; i < mime_types.size(); ++i) { |
| string16 mime_type = mime_types[i]; |
| std::string ascii_mime_type = StringToLowerASCII(UTF16ToASCII(mime_type)); |
| |
| TrimWhitespace(ascii_mime_type, TRIM_ALL, &ascii_mime_type); |
| if (ascii_mime_type.empty()) |
| continue; |
| |
| size_t old_extension_size = extensions->size(); |
| if (ascii_mime_type == "image/*") { |
| description_id = IDS_IMAGE_FILES; |
| net::GetImageExtensions(extensions); |
| } else if (ascii_mime_type == "audio/*") { |
| description_id = IDS_AUDIO_FILES; |
| net::GetAudioExtensions(extensions); |
| } else if (ascii_mime_type == "video/*") { |
| description_id = IDS_VIDEO_FILES; |
| net::GetVideoExtensions(extensions); |
| } else { |
| net::GetExtensionsForMimeType(ascii_mime_type, extensions); |
| } |
| |
| if (extensions->size() > old_extension_size) |
| valid_type_count++; |
| } |
| |
| // If no valid extension is added, bail out. |
| if (valid_type_count == 0) |
| return NULL; |
| |
| // Use a generic description "Custom Files" if either of the following is |
| // true: |
| // 1) There're multiple types specified, like "audio/*,video/*" |
| // 2) There're multiple extensions for a MIME type without parameter, like |
| // "ehtml,shtml,htm,html" for "text/html". On Windows, the select file |
| // dialog uses the first extension in the list to form the description, |
| // like "EHTML Files". This is not what we want. |
| if (valid_type_count > 1 || |
| (valid_type_count == 1 && description_id == 0 && extensions->size() > 1)) |
| description_id = IDS_CUSTOM_FILES; |
| |
| if (description_id) { |
| file_type->extension_description_overrides.push_back( |
| l10n_util::GetStringUTF16(description_id)); |
| } |
| |
| return file_type.release(); |
| } |
| |
| void FileSelectHelper::RunFileChooser( |
| RenderViewHost* render_view_host, |
| TabContents* tab_contents, |
| const ViewHostMsg_RunFileChooser_Params& params) { |
| DCHECK(!render_view_host_); |
| render_view_host_ = render_view_host; |
| notification_registrar_.RemoveAll(); |
| notification_registrar_.Add(this, |
| NotificationType::RENDER_WIDGET_HOST_DESTROYED, |
| Source<RenderViewHost>(render_view_host)); |
| |
| if (!select_file_dialog_.get()) |
| select_file_dialog_ = SelectFileDialog::Create(this); |
| |
| switch (params.mode) { |
| case ViewHostMsg_RunFileChooser_Mode::Open: |
| dialog_type_ = SelectFileDialog::SELECT_OPEN_FILE; |
| break; |
| case ViewHostMsg_RunFileChooser_Mode::OpenMultiple: |
| dialog_type_ = SelectFileDialog::SELECT_OPEN_MULTI_FILE; |
| break; |
| case ViewHostMsg_RunFileChooser_Mode::OpenFolder: |
| dialog_type_ = SelectFileDialog::SELECT_FOLDER; |
| break; |
| case ViewHostMsg_RunFileChooser_Mode::Save: |
| dialog_type_ = SelectFileDialog::SELECT_SAVEAS_FILE; |
| break; |
| default: |
| dialog_type_ = SelectFileDialog::SELECT_OPEN_FILE; // Prevent warning. |
| NOTREACHED(); |
| } |
| scoped_ptr<SelectFileDialog::FileTypeInfo> file_types( |
| GetFileTypesFromAcceptType(params.accept_types)); |
| FilePath default_file_name = params.default_file_name; |
| if (default_file_name.empty()) |
| default_file_name = profile_->last_selected_directory(); |
| |
| gfx::NativeWindow owning_window = |
| platform_util::GetTopLevel(render_view_host_->view()->GetNativeView()); |
| |
| select_file_dialog_->SelectFile(dialog_type_, |
| params.title, |
| default_file_name, |
| file_types.get(), |
| file_types.get() ? 1 : 0, // 1-based index. |
| FILE_PATH_LITERAL(""), |
| tab_contents, |
| owning_window, |
| NULL); |
| } |
| |
| void FileSelectHelper::EnumerateDirectory(int request_id, |
| RenderViewHost* render_view_host, |
| const FilePath& path) { |
| DCHECK_NE(kFileSelectEnumerationId, request_id); |
| StartNewEnumeration(path, request_id, render_view_host); |
| } |
| |
| void FileSelectHelper::Observe(NotificationType type, |
| const NotificationSource& source, |
| const NotificationDetails& details) { |
| DCHECK(type == NotificationType::RENDER_WIDGET_HOST_DESTROYED); |
| DCHECK(Details<RenderViewHost>(details).ptr() == render_view_host_); |
| render_view_host_ = NULL; |
| } |
| |
| FileSelectObserver::FileSelectObserver(TabContents* tab_contents) |
| : TabContentsObserver(tab_contents) { |
| } |
| |
| FileSelectObserver::~FileSelectObserver() { |
| } |
| |
| bool FileSelectObserver::OnMessageReceived(const IPC::Message& message) { |
| bool handled = true; |
| IPC_BEGIN_MESSAGE_MAP(FileSelectObserver, message) |
| IPC_MESSAGE_HANDLER(ViewHostMsg_RunFileChooser, OnRunFileChooser) |
| IPC_MESSAGE_HANDLER(ViewHostMsg_EnumerateDirectory, OnEnumerateDirectory) |
| IPC_MESSAGE_UNHANDLED(handled = false) |
| IPC_END_MESSAGE_MAP() |
| |
| return handled; |
| } |
| |
| void FileSelectObserver::OnRunFileChooser( |
| const ViewHostMsg_RunFileChooser_Params& params) { |
| if (!file_select_helper_.get()) |
| file_select_helper_.reset(new FileSelectHelper(tab_contents()->profile())); |
| file_select_helper_->RunFileChooser(tab_contents()->render_view_host(), |
| tab_contents(), |
| params); |
| } |
| |
| void FileSelectObserver::OnEnumerateDirectory(int request_id, |
| const FilePath& path) { |
| ChildProcessSecurityPolicy* policy = |
| ChildProcessSecurityPolicy::GetInstance(); |
| if (!policy->CanReadDirectory( |
| tab_contents()->render_view_host()->process()->id(), |
| path)) { |
| return; |
| } |
| |
| if (!file_select_helper_.get()) |
| file_select_helper_.reset(new FileSelectHelper(tab_contents()->profile())); |
| file_select_helper_->EnumerateDirectory(request_id, |
| tab_contents()->render_view_host(), |
| path); |
| } |