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 47b819ddbfSLukasz Majewski typedef int (*dfu_state_fn) (struct f_dfu *, 48b819ddbfSLukasz Majewski const struct usb_ctrlrequest *, 49b819ddbfSLukasz Majewski struct usb_gadget *, 50b819ddbfSLukasz Majewski struct usb_request *); 51b819ddbfSLukasz Majewski 52b819ddbfSLukasz Majewski static inline struct f_dfu *func_to_dfu(struct usb_function *f) 53b819ddbfSLukasz Majewski { 54b819ddbfSLukasz Majewski return container_of(f, struct f_dfu, usb_function); 55b819ddbfSLukasz Majewski } 56b819ddbfSLukasz Majewski 57b819ddbfSLukasz Majewski static const struct dfu_function_descriptor dfu_func = { 58b819ddbfSLukasz Majewski .bLength = sizeof dfu_func, 59b819ddbfSLukasz Majewski .bDescriptorType = DFU_DT_FUNC, 60b819ddbfSLukasz Majewski .bmAttributes = DFU_BIT_WILL_DETACH | 61b819ddbfSLukasz Majewski DFU_BIT_MANIFESTATION_TOLERANT | 62b819ddbfSLukasz Majewski DFU_BIT_CAN_UPLOAD | 63b819ddbfSLukasz Majewski DFU_BIT_CAN_DNLOAD, 64b819ddbfSLukasz Majewski .wDetachTimeOut = 0, 65b819ddbfSLukasz Majewski .wTransferSize = DFU_USB_BUFSIZ, 66b819ddbfSLukasz Majewski .bcdDFUVersion = __constant_cpu_to_le16(0x0110), 67b819ddbfSLukasz Majewski }; 68b819ddbfSLukasz Majewski 69b819ddbfSLukasz Majewski static struct usb_interface_descriptor dfu_intf_runtime = { 70b819ddbfSLukasz Majewski .bLength = sizeof dfu_intf_runtime, 71b819ddbfSLukasz Majewski .bDescriptorType = USB_DT_INTERFACE, 72b819ddbfSLukasz Majewski .bNumEndpoints = 0, 73b819ddbfSLukasz Majewski .bInterfaceClass = USB_CLASS_APP_SPEC, 74b819ddbfSLukasz Majewski .bInterfaceSubClass = 1, 75b819ddbfSLukasz Majewski .bInterfaceProtocol = 1, 76b819ddbfSLukasz Majewski /* .iInterface = DYNAMIC */ 77b819ddbfSLukasz Majewski }; 78b819ddbfSLukasz Majewski 79b819ddbfSLukasz Majewski static struct usb_descriptor_header *dfu_runtime_descs[] = { 80b819ddbfSLukasz Majewski (struct usb_descriptor_header *) &dfu_intf_runtime, 81b819ddbfSLukasz Majewski NULL, 82b819ddbfSLukasz Majewski }; 83b819ddbfSLukasz Majewski 84b819ddbfSLukasz Majewski static const char dfu_name[] = "Device Firmware Upgrade"; 85b819ddbfSLukasz Majewski 86b819ddbfSLukasz Majewski /* 87b819ddbfSLukasz Majewski * static strings, in UTF-8 88b819ddbfSLukasz Majewski * 89b819ddbfSLukasz Majewski * dfu_generic configuration 90b819ddbfSLukasz Majewski */ 91b819ddbfSLukasz Majewski static struct usb_string strings_dfu_generic[] = { 92b819ddbfSLukasz Majewski [0].s = dfu_name, 93b819ddbfSLukasz Majewski { } /* end of list */ 94b819ddbfSLukasz Majewski }; 95b819ddbfSLukasz Majewski 96b819ddbfSLukasz Majewski static struct usb_gadget_strings stringtab_dfu_generic = { 97b819ddbfSLukasz Majewski .language = 0x0409, /* en-us */ 98b819ddbfSLukasz Majewski .strings = strings_dfu_generic, 99b819ddbfSLukasz Majewski }; 100b819ddbfSLukasz Majewski 101b819ddbfSLukasz Majewski static struct usb_gadget_strings *dfu_generic_strings[] = { 102b819ddbfSLukasz Majewski &stringtab_dfu_generic, 103b819ddbfSLukasz Majewski NULL, 104b819ddbfSLukasz Majewski }; 105b819ddbfSLukasz Majewski 106b819ddbfSLukasz Majewski /* 107b819ddbfSLukasz Majewski * usb_function specific 108b819ddbfSLukasz Majewski */ 109b819ddbfSLukasz Majewski static struct usb_gadget_strings stringtab_dfu = { 110b819ddbfSLukasz Majewski .language = 0x0409, /* en-us */ 111b819ddbfSLukasz Majewski /* 112b819ddbfSLukasz Majewski * .strings 113b819ddbfSLukasz Majewski * 114b819ddbfSLukasz Majewski * assigned during initialization, 115b819ddbfSLukasz Majewski * depends on number of flash entities 116b819ddbfSLukasz Majewski * 117b819ddbfSLukasz Majewski */ 118b819ddbfSLukasz Majewski }; 119b819ddbfSLukasz Majewski 120b819ddbfSLukasz Majewski static struct usb_gadget_strings *dfu_strings[] = { 121b819ddbfSLukasz Majewski &stringtab_dfu, 122b819ddbfSLukasz Majewski NULL, 123b819ddbfSLukasz Majewski }; 124b819ddbfSLukasz Majewski 12533fac4a6SLukasz Majewski static void dfu_set_poll_timeout(struct dfu_status *dstat, unsigned int ms) 12633fac4a6SLukasz Majewski { 12733fac4a6SLukasz Majewski /* 12833fac4a6SLukasz Majewski * The bwPollTimeout DFU_GETSTATUS request payload provides information 12933fac4a6SLukasz Majewski * about minimum time, in milliseconds, that the host should wait before 13033fac4a6SLukasz Majewski * sending a subsequent DFU_GETSTATUS request 13133fac4a6SLukasz Majewski * 13233fac4a6SLukasz Majewski * This permits the device to vary the delay depending on its need to 13333fac4a6SLukasz Majewski * erase or program the memory 13433fac4a6SLukasz Majewski * 13533fac4a6SLukasz Majewski */ 13633fac4a6SLukasz Majewski 13733fac4a6SLukasz Majewski unsigned char *p = (unsigned char *)&ms; 13833fac4a6SLukasz Majewski 13933fac4a6SLukasz Majewski if (!ms || (ms & ~DFU_POLL_TIMEOUT_MASK)) { 14033fac4a6SLukasz Majewski dstat->bwPollTimeout[0] = 0; 14133fac4a6SLukasz Majewski dstat->bwPollTimeout[1] = 0; 14233fac4a6SLukasz Majewski dstat->bwPollTimeout[2] = 0; 14333fac4a6SLukasz Majewski 14433fac4a6SLukasz Majewski return; 14533fac4a6SLukasz Majewski } 14633fac4a6SLukasz Majewski 14733fac4a6SLukasz Majewski dstat->bwPollTimeout[0] = *p++; 14833fac4a6SLukasz Majewski dstat->bwPollTimeout[1] = *p++; 14933fac4a6SLukasz Majewski dstat->bwPollTimeout[2] = *p; 15033fac4a6SLukasz Majewski } 15133fac4a6SLukasz Majewski 152b819ddbfSLukasz Majewski /*-------------------------------------------------------------------------*/ 153b819ddbfSLukasz Majewski 154b819ddbfSLukasz Majewski static void dnload_request_complete(struct usb_ep *ep, struct usb_request *req) 155b819ddbfSLukasz Majewski { 156b819ddbfSLukasz Majewski struct f_dfu *f_dfu = req->context; 1573ee9593fSStephen Warren int ret; 158b819ddbfSLukasz Majewski 1593ee9593fSStephen Warren ret = dfu_write(dfu_get_entity(f_dfu->altsetting), req->buf, 160b819ddbfSLukasz Majewski req->length, f_dfu->blk_seq_num); 1613ee9593fSStephen Warren if (ret) { 1623ee9593fSStephen Warren f_dfu->dfu_status = DFU_STATUS_errUNKNOWN; 1633ee9593fSStephen Warren f_dfu->dfu_state = DFU_STATE_dfuERROR; 1643ee9593fSStephen Warren } 165001a8319SHeiko Schocher } 166b819ddbfSLukasz Majewski 167001a8319SHeiko Schocher static void dnload_request_flush(struct usb_ep *ep, struct usb_request *req) 168001a8319SHeiko Schocher { 169001a8319SHeiko Schocher struct f_dfu *f_dfu = req->context; 1703ee9593fSStephen Warren int ret; 171001a8319SHeiko Schocher 1723ee9593fSStephen Warren ret = dfu_flush(dfu_get_entity(f_dfu->altsetting), req->buf, 173001a8319SHeiko Schocher req->length, f_dfu->blk_seq_num); 1743ee9593fSStephen Warren if (ret) { 1753ee9593fSStephen Warren f_dfu->dfu_status = DFU_STATUS_errUNKNOWN; 1763ee9593fSStephen Warren f_dfu->dfu_state = DFU_STATE_dfuERROR; 1773ee9593fSStephen Warren } 178b819ddbfSLukasz Majewski } 179b819ddbfSLukasz Majewski 180fc25fa27SHeiko Schocher static inline int dfu_get_manifest_timeout(struct dfu_entity *dfu) 181fc25fa27SHeiko Schocher { 182fc25fa27SHeiko Schocher return dfu->poll_timeout ? dfu->poll_timeout(dfu) : 183fc25fa27SHeiko Schocher DFU_MANIFEST_POLL_TIMEOUT; 184fc25fa27SHeiko Schocher } 185fc25fa27SHeiko Schocher 186b819ddbfSLukasz Majewski static void handle_getstatus(struct usb_request *req) 187b819ddbfSLukasz Majewski { 188b819ddbfSLukasz Majewski struct dfu_status *dstat = (struct dfu_status *)req->buf; 189b819ddbfSLukasz Majewski struct f_dfu *f_dfu = req->context; 190fc25fa27SHeiko Schocher struct dfu_entity *dfu = dfu_get_entity(f_dfu->altsetting); 191b819ddbfSLukasz Majewski 192001a8319SHeiko Schocher dfu_set_poll_timeout(dstat, 0); 193001a8319SHeiko Schocher 194b819ddbfSLukasz Majewski switch (f_dfu->dfu_state) { 195b819ddbfSLukasz Majewski case DFU_STATE_dfuDNLOAD_SYNC: 196b819ddbfSLukasz Majewski case DFU_STATE_dfuDNBUSY: 197b819ddbfSLukasz Majewski f_dfu->dfu_state = DFU_STATE_dfuDNLOAD_IDLE; 198b819ddbfSLukasz Majewski break; 199b819ddbfSLukasz Majewski case DFU_STATE_dfuMANIFEST_SYNC: 200001a8319SHeiko Schocher f_dfu->dfu_state = DFU_STATE_dfuMANIFEST; 201b819ddbfSLukasz Majewski break; 202001a8319SHeiko Schocher case DFU_STATE_dfuMANIFEST: 203fc25fa27SHeiko Schocher dfu_set_poll_timeout(dstat, dfu_get_manifest_timeout(dfu)); 204fc25fa27SHeiko Schocher break; 205b819ddbfSLukasz Majewski default: 206b819ddbfSLukasz Majewski break; 207b819ddbfSLukasz Majewski } 208b819ddbfSLukasz Majewski 20933fac4a6SLukasz Majewski if (f_dfu->poll_timeout) 21033fac4a6SLukasz Majewski if (!(f_dfu->blk_seq_num % 21133fac4a6SLukasz Majewski (dfu_get_buf_size() / DFU_USB_BUFSIZ))) 21233fac4a6SLukasz Majewski dfu_set_poll_timeout(dstat, f_dfu->poll_timeout); 21333fac4a6SLukasz Majewski 214b819ddbfSLukasz Majewski /* send status response */ 215b819ddbfSLukasz Majewski dstat->bStatus = f_dfu->dfu_status; 216b819ddbfSLukasz Majewski dstat->bState = f_dfu->dfu_state; 217b819ddbfSLukasz Majewski dstat->iString = 0; 218b819ddbfSLukasz Majewski } 219b819ddbfSLukasz Majewski 220b819ddbfSLukasz Majewski static void handle_getstate(struct usb_request *req) 221b819ddbfSLukasz Majewski { 222b819ddbfSLukasz Majewski struct f_dfu *f_dfu = req->context; 223b819ddbfSLukasz Majewski 224b819ddbfSLukasz Majewski ((u8 *)req->buf)[0] = f_dfu->dfu_state; 225b819ddbfSLukasz Majewski req->actual = sizeof(u8); 226b819ddbfSLukasz Majewski } 227b819ddbfSLukasz Majewski 228b819ddbfSLukasz Majewski static inline void to_dfu_mode(struct f_dfu *f_dfu) 229b819ddbfSLukasz Majewski { 230b819ddbfSLukasz Majewski f_dfu->usb_function.strings = dfu_strings; 231b819ddbfSLukasz Majewski f_dfu->usb_function.hs_descriptors = f_dfu->function; 2329f3b8ed1SHeiko Schocher f_dfu->usb_function.descriptors = f_dfu->function; 233ad5f9778SHeiko Schocher f_dfu->dfu_state = DFU_STATE_dfuIDLE; 234b819ddbfSLukasz Majewski } 235b819ddbfSLukasz Majewski 236b819ddbfSLukasz Majewski static inline void to_runtime_mode(struct f_dfu *f_dfu) 237b819ddbfSLukasz Majewski { 238b819ddbfSLukasz Majewski f_dfu->usb_function.strings = NULL; 239b819ddbfSLukasz Majewski f_dfu->usb_function.hs_descriptors = dfu_runtime_descs; 2409f3b8ed1SHeiko Schocher f_dfu->usb_function.descriptors = dfu_runtime_descs; 241b819ddbfSLukasz Majewski } 242b819ddbfSLukasz Majewski 243b819ddbfSLukasz Majewski static int handle_upload(struct usb_request *req, u16 len) 244b819ddbfSLukasz Majewski { 245b819ddbfSLukasz Majewski struct f_dfu *f_dfu = req->context; 246b819ddbfSLukasz Majewski 247b819ddbfSLukasz Majewski return dfu_read(dfu_get_entity(f_dfu->altsetting), req->buf, 248b819ddbfSLukasz Majewski req->length, f_dfu->blk_seq_num); 249b819ddbfSLukasz Majewski } 250b819ddbfSLukasz Majewski 251b819ddbfSLukasz Majewski static int handle_dnload(struct usb_gadget *gadget, u16 len) 252b819ddbfSLukasz Majewski { 253b819ddbfSLukasz Majewski struct usb_composite_dev *cdev = get_gadget_data(gadget); 254b819ddbfSLukasz Majewski struct usb_request *req = cdev->req; 255b819ddbfSLukasz Majewski struct f_dfu *f_dfu = req->context; 256b819ddbfSLukasz Majewski 257b819ddbfSLukasz Majewski if (len == 0) 258b819ddbfSLukasz Majewski f_dfu->dfu_state = DFU_STATE_dfuMANIFEST_SYNC; 259b819ddbfSLukasz Majewski 260b819ddbfSLukasz Majewski req->complete = dnload_request_complete; 261b819ddbfSLukasz Majewski 262b819ddbfSLukasz Majewski return len; 263b819ddbfSLukasz Majewski } 264b819ddbfSLukasz Majewski 265b819ddbfSLukasz Majewski /*-------------------------------------------------------------------------*/ 266b819ddbfSLukasz Majewski /* DFU state machine */ 267b819ddbfSLukasz Majewski static int state_app_idle(struct f_dfu *f_dfu, 268b819ddbfSLukasz Majewski const struct usb_ctrlrequest *ctrl, 269b819ddbfSLukasz Majewski struct usb_gadget *gadget, 270b819ddbfSLukasz Majewski struct usb_request *req) 271b819ddbfSLukasz Majewski { 272b819ddbfSLukasz Majewski int value = 0; 273b819ddbfSLukasz Majewski 274b819ddbfSLukasz Majewski switch (ctrl->bRequest) { 275b819ddbfSLukasz Majewski case USB_REQ_DFU_GETSTATUS: 276b819ddbfSLukasz Majewski handle_getstatus(req); 277b819ddbfSLukasz Majewski value = RET_STAT_LEN; 278b819ddbfSLukasz Majewski break; 279b819ddbfSLukasz Majewski case USB_REQ_DFU_GETSTATE: 280b819ddbfSLukasz Majewski handle_getstate(req); 281b819ddbfSLukasz Majewski break; 282b819ddbfSLukasz Majewski case USB_REQ_DFU_DETACH: 283b819ddbfSLukasz Majewski f_dfu->dfu_state = DFU_STATE_appDETACH; 284b819ddbfSLukasz Majewski to_dfu_mode(f_dfu); 285b819ddbfSLukasz Majewski value = RET_ZLP; 286b819ddbfSLukasz Majewski break; 287b819ddbfSLukasz Majewski default: 288b819ddbfSLukasz Majewski value = RET_STALL; 289b819ddbfSLukasz Majewski break; 290b819ddbfSLukasz Majewski } 291b819ddbfSLukasz Majewski 292b819ddbfSLukasz Majewski return value; 293b819ddbfSLukasz Majewski } 294b819ddbfSLukasz Majewski 295b819ddbfSLukasz Majewski static int state_app_detach(struct f_dfu *f_dfu, 296b819ddbfSLukasz Majewski const struct usb_ctrlrequest *ctrl, 297b819ddbfSLukasz Majewski struct usb_gadget *gadget, 298b819ddbfSLukasz Majewski struct usb_request *req) 299b819ddbfSLukasz Majewski { 300b819ddbfSLukasz Majewski int value = 0; 301b819ddbfSLukasz Majewski 302b819ddbfSLukasz Majewski switch (ctrl->bRequest) { 303b819ddbfSLukasz Majewski case USB_REQ_DFU_GETSTATUS: 304b819ddbfSLukasz Majewski handle_getstatus(req); 305b819ddbfSLukasz Majewski value = RET_STAT_LEN; 306b819ddbfSLukasz Majewski break; 307b819ddbfSLukasz Majewski case USB_REQ_DFU_GETSTATE: 308b819ddbfSLukasz Majewski handle_getstate(req); 309b819ddbfSLukasz Majewski break; 310b819ddbfSLukasz Majewski default: 311b819ddbfSLukasz Majewski f_dfu->dfu_state = DFU_STATE_appIDLE; 312b819ddbfSLukasz Majewski value = RET_STALL; 313b819ddbfSLukasz Majewski break; 314b819ddbfSLukasz Majewski } 315b819ddbfSLukasz Majewski 316b819ddbfSLukasz Majewski return value; 317b819ddbfSLukasz Majewski } 318b819ddbfSLukasz Majewski 319b819ddbfSLukasz Majewski static int state_dfu_idle(struct f_dfu *f_dfu, 320b819ddbfSLukasz Majewski const struct usb_ctrlrequest *ctrl, 321b819ddbfSLukasz Majewski struct usb_gadget *gadget, 322b819ddbfSLukasz Majewski struct usb_request *req) 323b819ddbfSLukasz Majewski { 324b819ddbfSLukasz Majewski u16 w_value = le16_to_cpu(ctrl->wValue); 325b819ddbfSLukasz Majewski u16 len = le16_to_cpu(ctrl->wLength); 326b819ddbfSLukasz Majewski int value = 0; 327b819ddbfSLukasz Majewski 328b819ddbfSLukasz Majewski switch (ctrl->bRequest) { 329b819ddbfSLukasz Majewski case USB_REQ_DFU_DNLOAD: 330b819ddbfSLukasz Majewski if (len == 0) { 331b819ddbfSLukasz Majewski f_dfu->dfu_state = DFU_STATE_dfuERROR; 332b819ddbfSLukasz Majewski value = RET_STALL; 333b819ddbfSLukasz Majewski break; 334b819ddbfSLukasz Majewski } 335b819ddbfSLukasz Majewski f_dfu->dfu_state = DFU_STATE_dfuDNLOAD_SYNC; 336b819ddbfSLukasz Majewski f_dfu->blk_seq_num = w_value; 337b819ddbfSLukasz Majewski value = handle_dnload(gadget, len); 338b819ddbfSLukasz Majewski break; 339b819ddbfSLukasz Majewski case USB_REQ_DFU_UPLOAD: 340b819ddbfSLukasz Majewski f_dfu->dfu_state = DFU_STATE_dfuUPLOAD_IDLE; 341b819ddbfSLukasz Majewski f_dfu->blk_seq_num = 0; 342b819ddbfSLukasz Majewski value = handle_upload(req, len); 343b819ddbfSLukasz Majewski break; 344b819ddbfSLukasz Majewski case USB_REQ_DFU_ABORT: 345b819ddbfSLukasz Majewski /* no zlp? */ 346b819ddbfSLukasz Majewski value = RET_ZLP; 347b819ddbfSLukasz Majewski break; 348b819ddbfSLukasz Majewski case USB_REQ_DFU_GETSTATUS: 349b819ddbfSLukasz Majewski handle_getstatus(req); 350b819ddbfSLukasz Majewski value = RET_STAT_LEN; 351b819ddbfSLukasz Majewski break; 352b819ddbfSLukasz Majewski case USB_REQ_DFU_GETSTATE: 353b819ddbfSLukasz Majewski handle_getstate(req); 354b819ddbfSLukasz Majewski break; 355b819ddbfSLukasz Majewski case USB_REQ_DFU_DETACH: 356b819ddbfSLukasz Majewski /* 357b819ddbfSLukasz Majewski * Proprietary extension: 'detach' from idle mode and 358b819ddbfSLukasz Majewski * get back to runtime mode in case of USB Reset. As 359b819ddbfSLukasz Majewski * much as I dislike this, we just can't use every USB 360b819ddbfSLukasz Majewski * bus reset to switch back to runtime mode, since at 361b819ddbfSLukasz Majewski * least the Linux USB stack likes to send a number of 362b819ddbfSLukasz Majewski * resets in a row :( 363b819ddbfSLukasz Majewski */ 364b819ddbfSLukasz Majewski f_dfu->dfu_state = 365b819ddbfSLukasz Majewski DFU_STATE_dfuMANIFEST_WAIT_RST; 366b819ddbfSLukasz Majewski to_runtime_mode(f_dfu); 367b819ddbfSLukasz Majewski f_dfu->dfu_state = DFU_STATE_appIDLE; 3686bed7ce5SLukasz Majewski 369fe1b28c9SRob Herring g_dnl_trigger_detach(); 370b819ddbfSLukasz Majewski break; 371b819ddbfSLukasz Majewski default: 372b819ddbfSLukasz Majewski f_dfu->dfu_state = DFU_STATE_dfuERROR; 373b819ddbfSLukasz Majewski value = RET_STALL; 374b819ddbfSLukasz Majewski break; 375b819ddbfSLukasz Majewski } 376b819ddbfSLukasz Majewski 377b819ddbfSLukasz Majewski return value; 378b819ddbfSLukasz Majewski } 379b819ddbfSLukasz Majewski 380b819ddbfSLukasz Majewski static int state_dfu_dnload_sync(struct f_dfu *f_dfu, 381b819ddbfSLukasz Majewski const struct usb_ctrlrequest *ctrl, 382b819ddbfSLukasz Majewski struct usb_gadget *gadget, 383b819ddbfSLukasz Majewski struct usb_request *req) 384b819ddbfSLukasz Majewski { 385b819ddbfSLukasz Majewski int value = 0; 386b819ddbfSLukasz Majewski 387b819ddbfSLukasz Majewski switch (ctrl->bRequest) { 388b819ddbfSLukasz Majewski case USB_REQ_DFU_GETSTATUS: 389b819ddbfSLukasz Majewski handle_getstatus(req); 390b819ddbfSLukasz Majewski value = RET_STAT_LEN; 391b819ddbfSLukasz Majewski break; 392b819ddbfSLukasz Majewski case USB_REQ_DFU_GETSTATE: 393b819ddbfSLukasz Majewski handle_getstate(req); 394b819ddbfSLukasz Majewski break; 395b819ddbfSLukasz Majewski default: 396b819ddbfSLukasz Majewski f_dfu->dfu_state = DFU_STATE_dfuERROR; 397b819ddbfSLukasz Majewski value = RET_STALL; 398b819ddbfSLukasz Majewski break; 399b819ddbfSLukasz Majewski } 400b819ddbfSLukasz Majewski 401b819ddbfSLukasz Majewski return value; 402b819ddbfSLukasz Majewski } 403b819ddbfSLukasz Majewski 404b819ddbfSLukasz Majewski static int state_dfu_dnbusy(struct f_dfu *f_dfu, 405b819ddbfSLukasz Majewski const struct usb_ctrlrequest *ctrl, 406b819ddbfSLukasz Majewski struct usb_gadget *gadget, 407b819ddbfSLukasz Majewski struct usb_request *req) 408b819ddbfSLukasz Majewski { 409b819ddbfSLukasz Majewski int value = 0; 410b819ddbfSLukasz Majewski 411b819ddbfSLukasz Majewski switch (ctrl->bRequest) { 412b819ddbfSLukasz Majewski case USB_REQ_DFU_GETSTATUS: 413b819ddbfSLukasz Majewski handle_getstatus(req); 414b819ddbfSLukasz Majewski value = RET_STAT_LEN; 415b819ddbfSLukasz Majewski break; 416b819ddbfSLukasz Majewski default: 417b819ddbfSLukasz Majewski f_dfu->dfu_state = DFU_STATE_dfuERROR; 418b819ddbfSLukasz Majewski value = RET_STALL; 419b819ddbfSLukasz Majewski break; 420b819ddbfSLukasz Majewski } 421b819ddbfSLukasz Majewski 422b819ddbfSLukasz Majewski return value; 423b819ddbfSLukasz Majewski } 424b819ddbfSLukasz Majewski 425b819ddbfSLukasz Majewski static int state_dfu_dnload_idle(struct f_dfu *f_dfu, 426b819ddbfSLukasz Majewski const struct usb_ctrlrequest *ctrl, 427b819ddbfSLukasz Majewski struct usb_gadget *gadget, 428b819ddbfSLukasz Majewski struct usb_request *req) 429b819ddbfSLukasz Majewski { 430b819ddbfSLukasz Majewski u16 w_value = le16_to_cpu(ctrl->wValue); 431b819ddbfSLukasz Majewski u16 len = le16_to_cpu(ctrl->wLength); 432b819ddbfSLukasz Majewski int value = 0; 433b819ddbfSLukasz Majewski 434b819ddbfSLukasz Majewski switch (ctrl->bRequest) { 435b819ddbfSLukasz Majewski case USB_REQ_DFU_DNLOAD: 436b819ddbfSLukasz Majewski f_dfu->dfu_state = DFU_STATE_dfuDNLOAD_SYNC; 437b819ddbfSLukasz Majewski f_dfu->blk_seq_num = w_value; 438b819ddbfSLukasz Majewski value = handle_dnload(gadget, len); 439b819ddbfSLukasz Majewski break; 440b819ddbfSLukasz Majewski case USB_REQ_DFU_ABORT: 441b819ddbfSLukasz Majewski f_dfu->dfu_state = DFU_STATE_dfuIDLE; 442b819ddbfSLukasz Majewski value = RET_ZLP; 443b819ddbfSLukasz Majewski break; 444b819ddbfSLukasz Majewski case USB_REQ_DFU_GETSTATUS: 445b819ddbfSLukasz Majewski handle_getstatus(req); 446b819ddbfSLukasz Majewski value = RET_STAT_LEN; 447b819ddbfSLukasz Majewski break; 448b819ddbfSLukasz Majewski case USB_REQ_DFU_GETSTATE: 449b819ddbfSLukasz Majewski handle_getstate(req); 450b819ddbfSLukasz Majewski break; 451b819ddbfSLukasz Majewski default: 452b819ddbfSLukasz Majewski f_dfu->dfu_state = DFU_STATE_dfuERROR; 453b819ddbfSLukasz Majewski value = RET_STALL; 454b819ddbfSLukasz Majewski break; 455b819ddbfSLukasz Majewski } 456b819ddbfSLukasz Majewski 457b819ddbfSLukasz Majewski return value; 458b819ddbfSLukasz Majewski } 459b819ddbfSLukasz Majewski 460b819ddbfSLukasz Majewski static int state_dfu_manifest_sync(struct f_dfu *f_dfu, 461b819ddbfSLukasz Majewski const struct usb_ctrlrequest *ctrl, 462b819ddbfSLukasz Majewski struct usb_gadget *gadget, 463b819ddbfSLukasz Majewski struct usb_request *req) 464b819ddbfSLukasz Majewski { 465b819ddbfSLukasz Majewski int value = 0; 466b819ddbfSLukasz Majewski 467b819ddbfSLukasz Majewski switch (ctrl->bRequest) { 468b819ddbfSLukasz Majewski case USB_REQ_DFU_GETSTATUS: 469b819ddbfSLukasz Majewski /* We're MainfestationTolerant */ 470001a8319SHeiko Schocher f_dfu->dfu_state = DFU_STATE_dfuMANIFEST; 471b819ddbfSLukasz Majewski handle_getstatus(req); 472b819ddbfSLukasz Majewski f_dfu->blk_seq_num = 0; 473b819ddbfSLukasz Majewski value = RET_STAT_LEN; 474001a8319SHeiko Schocher req->complete = dnload_request_flush; 475b819ddbfSLukasz Majewski break; 476b819ddbfSLukasz Majewski case USB_REQ_DFU_GETSTATE: 477b819ddbfSLukasz Majewski handle_getstate(req); 478b819ddbfSLukasz Majewski break; 479b819ddbfSLukasz Majewski default: 480b819ddbfSLukasz Majewski f_dfu->dfu_state = DFU_STATE_dfuERROR; 481b819ddbfSLukasz Majewski value = RET_STALL; 482b819ddbfSLukasz Majewski break; 483b819ddbfSLukasz Majewski } 484b819ddbfSLukasz Majewski 485b819ddbfSLukasz Majewski return value; 486b819ddbfSLukasz Majewski } 487b819ddbfSLukasz Majewski 488001a8319SHeiko Schocher static int state_dfu_manifest(struct f_dfu *f_dfu, 489001a8319SHeiko Schocher const struct usb_ctrlrequest *ctrl, 490001a8319SHeiko Schocher struct usb_gadget *gadget, 491001a8319SHeiko Schocher struct usb_request *req) 492001a8319SHeiko Schocher { 493001a8319SHeiko Schocher int value = 0; 494001a8319SHeiko Schocher 495001a8319SHeiko Schocher switch (ctrl->bRequest) { 496001a8319SHeiko Schocher case USB_REQ_DFU_GETSTATUS: 497001a8319SHeiko Schocher /* We're MainfestationTolerant */ 498001a8319SHeiko Schocher f_dfu->dfu_state = DFU_STATE_dfuIDLE; 499001a8319SHeiko Schocher handle_getstatus(req); 500001a8319SHeiko Schocher f_dfu->blk_seq_num = 0; 501001a8319SHeiko Schocher value = RET_STAT_LEN; 502001a8319SHeiko Schocher puts("DOWNLOAD ... OK\nCtrl+C to exit ...\n"); 503001a8319SHeiko Schocher break; 504001a8319SHeiko Schocher case USB_REQ_DFU_GETSTATE: 505001a8319SHeiko Schocher handle_getstate(req); 506001a8319SHeiko Schocher break; 507001a8319SHeiko Schocher default: 508001a8319SHeiko Schocher f_dfu->dfu_state = DFU_STATE_dfuERROR; 509001a8319SHeiko Schocher value = RET_STALL; 510001a8319SHeiko Schocher break; 511001a8319SHeiko Schocher } 512001a8319SHeiko Schocher return value; 513001a8319SHeiko Schocher } 514001a8319SHeiko Schocher 515b819ddbfSLukasz Majewski static int state_dfu_upload_idle(struct f_dfu *f_dfu, 516b819ddbfSLukasz Majewski const struct usb_ctrlrequest *ctrl, 517b819ddbfSLukasz Majewski struct usb_gadget *gadget, 518b819ddbfSLukasz Majewski struct usb_request *req) 519b819ddbfSLukasz Majewski { 520b819ddbfSLukasz Majewski u16 w_value = le16_to_cpu(ctrl->wValue); 521b819ddbfSLukasz Majewski u16 len = le16_to_cpu(ctrl->wLength); 522b819ddbfSLukasz Majewski int value = 0; 523b819ddbfSLukasz Majewski 524b819ddbfSLukasz Majewski switch (ctrl->bRequest) { 525b819ddbfSLukasz Majewski case USB_REQ_DFU_UPLOAD: 526b819ddbfSLukasz Majewski /* state transition if less data then requested */ 527b819ddbfSLukasz Majewski f_dfu->blk_seq_num = w_value; 528b819ddbfSLukasz Majewski value = handle_upload(req, len); 529b819ddbfSLukasz Majewski if (value >= 0 && value < len) 530b819ddbfSLukasz Majewski f_dfu->dfu_state = DFU_STATE_dfuIDLE; 531b819ddbfSLukasz Majewski break; 532b819ddbfSLukasz Majewski case USB_REQ_DFU_ABORT: 533b819ddbfSLukasz Majewski f_dfu->dfu_state = DFU_STATE_dfuIDLE; 534b819ddbfSLukasz Majewski /* no zlp? */ 535b819ddbfSLukasz Majewski value = RET_ZLP; 536b819ddbfSLukasz Majewski break; 537b819ddbfSLukasz Majewski case USB_REQ_DFU_GETSTATUS: 538b819ddbfSLukasz Majewski handle_getstatus(req); 539b819ddbfSLukasz Majewski value = RET_STAT_LEN; 540b819ddbfSLukasz Majewski break; 541b819ddbfSLukasz Majewski case USB_REQ_DFU_GETSTATE: 542b819ddbfSLukasz Majewski handle_getstate(req); 543b819ddbfSLukasz Majewski break; 544b819ddbfSLukasz Majewski default: 545b819ddbfSLukasz Majewski f_dfu->dfu_state = DFU_STATE_dfuERROR; 546b819ddbfSLukasz Majewski value = RET_STALL; 547b819ddbfSLukasz Majewski break; 548b819ddbfSLukasz Majewski } 549b819ddbfSLukasz Majewski 550b819ddbfSLukasz Majewski return value; 551b819ddbfSLukasz Majewski } 552b819ddbfSLukasz Majewski 553b819ddbfSLukasz Majewski static int state_dfu_error(struct f_dfu *f_dfu, 554b819ddbfSLukasz Majewski const struct usb_ctrlrequest *ctrl, 555b819ddbfSLukasz Majewski struct usb_gadget *gadget, 556b819ddbfSLukasz Majewski struct usb_request *req) 557b819ddbfSLukasz Majewski { 558b819ddbfSLukasz Majewski int value = 0; 559b819ddbfSLukasz Majewski 560b819ddbfSLukasz Majewski switch (ctrl->bRequest) { 561b819ddbfSLukasz Majewski case USB_REQ_DFU_GETSTATUS: 562b819ddbfSLukasz Majewski handle_getstatus(req); 563b819ddbfSLukasz Majewski value = RET_STAT_LEN; 564b819ddbfSLukasz Majewski break; 565b819ddbfSLukasz Majewski case USB_REQ_DFU_GETSTATE: 566b819ddbfSLukasz Majewski handle_getstate(req); 567b819ddbfSLukasz Majewski break; 568b819ddbfSLukasz Majewski case USB_REQ_DFU_CLRSTATUS: 569b819ddbfSLukasz Majewski f_dfu->dfu_state = DFU_STATE_dfuIDLE; 570b819ddbfSLukasz Majewski f_dfu->dfu_status = DFU_STATUS_OK; 571b819ddbfSLukasz Majewski /* no zlp? */ 572b819ddbfSLukasz Majewski value = RET_ZLP; 573b819ddbfSLukasz Majewski break; 574b819ddbfSLukasz Majewski default: 575b819ddbfSLukasz Majewski f_dfu->dfu_state = DFU_STATE_dfuERROR; 576b819ddbfSLukasz Majewski value = RET_STALL; 577b819ddbfSLukasz Majewski break; 578b819ddbfSLukasz Majewski } 579b819ddbfSLukasz Majewski 580b819ddbfSLukasz Majewski return value; 581b819ddbfSLukasz Majewski } 582b819ddbfSLukasz Majewski 583b819ddbfSLukasz Majewski static dfu_state_fn dfu_state[] = { 584b819ddbfSLukasz Majewski state_app_idle, /* DFU_STATE_appIDLE */ 585b819ddbfSLukasz Majewski state_app_detach, /* DFU_STATE_appDETACH */ 586b819ddbfSLukasz Majewski state_dfu_idle, /* DFU_STATE_dfuIDLE */ 587b819ddbfSLukasz Majewski state_dfu_dnload_sync, /* DFU_STATE_dfuDNLOAD_SYNC */ 588b819ddbfSLukasz Majewski state_dfu_dnbusy, /* DFU_STATE_dfuDNBUSY */ 589b819ddbfSLukasz Majewski state_dfu_dnload_idle, /* DFU_STATE_dfuDNLOAD_IDLE */ 590b819ddbfSLukasz Majewski state_dfu_manifest_sync, /* DFU_STATE_dfuMANIFEST_SYNC */ 591001a8319SHeiko Schocher state_dfu_manifest, /* DFU_STATE_dfuMANIFEST */ 592b819ddbfSLukasz Majewski NULL, /* DFU_STATE_dfuMANIFEST_WAIT_RST */ 593b819ddbfSLukasz Majewski state_dfu_upload_idle, /* DFU_STATE_dfuUPLOAD_IDLE */ 594b819ddbfSLukasz Majewski state_dfu_error /* DFU_STATE_dfuERROR */ 595b819ddbfSLukasz Majewski }; 596b819ddbfSLukasz Majewski 597b819ddbfSLukasz Majewski static int 598b819ddbfSLukasz Majewski dfu_handle(struct usb_function *f, const struct usb_ctrlrequest *ctrl) 599b819ddbfSLukasz Majewski { 600b819ddbfSLukasz Majewski struct usb_gadget *gadget = f->config->cdev->gadget; 601b819ddbfSLukasz Majewski struct usb_request *req = f->config->cdev->req; 602b819ddbfSLukasz Majewski struct f_dfu *f_dfu = f->config->cdev->req->context; 603b819ddbfSLukasz Majewski u16 len = le16_to_cpu(ctrl->wLength); 604b819ddbfSLukasz Majewski u16 w_value = le16_to_cpu(ctrl->wValue); 605b819ddbfSLukasz Majewski int value = 0; 606b819ddbfSLukasz Majewski u8 req_type = ctrl->bRequestType & USB_TYPE_MASK; 607b819ddbfSLukasz Majewski 608b819ddbfSLukasz Majewski debug("w_value: 0x%x len: 0x%x\n", w_value, len); 609b819ddbfSLukasz Majewski debug("req_type: 0x%x ctrl->bRequest: 0x%x f_dfu->dfu_state: 0x%x\n", 610b819ddbfSLukasz Majewski req_type, ctrl->bRequest, f_dfu->dfu_state); 611b819ddbfSLukasz Majewski 612b819ddbfSLukasz Majewski if (req_type == USB_TYPE_STANDARD) { 613b819ddbfSLukasz Majewski if (ctrl->bRequest == USB_REQ_GET_DESCRIPTOR && 614b819ddbfSLukasz Majewski (w_value >> 8) == DFU_DT_FUNC) { 615b819ddbfSLukasz Majewski value = min(len, (u16) sizeof(dfu_func)); 616b819ddbfSLukasz Majewski memcpy(req->buf, &dfu_func, value); 617b819ddbfSLukasz Majewski } 618b819ddbfSLukasz Majewski } else /* DFU specific request */ 619b819ddbfSLukasz Majewski value = dfu_state[f_dfu->dfu_state] (f_dfu, ctrl, gadget, req); 620b819ddbfSLukasz Majewski 621b819ddbfSLukasz Majewski if (value >= 0) { 622b819ddbfSLukasz Majewski req->length = value; 623b819ddbfSLukasz Majewski req->zero = value < len; 624b819ddbfSLukasz Majewski value = usb_ep_queue(gadget->ep0, req, 0); 625b819ddbfSLukasz Majewski if (value < 0) { 626b819ddbfSLukasz Majewski debug("ep_queue --> %d\n", value); 627b819ddbfSLukasz Majewski req->status = 0; 628b819ddbfSLukasz Majewski } 629b819ddbfSLukasz Majewski } 630b819ddbfSLukasz Majewski 631b819ddbfSLukasz Majewski return value; 632b819ddbfSLukasz Majewski } 633b819ddbfSLukasz Majewski 634b819ddbfSLukasz Majewski /*-------------------------------------------------------------------------*/ 635b819ddbfSLukasz Majewski 636b819ddbfSLukasz Majewski static int 637b819ddbfSLukasz Majewski dfu_prepare_strings(struct f_dfu *f_dfu, int n) 638b819ddbfSLukasz Majewski { 639b819ddbfSLukasz Majewski struct dfu_entity *de = NULL; 640b819ddbfSLukasz Majewski int i = 0; 641b819ddbfSLukasz Majewski 642b819ddbfSLukasz Majewski f_dfu->strings = calloc(sizeof(struct usb_string), n + 1); 643b819ddbfSLukasz Majewski if (!f_dfu->strings) 644b819ddbfSLukasz Majewski goto enomem; 645b819ddbfSLukasz Majewski 646b819ddbfSLukasz Majewski for (i = 0; i < n; ++i) { 647b819ddbfSLukasz Majewski de = dfu_get_entity(i); 648b819ddbfSLukasz Majewski f_dfu->strings[i].s = de->name; 649b819ddbfSLukasz Majewski } 650b819ddbfSLukasz Majewski 651b819ddbfSLukasz Majewski f_dfu->strings[i].id = 0; 652b819ddbfSLukasz Majewski f_dfu->strings[i].s = NULL; 653b819ddbfSLukasz Majewski 654b819ddbfSLukasz Majewski return 0; 655b819ddbfSLukasz Majewski 656b819ddbfSLukasz Majewski enomem: 657b819ddbfSLukasz Majewski while (i) 658b819ddbfSLukasz Majewski f_dfu->strings[--i].s = NULL; 659b819ddbfSLukasz Majewski 660b819ddbfSLukasz Majewski free(f_dfu->strings); 661b819ddbfSLukasz Majewski 662b819ddbfSLukasz Majewski return -ENOMEM; 663b819ddbfSLukasz Majewski } 664b819ddbfSLukasz Majewski 665b819ddbfSLukasz Majewski static int dfu_prepare_function(struct f_dfu *f_dfu, int n) 666b819ddbfSLukasz Majewski { 667b819ddbfSLukasz Majewski struct usb_interface_descriptor *d; 668b819ddbfSLukasz Majewski int i = 0; 669b819ddbfSLukasz Majewski 670e059a400SLukasz Majewski f_dfu->function = calloc(sizeof(struct usb_descriptor_header *), n + 1); 671b819ddbfSLukasz Majewski if (!f_dfu->function) 672b819ddbfSLukasz Majewski goto enomem; 673b819ddbfSLukasz Majewski 674b819ddbfSLukasz Majewski for (i = 0; i < n; ++i) { 675b819ddbfSLukasz Majewski d = calloc(sizeof(*d), 1); 676b819ddbfSLukasz Majewski if (!d) 677b819ddbfSLukasz Majewski goto enomem; 678b819ddbfSLukasz Majewski 679b819ddbfSLukasz Majewski d->bLength = sizeof(*d); 680b819ddbfSLukasz Majewski d->bDescriptorType = USB_DT_INTERFACE; 681b819ddbfSLukasz Majewski d->bAlternateSetting = i; 682b819ddbfSLukasz Majewski d->bNumEndpoints = 0; 683b819ddbfSLukasz Majewski d->bInterfaceClass = USB_CLASS_APP_SPEC; 684b819ddbfSLukasz Majewski d->bInterfaceSubClass = 1; 685b819ddbfSLukasz Majewski d->bInterfaceProtocol = 2; 686b819ddbfSLukasz Majewski 687b819ddbfSLukasz Majewski f_dfu->function[i] = (struct usb_descriptor_header *)d; 688b819ddbfSLukasz Majewski } 689b819ddbfSLukasz Majewski f_dfu->function[i] = NULL; 690b819ddbfSLukasz Majewski 691b819ddbfSLukasz Majewski return 0; 692b819ddbfSLukasz Majewski 693b819ddbfSLukasz Majewski enomem: 694b819ddbfSLukasz Majewski while (i) { 695b819ddbfSLukasz Majewski free(f_dfu->function[--i]); 696b819ddbfSLukasz Majewski f_dfu->function[i] = NULL; 697b819ddbfSLukasz Majewski } 698b819ddbfSLukasz Majewski free(f_dfu->function); 699b819ddbfSLukasz Majewski 700b819ddbfSLukasz Majewski return -ENOMEM; 701b819ddbfSLukasz Majewski } 702b819ddbfSLukasz Majewski 703b819ddbfSLukasz Majewski static int dfu_bind(struct usb_configuration *c, struct usb_function *f) 704b819ddbfSLukasz Majewski { 705b819ddbfSLukasz Majewski struct usb_composite_dev *cdev = c->cdev; 706b819ddbfSLukasz Majewski struct f_dfu *f_dfu = func_to_dfu(f); 707b819ddbfSLukasz Majewski int alt_num = dfu_get_alt_number(); 708b819ddbfSLukasz Majewski int rv, id, i; 709b819ddbfSLukasz Majewski 710b819ddbfSLukasz Majewski id = usb_interface_id(c, f); 711b819ddbfSLukasz Majewski if (id < 0) 712b819ddbfSLukasz Majewski return id; 713b819ddbfSLukasz Majewski dfu_intf_runtime.bInterfaceNumber = id; 714b819ddbfSLukasz Majewski 715b819ddbfSLukasz Majewski f_dfu->dfu_state = DFU_STATE_appIDLE; 716b819ddbfSLukasz Majewski f_dfu->dfu_status = DFU_STATUS_OK; 717b819ddbfSLukasz Majewski 718b819ddbfSLukasz Majewski rv = dfu_prepare_function(f_dfu, alt_num); 719b819ddbfSLukasz Majewski if (rv) 720b819ddbfSLukasz Majewski goto error; 721b819ddbfSLukasz Majewski 722b819ddbfSLukasz Majewski rv = dfu_prepare_strings(f_dfu, alt_num); 723b819ddbfSLukasz Majewski if (rv) 724b819ddbfSLukasz Majewski goto error; 725b819ddbfSLukasz Majewski for (i = 0; i < alt_num; i++) { 726b819ddbfSLukasz Majewski id = usb_string_id(cdev); 727b819ddbfSLukasz Majewski if (id < 0) 728b819ddbfSLukasz Majewski return id; 729b819ddbfSLukasz Majewski f_dfu->strings[i].id = id; 730b819ddbfSLukasz Majewski ((struct usb_interface_descriptor *)f_dfu->function[i]) 731b819ddbfSLukasz Majewski ->iInterface = id; 732b819ddbfSLukasz Majewski } 733b819ddbfSLukasz Majewski 734ad5f9778SHeiko Schocher to_dfu_mode(f_dfu); 735ad5f9778SHeiko Schocher 736b819ddbfSLukasz Majewski stringtab_dfu.strings = f_dfu->strings; 737b819ddbfSLukasz Majewski 738b819ddbfSLukasz Majewski cdev->req->context = f_dfu; 739b819ddbfSLukasz Majewski 740b819ddbfSLukasz Majewski error: 741b819ddbfSLukasz Majewski return rv; 742b819ddbfSLukasz Majewski } 743b819ddbfSLukasz Majewski 744b819ddbfSLukasz Majewski static void dfu_unbind(struct usb_configuration *c, struct usb_function *f) 745b819ddbfSLukasz Majewski { 746b819ddbfSLukasz Majewski struct f_dfu *f_dfu = func_to_dfu(f); 747b819ddbfSLukasz Majewski int alt_num = dfu_get_alt_number(); 748b819ddbfSLukasz Majewski int i; 749b819ddbfSLukasz Majewski 750b819ddbfSLukasz Majewski if (f_dfu->strings) { 751b819ddbfSLukasz Majewski i = alt_num; 752b819ddbfSLukasz Majewski while (i) 753b819ddbfSLukasz Majewski f_dfu->strings[--i].s = NULL; 754b819ddbfSLukasz Majewski 755b819ddbfSLukasz Majewski free(f_dfu->strings); 756b819ddbfSLukasz Majewski } 757b819ddbfSLukasz Majewski 758b819ddbfSLukasz Majewski if (f_dfu->function) { 759b819ddbfSLukasz Majewski i = alt_num; 760b819ddbfSLukasz Majewski while (i) { 761b819ddbfSLukasz Majewski free(f_dfu->function[--i]); 762b819ddbfSLukasz Majewski f_dfu->function[i] = NULL; 763b819ddbfSLukasz Majewski } 764b819ddbfSLukasz Majewski free(f_dfu->function); 765b819ddbfSLukasz Majewski } 766b819ddbfSLukasz Majewski 767b819ddbfSLukasz Majewski free(f_dfu); 768b819ddbfSLukasz Majewski } 769b819ddbfSLukasz Majewski 770b819ddbfSLukasz Majewski static int dfu_set_alt(struct usb_function *f, unsigned intf, unsigned alt) 771b819ddbfSLukasz Majewski { 772b819ddbfSLukasz Majewski struct f_dfu *f_dfu = func_to_dfu(f); 773b819ddbfSLukasz Majewski 774b819ddbfSLukasz Majewski debug("%s: intf:%d alt:%d\n", __func__, intf, alt); 775b819ddbfSLukasz Majewski 776b819ddbfSLukasz Majewski f_dfu->altsetting = alt; 7773ee9593fSStephen Warren f_dfu->dfu_state = DFU_STATE_dfuIDLE; 7783ee9593fSStephen Warren f_dfu->dfu_status = DFU_STATUS_OK; 779b819ddbfSLukasz Majewski 780b819ddbfSLukasz Majewski return 0; 781b819ddbfSLukasz Majewski } 782b819ddbfSLukasz Majewski 783*1fd81b7cSStefan Roese static int __dfu_get_alt(struct usb_function *f, unsigned intf) 784*1fd81b7cSStefan Roese { 785*1fd81b7cSStefan Roese struct f_dfu *f_dfu = func_to_dfu(f); 786*1fd81b7cSStefan Roese 787*1fd81b7cSStefan Roese return f_dfu->altsetting; 788*1fd81b7cSStefan Roese } 789*1fd81b7cSStefan Roese 790b819ddbfSLukasz Majewski /* TODO: is this really what we need here? */ 791b819ddbfSLukasz Majewski static void dfu_disable(struct usb_function *f) 792b819ddbfSLukasz Majewski { 793b819ddbfSLukasz Majewski struct f_dfu *f_dfu = func_to_dfu(f); 794b819ddbfSLukasz Majewski if (f_dfu->config == 0) 795b819ddbfSLukasz Majewski return; 796b819ddbfSLukasz Majewski 797b819ddbfSLukasz Majewski debug("%s: reset config\n", __func__); 798b819ddbfSLukasz Majewski 799b819ddbfSLukasz Majewski f_dfu->config = 0; 800b819ddbfSLukasz Majewski } 801b819ddbfSLukasz Majewski 802b819ddbfSLukasz Majewski static int dfu_bind_config(struct usb_configuration *c) 803b819ddbfSLukasz Majewski { 804b819ddbfSLukasz Majewski struct f_dfu *f_dfu; 805b819ddbfSLukasz Majewski int status; 806b819ddbfSLukasz Majewski 807b819ddbfSLukasz Majewski f_dfu = calloc(sizeof(*f_dfu), 1); 808b819ddbfSLukasz Majewski if (!f_dfu) 809b819ddbfSLukasz Majewski return -ENOMEM; 810b819ddbfSLukasz Majewski f_dfu->usb_function.name = "dfu"; 811b819ddbfSLukasz Majewski f_dfu->usb_function.hs_descriptors = dfu_runtime_descs; 8129f3b8ed1SHeiko Schocher f_dfu->usb_function.descriptors = dfu_runtime_descs; 813b819ddbfSLukasz Majewski f_dfu->usb_function.bind = dfu_bind; 814b819ddbfSLukasz Majewski f_dfu->usb_function.unbind = dfu_unbind; 815b819ddbfSLukasz Majewski f_dfu->usb_function.set_alt = dfu_set_alt; 816*1fd81b7cSStefan Roese f_dfu->usb_function.get_alt = __dfu_get_alt; 817b819ddbfSLukasz Majewski f_dfu->usb_function.disable = dfu_disable; 81877b95042SLukasz Majewski f_dfu->usb_function.strings = dfu_generic_strings; 81977b95042SLukasz Majewski f_dfu->usb_function.setup = dfu_handle; 82033fac4a6SLukasz Majewski f_dfu->poll_timeout = DFU_DEFAULT_POLL_TIMEOUT; 821b819ddbfSLukasz Majewski 822b819ddbfSLukasz Majewski status = usb_add_function(c, &f_dfu->usb_function); 823b819ddbfSLukasz Majewski if (status) 824b819ddbfSLukasz Majewski free(f_dfu); 825b819ddbfSLukasz Majewski 826b819ddbfSLukasz Majewski return status; 827b819ddbfSLukasz Majewski } 828b819ddbfSLukasz Majewski 829b819ddbfSLukasz Majewski int dfu_add(struct usb_configuration *c) 830b819ddbfSLukasz Majewski { 831b819ddbfSLukasz Majewski int id; 832b819ddbfSLukasz Majewski 833b819ddbfSLukasz Majewski id = usb_string_id(c->cdev); 834b819ddbfSLukasz Majewski if (id < 0) 835b819ddbfSLukasz Majewski return id; 836b819ddbfSLukasz Majewski strings_dfu_generic[0].id = id; 837b819ddbfSLukasz Majewski dfu_intf_runtime.iInterface = id; 838b819ddbfSLukasz Majewski 839b819ddbfSLukasz Majewski debug("%s: cdev: 0x%p gadget:0x%p gadget->ep0: 0x%p\n", __func__, 840b819ddbfSLukasz Majewski c->cdev, c->cdev->gadget, c->cdev->gadget->ep0); 841b819ddbfSLukasz Majewski 842b819ddbfSLukasz Majewski return dfu_bind_config(c); 843b819ddbfSLukasz Majewski } 844c4d0e856SMateusz Zalega 845c4d0e856SMateusz Zalega DECLARE_GADGET_BIND_CALLBACK(usb_dnl_dfu, dfu_add); 846