diff options
Diffstat (limited to 'tools')
-rw-r--r-- | tools/usb/ffs-test.c | 554 | ||||
-rw-r--r-- | tools/usb/testusb.c | 547 |
2 files changed, 1101 insertions, 0 deletions
diff --git a/tools/usb/ffs-test.c b/tools/usb/ffs-test.c new file mode 100644 index 000000000000..bbe2e3a2ea62 --- /dev/null +++ b/tools/usb/ffs-test.c | |||
@@ -0,0 +1,554 @@ | |||
1 | /* | ||
2 | * ffs-test.c.c -- user mode filesystem api for usb composite function | ||
3 | * | ||
4 | * Copyright (C) 2010 Samsung Electronics | ||
5 | * Author: Michal Nazarewicz <m.nazarewicz@samsung.com> | ||
6 | * | ||
7 | * This program is free software; you can redistribute it and/or modify | ||
8 | * it under the terms of the GNU General Public License as published by | ||
9 | * the Free Software Foundation; either version 2 of the License, or | ||
10 | * (at your option) any later version. | ||
11 | * | ||
12 | * This program is distributed in the hope that it will be useful, | ||
13 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
14 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | ||
15 | * GNU General Public License for more details. | ||
16 | * | ||
17 | * You should have received a copy of the GNU General Public License | ||
18 | * along with this program; if not, write to the Free Software | ||
19 | * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA | ||
20 | */ | ||
21 | |||
22 | /* $(CROSS_COMPILE)cc -Wall -Wextra -g -o ffs-test ffs-test.c -lpthread */ | ||
23 | |||
24 | |||
25 | #define _BSD_SOURCE /* for endian.h */ | ||
26 | |||
27 | #include <endian.h> | ||
28 | #include <errno.h> | ||
29 | #include <fcntl.h> | ||
30 | #include <pthread.h> | ||
31 | #include <stdarg.h> | ||
32 | #include <stdio.h> | ||
33 | #include <stdlib.h> | ||
34 | #include <string.h> | ||
35 | #include <sys/ioctl.h> | ||
36 | #include <sys/stat.h> | ||
37 | #include <sys/types.h> | ||
38 | #include <unistd.h> | ||
39 | |||
40 | #include <linux/usb/functionfs.h> | ||
41 | |||
42 | |||
43 | /******************** Little Endian Handling ********************************/ | ||
44 | |||
45 | #define cpu_to_le16(x) htole16(x) | ||
46 | #define cpu_to_le32(x) htole32(x) | ||
47 | #define le32_to_cpu(x) le32toh(x) | ||
48 | #define le16_to_cpu(x) le16toh(x) | ||
49 | |||
50 | static inline __u16 get_unaligned_le16(const void *_ptr) | ||
51 | { | ||
52 | const __u8 *ptr = _ptr; | ||
53 | return ptr[0] | (ptr[1] << 8); | ||
54 | } | ||
55 | |||
56 | static inline __u32 get_unaligned_le32(const void *_ptr) | ||
57 | { | ||
58 | const __u8 *ptr = _ptr; | ||
59 | return ptr[0] | (ptr[1] << 8) | (ptr[2] << 16) | (ptr[3] << 24); | ||
60 | } | ||
61 | |||
62 | static inline void put_unaligned_le16(__u16 val, void *_ptr) | ||
63 | { | ||
64 | __u8 *ptr = _ptr; | ||
65 | *ptr++ = val; | ||
66 | *ptr++ = val >> 8; | ||
67 | } | ||
68 | |||
69 | static inline void put_unaligned_le32(__u32 val, void *_ptr) | ||
70 | { | ||
71 | __u8 *ptr = _ptr; | ||
72 | *ptr++ = val; | ||
73 | *ptr++ = val >> 8; | ||
74 | *ptr++ = val >> 16; | ||
75 | *ptr++ = val >> 24; | ||
76 | } | ||
77 | |||
78 | |||
79 | /******************** Messages and Errors ***********************************/ | ||
80 | |||
81 | static const char argv0[] = "ffs-test"; | ||
82 | |||
83 | static unsigned verbosity = 7; | ||
84 | |||
85 | static void _msg(unsigned level, const char *fmt, ...) | ||
86 | { | ||
87 | if (level < 2) | ||
88 | level = 2; | ||
89 | else if (level > 7) | ||
90 | level = 7; | ||
91 | |||
92 | if (level <= verbosity) { | ||
93 | static const char levels[8][6] = { | ||
94 | [2] = "crit:", | ||
95 | [3] = "err: ", | ||
96 | [4] = "warn:", | ||
97 | [5] = "note:", | ||
98 | [6] = "info:", | ||
99 | [7] = "dbg: " | ||
100 | }; | ||
101 | |||
102 | int _errno = errno; | ||
103 | va_list ap; | ||
104 | |||
105 | fprintf(stderr, "%s: %s ", argv0, levels[level]); | ||
106 | va_start(ap, fmt); | ||
107 | vfprintf(stderr, fmt, ap); | ||
108 | va_end(ap); | ||
109 | |||
110 | if (fmt[strlen(fmt) - 1] != '\n') { | ||
111 | char buffer[128]; | ||
112 | strerror_r(_errno, buffer, sizeof buffer); | ||
113 | fprintf(stderr, ": (-%d) %s\n", _errno, buffer); | ||
114 | } | ||
115 | |||
116 | fflush(stderr); | ||
117 | } | ||
118 | } | ||
119 | |||
120 | #define die(...) (_msg(2, __VA_ARGS__), exit(1)) | ||
121 | #define err(...) _msg(3, __VA_ARGS__) | ||
122 | #define warn(...) _msg(4, __VA_ARGS__) | ||
123 | #define note(...) _msg(5, __VA_ARGS__) | ||
124 | #define info(...) _msg(6, __VA_ARGS__) | ||
125 | #define debug(...) _msg(7, __VA_ARGS__) | ||
126 | |||
127 | #define die_on(cond, ...) do { \ | ||
128 | if (cond) \ | ||
129 | die(__VA_ARGS__); \ | ||
130 | } while (0) | ||
131 | |||
132 | |||
133 | /******************** Descriptors and Strings *******************************/ | ||
134 | |||
135 | static const struct { | ||
136 | struct usb_functionfs_descs_head header; | ||
137 | struct { | ||
138 | struct usb_interface_descriptor intf; | ||
139 | struct usb_endpoint_descriptor_no_audio sink; | ||
140 | struct usb_endpoint_descriptor_no_audio source; | ||
141 | } __attribute__((packed)) fs_descs, hs_descs; | ||
142 | } __attribute__((packed)) descriptors = { | ||
143 | .header = { | ||
144 | .magic = cpu_to_le32(FUNCTIONFS_DESCRIPTORS_MAGIC), | ||
145 | .length = cpu_to_le32(sizeof descriptors), | ||
146 | .fs_count = 3, | ||
147 | .hs_count = 3, | ||
148 | }, | ||
149 | .fs_descs = { | ||
150 | .intf = { | ||
151 | .bLength = sizeof descriptors.fs_descs.intf, | ||
152 | .bDescriptorType = USB_DT_INTERFACE, | ||
153 | .bNumEndpoints = 2, | ||
154 | .bInterfaceClass = USB_CLASS_VENDOR_SPEC, | ||
155 | .iInterface = 1, | ||
156 | }, | ||
157 | .sink = { | ||
158 | .bLength = sizeof descriptors.fs_descs.sink, | ||
159 | .bDescriptorType = USB_DT_ENDPOINT, | ||
160 | .bEndpointAddress = 1 | USB_DIR_IN, | ||
161 | .bmAttributes = USB_ENDPOINT_XFER_BULK, | ||
162 | /* .wMaxPacketSize = autoconfiguration (kernel) */ | ||
163 | }, | ||
164 | .source = { | ||
165 | .bLength = sizeof descriptors.fs_descs.source, | ||
166 | .bDescriptorType = USB_DT_ENDPOINT, | ||
167 | .bEndpointAddress = 2 | USB_DIR_OUT, | ||
168 | .bmAttributes = USB_ENDPOINT_XFER_BULK, | ||
169 | /* .wMaxPacketSize = autoconfiguration (kernel) */ | ||
170 | }, | ||
171 | }, | ||
172 | .hs_descs = { | ||
173 | .intf = { | ||
174 | .bLength = sizeof descriptors.fs_descs.intf, | ||
175 | .bDescriptorType = USB_DT_INTERFACE, | ||
176 | .bNumEndpoints = 2, | ||
177 | .bInterfaceClass = USB_CLASS_VENDOR_SPEC, | ||
178 | .iInterface = 1, | ||
179 | }, | ||
180 | .sink = { | ||
181 | .bLength = sizeof descriptors.hs_descs.sink, | ||
182 | .bDescriptorType = USB_DT_ENDPOINT, | ||
183 | .bEndpointAddress = 1 | USB_DIR_IN, | ||
184 | .bmAttributes = USB_ENDPOINT_XFER_BULK, | ||
185 | .wMaxPacketSize = cpu_to_le16(512), | ||
186 | }, | ||
187 | .source = { | ||
188 | .bLength = sizeof descriptors.hs_descs.source, | ||
189 | .bDescriptorType = USB_DT_ENDPOINT, | ||
190 | .bEndpointAddress = 2 | USB_DIR_OUT, | ||
191 | .bmAttributes = USB_ENDPOINT_XFER_BULK, | ||
192 | .wMaxPacketSize = cpu_to_le16(512), | ||
193 | .bInterval = 1, /* NAK every 1 uframe */ | ||
194 | }, | ||
195 | }, | ||
196 | }; | ||
197 | |||
198 | |||
199 | #define STR_INTERFACE_ "Source/Sink" | ||
200 | |||
201 | static const struct { | ||
202 | struct usb_functionfs_strings_head header; | ||
203 | struct { | ||
204 | __le16 code; | ||
205 | const char str1[sizeof STR_INTERFACE_]; | ||
206 | } __attribute__((packed)) lang0; | ||
207 | } __attribute__((packed)) strings = { | ||
208 | .header = { | ||
209 | .magic = cpu_to_le32(FUNCTIONFS_STRINGS_MAGIC), | ||
210 | .length = cpu_to_le32(sizeof strings), | ||
211 | .str_count = cpu_to_le32(1), | ||
212 | .lang_count = cpu_to_le32(1), | ||
213 | }, | ||
214 | .lang0 = { | ||
215 | cpu_to_le16(0x0409), /* en-us */ | ||
216 | STR_INTERFACE_, | ||
217 | }, | ||
218 | }; | ||
219 | |||
220 | #define STR_INTERFACE strings.lang0.str1 | ||
221 | |||
222 | |||
223 | /******************** Files and Threads Handling ****************************/ | ||
224 | |||
225 | struct thread; | ||
226 | |||
227 | static ssize_t read_wrap(struct thread *t, void *buf, size_t nbytes); | ||
228 | static ssize_t write_wrap(struct thread *t, const void *buf, size_t nbytes); | ||
229 | static ssize_t ep0_consume(struct thread *t, const void *buf, size_t nbytes); | ||
230 | static ssize_t fill_in_buf(struct thread *t, void *buf, size_t nbytes); | ||
231 | static ssize_t empty_out_buf(struct thread *t, const void *buf, size_t nbytes); | ||
232 | |||
233 | |||
234 | static struct thread { | ||
235 | const char *const filename; | ||
236 | size_t buf_size; | ||
237 | |||
238 | ssize_t (*in)(struct thread *, void *, size_t); | ||
239 | const char *const in_name; | ||
240 | |||
241 | ssize_t (*out)(struct thread *, const void *, size_t); | ||
242 | const char *const out_name; | ||
243 | |||
244 | int fd; | ||
245 | pthread_t id; | ||
246 | void *buf; | ||
247 | ssize_t status; | ||
248 | } threads[] = { | ||
249 | { | ||
250 | "ep0", 4 * sizeof(struct usb_functionfs_event), | ||
251 | read_wrap, NULL, | ||
252 | ep0_consume, "<consume>", | ||
253 | 0, 0, NULL, 0 | ||
254 | }, | ||
255 | { | ||
256 | "ep1", 8 * 1024, | ||
257 | fill_in_buf, "<in>", | ||
258 | write_wrap, NULL, | ||
259 | 0, 0, NULL, 0 | ||
260 | }, | ||
261 | { | ||
262 | "ep2", 8 * 1024, | ||
263 | read_wrap, NULL, | ||
264 | empty_out_buf, "<out>", | ||
265 | 0, 0, NULL, 0 | ||
266 | }, | ||
267 | }; | ||
268 | |||
269 | |||
270 | static void init_thread(struct thread *t) | ||
271 | { | ||
272 | t->buf = malloc(t->buf_size); | ||
273 | die_on(!t->buf, "malloc"); | ||
274 | |||
275 | t->fd = open(t->filename, O_RDWR); | ||
276 | die_on(t->fd < 0, "%s", t->filename); | ||
277 | } | ||
278 | |||
279 | static void cleanup_thread(void *arg) | ||
280 | { | ||
281 | struct thread *t = arg; | ||
282 | int ret, fd; | ||
283 | |||
284 | fd = t->fd; | ||
285 | if (t->fd < 0) | ||
286 | return; | ||
287 | t->fd = -1; | ||
288 | |||
289 | /* test the FIFO ioctls (non-ep0 code paths) */ | ||
290 | if (t != threads) { | ||
291 | ret = ioctl(fd, FUNCTIONFS_FIFO_STATUS); | ||
292 | if (ret < 0) { | ||
293 | /* ENODEV reported after disconnect */ | ||
294 | if (errno != ENODEV) | ||
295 | err("%s: get fifo status", t->filename); | ||
296 | } else if (ret) { | ||
297 | warn("%s: unclaimed = %d\n", t->filename, ret); | ||
298 | if (ioctl(fd, FUNCTIONFS_FIFO_FLUSH) < 0) | ||
299 | err("%s: fifo flush", t->filename); | ||
300 | } | ||
301 | } | ||
302 | |||
303 | if (close(fd) < 0) | ||
304 | err("%s: close", t->filename); | ||
305 | |||
306 | free(t->buf); | ||
307 | t->buf = NULL; | ||
308 | } | ||
309 | |||
310 | static void *start_thread_helper(void *arg) | ||
311 | { | ||
312 | const char *name, *op, *in_name, *out_name; | ||
313 | struct thread *t = arg; | ||
314 | ssize_t ret; | ||
315 | |||
316 | info("%s: starts\n", t->filename); | ||
317 | in_name = t->in_name ? t->in_name : t->filename; | ||
318 | out_name = t->out_name ? t->out_name : t->filename; | ||
319 | |||
320 | pthread_cleanup_push(cleanup_thread, arg); | ||
321 | |||
322 | for (;;) { | ||
323 | pthread_testcancel(); | ||
324 | |||
325 | ret = t->in(t, t->buf, t->buf_size); | ||
326 | if (ret > 0) { | ||
327 | ret = t->out(t, t->buf, t->buf_size); | ||
328 | name = out_name; | ||
329 | op = "write"; | ||
330 | } else { | ||
331 | name = in_name; | ||
332 | op = "read"; | ||
333 | } | ||
334 | |||
335 | if (ret > 0) { | ||
336 | /* nop */ | ||
337 | } else if (!ret) { | ||
338 | debug("%s: %s: EOF", name, op); | ||
339 | break; | ||
340 | } else if (errno == EINTR || errno == EAGAIN) { | ||
341 | debug("%s: %s", name, op); | ||
342 | } else { | ||
343 | warn("%s: %s", name, op); | ||
344 | break; | ||
345 | } | ||
346 | } | ||
347 | |||
348 | pthread_cleanup_pop(1); | ||
349 | |||
350 | t->status = ret; | ||
351 | info("%s: ends\n", t->filename); | ||
352 | return NULL; | ||
353 | } | ||
354 | |||
355 | static void start_thread(struct thread *t) | ||
356 | { | ||
357 | debug("%s: starting\n", t->filename); | ||
358 | |||
359 | die_on(pthread_create(&t->id, NULL, start_thread_helper, t) < 0, | ||
360 | "pthread_create(%s)", t->filename); | ||
361 | } | ||
362 | |||
363 | static void join_thread(struct thread *t) | ||
364 | { | ||
365 | int ret = pthread_join(t->id, NULL); | ||
366 | |||
367 | if (ret < 0) | ||
368 | err("%s: joining thread", t->filename); | ||
369 | else | ||
370 | debug("%s: joined\n", t->filename); | ||
371 | } | ||
372 | |||
373 | |||
374 | static ssize_t read_wrap(struct thread *t, void *buf, size_t nbytes) | ||
375 | { | ||
376 | return read(t->fd, buf, nbytes); | ||
377 | } | ||
378 | |||
379 | static ssize_t write_wrap(struct thread *t, const void *buf, size_t nbytes) | ||
380 | { | ||
381 | return write(t->fd, buf, nbytes); | ||
382 | } | ||
383 | |||
384 | |||
385 | /******************** Empty/Fill buffer routines ****************************/ | ||
386 | |||
387 | /* 0 -- stream of zeros, 1 -- i % 63, 2 -- pipe */ | ||
388 | enum pattern { PAT_ZERO, PAT_SEQ, PAT_PIPE }; | ||
389 | static enum pattern pattern; | ||
390 | |||
391 | static ssize_t | ||
392 | fill_in_buf(struct thread *ignore, void *buf, size_t nbytes) | ||
393 | { | ||
394 | size_t i; | ||
395 | __u8 *p; | ||
396 | |||
397 | (void)ignore; | ||
398 | |||
399 | switch (pattern) { | ||
400 | case PAT_ZERO: | ||
401 | memset(buf, 0, nbytes); | ||
402 | break; | ||
403 | |||
404 | case PAT_SEQ: | ||
405 | for (p = buf, i = 0; i < nbytes; ++i, ++p) | ||
406 | *p = i % 63; | ||
407 | break; | ||
408 | |||
409 | case PAT_PIPE: | ||
410 | return fread(buf, 1, nbytes, stdin); | ||
411 | } | ||
412 | |||
413 | return nbytes; | ||
414 | } | ||
415 | |||
416 | static ssize_t | ||
417 | empty_out_buf(struct thread *ignore, const void *buf, size_t nbytes) | ||
418 | { | ||
419 | const __u8 *p; | ||
420 | __u8 expected; | ||
421 | ssize_t ret; | ||
422 | size_t len; | ||
423 | |||
424 | (void)ignore; | ||
425 | |||
426 | switch (pattern) { | ||
427 | case PAT_ZERO: | ||
428 | expected = 0; | ||
429 | for (p = buf, len = 0; len < nbytes; ++p, ++len) | ||
430 | if (*p) | ||
431 | goto invalid; | ||
432 | break; | ||
433 | |||
434 | case PAT_SEQ: | ||
435 | for (p = buf, len = 0; len < nbytes; ++p, ++len) | ||
436 | if (*p != len % 63) { | ||
437 | expected = len % 63; | ||
438 | goto invalid; | ||
439 | } | ||
440 | break; | ||
441 | |||
442 | case PAT_PIPE: | ||
443 | ret = fwrite(buf, nbytes, 1, stdout); | ||
444 | if (ret > 0) | ||
445 | fflush(stdout); | ||
446 | break; | ||
447 | |||
448 | invalid: | ||
449 | err("bad OUT byte %zd, expected %02x got %02x\n", | ||
450 | len, expected, *p); | ||
451 | for (p = buf, len = 0; len < nbytes; ++p, ++len) { | ||
452 | if (0 == (len % 32)) | ||
453 | fprintf(stderr, "%4d:", len); | ||
454 | fprintf(stderr, " %02x", *p); | ||
455 | if (31 == (len % 32)) | ||
456 | fprintf(stderr, "\n"); | ||
457 | } | ||
458 | fflush(stderr); | ||
459 | errno = EILSEQ; | ||
460 | return -1; | ||
461 | } | ||
462 | |||
463 | return len; | ||
464 | } | ||
465 | |||
466 | |||
467 | /******************** Endpoints routines ************************************/ | ||
468 | |||
469 | static void handle_setup(const struct usb_ctrlrequest *setup) | ||
470 | { | ||
471 | printf("bRequestType = %d\n", setup->bRequestType); | ||
472 | printf("bRequest = %d\n", setup->bRequest); | ||
473 | printf("wValue = %d\n", le16_to_cpu(setup->wValue)); | ||
474 | printf("wIndex = %d\n", le16_to_cpu(setup->wIndex)); | ||
475 | printf("wLength = %d\n", le16_to_cpu(setup->wLength)); | ||
476 | } | ||
477 | |||
478 | static ssize_t | ||
479 | ep0_consume(struct thread *ignore, const void *buf, size_t nbytes) | ||
480 | { | ||
481 | static const char *const names[] = { | ||
482 | [FUNCTIONFS_BIND] = "BIND", | ||
483 | [FUNCTIONFS_UNBIND] = "UNBIND", | ||
484 | [FUNCTIONFS_ENABLE] = "ENABLE", | ||
485 | [FUNCTIONFS_DISABLE] = "DISABLE", | ||
486 | [FUNCTIONFS_SETUP] = "SETUP", | ||
487 | [FUNCTIONFS_SUSPEND] = "SUSPEND", | ||
488 | [FUNCTIONFS_RESUME] = "RESUME", | ||
489 | }; | ||
490 | |||
491 | const struct usb_functionfs_event *event = buf; | ||
492 | size_t n; | ||
493 | |||
494 | (void)ignore; | ||
495 | |||
496 | for (n = nbytes / sizeof *event; n; --n, ++event) | ||
497 | switch (event->type) { | ||
498 | case FUNCTIONFS_BIND: | ||
499 | case FUNCTIONFS_UNBIND: | ||
500 | case FUNCTIONFS_ENABLE: | ||
501 | case FUNCTIONFS_DISABLE: | ||
502 | case FUNCTIONFS_SETUP: | ||
503 | case FUNCTIONFS_SUSPEND: | ||
504 | case FUNCTIONFS_RESUME: | ||
505 | printf("Event %s\n", names[event->type]); | ||
506 | if (event->type == FUNCTIONFS_SETUP) | ||
507 | handle_setup(&event->u.setup); | ||
508 | break; | ||
509 | |||
510 | default: | ||
511 | printf("Event %03u (unknown)\n", event->type); | ||
512 | } | ||
513 | |||
514 | return nbytes; | ||
515 | } | ||
516 | |||
517 | static void ep0_init(struct thread *t) | ||
518 | { | ||
519 | ssize_t ret; | ||
520 | |||
521 | info("%s: writing descriptors\n", t->filename); | ||
522 | ret = write(t->fd, &descriptors, sizeof descriptors); | ||
523 | die_on(ret < 0, "%s: write: descriptors", t->filename); | ||
524 | |||
525 | info("%s: writing strings\n", t->filename); | ||
526 | ret = write(t->fd, &strings, sizeof strings); | ||
527 | die_on(ret < 0, "%s: write: strings", t->filename); | ||
528 | } | ||
529 | |||
530 | |||
531 | /******************** Main **************************************************/ | ||
532 | |||
533 | int main(void) | ||
534 | { | ||
535 | unsigned i; | ||
536 | |||
537 | /* XXX TODO: Argument parsing missing */ | ||
538 | |||
539 | init_thread(threads); | ||
540 | ep0_init(threads); | ||
541 | |||
542 | for (i = 1; i < sizeof threads / sizeof *threads; ++i) | ||
543 | init_thread(threads + i); | ||
544 | |||
545 | for (i = 1; i < sizeof threads / sizeof *threads; ++i) | ||
546 | start_thread(threads + i); | ||
547 | |||
548 | start_thread_helper(threads); | ||
549 | |||
550 | for (i = 1; i < sizeof threads / sizeof *threads; ++i) | ||
551 | join_thread(threads + i); | ||
552 | |||
553 | return 0; | ||
554 | } | ||
diff --git a/tools/usb/testusb.c b/tools/usb/testusb.c new file mode 100644 index 000000000000..f08e89463842 --- /dev/null +++ b/tools/usb/testusb.c | |||
@@ -0,0 +1,547 @@ | |||
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 <m.nazarewicz@samsung.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 | |||
55 | struct 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 | |||
79 | struct 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 | |||
96 | struct 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 | |||
109 | enum 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 | |||
117 | static 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 | |||
128 | struct 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 | }; | ||
139 | static struct testdev *testdevs; | ||
140 | |||
141 | static 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 | |||
164 | static 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 | |||
246 | static 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 | /* ignore /proc/bus/usb/{devices,drivers} */ | ||
257 | if (strrchr(name, '/')[1] == 'd') | ||
258 | return 0; | ||
259 | |||
260 | fd = fopen(name, "rb"); | ||
261 | if (!fd) { | ||
262 | perror(name); | ||
263 | return 0; | ||
264 | } | ||
265 | |||
266 | ifnum = testdev_ifnum(fd); | ||
267 | fclose(fd); | ||
268 | if (ifnum < 0) | ||
269 | return 0; | ||
270 | |||
271 | entry = calloc(1, sizeof *entry); | ||
272 | if (!entry) | ||
273 | goto nomem; | ||
274 | |||
275 | entry->name = strdup(name); | ||
276 | if (!entry->name) { | ||
277 | free(entry); | ||
278 | nomem: | ||
279 | perror("malloc"); | ||
280 | return 0; | ||
281 | } | ||
282 | |||
283 | entry->ifnum = ifnum; | ||
284 | |||
285 | /* FIXME ask usbfs what speed; update USBDEVFS_CONNECTINFO so | ||
286 | * it tells about high speed etc */ | ||
287 | |||
288 | fprintf(stderr, "%s speed\t%s\t%u\n", | ||
289 | speed(entry->speed), entry->name, entry->ifnum); | ||
290 | |||
291 | entry->next = testdevs; | ||
292 | testdevs = entry; | ||
293 | return 0; | ||
294 | } | ||
295 | |||
296 | static int | ||
297 | usbdev_ioctl (int fd, int ifno, unsigned request, void *param) | ||
298 | { | ||
299 | struct usbdevfs_ioctl wrapper; | ||
300 | |||
301 | wrapper.ifno = ifno; | ||
302 | wrapper.ioctl_code = request; | ||
303 | wrapper.data = param; | ||
304 | |||
305 | return ioctl (fd, USBDEVFS_IOCTL, &wrapper); | ||
306 | } | ||
307 | |||
308 | static void *handle_testdev (void *arg) | ||
309 | { | ||
310 | struct testdev *dev = arg; | ||
311 | int fd, i; | ||
312 | int status; | ||
313 | |||
314 | if ((fd = open (dev->name, O_RDWR)) < 0) { | ||
315 | perror ("can't open dev file r/w"); | ||
316 | return 0; | ||
317 | } | ||
318 | |||
319 | restart: | ||
320 | for (i = 0; i < TEST_CASES; i++) { | ||
321 | if (dev->test != -1 && dev->test != i) | ||
322 | continue; | ||
323 | dev->param.test_num = i; | ||
324 | |||
325 | status = usbdev_ioctl (fd, dev->ifnum, | ||
326 | USBTEST_REQUEST, &dev->param); | ||
327 | if (status < 0 && errno == EOPNOTSUPP) | ||
328 | continue; | ||
329 | |||
330 | /* FIXME need a "syslog it" option for background testing */ | ||
331 | |||
332 | /* NOTE: each thread emits complete lines; no fragments! */ | ||
333 | if (status < 0) { | ||
334 | char buf [80]; | ||
335 | int err = errno; | ||
336 | |||
337 | if (strerror_r (errno, buf, sizeof buf)) { | ||
338 | snprintf (buf, sizeof buf, "error %d", err); | ||
339 | errno = err; | ||
340 | } | ||
341 | printf ("%s test %d --> %d (%s)\n", | ||
342 | dev->name, i, errno, buf); | ||
343 | } else | ||
344 | printf ("%s test %d, %4d.%.06d secs\n", dev->name, i, | ||
345 | (int) dev->param.duration.tv_sec, | ||
346 | (int) dev->param.duration.tv_usec); | ||
347 | |||
348 | fflush (stdout); | ||
349 | } | ||
350 | if (dev->forever) | ||
351 | goto restart; | ||
352 | |||
353 | close (fd); | ||
354 | return arg; | ||
355 | } | ||
356 | |||
357 | static const char *usbfs_dir_find(void) | ||
358 | { | ||
359 | static char usbfs_path_0[] = "/dev/usb/devices"; | ||
360 | static char usbfs_path_1[] = "/proc/bus/usb/devices"; | ||
361 | |||
362 | static char *const usbfs_paths[] = { | ||
363 | usbfs_path_0, usbfs_path_1 | ||
364 | }; | ||
365 | |||
366 | static char *const * | ||
367 | end = usbfs_paths + sizeof usbfs_paths / sizeof *usbfs_paths; | ||
368 | |||
369 | char *const *it = usbfs_paths; | ||
370 | do { | ||
371 | int fd = open(*it, O_RDONLY); | ||
372 | close(fd); | ||
373 | if (fd >= 0) { | ||
374 | strrchr(*it, '/')[0] = '\0'; | ||
375 | return *it; | ||
376 | } | ||
377 | } while (++it != end); | ||
378 | |||
379 | return NULL; | ||
380 | } | ||
381 | |||
382 | static int parse_num(unsigned *num, const char *str) | ||
383 | { | ||
384 | unsigned long val; | ||
385 | char *end; | ||
386 | |||
387 | errno = 0; | ||
388 | val = strtoul(str, &end, 0); | ||
389 | if (errno || *end || val > UINT_MAX) | ||
390 | return -1; | ||
391 | *num = val; | ||
392 | return 0; | ||
393 | } | ||
394 | |||
395 | int main (int argc, char **argv) | ||
396 | { | ||
397 | |||
398 | int c; | ||
399 | struct testdev *entry; | ||
400 | char *device; | ||
401 | const char *usbfs_dir = NULL; | ||
402 | int all = 0, forever = 0, not = 0; | ||
403 | int test = -1 /* all */; | ||
404 | struct usbtest_param param; | ||
405 | |||
406 | /* pick defaults that works with all speeds, without short packets. | ||
407 | * | ||
408 | * Best per-frame data rates: | ||
409 | * high speed, bulk 512 * 13 * 8 = 53248 | ||
410 | * interrupt 1024 * 3 * 8 = 24576 | ||
411 | * full speed, bulk/intr 64 * 19 = 1216 | ||
412 | * interrupt 64 * 1 = 64 | ||
413 | * low speed, interrupt 8 * 1 = 8 | ||
414 | */ | ||
415 | param.iterations = 1000; | ||
416 | param.length = 512; | ||
417 | param.vary = 512; | ||
418 | param.sglen = 32; | ||
419 | |||
420 | /* for easy use when hotplugging */ | ||
421 | device = getenv ("DEVICE"); | ||
422 | |||
423 | while ((c = getopt (argc, argv, "D:aA:c:g:hns:t:v:")) != EOF) | ||
424 | switch (c) { | ||
425 | case 'D': /* device, if only one */ | ||
426 | device = optarg; | ||
427 | continue; | ||
428 | case 'A': /* use all devices with specified usbfs dir */ | ||
429 | usbfs_dir = optarg; | ||
430 | /* FALL THROUGH */ | ||
431 | case 'a': /* use all devices */ | ||
432 | device = NULL; | ||
433 | all = 1; | ||
434 | continue; | ||
435 | case 'c': /* count iterations */ | ||
436 | if (parse_num(¶m.iterations, optarg)) | ||
437 | goto usage; | ||
438 | continue; | ||
439 | case 'g': /* scatter/gather entries */ | ||
440 | if (parse_num(¶m.sglen, optarg)) | ||
441 | goto usage; | ||
442 | continue; | ||
443 | case 'l': /* loop forever */ | ||
444 | forever = 1; | ||
445 | continue; | ||
446 | case 'n': /* no test running! */ | ||
447 | not = 1; | ||
448 | continue; | ||
449 | case 's': /* size of packet */ | ||
450 | if (parse_num(¶m.length, optarg)) | ||
451 | goto usage; | ||
452 | continue; | ||
453 | case 't': /* run just one test */ | ||
454 | test = atoi (optarg); | ||
455 | if (test < 0) | ||
456 | goto usage; | ||
457 | continue; | ||
458 | case 'v': /* vary packet size by ... */ | ||
459 | if (parse_num(¶m.vary, optarg)) | ||
460 | goto usage; | ||
461 | continue; | ||
462 | case '?': | ||
463 | case 'h': | ||
464 | default: | ||
465 | usage: | ||
466 | fprintf (stderr, "usage: %s [-n] [-D dev | -a | -A usbfs-dir]\n" | ||
467 | "\t[-c iterations] [-t testnum]\n" | ||
468 | "\t[-s packetsize] [-g sglen] [-v vary]\n", | ||
469 | argv [0]); | ||
470 | return 1; | ||
471 | } | ||
472 | if (optind != argc) | ||
473 | goto usage; | ||
474 | if (!all && !device) { | ||
475 | fprintf (stderr, "must specify '-a' or '-D dev', " | ||
476 | "or DEVICE=/proc/bus/usb/BBB/DDD in env\n"); | ||
477 | goto usage; | ||
478 | } | ||
479 | |||
480 | /* Find usbfs mount point */ | ||
481 | if (!usbfs_dir) { | ||
482 | usbfs_dir = usbfs_dir_find(); | ||
483 | if (!usbfs_dir) { | ||
484 | fputs ("usbfs files are missing\n", stderr); | ||
485 | return -1; | ||
486 | } | ||
487 | } | ||
488 | |||
489 | /* collect and list the test devices */ | ||
490 | if (ftw (usbfs_dir, find_testdev, 3) != 0) { | ||
491 | fputs ("ftw failed; is usbfs missing?\n", stderr); | ||
492 | return -1; | ||
493 | } | ||
494 | |||
495 | /* quit, run single test, or create test threads */ | ||
496 | if (!testdevs && !device) { | ||
497 | fputs ("no test devices recognized\n", stderr); | ||
498 | return -1; | ||
499 | } | ||
500 | if (not) | ||
501 | return 0; | ||
502 | if (testdevs && testdevs->next == 0 && !device) | ||
503 | device = testdevs->name; | ||
504 | for (entry = testdevs; entry; entry = entry->next) { | ||
505 | int status; | ||
506 | |||
507 | entry->param = param; | ||
508 | entry->forever = forever; | ||
509 | entry->test = test; | ||
510 | |||
511 | if (device) { | ||
512 | if (strcmp (entry->name, device)) | ||
513 | continue; | ||
514 | return handle_testdev (entry) != entry; | ||
515 | } | ||
516 | status = pthread_create (&entry->thread, 0, handle_testdev, entry); | ||
517 | if (status) { | ||
518 | perror ("pthread_create"); | ||
519 | continue; | ||
520 | } | ||
521 | } | ||
522 | if (device) { | ||
523 | struct testdev dev; | ||
524 | |||
525 | /* kernel can recognize test devices we don't */ | ||
526 | fprintf (stderr, "%s: %s may see only control tests\n", | ||
527 | argv [0], device); | ||
528 | |||
529 | memset (&dev, 0, sizeof dev); | ||
530 | dev.name = device; | ||
531 | dev.param = param; | ||
532 | dev.forever = forever; | ||
533 | dev.test = test; | ||
534 | return handle_testdev (&dev) != &dev; | ||
535 | } | ||
536 | |||
537 | /* wait for tests to complete */ | ||
538 | for (entry = testdevs; entry; entry = entry->next) { | ||
539 | void *retval; | ||
540 | |||
541 | if (pthread_join (entry->thread, &retval)) | ||
542 | perror ("pthread_join"); | ||
543 | /* testing errors discarded! */ | ||
544 | } | ||
545 | |||
546 | return 0; | ||
547 | } | ||