Index: chrome/browser/usb/usb_chooser_bubble_delegate.cc |
diff --git a/chrome/browser/usb/usb_chooser_bubble_delegate.cc b/chrome/browser/usb/usb_chooser_bubble_delegate.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..6815fab81b5c0c595fa36a58405e109d918e466b |
--- /dev/null |
+++ b/chrome/browser/usb/usb_chooser_bubble_delegate.cc |
@@ -0,0 +1,152 @@ |
+// 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/usb_chooser_bubble_delegate.h" |
+ |
+#include "base/bind.h" |
+#include "base/stl_util.h" |
+#include "chrome/browser/profiles/profile.h" |
+#include "chrome/browser/usb/usb_chooser_context.h" |
+#include "chrome/browser/usb/usb_chooser_context_factory.h" |
+#include "content/public/browser/render_frame_host.h" |
+#include "content/public/browser/web_contents.h" |
+#include "device/core/device_client.h" |
+#include "device/devices_app/usb/type_converters.h" |
+#include "device/usb/usb_device.h" |
+#include "device/usb/usb_device_filter.h" |
+#include "url/gurl.h" |
+ |
+namespace { |
+ |
+// Check if the origin is in the description set. |
+bool FindOriginInDescriptorSet(const device::WebUsbDescriptorSet* set, |
+ const GURL& origin) { |
+ if (!set) |
+ return false; |
+ |
+ if (ContainsValue(set->origins, origin)) |
+ return true; |
+ |
+ for (const auto& config : set->configurations) { |
+ if (ContainsValue(config.origins, origin)) |
+ return true; |
+ |
+ for (const auto& function : config.functions) { |
+ if (ContainsValue(function.origins, origin)) |
+ return true; |
+ } |
+ } |
+ |
+ return false; |
+} |
+ |
+} // namespace |
+ |
+UsbChooserBubbleDelegate::UsbChooserBubbleDelegate( |
+ Browser* browser, |
+ mojo::Array<device::usb::DeviceFilterPtr> device_filters, |
+ content::RenderFrameHost* render_frame_host, |
+ const webusb::WebUsbPermissionBubble::GetPermissionCallback& callback) |
+ : ChooserBubbleDelegate(browser), |
+ render_frame_host_(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>>(); |
+ |
+ usb_service->GetDevices(base::Bind( |
+ &UsbChooserBubbleDelegate::GotUsbDeviceList, weak_factory_.GetWeakPtr())); |
+} |
+ |
+UsbChooserBubbleDelegate::~UsbChooserBubbleDelegate() { |
+ if (!callback_.is_null()) |
+ callback_.Run(nullptr); |
+} |
+ |
+const std::vector<base::string16>& UsbChooserBubbleDelegate::GetOptions() |
+ const { |
+ return devices_names_; |
+} |
+ |
+void UsbChooserBubbleDelegate::Select(int index) { |
+ size_t idx = static_cast<size_t>(index); |
+ size_t num_options = devices_.size(); |
+ DCHECK_LT(idx, num_options); |
+ if (idx < num_options) { |
+ 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, |
+ devices_[idx]->guid()); |
+ |
+ device::usb::DeviceInfoPtr device_info_ptr = |
+ device::usb::DeviceInfo::From(*devices_[idx]); |
+ callback_.Run(device_info_ptr.Pass()); |
+ } else { |
+ callback_.Run(nullptr); |
+ } |
+ callback_.reset(); // Reset |callback_| so that it is only run once. |
+} |
+ |
+void UsbChooserBubbleDelegate::Cancel() {} |
+ |
+void UsbChooserBubbleDelegate::Close() {} |
+ |
+void UsbChooserBubbleDelegate::OnDeviceAdded( |
+ scoped_refptr<device::UsbDevice> device) { |
+ DCHECK(!ContainsValue(devices_, device)); |
+ if (device::UsbDeviceFilter::MatchesAny(device, filters_) && |
+ FindOriginInDescriptorSet( |
+ device->webusb_allowed_origins(), |
+ render_frame_host_->GetLastCommittedURL().GetOrigin())) { |
+ devices_.push_back(device); |
+ devices_names_.push_back(device->product_string()); |
+ if (observer()) |
+ observer()->OnOptionAdded(static_cast<int>(devices_names_.size()) - 1); |
+ } |
+} |
+ |
+void UsbChooserBubbleDelegate::OnDeviceRemoved( |
+ scoped_refptr<device::UsbDevice> device) { |
+ auto iter = std::find(devices_.begin(), devices_.end(), device); |
+ if (iter != devices_.end()) { |
+ int index = iter - devices_.begin(); |
+ devices_.erase(iter); |
+ devices_names_.erase(devices_names_.begin() + index); |
+ if (observer()) |
+ observer()->OnOptionRemoved(index); |
+ } |
+} |
+ |
+// Get a list of devices that can be shown in the chooser bubble UI for |
+// user to grant permsssion. |
+void UsbChooserBubbleDelegate::GotUsbDeviceList( |
+ const std::vector<scoped_refptr<device::UsbDevice>>& devices) { |
+ for (const auto& device : devices) { |
+ if (device::UsbDeviceFilter::MatchesAny(device, filters_) && |
+ FindOriginInDescriptorSet( |
+ device->webusb_allowed_origins(), |
+ render_frame_host_->GetLastCommittedURL().GetOrigin())) { |
+ devices_.push_back(device); |
+ devices_names_.push_back(device->product_string()); |
+ } |
+ } |
+ if (observer()) |
+ observer()->OnOptionsInitialized(); |
+} |