Home | History | Annotate | Download | only in storage_monitor 
      1 // Copyright 2014 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 #ifndef COMPONENTS_STORAGE_MONITOR_STORAGE_MONITOR_MAC_H_
      6 #define COMPONENTS_STORAGE_MONITOR_STORAGE_MONITOR_MAC_H_
      7 
      8 #include <DiskArbitration/DiskArbitration.h>
      9 #include <map>
     10 
     11 #include "base/mac/scoped_cftyperef.h"
     12 #include "base/memory/weak_ptr.h"
     13 #include "components/storage_monitor/storage_monitor.h"
     14 
     15 namespace storage_monitor {
     16 
     17 class ImageCaptureDeviceManager;
     18 
     19 // This class posts notifications to listeners when a new disk
     20 // is attached, removed, or changed.
     21 class StorageMonitorMac : public StorageMonitor,
     22                           public base::SupportsWeakPtr<StorageMonitorMac> {
     23  public:
     24   enum UpdateType {
     25     UPDATE_DEVICE_ADDED,
     26     UPDATE_DEVICE_CHANGED,
     27     UPDATE_DEVICE_REMOVED,
     28   };
     29 
     30   // Should only be called by browser start up code.  Use GetInstance() instead.
     31   StorageMonitorMac();
     32 
     33   virtual ~StorageMonitorMac();
     34 
     35   virtual void Init() OVERRIDE;
     36 
     37   void UpdateDisk(const std::string& bsd_name,
     38                   const StorageInfo& info,
     39                   UpdateType update_type);
     40 
     41   virtual bool GetStorageInfoForPath(
     42       const base::FilePath& path,
     43       StorageInfo* device_info) const OVERRIDE;
     44 
     45   virtual void EjectDevice(
     46       const std::string& device_id,
     47       base::Callback<void(EjectStatus)> callback) OVERRIDE;
     48 
     49  private:
     50   static void DiskAppearedCallback(DADiskRef disk, void* context);
     51   static void DiskDisappearedCallback(DADiskRef disk, void* context);
     52   static void DiskDescriptionChangedCallback(DADiskRef disk,
     53                                              CFArrayRef keys,
     54                                              void *context);
     55   void GetDiskInfoAndUpdate(DADiskRef disk, UpdateType update_type);
     56 
     57   bool ShouldPostNotificationForDisk(const StorageInfo& info) const;
     58   bool FindDiskWithMountPoint(const base::FilePath& mount_point,
     59                               StorageInfo* info) const;
     60 
     61   base::ScopedCFTypeRef<DASessionRef> session_;
     62   // Maps disk bsd names to disk info objects. This map tracks all mountable
     63   // devices on the system, though only notifications for removable devices are
     64   // posted.
     65   std::map<std::string, StorageInfo> disk_info_map_;
     66 
     67   int pending_disk_updates_;
     68 
     69   scoped_ptr<ImageCaptureDeviceManager> image_capture_device_manager_;
     70 
     71   DISALLOW_COPY_AND_ASSIGN(StorageMonitorMac);
     72 };
     73 
     74 }  // namespace storage_monitor
     75 
     76 #endif  // COMPONENTS_STORAGE_MONITOR_STORAGE_MONITOR_MAC_H_
     77