1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
|
//
// Copyright 2010 Ettus Research LLC
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.
//
#include <uhd/types/usb_descriptor.hpp>
#include <uhd/transport/usb_control.hpp>
#include <libusb-1.0/libusb.h>
#include <boost/asio.hpp>
#include <stdexcept>
#include <iostream>
using namespace uhd::transport;
static int libusb_debug_level = 0;
static int libusb_timeout = 0;
/***********************************************************************
* libusb-1.0 implementation of USB control transport
**********************************************************************/
class libusb_control_impl : public usb_control {
public:
libusb_control_impl(uhd::usb_descriptor_t descriptor);
~libusb_control_impl();
size_t submit(boost::uint8_t request_type,
boost::uint8_t request,
boost::uint16_t value,
boost::uint16_t index,
unsigned char *buff,
boost::uint16_t length);
static uhd::usb_descriptor_t create_descriptor(libusb_device *dev);
static std::string get_serial(libusb_device *dev);
private:
uhd::usb_descriptor_t _descriptor;
libusb_context *_ctx;
libusb_device_handle *_dev_handle;
bool open_device();
bool open_interface();
bool compare_device(libusb_device *dev, uhd::usb_descriptor_t descriptor);
};
libusb_control_impl::libusb_control_impl(uhd::usb_descriptor_t descriptor)
: _descriptor(descriptor), _ctx(NULL), _dev_handle(NULL)
{
if (libusb_init(&_ctx) < 0)
throw std::runtime_error("USB: failed to initialize libusb");
libusb_set_debug(_ctx, libusb_debug_level);
if (!open_device())
throw std::runtime_error("USB: failed to open device");
if (!open_interface())
throw std::runtime_error("USB: failed to open device interface");
}
libusb_control_impl::~libusb_control_impl()
{
libusb_close(_dev_handle);
libusb_exit(_ctx);
}
uhd::usb_descriptor_t libusb_control_impl::create_descriptor(libusb_device *dev)
{
libusb_device_descriptor desc;
if (libusb_get_device_descriptor(dev, &desc) < 0)
throw std::runtime_error("USB: failed to get device descriptor");
uhd::usb_descriptor_t descriptor;
descriptor.serial = get_serial(dev);
descriptor.product_id = desc.idProduct;
descriptor.vendor_id = desc.idVendor;
descriptor.device_addr = libusb_get_device_address(dev);
return descriptor;
}
std::string libusb_control_impl::get_serial(libusb_device *dev)
{
unsigned char buff[32];
libusb_device_descriptor desc;
if (libusb_get_device_descriptor(dev, &desc) < 0)
return "";
if (desc.iSerialNumber == 0)
return "";
//open the device because we have to
libusb_device_handle *dev_handle;
if (libusb_open(dev, &dev_handle) < 0)
return "";
if (libusb_get_string_descriptor_ascii(dev_handle, desc.iSerialNumber,
buff, sizeof(buff)) < 0) {
return "";
}
libusb_close(dev_handle);
return (char*) buff;
}
bool libusb_control_impl::open_device()
{
libusb_device **list;
libusb_device *dev;
ssize_t cnt = libusb_get_device_list(_ctx, &list);
if (cnt < 0)
return cnt;
ssize_t i = 0;
for (i = 0; i < cnt; i++) {
dev = list[i];
if (compare_device(dev, _descriptor))
goto found;
}
return false;
found:
int ret;
if ((ret = libusb_open(dev, &_dev_handle)) < 0)
return false;
else
return true;
}
bool libusb_control_impl::compare_device(libusb_device *dev,
uhd::usb_descriptor_t descriptor)
{
std::string serial = descriptor.serial;
boost::uint16_t vendor_id = descriptor.vendor_id;
boost::uint16_t product_id = descriptor.product_id;
boost::uint8_t device_addr = descriptor.device_addr;
libusb_device_descriptor libusb_desc;
if (libusb_get_device_descriptor(dev, &libusb_desc) < 0)
return false;
if (serial != get_serial(dev))
return false;
if (vendor_id != libusb_desc.idVendor)
return false;
if (product_id != libusb_desc.idProduct)
return false;
if (device_addr != libusb_get_device_address(dev))
return false;
return true;
}
bool libusb_control_impl::open_interface()
{
if (libusb_claim_interface(_dev_handle, 0) < 0)
return false;
else
return true;
}
size_t libusb_control_impl::submit(boost::uint8_t request_type,
boost::uint8_t request,
boost::uint16_t value,
boost::uint16_t index,
unsigned char *buff,
boost::uint16_t length)
{
return libusb_control_transfer(_dev_handle,
request_type,
request,
value,
index,
buff,
length,
libusb_timeout);
}
/***********************************************************************
* USB control public make functions
**********************************************************************/
usb_control::sptr usb_control::make(uhd::usb_descriptor_t descriptor)
{
return sptr(new libusb_control_impl(descriptor));
}
uhd::usb_descriptors_t usb_control::get_device_list()
{
libusb_device **list;
uhd::usb_descriptors_t descriptors;
if (libusb_init(NULL) < 0)
throw std::runtime_error("USB: failed to initialize libusb");
ssize_t cnt = libusb_get_device_list(NULL, &list);
if (cnt < 0)
throw std::runtime_error("USB: failed to get device list");
ssize_t i = 0;
for (i = 0; i < cnt; i++) {
libusb_device *dev = list[i];
descriptors.push_back(libusb_control_impl::create_descriptor(dev));
}
libusb_free_device_list(list, 0);
libusb_exit(NULL);
return descriptors;
}
|
