Loading...
1// SPDX-License-Identifier: GPL-2.0-or-later
2/* $(CROSS_COMPILE)cc -Wall -Wextra -g -lpthread -o testusb testusb.c */
3
4/*
5 * Copyright (c) 2002 by David Brownell
6 * Copyright (c) 2010 by Samsung Electronics
7 * Author: Michal Nazarewicz <mina86@mina86.com>
8 */
9
10/*
11 * This program issues ioctls to perform the tests implemented by the
12 * kernel driver. It can generate a variety of transfer patterns; you
13 * should make sure to test both regular streaming and mixes of
14 * transfer sizes (including short transfers).
15 *
16 * For more information on how this can be used and on USB testing
17 * refer to <URL:http://www.linux-usb.org/usbtest/>.
18 */
19
20#include <stdio.h>
21#include <string.h>
22#include <ftw.h>
23#include <stdlib.h>
24#include <pthread.h>
25#include <unistd.h>
26#include <errno.h>
27#include <limits.h>
28
29#include <sys/types.h>
30#include <sys/stat.h>
31#include <fcntl.h>
32
33#include <sys/ioctl.h>
34#include <linux/usbdevice_fs.h>
35
36/*-------------------------------------------------------------------------*/
37
38#define TEST_CASES 30
39
40// FIXME make these public somewhere; usbdevfs.h?
41
42struct usbtest_param {
43 // inputs
44 unsigned test_num; /* 0..(TEST_CASES-1) */
45 unsigned iterations;
46 unsigned length;
47 unsigned vary;
48 unsigned sglen;
49
50 // outputs
51 struct timeval duration;
52};
53#define USBTEST_REQUEST _IOWR('U', 100, struct usbtest_param)
54
55/*-------------------------------------------------------------------------*/
56
57/* #include <linux/usb_ch9.h> */
58
59#define USB_DT_DEVICE 0x01
60#define USB_DT_INTERFACE 0x04
61
62#define USB_CLASS_PER_INTERFACE 0 /* for DeviceClass */
63#define USB_CLASS_VENDOR_SPEC 0xff
64
65
66struct usb_device_descriptor {
67 __u8 bLength;
68 __u8 bDescriptorType;
69 __u16 bcdUSB;
70 __u8 bDeviceClass;
71 __u8 bDeviceSubClass;
72 __u8 bDeviceProtocol;
73 __u8 bMaxPacketSize0;
74 __u16 idVendor;
75 __u16 idProduct;
76 __u16 bcdDevice;
77 __u8 iManufacturer;
78 __u8 iProduct;
79 __u8 iSerialNumber;
80 __u8 bNumConfigurations;
81} __attribute__ ((packed));
82
83struct usb_interface_descriptor {
84 __u8 bLength;
85 __u8 bDescriptorType;
86
87 __u8 bInterfaceNumber;
88 __u8 bAlternateSetting;
89 __u8 bNumEndpoints;
90 __u8 bInterfaceClass;
91 __u8 bInterfaceSubClass;
92 __u8 bInterfaceProtocol;
93 __u8 iInterface;
94} __attribute__ ((packed));
95
96enum usb_device_speed {
97 USB_SPEED_UNKNOWN = 0, /* enumerating */
98 USB_SPEED_LOW, USB_SPEED_FULL, /* usb 1.1 */
99 USB_SPEED_HIGH, /* usb 2.0 */
100 USB_SPEED_WIRELESS, /* wireless (usb 2.5) */
101 USB_SPEED_SUPER, /* usb 3.0 */
102 USB_SPEED_SUPER_PLUS, /* usb 3.1 */
103};
104
105/*-------------------------------------------------------------------------*/
106
107static char *speed (enum usb_device_speed s)
108{
109 switch (s) {
110 case USB_SPEED_UNKNOWN: return "unknown";
111 case USB_SPEED_LOW: return "low";
112 case USB_SPEED_FULL: return "full";
113 case USB_SPEED_HIGH: return "high";
114 case USB_SPEED_WIRELESS: return "wireless";
115 case USB_SPEED_SUPER: return "super";
116 case USB_SPEED_SUPER_PLUS: return "super-plus";
117 default: return "??";
118 }
119}
120
121struct testdev {
122 struct testdev *next;
123 char *name;
124 pthread_t thread;
125 enum usb_device_speed speed;
126 unsigned ifnum : 8;
127 unsigned forever : 1;
128 int test;
129
130 struct usbtest_param param;
131};
132static struct testdev *testdevs;
133
134static int testdev_ffs_ifnum(FILE *fd)
135{
136 union {
137 char buf[255];
138 struct usb_interface_descriptor intf;
139 } u;
140
141 for (;;) {
142 if (fread(u.buf, 1, 1, fd) != 1)
143 return -1;
144 if (fread(u.buf + 1, (unsigned char)u.buf[0] - 1, 1, fd) != 1)
145 return -1;
146
147 if (u.intf.bLength == sizeof u.intf
148 && u.intf.bDescriptorType == USB_DT_INTERFACE
149 && u.intf.bNumEndpoints == 2
150 && u.intf.bInterfaceClass == USB_CLASS_VENDOR_SPEC
151 && u.intf.bInterfaceSubClass == 0
152 && u.intf.bInterfaceProtocol == 0)
153 return (unsigned char)u.intf.bInterfaceNumber;
154 }
155}
156
157static int testdev_ifnum(FILE *fd)
158{
159 struct usb_device_descriptor dev;
160
161 if (fread(&dev, sizeof dev, 1, fd) != 1)
162 return -1;
163
164 if (dev.bLength != sizeof dev || dev.bDescriptorType != USB_DT_DEVICE)
165 return -1;
166
167 /* FX2 with (tweaked) bulksrc firmware */
168 if (dev.idVendor == 0x0547 && dev.idProduct == 0x1002)
169 return 0;
170
171 /*----------------------------------------------------*/
172
173 /* devices that start up using the EZ-USB default device and
174 * which we can use after loading simple firmware. hotplug
175 * can fxload it, and then run this test driver.
176 *
177 * we return false positives in two cases:
178 * - the device has a "real" driver (maybe usb-serial) that
179 * renumerates. the device should vanish quickly.
180 * - the device doesn't have the test firmware installed.
181 */
182
183 /* generic EZ-USB FX controller */
184 if (dev.idVendor == 0x0547 && dev.idProduct == 0x2235)
185 return 0;
186
187 /* generic EZ-USB FX2 controller */
188 if (dev.idVendor == 0x04b4 && dev.idProduct == 0x8613)
189 return 0;
190
191 /* CY3671 development board with EZ-USB FX */
192 if (dev.idVendor == 0x0547 && dev.idProduct == 0x0080)
193 return 0;
194
195 /* Keyspan 19Qi uses an21xx (original EZ-USB) */
196 if (dev.idVendor == 0x06cd && dev.idProduct == 0x010b)
197 return 0;
198
199 /*----------------------------------------------------*/
200
201 /* "gadget zero", Linux-USB test software */
202 if (dev.idVendor == 0x0525 && dev.idProduct == 0xa4a0)
203 return 0;
204
205 /* user mode subset of that */
206 if (dev.idVendor == 0x0525 && dev.idProduct == 0xa4a4)
207 return testdev_ffs_ifnum(fd);
208 /* return 0; */
209
210 /* iso version of usermode code */
211 if (dev.idVendor == 0x0525 && dev.idProduct == 0xa4a3)
212 return 0;
213
214 /* some GPL'd test firmware uses these IDs */
215
216 if (dev.idVendor == 0xfff0 && dev.idProduct == 0xfff0)
217 return 0;
218
219 /*----------------------------------------------------*/
220
221 /* iBOT2 high speed webcam */
222 if (dev.idVendor == 0x0b62 && dev.idProduct == 0x0059)
223 return 0;
224
225 /*----------------------------------------------------*/
226
227 /* the FunctionFS gadget can have the source/sink interface
228 * anywhere. We look for an interface descriptor that match
229 * what we expect. We ignore configuratiens thou. */
230
231 if (dev.idVendor == 0x0525 && dev.idProduct == 0xa4ac
232 && (dev.bDeviceClass == USB_CLASS_PER_INTERFACE
233 || dev.bDeviceClass == USB_CLASS_VENDOR_SPEC))
234 return testdev_ffs_ifnum(fd);
235
236 return -1;
237}
238
239static int find_testdev(const char *name, const struct stat *sb, int flag)
240{
241 FILE *fd;
242 int ifnum;
243 struct testdev *entry;
244
245 (void)sb; /* unused */
246
247 if (flag != FTW_F)
248 return 0;
249
250 fd = fopen(name, "rb");
251 if (!fd) {
252 perror(name);
253 return 0;
254 }
255
256 ifnum = testdev_ifnum(fd);
257 fclose(fd);
258 if (ifnum < 0)
259 return 0;
260
261 entry = calloc(1, sizeof *entry);
262 if (!entry)
263 goto nomem;
264
265 entry->name = strdup(name);
266 if (!entry->name) {
267 free(entry);
268nomem:
269 perror("malloc");
270 return 0;
271 }
272
273 entry->ifnum = ifnum;
274 entry->next = testdevs;
275 testdevs = entry;
276 return 0;
277}
278
279static int
280usbdev_ioctl (int fd, int ifno, unsigned request, void *param)
281{
282 struct usbdevfs_ioctl wrapper;
283
284 wrapper.ifno = ifno;
285 wrapper.ioctl_code = request;
286 wrapper.data = param;
287
288 return ioctl (fd, USBDEVFS_IOCTL, &wrapper);
289}
290
291static void *handle_testdev (void *arg)
292{
293 struct testdev *dev = arg;
294 int fd, i;
295 int status;
296
297 if ((fd = open (dev->name, O_RDWR)) < 0) {
298 perror ("can't open dev file r/w");
299 return 0;
300 }
301
302 status = ioctl(fd, USBDEVFS_GET_SPEED, NULL);
303 if (status < 0)
304 fprintf(stderr, "USBDEVFS_GET_SPEED failed %d\n", status);
305 else
306 dev->speed = status;
307 fprintf(stderr, "%s speed\t%s\t%u\n",
308 speed(dev->speed), dev->name, dev->ifnum);
309
310restart:
311 for (i = 0; i < TEST_CASES; i++) {
312 if (dev->test != -1 && dev->test != i)
313 continue;
314 dev->param.test_num = i;
315
316 status = usbdev_ioctl (fd, dev->ifnum,
317 USBTEST_REQUEST, &dev->param);
318 if (status < 0 && errno == EOPNOTSUPP)
319 continue;
320
321 /* FIXME need a "syslog it" option for background testing */
322
323 /* NOTE: each thread emits complete lines; no fragments! */
324 if (status < 0) {
325 char buf [80];
326 int err = errno;
327
328 if (strerror_r (errno, buf, sizeof buf)) {
329 snprintf (buf, sizeof buf, "error %d", err);
330 errno = err;
331 }
332 printf ("%s test %d --> %d (%s)\n",
333 dev->name, i, errno, buf);
334 } else
335 printf ("%s test %d, %4d.%.06d secs\n", dev->name, i,
336 (int) dev->param.duration.tv_sec,
337 (int) dev->param.duration.tv_usec);
338
339 fflush (stdout);
340 }
341 if (dev->forever)
342 goto restart;
343
344 close (fd);
345 return arg;
346}
347
348static const char *usb_dir_find(void)
349{
350 static char udev_usb_path[] = "/dev/bus/usb";
351
352 if (access(udev_usb_path, F_OK) == 0)
353 return udev_usb_path;
354
355 return NULL;
356}
357
358static int parse_num(unsigned *num, const char *str)
359{
360 unsigned long val;
361 char *end;
362
363 errno = 0;
364 val = strtoul(str, &end, 0);
365 if (errno || *end || val > UINT_MAX)
366 return -1;
367 *num = val;
368 return 0;
369}
370
371int main (int argc, char **argv)
372{
373
374 int c;
375 struct testdev *entry;
376 char *device;
377 const char *usb_dir = NULL;
378 int all = 0, forever = 0, not = 0;
379 int test = -1 /* all */;
380 struct usbtest_param param;
381
382 /* pick defaults that works with all speeds, without short packets.
383 *
384 * Best per-frame data rates:
385 * super speed,bulk 1024 * 16 * 8 = 131072
386 * interrupt 1024 * 3 * 8 = 24576
387 * high speed, bulk 512 * 13 * 8 = 53248
388 * interrupt 1024 * 3 * 8 = 24576
389 * full speed, bulk/intr 64 * 19 = 1216
390 * interrupt 64 * 1 = 64
391 * low speed, interrupt 8 * 1 = 8
392 */
393 param.iterations = 1000;
394 param.length = 1024;
395 param.vary = 1024;
396 param.sglen = 32;
397
398 /* for easy use when hotplugging */
399 device = getenv ("DEVICE");
400
401 while ((c = getopt (argc, argv, "D:aA:c:g:hlns:t:v:")) != EOF)
402 switch (c) {
403 case 'D': /* device, if only one */
404 device = optarg;
405 continue;
406 case 'A': /* use all devices with specified USB dir */
407 usb_dir = optarg;
408 /* FALL THROUGH */
409 case 'a': /* use all devices */
410 device = NULL;
411 all = 1;
412 continue;
413 case 'c': /* count iterations */
414 if (parse_num(¶m.iterations, optarg))
415 goto usage;
416 continue;
417 case 'g': /* scatter/gather entries */
418 if (parse_num(¶m.sglen, optarg))
419 goto usage;
420 continue;
421 case 'l': /* loop forever */
422 forever = 1;
423 continue;
424 case 'n': /* no test running! */
425 not = 1;
426 continue;
427 case 's': /* size of packet */
428 if (parse_num(¶m.length, optarg))
429 goto usage;
430 continue;
431 case 't': /* run just one test */
432 test = atoi (optarg);
433 if (test < 0)
434 goto usage;
435 continue;
436 case 'v': /* vary packet size by ... */
437 if (parse_num(¶m.vary, optarg))
438 goto usage;
439 continue;
440 case '?':
441 case 'h':
442 default:
443usage:
444 fprintf (stderr,
445 "usage: %s [options]\n"
446 "Options:\n"
447 "\t-D dev only test specific device\n"
448 "\t-A usb-dir\n"
449 "\t-a test all recognized devices\n"
450 "\t-l loop forever(for stress test)\n"
451 "\t-t testnum only run specified case\n"
452 "\t-n no test running, show devices to be tested\n"
453 "Case arguments:\n"
454 "\t-c iterations default 1000\n"
455 "\t-s transfer length default 1024\n"
456 "\t-g sglen default 32\n"
457 "\t-v vary default 1024\n",
458 argv[0]);
459 return 1;
460 }
461 if (optind != argc)
462 goto usage;
463 if (!all && !device) {
464 fprintf (stderr, "must specify '-a' or '-D dev', "
465 "or DEVICE=/dev/bus/usb/BBB/DDD in env\n");
466 goto usage;
467 }
468
469 /* Find usb device subdirectory */
470 if (!usb_dir) {
471 usb_dir = usb_dir_find();
472 if (!usb_dir) {
473 fputs ("USB device files are missing\n", stderr);
474 return -1;
475 }
476 }
477
478 /* collect and list the test devices */
479 if (ftw (usb_dir, find_testdev, 3) != 0) {
480 fputs ("ftw failed; are USB device files missing?\n", stderr);
481 return -1;
482 }
483
484 /* quit, run single test, or create test threads */
485 if (!testdevs && !device) {
486 fputs ("no test devices recognized\n", stderr);
487 return -1;
488 }
489 if (not)
490 return 0;
491 if (testdevs && !testdevs->next && !device)
492 device = testdevs->name;
493 for (entry = testdevs; entry; entry = entry->next) {
494 int status;
495
496 entry->param = param;
497 entry->forever = forever;
498 entry->test = test;
499
500 if (device) {
501 if (strcmp (entry->name, device))
502 continue;
503 return handle_testdev (entry) != entry;
504 }
505 status = pthread_create (&entry->thread, 0, handle_testdev, entry);
506 if (status)
507 perror ("pthread_create");
508 }
509 if (device) {
510 struct testdev dev;
511
512 /* kernel can recognize test devices we don't */
513 fprintf (stderr, "%s: %s may see only control tests\n",
514 argv [0], device);
515
516 memset (&dev, 0, sizeof dev);
517 dev.name = device;
518 dev.param = param;
519 dev.forever = forever;
520 dev.test = test;
521 return handle_testdev (&dev) != &dev;
522 }
523
524 /* wait for tests to complete */
525 for (entry = testdevs; entry; entry = entry->next) {
526 void *retval;
527
528 if (pthread_join (entry->thread, &retval))
529 perror ("pthread_join");
530 /* testing errors discarded! */
531 }
532
533 return 0;
534}
1/* $(CROSS_COMPILE)cc -Wall -Wextra -g -lpthread -o testusb testusb.c */
2
3/*
4 * Copyright (c) 2002 by David Brownell
5 * Copyright (c) 2010 by Samsung Electronics
6 * Author: Michal Nazarewicz <mina86@mina86.com>
7 *
8 * This program is free software; you can redistribute it and/or modify it
9 * under the terms of the GNU General Public License as published by the
10 * Free Software Foundation; either version 2 of the License, or (at your
11 * option) any later version.
12 *
13 * This program is distributed in the hope that it will be useful, but
14 * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
15 * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
16 * for more details.
17 *
18 * You should have received a copy of the GNU General Public License
19 * along with this program; if not, write to the Free Software Foundation,
20 * Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
21 */
22
23/*
24 * This program issues ioctls to perform the tests implemented by the
25 * kernel driver. It can generate a variety of transfer patterns; you
26 * should make sure to test both regular streaming and mixes of
27 * transfer sizes (including short transfers).
28 *
29 * For more information on how this can be used and on USB testing
30 * refer to <URL:http://www.linux-usb.org/usbtest/>.
31 */
32
33#include <stdio.h>
34#include <string.h>
35#include <ftw.h>
36#include <stdlib.h>
37#include <pthread.h>
38#include <unistd.h>
39#include <errno.h>
40#include <limits.h>
41
42#include <sys/types.h>
43#include <sys/stat.h>
44#include <fcntl.h>
45
46#include <sys/ioctl.h>
47#include <linux/usbdevice_fs.h>
48
49/*-------------------------------------------------------------------------*/
50
51#define TEST_CASES 30
52
53// FIXME make these public somewhere; usbdevfs.h?
54
55struct usbtest_param {
56 // inputs
57 unsigned test_num; /* 0..(TEST_CASES-1) */
58 unsigned iterations;
59 unsigned length;
60 unsigned vary;
61 unsigned sglen;
62
63 // outputs
64 struct timeval duration;
65};
66#define USBTEST_REQUEST _IOWR('U', 100, struct usbtest_param)
67
68/*-------------------------------------------------------------------------*/
69
70/* #include <linux/usb_ch9.h> */
71
72#define USB_DT_DEVICE 0x01
73#define USB_DT_INTERFACE 0x04
74
75#define USB_CLASS_PER_INTERFACE 0 /* for DeviceClass */
76#define USB_CLASS_VENDOR_SPEC 0xff
77
78
79struct usb_device_descriptor {
80 __u8 bLength;
81 __u8 bDescriptorType;
82 __u16 bcdUSB;
83 __u8 bDeviceClass;
84 __u8 bDeviceSubClass;
85 __u8 bDeviceProtocol;
86 __u8 bMaxPacketSize0;
87 __u16 idVendor;
88 __u16 idProduct;
89 __u16 bcdDevice;
90 __u8 iManufacturer;
91 __u8 iProduct;
92 __u8 iSerialNumber;
93 __u8 bNumConfigurations;
94} __attribute__ ((packed));
95
96struct usb_interface_descriptor {
97 __u8 bLength;
98 __u8 bDescriptorType;
99
100 __u8 bInterfaceNumber;
101 __u8 bAlternateSetting;
102 __u8 bNumEndpoints;
103 __u8 bInterfaceClass;
104 __u8 bInterfaceSubClass;
105 __u8 bInterfaceProtocol;
106 __u8 iInterface;
107} __attribute__ ((packed));
108
109enum usb_device_speed {
110 USB_SPEED_UNKNOWN = 0, /* enumerating */
111 USB_SPEED_LOW, USB_SPEED_FULL, /* usb 1.1 */
112 USB_SPEED_HIGH /* usb 2.0 */
113};
114
115/*-------------------------------------------------------------------------*/
116
117static char *speed (enum usb_device_speed s)
118{
119 switch (s) {
120 case USB_SPEED_UNKNOWN: return "unknown";
121 case USB_SPEED_LOW: return "low";
122 case USB_SPEED_FULL: return "full";
123 case USB_SPEED_HIGH: return "high";
124 default: return "??";
125 }
126}
127
128struct testdev {
129 struct testdev *next;
130 char *name;
131 pthread_t thread;
132 enum usb_device_speed speed;
133 unsigned ifnum : 8;
134 unsigned forever : 1;
135 int test;
136
137 struct usbtest_param param;
138};
139static struct testdev *testdevs;
140
141static int testdev_ffs_ifnum(FILE *fd)
142{
143 union {
144 char buf[255];
145 struct usb_interface_descriptor intf;
146 } u;
147
148 for (;;) {
149 if (fread(u.buf, 1, 1, fd) != 1)
150 return -1;
151 if (fread(u.buf + 1, (unsigned char)u.buf[0] - 1, 1, fd) != 1)
152 return -1;
153
154 if (u.intf.bLength == sizeof u.intf
155 && u.intf.bDescriptorType == USB_DT_INTERFACE
156 && u.intf.bNumEndpoints == 2
157 && u.intf.bInterfaceClass == USB_CLASS_VENDOR_SPEC
158 && u.intf.bInterfaceSubClass == 0
159 && u.intf.bInterfaceProtocol == 0)
160 return (unsigned char)u.intf.bInterfaceNumber;
161 }
162}
163
164static int testdev_ifnum(FILE *fd)
165{
166 struct usb_device_descriptor dev;
167
168 if (fread(&dev, sizeof dev, 1, fd) != 1)
169 return -1;
170
171 if (dev.bLength != sizeof dev || dev.bDescriptorType != USB_DT_DEVICE)
172 return -1;
173
174 /* FX2 with (tweaked) bulksrc firmware */
175 if (dev.idVendor == 0x0547 && dev.idProduct == 0x1002)
176 return 0;
177
178 /*----------------------------------------------------*/
179
180 /* devices that start up using the EZ-USB default device and
181 * which we can use after loading simple firmware. hotplug
182 * can fxload it, and then run this test driver.
183 *
184 * we return false positives in two cases:
185 * - the device has a "real" driver (maybe usb-serial) that
186 * renumerates. the device should vanish quickly.
187 * - the device doesn't have the test firmware installed.
188 */
189
190 /* generic EZ-USB FX controller */
191 if (dev.idVendor == 0x0547 && dev.idProduct == 0x2235)
192 return 0;
193
194 /* generic EZ-USB FX2 controller */
195 if (dev.idVendor == 0x04b4 && dev.idProduct == 0x8613)
196 return 0;
197
198 /* CY3671 development board with EZ-USB FX */
199 if (dev.idVendor == 0x0547 && dev.idProduct == 0x0080)
200 return 0;
201
202 /* Keyspan 19Qi uses an21xx (original EZ-USB) */
203 if (dev.idVendor == 0x06cd && dev.idProduct == 0x010b)
204 return 0;
205
206 /*----------------------------------------------------*/
207
208 /* "gadget zero", Linux-USB test software */
209 if (dev.idVendor == 0x0525 && dev.idProduct == 0xa4a0)
210 return 0;
211
212 /* user mode subset of that */
213 if (dev.idVendor == 0x0525 && dev.idProduct == 0xa4a4)
214 return testdev_ffs_ifnum(fd);
215 /* return 0; */
216
217 /* iso version of usermode code */
218 if (dev.idVendor == 0x0525 && dev.idProduct == 0xa4a3)
219 return 0;
220
221 /* some GPL'd test firmware uses these IDs */
222
223 if (dev.idVendor == 0xfff0 && dev.idProduct == 0xfff0)
224 return 0;
225
226 /*----------------------------------------------------*/
227
228 /* iBOT2 high speed webcam */
229 if (dev.idVendor == 0x0b62 && dev.idProduct == 0x0059)
230 return 0;
231
232 /*----------------------------------------------------*/
233
234 /* the FunctionFS gadget can have the source/sink interface
235 * anywhere. We look for an interface descriptor that match
236 * what we expect. We ignore configuratiens thou. */
237
238 if (dev.idVendor == 0x0525 && dev.idProduct == 0xa4ac
239 && (dev.bDeviceClass == USB_CLASS_PER_INTERFACE
240 || dev.bDeviceClass == USB_CLASS_VENDOR_SPEC))
241 return testdev_ffs_ifnum(fd);
242
243 return -1;
244}
245
246static int find_testdev(const char *name, const struct stat *sb, int flag)
247{
248 FILE *fd;
249 int ifnum;
250 struct testdev *entry;
251
252 (void)sb; /* unused */
253
254 if (flag != FTW_F)
255 return 0;
256
257 fd = fopen(name, "rb");
258 if (!fd) {
259 perror(name);
260 return 0;
261 }
262
263 ifnum = testdev_ifnum(fd);
264 fclose(fd);
265 if (ifnum < 0)
266 return 0;
267
268 entry = calloc(1, sizeof *entry);
269 if (!entry)
270 goto nomem;
271
272 entry->name = strdup(name);
273 if (!entry->name) {
274 free(entry);
275nomem:
276 perror("malloc");
277 return 0;
278 }
279
280 entry->ifnum = ifnum;
281
282 /* FIXME update USBDEVFS_CONNECTINFO so it tells about high speed etc */
283
284 fprintf(stderr, "%s speed\t%s\t%u\n",
285 speed(entry->speed), entry->name, entry->ifnum);
286
287 entry->next = testdevs;
288 testdevs = entry;
289 return 0;
290}
291
292static int
293usbdev_ioctl (int fd, int ifno, unsigned request, void *param)
294{
295 struct usbdevfs_ioctl wrapper;
296
297 wrapper.ifno = ifno;
298 wrapper.ioctl_code = request;
299 wrapper.data = param;
300
301 return ioctl (fd, USBDEVFS_IOCTL, &wrapper);
302}
303
304static void *handle_testdev (void *arg)
305{
306 struct testdev *dev = arg;
307 int fd, i;
308 int status;
309
310 if ((fd = open (dev->name, O_RDWR)) < 0) {
311 perror ("can't open dev file r/w");
312 return 0;
313 }
314
315restart:
316 for (i = 0; i < TEST_CASES; i++) {
317 if (dev->test != -1 && dev->test != i)
318 continue;
319 dev->param.test_num = i;
320
321 status = usbdev_ioctl (fd, dev->ifnum,
322 USBTEST_REQUEST, &dev->param);
323 if (status < 0 && errno == EOPNOTSUPP)
324 continue;
325
326 /* FIXME need a "syslog it" option for background testing */
327
328 /* NOTE: each thread emits complete lines; no fragments! */
329 if (status < 0) {
330 char buf [80];
331 int err = errno;
332
333 if (strerror_r (errno, buf, sizeof buf)) {
334 snprintf (buf, sizeof buf, "error %d", err);
335 errno = err;
336 }
337 printf ("%s test %d --> %d (%s)\n",
338 dev->name, i, errno, buf);
339 } else
340 printf ("%s test %d, %4d.%.06d secs\n", dev->name, i,
341 (int) dev->param.duration.tv_sec,
342 (int) dev->param.duration.tv_usec);
343
344 fflush (stdout);
345 }
346 if (dev->forever)
347 goto restart;
348
349 close (fd);
350 return arg;
351}
352
353static const char *usb_dir_find(void)
354{
355 static char udev_usb_path[] = "/dev/bus/usb";
356
357 if (access(udev_usb_path, F_OK) == 0)
358 return udev_usb_path;
359
360 return NULL;
361}
362
363static int parse_num(unsigned *num, const char *str)
364{
365 unsigned long val;
366 char *end;
367
368 errno = 0;
369 val = strtoul(str, &end, 0);
370 if (errno || *end || val > UINT_MAX)
371 return -1;
372 *num = val;
373 return 0;
374}
375
376int main (int argc, char **argv)
377{
378
379 int c;
380 struct testdev *entry;
381 char *device;
382 const char *usb_dir = NULL;
383 int all = 0, forever = 0, not = 0;
384 int test = -1 /* all */;
385 struct usbtest_param param;
386
387 /* pick defaults that works with all speeds, without short packets.
388 *
389 * Best per-frame data rates:
390 * super speed,bulk 1024 * 16 * 8 = 131072
391 * interrupt 1024 * 3 * 8 = 24576
392 * high speed, bulk 512 * 13 * 8 = 53248
393 * interrupt 1024 * 3 * 8 = 24576
394 * full speed, bulk/intr 64 * 19 = 1216
395 * interrupt 64 * 1 = 64
396 * low speed, interrupt 8 * 1 = 8
397 */
398 param.iterations = 1000;
399 param.length = 1024;
400 param.vary = 1024;
401 param.sglen = 32;
402
403 /* for easy use when hotplugging */
404 device = getenv ("DEVICE");
405
406 while ((c = getopt (argc, argv, "D:aA:c:g:hlns:t:v:")) != EOF)
407 switch (c) {
408 case 'D': /* device, if only one */
409 device = optarg;
410 continue;
411 case 'A': /* use all devices with specified USB dir */
412 usb_dir = optarg;
413 /* FALL THROUGH */
414 case 'a': /* use all devices */
415 device = NULL;
416 all = 1;
417 continue;
418 case 'c': /* count iterations */
419 if (parse_num(¶m.iterations, optarg))
420 goto usage;
421 continue;
422 case 'g': /* scatter/gather entries */
423 if (parse_num(¶m.sglen, optarg))
424 goto usage;
425 continue;
426 case 'l': /* loop forever */
427 forever = 1;
428 continue;
429 case 'n': /* no test running! */
430 not = 1;
431 continue;
432 case 's': /* size of packet */
433 if (parse_num(¶m.length, optarg))
434 goto usage;
435 continue;
436 case 't': /* run just one test */
437 test = atoi (optarg);
438 if (test < 0)
439 goto usage;
440 continue;
441 case 'v': /* vary packet size by ... */
442 if (parse_num(¶m.vary, optarg))
443 goto usage;
444 continue;
445 case '?':
446 case 'h':
447 default:
448usage:
449 fprintf (stderr,
450 "usage: %s [options]\n"
451 "Options:\n"
452 "\t-D dev only test specific device\n"
453 "\t-A usb-dir\n"
454 "\t-a test all recognized devices\n"
455 "\t-l loop forever(for stress test)\n"
456 "\t-t testnum only run specified case\n"
457 "\t-n no test running, show devices to be tested\n"
458 "Case arguments:\n"
459 "\t-c iterations default 1000\n"
460 "\t-s transfer length default 1024\n"
461 "\t-g sglen default 32\n"
462 "\t-v vary default 1024\n",
463 argv[0]);
464 return 1;
465 }
466 if (optind != argc)
467 goto usage;
468 if (!all && !device) {
469 fprintf (stderr, "must specify '-a' or '-D dev', "
470 "or DEVICE=/dev/bus/usb/BBB/DDD in env\n");
471 goto usage;
472 }
473
474 /* Find usb device subdirectory */
475 if (!usb_dir) {
476 usb_dir = usb_dir_find();
477 if (!usb_dir) {
478 fputs ("USB device files are missing\n", stderr);
479 return -1;
480 }
481 }
482
483 /* collect and list the test devices */
484 if (ftw (usb_dir, find_testdev, 3) != 0) {
485 fputs ("ftw failed; are USB device files missing?\n", stderr);
486 return -1;
487 }
488
489 /* quit, run single test, or create test threads */
490 if (!testdevs && !device) {
491 fputs ("no test devices recognized\n", stderr);
492 return -1;
493 }
494 if (not)
495 return 0;
496 if (testdevs && testdevs->next == 0 && !device)
497 device = testdevs->name;
498 for (entry = testdevs; entry; entry = entry->next) {
499 int status;
500
501 entry->param = param;
502 entry->forever = forever;
503 entry->test = test;
504
505 if (device) {
506 if (strcmp (entry->name, device))
507 continue;
508 return handle_testdev (entry) != entry;
509 }
510 status = pthread_create (&entry->thread, 0, handle_testdev, entry);
511 if (status)
512 perror ("pthread_create");
513 }
514 if (device) {
515 struct testdev dev;
516
517 /* kernel can recognize test devices we don't */
518 fprintf (stderr, "%s: %s may see only control tests\n",
519 argv [0], device);
520
521 memset (&dev, 0, sizeof dev);
522 dev.name = device;
523 dev.param = param;
524 dev.forever = forever;
525 dev.test = test;
526 return handle_testdev (&dev) != &dev;
527 }
528
529 /* wait for tests to complete */
530 for (entry = testdevs; entry; entry = entry->next) {
531 void *retval;
532
533 if (pthread_join (entry->thread, &retval))
534 perror ("pthread_join");
535 /* testing errors discarded! */
536 }
537
538 return 0;
539}