chrome/browser/media_galleries/linux/mtp_device_task_helper.cc

   1 // Copyright 2013 The Chromium Authors. All rights reserved.
   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/media_galleries/linux/mtp_device_task_helper.h"
   6
   7 #include <algorithm>
   8
   9 #include "base/logging.h"
  10 #include "base/numerics/safe_conversions.h"
  11 #include "base/strings/string_number_conversions.h"
  12 #include "chrome/browser/media_galleries/linux/mtp_device_object_enumerator.h"
  13 #include "chrome/browser/media_galleries/linux/mtp_read_file_worker.h"
  14 #include "chrome/browser/media_galleries/linux/snapshot_file_details.h"
  15 #include "components/storage_monitor/storage_monitor.h"
  16 #include "content/public/browser/browser_thread.h"
  17 #include "device/media_transfer_protocol/media_transfer_protocol_manager.h"
  18 #include "net/base/io_buffer.h"
  19 #include "third_party/cros_system_api/dbus/service_constants.h"
  20 #include "webkit/browser/fileapi/async_file_util.h"
  21 #include "webkit/common/fileapi/file_system_util.h"
  22
  23 using storage_monitor::StorageMonitor;
  24
  25 namespace {
  26
  27 // Does nothing.
  28 // This method is used to handle the results of
  29 // MediaTransferProtocolManager::CloseStorage method call.
  30 void DoNothing(bool error) {
  31 }
  32
  33 device::MediaTransferProtocolManager* GetMediaTransferProtocolManager() {
  34   return StorageMonitor::GetInstance()->media_transfer_protocol_manager();
  35 }
  36
  37 base::File::Info FileInfoFromMTPFileEntry(const MtpFileEntry& file_entry) {
  38   base::File::Info file_entry_info;
  39   file_entry_info.size = file_entry.file_size();
  40   file_entry_info.is_directory =
  41       file_entry.file_type() == MtpFileEntry::FILE_TYPE_FOLDER;
  42   file_entry_info.is_symbolic_link = false;
  43   file_entry_info.last_modified =
  44       base::Time::FromTimeT(file_entry.modification_time());
  45   file_entry_info.last_accessed = file_entry_info.last_modified;
  46   file_entry_info.creation_time = base::Time();
  47   return file_entry_info;
  48 }
  49
  50 }  // namespace
  51
  52 MTPDeviceTaskHelper::MTPDeviceTaskHelper()
  53     : weak_ptr_factory_(this) {
  54   DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
  55 }
  56
  57 MTPDeviceTaskHelper::~MTPDeviceTaskHelper() {
  58   DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
  59 }
  60
  61 void MTPDeviceTaskHelper::OpenStorage(const std::string& storage_name,
  62                                       const OpenStorageCallback& callback) {
  63   DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
  64   DCHECK(!storage_name.empty());
  65   if (!device_handle_.empty()) {
  66     content::BrowserThread::PostTask(content::BrowserThread::IO,
  67                                      FROM_HERE,
  68                                      base::Bind(callback, true));
  69     return;
  70   }
  71   GetMediaTransferProtocolManager()->OpenStorage(
  72       storage_name, mtpd::kReadOnlyMode,
  73       base::Bind(&MTPDeviceTaskHelper::OnDidOpenStorage,
  74                  weak_ptr_factory_.GetWeakPtr(),
  75                  callback));
  76 }
  77
  78 void MTPDeviceTaskHelper::GetFileInfoById(
  79     uint32 file_id,
  80     const GetFileInfoSuccessCallback& success_callback,
  81     const ErrorCallback& error_callback) {
  82   DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
  83   if (device_handle_.empty())
  84     return HandleDeviceError(error_callback, base::File::FILE_ERROR_FAILED);
  85
  86   GetMediaTransferProtocolManager()->GetFileInfoById(
  87       device_handle_, file_id,
  88       base::Bind(&MTPDeviceTaskHelper::OnGetFileInfo,
  89                  weak_ptr_factory_.GetWeakPtr(),
  90                  success_callback,
  91                  error_callback));
  92 }
  93
  94 void MTPDeviceTaskHelper::ReadDirectoryById(
  95     uint32 dir_id,
  96     const ReadDirectorySuccessCallback& success_callback,
  97     const ErrorCallback& error_callback) {
  98   DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
  99   if (device_handle_.empty())
 100     return HandleDeviceError(error_callback, base::File::FILE_ERROR_FAILED);
 101
 102   GetMediaTransferProtocolManager()->ReadDirectoryById(
 103       device_handle_, dir_id,
 104       base::Bind(&MTPDeviceTaskHelper::OnDidReadDirectoryById,
 105                  weak_ptr_factory_.GetWeakPtr(),
 106                  success_callback,
 107                  error_callback));
 108 }
 109
 110 void MTPDeviceTaskHelper::WriteDataIntoSnapshotFile(
 111     const SnapshotRequestInfo& request_info,
 112     const base::File::Info& snapshot_file_info) {
 113   DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
 114   if (device_handle_.empty()) {
 115     return HandleDeviceError(request_info.error_callback,
 116                              base::File::FILE_ERROR_FAILED);
 117   }
 118
 119   if (!read_file_worker_)
 120     read_file_worker_.reset(new MTPReadFileWorker(device_handle_));
 121   read_file_worker_->WriteDataIntoSnapshotFile(request_info,
 122                                                snapshot_file_info);
 123 }
 124
 125 void MTPDeviceTaskHelper::ReadBytes(
 126     const MTPDeviceAsyncDelegate::ReadBytesRequest& request) {
 127   DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
 128   if (device_handle_.empty()) {
 129     return HandleDeviceError(request.error_callback,
 130                              base::File::FILE_ERROR_FAILED);
 131   }
 132
 133   GetMediaTransferProtocolManager()->GetFileInfoById(
 134       device_handle_, request.file_id,
 135       base::Bind(&MTPDeviceTaskHelper::OnGetFileInfoToReadBytes,
 136                  weak_ptr_factory_.GetWeakPtr(), request));
 137 }
 138
 139 void MTPDeviceTaskHelper::CloseStorage() const {
 140   DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
 141   if (device_handle_.empty())
 142     return;
 143   GetMediaTransferProtocolManager()->CloseStorage(device_handle_,
 144                                                   base::Bind(&DoNothing));
 145 }
 146
 147 void MTPDeviceTaskHelper::OnDidOpenStorage(
 148     const OpenStorageCallback& completion_callback,
 149     const std::string& device_handle,
 150     bool error) {
 151   DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
 152   device_handle_ = device_handle;
 153   content::BrowserThread::PostTask(content::BrowserThread::IO,
 154                                    FROM_HERE,
 155                                    base::Bind(completion_callback, !error));
 156 }
 157
 158 void MTPDeviceTaskHelper::OnGetFileInfo(
 159     const GetFileInfoSuccessCallback& success_callback,
 160     const ErrorCallback& error_callback,
 161     const MtpFileEntry& file_entry,
 162     bool error) const {
 163   DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
 164   if (error) {
 165     return HandleDeviceError(error_callback,
 166                              base::File::FILE_ERROR_NOT_FOUND);
 167   }
 168
 169   content::BrowserThread::PostTask(
 170       content::BrowserThread::IO,
 171       FROM_HERE,
 172       base::Bind(success_callback, FileInfoFromMTPFileEntry(file_entry)));
 173 }
 174
 175 void MTPDeviceTaskHelper::OnDidReadDirectoryById(
 176     const ReadDirectorySuccessCallback& success_callback,
 177     const ErrorCallback& error_callback,
 178     const std::vector<MtpFileEntry>& file_entries,
 179     bool error) const {
 180   DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
 181   if (error)
 182     return HandleDeviceError(error_callback, base::File::FILE_ERROR_FAILED);
 183
 184   fileapi::AsyncFileUtil::EntryList entries;
 185   base::FilePath current;
 186   MTPDeviceObjectEnumerator file_enum(file_entries);
 187   while (!(current = file_enum.Next()).empty()) {
 188     fileapi::DirectoryEntry entry;
 189     entry.name = fileapi::VirtualPath::BaseName(current).value();
 190     uint32 file_id = 0;
 191     bool ret = file_enum.GetEntryId(&file_id);
 192     DCHECK(ret);
 193     entry.name.push_back(',');
 194     entry.name += base::UintToString(file_id);
 195     entry.is_directory = file_enum.IsDirectory();
 196     entry.size = file_enum.Size();
 197     entry.last_modified_time = file_enum.LastModifiedTime();
 198     entries.push_back(entry);
 199   }
 200   content::BrowserThread::PostTask(content::BrowserThread::IO,
 201                                    FROM_HERE,
 202                                    base::Bind(success_callback, entries));
 203 }
 204
 205 void MTPDeviceTaskHelper::OnGetFileInfoToReadBytes(
 206     const MTPDeviceAsyncDelegate::ReadBytesRequest& request,
 207     const MtpFileEntry& file_entry,
 208     bool error) {
 209   DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
 210   DCHECK(request.buf);
 211   DCHECK_GE(request.buf_len, 0);
 212   DCHECK_GE(request.offset, 0);
 213   if (error) {
 214     return HandleDeviceError(request.error_callback,
 215                              base::File::FILE_ERROR_FAILED);
 216   }
 217
 218   base::File::Info file_info = FileInfoFromMTPFileEntry(file_entry);
 219   if (file_info.is_directory) {
 220     return HandleDeviceError(request.error_callback,
 221                              base::File::FILE_ERROR_NOT_A_FILE);
 222   } else if (file_info.size < 0 || file_info.size > kuint32max ||
 223              request.offset > file_info.size) {
 224     return HandleDeviceError(request.error_callback,
 225                              base::File::FILE_ERROR_FAILED);
 226   } else if (request.offset == file_info.size) {
 227     content::BrowserThread::PostTask(content::BrowserThread::IO,
 228                                      FROM_HERE,
 229                                      base::Bind(request.success_callback,
 230                                                 file_info, 0u));
 231     return;
 232   }
 233
 234   uint32 bytes_to_read = std::min(
 235       base::checked_cast<uint32>(request.buf_len),
 236       base::saturated_cast<uint32>(file_info.size - request.offset));
 237
 238   GetMediaTransferProtocolManager()->ReadFileChunkById(
 239       device_handle_,
 240       request.file_id,
 241       base::checked_cast<uint32>(request.offset),
 242       bytes_to_read,
 243       base::Bind(&MTPDeviceTaskHelper::OnDidReadBytes,
 244                  weak_ptr_factory_.GetWeakPtr(), request, file_info));
 245 }
 246
 247 void MTPDeviceTaskHelper::OnDidReadBytes(
 248     const MTPDeviceAsyncDelegate::ReadBytesRequest& request,
 249     const base::File::Info& file_info,
 250     const std::string& data,
 251     bool error) const {
 252   DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
 253   if (error) {
 254     return HandleDeviceError(request.error_callback,
 255                              base::File::FILE_ERROR_FAILED);
 256   }
 257
 258   CHECK_LE(base::checked_cast<int>(data.length()), request.buf_len);
 259   std::copy(data.begin(), data.end(), request.buf->data());
 260
 261   content::BrowserThread::PostTask(content::BrowserThread::IO,
 262                                    FROM_HERE,
 263                                    base::Bind(request.success_callback,
 264                                               file_info, data.length()));
 265 }
 266
 267 void MTPDeviceTaskHelper::HandleDeviceError(
 268     const ErrorCallback& error_callback,
 269     base::File::Error error) const {
 270   DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
 271   content::BrowserThread::PostTask(content::BrowserThread::IO,
 272                                    FROM_HERE,
 273                                    base::Bind(error_callback, error));
 274 }