| 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 #ifndef CHROME_BROWSER_STORAGE_MONITOR_REMOVABLE_DEVICE_NOTIFICATIONS_MAC_H_ | 5 #ifndef CHROME_BROWSER_STORAGE_MONITOR_STORAGE_MONITOR_MAC_H_ |
| 6 #define CHROME_BROWSER_STORAGE_MONITOR_REMOVABLE_DEVICE_NOTIFICATIONS_MAC_H_ | 6 #define CHROME_BROWSER_STORAGE_MONITOR_STORAGE_MONITOR_MAC_H_ |
| 7 | 7 |
| 8 #include <DiskArbitration/DiskArbitration.h> | 8 #include <DiskArbitration/DiskArbitration.h> |
| 9 #include <map> | 9 #include <map> |
| 10 | 10 |
| 11 #include "base/mac/scoped_cftyperef.h" | 11 #include "base/mac/scoped_cftyperef.h" |
| 12 #include "base/memory/ref_counted.h" | 12 #include "base/memory/ref_counted.h" |
| 13 #include "chrome/browser/storage_monitor/disk_info_mac.h" | 13 #include "chrome/browser/storage_monitor/disk_info_mac.h" |
| 14 #include "chrome/browser/storage_monitor/storage_monitor.h" | 14 #include "chrome/browser/storage_monitor/storage_monitor.h" |
| 15 | 15 |
| 16 namespace chrome { | 16 namespace chrome { |
| 17 | 17 |
| 18 // This class posts notifications to listeners when a new disk | 18 // This class posts notifications to listeners when a new disk |
| 19 // is attached, removed, or changed. | 19 // is attached, removed, or changed. |
| 20 class RemovableDeviceNotificationsMac | 20 class StorageMonitorMac |
| 21 : public StorageMonitor, | 21 : public StorageMonitor, |
| 22 public base::RefCountedThreadSafe<RemovableDeviceNotificationsMac> { | 22 public base::RefCountedThreadSafe<StorageMonitorMac> { |
| 23 public: | 23 public: |
| 24 enum UpdateType { | 24 enum UpdateType { |
| 25 UPDATE_DEVICE_ADDED, | 25 UPDATE_DEVICE_ADDED, |
| 26 UPDATE_DEVICE_CHANGED, | 26 UPDATE_DEVICE_CHANGED, |
| 27 UPDATE_DEVICE_REMOVED, | 27 UPDATE_DEVICE_REMOVED, |
| 28 }; | 28 }; |
| 29 | 29 |
| 30 // Should only be called by browser start up code. Use GetInstance() instead. | 30 // Should only be called by browser start up code. Use GetInstance() instead. |
| 31 RemovableDeviceNotificationsMac(); | 31 StorageMonitorMac(); |
| 32 | 32 |
| 33 void UpdateDisk(const DiskInfoMac& info, UpdateType update_type); | 33 void UpdateDisk(const DiskInfoMac& info, UpdateType update_type); |
| 34 | 34 |
| 35 virtual bool GetStorageInfoForPath( | 35 virtual bool GetStorageInfoForPath( |
| 36 const base::FilePath& path, | 36 const base::FilePath& path, |
| 37 StorageInfo* device_info) const OVERRIDE; | 37 StorageInfo* device_info) const OVERRIDE; |
| 38 | 38 |
| 39 // Returns the storage size of the device present at |location|. If the | 39 // Returns the storage size of the device present at |location|. If the |
| 40 // device information is unavailable, returns zero. |location| must be a | 40 // device information is unavailable, returns zero. |location| must be a |
| 41 // top-level mount point. | 41 // top-level mount point. |
| 42 virtual uint64 GetStorageSize(const std::string& location) const OVERRIDE; | 42 virtual uint64 GetStorageSize(const std::string& location) const OVERRIDE; |
| 43 | 43 |
| 44 private: | 44 private: |
| 45 friend class base::RefCountedThreadSafe<RemovableDeviceNotificationsMac>; | 45 friend class base::RefCountedThreadSafe<StorageMonitorMac>; |
| 46 virtual ~RemovableDeviceNotificationsMac(); | 46 virtual ~StorageMonitorMac(); |
| 47 | 47 |
| 48 static void DiskAppearedCallback(DADiskRef disk, void* context); | 48 static void DiskAppearedCallback(DADiskRef disk, void* context); |
| 49 static void DiskDisappearedCallback(DADiskRef disk, void* context); | 49 static void DiskDisappearedCallback(DADiskRef disk, void* context); |
| 50 static void DiskDescriptionChangedCallback(DADiskRef disk, | 50 static void DiskDescriptionChangedCallback(DADiskRef disk, |
| 51 CFArrayRef keys, | 51 CFArrayRef keys, |
| 52 void *context); | 52 void *context); |
| 53 | 53 |
| 54 bool ShouldPostNotificationForDisk(const DiskInfoMac& info) const; | 54 bool ShouldPostNotificationForDisk(const DiskInfoMac& info) const; |
| 55 bool FindDiskWithMountPoint(const base::FilePath& mount_point, | 55 bool FindDiskWithMountPoint(const base::FilePath& mount_point, |
| 56 DiskInfoMac* info) const; | 56 DiskInfoMac* info) const; |
| 57 | 57 |
| 58 base::mac::ScopedCFTypeRef<DASessionRef> session_; | 58 base::mac::ScopedCFTypeRef<DASessionRef> session_; |
| 59 // Maps disk bsd names to disk info objects. This map tracks all mountable | 59 // Maps disk bsd names to disk info objects. This map tracks all mountable |
| 60 // devices on the system though only notifications for removable devices are | 60 // devices on the system though only notifications for removable devices are |
| 61 // posted. | 61 // posted. |
| 62 std::map<std::string, DiskInfoMac> disk_info_map_; | 62 std::map<std::string, DiskInfoMac> disk_info_map_; |
| 63 | 63 |
| 64 DISALLOW_COPY_AND_ASSIGN(RemovableDeviceNotificationsMac); | 64 DISALLOW_COPY_AND_ASSIGN(StorageMonitorMac); |
| 65 }; | 65 }; |
| 66 | 66 |
| 67 } // namespace chrome | 67 } // namespace chrome |
| 68 | 68 |
| 69 #endif // CHROME_BROWSER_STORAGE_MONITOR_REMOVABLE_DEVICE_NOTIFICATIONS_MAC_H_ | 69 #endif // CHROME_BROWSER_STORAGE_MONITOR_STORAGE_MONITOR_MAC_H_ |
| OLD | NEW |