OLD | NEW |
| (Empty) |
1 // Copyright (c) 2012 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 | |
5 #include "chrome/browser/usb/usb_service.h" | |
6 | |
7 #include <set> | |
8 #include <vector> | |
9 | |
10 #include "base/lazy_instance.h" | |
11 #include "base/stl_util.h" | |
12 #include "chrome/browser/usb/usb_context.h" | |
13 #include "chrome/browser/usb/usb_device.h" | |
14 #include "content/public/browser/browser_thread.h" | |
15 #include "third_party/libusb/src/libusb/libusb.h" | |
16 | |
17 namespace { | |
18 | |
19 base::LazyInstance<scoped_ptr<UsbService> >::Leaky g_usb_service_instance = | |
20 LAZY_INSTANCE_INITIALIZER; | |
21 | |
22 } // namespace | |
23 | |
24 // static | |
25 UsbService* UsbService::GetInstance() { | |
26 DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::FILE)); | |
27 UsbService* instance = g_usb_service_instance.Get().get(); | |
28 if (!instance) { | |
29 PlatformUsbContext context = NULL; | |
30 if (libusb_init(&context) != LIBUSB_SUCCESS) | |
31 return NULL; | |
32 if (!context) | |
33 return NULL; | |
34 | |
35 instance = new UsbService(context); | |
36 g_usb_service_instance.Get().reset(instance); | |
37 } | |
38 return instance; | |
39 } | |
40 | |
41 scoped_refptr<UsbDevice> UsbService::GetDeviceById(uint32 unique_id) { | |
42 DCHECK(CalledOnValidThread()); | |
43 RefreshDevices(); | |
44 for (DeviceMap::iterator it = devices_.begin(); it != devices_.end(); ++it) { | |
45 if (it->second->unique_id() == unique_id) | |
46 return it->second; | |
47 } | |
48 return NULL; | |
49 } | |
50 | |
51 void UsbService::GetDevices(std::vector<scoped_refptr<UsbDevice> >* devices) { | |
52 DCHECK(CalledOnValidThread()); | |
53 STLClearObject(devices); | |
54 RefreshDevices(); | |
55 | |
56 for (DeviceMap::iterator it = devices_.begin(); it != devices_.end(); ++it) { | |
57 devices->push_back(it->second); | |
58 } | |
59 } | |
60 | |
61 void UsbService::WillDestroyCurrentMessageLoop() { | |
62 DCHECK(CalledOnValidThread()); | |
63 g_usb_service_instance.Get().reset(NULL); | |
64 } | |
65 | |
66 UsbService::UsbService(PlatformUsbContext context) | |
67 : context_(new UsbContext(context)), next_unique_id_(0) { | |
68 base::MessageLoop::current()->AddDestructionObserver(this); | |
69 } | |
70 | |
71 UsbService::~UsbService() { | |
72 base::MessageLoop::current()->RemoveDestructionObserver(this); | |
73 for (DeviceMap::iterator it = devices_.begin(); it != devices_.end(); ++it) { | |
74 it->second->OnDisconnect(); | |
75 } | |
76 } | |
77 | |
78 void UsbService::RefreshDevices() { | |
79 DCHECK(CalledOnValidThread()); | |
80 | |
81 libusb_device** platform_devices = NULL; | |
82 const ssize_t device_count = | |
83 libusb_get_device_list(context_->context(), &platform_devices); | |
84 | |
85 std::set<UsbDevice*> connected_devices; | |
86 std::vector<PlatformUsbDevice> disconnected_devices; | |
87 | |
88 // Populates new devices. | |
89 for (ssize_t i = 0; i < device_count; ++i) { | |
90 if (!ContainsKey(devices_, platform_devices[i])) { | |
91 libusb_device_descriptor descriptor; | |
92 // This test is needed. A valid vendor/produce pair is required. | |
93 if (0 != libusb_get_device_descriptor(platform_devices[i], &descriptor)) | |
94 continue; | |
95 UsbDevice* new_device = new UsbDevice(context_, | |
96 platform_devices[i], | |
97 descriptor.idVendor, | |
98 descriptor.idProduct, | |
99 ++next_unique_id_); | |
100 devices_[platform_devices[i]] = new_device; | |
101 connected_devices.insert(new_device); | |
102 } else { | |
103 connected_devices.insert(devices_[platform_devices[i]].get()); | |
104 } | |
105 } | |
106 | |
107 // Find disconnected devices. | |
108 for (DeviceMap::iterator it = devices_.begin(); it != devices_.end(); ++it) { | |
109 if (!ContainsKey(connected_devices, it->second)) { | |
110 disconnected_devices.push_back(it->first); | |
111 } | |
112 } | |
113 | |
114 // Remove disconnected devices from devices_. | |
115 for (size_t i = 0; i < disconnected_devices.size(); ++i) { | |
116 // UsbDevice will be destroyed after this. The corresponding | |
117 // PlatformUsbDevice will be unref'ed during this process. | |
118 devices_.erase(disconnected_devices[i]); | |
119 } | |
120 | |
121 libusb_free_device_list(platform_devices, true); | |
122 } | |
OLD | NEW |