drivers/greybus/es2.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 /*
   3  * Greybus "AP" USB driver for "ES2" controller chips
   4  *
   5  * Copyright 2014-2015 Google Inc.
   6  * Copyright 2014-2015 Linaro Ltd.
   7  */
   8 #include <linux/kthread.h>
   9 #include <linux/sizes.h>
  10 #include <linux/usb.h>
  11 #include <linux/kfifo.h>
  12 #include <linux/debugfs.h>
  13 #include <linux/list.h>
  14 #include <linux/greybus.h>
  15 #include <asm/unaligned.h>
  16
  17 #include "arpc.h"
  18 #include "greybus_trace.h"
  19
  20
  21 /* Default timeout for USB vendor requests. */
  22 #define ES2_USB_CTRL_TIMEOUT    500
  23
  24 /* Default timeout for ARPC CPort requests */
  25 #define ES2_ARPC_CPORT_TIMEOUT  500
  26
  27 /* Fixed CPort numbers */
  28 #define ES2_CPORT_CDSI0         16
  29 #define ES2_CPORT_CDSI1         17
  30
  31 /* Memory sizes for the buffers sent to/from the ES2 controller */
  32 #define ES2_GBUF_MSG_SIZE_MAX   2048
  33
  34 /* Memory sizes for the ARPC buffers */
  35 #define ARPC_OUT_SIZE_MAX       U16_MAX
  36 #define ARPC_IN_SIZE_MAX        128
  37
  38 static const struct usb_device_id id_table[] = {
  39         { USB_DEVICE(0x18d1, 0x1eaf) },
  40         { },
  41 };
  42 MODULE_DEVICE_TABLE(usb, id_table);
  43
  44 #define APB1_LOG_SIZE           SZ_16K
  45
  46 /*
  47  * Number of CPort IN urbs in flight at any point in time.
  48  * Adjust if we are having stalls in the USB buffer due to not enough urbs in
  49  * flight.
  50  */
  51 #define NUM_CPORT_IN_URB        4
  52
  53 /* Number of CPort OUT urbs in flight at any point in time.
  54  * Adjust if we get messages saying we are out of urbs in the system log.
  55  */
  56 #define NUM_CPORT_OUT_URB       8
  57
  58 /*
  59  * Number of ARPC in urbs in flight at any point in time.
  60  */
  61 #define NUM_ARPC_IN_URB         2
  62
  63 /*
  64  * @endpoint: bulk in endpoint for CPort data
  65  * @urb: array of urbs for the CPort in messages
  66  * @buffer: array of buffers for the @cport_in_urb urbs
  67  */
  68 struct es2_cport_in {
  69         __u8 endpoint;
  70         struct urb *urb[NUM_CPORT_IN_URB];
  71         u8 *buffer[NUM_CPORT_IN_URB];
  72 };
  73
  74 /**
  75  * es2_ap_dev - ES2 USB Bridge to AP structure
  76  * @usb_dev: pointer to the USB device we are.
  77  * @usb_intf: pointer to the USB interface we are bound to.
  78  * @hd: pointer to our gb_host_device structure
  79
  80  * @cport_in: endpoint, urbs and buffer for cport in messages
  81  * @cport_out_endpoint: endpoint for for cport out messages
  82  * @cport_out_urb: array of urbs for the CPort out messages
  83  * @cport_out_urb_busy: array of flags to see if the @cport_out_urb is busy or
  84  *                      not.
  85  * @cport_out_urb_cancelled: array of flags indicating whether the
  86  *                      corresponding @cport_out_urb is being cancelled
  87  * @cport_out_urb_lock: locks the @cport_out_urb_busy "list"
  88  *
  89  * @apb_log_task: task pointer for logging thread
  90  * @apb_log_dentry: file system entry for the log file interface
  91  * @apb_log_enable_dentry: file system entry for enabling logging
  92  * @apb_log_fifo: kernel FIFO to carry logged data
  93  * @arpc_urb: array of urbs for the ARPC in messages
  94  * @arpc_buffer: array of buffers for the @arpc_urb urbs
  95  * @arpc_endpoint_in: bulk in endpoint for APBridgeA RPC
  96  * @arpc_id_cycle: gives an unique id to ARPC
  97  * @arpc_lock: locks ARPC list
  98  * @arpcs: list of in progress ARPCs
  99  */
 100 struct es2_ap_dev {
 101         struct usb_device *usb_dev;
 102         struct usb_interface *usb_intf;
 103         struct gb_host_device *hd;
 104
 105         struct es2_cport_in cport_in;
 106         __u8 cport_out_endpoint;
 107         struct urb *cport_out_urb[NUM_CPORT_OUT_URB];
 108         bool cport_out_urb_busy[NUM_CPORT_OUT_URB];
 109         bool cport_out_urb_cancelled[NUM_CPORT_OUT_URB];
 110         spinlock_t cport_out_urb_lock;
 111
 112         bool cdsi1_in_use;
 113
 114         struct task_struct *apb_log_task;
 115         struct dentry *apb_log_dentry;
 116         struct dentry *apb_log_enable_dentry;
 117         DECLARE_KFIFO(apb_log_fifo, char, APB1_LOG_SIZE);
 118
 119         __u8 arpc_endpoint_in;
 120         struct urb *arpc_urb[NUM_ARPC_IN_URB];
 121         u8 *arpc_buffer[NUM_ARPC_IN_URB];
 122
 123         int arpc_id_cycle;
 124         spinlock_t arpc_lock;
 125         struct list_head arpcs;
 126 };
 127
 128 struct arpc {
 129         struct list_head list;
 130         struct arpc_request_message *req;
 131         struct arpc_response_message *resp;
 132         struct completion response_received;
 133         bool active;
 134 };
 135
 136 static inline struct es2_ap_dev *hd_to_es2(struct gb_host_device *hd)
 137 {
 138         return (struct es2_ap_dev *)&hd->hd_priv;
 139 }
 140
 141 static void cport_out_callback(struct urb *urb);
 142 static void usb_log_enable(struct es2_ap_dev *es2);
 143 static void usb_log_disable(struct es2_ap_dev *es2);
 144 static int arpc_sync(struct es2_ap_dev *es2, u8 type, void *payload,
 145                      size_t size, int *result, unsigned int timeout);
 146
 147 static int output_sync(struct es2_ap_dev *es2, void *req, u16 size, u8 cmd)
 148 {
 149         struct usb_device *udev = es2->usb_dev;
 150         u8 *data;
 151         int retval;
 152
 153         data = kmemdup(req, size, GFP_KERNEL);
 154         if (!data)
 155                 return -ENOMEM;
 156
 157         retval = usb_control_msg(udev, usb_sndctrlpipe(udev, 0),
 158                                  cmd,
 159                                  USB_DIR_OUT | USB_TYPE_VENDOR |
 160                                  USB_RECIP_INTERFACE,
 161                                  0, 0, data, size, ES2_USB_CTRL_TIMEOUT);
 162         if (retval < 0)
 163                 dev_err(&udev->dev, "%s: return error %d\n", __func__, retval);
 164         else
 165                 retval = 0;
 166
 167         kfree(data);
 168         return retval;
 169 }
 170
 171 static void ap_urb_complete(struct urb *urb)
 172 {
 173         struct usb_ctrlrequest *dr = urb->context;
 174
 175         kfree(dr);
 176         usb_free_urb(urb);
 177 }
 178
 179 static int output_async(struct es2_ap_dev *es2, void *req, u16 size, u8 cmd)
 180 {
 181         struct usb_device *udev = es2->usb_dev;
 182         struct urb *urb;
 183         struct usb_ctrlrequest *dr;
 184         u8 *buf;
 185         int retval;
 186
 187         urb = usb_alloc_urb(0, GFP_ATOMIC);
 188         if (!urb)
 189                 return -ENOMEM;
 190
 191         dr = kmalloc(sizeof(*dr) + size, GFP_ATOMIC);
 192         if (!dr) {
 193                 usb_free_urb(urb);
 194                 return -ENOMEM;
 195         }
 196
 197         buf = (u8 *)dr + sizeof(*dr);
 198         memcpy(buf, req, size);
 199
 200         dr->bRequest = cmd;
 201         dr->bRequestType = USB_DIR_OUT | USB_TYPE_VENDOR | USB_RECIP_INTERFACE;
 202         dr->wValue = 0;
 203         dr->wIndex = 0;
 204         dr->wLength = cpu_to_le16(size);
 205
 206         usb_fill_control_urb(urb, udev, usb_sndctrlpipe(udev, 0),
 207                              (unsigned char *)dr, buf, size,
 208                              ap_urb_complete, dr);
 209         retval = usb_submit_urb(urb, GFP_ATOMIC);
 210         if (retval) {
 211                 usb_free_urb(urb);
 212                 kfree(dr);
 213         }
 214         return retval;
 215 }
 216
 217 static int output(struct gb_host_device *hd, void *req, u16 size, u8 cmd,
 218                   bool async)
 219 {
 220         struct es2_ap_dev *es2 = hd_to_es2(hd);
 221
 222         if (async)
 223                 return output_async(es2, req, size, cmd);
 224
 225         return output_sync(es2, req, size, cmd);
 226 }
 227
 228 static int es2_cport_in_enable(struct es2_ap_dev *es2,
 229                                struct es2_cport_in *cport_in)
 230 {
 231         struct urb *urb;
 232         int ret;
 233         int i;
 234
 235         for (i = 0; i < NUM_CPORT_IN_URB; ++i) {
 236                 urb = cport_in->urb[i];
 237
 238                 ret = usb_submit_urb(urb, GFP_KERNEL);
 239                 if (ret) {
 240                         dev_err(&es2->usb_dev->dev,
 241                                 "failed to submit in-urb: %d\n", ret);
 242                         goto err_kill_urbs;
 243                 }
 244         }
 245
 246         return 0;
 247
 248 err_kill_urbs:
 249         for (--i; i >= 0; --i) {
 250                 urb = cport_in->urb[i];
 251                 usb_kill_urb(urb);
 252         }
 253
 254         return ret;
 255 }
 256
 257 static void es2_cport_in_disable(struct es2_ap_dev *es2,
 258                                  struct es2_cport_in *cport_in)
 259 {
 260         struct urb *urb;
 261         int i;
 262
 263         for (i = 0; i < NUM_CPORT_IN_URB; ++i) {
 264                 urb = cport_in->urb[i];
 265                 usb_kill_urb(urb);
 266         }
 267 }
 268
 269 static int es2_arpc_in_enable(struct es2_ap_dev *es2)
 270 {
 271         struct urb *urb;
 272         int ret;
 273         int i;
 274
 275         for (i = 0; i < NUM_ARPC_IN_URB; ++i) {
 276                 urb = es2->arpc_urb[i];
 277
 278                 ret = usb_submit_urb(urb, GFP_KERNEL);
 279                 if (ret) {
 280                         dev_err(&es2->usb_dev->dev,
 281                                 "failed to submit arpc in-urb: %d\n", ret);
 282                         goto err_kill_urbs;
 283                 }
 284         }
 285
 286         return 0;
 287
 288 err_kill_urbs:
 289         for (--i; i >= 0; --i) {
 290                 urb = es2->arpc_urb[i];
 291                 usb_kill_urb(urb);
 292         }
 293
 294         return ret;
 295 }
 296
 297 static void es2_arpc_in_disable(struct es2_ap_dev *es2)
 298 {
 299         struct urb *urb;
 300         int i;
 301
 302         for (i = 0; i < NUM_ARPC_IN_URB; ++i) {
 303                 urb = es2->arpc_urb[i];
 304                 usb_kill_urb(urb);
 305         }
 306 }
 307
 308 static struct urb *next_free_urb(struct es2_ap_dev *es2, gfp_t gfp_mask)
 309 {
 310         struct urb *urb = NULL;
 311         unsigned long flags;
 312         int i;
 313
 314         spin_lock_irqsave(&es2->cport_out_urb_lock, flags);
 315
 316         /* Look in our pool of allocated urbs first, as that's the "fastest" */
 317         for (i = 0; i < NUM_CPORT_OUT_URB; ++i) {
 318                 if (!es2->cport_out_urb_busy[i] &&
 319                     !es2->cport_out_urb_cancelled[i]) {
 320                         es2->cport_out_urb_busy[i] = true;
 321                         urb = es2->cport_out_urb[i];
 322                         break;
 323                 }
 324         }
 325         spin_unlock_irqrestore(&es2->cport_out_urb_lock, flags);
 326         if (urb)
 327                 return urb;
 328
 329         /*
 330          * Crap, pool is empty, complain to the syslog and go allocate one
 331          * dynamically as we have to succeed.
 332          */
 333         dev_dbg(&es2->usb_dev->dev,
 334                 "No free CPort OUT urbs, having to dynamically allocate one!\n");
 335         return usb_alloc_urb(0, gfp_mask);
 336 }
 337
 338 static void free_urb(struct es2_ap_dev *es2, struct urb *urb)
 339 {
 340         unsigned long flags;
 341         int i;
 342         /*
 343          * See if this was an urb in our pool, if so mark it "free", otherwise
 344          * we need to free it ourselves.
 345          */
 346         spin_lock_irqsave(&es2->cport_out_urb_lock, flags);
 347         for (i = 0; i < NUM_CPORT_OUT_URB; ++i) {
 348                 if (urb == es2->cport_out_urb[i]) {
 349                         es2->cport_out_urb_busy[i] = false;
 350                         urb = NULL;
 351                         break;
 352                 }
 353         }
 354         spin_unlock_irqrestore(&es2->cport_out_urb_lock, flags);
 355
 356         /* If urb is not NULL, then we need to free this urb */
 357         usb_free_urb(urb);
 358 }
 359
 360 /*
 361  * We (ab)use the operation-message header pad bytes to transfer the
 362  * cport id in order to minimise overhead.
 363  */
 364 static void
 365 gb_message_cport_pack(struct gb_operation_msg_hdr *header, u16 cport_id)
 366 {
 367         header->pad[0] = cport_id;
 368 }
 369
 370 /* Clear the pad bytes used for the CPort id */
 371 static void gb_message_cport_clear(struct gb_operation_msg_hdr *header)
 372 {
 373         header->pad[0] = 0;
 374 }
 375
 376 /* Extract the CPort id packed into the header, and clear it */
 377 static u16 gb_message_cport_unpack(struct gb_operation_msg_hdr *header)
 378 {
 379         u16 cport_id = header->pad[0];
 380
 381         gb_message_cport_clear(header);
 382
 383         return cport_id;
 384 }
 385
 386 /*
 387  * Returns zero if the message was successfully queued, or a negative errno
 388  * otherwise.
 389  */
 390 static int message_send(struct gb_host_device *hd, u16 cport_id,
 391                         struct gb_message *message, gfp_t gfp_mask)
 392 {
 393         struct es2_ap_dev *es2 = hd_to_es2(hd);
 394         struct usb_device *udev = es2->usb_dev;
 395         size_t buffer_size;
 396         int retval;
 397         struct urb *urb;
 398         unsigned long flags;
 399
 400         /*
 401          * The data actually transferred will include an indication
 402          * of where the data should be sent.  Do one last check of
 403          * the target CPort id before filling it in.
 404          */
 405         if (!cport_id_valid(hd, cport_id)) {
 406                 dev_err(&udev->dev, "invalid cport %u\n", cport_id);
 407                 return -EINVAL;
 408         }
 409
 410         /* Find a free urb */
 411         urb = next_free_urb(es2, gfp_mask);
 412         if (!urb)
 413                 return -ENOMEM;
 414
 415         spin_lock_irqsave(&es2->cport_out_urb_lock, flags);
 416         message->hcpriv = urb;
 417         spin_unlock_irqrestore(&es2->cport_out_urb_lock, flags);
 418
 419         /* Pack the cport id into the message header */
 420         gb_message_cport_pack(message->header, cport_id);
 421
 422         buffer_size = sizeof(*message->header) + message->payload_size;
 423
 424         usb_fill_bulk_urb(urb, udev,
 425                           usb_sndbulkpipe(udev,
 426                                           es2->cport_out_endpoint),
 427                           message->buffer, buffer_size,
 428                           cport_out_callback, message);
 429         urb->transfer_flags |= URB_ZERO_PACKET;
 430
 431         trace_gb_message_submit(message);
 432
 433         retval = usb_submit_urb(urb, gfp_mask);
 434         if (retval) {
 435                 dev_err(&udev->dev, "failed to submit out-urb: %d\n", retval);
 436
 437                 spin_lock_irqsave(&es2->cport_out_urb_lock, flags);
 438                 message->hcpriv = NULL;
 439                 spin_unlock_irqrestore(&es2->cport_out_urb_lock, flags);
 440
 441                 free_urb(es2, urb);
 442                 gb_message_cport_clear(message->header);
 443
 444                 return retval;
 445         }
 446
 447         return 0;
 448 }
 449
 450 /*
 451  * Can not be called in atomic context.
 452  */
 453 static void message_cancel(struct gb_message *message)
 454 {
 455         struct gb_host_device *hd = message->operation->connection->hd;
 456         struct es2_ap_dev *es2 = hd_to_es2(hd);
 457         struct urb *urb;
 458         int i;
 459
 460         might_sleep();
 461
 462         spin_lock_irq(&es2->cport_out_urb_lock);
 463         urb = message->hcpriv;
 464
 465         /* Prevent dynamically allocated urb from being deallocated. */
 466         usb_get_urb(urb);
 467
 468         /* Prevent pre-allocated urb from being reused. */
 469         for (i = 0; i < NUM_CPORT_OUT_URB; ++i) {
 470                 if (urb == es2->cport_out_urb[i]) {
 471                         es2->cport_out_urb_cancelled[i] = true;
 472                         break;
 473                 }
 474         }
 475         spin_unlock_irq(&es2->cport_out_urb_lock);
 476
 477         usb_kill_urb(urb);
 478
 479         if (i < NUM_CPORT_OUT_URB) {
 480                 spin_lock_irq(&es2->cport_out_urb_lock);
 481                 es2->cport_out_urb_cancelled[i] = false;
 482                 spin_unlock_irq(&es2->cport_out_urb_lock);
 483         }
 484
 485         usb_free_urb(urb);
 486 }
 487
 488 static int es2_cport_allocate(struct gb_host_device *hd, int cport_id,
 489                               unsigned long flags)
 490 {
 491         struct es2_ap_dev *es2 = hd_to_es2(hd);
 492         struct ida *id_map = &hd->cport_id_map;
 493         int ida_start, ida_end;
 494
 495         switch (cport_id) {
 496         case ES2_CPORT_CDSI0:
 497         case ES2_CPORT_CDSI1:
 498                 dev_err(&hd->dev, "cport %d not available\n", cport_id);
 499                 return -EBUSY;
 500         }
 501
 502         if (flags & GB_CONNECTION_FLAG_OFFLOADED &&
 503             flags & GB_CONNECTION_FLAG_CDSI1) {
 504                 if (es2->cdsi1_in_use) {
 505                         dev_err(&hd->dev, "CDSI1 already in use\n");
 506                         return -EBUSY;
 507                 }
 508
 509                 es2->cdsi1_in_use = true;
 510
 511                 return ES2_CPORT_CDSI1;
 512         }
 513
 514         if (cport_id < 0) {
 515                 ida_start = 0;
 516                 ida_end = hd->num_cports;
 517         } else if (cport_id < hd->num_cports) {
 518                 ida_start = cport_id;
 519                 ida_end = cport_id + 1;
 520         } else {
 521                 dev_err(&hd->dev, "cport %d not available\n", cport_id);
 522                 return -EINVAL;
 523         }
 524
 525         return ida_simple_get(id_map, ida_start, ida_end, GFP_KERNEL);
 526 }
 527
 528 static void es2_cport_release(struct gb_host_device *hd, u16 cport_id)
 529 {
 530         struct es2_ap_dev *es2 = hd_to_es2(hd);
 531
 532         switch (cport_id) {
 533         case ES2_CPORT_CDSI1:
 534                 es2->cdsi1_in_use = false;
 535                 return;
 536         }
 537
 538         ida_simple_remove(&hd->cport_id_map, cport_id);
 539 }
 540
 541 static int cport_enable(struct gb_host_device *hd, u16 cport_id,
 542                         unsigned long flags)
 543 {
 544         struct es2_ap_dev *es2 = hd_to_es2(hd);
 545         struct usb_device *udev = es2->usb_dev;
 546         struct gb_apb_request_cport_flags *req;
 547         u32 connection_flags;
 548         int ret;
 549
 550         req = kzalloc(sizeof(*req), GFP_KERNEL);
 551         if (!req)
 552                 return -ENOMEM;
 553
 554         connection_flags = 0;
 555         if (flags & GB_CONNECTION_FLAG_CONTROL)
 556                 connection_flags |= GB_APB_CPORT_FLAG_CONTROL;
 557         if (flags & GB_CONNECTION_FLAG_HIGH_PRIO)
 558                 connection_flags |= GB_APB_CPORT_FLAG_HIGH_PRIO;
 559
 560         req->flags = cpu_to_le32(connection_flags);
 561
 562         dev_dbg(&hd->dev, "%s - cport = %u, flags = %02x\n", __func__,
 563                 cport_id, connection_flags);
 564
 565         ret = usb_control_msg(udev, usb_sndctrlpipe(udev, 0),
 566                               GB_APB_REQUEST_CPORT_FLAGS,
 567                               USB_DIR_OUT | USB_TYPE_VENDOR |
 568                               USB_RECIP_INTERFACE, cport_id, 0,
 569                               req, sizeof(*req), ES2_USB_CTRL_TIMEOUT);
 570         if (ret != sizeof(*req)) {
 571                 dev_err(&udev->dev, "failed to set cport flags for port %d\n",
 572                         cport_id);
 573                 if (ret >= 0)
 574                         ret = -EIO;
 575
 576                 goto out;
 577         }
 578
 579         ret = 0;
 580 out:
 581         kfree(req);
 582
 583         return ret;
 584 }
 585
 586 static int es2_cport_connected(struct gb_host_device *hd, u16 cport_id)
 587 {
 588         struct es2_ap_dev *es2 = hd_to_es2(hd);
 589         struct device *dev = &es2->usb_dev->dev;
 590         struct arpc_cport_connected_req req;
 591         int ret;
 592
 593         req.cport_id = cpu_to_le16(cport_id);
 594         ret = arpc_sync(es2, ARPC_TYPE_CPORT_CONNECTED, &req, sizeof(req),
 595                         NULL, ES2_ARPC_CPORT_TIMEOUT);
 596         if (ret) {
 597                 dev_err(dev, "failed to set connected state for cport %u: %d\n",
 598                         cport_id, ret);
 599                 return ret;
 600         }
 601
 602         return 0;
 603 }
 604
 605 static int es2_cport_flush(struct gb_host_device *hd, u16 cport_id)
 606 {
 607         struct es2_ap_dev *es2 = hd_to_es2(hd);
 608         struct device *dev = &es2->usb_dev->dev;
 609         struct arpc_cport_flush_req req;
 610         int ret;
 611
 612         req.cport_id = cpu_to_le16(cport_id);
 613         ret = arpc_sync(es2, ARPC_TYPE_CPORT_FLUSH, &req, sizeof(req),
 614                         NULL, ES2_ARPC_CPORT_TIMEOUT);
 615         if (ret) {
 616                 dev_err(dev, "failed to flush cport %u: %d\n", cport_id, ret);
 617                 return ret;
 618         }
 619
 620         return 0;
 621 }
 622
 623 static int es2_cport_shutdown(struct gb_host_device *hd, u16 cport_id,
 624                               u8 phase, unsigned int timeout)
 625 {
 626         struct es2_ap_dev *es2 = hd_to_es2(hd);
 627         struct device *dev = &es2->usb_dev->dev;
 628         struct arpc_cport_shutdown_req req;
 629         int result;
 630         int ret;
 631
 632         if (timeout > U16_MAX)
 633                 return -EINVAL;
 634
 635         req.cport_id = cpu_to_le16(cport_id);
 636         req.timeout = cpu_to_le16(timeout);
 637         req.phase = phase;
 638         ret = arpc_sync(es2, ARPC_TYPE_CPORT_SHUTDOWN, &req, sizeof(req),
 639                         &result, ES2_ARPC_CPORT_TIMEOUT + timeout);
 640         if (ret) {
 641                 dev_err(dev, "failed to send shutdown over cport %u: %d (%d)\n",
 642                         cport_id, ret, result);
 643                 return ret;
 644         }
 645
 646         return 0;
 647 }
 648
 649 static int es2_cport_quiesce(struct gb_host_device *hd, u16 cport_id,
 650                              size_t peer_space, unsigned int timeout)
 651 {
 652         struct es2_ap_dev *es2 = hd_to_es2(hd);
 653         struct device *dev = &es2->usb_dev->dev;
 654         struct arpc_cport_quiesce_req req;
 655         int result;
 656         int ret;
 657
 658         if (peer_space > U16_MAX)
 659                 return -EINVAL;
 660
 661         if (timeout > U16_MAX)
 662                 return -EINVAL;
 663
 664         req.cport_id = cpu_to_le16(cport_id);
 665         req.peer_space = cpu_to_le16(peer_space);
 666         req.timeout = cpu_to_le16(timeout);
 667         ret = arpc_sync(es2, ARPC_TYPE_CPORT_QUIESCE, &req, sizeof(req),
 668                         &result, ES2_ARPC_CPORT_TIMEOUT + timeout);
 669         if (ret) {
 670                 dev_err(dev, "failed to quiesce cport %u: %d (%d)\n",
 671                         cport_id, ret, result);
 672                 return ret;
 673         }
 674
 675         return 0;
 676 }
 677
 678 static int es2_cport_clear(struct gb_host_device *hd, u16 cport_id)
 679 {
 680         struct es2_ap_dev *es2 = hd_to_es2(hd);
 681         struct device *dev = &es2->usb_dev->dev;
 682         struct arpc_cport_clear_req req;
 683         int ret;
 684
 685         req.cport_id = cpu_to_le16(cport_id);
 686         ret = arpc_sync(es2, ARPC_TYPE_CPORT_CLEAR, &req, sizeof(req),
 687                         NULL, ES2_ARPC_CPORT_TIMEOUT);
 688         if (ret) {
 689                 dev_err(dev, "failed to clear cport %u: %d\n", cport_id, ret);
 690                 return ret;
 691         }
 692
 693         return 0;
 694 }
 695
 696 static int latency_tag_enable(struct gb_host_device *hd, u16 cport_id)
 697 {
 698         int retval;
 699         struct es2_ap_dev *es2 = hd_to_es2(hd);
 700         struct usb_device *udev = es2->usb_dev;
 701
 702         retval = usb_control_msg(udev, usb_sndctrlpipe(udev, 0),
 703                                  GB_APB_REQUEST_LATENCY_TAG_EN,
 704                                  USB_DIR_OUT | USB_TYPE_VENDOR |
 705                                  USB_RECIP_INTERFACE, cport_id, 0, NULL,
 706                                  0, ES2_USB_CTRL_TIMEOUT);
 707
 708         if (retval < 0)
 709                 dev_err(&udev->dev, "Cannot enable latency tag for cport %d\n",
 710                         cport_id);
 711         return retval;
 712 }
 713
 714 static int latency_tag_disable(struct gb_host_device *hd, u16 cport_id)
 715 {
 716         int retval;
 717         struct es2_ap_dev *es2 = hd_to_es2(hd);
 718         struct usb_device *udev = es2->usb_dev;
 719
 720         retval = usb_control_msg(udev, usb_sndctrlpipe(udev, 0),
 721                                  GB_APB_REQUEST_LATENCY_TAG_DIS,
 722                                  USB_DIR_OUT | USB_TYPE_VENDOR |
 723                                  USB_RECIP_INTERFACE, cport_id, 0, NULL,
 724                                  0, ES2_USB_CTRL_TIMEOUT);
 725
 726         if (retval < 0)
 727                 dev_err(&udev->dev, "Cannot disable latency tag for cport %d\n",
 728                         cport_id);
 729         return retval;
 730 }
 731
 732 static struct gb_hd_driver es2_driver = {
 733         .hd_priv_size                   = sizeof(struct es2_ap_dev),
 734         .message_send                   = message_send,
 735         .message_cancel                 = message_cancel,
 736         .cport_allocate                 = es2_cport_allocate,
 737         .cport_release                  = es2_cport_release,
 738         .cport_enable                   = cport_enable,
 739         .cport_connected                = es2_cport_connected,
 740         .cport_flush                    = es2_cport_flush,
 741         .cport_shutdown                 = es2_cport_shutdown,
 742         .cport_quiesce                  = es2_cport_quiesce,
 743         .cport_clear                    = es2_cport_clear,
 744         .latency_tag_enable             = latency_tag_enable,
 745         .latency_tag_disable            = latency_tag_disable,
 746         .output                         = output,
 747 };
 748
 749 /* Common function to report consistent warnings based on URB status */
 750 static int check_urb_status(struct urb *urb)
 751 {
 752         struct device *dev = &urb->dev->dev;
 753         int status = urb->status;
 754
 755         switch (status) {
 756         case 0:
 757                 return 0;
 758
 759         case -EOVERFLOW:
 760                 dev_err(dev, "%s: overflow actual length is %d\n",
 761                         __func__, urb->actual_length);
 762                 /* fall through */
 763         case -ECONNRESET:
 764         case -ENOENT:
 765         case -ESHUTDOWN:
 766         case -EILSEQ:
 767         case -EPROTO:
 768                 /* device is gone, stop sending */
 769                 return status;
 770         }
 771         dev_err(dev, "%s: unknown status %d\n", __func__, status);
 772
 773         return -EAGAIN;
 774 }
 775
 776 static void es2_destroy(struct es2_ap_dev *es2)
 777 {
 778         struct usb_device *udev;
 779         struct urb *urb;
 780         int i;
 781
 782         debugfs_remove(es2->apb_log_enable_dentry);
 783         usb_log_disable(es2);
 784
 785         /* Tear down everything! */
 786         for (i = 0; i < NUM_CPORT_OUT_URB; ++i) {
 787                 urb = es2->cport_out_urb[i];
 788                 usb_kill_urb(urb);
 789                 usb_free_urb(urb);
 790                 es2->cport_out_urb[i] = NULL;
 791                 es2->cport_out_urb_busy[i] = false;     /* just to be anal */
 792         }
 793
 794         for (i = 0; i < NUM_ARPC_IN_URB; ++i) {
 795                 usb_free_urb(es2->arpc_urb[i]);
 796                 kfree(es2->arpc_buffer[i]);
 797                 es2->arpc_buffer[i] = NULL;
 798         }
 799
 800         for (i = 0; i < NUM_CPORT_IN_URB; ++i) {
 801                 usb_free_urb(es2->cport_in.urb[i]);
 802                 kfree(es2->cport_in.buffer[i]);
 803                 es2->cport_in.buffer[i] = NULL;
 804         }
 805
 806         /* release reserved CDSI0 and CDSI1 cports */
 807         gb_hd_cport_release_reserved(es2->hd, ES2_CPORT_CDSI1);
 808         gb_hd_cport_release_reserved(es2->hd, ES2_CPORT_CDSI0);
 809
 810         udev = es2->usb_dev;
 811         gb_hd_put(es2->hd);
 812
 813         usb_put_dev(udev);
 814 }
 815
 816 static void cport_in_callback(struct urb *urb)
 817 {
 818         struct gb_host_device *hd = urb->context;
 819         struct device *dev = &urb->dev->dev;
 820         struct gb_operation_msg_hdr *header;
 821         int status = check_urb_status(urb);
 822         int retval;
 823         u16 cport_id;
 824
 825         if (status) {
 826                 if ((status == -EAGAIN) || (status == -EPROTO))
 827                         goto exit;
 828
 829                 /* The urb is being unlinked */
 830                 if (status == -ENOENT || status == -ESHUTDOWN)
 831                         return;
 832
 833                 dev_err(dev, "urb cport in error %d (dropped)\n", status);
 834                 return;
 835         }
 836
 837         if (urb->actual_length < sizeof(*header)) {
 838                 dev_err(dev, "short message received\n");
 839                 goto exit;
 840         }
 841
 842         /* Extract the CPort id, which is packed in the message header */
 843         header = urb->transfer_buffer;
 844         cport_id = gb_message_cport_unpack(header);
 845
 846         if (cport_id_valid(hd, cport_id)) {
 847                 greybus_data_rcvd(hd, cport_id, urb->transfer_buffer,
 848                                   urb->actual_length);
 849         } else {
 850                 dev_err(dev, "invalid cport id %u received\n", cport_id);
 851         }
 852 exit:
 853         /* put our urb back in the request pool */
 854         retval = usb_submit_urb(urb, GFP_ATOMIC);
 855         if (retval)
 856                 dev_err(dev, "failed to resubmit in-urb: %d\n", retval);
 857 }
 858
 859 static void cport_out_callback(struct urb *urb)
 860 {
 861         struct gb_message *message = urb->context;
 862         struct gb_host_device *hd = message->operation->connection->hd;
 863         struct es2_ap_dev *es2 = hd_to_es2(hd);
 864         int status = check_urb_status(urb);
 865         unsigned long flags;
 866
 867         gb_message_cport_clear(message->header);
 868
 869         spin_lock_irqsave(&es2->cport_out_urb_lock, flags);
 870         message->hcpriv = NULL;
 871         spin_unlock_irqrestore(&es2->cport_out_urb_lock, flags);
 872
 873         /*
 874          * Tell the submitter that the message send (attempt) is
 875          * complete, and report the status.
 876          */
 877         greybus_message_sent(hd, message, status);
 878
 879         free_urb(es2, urb);
 880 }
 881
 882 static struct arpc *arpc_alloc(void *payload, u16 size, u8 type)
 883 {
 884         struct arpc *rpc;
 885
 886         if (size + sizeof(*rpc->req) > ARPC_OUT_SIZE_MAX)
 887                 return NULL;
 888
 889         rpc = kzalloc(sizeof(*rpc), GFP_KERNEL);
 890         if (!rpc)
 891                 return NULL;
 892
 893         INIT_LIST_HEAD(&rpc->list);
 894         rpc->req = kzalloc(sizeof(*rpc->req) + size, GFP_KERNEL);
 895         if (!rpc->req)
 896                 goto err_free_rpc;
 897
 898         rpc->resp = kzalloc(sizeof(*rpc->resp), GFP_KERNEL);
 899         if (!rpc->resp)
 900                 goto err_free_req;
 901
 902         rpc->req->type = type;
 903         rpc->req->size = cpu_to_le16(sizeof(*rpc->req) + size);
 904         memcpy(rpc->req->data, payload, size);
 905
 906         init_completion(&rpc->response_received);
 907
 908         return rpc;
 909
 910 err_free_req:
 911         kfree(rpc->req);
 912 err_free_rpc:
 913         kfree(rpc);
 914
 915         return NULL;
 916 }
 917
 918 static void arpc_free(struct arpc *rpc)
 919 {
 920         kfree(rpc->req);
 921         kfree(rpc->resp);
 922         kfree(rpc);
 923 }
 924
 925 static struct arpc *arpc_find(struct es2_ap_dev *es2, __le16 id)
 926 {
 927         struct arpc *rpc;
 928
 929         list_for_each_entry(rpc, &es2->arpcs, list) {
 930                 if (rpc->req->id == id)
 931                         return rpc;
 932         }
 933
 934         return NULL;
 935 }
 936
 937 static void arpc_add(struct es2_ap_dev *es2, struct arpc *rpc)
 938 {
 939         rpc->active = true;
 940         rpc->req->id = cpu_to_le16(es2->arpc_id_cycle++);
 941         list_add_tail(&rpc->list, &es2->arpcs);
 942 }
 943
 944 static void arpc_del(struct es2_ap_dev *es2, struct arpc *rpc)
 945 {
 946         if (rpc->active) {
 947                 rpc->active = false;
 948                 list_del(&rpc->list);
 949         }
 950 }
 951
 952 static int arpc_send(struct es2_ap_dev *es2, struct arpc *rpc, int timeout)
 953 {
 954         struct usb_device *udev = es2->usb_dev;
 955         int retval;
 956
 957         retval = usb_control_msg(udev, usb_sndctrlpipe(udev, 0),
 958                                  GB_APB_REQUEST_ARPC_RUN,
 959                                  USB_DIR_OUT | USB_TYPE_VENDOR |
 960                                  USB_RECIP_INTERFACE,
 961                                  0, 0,
 962                                  rpc->req, le16_to_cpu(rpc->req->size),
 963                                  ES2_USB_CTRL_TIMEOUT);
 964         if (retval != le16_to_cpu(rpc->req->size)) {
 965                 dev_err(&udev->dev,
 966                         "failed to send ARPC request %d: %d\n",
 967                         rpc->req->type, retval);
 968                 if (retval > 0)
 969                         retval = -EIO;
 970                 return retval;
 971         }
 972
 973         return 0;
 974 }
 975
 976 static int arpc_sync(struct es2_ap_dev *es2, u8 type, void *payload,
 977                      size_t size, int *result, unsigned int timeout)
 978 {
 979         struct arpc *rpc;
 980         unsigned long flags;
 981         int retval;
 982
 983         if (result)
 984                 *result = 0;
 985
 986         rpc = arpc_alloc(payload, size, type);
 987         if (!rpc)
 988                 return -ENOMEM;
 989
 990         spin_lock_irqsave(&es2->arpc_lock, flags);
 991         arpc_add(es2, rpc);
 992         spin_unlock_irqrestore(&es2->arpc_lock, flags);
 993
 994         retval = arpc_send(es2, rpc, timeout);
 995         if (retval)
 996                 goto out_arpc_del;
 997
 998         retval = wait_for_completion_interruptible_timeout(
 999                                                 &rpc->response_received,
1000                                                 msecs_to_jiffies(timeout));
1001         if (retval <= 0) {
1002                 if (!retval)
1003                         retval = -ETIMEDOUT;
1004                 goto out_arpc_del;
1005         }
1006
1007         if (rpc->resp->result) {
1008                 retval = -EREMOTEIO;
1009                 if (result)
1010                         *result = rpc->resp->result;
1011         } else {
1012                 retval = 0;
1013         }
1014
1015 out_arpc_del:
1016         spin_lock_irqsave(&es2->arpc_lock, flags);
1017         arpc_del(es2, rpc);
1018         spin_unlock_irqrestore(&es2->arpc_lock, flags);
1019         arpc_free(rpc);
1020
1021         if (retval < 0 && retval != -EREMOTEIO) {
1022                 dev_err(&es2->usb_dev->dev,
1023                         "failed to execute ARPC: %d\n", retval);
1024         }
1025
1026         return retval;
1027 }
1028
1029 static void arpc_in_callback(struct urb *urb)
1030 {
1031         struct es2_ap_dev *es2 = urb->context;
1032         struct device *dev = &urb->dev->dev;
1033         int status = check_urb_status(urb);
1034         struct arpc *rpc;
1035         struct arpc_response_message *resp;
1036         unsigned long flags;
1037         int retval;
1038
1039         if (status) {
1040                 if ((status == -EAGAIN) || (status == -EPROTO))
1041                         goto exit;
1042
1043                 /* The urb is being unlinked */
1044                 if (status == -ENOENT || status == -ESHUTDOWN)
1045                         return;
1046
1047                 dev_err(dev, "arpc in-urb error %d (dropped)\n", status);
1048                 return;
1049         }
1050
1051         if (urb->actual_length < sizeof(*resp)) {
1052                 dev_err(dev, "short aprc response received\n");
1053                 goto exit;
1054         }
1055
1056         resp = urb->transfer_buffer;
1057         spin_lock_irqsave(&es2->arpc_lock, flags);
1058         rpc = arpc_find(es2, resp->id);
1059         if (!rpc) {
1060                 dev_err(dev, "invalid arpc response id received: %u\n",
1061                         le16_to_cpu(resp->id));
1062                 spin_unlock_irqrestore(&es2->arpc_lock, flags);
1063                 goto exit;
1064         }
1065
1066         arpc_del(es2, rpc);
1067         memcpy(rpc->resp, resp, sizeof(*resp));
1068         complete(&rpc->response_received);
1069         spin_unlock_irqrestore(&es2->arpc_lock, flags);
1070
1071 exit:
1072         /* put our urb back in the request pool */
1073         retval = usb_submit_urb(urb, GFP_ATOMIC);
1074         if (retval)
1075                 dev_err(dev, "failed to resubmit arpc in-urb: %d\n", retval);
1076 }
1077
1078 #define APB1_LOG_MSG_SIZE       64
1079 static void apb_log_get(struct es2_ap_dev *es2, char *buf)
1080 {
1081         int retval;
1082
1083         do {
1084                 retval = usb_control_msg(es2->usb_dev,
1085                                          usb_rcvctrlpipe(es2->usb_dev, 0),
1086                                          GB_APB_REQUEST_LOG,
1087                                          USB_DIR_IN | USB_TYPE_VENDOR |
1088                                          USB_RECIP_INTERFACE,
1089                                          0x00, 0x00,
1090                                          buf,
1091                                          APB1_LOG_MSG_SIZE,
1092                                          ES2_USB_CTRL_TIMEOUT);
1093                 if (retval > 0)
1094                         kfifo_in(&es2->apb_log_fifo, buf, retval);
1095         } while (retval > 0);
1096 }
1097
1098 static int apb_log_poll(void *data)
1099 {
1100         struct es2_ap_dev *es2 = data;
1101         char *buf;
1102
1103         buf = kmalloc(APB1_LOG_MSG_SIZE, GFP_KERNEL);
1104         if (!buf)
1105                 return -ENOMEM;
1106
1107         while (!kthread_should_stop()) {
1108                 msleep(1000);
1109                 apb_log_get(es2, buf);
1110         }
1111
1112         kfree(buf);
1113
1114         return 0;
1115 }
1116
1117 static ssize_t apb_log_read(struct file *f, char __user *buf,
1118                             size_t count, loff_t *ppos)
1119 {
1120         struct es2_ap_dev *es2 = file_inode(f)->i_private;
1121         ssize_t ret;
1122         size_t copied;
1123         char *tmp_buf;
1124
1125         if (count > APB1_LOG_SIZE)
1126                 count = APB1_LOG_SIZE;
1127
1128         tmp_buf = kmalloc(count, GFP_KERNEL);
1129         if (!tmp_buf)
1130                 return -ENOMEM;
1131
1132         copied = kfifo_out(&es2->apb_log_fifo, tmp_buf, count);
1133         ret = simple_read_from_buffer(buf, count, ppos, tmp_buf, copied);
1134
1135         kfree(tmp_buf);
1136
1137         return ret;
1138 }
1139
1140 static const struct file_operations apb_log_fops = {
1141         .read   = apb_log_read,
1142 };
1143
1144 static void usb_log_enable(struct es2_ap_dev *es2)
1145 {
1146         if (!IS_ERR_OR_NULL(es2->apb_log_task))
1147                 return;
1148
1149         /* get log from APB1 */
1150         es2->apb_log_task = kthread_run(apb_log_poll, es2, "apb_log");
1151         if (IS_ERR(es2->apb_log_task))
1152                 return;
1153         /* XXX We will need to rename this per APB */
1154         es2->apb_log_dentry = debugfs_create_file("apb_log", 0444,
1155                                                   gb_debugfs_get(), es2,
1156                                                   &apb_log_fops);
1157 }
1158
1159 static void usb_log_disable(struct es2_ap_dev *es2)
1160 {
1161         if (IS_ERR_OR_NULL(es2->apb_log_task))
1162                 return;
1163
1164         debugfs_remove(es2->apb_log_dentry);
1165         es2->apb_log_dentry = NULL;
1166
1167         kthread_stop(es2->apb_log_task);
1168         es2->apb_log_task = NULL;
1169 }
1170
1171 static ssize_t apb_log_enable_read(struct file *f, char __user *buf,
1172                                    size_t count, loff_t *ppos)
1173 {
1174         struct es2_ap_dev *es2 = file_inode(f)->i_private;
1175         int enable = !IS_ERR_OR_NULL(es2->apb_log_task);
1176         char tmp_buf[3];
1177
1178         sprintf(tmp_buf, "%d\n", enable);
1179         return simple_read_from_buffer(buf, count, ppos, tmp_buf, 3);
1180 }
1181
1182 static ssize_t apb_log_enable_write(struct file *f, const char __user *buf,
1183                                     size_t count, loff_t *ppos)
1184 {
1185         int enable;
1186         ssize_t retval;
1187         struct es2_ap_dev *es2 = file_inode(f)->i_private;
1188
1189         retval = kstrtoint_from_user(buf, count, 10, &enable);
1190         if (retval)
1191                 return retval;
1192
1193         if (enable)
1194                 usb_log_enable(es2);
1195         else
1196                 usb_log_disable(es2);
1197
1198         return count;
1199 }
1200
1201 static const struct file_operations apb_log_enable_fops = {
1202         .read   = apb_log_enable_read,
1203         .write  = apb_log_enable_write,
1204 };
1205
1206 static int apb_get_cport_count(struct usb_device *udev)
1207 {
1208         int retval;
1209         __le16 *cport_count;
1210
1211         cport_count = kzalloc(sizeof(*cport_count), GFP_KERNEL);
1212         if (!cport_count)
1213                 return -ENOMEM;
1214
1215         retval = usb_control_msg(udev, usb_rcvctrlpipe(udev, 0),
1216                                  GB_APB_REQUEST_CPORT_COUNT,
1217                                  USB_DIR_IN | USB_TYPE_VENDOR |
1218                                  USB_RECIP_INTERFACE, 0, 0, cport_count,
1219                                  sizeof(*cport_count), ES2_USB_CTRL_TIMEOUT);
1220         if (retval != sizeof(*cport_count)) {
1221                 dev_err(&udev->dev, "Cannot retrieve CPort count: %d\n",
1222                         retval);
1223
1224                 if (retval >= 0)
1225                         retval = -EIO;
1226
1227                 goto out;
1228         }
1229
1230         retval = le16_to_cpu(*cport_count);
1231
1232         /* We need to fit a CPort ID in one byte of a message header */
1233         if (retval > U8_MAX) {
1234                 retval = U8_MAX;
1235                 dev_warn(&udev->dev, "Limiting number of CPorts to U8_MAX\n");
1236         }
1237
1238 out:
1239         kfree(cport_count);
1240         return retval;
1241 }
1242
1243 /*
1244  * The ES2 USB Bridge device has 15 endpoints
1245  * 1 Control - usual USB stuff + AP -> APBridgeA messages
1246  * 7 Bulk IN - CPort data in
1247  * 7 Bulk OUT - CPort data out
1248  */
1249 static int ap_probe(struct usb_interface *interface,
1250                     const struct usb_device_id *id)
1251 {
1252         struct es2_ap_dev *es2;
1253         struct gb_host_device *hd;
1254         struct usb_device *udev;
1255         struct usb_host_interface *iface_desc;
1256         struct usb_endpoint_descriptor *endpoint;
1257         __u8 ep_addr;
1258         int retval;
1259         int i;
1260         int num_cports;
1261         bool bulk_out_found = false;
1262         bool bulk_in_found = false;
1263         bool arpc_in_found = false;
1264
1265         udev = usb_get_dev(interface_to_usbdev(interface));
1266
1267         num_cports = apb_get_cport_count(udev);
1268         if (num_cports < 0) {
1269                 usb_put_dev(udev);
1270                 dev_err(&udev->dev, "Cannot retrieve CPort count: %d\n",
1271                         num_cports);
1272                 return num_cports;
1273         }
1274
1275         hd = gb_hd_create(&es2_driver, &udev->dev, ES2_GBUF_MSG_SIZE_MAX,
1276                           num_cports);
1277         if (IS_ERR(hd)) {
1278                 usb_put_dev(udev);
1279                 return PTR_ERR(hd);
1280         }
1281
1282         es2 = hd_to_es2(hd);
1283         es2->hd = hd;
1284         es2->usb_intf = interface;
1285         es2->usb_dev = udev;
1286         spin_lock_init(&es2->cport_out_urb_lock);
1287         INIT_KFIFO(es2->apb_log_fifo);
1288         usb_set_intfdata(interface, es2);
1289
1290         /*
1291          * Reserve the CDSI0 and CDSI1 CPorts so they won't be allocated
1292          * dynamically.
1293          */
1294         retval = gb_hd_cport_reserve(hd, ES2_CPORT_CDSI0);
1295         if (retval)
1296                 goto error;
1297         retval = gb_hd_cport_reserve(hd, ES2_CPORT_CDSI1);
1298         if (retval)
1299                 goto error;
1300
1301         /* find all bulk endpoints */
1302         iface_desc = interface->cur_altsetting;
1303         for (i = 0; i < iface_desc->desc.bNumEndpoints; ++i) {
1304                 endpoint = &iface_desc->endpoint[i].desc;
1305                 ep_addr = endpoint->bEndpointAddress;
1306
1307                 if (usb_endpoint_is_bulk_in(endpoint)) {
1308                         if (!bulk_in_found) {
1309                                 es2->cport_in.endpoint = ep_addr;
1310                                 bulk_in_found = true;
1311                         } else if (!arpc_in_found) {
1312                                 es2->arpc_endpoint_in = ep_addr;
1313                                 arpc_in_found = true;
1314                         } else {
1315                                 dev_warn(&udev->dev,
1316                                          "Unused bulk IN endpoint found: 0x%02x\n",
1317                                          ep_addr);
1318                         }
1319                         continue;
1320                 }
1321                 if (usb_endpoint_is_bulk_out(endpoint)) {
1322                         if (!bulk_out_found) {
1323                                 es2->cport_out_endpoint = ep_addr;
1324                                 bulk_out_found = true;
1325                         } else {
1326                                 dev_warn(&udev->dev,
1327                                          "Unused bulk OUT endpoint found: 0x%02x\n",
1328                                          ep_addr);
1329                         }
1330                         continue;
1331                 }
1332                 dev_warn(&udev->dev,
1333                          "Unknown endpoint type found, address 0x%02x\n",
1334                          ep_addr);
1335         }
1336         if (!bulk_in_found || !arpc_in_found || !bulk_out_found) {
1337                 dev_err(&udev->dev, "Not enough endpoints found in device, aborting!\n");
1338                 retval = -ENODEV;
1339                 goto error;
1340         }
1341
1342         /* Allocate buffers for our cport in messages */
1343         for (i = 0; i < NUM_CPORT_IN_URB; ++i) {
1344                 struct urb *urb;
1345                 u8 *buffer;
1346
1347                 urb = usb_alloc_urb(0, GFP_KERNEL);
1348                 if (!urb) {
1349                         retval = -ENOMEM;
1350                         goto error;
1351                 }
1352                 es2->cport_in.urb[i] = urb;
1353
1354                 buffer = kmalloc(ES2_GBUF_MSG_SIZE_MAX, GFP_KERNEL);
1355                 if (!buffer) {
1356                         retval = -ENOMEM;
1357                         goto error;
1358                 }
1359
1360                 usb_fill_bulk_urb(urb, udev,
1361                                   usb_rcvbulkpipe(udev, es2->cport_in.endpoint),
1362                                   buffer, ES2_GBUF_MSG_SIZE_MAX,
1363                                   cport_in_callback, hd);
1364
1365                 es2->cport_in.buffer[i] = buffer;
1366         }
1367
1368         /* Allocate buffers for ARPC in messages */
1369         for (i = 0; i < NUM_ARPC_IN_URB; ++i) {
1370                 struct urb *urb;
1371                 u8 *buffer;
1372
1373                 urb = usb_alloc_urb(0, GFP_KERNEL);
1374                 if (!urb) {
1375                         retval = -ENOMEM;
1376                         goto error;
1377                 }
1378                 es2->arpc_urb[i] = urb;
1379
1380                 buffer = kmalloc(ARPC_IN_SIZE_MAX, GFP_KERNEL);
1381                 if (!buffer) {
1382                         retval = -ENOMEM;
1383                         goto error;
1384                 }
1385
1386                 usb_fill_bulk_urb(urb, udev,
1387                                   usb_rcvbulkpipe(udev,
1388                                                   es2->arpc_endpoint_in),
1389                                   buffer, ARPC_IN_SIZE_MAX,
1390                                   arpc_in_callback, es2);
1391
1392                 es2->arpc_buffer[i] = buffer;
1393         }
1394
1395         /* Allocate urbs for our CPort OUT messages */
1396         for (i = 0; i < NUM_CPORT_OUT_URB; ++i) {
1397                 struct urb *urb;
1398
1399                 urb = usb_alloc_urb(0, GFP_KERNEL);
1400                 if (!urb) {
1401                         retval = -ENOMEM;
1402                         goto error;
1403                 }
1404
1405                 es2->cport_out_urb[i] = urb;
1406                 es2->cport_out_urb_busy[i] = false;     /* just to be anal */
1407         }
1408
1409         /* XXX We will need to rename this per APB */
1410         es2->apb_log_enable_dentry = debugfs_create_file("apb_log_enable",
1411                                                          0644,
1412                                                          gb_debugfs_get(), es2,
1413                                                          &apb_log_enable_fops);
1414
1415         INIT_LIST_HEAD(&es2->arpcs);
1416         spin_lock_init(&es2->arpc_lock);
1417
1418         retval = es2_arpc_in_enable(es2);
1419         if (retval)
1420                 goto error;
1421
1422         retval = gb_hd_add(hd);
1423         if (retval)
1424                 goto err_disable_arpc_in;
1425
1426         retval = es2_cport_in_enable(es2, &es2->cport_in);
1427         if (retval)
1428                 goto err_hd_del;
1429
1430         return 0;
1431
1432 err_hd_del:
1433         gb_hd_del(hd);
1434 err_disable_arpc_in:
1435         es2_arpc_in_disable(es2);
1436 error:
1437         es2_destroy(es2);
1438
1439         return retval;
1440 }
1441
1442 static void ap_disconnect(struct usb_interface *interface)
1443 {
1444         struct es2_ap_dev *es2 = usb_get_intfdata(interface);
1445
1446         gb_hd_del(es2->hd);
1447
1448         es2_cport_in_disable(es2, &es2->cport_in);
1449         es2_arpc_in_disable(es2);
1450
1451         es2_destroy(es2);
1452 }
1453
1454 static struct usb_driver es2_ap_driver = {
1455         .name =         "es2_ap_driver",
1456         .probe =        ap_probe,
1457         .disconnect =   ap_disconnect,
1458         .id_table =     id_table,
1459         .soft_unbind =  1,
1460 };
1461
1462 module_usb_driver(es2_ap_driver);
1463
1464 MODULE_LICENSE("GPL v2");
1465 MODULE_AUTHOR("Greg Kroah-Hartman <gregkh@linuxfoundation.org>");