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 "device/devices_app/usb/device_manager_impl.h" | 5 #include "device/devices_app/usb/device_manager_impl.h" |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/location.h" | 8 #include "base/location.h" |
9 #include "base/memory/scoped_ptr.h" | 9 #include "base/memory/scoped_ptr.h" |
10 #include "base/message_loop/message_loop.h" | 10 #include "base/message_loop/message_loop.h" |
(...skipping 33 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
44 UsbService* usb_service = DeviceClient::Get()->GetUsbService(); | 44 UsbService* usb_service = DeviceClient::Get()->GetUsbService(); |
45 if (usb_service) { | 45 if (usb_service) { |
46 usb_service->GetDevices(base::Bind(&OnGetDevicesOnServiceThread, callback, | 46 usb_service->GetDevices(base::Bind(&OnGetDevicesOnServiceThread, callback, |
47 callback_task_runner)); | 47 callback_task_runner)); |
48 } else { | 48 } else { |
49 callback_task_runner->PostTask(FROM_HERE, | 49 callback_task_runner->PostTask(FROM_HERE, |
50 base::Bind(callback, DeviceList())); | 50 base::Bind(callback, DeviceList())); |
51 } | 51 } |
52 } | 52 } |
53 | 53 |
| 54 void GetDeviceOnServiceThread( |
| 55 const mojo::String& guid, |
| 56 const base::Callback<void(DeviceInfoPtr)>& callback, |
| 57 scoped_refptr<base::TaskRunner> callback_task_runner) { |
| 58 DeviceInfoPtr device_info; |
| 59 DCHECK(DeviceClient::Get()); |
| 60 UsbService* usb_service = DeviceClient::Get()->GetUsbService(); |
| 61 if (usb_service) { |
| 62 scoped_refptr<UsbDevice> device = usb_service->GetDevice(guid); |
| 63 if (device) |
| 64 device_info = DeviceInfo::From(*device); |
| 65 } |
| 66 |
| 67 callback_task_runner->PostTask( |
| 68 FROM_HERE, base::Bind(callback, base::Passed(&device_info))); |
| 69 } |
| 70 |
54 void RunOpenDeviceCallback(const DeviceManager::OpenDeviceCallback& callback, | 71 void RunOpenDeviceCallback(const DeviceManager::OpenDeviceCallback& callback, |
55 OpenDeviceError error) { | 72 OpenDeviceError error) { |
56 callback.Run(error); | 73 callback.Run(error); |
57 } | 74 } |
58 | 75 |
59 void OnOpenDeviceOnServiceThread( | 76 void OnOpenDeviceOnServiceThread( |
60 mojo::InterfaceRequest<Device> device_request, | 77 mojo::InterfaceRequest<Device> device_request, |
61 const DeviceManager::OpenDeviceCallback& callback, | 78 const DeviceManager::OpenDeviceCallback& callback, |
62 scoped_refptr<base::TaskRunner> callback_task_runner, | 79 scoped_refptr<base::TaskRunner> callback_task_runner, |
63 scoped_refptr<UsbDeviceHandle> device_handle) { | 80 scoped_refptr<UsbDeviceHandle> device_handle) { |
(...skipping 140 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
204 void DeviceManagerImpl::GetDeviceChanges( | 221 void DeviceManagerImpl::GetDeviceChanges( |
205 const GetDeviceChangesCallback& callback) { | 222 const GetDeviceChangesCallback& callback) { |
206 device_change_callbacks_.push(callback); | 223 device_change_callbacks_.push(callback); |
207 MaybeRunDeviceChangesCallback(); | 224 MaybeRunDeviceChangesCallback(); |
208 } | 225 } |
209 | 226 |
210 void DeviceManagerImpl::OpenDevice( | 227 void DeviceManagerImpl::OpenDevice( |
211 const mojo::String& guid, | 228 const mojo::String& guid, |
212 mojo::InterfaceRequest<Device> device_request, | 229 mojo::InterfaceRequest<Device> device_request, |
213 const OpenDeviceCallback& callback) { | 230 const OpenDeviceCallback& callback) { |
214 mojo::Array<mojo::String> requested_guids(1); | 231 auto has_permission_callback = base::Bind( |
215 requested_guids[0] = guid; | 232 &DeviceManagerImpl::OnGotDeviceInfoForOpen, weak_factory_.GetWeakPtr(), |
| 233 base::Passed(&device_request), callback); |
| 234 service_task_runner_->PostTask( |
| 235 FROM_HERE, |
| 236 base::Bind(&GetDeviceOnServiceThread, guid, has_permission_callback, |
| 237 base::ThreadTaskRunnerHandle::Get())); |
| 238 } |
| 239 |
| 240 void DeviceManagerImpl::OnGotDeviceInfoForOpen( |
| 241 mojo::InterfaceRequest<Device> device_request, |
| 242 const OpenDeviceCallback& callback, |
| 243 DeviceInfoPtr device_info) { |
| 244 if (!device_info) { |
| 245 callback.Run(OPEN_DEVICE_ERROR_NOT_FOUND); |
| 246 return; |
| 247 } |
| 248 |
| 249 mojo::Array<DeviceInfoPtr> requested_devices(1); |
| 250 requested_devices[0] = device_info.Pass(); |
216 permission_provider_->HasDevicePermission( | 251 permission_provider_->HasDevicePermission( |
217 requested_guids.Pass(), | 252 requested_devices.Pass(), |
218 base::Bind(&DeviceManagerImpl::OnOpenDevicePermissionCheckComplete, | 253 base::Bind(&DeviceManagerImpl::OnOpenDevicePermissionCheckComplete, |
219 base::Unretained(this), base::Passed(&device_request), | 254 base::Unretained(this), base::Passed(&device_request), |
220 callback)); | 255 callback)); |
221 } | 256 } |
222 | 257 |
223 void DeviceManagerImpl::OnOpenDevicePermissionCheckComplete( | 258 void DeviceManagerImpl::OnOpenDevicePermissionCheckComplete( |
224 mojo::InterfaceRequest<Device> device_request, | 259 mojo::InterfaceRequest<Device> device_request, |
225 const OpenDeviceCallback& callback, | 260 const OpenDeviceCallback& callback, |
226 mojo::Array<mojo::String> allowed_guids) { | 261 mojo::Array<mojo::String> allowed_guids) { |
227 if (allowed_guids.size() == 0) { | 262 if (allowed_guids.size() == 0) { |
228 callback.Run(OPEN_DEVICE_ERROR_ACCESS_DENIED); | 263 callback.Run(OPEN_DEVICE_ERROR_ACCESS_DENIED); |
229 return; | 264 return; |
230 } | 265 } |
231 | 266 |
232 DCHECK(allowed_guids.size() == 1); | 267 DCHECK(allowed_guids.size() == 1); |
233 service_task_runner_->PostTask( | 268 service_task_runner_->PostTask( |
234 FROM_HERE, base::Bind(&OpenDeviceOnServiceThread, allowed_guids[0], | 269 FROM_HERE, base::Bind(&OpenDeviceOnServiceThread, allowed_guids[0], |
235 base::Passed(&device_request), callback, | 270 base::Passed(&device_request), callback, |
236 base::ThreadTaskRunnerHandle::Get())); | 271 base::ThreadTaskRunnerHandle::Get())); |
237 } | 272 } |
238 | 273 |
239 void DeviceManagerImpl::OnGetDevices(EnumerationOptionsPtr options, | 274 void DeviceManagerImpl::OnGetDevices(EnumerationOptionsPtr options, |
240 const GetDevicesCallback& callback, | 275 const GetDevicesCallback& callback, |
241 const DeviceList& devices) { | 276 const DeviceList& devices) { |
242 std::vector<UsbDeviceFilter> filters; | 277 std::vector<UsbDeviceFilter> filters; |
243 if (options) | 278 if (options) |
244 filters = options->filters.To<std::vector<UsbDeviceFilter>>(); | 279 filters = options->filters.To<std::vector<UsbDeviceFilter>>(); |
245 | 280 |
246 std::map<std::string, scoped_refptr<UsbDevice>> device_map; | 281 std::map<std::string, scoped_refptr<UsbDevice>> device_map; |
247 mojo::Array<mojo::String> requested_guids(0); | 282 mojo::Array<DeviceInfoPtr> requested_devices(0); |
248 for (const auto& device : devices) { | 283 for (const auto& device : devices) { |
249 if (filters.empty() || UsbDeviceFilter::MatchesAny(device, filters)) { | 284 if (filters.empty() || UsbDeviceFilter::MatchesAny(device, filters)) { |
250 device_map[device->guid()] = device; | 285 device_map[device->guid()] = device; |
251 requested_guids.push_back(device->guid()); | 286 requested_devices.push_back(DeviceInfo::From(*device)); |
252 } | 287 } |
253 } | 288 } |
254 | 289 |
255 permission_provider_->HasDevicePermission( | 290 permission_provider_->HasDevicePermission( |
256 requested_guids.Pass(), | 291 requested_devices.Pass(), |
257 base::Bind(&FilterAndConvertDevicesAndThen, device_map, callback)); | 292 base::Bind(&FilterAndConvertDevicesAndThen, device_map, callback)); |
258 } | 293 } |
259 | 294 |
260 void DeviceManagerImpl::OnDeviceAdded(scoped_refptr<UsbDevice> device) { | 295 void DeviceManagerImpl::OnDeviceAdded(scoped_refptr<UsbDevice> device) { |
261 DCHECK(!ContainsKey(devices_removed_, device->guid())); | 296 DCHECK(!ContainsKey(devices_removed_, device->guid())); |
262 devices_added_[device->guid()] = device; | 297 devices_added_[device->guid()] = device; |
263 MaybeRunDeviceChangesCallback(); | 298 MaybeRunDeviceChangesCallback(); |
264 } | 299 } |
265 | 300 |
266 void DeviceManagerImpl::OnDeviceRemoved(scoped_refptr<UsbDevice> device) { | 301 void DeviceManagerImpl::OnDeviceRemoved(scoped_refptr<UsbDevice> device) { |
267 if (devices_added_.erase(device->guid()) == 0) | 302 if (devices_added_.erase(device->guid()) == 0) |
268 devices_removed_[device->guid()] = device; | 303 devices_removed_[device->guid()] = device; |
269 MaybeRunDeviceChangesCallback(); | 304 MaybeRunDeviceChangesCallback(); |
270 } | 305 } |
271 | 306 |
272 void DeviceManagerImpl::MaybeRunDeviceChangesCallback() { | 307 void DeviceManagerImpl::MaybeRunDeviceChangesCallback() { |
273 if (!permission_request_pending_ && !device_change_callbacks_.empty()) { | 308 if (!permission_request_pending_ && !device_change_callbacks_.empty()) { |
274 DeviceMap devices_added; | 309 DeviceMap devices_added; |
275 devices_added.swap(devices_added_); | 310 devices_added.swap(devices_added_); |
276 DeviceMap devices_removed; | 311 DeviceMap devices_removed; |
277 devices_removed.swap(devices_removed_); | 312 devices_removed.swap(devices_removed_); |
278 | 313 |
279 mojo::Array<mojo::String> requested_guids(devices_added.size() + | 314 mojo::Array<DeviceInfoPtr> requested_devices(devices_added.size() + |
280 devices_removed.size()); | 315 devices_removed.size()); |
281 { | 316 { |
282 size_t i = 0; | 317 size_t i = 0; |
283 for (const auto& map_entry : devices_added) | 318 for (const auto& map_entry : devices_added) |
284 requested_guids[i++] = map_entry.first; | 319 requested_devices[i++] = DeviceInfo::From(*map_entry.second); |
285 for (const auto& map_entry : devices_removed) | 320 for (const auto& map_entry : devices_removed) |
286 requested_guids[i++] = map_entry.first; | 321 requested_devices[i++] = DeviceInfo::From(*map_entry.second); |
287 } | 322 } |
288 | 323 |
289 permission_request_pending_ = true; | 324 permission_request_pending_ = true; |
290 permission_provider_->HasDevicePermission( | 325 permission_provider_->HasDevicePermission( |
291 requested_guids.Pass(), | 326 requested_devices.Pass(), |
292 base::Bind(&DeviceManagerImpl::OnEnumerationPermissionCheckComplete, | 327 base::Bind(&DeviceManagerImpl::OnEnumerationPermissionCheckComplete, |
293 base::Unretained(this), devices_added, devices_removed)); | 328 base::Unretained(this), devices_added, devices_removed)); |
294 } | 329 } |
295 } | 330 } |
296 | 331 |
297 void DeviceManagerImpl::OnEnumerationPermissionCheckComplete( | 332 void DeviceManagerImpl::OnEnumerationPermissionCheckComplete( |
298 const DeviceMap& devices_added, | 333 const DeviceMap& devices_added, |
299 const DeviceMap& devices_removed, | 334 const DeviceMap& devices_removed, |
300 mojo::Array<mojo::String> allowed_guids) { | 335 mojo::Array<mojo::String> allowed_guids) { |
301 permission_request_pending_ = false; | 336 permission_request_pending_ = false; |
(...skipping 21 matching lines...) Expand all Loading... |
323 callback.Run(notification.Pass()); | 358 callback.Run(notification.Pass()); |
324 device_change_callbacks_.pop(); | 359 device_change_callbacks_.pop(); |
325 } | 360 } |
326 | 361 |
327 if (devices_added_.size() > 0 || !devices_removed_.empty()) | 362 if (devices_added_.size() > 0 || !devices_removed_.empty()) |
328 MaybeRunDeviceChangesCallback(); | 363 MaybeRunDeviceChangesCallback(); |
329 } | 364 } |
330 | 365 |
331 } // namespace usb | 366 } // namespace usb |
332 } // namespace device | 367 } // namespace device |
OLD | NEW |