Linux Audio

Check our new training course

Loading...
v3.1
 
  1/*
  2 * USB Attached SCSI
  3 * Note that this is not the same as the USB Mass Storage driver
  4 *
 
  5 * Copyright Matthew Wilcox for Intel Corp, 2010
  6 * Copyright Sarah Sharp for Intel Corp, 2010
  7 *
  8 * Distributed under the terms of the GNU GPL, version two.
  9 */
 10
 11#include <linux/blkdev.h>
 12#include <linux/slab.h>
 13#include <linux/types.h>
 
 14#include <linux/usb.h>
 
 
 15#include <linux/usb/storage.h>
 
 16
 17#include <scsi/scsi.h>
 
 18#include <scsi/scsi_dbg.h>
 19#include <scsi/scsi_cmnd.h>
 20#include <scsi/scsi_device.h>
 21#include <scsi/scsi_host.h>
 22#include <scsi/scsi_tcq.h>
 23
 24/* Common header for all IUs */
 25struct iu {
 26	__u8 iu_id;
 27	__u8 rsvd1;
 28	__be16 tag;
 29};
 30
 31enum {
 32	IU_ID_COMMAND		= 0x01,
 33	IU_ID_STATUS		= 0x03,
 34	IU_ID_RESPONSE		= 0x04,
 35	IU_ID_TASK_MGMT		= 0x05,
 36	IU_ID_READ_READY	= 0x06,
 37	IU_ID_WRITE_READY	= 0x07,
 38};
 39
 40struct command_iu {
 41	__u8 iu_id;
 42	__u8 rsvd1;
 43	__be16 tag;
 44	__u8 prio_attr;
 45	__u8 rsvd5;
 46	__u8 len;
 47	__u8 rsvd7;
 48	struct scsi_lun lun;
 49	__u8 cdb[16];	/* XXX: Overflow-checking tools may misunderstand */
 50};
 51
 52/*
 53 * Also used for the Read Ready and Write Ready IUs since they have the
 54 * same first four bytes
 55 */
 56struct sense_iu {
 57	__u8 iu_id;
 58	__u8 rsvd1;
 59	__be16 tag;
 60	__be16 status_qual;
 61	__u8 status;
 62	__u8 rsvd7[7];
 63	__be16 len;
 64	__u8 sense[SCSI_SENSE_BUFFERSIZE];
 65};
 66
 67/*
 68 * The r00-r01c specs define this version of the SENSE IU data structure.
 69 * It's still in use by several different firmware releases.
 70 */
 71struct sense_iu_old {
 72	__u8 iu_id;
 73	__u8 rsvd1;
 74	__be16 tag;
 75	__be16 len;
 76	__u8 status;
 77	__u8 service_response;
 78	__u8 sense[SCSI_SENSE_BUFFERSIZE];
 79};
 80
 81enum {
 82	CMD_PIPE_ID		= 1,
 83	STATUS_PIPE_ID		= 2,
 84	DATA_IN_PIPE_ID		= 3,
 85	DATA_OUT_PIPE_ID	= 4,
 86
 87	UAS_SIMPLE_TAG		= 0,
 88	UAS_HEAD_TAG		= 1,
 89	UAS_ORDERED_TAG		= 2,
 90	UAS_ACA			= 4,
 91};
 92
 93struct uas_dev_info {
 94	struct usb_interface *intf;
 95	struct usb_device *udev;
 96	int qdepth;
 
 
 
 
 97	unsigned cmd_pipe, status_pipe, data_in_pipe, data_out_pipe;
 98	unsigned use_streams:1;
 99	unsigned uas_sense_old:1;
 
 
 
100};
101
102enum {
103	ALLOC_STATUS_URB	= (1 << 0),
104	SUBMIT_STATUS_URB	= (1 << 1),
105	ALLOC_DATA_IN_URB	= (1 << 2),
106	SUBMIT_DATA_IN_URB	= (1 << 3),
107	ALLOC_DATA_OUT_URB	= (1 << 4),
108	SUBMIT_DATA_OUT_URB	= (1 << 5),
109	ALLOC_CMD_URB		= (1 << 6),
110	SUBMIT_CMD_URB		= (1 << 7),
 
 
 
 
111};
112
113/* Overrides scsi_pointer */
114struct uas_cmd_info {
115	unsigned int state;
116	unsigned int stream;
117	struct urb *cmd_urb;
118	struct urb *status_urb;
119	struct urb *data_in_urb;
120	struct urb *data_out_urb;
121	struct list_head list;
122};
123
124/* I hate forward declarations, but I actually have a loop */
125static int uas_submit_urbs(struct scsi_cmnd *cmnd,
126				struct uas_dev_info *devinfo, gfp_t gfp);
127
128static DEFINE_SPINLOCK(uas_work_lock);
129static LIST_HEAD(uas_work_list);
 
 
130
131static void uas_do_work(struct work_struct *work)
132{
 
 
133	struct uas_cmd_info *cmdinfo;
134	struct list_head list;
 
 
135
136	spin_lock_irq(&uas_work_lock);
137	list_replace_init(&uas_work_list, &list);
138	spin_unlock_irq(&uas_work_lock);
 
139
140	list_for_each_entry(cmdinfo, &list, list) {
141		struct scsi_pointer *scp = (void *)cmdinfo;
142		struct scsi_cmnd *cmnd = container_of(scp,
143							struct scsi_cmnd, SCp);
144		uas_submit_urbs(cmnd, cmnd->device->hostdata, GFP_NOIO);
 
 
 
 
 
 
 
 
 
 
145	}
 
 
 
 
 
 
 
 
 
 
 
 
 
146}
147
148static DECLARE_WORK(uas_work, uas_do_work);
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
149
150static void uas_sense(struct urb *urb, struct scsi_cmnd *cmnd)
151{
152	struct sense_iu *sense_iu = urb->transfer_buffer;
153	struct scsi_device *sdev = cmnd->device;
154
155	if (urb->actual_length > 16) {
156		unsigned len = be16_to_cpup(&sense_iu->len);
157		if (len + 16 != urb->actual_length) {
158			int newlen = min(len + 16, urb->actual_length) - 16;
159			if (newlen < 0)
160				newlen = 0;
161			sdev_printk(KERN_INFO, sdev, "%s: urb length %d "
162				"disagrees with IU sense data length %d, "
163				"using %d bytes of sense data\n", __func__,
164					urb->actual_length, len, newlen);
165			len = newlen;
166		}
167		memcpy(cmnd->sense_buffer, sense_iu->sense, len);
168	}
169
170	cmnd->result = sense_iu->status;
171	if (sdev->current_cmnd)
172		sdev->current_cmnd = NULL;
173	cmnd->scsi_done(cmnd);
174	usb_free_urb(urb);
175}
176
177static void uas_sense_old(struct urb *urb, struct scsi_cmnd *cmnd)
 
178{
179	struct sense_iu_old *sense_iu = urb->transfer_buffer;
180	struct scsi_device *sdev = cmnd->device;
181
182	if (urb->actual_length > 8) {
183		unsigned len = be16_to_cpup(&sense_iu->len) - 2;
184		if (len + 8 != urb->actual_length) {
185			int newlen = min(len + 8, urb->actual_length) - 8;
186			if (newlen < 0)
187				newlen = 0;
188			sdev_printk(KERN_INFO, sdev, "%s: urb length %d "
189				"disagrees with IU sense data length %d, "
190				"using %d bytes of sense data\n", __func__,
191					urb->actual_length, len, newlen);
192			len = newlen;
193		}
194		memcpy(cmnd->sense_buffer, sense_iu->sense, len);
195	}
 
 
 
196
197	cmnd->result = sense_iu->status;
198	if (sdev->current_cmnd)
199		sdev->current_cmnd = NULL;
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
200	cmnd->scsi_done(cmnd);
201	usb_free_urb(urb);
202}
203
204static void uas_xfer_data(struct urb *urb, struct scsi_cmnd *cmnd,
205							unsigned direction)
206{
207	struct uas_cmd_info *cmdinfo = (void *)&cmnd->SCp;
208	int err;
209
210	cmdinfo->state = direction | SUBMIT_STATUS_URB;
211	err = uas_submit_urbs(cmnd, cmnd->device->hostdata, GFP_ATOMIC);
212	if (err) {
213		spin_lock(&uas_work_lock);
214		list_add_tail(&cmdinfo->list, &uas_work_list);
215		spin_unlock(&uas_work_lock);
216		schedule_work(&uas_work);
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
217	}
 
 
218}
219
220static void uas_stat_cmplt(struct urb *urb)
221{
222	struct iu *iu = urb->transfer_buffer;
223	struct scsi_device *sdev = urb->context;
224	struct uas_dev_info *devinfo = sdev->hostdata;
 
 
225	struct scsi_cmnd *cmnd;
226	u16 tag;
 
 
 
 
227
228	if (urb->status) {
229		dev_err(&urb->dev->dev, "URB BAD STATUS %d\n", urb->status);
230		usb_free_urb(urb);
231		return;
 
 
 
 
 
232	}
233
234	tag = be16_to_cpup(&iu->tag) - 1;
235	if (sdev->current_cmnd)
236		cmnd = sdev->current_cmnd;
237	else
238		cmnd = scsi_find_tag(sdev, tag);
239	if (!cmnd)
240		return;
 
 
 
 
 
 
 
241
242	switch (iu->iu_id) {
243	case IU_ID_STATUS:
244		if (urb->actual_length < 16)
245			devinfo->uas_sense_old = 1;
246		if (devinfo->uas_sense_old)
247			uas_sense_old(urb, cmnd);
248		else
249			uas_sense(urb, cmnd);
 
 
250		break;
251	case IU_ID_READ_READY:
 
 
 
 
 
252		uas_xfer_data(urb, cmnd, SUBMIT_DATA_IN_URB);
253		break;
254	case IU_ID_WRITE_READY:
 
 
 
 
 
255		uas_xfer_data(urb, cmnd, SUBMIT_DATA_OUT_URB);
256		break;
 
 
 
 
 
 
 
 
 
 
257	default:
258		scmd_printk(KERN_ERR, cmnd,
259			"Bogus IU (%d) received on status pipe\n", iu->iu_id);
 
 
 
 
 
 
 
 
 
 
 
 
260	}
261}
262
263static void uas_data_cmplt(struct urb *urb)
264{
265	struct scsi_data_buffer *sdb = urb->context;
266	sdb->resid = sdb->length - urb->actual_length;
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
267	usb_free_urb(urb);
268}
269
270static struct urb *uas_alloc_data_urb(struct uas_dev_info *devinfo, gfp_t gfp,
271				unsigned int pipe, u16 stream_id,
272				struct scsi_data_buffer *sdb,
273				enum dma_data_direction dir)
274{
275	struct usb_device *udev = devinfo->udev;
 
276	struct urb *urb = usb_alloc_urb(0, gfp);
 
 
 
 
277
278	if (!urb)
279		goto out;
280	usb_fill_bulk_urb(urb, udev, pipe, NULL, sdb->length, uas_data_cmplt,
281									sdb);
282	if (devinfo->use_streams)
283		urb->stream_id = stream_id;
284	urb->num_sgs = udev->bus->sg_tablesize ? sdb->table.nents : 0;
285	urb->sg = sdb->table.sgl;
286 out:
287	return urb;
288}
289
290static struct urb *uas_alloc_sense_urb(struct uas_dev_info *devinfo, gfp_t gfp,
291					struct scsi_cmnd *cmnd, u16 stream_id)
292{
293	struct usb_device *udev = devinfo->udev;
 
294	struct urb *urb = usb_alloc_urb(0, gfp);
295	struct sense_iu *iu;
296
297	if (!urb)
298		goto out;
299
300	iu = kzalloc(sizeof(*iu), gfp);
301	if (!iu)
302		goto free;
303
304	usb_fill_bulk_urb(urb, udev, devinfo->status_pipe, iu, sizeof(*iu),
305						uas_stat_cmplt, cmnd->device);
306	urb->stream_id = stream_id;
 
307	urb->transfer_flags |= URB_FREE_BUFFER;
308 out:
309	return urb;
310 free:
311	usb_free_urb(urb);
312	return NULL;
313}
314
315static struct urb *uas_alloc_cmd_urb(struct uas_dev_info *devinfo, gfp_t gfp,
316					struct scsi_cmnd *cmnd, u16 stream_id)
317{
318	struct usb_device *udev = devinfo->udev;
319	struct scsi_device *sdev = cmnd->device;
 
320	struct urb *urb = usb_alloc_urb(0, gfp);
321	struct command_iu *iu;
322	int len;
323
324	if (!urb)
325		goto out;
326
327	len = cmnd->cmd_len - 16;
328	if (len < 0)
329		len = 0;
330	len = ALIGN(len, 4);
331	iu = kzalloc(sizeof(*iu) + len, gfp);
332	if (!iu)
333		goto free;
334
335	iu->iu_id = IU_ID_COMMAND;
336	iu->tag = cpu_to_be16(stream_id);
337	iu->prio_attr = UAS_SIMPLE_TAG;
338	iu->len = len;
339	int_to_scsilun(sdev->lun, &iu->lun);
340	memcpy(iu->cdb, cmnd->cmnd, cmnd->cmd_len);
341
342	usb_fill_bulk_urb(urb, udev, devinfo->cmd_pipe, iu, sizeof(*iu) + len,
343							usb_free_urb, NULL);
344	urb->transfer_flags |= URB_FREE_BUFFER;
345 out:
346	return urb;
347 free:
348	usb_free_urb(urb);
349	return NULL;
350}
351
352/*
353 * Why should I request the Status IU before sending the Command IU?  Spec
354 * says to, but also says the device may receive them in any order.  Seems
355 * daft to me.
356 */
357
358static int uas_submit_urbs(struct scsi_cmnd *cmnd,
359					struct uas_dev_info *devinfo, gfp_t gfp)
360{
361	struct uas_cmd_info *cmdinfo = (void *)&cmnd->SCp;
 
 
362
363	if (cmdinfo->state & ALLOC_STATUS_URB) {
364		cmdinfo->status_urb = uas_alloc_sense_urb(devinfo, gfp, cmnd,
365							  cmdinfo->stream);
366		if (!cmdinfo->status_urb)
367			return SCSI_MLQUEUE_DEVICE_BUSY;
368		cmdinfo->state &= ~ALLOC_STATUS_URB;
 
 
 
 
369	}
 
 
 
 
 
 
 
 
 
370
 
371	if (cmdinfo->state & SUBMIT_STATUS_URB) {
372		if (usb_submit_urb(cmdinfo->status_urb, gfp)) {
373			scmd_printk(KERN_INFO, cmnd,
374					"sense urb submission failure\n");
375			return SCSI_MLQUEUE_DEVICE_BUSY;
376		}
377		cmdinfo->state &= ~SUBMIT_STATUS_URB;
378	}
379
380	if (cmdinfo->state & ALLOC_DATA_IN_URB) {
381		cmdinfo->data_in_urb = uas_alloc_data_urb(devinfo, gfp,
382					devinfo->data_in_pipe, cmdinfo->stream,
383					scsi_in(cmnd), DMA_FROM_DEVICE);
384		if (!cmdinfo->data_in_urb)
385			return SCSI_MLQUEUE_DEVICE_BUSY;
386		cmdinfo->state &= ~ALLOC_DATA_IN_URB;
387	}
388
389	if (cmdinfo->state & SUBMIT_DATA_IN_URB) {
390		if (usb_submit_urb(cmdinfo->data_in_urb, gfp)) {
391			scmd_printk(KERN_INFO, cmnd,
392					"data in urb submission failure\n");
 
 
393			return SCSI_MLQUEUE_DEVICE_BUSY;
394		}
395		cmdinfo->state &= ~SUBMIT_DATA_IN_URB;
 
396	}
397
398	if (cmdinfo->state & ALLOC_DATA_OUT_URB) {
399		cmdinfo->data_out_urb = uas_alloc_data_urb(devinfo, gfp,
400					devinfo->data_out_pipe, cmdinfo->stream,
401					scsi_out(cmnd), DMA_TO_DEVICE);
402		if (!cmdinfo->data_out_urb)
403			return SCSI_MLQUEUE_DEVICE_BUSY;
404		cmdinfo->state &= ~ALLOC_DATA_OUT_URB;
405	}
406
407	if (cmdinfo->state & SUBMIT_DATA_OUT_URB) {
408		if (usb_submit_urb(cmdinfo->data_out_urb, gfp)) {
409			scmd_printk(KERN_INFO, cmnd,
410					"data out urb submission failure\n");
 
 
411			return SCSI_MLQUEUE_DEVICE_BUSY;
412		}
413		cmdinfo->state &= ~SUBMIT_DATA_OUT_URB;
 
414	}
415
416	if (cmdinfo->state & ALLOC_CMD_URB) {
417		cmdinfo->cmd_urb = uas_alloc_cmd_urb(devinfo, gfp, cmnd,
418							cmdinfo->stream);
419		if (!cmdinfo->cmd_urb)
420			return SCSI_MLQUEUE_DEVICE_BUSY;
421		cmdinfo->state &= ~ALLOC_CMD_URB;
422	}
423
424	if (cmdinfo->state & SUBMIT_CMD_URB) {
425		if (usb_submit_urb(cmdinfo->cmd_urb, gfp)) {
426			scmd_printk(KERN_INFO, cmnd,
427					"cmd urb submission failure\n");
 
 
428			return SCSI_MLQUEUE_DEVICE_BUSY;
429		}
 
430		cmdinfo->state &= ~SUBMIT_CMD_URB;
 
431	}
432
433	return 0;
434}
435
436static int uas_queuecommand_lck(struct scsi_cmnd *cmnd,
437					void (*done)(struct scsi_cmnd *))
438{
439	struct scsi_device *sdev = cmnd->device;
440	struct uas_dev_info *devinfo = sdev->hostdata;
441	struct uas_cmd_info *cmdinfo = (void *)&cmnd->SCp;
442	int err;
 
443
444	BUILD_BUG_ON(sizeof(struct uas_cmd_info) > sizeof(struct scsi_pointer));
445
446	if (!cmdinfo->status_urb && sdev->current_cmnd)
 
447		return SCSI_MLQUEUE_DEVICE_BUSY;
448
449	if (blk_rq_tagged(cmnd->request)) {
450		cmdinfo->stream = cmnd->request->tag + 1;
451	} else {
452		sdev->current_cmnd = cmnd;
453		cmdinfo->stream = 1;
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
454	}
455
456	cmnd->scsi_done = done;
457
458	cmdinfo->state = ALLOC_STATUS_URB | SUBMIT_STATUS_URB |
459			ALLOC_CMD_URB | SUBMIT_CMD_URB;
 
460
461	switch (cmnd->sc_data_direction) {
462	case DMA_FROM_DEVICE:
463		cmdinfo->state |= ALLOC_DATA_IN_URB | SUBMIT_DATA_IN_URB;
464		break;
465	case DMA_BIDIRECTIONAL:
466		cmdinfo->state |= ALLOC_DATA_IN_URB | SUBMIT_DATA_IN_URB;
 
467	case DMA_TO_DEVICE:
468		cmdinfo->state |= ALLOC_DATA_OUT_URB | SUBMIT_DATA_OUT_URB;
469	case DMA_NONE:
470		break;
471	}
472
473	if (!devinfo->use_streams) {
474		cmdinfo->state &= ~(SUBMIT_DATA_IN_URB | SUBMIT_DATA_OUT_URB);
475		cmdinfo->stream = 0;
476	}
477
478	err = uas_submit_urbs(cmnd, devinfo, GFP_ATOMIC);
479	if (err) {
480		/* If we did nothing, give up now */
481		if (cmdinfo->state & SUBMIT_STATUS_URB) {
482			usb_free_urb(cmdinfo->status_urb);
483			return SCSI_MLQUEUE_DEVICE_BUSY;
484		}
485		spin_lock(&uas_work_lock);
486		list_add_tail(&cmdinfo->list, &uas_work_list);
487		spin_unlock(&uas_work_lock);
488		schedule_work(&uas_work);
489	}
490
 
 
491	return 0;
492}
493
494static DEF_SCSI_QCMD(uas_queuecommand)
495
 
 
 
 
 
496static int uas_eh_abort_handler(struct scsi_cmnd *cmnd)
497{
498	struct scsi_device *sdev = cmnd->device;
499	sdev_printk(KERN_INFO, sdev, "%s tag %d\n", __func__,
500							cmnd->request->tag);
 
 
501
502/* XXX: Send ABORT TASK Task Management command */
503	return FAILED;
504}
505
506static int uas_eh_device_reset_handler(struct scsi_cmnd *cmnd)
507{
508	struct scsi_device *sdev = cmnd->device;
509	sdev_printk(KERN_INFO, sdev, "%s tag %d\n", __func__,
510							cmnd->request->tag);
511
512/* XXX: Send LOGICAL UNIT RESET Task Management command */
513	return FAILED;
514}
515
516static int uas_eh_target_reset_handler(struct scsi_cmnd *cmnd)
517{
518	struct scsi_device *sdev = cmnd->device;
519	sdev_printk(KERN_INFO, sdev, "%s tag %d\n", __func__,
520							cmnd->request->tag);
 
 
 
 
 
 
 
 
 
 
 
 
 
 
521
522/* XXX: Can we reset just the one USB interface?
523 * Would calling usb_set_interface() have the right effect?
524 */
525	return FAILED;
526}
527
528static int uas_eh_bus_reset_handler(struct scsi_cmnd *cmnd)
529{
530	struct scsi_device *sdev = cmnd->device;
531	struct uas_dev_info *devinfo = sdev->hostdata;
532	struct usb_device *udev = devinfo->udev;
 
 
533
534	sdev_printk(KERN_INFO, sdev, "%s tag %d\n", __func__,
535							cmnd->request->tag);
 
 
 
 
536
537	if (usb_reset_device(udev))
538		return SUCCESS;
539
540	return FAILED;
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
541}
542
543static int uas_slave_alloc(struct scsi_device *sdev)
544{
545	sdev->hostdata = (void *)sdev->host->hostdata[0];
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
546	return 0;
547}
548
549static int uas_slave_configure(struct scsi_device *sdev)
550{
551	struct uas_dev_info *devinfo = sdev->hostdata;
552	scsi_set_tag_type(sdev, MSG_ORDERED_TAG);
553	scsi_activate_tcq(sdev, devinfo->qdepth - 1);
 
 
 
 
 
 
 
554	return 0;
555}
556
557static struct scsi_host_template uas_host_template = {
558	.module = THIS_MODULE,
559	.name = "uas",
560	.queuecommand = uas_queuecommand,
 
561	.slave_alloc = uas_slave_alloc,
562	.slave_configure = uas_slave_configure,
563	.eh_abort_handler = uas_eh_abort_handler,
564	.eh_device_reset_handler = uas_eh_device_reset_handler,
565	.eh_target_reset_handler = uas_eh_target_reset_handler,
566	.eh_bus_reset_handler = uas_eh_bus_reset_handler,
567	.can_queue = 65536,	/* Is there a limit on the _host_ ? */
568	.this_id = -1,
569	.sg_tablesize = SG_NONE,
570	.cmd_per_lun = 1,	/* until we override it */
571	.skip_settle_delay = 1,
572	.ordered_tag = 1,
573};
574
 
 
 
 
 
 
575static struct usb_device_id uas_usb_ids[] = {
 
576	{ USB_INTERFACE_INFO(USB_CLASS_MASS_STORAGE, USB_SC_SCSI, USB_PR_BULK) },
577	{ USB_INTERFACE_INFO(USB_CLASS_MASS_STORAGE, USB_SC_SCSI, USB_PR_UAS) },
578	/* 0xaa is a prototype device I happen to have access to */
579	{ USB_INTERFACE_INFO(USB_CLASS_MASS_STORAGE, USB_SC_SCSI, 0xaa) },
580	{ }
581};
582MODULE_DEVICE_TABLE(usb, uas_usb_ids);
583
584static int uas_is_interface(struct usb_host_interface *intf)
585{
586	return (intf->desc.bInterfaceClass == USB_CLASS_MASS_STORAGE &&
587		intf->desc.bInterfaceSubClass == USB_SC_SCSI &&
588		intf->desc.bInterfaceProtocol == USB_PR_UAS);
589}
590
591static int uas_switch_interface(struct usb_device *udev,
592						struct usb_interface *intf)
593{
594	int i;
595
596	if (uas_is_interface(intf->cur_altsetting))
597		return 0;
598
599	for (i = 0; i < intf->num_altsetting; i++) {
600		struct usb_host_interface *alt = &intf->altsetting[i];
601		if (alt == intf->cur_altsetting)
602			continue;
603		if (uas_is_interface(alt))
604			return usb_set_interface(udev,
605						alt->desc.bInterfaceNumber,
606						alt->desc.bAlternateSetting);
607	}
608
609	return -ENODEV;
 
610}
611
612static void uas_configure_endpoints(struct uas_dev_info *devinfo)
613{
614	struct usb_host_endpoint *eps[4] = { };
615	struct usb_interface *intf = devinfo->intf;
616	struct usb_device *udev = devinfo->udev;
617	struct usb_host_endpoint *endpoint = intf->cur_altsetting->endpoint;
618	unsigned i, n_endpoints = intf->cur_altsetting->desc.bNumEndpoints;
619
620	devinfo->uas_sense_old = 0;
 
 
 
 
 
 
 
 
 
 
 
621
622	for (i = 0; i < n_endpoints; i++) {
623		unsigned char *extra = endpoint[i].extra;
624		int len = endpoint[i].extralen;
625		while (len > 1) {
626			if (extra[1] == USB_DT_PIPE_USAGE) {
627				unsigned pipe_id = extra[2];
628				if (pipe_id > 0 && pipe_id < 5)
629					eps[pipe_id - 1] = &endpoint[i];
630				break;
631			}
632			len -= extra[0];
633			extra += extra[0];
634		}
635	}
636
637	/*
638	 * Assume that if we didn't find a control pipe descriptor, we're
639	 * using a device with old firmware that happens to be set up like
640	 * this.
641	 */
642	if (!eps[0]) {
643		devinfo->cmd_pipe = usb_sndbulkpipe(udev, 1);
644		devinfo->status_pipe = usb_rcvbulkpipe(udev, 1);
645		devinfo->data_in_pipe = usb_rcvbulkpipe(udev, 2);
646		devinfo->data_out_pipe = usb_sndbulkpipe(udev, 2);
647
648		eps[1] = usb_pipe_endpoint(udev, devinfo->status_pipe);
649		eps[2] = usb_pipe_endpoint(udev, devinfo->data_in_pipe);
650		eps[3] = usb_pipe_endpoint(udev, devinfo->data_out_pipe);
651	} else {
652		devinfo->cmd_pipe = usb_sndbulkpipe(udev,
653						eps[0]->desc.bEndpointAddress);
654		devinfo->status_pipe = usb_rcvbulkpipe(udev,
655						eps[1]->desc.bEndpointAddress);
656		devinfo->data_in_pipe = usb_rcvbulkpipe(udev,
657						eps[2]->desc.bEndpointAddress);
658		devinfo->data_out_pipe = usb_sndbulkpipe(udev,
659						eps[3]->desc.bEndpointAddress);
660	}
661
662	devinfo->qdepth = usb_alloc_streams(devinfo->intf, eps + 1, 3, 256,
663								GFP_KERNEL);
664	if (devinfo->qdepth < 0) {
665		devinfo->qdepth = 256;
666		devinfo->use_streams = 0;
667	} else {
 
 
 
 
668		devinfo->use_streams = 1;
669	}
 
 
 
 
 
 
 
 
 
 
 
 
 
670}
671
672/*
673 * XXX: What I'd like to do here is register a SCSI host for each USB host in
674 * the system.  Follow usb-storage's design of registering a SCSI host for
675 * each USB device for the moment.  Can implement this by walking up the
676 * USB hierarchy until we find a USB host.
677 */
678static int uas_probe(struct usb_interface *intf, const struct usb_device_id *id)
679{
680	int result;
681	struct Scsi_Host *shost;
682	struct uas_dev_info *devinfo;
683	struct usb_device *udev = interface_to_usbdev(intf);
 
684
685	if (uas_switch_interface(udev, intf))
686		return -ENODEV;
687
688	devinfo = kmalloc(sizeof(struct uas_dev_info), GFP_KERNEL);
689	if (!devinfo)
690		return -ENOMEM;
691
692	result = -ENOMEM;
693	shost = scsi_host_alloc(&uas_host_template, sizeof(void *));
694	if (!shost)
695		goto free;
696
697	shost->max_cmd_len = 16 + 252;
698	shost->max_id = 1;
 
 
699	shost->sg_tablesize = udev->bus->sg_tablesize;
700
701	result = scsi_add_host(shost, &intf->dev);
702	if (result)
703		goto free;
704	shost->hostdata[0] = (unsigned long)devinfo;
705
706	devinfo->intf = intf;
707	devinfo->udev = udev;
708	uas_configure_endpoints(devinfo);
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
709
710	scsi_scan_host(shost);
711	usb_set_intfdata(intf, shost);
 
 
 
 
 
712	return result;
713 free:
714	kfree(devinfo);
 
 
 
 
715	if (shost)
716		scsi_host_put(shost);
717	return result;
718}
719
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
720static int uas_pre_reset(struct usb_interface *intf)
721{
722/* XXX: Need to return 1 if it's not our device in error handling */
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
723	return 0;
724}
725
726static int uas_post_reset(struct usb_interface *intf)
727{
728/* XXX: Need to return 1 if it's not our device in error handling */
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
729	return 0;
730}
731
732static void uas_disconnect(struct usb_interface *intf)
 
 
 
 
 
733{
734	struct usb_device *udev = interface_to_usbdev(intf);
735	struct usb_host_endpoint *eps[3];
736	struct Scsi_Host *shost = usb_get_intfdata(intf);
737	struct uas_dev_info *devinfo = (void *)shost->hostdata[0];
 
 
738
739	scsi_remove_host(shost);
 
 
 
 
 
 
740
741	eps[0] = usb_pipe_endpoint(udev, devinfo->status_pipe);
742	eps[1] = usb_pipe_endpoint(udev, devinfo->data_in_pipe);
743	eps[2] = usb_pipe_endpoint(udev, devinfo->data_out_pipe);
744	usb_free_streams(intf, eps, 3, GFP_KERNEL);
 
 
745
746	kfree(devinfo);
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
747}
748
749/*
750 * XXX: Should this plug into libusual so we can auto-upgrade devices from
751 * Bulk-Only to UAS?
 
752 */
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
753static struct usb_driver uas_driver = {
754	.name = "uas",
755	.probe = uas_probe,
756	.disconnect = uas_disconnect,
757	.pre_reset = uas_pre_reset,
758	.post_reset = uas_post_reset,
 
 
 
 
759	.id_table = uas_usb_ids,
760};
761
762static int uas_init(void)
763{
764	return usb_register(&uas_driver);
765}
766
767static void uas_exit(void)
768{
769	usb_deregister(&uas_driver);
770}
771
772module_init(uas_init);
773module_exit(uas_exit);
774
775MODULE_LICENSE("GPL");
776MODULE_AUTHOR("Matthew Wilcox and Sarah Sharp");
 
v4.17
   1// SPDX-License-Identifier: GPL-2.0
   2/*
   3 * USB Attached SCSI
   4 * Note that this is not the same as the USB Mass Storage driver
   5 *
   6 * Copyright Hans de Goede <hdegoede@redhat.com> for Red Hat, Inc. 2013 - 2016
   7 * Copyright Matthew Wilcox for Intel Corp, 2010
   8 * Copyright Sarah Sharp for Intel Corp, 2010
 
 
   9 */
  10
  11#include <linux/blkdev.h>
  12#include <linux/slab.h>
  13#include <linux/types.h>
  14#include <linux/module.h>
  15#include <linux/usb.h>
  16#include <linux/usb_usual.h>
  17#include <linux/usb/hcd.h>
  18#include <linux/usb/storage.h>
  19#include <linux/usb/uas.h>
  20
  21#include <scsi/scsi.h>
  22#include <scsi/scsi_eh.h>
  23#include <scsi/scsi_dbg.h>
  24#include <scsi/scsi_cmnd.h>
  25#include <scsi/scsi_device.h>
  26#include <scsi/scsi_host.h>
  27#include <scsi/scsi_tcq.h>
  28
  29#include "uas-detect.h"
  30#include "scsiglue.h"
 
 
 
 
 
 
 
 
 
 
 
 
 
  31
  32#define MAX_CMNDS 256
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
  33
  34struct uas_dev_info {
  35	struct usb_interface *intf;
  36	struct usb_device *udev;
  37	struct usb_anchor cmd_urbs;
  38	struct usb_anchor sense_urbs;
  39	struct usb_anchor data_urbs;
  40	unsigned long flags;
  41	int qdepth, resetting;
  42	unsigned cmd_pipe, status_pipe, data_in_pipe, data_out_pipe;
  43	unsigned use_streams:1;
  44	unsigned shutdown:1;
  45	struct scsi_cmnd *cmnd[MAX_CMNDS];
  46	spinlock_t lock;
  47	struct work_struct work;
  48};
  49
  50enum {
  51	SUBMIT_STATUS_URB	= BIT(1),
  52	ALLOC_DATA_IN_URB	= BIT(2),
  53	SUBMIT_DATA_IN_URB	= BIT(3),
  54	ALLOC_DATA_OUT_URB	= BIT(4),
  55	SUBMIT_DATA_OUT_URB	= BIT(5),
  56	ALLOC_CMD_URB		= BIT(6),
  57	SUBMIT_CMD_URB		= BIT(7),
  58	COMMAND_INFLIGHT        = BIT(8),
  59	DATA_IN_URB_INFLIGHT    = BIT(9),
  60	DATA_OUT_URB_INFLIGHT   = BIT(10),
  61	COMMAND_ABORTED         = BIT(11),
  62	IS_IN_WORK_LIST         = BIT(12),
  63};
  64
  65/* Overrides scsi_pointer */
  66struct uas_cmd_info {
  67	unsigned int state;
  68	unsigned int uas_tag;
  69	struct urb *cmd_urb;
 
  70	struct urb *data_in_urb;
  71	struct urb *data_out_urb;
 
  72};
  73
  74/* I hate forward declarations, but I actually have a loop */
  75static int uas_submit_urbs(struct scsi_cmnd *cmnd,
  76				struct uas_dev_info *devinfo);
  77static void uas_do_work(struct work_struct *work);
  78static int uas_try_complete(struct scsi_cmnd *cmnd, const char *caller);
  79static void uas_free_streams(struct uas_dev_info *devinfo);
  80static void uas_log_cmd_state(struct scsi_cmnd *cmnd, const char *prefix,
  81				int status);
  82
  83static void uas_do_work(struct work_struct *work)
  84{
  85	struct uas_dev_info *devinfo =
  86		container_of(work, struct uas_dev_info, work);
  87	struct uas_cmd_info *cmdinfo;
  88	struct scsi_cmnd *cmnd;
  89	unsigned long flags;
  90	int i, err;
  91
  92	spin_lock_irqsave(&devinfo->lock, flags);
  93
  94	if (devinfo->resetting)
  95		goto out;
  96
  97	for (i = 0; i < devinfo->qdepth; i++) {
  98		if (!devinfo->cmnd[i])
  99			continue;
 100
 101		cmnd = devinfo->cmnd[i];
 102		cmdinfo = (void *)&cmnd->SCp;
 103
 104		if (!(cmdinfo->state & IS_IN_WORK_LIST))
 105			continue;
 106
 107		err = uas_submit_urbs(cmnd, cmnd->device->hostdata);
 108		if (!err)
 109			cmdinfo->state &= ~IS_IN_WORK_LIST;
 110		else
 111			schedule_work(&devinfo->work);
 112	}
 113out:
 114	spin_unlock_irqrestore(&devinfo->lock, flags);
 115}
 116
 117static void uas_add_work(struct uas_cmd_info *cmdinfo)
 118{
 119	struct scsi_pointer *scp = (void *)cmdinfo;
 120	struct scsi_cmnd *cmnd = container_of(scp, struct scsi_cmnd, SCp);
 121	struct uas_dev_info *devinfo = cmnd->device->hostdata;
 122
 123	lockdep_assert_held(&devinfo->lock);
 124	cmdinfo->state |= IS_IN_WORK_LIST;
 125	schedule_work(&devinfo->work);
 126}
 127
 128static void uas_zap_pending(struct uas_dev_info *devinfo, int result)
 129{
 130	struct uas_cmd_info *cmdinfo;
 131	struct scsi_cmnd *cmnd;
 132	unsigned long flags;
 133	int i, err;
 134
 135	spin_lock_irqsave(&devinfo->lock, flags);
 136	for (i = 0; i < devinfo->qdepth; i++) {
 137		if (!devinfo->cmnd[i])
 138			continue;
 139
 140		cmnd = devinfo->cmnd[i];
 141		cmdinfo = (void *)&cmnd->SCp;
 142		uas_log_cmd_state(cmnd, __func__, 0);
 143		/* Sense urbs were killed, clear COMMAND_INFLIGHT manually */
 144		cmdinfo->state &= ~COMMAND_INFLIGHT;
 145		cmnd->result = result << 16;
 146		err = uas_try_complete(cmnd, __func__);
 147		WARN_ON(err != 0);
 148	}
 149	spin_unlock_irqrestore(&devinfo->lock, flags);
 150}
 151
 152static void uas_sense(struct urb *urb, struct scsi_cmnd *cmnd)
 153{
 154	struct sense_iu *sense_iu = urb->transfer_buffer;
 155	struct scsi_device *sdev = cmnd->device;
 156
 157	if (urb->actual_length > 16) {
 158		unsigned len = be16_to_cpup(&sense_iu->len);
 159		if (len + 16 != urb->actual_length) {
 160			int newlen = min(len + 16, urb->actual_length) - 16;
 161			if (newlen < 0)
 162				newlen = 0;
 163			sdev_printk(KERN_INFO, sdev, "%s: urb length %d "
 164				"disagrees with IU sense data length %d, "
 165				"using %d bytes of sense data\n", __func__,
 166					urb->actual_length, len, newlen);
 167			len = newlen;
 168		}
 169		memcpy(cmnd->sense_buffer, sense_iu->sense, len);
 170	}
 171
 172	cmnd->result = sense_iu->status;
 
 
 
 
 173}
 174
 175static void uas_log_cmd_state(struct scsi_cmnd *cmnd, const char *prefix,
 176			      int status)
 177{
 178	struct uas_cmd_info *ci = (void *)&cmnd->SCp;
 179	struct uas_cmd_info *cmdinfo = (void *)&cmnd->SCp;
 180
 181	scmd_printk(KERN_INFO, cmnd,
 182		    "%s %d uas-tag %d inflight:%s%s%s%s%s%s%s%s%s%s%s%s ",
 183		    prefix, status, cmdinfo->uas_tag,
 184		    (ci->state & SUBMIT_STATUS_URB)     ? " s-st"  : "",
 185		    (ci->state & ALLOC_DATA_IN_URB)     ? " a-in"  : "",
 186		    (ci->state & SUBMIT_DATA_IN_URB)    ? " s-in"  : "",
 187		    (ci->state & ALLOC_DATA_OUT_URB)    ? " a-out" : "",
 188		    (ci->state & SUBMIT_DATA_OUT_URB)   ? " s-out" : "",
 189		    (ci->state & ALLOC_CMD_URB)         ? " a-cmd" : "",
 190		    (ci->state & SUBMIT_CMD_URB)        ? " s-cmd" : "",
 191		    (ci->state & COMMAND_INFLIGHT)      ? " CMD"   : "",
 192		    (ci->state & DATA_IN_URB_INFLIGHT)  ? " IN"    : "",
 193		    (ci->state & DATA_OUT_URB_INFLIGHT) ? " OUT"   : "",
 194		    (ci->state & COMMAND_ABORTED)       ? " abort" : "",
 195		    (ci->state & IS_IN_WORK_LIST)       ? " work"  : "");
 196	scsi_print_command(cmnd);
 197}
 198
 199static void uas_free_unsubmitted_urbs(struct scsi_cmnd *cmnd)
 200{
 201	struct uas_cmd_info *cmdinfo;
 202
 203	if (!cmnd)
 204		return;
 205
 206	cmdinfo = (void *)&cmnd->SCp;
 207
 208	if (cmdinfo->state & SUBMIT_CMD_URB)
 209		usb_free_urb(cmdinfo->cmd_urb);
 210
 211	/* data urbs may have never gotten their submit flag set */
 212	if (!(cmdinfo->state & DATA_IN_URB_INFLIGHT))
 213		usb_free_urb(cmdinfo->data_in_urb);
 214	if (!(cmdinfo->state & DATA_OUT_URB_INFLIGHT))
 215		usb_free_urb(cmdinfo->data_out_urb);
 216}
 217
 218static int uas_try_complete(struct scsi_cmnd *cmnd, const char *caller)
 219{
 220	struct uas_cmd_info *cmdinfo = (void *)&cmnd->SCp;
 221	struct uas_dev_info *devinfo = (void *)cmnd->device->hostdata;
 222
 223	lockdep_assert_held(&devinfo->lock);
 224	if (cmdinfo->state & (COMMAND_INFLIGHT |
 225			      DATA_IN_URB_INFLIGHT |
 226			      DATA_OUT_URB_INFLIGHT |
 227			      COMMAND_ABORTED))
 228		return -EBUSY;
 229	devinfo->cmnd[cmdinfo->uas_tag - 1] = NULL;
 230	uas_free_unsubmitted_urbs(cmnd);
 231	cmnd->scsi_done(cmnd);
 232	return 0;
 233}
 234
 235static void uas_xfer_data(struct urb *urb, struct scsi_cmnd *cmnd,
 236			  unsigned direction)
 237{
 238	struct uas_cmd_info *cmdinfo = (void *)&cmnd->SCp;
 239	int err;
 240
 241	cmdinfo->state |= direction | SUBMIT_STATUS_URB;
 242	err = uas_submit_urbs(cmnd, cmnd->device->hostdata);
 243	if (err) {
 244		uas_add_work(cmdinfo);
 245	}
 246}
 247
 248static bool uas_evaluate_response_iu(struct response_iu *riu, struct scsi_cmnd *cmnd)
 249{
 250	u8 response_code = riu->response_code;
 251
 252	switch (response_code) {
 253	case RC_INCORRECT_LUN:
 254		cmnd->result = DID_BAD_TARGET << 16;
 255		break;
 256	case RC_TMF_SUCCEEDED:
 257		cmnd->result = DID_OK << 16;
 258		break;
 259	case RC_TMF_NOT_SUPPORTED:
 260		cmnd->result = DID_TARGET_FAILURE << 16;
 261		break;
 262	default:
 263		uas_log_cmd_state(cmnd, "response iu", response_code);
 264		cmnd->result = DID_ERROR << 16;
 265		break;
 266	}
 267
 268	return response_code == RC_TMF_SUCCEEDED;
 269}
 270
 271static void uas_stat_cmplt(struct urb *urb)
 272{
 273	struct iu *iu = urb->transfer_buffer;
 274	struct Scsi_Host *shost = urb->context;
 275	struct uas_dev_info *devinfo = (struct uas_dev_info *)shost->hostdata;
 276	struct urb *data_in_urb = NULL;
 277	struct urb *data_out_urb = NULL;
 278	struct scsi_cmnd *cmnd;
 279	struct uas_cmd_info *cmdinfo;
 280	unsigned long flags;
 281	unsigned int idx;
 282	int status = urb->status;
 283	bool success;
 284
 285	spin_lock_irqsave(&devinfo->lock, flags);
 286
 287	if (devinfo->resetting)
 288		goto out;
 289
 290	if (status) {
 291		if (status != -ENOENT && status != -ECONNRESET && status != -ESHUTDOWN)
 292			dev_err(&urb->dev->dev, "stat urb: status %d\n", status);
 293		goto out;
 294	}
 295
 296	idx = be16_to_cpup(&iu->tag) - 1;
 297	if (idx >= MAX_CMNDS || !devinfo->cmnd[idx]) {
 298		dev_err(&urb->dev->dev,
 299			"stat urb: no pending cmd for uas-tag %d\n", idx + 1);
 300		goto out;
 301	}
 302
 303	cmnd = devinfo->cmnd[idx];
 304	cmdinfo = (void *)&cmnd->SCp;
 305
 306	if (!(cmdinfo->state & COMMAND_INFLIGHT)) {
 307		uas_log_cmd_state(cmnd, "unexpected status cmplt", 0);
 308		goto out;
 309	}
 310
 311	switch (iu->iu_id) {
 312	case IU_ID_STATUS:
 313		uas_sense(urb, cmnd);
 314		if (cmnd->result != 0) {
 315			/* cancel data transfers on error */
 316			data_in_urb = usb_get_urb(cmdinfo->data_in_urb);
 317			data_out_urb = usb_get_urb(cmdinfo->data_out_urb);
 318		}
 319		cmdinfo->state &= ~COMMAND_INFLIGHT;
 320		uas_try_complete(cmnd, __func__);
 321		break;
 322	case IU_ID_READ_READY:
 323		if (!cmdinfo->data_in_urb ||
 324				(cmdinfo->state & DATA_IN_URB_INFLIGHT)) {
 325			uas_log_cmd_state(cmnd, "unexpected read rdy", 0);
 326			break;
 327		}
 328		uas_xfer_data(urb, cmnd, SUBMIT_DATA_IN_URB);
 329		break;
 330	case IU_ID_WRITE_READY:
 331		if (!cmdinfo->data_out_urb ||
 332				(cmdinfo->state & DATA_OUT_URB_INFLIGHT)) {
 333			uas_log_cmd_state(cmnd, "unexpected write rdy", 0);
 334			break;
 335		}
 336		uas_xfer_data(urb, cmnd, SUBMIT_DATA_OUT_URB);
 337		break;
 338	case IU_ID_RESPONSE:
 339		cmdinfo->state &= ~COMMAND_INFLIGHT;
 340		success = uas_evaluate_response_iu((struct response_iu *)iu, cmnd);
 341		if (!success) {
 342			/* Error, cancel data transfers */
 343			data_in_urb = usb_get_urb(cmdinfo->data_in_urb);
 344			data_out_urb = usb_get_urb(cmdinfo->data_out_urb);
 345		}
 346		uas_try_complete(cmnd, __func__);
 347		break;
 348	default:
 349		uas_log_cmd_state(cmnd, "bogus IU", iu->iu_id);
 350	}
 351out:
 352	usb_free_urb(urb);
 353	spin_unlock_irqrestore(&devinfo->lock, flags);
 354
 355	/* Unlinking of data urbs must be done without holding the lock */
 356	if (data_in_urb) {
 357		usb_unlink_urb(data_in_urb);
 358		usb_put_urb(data_in_urb);
 359	}
 360	if (data_out_urb) {
 361		usb_unlink_urb(data_out_urb);
 362		usb_put_urb(data_out_urb);
 363	}
 364}
 365
 366static void uas_data_cmplt(struct urb *urb)
 367{
 368	struct scsi_cmnd *cmnd = urb->context;
 369	struct uas_cmd_info *cmdinfo = (void *)&cmnd->SCp;
 370	struct uas_dev_info *devinfo = (void *)cmnd->device->hostdata;
 371	struct scsi_data_buffer *sdb = NULL;
 372	unsigned long flags;
 373	int status = urb->status;
 374
 375	spin_lock_irqsave(&devinfo->lock, flags);
 376
 377	if (cmdinfo->data_in_urb == urb) {
 378		sdb = scsi_in(cmnd);
 379		cmdinfo->state &= ~DATA_IN_URB_INFLIGHT;
 380		cmdinfo->data_in_urb = NULL;
 381	} else if (cmdinfo->data_out_urb == urb) {
 382		sdb = scsi_out(cmnd);
 383		cmdinfo->state &= ~DATA_OUT_URB_INFLIGHT;
 384		cmdinfo->data_out_urb = NULL;
 385	}
 386	if (sdb == NULL) {
 387		WARN_ON_ONCE(1);
 388		goto out;
 389	}
 390
 391	if (devinfo->resetting)
 392		goto out;
 393
 394	/* Data urbs should not complete before the cmd urb is submitted */
 395	if (cmdinfo->state & SUBMIT_CMD_URB) {
 396		uas_log_cmd_state(cmnd, "unexpected data cmplt", 0);
 397		goto out;
 398	}
 399
 400	if (status) {
 401		if (status != -ENOENT && status != -ECONNRESET && status != -ESHUTDOWN)
 402			uas_log_cmd_state(cmnd, "data cmplt err", status);
 403		/* error: no data transfered */
 404		sdb->resid = sdb->length;
 405	} else {
 406		sdb->resid = sdb->length - urb->actual_length;
 407	}
 408	uas_try_complete(cmnd, __func__);
 409out:
 410	usb_free_urb(urb);
 411	spin_unlock_irqrestore(&devinfo->lock, flags);
 412}
 413
 414static void uas_cmd_cmplt(struct urb *urb)
 415{
 416	if (urb->status)
 417		dev_err(&urb->dev->dev, "cmd cmplt err %d\n", urb->status);
 418
 419	usb_free_urb(urb);
 420}
 421
 422static struct urb *uas_alloc_data_urb(struct uas_dev_info *devinfo, gfp_t gfp,
 423				      struct scsi_cmnd *cmnd,
 424				      enum dma_data_direction dir)
 
 425{
 426	struct usb_device *udev = devinfo->udev;
 427	struct uas_cmd_info *cmdinfo = (void *)&cmnd->SCp;
 428	struct urb *urb = usb_alloc_urb(0, gfp);
 429	struct scsi_data_buffer *sdb = (dir == DMA_FROM_DEVICE)
 430		? scsi_in(cmnd) : scsi_out(cmnd);
 431	unsigned int pipe = (dir == DMA_FROM_DEVICE)
 432		? devinfo->data_in_pipe : devinfo->data_out_pipe;
 433
 434	if (!urb)
 435		goto out;
 436	usb_fill_bulk_urb(urb, udev, pipe, NULL, sdb->length,
 437			  uas_data_cmplt, cmnd);
 438	if (devinfo->use_streams)
 439		urb->stream_id = cmdinfo->uas_tag;
 440	urb->num_sgs = udev->bus->sg_tablesize ? sdb->table.nents : 0;
 441	urb->sg = sdb->table.sgl;
 442 out:
 443	return urb;
 444}
 445
 446static struct urb *uas_alloc_sense_urb(struct uas_dev_info *devinfo, gfp_t gfp,
 447				       struct scsi_cmnd *cmnd)
 448{
 449	struct usb_device *udev = devinfo->udev;
 450	struct uas_cmd_info *cmdinfo = (void *)&cmnd->SCp;
 451	struct urb *urb = usb_alloc_urb(0, gfp);
 452	struct sense_iu *iu;
 453
 454	if (!urb)
 455		goto out;
 456
 457	iu = kzalloc(sizeof(*iu), gfp);
 458	if (!iu)
 459		goto free;
 460
 461	usb_fill_bulk_urb(urb, udev, devinfo->status_pipe, iu, sizeof(*iu),
 462			  uas_stat_cmplt, cmnd->device->host);
 463	if (devinfo->use_streams)
 464		urb->stream_id = cmdinfo->uas_tag;
 465	urb->transfer_flags |= URB_FREE_BUFFER;
 466 out:
 467	return urb;
 468 free:
 469	usb_free_urb(urb);
 470	return NULL;
 471}
 472
 473static struct urb *uas_alloc_cmd_urb(struct uas_dev_info *devinfo, gfp_t gfp,
 474					struct scsi_cmnd *cmnd)
 475{
 476	struct usb_device *udev = devinfo->udev;
 477	struct scsi_device *sdev = cmnd->device;
 478	struct uas_cmd_info *cmdinfo = (void *)&cmnd->SCp;
 479	struct urb *urb = usb_alloc_urb(0, gfp);
 480	struct command_iu *iu;
 481	int len;
 482
 483	if (!urb)
 484		goto out;
 485
 486	len = cmnd->cmd_len - 16;
 487	if (len < 0)
 488		len = 0;
 489	len = ALIGN(len, 4);
 490	iu = kzalloc(sizeof(*iu) + len, gfp);
 491	if (!iu)
 492		goto free;
 493
 494	iu->iu_id = IU_ID_COMMAND;
 495	iu->tag = cpu_to_be16(cmdinfo->uas_tag);
 496	iu->prio_attr = UAS_SIMPLE_TAG;
 497	iu->len = len;
 498	int_to_scsilun(sdev->lun, &iu->lun);
 499	memcpy(iu->cdb, cmnd->cmnd, cmnd->cmd_len);
 500
 501	usb_fill_bulk_urb(urb, udev, devinfo->cmd_pipe, iu, sizeof(*iu) + len,
 502							uas_cmd_cmplt, NULL);
 503	urb->transfer_flags |= URB_FREE_BUFFER;
 504 out:
 505	return urb;
 506 free:
 507	usb_free_urb(urb);
 508	return NULL;
 509}
 510
 511/*
 512 * Why should I request the Status IU before sending the Command IU?  Spec
 513 * says to, but also says the device may receive them in any order.  Seems
 514 * daft to me.
 515 */
 516
 517static struct urb *uas_submit_sense_urb(struct scsi_cmnd *cmnd, gfp_t gfp)
 
 518{
 519	struct uas_dev_info *devinfo = cmnd->device->hostdata;
 520	struct urb *urb;
 521	int err;
 522
 523	urb = uas_alloc_sense_urb(devinfo, gfp, cmnd);
 524	if (!urb)
 525		return NULL;
 526	usb_anchor_urb(urb, &devinfo->sense_urbs);
 527	err = usb_submit_urb(urb, gfp);
 528	if (err) {
 529		usb_unanchor_urb(urb);
 530		uas_log_cmd_state(cmnd, "sense submit err", err);
 531		usb_free_urb(urb);
 532		return NULL;
 533	}
 534	return urb;
 535}
 536
 537static int uas_submit_urbs(struct scsi_cmnd *cmnd,
 538			   struct uas_dev_info *devinfo)
 539{
 540	struct uas_cmd_info *cmdinfo = (void *)&cmnd->SCp;
 541	struct urb *urb;
 542	int err;
 543
 544	lockdep_assert_held(&devinfo->lock);
 545	if (cmdinfo->state & SUBMIT_STATUS_URB) {
 546		urb = uas_submit_sense_urb(cmnd, GFP_ATOMIC);
 547		if (!urb)
 
 548			return SCSI_MLQUEUE_DEVICE_BUSY;
 
 549		cmdinfo->state &= ~SUBMIT_STATUS_URB;
 550	}
 551
 552	if (cmdinfo->state & ALLOC_DATA_IN_URB) {
 553		cmdinfo->data_in_urb = uas_alloc_data_urb(devinfo, GFP_ATOMIC,
 554							cmnd, DMA_FROM_DEVICE);
 
 555		if (!cmdinfo->data_in_urb)
 556			return SCSI_MLQUEUE_DEVICE_BUSY;
 557		cmdinfo->state &= ~ALLOC_DATA_IN_URB;
 558	}
 559
 560	if (cmdinfo->state & SUBMIT_DATA_IN_URB) {
 561		usb_anchor_urb(cmdinfo->data_in_urb, &devinfo->data_urbs);
 562		err = usb_submit_urb(cmdinfo->data_in_urb, GFP_ATOMIC);
 563		if (err) {
 564			usb_unanchor_urb(cmdinfo->data_in_urb);
 565			uas_log_cmd_state(cmnd, "data in submit err", err);
 566			return SCSI_MLQUEUE_DEVICE_BUSY;
 567		}
 568		cmdinfo->state &= ~SUBMIT_DATA_IN_URB;
 569		cmdinfo->state |= DATA_IN_URB_INFLIGHT;
 570	}
 571
 572	if (cmdinfo->state & ALLOC_DATA_OUT_URB) {
 573		cmdinfo->data_out_urb = uas_alloc_data_urb(devinfo, GFP_ATOMIC,
 574							cmnd, DMA_TO_DEVICE);
 
 575		if (!cmdinfo->data_out_urb)
 576			return SCSI_MLQUEUE_DEVICE_BUSY;
 577		cmdinfo->state &= ~ALLOC_DATA_OUT_URB;
 578	}
 579
 580	if (cmdinfo->state & SUBMIT_DATA_OUT_URB) {
 581		usb_anchor_urb(cmdinfo->data_out_urb, &devinfo->data_urbs);
 582		err = usb_submit_urb(cmdinfo->data_out_urb, GFP_ATOMIC);
 583		if (err) {
 584			usb_unanchor_urb(cmdinfo->data_out_urb);
 585			uas_log_cmd_state(cmnd, "data out submit err", err);
 586			return SCSI_MLQUEUE_DEVICE_BUSY;
 587		}
 588		cmdinfo->state &= ~SUBMIT_DATA_OUT_URB;
 589		cmdinfo->state |= DATA_OUT_URB_INFLIGHT;
 590	}
 591
 592	if (cmdinfo->state & ALLOC_CMD_URB) {
 593		cmdinfo->cmd_urb = uas_alloc_cmd_urb(devinfo, GFP_ATOMIC, cmnd);
 
 594		if (!cmdinfo->cmd_urb)
 595			return SCSI_MLQUEUE_DEVICE_BUSY;
 596		cmdinfo->state &= ~ALLOC_CMD_URB;
 597	}
 598
 599	if (cmdinfo->state & SUBMIT_CMD_URB) {
 600		usb_anchor_urb(cmdinfo->cmd_urb, &devinfo->cmd_urbs);
 601		err = usb_submit_urb(cmdinfo->cmd_urb, GFP_ATOMIC);
 602		if (err) {
 603			usb_unanchor_urb(cmdinfo->cmd_urb);
 604			uas_log_cmd_state(cmnd, "cmd submit err", err);
 605			return SCSI_MLQUEUE_DEVICE_BUSY;
 606		}
 607		cmdinfo->cmd_urb = NULL;
 608		cmdinfo->state &= ~SUBMIT_CMD_URB;
 609		cmdinfo->state |= COMMAND_INFLIGHT;
 610	}
 611
 612	return 0;
 613}
 614
 615static int uas_queuecommand_lck(struct scsi_cmnd *cmnd,
 616					void (*done)(struct scsi_cmnd *))
 617{
 618	struct scsi_device *sdev = cmnd->device;
 619	struct uas_dev_info *devinfo = sdev->hostdata;
 620	struct uas_cmd_info *cmdinfo = (void *)&cmnd->SCp;
 621	unsigned long flags;
 622	int idx, err;
 623
 624	BUILD_BUG_ON(sizeof(struct uas_cmd_info) > sizeof(struct scsi_pointer));
 625
 626	/* Re-check scsi_block_requests now that we've the host-lock */
 627	if (cmnd->device->host->host_self_blocked)
 628		return SCSI_MLQUEUE_DEVICE_BUSY;
 629
 630	if ((devinfo->flags & US_FL_NO_ATA_1X) &&
 631			(cmnd->cmnd[0] == ATA_12 || cmnd->cmnd[0] == ATA_16)) {
 632		memcpy(cmnd->sense_buffer, usb_stor_sense_invalidCDB,
 633		       sizeof(usb_stor_sense_invalidCDB));
 634		cmnd->result = SAM_STAT_CHECK_CONDITION;
 635		cmnd->scsi_done(cmnd);
 636		return 0;
 637	}
 638
 639	spin_lock_irqsave(&devinfo->lock, flags);
 640
 641	if (devinfo->resetting) {
 642		cmnd->result = DID_ERROR << 16;
 643		cmnd->scsi_done(cmnd);
 644		spin_unlock_irqrestore(&devinfo->lock, flags);
 645		return 0;
 646	}
 647
 648	/* Find a free uas-tag */
 649	for (idx = 0; idx < devinfo->qdepth; idx++) {
 650		if (!devinfo->cmnd[idx])
 651			break;
 652	}
 653	if (idx == devinfo->qdepth) {
 654		spin_unlock_irqrestore(&devinfo->lock, flags);
 655		return SCSI_MLQUEUE_DEVICE_BUSY;
 656	}
 657
 658	cmnd->scsi_done = done;
 659
 660	memset(cmdinfo, 0, sizeof(*cmdinfo));
 661	cmdinfo->uas_tag = idx + 1; /* uas-tag == usb-stream-id, so 1 based */
 662	cmdinfo->state = SUBMIT_STATUS_URB | ALLOC_CMD_URB | SUBMIT_CMD_URB;
 663
 664	switch (cmnd->sc_data_direction) {
 665	case DMA_FROM_DEVICE:
 666		cmdinfo->state |= ALLOC_DATA_IN_URB | SUBMIT_DATA_IN_URB;
 667		break;
 668	case DMA_BIDIRECTIONAL:
 669		cmdinfo->state |= ALLOC_DATA_IN_URB | SUBMIT_DATA_IN_URB;
 670		/* fall through */
 671	case DMA_TO_DEVICE:
 672		cmdinfo->state |= ALLOC_DATA_OUT_URB | SUBMIT_DATA_OUT_URB;
 673	case DMA_NONE:
 674		break;
 675	}
 676
 677	if (!devinfo->use_streams)
 678		cmdinfo->state &= ~(SUBMIT_DATA_IN_URB | SUBMIT_DATA_OUT_URB);
 
 
 679
 680	err = uas_submit_urbs(cmnd, devinfo);
 681	if (err) {
 682		/* If we did nothing, give up now */
 683		if (cmdinfo->state & SUBMIT_STATUS_URB) {
 684			spin_unlock_irqrestore(&devinfo->lock, flags);
 685			return SCSI_MLQUEUE_DEVICE_BUSY;
 686		}
 687		uas_add_work(cmdinfo);
 
 
 
 688	}
 689
 690	devinfo->cmnd[idx] = cmnd;
 691	spin_unlock_irqrestore(&devinfo->lock, flags);
 692	return 0;
 693}
 694
 695static DEF_SCSI_QCMD(uas_queuecommand)
 696
 697/*
 698 * For now we do not support actually sending an abort to the device, so
 699 * this eh always fails. Still we must define it to make sure that we've
 700 * dropped all references to the cmnd in question once this function exits.
 701 */
 702static int uas_eh_abort_handler(struct scsi_cmnd *cmnd)
 703{
 704	struct uas_cmd_info *cmdinfo = (void *)&cmnd->SCp;
 705	struct uas_dev_info *devinfo = (void *)cmnd->device->hostdata;
 706	struct urb *data_in_urb = NULL;
 707	struct urb *data_out_urb = NULL;
 708	unsigned long flags;
 709
 710	spin_lock_irqsave(&devinfo->lock, flags);
 
 
 711
 712	uas_log_cmd_state(cmnd, __func__, 0);
 
 
 
 
 713
 714	/* Ensure that try_complete does not call scsi_done */
 715	cmdinfo->state |= COMMAND_ABORTED;
 
 716
 717	/* Drop all refs to this cmnd, kill data urbs to break their ref */
 718	devinfo->cmnd[cmdinfo->uas_tag - 1] = NULL;
 719	if (cmdinfo->state & DATA_IN_URB_INFLIGHT)
 720		data_in_urb = usb_get_urb(cmdinfo->data_in_urb);
 721	if (cmdinfo->state & DATA_OUT_URB_INFLIGHT)
 722		data_out_urb = usb_get_urb(cmdinfo->data_out_urb);
 723
 724	uas_free_unsubmitted_urbs(cmnd);
 725
 726	spin_unlock_irqrestore(&devinfo->lock, flags);
 727
 728	if (data_in_urb) {
 729		usb_kill_urb(data_in_urb);
 730		usb_put_urb(data_in_urb);
 731	}
 732	if (data_out_urb) {
 733		usb_kill_urb(data_out_urb);
 734		usb_put_urb(data_out_urb);
 735	}
 736
 
 
 
 737	return FAILED;
 738}
 739
 740static int uas_eh_device_reset_handler(struct scsi_cmnd *cmnd)
 741{
 742	struct scsi_device *sdev = cmnd->device;
 743	struct uas_dev_info *devinfo = sdev->hostdata;
 744	struct usb_device *udev = devinfo->udev;
 745	unsigned long flags;
 746	int err;
 747
 748	err = usb_lock_device_for_reset(udev, devinfo->intf);
 749	if (err) {
 750		shost_printk(KERN_ERR, sdev->host,
 751			     "%s FAILED to get lock err %d\n", __func__, err);
 752		return FAILED;
 753	}
 754
 755	shost_printk(KERN_INFO, sdev->host, "%s start\n", __func__);
 
 756
 757	spin_lock_irqsave(&devinfo->lock, flags);
 758	devinfo->resetting = 1;
 759	spin_unlock_irqrestore(&devinfo->lock, flags);
 760
 761	usb_kill_anchored_urbs(&devinfo->cmd_urbs);
 762	usb_kill_anchored_urbs(&devinfo->sense_urbs);
 763	usb_kill_anchored_urbs(&devinfo->data_urbs);
 764	uas_zap_pending(devinfo, DID_RESET);
 765
 766	err = usb_reset_device(udev);
 767
 768	spin_lock_irqsave(&devinfo->lock, flags);
 769	devinfo->resetting = 0;
 770	spin_unlock_irqrestore(&devinfo->lock, flags);
 771
 772	usb_unlock_device(udev);
 773
 774	if (err) {
 775		shost_printk(KERN_INFO, sdev->host, "%s FAILED err %d\n",
 776			     __func__, err);
 777		return FAILED;
 778	}
 779
 780	shost_printk(KERN_INFO, sdev->host, "%s success\n", __func__);
 781	return SUCCESS;
 782}
 783
 784static int uas_target_alloc(struct scsi_target *starget)
 785{
 786	struct uas_dev_info *devinfo = (struct uas_dev_info *)
 787			dev_to_shost(starget->dev.parent)->hostdata;
 788
 789	if (devinfo->flags & US_FL_NO_REPORT_LUNS)
 790		starget->no_report_luns = 1;
 791
 792	return 0;
 793}
 794
 795static int uas_slave_alloc(struct scsi_device *sdev)
 796{
 797	struct uas_dev_info *devinfo =
 798		(struct uas_dev_info *)sdev->host->hostdata;
 799
 800	sdev->hostdata = devinfo;
 801
 802	/*
 803	 * USB has unusual DMA-alignment requirements: Although the
 804	 * starting address of each scatter-gather element doesn't matter,
 805	 * the length of each element except the last must be divisible
 806	 * by the Bulk maxpacket value.  There's currently no way to
 807	 * express this by block-layer constraints, so we'll cop out
 808	 * and simply require addresses to be aligned at 512-byte
 809	 * boundaries.  This is okay since most block I/O involves
 810	 * hardware sectors that are multiples of 512 bytes in length,
 811	 * and since host controllers up through USB 2.0 have maxpacket
 812	 * values no larger than 512.
 813	 *
 814	 * But it doesn't suffice for Wireless USB, where Bulk maxpacket
 815	 * values can be as large as 2048.  To make that work properly
 816	 * will require changes to the block layer.
 817	 */
 818	blk_queue_update_dma_alignment(sdev->request_queue, (512 - 1));
 819
 820	if (devinfo->flags & US_FL_MAX_SECTORS_64)
 821		blk_queue_max_hw_sectors(sdev->request_queue, 64);
 822	else if (devinfo->flags & US_FL_MAX_SECTORS_240)
 823		blk_queue_max_hw_sectors(sdev->request_queue, 240);
 824
 825	return 0;
 826}
 827
 828static int uas_slave_configure(struct scsi_device *sdev)
 829{
 830	struct uas_dev_info *devinfo = sdev->hostdata;
 831
 832	if (devinfo->flags & US_FL_NO_REPORT_OPCODES)
 833		sdev->no_report_opcodes = 1;
 834
 835	/* A few buggy USB-ATA bridges don't understand FUA */
 836	if (devinfo->flags & US_FL_BROKEN_FUA)
 837		sdev->broken_fua = 1;
 838
 839	scsi_change_queue_depth(sdev, devinfo->qdepth - 2);
 840	return 0;
 841}
 842
 843static struct scsi_host_template uas_host_template = {
 844	.module = THIS_MODULE,
 845	.name = "uas",
 846	.queuecommand = uas_queuecommand,
 847	.target_alloc = uas_target_alloc,
 848	.slave_alloc = uas_slave_alloc,
 849	.slave_configure = uas_slave_configure,
 850	.eh_abort_handler = uas_eh_abort_handler,
 851	.eh_device_reset_handler = uas_eh_device_reset_handler,
 
 
 
 852	.this_id = -1,
 853	.sg_tablesize = SG_NONE,
 
 854	.skip_settle_delay = 1,
 
 855};
 856
 857#define UNUSUAL_DEV(id_vendor, id_product, bcdDeviceMin, bcdDeviceMax, \
 858		    vendorName, productName, useProtocol, useTransport, \
 859		    initFunction, flags) \
 860{ USB_DEVICE_VER(id_vendor, id_product, bcdDeviceMin, bcdDeviceMax), \
 861	.driver_info = (flags) }
 862
 863static struct usb_device_id uas_usb_ids[] = {
 864#	include "unusual_uas.h"
 865	{ USB_INTERFACE_INFO(USB_CLASS_MASS_STORAGE, USB_SC_SCSI, USB_PR_BULK) },
 866	{ USB_INTERFACE_INFO(USB_CLASS_MASS_STORAGE, USB_SC_SCSI, USB_PR_UAS) },
 
 
 867	{ }
 868};
 869MODULE_DEVICE_TABLE(usb, uas_usb_ids);
 870
 871#undef UNUSUAL_DEV
 
 
 
 
 
 872
 873static int uas_switch_interface(struct usb_device *udev,
 874				struct usb_interface *intf)
 875{
 876	struct usb_host_interface *alt;
 877
 878	alt = uas_find_uas_alt_setting(intf);
 879	if (!alt)
 880		return -ENODEV;
 
 
 
 
 
 
 
 
 
 881
 882	return usb_set_interface(udev, alt->desc.bInterfaceNumber,
 883			alt->desc.bAlternateSetting);
 884}
 885
 886static int uas_configure_endpoints(struct uas_dev_info *devinfo)
 887{
 888	struct usb_host_endpoint *eps[4] = { };
 
 889	struct usb_device *udev = devinfo->udev;
 890	int r;
 
 891
 892	r = uas_find_endpoints(devinfo->intf->cur_altsetting, eps);
 893	if (r)
 894		return r;
 895
 896	devinfo->cmd_pipe = usb_sndbulkpipe(udev,
 897					    usb_endpoint_num(&eps[0]->desc));
 898	devinfo->status_pipe = usb_rcvbulkpipe(udev,
 899					    usb_endpoint_num(&eps[1]->desc));
 900	devinfo->data_in_pipe = usb_rcvbulkpipe(udev,
 901					    usb_endpoint_num(&eps[2]->desc));
 902	devinfo->data_out_pipe = usb_sndbulkpipe(udev,
 903					    usb_endpoint_num(&eps[3]->desc));
 904
 905	if (udev->speed < USB_SPEED_SUPER) {
 906		devinfo->qdepth = 32;
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 907		devinfo->use_streams = 0;
 908	} else {
 909		devinfo->qdepth = usb_alloc_streams(devinfo->intf, eps + 1,
 910						    3, MAX_CMNDS, GFP_NOIO);
 911		if (devinfo->qdepth < 0)
 912			return devinfo->qdepth;
 913		devinfo->use_streams = 1;
 914	}
 915
 916	return 0;
 917}
 918
 919static void uas_free_streams(struct uas_dev_info *devinfo)
 920{
 921	struct usb_device *udev = devinfo->udev;
 922	struct usb_host_endpoint *eps[3];
 923
 924	eps[0] = usb_pipe_endpoint(udev, devinfo->status_pipe);
 925	eps[1] = usb_pipe_endpoint(udev, devinfo->data_in_pipe);
 926	eps[2] = usb_pipe_endpoint(udev, devinfo->data_out_pipe);
 927	usb_free_streams(devinfo->intf, eps, 3, GFP_NOIO);
 928}
 929
 
 
 
 
 
 
 930static int uas_probe(struct usb_interface *intf, const struct usb_device_id *id)
 931{
 932	int result = -ENOMEM;
 933	struct Scsi_Host *shost = NULL;
 934	struct uas_dev_info *devinfo;
 935	struct usb_device *udev = interface_to_usbdev(intf);
 936	unsigned long dev_flags;
 937
 938	if (!uas_use_uas_driver(intf, id, &dev_flags))
 939		return -ENODEV;
 940
 941	if (uas_switch_interface(udev, intf))
 942		return -ENODEV;
 
 943
 944	shost = scsi_host_alloc(&uas_host_template,
 945				sizeof(struct uas_dev_info));
 946	if (!shost)
 947		goto set_alt0;
 948
 949	shost->max_cmd_len = 16 + 252;
 950	shost->max_id = 1;
 951	shost->max_lun = 256;
 952	shost->max_channel = 0;
 953	shost->sg_tablesize = udev->bus->sg_tablesize;
 954
 955	devinfo = (struct uas_dev_info *)shost->hostdata;
 
 
 
 
 956	devinfo->intf = intf;
 957	devinfo->udev = udev;
 958	devinfo->resetting = 0;
 959	devinfo->shutdown = 0;
 960	devinfo->flags = dev_flags;
 961	init_usb_anchor(&devinfo->cmd_urbs);
 962	init_usb_anchor(&devinfo->sense_urbs);
 963	init_usb_anchor(&devinfo->data_urbs);
 964	spin_lock_init(&devinfo->lock);
 965	INIT_WORK(&devinfo->work, uas_do_work);
 966
 967	result = uas_configure_endpoints(devinfo);
 968	if (result)
 969		goto set_alt0;
 970
 971	/*
 972	 * 1 tag is reserved for untagged commands +
 973	 * 1 tag to avoid off by one errors in some bridge firmwares
 974	 */
 975	shost->can_queue = devinfo->qdepth - 2;
 976
 
 977	usb_set_intfdata(intf, shost);
 978	result = scsi_add_host(shost, &intf->dev);
 979	if (result)
 980		goto free_streams;
 981
 982	scsi_scan_host(shost);
 983	return result;
 984
 985free_streams:
 986	uas_free_streams(devinfo);
 987	usb_set_intfdata(intf, NULL);
 988set_alt0:
 989	usb_set_interface(udev, intf->altsetting[0].desc.bInterfaceNumber, 0);
 990	if (shost)
 991		scsi_host_put(shost);
 992	return result;
 993}
 994
 995static int uas_cmnd_list_empty(struct uas_dev_info *devinfo)
 996{
 997	unsigned long flags;
 998	int i, r = 1;
 999
1000	spin_lock_irqsave(&devinfo->lock, flags);
1001
1002	for (i = 0; i < devinfo->qdepth; i++) {
1003		if (devinfo->cmnd[i]) {
1004			r = 0; /* Not empty */
1005			break;
1006		}
1007	}
1008
1009	spin_unlock_irqrestore(&devinfo->lock, flags);
1010
1011	return r;
1012}
1013
1014/*
1015 * Wait for any pending cmnds to complete, on usb-2 sense_urbs may temporarily
1016 * get empty while there still is more work to do due to sense-urbs completing
1017 * with a READ/WRITE_READY iu code, so keep waiting until the list gets empty.
1018 */
1019static int uas_wait_for_pending_cmnds(struct uas_dev_info *devinfo)
1020{
1021	unsigned long start_time;
1022	int r;
1023
1024	start_time = jiffies;
1025	do {
1026		flush_work(&devinfo->work);
1027
1028		r = usb_wait_anchor_empty_timeout(&devinfo->sense_urbs, 5000);
1029		if (r == 0)
1030			return -ETIME;
1031
1032		r = usb_wait_anchor_empty_timeout(&devinfo->data_urbs, 500);
1033		if (r == 0)
1034			return -ETIME;
1035
1036		if (time_after(jiffies, start_time + 5 * HZ))
1037			return -ETIME;
1038	} while (!uas_cmnd_list_empty(devinfo));
1039
1040	return 0;
1041}
1042
1043static int uas_pre_reset(struct usb_interface *intf)
1044{
1045	struct Scsi_Host *shost = usb_get_intfdata(intf);
1046	struct uas_dev_info *devinfo = (struct uas_dev_info *)shost->hostdata;
1047	unsigned long flags;
1048
1049	if (devinfo->shutdown)
1050		return 0;
1051
1052	/* Block new requests */
1053	spin_lock_irqsave(shost->host_lock, flags);
1054	scsi_block_requests(shost);
1055	spin_unlock_irqrestore(shost->host_lock, flags);
1056
1057	if (uas_wait_for_pending_cmnds(devinfo) != 0) {
1058		shost_printk(KERN_ERR, shost, "%s: timed out\n", __func__);
1059		scsi_unblock_requests(shost);
1060		return 1;
1061	}
1062
1063	uas_free_streams(devinfo);
1064
1065	return 0;
1066}
1067
1068static int uas_post_reset(struct usb_interface *intf)
1069{
1070	struct Scsi_Host *shost = usb_get_intfdata(intf);
1071	struct uas_dev_info *devinfo = (struct uas_dev_info *)shost->hostdata;
1072	unsigned long flags;
1073	int err;
1074
1075	if (devinfo->shutdown)
1076		return 0;
1077
1078	err = uas_configure_endpoints(devinfo);
1079	if (err && err != -ENODEV)
1080		shost_printk(KERN_ERR, shost,
1081			     "%s: alloc streams error %d after reset",
1082			     __func__, err);
1083
1084	/* we must unblock the host in every case lest we deadlock */
1085	spin_lock_irqsave(shost->host_lock, flags);
1086	scsi_report_bus_reset(shost, 0);
1087	spin_unlock_irqrestore(shost->host_lock, flags);
1088
1089	scsi_unblock_requests(shost);
1090
1091	return err ? 1 : 0;
1092}
1093
1094static int uas_suspend(struct usb_interface *intf, pm_message_t message)
1095{
1096	struct Scsi_Host *shost = usb_get_intfdata(intf);
1097	struct uas_dev_info *devinfo = (struct uas_dev_info *)shost->hostdata;
1098
1099	if (uas_wait_for_pending_cmnds(devinfo) != 0) {
1100		shost_printk(KERN_ERR, shost, "%s: timed out\n", __func__);
1101		return -ETIME;
1102	}
1103
1104	return 0;
1105}
1106
1107static int uas_resume(struct usb_interface *intf)
1108{
1109	return 0;
1110}
1111
1112static int uas_reset_resume(struct usb_interface *intf)
1113{
 
 
1114	struct Scsi_Host *shost = usb_get_intfdata(intf);
1115	struct uas_dev_info *devinfo = (struct uas_dev_info *)shost->hostdata;
1116	unsigned long flags;
1117	int err;
1118
1119	err = uas_configure_endpoints(devinfo);
1120	if (err) {
1121		shost_printk(KERN_ERR, shost,
1122			     "%s: alloc streams error %d after reset",
1123			     __func__, err);
1124		return -EIO;
1125	}
1126
1127	spin_lock_irqsave(shost->host_lock, flags);
1128	scsi_report_bus_reset(shost, 0);
1129	spin_unlock_irqrestore(shost->host_lock, flags);
1130
1131	return 0;
1132}
1133
1134static void uas_disconnect(struct usb_interface *intf)
1135{
1136	struct Scsi_Host *shost = usb_get_intfdata(intf);
1137	struct uas_dev_info *devinfo = (struct uas_dev_info *)shost->hostdata;
1138	unsigned long flags;
1139
1140	spin_lock_irqsave(&devinfo->lock, flags);
1141	devinfo->resetting = 1;
1142	spin_unlock_irqrestore(&devinfo->lock, flags);
1143
1144	cancel_work_sync(&devinfo->work);
1145	usb_kill_anchored_urbs(&devinfo->cmd_urbs);
1146	usb_kill_anchored_urbs(&devinfo->sense_urbs);
1147	usb_kill_anchored_urbs(&devinfo->data_urbs);
1148	uas_zap_pending(devinfo, DID_NO_CONNECT);
1149
1150	scsi_remove_host(shost);
1151	uas_free_streams(devinfo);
1152	scsi_host_put(shost);
1153}
1154
1155/*
1156 * Put the device back in usb-storage mode on shutdown, as some BIOS-es
1157 * hang on reboot when the device is still in uas mode. Note the reset is
1158 * necessary as some devices won't revert to usb-storage mode without it.
1159 */
1160static void uas_shutdown(struct device *dev)
1161{
1162	struct usb_interface *intf = to_usb_interface(dev);
1163	struct usb_device *udev = interface_to_usbdev(intf);
1164	struct Scsi_Host *shost = usb_get_intfdata(intf);
1165	struct uas_dev_info *devinfo = (struct uas_dev_info *)shost->hostdata;
1166
1167	if (system_state != SYSTEM_RESTART)
1168		return;
1169
1170	devinfo->shutdown = 1;
1171	uas_free_streams(devinfo);
1172	usb_set_interface(udev, intf->altsetting[0].desc.bInterfaceNumber, 0);
1173	usb_reset_device(udev);
1174}
1175
1176static struct usb_driver uas_driver = {
1177	.name = "uas",
1178	.probe = uas_probe,
1179	.disconnect = uas_disconnect,
1180	.pre_reset = uas_pre_reset,
1181	.post_reset = uas_post_reset,
1182	.suspend = uas_suspend,
1183	.resume = uas_resume,
1184	.reset_resume = uas_reset_resume,
1185	.drvwrap.driver.shutdown = uas_shutdown,
1186	.id_table = uas_usb_ids,
1187};
1188
1189module_usb_driver(uas_driver);
 
 
 
 
 
 
 
 
 
 
 
1190
1191MODULE_LICENSE("GPL");
1192MODULE_AUTHOR(
1193	"Hans de Goede <hdegoede@redhat.com>, Matthew Wilcox and Sarah Sharp");