device/usb/usb_service_impl.h

   1 // Copyright 2015 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 #include "device/usb/usb_service.h"
   6
   7 #include <map>
   8
   9 #include "base/memory/weak_ptr.h"
  10 #include "base/single_thread_task_runner.h"
  11 #include "device/usb/usb_context.h"
  12 #include "device/usb/usb_device_impl.h"
  13 #include "third_party/libusb/src/libusb/libusb.h"
  14
  15 namespace device {
  16
  17 typedef struct libusb_device* PlatformUsbDevice;
  18 typedef struct libusb_context* PlatformUsbContext;
  19
  20 class UsbServiceImpl : public UsbService {
  21  public:
  22   static UsbService* Create(
  23       scoped_refptr<base::SingleThreadTaskRunner> ui_task_runner);
  24
  25  private:
  26   explicit UsbServiceImpl(
  27       PlatformUsbContext context,
  28       scoped_refptr<base::SingleThreadTaskRunner> ui_task_runner);
  29   ~UsbServiceImpl() override;
  30
  31   // device::UsbService implementation
  32   scoped_refptr<UsbDevice> GetDeviceById(uint32 unique_id) override;
  33   void GetDevices(std::vector<scoped_refptr<UsbDevice>>* devices) override;
  34
  35   // Enumerate USB devices from OS and update devices_ map.
  36   void RefreshDevices();
  37
  38 #if defined(OS_WIN)
  39   void RefreshDevicesIfWinUsbDevice(const std::string& device_path);
  40 #endif  // OS_WIN
  41
  42   // Adds a new UsbDevice to the devices_ map based on the given libusb device.
  43   scoped_refptr<UsbDeviceImpl> AddDevice(PlatformUsbDevice platform_device);
  44
  45   // Handle hotplug events from libusb.
  46   static int LIBUSB_CALL HotplugCallback(libusb_context* context,
  47                                          PlatformUsbDevice device,
  48                                          libusb_hotplug_event event,
  49                                          void* user_data);
  50   // These functions release a reference to the provided platform device.
  51   void OnDeviceAdded(PlatformUsbDevice platform_device);
  52   void OnDeviceRemoved(PlatformUsbDevice platform_device);
  53
  54   scoped_refptr<UsbContext> context_;
  55   scoped_refptr<base::SingleThreadTaskRunner> task_runner_;
  56   scoped_refptr<base::SingleThreadTaskRunner> ui_task_runner_;
  57
  58 #if defined(OS_WIN)
  59   class UIThreadHelper;
  60   UIThreadHelper* ui_thread_helper_;
  61 #endif  // OS_WIN
  62
  63   // TODO(reillyg): Figure out a better solution for device IDs.
  64   uint32 next_unique_id_;
  65
  66   // When available the device list will be updated when new devices are
  67   // connected instead of only when a full enumeration is requested.
  68   // TODO(reillyg): Support this on all platforms. crbug.com/411715
  69   bool hotplug_enabled_;
  70   libusb_hotplug_callback_handle hotplug_handle_;
  71
  72   // The map from unique IDs to UsbDevices.
  73   typedef std::map<uint32, scoped_refptr<UsbDeviceImpl>> DeviceMap;
  74   DeviceMap devices_;
  75
  76   // The map from PlatformUsbDevices to UsbDevices.
  77   typedef std::map<PlatformUsbDevice, scoped_refptr<UsbDeviceImpl>>
  78       PlatformDeviceMap;
  79   PlatformDeviceMap platform_devices_;
  80
  81   base::WeakPtrFactory<UsbServiceImpl> weak_factory_;
  82
  83   DISALLOW_COPY_AND_ASSIGN(UsbServiceImpl);
  84 };
  85
  86 }  // namespace device