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