| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/browser/storage_monitor/storage_monitor_mac.h" | 5 #include "chrome/browser/storage_monitor/storage_monitor_mac.h" |
| 6 | 6 |
| 7 #include "base/mac/foundation_util.h" | 7 #include "base/mac/foundation_util.h" |
| 8 #include "base/mac/mac_util.h" | 8 #include "base/mac/mac_util.h" |
| 9 #include "base/strings/sys_string_conversions.h" | 9 #include "base/strings/sys_string_conversions.h" |
| 10 #include "base/utf_string_conversions.h" | 10 #include "base/utf_string_conversions.h" |
| (...skipping 29 matching lines...) Expand all Loading... |
| 40 return StorageInfo::FIXED_MASS_STORAGE; | 40 return StorageInfo::FIXED_MASS_STORAGE; |
| 41 if (has_dcim) | 41 if (has_dcim) |
| 42 return StorageInfo::REMOVABLE_MASS_STORAGE_WITH_DCIM; | 42 return StorageInfo::REMOVABLE_MASS_STORAGE_WITH_DCIM; |
| 43 return StorageInfo::REMOVABLE_MASS_STORAGE_NO_DCIM; | 43 return StorageInfo::REMOVABLE_MASS_STORAGE_NO_DCIM; |
| 44 } | 44 } |
| 45 | 45 |
| 46 StorageInfo BuildStorageInfo( | 46 StorageInfo BuildStorageInfo( |
| 47 CFDictionaryRef dict, std::string* bsd_name) { | 47 CFDictionaryRef dict, std::string* bsd_name) { |
| 48 DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::FILE)); | 48 DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::FILE)); |
| 49 | 49 |
| 50 StorageInfo info; | |
| 51 | |
| 52 CFStringRef device_bsd_name = base::mac::GetValueFromDictionary<CFStringRef>( | 50 CFStringRef device_bsd_name = base::mac::GetValueFromDictionary<CFStringRef>( |
| 53 dict, kDADiskDescriptionMediaBSDNameKey); | 51 dict, kDADiskDescriptionMediaBSDNameKey); |
| 54 if (device_bsd_name && bsd_name) | 52 if (device_bsd_name && bsd_name) |
| 55 *bsd_name = base::SysCFStringRefToUTF8(device_bsd_name); | 53 *bsd_name = base::SysCFStringRefToUTF8(device_bsd_name); |
| 56 | 54 |
| 57 CFURLRef url = base::mac::GetValueFromDictionary<CFURLRef>( | 55 CFURLRef url = base::mac::GetValueFromDictionary<CFURLRef>( |
| 58 dict, kDADiskDescriptionVolumePathKey); | 56 dict, kDADiskDescriptionVolumePathKey); |
| 59 NSURL* nsurl = base::mac::CFToNSCast(url); | 57 NSURL* nsurl = base::mac::CFToNSCast(url); |
| 60 info.location = base::mac::NSStringToFilePath([nsurl path]).value(); | 58 base::FilePath location = base::mac::NSStringToFilePath([nsurl path]); |
| 61 CFNumberRef size_number = | 59 CFNumberRef size_number = |
| 62 base::mac::GetValueFromDictionary<CFNumberRef>( | 60 base::mac::GetValueFromDictionary<CFNumberRef>( |
| 63 dict, kDADiskDescriptionMediaSizeKey); | 61 dict, kDADiskDescriptionMediaSizeKey); |
| 64 if (size_number) { | 62 uint64 size_in_bytes = 0; |
| 65 CFNumberGetValue(size_number, kCFNumberLongLongType, | 63 if (size_number) |
| 66 &(info.total_size_in_bytes)); | 64 CFNumberGetValue(size_number, kCFNumberLongLongType, &size_in_bytes); |
| 67 } | |
| 68 | 65 |
| 69 info.vendor_name = GetUTF16FromDictionary( | 66 string16 vendor = GetUTF16FromDictionary( |
| 70 dict, kDADiskDescriptionDeviceVendorKey); | 67 dict, kDADiskDescriptionDeviceVendorKey); |
| 71 info.model_name = GetUTF16FromDictionary( | 68 string16 model = GetUTF16FromDictionary( |
| 72 dict, kDADiskDescriptionDeviceModelKey); | 69 dict, kDADiskDescriptionDeviceModelKey); |
| 73 info.storage_label = GetUTF16FromDictionary( | 70 string16 label = GetUTF16FromDictionary( |
| 74 dict, kDADiskDescriptionVolumeNameKey); | 71 dict, kDADiskDescriptionVolumeNameKey); |
| 75 | 72 |
| 76 CFUUIDRef uuid = base::mac::GetValueFromDictionary<CFUUIDRef>( | 73 CFUUIDRef uuid = base::mac::GetValueFromDictionary<CFUUIDRef>( |
| 77 dict, kDADiskDescriptionVolumeUUIDKey); | 74 dict, kDADiskDescriptionVolumeUUIDKey); |
| 78 std::string unique_id; | 75 std::string unique_id; |
| 79 if (uuid) { | 76 if (uuid) { |
| 80 base::mac::ScopedCFTypeRef<CFStringRef> uuid_string( | 77 base::mac::ScopedCFTypeRef<CFStringRef> uuid_string( |
| 81 CFUUIDCreateString(NULL, uuid)); | 78 CFUUIDCreateString(NULL, uuid)); |
| 82 if (uuid_string.get()) | 79 if (uuid_string.get()) |
| 83 unique_id = base::SysCFStringRefToUTF8(uuid_string); | 80 unique_id = base::SysCFStringRefToUTF8(uuid_string); |
| 84 } | 81 } |
| 85 if (unique_id.empty()) { | 82 if (unique_id.empty()) { |
| 86 string16 revision = GetUTF16FromDictionary( | 83 string16 revision = GetUTF16FromDictionary( |
| 87 dict, kDADiskDescriptionDeviceRevisionKey); | 84 dict, kDADiskDescriptionDeviceRevisionKey); |
| 88 string16 unique_id2 = info.vendor_name; | 85 string16 unique_id2 = vendor; |
| 89 unique_id2 = JoinName(unique_id2, info.model_name); | 86 unique_id2 = JoinName(unique_id2, model); |
| 90 unique_id2 = JoinName(unique_id2, revision); | 87 unique_id2 = JoinName(unique_id2, revision); |
| 91 unique_id = UTF16ToUTF8(unique_id2); | 88 unique_id = UTF16ToUTF8(unique_id2); |
| 92 } | 89 } |
| 93 | 90 |
| 94 CFBooleanRef is_removable_ref = | 91 CFBooleanRef is_removable_ref = |
| 95 base::mac::GetValueFromDictionary<CFBooleanRef>( | 92 base::mac::GetValueFromDictionary<CFBooleanRef>( |
| 96 dict, kDADiskDescriptionMediaRemovableKey); | 93 dict, kDADiskDescriptionMediaRemovableKey); |
| 97 bool is_removable = is_removable_ref && CFBooleanGetValue(is_removable_ref); | 94 bool is_removable = is_removable_ref && CFBooleanGetValue(is_removable_ref); |
| 98 // Checking for DCIM only matters on removable devices. | 95 // Checking for DCIM only matters on removable devices. |
| 99 bool has_dcim = is_removable && | 96 bool has_dcim = is_removable && MediaStorageUtil::HasDcim(location); |
| 100 MediaStorageUtil::HasDcim(base::FilePath(info.location)); | |
| 101 StorageInfo::Type device_type = GetDeviceType(is_removable, has_dcim); | 97 StorageInfo::Type device_type = GetDeviceType(is_removable, has_dcim); |
| 98 std::string device_id; |
| 102 if (!unique_id.empty()) | 99 if (!unique_id.empty()) |
| 103 info.device_id = StorageInfo::MakeDeviceId(device_type, | 100 device_id = StorageInfo::MakeDeviceId(device_type, unique_id); |
| 104 unique_id); | |
| 105 | 101 |
| 106 return info; | 102 return StorageInfo(device_id, string16(), location.value(), label, vendor, |
| 103 model, size_in_bytes); |
| 107 } | 104 } |
| 108 | 105 |
| 109 void GetDiskInfoAndUpdateOnFileThread( | 106 void GetDiskInfoAndUpdateOnFileThread( |
| 110 const base::WeakPtr<StorageMonitorMac>& monitor, | 107 const base::WeakPtr<StorageMonitorMac>& monitor, |
| 111 base::mac::ScopedCFTypeRef<CFDictionaryRef> dict, | 108 base::mac::ScopedCFTypeRef<CFDictionaryRef> dict, |
| 112 StorageMonitorMac::UpdateType update_type) { | 109 StorageMonitorMac::UpdateType update_type) { |
| 113 DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::FILE)); | 110 DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::FILE)); |
| 114 | 111 |
| 115 std::string bsd_name; | 112 std::string bsd_name; |
| 116 StorageInfo info = BuildStorageInfo(dict, &bsd_name); | 113 StorageInfo info = BuildStorageInfo(dict, &bsd_name); |
| (...skipping 93 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 210 const std::string& bsd_name, | 207 const std::string& bsd_name, |
| 211 const StorageInfo& info, | 208 const StorageInfo& info, |
| 212 UpdateType update_type) { | 209 UpdateType update_type) { |
| 213 DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI)); | 210 DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI)); |
| 214 | 211 |
| 215 pending_disk_updates_--; | 212 pending_disk_updates_--; |
| 216 bool initialization_complete = false; | 213 bool initialization_complete = false; |
| 217 if (!IsInitialized() && pending_disk_updates_ == 0) | 214 if (!IsInitialized() && pending_disk_updates_ == 0) |
| 218 initialization_complete = true; | 215 initialization_complete = true; |
| 219 | 216 |
| 220 if (info.device_id.empty() || bsd_name.empty()) { | 217 if (info.device_id().empty() || bsd_name.empty()) { |
| 221 if (initialization_complete) | 218 if (initialization_complete) |
| 222 MarkInitialized(); | 219 MarkInitialized(); |
| 223 return; | 220 return; |
| 224 } | 221 } |
| 225 | 222 |
| 226 std::map<std::string, StorageInfo>::iterator it = | 223 std::map<std::string, StorageInfo>::iterator it = |
| 227 disk_info_map_.find(bsd_name); | 224 disk_info_map_.find(bsd_name); |
| 228 if (it != disk_info_map_.end()) { | 225 if (it != disk_info_map_.end()) { |
| 229 // If an attached notification was previously posted then post a detached | 226 // If an attached notification was previously posted then post a detached |
| 230 // notification now. This is used for devices that are being removed or | 227 // notification now. This is used for devices that are being removed or |
| 231 // devices that have changed. | 228 // devices that have changed. |
| 232 if (ShouldPostNotificationForDisk(it->second)) { | 229 if (ShouldPostNotificationForDisk(it->second)) { |
| 233 receiver()->ProcessDetach(it->second.device_id); | 230 receiver()->ProcessDetach(it->second.device_id()); |
| 234 } | 231 } |
| 235 } | 232 } |
| 236 | 233 |
| 237 if (update_type == UPDATE_DEVICE_REMOVED) { | 234 if (update_type == UPDATE_DEVICE_REMOVED) { |
| 238 if (it != disk_info_map_.end()) | 235 if (it != disk_info_map_.end()) |
| 239 disk_info_map_.erase(it); | 236 disk_info_map_.erase(it); |
| 240 } else { | 237 } else { |
| 241 disk_info_map_[bsd_name] = info; | 238 disk_info_map_[bsd_name] = info; |
| 242 MediaStorageUtil::RecordDeviceInfoHistogram(true, info.device_id, | 239 MediaStorageUtil::RecordDeviceInfoHistogram(true, info.device_id(), |
| 243 info.storage_label); | 240 info.storage_label()); |
| 244 if (ShouldPostNotificationForDisk(info)) | 241 if (ShouldPostNotificationForDisk(info)) |
| 245 receiver()->ProcessAttach(info); | 242 receiver()->ProcessAttach(info); |
| 246 } | 243 } |
| 247 | 244 |
| 248 // We're not really honestly sure we're done, but this looks the best we | 245 // We're not really honestly sure we're done, but this looks the best we |
| 249 // can do. Any misses should go out through notifications. | 246 // can do. Any misses should go out through notifications. |
| 250 if (initialization_complete) | 247 if (initialization_complete) |
| 251 MarkInitialized(); | 248 MarkInitialized(); |
| 252 } | 249 } |
| 253 | 250 |
| (...skipping 17 matching lines...) Expand all Loading... |
| 271 | 268 |
| 272 return false; | 269 return false; |
| 273 } | 270 } |
| 274 | 271 |
| 275 void StorageMonitorMac::EjectDevice( | 272 void StorageMonitorMac::EjectDevice( |
| 276 const std::string& device_id, | 273 const std::string& device_id, |
| 277 base::Callback<void(EjectStatus)> callback) { | 274 base::Callback<void(EjectStatus)> callback) { |
| 278 std::string bsd_name; | 275 std::string bsd_name; |
| 279 for (std::map<std::string, StorageInfo>::iterator | 276 for (std::map<std::string, StorageInfo>::iterator |
| 280 it = disk_info_map_.begin(); it != disk_info_map_.end(); ++it) { | 277 it = disk_info_map_.begin(); it != disk_info_map_.end(); ++it) { |
| 281 if (it->second.device_id == device_id) { | 278 if (it->second.device_id() == device_id) { |
| 282 bsd_name = it->first; | 279 bsd_name = it->first; |
| 283 disk_info_map_.erase(it); | 280 disk_info_map_.erase(it); |
| 284 break; | 281 break; |
| 285 } | 282 } |
| 286 } | 283 } |
| 287 | 284 |
| 288 if (bsd_name.empty()) { | 285 if (bsd_name.empty()) { |
| 289 callback.Run(EJECT_NO_SUCH_DEVICE); | 286 callback.Run(EJECT_NO_SUCH_DEVICE); |
| 290 return; | 287 return; |
| 291 } | 288 } |
| (...skipping 54 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 346 FROM_HERE, | 343 FROM_HERE, |
| 347 base::Bind(GetDiskInfoAndUpdateOnFileThread, | 344 base::Bind(GetDiskInfoAndUpdateOnFileThread, |
| 348 AsWeakPtr(), dict, update_type)); | 345 AsWeakPtr(), dict, update_type)); |
| 349 } | 346 } |
| 350 | 347 |
| 351 | 348 |
| 352 bool StorageMonitorMac::ShouldPostNotificationForDisk( | 349 bool StorageMonitorMac::ShouldPostNotificationForDisk( |
| 353 const StorageInfo& info) const { | 350 const StorageInfo& info) const { |
| 354 // Only post notifications about disks that have no empty fields and | 351 // Only post notifications about disks that have no empty fields and |
| 355 // are removable. Also exclude disk images (DMGs). | 352 // are removable. Also exclude disk images (DMGs). |
| 356 return !info.device_id.empty() && | 353 return !info.device_id().empty() && |
| 357 !info.location.empty() && | 354 !info.location().empty() && |
| 358 info.model_name != ASCIIToUTF16(kDiskImageModelName) && | 355 info.model_name() != ASCIIToUTF16(kDiskImageModelName) && |
| 359 StorageInfo::IsRemovableDevice(info.device_id) && | 356 StorageInfo::IsRemovableDevice(info.device_id()) && |
| 360 StorageInfo::IsMassStorageDevice(info.device_id); | 357 StorageInfo::IsMassStorageDevice(info.device_id()); |
| 361 } | 358 } |
| 362 | 359 |
| 363 bool StorageMonitorMac::FindDiskWithMountPoint( | 360 bool StorageMonitorMac::FindDiskWithMountPoint( |
| 364 const base::FilePath& mount_point, | 361 const base::FilePath& mount_point, |
| 365 StorageInfo* info) const { | 362 StorageInfo* info) const { |
| 366 for (std::map<std::string, StorageInfo>::const_iterator | 363 for (std::map<std::string, StorageInfo>::const_iterator |
| 367 it = disk_info_map_.begin(); it != disk_info_map_.end(); ++it) { | 364 it = disk_info_map_.begin(); it != disk_info_map_.end(); ++it) { |
| 368 if (it->second.location == mount_point.value()) { | 365 if (it->second.location() == mount_point.value()) { |
| 369 *info = it->second; | 366 *info = it->second; |
| 370 return true; | 367 return true; |
| 371 } | 368 } |
| 372 } | 369 } |
| 373 return false; | 370 return false; |
| 374 } | 371 } |
| 375 | 372 |
| 376 } // namespace chrome | 373 } // namespace chrome |
| OLD | NEW |