blob: 6b51cab675c5270fa2aca4756c197c90790347e1 [file] [log] [blame]
// 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.
#include "base/basictypes.h"
#include "base/memory/ref_counted.h"
#include "base/memory/scoped_ptr.h"
#include "components/storage_monitor/storage_monitor.h"
namespace base {
class FilePath;
namespace storage_monitor {
class PortableDeviceWatcherWin;
class TestStorageMonitorWin;
class VolumeMountWatcherWin;
class StorageMonitorWin : public StorageMonitor {
// 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);
virtual ~StorageMonitorWin();
// Must be called after the file thread is created.
virtual void Init() override;
// StorageMonitor:
virtual bool GetStorageInfoForPath(const base::FilePath& path,
StorageInfo* device_info) const override;
virtual bool GetMTPStorageInfoFromDeviceId(
const std::string& storage_device_id,
base::string16* device_location,
base::string16* storage_object_id) const override;
virtual void EjectDevice(
const std::string& device_id,
base::Callback<void(EjectStatus)> callback) override;
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.
scoped_ptr<VolumeMountWatcherWin> volume_mount_watcher_;
// The portable device watcher, used to manage media transfer protocol
// devices.
scoped_ptr<PortableDeviceWatcherWin> portable_device_watcher_;
} // namespace storage_monitor