Store USB device permissions in website settings.

chrome/browser/usb/web_usb_permission_provider.cc

 // Copyright 2015 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/usb/web_usb_permission_provider.h"
 
 #include "base/command_line.h"
 #include "chrome/browser/profiles/profile.h"
+#include "chrome/browser/usb/usb_chooser_permission_context.h"
+#include "chrome/browser/usb/usb_chooser_permission_context_factory.h"
 #include "chrome/common/chrome_switches.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/render_frame_host.h"
+#include "content/public/browser/render_process_host.h"
+#include "content/public/browser/web_contents.h"
 #include "device/core/device_client.h"
 
+using content::RenderFrameHost;
+using content::WebContents;
 using device::usb::WebUsbDescriptorSet;
 using device::usb::WebUsbConfigurationSubsetPtr;
 using device::usb::WebUsbFunctionSubsetPtr;
 
 namespace {
 
 bool FindOriginInDescriptorSet(const WebUsbDescriptorSet* set,
                                const GURL& origin,
                                const uint8_t* configuration_value,
                                const uint8_t* interface_number) {
@@ skipping 31 matching lines @@
 
 bool EnableWebUsbOnAnyOrigin() {
   return base::CommandLine::ForCurrentProcess()->HasSwitch(
       switches::kEnableWebUsbOnAnyOrigin);
 }
 
 }  // namespace
 
 // static
 void WebUSBPermissionProvider::Create(
-    content::RenderFrameHost* render_frame_host,
+    RenderFrameHost* render_frame_host,
     mojo::InterfaceRequest<PermissionProvider> request) {
   DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
   DCHECK(render_frame_host);
 
   // The created object is owned by its bindings.
   new WebUSBPermissionProvider(render_frame_host, request.Pass());
 }
 
 WebUSBPermissionProvider::~WebUSBPermissionProvider() {}
 
 WebUSBPermissionProvider::WebUSBPermissionProvider(
-    content::RenderFrameHost* render_frame_host,
+    RenderFrameHost* render_frame_host,
     mojo::InterfaceRequest<PermissionProvider> request)
     : render_frame_host_(render_frame_host) {
   bindings_.set_connection_error_handler(base::Bind(
       &WebUSBPermissionProvider::OnConnectionError, base::Unretained(this)));
   bindings_.AddBinding(this, request.Pass());
 }
 
 void WebUSBPermissionProvider::HasDevicePermission(
     mojo::Array<device::usb::DeviceInfoPtr> requested_devices,
     const HasDevicePermissionCallback& callback) {
   DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
-  GURL origin = render_frame_host_->GetLastCommittedURL().GetOrigin();
+  mojo::Array<mojo::String> allowed_guids(0);
+  WebContents* web_contents =
+      WebContents::FromRenderFrameHost(render_frame_host_);
+  if (!web_contents) {
+    callback.Run(allowed_guids.Pass());
+    return;
+  }
 
-  mojo::Array<mojo::String> allowed_guids(0);
+  GURL embedding_origin =
+      web_contents->GetMainFrame()->GetLastCommittedURL().GetOrigin();
+  GURL requesting_origin =
+      render_frame_host_->GetLastCommittedURL().GetOrigin();
+  Profile* profile = Profile::FromBrowserContext(
+      render_frame_host_->GetProcess()->GetBrowserContext());
+  UsbChooserPermissionContext* permission_context =
+      UsbChooserPermissionContextFactory::GetForProfile(profile);
+
   for (size_t i = 0; i < requested_devices.size(); ++i) {
     const device::usb::DeviceInfoPtr& device = requested_devices[i];
-    if (FindOriginInDescriptorSet(device->webusb_allowed_origins.get(), origin,
-                                  nullptr, nullptr) &&
-        EnableWebUsbOnAnyOrigin())
+    if (FindOriginInDescriptorSet(device->webusb_allowed_origins.get(),
+                                  requesting_origin, nullptr, nullptr) &&
+        (EnableWebUsbOnAnyOrigin() ||
+         permission_context->HasDevicePermission(
+             requesting_origin, embedding_origin, device->guid)))
       allowed_guids.push_back(device->guid);
   }
   callback.Run(allowed_guids.Pass());
 }
 
 void WebUSBPermissionProvider::HasConfigurationPermission(
     uint8_t requested_configuration_value,
     device::usb::DeviceInfoPtr device,
     const HasInterfacePermissionCallback& callback) {
   DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
@@ skipping 17 matching lines @@
 
 void WebUSBPermissionProvider::Bind(
     mojo::InterfaceRequest<device::usb::PermissionProvider> request) {
   bindings_.AddBinding(this, request.Pass());
 }
 
 void WebUSBPermissionProvider::OnConnectionError() {
   if (bindings_.empty())
     delete this;
 }