1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * f_sourcesink.c - USB peripheral source/sink configuration driver
4  *
5  * Copyright (C) 2003-2008 David Brownell
6  * Copyright (C) 2008 by Nokia Corporation
7  */
8 
9 /* #define VERBOSE_DEBUG */
10 
11 #include <linux/slab.h>
12 #include <linux/kernel.h>
13 #include <linux/device.h>
14 #include <linux/module.h>
15 #include <linux/usb/composite.h>
16 #include <linux/err.h>
17 
18 #include "g_zero.h"
19 #include "u_f.h"
20 
21 /*
22  * SOURCE/SINK FUNCTION ... a primary testing vehicle for USB peripheral
23  * controller drivers.
24  *
25  * This just sinks bulk packets OUT to the peripheral and sources them IN
26  * to the host, optionally with specific data patterns for integrity tests.
27  * As such it supports basic functionality and load tests.
28  *
29  * In terms of control messaging, this supports all the standard requests
30  * plus two that support control-OUT tests.  If the optional "autoresume"
31  * mode is enabled, it provides good functional coverage for the "USBCV"
32  * test harness from USB-IF.
33  */
34 struct f_sourcesink {
35 	struct usb_function	function;
36 
37 	struct usb_ep		*in_ep;
38 	struct usb_ep		*out_ep;
39 	struct usb_ep		*iso_in_ep;
40 	struct usb_ep		*iso_out_ep;
41 	int			cur_alt;
42 
43 	unsigned pattern;
44 	unsigned isoc_interval;
45 	unsigned isoc_maxpacket;
46 	unsigned isoc_mult;
47 	unsigned isoc_maxburst;
48 	unsigned buflen;
49 	unsigned bulk_qlen;
50 	unsigned iso_qlen;
51 };
52 
53 static inline struct f_sourcesink *func_to_ss(struct usb_function *f)
54 {
55 	return container_of(f, struct f_sourcesink, function);
56 }
57 
58 /*-------------------------------------------------------------------------*/
59 
60 static struct usb_interface_descriptor source_sink_intf_alt0 = {
61 	.bLength =		USB_DT_INTERFACE_SIZE,
62 	.bDescriptorType =	USB_DT_INTERFACE,
63 
64 	.bAlternateSetting =	0,
65 	.bNumEndpoints =	2,
66 	.bInterfaceClass =	USB_CLASS_VENDOR_SPEC,
67 	/* .iInterface		= DYNAMIC */
68 };
69 
70 static struct usb_interface_descriptor source_sink_intf_alt1 = {
71 	.bLength =		USB_DT_INTERFACE_SIZE,
72 	.bDescriptorType =	USB_DT_INTERFACE,
73 
74 	.bAlternateSetting =	1,
75 	.bNumEndpoints =	4,
76 	.bInterfaceClass =	USB_CLASS_VENDOR_SPEC,
77 	/* .iInterface		= DYNAMIC */
78 };
79 
80 /* full speed support: */
81 
82 static struct usb_endpoint_descriptor fs_source_desc = {
83 	.bLength =		USB_DT_ENDPOINT_SIZE,
84 	.bDescriptorType =	USB_DT_ENDPOINT,
85 
86 	.bEndpointAddress =	USB_DIR_IN,
87 	.bmAttributes =		USB_ENDPOINT_XFER_BULK,
88 };
89 
90 static struct usb_endpoint_descriptor fs_sink_desc = {
91 	.bLength =		USB_DT_ENDPOINT_SIZE,
92 	.bDescriptorType =	USB_DT_ENDPOINT,
93 
94 	.bEndpointAddress =	USB_DIR_OUT,
95 	.bmAttributes =		USB_ENDPOINT_XFER_BULK,
96 };
97 
98 static struct usb_endpoint_descriptor fs_iso_source_desc = {
99 	.bLength =		USB_DT_ENDPOINT_SIZE,
100 	.bDescriptorType =	USB_DT_ENDPOINT,
101 
102 	.bEndpointAddress =	USB_DIR_IN,
103 	.bmAttributes =		USB_ENDPOINT_XFER_ISOC,
104 	.wMaxPacketSize =	cpu_to_le16(1023),
105 	.bInterval =		4,
106 };
107 
108 static struct usb_endpoint_descriptor fs_iso_sink_desc = {
109 	.bLength =		USB_DT_ENDPOINT_SIZE,
110 	.bDescriptorType =	USB_DT_ENDPOINT,
111 
112 	.bEndpointAddress =	USB_DIR_OUT,
113 	.bmAttributes =		USB_ENDPOINT_XFER_ISOC,
114 	.wMaxPacketSize =	cpu_to_le16(1023),
115 	.bInterval =		4,
116 };
117 
118 static struct usb_descriptor_header *fs_source_sink_descs[] = {
119 	(struct usb_descriptor_header *) &source_sink_intf_alt0,
120 	(struct usb_descriptor_header *) &fs_sink_desc,
121 	(struct usb_descriptor_header *) &fs_source_desc,
122 	(struct usb_descriptor_header *) &source_sink_intf_alt1,
123 #define FS_ALT_IFC_1_OFFSET	3
124 	(struct usb_descriptor_header *) &fs_sink_desc,
125 	(struct usb_descriptor_header *) &fs_source_desc,
126 	(struct usb_descriptor_header *) &fs_iso_sink_desc,
127 	(struct usb_descriptor_header *) &fs_iso_source_desc,
128 	NULL,
129 };
130 
131 /* high speed support: */
132 
133 static struct usb_endpoint_descriptor hs_source_desc = {
134 	.bLength =		USB_DT_ENDPOINT_SIZE,
135 	.bDescriptorType =	USB_DT_ENDPOINT,
136 
137 	.bmAttributes =		USB_ENDPOINT_XFER_BULK,
138 	.wMaxPacketSize =	cpu_to_le16(512),
139 };
140 
141 static struct usb_endpoint_descriptor hs_sink_desc = {
142 	.bLength =		USB_DT_ENDPOINT_SIZE,
143 	.bDescriptorType =	USB_DT_ENDPOINT,
144 
145 	.bmAttributes =		USB_ENDPOINT_XFER_BULK,
146 	.wMaxPacketSize =	cpu_to_le16(512),
147 };
148 
149 static struct usb_endpoint_descriptor hs_iso_source_desc = {
150 	.bLength =		USB_DT_ENDPOINT_SIZE,
151 	.bDescriptorType =	USB_DT_ENDPOINT,
152 
153 	.bmAttributes =		USB_ENDPOINT_XFER_ISOC,
154 	.wMaxPacketSize =	cpu_to_le16(1024),
155 	.bInterval =		4,
156 };
157 
158 static struct usb_endpoint_descriptor hs_iso_sink_desc = {
159 	.bLength =		USB_DT_ENDPOINT_SIZE,
160 	.bDescriptorType =	USB_DT_ENDPOINT,
161 
162 	.bmAttributes =		USB_ENDPOINT_XFER_ISOC,
163 	.wMaxPacketSize =	cpu_to_le16(1024),
164 	.bInterval =		4,
165 };
166 
167 static struct usb_descriptor_header *hs_source_sink_descs[] = {
168 	(struct usb_descriptor_header *) &source_sink_intf_alt0,
169 	(struct usb_descriptor_header *) &hs_source_desc,
170 	(struct usb_descriptor_header *) &hs_sink_desc,
171 	(struct usb_descriptor_header *) &source_sink_intf_alt1,
172 #define HS_ALT_IFC_1_OFFSET	3
173 	(struct usb_descriptor_header *) &hs_source_desc,
174 	(struct usb_descriptor_header *) &hs_sink_desc,
175 	(struct usb_descriptor_header *) &hs_iso_source_desc,
176 	(struct usb_descriptor_header *) &hs_iso_sink_desc,
177 	NULL,
178 };
179 
180 /* super speed support: */
181 
182 static struct usb_endpoint_descriptor ss_source_desc = {
183 	.bLength =		USB_DT_ENDPOINT_SIZE,
184 	.bDescriptorType =	USB_DT_ENDPOINT,
185 
186 	.bmAttributes =		USB_ENDPOINT_XFER_BULK,
187 	.wMaxPacketSize =	cpu_to_le16(1024),
188 };
189 
190 static struct usb_ss_ep_comp_descriptor ss_source_comp_desc = {
191 	.bLength =		USB_DT_SS_EP_COMP_SIZE,
192 	.bDescriptorType =	USB_DT_SS_ENDPOINT_COMP,
193 
194 	.bMaxBurst =		0,
195 	.bmAttributes =		0,
196 	.wBytesPerInterval =	0,
197 };
198 
199 static struct usb_endpoint_descriptor ss_sink_desc = {
200 	.bLength =		USB_DT_ENDPOINT_SIZE,
201 	.bDescriptorType =	USB_DT_ENDPOINT,
202 
203 	.bmAttributes =		USB_ENDPOINT_XFER_BULK,
204 	.wMaxPacketSize =	cpu_to_le16(1024),
205 };
206 
207 static struct usb_ss_ep_comp_descriptor ss_sink_comp_desc = {
208 	.bLength =		USB_DT_SS_EP_COMP_SIZE,
209 	.bDescriptorType =	USB_DT_SS_ENDPOINT_COMP,
210 
211 	.bMaxBurst =		0,
212 	.bmAttributes =		0,
213 	.wBytesPerInterval =	0,
214 };
215 
216 static struct usb_endpoint_descriptor ss_iso_source_desc = {
217 	.bLength =		USB_DT_ENDPOINT_SIZE,
218 	.bDescriptorType =	USB_DT_ENDPOINT,
219 
220 	.bmAttributes =		USB_ENDPOINT_XFER_ISOC,
221 	.wMaxPacketSize =	cpu_to_le16(1024),
222 	.bInterval =		4,
223 };
224 
225 static struct usb_ss_ep_comp_descriptor ss_iso_source_comp_desc = {
226 	.bLength =		USB_DT_SS_EP_COMP_SIZE,
227 	.bDescriptorType =	USB_DT_SS_ENDPOINT_COMP,
228 
229 	.bMaxBurst =		0,
230 	.bmAttributes =		0,
231 	.wBytesPerInterval =	cpu_to_le16(1024),
232 };
233 
234 static struct usb_endpoint_descriptor ss_iso_sink_desc = {
235 	.bLength =		USB_DT_ENDPOINT_SIZE,
236 	.bDescriptorType =	USB_DT_ENDPOINT,
237 
238 	.bmAttributes =		USB_ENDPOINT_XFER_ISOC,
239 	.wMaxPacketSize =	cpu_to_le16(1024),
240 	.bInterval =		4,
241 };
242 
243 static struct usb_ss_ep_comp_descriptor ss_iso_sink_comp_desc = {
244 	.bLength =		USB_DT_SS_EP_COMP_SIZE,
245 	.bDescriptorType =	USB_DT_SS_ENDPOINT_COMP,
246 
247 	.bMaxBurst =		0,
248 	.bmAttributes =		0,
249 	.wBytesPerInterval =	cpu_to_le16(1024),
250 };
251 
252 static struct usb_descriptor_header *ss_source_sink_descs[] = {
253 	(struct usb_descriptor_header *) &source_sink_intf_alt0,
254 	(struct usb_descriptor_header *) &ss_source_desc,
255 	(struct usb_descriptor_header *) &ss_source_comp_desc,
256 	(struct usb_descriptor_header *) &ss_sink_desc,
257 	(struct usb_descriptor_header *) &ss_sink_comp_desc,
258 	(struct usb_descriptor_header *) &source_sink_intf_alt1,
259 #define SS_ALT_IFC_1_OFFSET	5
260 	(struct usb_descriptor_header *) &ss_source_desc,
261 	(struct usb_descriptor_header *) &ss_source_comp_desc,
262 	(struct usb_descriptor_header *) &ss_sink_desc,
263 	(struct usb_descriptor_header *) &ss_sink_comp_desc,
264 	(struct usb_descriptor_header *) &ss_iso_source_desc,
265 	(struct usb_descriptor_header *) &ss_iso_source_comp_desc,
266 	(struct usb_descriptor_header *) &ss_iso_sink_desc,
267 	(struct usb_descriptor_header *) &ss_iso_sink_comp_desc,
268 	NULL,
269 };
270 
271 /* function-specific strings: */
272 
273 static struct usb_string strings_sourcesink[] = {
274 	[0].s = "source and sink data",
275 	{  }			/* end of list */
276 };
277 
278 static struct usb_gadget_strings stringtab_sourcesink = {
279 	.language	= 0x0409,	/* en-us */
280 	.strings	= strings_sourcesink,
281 };
282 
283 static struct usb_gadget_strings *sourcesink_strings[] = {
284 	&stringtab_sourcesink,
285 	NULL,
286 };
287 
288 /*-------------------------------------------------------------------------*/
289 
290 static inline struct usb_request *ss_alloc_ep_req(struct usb_ep *ep, int len)
291 {
292 	return alloc_ep_req(ep, len);
293 }
294 
295 static void disable_ep(struct usb_composite_dev *cdev, struct usb_ep *ep)
296 {
297 	int			value;
298 
299 	value = usb_ep_disable(ep);
300 	if (value < 0)
301 		DBG(cdev, "disable %s --> %d\n", ep->name, value);
302 }
303 
304 void disable_endpoints(struct usb_composite_dev *cdev,
305 		struct usb_ep *in, struct usb_ep *out,
306 		struct usb_ep *iso_in, struct usb_ep *iso_out)
307 {
308 	disable_ep(cdev, in);
309 	disable_ep(cdev, out);
310 	if (iso_in)
311 		disable_ep(cdev, iso_in);
312 	if (iso_out)
313 		disable_ep(cdev, iso_out);
314 }
315 
316 static int
317 sourcesink_bind(struct usb_configuration *c, struct usb_function *f)
318 {
319 	struct usb_composite_dev *cdev = c->cdev;
320 	struct f_sourcesink	*ss = func_to_ss(f);
321 	int	id;
322 	int ret;
323 
324 	/* allocate interface ID(s) */
325 	id = usb_interface_id(c, f);
326 	if (id < 0)
327 		return id;
328 	source_sink_intf_alt0.bInterfaceNumber = id;
329 	source_sink_intf_alt1.bInterfaceNumber = id;
330 
331 	/* allocate bulk endpoints */
332 	ss->in_ep = usb_ep_autoconfig(cdev->gadget, &fs_source_desc);
333 	if (!ss->in_ep) {
334 autoconf_fail:
335 		ERROR(cdev, "%s: can't autoconfigure on %s\n",
336 			f->name, cdev->gadget->name);
337 		return -ENODEV;
338 	}
339 
340 	ss->out_ep = usb_ep_autoconfig(cdev->gadget, &fs_sink_desc);
341 	if (!ss->out_ep)
342 		goto autoconf_fail;
343 
344 	/* sanity check the isoc module parameters */
345 	if (ss->isoc_interval < 1)
346 		ss->isoc_interval = 1;
347 	if (ss->isoc_interval > 16)
348 		ss->isoc_interval = 16;
349 	if (ss->isoc_mult > 2)
350 		ss->isoc_mult = 2;
351 	if (ss->isoc_maxburst > 15)
352 		ss->isoc_maxburst = 15;
353 
354 	/* fill in the FS isoc descriptors from the module parameters */
355 	fs_iso_source_desc.wMaxPacketSize = ss->isoc_maxpacket > 1023 ?
356 						1023 : ss->isoc_maxpacket;
357 	fs_iso_source_desc.bInterval = ss->isoc_interval;
358 	fs_iso_sink_desc.wMaxPacketSize = ss->isoc_maxpacket > 1023 ?
359 						1023 : ss->isoc_maxpacket;
360 	fs_iso_sink_desc.bInterval = ss->isoc_interval;
361 
362 	/* allocate iso endpoints */
363 	ss->iso_in_ep = usb_ep_autoconfig(cdev->gadget, &fs_iso_source_desc);
364 	if (!ss->iso_in_ep)
365 		goto no_iso;
366 
367 	ss->iso_out_ep = usb_ep_autoconfig(cdev->gadget, &fs_iso_sink_desc);
368 	if (!ss->iso_out_ep) {
369 		usb_ep_autoconfig_release(ss->iso_in_ep);
370 		ss->iso_in_ep = NULL;
371 no_iso:
372 		/*
373 		 * We still want to work even if the UDC doesn't have isoc
374 		 * endpoints, so null out the alt interface that contains
375 		 * them and continue.
376 		 */
377 		fs_source_sink_descs[FS_ALT_IFC_1_OFFSET] = NULL;
378 		hs_source_sink_descs[HS_ALT_IFC_1_OFFSET] = NULL;
379 		ss_source_sink_descs[SS_ALT_IFC_1_OFFSET] = NULL;
380 	}
381 
382 	if (ss->isoc_maxpacket > 1024)
383 		ss->isoc_maxpacket = 1024;
384 
385 	/* support high speed hardware */
386 	hs_source_desc.bEndpointAddress = fs_source_desc.bEndpointAddress;
387 	hs_sink_desc.bEndpointAddress = fs_sink_desc.bEndpointAddress;
388 
389 	/*
390 	 * Fill in the HS isoc descriptors from the module parameters.
391 	 * We assume that the user knows what they are doing and won't
392 	 * give parameters that their UDC doesn't support.
393 	 */
394 	hs_iso_source_desc.wMaxPacketSize = ss->isoc_maxpacket;
395 	hs_iso_source_desc.wMaxPacketSize |= ss->isoc_mult << 11;
396 	hs_iso_source_desc.bInterval = ss->isoc_interval;
397 	hs_iso_source_desc.bEndpointAddress =
398 		fs_iso_source_desc.bEndpointAddress;
399 
400 	hs_iso_sink_desc.wMaxPacketSize = ss->isoc_maxpacket;
401 	hs_iso_sink_desc.wMaxPacketSize |= ss->isoc_mult << 11;
402 	hs_iso_sink_desc.bInterval = ss->isoc_interval;
403 	hs_iso_sink_desc.bEndpointAddress = fs_iso_sink_desc.bEndpointAddress;
404 
405 	/* support super speed hardware */
406 	ss_source_desc.bEndpointAddress =
407 		fs_source_desc.bEndpointAddress;
408 	ss_sink_desc.bEndpointAddress =
409 		fs_sink_desc.bEndpointAddress;
410 
411 	/*
412 	 * Fill in the SS isoc descriptors from the module parameters.
413 	 * We assume that the user knows what they are doing and won't
414 	 * give parameters that their UDC doesn't support.
415 	 */
416 	ss_iso_source_desc.wMaxPacketSize = ss->isoc_maxpacket;
417 	ss_iso_source_desc.bInterval = ss->isoc_interval;
418 	ss_iso_source_comp_desc.bmAttributes = ss->isoc_mult;
419 	ss_iso_source_comp_desc.bMaxBurst = ss->isoc_maxburst;
420 	ss_iso_source_comp_desc.wBytesPerInterval = ss->isoc_maxpacket *
421 		(ss->isoc_mult + 1) * (ss->isoc_maxburst + 1);
422 	ss_iso_source_desc.bEndpointAddress =
423 		fs_iso_source_desc.bEndpointAddress;
424 
425 	ss_iso_sink_desc.wMaxPacketSize = ss->isoc_maxpacket;
426 	ss_iso_sink_desc.bInterval = ss->isoc_interval;
427 	ss_iso_sink_comp_desc.bmAttributes = ss->isoc_mult;
428 	ss_iso_sink_comp_desc.bMaxBurst = ss->isoc_maxburst;
429 	ss_iso_sink_comp_desc.wBytesPerInterval = ss->isoc_maxpacket *
430 		(ss->isoc_mult + 1) * (ss->isoc_maxburst + 1);
431 	ss_iso_sink_desc.bEndpointAddress = fs_iso_sink_desc.bEndpointAddress;
432 
433 	ret = usb_assign_descriptors(f, fs_source_sink_descs,
434 			hs_source_sink_descs, ss_source_sink_descs, NULL);
435 	if (ret)
436 		return ret;
437 
438 	DBG(cdev, "%s speed %s: IN/%s, OUT/%s, ISO-IN/%s, ISO-OUT/%s\n",
439 	    (gadget_is_superspeed(c->cdev->gadget) ? "super" :
440 	     (gadget_is_dualspeed(c->cdev->gadget) ? "dual" : "full")),
441 			f->name, ss->in_ep->name, ss->out_ep->name,
442 			ss->iso_in_ep ? ss->iso_in_ep->name : "<none>",
443 			ss->iso_out_ep ? ss->iso_out_ep->name : "<none>");
444 	return 0;
445 }
446 
447 static void
448 sourcesink_free_func(struct usb_function *f)
449 {
450 	struct f_ss_opts *opts;
451 
452 	opts = container_of(f->fi, struct f_ss_opts, func_inst);
453 
454 	mutex_lock(&opts->lock);
455 	opts->refcnt--;
456 	mutex_unlock(&opts->lock);
457 
458 	usb_free_all_descriptors(f);
459 	kfree(func_to_ss(f));
460 }
461 
462 /* optionally require specific source/sink data patterns  */
463 static int check_read_data(struct f_sourcesink *ss, struct usb_request *req)
464 {
465 	unsigned		i;
466 	u8			*buf = req->buf;
467 	struct usb_composite_dev *cdev = ss->function.config->cdev;
468 	int max_packet_size = le16_to_cpu(ss->out_ep->desc->wMaxPacketSize);
469 
470 	if (ss->pattern == 2)
471 		return 0;
472 
473 	for (i = 0; i < req->actual; i++, buf++) {
474 		switch (ss->pattern) {
475 
476 		/* all-zeroes has no synchronization issues */
477 		case 0:
478 			if (*buf == 0)
479 				continue;
480 			break;
481 
482 		/* "mod63" stays in sync with short-terminated transfers,
483 		 * OR otherwise when host and gadget agree on how large
484 		 * each usb transfer request should be.  Resync is done
485 		 * with set_interface or set_config.  (We *WANT* it to
486 		 * get quickly out of sync if controllers or their drivers
487 		 * stutter for any reason, including buffer duplication...)
488 		 */
489 		case 1:
490 			if (*buf == (u8)((i % max_packet_size) % 63))
491 				continue;
492 			break;
493 		}
494 		ERROR(cdev, "bad OUT byte, buf[%d] = %d\n", i, *buf);
495 		usb_ep_set_halt(ss->out_ep);
496 		return -EINVAL;
497 	}
498 	return 0;
499 }
500 
501 static void reinit_write_data(struct usb_ep *ep, struct usb_request *req)
502 {
503 	unsigned	i;
504 	u8		*buf = req->buf;
505 	int max_packet_size = le16_to_cpu(ep->desc->wMaxPacketSize);
506 	struct f_sourcesink *ss = ep->driver_data;
507 
508 	switch (ss->pattern) {
509 	case 0:
510 		memset(req->buf, 0, req->length);
511 		break;
512 	case 1:
513 		for  (i = 0; i < req->length; i++)
514 			*buf++ = (u8) ((i % max_packet_size) % 63);
515 		break;
516 	case 2:
517 		break;
518 	}
519 }
520 
521 static void source_sink_complete(struct usb_ep *ep, struct usb_request *req)
522 {
523 	struct usb_composite_dev	*cdev;
524 	struct f_sourcesink		*ss = ep->driver_data;
525 	int				status = req->status;
526 
527 	/* driver_data will be null if ep has been disabled */
528 	if (!ss)
529 		return;
530 
531 	cdev = ss->function.config->cdev;
532 
533 	switch (status) {
534 
535 	case 0:				/* normal completion? */
536 		if (ep == ss->out_ep) {
537 			check_read_data(ss, req);
538 			if (ss->pattern != 2)
539 				memset(req->buf, 0x55, req->length);
540 		}
541 		break;
542 
543 	/* this endpoint is normally active while we're configured */
544 	case -ECONNABORTED:		/* hardware forced ep reset */
545 	case -ECONNRESET:		/* request dequeued */
546 	case -ESHUTDOWN:		/* disconnect from host */
547 		VDBG(cdev, "%s gone (%d), %d/%d\n", ep->name, status,
548 				req->actual, req->length);
549 		if (ep == ss->out_ep)
550 			check_read_data(ss, req);
551 		free_ep_req(ep, req);
552 		return;
553 
554 	case -EOVERFLOW:		/* buffer overrun on read means that
555 					 * we didn't provide a big enough
556 					 * buffer.
557 					 */
558 	default:
559 #if 1
560 		DBG(cdev, "%s complete --> %d, %d/%d\n", ep->name,
561 				status, req->actual, req->length);
562 #endif
563 	case -EREMOTEIO:		/* short read */
564 		break;
565 	}
566 
567 	status = usb_ep_queue(ep, req, GFP_ATOMIC);
568 	if (status) {
569 		ERROR(cdev, "kill %s:  resubmit %d bytes --> %d\n",
570 				ep->name, req->length, status);
571 		usb_ep_set_halt(ep);
572 		/* FIXME recover later ... somehow */
573 	}
574 }
575 
576 static int source_sink_start_ep(struct f_sourcesink *ss, bool is_in,
577 		bool is_iso, int speed)
578 {
579 	struct usb_ep		*ep;
580 	struct usb_request	*req;
581 	int			i, size, qlen, status = 0;
582 
583 	if (is_iso) {
584 		switch (speed) {
585 		case USB_SPEED_SUPER:
586 			size = ss->isoc_maxpacket *
587 					(ss->isoc_mult + 1) *
588 					(ss->isoc_maxburst + 1);
589 			break;
590 		case USB_SPEED_HIGH:
591 			size = ss->isoc_maxpacket * (ss->isoc_mult + 1);
592 			break;
593 		default:
594 			size = ss->isoc_maxpacket > 1023 ?
595 					1023 : ss->isoc_maxpacket;
596 			break;
597 		}
598 		ep = is_in ? ss->iso_in_ep : ss->iso_out_ep;
599 		qlen = ss->iso_qlen;
600 	} else {
601 		ep = is_in ? ss->in_ep : ss->out_ep;
602 		qlen = ss->bulk_qlen;
603 		size = ss->buflen;
604 	}
605 
606 	for (i = 0; i < qlen; i++) {
607 		req = ss_alloc_ep_req(ep, size);
608 		if (!req)
609 			return -ENOMEM;
610 
611 		req->complete = source_sink_complete;
612 		if (is_in)
613 			reinit_write_data(ep, req);
614 		else if (ss->pattern != 2)
615 			memset(req->buf, 0x55, req->length);
616 
617 		status = usb_ep_queue(ep, req, GFP_ATOMIC);
618 		if (status) {
619 			struct usb_composite_dev	*cdev;
620 
621 			cdev = ss->function.config->cdev;
622 			ERROR(cdev, "start %s%s %s --> %d\n",
623 			      is_iso ? "ISO-" : "", is_in ? "IN" : "OUT",
624 			      ep->name, status);
625 			free_ep_req(ep, req);
626 			return status;
627 		}
628 	}
629 
630 	return status;
631 }
632 
633 static void disable_source_sink(struct f_sourcesink *ss)
634 {
635 	struct usb_composite_dev	*cdev;
636 
637 	cdev = ss->function.config->cdev;
638 	disable_endpoints(cdev, ss->in_ep, ss->out_ep, ss->iso_in_ep,
639 			ss->iso_out_ep);
640 	VDBG(cdev, "%s disabled\n", ss->function.name);
641 }
642 
643 static int
644 enable_source_sink(struct usb_composite_dev *cdev, struct f_sourcesink *ss,
645 		int alt)
646 {
647 	int					result = 0;
648 	int					speed = cdev->gadget->speed;
649 	struct usb_ep				*ep;
650 
651 	/* one bulk endpoint writes (sources) zeroes IN (to the host) */
652 	ep = ss->in_ep;
653 	result = config_ep_by_speed(cdev->gadget, &(ss->function), ep);
654 	if (result)
655 		return result;
656 	result = usb_ep_enable(ep);
657 	if (result < 0)
658 		return result;
659 	ep->driver_data = ss;
660 
661 	result = source_sink_start_ep(ss, true, false, speed);
662 	if (result < 0) {
663 fail:
664 		ep = ss->in_ep;
665 		usb_ep_disable(ep);
666 		return result;
667 	}
668 
669 	/* one bulk endpoint reads (sinks) anything OUT (from the host) */
670 	ep = ss->out_ep;
671 	result = config_ep_by_speed(cdev->gadget, &(ss->function), ep);
672 	if (result)
673 		goto fail;
674 	result = usb_ep_enable(ep);
675 	if (result < 0)
676 		goto fail;
677 	ep->driver_data = ss;
678 
679 	result = source_sink_start_ep(ss, false, false, speed);
680 	if (result < 0) {
681 fail2:
682 		ep = ss->out_ep;
683 		usb_ep_disable(ep);
684 		goto fail;
685 	}
686 
687 	if (alt == 0)
688 		goto out;
689 
690 	/* one iso endpoint writes (sources) zeroes IN (to the host) */
691 	ep = ss->iso_in_ep;
692 	if (ep) {
693 		result = config_ep_by_speed(cdev->gadget, &(ss->function), ep);
694 		if (result)
695 			goto fail2;
696 		result = usb_ep_enable(ep);
697 		if (result < 0)
698 			goto fail2;
699 		ep->driver_data = ss;
700 
701 		result = source_sink_start_ep(ss, true, true, speed);
702 		if (result < 0) {
703 fail3:
704 			ep = ss->iso_in_ep;
705 			if (ep)
706 				usb_ep_disable(ep);
707 			goto fail2;
708 		}
709 	}
710 
711 	/* one iso endpoint reads (sinks) anything OUT (from the host) */
712 	ep = ss->iso_out_ep;
713 	if (ep) {
714 		result = config_ep_by_speed(cdev->gadget, &(ss->function), ep);
715 		if (result)
716 			goto fail3;
717 		result = usb_ep_enable(ep);
718 		if (result < 0)
719 			goto fail3;
720 		ep->driver_data = ss;
721 
722 		result = source_sink_start_ep(ss, false, true, speed);
723 		if (result < 0) {
724 			usb_ep_disable(ep);
725 			goto fail3;
726 		}
727 	}
728 out:
729 	ss->cur_alt = alt;
730 
731 	DBG(cdev, "%s enabled, alt intf %d\n", ss->function.name, alt);
732 	return result;
733 }
734 
735 static int sourcesink_set_alt(struct usb_function *f,
736 		unsigned intf, unsigned alt)
737 {
738 	struct f_sourcesink		*ss = func_to_ss(f);
739 	struct usb_composite_dev	*cdev = f->config->cdev;
740 
741 	disable_source_sink(ss);
742 	return enable_source_sink(cdev, ss, alt);
743 }
744 
745 static int sourcesink_get_alt(struct usb_function *f, unsigned intf)
746 {
747 	struct f_sourcesink		*ss = func_to_ss(f);
748 
749 	return ss->cur_alt;
750 }
751 
752 static void sourcesink_disable(struct usb_function *f)
753 {
754 	struct f_sourcesink	*ss = func_to_ss(f);
755 
756 	disable_source_sink(ss);
757 }
758 
759 /*-------------------------------------------------------------------------*/
760 
761 static int sourcesink_setup(struct usb_function *f,
762 		const struct usb_ctrlrequest *ctrl)
763 {
764 	struct usb_configuration        *c = f->config;
765 	struct usb_request	*req = c->cdev->req;
766 	int			value = -EOPNOTSUPP;
767 	u16			w_index = le16_to_cpu(ctrl->wIndex);
768 	u16			w_value = le16_to_cpu(ctrl->wValue);
769 	u16			w_length = le16_to_cpu(ctrl->wLength);
770 
771 	req->length = USB_COMP_EP0_BUFSIZ;
772 
773 	/* composite driver infrastructure handles everything except
774 	 * the two control test requests.
775 	 */
776 	switch (ctrl->bRequest) {
777 
778 	/*
779 	 * These are the same vendor-specific requests supported by
780 	 * Intel's USB 2.0 compliance test devices.  We exceed that
781 	 * device spec by allowing multiple-packet requests.
782 	 *
783 	 * NOTE:  the Control-OUT data stays in req->buf ... better
784 	 * would be copying it into a scratch buffer, so that other
785 	 * requests may safely intervene.
786 	 */
787 	case 0x5b:	/* control WRITE test -- fill the buffer */
788 		if (ctrl->bRequestType != (USB_DIR_OUT|USB_TYPE_VENDOR))
789 			goto unknown;
790 		if (w_value || w_index)
791 			break;
792 		/* just read that many bytes into the buffer */
793 		if (w_length > req->length)
794 			break;
795 		value = w_length;
796 		break;
797 	case 0x5c:	/* control READ test -- return the buffer */
798 		if (ctrl->bRequestType != (USB_DIR_IN|USB_TYPE_VENDOR))
799 			goto unknown;
800 		if (w_value || w_index)
801 			break;
802 		/* expect those bytes are still in the buffer; send back */
803 		if (w_length > req->length)
804 			break;
805 		value = w_length;
806 		break;
807 
808 	default:
809 unknown:
810 		VDBG(c->cdev,
811 			"unknown control req%02x.%02x v%04x i%04x l%d\n",
812 			ctrl->bRequestType, ctrl->bRequest,
813 			w_value, w_index, w_length);
814 	}
815 
816 	/* respond with data transfer or status phase? */
817 	if (value >= 0) {
818 		VDBG(c->cdev, "source/sink req%02x.%02x v%04x i%04x l%d\n",
819 			ctrl->bRequestType, ctrl->bRequest,
820 			w_value, w_index, w_length);
821 		req->zero = 0;
822 		req->length = value;
823 		value = usb_ep_queue(c->cdev->gadget->ep0, req, GFP_ATOMIC);
824 		if (value < 0)
825 			ERROR(c->cdev, "source/sink response, err %d\n",
826 					value);
827 	}
828 
829 	/* device either stalls (value < 0) or reports success */
830 	return value;
831 }
832 
833 static struct usb_function *source_sink_alloc_func(
834 		struct usb_function_instance *fi)
835 {
836 	struct f_sourcesink     *ss;
837 	struct f_ss_opts	*ss_opts;
838 
839 	ss = kzalloc(sizeof(*ss), GFP_KERNEL);
840 	if (!ss)
841 		return ERR_PTR(-ENOMEM);
842 
843 	ss_opts =  container_of(fi, struct f_ss_opts, func_inst);
844 
845 	mutex_lock(&ss_opts->lock);
846 	ss_opts->refcnt++;
847 	mutex_unlock(&ss_opts->lock);
848 
849 	ss->pattern = ss_opts->pattern;
850 	ss->isoc_interval = ss_opts->isoc_interval;
851 	ss->isoc_maxpacket = ss_opts->isoc_maxpacket;
852 	ss->isoc_mult = ss_opts->isoc_mult;
853 	ss->isoc_maxburst = ss_opts->isoc_maxburst;
854 	ss->buflen = ss_opts->bulk_buflen;
855 	ss->bulk_qlen = ss_opts->bulk_qlen;
856 	ss->iso_qlen = ss_opts->iso_qlen;
857 
858 	ss->function.name = "source/sink";
859 	ss->function.bind = sourcesink_bind;
860 	ss->function.set_alt = sourcesink_set_alt;
861 	ss->function.get_alt = sourcesink_get_alt;
862 	ss->function.disable = sourcesink_disable;
863 	ss->function.setup = sourcesink_setup;
864 	ss->function.strings = sourcesink_strings;
865 
866 	ss->function.free_func = sourcesink_free_func;
867 
868 	return &ss->function;
869 }
870 
871 static inline struct f_ss_opts *to_f_ss_opts(struct config_item *item)
872 {
873 	return container_of(to_config_group(item), struct f_ss_opts,
874 			    func_inst.group);
875 }
876 
877 static void ss_attr_release(struct config_item *item)
878 {
879 	struct f_ss_opts *ss_opts = to_f_ss_opts(item);
880 
881 	usb_put_function_instance(&ss_opts->func_inst);
882 }
883 
884 static struct configfs_item_operations ss_item_ops = {
885 	.release		= ss_attr_release,
886 };
887 
888 static ssize_t f_ss_opts_pattern_show(struct config_item *item, char *page)
889 {
890 	struct f_ss_opts *opts = to_f_ss_opts(item);
891 	int result;
892 
893 	mutex_lock(&opts->lock);
894 	result = sprintf(page, "%u\n", opts->pattern);
895 	mutex_unlock(&opts->lock);
896 
897 	return result;
898 }
899 
900 static ssize_t f_ss_opts_pattern_store(struct config_item *item,
901 				       const char *page, size_t len)
902 {
903 	struct f_ss_opts *opts = to_f_ss_opts(item);
904 	int ret;
905 	u8 num;
906 
907 	mutex_lock(&opts->lock);
908 	if (opts->refcnt) {
909 		ret = -EBUSY;
910 		goto end;
911 	}
912 
913 	ret = kstrtou8(page, 0, &num);
914 	if (ret)
915 		goto end;
916 
917 	if (num != 0 && num != 1 && num != 2) {
918 		ret = -EINVAL;
919 		goto end;
920 	}
921 
922 	opts->pattern = num;
923 	ret = len;
924 end:
925 	mutex_unlock(&opts->lock);
926 	return ret;
927 }
928 
929 CONFIGFS_ATTR(f_ss_opts_, pattern);
930 
931 static ssize_t f_ss_opts_isoc_interval_show(struct config_item *item, char *page)
932 {
933 	struct f_ss_opts *opts = to_f_ss_opts(item);
934 	int result;
935 
936 	mutex_lock(&opts->lock);
937 	result = sprintf(page, "%u\n", opts->isoc_interval);
938 	mutex_unlock(&opts->lock);
939 
940 	return result;
941 }
942 
943 static ssize_t f_ss_opts_isoc_interval_store(struct config_item *item,
944 				       const char *page, size_t len)
945 {
946 	struct f_ss_opts *opts = to_f_ss_opts(item);
947 	int ret;
948 	u8 num;
949 
950 	mutex_lock(&opts->lock);
951 	if (opts->refcnt) {
952 		ret = -EBUSY;
953 		goto end;
954 	}
955 
956 	ret = kstrtou8(page, 0, &num);
957 	if (ret)
958 		goto end;
959 
960 	if (num > 16) {
961 		ret = -EINVAL;
962 		goto end;
963 	}
964 
965 	opts->isoc_interval = num;
966 	ret = len;
967 end:
968 	mutex_unlock(&opts->lock);
969 	return ret;
970 }
971 
972 CONFIGFS_ATTR(f_ss_opts_, isoc_interval);
973 
974 static ssize_t f_ss_opts_isoc_maxpacket_show(struct config_item *item, char *page)
975 {
976 	struct f_ss_opts *opts = to_f_ss_opts(item);
977 	int result;
978 
979 	mutex_lock(&opts->lock);
980 	result = sprintf(page, "%u\n", opts->isoc_maxpacket);
981 	mutex_unlock(&opts->lock);
982 
983 	return result;
984 }
985 
986 static ssize_t f_ss_opts_isoc_maxpacket_store(struct config_item *item,
987 				       const char *page, size_t len)
988 {
989 	struct f_ss_opts *opts = to_f_ss_opts(item);
990 	int ret;
991 	u16 num;
992 
993 	mutex_lock(&opts->lock);
994 	if (opts->refcnt) {
995 		ret = -EBUSY;
996 		goto end;
997 	}
998 
999 	ret = kstrtou16(page, 0, &num);
1000 	if (ret)
1001 		goto end;
1002 
1003 	if (num > 1024) {
1004 		ret = -EINVAL;
1005 		goto end;
1006 	}
1007 
1008 	opts->isoc_maxpacket = num;
1009 	ret = len;
1010 end:
1011 	mutex_unlock(&opts->lock);
1012 	return ret;
1013 }
1014 
1015 CONFIGFS_ATTR(f_ss_opts_, isoc_maxpacket);
1016 
1017 static ssize_t f_ss_opts_isoc_mult_show(struct config_item *item, char *page)
1018 {
1019 	struct f_ss_opts *opts = to_f_ss_opts(item);
1020 	int result;
1021 
1022 	mutex_lock(&opts->lock);
1023 	result = sprintf(page, "%u\n", opts->isoc_mult);
1024 	mutex_unlock(&opts->lock);
1025 
1026 	return result;
1027 }
1028 
1029 static ssize_t f_ss_opts_isoc_mult_store(struct config_item *item,
1030 				       const char *page, size_t len)
1031 {
1032 	struct f_ss_opts *opts = to_f_ss_opts(item);
1033 	int ret;
1034 	u8 num;
1035 
1036 	mutex_lock(&opts->lock);
1037 	if (opts->refcnt) {
1038 		ret = -EBUSY;
1039 		goto end;
1040 	}
1041 
1042 	ret = kstrtou8(page, 0, &num);
1043 	if (ret)
1044 		goto end;
1045 
1046 	if (num > 2) {
1047 		ret = -EINVAL;
1048 		goto end;
1049 	}
1050 
1051 	opts->isoc_mult = num;
1052 	ret = len;
1053 end:
1054 	mutex_unlock(&opts->lock);
1055 	return ret;
1056 }
1057 
1058 CONFIGFS_ATTR(f_ss_opts_, isoc_mult);
1059 
1060 static ssize_t f_ss_opts_isoc_maxburst_show(struct config_item *item, char *page)
1061 {
1062 	struct f_ss_opts *opts = to_f_ss_opts(item);
1063 	int result;
1064 
1065 	mutex_lock(&opts->lock);
1066 	result = sprintf(page, "%u\n", opts->isoc_maxburst);
1067 	mutex_unlock(&opts->lock);
1068 
1069 	return result;
1070 }
1071 
1072 static ssize_t f_ss_opts_isoc_maxburst_store(struct config_item *item,
1073 				       const char *page, size_t len)
1074 {
1075 	struct f_ss_opts *opts = to_f_ss_opts(item);
1076 	int ret;
1077 	u8 num;
1078 
1079 	mutex_lock(&opts->lock);
1080 	if (opts->refcnt) {
1081 		ret = -EBUSY;
1082 		goto end;
1083 	}
1084 
1085 	ret = kstrtou8(page, 0, &num);
1086 	if (ret)
1087 		goto end;
1088 
1089 	if (num > 15) {
1090 		ret = -EINVAL;
1091 		goto end;
1092 	}
1093 
1094 	opts->isoc_maxburst = num;
1095 	ret = len;
1096 end:
1097 	mutex_unlock(&opts->lock);
1098 	return ret;
1099 }
1100 
1101 CONFIGFS_ATTR(f_ss_opts_, isoc_maxburst);
1102 
1103 static ssize_t f_ss_opts_bulk_buflen_show(struct config_item *item, char *page)
1104 {
1105 	struct f_ss_opts *opts = to_f_ss_opts(item);
1106 	int result;
1107 
1108 	mutex_lock(&opts->lock);
1109 	result = sprintf(page, "%u\n", opts->bulk_buflen);
1110 	mutex_unlock(&opts->lock);
1111 
1112 	return result;
1113 }
1114 
1115 static ssize_t f_ss_opts_bulk_buflen_store(struct config_item *item,
1116 					   const char *page, size_t len)
1117 {
1118 	struct f_ss_opts *opts = to_f_ss_opts(item);
1119 	int ret;
1120 	u32 num;
1121 
1122 	mutex_lock(&opts->lock);
1123 	if (opts->refcnt) {
1124 		ret = -EBUSY;
1125 		goto end;
1126 	}
1127 
1128 	ret = kstrtou32(page, 0, &num);
1129 	if (ret)
1130 		goto end;
1131 
1132 	opts->bulk_buflen = num;
1133 	ret = len;
1134 end:
1135 	mutex_unlock(&opts->lock);
1136 	return ret;
1137 }
1138 
1139 CONFIGFS_ATTR(f_ss_opts_, bulk_buflen);
1140 
1141 static ssize_t f_ss_opts_bulk_qlen_show(struct config_item *item, char *page)
1142 {
1143 	struct f_ss_opts *opts = to_f_ss_opts(item);
1144 	int result;
1145 
1146 	mutex_lock(&opts->lock);
1147 	result = sprintf(page, "%u\n", opts->bulk_qlen);
1148 	mutex_unlock(&opts->lock);
1149 
1150 	return result;
1151 }
1152 
1153 static ssize_t f_ss_opts_bulk_qlen_store(struct config_item *item,
1154 					   const char *page, size_t len)
1155 {
1156 	struct f_ss_opts *opts = to_f_ss_opts(item);
1157 	int ret;
1158 	u32 num;
1159 
1160 	mutex_lock(&opts->lock);
1161 	if (opts->refcnt) {
1162 		ret = -EBUSY;
1163 		goto end;
1164 	}
1165 
1166 	ret = kstrtou32(page, 0, &num);
1167 	if (ret)
1168 		goto end;
1169 
1170 	opts->bulk_qlen = num;
1171 	ret = len;
1172 end:
1173 	mutex_unlock(&opts->lock);
1174 	return ret;
1175 }
1176 
1177 CONFIGFS_ATTR(f_ss_opts_, bulk_qlen);
1178 
1179 static ssize_t f_ss_opts_iso_qlen_show(struct config_item *item, char *page)
1180 {
1181 	struct f_ss_opts *opts = to_f_ss_opts(item);
1182 	int result;
1183 
1184 	mutex_lock(&opts->lock);
1185 	result = sprintf(page, "%u\n", opts->iso_qlen);
1186 	mutex_unlock(&opts->lock);
1187 
1188 	return result;
1189 }
1190 
1191 static ssize_t f_ss_opts_iso_qlen_store(struct config_item *item,
1192 					   const char *page, size_t len)
1193 {
1194 	struct f_ss_opts *opts = to_f_ss_opts(item);
1195 	int ret;
1196 	u32 num;
1197 
1198 	mutex_lock(&opts->lock);
1199 	if (opts->refcnt) {
1200 		ret = -EBUSY;
1201 		goto end;
1202 	}
1203 
1204 	ret = kstrtou32(page, 0, &num);
1205 	if (ret)
1206 		goto end;
1207 
1208 	opts->iso_qlen = num;
1209 	ret = len;
1210 end:
1211 	mutex_unlock(&opts->lock);
1212 	return ret;
1213 }
1214 
1215 CONFIGFS_ATTR(f_ss_opts_, iso_qlen);
1216 
1217 static struct configfs_attribute *ss_attrs[] = {
1218 	&f_ss_opts_attr_pattern,
1219 	&f_ss_opts_attr_isoc_interval,
1220 	&f_ss_opts_attr_isoc_maxpacket,
1221 	&f_ss_opts_attr_isoc_mult,
1222 	&f_ss_opts_attr_isoc_maxburst,
1223 	&f_ss_opts_attr_bulk_buflen,
1224 	&f_ss_opts_attr_bulk_qlen,
1225 	&f_ss_opts_attr_iso_qlen,
1226 	NULL,
1227 };
1228 
1229 static const struct config_item_type ss_func_type = {
1230 	.ct_item_ops    = &ss_item_ops,
1231 	.ct_attrs	= ss_attrs,
1232 	.ct_owner       = THIS_MODULE,
1233 };
1234 
1235 static void source_sink_free_instance(struct usb_function_instance *fi)
1236 {
1237 	struct f_ss_opts *ss_opts;
1238 
1239 	ss_opts = container_of(fi, struct f_ss_opts, func_inst);
1240 	kfree(ss_opts);
1241 }
1242 
1243 static struct usb_function_instance *source_sink_alloc_inst(void)
1244 {
1245 	struct f_ss_opts *ss_opts;
1246 
1247 	ss_opts = kzalloc(sizeof(*ss_opts), GFP_KERNEL);
1248 	if (!ss_opts)
1249 		return ERR_PTR(-ENOMEM);
1250 	mutex_init(&ss_opts->lock);
1251 	ss_opts->func_inst.free_func_inst = source_sink_free_instance;
1252 	ss_opts->isoc_interval = GZERO_ISOC_INTERVAL;
1253 	ss_opts->isoc_maxpacket = GZERO_ISOC_MAXPACKET;
1254 	ss_opts->bulk_buflen = GZERO_BULK_BUFLEN;
1255 	ss_opts->bulk_qlen = GZERO_SS_BULK_QLEN;
1256 	ss_opts->iso_qlen = GZERO_SS_ISO_QLEN;
1257 
1258 	config_group_init_type_name(&ss_opts->func_inst.group, "",
1259 				    &ss_func_type);
1260 
1261 	return &ss_opts->func_inst;
1262 }
1263 DECLARE_USB_FUNCTION(SourceSink, source_sink_alloc_inst,
1264 		source_sink_alloc_func);
1265 
1266 static int __init sslb_modinit(void)
1267 {
1268 	int ret;
1269 
1270 	ret = usb_function_register(&SourceSinkusb_func);
1271 	if (ret)
1272 		return ret;
1273 	ret = lb_modinit();
1274 	if (ret)
1275 		usb_function_unregister(&SourceSinkusb_func);
1276 	return ret;
1277 }
1278 static void __exit sslb_modexit(void)
1279 {
1280 	usb_function_unregister(&SourceSinkusb_func);
1281 	lb_modexit();
1282 }
1283 module_init(sslb_modinit);
1284 module_exit(sslb_modexit);
1285 
1286 MODULE_LICENSE("GPL");
1287