Linux Audio

Check our new training course

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