Donna Wu | 2fd87fc9 | 2018-07-27 01:32:19 | [diff] [blame] | 1 | // Copyright 2018 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 | |
Donna Wu | 838ac36 | 2018-08-10 10:36:33 | [diff] [blame] | 5 | #include "chrome/browser/usb/web_usb_service_impl.h" |
Donna Wu | 2fd87fc9 | 2018-07-27 01:32:19 | [diff] [blame] | 6 | |
Donna Wu | c702987 | 2018-11-02 09:47:05 | [diff] [blame] | 7 | #include <memory> |
Donna Wu | 2fd87fc9 | 2018-07-27 01:32:19 | [diff] [blame] | 8 | #include <utility> |
| 9 | |
| 10 | #include "base/bind.h" |
Donna Wu | c702987 | 2018-11-02 09:47:05 | [diff] [blame] | 11 | #include "base/stl_util.h" |
Donna Wu | 50fe6b2 | 2018-08-30 06:27:43 | [diff] [blame] | 12 | #include "chrome/browser/profiles/profile.h" |
| 13 | #include "chrome/browser/usb/usb_blocklist.h" |
Donna Wu | 50fe6b2 | 2018-08-30 06:27:43 | [diff] [blame] | 14 | #include "chrome/browser/usb/usb_chooser_context_factory.h" |
| 15 | #include "chrome/browser/usb/usb_tab_helper.h" |
| 16 | #include "content/public/browser/browser_thread.h" |
Takuto Ikuta | a6629ba6 | 2019-01-30 23:31:41 | [diff] [blame^] | 17 | #include "device/usb/public/mojom/device_enumeration_options.mojom.h" |
| 18 | #include "media/mojo/interfaces/remoting_common.mojom.h" |
Donna Wu | 2fd87fc9 | 2018-07-27 01:32:19 | [diff] [blame] | 19 | |
Donna Wu | 838ac36 | 2018-08-10 10:36:33 | [diff] [blame] | 20 | WebUsbServiceImpl::WebUsbServiceImpl( |
Donna Wu | 50fe6b2 | 2018-08-30 06:27:43 | [diff] [blame] | 21 | content::RenderFrameHost* render_frame_host, |
Donna Wu | c98e12a | 2018-08-16 07:24:02 | [diff] [blame] | 22 | base::WeakPtr<WebUsbChooser> usb_chooser) |
Donna Wu | 50fe6b2 | 2018-08-30 06:27:43 | [diff] [blame] | 23 | : render_frame_host_(render_frame_host), |
Donna Wu | c98e12a | 2018-08-16 07:24:02 | [diff] [blame] | 24 | usb_chooser_(std::move(usb_chooser)), |
Ovidio Henriquez | 8b39653b | 2019-01-18 18:56:53 | [diff] [blame] | 25 | device_observer_(this), |
| 26 | permission_observer_(this), |
Donna Wu | 2fd87fc9 | 2018-07-27 01:32:19 | [diff] [blame] | 27 | weak_factory_(this) { |
Donna Wu | 35c12ea | 2018-09-20 06:08:18 | [diff] [blame] | 28 | DCHECK(render_frame_host_); |
| 29 | content::WebContents* web_contents = |
| 30 | content::WebContents::FromRenderFrameHost(render_frame_host_); |
Donna Wu | c702987 | 2018-11-02 09:47:05 | [diff] [blame] | 31 | // This class is destroyed on cross-origin navigations and so it is safe to |
| 32 | // cache these values. |
| 33 | requesting_origin_ = render_frame_host_->GetLastCommittedURL().GetOrigin(); |
| 34 | embedding_origin_ = |
| 35 | web_contents->GetMainFrame()->GetLastCommittedURL().GetOrigin(); |
Donna Wu | 35c12ea | 2018-09-20 06:08:18 | [diff] [blame] | 36 | Profile* profile = |
| 37 | Profile::FromBrowserContext(web_contents->GetBrowserContext()); |
| 38 | chooser_context_ = UsbChooserContextFactory::GetForProfile(profile); |
| 39 | DCHECK(chooser_context_); |
Donna Wu | 50fe6b2 | 2018-08-30 06:27:43 | [diff] [blame] | 40 | |
| 41 | bindings_.set_connection_error_handler(base::BindRepeating( |
| 42 | &WebUsbServiceImpl::OnBindingConnectionError, base::Unretained(this))); |
| 43 | } |
| 44 | |
| 45 | WebUsbServiceImpl::~WebUsbServiceImpl() = default; |
| 46 | |
| 47 | void WebUsbServiceImpl::BindRequest( |
| 48 | blink::mojom::WebUsbServiceRequest request) { |
| 49 | bindings_.AddBinding(this, std::move(request)); |
| 50 | |
Donna Wu | 35c12ea | 2018-09-20 06:08:18 | [diff] [blame] | 51 | // Listen to UsbChooserContext for add/remove device events from UsbService. |
Donna Wu | 838ac36 | 2018-08-10 10:36:33 | [diff] [blame] | 52 | // We can't set WebUsbServiceImpl as a UsbDeviceManagerClient because |
Donna Wu | 2fd87fc9 | 2018-07-27 01:32:19 | [diff] [blame] | 53 | // the OnDeviceRemoved event will be delivered here after it is delivered |
| 54 | // to UsbChooserContext, meaning that all ephemeral permission checks in |
| 55 | // OnDeviceRemoved() will fail. |
Ovidio Henriquez | 8b39653b | 2019-01-18 18:56:53 | [diff] [blame] | 56 | if (!device_observer_.IsObservingSources()) |
| 57 | device_observer_.Add(chooser_context_); |
| 58 | if (!permission_observer_.IsObservingSources()) |
| 59 | permission_observer_.Add(chooser_context_); |
Donna Wu | 2fd87fc9 | 2018-07-27 01:32:19 | [diff] [blame] | 60 | } |
| 61 | |
Donna Wu | 50fe6b2 | 2018-08-30 06:27:43 | [diff] [blame] | 62 | bool WebUsbServiceImpl::HasDevicePermission( |
| 63 | const device::mojom::UsbDeviceInfo& device_info) const { |
| 64 | DCHECK_CURRENTLY_ON(content::BrowserThread::UI); |
| 65 | DCHECK(render_frame_host_); |
Donna Wu | 35c12ea | 2018-09-20 06:08:18 | [diff] [blame] | 66 | if (!chooser_context_) |
| 67 | return false; |
Donna Wu | 50fe6b2 | 2018-08-30 06:27:43 | [diff] [blame] | 68 | |
Donna Wu | c702987 | 2018-11-02 09:47:05 | [diff] [blame] | 69 | return chooser_context_->HasDevicePermission(requesting_origin_, |
| 70 | embedding_origin_, device_info); |
Donna Wu | 50fe6b2 | 2018-08-30 06:27:43 | [diff] [blame] | 71 | } |
Donna Wu | 2fd87fc9 | 2018-07-27 01:32:19 | [diff] [blame] | 72 | |
Donna Wu | 838ac36 | 2018-08-10 10:36:33 | [diff] [blame] | 73 | void WebUsbServiceImpl::GetDevices(GetDevicesCallback callback) { |
Donna Wu | 35c12ea | 2018-09-20 06:08:18 | [diff] [blame] | 74 | if (!chooser_context_) |
| 75 | std::move(callback).Run(std::vector<device::mojom::UsbDeviceInfoPtr>()); |
| 76 | |
| 77 | chooser_context_->GetDevices(base::BindOnce(&WebUsbServiceImpl::OnGetDevices, |
| 78 | weak_factory_.GetWeakPtr(), |
| 79 | std::move(callback))); |
Donna Wu | caa44de | 2018-08-17 06:42:39 | [diff] [blame] | 80 | } |
| 81 | |
| 82 | void WebUsbServiceImpl::OnGetDevices( |
| 83 | GetDevicesCallback callback, |
| 84 | std::vector<device::mojom::UsbDeviceInfoPtr> device_info_list) { |
| 85 | std::vector<device::mojom::UsbDeviceInfoPtr> device_infos; |
| 86 | for (auto& device_info : device_info_list) { |
Donna Wu | 50fe6b2 | 2018-08-30 06:27:43 | [diff] [blame] | 87 | if (HasDevicePermission(*device_info)) |
Donna Wu | caa44de | 2018-08-17 06:42:39 | [diff] [blame] | 88 | device_infos.push_back(device_info.Clone()); |
Donna Wu | caa44de | 2018-08-17 06:42:39 | [diff] [blame] | 89 | } |
| 90 | std::move(callback).Run(std::move(device_infos)); |
Donna Wu | 2fd87fc9 | 2018-07-27 01:32:19 | [diff] [blame] | 91 | } |
| 92 | |
Donna Wu | 838ac36 | 2018-08-10 10:36:33 | [diff] [blame] | 93 | void WebUsbServiceImpl::GetDevice( |
Donna Wu | 2fd87fc9 | 2018-07-27 01:32:19 | [diff] [blame] | 94 | const std::string& guid, |
| 95 | device::mojom::UsbDeviceRequest device_request) { |
Donna Wu | 35c12ea | 2018-09-20 06:08:18 | [diff] [blame] | 96 | if (!chooser_context_) |
| 97 | return; |
| 98 | |
Donna Wu | c702987 | 2018-11-02 09:47:05 | [diff] [blame] | 99 | auto* device_info = chooser_context_->GetDeviceInfo(guid); |
| 100 | if (!device_info || !HasDevicePermission(*device_info)) |
| 101 | return; |
| 102 | |
| 103 | // Connect Blink to the native device and keep a binding to this for the |
| 104 | // UsbDeviceClient interface so we can receive DeviceOpened/Closed events. |
| 105 | // This binding will also be closed to notify the device service to close |
| 106 | // the connection if permission is revoked. |
Donna Wu | 656439c | 2018-08-13 09:31:16 | [diff] [blame] | 107 | device::mojom::UsbDeviceClientPtr device_client; |
Donna Wu | c702987 | 2018-11-02 09:47:05 | [diff] [blame] | 108 | device_client_bindings_[guid].AddBinding(this, |
| 109 | mojo::MakeRequest(&device_client)); |
Donna Wu | 35c12ea | 2018-09-20 06:08:18 | [diff] [blame] | 110 | chooser_context_->GetDevice(guid, std::move(device_request), |
| 111 | std::move(device_client)); |
Donna Wu | 2fd87fc9 | 2018-07-27 01:32:19 | [diff] [blame] | 112 | } |
| 113 | |
Donna Wu | c98e12a | 2018-08-16 07:24:02 | [diff] [blame] | 114 | void WebUsbServiceImpl::GetPermission( |
| 115 | std::vector<device::mojom::UsbDeviceFilterPtr> device_filters, |
| 116 | GetPermissionCallback callback) { |
| 117 | if (!usb_chooser_) |
| 118 | std::move(callback).Run(nullptr); |
| 119 | |
| 120 | usb_chooser_->GetPermission(std::move(device_filters), std::move(callback)); |
| 121 | } |
| 122 | |
Donna Wu | 838ac36 | 2018-08-10 10:36:33 | [diff] [blame] | 123 | void WebUsbServiceImpl::SetClient( |
Donna Wu | 7205a85 | 2018-09-20 04:21:13 | [diff] [blame] | 124 | device::mojom::UsbDeviceManagerClientAssociatedPtrInfo client) { |
| 125 | DCHECK(client); |
| 126 | |
| 127 | device::mojom::UsbDeviceManagerClientAssociatedPtr client_ptr; |
| 128 | client_ptr.Bind(std::move(client)); |
| 129 | clients_.AddPtr(std::move(client_ptr)); |
Donna Wu | 2fd87fc9 | 2018-07-27 01:32:19 | [diff] [blame] | 130 | } |
| 131 | |
Ovidio Henriquez | 8b39653b | 2019-01-18 18:56:53 | [diff] [blame] | 132 | void WebUsbServiceImpl::OnPermissionRevoked(const GURL& requesting_origin, |
| 133 | const GURL& embedding_origin) { |
| 134 | if (requesting_origin_ != requesting_origin || |
| 135 | embedding_origin_ != embedding_origin) { |
| 136 | return; |
| 137 | } |
| 138 | |
| 139 | // Close the connection between Blink and the device if the device lost |
| 140 | // permission. |
| 141 | base::EraseIf(device_client_bindings_, [this](const auto& entry) { |
| 142 | auto* device_info = chooser_context_->GetDeviceInfo(entry.first); |
| 143 | if (!device_info) |
| 144 | return true; |
| 145 | |
| 146 | return !HasDevicePermission(*device_info); |
| 147 | }); |
| 148 | } |
| 149 | |
Donna Wu | 35c12ea | 2018-09-20 06:08:18 | [diff] [blame] | 150 | void WebUsbServiceImpl::OnDeviceAdded( |
Donna Wu | c42ee16b | 2018-09-30 05:19:51 | [diff] [blame] | 151 | const device::mojom::UsbDeviceInfo& device_info) { |
| 152 | if (!HasDevicePermission(device_info)) |
Donna Wu | 50fe6b2 | 2018-08-30 06:27:43 | [diff] [blame] | 153 | return; |
| 154 | |
| 155 | clients_.ForAllPtrs( |
| 156 | [&device_info](device::mojom::UsbDeviceManagerClient* client) { |
Donna Wu | c42ee16b | 2018-09-30 05:19:51 | [diff] [blame] | 157 | client->OnDeviceAdded(device_info.Clone()); |
Donna Wu | 50fe6b2 | 2018-08-30 06:27:43 | [diff] [blame] | 158 | }); |
Donna Wu | 2fd87fc9 | 2018-07-27 01:32:19 | [diff] [blame] | 159 | } |
| 160 | |
Donna Wu | 838ac36 | 2018-08-10 10:36:33 | [diff] [blame] | 161 | void WebUsbServiceImpl::OnDeviceRemoved( |
Donna Wu | c42ee16b | 2018-09-30 05:19:51 | [diff] [blame] | 162 | const device::mojom::UsbDeviceInfo& device_info) { |
Donna Wu | c702987 | 2018-11-02 09:47:05 | [diff] [blame] | 163 | device_client_bindings_.erase(device_info.guid); |
Donna Wu | c42ee16b | 2018-09-30 05:19:51 | [diff] [blame] | 164 | if (!HasDevicePermission(device_info)) |
Donna Wu | 50fe6b2 | 2018-08-30 06:27:43 | [diff] [blame] | 165 | return; |
| 166 | |
| 167 | clients_.ForAllPtrs( |
| 168 | [&device_info](device::mojom::UsbDeviceManagerClient* client) { |
Donna Wu | c42ee16b | 2018-09-30 05:19:51 | [diff] [blame] | 169 | client->OnDeviceRemoved(device_info.Clone()); |
Donna Wu | 50fe6b2 | 2018-08-30 06:27:43 | [diff] [blame] | 170 | }); |
Donna Wu | 2fd87fc9 | 2018-07-27 01:32:19 | [diff] [blame] | 171 | } |
| 172 | |
Donna Wu | 35c12ea | 2018-09-20 06:08:18 | [diff] [blame] | 173 | void WebUsbServiceImpl::OnDeviceManagerConnectionError() { |
| 174 | // Close the connection with blink. |
| 175 | clients_.CloseAll(); |
| 176 | bindings_.CloseAllBindings(); |
| 177 | |
| 178 | // Remove itself from UsbChooserContext's ObserverList. |
Ovidio Henriquez | 8b39653b | 2019-01-18 18:56:53 | [diff] [blame] | 179 | device_observer_.RemoveAll(); |
| 180 | permission_observer_.RemoveAll(); |
Donna Wu | 35c12ea | 2018-09-20 06:08:18 | [diff] [blame] | 181 | } |
| 182 | |
Donna Wu | 656439c | 2018-08-13 09:31:16 | [diff] [blame] | 183 | // device::mojom::UsbDeviceClient implementation: |
| 184 | void WebUsbServiceImpl::OnDeviceOpened() { |
Donna Wu | 50fe6b2 | 2018-08-30 06:27:43 | [diff] [blame] | 185 | DCHECK_CURRENTLY_ON(content::BrowserThread::UI); |
| 186 | content::WebContents* web_contents = |
| 187 | content::WebContents::FromRenderFrameHost(render_frame_host_); |
| 188 | UsbTabHelper* tab_helper = UsbTabHelper::FromWebContents(web_contents); |
| 189 | tab_helper->IncrementConnectionCount(render_frame_host_); |
Donna Wu | 656439c | 2018-08-13 09:31:16 | [diff] [blame] | 190 | } |
| 191 | |
| 192 | void WebUsbServiceImpl::OnDeviceClosed() { |
Donna Wu | 50fe6b2 | 2018-08-30 06:27:43 | [diff] [blame] | 193 | DCHECK_CURRENTLY_ON(content::BrowserThread::UI); |
| 194 | content::WebContents* web_contents = |
| 195 | content::WebContents::FromRenderFrameHost(render_frame_host_); |
| 196 | UsbTabHelper* tab_helper = UsbTabHelper::FromWebContents(web_contents); |
| 197 | tab_helper->DecrementConnectionCount(render_frame_host_); |
Donna Wu | 656439c | 2018-08-13 09:31:16 | [diff] [blame] | 198 | } |
| 199 | |
Donna Wu | 50fe6b2 | 2018-08-30 06:27:43 | [diff] [blame] | 200 | void WebUsbServiceImpl::OnBindingConnectionError() { |
Ovidio Henriquez | 8b39653b | 2019-01-18 18:56:53 | [diff] [blame] | 201 | if (bindings_.empty()) { |
| 202 | device_observer_.RemoveAll(); |
| 203 | permission_observer_.RemoveAll(); |
| 204 | } |
Donna Wu | 2fd87fc9 | 2018-07-27 01:32:19 | [diff] [blame] | 205 | } |