WebUsb Android chooser UI

chrome/browser/ui/android/usb_chooser_android.cc

+// Copyright 2016 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/ui/android/usb_chooser_android.h"
+
+#include <stddef.h>
+
+#include <algorithm>
+
+#include "base/android/jni_android.h"
+#include "base/android/jni_array.h"
+#include "base/android/jni_string.h"
+#include "base/bind.h"
+#include "base/strings/utf_string_conversions.h"
+#include "chrome/browser/profiles/profile.h"
+#include "chrome/browser/ssl/chrome_security_state_model_client.h"
+#include "chrome/browser/ui/android/view_android_helper.h"
+#include "chrome/browser/usb/usb_chooser_context.h"
+#include "chrome/browser/usb/usb_chooser_context_factory.h"
+#include "chrome/browser/usb/web_usb_histograms.h"
+#include "chrome/common/url_constants.h"
+#include "content/public/browser/android/content_view_core.h"
+#include "content/public/browser/render_frame_host.h"
+#include "content/public/browser/web_contents.h"
+#include "device/core/device_client.h"
+#include "device/usb/mojo/type_converters.h"
+#include "device/usb/usb_device.h"
+#include "device/usb/usb_device_filter.h"
+#include "device/usb/webusb_descriptors.h"
+#include "jni/UsbChooserDialog_jni.h"
+#include "ui/android/window_android.h"
+#include "url/gurl.h"
+#include "url/origin.h"
+
+UsbChooserAndroid::UsbChooserAndroid(
+    mojo::Array<device::usb::DeviceFilterPtr> device_filters,
+    content::RenderFrameHost* render_frame_host,
+    const device::usb::ChooserService::GetPermissionCallback& callback)
+    : render_frame_host_(render_frame_host),
+      web_contents_(

[Yaron, 2016/03/07 21:51:49]

Don't think it's worth holding on to this given you're already caching
RenderFrameHost

[juncai, 2016/03/09 01:26:51]

Done.

+          content::WebContents::FromRenderFrameHost(render_frame_host)),
+      callback_(callback),
+      usb_service_observer_(this),
+      weak_factory_(this) {
+  device::UsbService* usb_service =
+      device::DeviceClient::Get()->GetUsbService();
+  if (!usb_service)
+    return;
+
+  if (!usb_service_observer_.IsObserving(usb_service))
+    usb_service_observer_.Add(usb_service);
+
+  if (!device_filters.is_null())
+    filters_ = device_filters.To<std::vector<device::UsbDeviceFilter>>();
+
+  // Create (and show) the UsbChooser dialog.
+  base::android::ScopedJavaLocalRef<jobject> window_android =
+      content::ContentViewCore::FromWebContents(web_contents_)
+          ->GetWindowAndroid()
+          ->GetJavaObject();
+  JNIEnv* env = base::android::AttachCurrentThread();
+  const url::Origin origin = render_frame_host->GetLastCommittedOrigin();
+  DCHECK(!origin.unique());
+  base::android::ScopedJavaLocalRef<jstring> origin_string =

[Yaron, 2016/03/07 21:51:49]

I believe this should use url_formatter::FormatUrlForSecurityDisplay

[Finnur, 2016/03/08 10:26:20]

Then BluetoothChooserAndroid ctor needs a corresponding change.

[juncai, 2016/03/09 01:26:51]

Done.

[juncai, 2016/03/09 01:26:51]

A patch was created to fix it at BluetoothChooserAndroid ctor:
https://codereview.chromium.org/1771353004/

+      base::android::ConvertUTF8ToJavaString(env, origin.Serialize());
+  ChromeSecurityStateModelClient* security_model_client =
+      ChromeSecurityStateModelClient::FromWebContents(web_contents_);
+  DCHECK(security_model_client);
+  java_dialog_.Reset(Java_UsbChooserDialog_create(
+      env, window_android.obj(), origin_string.obj(),
+      security_model_client->GetSecurityInfo().security_level,
+      reinterpret_cast<intptr_t>(this)));
+
+  usb_service->GetDevices(base::Bind(&UsbChooserAndroid::GotUsbDeviceList,
+                                     weak_factory_.GetWeakPtr()));
+}
+
+UsbChooserAndroid::~UsbChooserAndroid() {
+  if (!callback_.is_null())
+    callback_.Run(nullptr);
+
+  if (!java_dialog_.is_null()) {
+    Java_UsbChooserDialog_closeDialog(base::android::AttachCurrentThread(),
+                                      java_dialog_.obj());
+  }
+}
+
+void UsbChooserAndroid::OnDeviceAdded(scoped_refptr<device::UsbDevice> device) {
+  if (device::UsbDeviceFilter::MatchesAny(device, filters_) &&
+      FindInWebUsbAllowedOrigins(
+          device->webusb_allowed_origins(),
+          render_frame_host_->GetLastCommittedURL().GetOrigin())) {
+    devices_.push_back(device);
+    device_guids_.push_back(device->guid());
+    device_names_.push_back(device->product_string());
+    UpdateChooserDialog();
+  }
+}
+
+void UsbChooserAndroid::OnDeviceRemoved(
+    scoped_refptr<device::UsbDevice> device) {
+  auto it = std::find(devices_.begin(), devices_.end(), device);
+  if (it != devices_.end()) {
+    size_t index = it - devices_.begin();
+    devices_.erase(it);
+    device_guids_.erase(device_guids_.begin() + index);
+    device_names_.erase(device_names_.begin() + index);
+    UpdateChooserDialog();
+  }
+}
+
+void UsbChooserAndroid::Select(const std::string& guid) {
+  auto it = std::find(device_guids_.begin(), device_guids_.end(), guid);
+  if (it != device_guids_.end()) {
+    size_t index = it - device_guids_.begin();
+    content::WebContents* web_contents =
+        content::WebContents::FromRenderFrameHost(render_frame_host_);
+    GURL embedding_origin =
+        web_contents->GetMainFrame()->GetLastCommittedURL().GetOrigin();
+    Profile* profile =
+        Profile::FromBrowserContext(web_contents->GetBrowserContext());
+    UsbChooserContext* chooser_context =
+        UsbChooserContextFactory::GetForProfile(profile);
+    chooser_context->GrantDevicePermission(
+        render_frame_host_->GetLastCommittedURL().GetOrigin(), embedding_origin,
+        *it);
+    device::usb::DeviceInfoPtr device_info_ptr =
+        device::usb::DeviceInfo::From(*devices_[index]);
+    callback_.Run(std::move(device_info_ptr));
+    callback_.reset();  // Reset |callback_| so that it is only run once.
+    Java_UsbChooserDialog_closeDialog(base::android::AttachCurrentThread(),
+                                      java_dialog_.obj());
+
+    RecordWebUsbChooserClosure(
+        devices_[index]->serial_number().empty()
+            ? WEBUSB_CHOOSER_CLOSED_EPHEMERAL_PERMISSION_GRANTED
+            : WEBUSB_CHOOSER_CLOSED_PERMISSION_GRANTED);
+  }
+}
+
+void UsbChooserAndroid::Cancel() {
+  callback_.Run(nullptr);
+  callback_.reset();  // Reset |callback_| so that it is only run once.
+  Java_UsbChooserDialog_closeDialog(base::android::AttachCurrentThread(),
+                                    java_dialog_.obj());
+
+  RecordWebUsbChooserClosure(devices_.size() == 0
+                                 ? WEBUSB_CHOOSER_CLOSED_CANCELLED_NO_DEVICES
+                                 : WEBUSB_CHOOSER_CLOSED_CANCELLED);
+}
+
+// Get a list of devices that can be shown in the chooser bubble UI for
+// user to grant permsssion.
+void UsbChooserAndroid::GotUsbDeviceList(
+    const std::vector<scoped_refptr<device::UsbDevice>>& devices) {
+  for (const auto& device : devices) {
+    if (device::UsbDeviceFilter::MatchesAny(device, filters_) &&
+        FindInWebUsbAllowedOrigins(
+            device->webusb_allowed_origins(),
+            render_frame_host_->GetLastCommittedURL().GetOrigin())) {
+      devices_.push_back(device);
+      device_guids_.push_back(device->guid());
+      device_names_.push_back(device->product_string());
+    }
+  }
+  UpdateChooserDialog();
+}
+
+void UsbChooserAndroid::OnDialogFinished(
+    JNIEnv* env,
+    const JavaParamRef<jobject>& obj,
+    jint event_type,
+    const JavaParamRef<jstring>& device_id) {
+  // Values are defined in UsbChooserDialog as DIALOG_FINISHED constants.
+  switch (event_type) {
+    case 0:
+      Cancel();
+      return;
+    case 1:
+      Select(base::android::ConvertJavaStringToUTF8(env, device_id));
+      return;
+  }
+  NOTREACHED();
+}
+
+void UsbChooserAndroid::ShowUsbOverviewLink(
+    JNIEnv* env,
+    const base::android::JavaParamRef<jobject>& obj) {
+  OpenUrl(chrome::kChooserUsbOverviewURL);
+}
+
+void UsbChooserAndroid::UpdateChooserDialog() const {
+  JNIEnv* env = base::android::AttachCurrentThread();
+  base::android::ScopedJavaLocalRef<jobjectArray> guids =
+      base::android::ToJavaArrayOfStrings(env, device_guids_);
+  base::android::ScopedJavaLocalRef<jobjectArray> names =
+      base::android::ToJavaArrayOfStrings(env, device_names_);
+  Java_UsbChooserDialog_updateOptions(env, java_dialog_.obj(), guids.obj(),
+                                      names.obj());
+}
+
+void UsbChooserAndroid::OpenUrl(const std::string& url) {
+  web_contents_->OpenURL(content::OpenURLParams(
+      GURL(url), content::Referrer(), NEW_FOREGROUND_TAB,
+      ui::PAGE_TRANSITION_AUTO_TOPLEVEL, false /* is_renderer_initiated */));
+}
+
+// static
+bool UsbChooserAndroid::Register(JNIEnv* env) {
+  return RegisterNativesImpl(env);
+}