| // 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_WIN_H_ |
| #define COMPONENTS_STORAGE_MONITOR_STORAGE_MONITOR_WIN_H_ |
| |
| #include <memory> |
| |
| #include "base/macros.h" |
| #include "base/memory/ref_counted.h" |
| #include "components/storage_monitor/storage_monitor.h" |
| |
| #include <windows.h> |
| |
| namespace base { |
| class FilePath; |
| } |
| |
| namespace storage_monitor { |
| |
| class PortableDeviceWatcherWin; |
| class TestStorageMonitorWin; |
| class VolumeMountWatcherWin; |
| |
| class StorageMonitorWin : public StorageMonitor { |
| public: |
| // Should only be called by browser start up code. |
| // Use StorageMonitor::GetInstance() instead. |
| // To support unit tests, this constructor takes |volume_mount_watcher| and |
| // |portable_device_watcher| objects. These params are either constructed in |
| // unit tests or in StorageMonitorWin CreateInternal() function. |
| StorageMonitorWin(VolumeMountWatcherWin* volume_mount_watcher, |
| PortableDeviceWatcherWin* portable_device_watcher); |
| ~StorageMonitorWin() override; |
| |
| // Must be called after the file thread is created. |
| void Init() override; |
| |
| // StorageMonitor: |
| bool GetStorageInfoForPath(const base::FilePath& path, |
| StorageInfo* device_info) const override; |
| bool GetMTPStorageInfoFromDeviceId( |
| const std::string& storage_device_id, |
| base::string16* device_location, |
| base::string16* storage_object_id) const override; |
| |
| void EjectDevice( |
| const std::string& device_id, |
| base::Callback<void(EjectStatus)> callback) override; |
| |
| private: |
| class PortableDeviceNotifications; |
| friend class TestStorageMonitorWin; |
| |
| void MediaChangeNotificationRegister(); |
| |
| // Gets the removable storage information given a |device_path|. On success, |
| // returns true and fills in |info|. |
| bool GetDeviceInfo(const base::FilePath& device_path, |
| StorageInfo* info) const; |
| |
| static LRESULT CALLBACK WndProcThunk(HWND hwnd, UINT message, WPARAM wparam, |
| LPARAM lparam); |
| |
| LRESULT CALLBACK WndProc(HWND hwnd, UINT message, WPARAM wparam, |
| LPARAM lparam); |
| |
| void OnDeviceChange(UINT event_type, LPARAM data); |
| void OnMediaChange(WPARAM wparam, LPARAM lparam); |
| |
| // The window class of |window_|. |
| ATOM window_class_; |
| |
| // The handle of the module that contains the window procedure of |window_|. |
| HMODULE instance_; |
| HWND window_; |
| |
| // The handle of a registration for shell notifications. |
| ULONG shell_change_notify_id_; |
| |
| // The volume mount point watcher, used to manage the mounted devices. |
| std::unique_ptr<VolumeMountWatcherWin> volume_mount_watcher_; |
| |
| // The portable device watcher, used to manage media transfer protocol |
| // devices. |
| std::unique_ptr<PortableDeviceWatcherWin> portable_device_watcher_; |
| |
| DISALLOW_COPY_AND_ASSIGN(StorageMonitorWin); |
| }; |
| |
| } // namespace storage_monitor |
| |
| #endif // COMPONENTS_STORAGE_MONITOR_STORAGE_MONITOR_WIN_H_ |