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