Linux Audio

Check our new training course

Loading...
v3.5.6
 
  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/module.h>
 15#include <linux/usb.h>
 
 16#include <linux/usb/hcd.h>
 17#include <linux/usb/storage.h>
 18#include <linux/usb/uas.h>
 19
 20#include <scsi/scsi.h>
 
 21#include <scsi/scsi_dbg.h>
 22#include <scsi/scsi_cmnd.h>
 23#include <scsi/scsi_device.h>
 24#include <scsi/scsi_host.h>
 25#include <scsi/scsi_tcq.h>
 26
 27/*
 28 * The r00-r01c specs define this version of the SENSE IU data structure.
 29 * It's still in use by several different firmware releases.
 30 */
 31struct sense_iu_old {
 32	__u8 iu_id;
 33	__u8 rsvd1;
 34	__be16 tag;
 35	__be16 len;
 36	__u8 status;
 37	__u8 service_response;
 38	__u8 sense[SCSI_SENSE_BUFFERSIZE];
 39};
 40
 41struct uas_dev_info {
 42	struct usb_interface *intf;
 43	struct usb_device *udev;
 44	int qdepth;
 
 
 
 
 45	unsigned cmd_pipe, status_pipe, data_in_pipe, data_out_pipe;
 46	unsigned use_streams:1;
 47	unsigned uas_sense_old:1;
 48	struct scsi_cmnd *cmnd;
 49	struct urb *status_urb; /* used only if stream support is available */
 
 50};
 51
 52enum {
 53	ALLOC_STATUS_URB	= (1 << 0),
 54	SUBMIT_STATUS_URB	= (1 << 1),
 55	ALLOC_DATA_IN_URB	= (1 << 2),
 56	SUBMIT_DATA_IN_URB	= (1 << 3),
 57	ALLOC_DATA_OUT_URB	= (1 << 4),
 58	SUBMIT_DATA_OUT_URB	= (1 << 5),
 59	ALLOC_CMD_URB		= (1 << 6),
 60	SUBMIT_CMD_URB		= (1 << 7),
 
 
 
 
 61};
 62
 63/* Overrides scsi_pointer */
 64struct uas_cmd_info {
 65	unsigned int state;
 66	unsigned int stream;
 67	struct urb *cmd_urb;
 68	/* status_urb is used only if stream support isn't available */
 69	struct urb *status_urb;
 70	struct urb *data_in_urb;
 71	struct urb *data_out_urb;
 72	struct list_head list;
 73};
 74
 75/* I hate forward declarations, but I actually have a loop */
 76static int uas_submit_urbs(struct scsi_cmnd *cmnd,
 77				struct uas_dev_info *devinfo, gfp_t gfp);
 78static void uas_do_work(struct work_struct *work);
 79
 80static DECLARE_WORK(uas_work, uas_do_work);
 81static DEFINE_SPINLOCK(uas_work_lock);
 82static LIST_HEAD(uas_work_list);
 83
 84static void uas_do_work(struct work_struct *work)
 85{
 
 
 86	struct uas_cmd_info *cmdinfo;
 87	struct uas_cmd_info *temp;
 88	struct list_head list;
 89	int err;
 90
 91	spin_lock_irq(&uas_work_lock);
 92	list_replace_init(&uas_work_list, &list);
 93	spin_unlock_irq(&uas_work_lock);
 94
 95	list_for_each_entry_safe(cmdinfo, temp, &list, list) {
 96		struct scsi_pointer *scp = (void *)cmdinfo;
 97		struct scsi_cmnd *cmnd = container_of(scp,
 98							struct scsi_cmnd, SCp);
 99		err = uas_submit_urbs(cmnd, cmnd->device->hostdata, GFP_NOIO);
100		if (err) {
101			list_del(&cmdinfo->list);
102			spin_lock_irq(&uas_work_lock);
103			list_add_tail(&cmdinfo->list, &uas_work_list);
104			spin_unlock_irq(&uas_work_lock);
105			schedule_work(&uas_work);
106		}
 
 
 
 
107	}
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
108}
109
110static void uas_sense(struct urb *urb, struct scsi_cmnd *cmnd)
111{
112	struct sense_iu *sense_iu = urb->transfer_buffer;
113	struct scsi_device *sdev = cmnd->device;
114
115	if (urb->actual_length > 16) {
116		unsigned len = be16_to_cpup(&sense_iu->len);
117		if (len + 16 != urb->actual_length) {
118			int newlen = min(len + 16, urb->actual_length) - 16;
119			if (newlen < 0)
120				newlen = 0;
121			sdev_printk(KERN_INFO, sdev, "%s: urb length %d "
122				"disagrees with IU sense data length %d, "
123				"using %d bytes of sense data\n", __func__,
124					urb->actual_length, len, newlen);
125			len = newlen;
126		}
127		memcpy(cmnd->sense_buffer, sense_iu->sense, len);
128	}
129
130	cmnd->result = sense_iu->status;
131	cmnd->scsi_done(cmnd);
132}
133
134static void uas_sense_old(struct urb *urb, struct scsi_cmnd *cmnd)
 
135{
136	struct sense_iu_old *sense_iu = urb->transfer_buffer;
137	struct scsi_device *sdev = cmnd->device;
138
139	if (urb->actual_length > 8) {
140		unsigned len = be16_to_cpup(&sense_iu->len) - 2;
141		if (len + 8 != urb->actual_length) {
142			int newlen = min(len + 8, urb->actual_length) - 8;
143			if (newlen < 0)
144				newlen = 0;
145			sdev_printk(KERN_INFO, sdev, "%s: urb length %d "
146				"disagrees with IU sense data length %d, "
147				"using %d bytes of sense data\n", __func__,
148					urb->actual_length, len, newlen);
149			len = newlen;
150		}
151		memcpy(cmnd->sense_buffer, sense_iu->sense, len);
152	}
 
 
 
153
154	cmnd->result = sense_iu->status;
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
155	cmnd->scsi_done(cmnd);
 
156}
157
158static void uas_xfer_data(struct urb *urb, struct scsi_cmnd *cmnd,
159							unsigned direction)
160{
161	struct uas_cmd_info *cmdinfo = (void *)&cmnd->SCp;
162	int err;
163
164	cmdinfo->state = direction;
165	err = uas_submit_urbs(cmnd, cmnd->device->hostdata, GFP_ATOMIC);
166	if (err) {
167		spin_lock(&uas_work_lock);
168		list_add_tail(&cmdinfo->list, &uas_work_list);
169		spin_unlock(&uas_work_lock);
170		schedule_work(&uas_work);
171	}
172}
173
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
174static void uas_stat_cmplt(struct urb *urb)
175{
176	struct iu *iu = urb->transfer_buffer;
177	struct Scsi_Host *shost = urb->context;
178	struct uas_dev_info *devinfo = (void *)shost->hostdata[0];
 
 
179	struct scsi_cmnd *cmnd;
180	u16 tag;
181	int ret;
 
 
 
182
183	if (urb->status) {
184		dev_err(&urb->dev->dev, "URB BAD STATUS %d\n", urb->status);
185		if (devinfo->use_streams)
186			usb_free_urb(urb);
187		return;
 
 
 
 
188	}
189
190	tag = be16_to_cpup(&iu->tag) - 1;
191	if (tag == 0)
192		cmnd = devinfo->cmnd;
193	else
194		cmnd = scsi_host_find_tag(shost, tag - 1);
195	if (!cmnd) {
196		if (devinfo->use_streams) {
197			usb_free_urb(urb);
198			return;
199		}
200		ret = usb_submit_urb(urb, GFP_ATOMIC);
201		if (ret)
202			dev_err(&urb->dev->dev, "failed submit status urb\n");
203		return;
204	}
205
206	switch (iu->iu_id) {
207	case IU_ID_STATUS:
208		if (devinfo->cmnd == cmnd)
209			devinfo->cmnd = NULL;
210
211		if (urb->actual_length < 16)
212			devinfo->uas_sense_old = 1;
213		if (devinfo->uas_sense_old)
214			uas_sense_old(urb, cmnd);
215		else
216			uas_sense(urb, cmnd);
217		break;
218	case IU_ID_READ_READY:
 
 
 
 
 
219		uas_xfer_data(urb, cmnd, SUBMIT_DATA_IN_URB);
220		break;
221	case IU_ID_WRITE_READY:
 
 
 
 
 
222		uas_xfer_data(urb, cmnd, SUBMIT_DATA_OUT_URB);
223		break;
 
 
 
 
 
 
 
 
 
 
224	default:
225		scmd_printk(KERN_ERR, cmnd,
226			"Bogus IU (%d) received on status pipe\n", iu->iu_id);
227	}
 
 
 
228
229	if (devinfo->use_streams) {
230		usb_free_urb(urb);
231		return;
 
 
 
 
 
232	}
233
234	ret = usb_submit_urb(urb, GFP_ATOMIC);
235	if (ret)
236		dev_err(&urb->dev->dev, "failed submit status urb\n");
237}
238
239static void uas_data_cmplt(struct urb *urb)
240{
241	struct scsi_data_buffer *sdb = urb->context;
242	sdb->resid = sdb->length - urb->actual_length;
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
243	usb_free_urb(urb);
244}
245
246static struct urb *uas_alloc_data_urb(struct uas_dev_info *devinfo, gfp_t gfp,
247				unsigned int pipe, u16 stream_id,
248				struct scsi_data_buffer *sdb,
249				enum dma_data_direction dir)
250{
251	struct usb_device *udev = devinfo->udev;
 
252	struct urb *urb = usb_alloc_urb(0, gfp);
 
 
 
 
253
254	if (!urb)
255		goto out;
256	usb_fill_bulk_urb(urb, udev, pipe, NULL, sdb->length, uas_data_cmplt,
257									sdb);
258	if (devinfo->use_streams)
259		urb->stream_id = stream_id;
260	urb->num_sgs = udev->bus->sg_tablesize ? sdb->table.nents : 0;
261	urb->sg = sdb->table.sgl;
262 out:
263	return urb;
264}
265
266static struct urb *uas_alloc_sense_urb(struct uas_dev_info *devinfo, gfp_t gfp,
267		struct Scsi_Host *shost, u16 stream_id)
268{
269	struct usb_device *udev = devinfo->udev;
 
270	struct urb *urb = usb_alloc_urb(0, gfp);
271	struct sense_iu *iu;
272
273	if (!urb)
274		goto out;
275
276	iu = kzalloc(sizeof(*iu), gfp);
277	if (!iu)
278		goto free;
279
280	usb_fill_bulk_urb(urb, udev, devinfo->status_pipe, iu, sizeof(*iu),
281						uas_stat_cmplt, shost);
282	urb->stream_id = stream_id;
 
283	urb->transfer_flags |= URB_FREE_BUFFER;
284 out:
285	return urb;
286 free:
287	usb_free_urb(urb);
288	return NULL;
289}
290
291static struct urb *uas_alloc_cmd_urb(struct uas_dev_info *devinfo, gfp_t gfp,
292					struct scsi_cmnd *cmnd, u16 stream_id)
293{
294	struct usb_device *udev = devinfo->udev;
295	struct scsi_device *sdev = cmnd->device;
 
296	struct urb *urb = usb_alloc_urb(0, gfp);
297	struct command_iu *iu;
298	int len;
299
300	if (!urb)
301		goto out;
302
303	len = cmnd->cmd_len - 16;
304	if (len < 0)
305		len = 0;
306	len = ALIGN(len, 4);
307	iu = kzalloc(sizeof(*iu) + len, gfp);
308	if (!iu)
309		goto free;
310
311	iu->iu_id = IU_ID_COMMAND;
312	if (blk_rq_tagged(cmnd->request))
313		iu->tag = cpu_to_be16(cmnd->request->tag + 2);
314	else
315		iu->tag = cpu_to_be16(1);
316	iu->prio_attr = UAS_SIMPLE_TAG;
317	iu->len = len;
318	int_to_scsilun(sdev->lun, &iu->lun);
319	memcpy(iu->cdb, cmnd->cmnd, cmnd->cmd_len);
320
321	usb_fill_bulk_urb(urb, udev, devinfo->cmd_pipe, iu, sizeof(*iu) + len,
322							usb_free_urb, NULL);
323	urb->transfer_flags |= URB_FREE_BUFFER;
324 out:
325	return urb;
326 free:
327	usb_free_urb(urb);
328	return NULL;
329}
330
331/*
332 * Why should I request the Status IU before sending the Command IU?  Spec
333 * says to, but also says the device may receive them in any order.  Seems
334 * daft to me.
335 */
336
337static int uas_submit_urbs(struct scsi_cmnd *cmnd,
338					struct uas_dev_info *devinfo, gfp_t gfp)
339{
340	struct uas_cmd_info *cmdinfo = (void *)&cmnd->SCp;
 
 
341
342	if (cmdinfo->state & ALLOC_STATUS_URB) {
343		cmdinfo->status_urb = uas_alloc_sense_urb(devinfo, gfp,
344				cmnd->device->host, cmdinfo->stream);
345		if (!cmdinfo->status_urb)
346			return SCSI_MLQUEUE_DEVICE_BUSY;
347		cmdinfo->state &= ~ALLOC_STATUS_URB;
 
 
 
 
348	}
 
 
349
 
 
 
 
 
 
 
 
350	if (cmdinfo->state & SUBMIT_STATUS_URB) {
351		if (usb_submit_urb(cmdinfo->status_urb, gfp)) {
352			scmd_printk(KERN_INFO, cmnd,
353					"sense urb submission failure\n");
354			return SCSI_MLQUEUE_DEVICE_BUSY;
355		}
356		cmdinfo->state &= ~SUBMIT_STATUS_URB;
357	}
358
359	if (cmdinfo->state & ALLOC_DATA_IN_URB) {
360		cmdinfo->data_in_urb = uas_alloc_data_urb(devinfo, gfp,
361					devinfo->data_in_pipe, cmdinfo->stream,
362					scsi_in(cmnd), DMA_FROM_DEVICE);
363		if (!cmdinfo->data_in_urb)
364			return SCSI_MLQUEUE_DEVICE_BUSY;
365		cmdinfo->state &= ~ALLOC_DATA_IN_URB;
366	}
367
368	if (cmdinfo->state & SUBMIT_DATA_IN_URB) {
369		if (usb_submit_urb(cmdinfo->data_in_urb, gfp)) {
370			scmd_printk(KERN_INFO, cmnd,
371					"data in urb submission failure\n");
 
 
372			return SCSI_MLQUEUE_DEVICE_BUSY;
373		}
374		cmdinfo->state &= ~SUBMIT_DATA_IN_URB;
 
375	}
376
377	if (cmdinfo->state & ALLOC_DATA_OUT_URB) {
378		cmdinfo->data_out_urb = uas_alloc_data_urb(devinfo, gfp,
379					devinfo->data_out_pipe, cmdinfo->stream,
380					scsi_out(cmnd), DMA_TO_DEVICE);
381		if (!cmdinfo->data_out_urb)
382			return SCSI_MLQUEUE_DEVICE_BUSY;
383		cmdinfo->state &= ~ALLOC_DATA_OUT_URB;
384	}
385
386	if (cmdinfo->state & SUBMIT_DATA_OUT_URB) {
387		if (usb_submit_urb(cmdinfo->data_out_urb, gfp)) {
388			scmd_printk(KERN_INFO, cmnd,
389					"data out urb submission failure\n");
 
 
390			return SCSI_MLQUEUE_DEVICE_BUSY;
391		}
392		cmdinfo->state &= ~SUBMIT_DATA_OUT_URB;
 
393	}
394
395	if (cmdinfo->state & ALLOC_CMD_URB) {
396		cmdinfo->cmd_urb = uas_alloc_cmd_urb(devinfo, gfp, cmnd,
397							cmdinfo->stream);
398		if (!cmdinfo->cmd_urb)
399			return SCSI_MLQUEUE_DEVICE_BUSY;
400		cmdinfo->state &= ~ALLOC_CMD_URB;
401	}
402
403	if (cmdinfo->state & SUBMIT_CMD_URB) {
404		if (usb_submit_urb(cmdinfo->cmd_urb, gfp)) {
405			scmd_printk(KERN_INFO, cmnd,
406					"cmd urb submission failure\n");
 
 
407			return SCSI_MLQUEUE_DEVICE_BUSY;
408		}
 
409		cmdinfo->state &= ~SUBMIT_CMD_URB;
 
410	}
411
412	return 0;
413}
414
415static int uas_queuecommand_lck(struct scsi_cmnd *cmnd,
416					void (*done)(struct scsi_cmnd *))
417{
418	struct scsi_device *sdev = cmnd->device;
419	struct uas_dev_info *devinfo = sdev->hostdata;
420	struct uas_cmd_info *cmdinfo = (void *)&cmnd->SCp;
421	int err;
 
422
423	BUILD_BUG_ON(sizeof(struct uas_cmd_info) > sizeof(struct scsi_pointer));
424
425	if (devinfo->cmnd)
 
426		return SCSI_MLQUEUE_DEVICE_BUSY;
427
428	if (blk_rq_tagged(cmnd->request)) {
429		cmdinfo->stream = cmnd->request->tag + 2;
430	} else {
431		devinfo->cmnd = cmnd;
432		cmdinfo->stream = 1;
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
433	}
434
435	cmnd->scsi_done = done;
436
437	cmdinfo->state = ALLOC_STATUS_URB | SUBMIT_STATUS_URB |
438			ALLOC_CMD_URB | SUBMIT_CMD_URB;
 
439
440	switch (cmnd->sc_data_direction) {
441	case DMA_FROM_DEVICE:
442		cmdinfo->state |= ALLOC_DATA_IN_URB | SUBMIT_DATA_IN_URB;
443		break;
444	case DMA_BIDIRECTIONAL:
445		cmdinfo->state |= ALLOC_DATA_IN_URB | SUBMIT_DATA_IN_URB;
 
446	case DMA_TO_DEVICE:
447		cmdinfo->state |= ALLOC_DATA_OUT_URB | SUBMIT_DATA_OUT_URB;
448	case DMA_NONE:
449		break;
450	}
451
452	if (!devinfo->use_streams) {
453		cmdinfo->state &= ~(SUBMIT_DATA_IN_URB | SUBMIT_DATA_OUT_URB |
454				ALLOC_STATUS_URB | SUBMIT_STATUS_URB);
455		cmdinfo->stream = 0;
456	}
457
458	err = uas_submit_urbs(cmnd, devinfo, GFP_ATOMIC);
459	if (err) {
460		/* If we did nothing, give up now */
461		if (cmdinfo->state & SUBMIT_STATUS_URB) {
462			usb_free_urb(cmdinfo->status_urb);
463			return SCSI_MLQUEUE_DEVICE_BUSY;
464		}
465		spin_lock(&uas_work_lock);
466		list_add_tail(&cmdinfo->list, &uas_work_list);
467		spin_unlock(&uas_work_lock);
468		schedule_work(&uas_work);
469	}
470
 
 
471	return 0;
472}
473
474static DEF_SCSI_QCMD(uas_queuecommand)
475
 
 
 
 
 
476static int uas_eh_abort_handler(struct scsi_cmnd *cmnd)
477{
478	struct scsi_device *sdev = cmnd->device;
479	sdev_printk(KERN_INFO, sdev, "%s tag %d\n", __func__,
480							cmnd->request->tag);
 
 
481
482/* XXX: Send ABORT TASK Task Management command */
483	return FAILED;
484}
485
486static int uas_eh_device_reset_handler(struct scsi_cmnd *cmnd)
487{
488	struct scsi_device *sdev = cmnd->device;
489	sdev_printk(KERN_INFO, sdev, "%s tag %d\n", __func__,
490							cmnd->request->tag);
491
492/* XXX: Send LOGICAL UNIT RESET Task Management command */
493	return FAILED;
494}
495
496static int uas_eh_target_reset_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: Can we reset just the one USB interface?
503 * Would calling usb_set_interface() have the right effect?
504 */
505	return FAILED;
506}
507
508static int uas_eh_bus_reset_handler(struct scsi_cmnd *cmnd)
509{
510	struct scsi_device *sdev = cmnd->device;
511	struct uas_dev_info *devinfo = sdev->hostdata;
512	struct usb_device *udev = devinfo->udev;
 
 
513
514	sdev_printk(KERN_INFO, sdev, "%s tag %d\n", __func__,
515							cmnd->request->tag);
 
 
 
 
516
517	if (usb_reset_device(udev))
518		return SUCCESS;
519
520	return FAILED;
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
521}
522
523static int uas_slave_alloc(struct scsi_device *sdev)
524{
525	sdev->hostdata = (void *)sdev->host->hostdata[0];
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
526	return 0;
527}
528
529static int uas_slave_configure(struct scsi_device *sdev)
530{
531	struct uas_dev_info *devinfo = sdev->hostdata;
532	scsi_set_tag_type(sdev, MSG_ORDERED_TAG);
533	scsi_activate_tcq(sdev, devinfo->qdepth - 2);
 
 
 
 
 
 
 
534	return 0;
535}
536
537static struct scsi_host_template uas_host_template = {
538	.module = THIS_MODULE,
539	.name = "uas",
540	.queuecommand = uas_queuecommand,
 
541	.slave_alloc = uas_slave_alloc,
542	.slave_configure = uas_slave_configure,
543	.eh_abort_handler = uas_eh_abort_handler,
544	.eh_device_reset_handler = uas_eh_device_reset_handler,
545	.eh_target_reset_handler = uas_eh_target_reset_handler,
546	.eh_bus_reset_handler = uas_eh_bus_reset_handler,
547	.can_queue = 65536,	/* Is there a limit on the _host_ ? */
548	.this_id = -1,
549	.sg_tablesize = SG_NONE,
550	.cmd_per_lun = 1,	/* until we override it */
551	.skip_settle_delay = 1,
552	.ordered_tag = 1,
553};
554
 
 
 
 
 
 
555static struct usb_device_id uas_usb_ids[] = {
 
556	{ USB_INTERFACE_INFO(USB_CLASS_MASS_STORAGE, USB_SC_SCSI, USB_PR_BULK) },
557	{ USB_INTERFACE_INFO(USB_CLASS_MASS_STORAGE, USB_SC_SCSI, USB_PR_UAS) },
558	/* 0xaa is a prototype device I happen to have access to */
559	{ USB_INTERFACE_INFO(USB_CLASS_MASS_STORAGE, USB_SC_SCSI, 0xaa) },
560	{ }
561};
562MODULE_DEVICE_TABLE(usb, uas_usb_ids);
563
564static int uas_is_interface(struct usb_host_interface *intf)
565{
566	return (intf->desc.bInterfaceClass == USB_CLASS_MASS_STORAGE &&
567		intf->desc.bInterfaceSubClass == USB_SC_SCSI &&
568		intf->desc.bInterfaceProtocol == USB_PR_UAS);
569}
570
571static int uas_isnt_supported(struct usb_device *udev)
572{
573	struct usb_hcd *hcd = bus_to_hcd(udev->bus);
574
575	dev_warn(&udev->dev, "The driver for the USB controller %s does not "
576			"support scatter-gather which is\n",
577			hcd->driver->description);
578	dev_warn(&udev->dev, "required by the UAS driver. Please try an"
579			"alternative USB controller if you wish to use UAS.\n");
580	return -ENODEV;
581}
582
583static int uas_switch_interface(struct usb_device *udev,
584						struct usb_interface *intf)
585{
586	int i;
587	int sg_supported = udev->bus->sg_tablesize != 0;
588
589	for (i = 0; i < intf->num_altsetting; i++) {
590		struct usb_host_interface *alt = &intf->altsetting[i];
591
592		if (uas_is_interface(alt)) {
593			if (!sg_supported)
594				return uas_isnt_supported(udev);
595			return usb_set_interface(udev,
596						alt->desc.bInterfaceNumber,
597						alt->desc.bAlternateSetting);
598		}
599	}
600
601	return -ENODEV;
 
602}
603
604static void uas_configure_endpoints(struct uas_dev_info *devinfo)
605{
606	struct usb_host_endpoint *eps[4] = { };
607	struct usb_interface *intf = devinfo->intf;
608	struct usb_device *udev = devinfo->udev;
609	struct usb_host_endpoint *endpoint = intf->cur_altsetting->endpoint;
610	unsigned i, n_endpoints = intf->cur_altsetting->desc.bNumEndpoints;
611
612	devinfo->uas_sense_old = 0;
613	devinfo->cmnd = NULL;
 
 
 
 
 
 
 
 
 
 
614
615	for (i = 0; i < n_endpoints; i++) {
616		unsigned char *extra = endpoint[i].extra;
617		int len = endpoint[i].extralen;
618		while (len > 1) {
619			if (extra[1] == USB_DT_PIPE_USAGE) {
620				unsigned pipe_id = extra[2];
621				if (pipe_id > 0 && pipe_id < 5)
622					eps[pipe_id - 1] = &endpoint[i];
623				break;
624			}
625			len -= extra[0];
626			extra += extra[0];
627		}
628	}
629
630	/*
631	 * Assume that if we didn't find a control pipe descriptor, we're
632	 * using a device with old firmware that happens to be set up like
633	 * this.
634	 */
635	if (!eps[0]) {
636		devinfo->cmd_pipe = usb_sndbulkpipe(udev, 1);
637		devinfo->status_pipe = usb_rcvbulkpipe(udev, 1);
638		devinfo->data_in_pipe = usb_rcvbulkpipe(udev, 2);
639		devinfo->data_out_pipe = usb_sndbulkpipe(udev, 2);
640
641		eps[1] = usb_pipe_endpoint(udev, devinfo->status_pipe);
642		eps[2] = usb_pipe_endpoint(udev, devinfo->data_in_pipe);
643		eps[3] = usb_pipe_endpoint(udev, devinfo->data_out_pipe);
644	} else {
645		devinfo->cmd_pipe = usb_sndbulkpipe(udev,
646						eps[0]->desc.bEndpointAddress);
647		devinfo->status_pipe = usb_rcvbulkpipe(udev,
648						eps[1]->desc.bEndpointAddress);
649		devinfo->data_in_pipe = usb_rcvbulkpipe(udev,
650						eps[2]->desc.bEndpointAddress);
651		devinfo->data_out_pipe = usb_sndbulkpipe(udev,
652						eps[3]->desc.bEndpointAddress);
653	}
654
655	devinfo->qdepth = usb_alloc_streams(devinfo->intf, eps + 1, 3, 256,
656								GFP_KERNEL);
657	if (devinfo->qdepth < 0) {
658		devinfo->qdepth = 256;
659		devinfo->use_streams = 0;
660	} else {
 
 
 
 
661		devinfo->use_streams = 1;
662	}
663}
664
665static int uas_alloc_status_urb(struct uas_dev_info *devinfo,
666		struct Scsi_Host *shost)
667{
668	if (devinfo->use_streams) {
669		devinfo->status_urb = NULL;
670		return 0;
671	}
672
673	devinfo->status_urb = uas_alloc_sense_urb(devinfo, GFP_KERNEL,
674			shost, 0);
675	if (!devinfo->status_urb)
676		goto err_s_urb;
677
678	if (usb_submit_urb(devinfo->status_urb, GFP_KERNEL))
679		goto err_submit_urb;
680
681	return 0;
682err_submit_urb:
683	usb_free_urb(devinfo->status_urb);
684err_s_urb:
685	return -ENOMEM;
686}
687
688static void uas_free_streams(struct uas_dev_info *devinfo)
689{
690	struct usb_device *udev = devinfo->udev;
691	struct usb_host_endpoint *eps[3];
692
693	eps[0] = usb_pipe_endpoint(udev, devinfo->status_pipe);
694	eps[1] = usb_pipe_endpoint(udev, devinfo->data_in_pipe);
695	eps[2] = usb_pipe_endpoint(udev, devinfo->data_out_pipe);
696	usb_free_streams(devinfo->intf, eps, 3, GFP_KERNEL);
697}
698
699/*
700 * XXX: What I'd like to do here is register a SCSI host for each USB host in
701 * the system.  Follow usb-storage's design of registering a SCSI host for
702 * each USB device for the moment.  Can implement this by walking up the
703 * USB hierarchy until we find a USB host.
704 */
705static int uas_probe(struct usb_interface *intf, const struct usb_device_id *id)
706{
707	int result;
708	struct Scsi_Host *shost;
709	struct uas_dev_info *devinfo;
710	struct usb_device *udev = interface_to_usbdev(intf);
 
711
712	if (uas_switch_interface(udev, intf))
713		return -ENODEV;
714
715	devinfo = kmalloc(sizeof(struct uas_dev_info), GFP_KERNEL);
716	if (!devinfo)
717		return -ENOMEM;
718
719	result = -ENOMEM;
720	shost = scsi_host_alloc(&uas_host_template, sizeof(void *));
721	if (!shost)
722		goto free;
723
724	shost->max_cmd_len = 16 + 252;
725	shost->max_id = 1;
 
 
726	shost->sg_tablesize = udev->bus->sg_tablesize;
727
 
728	devinfo->intf = intf;
729	devinfo->udev = udev;
730	uas_configure_endpoints(devinfo);
 
 
 
 
 
 
 
731
732	result = scsi_init_shared_tag_map(shost, devinfo->qdepth - 2);
733	if (result)
734		goto free;
735
736	result = scsi_add_host(shost, &intf->dev);
737	if (result)
738		goto deconfig_eps;
739
740	shost->hostdata[0] = (unsigned long)devinfo;
741
742	result = uas_alloc_status_urb(devinfo, shost);
 
743	if (result)
744		goto err_alloc_status;
745
746	scsi_scan_host(shost);
747	usb_set_intfdata(intf, shost);
748	return result;
749
750err_alloc_status:
751	scsi_remove_host(shost);
752	shost = NULL;
753deconfig_eps:
754	uas_free_streams(devinfo);
755 free:
756	kfree(devinfo);
 
757	if (shost)
758		scsi_host_put(shost);
759	return result;
760}
761
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
762static int uas_pre_reset(struct usb_interface *intf)
763{
764/* XXX: Need to return 1 if it's not our device in error handling */
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
765	return 0;
766}
767
768static int uas_post_reset(struct usb_interface *intf)
769{
770/* XXX: Need to return 1 if it's not our device in error handling */
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
771	return 0;
772}
773
774static void uas_disconnect(struct usb_interface *intf)
775{
776	struct Scsi_Host *shost = usb_get_intfdata(intf);
777	struct uas_dev_info *devinfo = (void *)shost->hostdata[0];
 
 
 
 
 
 
 
 
 
 
 
778
779	scsi_remove_host(shost);
780	usb_kill_urb(devinfo->status_urb);
781	usb_free_urb(devinfo->status_urb);
782	uas_free_streams(devinfo);
783	kfree(devinfo);
784}
785
786/*
787 * XXX: Should this plug into libusual so we can auto-upgrade devices from
788 * Bulk-Only to UAS?
 
789 */
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
790static struct usb_driver uas_driver = {
791	.name = "uas",
792	.probe = uas_probe,
793	.disconnect = uas_disconnect,
794	.pre_reset = uas_pre_reset,
795	.post_reset = uas_post_reset,
 
 
 
 
796	.id_table = uas_usb_ids,
797};
798
799module_usb_driver(uas_driver);
800
801MODULE_LICENSE("GPL");
802MODULE_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");