Index: chrome/browser/media_gallery/linux/mtp_device_delegate_impl_linux.cc |
diff --git a/chrome/browser/media_gallery/linux/mtp_device_delegate_impl_linux.cc b/chrome/browser/media_gallery/linux/mtp_device_delegate_impl_linux.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..86fac656a4c9305b5deb1b99b27c445d6e1a1bb2 |
--- /dev/null |
+++ b/chrome/browser/media_gallery/linux/mtp_device_delegate_impl_linux.cc |
@@ -0,0 +1,216 @@ |
+// Copyright (c) 2012 The Chromium Authors. All rights reserved. |
Lei Zhang
2012/12/03 22:23:54
Can you tweak the git similarity parameter so this
kmadhusu
2012/12/03 23:30:00
I tried with different similiarity(e.g 60, 80, 75,
Lei Zhang
2012/12/03 23:46:27
Aww, too bad. It would be nice to preserve the his
|
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+#include "chrome/browser/media_gallery/linux/mtp_device_delegate_impl_linux.h" |
+ |
+#include "base/bind.h" |
+#include "base/file_path.h" |
+#include "base/file_util.h" |
+#include "base/sequenced_task_runner.h" |
+#include "base/string_util.h" |
+#include "base/threading/sequenced_worker_pool.h" |
+#include "chrome/browser/media_gallery/linux/get_mtp_file_info_worker.h" |
+#include "chrome/browser/media_gallery/linux/mtp_device_object_enumerator.h" |
+#include "chrome/browser/media_gallery/linux/mtp_device_operations_utils.h" |
+#include "chrome/browser/media_gallery/linux/open_mtp_storage_worker.h" |
+#include "chrome/browser/media_gallery/linux/read_mtp_directory_worker.h" |
+#include "chrome/browser/media_gallery/linux/read_mtp_file_worker.h" |
+#include "chrome/browser/media_gallery/linux/recursive_mtp_device_object_enumerator.h" |
+#include "chrome/browser/media_transfer_protocol/media_transfer_protocol_manager.h" |
+#include "content/public/browser/browser_thread.h" |
+ |
+using base::Bind; |
+using base::PlatformFileError; |
+using base::PlatformFileInfo; |
+using base::SequencedTaskRunner; |
+using content::BrowserThread; |
+using fileapi::FileSystemFileUtil; |
+ |
+namespace chrome { |
+ |
+namespace { |
+ |
+// File path separator constant. |
+const char kRootPath[] = "/"; |
+ |
+// Does nothing. |
+// This method is used to handle the results of |
+// MediaTransferProtocolManager::CloseStorage method call. |
+void DoNothing(bool error) { |
+} |
+ |
+// Closes the device storage on the UI thread. |
+void CloseStorageOnUIThread(const std::string& device_handle) { |
+ DCHECK(!device_handle.empty()); |
+ DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
+ GetMediaTransferProtocolManager()->CloseStorage(device_handle, |
+ Bind(&DoNothing)); |
+} |
+ |
+// Returns the device relative file path given |file_path|. |
+// E.g.: If the |file_path| is "/usb:2,2:12345/DCIM" and |registered_dev_path| |
+// is "/usb:2,2:12345", this function returns the device relative path which is |
+// "/DCIM". |
+std::string GetDeviceRelativePath(const std::string& registered_dev_path, |
+ const std::string& file_path) { |
+ DCHECK(!registered_dev_path.empty()); |
+ DCHECK(!file_path.empty()); |
+ |
+ std::string actual_file_path; |
+ if (registered_dev_path == file_path) { |
+ actual_file_path = kRootPath; |
+ } else { |
+ actual_file_path = file_path; |
+ ReplaceFirstSubstringAfterOffset(&actual_file_path, 0, |
+ registered_dev_path.c_str(), ""); |
+ } |
+ DCHECK(!actual_file_path.empty()); |
+ return actual_file_path; |
+} |
+ |
+} // namespace |
+ |
+MTPDeviceDelegateImplLinux::MTPDeviceDelegateImplLinux( |
+ const std::string& device_location) |
+ : device_path_(device_location), |
+ on_task_completed_event_(false, false), |
+ on_shutdown_event_(true, false) { |
+ CHECK(!device_path_.empty()); |
+ DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
+ base::SequencedWorkerPool* pool = BrowserThread::GetBlockingPool(); |
+ base::SequencedWorkerPool::SequenceToken media_sequence_token = |
+ pool->GetNamedSequenceToken("media-task-runner"); |
+ media_task_runner_ = pool->GetSequencedTaskRunner(media_sequence_token); |
+} |
+ |
+PlatformFileError MTPDeviceDelegateImplLinux::GetFileInfo( |
+ const FilePath& file_path, |
+ PlatformFileInfo* file_info) { |
+ if (!LazyInit()) |
+ return base::PLATFORM_FILE_ERROR_FAILED; |
+ |
+ scoped_refptr<GetMTPFileInfoWorker> worker(new GetMTPFileInfoWorker( |
+ device_handle_, GetDeviceRelativePath(device_path_, file_path.value()), |
+ media_task_runner_, &on_task_completed_event_, &on_shutdown_event_)); |
+ worker->Run(); |
+ return worker->get_file_info(file_info); |
+} |
+ |
+scoped_ptr<FileSystemFileUtil::AbstractFileEnumerator> |
+ MTPDeviceDelegateImplLinux::CreateFileEnumerator( |
+ const FilePath& root, |
+ bool recursive) { |
+ if (root.value().empty() || !LazyInit()) { |
+ return make_scoped_ptr(new FileSystemFileUtil::EmptyFileEnumerator()) |
+ .PassAs<FileSystemFileUtil::AbstractFileEnumerator>(); |
+ } |
+ |
+ scoped_refptr<ReadMTPDirectoryWorker> worker(new ReadMTPDirectoryWorker( |
+ device_handle_, GetDeviceRelativePath(device_path_, root.value()), |
+ media_task_runner_, &on_task_completed_event_, &on_shutdown_event_)); |
+ worker->Run(); |
+ |
+ if (worker->get_file_entries().empty()) { |
+ return make_scoped_ptr(new FileSystemFileUtil::EmptyFileEnumerator()) |
+ .PassAs<FileSystemFileUtil::AbstractFileEnumerator>(); |
+ } |
+ |
+ if (recursive) { |
+ return make_scoped_ptr(new RecursiveMTPDeviceObjectEnumerator( |
+ device_handle_, media_task_runner_, worker->get_file_entries(), |
+ &on_task_completed_event_, &on_shutdown_event_)) |
+ .PassAs<FileSystemFileUtil::AbstractFileEnumerator>(); |
+ } |
+ return make_scoped_ptr( |
+ new MTPDeviceObjectEnumerator(worker->get_file_entries())) |
+ .PassAs<FileSystemFileUtil::AbstractFileEnumerator>(); |
+} |
+ |
+PlatformFileError MTPDeviceDelegateImplLinux::CreateSnapshotFile( |
+ const FilePath& device_file_path, |
+ const FilePath& local_path, |
+ PlatformFileInfo* file_info) { |
+ if (!LazyInit()) |
+ return base::PLATFORM_FILE_ERROR_FAILED; |
+ |
+ PlatformFileError error = GetFileInfo(device_file_path, file_info); |
+ if (error != base::PLATFORM_FILE_OK) |
+ return error; |
+ |
+ if (file_info->size <= 0 || file_info->size > kuint32max) |
+ return base::PLATFORM_FILE_ERROR_FAILED; |
+ |
+ scoped_refptr<ReadMTPFileWorker> worker(new ReadMTPFileWorker( |
+ device_handle_, |
+ GetDeviceRelativePath(device_path_, device_file_path.value()), |
+ file_info->size, local_path, media_task_runner_, |
+ &on_task_completed_event_, &on_shutdown_event_)); |
+ worker->Run(); |
+ if (!worker->Succeeded()) |
+ return base::PLATFORM_FILE_ERROR_FAILED; |
+ |
+ // Modify the last modified time to null. This prevents the time stamp |
+ // verfication in LocalFileStreamReader. |
+ file_info->last_modified = base::Time(); |
+ return base::PLATFORM_FILE_OK; |
+} |
+ |
+SequencedTaskRunner* MTPDeviceDelegateImplLinux::GetMediaTaskRunner() { |
+ return media_task_runner_.get(); |
+} |
+ |
+void MTPDeviceDelegateImplLinux::CancelPendingTasksAndDeleteDelegate() { |
+ DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
+ // Caution: This function is called on the IO thread. Access only the thread |
+ // safe member variables in this function. Do all the clean up operations in |
+ // DeleteDelegateOnTaskRunner(). |
+ on_shutdown_event_.Signal(); |
+ on_task_completed_event_.Signal(); |
+ media_task_runner_->PostTask( |
+ FROM_HERE, |
+ base::Bind(&MTPDeviceDelegateImplLinux::DeleteDelegateOnTaskRunner, |
+ base::Unretained(this))); |
+} |
+ |
+base::WeakPtr<fileapi::MTPDeviceDelegate> MTPDeviceDelegateImplLinux:: |
+ GetAsWeakPtrOnIOThread() { |
+ DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
+ base::WeakPtr<fileapi::MTPDeviceDelegate> delegate = AsWeakPtr(); |
+ // The weak pointer is instantiated on the IO thread, but only accessed on |
+ // |media_task_runner_|. Therefore, detach from the current thread. |
+ DetachFromThread(); |
+ return delegate; |
+} |
+ |
+MTPDeviceDelegateImplLinux::~MTPDeviceDelegateImplLinux() { |
+ DCHECK(media_task_runner_->RunsTasksOnCurrentThread()); |
+ // Do all the clean up operations on DeleteDelegateOnTaskRunner(). |
+} |
+ |
+bool MTPDeviceDelegateImplLinux::LazyInit() { |
+ DCHECK(media_task_runner_); |
+ DCHECK(media_task_runner_->RunsTasksOnCurrentThread()); |
+ |
+ if (!device_handle_.empty()) |
+ return true; // Already successfully initialized. |
+ |
+ std::string storage_name; |
+ RemoveChars(device_path_, kRootPath, &storage_name); |
+ DCHECK(!storage_name.empty()); |
+ scoped_refptr<OpenMTPStorageWorker> worker(new OpenMTPStorageWorker( |
+ storage_name, media_task_runner_, &on_task_completed_event_, |
+ &on_shutdown_event_)); |
+ worker->Run(); |
+ device_handle_ = worker->device_handle(); |
+ return !device_handle_.empty(); |
+} |
+ |
+void MTPDeviceDelegateImplLinux::DeleteDelegateOnTaskRunner() { |
+ DCHECK(media_task_runner_->RunsTasksOnCurrentThread()); |
+ BrowserThread::PostTask(BrowserThread::UI, FROM_HERE, |
+ Bind(&CloseStorageOnUIThread, device_handle_)); |
+ delete this; |
+} |
+ |
+} // namespace chrome |