[Win, MediaGallery] Enumerate and handle mtp device attach/detach events.

chrome/browser/system_monitor/removable_device_notifications_window_win.cc

 // Copyright (c) 2012 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 "chrome/browser/system_monitor/removable_device_notifications_window_win.h"
 
 #include <windows.h>
 #include <dbt.h>
 #include <fileapi.h>
 
-#include "base/file_path.h"
 #include "base/win/wrapped_window_proc.h"
 #include "chrome/browser/system_monitor/media_storage_util.h"
-
-using base::SystemMonitor;
-using base::win::WrappedWindowProc;
+#include "chrome/browser/system_monitor/portable_device_watcher_win.h"
+#include "chrome/browser/system_monitor/removable_device_constants.h"
+#include "chrome/browser/system_monitor/volume_mount_watcher_win.h"
 
 namespace chrome {
 
 namespace {
 
 const char16 kWindowClassName[] = L"Chrome_RemovableDeviceNotificationWindow";
 
 RemovableDeviceNotificationsWindowWin*
     g_removable_device_notifications_window_win = NULL;
 
 }  // namespace
 
+// Manages portable device notification handle for
+// RemovableDeviceNotificationsWindowWin.
+class RemovableDeviceNotificationsWindowWin::PortableDeviceNotifications {
+ public:
+  explicit PortableDeviceNotifications(HWND hwnd);
+  ~PortableDeviceNotifications();
+
+ private:
+  // Register window handle to receive portable device notifications details.
+  void Register(HWND hwnd);
+
+  // Unregisters device notifications.
+  void Unregister();
+
+  HDEVNOTIFY notifications_;
+
+  DISALLOW_IMPLICIT_CONSTRUCTORS(PortableDeviceNotifications);
+};
+
+RemovableDeviceNotificationsWindowWin::PortableDeviceNotifications::
+PortableDeviceNotifications(HWND hwnd) {
+  Register(hwnd);
+}
+
+RemovableDeviceNotificationsWindowWin::PortableDeviceNotifications::
+~PortableDeviceNotifications() {
+  Unregister();
+}
+
+void RemovableDeviceNotificationsWindowWin::PortableDeviceNotifications::
+Register(HWND hwnd) {
+  GUID dev_interface_guid = GUID_NULL;
+  HRESULT hr = CLSIDFromString(kWPDDevInterfaceGUID, &dev_interface_guid);
+  if (FAILED(hr))
+    return;
+  DEV_BROADCAST_DEVICEINTERFACE db = {sizeof(DEV_BROADCAST_DEVICEINTERFACE),
+                                      DBT_DEVTYP_DEVICEINTERFACE, 0,
+                                      dev_interface_guid};
+  notifications_ = RegisterDeviceNotification(hwnd, &db,
+                                              DEVICE_NOTIFY_WINDOW_HANDLE);
+}
+
+void RemovableDeviceNotificationsWindowWin::PortableDeviceNotifications::
+Unregister() {
+  UnregisterDeviceNotification(notifications_);
+}
+
 // static
 RemovableDeviceNotificationsWindowWin*
 RemovableDeviceNotificationsWindowWin::Create() {
-  return new RemovableDeviceNotificationsWindowWin(new VolumeMountWatcherWin());
+  return new RemovableDeviceNotificationsWindowWin(
+      new VolumeMountWatcherWin(), new PortableDeviceWatcherWin());
 }
 
 RemovableDeviceNotificationsWindowWin::
 RemovableDeviceNotificationsWindowWin(
-    VolumeMountWatcherWin* volume_mount_watcher)
+    VolumeMountWatcherWin* volume_mount_watcher,
+    PortableDeviceWatcherWin* portable_device_watcher)
     : window_class_(0),
       instance_(NULL),
       window_(NULL),
-      volume_mount_watcher_(volume_mount_watcher) {
+      volume_mount_watcher_(volume_mount_watcher),
+      portable_device_watcher_(portable_device_watcher) {
   DCHECK(volume_mount_watcher_);
+  DCHECK(portable_device_watcher_);
   DCHECK(!g_removable_device_notifications_window_win);
   g_removable_device_notifications_window_win = this;
 }
 
 RemovableDeviceNotificationsWindowWin::
 ~RemovableDeviceNotificationsWindowWin() {
   if (window_)
     DestroyWindow(window_);
 
   if (window_class_)
     UnregisterClass(MAKEINTATOM(window_class_), instance_);
 
   DCHECK_EQ(this, g_removable_device_notifications_window_win);
   g_removable_device_notifications_window_win = NULL;
 }
 
 // static
 RemovableDeviceNotificationsWindowWin*
 RemovableDeviceNotificationsWindowWin::GetInstance() {
   DCHECK(g_removable_device_notifications_window_win);
   return g_removable_device_notifications_window_win;
 }
 
 void RemovableDeviceNotificationsWindowWin::Init() {
   volume_mount_watcher_->Init();
+  portable_device_watcher_->Init();
 
   WNDCLASSEX window_class;
   base::win::InitializeWindowClass(
       kWindowClassName,
-      &WrappedWindowProc<RemovableDeviceNotificationsWindowWin::WndProcThunk>,
+      &base::win::WrappedWindowProc<
+          RemovableDeviceNotificationsWindowWin::WndProcThunk>,
       0, 0, 0, NULL, NULL, NULL, NULL, NULL,
       &window_class);
   instance_ = window_class.hInstance;
   window_class_ = RegisterClassEx(&window_class);
   DCHECK(window_class_);
 
   window_ = CreateWindow(MAKEINTATOM(window_class_), 0, 0, 0, 0, 0, 0, 0, 0,
                          instance_, 0);
   SetWindowLongPtr(window_, GWLP_USERDATA, reinterpret_cast<LONG_PTR>(this));
+  portable_device_notifications_.reset(
+      new PortableDeviceNotifications(window_));
 }
 
 bool RemovableDeviceNotificationsWindowWin::GetDeviceInfoForPath(
     const FilePath& path,
     base::SystemMonitor::RemovableStorageInfo* device_info) {
   string16 location;
   std::string unique_id;
   string16 name;
   bool removable;
   if (!GetDeviceInfo(path, &location, &unique_id, &name, &removable))
     return false;
 
   // To compute the device id, the device type is needed.  For removable
   // devices, that requires knowing if there's a DCIM directory, which would
   // require bouncing over to the file thread.  Instead, just iterate the
-  // devices in SystemMonitor.
+  // devices in base::SystemMonitor.
   std::string device_id;
   if (removable) {
-    std::vector<SystemMonitor::RemovableStorageInfo> attached_devices =
-        SystemMonitor::Get()->GetAttachedRemovableStorage();
+    std::vector<base::SystemMonitor::RemovableStorageInfo> attached_devices =
+        base::SystemMonitor::Get()->GetAttachedRemovableStorage();
     bool found = false;
     for (size_t i = 0; i < attached_devices.size(); i++) {
       MediaStorageUtil::Type type;
       std::string id;
       MediaStorageUtil::CrackDeviceId(attached_devices[i].device_id, &type,
                                       &id);
       if (id == unique_id) {
         found = true;
         device_id = attached_devices[i].device_id;
         break;
@@ skipping 35 matching lines @@
       break;
   }
 
   return ::DefWindowProc(hwnd, message, wparam, lparam);
 }
 
 bool RemovableDeviceNotificationsWindowWin::GetDeviceInfo(
     const FilePath& device_path, string16* device_location,
     std::string* unique_id, string16* name, bool* removable) {
   return volume_mount_watcher_->GetDeviceInfo(device_path, device_location,
-                                              unique_id, name, removable);
+                                              unique_id, name, removable) ||
+         portable_device_watcher_->GetDeviceInfo(device_path, device_location,
+                                                 unique_id, name, removable);
 }
 
 void RemovableDeviceNotificationsWindowWin::OnDeviceChange(UINT event_type,
                                                            LPARAM data) {
   volume_mount_watcher_->OnWindowMessage(event_type, data);
+  portable_device_watcher_->OnWindowMessage(event_type, data);
 }
 
 }  // namespace chrome