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