1// Copyright (c) 2012 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/storage_monitor/media_storage_util.h" 6 7#include <vector> 8 9#include "base/callback.h" 10#include "base/file_util.h" 11#include "base/logging.h" 12#include "base/metrics/histogram.h" 13#include "base/strings/string_util.h" 14#include "base/strings/utf_string_conversions.h" 15#include "chrome/browser/storage_monitor/removable_device_constants.h" 16#include "chrome/browser/storage_monitor/storage_monitor.h" 17#include "content/public/browser/browser_thread.h" 18 19using content::BrowserThread; 20 21namespace { 22 23// MediaDeviceNotification.DeviceInfo histogram values. 24enum DeviceInfoHistogramBuckets { 25 MASS_STORAGE_DEVICE_NAME_AND_UUID_AVAILABLE, 26 MASS_STORAGE_DEVICE_UUID_MISSING, 27 MASS_STORAGE_DEVICE_NAME_MISSING, 28 MASS_STORAGE_DEVICE_NAME_AND_UUID_MISSING, 29 MTP_STORAGE_DEVICE_NAME_AND_UUID_AVAILABLE, 30 MTP_STORAGE_DEVICE_UUID_MISSING, 31 MTP_STORAGE_DEVICE_NAME_MISSING, 32 MTP_STORAGE_DEVICE_NAME_AND_UUID_MISSING, 33 DEVICE_INFO_BUCKET_BOUNDARY 34}; 35 36#if !defined(OS_WIN) 37const char kRootPath[] = "/"; 38#endif 39 40typedef std::vector<StorageInfo> StorageInfoList; 41 42base::FilePath::StringType FindRemovableStorageLocationById( 43 const std::string& device_id) { 44 StorageInfoList devices = 45 StorageMonitor::GetInstance()->GetAllAvailableStorages(); 46 for (StorageInfoList::const_iterator it = devices.begin(); 47 it != devices.end(); ++it) { 48 if (it->device_id() == device_id 49 && StorageInfo::IsRemovableDevice(device_id)) 50 return it->location(); 51 } 52 return base::FilePath::StringType(); 53} 54 55void FilterAttachedDevicesOnFileThread(MediaStorageUtil::DeviceIdSet* devices) { 56 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); 57 MediaStorageUtil::DeviceIdSet missing_devices; 58 59 for (MediaStorageUtil::DeviceIdSet::const_iterator it = devices->begin(); 60 it != devices->end(); 61 ++it) { 62 StorageInfo::Type type; 63 std::string unique_id; 64 if (!StorageInfo::CrackDeviceId(*it, &type, &unique_id)) { 65 missing_devices.insert(*it); 66 continue; 67 } 68 69 if (type == StorageInfo::FIXED_MASS_STORAGE || 70 type == StorageInfo::ITUNES || 71 type == StorageInfo::IPHOTO || 72 type == StorageInfo::PICASA) { 73 if (!base::PathExists(base::FilePath::FromUTF8Unsafe(unique_id))) 74 missing_devices.insert(*it); 75 continue; 76 } 77 78 if (!MediaStorageUtil::IsRemovableStorageAttached(*it)) 79 missing_devices.insert(*it); 80 } 81 82 for (MediaStorageUtil::DeviceIdSet::const_iterator it = 83 missing_devices.begin(); 84 it != missing_devices.end(); 85 ++it) { 86 devices->erase(*it); 87 } 88} 89 90} // namespace 91 92// static 93bool MediaStorageUtil::HasDcim(const base::FilePath& mount_point) { 94 DCHECK(!content::BrowserThread::CurrentlyOn(content::BrowserThread::UI)); 95 96 base::FilePath::StringType dcim_dir(kDCIMDirectoryName); 97 if (!base::DirectoryExists(mount_point.Append(dcim_dir))) { 98 // Check for lowercase 'dcim' as well. 99 base::FilePath dcim_path_lower( 100 mount_point.Append(StringToLowerASCII(dcim_dir))); 101 if (!base::DirectoryExists(dcim_path_lower)) 102 return false; 103 } 104 return true; 105} 106 107// static 108bool MediaStorageUtil::CanCreateFileSystem(const std::string& device_id, 109 const base::FilePath& path) { 110 StorageInfo::Type type; 111 if (!StorageInfo::CrackDeviceId(device_id, &type, NULL)) 112 return false; 113 114 if (type == StorageInfo::MAC_IMAGE_CAPTURE) 115 return true; 116 117 return path.IsAbsolute() && !path.ReferencesParent(); 118} 119 120// static 121void MediaStorageUtil::FilterAttachedDevices(DeviceIdSet* devices, 122 const base::Closure& done) { 123 if (BrowserThread::CurrentlyOn(BrowserThread::FILE)) { 124 FilterAttachedDevicesOnFileThread(devices); 125 done.Run(); 126 return; 127 } 128 BrowserThread::PostTaskAndReply(BrowserThread::FILE, 129 FROM_HERE, 130 base::Bind(&FilterAttachedDevicesOnFileThread, 131 devices), 132 done); 133} 134 135// TODO(kmadhusu) Write unit tests for GetDeviceInfoFromPath(). 136bool MediaStorageUtil::GetDeviceInfoFromPath(const base::FilePath& path, 137 StorageInfo* device_info, 138 base::FilePath* relative_path) { 139 DCHECK(device_info); 140 DCHECK(relative_path); 141 142 if (!path.IsAbsolute()) 143 return false; 144 145 StorageInfo info; 146 StorageMonitor* monitor = StorageMonitor::GetInstance(); 147 bool found_device = monitor->GetStorageInfoForPath(path, &info); 148 149 if (found_device && StorageInfo::IsRemovableDevice(info.device_id())) { 150 base::FilePath sub_folder_path; 151 base::FilePath device_path(info.location()); 152 if (path != device_path) { 153 bool success = device_path.AppendRelativePath(path, &sub_folder_path); 154 DCHECK(success); 155 } 156 157 *device_info = info; 158 *relative_path = sub_folder_path; 159 return true; 160 } 161 162 // On Posix systems, there's one root so any absolute path could be valid. 163 // TODO(gbillock): Delete this stanza? Posix systems should have the root 164 // volume information. If not, we should move the below into the 165 // right GetStorageInfoForPath implementations. 166#if !defined(OS_POSIX) 167 if (!found_device) 168 return false; 169#endif 170 171 // Handle non-removable devices. Note: this is just overwriting 172 // good values from StorageMonitor. 173 // TODO(gbillock): Make sure return values from that class are definitive, 174 // and don't do this here. 175 info.set_device_id( 176 StorageInfo::MakeDeviceId(StorageInfo::FIXED_MASS_STORAGE, 177 path.AsUTF8Unsafe())); 178 *device_info = info; 179 *relative_path = base::FilePath(); 180 return true; 181} 182 183// static 184base::FilePath MediaStorageUtil::FindDevicePathById( 185 const std::string& device_id) { 186 StorageInfo::Type type; 187 std::string unique_id; 188 if (!StorageInfo::CrackDeviceId(device_id, &type, &unique_id)) 189 return base::FilePath(); 190 191 if (type == StorageInfo::FIXED_MASS_STORAGE || 192 type == StorageInfo::ITUNES || 193 type == StorageInfo::IPHOTO || 194 type == StorageInfo::PICASA) { 195 // For this type, the unique_id is the path. 196 return base::FilePath::FromUTF8Unsafe(unique_id); 197 } 198 199 // For ImageCapture, the synthetic filesystem will be rooted at a fake 200 // top-level directory which is the device_id. 201 if (type == StorageInfo::MAC_IMAGE_CAPTURE) { 202#if !defined(OS_WIN) 203 return base::FilePath(kRootPath + device_id); 204#endif 205 } 206 207 DCHECK(type == StorageInfo::MTP_OR_PTP || 208 type == StorageInfo::REMOVABLE_MASS_STORAGE_WITH_DCIM || 209 type == StorageInfo::REMOVABLE_MASS_STORAGE_NO_DCIM); 210 return base::FilePath(FindRemovableStorageLocationById(device_id)); 211} 212 213// static 214void MediaStorageUtil::RecordDeviceInfoHistogram( 215 bool mass_storage, 216 const std::string& device_uuid, 217 const base::string16& device_label) { 218 unsigned int event_number = 0; 219 if (!mass_storage) 220 event_number = 4; 221 222 if (device_label.empty()) 223 event_number += 2; 224 225 if (device_uuid.empty()) 226 event_number += 1; 227 enum DeviceInfoHistogramBuckets event = 228 static_cast<enum DeviceInfoHistogramBuckets>(event_number); 229 if (event >= DEVICE_INFO_BUCKET_BOUNDARY) { 230 NOTREACHED(); 231 return; 232 } 233 UMA_HISTOGRAM_ENUMERATION("MediaDeviceNotifications.DeviceInfo", event, 234 DEVICE_INFO_BUCKET_BOUNDARY); 235} 236 237bool MediaStorageUtil::IsRemovableStorageAttached(const std::string& id) { 238 StorageInfoList devices = 239 StorageMonitor::GetInstance()->GetAllAvailableStorages(); 240 for (StorageInfoList::const_iterator it = devices.begin(); 241 it != devices.end(); ++it) { 242 if (StorageInfo::IsRemovableDevice(id) && it->device_id() == id) 243 return true; 244 } 245 return false; 246} 247