drivers/usb/gadget/f_thor.c

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