xref: /rk3399_rockchip-uboot/drivers/usb/gadget/f_dfu.c (revision 1a4f6af8bfd44c8ae6e87a81ff125eed47042cc5)
1b819ddbfSLukasz Majewski /*
2b819ddbfSLukasz Majewski  * f_dfu.c -- Device Firmware Update USB function
3b819ddbfSLukasz Majewski  *
4b819ddbfSLukasz Majewski  * Copyright (C) 2012 Samsung Electronics
5b819ddbfSLukasz Majewski  * authors: Andrzej Pietrasiewicz <andrzej.p@samsung.com>
6b819ddbfSLukasz Majewski  *          Lukasz Majewski <l.majewski@samsung.com>
7b819ddbfSLukasz Majewski  *
881c065d2SŁukasz Majewski  * Based on OpenMoko u-boot: drivers/usb/usbdfu.c
981c065d2SŁukasz Majewski  * (C) 2007 by OpenMoko, Inc.
1081c065d2SŁukasz Majewski  * Author: Harald Welte <laforge@openmoko.org>
1181c065d2SŁukasz Majewski  *
1281c065d2SŁukasz Majewski  * based on existing SAM7DFU code from OpenPCD:
1381c065d2SŁukasz Majewski  * (C) Copyright 2006 by Harald Welte <hwelte at hmw-consulting.de>
1481c065d2SŁukasz Majewski  *
151a459660SWolfgang Denk  * SPDX-License-Identifier:	GPL-2.0+
16b819ddbfSLukasz Majewski  */
17b819ddbfSLukasz Majewski 
18b819ddbfSLukasz Majewski #include <errno.h>
19b819ddbfSLukasz Majewski #include <common.h>
20b819ddbfSLukasz Majewski #include <malloc.h>
21b819ddbfSLukasz Majewski 
22b819ddbfSLukasz Majewski #include <linux/usb/ch9.h>
23b819ddbfSLukasz Majewski #include <linux/usb/gadget.h>
24b819ddbfSLukasz Majewski #include <linux/usb/composite.h>
25b819ddbfSLukasz Majewski 
26b819ddbfSLukasz Majewski #include <dfu.h>
27c4d0e856SMateusz Zalega #include <g_dnl.h>
28b819ddbfSLukasz Majewski #include "f_dfu.h"
29b819ddbfSLukasz Majewski 
30b819ddbfSLukasz Majewski struct f_dfu {
31b819ddbfSLukasz Majewski 	struct usb_function		usb_function;
32b819ddbfSLukasz Majewski 
33b819ddbfSLukasz Majewski 	struct usb_descriptor_header	**function;
34b819ddbfSLukasz Majewski 	struct usb_string		*strings;
35b819ddbfSLukasz Majewski 
36b819ddbfSLukasz Majewski 	/* when configured, we have one config */
37b819ddbfSLukasz Majewski 	u8				config;
38b819ddbfSLukasz Majewski 	u8				altsetting;
39b819ddbfSLukasz Majewski 	enum dfu_state			dfu_state;
40b819ddbfSLukasz Majewski 	unsigned int			dfu_status;
41b819ddbfSLukasz Majewski 
42b819ddbfSLukasz Majewski 	/* Send/received block number is handy for data integrity check */
43b819ddbfSLukasz Majewski 	int                             blk_seq_num;
4433fac4a6SLukasz Majewski 	unsigned int                    poll_timeout;
45b819ddbfSLukasz Majewski };
46b819ddbfSLukasz Majewski 
47fc18f8d1SLukasz Majewski struct dfu_entity *dfu_defer_flush;
48fc18f8d1SLukasz Majewski 
49b819ddbfSLukasz Majewski typedef int (*dfu_state_fn) (struct f_dfu *,
50b819ddbfSLukasz Majewski 			     const struct usb_ctrlrequest *,
51b819ddbfSLukasz Majewski 			     struct usb_gadget *,
52b819ddbfSLukasz Majewski 			     struct usb_request *);
53b819ddbfSLukasz Majewski 
func_to_dfu(struct usb_function * f)54b819ddbfSLukasz Majewski static inline struct f_dfu *func_to_dfu(struct usb_function *f)
55b819ddbfSLukasz Majewski {
56b819ddbfSLukasz Majewski 	return container_of(f, struct f_dfu, usb_function);
57b819ddbfSLukasz Majewski }
58b819ddbfSLukasz Majewski 
59b819ddbfSLukasz Majewski static const struct dfu_function_descriptor dfu_func = {
60b819ddbfSLukasz Majewski 	.bLength =		sizeof dfu_func,
61b819ddbfSLukasz Majewski 	.bDescriptorType =	DFU_DT_FUNC,
62b819ddbfSLukasz Majewski 	.bmAttributes =		DFU_BIT_WILL_DETACH |
63b819ddbfSLukasz Majewski 				DFU_BIT_MANIFESTATION_TOLERANT |
64b819ddbfSLukasz Majewski 				DFU_BIT_CAN_UPLOAD |
65b819ddbfSLukasz Majewski 				DFU_BIT_CAN_DNLOAD,
66b819ddbfSLukasz Majewski 	.wDetachTimeOut =	0,
67b819ddbfSLukasz Majewski 	.wTransferSize =	DFU_USB_BUFSIZ,
68b819ddbfSLukasz Majewski 	.bcdDFUVersion =	__constant_cpu_to_le16(0x0110),
69b819ddbfSLukasz Majewski };
70b819ddbfSLukasz Majewski 
71b819ddbfSLukasz Majewski static struct usb_interface_descriptor dfu_intf_runtime = {
72b819ddbfSLukasz Majewski 	.bLength =		sizeof dfu_intf_runtime,
73b819ddbfSLukasz Majewski 	.bDescriptorType =	USB_DT_INTERFACE,
74b819ddbfSLukasz Majewski 	.bNumEndpoints =	0,
75b819ddbfSLukasz Majewski 	.bInterfaceClass =	USB_CLASS_APP_SPEC,
76b819ddbfSLukasz Majewski 	.bInterfaceSubClass =	1,
77b819ddbfSLukasz Majewski 	.bInterfaceProtocol =	1,
78b819ddbfSLukasz Majewski 	/* .iInterface = DYNAMIC */
79b819ddbfSLukasz Majewski };
80b819ddbfSLukasz Majewski 
81b819ddbfSLukasz Majewski static struct usb_descriptor_header *dfu_runtime_descs[] = {
82b819ddbfSLukasz Majewski 	(struct usb_descriptor_header *) &dfu_intf_runtime,
83b819ddbfSLukasz Majewski 	NULL,
84b819ddbfSLukasz Majewski };
85b819ddbfSLukasz Majewski 
86b819ddbfSLukasz Majewski static const char dfu_name[] = "Device Firmware Upgrade";
87b819ddbfSLukasz Majewski 
88b819ddbfSLukasz Majewski /*
89b819ddbfSLukasz Majewski  * static strings, in UTF-8
90b819ddbfSLukasz Majewski  *
91b819ddbfSLukasz Majewski  * dfu_generic configuration
92b819ddbfSLukasz Majewski  */
93b819ddbfSLukasz Majewski static struct usb_string strings_dfu_generic[] = {
94b819ddbfSLukasz Majewski 	[0].s = dfu_name,
95b819ddbfSLukasz Majewski 	{  }			/* end of list */
96b819ddbfSLukasz Majewski };
97b819ddbfSLukasz Majewski 
98b819ddbfSLukasz Majewski static struct usb_gadget_strings stringtab_dfu_generic = {
99b819ddbfSLukasz Majewski 	.language	= 0x0409,	/* en-us */
100b819ddbfSLukasz Majewski 	.strings	= strings_dfu_generic,
101b819ddbfSLukasz Majewski };
102b819ddbfSLukasz Majewski 
103b819ddbfSLukasz Majewski static struct usb_gadget_strings *dfu_generic_strings[] = {
104b819ddbfSLukasz Majewski 	&stringtab_dfu_generic,
105b819ddbfSLukasz Majewski 	NULL,
106b819ddbfSLukasz Majewski };
107b819ddbfSLukasz Majewski 
108b819ddbfSLukasz Majewski /*
109b819ddbfSLukasz Majewski  * usb_function specific
110b819ddbfSLukasz Majewski  */
111b819ddbfSLukasz Majewski static struct usb_gadget_strings stringtab_dfu = {
112b819ddbfSLukasz Majewski 	.language	= 0x0409,	/* en-us */
113b819ddbfSLukasz Majewski 	/*
114b819ddbfSLukasz Majewski 	 * .strings
115b819ddbfSLukasz Majewski 	 *
116b819ddbfSLukasz Majewski 	 * assigned during initialization,
117b819ddbfSLukasz Majewski 	 * depends on number of flash entities
118b819ddbfSLukasz Majewski 	 *
119b819ddbfSLukasz Majewski 	 */
120b819ddbfSLukasz Majewski };
121b819ddbfSLukasz Majewski 
122b819ddbfSLukasz Majewski static struct usb_gadget_strings *dfu_strings[] = {
123b819ddbfSLukasz Majewski 	&stringtab_dfu,
124b819ddbfSLukasz Majewski 	NULL,
125b819ddbfSLukasz Majewski };
126b819ddbfSLukasz Majewski 
dfu_set_poll_timeout(struct dfu_status * dstat,unsigned int ms)12733fac4a6SLukasz Majewski static void dfu_set_poll_timeout(struct dfu_status *dstat, unsigned int ms)
12833fac4a6SLukasz Majewski {
12933fac4a6SLukasz Majewski 	/*
13033fac4a6SLukasz Majewski 	 * The bwPollTimeout DFU_GETSTATUS request payload provides information
13133fac4a6SLukasz Majewski 	 * about minimum time, in milliseconds, that the host should wait before
13233fac4a6SLukasz Majewski 	 * sending a subsequent DFU_GETSTATUS request
13333fac4a6SLukasz Majewski 	 *
13433fac4a6SLukasz Majewski 	 * This permits the device to vary the delay depending on its need to
13533fac4a6SLukasz Majewski 	 * erase or program the memory
13633fac4a6SLukasz Majewski 	 *
13733fac4a6SLukasz Majewski 	 */
13833fac4a6SLukasz Majewski 
13933fac4a6SLukasz Majewski 	unsigned char *p = (unsigned char *)&ms;
14033fac4a6SLukasz Majewski 
14133fac4a6SLukasz Majewski 	if (!ms || (ms & ~DFU_POLL_TIMEOUT_MASK)) {
14233fac4a6SLukasz Majewski 		dstat->bwPollTimeout[0] = 0;
14333fac4a6SLukasz Majewski 		dstat->bwPollTimeout[1] = 0;
14433fac4a6SLukasz Majewski 		dstat->bwPollTimeout[2] = 0;
14533fac4a6SLukasz Majewski 
14633fac4a6SLukasz Majewski 		return;
14733fac4a6SLukasz Majewski 	}
14833fac4a6SLukasz Majewski 
14933fac4a6SLukasz Majewski 	dstat->bwPollTimeout[0] = *p++;
15033fac4a6SLukasz Majewski 	dstat->bwPollTimeout[1] = *p++;
15133fac4a6SLukasz Majewski 	dstat->bwPollTimeout[2] = *p;
15233fac4a6SLukasz Majewski }
15333fac4a6SLukasz Majewski 
154b819ddbfSLukasz Majewski /*-------------------------------------------------------------------------*/
155b819ddbfSLukasz Majewski 
dnload_request_complete(struct usb_ep * ep,struct usb_request * req)156b819ddbfSLukasz Majewski static void dnload_request_complete(struct usb_ep *ep, struct usb_request *req)
157b819ddbfSLukasz Majewski {
158b819ddbfSLukasz Majewski 	struct f_dfu *f_dfu = req->context;
1593ee9593fSStephen Warren 	int ret;
160b819ddbfSLukasz Majewski 
1613ee9593fSStephen Warren 	ret = dfu_write(dfu_get_entity(f_dfu->altsetting), req->buf,
16200e9d696SFelipe Balbi 			req->actual, f_dfu->blk_seq_num);
1633ee9593fSStephen Warren 	if (ret) {
1643ee9593fSStephen Warren 		f_dfu->dfu_status = DFU_STATUS_errUNKNOWN;
1653ee9593fSStephen Warren 		f_dfu->dfu_state = DFU_STATE_dfuERROR;
1663ee9593fSStephen Warren 	}
167001a8319SHeiko Schocher }
168b819ddbfSLukasz Majewski 
dnload_request_flush(struct usb_ep * ep,struct usb_request * req)169001a8319SHeiko Schocher static void dnload_request_flush(struct usb_ep *ep, struct usb_request *req)
170001a8319SHeiko Schocher {
171001a8319SHeiko Schocher 	struct f_dfu *f_dfu = req->context;
172fc18f8d1SLukasz Majewski 	dfu_set_defer_flush(dfu_get_entity(f_dfu->altsetting));
173b819ddbfSLukasz Majewski }
174b819ddbfSLukasz Majewski 
dfu_get_manifest_timeout(struct dfu_entity * dfu)175fc25fa27SHeiko Schocher static inline int dfu_get_manifest_timeout(struct dfu_entity *dfu)
176fc25fa27SHeiko Schocher {
177fc25fa27SHeiko Schocher 	return dfu->poll_timeout ? dfu->poll_timeout(dfu) :
178fc25fa27SHeiko Schocher 		DFU_MANIFEST_POLL_TIMEOUT;
179fc25fa27SHeiko Schocher }
180fc25fa27SHeiko Schocher 
handle_getstatus(struct usb_request * req)181d4287766SPatrick Delaunay static int handle_getstatus(struct usb_request *req)
182b819ddbfSLukasz Majewski {
183b819ddbfSLukasz Majewski 	struct dfu_status *dstat = (struct dfu_status *)req->buf;
184b819ddbfSLukasz Majewski 	struct f_dfu *f_dfu = req->context;
185fc25fa27SHeiko Schocher 	struct dfu_entity *dfu = dfu_get_entity(f_dfu->altsetting);
186b819ddbfSLukasz Majewski 
187001a8319SHeiko Schocher 	dfu_set_poll_timeout(dstat, 0);
188001a8319SHeiko Schocher 
189b819ddbfSLukasz Majewski 	switch (f_dfu->dfu_state) {
190b819ddbfSLukasz Majewski 	case DFU_STATE_dfuDNLOAD_SYNC:
191b819ddbfSLukasz Majewski 	case DFU_STATE_dfuDNBUSY:
192b819ddbfSLukasz Majewski 		f_dfu->dfu_state = DFU_STATE_dfuDNLOAD_IDLE;
193b819ddbfSLukasz Majewski 		break;
194b819ddbfSLukasz Majewski 	case DFU_STATE_dfuMANIFEST_SYNC:
195001a8319SHeiko Schocher 		f_dfu->dfu_state = DFU_STATE_dfuMANIFEST;
196b819ddbfSLukasz Majewski 		break;
197001a8319SHeiko Schocher 	case DFU_STATE_dfuMANIFEST:
198fc25fa27SHeiko Schocher 		dfu_set_poll_timeout(dstat, dfu_get_manifest_timeout(dfu));
199fc25fa27SHeiko Schocher 		break;
200b819ddbfSLukasz Majewski 	default:
201b819ddbfSLukasz Majewski 		break;
202b819ddbfSLukasz Majewski 	}
203b819ddbfSLukasz Majewski 
20433fac4a6SLukasz Majewski 	if (f_dfu->poll_timeout)
20533fac4a6SLukasz Majewski 		if (!(f_dfu->blk_seq_num %
20633fac4a6SLukasz Majewski 		      (dfu_get_buf_size() / DFU_USB_BUFSIZ)))
20733fac4a6SLukasz Majewski 			dfu_set_poll_timeout(dstat, f_dfu->poll_timeout);
20833fac4a6SLukasz Majewski 
209b819ddbfSLukasz Majewski 	/* send status response */
210b819ddbfSLukasz Majewski 	dstat->bStatus = f_dfu->dfu_status;
211b819ddbfSLukasz Majewski 	dstat->bState = f_dfu->dfu_state;
212b819ddbfSLukasz Majewski 	dstat->iString = 0;
213d4287766SPatrick Delaunay 
214d4287766SPatrick Delaunay 	return sizeof(struct dfu_status);
215b819ddbfSLukasz Majewski }
216b819ddbfSLukasz Majewski 
handle_getstate(struct usb_request * req)217f11bb252SPatrick Delaunay static int handle_getstate(struct usb_request *req)
218b819ddbfSLukasz Majewski {
219b819ddbfSLukasz Majewski 	struct f_dfu *f_dfu = req->context;
220b819ddbfSLukasz Majewski 
221b819ddbfSLukasz Majewski 	((u8 *)req->buf)[0] = f_dfu->dfu_state;
222f11bb252SPatrick Delaunay 	return sizeof(u8);
223b819ddbfSLukasz Majewski }
224b819ddbfSLukasz Majewski 
to_dfu_mode(struct f_dfu * f_dfu)225b819ddbfSLukasz Majewski static inline void to_dfu_mode(struct f_dfu *f_dfu)
226b819ddbfSLukasz Majewski {
227b819ddbfSLukasz Majewski 	f_dfu->usb_function.strings = dfu_strings;
228b819ddbfSLukasz Majewski 	f_dfu->usb_function.hs_descriptors = f_dfu->function;
2299f3b8ed1SHeiko Schocher 	f_dfu->usb_function.descriptors = f_dfu->function;
230ad5f9778SHeiko Schocher 	f_dfu->dfu_state = DFU_STATE_dfuIDLE;
231b819ddbfSLukasz Majewski }
232b819ddbfSLukasz Majewski 
to_runtime_mode(struct f_dfu * f_dfu)233b819ddbfSLukasz Majewski static inline void to_runtime_mode(struct f_dfu *f_dfu)
234b819ddbfSLukasz Majewski {
235b819ddbfSLukasz Majewski 	f_dfu->usb_function.strings = NULL;
236b819ddbfSLukasz Majewski 	f_dfu->usb_function.hs_descriptors = dfu_runtime_descs;
2379f3b8ed1SHeiko Schocher 	f_dfu->usb_function.descriptors = dfu_runtime_descs;
238b819ddbfSLukasz Majewski }
239b819ddbfSLukasz Majewski 
handle_upload(struct usb_request * req,u16 len)240b819ddbfSLukasz Majewski static int handle_upload(struct usb_request *req, u16 len)
241b819ddbfSLukasz Majewski {
242b819ddbfSLukasz Majewski 	struct f_dfu *f_dfu = req->context;
243b819ddbfSLukasz Majewski 
244b819ddbfSLukasz Majewski 	return dfu_read(dfu_get_entity(f_dfu->altsetting), req->buf,
245b819ddbfSLukasz Majewski 			req->length, f_dfu->blk_seq_num);
246b819ddbfSLukasz Majewski }
247b819ddbfSLukasz Majewski 
handle_dnload(struct usb_gadget * gadget,u16 len)248b819ddbfSLukasz Majewski static int handle_dnload(struct usb_gadget *gadget, u16 len)
249b819ddbfSLukasz Majewski {
250b819ddbfSLukasz Majewski 	struct usb_composite_dev *cdev = get_gadget_data(gadget);
251b819ddbfSLukasz Majewski 	struct usb_request *req = cdev->req;
252b819ddbfSLukasz Majewski 	struct f_dfu *f_dfu = req->context;
253b819ddbfSLukasz Majewski 
254b819ddbfSLukasz Majewski 	if (len == 0)
255b819ddbfSLukasz Majewski 		f_dfu->dfu_state = DFU_STATE_dfuMANIFEST_SYNC;
256b819ddbfSLukasz Majewski 
257b819ddbfSLukasz Majewski 	req->complete = dnload_request_complete;
258b819ddbfSLukasz Majewski 
259b819ddbfSLukasz Majewski 	return len;
260b819ddbfSLukasz Majewski }
261b819ddbfSLukasz Majewski 
262b819ddbfSLukasz Majewski /*-------------------------------------------------------------------------*/
263b819ddbfSLukasz Majewski /* DFU state machine  */
state_app_idle(struct f_dfu * f_dfu,const struct usb_ctrlrequest * ctrl,struct usb_gadget * gadget,struct usb_request * req)264b819ddbfSLukasz Majewski static int state_app_idle(struct f_dfu *f_dfu,
265b819ddbfSLukasz Majewski 			  const struct usb_ctrlrequest *ctrl,
266b819ddbfSLukasz Majewski 			  struct usb_gadget *gadget,
267b819ddbfSLukasz Majewski 			  struct usb_request *req)
268b819ddbfSLukasz Majewski {
269b819ddbfSLukasz Majewski 	int value = 0;
270b819ddbfSLukasz Majewski 
271b819ddbfSLukasz Majewski 	switch (ctrl->bRequest) {
272b819ddbfSLukasz Majewski 	case USB_REQ_DFU_GETSTATUS:
273d4287766SPatrick Delaunay 		value = handle_getstatus(req);
274b819ddbfSLukasz Majewski 		break;
275b819ddbfSLukasz Majewski 	case USB_REQ_DFU_GETSTATE:
276f11bb252SPatrick Delaunay 		value = handle_getstate(req);
277b819ddbfSLukasz Majewski 		break;
278b819ddbfSLukasz Majewski 	case USB_REQ_DFU_DETACH:
279b819ddbfSLukasz Majewski 		f_dfu->dfu_state = DFU_STATE_appDETACH;
280b819ddbfSLukasz Majewski 		to_dfu_mode(f_dfu);
281b819ddbfSLukasz Majewski 		value = RET_ZLP;
282b819ddbfSLukasz Majewski 		break;
283b819ddbfSLukasz Majewski 	default:
284b819ddbfSLukasz Majewski 		value = RET_STALL;
285b819ddbfSLukasz Majewski 		break;
286b819ddbfSLukasz Majewski 	}
287b819ddbfSLukasz Majewski 
288b819ddbfSLukasz Majewski 	return value;
289b819ddbfSLukasz Majewski }
290b819ddbfSLukasz Majewski 
state_app_detach(struct f_dfu * f_dfu,const struct usb_ctrlrequest * ctrl,struct usb_gadget * gadget,struct usb_request * req)291b819ddbfSLukasz Majewski static int state_app_detach(struct f_dfu *f_dfu,
292b819ddbfSLukasz Majewski 			    const struct usb_ctrlrequest *ctrl,
293b819ddbfSLukasz Majewski 			    struct usb_gadget *gadget,
294b819ddbfSLukasz Majewski 			    struct usb_request *req)
295b819ddbfSLukasz Majewski {
296b819ddbfSLukasz Majewski 	int value = 0;
297b819ddbfSLukasz Majewski 
298b819ddbfSLukasz Majewski 	switch (ctrl->bRequest) {
299b819ddbfSLukasz Majewski 	case USB_REQ_DFU_GETSTATUS:
300d4287766SPatrick Delaunay 		value = handle_getstatus(req);
301b819ddbfSLukasz Majewski 		break;
302b819ddbfSLukasz Majewski 	case USB_REQ_DFU_GETSTATE:
303f11bb252SPatrick Delaunay 		value = handle_getstate(req);
304b819ddbfSLukasz Majewski 		break;
305b819ddbfSLukasz Majewski 	default:
306b819ddbfSLukasz Majewski 		f_dfu->dfu_state = DFU_STATE_appIDLE;
307b819ddbfSLukasz Majewski 		value = RET_STALL;
308b819ddbfSLukasz Majewski 		break;
309b819ddbfSLukasz Majewski 	}
310b819ddbfSLukasz Majewski 
311b819ddbfSLukasz Majewski 	return value;
312b819ddbfSLukasz Majewski }
313b819ddbfSLukasz Majewski 
state_dfu_idle(struct f_dfu * f_dfu,const struct usb_ctrlrequest * ctrl,struct usb_gadget * gadget,struct usb_request * req)314b819ddbfSLukasz Majewski static int state_dfu_idle(struct f_dfu *f_dfu,
315b819ddbfSLukasz Majewski 			  const struct usb_ctrlrequest *ctrl,
316b819ddbfSLukasz Majewski 			  struct usb_gadget *gadget,
317b819ddbfSLukasz Majewski 			  struct usb_request *req)
318b819ddbfSLukasz Majewski {
319b819ddbfSLukasz Majewski 	u16 w_value = le16_to_cpu(ctrl->wValue);
320b819ddbfSLukasz Majewski 	u16 len = le16_to_cpu(ctrl->wLength);
321b819ddbfSLukasz Majewski 	int value = 0;
322b819ddbfSLukasz Majewski 
323b819ddbfSLukasz Majewski 	switch (ctrl->bRequest) {
324b819ddbfSLukasz Majewski 	case USB_REQ_DFU_DNLOAD:
325b819ddbfSLukasz Majewski 		if (len == 0) {
326b819ddbfSLukasz Majewski 			f_dfu->dfu_state = DFU_STATE_dfuERROR;
327b819ddbfSLukasz Majewski 			value = RET_STALL;
328b819ddbfSLukasz Majewski 			break;
329b819ddbfSLukasz Majewski 		}
330b819ddbfSLukasz Majewski 		f_dfu->dfu_state = DFU_STATE_dfuDNLOAD_SYNC;
331b819ddbfSLukasz Majewski 		f_dfu->blk_seq_num = w_value;
332b819ddbfSLukasz Majewski 		value = handle_dnload(gadget, len);
333b819ddbfSLukasz Majewski 		break;
334b819ddbfSLukasz Majewski 	case USB_REQ_DFU_UPLOAD:
335b819ddbfSLukasz Majewski 		f_dfu->dfu_state = DFU_STATE_dfuUPLOAD_IDLE;
336b819ddbfSLukasz Majewski 		f_dfu->blk_seq_num = 0;
337b819ddbfSLukasz Majewski 		value = handle_upload(req, len);
338b819ddbfSLukasz Majewski 		break;
339b819ddbfSLukasz Majewski 	case USB_REQ_DFU_ABORT:
340b819ddbfSLukasz Majewski 		/* no zlp? */
341b819ddbfSLukasz Majewski 		value = RET_ZLP;
342b819ddbfSLukasz Majewski 		break;
343b819ddbfSLukasz Majewski 	case USB_REQ_DFU_GETSTATUS:
344d4287766SPatrick Delaunay 		value = handle_getstatus(req);
345b819ddbfSLukasz Majewski 		break;
346b819ddbfSLukasz Majewski 	case USB_REQ_DFU_GETSTATE:
347f11bb252SPatrick Delaunay 		value = handle_getstate(req);
348b819ddbfSLukasz Majewski 		break;
349b819ddbfSLukasz Majewski 	case USB_REQ_DFU_DETACH:
350b819ddbfSLukasz Majewski 		/*
351b819ddbfSLukasz Majewski 		 * Proprietary extension: 'detach' from idle mode and
352b819ddbfSLukasz Majewski 		 * get back to runtime mode in case of USB Reset.  As
353b819ddbfSLukasz Majewski 		 * much as I dislike this, we just can't use every USB
354b819ddbfSLukasz Majewski 		 * bus reset to switch back to runtime mode, since at
355b819ddbfSLukasz Majewski 		 * least the Linux USB stack likes to send a number of
356b819ddbfSLukasz Majewski 		 * resets in a row :(
357b819ddbfSLukasz Majewski 		 */
358b819ddbfSLukasz Majewski 		f_dfu->dfu_state =
359b819ddbfSLukasz Majewski 			DFU_STATE_dfuMANIFEST_WAIT_RST;
360b819ddbfSLukasz Majewski 		to_runtime_mode(f_dfu);
361b819ddbfSLukasz Majewski 		f_dfu->dfu_state = DFU_STATE_appIDLE;
3626bed7ce5SLukasz Majewski 
363fe1b28c9SRob Herring 		g_dnl_trigger_detach();
364b819ddbfSLukasz Majewski 		break;
365b819ddbfSLukasz Majewski 	default:
366b819ddbfSLukasz Majewski 		f_dfu->dfu_state = DFU_STATE_dfuERROR;
367b819ddbfSLukasz Majewski 		value = RET_STALL;
368b819ddbfSLukasz Majewski 		break;
369b819ddbfSLukasz Majewski 	}
370b819ddbfSLukasz Majewski 
371b819ddbfSLukasz Majewski 	return value;
372b819ddbfSLukasz Majewski }
373b819ddbfSLukasz Majewski 
state_dfu_dnload_sync(struct f_dfu * f_dfu,const struct usb_ctrlrequest * ctrl,struct usb_gadget * gadget,struct usb_request * req)374b819ddbfSLukasz Majewski static int state_dfu_dnload_sync(struct f_dfu *f_dfu,
375b819ddbfSLukasz Majewski 				 const struct usb_ctrlrequest *ctrl,
376b819ddbfSLukasz Majewski 				 struct usb_gadget *gadget,
377b819ddbfSLukasz Majewski 				 struct usb_request *req)
378b819ddbfSLukasz Majewski {
379b819ddbfSLukasz Majewski 	int value = 0;
380b819ddbfSLukasz Majewski 
381b819ddbfSLukasz Majewski 	switch (ctrl->bRequest) {
382b819ddbfSLukasz Majewski 	case USB_REQ_DFU_GETSTATUS:
383d4287766SPatrick Delaunay 		value = handle_getstatus(req);
384b819ddbfSLukasz Majewski 		break;
385b819ddbfSLukasz Majewski 	case USB_REQ_DFU_GETSTATE:
386f11bb252SPatrick Delaunay 		value = handle_getstate(req);
387b819ddbfSLukasz Majewski 		break;
388b819ddbfSLukasz Majewski 	default:
389b819ddbfSLukasz Majewski 		f_dfu->dfu_state = DFU_STATE_dfuERROR;
390b819ddbfSLukasz Majewski 		value = RET_STALL;
391b819ddbfSLukasz Majewski 		break;
392b819ddbfSLukasz Majewski 	}
393b819ddbfSLukasz Majewski 
394b819ddbfSLukasz Majewski 	return value;
395b819ddbfSLukasz Majewski }
396b819ddbfSLukasz Majewski 
state_dfu_dnbusy(struct f_dfu * f_dfu,const struct usb_ctrlrequest * ctrl,struct usb_gadget * gadget,struct usb_request * req)397b819ddbfSLukasz Majewski static int state_dfu_dnbusy(struct f_dfu *f_dfu,
398b819ddbfSLukasz Majewski 			    const struct usb_ctrlrequest *ctrl,
399b819ddbfSLukasz Majewski 			    struct usb_gadget *gadget,
400b819ddbfSLukasz Majewski 			    struct usb_request *req)
401b819ddbfSLukasz Majewski {
402b819ddbfSLukasz Majewski 	int value = 0;
403b819ddbfSLukasz Majewski 
404b819ddbfSLukasz Majewski 	switch (ctrl->bRequest) {
405b819ddbfSLukasz Majewski 	case USB_REQ_DFU_GETSTATUS:
406d4287766SPatrick Delaunay 		value = handle_getstatus(req);
407b819ddbfSLukasz Majewski 		break;
408b819ddbfSLukasz Majewski 	default:
409b819ddbfSLukasz Majewski 		f_dfu->dfu_state = DFU_STATE_dfuERROR;
410b819ddbfSLukasz Majewski 		value = RET_STALL;
411b819ddbfSLukasz Majewski 		break;
412b819ddbfSLukasz Majewski 	}
413b819ddbfSLukasz Majewski 
414b819ddbfSLukasz Majewski 	return value;
415b819ddbfSLukasz Majewski }
416b819ddbfSLukasz Majewski 
state_dfu_dnload_idle(struct f_dfu * f_dfu,const struct usb_ctrlrequest * ctrl,struct usb_gadget * gadget,struct usb_request * req)417b819ddbfSLukasz Majewski static int state_dfu_dnload_idle(struct f_dfu *f_dfu,
418b819ddbfSLukasz Majewski 				 const struct usb_ctrlrequest *ctrl,
419b819ddbfSLukasz Majewski 				 struct usb_gadget *gadget,
420b819ddbfSLukasz Majewski 				 struct usb_request *req)
421b819ddbfSLukasz Majewski {
422b819ddbfSLukasz Majewski 	u16 w_value = le16_to_cpu(ctrl->wValue);
423b819ddbfSLukasz Majewski 	u16 len = le16_to_cpu(ctrl->wLength);
424b819ddbfSLukasz Majewski 	int value = 0;
425b819ddbfSLukasz Majewski 
426b819ddbfSLukasz Majewski 	switch (ctrl->bRequest) {
427b819ddbfSLukasz Majewski 	case USB_REQ_DFU_DNLOAD:
428b819ddbfSLukasz Majewski 		f_dfu->dfu_state = DFU_STATE_dfuDNLOAD_SYNC;
429b819ddbfSLukasz Majewski 		f_dfu->blk_seq_num = w_value;
430b819ddbfSLukasz Majewski 		value = handle_dnload(gadget, len);
431b819ddbfSLukasz Majewski 		break;
432b819ddbfSLukasz Majewski 	case USB_REQ_DFU_ABORT:
433b819ddbfSLukasz Majewski 		f_dfu->dfu_state = DFU_STATE_dfuIDLE;
434b819ddbfSLukasz Majewski 		value = RET_ZLP;
435b819ddbfSLukasz Majewski 		break;
436b819ddbfSLukasz Majewski 	case USB_REQ_DFU_GETSTATUS:
437d4287766SPatrick Delaunay 		value = handle_getstatus(req);
438b819ddbfSLukasz Majewski 		break;
439b819ddbfSLukasz Majewski 	case USB_REQ_DFU_GETSTATE:
440f11bb252SPatrick Delaunay 		value = handle_getstate(req);
441b819ddbfSLukasz Majewski 		break;
442b819ddbfSLukasz Majewski 	default:
443b819ddbfSLukasz Majewski 		f_dfu->dfu_state = DFU_STATE_dfuERROR;
444b819ddbfSLukasz Majewski 		value = RET_STALL;
445b819ddbfSLukasz Majewski 		break;
446b819ddbfSLukasz Majewski 	}
447b819ddbfSLukasz Majewski 
448b819ddbfSLukasz Majewski 	return value;
449b819ddbfSLukasz Majewski }
450b819ddbfSLukasz Majewski 
state_dfu_manifest_sync(struct f_dfu * f_dfu,const struct usb_ctrlrequest * ctrl,struct usb_gadget * gadget,struct usb_request * req)451b819ddbfSLukasz Majewski static int state_dfu_manifest_sync(struct f_dfu *f_dfu,
452b819ddbfSLukasz Majewski 				   const struct usb_ctrlrequest *ctrl,
453b819ddbfSLukasz Majewski 				   struct usb_gadget *gadget,
454b819ddbfSLukasz Majewski 				   struct usb_request *req)
455b819ddbfSLukasz Majewski {
456b819ddbfSLukasz Majewski 	int value = 0;
457b819ddbfSLukasz Majewski 
458b819ddbfSLukasz Majewski 	switch (ctrl->bRequest) {
459b819ddbfSLukasz Majewski 	case USB_REQ_DFU_GETSTATUS:
460b819ddbfSLukasz Majewski 		/* We're MainfestationTolerant */
461001a8319SHeiko Schocher 		f_dfu->dfu_state = DFU_STATE_dfuMANIFEST;
462d4287766SPatrick Delaunay 		value = handle_getstatus(req);
463b819ddbfSLukasz Majewski 		f_dfu->blk_seq_num = 0;
464001a8319SHeiko Schocher 		req->complete = dnload_request_flush;
465b819ddbfSLukasz Majewski 		break;
466b819ddbfSLukasz Majewski 	case USB_REQ_DFU_GETSTATE:
467f11bb252SPatrick Delaunay 		value = handle_getstate(req);
468b819ddbfSLukasz Majewski 		break;
469b819ddbfSLukasz Majewski 	default:
470b819ddbfSLukasz Majewski 		f_dfu->dfu_state = DFU_STATE_dfuERROR;
471b819ddbfSLukasz Majewski 		value = RET_STALL;
472b819ddbfSLukasz Majewski 		break;
473b819ddbfSLukasz Majewski 	}
474b819ddbfSLukasz Majewski 
475b819ddbfSLukasz Majewski 	return value;
476b819ddbfSLukasz Majewski }
477b819ddbfSLukasz Majewski 
state_dfu_manifest(struct f_dfu * f_dfu,const struct usb_ctrlrequest * ctrl,struct usb_gadget * gadget,struct usb_request * req)478001a8319SHeiko Schocher static int state_dfu_manifest(struct f_dfu *f_dfu,
479001a8319SHeiko Schocher 			      const struct usb_ctrlrequest *ctrl,
480001a8319SHeiko Schocher 			      struct usb_gadget *gadget,
481001a8319SHeiko Schocher 			      struct usb_request *req)
482001a8319SHeiko Schocher {
483001a8319SHeiko Schocher 	int value = 0;
484001a8319SHeiko Schocher 
485001a8319SHeiko Schocher 	switch (ctrl->bRequest) {
486001a8319SHeiko Schocher 	case USB_REQ_DFU_GETSTATUS:
487001a8319SHeiko Schocher 		/* We're MainfestationTolerant */
488001a8319SHeiko Schocher 		f_dfu->dfu_state = DFU_STATE_dfuIDLE;
489d4287766SPatrick Delaunay 		value = handle_getstatus(req);
490001a8319SHeiko Schocher 		f_dfu->blk_seq_num = 0;
491001a8319SHeiko Schocher 		puts("DOWNLOAD ... OK\nCtrl+C to exit ...\n");
492001a8319SHeiko Schocher 		break;
493001a8319SHeiko Schocher 	case USB_REQ_DFU_GETSTATE:
494f11bb252SPatrick Delaunay 		value = handle_getstate(req);
495001a8319SHeiko Schocher 		break;
496001a8319SHeiko Schocher 	default:
497001a8319SHeiko Schocher 		f_dfu->dfu_state = DFU_STATE_dfuERROR;
498001a8319SHeiko Schocher 		value = RET_STALL;
499001a8319SHeiko Schocher 		break;
500001a8319SHeiko Schocher 	}
501001a8319SHeiko Schocher 	return value;
502001a8319SHeiko Schocher }
503001a8319SHeiko Schocher 
state_dfu_upload_idle(struct f_dfu * f_dfu,const struct usb_ctrlrequest * ctrl,struct usb_gadget * gadget,struct usb_request * req)504b819ddbfSLukasz Majewski static int state_dfu_upload_idle(struct f_dfu *f_dfu,
505b819ddbfSLukasz Majewski 				 const struct usb_ctrlrequest *ctrl,
506b819ddbfSLukasz Majewski 				 struct usb_gadget *gadget,
507b819ddbfSLukasz Majewski 				 struct usb_request *req)
508b819ddbfSLukasz Majewski {
509b819ddbfSLukasz Majewski 	u16 w_value = le16_to_cpu(ctrl->wValue);
510b819ddbfSLukasz Majewski 	u16 len = le16_to_cpu(ctrl->wLength);
511b819ddbfSLukasz Majewski 	int value = 0;
512b819ddbfSLukasz Majewski 
513b819ddbfSLukasz Majewski 	switch (ctrl->bRequest) {
514b819ddbfSLukasz Majewski 	case USB_REQ_DFU_UPLOAD:
515b819ddbfSLukasz Majewski 		/* state transition if less data then requested */
516b819ddbfSLukasz Majewski 		f_dfu->blk_seq_num = w_value;
517b819ddbfSLukasz Majewski 		value = handle_upload(req, len);
518b819ddbfSLukasz Majewski 		if (value >= 0 && value < len)
519b819ddbfSLukasz Majewski 			f_dfu->dfu_state = DFU_STATE_dfuIDLE;
520b819ddbfSLukasz Majewski 		break;
521b819ddbfSLukasz Majewski 	case USB_REQ_DFU_ABORT:
522b819ddbfSLukasz Majewski 		f_dfu->dfu_state = DFU_STATE_dfuIDLE;
523b819ddbfSLukasz Majewski 		/* no zlp? */
524b819ddbfSLukasz Majewski 		value = RET_ZLP;
525b819ddbfSLukasz Majewski 		break;
526b819ddbfSLukasz Majewski 	case USB_REQ_DFU_GETSTATUS:
527d4287766SPatrick Delaunay 		value = handle_getstatus(req);
528b819ddbfSLukasz Majewski 		break;
529b819ddbfSLukasz Majewski 	case USB_REQ_DFU_GETSTATE:
530f11bb252SPatrick Delaunay 		value = handle_getstate(req);
531b819ddbfSLukasz Majewski 		break;
532b819ddbfSLukasz Majewski 	default:
533b819ddbfSLukasz Majewski 		f_dfu->dfu_state = DFU_STATE_dfuERROR;
534b819ddbfSLukasz Majewski 		value = RET_STALL;
535b819ddbfSLukasz Majewski 		break;
536b819ddbfSLukasz Majewski 	}
537b819ddbfSLukasz Majewski 
538b819ddbfSLukasz Majewski 	return value;
539b819ddbfSLukasz Majewski }
540b819ddbfSLukasz Majewski 
state_dfu_error(struct f_dfu * f_dfu,const struct usb_ctrlrequest * ctrl,struct usb_gadget * gadget,struct usb_request * req)541b819ddbfSLukasz Majewski static int state_dfu_error(struct f_dfu *f_dfu,
542b819ddbfSLukasz Majewski 				 const struct usb_ctrlrequest *ctrl,
543b819ddbfSLukasz Majewski 				 struct usb_gadget *gadget,
544b819ddbfSLukasz Majewski 				 struct usb_request *req)
545b819ddbfSLukasz Majewski {
546b819ddbfSLukasz Majewski 	int value = 0;
547b819ddbfSLukasz Majewski 
548b819ddbfSLukasz Majewski 	switch (ctrl->bRequest) {
549b819ddbfSLukasz Majewski 	case USB_REQ_DFU_GETSTATUS:
550d4287766SPatrick Delaunay 		value = handle_getstatus(req);
551b819ddbfSLukasz Majewski 		break;
552b819ddbfSLukasz Majewski 	case USB_REQ_DFU_GETSTATE:
553f11bb252SPatrick Delaunay 		value = handle_getstate(req);
554b819ddbfSLukasz Majewski 		break;
555b819ddbfSLukasz Majewski 	case USB_REQ_DFU_CLRSTATUS:
556b819ddbfSLukasz Majewski 		f_dfu->dfu_state = DFU_STATE_dfuIDLE;
557b819ddbfSLukasz Majewski 		f_dfu->dfu_status = DFU_STATUS_OK;
558b819ddbfSLukasz Majewski 		/* no zlp? */
559b819ddbfSLukasz Majewski 		value = RET_ZLP;
560b819ddbfSLukasz Majewski 		break;
561b819ddbfSLukasz Majewski 	default:
562b819ddbfSLukasz Majewski 		f_dfu->dfu_state = DFU_STATE_dfuERROR;
563b819ddbfSLukasz Majewski 		value = RET_STALL;
564b819ddbfSLukasz Majewski 		break;
565b819ddbfSLukasz Majewski 	}
566b819ddbfSLukasz Majewski 
567b819ddbfSLukasz Majewski 	return value;
568b819ddbfSLukasz Majewski }
569b819ddbfSLukasz Majewski 
570b819ddbfSLukasz Majewski static dfu_state_fn dfu_state[] = {
571b819ddbfSLukasz Majewski 	state_app_idle,          /* DFU_STATE_appIDLE */
572b819ddbfSLukasz Majewski 	state_app_detach,        /* DFU_STATE_appDETACH */
573b819ddbfSLukasz Majewski 	state_dfu_idle,          /* DFU_STATE_dfuIDLE */
574b819ddbfSLukasz Majewski 	state_dfu_dnload_sync,   /* DFU_STATE_dfuDNLOAD_SYNC */
575b819ddbfSLukasz Majewski 	state_dfu_dnbusy,        /* DFU_STATE_dfuDNBUSY */
576b819ddbfSLukasz Majewski 	state_dfu_dnload_idle,   /* DFU_STATE_dfuDNLOAD_IDLE */
577b819ddbfSLukasz Majewski 	state_dfu_manifest_sync, /* DFU_STATE_dfuMANIFEST_SYNC */
578001a8319SHeiko Schocher 	state_dfu_manifest,	 /* DFU_STATE_dfuMANIFEST */
579b819ddbfSLukasz Majewski 	NULL,                    /* DFU_STATE_dfuMANIFEST_WAIT_RST */
580b819ddbfSLukasz Majewski 	state_dfu_upload_idle,   /* DFU_STATE_dfuUPLOAD_IDLE */
581b819ddbfSLukasz Majewski 	state_dfu_error          /* DFU_STATE_dfuERROR */
582b819ddbfSLukasz Majewski };
583b819ddbfSLukasz Majewski 
584b819ddbfSLukasz Majewski static int
dfu_handle(struct usb_function * f,const struct usb_ctrlrequest * ctrl)585b819ddbfSLukasz Majewski dfu_handle(struct usb_function *f, const struct usb_ctrlrequest *ctrl)
586b819ddbfSLukasz Majewski {
587b819ddbfSLukasz Majewski 	struct usb_gadget *gadget = f->config->cdev->gadget;
588b819ddbfSLukasz Majewski 	struct usb_request *req = f->config->cdev->req;
589b819ddbfSLukasz Majewski 	struct f_dfu *f_dfu = f->config->cdev->req->context;
590b819ddbfSLukasz Majewski 	u16 len = le16_to_cpu(ctrl->wLength);
591b819ddbfSLukasz Majewski 	u16 w_value = le16_to_cpu(ctrl->wValue);
592b819ddbfSLukasz Majewski 	int value = 0;
593b819ddbfSLukasz Majewski 	u8 req_type = ctrl->bRequestType & USB_TYPE_MASK;
594b819ddbfSLukasz Majewski 
595b819ddbfSLukasz Majewski 	debug("w_value: 0x%x len: 0x%x\n", w_value, len);
596b819ddbfSLukasz Majewski 	debug("req_type: 0x%x ctrl->bRequest: 0x%x f_dfu->dfu_state: 0x%x\n",
597b819ddbfSLukasz Majewski 	       req_type, ctrl->bRequest, f_dfu->dfu_state);
598b819ddbfSLukasz Majewski 
599b819ddbfSLukasz Majewski 	if (req_type == USB_TYPE_STANDARD) {
600b819ddbfSLukasz Majewski 		if (ctrl->bRequest == USB_REQ_GET_DESCRIPTOR &&
601b819ddbfSLukasz Majewski 		    (w_value >> 8) == DFU_DT_FUNC) {
602b819ddbfSLukasz Majewski 			value = min(len, (u16) sizeof(dfu_func));
603b819ddbfSLukasz Majewski 			memcpy(req->buf, &dfu_func, value);
604b819ddbfSLukasz Majewski 		}
605b819ddbfSLukasz Majewski 	} else /* DFU specific request */
606b819ddbfSLukasz Majewski 		value = dfu_state[f_dfu->dfu_state] (f_dfu, ctrl, gadget, req);
607b819ddbfSLukasz Majewski 
608b819ddbfSLukasz Majewski 	if (value >= 0) {
609b819ddbfSLukasz Majewski 		req->length = value;
610b819ddbfSLukasz Majewski 		req->zero = value < len;
611b819ddbfSLukasz Majewski 		value = usb_ep_queue(gadget->ep0, req, 0);
612b819ddbfSLukasz Majewski 		if (value < 0) {
613b819ddbfSLukasz Majewski 			debug("ep_queue --> %d\n", value);
614b819ddbfSLukasz Majewski 			req->status = 0;
615b819ddbfSLukasz Majewski 		}
616b819ddbfSLukasz Majewski 	}
617b819ddbfSLukasz Majewski 
618b819ddbfSLukasz Majewski 	return value;
619b819ddbfSLukasz Majewski }
620b819ddbfSLukasz Majewski 
621b819ddbfSLukasz Majewski /*-------------------------------------------------------------------------*/
622b819ddbfSLukasz Majewski 
623b819ddbfSLukasz Majewski static int
dfu_prepare_strings(struct f_dfu * f_dfu,int n)624b819ddbfSLukasz Majewski dfu_prepare_strings(struct f_dfu *f_dfu, int n)
625b819ddbfSLukasz Majewski {
626b819ddbfSLukasz Majewski 	struct dfu_entity *de = NULL;
627b819ddbfSLukasz Majewski 	int i = 0;
628b819ddbfSLukasz Majewski 
629b819ddbfSLukasz Majewski 	f_dfu->strings = calloc(sizeof(struct usb_string), n + 1);
630b819ddbfSLukasz Majewski 	if (!f_dfu->strings)
63112ff19dbSPeng Fan 		return -ENOMEM;
632b819ddbfSLukasz Majewski 
633b819ddbfSLukasz Majewski 	for (i = 0; i < n; ++i) {
634b819ddbfSLukasz Majewski 		de = dfu_get_entity(i);
635b819ddbfSLukasz Majewski 		f_dfu->strings[i].s = de->name;
636b819ddbfSLukasz Majewski 	}
637b819ddbfSLukasz Majewski 
638b819ddbfSLukasz Majewski 	f_dfu->strings[i].id = 0;
639b819ddbfSLukasz Majewski 	f_dfu->strings[i].s = NULL;
640b819ddbfSLukasz Majewski 
641b819ddbfSLukasz Majewski 	return 0;
642b819ddbfSLukasz Majewski }
643b819ddbfSLukasz Majewski 
dfu_prepare_function(struct f_dfu * f_dfu,int n)644b819ddbfSLukasz Majewski static int dfu_prepare_function(struct f_dfu *f_dfu, int n)
645b819ddbfSLukasz Majewski {
646b819ddbfSLukasz Majewski 	struct usb_interface_descriptor *d;
647b819ddbfSLukasz Majewski 	int i = 0;
648b819ddbfSLukasz Majewski 
6498987012fSPatrick Delaunay 	f_dfu->function = calloc(sizeof(struct usb_descriptor_header *), n + 2);
650b819ddbfSLukasz Majewski 	if (!f_dfu->function)
651b819ddbfSLukasz Majewski 		goto enomem;
652b819ddbfSLukasz Majewski 
653b819ddbfSLukasz Majewski 	for (i = 0; i < n; ++i) {
654b819ddbfSLukasz Majewski 		d = calloc(sizeof(*d), 1);
655b819ddbfSLukasz Majewski 		if (!d)
656b819ddbfSLukasz Majewski 			goto enomem;
657b819ddbfSLukasz Majewski 
658b819ddbfSLukasz Majewski 		d->bLength =		sizeof(*d);
659b819ddbfSLukasz Majewski 		d->bDescriptorType =	USB_DT_INTERFACE;
660b819ddbfSLukasz Majewski 		d->bAlternateSetting =	i;
661b819ddbfSLukasz Majewski 		d->bNumEndpoints =	0;
662b819ddbfSLukasz Majewski 		d->bInterfaceClass =	USB_CLASS_APP_SPEC;
663b819ddbfSLukasz Majewski 		d->bInterfaceSubClass =	1;
664b819ddbfSLukasz Majewski 		d->bInterfaceProtocol =	2;
665b819ddbfSLukasz Majewski 
666b819ddbfSLukasz Majewski 		f_dfu->function[i] = (struct usb_descriptor_header *)d;
667b819ddbfSLukasz Majewski 	}
6688987012fSPatrick Delaunay 
6698987012fSPatrick Delaunay 	/* add DFU Functional Descriptor */
6708987012fSPatrick Delaunay 	f_dfu->function[i] = calloc(sizeof(dfu_func), 1);
6718987012fSPatrick Delaunay 	if (!f_dfu->function[i])
6728987012fSPatrick Delaunay 		goto enomem;
6738987012fSPatrick Delaunay 	memcpy(f_dfu->function[i], &dfu_func, sizeof(dfu_func));
6748987012fSPatrick Delaunay 
6758987012fSPatrick Delaunay 	i++;
676b819ddbfSLukasz Majewski 	f_dfu->function[i] = NULL;
677b819ddbfSLukasz Majewski 
678b819ddbfSLukasz Majewski 	return 0;
679b819ddbfSLukasz Majewski 
680b819ddbfSLukasz Majewski enomem:
681b819ddbfSLukasz Majewski 	while (i) {
682b819ddbfSLukasz Majewski 		free(f_dfu->function[--i]);
683b819ddbfSLukasz Majewski 		f_dfu->function[i] = NULL;
684b819ddbfSLukasz Majewski 	}
685b819ddbfSLukasz Majewski 	free(f_dfu->function);
686b819ddbfSLukasz Majewski 
687b819ddbfSLukasz Majewski 	return -ENOMEM;
688b819ddbfSLukasz Majewski }
689b819ddbfSLukasz Majewski 
dfu_bind(struct usb_configuration * c,struct usb_function * f)690b819ddbfSLukasz Majewski static int dfu_bind(struct usb_configuration *c, struct usb_function *f)
691b819ddbfSLukasz Majewski {
692b819ddbfSLukasz Majewski 	struct usb_composite_dev *cdev = c->cdev;
693b819ddbfSLukasz Majewski 	struct f_dfu *f_dfu = func_to_dfu(f);
6949bf9e813SFelipe Balbi 	const char *s;
695b819ddbfSLukasz Majewski 	int alt_num = dfu_get_alt_number();
696b819ddbfSLukasz Majewski 	int rv, id, i;
697b819ddbfSLukasz Majewski 
698b819ddbfSLukasz Majewski 	id = usb_interface_id(c, f);
699b819ddbfSLukasz Majewski 	if (id < 0)
700b819ddbfSLukasz Majewski 		return id;
701b819ddbfSLukasz Majewski 	dfu_intf_runtime.bInterfaceNumber = id;
702b819ddbfSLukasz Majewski 
703b819ddbfSLukasz Majewski 	f_dfu->dfu_state = DFU_STATE_appIDLE;
704b819ddbfSLukasz Majewski 	f_dfu->dfu_status = DFU_STATUS_OK;
705b819ddbfSLukasz Majewski 
706b819ddbfSLukasz Majewski 	rv = dfu_prepare_function(f_dfu, alt_num);
707b819ddbfSLukasz Majewski 	if (rv)
708b819ddbfSLukasz Majewski 		goto error;
709b819ddbfSLukasz Majewski 
710b819ddbfSLukasz Majewski 	rv = dfu_prepare_strings(f_dfu, alt_num);
711b819ddbfSLukasz Majewski 	if (rv)
712b819ddbfSLukasz Majewski 		goto error;
713b819ddbfSLukasz Majewski 	for (i = 0; i < alt_num; i++) {
714b819ddbfSLukasz Majewski 		id = usb_string_id(cdev);
715b819ddbfSLukasz Majewski 		if (id < 0)
716b819ddbfSLukasz Majewski 			return id;
717b819ddbfSLukasz Majewski 		f_dfu->strings[i].id = id;
718b819ddbfSLukasz Majewski 		((struct usb_interface_descriptor *)f_dfu->function[i])
719b819ddbfSLukasz Majewski 			->iInterface = id;
720b819ddbfSLukasz Majewski 	}
721b819ddbfSLukasz Majewski 
722ad5f9778SHeiko Schocher 	to_dfu_mode(f_dfu);
723ad5f9778SHeiko Schocher 
724b819ddbfSLukasz Majewski 	stringtab_dfu.strings = f_dfu->strings;
725b819ddbfSLukasz Majewski 
726b819ddbfSLukasz Majewski 	cdev->req->context = f_dfu;
727b819ddbfSLukasz Majewski 
72800caae6dSSimon Glass 	s = env_get("serial#");
7299bf9e813SFelipe Balbi 	if (s)
7309bf9e813SFelipe Balbi 		g_dnl_set_serialnumber((char *)s);
7319bf9e813SFelipe Balbi 
732b819ddbfSLukasz Majewski error:
733b819ddbfSLukasz Majewski 	return rv;
734b819ddbfSLukasz Majewski }
735b819ddbfSLukasz Majewski 
dfu_unbind(struct usb_configuration * c,struct usb_function * f)736b819ddbfSLukasz Majewski static void dfu_unbind(struct usb_configuration *c, struct usb_function *f)
737b819ddbfSLukasz Majewski {
738b819ddbfSLukasz Majewski 	struct f_dfu *f_dfu = func_to_dfu(f);
739b819ddbfSLukasz Majewski 	int alt_num = dfu_get_alt_number();
740b819ddbfSLukasz Majewski 	int i;
741b819ddbfSLukasz Majewski 
742b819ddbfSLukasz Majewski 	if (f_dfu->strings) {
743b819ddbfSLukasz Majewski 		i = alt_num;
744b819ddbfSLukasz Majewski 		while (i)
745b819ddbfSLukasz Majewski 			f_dfu->strings[--i].s = NULL;
746b819ddbfSLukasz Majewski 
747b819ddbfSLukasz Majewski 		free(f_dfu->strings);
748b819ddbfSLukasz Majewski 	}
749b819ddbfSLukasz Majewski 
750b819ddbfSLukasz Majewski 	if (f_dfu->function) {
751b819ddbfSLukasz Majewski 		i = alt_num;
752*2b5fff1cSRalph Siemsen 		i++; /* free DFU Functional Descriptor */
753b819ddbfSLukasz Majewski 		while (i) {
754b819ddbfSLukasz Majewski 			free(f_dfu->function[--i]);
755b819ddbfSLukasz Majewski 			f_dfu->function[i] = NULL;
756b819ddbfSLukasz Majewski 		}
757b819ddbfSLukasz Majewski 		free(f_dfu->function);
758b819ddbfSLukasz Majewski 	}
759b819ddbfSLukasz Majewski 
760b819ddbfSLukasz Majewski 	free(f_dfu);
761b819ddbfSLukasz Majewski }
762b819ddbfSLukasz Majewski 
dfu_set_alt(struct usb_function * f,unsigned intf,unsigned alt)763b819ddbfSLukasz Majewski static int dfu_set_alt(struct usb_function *f, unsigned intf, unsigned alt)
764b819ddbfSLukasz Majewski {
765b819ddbfSLukasz Majewski 	struct f_dfu *f_dfu = func_to_dfu(f);
766b819ddbfSLukasz Majewski 
767b819ddbfSLukasz Majewski 	debug("%s: intf:%d alt:%d\n", __func__, intf, alt);
768b819ddbfSLukasz Majewski 
769b819ddbfSLukasz Majewski 	f_dfu->altsetting = alt;
7703ee9593fSStephen Warren 	f_dfu->dfu_state = DFU_STATE_dfuIDLE;
7713ee9593fSStephen Warren 	f_dfu->dfu_status = DFU_STATUS_OK;
772b819ddbfSLukasz Majewski 
773b819ddbfSLukasz Majewski 	return 0;
774b819ddbfSLukasz Majewski }
775b819ddbfSLukasz Majewski 
__dfu_get_alt(struct usb_function * f,unsigned intf)7761fd81b7cSStefan Roese static int __dfu_get_alt(struct usb_function *f, unsigned intf)
7771fd81b7cSStefan Roese {
7781fd81b7cSStefan Roese 	struct f_dfu *f_dfu = func_to_dfu(f);
7791fd81b7cSStefan Roese 
7801fd81b7cSStefan Roese 	return f_dfu->altsetting;
7811fd81b7cSStefan Roese }
7821fd81b7cSStefan Roese 
783b819ddbfSLukasz Majewski /* TODO: is this really what we need here? */
dfu_disable(struct usb_function * f)784b819ddbfSLukasz Majewski static void dfu_disable(struct usb_function *f)
785b819ddbfSLukasz Majewski {
786b819ddbfSLukasz Majewski 	struct f_dfu *f_dfu = func_to_dfu(f);
787b819ddbfSLukasz Majewski 	if (f_dfu->config == 0)
788b819ddbfSLukasz Majewski 		return;
789b819ddbfSLukasz Majewski 
790b819ddbfSLukasz Majewski 	debug("%s: reset config\n", __func__);
791b819ddbfSLukasz Majewski 
792b819ddbfSLukasz Majewski 	f_dfu->config = 0;
793b819ddbfSLukasz Majewski }
794b819ddbfSLukasz Majewski 
dfu_bind_config(struct usb_configuration * c)795b819ddbfSLukasz Majewski static int dfu_bind_config(struct usb_configuration *c)
796b819ddbfSLukasz Majewski {
797b819ddbfSLukasz Majewski 	struct f_dfu *f_dfu;
798b819ddbfSLukasz Majewski 	int status;
799b819ddbfSLukasz Majewski 
800b819ddbfSLukasz Majewski 	f_dfu = calloc(sizeof(*f_dfu), 1);
801b819ddbfSLukasz Majewski 	if (!f_dfu)
802b819ddbfSLukasz Majewski 		return -ENOMEM;
803b819ddbfSLukasz Majewski 	f_dfu->usb_function.name = "dfu";
804b819ddbfSLukasz Majewski 	f_dfu->usb_function.hs_descriptors = dfu_runtime_descs;
8059f3b8ed1SHeiko Schocher 	f_dfu->usb_function.descriptors = dfu_runtime_descs;
806b819ddbfSLukasz Majewski 	f_dfu->usb_function.bind = dfu_bind;
807b819ddbfSLukasz Majewski 	f_dfu->usb_function.unbind = dfu_unbind;
808b819ddbfSLukasz Majewski 	f_dfu->usb_function.set_alt = dfu_set_alt;
8091fd81b7cSStefan Roese 	f_dfu->usb_function.get_alt = __dfu_get_alt;
810b819ddbfSLukasz Majewski 	f_dfu->usb_function.disable = dfu_disable;
81177b95042SLukasz Majewski 	f_dfu->usb_function.strings = dfu_generic_strings;
81277b95042SLukasz Majewski 	f_dfu->usb_function.setup = dfu_handle;
81333fac4a6SLukasz Majewski 	f_dfu->poll_timeout = DFU_DEFAULT_POLL_TIMEOUT;
814b819ddbfSLukasz Majewski 
815b819ddbfSLukasz Majewski 	status = usb_add_function(c, &f_dfu->usb_function);
816b819ddbfSLukasz Majewski 	if (status)
817b819ddbfSLukasz Majewski 		free(f_dfu);
818b819ddbfSLukasz Majewski 
819b819ddbfSLukasz Majewski 	return status;
820b819ddbfSLukasz Majewski }
821b819ddbfSLukasz Majewski 
dfu_add(struct usb_configuration * c)822b819ddbfSLukasz Majewski int dfu_add(struct usb_configuration *c)
823b819ddbfSLukasz Majewski {
824b819ddbfSLukasz Majewski 	int id;
825b819ddbfSLukasz Majewski 
826b819ddbfSLukasz Majewski 	id = usb_string_id(c->cdev);
827b819ddbfSLukasz Majewski 	if (id < 0)
828b819ddbfSLukasz Majewski 		return id;
829b819ddbfSLukasz Majewski 	strings_dfu_generic[0].id = id;
830b819ddbfSLukasz Majewski 	dfu_intf_runtime.iInterface = id;
831b819ddbfSLukasz Majewski 
832b819ddbfSLukasz Majewski 	debug("%s: cdev: 0x%p gadget:0x%p gadget->ep0: 0x%p\n", __func__,
833b819ddbfSLukasz Majewski 	       c->cdev, c->cdev->gadget, c->cdev->gadget->ep0);
834b819ddbfSLukasz Majewski 
835b819ddbfSLukasz Majewski 	return dfu_bind_config(c);
836b819ddbfSLukasz Majewski }
837c4d0e856SMateusz Zalega 
838c4d0e856SMateusz Zalega DECLARE_GADGET_BIND_CALLBACK(usb_dnl_dfu, dfu_add);
839