1 /*
2  * Driver for SCM Microsystems (a.k.a. Shuttle) USB-ATAPI cable
3  *
4  * Current development and maintenance by:
5  *   (c) 2000, 2001 Robert Baruch (autophile@starband.net)
6  *   (c) 2004, 2005 Daniel Drake <dsd@gentoo.org>
7  *
8  * Developed with the assistance of:
9  *   (c) 2002 Alan Stern <stern@rowland.org>
10  *
11  * Flash support based on earlier work by:
12  *   (c) 2002 Thomas Kreiling <usbdev@sm04.de>
13  *
14  * Many originally ATAPI devices were slightly modified to meet the USB
15  * market by using some kind of translation from ATAPI to USB on the host,
16  * and the peripheral would translate from USB back to ATAPI.
17  *
18  * SCM Microsystems (www.scmmicro.com) makes a device, sold to OEM's only,
19  * which does the USB-to-ATAPI conversion.  By obtaining the data sheet on
20  * their device under nondisclosure agreement, I have been able to write
21  * this driver for Linux.
22  *
23  * The chip used in the device can also be used for EPP and ISA translation
24  * as well. This driver is only guaranteed to work with the ATAPI
25  * translation.
26  *
27  * See the Kconfig help text for a list of devices known to be supported by
28  * this driver.
29  *
30  * This program is free software; you can redistribute it and/or modify it
31  * under the terms of the GNU General Public License as published by the
32  * Free Software Foundation; either version 2, or (at your option) any
33  * later version.
34  *
35  * This program is distributed in the hope that it will be useful, but
36  * WITHOUT ANY WARRANTY; without even the implied warranty of
37  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
38  * General Public License for more details.
39  *
40  * You should have received a copy of the GNU General Public License along
41  * with this program; if not, write to the Free Software Foundation, Inc.,
42  * 675 Mass Ave, Cambridge, MA 02139, USA.
43  */
44 
45 #include <linux/errno.h>
46 #include <linux/module.h>
47 #include <linux/slab.h>
48 #include <linux/cdrom.h>
49 
50 #include <scsi/scsi.h>
51 #include <scsi/scsi_cmnd.h>
52 
53 #include "usb.h"
54 #include "transport.h"
55 #include "protocol.h"
56 #include "debug.h"
57 #include "scsiglue.h"
58 
59 #define DRV_NAME "ums-usbat"
60 
61 MODULE_DESCRIPTION("Driver for SCM Microsystems (a.k.a. Shuttle) USB-ATAPI cable");
62 MODULE_AUTHOR("Daniel Drake <dsd@gentoo.org>, Robert Baruch <autophile@starband.net>");
63 MODULE_LICENSE("GPL");
64 
65 /* Supported device types */
66 #define USBAT_DEV_HP8200	0x01
67 #define USBAT_DEV_FLASH		0x02
68 
69 #define USBAT_EPP_PORT		0x10
70 #define USBAT_EPP_REGISTER	0x30
71 #define USBAT_ATA		0x40
72 #define USBAT_ISA		0x50
73 
74 /* Commands (need to be logically OR'd with an access type */
75 #define USBAT_CMD_READ_REG		0x00
76 #define USBAT_CMD_WRITE_REG		0x01
77 #define USBAT_CMD_READ_BLOCK	0x02
78 #define USBAT_CMD_WRITE_BLOCK	0x03
79 #define USBAT_CMD_COND_READ_BLOCK	0x04
80 #define USBAT_CMD_COND_WRITE_BLOCK	0x05
81 #define USBAT_CMD_WRITE_REGS	0x07
82 
83 /* Commands (these don't need an access type) */
84 #define USBAT_CMD_EXEC_CMD	0x80
85 #define USBAT_CMD_SET_FEAT	0x81
86 #define USBAT_CMD_UIO		0x82
87 
88 /* Methods of accessing UIO register */
89 #define USBAT_UIO_READ	1
90 #define USBAT_UIO_WRITE	0
91 
92 /* Qualifier bits */
93 #define USBAT_QUAL_FCQ	0x20	/* full compare */
94 #define USBAT_QUAL_ALQ	0x10	/* auto load subcount */
95 
96 /* USBAT Flash Media status types */
97 #define USBAT_FLASH_MEDIA_NONE	0
98 #define USBAT_FLASH_MEDIA_CF	1
99 
100 /* USBAT Flash Media change types */
101 #define USBAT_FLASH_MEDIA_SAME	0
102 #define USBAT_FLASH_MEDIA_CHANGED	1
103 
104 /* USBAT ATA registers */
105 #define USBAT_ATA_DATA      0x10  /* read/write data (R/W) */
106 #define USBAT_ATA_FEATURES  0x11  /* set features (W) */
107 #define USBAT_ATA_ERROR     0x11  /* error (R) */
108 #define USBAT_ATA_SECCNT    0x12  /* sector count (R/W) */
109 #define USBAT_ATA_SECNUM    0x13  /* sector number (R/W) */
110 #define USBAT_ATA_LBA_ME    0x14  /* cylinder low (R/W) */
111 #define USBAT_ATA_LBA_HI    0x15  /* cylinder high (R/W) */
112 #define USBAT_ATA_DEVICE    0x16  /* head/device selection (R/W) */
113 #define USBAT_ATA_STATUS    0x17  /* device status (R) */
114 #define USBAT_ATA_CMD       0x17  /* device command (W) */
115 #define USBAT_ATA_ALTSTATUS 0x0E  /* status (no clear IRQ) (R) */
116 
117 /* USBAT User I/O Data registers */
118 #define USBAT_UIO_EPAD		0x80 /* Enable Peripheral Control Signals */
119 #define USBAT_UIO_CDT		0x40 /* Card Detect (Read Only) */
120 				     /* CDT = ACKD & !UI1 & !UI0 */
121 #define USBAT_UIO_1		0x20 /* I/O 1 */
122 #define USBAT_UIO_0		0x10 /* I/O 0 */
123 #define USBAT_UIO_EPP_ATA	0x08 /* 1=EPP mode, 0=ATA mode */
124 #define USBAT_UIO_UI1		0x04 /* Input 1 */
125 #define USBAT_UIO_UI0		0x02 /* Input 0 */
126 #define USBAT_UIO_INTR_ACK	0x01 /* Interrupt (ATA/ISA)/Acknowledge (EPP) */
127 
128 /* USBAT User I/O Enable registers */
129 #define USBAT_UIO_DRVRST	0x80 /* Reset Peripheral */
130 #define USBAT_UIO_ACKD		0x40 /* Enable Card Detect */
131 #define USBAT_UIO_OE1		0x20 /* I/O 1 set=output/clr=input */
132 				     /* If ACKD=1, set OE1 to 1 also. */
133 #define USBAT_UIO_OE0		0x10 /* I/O 0 set=output/clr=input */
134 #define USBAT_UIO_ADPRST	0x01 /* Reset SCM chip */
135 
136 /* USBAT Features */
137 #define USBAT_FEAT_ETEN	0x80	/* External trigger enable */
138 #define USBAT_FEAT_U1	0x08
139 #define USBAT_FEAT_U0	0x04
140 #define USBAT_FEAT_ET1	0x02
141 #define USBAT_FEAT_ET2	0x01
142 
143 struct usbat_info {
144 	int devicetype;
145 
146 	/* Used for Flash readers only */
147 	unsigned long sectors;     /* total sector count */
148 	unsigned long ssize;       /* sector size in bytes */
149 
150 	unsigned char sense_key;
151 	unsigned long sense_asc;   /* additional sense code */
152 	unsigned long sense_ascq;  /* additional sense code qualifier */
153 };
154 
155 #define short_pack(LSB,MSB) ( ((u16)(LSB)) | ( ((u16)(MSB))<<8 ) )
156 #define LSB_of(s) ((s)&0xFF)
157 #define MSB_of(s) ((s)>>8)
158 
159 static int transferred = 0;
160 
161 static int usbat_flash_transport(struct scsi_cmnd * srb, struct us_data *us);
162 static int usbat_hp8200e_transport(struct scsi_cmnd *srb, struct us_data *us);
163 
164 static int init_usbat_cd(struct us_data *us);
165 static int init_usbat_flash(struct us_data *us);
166 
167 
168 /*
169  * The table of devices
170  */
171 #define UNUSUAL_DEV(id_vendor, id_product, bcdDeviceMin, bcdDeviceMax, \
172 		    vendorName, productName, useProtocol, useTransport, \
173 		    initFunction, flags) \
174 { USB_DEVICE_VER(id_vendor, id_product, bcdDeviceMin, bcdDeviceMax), \
175   .driver_info = (flags) }
176 
177 static struct usb_device_id usbat_usb_ids[] = {
178 #	include "unusual_usbat.h"
179 	{ }		/* Terminating entry */
180 };
181 MODULE_DEVICE_TABLE(usb, usbat_usb_ids);
182 
183 #undef UNUSUAL_DEV
184 
185 /*
186  * The flags table
187  */
188 #define UNUSUAL_DEV(idVendor, idProduct, bcdDeviceMin, bcdDeviceMax, \
189 		    vendor_name, product_name, use_protocol, use_transport, \
190 		    init_function, Flags) \
191 { \
192 	.vendorName = vendor_name,	\
193 	.productName = product_name,	\
194 	.useProtocol = use_protocol,	\
195 	.useTransport = use_transport,	\
196 	.initFunction = init_function,	\
197 }
198 
199 static struct us_unusual_dev usbat_unusual_dev_list[] = {
200 #	include "unusual_usbat.h"
201 	{ }		/* Terminating entry */
202 };
203 
204 #undef UNUSUAL_DEV
205 
206 /*
207  * Convenience function to produce an ATA read/write sectors command
208  * Use cmd=0x20 for read, cmd=0x30 for write
209  */
210 static void usbat_pack_ata_sector_cmd(unsigned char *buf,
211 					unsigned char thistime,
212 					u32 sector, unsigned char cmd)
213 {
214 	buf[0] = 0;
215 	buf[1] = thistime;
216 	buf[2] = sector & 0xFF;
217 	buf[3] = (sector >>  8) & 0xFF;
218 	buf[4] = (sector >> 16) & 0xFF;
219 	buf[5] = 0xE0 | ((sector >> 24) & 0x0F);
220 	buf[6] = cmd;
221 }
222 
223 /*
224  * Convenience function to get the device type (flash or hp8200)
225  */
226 static int usbat_get_device_type(struct us_data *us)
227 {
228 	return ((struct usbat_info*)us->extra)->devicetype;
229 }
230 
231 /*
232  * Read a register from the device
233  */
234 static int usbat_read(struct us_data *us,
235 		      unsigned char access,
236 		      unsigned char reg,
237 		      unsigned char *content)
238 {
239 	return usb_stor_ctrl_transfer(us,
240 		us->recv_ctrl_pipe,
241 		access | USBAT_CMD_READ_REG,
242 		0xC0,
243 		(u16)reg,
244 		0,
245 		content,
246 		1);
247 }
248 
249 /*
250  * Write to a register on the device
251  */
252 static int usbat_write(struct us_data *us,
253 		       unsigned char access,
254 		       unsigned char reg,
255 		       unsigned char content)
256 {
257 	return usb_stor_ctrl_transfer(us,
258 		us->send_ctrl_pipe,
259 		access | USBAT_CMD_WRITE_REG,
260 		0x40,
261 		short_pack(reg, content),
262 		0,
263 		NULL,
264 		0);
265 }
266 
267 /*
268  * Convenience function to perform a bulk read
269  */
270 static int usbat_bulk_read(struct us_data *us,
271 			   void* buf,
272 			   unsigned int len,
273 			   int use_sg)
274 {
275 	if (len == 0)
276 		return USB_STOR_XFER_GOOD;
277 
278 	usb_stor_dbg(us, "len = %d\n", len);
279 	return usb_stor_bulk_transfer_sg(us, us->recv_bulk_pipe, buf, len, use_sg, NULL);
280 }
281 
282 /*
283  * Convenience function to perform a bulk write
284  */
285 static int usbat_bulk_write(struct us_data *us,
286 			    void* buf,
287 			    unsigned int len,
288 			    int use_sg)
289 {
290 	if (len == 0)
291 		return USB_STOR_XFER_GOOD;
292 
293 	usb_stor_dbg(us, "len = %d\n", len);
294 	return usb_stor_bulk_transfer_sg(us, us->send_bulk_pipe, buf, len, use_sg, NULL);
295 }
296 
297 /*
298  * Some USBAT-specific commands can only be executed over a command transport
299  * This transport allows one (len=8) or two (len=16) vendor-specific commands
300  * to be executed.
301  */
302 static int usbat_execute_command(struct us_data *us,
303 								 unsigned char *commands,
304 								 unsigned int len)
305 {
306 	return usb_stor_ctrl_transfer(us, us->send_ctrl_pipe,
307 								  USBAT_CMD_EXEC_CMD, 0x40, 0, 0,
308 								  commands, len);
309 }
310 
311 /*
312  * Read the status register
313  */
314 static int usbat_get_status(struct us_data *us, unsigned char *status)
315 {
316 	int rc;
317 	rc = usbat_read(us, USBAT_ATA, USBAT_ATA_STATUS, status);
318 
319 	usb_stor_dbg(us, "0x%02X\n", *status);
320 	return rc;
321 }
322 
323 /*
324  * Check the device status
325  */
326 static int usbat_check_status(struct us_data *us)
327 {
328 	unsigned char *reply = us->iobuf;
329 	int rc;
330 
331 	rc = usbat_get_status(us, reply);
332 	if (rc != USB_STOR_XFER_GOOD)
333 		return USB_STOR_TRANSPORT_FAILED;
334 
335 	/* error/check condition (0x51 is ok) */
336 	if (*reply & 0x01 && *reply != 0x51)
337 		return USB_STOR_TRANSPORT_FAILED;
338 
339 	/* device fault */
340 	if (*reply & 0x20)
341 		return USB_STOR_TRANSPORT_FAILED;
342 
343 	return USB_STOR_TRANSPORT_GOOD;
344 }
345 
346 /*
347  * Stores critical information in internal registers in preparation for the execution
348  * of a conditional usbat_read_blocks or usbat_write_blocks call.
349  */
350 static int usbat_set_shuttle_features(struct us_data *us,
351 				      unsigned char external_trigger,
352 				      unsigned char epp_control,
353 				      unsigned char mask_byte,
354 				      unsigned char test_pattern,
355 				      unsigned char subcountH,
356 				      unsigned char subcountL)
357 {
358 	unsigned char *command = us->iobuf;
359 
360 	command[0] = 0x40;
361 	command[1] = USBAT_CMD_SET_FEAT;
362 
363 	/*
364 	 * The only bit relevant to ATA access is bit 6
365 	 * which defines 8 bit data access (set) or 16 bit (unset)
366 	 */
367 	command[2] = epp_control;
368 
369 	/*
370 	 * If FCQ is set in the qualifier (defined in R/W cmd), then bits U0, U1,
371 	 * ET1 and ET2 define an external event to be checked for on event of a
372 	 * _read_blocks or _write_blocks operation. The read/write will not take
373 	 * place unless the defined trigger signal is active.
374 	 */
375 	command[3] = external_trigger;
376 
377 	/*
378 	 * The resultant byte of the mask operation (see mask_byte) is compared for
379 	 * equivalence with this test pattern. If equal, the read/write will take
380 	 * place.
381 	 */
382 	command[4] = test_pattern;
383 
384 	/*
385 	 * This value is logically ANDed with the status register field specified
386 	 * in the read/write command.
387 	 */
388 	command[5] = mask_byte;
389 
390 	/*
391 	 * If ALQ is set in the qualifier, this field contains the address of the
392 	 * registers where the byte count should be read for transferring the data.
393 	 * If ALQ is not set, then this field contains the number of bytes to be
394 	 * transferred.
395 	 */
396 	command[6] = subcountL;
397 	command[7] = subcountH;
398 
399 	return usbat_execute_command(us, command, 8);
400 }
401 
402 /*
403  * Block, waiting for an ATA device to become not busy or to report
404  * an error condition.
405  */
406 static int usbat_wait_not_busy(struct us_data *us, int minutes)
407 {
408 	int i;
409 	int result;
410 	unsigned char *status = us->iobuf;
411 
412 	/*
413 	 * Synchronizing cache on a CDR could take a heck of a long time,
414 	 * but probably not more than 10 minutes or so. On the other hand,
415 	 * doing a full blank on a CDRW at speed 1 will take about 75
416 	 * minutes!
417 	 */
418 
419 	for (i=0; i<1200+minutes*60; i++) {
420 
421  		result = usbat_get_status(us, status);
422 
423 		if (result!=USB_STOR_XFER_GOOD)
424 			return USB_STOR_TRANSPORT_ERROR;
425 		if (*status & 0x01) { /* check condition */
426 			result = usbat_read(us, USBAT_ATA, 0x10, status);
427 			return USB_STOR_TRANSPORT_FAILED;
428 		}
429 		if (*status & 0x20) /* device fault */
430 			return USB_STOR_TRANSPORT_FAILED;
431 
432 		if ((*status & 0x80)==0x00) { /* not busy */
433 			usb_stor_dbg(us, "Waited not busy for %d steps\n", i);
434 			return USB_STOR_TRANSPORT_GOOD;
435 		}
436 
437 		if (i<500)
438 			msleep(10); /* 5 seconds */
439 		else if (i<700)
440 			msleep(50); /* 10 seconds */
441 		else if (i<1200)
442 			msleep(100); /* 50 seconds */
443 		else
444 			msleep(1000); /* X minutes */
445 	}
446 
447 	usb_stor_dbg(us, "Waited not busy for %d minutes, timing out\n",
448 		     minutes);
449 	return USB_STOR_TRANSPORT_FAILED;
450 }
451 
452 /*
453  * Read block data from the data register
454  */
455 static int usbat_read_block(struct us_data *us,
456 			    void* buf,
457 			    unsigned short len,
458 			    int use_sg)
459 {
460 	int result;
461 	unsigned char *command = us->iobuf;
462 
463 	if (!len)
464 		return USB_STOR_TRANSPORT_GOOD;
465 
466 	command[0] = 0xC0;
467 	command[1] = USBAT_ATA | USBAT_CMD_READ_BLOCK;
468 	command[2] = USBAT_ATA_DATA;
469 	command[3] = 0;
470 	command[4] = 0;
471 	command[5] = 0;
472 	command[6] = LSB_of(len);
473 	command[7] = MSB_of(len);
474 
475 	result = usbat_execute_command(us, command, 8);
476 	if (result != USB_STOR_XFER_GOOD)
477 		return USB_STOR_TRANSPORT_ERROR;
478 
479 	result = usbat_bulk_read(us, buf, len, use_sg);
480 	return (result == USB_STOR_XFER_GOOD ?
481 			USB_STOR_TRANSPORT_GOOD : USB_STOR_TRANSPORT_ERROR);
482 }
483 
484 /*
485  * Write block data via the data register
486  */
487 static int usbat_write_block(struct us_data *us,
488 			     unsigned char access,
489 			     void* buf,
490 			     unsigned short len,
491 			     int minutes,
492 			     int use_sg)
493 {
494 	int result;
495 	unsigned char *command = us->iobuf;
496 
497 	if (!len)
498 		return USB_STOR_TRANSPORT_GOOD;
499 
500 	command[0] = 0x40;
501 	command[1] = access | USBAT_CMD_WRITE_BLOCK;
502 	command[2] = USBAT_ATA_DATA;
503 	command[3] = 0;
504 	command[4] = 0;
505 	command[5] = 0;
506 	command[6] = LSB_of(len);
507 	command[7] = MSB_of(len);
508 
509 	result = usbat_execute_command(us, command, 8);
510 
511 	if (result != USB_STOR_XFER_GOOD)
512 		return USB_STOR_TRANSPORT_ERROR;
513 
514 	result = usbat_bulk_write(us, buf, len, use_sg);
515 	if (result != USB_STOR_XFER_GOOD)
516 		return USB_STOR_TRANSPORT_ERROR;
517 
518 	return usbat_wait_not_busy(us, minutes);
519 }
520 
521 /*
522  * Process read and write requests
523  */
524 static int usbat_hp8200e_rw_block_test(struct us_data *us,
525 				       unsigned char access,
526 				       unsigned char *registers,
527 				       unsigned char *data_out,
528 				       unsigned short num_registers,
529 				       unsigned char data_reg,
530 				       unsigned char status_reg,
531 				       unsigned char timeout,
532 				       unsigned char qualifier,
533 				       int direction,
534 				       void *buf,
535 				       unsigned short len,
536 				       int use_sg,
537 				       int minutes)
538 {
539 	int result;
540 	unsigned int pipe = (direction == DMA_FROM_DEVICE) ?
541 			us->recv_bulk_pipe : us->send_bulk_pipe;
542 
543 	unsigned char *command = us->iobuf;
544 	int i, j;
545 	int cmdlen;
546 	unsigned char *data = us->iobuf;
547 	unsigned char *status = us->iobuf;
548 
549 	BUG_ON(num_registers > US_IOBUF_SIZE/2);
550 
551 	for (i=0; i<20; i++) {
552 
553 		/*
554 		 * The first time we send the full command, which consists
555 		 * of downloading the SCSI command followed by downloading
556 		 * the data via a write-and-test.  Any other time we only
557 		 * send the command to download the data -- the SCSI command
558 		 * is still 'active' in some sense in the device.
559 		 *
560 		 * We're only going to try sending the data 10 times. After
561 		 * that, we just return a failure.
562 		 */
563 
564 		if (i==0) {
565 			cmdlen = 16;
566 			/*
567 			 * Write to multiple registers
568 			 * Not really sure the 0x07, 0x17, 0xfc, 0xe7 is
569 			 * necessary here, but that's what came out of the
570 			 * trace every single time.
571 			 */
572 			command[0] = 0x40;
573 			command[1] = access | USBAT_CMD_WRITE_REGS;
574 			command[2] = 0x07;
575 			command[3] = 0x17;
576 			command[4] = 0xFC;
577 			command[5] = 0xE7;
578 			command[6] = LSB_of(num_registers*2);
579 			command[7] = MSB_of(num_registers*2);
580 		} else
581 			cmdlen = 8;
582 
583 		/* Conditionally read or write blocks */
584 		command[cmdlen-8] = (direction==DMA_TO_DEVICE ? 0x40 : 0xC0);
585 		command[cmdlen-7] = access |
586 				(direction==DMA_TO_DEVICE ?
587 				 USBAT_CMD_COND_WRITE_BLOCK : USBAT_CMD_COND_READ_BLOCK);
588 		command[cmdlen-6] = data_reg;
589 		command[cmdlen-5] = status_reg;
590 		command[cmdlen-4] = timeout;
591 		command[cmdlen-3] = qualifier;
592 		command[cmdlen-2] = LSB_of(len);
593 		command[cmdlen-1] = MSB_of(len);
594 
595 		result = usbat_execute_command(us, command, cmdlen);
596 
597 		if (result != USB_STOR_XFER_GOOD)
598 			return USB_STOR_TRANSPORT_ERROR;
599 
600 		if (i==0) {
601 
602 			for (j=0; j<num_registers; j++) {
603 				data[j<<1] = registers[j];
604 				data[1+(j<<1)] = data_out[j];
605 			}
606 
607 			result = usbat_bulk_write(us, data, num_registers*2, 0);
608 			if (result != USB_STOR_XFER_GOOD)
609 				return USB_STOR_TRANSPORT_ERROR;
610 
611 		}
612 
613 		result = usb_stor_bulk_transfer_sg(us,
614 			pipe, buf, len, use_sg, NULL);
615 
616 		/*
617 		 * If we get a stall on the bulk download, we'll retry
618 		 * the bulk download -- but not the SCSI command because
619 		 * in some sense the SCSI command is still 'active' and
620 		 * waiting for the data. Don't ask me why this should be;
621 		 * I'm only following what the Windoze driver did.
622 		 *
623 		 * Note that a stall for the test-and-read/write command means
624 		 * that the test failed. In this case we're testing to make
625 		 * sure that the device is error-free
626 		 * (i.e. bit 0 -- CHK -- of status is 0). The most likely
627 		 * hypothesis is that the USBAT chip somehow knows what
628 		 * the device will accept, but doesn't give the device any
629 		 * data until all data is received. Thus, the device would
630 		 * still be waiting for the first byte of data if a stall
631 		 * occurs, even if the stall implies that some data was
632 		 * transferred.
633 		 */
634 
635 		if (result == USB_STOR_XFER_SHORT ||
636 				result == USB_STOR_XFER_STALLED) {
637 
638 			/*
639 			 * If we're reading and we stalled, then clear
640 			 * the bulk output pipe only the first time.
641 			 */
642 
643 			if (direction==DMA_FROM_DEVICE && i==0) {
644 				if (usb_stor_clear_halt(us,
645 						us->send_bulk_pipe) < 0)
646 					return USB_STOR_TRANSPORT_ERROR;
647 			}
648 
649 			/*
650 			 * Read status: is the device angry, or just busy?
651 			 */
652 
653  			result = usbat_read(us, USBAT_ATA,
654 				direction==DMA_TO_DEVICE ?
655 					USBAT_ATA_STATUS : USBAT_ATA_ALTSTATUS,
656 				status);
657 
658 			if (result!=USB_STOR_XFER_GOOD)
659 				return USB_STOR_TRANSPORT_ERROR;
660 			if (*status & 0x01) /* check condition */
661 				return USB_STOR_TRANSPORT_FAILED;
662 			if (*status & 0x20) /* device fault */
663 				return USB_STOR_TRANSPORT_FAILED;
664 
665 			usb_stor_dbg(us, "Redoing %s\n",
666 				     direction == DMA_TO_DEVICE
667 				     ? "write" : "read");
668 
669 		} else if (result != USB_STOR_XFER_GOOD)
670 			return USB_STOR_TRANSPORT_ERROR;
671 		else
672 			return usbat_wait_not_busy(us, minutes);
673 
674 	}
675 
676 	usb_stor_dbg(us, "Bummer! %s bulk data 20 times failed\n",
677 		     direction == DMA_TO_DEVICE ? "Writing" : "Reading");
678 
679 	return USB_STOR_TRANSPORT_FAILED;
680 }
681 
682 /*
683  * Write to multiple registers:
684  * Allows us to write specific data to any registers. The data to be written
685  * gets packed in this sequence: reg0, data0, reg1, data1, ..., regN, dataN
686  * which gets sent through bulk out.
687  * Not designed for large transfers of data!
688  */
689 static int usbat_multiple_write(struct us_data *us,
690 				unsigned char *registers,
691 				unsigned char *data_out,
692 				unsigned short num_registers)
693 {
694 	int i, result;
695 	unsigned char *data = us->iobuf;
696 	unsigned char *command = us->iobuf;
697 
698 	BUG_ON(num_registers > US_IOBUF_SIZE/2);
699 
700 	/* Write to multiple registers, ATA access */
701 	command[0] = 0x40;
702 	command[1] = USBAT_ATA | USBAT_CMD_WRITE_REGS;
703 
704 	/* No relevance */
705 	command[2] = 0;
706 	command[3] = 0;
707 	command[4] = 0;
708 	command[5] = 0;
709 
710 	/* Number of bytes to be transferred (incl. addresses and data) */
711 	command[6] = LSB_of(num_registers*2);
712 	command[7] = MSB_of(num_registers*2);
713 
714 	/* The setup command */
715 	result = usbat_execute_command(us, command, 8);
716 	if (result != USB_STOR_XFER_GOOD)
717 		return USB_STOR_TRANSPORT_ERROR;
718 
719 	/* Create the reg/data, reg/data sequence */
720 	for (i=0; i<num_registers; i++) {
721 		data[i<<1] = registers[i];
722 		data[1+(i<<1)] = data_out[i];
723 	}
724 
725 	/* Send the data */
726 	result = usbat_bulk_write(us, data, num_registers*2, 0);
727 	if (result != USB_STOR_XFER_GOOD)
728 		return USB_STOR_TRANSPORT_ERROR;
729 
730 	if (usbat_get_device_type(us) == USBAT_DEV_HP8200)
731 		return usbat_wait_not_busy(us, 0);
732 	else
733 		return USB_STOR_TRANSPORT_GOOD;
734 }
735 
736 /*
737  * Conditionally read blocks from device:
738  * Allows us to read blocks from a specific data register, based upon the
739  * condition that a status register can be successfully masked with a status
740  * qualifier. If this condition is not initially met, the read will wait
741  * up until a maximum amount of time has elapsed, as specified by timeout.
742  * The read will start when the condition is met, otherwise the command aborts.
743  *
744  * The qualifier defined here is not the value that is masked, it defines
745  * conditions for the write to take place. The actual masked qualifier (and
746  * other related details) are defined beforehand with _set_shuttle_features().
747  */
748 static int usbat_read_blocks(struct us_data *us,
749 			     void* buffer,
750 			     int len,
751 			     int use_sg)
752 {
753 	int result;
754 	unsigned char *command = us->iobuf;
755 
756 	command[0] = 0xC0;
757 	command[1] = USBAT_ATA | USBAT_CMD_COND_READ_BLOCK;
758 	command[2] = USBAT_ATA_DATA;
759 	command[3] = USBAT_ATA_STATUS;
760 	command[4] = 0xFD; /* Timeout (ms); */
761 	command[5] = USBAT_QUAL_FCQ;
762 	command[6] = LSB_of(len);
763 	command[7] = MSB_of(len);
764 
765 	/* Multiple block read setup command */
766 	result = usbat_execute_command(us, command, 8);
767 	if (result != USB_STOR_XFER_GOOD)
768 		return USB_STOR_TRANSPORT_FAILED;
769 
770 	/* Read the blocks we just asked for */
771 	result = usbat_bulk_read(us, buffer, len, use_sg);
772 	if (result != USB_STOR_XFER_GOOD)
773 		return USB_STOR_TRANSPORT_FAILED;
774 
775 	return USB_STOR_TRANSPORT_GOOD;
776 }
777 
778 /*
779  * Conditionally write blocks to device:
780  * Allows us to write blocks to a specific data register, based upon the
781  * condition that a status register can be successfully masked with a status
782  * qualifier. If this condition is not initially met, the write will wait
783  * up until a maximum amount of time has elapsed, as specified by timeout.
784  * The read will start when the condition is met, otherwise the command aborts.
785  *
786  * The qualifier defined here is not the value that is masked, it defines
787  * conditions for the write to take place. The actual masked qualifier (and
788  * other related details) are defined beforehand with _set_shuttle_features().
789  */
790 static int usbat_write_blocks(struct us_data *us,
791 			      void* buffer,
792 			      int len,
793 			      int use_sg)
794 {
795 	int result;
796 	unsigned char *command = us->iobuf;
797 
798 	command[0] = 0x40;
799 	command[1] = USBAT_ATA | USBAT_CMD_COND_WRITE_BLOCK;
800 	command[2] = USBAT_ATA_DATA;
801 	command[3] = USBAT_ATA_STATUS;
802 	command[4] = 0xFD; /* Timeout (ms) */
803 	command[5] = USBAT_QUAL_FCQ;
804 	command[6] = LSB_of(len);
805 	command[7] = MSB_of(len);
806 
807 	/* Multiple block write setup command */
808 	result = usbat_execute_command(us, command, 8);
809 	if (result != USB_STOR_XFER_GOOD)
810 		return USB_STOR_TRANSPORT_FAILED;
811 
812 	/* Write the data */
813 	result = usbat_bulk_write(us, buffer, len, use_sg);
814 	if (result != USB_STOR_XFER_GOOD)
815 		return USB_STOR_TRANSPORT_FAILED;
816 
817 	return USB_STOR_TRANSPORT_GOOD;
818 }
819 
820 /*
821  * Read the User IO register
822  */
823 static int usbat_read_user_io(struct us_data *us, unsigned char *data_flags)
824 {
825 	int result;
826 
827 	result = usb_stor_ctrl_transfer(us,
828 		us->recv_ctrl_pipe,
829 		USBAT_CMD_UIO,
830 		0xC0,
831 		0,
832 		0,
833 		data_flags,
834 		USBAT_UIO_READ);
835 
836 	usb_stor_dbg(us, "UIO register reads %02X\n", *data_flags);
837 
838 	return result;
839 }
840 
841 /*
842  * Write to the User IO register
843  */
844 static int usbat_write_user_io(struct us_data *us,
845 			       unsigned char enable_flags,
846 			       unsigned char data_flags)
847 {
848 	return usb_stor_ctrl_transfer(us,
849 		us->send_ctrl_pipe,
850 		USBAT_CMD_UIO,
851 		0x40,
852 		short_pack(enable_flags, data_flags),
853 		0,
854 		NULL,
855 		USBAT_UIO_WRITE);
856 }
857 
858 /*
859  * Reset the device
860  * Often needed on media change.
861  */
862 static int usbat_device_reset(struct us_data *us)
863 {
864 	int rc;
865 
866 	/*
867 	 * Reset peripheral, enable peripheral control signals
868 	 * (bring reset signal up)
869 	 */
870 	rc = usbat_write_user_io(us,
871 							 USBAT_UIO_DRVRST | USBAT_UIO_OE1 | USBAT_UIO_OE0,
872 							 USBAT_UIO_EPAD | USBAT_UIO_1);
873 	if (rc != USB_STOR_XFER_GOOD)
874 		return USB_STOR_TRANSPORT_ERROR;
875 
876 	/*
877 	 * Enable peripheral control signals
878 	 * (bring reset signal down)
879 	 */
880 	rc = usbat_write_user_io(us,
881 							 USBAT_UIO_OE1  | USBAT_UIO_OE0,
882 							 USBAT_UIO_EPAD | USBAT_UIO_1);
883 	if (rc != USB_STOR_XFER_GOOD)
884 		return USB_STOR_TRANSPORT_ERROR;
885 
886 	return USB_STOR_TRANSPORT_GOOD;
887 }
888 
889 /*
890  * Enable card detect
891  */
892 static int usbat_device_enable_cdt(struct us_data *us)
893 {
894 	int rc;
895 
896 	/* Enable peripheral control signals and card detect */
897 	rc = usbat_write_user_io(us,
898 							 USBAT_UIO_ACKD | USBAT_UIO_OE1  | USBAT_UIO_OE0,
899 							 USBAT_UIO_EPAD | USBAT_UIO_1);
900 	if (rc != USB_STOR_XFER_GOOD)
901 		return USB_STOR_TRANSPORT_ERROR;
902 
903 	return USB_STOR_TRANSPORT_GOOD;
904 }
905 
906 /*
907  * Determine if media is present.
908  */
909 static int usbat_flash_check_media_present(struct us_data *us,
910 					   unsigned char *uio)
911 {
912 	if (*uio & USBAT_UIO_UI0) {
913 		usb_stor_dbg(us, "no media detected\n");
914 		return USBAT_FLASH_MEDIA_NONE;
915 	}
916 
917 	return USBAT_FLASH_MEDIA_CF;
918 }
919 
920 /*
921  * Determine if media has changed since last operation
922  */
923 static int usbat_flash_check_media_changed(struct us_data *us,
924 					   unsigned char *uio)
925 {
926 	if (*uio & USBAT_UIO_0) {
927 		usb_stor_dbg(us, "media change detected\n");
928 		return USBAT_FLASH_MEDIA_CHANGED;
929 	}
930 
931 	return USBAT_FLASH_MEDIA_SAME;
932 }
933 
934 /*
935  * Check for media change / no media and handle the situation appropriately
936  */
937 static int usbat_flash_check_media(struct us_data *us,
938 				   struct usbat_info *info)
939 {
940 	int rc;
941 	unsigned char *uio = us->iobuf;
942 
943 	rc = usbat_read_user_io(us, uio);
944 	if (rc != USB_STOR_XFER_GOOD)
945 		return USB_STOR_TRANSPORT_ERROR;
946 
947 	/* Check for media existence */
948 	rc = usbat_flash_check_media_present(us, uio);
949 	if (rc == USBAT_FLASH_MEDIA_NONE) {
950 		info->sense_key = 0x02;
951 		info->sense_asc = 0x3A;
952 		info->sense_ascq = 0x00;
953 		return USB_STOR_TRANSPORT_FAILED;
954 	}
955 
956 	/* Check for media change */
957 	rc = usbat_flash_check_media_changed(us, uio);
958 	if (rc == USBAT_FLASH_MEDIA_CHANGED) {
959 
960 		/* Reset and re-enable card detect */
961 		rc = usbat_device_reset(us);
962 		if (rc != USB_STOR_TRANSPORT_GOOD)
963 			return rc;
964 		rc = usbat_device_enable_cdt(us);
965 		if (rc != USB_STOR_TRANSPORT_GOOD)
966 			return rc;
967 
968 		msleep(50);
969 
970 		rc = usbat_read_user_io(us, uio);
971 		if (rc != USB_STOR_XFER_GOOD)
972 			return USB_STOR_TRANSPORT_ERROR;
973 
974 		info->sense_key = UNIT_ATTENTION;
975 		info->sense_asc = 0x28;
976 		info->sense_ascq = 0x00;
977 		return USB_STOR_TRANSPORT_FAILED;
978 	}
979 
980 	return USB_STOR_TRANSPORT_GOOD;
981 }
982 
983 /*
984  * Determine whether we are controlling a flash-based reader/writer,
985  * or a HP8200-based CD drive.
986  * Sets transport functions as appropriate.
987  */
988 static int usbat_identify_device(struct us_data *us,
989 				 struct usbat_info *info)
990 {
991 	int rc;
992 	unsigned char status;
993 
994 	if (!us || !info)
995 		return USB_STOR_TRANSPORT_ERROR;
996 
997 	rc = usbat_device_reset(us);
998 	if (rc != USB_STOR_TRANSPORT_GOOD)
999 		return rc;
1000 	msleep(500);
1001 
1002 	/*
1003 	 * In attempt to distinguish between HP CDRW's and Flash readers, we now
1004 	 * execute the IDENTIFY PACKET DEVICE command. On ATA devices (i.e. flash
1005 	 * readers), this command should fail with error. On ATAPI devices (i.e.
1006 	 * CDROM drives), it should succeed.
1007 	 */
1008 	rc = usbat_write(us, USBAT_ATA, USBAT_ATA_CMD, 0xA1);
1009  	if (rc != USB_STOR_XFER_GOOD)
1010  		return USB_STOR_TRANSPORT_ERROR;
1011 
1012 	rc = usbat_get_status(us, &status);
1013  	if (rc != USB_STOR_XFER_GOOD)
1014  		return USB_STOR_TRANSPORT_ERROR;
1015 
1016 	/* Check for error bit, or if the command 'fell through' */
1017 	if (status == 0xA1 || !(status & 0x01)) {
1018 		/* Device is HP 8200 */
1019 		usb_stor_dbg(us, "Detected HP8200 CDRW\n");
1020 		info->devicetype = USBAT_DEV_HP8200;
1021 	} else {
1022 		/* Device is a CompactFlash reader/writer */
1023 		usb_stor_dbg(us, "Detected Flash reader/writer\n");
1024 		info->devicetype = USBAT_DEV_FLASH;
1025 	}
1026 
1027 	return USB_STOR_TRANSPORT_GOOD;
1028 }
1029 
1030 /*
1031  * Set the transport function based on the device type
1032  */
1033 static int usbat_set_transport(struct us_data *us,
1034 			       struct usbat_info *info,
1035 			       int devicetype)
1036 {
1037 
1038 	if (!info->devicetype)
1039 		info->devicetype = devicetype;
1040 
1041 	if (!info->devicetype)
1042 		usbat_identify_device(us, info);
1043 
1044 	switch (info->devicetype) {
1045 	default:
1046 		return USB_STOR_TRANSPORT_ERROR;
1047 
1048 	case  USBAT_DEV_HP8200:
1049 		us->transport = usbat_hp8200e_transport;
1050 		break;
1051 
1052 	case USBAT_DEV_FLASH:
1053 		us->transport = usbat_flash_transport;
1054 		break;
1055 	}
1056 
1057 	return 0;
1058 }
1059 
1060 /*
1061  * Read the media capacity
1062  */
1063 static int usbat_flash_get_sector_count(struct us_data *us,
1064 					struct usbat_info *info)
1065 {
1066 	unsigned char registers[3] = {
1067 		USBAT_ATA_SECCNT,
1068 		USBAT_ATA_DEVICE,
1069 		USBAT_ATA_CMD,
1070 	};
1071 	unsigned char  command[3] = { 0x01, 0xA0, 0xEC };
1072 	unsigned char *reply;
1073 	unsigned char status;
1074 	int rc;
1075 
1076 	if (!us || !info)
1077 		return USB_STOR_TRANSPORT_ERROR;
1078 
1079 	reply = kmalloc(512, GFP_NOIO);
1080 	if (!reply)
1081 		return USB_STOR_TRANSPORT_ERROR;
1082 
1083 	/* ATA command : IDENTIFY DEVICE */
1084 	rc = usbat_multiple_write(us, registers, command, 3);
1085 	if (rc != USB_STOR_XFER_GOOD) {
1086 		usb_stor_dbg(us, "Gah! identify_device failed\n");
1087 		rc = USB_STOR_TRANSPORT_ERROR;
1088 		goto leave;
1089 	}
1090 
1091 	/* Read device status */
1092 	if (usbat_get_status(us, &status) != USB_STOR_XFER_GOOD) {
1093 		rc = USB_STOR_TRANSPORT_ERROR;
1094 		goto leave;
1095 	}
1096 
1097 	msleep(100);
1098 
1099 	/* Read the device identification data */
1100 	rc = usbat_read_block(us, reply, 512, 0);
1101 	if (rc != USB_STOR_TRANSPORT_GOOD)
1102 		goto leave;
1103 
1104 	info->sectors = ((u32)(reply[117]) << 24) |
1105 		((u32)(reply[116]) << 16) |
1106 		((u32)(reply[115]) <<  8) |
1107 		((u32)(reply[114])      );
1108 
1109 	rc = USB_STOR_TRANSPORT_GOOD;
1110 
1111  leave:
1112 	kfree(reply);
1113 	return rc;
1114 }
1115 
1116 /*
1117  * Read data from device
1118  */
1119 static int usbat_flash_read_data(struct us_data *us,
1120 								 struct usbat_info *info,
1121 								 u32 sector,
1122 								 u32 sectors)
1123 {
1124 	unsigned char registers[7] = {
1125 		USBAT_ATA_FEATURES,
1126 		USBAT_ATA_SECCNT,
1127 		USBAT_ATA_SECNUM,
1128 		USBAT_ATA_LBA_ME,
1129 		USBAT_ATA_LBA_HI,
1130 		USBAT_ATA_DEVICE,
1131 		USBAT_ATA_STATUS,
1132 	};
1133 	unsigned char command[7];
1134 	unsigned char *buffer;
1135 	unsigned char  thistime;
1136 	unsigned int totallen, alloclen;
1137 	int len, result;
1138 	unsigned int sg_offset = 0;
1139 	struct scatterlist *sg = NULL;
1140 
1141 	result = usbat_flash_check_media(us, info);
1142 	if (result != USB_STOR_TRANSPORT_GOOD)
1143 		return result;
1144 
1145 	/*
1146 	 * we're working in LBA mode.  according to the ATA spec,
1147 	 * we can support up to 28-bit addressing.  I don't know if Jumpshot
1148 	 * supports beyond 24-bit addressing.  It's kind of hard to test
1149 	 * since it requires > 8GB CF card.
1150 	 */
1151 
1152 	if (sector > 0x0FFFFFFF)
1153 		return USB_STOR_TRANSPORT_ERROR;
1154 
1155 	totallen = sectors * info->ssize;
1156 
1157 	/*
1158 	 * Since we don't read more than 64 KB at a time, we have to create
1159 	 * a bounce buffer and move the data a piece at a time between the
1160 	 * bounce buffer and the actual transfer buffer.
1161 	 */
1162 
1163 	alloclen = min(totallen, 65536u);
1164 	buffer = kmalloc(alloclen, GFP_NOIO);
1165 	if (buffer == NULL)
1166 		return USB_STOR_TRANSPORT_ERROR;
1167 
1168 	do {
1169 		/*
1170 		 * loop, never allocate or transfer more than 64k at once
1171 		 * (min(128k, 255*info->ssize) is the real limit)
1172 		 */
1173 		len = min(totallen, alloclen);
1174 		thistime = (len / info->ssize) & 0xff;
1175 
1176 		/* ATA command 0x20 (READ SECTORS) */
1177 		usbat_pack_ata_sector_cmd(command, thistime, sector, 0x20);
1178 
1179 		/* Write/execute ATA read command */
1180 		result = usbat_multiple_write(us, registers, command, 7);
1181 		if (result != USB_STOR_TRANSPORT_GOOD)
1182 			goto leave;
1183 
1184 		/* Read the data we just requested */
1185 		result = usbat_read_blocks(us, buffer, len, 0);
1186 		if (result != USB_STOR_TRANSPORT_GOOD)
1187 			goto leave;
1188 
1189 		usb_stor_dbg(us, "%d bytes\n", len);
1190 
1191 		/* Store the data in the transfer buffer */
1192 		usb_stor_access_xfer_buf(buffer, len, us->srb,
1193 					 &sg, &sg_offset, TO_XFER_BUF);
1194 
1195 		sector += thistime;
1196 		totallen -= len;
1197 	} while (totallen > 0);
1198 
1199 	kfree(buffer);
1200 	return USB_STOR_TRANSPORT_GOOD;
1201 
1202 leave:
1203 	kfree(buffer);
1204 	return USB_STOR_TRANSPORT_ERROR;
1205 }
1206 
1207 /*
1208  * Write data to device
1209  */
1210 static int usbat_flash_write_data(struct us_data *us,
1211 								  struct usbat_info *info,
1212 								  u32 sector,
1213 								  u32 sectors)
1214 {
1215 	unsigned char registers[7] = {
1216 		USBAT_ATA_FEATURES,
1217 		USBAT_ATA_SECCNT,
1218 		USBAT_ATA_SECNUM,
1219 		USBAT_ATA_LBA_ME,
1220 		USBAT_ATA_LBA_HI,
1221 		USBAT_ATA_DEVICE,
1222 		USBAT_ATA_STATUS,
1223 	};
1224 	unsigned char command[7];
1225 	unsigned char *buffer;
1226 	unsigned char  thistime;
1227 	unsigned int totallen, alloclen;
1228 	int len, result;
1229 	unsigned int sg_offset = 0;
1230 	struct scatterlist *sg = NULL;
1231 
1232 	result = usbat_flash_check_media(us, info);
1233 	if (result != USB_STOR_TRANSPORT_GOOD)
1234 		return result;
1235 
1236 	/*
1237 	 * we're working in LBA mode.  according to the ATA spec,
1238 	 * we can support up to 28-bit addressing.  I don't know if the device
1239 	 * supports beyond 24-bit addressing.  It's kind of hard to test
1240 	 * since it requires > 8GB media.
1241 	 */
1242 
1243 	if (sector > 0x0FFFFFFF)
1244 		return USB_STOR_TRANSPORT_ERROR;
1245 
1246 	totallen = sectors * info->ssize;
1247 
1248 	/*
1249 	 * Since we don't write more than 64 KB at a time, we have to create
1250 	 * a bounce buffer and move the data a piece at a time between the
1251 	 * bounce buffer and the actual transfer buffer.
1252 	 */
1253 
1254 	alloclen = min(totallen, 65536u);
1255 	buffer = kmalloc(alloclen, GFP_NOIO);
1256 	if (buffer == NULL)
1257 		return USB_STOR_TRANSPORT_ERROR;
1258 
1259 	do {
1260 		/*
1261 		 * loop, never allocate or transfer more than 64k at once
1262 		 * (min(128k, 255*info->ssize) is the real limit)
1263 		 */
1264 		len = min(totallen, alloclen);
1265 		thistime = (len / info->ssize) & 0xff;
1266 
1267 		/* Get the data from the transfer buffer */
1268 		usb_stor_access_xfer_buf(buffer, len, us->srb,
1269 					 &sg, &sg_offset, FROM_XFER_BUF);
1270 
1271 		/* ATA command 0x30 (WRITE SECTORS) */
1272 		usbat_pack_ata_sector_cmd(command, thistime, sector, 0x30);
1273 
1274 		/* Write/execute ATA write command */
1275 		result = usbat_multiple_write(us, registers, command, 7);
1276 		if (result != USB_STOR_TRANSPORT_GOOD)
1277 			goto leave;
1278 
1279 		/* Write the data */
1280 		result = usbat_write_blocks(us, buffer, len, 0);
1281 		if (result != USB_STOR_TRANSPORT_GOOD)
1282 			goto leave;
1283 
1284 		sector += thistime;
1285 		totallen -= len;
1286 	} while (totallen > 0);
1287 
1288 	kfree(buffer);
1289 	return result;
1290 
1291 leave:
1292 	kfree(buffer);
1293 	return USB_STOR_TRANSPORT_ERROR;
1294 }
1295 
1296 /*
1297  * Squeeze a potentially huge (> 65535 byte) read10 command into
1298  * a little ( <= 65535 byte) ATAPI pipe
1299  */
1300 static int usbat_hp8200e_handle_read10(struct us_data *us,
1301 				       unsigned char *registers,
1302 				       unsigned char *data,
1303 				       struct scsi_cmnd *srb)
1304 {
1305 	int result = USB_STOR_TRANSPORT_GOOD;
1306 	unsigned char *buffer;
1307 	unsigned int len;
1308 	unsigned int sector;
1309 	unsigned int sg_offset = 0;
1310 	struct scatterlist *sg = NULL;
1311 
1312 	usb_stor_dbg(us, "transfersize %d\n", srb->transfersize);
1313 
1314 	if (scsi_bufflen(srb) < 0x10000) {
1315 
1316 		result = usbat_hp8200e_rw_block_test(us, USBAT_ATA,
1317 			registers, data, 19,
1318 			USBAT_ATA_DATA, USBAT_ATA_STATUS, 0xFD,
1319 			(USBAT_QUAL_FCQ | USBAT_QUAL_ALQ),
1320 			DMA_FROM_DEVICE,
1321 			scsi_sglist(srb),
1322 			scsi_bufflen(srb), scsi_sg_count(srb), 1);
1323 
1324 		return result;
1325 	}
1326 
1327 	/*
1328 	 * Since we're requesting more data than we can handle in
1329 	 * a single read command (max is 64k-1), we will perform
1330 	 * multiple reads, but each read must be in multiples of
1331 	 * a sector.  Luckily the sector size is in srb->transfersize
1332 	 * (see linux/drivers/scsi/sr.c).
1333 	 */
1334 
1335 	if (data[7+0] == GPCMD_READ_CD) {
1336 		len = short_pack(data[7+9], data[7+8]);
1337 		len <<= 16;
1338 		len |= data[7+7];
1339 		usb_stor_dbg(us, "GPCMD_READ_CD: len %d\n", len);
1340 		srb->transfersize = scsi_bufflen(srb)/len;
1341 	}
1342 
1343 	if (!srb->transfersize)  {
1344 		srb->transfersize = 2048; /* A guess */
1345 		usb_stor_dbg(us, "transfersize 0, forcing %d\n",
1346 			     srb->transfersize);
1347 	}
1348 
1349 	/*
1350 	 * Since we only read in one block at a time, we have to create
1351 	 * a bounce buffer and move the data a piece at a time between the
1352 	 * bounce buffer and the actual transfer buffer.
1353 	 */
1354 
1355 	len = (65535/srb->transfersize) * srb->transfersize;
1356 	usb_stor_dbg(us, "Max read is %d bytes\n", len);
1357 	len = min(len, scsi_bufflen(srb));
1358 	buffer = kmalloc(len, GFP_NOIO);
1359 	if (buffer == NULL) /* bloody hell! */
1360 		return USB_STOR_TRANSPORT_FAILED;
1361 	sector = short_pack(data[7+3], data[7+2]);
1362 	sector <<= 16;
1363 	sector |= short_pack(data[7+5], data[7+4]);
1364 	transferred = 0;
1365 
1366 	while (transferred != scsi_bufflen(srb)) {
1367 
1368 		if (len > scsi_bufflen(srb) - transferred)
1369 			len = scsi_bufflen(srb) - transferred;
1370 
1371 		data[3] = len&0xFF; 	  /* (cylL) = expected length (L) */
1372 		data[4] = (len>>8)&0xFF;  /* (cylH) = expected length (H) */
1373 
1374 		/* Fix up the SCSI command sector and num sectors */
1375 
1376 		data[7+2] = MSB_of(sector>>16); /* SCSI command sector */
1377 		data[7+3] = LSB_of(sector>>16);
1378 		data[7+4] = MSB_of(sector&0xFFFF);
1379 		data[7+5] = LSB_of(sector&0xFFFF);
1380 		if (data[7+0] == GPCMD_READ_CD)
1381 			data[7+6] = 0;
1382 		data[7+7] = MSB_of(len / srb->transfersize); /* SCSI command */
1383 		data[7+8] = LSB_of(len / srb->transfersize); /* num sectors */
1384 
1385 		result = usbat_hp8200e_rw_block_test(us, USBAT_ATA,
1386 			registers, data, 19,
1387 			USBAT_ATA_DATA, USBAT_ATA_STATUS, 0xFD,
1388 			(USBAT_QUAL_FCQ | USBAT_QUAL_ALQ),
1389 			DMA_FROM_DEVICE,
1390 			buffer,
1391 			len, 0, 1);
1392 
1393 		if (result != USB_STOR_TRANSPORT_GOOD)
1394 			break;
1395 
1396 		/* Store the data in the transfer buffer */
1397 		usb_stor_access_xfer_buf(buffer, len, srb,
1398 				 &sg, &sg_offset, TO_XFER_BUF);
1399 
1400 		/* Update the amount transferred and the sector number */
1401 
1402 		transferred += len;
1403 		sector += len / srb->transfersize;
1404 
1405 	} /* while transferred != scsi_bufflen(srb) */
1406 
1407 	kfree(buffer);
1408 	return result;
1409 }
1410 
1411 static int usbat_select_and_test_registers(struct us_data *us)
1412 {
1413 	int selector;
1414 	unsigned char *status = us->iobuf;
1415 
1416 	/* try device = master, then device = slave. */
1417 	for (selector = 0xA0; selector <= 0xB0; selector += 0x10) {
1418 		if (usbat_write(us, USBAT_ATA, USBAT_ATA_DEVICE, selector) !=
1419 				USB_STOR_XFER_GOOD)
1420 			return USB_STOR_TRANSPORT_ERROR;
1421 
1422 		if (usbat_read(us, USBAT_ATA, USBAT_ATA_STATUS, status) !=
1423 				USB_STOR_XFER_GOOD)
1424 			return USB_STOR_TRANSPORT_ERROR;
1425 
1426 		if (usbat_read(us, USBAT_ATA, USBAT_ATA_DEVICE, status) !=
1427 				USB_STOR_XFER_GOOD)
1428 			return USB_STOR_TRANSPORT_ERROR;
1429 
1430 		if (usbat_read(us, USBAT_ATA, USBAT_ATA_LBA_ME, status) !=
1431 				USB_STOR_XFER_GOOD)
1432 			return USB_STOR_TRANSPORT_ERROR;
1433 
1434 		if (usbat_read(us, USBAT_ATA, USBAT_ATA_LBA_HI, status) !=
1435 				USB_STOR_XFER_GOOD)
1436 			return USB_STOR_TRANSPORT_ERROR;
1437 
1438 		if (usbat_write(us, USBAT_ATA, USBAT_ATA_LBA_ME, 0x55) !=
1439 				USB_STOR_XFER_GOOD)
1440 			return USB_STOR_TRANSPORT_ERROR;
1441 
1442 		if (usbat_write(us, USBAT_ATA, USBAT_ATA_LBA_HI, 0xAA) !=
1443 				USB_STOR_XFER_GOOD)
1444 			return USB_STOR_TRANSPORT_ERROR;
1445 
1446 		if (usbat_read(us, USBAT_ATA, USBAT_ATA_LBA_ME, status) !=
1447 				USB_STOR_XFER_GOOD)
1448 			return USB_STOR_TRANSPORT_ERROR;
1449 
1450 		if (usbat_read(us, USBAT_ATA, USBAT_ATA_LBA_ME, status) !=
1451 				USB_STOR_XFER_GOOD)
1452 			return USB_STOR_TRANSPORT_ERROR;
1453 	}
1454 
1455 	return USB_STOR_TRANSPORT_GOOD;
1456 }
1457 
1458 /*
1459  * Initialize the USBAT processor and the storage device
1460  */
1461 static int init_usbat(struct us_data *us, int devicetype)
1462 {
1463 	int rc;
1464 	struct usbat_info *info;
1465 	unsigned char subcountH = USBAT_ATA_LBA_HI;
1466 	unsigned char subcountL = USBAT_ATA_LBA_ME;
1467 	unsigned char *status = us->iobuf;
1468 
1469 	us->extra = kzalloc(sizeof(struct usbat_info), GFP_NOIO);
1470 	if (!us->extra)
1471 		return 1;
1472 
1473 	info = (struct usbat_info *) (us->extra);
1474 
1475 	/* Enable peripheral control signals */
1476 	rc = usbat_write_user_io(us,
1477 				 USBAT_UIO_OE1 | USBAT_UIO_OE0,
1478 				 USBAT_UIO_EPAD | USBAT_UIO_1);
1479 	if (rc != USB_STOR_XFER_GOOD)
1480 		return USB_STOR_TRANSPORT_ERROR;
1481 
1482 	usb_stor_dbg(us, "INIT 1\n");
1483 
1484 	msleep(2000);
1485 
1486 	rc = usbat_read_user_io(us, status);
1487 	if (rc != USB_STOR_TRANSPORT_GOOD)
1488 		return rc;
1489 
1490 	usb_stor_dbg(us, "INIT 2\n");
1491 
1492 	rc = usbat_read_user_io(us, status);
1493 	if (rc != USB_STOR_XFER_GOOD)
1494 		return USB_STOR_TRANSPORT_ERROR;
1495 
1496 	rc = usbat_read_user_io(us, status);
1497 	if (rc != USB_STOR_XFER_GOOD)
1498 		return USB_STOR_TRANSPORT_ERROR;
1499 
1500 	usb_stor_dbg(us, "INIT 3\n");
1501 
1502 	rc = usbat_select_and_test_registers(us);
1503 	if (rc != USB_STOR_TRANSPORT_GOOD)
1504 		return rc;
1505 
1506 	usb_stor_dbg(us, "INIT 4\n");
1507 
1508 	rc = usbat_read_user_io(us, status);
1509 	if (rc != USB_STOR_XFER_GOOD)
1510 		return USB_STOR_TRANSPORT_ERROR;
1511 
1512 	usb_stor_dbg(us, "INIT 5\n");
1513 
1514 	/* Enable peripheral control signals and card detect */
1515 	rc = usbat_device_enable_cdt(us);
1516 	if (rc != USB_STOR_TRANSPORT_GOOD)
1517 		return rc;
1518 
1519 	usb_stor_dbg(us, "INIT 6\n");
1520 
1521 	rc = usbat_read_user_io(us, status);
1522 	if (rc != USB_STOR_XFER_GOOD)
1523 		return USB_STOR_TRANSPORT_ERROR;
1524 
1525 	usb_stor_dbg(us, "INIT 7\n");
1526 
1527 	msleep(1400);
1528 
1529 	rc = usbat_read_user_io(us, status);
1530 	if (rc != USB_STOR_XFER_GOOD)
1531 		return USB_STOR_TRANSPORT_ERROR;
1532 
1533 	usb_stor_dbg(us, "INIT 8\n");
1534 
1535 	rc = usbat_select_and_test_registers(us);
1536 	if (rc != USB_STOR_TRANSPORT_GOOD)
1537 		return rc;
1538 
1539 	usb_stor_dbg(us, "INIT 9\n");
1540 
1541 	/* At this point, we need to detect which device we are using */
1542 	if (usbat_set_transport(us, info, devicetype))
1543 		return USB_STOR_TRANSPORT_ERROR;
1544 
1545 	usb_stor_dbg(us, "INIT 10\n");
1546 
1547 	if (usbat_get_device_type(us) == USBAT_DEV_FLASH) {
1548 		subcountH = 0x02;
1549 		subcountL = 0x00;
1550 	}
1551 	rc = usbat_set_shuttle_features(us, (USBAT_FEAT_ETEN | USBAT_FEAT_ET2 | USBAT_FEAT_ET1),
1552 									0x00, 0x88, 0x08, subcountH, subcountL);
1553 	if (rc != USB_STOR_XFER_GOOD)
1554 		return USB_STOR_TRANSPORT_ERROR;
1555 
1556 	usb_stor_dbg(us, "INIT 11\n");
1557 
1558 	return USB_STOR_TRANSPORT_GOOD;
1559 }
1560 
1561 /*
1562  * Transport for the HP 8200e
1563  */
1564 static int usbat_hp8200e_transport(struct scsi_cmnd *srb, struct us_data *us)
1565 {
1566 	int result;
1567 	unsigned char *status = us->iobuf;
1568 	unsigned char registers[32];
1569 	unsigned char data[32];
1570 	unsigned int len;
1571 	int i;
1572 
1573 	len = scsi_bufflen(srb);
1574 
1575 	/*
1576 	 * Send A0 (ATA PACKET COMMAND).
1577 	 * Note: I guess we're never going to get any of the ATA
1578 	 * commands... just ATA Packet Commands.
1579  	 */
1580 
1581 	registers[0] = USBAT_ATA_FEATURES;
1582 	registers[1] = USBAT_ATA_SECCNT;
1583 	registers[2] = USBAT_ATA_SECNUM;
1584 	registers[3] = USBAT_ATA_LBA_ME;
1585 	registers[4] = USBAT_ATA_LBA_HI;
1586 	registers[5] = USBAT_ATA_DEVICE;
1587 	registers[6] = USBAT_ATA_CMD;
1588 	data[0] = 0x00;
1589 	data[1] = 0x00;
1590 	data[2] = 0x00;
1591 	data[3] = len&0xFF; 		/* (cylL) = expected length (L) */
1592 	data[4] = (len>>8)&0xFF; 	/* (cylH) = expected length (H) */
1593 	data[5] = 0xB0; 		/* (device sel) = slave */
1594 	data[6] = 0xA0; 		/* (command) = ATA PACKET COMMAND */
1595 
1596 	for (i=7; i<19; i++) {
1597 		registers[i] = 0x10;
1598 		data[i] = (i-7 >= srb->cmd_len) ? 0 : srb->cmnd[i-7];
1599 	}
1600 
1601 	result = usbat_get_status(us, status);
1602 	usb_stor_dbg(us, "Status = %02X\n", *status);
1603 	if (result != USB_STOR_XFER_GOOD)
1604 		return USB_STOR_TRANSPORT_ERROR;
1605 	if (srb->cmnd[0] == TEST_UNIT_READY)
1606 		transferred = 0;
1607 
1608 	if (srb->sc_data_direction == DMA_TO_DEVICE) {
1609 
1610 		result = usbat_hp8200e_rw_block_test(us, USBAT_ATA,
1611 			registers, data, 19,
1612 			USBAT_ATA_DATA, USBAT_ATA_STATUS, 0xFD,
1613 			(USBAT_QUAL_FCQ | USBAT_QUAL_ALQ),
1614 			DMA_TO_DEVICE,
1615 			scsi_sglist(srb),
1616 			len, scsi_sg_count(srb), 10);
1617 
1618 		if (result == USB_STOR_TRANSPORT_GOOD) {
1619 			transferred += len;
1620 			usb_stor_dbg(us, "Wrote %08X bytes\n", transferred);
1621 		}
1622 
1623 		return result;
1624 
1625 	} else if (srb->cmnd[0] == READ_10 ||
1626 		   srb->cmnd[0] == GPCMD_READ_CD) {
1627 
1628 		return usbat_hp8200e_handle_read10(us, registers, data, srb);
1629 
1630 	}
1631 
1632 	if (len > 0xFFFF) {
1633 		usb_stor_dbg(us, "Error: len = %08X... what do I do now?\n",
1634 			     len);
1635 		return USB_STOR_TRANSPORT_ERROR;
1636 	}
1637 
1638 	result = usbat_multiple_write(us, registers, data, 7);
1639 
1640 	if (result != USB_STOR_TRANSPORT_GOOD)
1641 		return result;
1642 
1643 	/*
1644 	 * Write the 12-byte command header.
1645 	 *
1646 	 * If the command is BLANK then set the timer for 75 minutes.
1647 	 * Otherwise set it for 10 minutes.
1648 	 *
1649 	 * NOTE: THE 8200 DOCUMENTATION STATES THAT BLANKING A CDRW
1650 	 * AT SPEED 4 IS UNRELIABLE!!!
1651 	 */
1652 
1653 	result = usbat_write_block(us, USBAT_ATA, srb->cmnd, 12,
1654 				   srb->cmnd[0] == GPCMD_BLANK ? 75 : 10, 0);
1655 
1656 	if (result != USB_STOR_TRANSPORT_GOOD)
1657 		return result;
1658 
1659 	/* If there is response data to be read in then do it here. */
1660 
1661 	if (len != 0 && (srb->sc_data_direction == DMA_FROM_DEVICE)) {
1662 
1663 		/* How many bytes to read in? Check cylL register */
1664 
1665 		if (usbat_read(us, USBAT_ATA, USBAT_ATA_LBA_ME, status) !=
1666 		    	USB_STOR_XFER_GOOD) {
1667 			return USB_STOR_TRANSPORT_ERROR;
1668 		}
1669 
1670 		if (len > 0xFF) { /* need to read cylH also */
1671 			len = *status;
1672 			if (usbat_read(us, USBAT_ATA, USBAT_ATA_LBA_HI, status) !=
1673 				    USB_STOR_XFER_GOOD) {
1674 				return USB_STOR_TRANSPORT_ERROR;
1675 			}
1676 			len += ((unsigned int) *status)<<8;
1677 		}
1678 		else
1679 			len = *status;
1680 
1681 
1682 		result = usbat_read_block(us, scsi_sglist(srb), len,
1683 			                                   scsi_sg_count(srb));
1684 	}
1685 
1686 	return result;
1687 }
1688 
1689 /*
1690  * Transport for USBAT02-based CompactFlash and similar storage devices
1691  */
1692 static int usbat_flash_transport(struct scsi_cmnd * srb, struct us_data *us)
1693 {
1694 	int rc;
1695 	struct usbat_info *info = (struct usbat_info *) (us->extra);
1696 	unsigned long block, blocks;
1697 	unsigned char *ptr = us->iobuf;
1698 	static unsigned char inquiry_response[36] = {
1699 		0x00, 0x80, 0x00, 0x01, 0x1F, 0x00, 0x00, 0x00
1700 	};
1701 
1702 	if (srb->cmnd[0] == INQUIRY) {
1703 		usb_stor_dbg(us, "INQUIRY - Returning bogus response\n");
1704 		memcpy(ptr, inquiry_response, sizeof(inquiry_response));
1705 		fill_inquiry_response(us, ptr, 36);
1706 		return USB_STOR_TRANSPORT_GOOD;
1707 	}
1708 
1709 	if (srb->cmnd[0] == READ_CAPACITY) {
1710 		rc = usbat_flash_check_media(us, info);
1711 		if (rc != USB_STOR_TRANSPORT_GOOD)
1712 			return rc;
1713 
1714 		rc = usbat_flash_get_sector_count(us, info);
1715 		if (rc != USB_STOR_TRANSPORT_GOOD)
1716 			return rc;
1717 
1718 		/* hard coded 512 byte sectors as per ATA spec */
1719 		info->ssize = 0x200;
1720 		usb_stor_dbg(us, "READ_CAPACITY: %ld sectors, %ld bytes per sector\n",
1721 			     info->sectors, info->ssize);
1722 
1723 		/*
1724 		 * build the reply
1725 		 * note: must return the sector number of the last sector,
1726 		 * *not* the total number of sectors
1727 		 */
1728 		((__be32 *) ptr)[0] = cpu_to_be32(info->sectors - 1);
1729 		((__be32 *) ptr)[1] = cpu_to_be32(info->ssize);
1730 		usb_stor_set_xfer_buf(ptr, 8, srb);
1731 
1732 		return USB_STOR_TRANSPORT_GOOD;
1733 	}
1734 
1735 	if (srb->cmnd[0] == MODE_SELECT_10) {
1736 		usb_stor_dbg(us, "Gah! MODE_SELECT_10\n");
1737 		return USB_STOR_TRANSPORT_ERROR;
1738 	}
1739 
1740 	if (srb->cmnd[0] == READ_10) {
1741 		block = ((u32)(srb->cmnd[2]) << 24) | ((u32)(srb->cmnd[3]) << 16) |
1742 				((u32)(srb->cmnd[4]) <<  8) | ((u32)(srb->cmnd[5]));
1743 
1744 		blocks = ((u32)(srb->cmnd[7]) << 8) | ((u32)(srb->cmnd[8]));
1745 
1746 		usb_stor_dbg(us, "READ_10: read block 0x%04lx  count %ld\n",
1747 			     block, blocks);
1748 		return usbat_flash_read_data(us, info, block, blocks);
1749 	}
1750 
1751 	if (srb->cmnd[0] == READ_12) {
1752 		/*
1753 		 * I don't think we'll ever see a READ_12 but support it anyway
1754 		 */
1755 		block = ((u32)(srb->cmnd[2]) << 24) | ((u32)(srb->cmnd[3]) << 16) |
1756 		        ((u32)(srb->cmnd[4]) <<  8) | ((u32)(srb->cmnd[5]));
1757 
1758 		blocks = ((u32)(srb->cmnd[6]) << 24) | ((u32)(srb->cmnd[7]) << 16) |
1759 		         ((u32)(srb->cmnd[8]) <<  8) | ((u32)(srb->cmnd[9]));
1760 
1761 		usb_stor_dbg(us, "READ_12: read block 0x%04lx  count %ld\n",
1762 			     block, blocks);
1763 		return usbat_flash_read_data(us, info, block, blocks);
1764 	}
1765 
1766 	if (srb->cmnd[0] == WRITE_10) {
1767 		block = ((u32)(srb->cmnd[2]) << 24) | ((u32)(srb->cmnd[3]) << 16) |
1768 		        ((u32)(srb->cmnd[4]) <<  8) | ((u32)(srb->cmnd[5]));
1769 
1770 		blocks = ((u32)(srb->cmnd[7]) << 8) | ((u32)(srb->cmnd[8]));
1771 
1772 		usb_stor_dbg(us, "WRITE_10: write block 0x%04lx  count %ld\n",
1773 			     block, blocks);
1774 		return usbat_flash_write_data(us, info, block, blocks);
1775 	}
1776 
1777 	if (srb->cmnd[0] == WRITE_12) {
1778 		/*
1779 		 * I don't think we'll ever see a WRITE_12 but support it anyway
1780 		 */
1781 		block = ((u32)(srb->cmnd[2]) << 24) | ((u32)(srb->cmnd[3]) << 16) |
1782 		        ((u32)(srb->cmnd[4]) <<  8) | ((u32)(srb->cmnd[5]));
1783 
1784 		blocks = ((u32)(srb->cmnd[6]) << 24) | ((u32)(srb->cmnd[7]) << 16) |
1785 		         ((u32)(srb->cmnd[8]) <<  8) | ((u32)(srb->cmnd[9]));
1786 
1787 		usb_stor_dbg(us, "WRITE_12: write block 0x%04lx  count %ld\n",
1788 			     block, blocks);
1789 		return usbat_flash_write_data(us, info, block, blocks);
1790 	}
1791 
1792 
1793 	if (srb->cmnd[0] == TEST_UNIT_READY) {
1794 		usb_stor_dbg(us, "TEST_UNIT_READY\n");
1795 
1796 		rc = usbat_flash_check_media(us, info);
1797 		if (rc != USB_STOR_TRANSPORT_GOOD)
1798 			return rc;
1799 
1800 		return usbat_check_status(us);
1801 	}
1802 
1803 	if (srb->cmnd[0] == REQUEST_SENSE) {
1804 		usb_stor_dbg(us, "REQUEST_SENSE\n");
1805 
1806 		memset(ptr, 0, 18);
1807 		ptr[0] = 0xF0;
1808 		ptr[2] = info->sense_key;
1809 		ptr[7] = 11;
1810 		ptr[12] = info->sense_asc;
1811 		ptr[13] = info->sense_ascq;
1812 		usb_stor_set_xfer_buf(ptr, 18, srb);
1813 
1814 		return USB_STOR_TRANSPORT_GOOD;
1815 	}
1816 
1817 	if (srb->cmnd[0] == ALLOW_MEDIUM_REMOVAL) {
1818 		/*
1819 		 * sure.  whatever.  not like we can stop the user from popping
1820 		 * the media out of the device (no locking doors, etc)
1821 		 */
1822 		return USB_STOR_TRANSPORT_GOOD;
1823 	}
1824 
1825 	usb_stor_dbg(us, "Gah! Unknown command: %d (0x%x)\n",
1826 		     srb->cmnd[0], srb->cmnd[0]);
1827 	info->sense_key = 0x05;
1828 	info->sense_asc = 0x20;
1829 	info->sense_ascq = 0x00;
1830 	return USB_STOR_TRANSPORT_FAILED;
1831 }
1832 
1833 static int init_usbat_cd(struct us_data *us)
1834 {
1835 	return init_usbat(us, USBAT_DEV_HP8200);
1836 }
1837 
1838 static int init_usbat_flash(struct us_data *us)
1839 {
1840 	return init_usbat(us, USBAT_DEV_FLASH);
1841 }
1842 
1843 static struct scsi_host_template usbat_host_template;
1844 
1845 static int usbat_probe(struct usb_interface *intf,
1846 			 const struct usb_device_id *id)
1847 {
1848 	struct us_data *us;
1849 	int result;
1850 
1851 	result = usb_stor_probe1(&us, intf, id,
1852 			(id - usbat_usb_ids) + usbat_unusual_dev_list,
1853 			&usbat_host_template);
1854 	if (result)
1855 		return result;
1856 
1857 	/*
1858 	 * The actual transport will be determined later by the
1859 	 * initialization routine; this is just a placeholder.
1860 	 */
1861 	us->transport_name = "Shuttle USBAT";
1862 	us->transport = usbat_flash_transport;
1863 	us->transport_reset = usb_stor_CB_reset;
1864 	us->max_lun = 0;
1865 
1866 	result = usb_stor_probe2(us);
1867 	return result;
1868 }
1869 
1870 static struct usb_driver usbat_driver = {
1871 	.name =		DRV_NAME,
1872 	.probe =	usbat_probe,
1873 	.disconnect =	usb_stor_disconnect,
1874 	.suspend =	usb_stor_suspend,
1875 	.resume =	usb_stor_resume,
1876 	.reset_resume =	usb_stor_reset_resume,
1877 	.pre_reset =	usb_stor_pre_reset,
1878 	.post_reset =	usb_stor_post_reset,
1879 	.id_table =	usbat_usb_ids,
1880 	.soft_unbind =	1,
1881 	.no_dynamic_id = 1,
1882 };
1883 
1884 module_usb_stor_driver(usbat_driver, usbat_host_template, DRV_NAME);
1885