drivers/usb/misc/usbtest.c

   1 #include <linux/kernel.h>
   2 #include <linux/errno.h>
   3 #include <linux/init.h>
   4 #include <linux/slab.h>
   5 #include <linux/mm.h>
   6 #include <linux/module.h>
   7 #include <linux/moduleparam.h>
   8 #include <linux/scatterlist.h>
   9 #include <linux/mutex.h>
  10 #include <linux/timer.h>
  11 #include <linux/usb.h>
  12
  13 #define SIMPLE_IO_TIMEOUT       10000   /* in milliseconds */
  14
  15 /*-------------------------------------------------------------------------*/
  16
  17 static int override_alt = -1;
  18 module_param_named(alt, override_alt, int, 0644);
  19 MODULE_PARM_DESC(alt, ">= 0 to override altsetting selection");
  20
  21 /*-------------------------------------------------------------------------*/
  22
  23 /* FIXME make these public somewhere; usbdevfs.h? */
  24 struct usbtest_param {
  25         /* inputs */
  26         unsigned                test_num;       /* 0..(TEST_CASES-1) */
  27         unsigned                iterations;
  28         unsigned                length;
  29         unsigned                vary;
  30         unsigned                sglen;
  31
  32         /* outputs */
  33         struct timeval          duration;
  34 };
  35 #define USBTEST_REQUEST _IOWR('U', 100, struct usbtest_param)
  36
  37 /*-------------------------------------------------------------------------*/
  38
  39 #define GENERIC         /* let probe() bind using module params */
  40
  41 /* Some devices that can be used for testing will have "real" drivers.
  42  * Entries for those need to be enabled here by hand, after disabling
  43  * that "real" driver.
  44  */
  45 //#define       IBOT2           /* grab iBOT2 webcams */
  46 //#define       KEYSPAN_19Qi    /* grab un-renumerated serial adapter */
  47
  48 /*-------------------------------------------------------------------------*/
  49
  50 struct usbtest_info {
  51         const char              *name;
  52         u8                      ep_in;          /* bulk/intr source */
  53         u8                      ep_out;         /* bulk/intr sink */
  54         unsigned                autoconf:1;
  55         unsigned                ctrl_out:1;
  56         unsigned                iso:1;          /* try iso in/out */
  57         int                     alt;
  58 };
  59
  60 /* this is accessed only through usbfs ioctl calls.
  61  * one ioctl to issue a test ... one lock per device.
  62  * tests create other threads if they need them.
  63  * urbs and buffers are allocated dynamically,
  64  * and data generated deterministically.
  65  */
  66 struct usbtest_dev {
  67         struct usb_interface    *intf;
  68         struct usbtest_info     *info;
  69         int                     in_pipe;
  70         int                     out_pipe;
  71         int                     in_iso_pipe;
  72         int                     out_iso_pipe;
  73         struct usb_endpoint_descriptor  *iso_in, *iso_out;
  74         struct mutex            lock;
  75
  76 #define TBUF_SIZE       256
  77         u8                      *buf;
  78 };
  79
  80 static struct usb_device *testdev_to_usbdev(struct usbtest_dev *test)
  81 {
  82         return interface_to_usbdev(test->intf);
  83 }
  84
  85 /* set up all urbs so they can be used with either bulk or interrupt */
  86 #define INTERRUPT_RATE          1       /* msec/transfer */
  87
  88 #define ERROR(tdev, fmt, args...) \
  89         dev_err(&(tdev)->intf->dev , fmt , ## args)
  90 #define WARNING(tdev, fmt, args...) \
  91         dev_warn(&(tdev)->intf->dev , fmt , ## args)
  92
  93 #define GUARD_BYTE      0xA5
  94
  95 /*-------------------------------------------------------------------------*/
  96
  97 static int
  98 get_endpoints(struct usbtest_dev *dev, struct usb_interface *intf)
  99 {
 100         int                             tmp;
 101         struct usb_host_interface       *alt;
 102         struct usb_host_endpoint        *in, *out;
 103         struct usb_host_endpoint        *iso_in, *iso_out;
 104         struct usb_device               *udev;
 105
 106         for (tmp = 0; tmp < intf->num_altsetting; tmp++) {
 107                 unsigned        ep;
 108
 109                 in = out = NULL;
 110                 iso_in = iso_out = NULL;
 111                 alt = intf->altsetting + tmp;
 112
 113                 if (override_alt >= 0 &&
 114                                 override_alt != alt->desc.bAlternateSetting)
 115                         continue;
 116
 117                 /* take the first altsetting with in-bulk + out-bulk;
 118                  * ignore other endpoints and altsettings.
 119                  */
 120                 for (ep = 0; ep < alt->desc.bNumEndpoints; ep++) {
 121                         struct usb_host_endpoint        *e;
 122
 123                         e = alt->endpoint + ep;
 124                         switch (e->desc.bmAttributes) {
 125                         case USB_ENDPOINT_XFER_BULK:
 126                                 break;
 127                         case USB_ENDPOINT_XFER_ISOC:
 128                                 if (dev->info->iso)
 129                                         goto try_iso;
 130                                 /* FALLTHROUGH */
 131                         default:
 132                                 continue;
 133                         }
 134                         if (usb_endpoint_dir_in(&e->desc)) {
 135                                 if (!in)
 136                                         in = e;
 137                         } else {
 138                                 if (!out)
 139                                         out = e;
 140                         }
 141                         continue;
 142 try_iso:
 143                         if (usb_endpoint_dir_in(&e->desc)) {
 144                                 if (!iso_in)
 145                                         iso_in = e;
 146                         } else {
 147                                 if (!iso_out)
 148                                         iso_out = e;
 149                         }
 150                 }
 151                 if ((in && out)  ||  iso_in || iso_out)
 152                         goto found;
 153         }
 154         return -EINVAL;
 155
 156 found:
 157         udev = testdev_to_usbdev(dev);
 158         dev->info->alt = alt->desc.bAlternateSetting;
 159         if (alt->desc.bAlternateSetting != 0) {
 160                 tmp = usb_set_interface(udev,
 161                                 alt->desc.bInterfaceNumber,
 162                                 alt->desc.bAlternateSetting);
 163                 if (tmp < 0)
 164                         return tmp;
 165         }
 166
 167         if (in) {
 168                 dev->in_pipe = usb_rcvbulkpipe(udev,
 169                         in->desc.bEndpointAddress & USB_ENDPOINT_NUMBER_MASK);
 170                 dev->out_pipe = usb_sndbulkpipe(udev,
 171                         out->desc.bEndpointAddress & USB_ENDPOINT_NUMBER_MASK);
 172         }
 173         if (iso_in) {
 174                 dev->iso_in = &iso_in->desc;
 175                 dev->in_iso_pipe = usb_rcvisocpipe(udev,
 176                                 iso_in->desc.bEndpointAddress
 177                                         & USB_ENDPOINT_NUMBER_MASK);
 178         }
 179
 180         if (iso_out) {
 181                 dev->iso_out = &iso_out->desc;
 182                 dev->out_iso_pipe = usb_sndisocpipe(udev,
 183                                 iso_out->desc.bEndpointAddress
 184                                         & USB_ENDPOINT_NUMBER_MASK);
 185         }
 186         return 0;
 187 }
 188
 189 /*-------------------------------------------------------------------------*/
 190
 191 /* Support for testing basic non-queued I/O streams.
 192  *
 193  * These just package urbs as requests that can be easily canceled.
 194  * Each urb's data buffer is dynamically allocated; callers can fill
 195  * them with non-zero test data (or test for it) when appropriate.
 196  */
 197
 198 static void simple_callback(struct urb *urb)
 199 {
 200         complete(urb->context);
 201 }
 202
 203 static struct urb *usbtest_alloc_urb(
 204         struct usb_device       *udev,
 205         int                     pipe,
 206         unsigned long           bytes,
 207         unsigned                transfer_flags,
 208         unsigned                offset)
 209 {
 210         struct urb              *urb;
 211
 212         urb = usb_alloc_urb(0, GFP_KERNEL);
 213         if (!urb)
 214                 return urb;
 215         usb_fill_bulk_urb(urb, udev, pipe, NULL, bytes, simple_callback, NULL);
 216         urb->interval = (udev->speed == USB_SPEED_HIGH)
 217                         ? (INTERRUPT_RATE << 3)
 218                         : INTERRUPT_RATE;
 219         urb->transfer_flags = transfer_flags;
 220         if (usb_pipein(pipe))
 221                 urb->transfer_flags |= URB_SHORT_NOT_OK;
 222
 223         if (urb->transfer_flags & URB_NO_TRANSFER_DMA_MAP)
 224                 urb->transfer_buffer = usb_alloc_coherent(udev, bytes + offset,
 225                         GFP_KERNEL, &urb->transfer_dma);
 226         else
 227                 urb->transfer_buffer = kmalloc(bytes + offset, GFP_KERNEL);
 228
 229         if (!urb->transfer_buffer) {
 230                 usb_free_urb(urb);
 231                 return NULL;
 232         }
 233
 234         /* To test unaligned transfers add an offset and fill the
 235                 unused memory with a guard value */
 236         if (offset) {
 237                 memset(urb->transfer_buffer, GUARD_BYTE, offset);
 238                 urb->transfer_buffer += offset;
 239                 if (urb->transfer_flags & URB_NO_TRANSFER_DMA_MAP)
 240                         urb->transfer_dma += offset;
 241         }
 242
 243         /* For inbound transfers use guard byte so that test fails if
 244                 data not correctly copied */
 245         memset(urb->transfer_buffer,
 246                         usb_pipein(urb->pipe) ? GUARD_BYTE : 0,
 247                         bytes);
 248         return urb;
 249 }
 250
 251 static struct urb *simple_alloc_urb(
 252         struct usb_device       *udev,
 253         int                     pipe,
 254         unsigned long           bytes)
 255 {
 256         return usbtest_alloc_urb(udev, pipe, bytes, URB_NO_TRANSFER_DMA_MAP, 0);
 257 }
 258
 259 static unsigned pattern;
 260 static unsigned mod_pattern;
 261 module_param_named(pattern, mod_pattern, uint, S_IRUGO | S_IWUSR);
 262 MODULE_PARM_DESC(mod_pattern, "i/o pattern (0 == zeroes)");
 263
 264 static inline void simple_fill_buf(struct urb *urb)
 265 {
 266         unsigned        i;
 267         u8              *buf = urb->transfer_buffer;
 268         unsigned        len = urb->transfer_buffer_length;
 269
 270         switch (pattern) {
 271         default:
 272                 /* FALLTHROUGH */
 273         case 0:
 274                 memset(buf, 0, len);
 275                 break;
 276         case 1:                 /* mod63 */
 277                 for (i = 0; i < len; i++)
 278                         *buf++ = (u8) (i % 63);
 279                 break;
 280         }
 281 }
 282
 283 static inline unsigned long buffer_offset(void *buf)
 284 {
 285         return (unsigned long)buf & (ARCH_KMALLOC_MINALIGN - 1);
 286 }
 287
 288 static int check_guard_bytes(struct usbtest_dev *tdev, struct urb *urb)
 289 {
 290         u8 *buf = urb->transfer_buffer;
 291         u8 *guard = buf - buffer_offset(buf);
 292         unsigned i;
 293
 294         for (i = 0; guard < buf; i++, guard++) {
 295                 if (*guard != GUARD_BYTE) {
 296                         ERROR(tdev, "guard byte[%d] %d (not %d)\n",
 297                                 i, *guard, GUARD_BYTE);
 298                         return -EINVAL;
 299                 }
 300         }
 301         return 0;
 302 }
 303
 304 static int simple_check_buf(struct usbtest_dev *tdev, struct urb *urb)
 305 {
 306         unsigned        i;
 307         u8              expected;
 308         u8              *buf = urb->transfer_buffer;
 309         unsigned        len = urb->actual_length;
 310
 311         int ret = check_guard_bytes(tdev, urb);
 312         if (ret)
 313                 return ret;
 314
 315         for (i = 0; i < len; i++, buf++) {
 316                 switch (pattern) {
 317                 /* all-zeroes has no synchronization issues */
 318                 case 0:
 319                         expected = 0;
 320                         break;
 321                 /* mod63 stays in sync with short-terminated transfers,
 322                  * or otherwise when host and gadget agree on how large
 323                  * each usb transfer request should be.  resync is done
 324                  * with set_interface or set_config.
 325                  */
 326                 case 1:                 /* mod63 */
 327                         expected = i % 63;
 328                         break;
 329                 /* always fail unsupported patterns */
 330                 default:
 331                         expected = !*buf;
 332                         break;
 333                 }
 334                 if (*buf == expected)
 335                         continue;
 336                 ERROR(tdev, "buf[%d] = %d (not %d)\n", i, *buf, expected);
 337                 return -EINVAL;
 338         }
 339         return 0;
 340 }
 341
 342 static void simple_free_urb(struct urb *urb)
 343 {
 344         unsigned long offset = buffer_offset(urb->transfer_buffer);
 345
 346         if (urb->transfer_flags & URB_NO_TRANSFER_DMA_MAP)
 347                 usb_free_coherent(
 348                         urb->dev,
 349                         urb->transfer_buffer_length + offset,
 350                         urb->transfer_buffer - offset,
 351                         urb->transfer_dma - offset);
 352         else
 353                 kfree(urb->transfer_buffer - offset);
 354         usb_free_urb(urb);
 355 }
 356
 357 static int simple_io(
 358         struct usbtest_dev      *tdev,
 359         struct urb              *urb,
 360         int                     iterations,
 361         int                     vary,
 362         int                     expected,
 363         const char              *label
 364 )
 365 {
 366         struct usb_device       *udev = urb->dev;
 367         int                     max = urb->transfer_buffer_length;
 368         struct completion       completion;
 369         int                     retval = 0;
 370         unsigned long           expire;
 371
 372         urb->context = &completion;
 373         while (retval == 0 && iterations-- > 0) {
 374                 init_completion(&completion);
 375                 if (usb_pipeout(urb->pipe)) {
 376                         simple_fill_buf(urb);
 377                         urb->transfer_flags |= URB_ZERO_PACKET;
 378                 }
 379                 retval = usb_submit_urb(urb, GFP_KERNEL);
 380                 if (retval != 0)
 381                         break;
 382
 383                 expire = msecs_to_jiffies(SIMPLE_IO_TIMEOUT);
 384                 if (!wait_for_completion_timeout(&completion, expire)) {
 385                         usb_kill_urb(urb);
 386                         retval = (urb->status == -ENOENT ?
 387                                   -ETIMEDOUT : urb->status);
 388                 } else {
 389                         retval = urb->status;
 390                 }
 391
 392                 urb->dev = udev;
 393                 if (retval == 0 && usb_pipein(urb->pipe))
 394                         retval = simple_check_buf(tdev, urb);
 395
 396                 if (vary) {
 397                         int     len = urb->transfer_buffer_length;
 398
 399                         len += vary;
 400                         len %= max;
 401                         if (len == 0)
 402                                 len = (vary < max) ? vary : max;
 403                         urb->transfer_buffer_length = len;
 404                 }
 405
 406                 /* FIXME if endpoint halted, clear halt (and log) */
 407         }
 408         urb->transfer_buffer_length = max;
 409
 410         if (expected != retval)
 411                 dev_err(&udev->dev,
 412                         "%s failed, iterations left %d, status %d (not %d)\n",
 413                                 label, iterations, retval, expected);
 414         return retval;
 415 }
 416
 417
 418 /*-------------------------------------------------------------------------*/
 419
 420 /* We use scatterlist primitives to test queued I/O.
 421  * Yes, this also tests the scatterlist primitives.
 422  */
 423
 424 static void free_sglist(struct scatterlist *sg, int nents)
 425 {
 426         unsigned                i;
 427
 428         if (!sg)
 429                 return;
 430         for (i = 0; i < nents; i++) {
 431                 if (!sg_page(&sg[i]))
 432                         continue;
 433                 kfree(sg_virt(&sg[i]));
 434         }
 435         kfree(sg);
 436 }
 437
 438 static struct scatterlist *
 439 alloc_sglist(int nents, int max, int vary)
 440 {
 441         struct scatterlist      *sg;
 442         unsigned                i;
 443         unsigned                size = max;
 444
 445         if (max == 0)
 446                 return NULL;
 447
 448         sg = kmalloc_array(nents, sizeof *sg, GFP_KERNEL);
 449         if (!sg)
 450                 return NULL;
 451         sg_init_table(sg, nents);
 452
 453         for (i = 0; i < nents; i++) {
 454                 char            *buf;
 455                 unsigned        j;
 456
 457                 buf = kzalloc(size, GFP_KERNEL);
 458                 if (!buf) {
 459                         free_sglist(sg, i);
 460                         return NULL;
 461                 }
 462
 463                 /* kmalloc pages are always physically contiguous! */
 464                 sg_set_buf(&sg[i], buf, size);
 465
 466                 switch (pattern) {
 467                 case 0:
 468                         /* already zeroed */
 469                         break;
 470                 case 1:
 471                         for (j = 0; j < size; j++)
 472                                 *buf++ = (u8) (j % 63);
 473                         break;
 474                 }
 475
 476                 if (vary) {
 477                         size += vary;
 478                         size %= max;
 479                         if (size == 0)
 480                                 size = (vary < max) ? vary : max;
 481                 }
 482         }
 483
 484         return sg;
 485 }
 486
 487 static void sg_timeout(unsigned long _req)
 488 {
 489         struct usb_sg_request   *req = (struct usb_sg_request *) _req;
 490
 491         req->status = -ETIMEDOUT;
 492         usb_sg_cancel(req);
 493 }
 494
 495 static int perform_sglist(
 496         struct usbtest_dev      *tdev,
 497         unsigned                iterations,
 498         int                     pipe,
 499         struct usb_sg_request   *req,
 500         struct scatterlist      *sg,
 501         int                     nents
 502 )
 503 {
 504         struct usb_device       *udev = testdev_to_usbdev(tdev);
 505         int                     retval = 0;
 506         struct timer_list       sg_timer;
 507
 508         setup_timer_on_stack(&sg_timer, sg_timeout, (unsigned long) req);
 509
 510         while (retval == 0 && iterations-- > 0) {
 511                 retval = usb_sg_init(req, udev, pipe,
 512                                 (udev->speed == USB_SPEED_HIGH)
 513                                         ? (INTERRUPT_RATE << 3)
 514                                         : INTERRUPT_RATE,
 515                                 sg, nents, 0, GFP_KERNEL);
 516
 517                 if (retval)
 518                         break;
 519                 mod_timer(&sg_timer, jiffies +
 520                                 msecs_to_jiffies(SIMPLE_IO_TIMEOUT));
 521                 usb_sg_wait(req);
 522                 del_timer_sync(&sg_timer);
 523                 retval = req->status;
 524
 525                 /* FIXME check resulting data pattern */
 526
 527                 /* FIXME if endpoint halted, clear halt (and log) */
 528         }
 529
 530         /* FIXME for unlink or fault handling tests, don't report
 531          * failure if retval is as we expected ...
 532          */
 533         if (retval)
 534                 ERROR(tdev, "perform_sglist failed, "
 535                                 "iterations left %d, status %d\n",
 536                                 iterations, retval);
 537         return retval;
 538 }
 539
 540
 541 /*-------------------------------------------------------------------------*/
 542
 543 /* unqueued control message testing
 544  *
 545  * there's a nice set of device functional requirements in chapter 9 of the
 546  * usb 2.0 spec, which we can apply to ANY device, even ones that don't use
 547  * special test firmware.
 548  *
 549  * we know the device is configured (or suspended) by the time it's visible
 550  * through usbfs.  we can't change that, so we won't test enumeration (which
 551  * worked 'well enough' to get here, this time), power management (ditto),
 552  * or remote wakeup (which needs human interaction).
 553  */
 554
 555 static unsigned realworld = 1;
 556 module_param(realworld, uint, 0);
 557 MODULE_PARM_DESC(realworld, "clear to demand stricter spec compliance");
 558
 559 static int get_altsetting(struct usbtest_dev *dev)
 560 {
 561         struct usb_interface    *iface = dev->intf;
 562         struct usb_device       *udev = interface_to_usbdev(iface);
 563         int                     retval;
 564
 565         retval = usb_control_msg(udev, usb_rcvctrlpipe(udev, 0),
 566                         USB_REQ_GET_INTERFACE, USB_DIR_IN|USB_RECIP_INTERFACE,
 567                         0, iface->altsetting[0].desc.bInterfaceNumber,
 568                         dev->buf, 1, USB_CTRL_GET_TIMEOUT);
 569         switch (retval) {
 570         case 1:
 571                 return dev->buf[0];
 572         case 0:
 573                 retval = -ERANGE;
 574                 /* FALLTHROUGH */
 575         default:
 576                 return retval;
 577         }
 578 }
 579
 580 static int set_altsetting(struct usbtest_dev *dev, int alternate)
 581 {
 582         struct usb_interface            *iface = dev->intf;
 583         struct usb_device               *udev;
 584
 585         if (alternate < 0 || alternate >= 256)
 586                 return -EINVAL;
 587
 588         udev = interface_to_usbdev(iface);
 589         return usb_set_interface(udev,
 590                         iface->altsetting[0].desc.bInterfaceNumber,
 591                         alternate);
 592 }
 593
 594 static int is_good_config(struct usbtest_dev *tdev, int len)
 595 {
 596         struct usb_config_descriptor    *config;
 597
 598         if (len < sizeof *config)
 599                 return 0;
 600         config = (struct usb_config_descriptor *) tdev->buf;
 601
 602         switch (config->bDescriptorType) {
 603         case USB_DT_CONFIG:
 604         case USB_DT_OTHER_SPEED_CONFIG:
 605                 if (config->bLength != 9) {
 606                         ERROR(tdev, "bogus config descriptor length\n");
 607                         return 0;
 608                 }
 609                 /* this bit 'must be 1' but often isn't */
 610                 if (!realworld && !(config->bmAttributes & 0x80)) {
 611                         ERROR(tdev, "high bit of config attributes not set\n");
 612                         return 0;
 613                 }
 614                 if (config->bmAttributes & 0x1f) {      /* reserved == 0 */
 615                         ERROR(tdev, "reserved config bits set\n");
 616                         return 0;
 617                 }
 618                 break;
 619         default:
 620                 return 0;
 621         }
 622
 623         if (le16_to_cpu(config->wTotalLength) == len)   /* read it all */
 624                 return 1;
 625         if (le16_to_cpu(config->wTotalLength) >= TBUF_SIZE)     /* max partial read */
 626                 return 1;
 627         ERROR(tdev, "bogus config descriptor read size\n");
 628         return 0;
 629 }
 630
 631 /* sanity test for standard requests working with usb_control_mesg() and some
 632  * of the utility functions which use it.
 633  *
 634  * this doesn't test how endpoint halts behave or data toggles get set, since
 635  * we won't do I/O to bulk/interrupt endpoints here (which is how to change
 636  * halt or toggle).  toggle testing is impractical without support from hcds.
 637  *
 638  * this avoids failing devices linux would normally work with, by not testing
 639  * config/altsetting operations for devices that only support their defaults.
 640  * such devices rarely support those needless operations.
 641  *
 642  * NOTE that since this is a sanity test, it's not examining boundary cases
 643  * to see if usbcore, hcd, and device all behave right.  such testing would
 644  * involve varied read sizes and other operation sequences.
 645  */
 646 static int ch9_postconfig(struct usbtest_dev *dev)
 647 {
 648         struct usb_interface    *iface = dev->intf;
 649         struct usb_device       *udev = interface_to_usbdev(iface);
 650         int                     i, alt, retval;
 651
 652         /* [9.2.3] if there's more than one altsetting, we need to be able to
 653          * set and get each one.  mostly trusts the descriptors from usbcore.
 654          */
 655         for (i = 0; i < iface->num_altsetting; i++) {
 656
 657                 /* 9.2.3 constrains the range here */
 658                 alt = iface->altsetting[i].desc.bAlternateSetting;
 659                 if (alt < 0 || alt >= iface->num_altsetting) {
 660                         dev_err(&iface->dev,
 661                                         "invalid alt [%d].bAltSetting = %d\n",
 662                                         i, alt);
 663                 }
 664
 665                 /* [real world] get/set unimplemented if there's only one */
 666                 if (realworld && iface->num_altsetting == 1)
 667                         continue;
 668
 669                 /* [9.4.10] set_interface */
 670                 retval = set_altsetting(dev, alt);
 671                 if (retval) {
 672                         dev_err(&iface->dev, "can't set_interface = %d, %d\n",
 673                                         alt, retval);
 674                         return retval;
 675                 }
 676
 677                 /* [9.4.4] get_interface always works */
 678                 retval = get_altsetting(dev);
 679                 if (retval != alt) {
 680                         dev_err(&iface->dev, "get alt should be %d, was %d\n",
 681                                         alt, retval);
 682                         return (retval < 0) ? retval : -EDOM;
 683                 }
 684
 685         }
 686
 687         /* [real world] get_config unimplemented if there's only one */
 688         if (!realworld || udev->descriptor.bNumConfigurations != 1) {
 689                 int     expected = udev->actconfig->desc.bConfigurationValue;
 690
 691                 /* [9.4.2] get_configuration always works
 692                  * ... although some cheap devices (like one TI Hub I've got)
 693                  * won't return config descriptors except before set_config.
 694                  */
 695                 retval = usb_control_msg(udev, usb_rcvctrlpipe(udev, 0),
 696                                 USB_REQ_GET_CONFIGURATION,
 697                                 USB_DIR_IN | USB_RECIP_DEVICE,
 698                                 0, 0, dev->buf, 1, USB_CTRL_GET_TIMEOUT);
 699                 if (retval != 1 || dev->buf[0] != expected) {
 700                         dev_err(&iface->dev, "get config --> %d %d (1 %d)\n",
 701                                 retval, dev->buf[0], expected);
 702                         return (retval < 0) ? retval : -EDOM;
 703                 }
 704         }
 705
 706         /* there's always [9.4.3] a device descriptor [9.6.1] */
 707         retval = usb_get_descriptor(udev, USB_DT_DEVICE, 0,
 708                         dev->buf, sizeof udev->descriptor);
 709         if (retval != sizeof udev->descriptor) {
 710                 dev_err(&iface->dev, "dev descriptor --> %d\n", retval);
 711                 return (retval < 0) ? retval : -EDOM;
 712         }
 713
 714         /* there's always [9.4.3] at least one config descriptor [9.6.3] */
 715         for (i = 0; i < udev->descriptor.bNumConfigurations; i++) {
 716                 retval = usb_get_descriptor(udev, USB_DT_CONFIG, i,
 717                                 dev->buf, TBUF_SIZE);
 718                 if (!is_good_config(dev, retval)) {
 719                         dev_err(&iface->dev,
 720                                         "config [%d] descriptor --> %d\n",
 721                                         i, retval);
 722                         return (retval < 0) ? retval : -EDOM;
 723                 }
 724
 725                 /* FIXME cross-checking udev->config[i] to make sure usbcore
 726                  * parsed it right (etc) would be good testing paranoia
 727                  */
 728         }
 729
 730         /* and sometimes [9.2.6.6] speed dependent descriptors */
 731         if (le16_to_cpu(udev->descriptor.bcdUSB) == 0x0200) {
 732                 struct usb_qualifier_descriptor *d = NULL;
 733
 734                 /* device qualifier [9.6.2] */
 735                 retval = usb_get_descriptor(udev,
 736                                 USB_DT_DEVICE_QUALIFIER, 0, dev->buf,
 737                                 sizeof(struct usb_qualifier_descriptor));
 738                 if (retval == -EPIPE) {
 739                         if (udev->speed == USB_SPEED_HIGH) {
 740                                 dev_err(&iface->dev,
 741                                                 "hs dev qualifier --> %d\n",
 742                                                 retval);
 743                                 return (retval < 0) ? retval : -EDOM;
 744                         }
 745                         /* usb2.0 but not high-speed capable; fine */
 746                 } else if (retval != sizeof(struct usb_qualifier_descriptor)) {
 747                         dev_err(&iface->dev, "dev qualifier --> %d\n", retval);
 748                         return (retval < 0) ? retval : -EDOM;
 749                 } else
 750                         d = (struct usb_qualifier_descriptor *) dev->buf;
 751
 752                 /* might not have [9.6.2] any other-speed configs [9.6.4] */
 753                 if (d) {
 754                         unsigned max = d->bNumConfigurations;
 755                         for (i = 0; i < max; i++) {
 756                                 retval = usb_get_descriptor(udev,
 757                                         USB_DT_OTHER_SPEED_CONFIG, i,
 758                                         dev->buf, TBUF_SIZE);
 759                                 if (!is_good_config(dev, retval)) {
 760                                         dev_err(&iface->dev,
 761                                                 "other speed config --> %d\n",
 762                                                 retval);
 763                                         return (retval < 0) ? retval : -EDOM;
 764                                 }
 765                         }
 766                 }
 767         }
 768         /* FIXME fetch strings from at least the device descriptor */
 769
 770         /* [9.4.5] get_status always works */
 771         retval = usb_get_status(udev, USB_RECIP_DEVICE, 0, dev->buf);
 772         if (retval != 2) {
 773                 dev_err(&iface->dev, "get dev status --> %d\n", retval);
 774                 return (retval < 0) ? retval : -EDOM;
 775         }
 776
 777         /* FIXME configuration.bmAttributes says if we could try to set/clear
 778          * the device's remote wakeup feature ... if we can, test that here
 779          */
 780
 781         retval = usb_get_status(udev, USB_RECIP_INTERFACE,
 782                         iface->altsetting[0].desc.bInterfaceNumber, dev->buf);
 783         if (retval != 2) {
 784                 dev_err(&iface->dev, "get interface status --> %d\n", retval);
 785                 return (retval < 0) ? retval : -EDOM;
 786         }
 787         /* FIXME get status for each endpoint in the interface */
 788
 789         return 0;
 790 }
 791
 792 /*-------------------------------------------------------------------------*/
 793
 794 /* use ch9 requests to test whether:
 795  *   (a) queues work for control, keeping N subtests queued and
 796  *       active (auto-resubmit) for M loops through the queue.
 797  *   (b) protocol stalls (control-only) will autorecover.
 798  *       it's not like bulk/intr; no halt clearing.
 799  *   (c) short control reads are reported and handled.
 800  *   (d) queues are always processed in-order
 801  */
 802
 803 struct ctrl_ctx {
 804         spinlock_t              lock;
 805         struct usbtest_dev      *dev;
 806         struct completion       complete;
 807         unsigned                count;
 808         unsigned                pending;
 809         int                     status;
 810         struct urb              **urb;
 811         struct usbtest_param    *param;
 812         int                     last;
 813 };
 814
 815 #define NUM_SUBCASES    15              /* how many test subcases here? */
 816
 817 struct subcase {
 818         struct usb_ctrlrequest  setup;
 819         int                     number;
 820         int                     expected;
 821 };
 822
 823 static void ctrl_complete(struct urb *urb)
 824 {
 825         struct ctrl_ctx         *ctx = urb->context;
 826         struct usb_ctrlrequest  *reqp;
 827         struct subcase          *subcase;
 828         int                     status = urb->status;
 829
 830         reqp = (struct usb_ctrlrequest *)urb->setup_packet;
 831         subcase = container_of(reqp, struct subcase, setup);
 832
 833         spin_lock(&ctx->lock);
 834         ctx->count--;
 835         ctx->pending--;
 836
 837         /* queue must transfer and complete in fifo order, unless
 838          * usb_unlink_urb() is used to unlink something not at the
 839          * physical queue head (not tested).
 840          */
 841         if (subcase->number > 0) {
 842                 if ((subcase->number - ctx->last) != 1) {
 843                         ERROR(ctx->dev,
 844                                 "subcase %d completed out of order, last %d\n",
 845                                 subcase->number, ctx->last);
 846                         status = -EDOM;
 847                         ctx->last = subcase->number;
 848                         goto error;
 849                 }
 850         }
 851         ctx->last = subcase->number;
 852
 853         /* succeed or fault in only one way? */
 854         if (status == subcase->expected)
 855                 status = 0;
 856
 857         /* async unlink for cleanup? */
 858         else if (status != -ECONNRESET) {
 859
 860                 /* some faults are allowed, not required */
 861                 if (subcase->expected > 0 && (
 862                           ((status == -subcase->expected        /* happened */
 863                            || status == 0))))                   /* didn't */
 864                         status = 0;
 865                 /* sometimes more than one fault is allowed */
 866                 else if (subcase->number == 12 && status == -EPIPE)
 867                         status = 0;
 868                 else
 869                         ERROR(ctx->dev, "subtest %d error, status %d\n",
 870                                         subcase->number, status);
 871         }
 872
 873         /* unexpected status codes mean errors; ideally, in hardware */
 874         if (status) {
 875 error:
 876                 if (ctx->status == 0) {
 877                         int             i;
 878
 879                         ctx->status = status;
 880                         ERROR(ctx->dev, "control queue %02x.%02x, err %d, "
 881                                         "%d left, subcase %d, len %d/%d\n",
 882                                         reqp->bRequestType, reqp->bRequest,
 883                                         status, ctx->count, subcase->number,
 884                                         urb->actual_length,
 885                                         urb->transfer_buffer_length);
 886
 887                         /* FIXME this "unlink everything" exit route should
 888                          * be a separate test case.
 889                          */
 890
 891                         /* unlink whatever's still pending */
 892                         for (i = 1; i < ctx->param->sglen; i++) {
 893                                 struct urb *u = ctx->urb[
 894                                                         (i + subcase->number)
 895                                                         % ctx->param->sglen];
 896
 897                                 if (u == urb || !u->dev)
 898                                         continue;
 899                                 spin_unlock(&ctx->lock);
 900                                 status = usb_unlink_urb(u);
 901                                 spin_lock(&ctx->lock);
 902                                 switch (status) {
 903                                 case -EINPROGRESS:
 904                                 case -EBUSY:
 905                                 case -EIDRM:
 906                                         continue;
 907                                 default:
 908                                         ERROR(ctx->dev, "urb unlink --> %d\n",
 909                                                         status);
 910                                 }
 911                         }
 912                         status = ctx->status;
 913                 }
 914         }
 915
 916         /* resubmit if we need to, else mark this as done */
 917         if ((status == 0) && (ctx->pending < ctx->count)) {
 918                 status = usb_submit_urb(urb, GFP_ATOMIC);
 919                 if (status != 0) {
 920                         ERROR(ctx->dev,
 921                                 "can't resubmit ctrl %02x.%02x, err %d\n",
 922                                 reqp->bRequestType, reqp->bRequest, status);
 923                         urb->dev = NULL;
 924                 } else
 925                         ctx->pending++;
 926         } else
 927                 urb->dev = NULL;
 928
 929         /* signal completion when nothing's queued */
 930         if (ctx->pending == 0)
 931                 complete(&ctx->complete);
 932         spin_unlock(&ctx->lock);
 933 }
 934
 935 static int
 936 test_ctrl_queue(struct usbtest_dev *dev, struct usbtest_param *param)
 937 {
 938         struct usb_device       *udev = testdev_to_usbdev(dev);
 939         struct urb              **urb;
 940         struct ctrl_ctx         context;
 941         int                     i;
 942
 943         if (param->sglen == 0 || param->iterations > UINT_MAX / param->sglen)
 944                 return -EOPNOTSUPP;
 945
 946         spin_lock_init(&context.lock);
 947         context.dev = dev;
 948         init_completion(&context.complete);
 949         context.count = param->sglen * param->iterations;
 950         context.pending = 0;
 951         context.status = -ENOMEM;
 952         context.param = param;
 953         context.last = -1;
 954
 955         /* allocate and init the urbs we'll queue.
 956          * as with bulk/intr sglists, sglen is the queue depth; it also
 957          * controls which subtests run (more tests than sglen) or rerun.
 958          */
 959         urb = kcalloc(param->sglen, sizeof(struct urb *), GFP_KERNEL);
 960         if (!urb)
 961                 return -ENOMEM;
 962         for (i = 0; i < param->sglen; i++) {
 963                 int                     pipe = usb_rcvctrlpipe(udev, 0);
 964                 unsigned                len;
 965                 struct urb              *u;
 966                 struct usb_ctrlrequest  req;
 967                 struct subcase          *reqp;
 968
 969                 /* sign of this variable means:
 970                  *  -: tested code must return this (negative) error code
 971                  *  +: tested code may return this (negative too) error code
 972                  */
 973                 int                     expected = 0;
 974
 975                 /* requests here are mostly expected to succeed on any
 976                  * device, but some are chosen to trigger protocol stalls
 977                  * or short reads.
 978                  */
 979                 memset(&req, 0, sizeof req);
 980                 req.bRequest = USB_REQ_GET_DESCRIPTOR;
 981                 req.bRequestType = USB_DIR_IN|USB_RECIP_DEVICE;
 982
 983                 switch (i % NUM_SUBCASES) {
 984                 case 0:         /* get device descriptor */
 985                         req.wValue = cpu_to_le16(USB_DT_DEVICE << 8);
 986                         len = sizeof(struct usb_device_descriptor);
 987                         break;
 988                 case 1:         /* get first config descriptor (only) */
 989                         req.wValue = cpu_to_le16((USB_DT_CONFIG << 8) | 0);
 990                         len = sizeof(struct usb_config_descriptor);
 991                         break;
 992                 case 2:         /* get altsetting (OFTEN STALLS) */
 993                         req.bRequest = USB_REQ_GET_INTERFACE;
 994                         req.bRequestType = USB_DIR_IN|USB_RECIP_INTERFACE;
 995                         /* index = 0 means first interface */
 996                         len = 1;
 997                         expected = EPIPE;
 998                         break;
 999                 case 3:         /* get interface status */
1000                         req.bRequest = USB_REQ_GET_STATUS;
1001                         req.bRequestType = USB_DIR_IN|USB_RECIP_INTERFACE;
1002                         /* interface 0 */
1003                         len = 2;
1004                         break;
1005                 case 4:         /* get device status */
1006                         req.bRequest = USB_REQ_GET_STATUS;
1007                         req.bRequestType = USB_DIR_IN|USB_RECIP_DEVICE;
1008                         len = 2;
1009                         break;
1010                 case 5:         /* get device qualifier (MAY STALL) */
1011                         req.wValue = cpu_to_le16 (USB_DT_DEVICE_QUALIFIER << 8);
1012                         len = sizeof(struct usb_qualifier_descriptor);
1013                         if (udev->speed != USB_SPEED_HIGH)
1014                                 expected = EPIPE;
1015                         break;
1016                 case 6:         /* get first config descriptor, plus interface */
1017                         req.wValue = cpu_to_le16((USB_DT_CONFIG << 8) | 0);
1018                         len = sizeof(struct usb_config_descriptor);
1019                         len += sizeof(struct usb_interface_descriptor);
1020                         break;
1021                 case 7:         /* get interface descriptor (ALWAYS STALLS) */
1022                         req.wValue = cpu_to_le16 (USB_DT_INTERFACE << 8);
1023                         /* interface == 0 */
1024                         len = sizeof(struct usb_interface_descriptor);
1025                         expected = -EPIPE;
1026                         break;
1027                 /* NOTE: two consecutive stalls in the queue here.
1028                  *  that tests fault recovery a bit more aggressively. */
1029                 case 8:         /* clear endpoint halt (MAY STALL) */
1030                         req.bRequest = USB_REQ_CLEAR_FEATURE;
1031                         req.bRequestType = USB_RECIP_ENDPOINT;
1032                         /* wValue 0 == ep halt */
1033                         /* wIndex 0 == ep0 (shouldn't halt!) */
1034                         len = 0;
1035                         pipe = usb_sndctrlpipe(udev, 0);
1036                         expected = EPIPE;
1037                         break;
1038                 case 9:         /* get endpoint status */
1039                         req.bRequest = USB_REQ_GET_STATUS;
1040                         req.bRequestType = USB_DIR_IN|USB_RECIP_ENDPOINT;
1041                         /* endpoint 0 */
1042                         len = 2;
1043                         break;
1044                 case 10:        /* trigger short read (EREMOTEIO) */
1045                         req.wValue = cpu_to_le16((USB_DT_CONFIG << 8) | 0);
1046                         len = 1024;
1047                         expected = -EREMOTEIO;
1048                         break;
1049                 /* NOTE: two consecutive _different_ faults in the queue. */
1050                 case 11:        /* get endpoint descriptor (ALWAYS STALLS) */
1051                         req.wValue = cpu_to_le16(USB_DT_ENDPOINT << 8);
1052                         /* endpoint == 0 */
1053                         len = sizeof(struct usb_interface_descriptor);
1054                         expected = EPIPE;
1055                         break;
1056                 /* NOTE: sometimes even a third fault in the queue! */
1057                 case 12:        /* get string 0 descriptor (MAY STALL) */
1058                         req.wValue = cpu_to_le16(USB_DT_STRING << 8);
1059                         /* string == 0, for language IDs */
1060                         len = sizeof(struct usb_interface_descriptor);
1061                         /* may succeed when > 4 languages */
1062                         expected = EREMOTEIO;   /* or EPIPE, if no strings */
1063                         break;
1064                 case 13:        /* short read, resembling case 10 */
1065                         req.wValue = cpu_to_le16((USB_DT_CONFIG << 8) | 0);
1066                         /* last data packet "should" be DATA1, not DATA0 */
1067                         if (udev->speed == USB_SPEED_SUPER)
1068                                 len = 1024 - 512;
1069                         else
1070                                 len = 1024 - udev->descriptor.bMaxPacketSize0;
1071                         expected = -EREMOTEIO;
1072                         break;
1073                 case 14:        /* short read; try to fill the last packet */
1074                         req.wValue = cpu_to_le16((USB_DT_DEVICE << 8) | 0);
1075                         /* device descriptor size == 18 bytes */
1076                         len = udev->descriptor.bMaxPacketSize0;
1077                         if (udev->speed == USB_SPEED_SUPER)
1078                                 len = 512;
1079                         switch (len) {
1080                         case 8:
1081                                 len = 24;
1082                                 break;
1083                         case 16:
1084                                 len = 32;
1085                                 break;
1086                         }
1087                         expected = -EREMOTEIO;
1088                         break;
1089                 default:
1090                         ERROR(dev, "bogus number of ctrl queue testcases!\n");
1091                         context.status = -EINVAL;
1092                         goto cleanup;
1093                 }
1094                 req.wLength = cpu_to_le16(len);
1095                 urb[i] = u = simple_alloc_urb(udev, pipe, len);
1096                 if (!u)
1097                         goto cleanup;
1098
1099                 reqp = kmalloc(sizeof *reqp, GFP_KERNEL);
1100                 if (!reqp)
1101                         goto cleanup;
1102                 reqp->setup = req;
1103                 reqp->number = i % NUM_SUBCASES;
1104                 reqp->expected = expected;
1105                 u->setup_packet = (char *) &reqp->setup;
1106
1107                 u->context = &context;
1108                 u->complete = ctrl_complete;
1109         }
1110
1111         /* queue the urbs */
1112         context.urb = urb;
1113         spin_lock_irq(&context.lock);
1114         for (i = 0; i < param->sglen; i++) {
1115                 context.status = usb_submit_urb(urb[i], GFP_ATOMIC);
1116                 if (context.status != 0) {
1117                         ERROR(dev, "can't submit urb[%d], status %d\n",
1118                                         i, context.status);
1119                         context.count = context.pending;
1120                         break;
1121                 }
1122                 context.pending++;
1123         }
1124         spin_unlock_irq(&context.lock);
1125
1126         /* FIXME  set timer and time out; provide a disconnect hook */
1127
1128         /* wait for the last one to complete */
1129         if (context.pending > 0)
1130                 wait_for_completion(&context.complete);
1131
1132 cleanup:
1133         for (i = 0; i < param->sglen; i++) {
1134                 if (!urb[i])
1135                         continue;
1136                 urb[i]->dev = udev;
1137                 kfree(urb[i]->setup_packet);
1138                 simple_free_urb(urb[i]);
1139         }
1140         kfree(urb);
1141         return context.status;
1142 }
1143 #undef NUM_SUBCASES
1144
1145
1146 /*-------------------------------------------------------------------------*/
1147
1148 static void unlink1_callback(struct urb *urb)
1149 {
1150         int     status = urb->status;
1151
1152         /* we "know" -EPIPE (stall) never happens */
1153         if (!status)
1154                 status = usb_submit_urb(urb, GFP_ATOMIC);
1155         if (status) {
1156                 urb->status = status;
1157                 complete(urb->context);
1158         }
1159 }
1160
1161 static int unlink1(struct usbtest_dev *dev, int pipe, int size, int async)
1162 {
1163         struct urb              *urb;
1164         struct completion       completion;
1165         int                     retval = 0;
1166
1167         init_completion(&completion);
1168         urb = simple_alloc_urb(testdev_to_usbdev(dev), pipe, size);
1169         if (!urb)
1170                 return -ENOMEM;
1171         urb->context = &completion;
1172         urb->complete = unlink1_callback;
1173
1174         if (usb_pipeout(urb->pipe)) {
1175                 simple_fill_buf(urb);
1176                 urb->transfer_flags |= URB_ZERO_PACKET;
1177         }
1178
1179         /* keep the endpoint busy.  there are lots of hc/hcd-internal
1180          * states, and testing should get to all of them over time.
1181          *
1182          * FIXME want additional tests for when endpoint is STALLing
1183          * due to errors, or is just NAKing requests.
1184          */
1185         retval = usb_submit_urb(urb, GFP_KERNEL);
1186         if (retval != 0) {
1187                 dev_err(&dev->intf->dev, "submit fail %d\n", retval);
1188                 return retval;
1189         }
1190
1191         /* unlinking that should always work.  variable delay tests more
1192          * hcd states and code paths, even with little other system load.
1193          */
1194         msleep(jiffies % (2 * INTERRUPT_RATE));
1195         if (async) {
1196                 while (!completion_done(&completion)) {
1197                         retval = usb_unlink_urb(urb);
1198
1199                         switch (retval) {
1200                         case -EBUSY:
1201                         case -EIDRM:
1202                                 /* we can't unlink urbs while they're completing
1203                                  * or if they've completed, and we haven't
1204                                  * resubmitted. "normal" drivers would prevent
1205                                  * resubmission, but since we're testing unlink
1206                                  * paths, we can't.
1207                                  */
1208                                 ERROR(dev, "unlink retry\n");
1209                                 continue;
1210                         case 0:
1211                         case -EINPROGRESS:
1212                                 break;
1213
1214                         default:
1215                                 dev_err(&dev->intf->dev,
1216                                         "unlink fail %d\n", retval);
1217                                 return retval;
1218                         }
1219
1220                         break;
1221                 }
1222         } else
1223                 usb_kill_urb(urb);
1224
1225         wait_for_completion(&completion);
1226         retval = urb->status;
1227         simple_free_urb(urb);
1228
1229         if (async)
1230                 return (retval == -ECONNRESET) ? 0 : retval - 1000;
1231         else
1232                 return (retval == -ENOENT || retval == -EPERM) ?
1233                                 0 : retval - 2000;
1234 }
1235
1236 static int unlink_simple(struct usbtest_dev *dev, int pipe, int len)
1237 {
1238         int                     retval = 0;
1239
1240         /* test sync and async paths */
1241         retval = unlink1(dev, pipe, len, 1);
1242         if (!retval)
1243                 retval = unlink1(dev, pipe, len, 0);
1244         return retval;
1245 }
1246
1247 /*-------------------------------------------------------------------------*/
1248
1249 struct queued_ctx {
1250         struct completion       complete;
1251         atomic_t                pending;
1252         unsigned                num;
1253         int                     status;
1254         struct urb              **urbs;
1255 };
1256
1257 static void unlink_queued_callback(struct urb *urb)
1258 {
1259         int                     status = urb->status;
1260         struct queued_ctx       *ctx = urb->context;
1261
1262         if (ctx->status)
1263                 goto done;
1264         if (urb == ctx->urbs[ctx->num - 4] || urb == ctx->urbs[ctx->num - 2]) {
1265                 if (status == -ECONNRESET)
1266                         goto done;
1267                 /* What error should we report if the URB completed normally? */
1268         }
1269         if (status != 0)
1270                 ctx->status = status;
1271
1272  done:
1273         if (atomic_dec_and_test(&ctx->pending))
1274                 complete(&ctx->complete);
1275 }
1276
1277 static int unlink_queued(struct usbtest_dev *dev, int pipe, unsigned num,
1278                 unsigned size)
1279 {
1280         struct queued_ctx       ctx;
1281         struct usb_device       *udev = testdev_to_usbdev(dev);
1282         void                    *buf;
1283         dma_addr_t              buf_dma;
1284         int                     i;
1285         int                     retval = -ENOMEM;
1286
1287         init_completion(&ctx.complete);
1288         atomic_set(&ctx.pending, 1);    /* One more than the actual value */
1289         ctx.num = num;
1290         ctx.status = 0;
1291
1292         buf = usb_alloc_coherent(udev, size, GFP_KERNEL, &buf_dma);
1293         if (!buf)
1294                 return retval;
1295         memset(buf, 0, size);
1296
1297         /* Allocate and init the urbs we'll queue */
1298         ctx.urbs = kcalloc(num, sizeof(struct urb *), GFP_KERNEL);
1299         if (!ctx.urbs)
1300                 goto free_buf;
1301         for (i = 0; i < num; i++) {
1302                 ctx.urbs[i] = usb_alloc_urb(0, GFP_KERNEL);
1303                 if (!ctx.urbs[i])
1304                         goto free_urbs;
1305                 usb_fill_bulk_urb(ctx.urbs[i], udev, pipe, buf, size,
1306                                 unlink_queued_callback, &ctx);
1307                 ctx.urbs[i]->transfer_dma = buf_dma;
1308                 ctx.urbs[i]->transfer_flags = URB_NO_TRANSFER_DMA_MAP;
1309
1310                 if (usb_pipeout(ctx.urbs[i]->pipe)) {
1311                         simple_fill_buf(ctx.urbs[i]);
1312                         ctx.urbs[i]->transfer_flags |= URB_ZERO_PACKET;
1313                 }
1314         }
1315
1316         /* Submit all the URBs and then unlink URBs num - 4 and num - 2. */
1317         for (i = 0; i < num; i++) {
1318                 atomic_inc(&ctx.pending);
1319                 retval = usb_submit_urb(ctx.urbs[i], GFP_KERNEL);
1320                 if (retval != 0) {
1321                         dev_err(&dev->intf->dev, "submit urbs[%d] fail %d\n",
1322                                         i, retval);
1323                         atomic_dec(&ctx.pending);
1324                         ctx.status = retval;
1325                         break;
1326                 }
1327         }
1328         if (i == num) {
1329                 usb_unlink_urb(ctx.urbs[num - 4]);
1330                 usb_unlink_urb(ctx.urbs[num - 2]);
1331         } else {
1332                 while (--i >= 0)
1333                         usb_unlink_urb(ctx.urbs[i]);
1334         }
1335
1336         if (atomic_dec_and_test(&ctx.pending))          /* The extra count */
1337                 complete(&ctx.complete);
1338         wait_for_completion(&ctx.complete);
1339         retval = ctx.status;
1340
1341  free_urbs:
1342         for (i = 0; i < num; i++)
1343                 usb_free_urb(ctx.urbs[i]);
1344         kfree(ctx.urbs);
1345  free_buf:
1346         usb_free_coherent(udev, size, buf, buf_dma);
1347         return retval;
1348 }
1349
1350 /*-------------------------------------------------------------------------*/
1351
1352 static int verify_not_halted(struct usbtest_dev *tdev, int ep, struct urb *urb)
1353 {
1354         int     retval;
1355         u16     status;
1356
1357         /* shouldn't look or act halted */
1358         retval = usb_get_status(urb->dev, USB_RECIP_ENDPOINT, ep, &status);
1359         if (retval < 0) {
1360                 ERROR(tdev, "ep %02x couldn't get no-halt status, %d\n",
1361                                 ep, retval);
1362                 return retval;
1363         }
1364         if (status != 0) {
1365                 ERROR(tdev, "ep %02x bogus status: %04x != 0\n", ep, status);
1366                 return -EINVAL;
1367         }
1368         retval = simple_io(tdev, urb, 1, 0, 0, __func__);
1369         if (retval != 0)
1370                 return -EINVAL;
1371         return 0;
1372 }
1373
1374 static int verify_halted(struct usbtest_dev *tdev, int ep, struct urb *urb)
1375 {
1376         int     retval;
1377         u16     status;
1378
1379         /* should look and act halted */
1380         retval = usb_get_status(urb->dev, USB_RECIP_ENDPOINT, ep, &status);
1381         if (retval < 0) {
1382                 ERROR(tdev, "ep %02x couldn't get halt status, %d\n",
1383                                 ep, retval);
1384                 return retval;
1385         }
1386         le16_to_cpus(&status);
1387         if (status != 1) {
1388                 ERROR(tdev, "ep %02x bogus status: %04x != 1\n", ep, status);
1389                 return -EINVAL;
1390         }
1391         retval = simple_io(tdev, urb, 1, 0, -EPIPE, __func__);
1392         if (retval != -EPIPE)
1393                 return -EINVAL;
1394         retval = simple_io(tdev, urb, 1, 0, -EPIPE, "verify_still_halted");
1395         if (retval != -EPIPE)
1396                 return -EINVAL;
1397         return 0;
1398 }
1399
1400 static int test_halt(struct usbtest_dev *tdev, int ep, struct urb *urb)
1401 {
1402         int     retval;
1403
1404         /* shouldn't look or act halted now */
1405         retval = verify_not_halted(tdev, ep, urb);
1406         if (retval < 0)
1407                 return retval;
1408
1409         /* set halt (protocol test only), verify it worked */
1410         retval = usb_control_msg(urb->dev, usb_sndctrlpipe(urb->dev, 0),
1411                         USB_REQ_SET_FEATURE, USB_RECIP_ENDPOINT,
1412                         USB_ENDPOINT_HALT, ep,
1413                         NULL, 0, USB_CTRL_SET_TIMEOUT);
1414         if (retval < 0) {
1415                 ERROR(tdev, "ep %02x couldn't set halt, %d\n", ep, retval);
1416                 return retval;
1417         }
1418         retval = verify_halted(tdev, ep, urb);
1419         if (retval < 0)
1420                 return retval;
1421
1422         /* clear halt (tests API + protocol), verify it worked */
1423         retval = usb_clear_halt(urb->dev, urb->pipe);
1424         if (retval < 0) {
1425                 ERROR(tdev, "ep %02x couldn't clear halt, %d\n", ep, retval);
1426                 return retval;
1427         }
1428         retval = verify_not_halted(tdev, ep, urb);
1429         if (retval < 0)
1430                 return retval;
1431
1432         /* NOTE:  could also verify SET_INTERFACE clear halts ... */
1433
1434         return 0;
1435 }
1436
1437 static int halt_simple(struct usbtest_dev *dev)
1438 {
1439         int                     ep;
1440         int                     retval = 0;
1441         struct urb              *urb;
1442         struct usb_device       *udev = testdev_to_usbdev(dev);
1443
1444         if (udev->speed == USB_SPEED_SUPER)
1445                 urb = simple_alloc_urb(udev, 0, 1024);
1446         else
1447                 urb = simple_alloc_urb(udev, 0, 512);
1448         if (urb == NULL)
1449                 return -ENOMEM;
1450
1451         if (dev->in_pipe) {
1452                 ep = usb_pipeendpoint(dev->in_pipe) | USB_DIR_IN;
1453                 urb->pipe = dev->in_pipe;
1454                 retval = test_halt(dev, ep, urb);
1455                 if (retval < 0)
1456                         goto done;
1457         }
1458
1459         if (dev->out_pipe) {
1460                 ep = usb_pipeendpoint(dev->out_pipe);
1461                 urb->pipe = dev->out_pipe;
1462                 retval = test_halt(dev, ep, urb);
1463         }
1464 done:
1465         simple_free_urb(urb);
1466         return retval;
1467 }
1468
1469 /*-------------------------------------------------------------------------*/
1470
1471 /* Control OUT tests use the vendor control requests from Intel's
1472  * USB 2.0 compliance test device:  write a buffer, read it back.
1473  *
1474  * Intel's spec only _requires_ that it work for one packet, which
1475  * is pretty weak.   Some HCDs place limits here; most devices will
1476  * need to be able to handle more than one OUT data packet.  We'll
1477  * try whatever we're told to try.
1478  */
1479 static int ctrl_out(struct usbtest_dev *dev,
1480                 unsigned count, unsigned length, unsigned vary, unsigned offset)
1481 {
1482         unsigned                i, j, len;
1483         int                     retval;
1484         u8                      *buf;
1485         char                    *what = "?";
1486         struct usb_device       *udev;
1487
1488         if (length < 1 || length > 0xffff || vary >= length)
1489                 return -EINVAL;
1490
1491         buf = kmalloc(length + offset, GFP_KERNEL);
1492         if (!buf)
1493                 return -ENOMEM;
1494
1495         buf += offset;
1496         udev = testdev_to_usbdev(dev);
1497         len = length;
1498         retval = 0;
1499
1500         /* NOTE:  hardware might well act differently if we pushed it
1501          * with lots back-to-back queued requests.
1502          */
1503         for (i = 0; i < count; i++) {
1504                 /* write patterned data */
1505                 for (j = 0; j < len; j++)
1506                         buf[j] = i + j;
1507                 retval = usb_control_msg(udev, usb_sndctrlpipe(udev, 0),
1508                                 0x5b, USB_DIR_OUT|USB_TYPE_VENDOR,
1509                                 0, 0, buf, len, USB_CTRL_SET_TIMEOUT);
1510                 if (retval != len) {
1511                         what = "write";
1512                         if (retval >= 0) {
1513                                 ERROR(dev, "ctrl_out, wlen %d (expected %d)\n",
1514                                                 retval, len);
1515                                 retval = -EBADMSG;
1516                         }
1517                         break;
1518                 }
1519
1520                 /* read it back -- assuming nothing intervened!!  */
1521                 retval = usb_control_msg(udev, usb_rcvctrlpipe(udev, 0),
1522                                 0x5c, USB_DIR_IN|USB_TYPE_VENDOR,
1523                                 0, 0, buf, len, USB_CTRL_GET_TIMEOUT);
1524                 if (retval != len) {
1525                         what = "read";
1526                         if (retval >= 0) {
1527                                 ERROR(dev, "ctrl_out, rlen %d (expected %d)\n",
1528                                                 retval, len);
1529                                 retval = -EBADMSG;
1530                         }
1531                         break;
1532                 }
1533
1534                 /* fail if we can't verify */
1535                 for (j = 0; j < len; j++) {
1536                         if (buf[j] != (u8) (i + j)) {
1537                                 ERROR(dev, "ctrl_out, byte %d is %d not %d\n",
1538                                         j, buf[j], (u8) i + j);
1539                                 retval = -EBADMSG;
1540                                 break;
1541                         }
1542                 }
1543                 if (retval < 0) {
1544                         what = "verify";
1545                         break;
1546                 }
1547
1548                 len += vary;
1549
1550                 /* [real world] the "zero bytes IN" case isn't really used.
1551                  * hardware can easily trip up in this weird case, since its
1552                  * status stage is IN, not OUT like other ep0in transfers.
1553                  */
1554                 if (len > length)
1555                         len = realworld ? 1 : 0;
1556         }
1557
1558         if (retval < 0)
1559                 ERROR(dev, "ctrl_out %s failed, code %d, count %d\n",
1560                         what, retval, i);
1561
1562         kfree(buf - offset);
1563         return retval;
1564 }
1565
1566 /*-------------------------------------------------------------------------*/
1567
1568 /* ISO tests ... mimics common usage
1569  *  - buffer length is split into N packets (mostly maxpacket sized)
1570  *  - multi-buffers according to sglen
1571  */
1572
1573 struct iso_context {
1574         unsigned                count;
1575         unsigned                pending;
1576         spinlock_t              lock;
1577         struct completion       done;
1578         int                     submit_error;
1579         unsigned long           errors;
1580         unsigned long           packet_count;
1581         struct usbtest_dev      *dev;
1582 };
1583
1584 static void iso_callback(struct urb *urb)
1585 {
1586         struct iso_context      *ctx = urb->context;
1587
1588         spin_lock(&ctx->lock);
1589         ctx->count--;
1590
1591         ctx->packet_count += urb->number_of_packets;
1592         if (urb->error_count > 0)
1593                 ctx->errors += urb->error_count;
1594         else if (urb->status != 0)
1595                 ctx->errors += urb->number_of_packets;
1596         else if (urb->actual_length != urb->transfer_buffer_length)
1597                 ctx->errors++;
1598         else if (check_guard_bytes(ctx->dev, urb) != 0)
1599                 ctx->errors++;
1600
1601         if (urb->status == 0 && ctx->count > (ctx->pending - 1)
1602                         && !ctx->submit_error) {
1603                 int status = usb_submit_urb(urb, GFP_ATOMIC);
1604                 switch (status) {
1605                 case 0:
1606                         goto done;
1607                 default:
1608                         dev_err(&ctx->dev->intf->dev,
1609                                         "iso resubmit err %d\n",
1610                                         status);
1611                         /* FALLTHROUGH */
1612                 case -ENODEV:                   /* disconnected */
1613                 case -ESHUTDOWN:                /* endpoint disabled */
1614                         ctx->submit_error = 1;
1615                         break;
1616                 }
1617         }
1618
1619         ctx->pending--;
1620         if (ctx->pending == 0) {
1621                 if (ctx->errors)
1622                         dev_err(&ctx->dev->intf->dev,
1623                                 "iso test, %lu errors out of %lu\n",
1624                                 ctx->errors, ctx->packet_count);
1625                 complete(&ctx->done);
1626         }
1627 done:
1628         spin_unlock(&ctx->lock);
1629 }
1630
1631 static struct urb *iso_alloc_urb(
1632         struct usb_device       *udev,
1633         int                     pipe,
1634         struct usb_endpoint_descriptor  *desc,
1635         long                    bytes,
1636         unsigned offset
1637 )
1638 {
1639         struct urb              *urb;
1640         unsigned                i, maxp, packets;
1641
1642         if (bytes < 0 || !desc)
1643                 return NULL;
1644         maxp = 0x7ff & usb_endpoint_maxp(desc);
1645         maxp *= 1 + (0x3 & (usb_endpoint_maxp(desc) >> 11));
1646         packets = DIV_ROUND_UP(bytes, maxp);
1647
1648         urb = usb_alloc_urb(packets, GFP_KERNEL);
1649         if (!urb)
1650                 return urb;
1651         urb->dev = udev;
1652         urb->pipe = pipe;
1653
1654         urb->number_of_packets = packets;
1655         urb->transfer_buffer_length = bytes;
1656         urb->transfer_buffer = usb_alloc_coherent(udev, bytes + offset,
1657                                                         GFP_KERNEL,
1658                                                         &urb->transfer_dma);
1659         if (!urb->transfer_buffer) {
1660                 usb_free_urb(urb);
1661                 return NULL;
1662         }
1663         if (offset) {
1664                 memset(urb->transfer_buffer, GUARD_BYTE, offset);
1665                 urb->transfer_buffer += offset;
1666                 urb->transfer_dma += offset;
1667         }
1668         /* For inbound transfers use guard byte so that test fails if
1669                 data not correctly copied */
1670         memset(urb->transfer_buffer,
1671                         usb_pipein(urb->pipe) ? GUARD_BYTE : 0,
1672                         bytes);
1673
1674         for (i = 0; i < packets; i++) {
1675                 /* here, only the last packet will be short */
1676                 urb->iso_frame_desc[i].length = min((unsigned) bytes, maxp);
1677                 bytes -= urb->iso_frame_desc[i].length;
1678
1679                 urb->iso_frame_desc[i].offset = maxp * i;
1680         }
1681
1682         urb->complete = iso_callback;
1683         /* urb->context = SET BY CALLER */
1684         urb->interval = 1 << (desc->bInterval - 1);
1685         urb->transfer_flags = URB_ISO_ASAP | URB_NO_TRANSFER_DMA_MAP;
1686         return urb;
1687 }
1688
1689 static int
1690 test_iso_queue(struct usbtest_dev *dev, struct usbtest_param *param,
1691                 int pipe, struct usb_endpoint_descriptor *desc, unsigned offset)
1692 {
1693         struct iso_context      context;
1694         struct usb_device       *udev;
1695         unsigned                i;
1696         unsigned long           packets = 0;
1697         int                     status = 0;
1698         struct urb              *urbs[10];      /* FIXME no limit */
1699
1700         if (param->sglen > 10)
1701                 return -EDOM;
1702
1703         memset(&context, 0, sizeof context);
1704         context.count = param->iterations * param->sglen;
1705         context.dev = dev;
1706         init_completion(&context.done);
1707         spin_lock_init(&context.lock);
1708
1709         memset(urbs, 0, sizeof urbs);
1710         udev = testdev_to_usbdev(dev);
1711         dev_info(&dev->intf->dev,
1712                 "... iso period %d %sframes, wMaxPacket %04x\n",
1713                 1 << (desc->bInterval - 1),
1714                 (udev->speed == USB_SPEED_HIGH) ? "micro" : "",
1715                 usb_endpoint_maxp(desc));
1716
1717         for (i = 0; i < param->sglen; i++) {
1718                 urbs[i] = iso_alloc_urb(udev, pipe, desc,
1719                                         param->length, offset);
1720                 if (!urbs[i]) {
1721                         status = -ENOMEM;
1722                         goto fail;
1723                 }
1724                 packets += urbs[i]->number_of_packets;
1725                 urbs[i]->context = &context;
1726         }
1727         packets *= param->iterations;
1728         dev_info(&dev->intf->dev,
1729                 "... total %lu msec (%lu packets)\n",
1730                 (packets * (1 << (desc->bInterval - 1)))
1731                         / ((udev->speed == USB_SPEED_HIGH) ? 8 : 1),
1732                 packets);
1733
1734         spin_lock_irq(&context.lock);
1735         for (i = 0; i < param->sglen; i++) {
1736                 ++context.pending;
1737                 status = usb_submit_urb(urbs[i], GFP_ATOMIC);
1738                 if (status < 0) {
1739                         ERROR(dev, "submit iso[%d], error %d\n", i, status);
1740                         if (i == 0) {
1741                                 spin_unlock_irq(&context.lock);
1742                                 goto fail;
1743                         }
1744
1745                         simple_free_urb(urbs[i]);
1746                         urbs[i] = NULL;
1747                         context.pending--;
1748                         context.submit_error = 1;
1749                         break;
1750                 }
1751         }
1752         spin_unlock_irq(&context.lock);
1753
1754         wait_for_completion(&context.done);
1755
1756         for (i = 0; i < param->sglen; i++) {
1757                 if (urbs[i])
1758                         simple_free_urb(urbs[i]);
1759         }
1760         /*
1761          * Isochronous transfers are expected to fail sometimes.  As an
1762          * arbitrary limit, we will report an error if any submissions
1763          * fail or if the transfer failure rate is > 10%.
1764          */
1765         if (status != 0)
1766                 ;
1767         else if (context.submit_error)
1768                 status = -EACCES;
1769         else if (context.errors > context.packet_count / 10)
1770                 status = -EIO;
1771         return status;
1772
1773 fail:
1774         for (i = 0; i < param->sglen; i++) {
1775                 if (urbs[i])
1776                         simple_free_urb(urbs[i]);
1777         }
1778         return status;
1779 }
1780
1781 static int test_unaligned_bulk(
1782         struct usbtest_dev *tdev,
1783         int pipe,
1784         unsigned length,
1785         int iterations,
1786         unsigned transfer_flags,
1787         const char *label)
1788 {
1789         int retval;
1790         struct urb *urb = usbtest_alloc_urb(
1791                 testdev_to_usbdev(tdev), pipe, length, transfer_flags, 1);
1792
1793         if (!urb)
1794                 return -ENOMEM;
1795
1796         retval = simple_io(tdev, urb, iterations, 0, 0, label);
1797         simple_free_urb(urb);
1798         return retval;
1799 }
1800
1801 /*-------------------------------------------------------------------------*/
1802
1803 /* We only have this one interface to user space, through usbfs.
1804  * User mode code can scan usbfs to find N different devices (maybe on
1805  * different busses) to use when testing, and allocate one thread per
1806  * test.  So discovery is simplified, and we have no device naming issues.
1807  *
1808  * Don't use these only as stress/load tests.  Use them along with with
1809  * other USB bus activity:  plugging, unplugging, mousing, mp3 playback,
1810  * video capture, and so on.  Run different tests at different times, in
1811  * different sequences.  Nothing here should interact with other devices,
1812  * except indirectly by consuming USB bandwidth and CPU resources for test
1813  * threads and request completion.  But the only way to know that for sure
1814  * is to test when HC queues are in use by many devices.
1815  *
1816  * WARNING:  Because usbfs grabs udev->dev.sem before calling this ioctl(),
1817  * it locks out usbcore in certain code paths.  Notably, if you disconnect
1818  * the device-under-test, khubd will wait block forever waiting for the
1819  * ioctl to complete ... so that usb_disconnect() can abort the pending
1820  * urbs and then call usbtest_disconnect().  To abort a test, you're best
1821  * off just killing the userspace task and waiting for it to exit.
1822  */
1823
1824 static int
1825 usbtest_ioctl(struct usb_interface *intf, unsigned int code, void *buf)
1826 {
1827         struct usbtest_dev      *dev = usb_get_intfdata(intf);
1828         struct usb_device       *udev = testdev_to_usbdev(dev);
1829         struct usbtest_param    *param = buf;
1830         int                     retval = -EOPNOTSUPP;
1831         struct urb              *urb;
1832         struct scatterlist      *sg;
1833         struct usb_sg_request   req;
1834         struct timeval          start;
1835         unsigned                i;
1836
1837         /* FIXME USBDEVFS_CONNECTINFO doesn't say how fast the device is. */
1838
1839         pattern = mod_pattern;
1840
1841         if (code != USBTEST_REQUEST)
1842                 return -EOPNOTSUPP;
1843
1844         if (param->iterations <= 0)
1845                 return -EINVAL;
1846
1847         if (mutex_lock_interruptible(&dev->lock))
1848                 return -ERESTARTSYS;
1849
1850         /* FIXME: What if a system sleep starts while a test is running? */
1851
1852         /* some devices, like ez-usb default devices, need a non-default
1853          * altsetting to have any active endpoints.  some tests change
1854          * altsettings; force a default so most tests don't need to check.
1855          */
1856         if (dev->info->alt >= 0) {
1857                 int     res;
1858
1859                 if (intf->altsetting->desc.bInterfaceNumber) {
1860                         mutex_unlock(&dev->lock);
1861                         return -ENODEV;
1862                 }
1863                 res = set_altsetting(dev, dev->info->alt);
1864                 if (res) {
1865                         dev_err(&intf->dev,
1866                                         "set altsetting to %d failed, %d\n",
1867                                         dev->info->alt, res);
1868                         mutex_unlock(&dev->lock);
1869                         return res;
1870                 }
1871         }
1872
1873         /*
1874          * Just a bunch of test cases that every HCD is expected to handle.
1875          *
1876          * Some may need specific firmware, though it'd be good to have
1877          * one firmware image to handle all the test cases.
1878          *
1879          * FIXME add more tests!  cancel requests, verify the data, control
1880          * queueing, concurrent read+write threads, and so on.
1881          */
1882         do_gettimeofday(&start);
1883         switch (param->test_num) {
1884
1885         case 0:
1886                 dev_info(&intf->dev, "TEST 0:  NOP\n");
1887                 retval = 0;
1888                 break;
1889
1890         /* Simple non-queued bulk I/O tests */
1891         case 1:
1892                 if (dev->out_pipe == 0)
1893                         break;
1894                 dev_info(&intf->dev,
1895                                 "TEST 1:  write %d bytes %u times\n",
1896                                 param->length, param->iterations);
1897                 urb = simple_alloc_urb(udev, dev->out_pipe, param->length);
1898                 if (!urb) {
1899                         retval = -ENOMEM;
1900                         break;
1901                 }
1902                 /* FIRMWARE:  bulk sink (maybe accepts short writes) */
1903                 retval = simple_io(dev, urb, param->iterations, 0, 0, "test1");
1904                 simple_free_urb(urb);
1905                 break;
1906         case 2:
1907                 if (dev->in_pipe == 0)
1908                         break;
1909                 dev_info(&intf->dev,
1910                                 "TEST 2:  read %d bytes %u times\n",
1911                                 param->length, param->iterations);
1912                 urb = simple_alloc_urb(udev, dev->in_pipe, param->length);
1913                 if (!urb) {
1914                         retval = -ENOMEM;
1915                         break;
1916                 }
1917                 /* FIRMWARE:  bulk source (maybe generates short writes) */
1918                 retval = simple_io(dev, urb, param->iterations, 0, 0, "test2");
1919                 simple_free_urb(urb);
1920                 break;
1921         case 3:
1922                 if (dev->out_pipe == 0 || param->vary == 0)
1923                         break;
1924                 dev_info(&intf->dev,
1925                                 "TEST 3:  write/%d 0..%d bytes %u times\n",
1926                                 param->vary, param->length, param->iterations);
1927                 urb = simple_alloc_urb(udev, dev->out_pipe, param->length);
1928                 if (!urb) {
1929                         retval = -ENOMEM;
1930                         break;
1931                 }
1932                 /* FIRMWARE:  bulk sink (maybe accepts short writes) */
1933                 retval = simple_io(dev, urb, param->iterations, param->vary,
1934                                         0, "test3");
1935                 simple_free_urb(urb);
1936                 break;
1937         case 4:
1938                 if (dev->in_pipe == 0 || param->vary == 0)
1939                         break;
1940                 dev_info(&intf->dev,
1941                                 "TEST 4:  read/%d 0..%d bytes %u times\n",
1942                                 param->vary, param->length, param->iterations);
1943                 urb = simple_alloc_urb(udev, dev->in_pipe, param->length);
1944                 if (!urb) {
1945                         retval = -ENOMEM;
1946                         break;
1947                 }
1948                 /* FIRMWARE:  bulk source (maybe generates short writes) */
1949                 retval = simple_io(dev, urb, param->iterations, param->vary,
1950                                         0, "test4");
1951                 simple_free_urb(urb);
1952                 break;
1953
1954         /* Queued bulk I/O tests */
1955         case 5:
1956                 if (dev->out_pipe == 0 || param->sglen == 0)
1957                         break;
1958                 dev_info(&intf->dev,
1959                         "TEST 5:  write %d sglists %d entries of %d bytes\n",
1960                                 param->iterations,
1961                                 param->sglen, param->length);
1962                 sg = alloc_sglist(param->sglen, param->length, 0);
1963                 if (!sg) {
1964                         retval = -ENOMEM;
1965                         break;
1966                 }
1967                 /* FIRMWARE:  bulk sink (maybe accepts short writes) */
1968                 retval = perform_sglist(dev, param->iterations, dev->out_pipe,
1969                                 &req, sg, param->sglen);
1970                 free_sglist(sg, param->sglen);
1971                 break;
1972
1973         case 6:
1974                 if (dev->in_pipe == 0 || param->sglen == 0)
1975                         break;
1976                 dev_info(&intf->dev,
1977                         "TEST 6:  read %d sglists %d entries of %d bytes\n",
1978                                 param->iterations,
1979                                 param->sglen, param->length);
1980                 sg = alloc_sglist(param->sglen, param->length, 0);
1981                 if (!sg) {
1982                         retval = -ENOMEM;
1983                         break;
1984                 }
1985                 /* FIRMWARE:  bulk source (maybe generates short writes) */
1986                 retval = perform_sglist(dev, param->iterations, dev->in_pipe,
1987                                 &req, sg, param->sglen);
1988                 free_sglist(sg, param->sglen);
1989                 break;
1990         case 7:
1991                 if (dev->out_pipe == 0 || param->sglen == 0 || param->vary == 0)
1992                         break;
1993                 dev_info(&intf->dev,
1994                         "TEST 7:  write/%d %d sglists %d entries 0..%d bytes\n",
1995                                 param->vary, param->iterations,
1996                                 param->sglen, param->length);
1997                 sg = alloc_sglist(param->sglen, param->length, param->vary);
1998                 if (!sg) {
1999                         retval = -ENOMEM;
2000                         break;
2001                 }
2002                 /* FIRMWARE:  bulk sink (maybe accepts short writes) */
2003                 retval = perform_sglist(dev, param->iterations, dev->out_pipe,
2004                                 &req, sg, param->sglen);
2005                 free_sglist(sg, param->sglen);
2006                 break;
2007         case 8:
2008                 if (dev->in_pipe == 0 || param->sglen == 0 || param->vary == 0)
2009                         break;
2010                 dev_info(&intf->dev,
2011                         "TEST 8:  read/%d %d sglists %d entries 0..%d bytes\n",
2012                                 param->vary, param->iterations,
2013                                 param->sglen, param->length);
2014                 sg = alloc_sglist(param->sglen, param->length, param->vary);
2015                 if (!sg) {
2016                         retval = -ENOMEM;
2017                         break;
2018                 }
2019                 /* FIRMWARE:  bulk source (maybe generates short writes) */
2020                 retval = perform_sglist(dev, param->iterations, dev->in_pipe,
2021                                 &req, sg, param->sglen);
2022                 free_sglist(sg, param->sglen);
2023                 break;
2024
2025         /* non-queued sanity tests for control (chapter 9 subset) */
2026         case 9:
2027                 retval = 0;
2028                 dev_info(&intf->dev,
2029                         "TEST 9:  ch9 (subset) control tests, %d times\n",
2030                                 param->iterations);
2031                 for (i = param->iterations; retval == 0 && i--; /* NOP */)
2032                         retval = ch9_postconfig(dev);
2033                 if (retval)
2034                         dev_err(&intf->dev, "ch9 subset failed, "
2035                                         "iterations left %d\n", i);
2036                 break;
2037
2038         /* queued control messaging */
2039         case 10:
2040                 retval = 0;
2041                 dev_info(&intf->dev,
2042                                 "TEST 10:  queue %d control calls, %d times\n",
2043                                 param->sglen,
2044                                 param->iterations);
2045                 retval = test_ctrl_queue(dev, param);
2046                 break;
2047
2048         /* simple non-queued unlinks (ring with one urb) */
2049         case 11:
2050                 if (dev->in_pipe == 0 || !param->length)
2051                         break;
2052                 retval = 0;
2053                 dev_info(&intf->dev, "TEST 11:  unlink %d reads of %d\n",
2054                                 param->iterations, param->length);
2055                 for (i = param->iterations; retval == 0 && i--; /* NOP */)
2056                         retval = unlink_simple(dev, dev->in_pipe,
2057                                                 param->length);
2058                 if (retval)
2059                         dev_err(&intf->dev, "unlink reads failed %d, "
2060                                 "iterations left %d\n", retval, i);
2061                 break;
2062         case 12:
2063                 if (dev->out_pipe == 0 || !param->length)
2064                         break;
2065                 retval = 0;
2066                 dev_info(&intf->dev, "TEST 12:  unlink %d writes of %d\n",
2067                                 param->iterations, param->length);
2068                 for (i = param->iterations; retval == 0 && i--; /* NOP */)
2069                         retval = unlink_simple(dev, dev->out_pipe,
2070                                                 param->length);
2071                 if (retval)
2072                         dev_err(&intf->dev, "unlink writes failed %d, "
2073                                 "iterations left %d\n", retval, i);
2074                 break;
2075
2076         /* ep halt tests */
2077         case 13:
2078                 if (dev->out_pipe == 0 && dev->in_pipe == 0)
2079                         break;
2080                 retval = 0;
2081                 dev_info(&intf->dev, "TEST 13:  set/clear %d halts\n",
2082                                 param->iterations);
2083                 for (i = param->iterations; retval == 0 && i--; /* NOP */)
2084                         retval = halt_simple(dev);
2085
2086                 if (retval)
2087                         ERROR(dev, "halts failed, iterations left %d\n", i);
2088                 break;
2089
2090         /* control write tests */
2091         case 14:
2092                 if (!dev->info->ctrl_out)
2093                         break;
2094                 dev_info(&intf->dev, "TEST 14:  %d ep0out, %d..%d vary %d\n",
2095                                 param->iterations,
2096                                 realworld ? 1 : 0, param->length,
2097                                 param->vary);
2098                 retval = ctrl_out(dev, param->iterations,
2099                                 param->length, param->vary, 0);
2100                 break;
2101
2102         /* iso write tests */
2103         case 15:
2104                 if (dev->out_iso_pipe == 0 || param->sglen == 0)
2105                         break;
2106                 dev_info(&intf->dev,
2107                         "TEST 15:  write %d iso, %d entries of %d bytes\n",
2108                                 param->iterations,
2109                                 param->sglen, param->length);
2110                 /* FIRMWARE:  iso sink */
2111                 retval = test_iso_queue(dev, param,
2112                                 dev->out_iso_pipe, dev->iso_out, 0);
2113                 break;
2114
2115         /* iso read tests */
2116         case 16:
2117                 if (dev->in_iso_pipe == 0 || param->sglen == 0)
2118                         break;
2119                 dev_info(&intf->dev,
2120                         "TEST 16:  read %d iso, %d entries of %d bytes\n",
2121                                 param->iterations,
2122                                 param->sglen, param->length);
2123                 /* FIRMWARE:  iso source */
2124                 retval = test_iso_queue(dev, param,
2125                                 dev->in_iso_pipe, dev->iso_in, 0);
2126                 break;
2127
2128         /* FIXME scatterlist cancel (needs helper thread) */
2129
2130         /* Tests for bulk I/O using DMA mapping by core and odd address */
2131         case 17:
2132                 if (dev->out_pipe == 0)
2133                         break;
2134                 dev_info(&intf->dev,
2135                         "TEST 17:  write odd addr %d bytes %u times core map\n",
2136                         param->length, param->iterations);
2137
2138                 retval = test_unaligned_bulk(
2139                                 dev, dev->out_pipe,
2140                                 param->length, param->iterations,
2141                                 0, "test17");
2142                 break;
2143
2144         case 18:
2145                 if (dev->in_pipe == 0)
2146                         break;
2147                 dev_info(&intf->dev,
2148                         "TEST 18:  read odd addr %d bytes %u times core map\n",
2149                         param->length, param->iterations);
2150
2151                 retval = test_unaligned_bulk(
2152                                 dev, dev->in_pipe,
2153                                 param->length, param->iterations,
2154                                 0, "test18");
2155                 break;
2156
2157         /* Tests for bulk I/O using premapped coherent buffer and odd address */
2158         case 19:
2159                 if (dev->out_pipe == 0)
2160                         break;
2161                 dev_info(&intf->dev,
2162                         "TEST 19:  write odd addr %d bytes %u times premapped\n",
2163                         param->length, param->iterations);
2164
2165                 retval = test_unaligned_bulk(
2166                                 dev, dev->out_pipe,
2167                                 param->length, param->iterations,
2168                                 URB_NO_TRANSFER_DMA_MAP, "test19");
2169                 break;
2170
2171         case 20:
2172                 if (dev->in_pipe == 0)
2173                         break;
2174                 dev_info(&intf->dev,
2175                         "TEST 20:  read odd addr %d bytes %u times premapped\n",
2176                         param->length, param->iterations);
2177
2178                 retval = test_unaligned_bulk(
2179                                 dev, dev->in_pipe,
2180                                 param->length, param->iterations,
2181                                 URB_NO_TRANSFER_DMA_MAP, "test20");
2182                 break;
2183
2184         /* control write tests with unaligned buffer */
2185         case 21:
2186                 if (!dev->info->ctrl_out)
2187                         break;
2188                 dev_info(&intf->dev,
2189                                 "TEST 21:  %d ep0out odd addr, %d..%d vary %d\n",
2190                                 param->iterations,
2191                                 realworld ? 1 : 0, param->length,
2192                                 param->vary);
2193                 retval = ctrl_out(dev, param->iterations,
2194                                 param->length, param->vary, 1);
2195                 break;
2196
2197         /* unaligned iso tests */
2198         case 22:
2199                 if (dev->out_iso_pipe == 0 || param->sglen == 0)
2200                         break;
2201                 dev_info(&intf->dev,
2202                         "TEST 22:  write %d iso odd, %d entries of %d bytes\n",
2203                                 param->iterations,
2204                                 param->sglen, param->length);
2205                 retval = test_iso_queue(dev, param,
2206                                 dev->out_iso_pipe, dev->iso_out, 1);
2207                 break;
2208
2209         case 23:
2210                 if (dev->in_iso_pipe == 0 || param->sglen == 0)
2211                         break;
2212                 dev_info(&intf->dev,
2213                         "TEST 23:  read %d iso odd, %d entries of %d bytes\n",
2214                                 param->iterations,
2215                                 param->sglen, param->length);
2216                 retval = test_iso_queue(dev, param,
2217                                 dev->in_iso_pipe, dev->iso_in, 1);
2218                 break;
2219
2220         /* unlink URBs from a bulk-OUT queue */
2221         case 24:
2222                 if (dev->out_pipe == 0 || !param->length || param->sglen < 4)
2223                         break;
2224                 retval = 0;
2225                 dev_info(&intf->dev, "TEST 24:  unlink from %d queues of "
2226                                 "%d %d-byte writes\n",
2227                                 param->iterations, param->sglen, param->length);
2228                 for (i = param->iterations; retval == 0 && i > 0; --i) {
2229                         retval = unlink_queued(dev, dev->out_pipe,
2230                                                 param->sglen, param->length);
2231                         if (retval) {
2232                                 dev_err(&intf->dev,
2233                                         "unlink queued writes failed %d, "
2234                                         "iterations left %d\n", retval, i);
2235                                 break;
2236                         }
2237                 }
2238                 break;
2239
2240         }
2241         do_gettimeofday(&param->duration);
2242         param->duration.tv_sec -= start.tv_sec;
2243         param->duration.tv_usec -= start.tv_usec;
2244         if (param->duration.tv_usec < 0) {
2245                 param->duration.tv_usec += 1000 * 1000;
2246                 param->duration.tv_sec -= 1;
2247         }
2248         mutex_unlock(&dev->lock);
2249         return retval;
2250 }
2251
2252 /*-------------------------------------------------------------------------*/
2253
2254 static unsigned force_interrupt;
2255 module_param(force_interrupt, uint, 0);
2256 MODULE_PARM_DESC(force_interrupt, "0 = test default; else interrupt");
2257
2258 #ifdef  GENERIC
2259 static unsigned short vendor;
2260 module_param(vendor, ushort, 0);
2261 MODULE_PARM_DESC(vendor, "vendor code (from usb-if)");
2262
2263 static unsigned short product;
2264 module_param(product, ushort, 0);
2265 MODULE_PARM_DESC(product, "product code (from vendor)");
2266 #endif
2267
2268 static int
2269 usbtest_probe(struct usb_interface *intf, const struct usb_device_id *id)
2270 {
2271         struct usb_device       *udev;
2272         struct usbtest_dev      *dev;
2273         struct usbtest_info     *info;
2274         char                    *rtest, *wtest;
2275         char                    *irtest, *iwtest;
2276
2277         udev = interface_to_usbdev(intf);
2278
2279 #ifdef  GENERIC
2280         /* specify devices by module parameters? */
2281         if (id->match_flags == 0) {
2282                 /* vendor match required, product match optional */
2283                 if (!vendor || le16_to_cpu(udev->descriptor.idVendor) != (u16)vendor)
2284                         return -ENODEV;
2285                 if (product && le16_to_cpu(udev->descriptor.idProduct) != (u16)product)
2286                         return -ENODEV;
2287                 dev_info(&intf->dev, "matched module params, "
2288                                         "vend=0x%04x prod=0x%04x\n",
2289                                 le16_to_cpu(udev->descriptor.idVendor),
2290                                 le16_to_cpu(udev->descriptor.idProduct));
2291         }
2292 #endif
2293
2294         dev = kzalloc(sizeof(*dev), GFP_KERNEL);
2295         if (!dev)
2296                 return -ENOMEM;
2297         info = (struct usbtest_info *) id->driver_info;
2298         dev->info = info;
2299         mutex_init(&dev->lock);
2300
2301         dev->intf = intf;
2302
2303         /* cacheline-aligned scratch for i/o */
2304         dev->buf = kmalloc(TBUF_SIZE, GFP_KERNEL);
2305         if (dev->buf == NULL) {
2306                 kfree(dev);
2307                 return -ENOMEM;
2308         }
2309
2310         /* NOTE this doesn't yet test the handful of difference that are
2311          * visible with high speed interrupts:  bigger maxpacket (1K) and
2312          * "high bandwidth" modes (up to 3 packets/uframe).
2313          */
2314         rtest = wtest = "";
2315         irtest = iwtest = "";
2316         if (force_interrupt || udev->speed == USB_SPEED_LOW) {
2317                 if (info->ep_in) {
2318                         dev->in_pipe = usb_rcvintpipe(udev, info->ep_in);
2319                         rtest = " intr-in";
2320                 }
2321                 if (info->ep_out) {
2322                         dev->out_pipe = usb_sndintpipe(udev, info->ep_out);
2323                         wtest = " intr-out";
2324                 }
2325         } else {
2326                 if (override_alt >= 0 || info->autoconf) {
2327                         int status;
2328
2329                         status = get_endpoints(dev, intf);
2330                         if (status < 0) {
2331                                 WARNING(dev, "couldn't get endpoints, %d\n",
2332                                                 status);
2333                                 kfree(dev->buf);
2334                                 kfree(dev);
2335                                 return status;
2336                         }
2337                         /* may find bulk or ISO pipes */
2338                 } else {
2339                         if (info->ep_in)
2340                                 dev->in_pipe = usb_rcvbulkpipe(udev,
2341                                                         info->ep_in);
2342                         if (info->ep_out)
2343                                 dev->out_pipe = usb_sndbulkpipe(udev,
2344                                                         info->ep_out);
2345                 }
2346                 if (dev->in_pipe)
2347                         rtest = " bulk-in";
2348                 if (dev->out_pipe)
2349                         wtest = " bulk-out";
2350                 if (dev->in_iso_pipe)
2351                         irtest = " iso-in";
2352                 if (dev->out_iso_pipe)
2353                         iwtest = " iso-out";
2354         }
2355
2356         usb_set_intfdata(intf, dev);
2357         dev_info(&intf->dev, "%s\n", info->name);
2358         dev_info(&intf->dev, "%s {control%s%s%s%s%s} tests%s\n",
2359                         usb_speed_string(udev->speed),
2360                         info->ctrl_out ? " in/out" : "",
2361                         rtest, wtest,
2362                         irtest, iwtest,
2363                         info->alt >= 0 ? " (+alt)" : "");
2364         return 0;
2365 }
2366
2367 static int usbtest_suspend(struct usb_interface *intf, pm_message_t message)
2368 {
2369         return 0;
2370 }
2371
2372 static int usbtest_resume(struct usb_interface *intf)
2373 {
2374         return 0;
2375 }
2376
2377
2378 static void usbtest_disconnect(struct usb_interface *intf)
2379 {
2380         struct usbtest_dev      *dev = usb_get_intfdata(intf);
2381
2382         usb_set_intfdata(intf, NULL);
2383         dev_dbg(&intf->dev, "disconnect\n");
2384         kfree(dev);
2385 }
2386
2387 /* Basic testing only needs a device that can source or sink bulk traffic.
2388  * Any device can test control transfers (default with GENERIC binding).
2389  *
2390  * Several entries work with the default EP0 implementation that's built
2391  * into EZ-USB chips.  There's a default vendor ID which can be overridden
2392  * by (very) small config EEPROMS, but otherwise all these devices act
2393  * identically until firmware is loaded:  only EP0 works.  It turns out
2394  * to be easy to make other endpoints work, without modifying that EP0
2395  * behavior.  For now, we expect that kind of firmware.
2396  */
2397
2398 /* an21xx or fx versions of ez-usb */
2399 static struct usbtest_info ez1_info = {
2400         .name           = "EZ-USB device",
2401         .ep_in          = 2,
2402         .ep_out         = 2,
2403         .alt            = 1,
2404 };
2405
2406 /* fx2 version of ez-usb */
2407 static struct usbtest_info ez2_info = {
2408         .name           = "FX2 device",
2409         .ep_in          = 6,
2410         .ep_out         = 2,
2411         .alt            = 1,
2412 };
2413
2414 /* ezusb family device with dedicated usb test firmware,
2415  */
2416 static struct usbtest_info fw_info = {
2417         .name           = "usb test device",
2418         .ep_in          = 2,
2419         .ep_out         = 2,
2420         .alt            = 1,
2421         .autoconf       = 1,            /* iso and ctrl_out need autoconf */
2422         .ctrl_out       = 1,
2423         .iso            = 1,            /* iso_ep's are #8 in/out */
2424 };
2425
2426 /* peripheral running Linux and 'zero.c' test firmware, or
2427  * its user-mode cousin. different versions of this use
2428  * different hardware with the same vendor/product codes.
2429  * host side MUST rely on the endpoint descriptors.
2430  */
2431 static struct usbtest_info gz_info = {
2432         .name           = "Linux gadget zero",
2433         .autoconf       = 1,
2434         .ctrl_out       = 1,
2435         .iso            = 1,
2436         .alt            = 0,
2437 };
2438
2439 static struct usbtest_info um_info = {
2440         .name           = "Linux user mode test driver",
2441         .autoconf       = 1,
2442         .alt            = -1,
2443 };
2444
2445 static struct usbtest_info um2_info = {
2446         .name           = "Linux user mode ISO test driver",
2447         .autoconf       = 1,
2448         .iso            = 1,
2449         .alt            = -1,
2450 };
2451
2452 #ifdef IBOT2
2453 /* this is a nice source of high speed bulk data;
2454  * uses an FX2, with firmware provided in the device
2455  */
2456 static struct usbtest_info ibot2_info = {
2457         .name           = "iBOT2 webcam",
2458         .ep_in          = 2,
2459         .alt            = -1,
2460 };
2461 #endif
2462
2463 #ifdef GENERIC
2464 /* we can use any device to test control traffic */
2465 static struct usbtest_info generic_info = {
2466         .name           = "Generic USB device",
2467         .alt            = -1,
2468 };
2469 #endif
2470
2471
2472 static const struct usb_device_id id_table[] = {
2473
2474         /*-------------------------------------------------------------*/
2475
2476         /* EZ-USB devices which download firmware to replace (or in our
2477          * case augment) the default device implementation.
2478          */
2479
2480         /* generic EZ-USB FX controller */
2481         { USB_DEVICE(0x0547, 0x2235),
2482                 .driver_info = (unsigned long) &ez1_info,
2483         },
2484
2485         /* CY3671 development board with EZ-USB FX */
2486         { USB_DEVICE(0x0547, 0x0080),
2487                 .driver_info = (unsigned long) &ez1_info,
2488         },
2489
2490         /* generic EZ-USB FX2 controller (or development board) */
2491         { USB_DEVICE(0x04b4, 0x8613),
2492                 .driver_info = (unsigned long) &ez2_info,
2493         },
2494
2495         /* re-enumerated usb test device firmware */
2496         { USB_DEVICE(0xfff0, 0xfff0),
2497                 .driver_info = (unsigned long) &fw_info,
2498         },
2499
2500         /* "Gadget Zero" firmware runs under Linux */
2501         { USB_DEVICE(0x0525, 0xa4a0),
2502                 .driver_info = (unsigned long) &gz_info,
2503         },
2504
2505         /* so does a user-mode variant */
2506         { USB_DEVICE(0x0525, 0xa4a4),
2507                 .driver_info = (unsigned long) &um_info,
2508         },
2509
2510         /* ... and a user-mode variant that talks iso */
2511         { USB_DEVICE(0x0525, 0xa4a3),
2512                 .driver_info = (unsigned long) &um2_info,
2513         },
2514
2515 #ifdef KEYSPAN_19Qi
2516         /* Keyspan 19qi uses an21xx (original EZ-USB) */
2517         /* this does not coexist with the real Keyspan 19qi driver! */
2518         { USB_DEVICE(0x06cd, 0x010b),
2519                 .driver_info = (unsigned long) &ez1_info,
2520         },
2521 #endif
2522
2523         /*-------------------------------------------------------------*/
2524
2525 #ifdef IBOT2
2526         /* iBOT2 makes a nice source of high speed bulk-in data */
2527         /* this does not coexist with a real iBOT2 driver! */
2528         { USB_DEVICE(0x0b62, 0x0059),
2529                 .driver_info = (unsigned long) &ibot2_info,
2530         },
2531 #endif
2532
2533         /*-------------------------------------------------------------*/
2534
2535 #ifdef GENERIC
2536         /* module params can specify devices to use for control tests */
2537         { .driver_info = (unsigned long) &generic_info, },
2538 #endif
2539
2540         /*-------------------------------------------------------------*/
2541
2542         { }
2543 };
2544 MODULE_DEVICE_TABLE(usb, id_table);
2545
2546 static struct usb_driver usbtest_driver = {
2547         .name =         "usbtest",
2548         .id_table =     id_table,
2549         .probe =        usbtest_probe,
2550         .unlocked_ioctl = usbtest_ioctl,
2551         .disconnect =   usbtest_disconnect,
2552         .suspend =      usbtest_suspend,
2553         .resume =       usbtest_resume,
2554 };
2555
2556 /*-------------------------------------------------------------------------*/
2557
2558 static int __init usbtest_init(void)
2559 {
2560 #ifdef GENERIC
2561         if (vendor)
2562                 pr_debug("params: vend=0x%04x prod=0x%04x\n", vendor, product);
2563 #endif
2564         return usb_register(&usbtest_driver);
2565 }
2566 module_init(usbtest_init);
2567
2568 static void __exit usbtest_exit(void)
2569 {
2570         usb_deregister(&usbtest_driver);
2571 }
2572 module_exit(usbtest_exit);
2573
2574 MODULE_DESCRIPTION("USB Core/HCD Testing Driver");
2575 MODULE_LICENSE("GPL");
2576