content/renderer/usb/web_usb_client_impl.cc

   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 "content/renderer/usb/web_usb_client_impl.h"
   6
   7 #include "base/bind.h"
   8 #include "base/callback.h"
   9 #include "base/memory/scoped_ptr.h"
  10 #include "base/move.h"
  11 #include "base/strings/utf_string_conversions.h"
  12 #include "content/child/scoped_web_callbacks.h"
  13 #include "content/renderer/usb/type_converters.h"
  14 #include "content/renderer/usb/web_usb_device_impl.h"
  15 #include "device/devices_app/public/cpp/constants.h"
  16 #include "mojo/application/public/cpp/connect.h"
  17 #include "mojo/application/public/interfaces/shell.mojom.h"
  18 #include "third_party/WebKit/public/platform/WebCallbacks.h"
  19 #include "third_party/WebKit/public/platform/WebPassOwnPtr.h"
  20 #include "third_party/WebKit/public/platform/modules/webusb/WebUSBDeviceFilter.h"
  21 #include "third_party/WebKit/public/platform/modules/webusb/WebUSBDeviceInfo.h"
  22 #include "third_party/WebKit/public/platform/modules/webusb/WebUSBDeviceRequestOptions.h"
  23 #include "third_party/WebKit/public/platform/modules/webusb/WebUSBError.h"
  24 #include "third_party/mojo/src/mojo/public/cpp/bindings/array.h"
  25 #include "third_party/mojo/src/mojo/public/cpp/bindings/interface_request.h"
  26
  27 namespace content {
  28
  29 namespace {
  30
  31 const char kNoServiceError[] = "USB service unavailable.";
  32
  33 // Generic default rejection handler for any WebUSB callbacks type. Assumes
  34 // |CallbacksType| is a blink::WebCallbacks<T, const blink::WebUSBError&>
  35 // for any type |T|.
  36 template <typename CallbacksType>
  37 void RejectCallbacksWithError(const blink::WebUSBError& error,
  38                               scoped_ptr<CallbacksType> callbacks) {
  39   callbacks->onError(error);
  40 }
  41
  42 // Create a new ScopedWebCallbacks for WebUSB client callbacks, defaulting to
  43 // a "no service" rejection.
  44 template <typename CallbacksType>
  45 ScopedWebCallbacks<CallbacksType> MakeScopedUSBCallbacks(
  46     CallbacksType* callbacks) {
  47   return make_scoped_web_callbacks(
  48       callbacks,
  49       base::Bind(&RejectCallbacksWithError<CallbacksType>,
  50                  blink::WebUSBError(blink::WebUSBError::Error::Service,
  51                                     base::UTF8ToUTF16(kNoServiceError))));
  52 }
  53
  54 void OnGetDevicesComplete(
  55     ScopedWebCallbacks<blink::WebUSBClientGetDevicesCallbacks> scoped_callbacks,
  56     mojo::ServiceProvider* device_services,
  57     mojo::Array<device::usb::DeviceInfoPtr> results) {
  58   device::usb::DeviceManagerPtr device_manager;
  59   mojo::ConnectToService(device_services, &device_manager);
  60   blink::WebVector<blink::WebUSBDevice*>* devices =
  61       new blink::WebVector<blink::WebUSBDevice*>(results.size());
  62   for (size_t i = 0; i < results.size(); ++i) {
  63     device::usb::DevicePtr device;
  64     device_manager->GetDevice(results[i]->guid, mojo::GetProxy(&device));
  65     (*devices)[i] = new WebUSBDeviceImpl(
  66         device.Pass(), mojo::ConvertTo<blink::WebUSBDeviceInfo>(results[i]));
  67   }
  68   scoped_callbacks.PassCallbacks()->onSuccess(blink::adoptWebPtr(devices));
  69 }
  70
  71 }  // namespace
  72
  73 WebUSBClientImpl::WebUSBClientImpl(mojo::ServiceProviderPtr device_services)
  74     : device_services_(device_services.Pass()) {
  75   mojo::ConnectToService(device_services_.get(), &device_manager_);
  76 }
  77
  78 WebUSBClientImpl::~WebUSBClientImpl() {}
  79
  80 void WebUSBClientImpl::getDevices(
  81     blink::WebUSBClientGetDevicesCallbacks* callbacks) {
  82   auto scoped_callbacks = MakeScopedUSBCallbacks(callbacks);
  83   // TODO(rockot): Remove this once DeviceManager is updated. It should no
  84   // longer take enumeration options.
  85   device::usb::EnumerationOptionsPtr options =
  86       device::usb::EnumerationOptions::New();
  87   options->filters = mojo::Array<device::usb::DeviceFilterPtr>::New(0);
  88   device_manager_->GetDevices(
  89       options.Pass(),
  90       base::Bind(&OnGetDevicesComplete, base::Passed(&scoped_callbacks),
  91                  base::Unretained(device_services_.get())));
  92 }
  93
  94 void WebUSBClientImpl::requestDevice(
  95     const blink::WebUSBDeviceRequestOptions& options,
  96     blink::WebUSBClientRequestDeviceCallbacks* callbacks) {
  97   callbacks->onError(blink::WebUSBError(blink::WebUSBError::Error::Service,
  98                                         base::UTF8ToUTF16("Not implemented.")));
  99   delete callbacks;
 100 }
 101
 102 void WebUSBClientImpl::setObserver(Observer* observer) {
 103   if (!observer_) {
 104     // Set up two sequential calls to GetDeviceChanges to avoid latency.
 105     device_manager_->GetDeviceChanges(base::Bind(
 106         &WebUSBClientImpl::OnDeviceChangeNotification, base::Unretained(this)));
 107     device_manager_->GetDeviceChanges(base::Bind(
 108         &WebUSBClientImpl::OnDeviceChangeNotification, base::Unretained(this)));
 109   }
 110
 111   observer_ = observer;
 112 }
 113
 114 void WebUSBClientImpl::OnDeviceChangeNotification(
 115     device::usb::DeviceChangeNotificationPtr notification) {
 116   if (!observer_)
 117     return;
 118
 119   device_manager_->GetDeviceChanges(base::Bind(
 120       &WebUSBClientImpl::OnDeviceChangeNotification, base::Unretained(this)));
 121   for (size_t i = 0; i < notification->devices_added.size(); ++i) {
 122     const device::usb::DeviceInfoPtr& device_info =
 123         notification->devices_added[i];
 124     device::usb::DevicePtr device;
 125     device_manager_->GetDevice(device_info->guid, mojo::GetProxy(&device));
 126     observer_->onDeviceConnected(blink::adoptWebPtr(new WebUSBDeviceImpl(
 127         device.Pass(), mojo::ConvertTo<blink::WebUSBDeviceInfo>(device_info))));
 128   }
 129   for (size_t i = 0; i < notification->devices_removed.size(); ++i) {
 130     const device::usb::DeviceInfoPtr& device_info =
 131         notification->devices_removed[i];
 132     device::usb::DevicePtr device;
 133     device_manager_->GetDevice(device_info->guid, mojo::GetProxy(&device));
 134     observer_->onDeviceDisconnected(blink::adoptWebPtr(new WebUSBDeviceImpl(
 135         device.Pass(), mojo::ConvertTo<blink::WebUSBDeviceInfo>(device_info))));
 136   }
 137 }
 138
 139 }  // namespace content