| OLD | NEW |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/browser/usb/usb_chooser_bubble_controller.h" | 5 #include "chrome/browser/usb/usb_chooser_controller.h" |
| 6 | 6 |
| 7 #include <stddef.h> | 7 #include <stddef.h> |
| 8 #include <utility> | 8 #include <utility> |
| 9 | 9 |
| 10 #include "base/bind.h" | 10 #include "base/bind.h" |
| 11 #include "base/command_line.h" | 11 #include "base/command_line.h" |
| 12 #include "chrome/browser/net/referrer.h" |
| 12 #include "chrome/browser/profiles/profile.h" | 13 #include "chrome/browser/profiles/profile.h" |
| 14 #include "chrome/browser/profiles/profile_manager.h" |
| 15 #include "chrome/browser/ui/browser.h" |
| 16 #include "chrome/browser/ui/scoped_tabbed_browser_displayer.h" |
| 13 #include "chrome/browser/usb/usb_chooser_context.h" | 17 #include "chrome/browser/usb/usb_chooser_context.h" |
| 14 #include "chrome/browser/usb/usb_chooser_context_factory.h" | 18 #include "chrome/browser/usb/usb_chooser_context_factory.h" |
| 15 #include "chrome/browser/usb/web_usb_histograms.h" | 19 #include "chrome/browser/usb/web_usb_histograms.h" |
| 16 #include "chrome/common/chrome_switches.h" | 20 #include "chrome/common/chrome_switches.h" |
| 17 #include "chrome/common/url_constants.h" | 21 #include "chrome/common/url_constants.h" |
| 18 #include "components/bubble/bubble_controller.h" | |
| 19 #include "content/public/browser/render_frame_host.h" | 22 #include "content/public/browser/render_frame_host.h" |
| 20 #include "content/public/browser/web_contents.h" | 23 #include "content/public/browser/web_contents.h" |
| 21 #include "device/core/device_client.h" | 24 #include "device/core/device_client.h" |
| 22 #include "device/usb/mojo/type_converters.h" | 25 #include "device/usb/mojo/type_converters.h" |
| 23 #include "device/usb/usb_device.h" | 26 #include "device/usb/usb_device.h" |
| 24 #include "device/usb/usb_device_filter.h" | 27 #include "device/usb/usb_device_filter.h" |
| 25 #include "device/usb/webusb_descriptors.h" | 28 #include "device/usb/webusb_descriptors.h" |
| 26 #include "url/gurl.h" | 29 #include "url/gurl.h" |
| 27 | 30 |
| 28 UsbChooserBubbleController::UsbChooserBubbleController( | 31 namespace { |
| 32 |
| 33 Browser* GetBrowser() { |
| 34 chrome::ScopedTabbedBrowserDisplayer browser_displayer( |
| 35 ProfileManager::GetActiveUserProfile()); |
| 36 DCHECK(browser_displayer.browser()); |
| 37 return browser_displayer.browser(); |
| 38 } |
| 39 |
| 40 } // namespace |
| 41 |
| 42 UsbChooserController::UsbChooserController( |
| 29 content::RenderFrameHost* owner, | 43 content::RenderFrameHost* owner, |
| 30 mojo::Array<device::usb::DeviceFilterPtr> device_filters, | 44 mojo::Array<device::usb::DeviceFilterPtr> device_filters, |
| 31 content::RenderFrameHost* render_frame_host, | 45 content::RenderFrameHost* render_frame_host, |
| 32 const device::usb::ChooserService::GetPermissionCallback& callback) | 46 const device::usb::ChooserService::GetPermissionCallback& callback) |
| 33 : ChooserBubbleController(owner), | 47 : ChooserController(owner), |
| 34 render_frame_host_(render_frame_host), | 48 render_frame_host_(render_frame_host), |
| 35 callback_(callback), | 49 callback_(callback), |
| 36 usb_service_observer_(this), | 50 usb_service_observer_(this), |
| 37 weak_factory_(this) { | 51 weak_factory_(this) { |
| 38 device::UsbService* usb_service = | 52 device::UsbService* usb_service = |
| 39 device::DeviceClient::Get()->GetUsbService(); | 53 device::DeviceClient::Get()->GetUsbService(); |
| 40 if (!usb_service) | 54 if (!usb_service) |
| 41 return; | 55 return; |
| 42 | 56 |
| 43 if (!usb_service_observer_.IsObserving(usb_service)) | 57 if (!usb_service_observer_.IsObserving(usb_service)) |
| 44 usb_service_observer_.Add(usb_service); | 58 usb_service_observer_.Add(usb_service); |
| 45 | 59 |
| 46 if (!device_filters.is_null()) | 60 if (!device_filters.is_null()) |
| 47 filters_ = device_filters.To<std::vector<device::UsbDeviceFilter>>(); | 61 filters_ = device_filters.To<std::vector<device::UsbDeviceFilter>>(); |
| 48 | 62 |
| 49 usb_service->GetDevices( | 63 usb_service->GetDevices(base::Bind(&UsbChooserController::GotUsbDeviceList, |
| 50 base::Bind(&UsbChooserBubbleController::GotUsbDeviceList, | 64 weak_factory_.GetWeakPtr())); |
| 51 weak_factory_.GetWeakPtr())); | |
| 52 } | 65 } |
| 53 | 66 |
| 54 UsbChooserBubbleController::~UsbChooserBubbleController() { | 67 UsbChooserController::~UsbChooserController() { |
| 55 if (!callback_.is_null()) | 68 if (!callback_.is_null()) |
| 56 callback_.Run(nullptr); | 69 callback_.Run(nullptr); |
| 57 } | 70 } |
| 58 | 71 |
| 59 size_t UsbChooserBubbleController::NumOptions() const { | 72 size_t UsbChooserController::NumOptions() const { |
| 60 return devices_.size(); | 73 return devices_.size(); |
| 61 } | 74 } |
| 62 | 75 |
| 63 const base::string16& UsbChooserBubbleController::GetOption( | 76 const base::string16& UsbChooserController::GetOption(size_t index) const { |
| 64 size_t index) const { | |
| 65 DCHECK_LT(index, devices_.size()); | 77 DCHECK_LT(index, devices_.size()); |
| 66 return devices_[index].second; | 78 return devices_[index].second; |
| 67 } | 79 } |
| 68 | 80 |
| 69 void UsbChooserBubbleController::Select(size_t index) { | 81 void UsbChooserController::Select(size_t index) { |
| 70 DCHECK_LT(index, devices_.size()); | 82 DCHECK_LT(index, devices_.size()); |
| 71 content::WebContents* web_contents = | 83 content::WebContents* web_contents = |
| 72 content::WebContents::FromRenderFrameHost(render_frame_host_); | 84 content::WebContents::FromRenderFrameHost(render_frame_host_); |
| 73 GURL embedding_origin = | 85 GURL embedding_origin = |
| 74 web_contents->GetMainFrame()->GetLastCommittedURL().GetOrigin(); | 86 web_contents->GetMainFrame()->GetLastCommittedURL().GetOrigin(); |
| 75 Profile* profile = | 87 Profile* profile = |
| 76 Profile::FromBrowserContext(web_contents->GetBrowserContext()); | 88 Profile::FromBrowserContext(web_contents->GetBrowserContext()); |
| 77 UsbChooserContext* chooser_context = | 89 UsbChooserContext* chooser_context = |
| 78 UsbChooserContextFactory::GetForProfile(profile); | 90 UsbChooserContextFactory::GetForProfile(profile); |
| 79 chooser_context->GrantDevicePermission( | 91 chooser_context->GrantDevicePermission( |
| 80 render_frame_host_->GetLastCommittedURL().GetOrigin(), embedding_origin, | 92 render_frame_host_->GetLastCommittedURL().GetOrigin(), embedding_origin, |
| 81 devices_[index].first->guid()); | 93 devices_[index].first->guid()); |
| 82 | 94 |
| 83 device::usb::DeviceInfoPtr device_info_ptr = | 95 device::usb::DeviceInfoPtr device_info_ptr = |
| 84 device::usb::DeviceInfo::From(*devices_[index].first); | 96 device::usb::DeviceInfo::From(*devices_[index].first); |
| 85 callback_.Run(std::move(device_info_ptr)); | 97 callback_.Run(std::move(device_info_ptr)); |
| 86 callback_.reset(); // Reset |callback_| so that it is only run once. | 98 callback_.reset(); // Reset |callback_| so that it is only run once. |
| 87 | 99 |
| 88 RecordWebUsbChooserClosure( | 100 RecordWebUsbChooserClosure( |
| 89 devices_[index].first->serial_number().empty() | 101 devices_[index].first->serial_number().empty() |
| 90 ? WEBUSB_CHOOSER_CLOSED_EPHEMERAL_PERMISSION_GRANTED | 102 ? WEBUSB_CHOOSER_CLOSED_EPHEMERAL_PERMISSION_GRANTED |
| 91 : WEBUSB_CHOOSER_CLOSED_PERMISSION_GRANTED); | 103 : WEBUSB_CHOOSER_CLOSED_PERMISSION_GRANTED); |
| 92 | |
| 93 if (bubble_reference_) | |
| 94 bubble_reference_->CloseBubble(BUBBLE_CLOSE_ACCEPTED); | |
| 95 } | 104 } |
| 96 | 105 |
| 97 void UsbChooserBubbleController::Cancel() { | 106 void UsbChooserController::Cancel() { |
| 98 RecordWebUsbChooserClosure(devices_.size() == 0 | 107 RecordWebUsbChooserClosure(devices_.size() == 0 |
| 99 ? WEBUSB_CHOOSER_CLOSED_CANCELLED_NO_DEVICES | 108 ? WEBUSB_CHOOSER_CLOSED_CANCELLED_NO_DEVICES |
| 100 : WEBUSB_CHOOSER_CLOSED_CANCELLED); | 109 : WEBUSB_CHOOSER_CLOSED_CANCELLED); |
| 101 | |
| 102 if (bubble_reference_) | |
| 103 bubble_reference_->CloseBubble(BUBBLE_CLOSE_CANCELED); | |
| 104 } | 110 } |
| 105 | 111 |
| 106 void UsbChooserBubbleController::Close() {} | 112 void UsbChooserController::Close() {} |
| 107 | 113 |
| 108 void UsbChooserBubbleController::OnDeviceAdded( | 114 void UsbChooserController::OpenHelpCenterUrl() const { |
| 115 GetBrowser()->OpenURL(content::OpenURLParams( |
| 116 GURL(chrome::kChooserUsbOverviewURL), content::Referrer(), |
| 117 NEW_FOREGROUND_TAB, ui::PAGE_TRANSITION_AUTO_TOPLEVEL, |
| 118 false /* is_renderer_initialized */)); |
| 119 } |
| 120 |
| 121 void UsbChooserController::OnDeviceAdded( |
| 109 scoped_refptr<device::UsbDevice> device) { | 122 scoped_refptr<device::UsbDevice> device) { |
| 110 if (DisplayDevice(device)) { | 123 if (DisplayDevice(device)) { |
| 111 devices_.push_back(std::make_pair(device, device->product_string())); | 124 devices_.push_back(std::make_pair(device, device->product_string())); |
| 112 if (observer()) | 125 if (observer()) |
| 113 observer()->OnOptionAdded(devices_.size() - 1); | 126 observer()->OnOptionAdded(devices_.size() - 1); |
| 114 } | 127 } |
| 115 } | 128 } |
| 116 | 129 |
| 117 GURL UsbChooserBubbleController::GetHelpCenterUrl() const { | 130 void UsbChooserController::OnDeviceRemoved( |
| 118 return GURL(chrome::kChooserUsbOverviewURL); | |
| 119 } | |
| 120 | |
| 121 void UsbChooserBubbleController::OnDeviceRemoved( | |
| 122 scoped_refptr<device::UsbDevice> device) { | 131 scoped_refptr<device::UsbDevice> device) { |
| 123 for (auto it = devices_.begin(); it != devices_.end(); ++it) { | 132 for (auto it = devices_.begin(); it != devices_.end(); ++it) { |
| 124 if (it->first == device) { | 133 if (it->first == device) { |
| 125 size_t index = it - devices_.begin(); | 134 size_t index = it - devices_.begin(); |
| 126 devices_.erase(it); | 135 devices_.erase(it); |
| 127 if (observer()) | 136 if (observer()) |
| 128 observer()->OnOptionRemoved(index); | 137 observer()->OnOptionRemoved(index); |
| 129 return; | 138 return; |
| 130 } | 139 } |
| 131 } | 140 } |
| 132 } | 141 } |
| 133 | 142 |
| 134 void UsbChooserBubbleController::set_bubble_reference( | |
| 135 BubbleReference bubble_reference) { | |
| 136 bubble_reference_ = bubble_reference; | |
| 137 } | |
| 138 | |
| 139 // Get a list of devices that can be shown in the chooser bubble UI for | 143 // Get a list of devices that can be shown in the chooser bubble UI for |
| 140 // user to grant permsssion. | 144 // user to grant permsssion. |
| 141 void UsbChooserBubbleController::GotUsbDeviceList( | 145 void UsbChooserController::GotUsbDeviceList( |
| 142 const std::vector<scoped_refptr<device::UsbDevice>>& devices) { | 146 const std::vector<scoped_refptr<device::UsbDevice>>& devices) { |
| 143 for (const auto& device : devices) { | 147 for (const auto& device : devices) { |
| 144 if (DisplayDevice(device)) | 148 if (DisplayDevice(device)) |
| 145 devices_.push_back(std::make_pair(device, device->product_string())); | 149 devices_.push_back(std::make_pair(device, device->product_string())); |
| 146 } | 150 } |
| 147 if (observer()) | 151 if (observer()) |
| 148 observer()->OnOptionsInitialized(); | 152 observer()->OnOptionsInitialized(); |
| 149 } | 153 } |
| 150 | 154 |
| 151 bool UsbChooserBubbleController::DisplayDevice( | 155 bool UsbChooserController::DisplayDevice( |
| 152 scoped_refptr<device::UsbDevice> device) const { | 156 scoped_refptr<device::UsbDevice> device) const { |
| 153 return device::UsbDeviceFilter::MatchesAny(device, filters_) && | 157 return device::UsbDeviceFilter::MatchesAny(device, filters_) && |
| 154 (base::CommandLine::ForCurrentProcess()->HasSwitch( | 158 (base::CommandLine::ForCurrentProcess()->HasSwitch( |
| 155 switches::kDisableWebUsbSecurity) || | 159 switches::kDisableWebUsbSecurity) || |
| 156 device::FindInWebUsbAllowedOrigins( | 160 device::FindInWebUsbAllowedOrigins( |
| 157 device->webusb_allowed_origins(), | 161 device->webusb_allowed_origins(), |
| 158 render_frame_host_->GetLastCommittedURL().GetOrigin())); | 162 render_frame_host_->GetLastCommittedURL().GetOrigin())); |
| 159 } | 163 } |
| OLD | NEW |