1 /* 2 * f_thor.c -- USB TIZEN THOR Downloader gadget function 3 * 4 * Copyright (C) 2013 Samsung Electronics 5 * Lukasz Majewski <l.majewski@samsung.com> 6 * 7 * Based on code from: 8 * git://review.tizen.org/kernel/u-boot 9 * 10 * Developed by: 11 * Copyright (C) 2009 Samsung Electronics 12 * Minkyu Kang <mk7.kang@samsung.com> 13 * Sanghee Kim <sh0130.kim@samsung.com> 14 * 15 * SPDX-License-Identifier: GPL-2.0+ 16 */ 17 18 #include <errno.h> 19 #include <common.h> 20 #include <console.h> 21 #include <malloc.h> 22 #include <memalign.h> 23 #include <version.h> 24 #include <linux/usb/ch9.h> 25 #include <linux/usb/gadget.h> 26 #include <linux/usb/composite.h> 27 #include <linux/usb/cdc.h> 28 #include <g_dnl.h> 29 #include <dfu.h> 30 31 #include "f_thor.h" 32 33 static void thor_tx_data(unsigned char *data, int len); 34 static void thor_set_dma(void *addr, int len); 35 static int thor_rx_data(void); 36 37 static struct f_thor *thor_func; 38 static inline struct f_thor *func_to_thor(struct usb_function *f) 39 { 40 return container_of(f, struct f_thor, usb_function); 41 } 42 43 DEFINE_CACHE_ALIGN_BUFFER(unsigned char, thor_tx_data_buf, 44 sizeof(struct rsp_box)); 45 DEFINE_CACHE_ALIGN_BUFFER(unsigned char, thor_rx_data_buf, 46 sizeof(struct rqt_box)); 47 48 /* ********************************************************** */ 49 /* THOR protocol - transmission handling */ 50 /* ********************************************************** */ 51 DEFINE_CACHE_ALIGN_BUFFER(char, f_name, F_NAME_BUF_SIZE + 1); 52 static unsigned long long int thor_file_size; 53 static int alt_setting_num; 54 55 static void send_rsp(const struct rsp_box *rsp) 56 { 57 memcpy(thor_tx_data_buf, rsp, sizeof(struct rsp_box)); 58 thor_tx_data(thor_tx_data_buf, sizeof(struct rsp_box)); 59 60 debug("-RSP: %d, %d\n", rsp->rsp, rsp->rsp_data); 61 } 62 63 static void send_data_rsp(s32 ack, s32 count) 64 { 65 ALLOC_CACHE_ALIGN_BUFFER(struct data_rsp_box, rsp, 66 sizeof(struct data_rsp_box)); 67 68 rsp->ack = ack; 69 rsp->count = count; 70 71 memcpy(thor_tx_data_buf, rsp, sizeof(struct data_rsp_box)); 72 thor_tx_data(thor_tx_data_buf, sizeof(struct data_rsp_box)); 73 74 debug("-DATA RSP: %d, %d\n", ack, count); 75 } 76 77 static int process_rqt_info(const struct rqt_box *rqt) 78 { 79 ALLOC_CACHE_ALIGN_BUFFER(struct rsp_box, rsp, sizeof(struct rsp_box)); 80 memset(rsp, 0, sizeof(struct rsp_box)); 81 82 rsp->rsp = rqt->rqt; 83 rsp->rsp_data = rqt->rqt_data; 84 85 switch (rqt->rqt_data) { 86 case RQT_INFO_VER_PROTOCOL: 87 rsp->int_data[0] = VER_PROTOCOL_MAJOR; 88 rsp->int_data[1] = VER_PROTOCOL_MINOR; 89 break; 90 case RQT_INIT_VER_HW: 91 snprintf(rsp->str_data[0], sizeof(rsp->str_data[0]), 92 "%x", checkboard()); 93 break; 94 case RQT_INIT_VER_BOOT: 95 sprintf(rsp->str_data[0], "%s", U_BOOT_VERSION); 96 break; 97 case RQT_INIT_VER_KERNEL: 98 sprintf(rsp->str_data[0], "%s", "k unknown"); 99 break; 100 case RQT_INIT_VER_PLATFORM: 101 sprintf(rsp->str_data[0], "%s", "p unknown"); 102 break; 103 case RQT_INIT_VER_CSC: 104 sprintf(rsp->str_data[0], "%s", "c unknown"); 105 break; 106 default: 107 return -EINVAL; 108 } 109 110 send_rsp(rsp); 111 return true; 112 } 113 114 static int process_rqt_cmd(const struct rqt_box *rqt) 115 { 116 ALLOC_CACHE_ALIGN_BUFFER(struct rsp_box, rsp, sizeof(struct rsp_box)); 117 memset(rsp, 0, sizeof(struct rsp_box)); 118 119 rsp->rsp = rqt->rqt; 120 rsp->rsp_data = rqt->rqt_data; 121 122 switch (rqt->rqt_data) { 123 case RQT_CMD_REBOOT: 124 debug("TARGET RESET\n"); 125 send_rsp(rsp); 126 g_dnl_unregister(); 127 dfu_free_entities(); 128 #ifdef CONFIG_THOR_RESET_OFF 129 return RESET_DONE; 130 #endif 131 run_command("reset", 0); 132 break; 133 case RQT_CMD_POWEROFF: 134 case RQT_CMD_EFSCLEAR: 135 send_rsp(rsp); 136 default: 137 printf("Command not supported -> cmd: %d\n", rqt->rqt_data); 138 return -EINVAL; 139 } 140 141 return true; 142 } 143 144 static long long int download_head(unsigned long long total, 145 unsigned int packet_size, 146 long long int *left, 147 int *cnt) 148 { 149 long long int rcv_cnt = 0, left_to_rcv, ret_rcv; 150 struct dfu_entity *dfu_entity = dfu_get_entity(alt_setting_num); 151 void *transfer_buffer = dfu_get_buf(dfu_entity); 152 void *buf = transfer_buffer; 153 int usb_pkt_cnt = 0, ret; 154 155 /* 156 * Files smaller than THOR_STORE_UNIT_SIZE (now 32 MiB) are stored on 157 * the medium. 158 * The packet response is sent on the purpose after successful data 159 * chunk write. There is a room for improvement when asynchronous write 160 * is performed. 161 */ 162 while (total - rcv_cnt >= packet_size) { 163 thor_set_dma(buf, packet_size); 164 buf += packet_size; 165 ret_rcv = thor_rx_data(); 166 if (ret_rcv < 0) 167 return ret_rcv; 168 rcv_cnt += ret_rcv; 169 debug("%d: RCV data count: %llu cnt: %d\n", usb_pkt_cnt, 170 rcv_cnt, *cnt); 171 172 if ((rcv_cnt % THOR_STORE_UNIT_SIZE) == 0) { 173 ret = dfu_write(dfu_get_entity(alt_setting_num), 174 transfer_buffer, THOR_STORE_UNIT_SIZE, 175 (*cnt)++); 176 if (ret) { 177 pr_err("DFU write failed [%d] cnt: %d", 178 ret, *cnt); 179 return ret; 180 } 181 buf = transfer_buffer; 182 } 183 send_data_rsp(0, ++usb_pkt_cnt); 184 } 185 186 /* Calculate the amount of data to arrive from PC (in bytes) */ 187 left_to_rcv = total - rcv_cnt; 188 189 /* 190 * Calculate number of data already received. but not yet stored 191 * on the medium (they are smaller than THOR_STORE_UNIT_SIZE) 192 */ 193 *left = left_to_rcv + buf - transfer_buffer; 194 debug("%s: left: %llu left_to_rcv: %llu buf: 0x%p\n", __func__, 195 *left, left_to_rcv, buf); 196 197 if (left_to_rcv) { 198 thor_set_dma(buf, packet_size); 199 ret_rcv = thor_rx_data(); 200 if (ret_rcv < 0) 201 return ret_rcv; 202 rcv_cnt += ret_rcv; 203 send_data_rsp(0, ++usb_pkt_cnt); 204 } 205 206 debug("%s: %llu total: %llu cnt: %d\n", __func__, rcv_cnt, total, *cnt); 207 208 return rcv_cnt; 209 } 210 211 static int download_tail(long long int left, int cnt) 212 { 213 struct dfu_entity *dfu_entity; 214 void *transfer_buffer; 215 int ret; 216 217 debug("%s: left: %llu cnt: %d\n", __func__, left, cnt); 218 219 dfu_entity = dfu_get_entity(alt_setting_num); 220 if (!dfu_entity) { 221 pr_err("Alt setting: %d entity not found!\n", alt_setting_num); 222 return -ENOENT; 223 } 224 225 transfer_buffer = dfu_get_buf(dfu_entity); 226 if (!transfer_buffer) { 227 pr_err("Transfer buffer not allocated!"); 228 return -ENXIO; 229 } 230 231 if (left) { 232 ret = dfu_write(dfu_entity, transfer_buffer, left, cnt++); 233 if (ret) { 234 pr_err("DFU write failed [%d]: left: %llu", ret, left); 235 return ret; 236 } 237 } 238 239 /* 240 * To store last "packet" or write file from buffer to filesystem 241 * DFU storage backend requires dfu_flush 242 * 243 * This also frees memory malloc'ed by dfu_get_buf(), so no explicit 244 * need fo call dfu_free_buf() is needed. 245 */ 246 ret = dfu_flush(dfu_entity, transfer_buffer, 0, cnt); 247 if (ret) 248 pr_err("DFU flush failed!"); 249 250 return ret; 251 } 252 253 static long long int process_rqt_download(const struct rqt_box *rqt) 254 { 255 ALLOC_CACHE_ALIGN_BUFFER(struct rsp_box, rsp, sizeof(struct rsp_box)); 256 static long long int left, ret_head; 257 int file_type, ret = 0; 258 static int cnt; 259 260 memset(rsp, 0, sizeof(struct rsp_box)); 261 rsp->rsp = rqt->rqt; 262 rsp->rsp_data = rqt->rqt_data; 263 264 switch (rqt->rqt_data) { 265 case RQT_DL_INIT: 266 thor_file_size = (unsigned long long int)rqt->int_data[0] + 267 (((unsigned long long int)rqt->int_data[1]) 268 << 32); 269 debug("INIT: total %llu bytes\n", thor_file_size); 270 break; 271 case RQT_DL_FILE_INFO: 272 file_type = rqt->int_data[0]; 273 if (file_type == FILE_TYPE_PIT) { 274 puts("PIT table file - not supported\n"); 275 rsp->ack = -ENOTSUPP; 276 ret = rsp->ack; 277 break; 278 } 279 280 thor_file_size = (unsigned long long int)rqt->int_data[1] + 281 (((unsigned long long int)rqt->int_data[2]) 282 << 32); 283 memcpy(f_name, rqt->str_data[0], F_NAME_BUF_SIZE); 284 f_name[F_NAME_BUF_SIZE] = '\0'; 285 286 debug("INFO: name(%s, %d), size(%llu), type(%d)\n", 287 f_name, 0, thor_file_size, file_type); 288 289 rsp->int_data[0] = THOR_PACKET_SIZE; 290 291 alt_setting_num = dfu_get_alt(f_name); 292 if (alt_setting_num < 0) { 293 pr_err("Alt setting [%d] to write not found!", 294 alt_setting_num); 295 rsp->ack = -ENODEV; 296 ret = rsp->ack; 297 } 298 break; 299 case RQT_DL_FILE_START: 300 send_rsp(rsp); 301 ret_head = download_head(thor_file_size, THOR_PACKET_SIZE, 302 &left, &cnt); 303 if (ret_head < 0) { 304 left = 0; 305 cnt = 0; 306 } 307 return ret_head; 308 case RQT_DL_FILE_END: 309 debug("DL FILE_END\n"); 310 rsp->ack = download_tail(left, cnt); 311 ret = rsp->ack; 312 left = 0; 313 cnt = 0; 314 break; 315 case RQT_DL_EXIT: 316 debug("DL EXIT\n"); 317 break; 318 default: 319 pr_err("Operation not supported: %d", rqt->rqt_data); 320 ret = -ENOTSUPP; 321 } 322 323 send_rsp(rsp); 324 return ret; 325 } 326 327 static int process_data(void) 328 { 329 ALLOC_CACHE_ALIGN_BUFFER(struct rqt_box, rqt, sizeof(struct rqt_box)); 330 int ret = -EINVAL; 331 332 memcpy(rqt, thor_rx_data_buf, sizeof(struct rqt_box)); 333 334 debug("+RQT: %d, %d\n", rqt->rqt, rqt->rqt_data); 335 336 switch (rqt->rqt) { 337 case RQT_INFO: 338 ret = process_rqt_info(rqt); 339 break; 340 case RQT_CMD: 341 ret = process_rqt_cmd(rqt); 342 break; 343 case RQT_DL: 344 ret = (int) process_rqt_download(rqt); 345 break; 346 case RQT_UL: 347 puts("RQT: UPLOAD not supported!\n"); 348 break; 349 default: 350 pr_err("unknown request (%d)", rqt->rqt); 351 } 352 353 return ret; 354 } 355 356 /* ********************************************************** */ 357 /* THOR USB Function */ 358 /* ********************************************************** */ 359 360 static inline struct usb_endpoint_descriptor * 361 ep_desc(struct usb_gadget *g, struct usb_endpoint_descriptor *hs, 362 struct usb_endpoint_descriptor *fs) 363 { 364 if (gadget_is_dualspeed(g) && g->speed == USB_SPEED_HIGH) 365 return hs; 366 return fs; 367 } 368 369 static struct usb_interface_descriptor thor_downloader_intf_data = { 370 .bLength = sizeof(thor_downloader_intf_data), 371 .bDescriptorType = USB_DT_INTERFACE, 372 373 .bNumEndpoints = 2, 374 .bInterfaceClass = USB_CLASS_CDC_DATA, 375 }; 376 377 static struct usb_endpoint_descriptor fs_in_desc = { 378 .bLength = USB_DT_ENDPOINT_SIZE, 379 .bDescriptorType = USB_DT_ENDPOINT, 380 381 .bEndpointAddress = USB_DIR_IN, 382 .bmAttributes = USB_ENDPOINT_XFER_BULK, 383 }; 384 385 static struct usb_endpoint_descriptor fs_out_desc = { 386 .bLength = USB_DT_ENDPOINT_SIZE, 387 .bDescriptorType = USB_DT_ENDPOINT, 388 389 .bEndpointAddress = USB_DIR_OUT, 390 .bmAttributes = USB_ENDPOINT_XFER_BULK, 391 }; 392 393 /* CDC configuration */ 394 static struct usb_interface_descriptor thor_downloader_intf_int = { 395 .bLength = sizeof(thor_downloader_intf_int), 396 .bDescriptorType = USB_DT_INTERFACE, 397 398 .bNumEndpoints = 1, 399 .bInterfaceClass = USB_CLASS_COMM, 400 /* 0x02 Abstract Line Control Model */ 401 .bInterfaceSubClass = USB_CDC_SUBCLASS_ACM, 402 /* 0x01 Common AT commands */ 403 .bInterfaceProtocol = USB_CDC_ACM_PROTO_AT_V25TER, 404 }; 405 406 static struct usb_cdc_header_desc thor_downloader_cdc_header = { 407 .bLength = sizeof(thor_downloader_cdc_header), 408 .bDescriptorType = 0x24, /* CS_INTERFACE */ 409 .bDescriptorSubType = 0x00, 410 .bcdCDC = 0x0110, 411 }; 412 413 static struct usb_cdc_call_mgmt_descriptor thor_downloader_cdc_call = { 414 .bLength = sizeof(thor_downloader_cdc_call), 415 .bDescriptorType = 0x24, /* CS_INTERFACE */ 416 .bDescriptorSubType = 0x01, 417 .bmCapabilities = 0x00, 418 .bDataInterface = 0x01, 419 }; 420 421 static struct usb_cdc_acm_descriptor thor_downloader_cdc_abstract = { 422 .bLength = sizeof(thor_downloader_cdc_abstract), 423 .bDescriptorType = 0x24, /* CS_INTERFACE */ 424 .bDescriptorSubType = 0x02, 425 .bmCapabilities = 0x00, 426 }; 427 428 static struct usb_cdc_union_desc thor_downloader_cdc_union = { 429 .bLength = sizeof(thor_downloader_cdc_union), 430 .bDescriptorType = 0x24, /* CS_INTERFACE */ 431 .bDescriptorSubType = USB_CDC_UNION_TYPE, 432 }; 433 434 static struct usb_endpoint_descriptor fs_int_desc = { 435 .bLength = USB_DT_ENDPOINT_SIZE, 436 .bDescriptorType = USB_DT_ENDPOINT, 437 438 .bEndpointAddress = 3 | USB_DIR_IN, 439 .bmAttributes = USB_ENDPOINT_XFER_INT, 440 .wMaxPacketSize = __constant_cpu_to_le16(16), 441 442 .bInterval = 0x9, 443 }; 444 445 static struct usb_interface_assoc_descriptor 446 thor_iad_descriptor = { 447 .bLength = sizeof(thor_iad_descriptor), 448 .bDescriptorType = USB_DT_INTERFACE_ASSOCIATION, 449 450 .bFirstInterface = 0, 451 .bInterfaceCount = 2, /* control + data */ 452 .bFunctionClass = USB_CLASS_COMM, 453 .bFunctionSubClass = USB_CDC_SUBCLASS_ACM, 454 .bFunctionProtocol = USB_CDC_PROTO_NONE, 455 }; 456 457 static struct usb_endpoint_descriptor hs_in_desc = { 458 .bLength = USB_DT_ENDPOINT_SIZE, 459 .bDescriptorType = USB_DT_ENDPOINT, 460 461 .bmAttributes = USB_ENDPOINT_XFER_BULK, 462 .wMaxPacketSize = __constant_cpu_to_le16(512), 463 }; 464 465 static struct usb_endpoint_descriptor hs_out_desc = { 466 .bLength = USB_DT_ENDPOINT_SIZE, 467 .bDescriptorType = USB_DT_ENDPOINT, 468 469 .bmAttributes = USB_ENDPOINT_XFER_BULK, 470 .wMaxPacketSize = __constant_cpu_to_le16(512), 471 }; 472 473 static struct usb_endpoint_descriptor hs_int_desc = { 474 .bLength = USB_DT_ENDPOINT_SIZE, 475 .bDescriptorType = USB_DT_ENDPOINT, 476 477 .bmAttributes = USB_ENDPOINT_XFER_INT, 478 .wMaxPacketSize = __constant_cpu_to_le16(16), 479 480 .bInterval = 0x9, 481 }; 482 483 /* 484 * This attribute vendor descriptor is necessary for correct operation with 485 * Windows version of THOR download program 486 * 487 * It prevents windows driver from sending zero lenght packet (ZLP) after 488 * each THOR_PACKET_SIZE. This assures consistent behaviour with libusb 489 */ 490 static struct usb_cdc_attribute_vendor_descriptor thor_downloader_cdc_av = { 491 .bLength = sizeof(thor_downloader_cdc_av), 492 .bDescriptorType = 0x24, 493 .bDescriptorSubType = 0x80, 494 .DAUType = 0x0002, 495 .DAULength = 0x0001, 496 .DAUValue = 0x00, 497 }; 498 499 static const struct usb_descriptor_header *hs_thor_downloader_function[] = { 500 (struct usb_descriptor_header *)&thor_iad_descriptor, 501 502 (struct usb_descriptor_header *)&thor_downloader_intf_int, 503 (struct usb_descriptor_header *)&thor_downloader_cdc_header, 504 (struct usb_descriptor_header *)&thor_downloader_cdc_call, 505 (struct usb_descriptor_header *)&thor_downloader_cdc_abstract, 506 (struct usb_descriptor_header *)&thor_downloader_cdc_union, 507 (struct usb_descriptor_header *)&hs_int_desc, 508 509 (struct usb_descriptor_header *)&thor_downloader_intf_data, 510 (struct usb_descriptor_header *)&thor_downloader_cdc_av, 511 (struct usb_descriptor_header *)&hs_in_desc, 512 (struct usb_descriptor_header *)&hs_out_desc, 513 NULL, 514 }; 515 516 /*-------------------------------------------------------------------------*/ 517 static struct usb_request *alloc_ep_req(struct usb_ep *ep, unsigned length) 518 { 519 struct usb_request *req; 520 521 req = usb_ep_alloc_request(ep, 0); 522 if (!req) 523 return req; 524 525 req->length = length; 526 req->buf = memalign(CONFIG_SYS_CACHELINE_SIZE, length); 527 if (!req->buf) { 528 usb_ep_free_request(ep, req); 529 req = NULL; 530 } 531 532 return req; 533 } 534 535 static int thor_rx_data(void) 536 { 537 struct thor_dev *dev = thor_func->dev; 538 int data_to_rx, tmp, status; 539 540 data_to_rx = dev->out_req->length; 541 tmp = data_to_rx; 542 do { 543 dev->out_req->length = data_to_rx; 544 debug("dev->out_req->length:%d dev->rxdata:%d\n", 545 dev->out_req->length, dev->rxdata); 546 547 status = usb_ep_queue(dev->out_ep, dev->out_req, 0); 548 if (status) { 549 pr_err("kill %s: resubmit %d bytes --> %d", 550 dev->out_ep->name, dev->out_req->length, status); 551 usb_ep_set_halt(dev->out_ep); 552 return -EAGAIN; 553 } 554 555 while (!dev->rxdata) { 556 usb_gadget_handle_interrupts(0); 557 if (ctrlc()) 558 return -1; 559 } 560 dev->rxdata = 0; 561 data_to_rx -= dev->out_req->actual; 562 } while (data_to_rx); 563 564 return tmp; 565 } 566 567 static void thor_tx_data(unsigned char *data, int len) 568 { 569 struct thor_dev *dev = thor_func->dev; 570 unsigned char *ptr = dev->in_req->buf; 571 int status; 572 573 memset(ptr, 0, len); 574 memcpy(ptr, data, len); 575 576 dev->in_req->length = len; 577 578 debug("%s: dev->in_req->length:%d to_cpy:%zd\n", __func__, 579 dev->in_req->length, sizeof(data)); 580 581 status = usb_ep_queue(dev->in_ep, dev->in_req, 0); 582 if (status) { 583 pr_err("kill %s: resubmit %d bytes --> %d", 584 dev->in_ep->name, dev->in_req->length, status); 585 usb_ep_set_halt(dev->in_ep); 586 } 587 588 /* Wait until tx interrupt received */ 589 while (!dev->txdata) 590 usb_gadget_handle_interrupts(0); 591 592 dev->txdata = 0; 593 } 594 595 static void thor_rx_tx_complete(struct usb_ep *ep, struct usb_request *req) 596 { 597 struct thor_dev *dev = thor_func->dev; 598 int status = req->status; 599 600 debug("%s: ep_ptr:%p, req_ptr:%p\n", __func__, ep, req); 601 switch (status) { 602 case 0: 603 if (ep == dev->out_ep) 604 dev->rxdata = 1; 605 else 606 dev->txdata = 1; 607 608 break; 609 610 /* this endpoint is normally active while we're configured */ 611 case -ECONNABORTED: /* hardware forced ep reset */ 612 case -ECONNRESET: /* request dequeued */ 613 case -ESHUTDOWN: /* disconnect from host */ 614 case -EREMOTEIO: /* short read */ 615 case -EOVERFLOW: 616 pr_err("ERROR:%d", status); 617 break; 618 } 619 620 debug("%s complete --> %d, %d/%d\n", ep->name, 621 status, req->actual, req->length); 622 } 623 624 static void thor_setup_complete(struct usb_ep *ep, struct usb_request *req) 625 { 626 if (req->status || req->actual != req->length) 627 debug("setup complete --> %d, %d/%d\n", 628 req->status, req->actual, req->length); 629 } 630 631 static int 632 thor_func_setup(struct usb_function *f, const struct usb_ctrlrequest *ctrl) 633 { 634 struct thor_dev *dev = thor_func->dev; 635 struct usb_request *req = dev->req; 636 struct usb_gadget *gadget = dev->gadget; 637 int value = 0; 638 639 u16 len = le16_to_cpu(ctrl->wLength); 640 641 debug("Req_Type: 0x%x Req: 0x%x wValue: 0x%x wIndex: 0x%x wLen: 0x%x\n", 642 ctrl->bRequestType, ctrl->bRequest, ctrl->wValue, ctrl->wIndex, 643 ctrl->wLength); 644 645 switch (ctrl->bRequest) { 646 case USB_CDC_REQ_SET_CONTROL_LINE_STATE: 647 value = 0; 648 break; 649 case USB_CDC_REQ_SET_LINE_CODING: 650 value = len; 651 /* Line Coding set done = configuration done */ 652 thor_func->dev->configuration_done = 1; 653 break; 654 655 default: 656 pr_err("thor_setup: unknown request: %d", ctrl->bRequest); 657 } 658 659 if (value >= 0) { 660 req->length = value; 661 req->zero = value < len; 662 value = usb_ep_queue(gadget->ep0, req, 0); 663 if (value < 0) { 664 debug("%s: ep_queue: %d\n", __func__, value); 665 req->status = 0; 666 } 667 } 668 669 return value; 670 } 671 672 /* Specific to the THOR protocol */ 673 static void thor_set_dma(void *addr, int len) 674 { 675 struct thor_dev *dev = thor_func->dev; 676 677 debug("in_req:%p, out_req:%p\n", dev->in_req, dev->out_req); 678 debug("addr:%p, len:%d\n", addr, len); 679 680 dev->out_req->buf = addr; 681 dev->out_req->length = len; 682 } 683 684 int thor_init(void) 685 { 686 struct thor_dev *dev = thor_func->dev; 687 688 /* Wait for a device enumeration and configuration settings */ 689 debug("THOR enumeration/configuration setting....\n"); 690 while (!dev->configuration_done) 691 usb_gadget_handle_interrupts(0); 692 693 thor_set_dma(thor_rx_data_buf, strlen("THOR")); 694 /* detect the download request from Host PC */ 695 if (thor_rx_data() < 0) { 696 printf("%s: Data not received!\n", __func__); 697 return -1; 698 } 699 700 if (!strncmp((char *)thor_rx_data_buf, "THOR", strlen("THOR"))) { 701 puts("Download request from the Host PC\n"); 702 udelay(30 * 1000); /* 30 ms */ 703 704 strcpy((char *)thor_tx_data_buf, "ROHT"); 705 thor_tx_data(thor_tx_data_buf, strlen("ROHT")); 706 } else { 707 puts("Wrong reply information\n"); 708 return -1; 709 } 710 711 return 0; 712 } 713 714 int thor_handle(void) 715 { 716 int ret; 717 718 /* receive the data from Host PC */ 719 while (1) { 720 thor_set_dma(thor_rx_data_buf, sizeof(struct rqt_box)); 721 ret = thor_rx_data(); 722 723 if (ret > 0) { 724 ret = process_data(); 725 #ifdef CONFIG_THOR_RESET_OFF 726 if (ret == RESET_DONE) 727 break; 728 #endif 729 if (ret < 0) 730 return ret; 731 } else { 732 printf("%s: No data received!\n", __func__); 733 break; 734 } 735 } 736 737 return 0; 738 } 739 740 static void free_ep_req(struct usb_ep *ep, struct usb_request *req) 741 { 742 if (req->buf) 743 free(req->buf); 744 usb_ep_free_request(ep, req); 745 } 746 747 static int thor_func_bind(struct usb_configuration *c, struct usb_function *f) 748 { 749 struct usb_gadget *gadget = c->cdev->gadget; 750 struct f_thor *f_thor = func_to_thor(f); 751 struct thor_dev *dev; 752 struct usb_ep *ep; 753 int status; 754 755 thor_func = f_thor; 756 dev = memalign(CONFIG_SYS_CACHELINE_SIZE, sizeof(*dev)); 757 if (!dev) 758 return -ENOMEM; 759 760 memset(dev, 0, sizeof(*dev)); 761 dev->gadget = gadget; 762 f_thor->dev = dev; 763 764 debug("%s: usb_configuration: 0x%p usb_function: 0x%p\n", 765 __func__, c, f); 766 debug("f_thor: 0x%p thor: 0x%p\n", f_thor, dev); 767 768 /* EP0 */ 769 /* preallocate control response and buffer */ 770 dev->req = usb_ep_alloc_request(gadget->ep0, 0); 771 if (!dev->req) { 772 status = -ENOMEM; 773 goto fail; 774 } 775 dev->req->buf = memalign(CONFIG_SYS_CACHELINE_SIZE, 776 THOR_PACKET_SIZE); 777 if (!dev->req->buf) { 778 status = -ENOMEM; 779 goto fail; 780 } 781 782 dev->req->complete = thor_setup_complete; 783 784 /* DYNAMIC interface numbers assignments */ 785 status = usb_interface_id(c, f); 786 787 if (status < 0) 788 goto fail; 789 790 thor_downloader_intf_int.bInterfaceNumber = status; 791 thor_downloader_cdc_union.bMasterInterface0 = status; 792 793 status = usb_interface_id(c, f); 794 795 if (status < 0) 796 goto fail; 797 798 thor_downloader_intf_data.bInterfaceNumber = status; 799 thor_downloader_cdc_union.bSlaveInterface0 = status; 800 801 /* allocate instance-specific endpoints */ 802 ep = usb_ep_autoconfig(gadget, &fs_in_desc); 803 if (!ep) { 804 status = -ENODEV; 805 goto fail; 806 } 807 808 if (gadget_is_dualspeed(gadget)) { 809 hs_in_desc.bEndpointAddress = 810 fs_in_desc.bEndpointAddress; 811 } 812 813 dev->in_ep = ep; /* Store IN EP for enabling @ setup */ 814 ep->driver_data = dev; 815 816 ep = usb_ep_autoconfig(gadget, &fs_out_desc); 817 if (!ep) { 818 status = -ENODEV; 819 goto fail; 820 } 821 822 if (gadget_is_dualspeed(gadget)) 823 hs_out_desc.bEndpointAddress = 824 fs_out_desc.bEndpointAddress; 825 826 dev->out_ep = ep; /* Store OUT EP for enabling @ setup */ 827 ep->driver_data = dev; 828 829 ep = usb_ep_autoconfig(gadget, &fs_int_desc); 830 if (!ep) { 831 status = -ENODEV; 832 goto fail; 833 } 834 835 dev->int_ep = ep; 836 ep->driver_data = dev; 837 838 if (gadget_is_dualspeed(gadget)) { 839 hs_int_desc.bEndpointAddress = 840 fs_int_desc.bEndpointAddress; 841 842 f->hs_descriptors = (struct usb_descriptor_header **) 843 &hs_thor_downloader_function; 844 845 if (!f->hs_descriptors) 846 goto fail; 847 } 848 849 debug("%s: out_ep:%p out_req:%p\n", __func__, 850 dev->out_ep, dev->out_req); 851 852 return 0; 853 854 fail: 855 if (dev->req) 856 free_ep_req(gadget->ep0, dev->req); 857 free(dev); 858 return status; 859 } 860 861 static void thor_unbind(struct usb_configuration *c, struct usb_function *f) 862 { 863 struct f_thor *f_thor = func_to_thor(f); 864 struct thor_dev *dev = f_thor->dev; 865 866 free_ep_req(dev->gadget->ep0, dev->req); 867 free(dev); 868 memset(thor_func, 0, sizeof(*thor_func)); 869 thor_func = NULL; 870 } 871 872 static void thor_func_disable(struct usb_function *f) 873 { 874 struct f_thor *f_thor = func_to_thor(f); 875 struct thor_dev *dev = f_thor->dev; 876 877 debug("%s:\n", __func__); 878 879 /* Avoid freeing memory when ep is still claimed */ 880 if (dev->in_ep->driver_data) { 881 usb_ep_disable(dev->in_ep); 882 free_ep_req(dev->in_ep, dev->in_req); 883 dev->in_ep->driver_data = NULL; 884 } 885 886 if (dev->out_ep->driver_data) { 887 usb_ep_disable(dev->out_ep); 888 usb_ep_free_request(dev->out_ep, dev->out_req); 889 dev->out_ep->driver_data = NULL; 890 } 891 892 if (dev->int_ep->driver_data) { 893 usb_ep_disable(dev->int_ep); 894 dev->int_ep->driver_data = NULL; 895 } 896 } 897 898 static int thor_eps_setup(struct usb_function *f) 899 { 900 struct usb_composite_dev *cdev = f->config->cdev; 901 struct usb_gadget *gadget = cdev->gadget; 902 struct thor_dev *dev = thor_func->dev; 903 struct usb_endpoint_descriptor *d; 904 struct usb_request *req; 905 struct usb_ep *ep; 906 int result; 907 908 ep = dev->in_ep; 909 d = ep_desc(gadget, &hs_in_desc, &fs_in_desc); 910 debug("(d)bEndpointAddress: 0x%x\n", d->bEndpointAddress); 911 912 result = usb_ep_enable(ep, d); 913 if (result) 914 goto err; 915 916 ep->driver_data = cdev; /* claim */ 917 req = alloc_ep_req(ep, THOR_PACKET_SIZE); 918 if (!req) { 919 result = -EIO; 920 goto err_disable_in_ep; 921 } 922 923 memset(req->buf, 0, req->length); 924 req->complete = thor_rx_tx_complete; 925 dev->in_req = req; 926 ep = dev->out_ep; 927 d = ep_desc(gadget, &hs_out_desc, &fs_out_desc); 928 debug("(d)bEndpointAddress: 0x%x\n", d->bEndpointAddress); 929 930 result = usb_ep_enable(ep, d); 931 if (result) 932 goto err_free_in_req; 933 934 ep->driver_data = cdev; /* claim */ 935 req = usb_ep_alloc_request(ep, 0); 936 if (!req) { 937 result = -EIO; 938 goto err_disable_out_ep; 939 } 940 941 req->complete = thor_rx_tx_complete; 942 dev->out_req = req; 943 /* ACM control EP */ 944 ep = dev->int_ep; 945 d = ep_desc(gadget, &hs_int_desc, &fs_int_desc); 946 debug("(d)bEndpointAddress: 0x%x\n", d->bEndpointAddress); 947 948 result = usb_ep_enable(ep, d); 949 if (result) 950 goto err; 951 952 ep->driver_data = cdev; /* claim */ 953 954 return 0; 955 956 err_disable_out_ep: 957 usb_ep_disable(dev->out_ep); 958 959 err_free_in_req: 960 free_ep_req(dev->in_ep, dev->in_req); 961 dev->in_req = NULL; 962 963 err_disable_in_ep: 964 usb_ep_disable(dev->in_ep); 965 966 err: 967 return result; 968 } 969 970 static int thor_func_set_alt(struct usb_function *f, 971 unsigned intf, unsigned alt) 972 { 973 struct thor_dev *dev = thor_func->dev; 974 int result; 975 976 debug("%s: func: %s intf: %d alt: %d\n", 977 __func__, f->name, intf, alt); 978 979 switch (intf) { 980 case 0: 981 debug("ACM INTR interface\n"); 982 break; 983 case 1: 984 debug("Communication Data interface\n"); 985 result = thor_eps_setup(f); 986 if (result) 987 pr_err("%s: EPs setup failed!", __func__); 988 dev->configuration_done = 1; 989 break; 990 } 991 992 return 0; 993 } 994 995 static int thor_func_init(struct usb_configuration *c) 996 { 997 struct f_thor *f_thor; 998 int status; 999 1000 debug("%s: cdev: 0x%p\n", __func__, c->cdev); 1001 1002 f_thor = memalign(CONFIG_SYS_CACHELINE_SIZE, sizeof(*f_thor)); 1003 if (!f_thor) 1004 return -ENOMEM; 1005 1006 memset(f_thor, 0, sizeof(*f_thor)); 1007 1008 f_thor->usb_function.name = "f_thor"; 1009 f_thor->usb_function.bind = thor_func_bind; 1010 f_thor->usb_function.unbind = thor_unbind; 1011 f_thor->usb_function.setup = thor_func_setup; 1012 f_thor->usb_function.set_alt = thor_func_set_alt; 1013 f_thor->usb_function.disable = thor_func_disable; 1014 1015 status = usb_add_function(c, &f_thor->usb_function); 1016 if (status) 1017 free(f_thor); 1018 1019 return status; 1020 } 1021 1022 int thor_add(struct usb_configuration *c) 1023 { 1024 debug("%s:\n", __func__); 1025 return thor_func_init(c); 1026 } 1027 1028 DECLARE_GADGET_BIND_CALLBACK(usb_dnl_thor, thor_add); 1029