drivers/usb/storage/usb.c

   1 /* Driver for USB Mass Storage compliant devices
   2  *
   3  * Current development and maintenance by:
   4  *   (c) 1999-2003 Matthew Dharm (mdharm-usb@one-eyed-alien.net)
   5  *
   6  * Developed with the assistance of:
   7  *   (c) 2000 David L. Brown, Jr. (usb-storage@davidb.org)
   8  *   (c) 2003-2009 Alan Stern (stern@rowland.harvard.edu)
   9  *
  10  * Initial work by:
  11  *   (c) 1999 Michael Gee (michael@linuxspecific.com)
  12  *
  13  * usb_device_id support by Adam J. Richter (adam@yggdrasil.com):
  14  *   (c) 2000 Yggdrasil Computing, Inc.
  15  *
  16  * This driver is based on the 'USB Mass Storage Class' document. This
  17  * describes in detail the protocol used to communicate with such
  18  * devices.  Clearly, the designers had SCSI and ATAPI commands in
  19  * mind when they created this document.  The commands are all very
  20  * similar to commands in the SCSI-II and ATAPI specifications.
  21  *
  22  * It is important to note that in a number of cases this class
  23  * exhibits class-specific exemptions from the USB specification.
  24  * Notably the usage of NAK, STALL and ACK differs from the norm, in
  25  * that they are used to communicate wait, failed and OK on commands.
  26  *
  27  * Also, for certain devices, the interrupt endpoint is used to convey
  28  * status of a command.
  29  *
  30  * Please see http://www.one-eyed-alien.net/~mdharm/linux-usb for more
  31  * information about this driver.
  32  *
  33  * This program is free software; you can redistribute it and/or modify it
  34  * under the terms of the GNU General Public License as published by the
  35  * Free Software Foundation; either version 2, or (at your option) any
  36  * later version.
  37  *
  38  * This program is distributed in the hope that it will be useful, but
  39  * WITHOUT ANY WARRANTY; without even the implied warranty of
  40  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  41  * General Public License for more details.
  42  *
  43  * You should have received a copy of the GNU General Public License along
  44  * with this program; if not, write to the Free Software Foundation, Inc.,
  45  * 675 Mass Ave, Cambridge, MA 02139, USA.
  46  */
  47
  48 #ifdef CONFIG_USB_STORAGE_DEBUG
  49 #define DEBUG
  50 #endif
  51
  52 #include <linux/sched.h>
  53 #include <linux/errno.h>
  54 #include <linux/freezer.h>
  55 #include <linux/module.h>
  56 #include <linux/init.h>
  57 #include <linux/slab.h>
  58 #include <linux/kthread.h>
  59 #include <linux/mutex.h>
  60 #include <linux/utsname.h>
  61
  62 #include <scsi/scsi.h>
  63 #include <scsi/scsi_cmnd.h>
  64 #include <scsi/scsi_device.h>
  65
  66 #include "usb.h"
  67 #include "scsiglue.h"
  68 #include "transport.h"
  69 #include "protocol.h"
  70 #include "debug.h"
  71 #include "initializers.h"
  72
  73 #include "sierra_ms.h"
  74 #include "option_ms.h"
  75
  76 /* Some informational data */
  77 MODULE_AUTHOR("Matthew Dharm <mdharm-usb@one-eyed-alien.net>");
  78 MODULE_DESCRIPTION("USB Mass Storage driver for Linux");
  79 MODULE_LICENSE("GPL");
  80
  81 static unsigned int delay_use = 5;
  82 module_param(delay_use, uint, S_IRUGO | S_IWUSR);
  83 MODULE_PARM_DESC(delay_use, "seconds to delay before using a new device");
  84
  85 static char quirks[128];
  86 module_param_string(quirks, quirks, sizeof(quirks), S_IRUGO | S_IWUSR);
  87 MODULE_PARM_DESC(quirks, "supplemental list of device IDs and their quirks");
  88
  89
  90 /*
  91  * The entries in this table correspond, line for line,
  92  * with the entries in usb_storage_usb_ids[], defined in usual-tables.c.
  93  */
  94
  95 /* The vendor name should be kept at eight characters or less, and
  96  * the product name should be kept at 16 characters or less. If a device
  97  * has the US_FL_FIX_INQUIRY flag, then the vendor and product names
  98  * normally generated by a device thorugh the INQUIRY response will be
  99  * taken from this list, and this is the reason for the above size
 100  * restriction. However, if the flag is not present, then you
 101  * are free to use as many characters as you like.
 102  */
 103
 104 #define UNUSUAL_DEV(idVendor, idProduct, bcdDeviceMin, bcdDeviceMax, \
 105                     vendor_name, product_name, use_protocol, use_transport, \
 106                     init_function, Flags) \
 107 { \
 108         .vendorName = vendor_name,      \
 109         .productName = product_name,    \
 110         .useProtocol = use_protocol,    \
 111         .useTransport = use_transport,  \
 112         .initFunction = init_function,  \
 113 }
 114
 115 #define COMPLIANT_DEV   UNUSUAL_DEV
 116
 117 #define USUAL_DEV(use_protocol, use_transport, use_type) \
 118 { \
 119         .useProtocol = use_protocol,    \
 120         .useTransport = use_transport,  \
 121 }
 122
 123 static struct us_unusual_dev us_unusual_dev_list[] = {
 124 #       include "unusual_devs.h"
 125         { }             /* Terminating entry */
 126 };
 127
 128 #undef UNUSUAL_DEV
 129 #undef COMPLIANT_DEV
 130 #undef USUAL_DEV
 131
 132
 133 #ifdef CONFIG_PM        /* Minimal support for suspend and resume */
 134
 135 int usb_stor_suspend(struct usb_interface *iface, pm_message_t message)
 136 {
 137         struct us_data *us = usb_get_intfdata(iface);
 138
 139         /* Wait until no command is running */
 140         mutex_lock(&us->dev_mutex);
 141
 142         US_DEBUGP("%s\n", __func__);
 143         if (us->suspend_resume_hook)
 144                 (us->suspend_resume_hook)(us, US_SUSPEND);
 145
 146         /* When runtime PM is working, we'll set a flag to indicate
 147          * whether we should autoresume when a SCSI request arrives. */
 148
 149         mutex_unlock(&us->dev_mutex);
 150         return 0;
 151 }
 152 EXPORT_SYMBOL_GPL(usb_stor_suspend);
 153
 154 int usb_stor_resume(struct usb_interface *iface)
 155 {
 156         struct us_data *us = usb_get_intfdata(iface);
 157
 158         mutex_lock(&us->dev_mutex);
 159
 160         US_DEBUGP("%s\n", __func__);
 161         if (us->suspend_resume_hook)
 162                 (us->suspend_resume_hook)(us, US_RESUME);
 163
 164         mutex_unlock(&us->dev_mutex);
 165         return 0;
 166 }
 167 EXPORT_SYMBOL_GPL(usb_stor_resume);
 168
 169 int usb_stor_reset_resume(struct usb_interface *iface)
 170 {
 171         struct us_data *us = usb_get_intfdata(iface);
 172
 173         US_DEBUGP("%s\n", __func__);
 174
 175         /* Report the reset to the SCSI core */
 176         usb_stor_report_bus_reset(us);
 177
 178         /* FIXME: Notify the subdrivers that they need to reinitialize
 179          * the device */
 180         return 0;
 181 }
 182 EXPORT_SYMBOL_GPL(usb_stor_reset_resume);
 183
 184 #endif /* CONFIG_PM */
 185
 186 /*
 187  * The next two routines get called just before and just after
 188  * a USB port reset, whether from this driver or a different one.
 189  */
 190
 191 int usb_stor_pre_reset(struct usb_interface *iface)
 192 {
 193         struct us_data *us = usb_get_intfdata(iface);
 194
 195         US_DEBUGP("%s\n", __func__);
 196
 197         /* Make sure no command runs during the reset */
 198         mutex_lock(&us->dev_mutex);
 199         return 0;
 200 }
 201 EXPORT_SYMBOL_GPL(usb_stor_pre_reset);
 202
 203 int usb_stor_post_reset(struct usb_interface *iface)
 204 {
 205         struct us_data *us = usb_get_intfdata(iface);
 206
 207         US_DEBUGP("%s\n", __func__);
 208
 209         /* Report the reset to the SCSI core */
 210         usb_stor_report_bus_reset(us);
 211
 212         /* FIXME: Notify the subdrivers that they need to reinitialize
 213          * the device */
 214
 215         mutex_unlock(&us->dev_mutex);
 216         return 0;
 217 }
 218 EXPORT_SYMBOL_GPL(usb_stor_post_reset);
 219
 220 /*
 221  * fill_inquiry_response takes an unsigned char array (which must
 222  * be at least 36 characters) and populates the vendor name,
 223  * product name, and revision fields. Then the array is copied
 224  * into the SCSI command's response buffer (oddly enough
 225  * called request_buffer). data_len contains the length of the
 226  * data array, which again must be at least 36.
 227  */
 228
 229 void fill_inquiry_response(struct us_data *us, unsigned char *data,
 230                 unsigned int data_len)
 231 {
 232         if (data_len<36) // You lose.
 233                 return;
 234
 235         memset(data+8, ' ', 28);
 236         if(data[0]&0x20) { /* USB device currently not connected. Return
 237                               peripheral qualifier 001b ("...however, the
 238                               physical device is not currently connected
 239                               to this logical unit") and leave vendor and
 240                               product identification empty. ("If the target
 241                               does store some of the INQUIRY data on the
 242                               device, it may return zeros or ASCII spaces
 243                               (20h) in those fields until the data is
 244                               available from the device."). */
 245         } else {
 246                 u16 bcdDevice = le16_to_cpu(us->pusb_dev->descriptor.bcdDevice);
 247                 int n;
 248
 249                 n = strlen(us->unusual_dev->vendorName);
 250                 memcpy(data+8, us->unusual_dev->vendorName, min(8, n));
 251                 n = strlen(us->unusual_dev->productName);
 252                 memcpy(data+16, us->unusual_dev->productName, min(16, n));
 253
 254                 data[32] = 0x30 + ((bcdDevice>>12) & 0x0F);
 255                 data[33] = 0x30 + ((bcdDevice>>8) & 0x0F);
 256                 data[34] = 0x30 + ((bcdDevice>>4) & 0x0F);
 257                 data[35] = 0x30 + ((bcdDevice) & 0x0F);
 258         }
 259
 260         usb_stor_set_xfer_buf(data, data_len, us->srb);
 261 }
 262 EXPORT_SYMBOL_GPL(fill_inquiry_response);
 263
 264 static int usb_stor_control_thread(void * __us)
 265 {
 266         struct us_data *us = (struct us_data *)__us;
 267         struct Scsi_Host *host = us_to_host(us);
 268
 269         for(;;) {
 270                 US_DEBUGP("*** thread sleeping.\n");
 271                 if (wait_for_completion_interruptible(&us->cmnd_ready))
 272                         break;
 273
 274                 US_DEBUGP("*** thread awakened.\n");
 275
 276                 /* lock the device pointers */
 277                 mutex_lock(&(us->dev_mutex));
 278
 279                 /* lock access to the state */
 280                 scsi_lock(host);
 281
 282                 /* When we are called with no command pending, we're done */
 283                 if (us->srb == NULL) {
 284                         scsi_unlock(host);
 285                         mutex_unlock(&us->dev_mutex);
 286                         US_DEBUGP("-- exiting\n");
 287                         break;
 288                 }
 289
 290                 /* has the command timed out *already* ? */
 291                 if (test_bit(US_FLIDX_TIMED_OUT, &us->dflags)) {
 292                         us->srb->result = DID_ABORT << 16;
 293                         goto SkipForAbort;
 294                 }
 295
 296                 scsi_unlock(host);
 297
 298                 /* reject the command if the direction indicator
 299                  * is UNKNOWN
 300                  */
 301                 if (us->srb->sc_data_direction == DMA_BIDIRECTIONAL) {
 302                         US_DEBUGP("UNKNOWN data direction\n");
 303                         us->srb->result = DID_ERROR << 16;
 304                 }
 305
 306                 /* reject if target != 0 or if LUN is higher than
 307                  * the maximum known LUN
 308                  */
 309                 else if (us->srb->device->id &&
 310                                 !(us->fflags & US_FL_SCM_MULT_TARG)) {
 311                         US_DEBUGP("Bad target number (%d:%d)\n",
 312                                   us->srb->device->id, us->srb->device->lun);
 313                         us->srb->result = DID_BAD_TARGET << 16;
 314                 }
 315
 316                 else if (us->srb->device->lun > us->max_lun) {
 317                         US_DEBUGP("Bad LUN (%d:%d)\n",
 318                                   us->srb->device->id, us->srb->device->lun);
 319                         us->srb->result = DID_BAD_TARGET << 16;
 320                 }
 321
 322                 /* Handle those devices which need us to fake
 323                  * their inquiry data */
 324                 else if ((us->srb->cmnd[0] == INQUIRY) &&
 325                             (us->fflags & US_FL_FIX_INQUIRY)) {
 326                         unsigned char data_ptr[36] = {
 327                             0x00, 0x80, 0x02, 0x02,
 328                             0x1F, 0x00, 0x00, 0x00};
 329
 330                         US_DEBUGP("Faking INQUIRY command\n");
 331                         fill_inquiry_response(us, data_ptr, 36);
 332                         us->srb->result = SAM_STAT_GOOD;
 333                 }
 334
 335                 /* we've got a command, let's do it! */
 336                 else {
 337                         US_DEBUG(usb_stor_show_command(us->srb));
 338                         us->proto_handler(us->srb, us);
 339                 }
 340
 341                 /* lock access to the state */
 342                 scsi_lock(host);
 343
 344                 /* indicate that the command is done */
 345                 if (us->srb->result != DID_ABORT << 16) {
 346                         US_DEBUGP("scsi cmd done, result=0x%x\n",
 347                                    us->srb->result);
 348                         us->srb->scsi_done(us->srb);
 349                 } else {
 350 SkipForAbort:
 351                         US_DEBUGP("scsi command aborted\n");
 352                 }
 353
 354                 /* If an abort request was received we need to signal that
 355                  * the abort has finished.  The proper test for this is
 356                  * the TIMED_OUT flag, not srb->result == DID_ABORT, because
 357                  * the timeout might have occurred after the command had
 358                  * already completed with a different result code. */
 359                 if (test_bit(US_FLIDX_TIMED_OUT, &us->dflags)) {
 360                         complete(&(us->notify));
 361
 362                         /* Allow USB transfers to resume */
 363                         clear_bit(US_FLIDX_ABORTING, &us->dflags);
 364                         clear_bit(US_FLIDX_TIMED_OUT, &us->dflags);
 365                 }
 366
 367                 /* finished working on this command */
 368                 us->srb = NULL;
 369                 scsi_unlock(host);
 370
 371                 /* unlock the device pointers */
 372                 mutex_unlock(&us->dev_mutex);
 373         } /* for (;;) */
 374
 375         /* Wait until we are told to stop */
 376         for (;;) {
 377                 set_current_state(TASK_INTERRUPTIBLE);
 378                 if (kthread_should_stop())
 379                         break;
 380                 schedule();
 381         }
 382         __set_current_state(TASK_RUNNING);
 383         return 0;
 384 }
 385
 386 /***********************************************************************
 387  * Device probing and disconnecting
 388  ***********************************************************************/
 389
 390 /* Associate our private data with the USB device */
 391 static int associate_dev(struct us_data *us, struct usb_interface *intf)
 392 {
 393         US_DEBUGP("-- %s\n", __func__);
 394
 395         /* Fill in the device-related fields */
 396         us->pusb_dev = interface_to_usbdev(intf);
 397         us->pusb_intf = intf;
 398         us->ifnum = intf->cur_altsetting->desc.bInterfaceNumber;
 399         US_DEBUGP("Vendor: 0x%04x, Product: 0x%04x, Revision: 0x%04x\n",
 400                         le16_to_cpu(us->pusb_dev->descriptor.idVendor),
 401                         le16_to_cpu(us->pusb_dev->descriptor.idProduct),
 402                         le16_to_cpu(us->pusb_dev->descriptor.bcdDevice));
 403         US_DEBUGP("Interface Subclass: 0x%02x, Protocol: 0x%02x\n",
 404                         intf->cur_altsetting->desc.bInterfaceSubClass,
 405                         intf->cur_altsetting->desc.bInterfaceProtocol);
 406
 407         /* Store our private data in the interface */
 408         usb_set_intfdata(intf, us);
 409
 410         /* Allocate the device-related DMA-mapped buffers */
 411         us->cr = usb_buffer_alloc(us->pusb_dev, sizeof(*us->cr),
 412                         GFP_KERNEL, &us->cr_dma);
 413         if (!us->cr) {
 414                 US_DEBUGP("usb_ctrlrequest allocation failed\n");
 415                 return -ENOMEM;
 416         }
 417
 418         us->iobuf = usb_buffer_alloc(us->pusb_dev, US_IOBUF_SIZE,
 419                         GFP_KERNEL, &us->iobuf_dma);
 420         if (!us->iobuf) {
 421                 US_DEBUGP("I/O buffer allocation failed\n");
 422                 return -ENOMEM;
 423         }
 424         return 0;
 425 }
 426
 427 /* Works only for digits and letters, but small and fast */
 428 #define TOLOWER(x) ((x) | 0x20)
 429
 430 /* Adjust device flags based on the "quirks=" module parameter */
 431 static void adjust_quirks(struct us_data *us)
 432 {
 433         char *p;
 434         u16 vid = le16_to_cpu(us->pusb_dev->descriptor.idVendor);
 435         u16 pid = le16_to_cpu(us->pusb_dev->descriptor.idProduct);
 436         unsigned f = 0;
 437         unsigned int mask = (US_FL_SANE_SENSE | US_FL_FIX_CAPACITY |
 438                         US_FL_CAPACITY_HEURISTICS | US_FL_IGNORE_DEVICE |
 439                         US_FL_NOT_LOCKABLE | US_FL_MAX_SECTORS_64 |
 440                         US_FL_CAPACITY_OK | US_FL_IGNORE_RESIDUE |
 441                         US_FL_SINGLE_LUN | US_FL_NO_WP_DETECT);
 442
 443         p = quirks;
 444         while (*p) {
 445                 /* Each entry consists of VID:PID:flags */
 446                 if (vid == simple_strtoul(p, &p, 16) &&
 447                                 *p == ':' &&
 448                                 pid == simple_strtoul(p+1, &p, 16) &&
 449                                 *p == ':')
 450                         break;
 451
 452                 /* Move forward to the next entry */
 453                 while (*p) {
 454                         if (*p++ == ',')
 455                                 break;
 456                 }
 457         }
 458         if (!*p)        /* No match */
 459                 return;
 460
 461         /* Collect the flags */
 462         while (*++p && *p != ',') {
 463                 switch (TOLOWER(*p)) {
 464                 case 'a':
 465                         f |= US_FL_SANE_SENSE;
 466                         break;
 467                 case 'b':
 468                         f |= US_FL_BAD_SENSE;
 469                         break;
 470                 case 'c':
 471                         f |= US_FL_FIX_CAPACITY;
 472                         break;
 473                 case 'h':
 474                         f |= US_FL_CAPACITY_HEURISTICS;
 475                         break;
 476                 case 'i':
 477                         f |= US_FL_IGNORE_DEVICE;
 478                         break;
 479                 case 'l':
 480                         f |= US_FL_NOT_LOCKABLE;
 481                         break;
 482                 case 'm':
 483                         f |= US_FL_MAX_SECTORS_64;
 484                         break;
 485                 case 'o':
 486                         f |= US_FL_CAPACITY_OK;
 487                         break;
 488                 case 'r':
 489                         f |= US_FL_IGNORE_RESIDUE;
 490                         break;
 491                 case 's':
 492                         f |= US_FL_SINGLE_LUN;
 493                         break;
 494                 case 'w':
 495                         f |= US_FL_NO_WP_DETECT;
 496                         break;
 497                 /* Ignore unrecognized flag characters */
 498                 }
 499         }
 500         us->fflags = (us->fflags & ~mask) | f;
 501         dev_info(&us->pusb_intf->dev, "Quirks match for "
 502                         "vid %04x pid %04x: %x\n",
 503                         vid, pid, f);
 504 }
 505
 506 /* Get the unusual_devs entries and the string descriptors */
 507 static int get_device_info(struct us_data *us, const struct usb_device_id *id,
 508                 struct us_unusual_dev *unusual_dev)
 509 {
 510         struct usb_device *dev = us->pusb_dev;
 511         struct usb_interface_descriptor *idesc =
 512                 &us->pusb_intf->cur_altsetting->desc;
 513
 514         /* Store the entries */
 515         us->unusual_dev = unusual_dev;
 516         us->subclass = (unusual_dev->useProtocol == US_SC_DEVICE) ?
 517                         idesc->bInterfaceSubClass :
 518                         unusual_dev->useProtocol;
 519         us->protocol = (unusual_dev->useTransport == US_PR_DEVICE) ?
 520                         idesc->bInterfaceProtocol :
 521                         unusual_dev->useTransport;
 522         us->fflags = USB_US_ORIG_FLAGS(id->driver_info);
 523         adjust_quirks(us);
 524
 525         if (us->fflags & US_FL_IGNORE_DEVICE) {
 526                 printk(KERN_INFO USB_STORAGE "device ignored\n");
 527                 return -ENODEV;
 528         }
 529
 530         /*
 531          * This flag is only needed when we're in high-speed, so let's
 532          * disable it if we're in full-speed
 533          */
 534         if (dev->speed != USB_SPEED_HIGH)
 535                 us->fflags &= ~US_FL_GO_SLOW;
 536
 537         /* Log a message if a non-generic unusual_dev entry contains an
 538          * unnecessary subclass or protocol override.  This may stimulate
 539          * reports from users that will help us remove unneeded entries
 540          * from the unusual_devs.h table.
 541          */
 542         if (id->idVendor || id->idProduct) {
 543                 static const char *msgs[3] = {
 544                         "an unneeded SubClass entry",
 545                         "an unneeded Protocol entry",
 546                         "unneeded SubClass and Protocol entries"};
 547                 struct usb_device_descriptor *ddesc = &dev->descriptor;
 548                 int msg = -1;
 549
 550                 if (unusual_dev->useProtocol != US_SC_DEVICE &&
 551                         us->subclass == idesc->bInterfaceSubClass)
 552                         msg += 1;
 553                 if (unusual_dev->useTransport != US_PR_DEVICE &&
 554                         us->protocol == idesc->bInterfaceProtocol)
 555                         msg += 2;
 556                 if (msg >= 0 && !(us->fflags & US_FL_NEED_OVERRIDE))
 557                         printk(KERN_NOTICE USB_STORAGE "This device "
 558                                 "(%04x,%04x,%04x S %02x P %02x)"
 559                                 " has %s in unusual_devs.h (kernel"
 560                                 " %s)\n"
 561                                 "   Please send a copy of this message to "
 562                                 "<linux-usb@vger.kernel.org> and "
 563                                 "<usb-storage@lists.one-eyed-alien.net>\n",
 564                                 le16_to_cpu(ddesc->idVendor),
 565                                 le16_to_cpu(ddesc->idProduct),
 566                                 le16_to_cpu(ddesc->bcdDevice),
 567                                 idesc->bInterfaceSubClass,
 568                                 idesc->bInterfaceProtocol,
 569                                 msgs[msg],
 570                                 utsname()->release);
 571         }
 572
 573         return 0;
 574 }
 575
 576 /* Get the transport settings */
 577 static void get_transport(struct us_data *us)
 578 {
 579         switch (us->protocol) {
 580         case US_PR_CB:
 581                 us->transport_name = "Control/Bulk";
 582                 us->transport = usb_stor_CB_transport;
 583                 us->transport_reset = usb_stor_CB_reset;
 584                 us->max_lun = 7;
 585                 break;
 586
 587         case US_PR_CBI:
 588                 us->transport_name = "Control/Bulk/Interrupt";
 589                 us->transport = usb_stor_CB_transport;
 590                 us->transport_reset = usb_stor_CB_reset;
 591                 us->max_lun = 7;
 592                 break;
 593
 594         case US_PR_BULK:
 595                 us->transport_name = "Bulk";
 596                 us->transport = usb_stor_Bulk_transport;
 597                 us->transport_reset = usb_stor_Bulk_reset;
 598                 break;
 599         }
 600 }
 601
 602 /* Get the protocol settings */
 603 static void get_protocol(struct us_data *us)
 604 {
 605         switch (us->subclass) {
 606         case US_SC_RBC:
 607                 us->protocol_name = "Reduced Block Commands (RBC)";
 608                 us->proto_handler = usb_stor_transparent_scsi_command;
 609                 break;
 610
 611         case US_SC_8020:
 612                 us->protocol_name = "8020i";
 613                 us->proto_handler = usb_stor_pad12_command;
 614                 us->max_lun = 0;
 615                 break;
 616
 617         case US_SC_QIC:
 618                 us->protocol_name = "QIC-157";
 619                 us->proto_handler = usb_stor_pad12_command;
 620                 us->max_lun = 0;
 621                 break;
 622
 623         case US_SC_8070:
 624                 us->protocol_name = "8070i";
 625                 us->proto_handler = usb_stor_pad12_command;
 626                 us->max_lun = 0;
 627                 break;
 628
 629         case US_SC_SCSI:
 630                 us->protocol_name = "Transparent SCSI";
 631                 us->proto_handler = usb_stor_transparent_scsi_command;
 632                 break;
 633
 634         case US_SC_UFI:
 635                 us->protocol_name = "Uniform Floppy Interface (UFI)";
 636                 us->proto_handler = usb_stor_ufi_command;
 637                 break;
 638         }
 639 }
 640
 641 /* Get the pipe settings */
 642 static int get_pipes(struct us_data *us)
 643 {
 644         struct usb_host_interface *altsetting =
 645                 us->pusb_intf->cur_altsetting;
 646         int i;
 647         struct usb_endpoint_descriptor *ep;
 648         struct usb_endpoint_descriptor *ep_in = NULL;
 649         struct usb_endpoint_descriptor *ep_out = NULL;
 650         struct usb_endpoint_descriptor *ep_int = NULL;
 651
 652         /*
 653          * Find the first endpoint of each type we need.
 654          * We are expecting a minimum of 2 endpoints - in and out (bulk).
 655          * An optional interrupt-in is OK (necessary for CBI protocol).
 656          * We will ignore any others.
 657          */
 658         for (i = 0; i < altsetting->desc.bNumEndpoints; i++) {
 659                 ep = &altsetting->endpoint[i].desc;
 660
 661                 if (usb_endpoint_xfer_bulk(ep)) {
 662                         if (usb_endpoint_dir_in(ep)) {
 663                                 if (!ep_in)
 664                                         ep_in = ep;
 665                         } else {
 666                                 if (!ep_out)
 667                                         ep_out = ep;
 668                         }
 669                 }
 670
 671                 else if (usb_endpoint_is_int_in(ep)) {
 672                         if (!ep_int)
 673                                 ep_int = ep;
 674                 }
 675         }
 676
 677         if (!ep_in || !ep_out || (us->protocol == US_PR_CBI && !ep_int)) {
 678                 US_DEBUGP("Endpoint sanity check failed! Rejecting dev.\n");
 679                 return -EIO;
 680         }
 681
 682         /* Calculate and store the pipe values */
 683         us->send_ctrl_pipe = usb_sndctrlpipe(us->pusb_dev, 0);
 684         us->recv_ctrl_pipe = usb_rcvctrlpipe(us->pusb_dev, 0);
 685         us->send_bulk_pipe = usb_sndbulkpipe(us->pusb_dev,
 686                 usb_endpoint_num(ep_out));
 687         us->recv_bulk_pipe = usb_rcvbulkpipe(us->pusb_dev,
 688                 usb_endpoint_num(ep_in));
 689         if (ep_int) {
 690                 us->recv_intr_pipe = usb_rcvintpipe(us->pusb_dev,
 691                         usb_endpoint_num(ep_int));
 692                 us->ep_bInterval = ep_int->bInterval;
 693         }
 694         return 0;
 695 }
 696
 697 /* Initialize all the dynamic resources we need */
 698 static int usb_stor_acquire_resources(struct us_data *us)
 699 {
 700         int p;
 701         struct task_struct *th;
 702
 703         us->current_urb = usb_alloc_urb(0, GFP_KERNEL);
 704         if (!us->current_urb) {
 705                 US_DEBUGP("URB allocation failed\n");
 706                 return -ENOMEM;
 707         }
 708
 709         /* Just before we start our control thread, initialize
 710          * the device if it needs initialization */
 711         if (us->unusual_dev->initFunction) {
 712                 p = us->unusual_dev->initFunction(us);
 713                 if (p)
 714                         return p;
 715         }
 716
 717         /* Start up our control thread */
 718         th = kthread_run(usb_stor_control_thread, us, "usb-storage");
 719         if (IS_ERR(th)) {
 720                 printk(KERN_WARNING USB_STORAGE
 721                        "Unable to start control thread\n");
 722                 return PTR_ERR(th);
 723         }
 724         us->ctl_thread = th;
 725
 726         return 0;
 727 }
 728
 729 /* Release all our dynamic resources */
 730 static void usb_stor_release_resources(struct us_data *us)
 731 {
 732         US_DEBUGP("-- %s\n", __func__);
 733
 734         /* Tell the control thread to exit.  The SCSI host must
 735          * already have been removed and the DISCONNECTING flag set
 736          * so that we won't accept any more commands.
 737          */
 738         US_DEBUGP("-- sending exit command to thread\n");
 739         complete(&us->cmnd_ready);
 740         if (us->ctl_thread)
 741                 kthread_stop(us->ctl_thread);
 742
 743         /* Call the destructor routine, if it exists */
 744         if (us->extra_destructor) {
 745                 US_DEBUGP("-- calling extra_destructor()\n");
 746                 us->extra_destructor(us->extra);
 747         }
 748
 749         /* Free the extra data and the URB */
 750         kfree(us->extra);
 751         usb_free_urb(us->current_urb);
 752 }
 753
 754 /* Dissociate from the USB device */
 755 static void dissociate_dev(struct us_data *us)
 756 {
 757         US_DEBUGP("-- %s\n", __func__);
 758
 759         /* Free the device-related DMA-mapped buffers */
 760         if (us->cr)
 761                 usb_buffer_free(us->pusb_dev, sizeof(*us->cr), us->cr,
 762                                 us->cr_dma);
 763         if (us->iobuf)
 764                 usb_buffer_free(us->pusb_dev, US_IOBUF_SIZE, us->iobuf,
 765                                 us->iobuf_dma);
 766
 767         /* Remove our private data from the interface */
 768         usb_set_intfdata(us->pusb_intf, NULL);
 769 }
 770
 771 /* First stage of disconnect processing: stop SCSI scanning,
 772  * remove the host, and stop accepting new commands
 773  */
 774 static void quiesce_and_remove_host(struct us_data *us)
 775 {
 776         struct Scsi_Host *host = us_to_host(us);
 777
 778         /* If the device is really gone, cut short reset delays */
 779         if (us->pusb_dev->state == USB_STATE_NOTATTACHED)
 780                 set_bit(US_FLIDX_DISCONNECTING, &us->dflags);
 781
 782         /* Prevent SCSI-scanning (if it hasn't started yet)
 783          * and wait for the SCSI-scanning thread to stop.
 784          */
 785         set_bit(US_FLIDX_DONT_SCAN, &us->dflags);
 786         wake_up(&us->delay_wait);
 787         wait_for_completion(&us->scanning_done);
 788
 789         /* Removing the host will perform an orderly shutdown: caches
 790          * synchronized, disks spun down, etc.
 791          */
 792         scsi_remove_host(host);
 793
 794         /* Prevent any new commands from being accepted and cut short
 795          * reset delays.
 796          */
 797         scsi_lock(host);
 798         set_bit(US_FLIDX_DISCONNECTING, &us->dflags);
 799         scsi_unlock(host);
 800         wake_up(&us->delay_wait);
 801 }
 802
 803 /* Second stage of disconnect processing: deallocate all resources */
 804 static void release_everything(struct us_data *us)
 805 {
 806         usb_stor_release_resources(us);
 807         dissociate_dev(us);
 808
 809         /* Drop our reference to the host; the SCSI core will free it
 810          * (and "us" along with it) when the refcount becomes 0. */
 811         scsi_host_put(us_to_host(us));
 812 }
 813
 814 /* Thread to carry out delayed SCSI-device scanning */
 815 static int usb_stor_scan_thread(void * __us)
 816 {
 817         struct us_data *us = (struct us_data *)__us;
 818
 819         dev_dbg(&us->pusb_intf->dev, "device found\n");
 820
 821         set_freezable();
 822         /* Wait for the timeout to expire or for a disconnect */
 823         if (delay_use > 0) {
 824                 dev_dbg(&us->pusb_intf->dev, "waiting for device to settle "
 825                                 "before scanning\n");
 826                 wait_event_freezable_timeout(us->delay_wait,
 827                                 test_bit(US_FLIDX_DONT_SCAN, &us->dflags),
 828                                 delay_use * HZ);
 829         }
 830
 831         /* If the device is still connected, perform the scanning */
 832         if (!test_bit(US_FLIDX_DONT_SCAN, &us->dflags)) {
 833
 834                 /* For bulk-only devices, determine the max LUN value */
 835                 if (us->protocol == US_PR_BULK &&
 836                                 !(us->fflags & US_FL_SINGLE_LUN)) {
 837                         mutex_lock(&us->dev_mutex);
 838                         us->max_lun = usb_stor_Bulk_max_lun(us);
 839                         mutex_unlock(&us->dev_mutex);
 840                 }
 841                 scsi_scan_host(us_to_host(us));
 842                 dev_dbg(&us->pusb_intf->dev, "scan complete\n");
 843
 844                 /* Should we unbind if no devices were detected? */
 845         }
 846
 847         complete_and_exit(&us->scanning_done, 0);
 848 }
 849
 850 static unsigned int usb_stor_sg_tablesize(struct usb_interface *intf)
 851 {
 852         struct usb_device *usb_dev = interface_to_usbdev(intf);
 853
 854         if (usb_dev->bus->sg_tablesize) {
 855                 return usb_dev->bus->sg_tablesize;
 856         }
 857         return SG_ALL;
 858 }
 859
 860 /* First part of general USB mass-storage probing */
 861 int usb_stor_probe1(struct us_data **pus,
 862                 struct usb_interface *intf,
 863                 const struct usb_device_id *id,
 864                 struct us_unusual_dev *unusual_dev)
 865 {
 866         struct Scsi_Host *host;
 867         struct us_data *us;
 868         int result;
 869
 870         US_DEBUGP("USB Mass Storage device detected\n");
 871
 872         /*
 873          * Ask the SCSI layer to allocate a host structure, with extra
 874          * space at the end for our private us_data structure.
 875          */
 876         host = scsi_host_alloc(&usb_stor_host_template, sizeof(*us));
 877         if (!host) {
 878                 printk(KERN_WARNING USB_STORAGE
 879                         "Unable to allocate the scsi host\n");
 880                 return -ENOMEM;
 881         }
 882
 883         /*
 884          * Allow 16-byte CDBs and thus > 2TB
 885          */
 886         host->max_cmd_len = 16;
 887         host->sg_tablesize = usb_stor_sg_tablesize(intf);
 888         *pus = us = host_to_us(host);
 889         memset(us, 0, sizeof(struct us_data));
 890         mutex_init(&(us->dev_mutex));
 891         init_completion(&us->cmnd_ready);
 892         init_completion(&(us->notify));
 893         init_waitqueue_head(&us->delay_wait);
 894         init_completion(&us->scanning_done);
 895
 896         /* Associate the us_data structure with the USB device */
 897         result = associate_dev(us, intf);
 898         if (result)
 899                 goto BadDevice;
 900
 901         /* Get the unusual_devs entries and the descriptors */
 902         result = get_device_info(us, id, unusual_dev);
 903         if (result)
 904                 goto BadDevice;
 905
 906         /* Get standard transport and protocol settings */
 907         get_transport(us);
 908         get_protocol(us);
 909
 910         /* Give the caller a chance to fill in specialized transport
 911          * or protocol settings.
 912          */
 913         return 0;
 914
 915 BadDevice:
 916         US_DEBUGP("storage_probe() failed\n");
 917         release_everything(us);
 918         return result;
 919 }
 920 EXPORT_SYMBOL_GPL(usb_stor_probe1);
 921
 922 /* Second part of general USB mass-storage probing */
 923 int usb_stor_probe2(struct us_data *us)
 924 {
 925         struct task_struct *th;
 926         int result;
 927
 928         /* Make sure the transport and protocol have both been set */
 929         if (!us->transport || !us->proto_handler) {
 930                 result = -ENXIO;
 931                 goto BadDevice;
 932         }
 933         US_DEBUGP("Transport: %s\n", us->transport_name);
 934         US_DEBUGP("Protocol: %s\n", us->protocol_name);
 935
 936         /* fix for single-lun devices */
 937         if (us->fflags & US_FL_SINGLE_LUN)
 938                 us->max_lun = 0;
 939
 940         /* Find the endpoints and calculate pipe values */
 941         result = get_pipes(us);
 942         if (result)
 943                 goto BadDevice;
 944
 945         /* Acquire all the other resources and add the host */
 946         result = usb_stor_acquire_resources(us);
 947         if (result)
 948                 goto BadDevice;
 949         snprintf(us->scsi_name, sizeof(us->scsi_name), "usb-storage %s",
 950                                         dev_name(&us->pusb_intf->dev));
 951         result = scsi_add_host(us_to_host(us), &us->pusb_intf->dev);
 952         if (result) {
 953                 printk(KERN_WARNING USB_STORAGE
 954                         "Unable to add the scsi host\n");
 955                 goto BadDevice;
 956         }
 957
 958         /* Start up the thread for delayed SCSI-device scanning */
 959         th = kthread_create(usb_stor_scan_thread, us, "usb-stor-scan");
 960         if (IS_ERR(th)) {
 961                 printk(KERN_WARNING USB_STORAGE
 962                        "Unable to start the device-scanning thread\n");
 963                 complete(&us->scanning_done);
 964                 quiesce_and_remove_host(us);
 965                 result = PTR_ERR(th);
 966                 goto BadDevice;
 967         }
 968
 969         wake_up_process(th);
 970
 971         return 0;
 972
 973         /* We come here if there are any problems */
 974 BadDevice:
 975         US_DEBUGP("storage_probe() failed\n");
 976         release_everything(us);
 977         return result;
 978 }
 979 EXPORT_SYMBOL_GPL(usb_stor_probe2);
 980
 981 /* Handle a USB mass-storage disconnect */
 982 void usb_stor_disconnect(struct usb_interface *intf)
 983 {
 984         struct us_data *us = usb_get_intfdata(intf);
 985
 986         US_DEBUGP("storage_disconnect() called\n");
 987         quiesce_and_remove_host(us);
 988         release_everything(us);
 989 }
 990 EXPORT_SYMBOL_GPL(usb_stor_disconnect);
 991
 992 /* The main probe routine for standard devices */
 993 static int storage_probe(struct usb_interface *intf,
 994                          const struct usb_device_id *id)
 995 {
 996         struct us_data *us;
 997         int result;
 998
 999         /*
1000          * If libusual is configured, let it decide whether a standard
1001          * device should be handled by usb-storage or by ub.
1002          * If the device isn't standard (is handled by a subdriver
1003          * module) then don't accept it.
1004          */
1005         if (usb_usual_check_type(id, USB_US_TYPE_STOR) ||
1006                         usb_usual_ignore_device(intf))
1007                 return -ENXIO;
1008
1009         /*
1010          * Call the general probe procedures.
1011          *
1012          * The unusual_dev_list array is parallel to the usb_storage_usb_ids
1013          * table, so we use the index of the id entry to find the
1014          * corresponding unusual_devs entry.
1015          */
1016         result = usb_stor_probe1(&us, intf, id,
1017                         (id - usb_storage_usb_ids) + us_unusual_dev_list);
1018         if (result)
1019                 return result;
1020
1021         /* No special transport or protocol settings in the main module */
1022
1023         result = usb_stor_probe2(us);
1024         return result;
1025 }
1026
1027 /***********************************************************************
1028  * Initialization and registration
1029  ***********************************************************************/
1030
1031 static struct usb_driver usb_storage_driver = {
1032         .name =         "usb-storage",
1033         .probe =        storage_probe,
1034         .disconnect =   usb_stor_disconnect,
1035         .suspend =      usb_stor_suspend,
1036         .resume =       usb_stor_resume,
1037         .reset_resume = usb_stor_reset_resume,
1038         .pre_reset =    usb_stor_pre_reset,
1039         .post_reset =   usb_stor_post_reset,
1040         .id_table =     usb_storage_usb_ids,
1041         .soft_unbind =  1,
1042 };
1043
1044 static int __init usb_stor_init(void)
1045 {
1046         int retval;
1047
1048         printk(KERN_INFO "Initializing USB Mass Storage driver...\n");
1049
1050         /* register the driver, return usb_register return code if error */
1051         retval = usb_register(&usb_storage_driver);
1052         if (retval == 0) {
1053                 printk(KERN_INFO "USB Mass Storage support registered.\n");
1054                 usb_usual_set_present(USB_US_TYPE_STOR);
1055         }
1056         return retval;
1057 }
1058
1059 static void __exit usb_stor_exit(void)
1060 {
1061         US_DEBUGP("usb_stor_exit() called\n");
1062
1063         /* Deregister the driver
1064          * This will cause disconnect() to be called for each
1065          * attached unit
1066          */
1067         US_DEBUGP("-- calling usb_deregister()\n");
1068         usb_deregister(&usb_storage_driver) ;
1069
1070         usb_usual_clear_present(USB_US_TYPE_STOR);
1071 }
1072
1073 module_init(usb_stor_init);
1074 module_exit(usb_stor_exit);