// SPDX-License-Identifier: ISC
/*
 * Copyright (c) 2011 Broadcom Corporation
 */

#include <linux/kernel.h>
#include <linux/module.h>
#include <linux/firmware.h>
#include <linux/usb.h>
#include <linux/vmalloc.h>

#include <brcmu_utils.h>
#include <brcm_hw_ids.h>
#include <brcmu_wifi.h>
#include "bus.h"
#include "debug.h"
#include "firmware.h"
#include "usb.h"
#include "core.h"
#include "common.h"
#include "bcdc.h"


#define IOCTL_RESP_TIMEOUT              msecs_to_jiffies(2000)

#define BRCMF_USB_RESET_GETVER_SPINWAIT 100     /* in unit of ms */
#define BRCMF_USB_RESET_GETVER_LOOP_CNT 10

#define BRCMF_POSTBOOT_ID               0xA123  /* ID to detect if dongle
                                                   has boot up */
#define BRCMF_USB_NRXQ                  50
#define BRCMF_USB_NTXQ                  50

#define BRCMF_USB_CBCTL_WRITE           0
#define BRCMF_USB_CBCTL_READ            1
#define BRCMF_USB_MAX_PKT_SIZE          1600

BRCMF_FW_DEF(43143, "brcmfmac43143");
BRCMF_FW_DEF(43236B, "brcmfmac43236b");
BRCMF_FW_DEF(43242A, "brcmfmac43242a");
BRCMF_FW_DEF(43569, "brcmfmac43569");
BRCMF_FW_DEF(4373, "brcmfmac4373");

static const struct brcmf_firmware_mapping brcmf_usb_fwnames[] = {
        BRCMF_FW_ENTRY(BRCM_CC_43143_CHIP_ID, 0xFFFFFFFF, 43143),
        BRCMF_FW_ENTRY(BRCM_CC_43235_CHIP_ID, 0x00000008, 43236B),
        BRCMF_FW_ENTRY(BRCM_CC_43236_CHIP_ID, 0x00000008, 43236B),
        BRCMF_FW_ENTRY(BRCM_CC_43238_CHIP_ID, 0x00000008, 43236B),
        BRCMF_FW_ENTRY(BRCM_CC_43242_CHIP_ID, 0xFFFFFFFF, 43242A),
        BRCMF_FW_ENTRY(BRCM_CC_43566_CHIP_ID, 0xFFFFFFFF, 43569),
        BRCMF_FW_ENTRY(BRCM_CC_43569_CHIP_ID, 0xFFFFFFFF, 43569),
        BRCMF_FW_ENTRY(CY_CC_4373_CHIP_ID, 0xFFFFFFFF, 4373)
};

#define TRX_MAGIC               0x30524448      /* "HDR0" */
#define TRX_MAX_OFFSET          3               /* Max number of file offsets */
#define TRX_UNCOMP_IMAGE        0x20            /* Trx holds uncompressed img */
#define TRX_RDL_CHUNK           1500            /* size of each dl transfer */
#define TRX_OFFSETS_DLFWLEN_IDX 0

/* Control messages: bRequest values */
#define DL_GETSTATE     0       /* returns the rdl_state_t struct */
#define DL_CHECK_CRC    1       /* currently unused */
#define DL_GO           2       /* execute downloaded image */
#define DL_START        3       /* initialize dl state */
#define DL_REBOOT       4       /* reboot the device in 2 seconds */
#define DL_GETVER       5       /* returns the bootrom_id_t struct */
#define DL_GO_PROTECTED 6       /* execute the downloaded code and set reset
                                 * event to occur in 2 seconds.  It is the
                                 * responsibility of the downloaded code to
                                 * clear this event
                                 */
#define DL_EXEC         7       /* jump to a supplied address */
#define DL_RESETCFG     8       /* To support single enum on dongle
                                 * - Not used by bootloader
                                 */
#define DL_DEFER_RESP_OK 9      /* Potentially defer the response to setup
                                 * if resp unavailable
                                 */

/* states */
#define DL_WAITING      0       /* waiting to rx first pkt */
#define DL_READY        1       /* hdr was good, waiting for more of the
                                 * compressed image
                                 */
#define DL_BAD_HDR      2       /* hdr was corrupted */
#define DL_BAD_CRC      3       /* compressed image was corrupted */
#define DL_RUNNABLE     4       /* download was successful,waiting for go cmd */
#define DL_START_FAIL   5       /* failed to initialize correctly */
#define DL_NVRAM_TOOBIG 6       /* host specified nvram data exceeds DL_NVRAM
                                 * value
                                 */
#define DL_IMAGE_TOOBIG 7       /* firmware image too big */


struct trx_header_le {
        __le32 magic;           /* "HDR0" */
        __le32 len;             /* Length of file including header */
        __le32 crc32;           /* CRC from flag_version to end of file */
        __le32 flag_version;    /* 0:15 flags, 16:31 version */
        __le32 offsets[TRX_MAX_OFFSET]; /* Offsets of partitions from start of
                                         * header
                                         */
};

struct rdl_state_le {
        __le32 state;
        __le32 bytes;
};

struct bootrom_id_le {
        __le32 chip;            /* Chip id */
        __le32 chiprev;         /* Chip rev */
        __le32 ramsize;         /* Size of  RAM */
        __le32 remapbase;       /* Current remap base address */
        __le32 boardtype;       /* Type of board */
        __le32 boardrev;        /* Board revision */
};

struct brcmf_usb_image {
        struct list_head list;
        s8 *fwname;
        u8 *image;
        int image_len;
};

struct brcmf_usbdev_info {
        struct brcmf_usbdev bus_pub; /* MUST BE FIRST */
        spinlock_t qlock;
        struct list_head rx_freeq;
        struct list_head rx_postq;
        struct list_head tx_freeq;
        struct list_head tx_postq;
        uint rx_pipe, tx_pipe;

        int rx_low_watermark;
        int tx_low_watermark;
        int tx_high_watermark;
        int tx_freecount;
        bool tx_flowblock;
        spinlock_t tx_flowblock_lock;

        struct brcmf_usbreq *tx_reqs;
        struct brcmf_usbreq *rx_reqs;

        char fw_name[BRCMF_FW_NAME_LEN];
        const u8 *image;        /* buffer for combine fw and nvram */
        int image_len;

        struct usb_device *usbdev;
        struct device *dev;
        struct completion dev_init_done;

        int ctl_in_pipe, ctl_out_pipe;
        struct urb *ctl_urb; /* URB for control endpoint */
        struct usb_ctrlrequest ctl_write;
        struct usb_ctrlrequest ctl_read;
        u32 ctl_urb_actual_length;
        int ctl_urb_status;
        int ctl_completed;
        wait_queue_head_t ioctl_resp_wait;
        ulong ctl_op;
        u8 ifnum;

        struct urb *bulk_urb; /* used for FW download */

        bool wowl_enabled;
        struct brcmf_mp_device *settings;
};

static void brcmf_usb_rx_refill(struct brcmf_usbdev_info *devinfo,
                                struct brcmf_usbreq  *req);

static struct brcmf_usbdev *brcmf_usb_get_buspub(struct device *dev)
{
        struct brcmf_bus *bus_if = dev_get_drvdata(dev);
        return bus_if->bus_priv.usb;
}

static struct brcmf_usbdev_info *brcmf_usb_get_businfo(struct device *dev)
{
        return brcmf_usb_get_buspub(dev)->devinfo;
}

static int brcmf_usb_ioctl_resp_wait(struct brcmf_usbdev_info *devinfo)
{
        return wait_event_timeout(devinfo->ioctl_resp_wait,
                                  devinfo->ctl_completed, IOCTL_RESP_TIMEOUT);
}

static void brcmf_usb_ioctl_resp_wake(struct brcmf_usbdev_info *devinfo)
{
        wake_up(&devinfo->ioctl_resp_wait);
}

static void
brcmf_usb_ctl_complete(struct brcmf_usbdev_info *devinfo, int type, int status)
{
        brcmf_dbg(USB, "Enter, status=%d\n", status);

        if (unlikely(devinfo == NULL))
                return;

        if (type == BRCMF_USB_CBCTL_READ) {
                if (status == 0)
                        devinfo->bus_pub.stats.rx_ctlpkts++;
                else
                        devinfo->bus_pub.stats.rx_ctlerrs++;
        } else if (type == BRCMF_USB_CBCTL_WRITE) {
                if (status == 0)
                        devinfo->bus_pub.stats.tx_ctlpkts++;
                else
                        devinfo->bus_pub.stats.tx_ctlerrs++;
        }

        devinfo->ctl_urb_status = status;
        devinfo->ctl_completed = true;
        brcmf_usb_ioctl_resp_wake(devinfo);
}

static void
brcmf_usb_ctlread_complete(struct urb *urb)
{
        struct brcmf_usbdev_info *devinfo =
                (struct brcmf_usbdev_info *)urb->context;

        brcmf_dbg(USB, "Enter\n");
        devinfo->ctl_urb_actual_length = urb->actual_length;
        brcmf_usb_ctl_complete(devinfo, BRCMF_USB_CBCTL_READ,
                urb->status);
}

static void
brcmf_usb_ctlwrite_complete(struct urb *urb)
{
        struct brcmf_usbdev_info *devinfo =
                (struct brcmf_usbdev_info *)urb->context;

        brcmf_dbg(USB, "Enter\n");
        brcmf_usb_ctl_complete(devinfo, BRCMF_USB_CBCTL_WRITE,
                urb->status);
}

static int
brcmf_usb_send_ctl(struct brcmf_usbdev_info *devinfo, u8 *buf, int len)
{
        int ret;
        u16 size;

        brcmf_dbg(USB, "Enter\n");
        if (devinfo == NULL || buf == NULL ||
            len == 0 || devinfo->ctl_urb == NULL)
                return -EINVAL;

        size = len;
        devinfo->ctl_write.wLength = cpu_to_le16p(&size);
        devinfo->ctl_urb->transfer_buffer_length = size;
        devinfo->ctl_urb_status = 0;
        devinfo->ctl_urb_actual_length = 0;

        usb_fill_control_urb(devinfo->ctl_urb,
                devinfo->usbdev,
                devinfo->ctl_out_pipe,
                (unsigned char *) &devinfo->ctl_write,
                buf, size,
                (usb_complete_t)brcmf_usb_ctlwrite_complete,
                devinfo);

        ret = usb_submit_urb(devinfo->ctl_urb, GFP_ATOMIC);
        if (ret < 0)
                brcmf_err("usb_submit_urb failed %d\n", ret);

        return ret;
}

static int
brcmf_usb_recv_ctl(struct brcmf_usbdev_info *devinfo, u8 *buf, int len)
{
        int ret;
        u16 size;

        brcmf_dbg(USB, "Enter\n");
        if ((devinfo == NULL) || (buf == NULL) || (len == 0)
                || (devinfo->ctl_urb == NULL))
                return -EINVAL;

        size = len;
        devinfo->ctl_read.wLength = cpu_to_le16p(&size);
        devinfo->ctl_urb->transfer_buffer_length = size;

        devinfo->ctl_read.bRequestType = USB_DIR_IN
                | USB_TYPE_CLASS | USB_RECIP_INTERFACE;
        devinfo->ctl_read.bRequest = 1;

        usb_fill_control_urb(devinfo->ctl_urb,
                devinfo->usbdev,
                devinfo->ctl_in_pipe,
                (unsigned char *) &devinfo->ctl_read,
                buf, size,
                (usb_complete_t)brcmf_usb_ctlread_complete,
                devinfo);

        ret = usb_submit_urb(devinfo->ctl_urb, GFP_ATOMIC);
        if (ret < 0)
                brcmf_err("usb_submit_urb failed %d\n", ret);

        return ret;
}

static int brcmf_usb_tx_ctlpkt(struct device *dev, u8 *buf, u32 len)
{
        int err = 0;
        int timeout = 0;
        struct brcmf_usbdev_info *devinfo = brcmf_usb_get_businfo(dev);

        brcmf_dbg(USB, "Enter\n");
        if (devinfo->bus_pub.state != BRCMFMAC_USB_STATE_UP)
                return -EIO;

        if (test_and_set_bit(0, &devinfo->ctl_op))
                return -EIO;

        devinfo->ctl_completed = false;
        err = brcmf_usb_send_ctl(devinfo, buf, len);
        if (err) {
                brcmf_err("fail %d bytes: %d\n", err, len);
                clear_bit(0, &devinfo->ctl_op);
                return err;
        }
        timeout = brcmf_usb_ioctl_resp_wait(devinfo);
        clear_bit(0, &devinfo->ctl_op);
        if (!timeout) {
                brcmf_err("Txctl wait timed out\n");
                err = -EIO;
        }
        return err;
}

static int brcmf_usb_rx_ctlpkt(struct device *dev, u8 *buf, u32 len)
{
        int err = 0;
        int timeout = 0;
        struct brcmf_usbdev_info *devinfo = brcmf_usb_get_businfo(dev);

        brcmf_dbg(USB, "Enter\n");
        if (devinfo->bus_pub.state != BRCMFMAC_USB_STATE_UP)
                return -EIO;

        if (test_and_set_bit(0, &devinfo->ctl_op))
                return -EIO;

        devinfo->ctl_completed = false;
        err = brcmf_usb_recv_ctl(devinfo, buf, len);
        if (err) {
                brcmf_err("fail %d bytes: %d\n", err, len);
                clear_bit(0, &devinfo->ctl_op);
                return err;
        }
        timeout = brcmf_usb_ioctl_resp_wait(devinfo);
        err = devinfo->ctl_urb_status;
        clear_bit(0, &devinfo->ctl_op);
        if (!timeout) {
                brcmf_err("rxctl wait timed out\n");
                err = -EIO;
        }
        if (!err)
                return devinfo->ctl_urb_actual_length;
        else
                return err;
}

static struct brcmf_usbreq *brcmf_usb_deq(struct brcmf_usbdev_info *devinfo,
                                          struct list_head *q, int *counter)
{
        unsigned long flags;
        struct brcmf_usbreq  *req;
        spin_lock_irqsave(&devinfo->qlock, flags);
        if (list_empty(q)) {
                spin_unlock_irqrestore(&devinfo->qlock, flags);
                return NULL;
        }
        req = list_entry(q->next, struct brcmf_usbreq, list);
        list_del_init(q->next);
        if (counter)
                (*counter)--;
        spin_unlock_irqrestore(&devinfo->qlock, flags);
        return req;

}

static void brcmf_usb_enq(struct brcmf_usbdev_info *devinfo,
                          struct list_head *q, struct brcmf_usbreq *req,
                          int *counter)
{
        unsigned long flags;
        spin_lock_irqsave(&devinfo->qlock, flags);
        list_add_tail(&req->list, q);
        if (counter)
                (*counter)++;
        spin_unlock_irqrestore(&devinfo->qlock, flags);
}

static struct brcmf_usbreq *
brcmf_usbdev_qinit(struct list_head *q, int qsize)
{
        int i;
        struct brcmf_usbreq *req, *reqs;

        reqs = kcalloc(qsize, sizeof(struct brcmf_usbreq), GFP_ATOMIC);
        if (reqs == NULL)
                return NULL;

        req = reqs;

        for (i = 0; i < qsize; i++) {
                req->urb = usb_alloc_urb(0, GFP_ATOMIC);
                if (!req->urb)
                        goto fail;

                INIT_LIST_HEAD(&req->list);
                list_add_tail(&req->list, q);
                req++;
        }
        return reqs;
fail:
        brcmf_err("fail!\n");
        while (!list_empty(q)) {
                req = list_entry(q->next, struct brcmf_usbreq, list);
                if (req)
                        usb_free_urb(req->urb);
                list_del(q->next);
        }
        return NULL;

}

static void brcmf_usb_free_q(struct list_head *q)
{
        struct brcmf_usbreq *req, *next;

        list_for_each_entry_safe(req, next, q, list) {
                if (!req->urb) {
                        brcmf_err("bad req\n");
                        break;
                }
                usb_free_urb(req->urb);
                list_del_init(&req->list);
        }
}

static void brcmf_usb_del_fromq(struct brcmf_usbdev_info *devinfo,
                                struct brcmf_usbreq *req)
{
        unsigned long flags;

        spin_lock_irqsave(&devinfo->qlock, flags);
        list_del_init(&req->list);
        spin_unlock_irqrestore(&devinfo->qlock, flags);
}


static void brcmf_usb_tx_complete(struct urb *urb)
{
        struct brcmf_usbreq *req = (struct brcmf_usbreq *)urb->context;
        struct brcmf_usbdev_info *devinfo = req->devinfo;
        unsigned long flags;

        brcmf_dbg(USB, "Enter, urb->status=%d, skb=%p\n", urb->status,
                  req->skb);
        brcmf_usb_del_fromq(devinfo, req);

        brcmf_proto_bcdc_txcomplete(devinfo->dev, req->skb, urb->status == 0);
        req->skb = NULL;
        brcmf_usb_enq(devinfo, &devinfo->tx_freeq, req, &devinfo->tx_freecount);
        spin_lock_irqsave(&devinfo->tx_flowblock_lock, flags);
        if (devinfo->tx_freecount > devinfo->tx_high_watermark &&
                devinfo->tx_flowblock) {
                brcmf_proto_bcdc_txflowblock(devinfo->dev, false);
                devinfo->tx_flowblock = false;
        }
        spin_unlock_irqrestore(&devinfo->tx_flowblock_lock, flags);
}

static void brcmf_usb_rx_complete(struct urb *urb)
{
        struct brcmf_usbreq  *req = (struct brcmf_usbreq *)urb->context;
        struct brcmf_usbdev_info *devinfo = req->devinfo;
        struct sk_buff *skb;

        brcmf_dbg(USB, "Enter, urb->status=%d\n", urb->status);
        brcmf_usb_del_fromq(devinfo, req);
        skb = req->skb;
        req->skb = NULL;

        /* zero length packets indicate usb "failure". Do not refill */
        if (urb->status != 0 || !urb->actual_length) {
                brcmu_pkt_buf_free_skb(skb);
                brcmf_usb_enq(devinfo, &devinfo->rx_freeq, req, NULL);
                return;
        }

        if (devinfo->bus_pub.state == BRCMFMAC_USB_STATE_UP) {
                skb_put(skb, urb->actual_length);
                brcmf_rx_frame(devinfo->dev, skb, true);
                brcmf_usb_rx_refill(devinfo, req);
        } else {
                brcmu_pkt_buf_free_skb(skb);
                brcmf_usb_enq(devinfo, &devinfo->rx_freeq, req, NULL);
        }
        return;

}

static void brcmf_usb_rx_refill(struct brcmf_usbdev_info *devinfo,
                                struct brcmf_usbreq  *req)
{
        struct sk_buff *skb;
        int ret;

        if (!req || !devinfo)
                return;

        skb = dev_alloc_skb(devinfo->bus_pub.bus_mtu);
        if (!skb) {
                brcmf_usb_enq(devinfo, &devinfo->rx_freeq, req, NULL);
                return;
        }
        req->skb = skb;

        usb_fill_bulk_urb(req->urb, devinfo->usbdev, devinfo->rx_pipe,
                          skb->data, skb_tailroom(skb), brcmf_usb_rx_complete,
                          req);
        req->devinfo = devinfo;
        brcmf_usb_enq(devinfo, &devinfo->rx_postq, req, NULL);

        ret = usb_submit_urb(req->urb, GFP_ATOMIC);
        if (ret) {
                brcmf_usb_del_fromq(devinfo, req);
                brcmu_pkt_buf_free_skb(req->skb);
                req->skb = NULL;
                brcmf_usb_enq(devinfo, &devinfo->rx_freeq, req, NULL);
        }
        return;
}

static void brcmf_usb_rx_fill_all(struct brcmf_usbdev_info *devinfo)
{
        struct brcmf_usbreq *req;

        if (devinfo->bus_pub.state != BRCMFMAC_USB_STATE_UP) {
                brcmf_err("bus is not up=%d\n", devinfo->bus_pub.state);
                return;
        }
        while ((req = brcmf_usb_deq(devinfo, &devinfo->rx_freeq, NULL)) != NULL)
                brcmf_usb_rx_refill(devinfo, req);
}

static void
brcmf_usb_state_change(struct brcmf_usbdev_info *devinfo, int state)
{
        struct brcmf_bus *bcmf_bus = devinfo->bus_pub.bus;

        brcmf_dbg(USB, "Enter, current state=%d, new state=%d\n",
                  devinfo->bus_pub.state, state);

        if (devinfo->bus_pub.state == state)
                return;

        devinfo->bus_pub.state = state;

        /* update state of upper layer */
        if (state == BRCMFMAC_USB_STATE_DOWN) {
                brcmf_dbg(USB, "DBUS is down\n");
                brcmf_bus_change_state(bcmf_bus, BRCMF_BUS_DOWN);
        } else if (state == BRCMFMAC_USB_STATE_UP) {
                brcmf_dbg(USB, "DBUS is up\n");
                brcmf_bus_change_state(bcmf_bus, BRCMF_BUS_UP);
        } else {
                brcmf_dbg(USB, "DBUS current state=%d\n", state);
        }
}

static int brcmf_usb_tx(struct device *dev, struct sk_buff *skb)
{
        struct brcmf_usbdev_info *devinfo = brcmf_usb_get_businfo(dev);
        struct brcmf_usbreq  *req;
        int ret;
        unsigned long flags;

        brcmf_dbg(USB, "Enter, skb=%p\n", skb);
        if (devinfo->bus_pub.state != BRCMFMAC_USB_STATE_UP) {
                ret = -EIO;
                goto fail;
        }

        req = brcmf_usb_deq(devinfo, &devinfo->tx_freeq,
                                        &devinfo->tx_freecount);
        if (!req) {
                brcmf_err("no req to send\n");
                ret = -ENOMEM;
                goto fail;
        }

        req->skb = skb;
        req->devinfo = devinfo;
        usb_fill_bulk_urb(req->urb, devinfo->usbdev, devinfo->tx_pipe,
                          skb->data, skb->len, brcmf_usb_tx_complete, req);
        req->urb->transfer_flags |= URB_ZERO_PACKET;
        brcmf_usb_enq(devinfo, &devinfo->tx_postq, req, NULL);
        ret = usb_submit_urb(req->urb, GFP_ATOMIC);
        if (ret) {
                brcmf_err("brcmf_usb_tx usb_submit_urb FAILED\n");
                brcmf_usb_del_fromq(devinfo, req);
                req->skb = NULL;
                brcmf_usb_enq(devinfo, &devinfo->tx_freeq, req,
                              &devinfo->tx_freecount);
                goto fail;
        }

        spin_lock_irqsave(&devinfo->tx_flowblock_lock, flags);
        if (devinfo->tx_freecount < devinfo->tx_low_watermark &&
            !devinfo->tx_flowblock) {
                brcmf_proto_bcdc_txflowblock(dev, true);
                devinfo->tx_flowblock = true;
        }
        spin_unlock_irqrestore(&devinfo->tx_flowblock_lock, flags);
        return 0;

fail:
        return ret;
}


static int brcmf_usb_up(struct device *dev)
{
        struct brcmf_usbdev_info *devinfo = brcmf_usb_get_businfo(dev);

        brcmf_dbg(USB, "Enter\n");
        if (devinfo->bus_pub.state == BRCMFMAC_USB_STATE_UP)
                return 0;

        /* Success, indicate devinfo is fully up */
        brcmf_usb_state_change(devinfo, BRCMFMAC_USB_STATE_UP);

        if (devinfo->ctl_urb) {
                devinfo->ctl_in_pipe = usb_rcvctrlpipe(devinfo->usbdev, 0);
                devinfo->ctl_out_pipe = usb_sndctrlpipe(devinfo->usbdev, 0);

                /* CTL Write */
                devinfo->ctl_write.bRequestType =
                        USB_DIR_OUT | USB_TYPE_CLASS | USB_RECIP_INTERFACE;
                devinfo->ctl_write.bRequest = 0;
                devinfo->ctl_write.wValue = cpu_to_le16(0);
                devinfo->ctl_write.wIndex = cpu_to_le16(devinfo->ifnum);

                /* CTL Read */
                devinfo->ctl_read.bRequestType =
                        USB_DIR_IN | USB_TYPE_CLASS | USB_RECIP_INTERFACE;
                devinfo->ctl_read.bRequest = 1;
                devinfo->ctl_read.wValue = cpu_to_le16(0);
                devinfo->ctl_read.wIndex = cpu_to_le16(devinfo->ifnum);
        }
        brcmf_usb_rx_fill_all(devinfo);
        return 0;
}

static void brcmf_cancel_all_urbs(struct brcmf_usbdev_info *devinfo)
{
        int i;

        if (devinfo->ctl_urb)
                usb_kill_urb(devinfo->ctl_urb);
        if (devinfo->bulk_urb)
                usb_kill_urb(devinfo->bulk_urb);
        if (devinfo->tx_reqs)
                for (i = 0; i < devinfo->bus_pub.ntxq; i++)
                        usb_kill_urb(devinfo->tx_reqs[i].urb);
        if (devinfo->rx_reqs)
                for (i = 0; i < devinfo->bus_pub.nrxq; i++)
                        usb_kill_urb(devinfo->rx_reqs[i].urb);
}

static void brcmf_usb_down(struct device *dev)
{
        struct brcmf_usbdev_info *devinfo = brcmf_usb_get_businfo(dev);

        brcmf_dbg(USB, "Enter\n");
        if (devinfo == NULL)
                return;

        if (devinfo->bus_pub.state == BRCMFMAC_USB_STATE_DOWN)
                return;

        brcmf_usb_state_change(devinfo, BRCMFMAC_USB_STATE_DOWN);

        brcmf_cancel_all_urbs(devinfo);
}

static void
brcmf_usb_sync_complete(struct urb *urb)
{
        struct brcmf_usbdev_info *devinfo =
                        (struct brcmf_usbdev_info *)urb->context;

        devinfo->ctl_completed = true;
        brcmf_usb_ioctl_resp_wake(devinfo);
}

static int brcmf_usb_dl_cmd(struct brcmf_usbdev_info *devinfo, u8 cmd,
                            void *buffer, int buflen)
{
        int ret;
        char *tmpbuf;
        u16 size;

        if ((!devinfo) || (devinfo->ctl_urb == NULL))
                return -EINVAL;

        tmpbuf = kmalloc(buflen, GFP_ATOMIC);
        if (!tmpbuf)
                return -ENOMEM;

        size = buflen;
        devinfo->ctl_urb->transfer_buffer_length = size;

        devinfo->ctl_read.wLength = cpu_to_le16p(&size);
        devinfo->ctl_read.bRequestType = USB_DIR_IN | USB_TYPE_VENDOR |
                USB_RECIP_INTERFACE;
        devinfo->ctl_read.bRequest = cmd;

        usb_fill_control_urb(devinfo->ctl_urb,
                devinfo->usbdev,
                usb_rcvctrlpipe(devinfo->usbdev, 0),
                (unsigned char *) &devinfo->ctl_read,
                (void *) tmpbuf, size,
                (usb_complete_t)brcmf_usb_sync_complete, devinfo);

        devinfo->ctl_completed = false;
        ret = usb_submit_urb(devinfo->ctl_urb, GFP_ATOMIC);
        if (ret < 0) {
                brcmf_err("usb_submit_urb failed %d\n", ret);
                goto finalize;
        }

        if (!brcmf_usb_ioctl_resp_wait(devinfo)) {
                usb_kill_urb(devinfo->ctl_urb);
                ret = -ETIMEDOUT;
        } else {
                memcpy(buffer, tmpbuf, buflen);
        }

finalize:
        kfree(tmpbuf);
        return ret;
}

static bool
brcmf_usb_dlneeded(struct brcmf_usbdev_info *devinfo)
{
        struct bootrom_id_le id;
        u32 chipid, chiprev;

        brcmf_dbg(USB, "Enter\n");

        if (devinfo == NULL)
                return false;

        /* Check if firmware downloaded already by querying runtime ID */
        id.chip = cpu_to_le32(0xDEAD);
        brcmf_usb_dl_cmd(devinfo, DL_GETVER, &id, sizeof(id));

        chipid = le32_to_cpu(id.chip);
        chiprev = le32_to_cpu(id.chiprev);

        if ((chipid & 0x4300) == 0x4300)
                brcmf_dbg(USB, "chip %x rev 0x%x\n", chipid, chiprev);
        else
                brcmf_dbg(USB, "chip %d rev 0x%x\n", chipid, chiprev);
        if (chipid == BRCMF_POSTBOOT_ID) {
                brcmf_dbg(USB, "firmware already downloaded\n");
                brcmf_usb_dl_cmd(devinfo, DL_RESETCFG, &id, sizeof(id));
                return false;
        } else {
                devinfo->bus_pub.devid = chipid;
                devinfo->bus_pub.chiprev = chiprev;
        }
        return true;
}

static int
brcmf_usb_resetcfg(struct brcmf_usbdev_info *devinfo)
{
        struct bootrom_id_le id;
        u32 loop_cnt;
        int err;

        brcmf_dbg(USB, "Enter\n");

        loop_cnt = 0;
        do {
                mdelay(BRCMF_USB_RESET_GETVER_SPINWAIT);
                loop_cnt++;
                id.chip = cpu_to_le32(0xDEAD);       /* Get the ID */
                err = brcmf_usb_dl_cmd(devinfo, DL_GETVER, &id, sizeof(id));
                if ((err) && (err != -ETIMEDOUT))
                        return err;
                if (id.chip == cpu_to_le32(BRCMF_POSTBOOT_ID))
                        break;
        } while (loop_cnt < BRCMF_USB_RESET_GETVER_LOOP_CNT);

        if (id.chip == cpu_to_le32(BRCMF_POSTBOOT_ID)) {
                brcmf_dbg(USB, "postboot chip 0x%x/rev 0x%x\n",
                          le32_to_cpu(id.chip), le32_to_cpu(id.chiprev));

                brcmf_usb_dl_cmd(devinfo, DL_RESETCFG, &id, sizeof(id));
                return 0;
        } else {
                brcmf_err("Cannot talk to Dongle. Firmware is not UP, %d ms\n",
                          BRCMF_USB_RESET_GETVER_SPINWAIT * loop_cnt);
                return -EINVAL;
        }
}


static int
brcmf_usb_dl_send_bulk(struct brcmf_usbdev_info *devinfo, void *buffer, int len)
{
        int ret;

        if ((devinfo == NULL) || (devinfo->bulk_urb == NULL))
                return -EINVAL;

        /* Prepare the URB */
        usb_fill_bulk_urb(devinfo->bulk_urb, devinfo->usbdev,
                          devinfo->tx_pipe, buffer, len,
                          (usb_complete_t)brcmf_usb_sync_complete, devinfo);

        devinfo->bulk_urb->transfer_flags |= URB_ZERO_PACKET;

        devinfo->ctl_completed = false;
        ret = usb_submit_urb(devinfo->bulk_urb, GFP_ATOMIC);
        if (ret) {
                brcmf_err("usb_submit_urb failed %d\n", ret);
                return ret;
        }
        ret = brcmf_usb_ioctl_resp_wait(devinfo);
        return (ret == 0);
}

static int
brcmf_usb_dl_writeimage(struct brcmf_usbdev_info *devinfo, u8 *fw, int fwlen)
{
        unsigned int sendlen, sent, dllen;
        char *bulkchunk = NULL, *dlpos;
        struct rdl_state_le state;
        u32 rdlstate, rdlbytes;
        int err = 0;

        brcmf_dbg(USB, "Enter, fw %p, len %d\n", fw, fwlen);

        bulkchunk = kmalloc(TRX_RDL_CHUNK, GFP_ATOMIC);
        if (bulkchunk == NULL) {
                err = -ENOMEM;
                goto fail;
        }

        /* 1) Prepare USB boot loader for runtime image */
        brcmf_usb_dl_cmd(devinfo, DL_START, &state, sizeof(state));

        rdlstate = le32_to_cpu(state.state);
        rdlbytes = le32_to_cpu(state.bytes);

        /* 2) Check we are in the Waiting state */
        if (rdlstate != DL_WAITING) {
                brcmf_err("Failed to DL_START\n");
                err = -EINVAL;
                goto fail;
        }
        sent = 0;
        dlpos = fw;
        dllen = fwlen;

        /* Get chip id and rev */
        while (rdlbytes != dllen) {
                /* Wait until the usb device reports it received all
                 * the bytes we sent */
                if ((rdlbytes == sent) && (rdlbytes != dllen)) {
                        if ((dllen-sent) < TRX_RDL_CHUNK)
                                sendlen = dllen-sent;
                        else
                                sendlen = TRX_RDL_CHUNK;

                        /* simply avoid having to send a ZLP by ensuring we
                         * never have an even
                         * multiple of 64
                         */
                        if (!(sendlen % 64))
                                sendlen -= 4;

                        /* send data */
                        memcpy(bulkchunk, dlpos, sendlen);
                        if (brcmf_usb_dl_send_bulk(devinfo, bulkchunk,
                                                   sendlen)) {
                                brcmf_err("send_bulk failed\n");
                                err = -EINVAL;
                                goto fail;
                        }

                        dlpos += sendlen;
                        sent += sendlen;
                }
                err = brcmf_usb_dl_cmd(devinfo, DL_GETSTATE, &state,
                                       sizeof(state));
                if (err) {
                        brcmf_err("DL_GETSTATE Failed\n");
                        goto fail;
                }

                rdlstate = le32_to_cpu(state.state);
                rdlbytes = le32_to_cpu(state.bytes);

                /* restart if an error is reported */
                if (rdlstate == DL_BAD_HDR || rdlstate == DL_BAD_CRC) {
                        brcmf_err("Bad Hdr or Bad CRC state %d\n",
                                  rdlstate);
                        err = -EINVAL;
                        goto fail;
                }
        }

fail:
        kfree(bulkchunk);
        brcmf_dbg(USB, "Exit, err=%d\n", err);
        return err;
}

static int brcmf_usb_dlstart(struct brcmf_usbdev_info *devinfo, u8 *fw, int len)
{
        int err;

        brcmf_dbg(USB, "Enter\n");

        if (devinfo == NULL)
                return -EINVAL;

        if (devinfo->bus_pub.devid == 0xDEAD)
                return -EINVAL;

        err = brcmf_usb_dl_writeimage(devinfo, fw, len);
        if (err == 0)
                devinfo->bus_pub.state = BRCMFMAC_USB_STATE_DL_DONE;
        else
                devinfo->bus_pub.state = BRCMFMAC_USB_STATE_DL_FAIL;
        brcmf_dbg(USB, "Exit, err=%d\n", err);

        return err;
}

static int brcmf_usb_dlrun(struct brcmf_usbdev_info *devinfo)
{
        struct rdl_state_le state;

        brcmf_dbg(USB, "Enter\n");
        if (!devinfo)
                return -EINVAL;

        if (devinfo->bus_pub.devid == 0xDEAD)
                return -EINVAL;

        /* Check we are runnable */
        state.state = 0;
        brcmf_usb_dl_cmd(devinfo, DL_GETSTATE, &state, sizeof(state));

        /* Start the image */
        if (state.state == cpu_to_le32(DL_RUNNABLE)) {
                if (brcmf_usb_dl_cmd(devinfo, DL_GO, &state, sizeof(state)))
                        return -ENODEV;
                if (brcmf_usb_resetcfg(devinfo))
                        return -ENODEV;
                /* The Dongle may go for re-enumeration. */
        } else {
                brcmf_err("Dongle not runnable\n");
                return -EINVAL;
        }
        brcmf_dbg(USB, "Exit\n");
        return 0;
}

static int
brcmf_usb_fw_download(struct brcmf_usbdev_info *devinfo)
{
        int err;

        brcmf_dbg(USB, "Enter\n");
        if (devinfo == NULL)
                return -ENODEV;

        if (!devinfo->image) {
                brcmf_err("No firmware!\n");
                return -ENOENT;

        }

        err = brcmf_usb_dlstart(devinfo,
                (u8 *)devinfo->image, devinfo->image_len);
        if (err == 0)
                err = brcmf_usb_dlrun(devinfo);
        return err;
}


static void brcmf_usb_detach(struct brcmf_usbdev_info *devinfo)
{
        brcmf_dbg(USB, "Enter, devinfo %p\n", devinfo);

        /* free the URBS */
        brcmf_usb_free_q(&devinfo->rx_freeq);
        brcmf_usb_free_q(&devinfo->tx_freeq);

        usb_free_urb(devinfo->ctl_urb);
        usb_free_urb(devinfo->bulk_urb);

        kfree(devinfo->tx_reqs);
        kfree(devinfo->rx_reqs);

        if (devinfo->settings)
                brcmf_release_module_param(devinfo->settings);
}


static int check_file(const u8 *headers)
{
        struct trx_header_le *trx;
        int actual_len = -1;

        brcmf_dbg(USB, "Enter\n");
        /* Extract trx header */
        trx = (struct trx_header_le *) headers;
        if (trx->magic != cpu_to_le32(TRX_MAGIC))
                return -1;

        headers += sizeof(struct trx_header_le);

        if (le32_to_cpu(trx->flag_version) & TRX_UNCOMP_IMAGE) {
                actual_len = le32_to_cpu(trx->offsets[TRX_OFFSETS_DLFWLEN_IDX]);
                return actual_len + sizeof(struct trx_header_le);
        }
        return -1;
}


static
struct brcmf_usbdev *brcmf_usb_attach(struct brcmf_usbdev_info *devinfo,
                                      int nrxq, int ntxq)
{
        brcmf_dbg(USB, "Enter\n");

        devinfo->bus_pub.nrxq = nrxq;
        devinfo->rx_low_watermark = nrxq / 2;
        devinfo->bus_pub.devinfo = devinfo;
        devinfo->bus_pub.ntxq = ntxq;
        devinfo->bus_pub.state = BRCMFMAC_USB_STATE_DOWN;

        /* flow control when too many tx urbs posted */
        devinfo->tx_low_watermark = ntxq / 4;
        devinfo->tx_high_watermark = devinfo->tx_low_watermark * 3;
        devinfo->bus_pub.bus_mtu = BRCMF_USB_MAX_PKT_SIZE;

        /* Initialize other structure content */
        init_waitqueue_head(&devinfo->ioctl_resp_wait);

        /* Initialize the spinlocks */
        spin_lock_init(&devinfo->qlock);
        spin_lock_init(&devinfo->tx_flowblock_lock);

        INIT_LIST_HEAD(&devinfo->rx_freeq);
        INIT_LIST_HEAD(&devinfo->rx_postq);

        INIT_LIST_HEAD(&devinfo->tx_freeq);
        INIT_LIST_HEAD(&devinfo->tx_postq);

        devinfo->tx_flowblock = false;

        devinfo->rx_reqs = brcmf_usbdev_qinit(&devinfo->rx_freeq, nrxq);
        if (!devinfo->rx_reqs)
                goto error;

        devinfo->tx_reqs = brcmf_usbdev_qinit(&devinfo->tx_freeq, ntxq);
        if (!devinfo->tx_reqs)
                goto error;
        devinfo->tx_freecount = ntxq;

        devinfo->ctl_urb = usb_alloc_urb(0, GFP_ATOMIC);
        if (!devinfo->ctl_urb)
                goto error;
        devinfo->bulk_urb = usb_alloc_urb(0, GFP_ATOMIC);
        if (!devinfo->bulk_urb)
                goto error;

        return &devinfo->bus_pub;

error:
        brcmf_err("failed!\n");
        brcmf_usb_detach(devinfo);
        return NULL;
}

static void brcmf_usb_wowl_config(struct device *dev, bool enabled)
{
        struct brcmf_usbdev_info *devinfo = brcmf_usb_get_businfo(dev);

        brcmf_dbg(USB, "Configuring WOWL, enabled=%d\n", enabled);
        devinfo->wowl_enabled = enabled;
        if (enabled)
                device_set_wakeup_enable(devinfo->dev, true);
        else
                device_set_wakeup_enable(devinfo->dev, false);
}

static
int brcmf_usb_get_fwname(struct device *dev, const char *ext, u8 *fw_name)
{
        struct brcmf_bus *bus = dev_get_drvdata(dev);
        struct brcmf_fw_request *fwreq;
        struct brcmf_fw_name fwnames[] = {
                { ext, fw_name },
        };

        fwreq = brcmf_fw_alloc_request(bus->chip, bus->chiprev,
                                       brcmf_usb_fwnames,
                                       ARRAY_SIZE(brcmf_usb_fwnames),
                                       fwnames, ARRAY_SIZE(fwnames));
        if (!fwreq)
                return -ENOMEM;

        kfree(fwreq);
        return 0;
}

static const struct brcmf_bus_ops brcmf_usb_bus_ops = {
        .preinit = brcmf_usb_up,
        .stop = brcmf_usb_down,
        .txdata = brcmf_usb_tx,
        .txctl = brcmf_usb_tx_ctlpkt,
        .rxctl = brcmf_usb_rx_ctlpkt,
        .wowl_config = brcmf_usb_wowl_config,
        .get_fwname = brcmf_usb_get_fwname,
};

#define BRCMF_USB_FW_CODE       0

static void brcmf_usb_probe_phase2(struct device *dev, int ret,
                                   struct brcmf_fw_request *fwreq)
{
        struct brcmf_bus *bus = dev_get_drvdata(dev);
        struct brcmf_usbdev_info *devinfo = bus->bus_priv.usb->devinfo;
        const struct firmware *fw;

        if (ret)
                goto error;

        brcmf_dbg(USB, "Start fw downloading\n");

        fw = fwreq->items[BRCMF_USB_FW_CODE].binary;
        kfree(fwreq);

        ret = check_file(fw->data);
        if (ret < 0) {
                brcmf_err("invalid firmware\n");
                release_firmware(fw);
                goto error;
        }

        devinfo->image = fw->data;
        devinfo->image_len = fw->size;

        ret = brcmf_usb_fw_download(devinfo);
        release_firmware(fw);
        if (ret)
                goto error;

        ret = brcmf_alloc(devinfo->dev, devinfo->settings);
        if (ret)
                goto error;

        /* Attach to the common driver interface */
        ret = brcmf_attach(devinfo->dev);
        if (ret)
                goto error;

        complete(&devinfo->dev_init_done);
        return;
error:
        brcmf_dbg(TRACE, "failed: dev=%s, err=%d\n", dev_name(dev), ret);
        complete(&devinfo->dev_init_done);
        device_release_driver(dev);
}

static struct brcmf_fw_request *
brcmf_usb_prepare_fw_request(struct brcmf_usbdev_info *devinfo)
{
        struct brcmf_fw_request *fwreq;
        struct brcmf_fw_name fwnames[] = {
                { ".bin", devinfo->fw_name },
        };

        fwreq = brcmf_fw_alloc_request(devinfo->bus_pub.devid,
                                       devinfo->bus_pub.chiprev,
                                       brcmf_usb_fwnames,
                                       ARRAY_SIZE(brcmf_usb_fwnames),
                                       fwnames, ARRAY_SIZE(fwnames));
        if (!fwreq)
                return NULL;

        fwreq->items[BRCMF_USB_FW_CODE].type = BRCMF_FW_TYPE_BINARY;

        return fwreq;
}

static int brcmf_usb_probe_cb(struct brcmf_usbdev_info *devinfo)
{
        struct brcmf_bus *bus = NULL;
        struct brcmf_usbdev *bus_pub = NULL;
        struct device *dev = devinfo->dev;
        struct brcmf_fw_request *fwreq;
        int ret;

        brcmf_dbg(USB, "Enter\n");
        bus_pub = brcmf_usb_attach(devinfo, BRCMF_USB_NRXQ, BRCMF_USB_NTXQ);
        if (!bus_pub)
                return -ENODEV;

        bus = kzalloc(sizeof(struct brcmf_bus), GFP_ATOMIC);
        if (!bus) {
                ret = -ENOMEM;
                goto fail;
        }

        bus->dev = dev;
        bus_pub->bus = bus;
        bus->bus_priv.usb = bus_pub;
        dev_set_drvdata(dev, bus);
        bus->ops = &brcmf_usb_bus_ops;
        bus->proto_type = BRCMF_PROTO_BCDC;
        bus->always_use_fws_queue = true;
#ifdef CONFIG_PM
        bus->wowl_supported = true;
#endif

        devinfo->settings = brcmf_get_module_param(bus->dev, BRCMF_BUSTYPE_USB,
                                                   bus_pub->devid,
                                                   bus_pub->chiprev);
        if (!devinfo->settings) {
                ret = -ENOMEM;
                goto fail;
        }

        if (!brcmf_usb_dlneeded(devinfo)) {
                ret = brcmf_alloc(devinfo->dev, devinfo->settings);
                if (ret)
                        goto fail;
                ret = brcmf_attach(devinfo->dev);
                if (ret)
                        goto fail;
                /* we are done */
                complete(&devinfo->dev_init_done);
                return 0;
        }
        bus->chip = bus_pub->devid;
        bus->chiprev = bus_pub->chiprev;

        fwreq = brcmf_usb_prepare_fw_request(devinfo);
        if (!fwreq) {
                ret = -ENOMEM;
                goto fail;
        }

        /* request firmware here */
        ret = brcmf_fw_get_firmwares(dev, fwreq, brcmf_usb_probe_phase2);
        if (ret) {
                brcmf_err("firmware request failed: %d\n", ret);
                kfree(fwreq);
                goto fail;
        }

        return 0;

fail:
        /* Release resources in reverse order */
        brcmf_free(devinfo->dev);
        kfree(bus);
        brcmf_usb_detach(devinfo);
        return ret;
}

static void
brcmf_usb_disconnect_cb(struct brcmf_usbdev_info *devinfo)
{
        if (!devinfo)
                return;
        brcmf_dbg(USB, "Enter, bus_pub %p\n", devinfo);

        brcmf_detach(devinfo->dev);
        brcmf_free(devinfo->dev);
        kfree(devinfo->bus_pub.bus);
        brcmf_usb_detach(devinfo);
}

static int
brcmf_usb_probe(struct usb_interface *intf, const struct usb_device_id *id)
{
        struct usb_device *usb = interface_to_usbdev(intf);
        struct brcmf_usbdev_info *devinfo;
        struct usb_interface_descriptor *desc;
        struct usb_endpoint_descriptor *endpoint;
        int ret = 0;
        u32 num_of_eps;
        u8 endpoint_num, ep;

        brcmf_dbg(USB, "Enter 0x%04x:0x%04x\n", id->idVendor, id->idProduct);

        devinfo = kzalloc(sizeof(*devinfo), GFP_ATOMIC);
        if (devinfo == NULL)
                return -ENOMEM;

        devinfo->usbdev = usb;
        devinfo->dev = &usb->dev;
        /* Init completion, to protect for disconnect while still loading.
         * Necessary because of the asynchronous firmware load construction
         */
        init_completion(&devinfo->dev_init_done);

        usb_set_intfdata(intf, devinfo);

        /* Check that the device supports only one configuration */
        if (usb->descriptor.bNumConfigurations != 1) {
                brcmf_err("Number of configurations: %d not supported\n",
                          usb->descriptor.bNumConfigurations);
                ret = -ENODEV;
                goto fail;
        }

        if ((usb->descriptor.bDeviceClass != USB_CLASS_VENDOR_SPEC) &&
            (usb->descriptor.bDeviceClass != USB_CLASS_MISC) &&
            (usb->descriptor.bDeviceClass != USB_CLASS_WIRELESS_CONTROLLER)) {
                brcmf_err("Device class: 0x%x not supported\n",
                          usb->descriptor.bDeviceClass);
                ret = -ENODEV;
                goto fail;
        }

        desc = &intf->altsetting[0].desc;
        if ((desc->bInterfaceClass != USB_CLASS_VENDOR_SPEC) ||
            (desc->bInterfaceSubClass != 2) ||
            (desc->bInterfaceProtocol != 0xff)) {
                brcmf_err("non WLAN interface %d: 0x%x:0x%x:0x%x\n",
                          desc->bInterfaceNumber, desc->bInterfaceClass,
                          desc->bInterfaceSubClass, desc->bInterfaceProtocol);
                ret = -ENODEV;
                goto fail;
        }

        num_of_eps = desc->bNumEndpoints;
        for (ep = 0; ep < num_of_eps; ep++) {
                endpoint = &intf->altsetting[0].endpoint[ep].desc;
                endpoint_num = usb_endpoint_num(endpoint);
                if (!usb_endpoint_xfer_bulk(endpoint))
                        continue;
                if (usb_endpoint_dir_in(endpoint)) {
                        if (!devinfo->rx_pipe)
                                devinfo->rx_pipe =
                                        usb_rcvbulkpipe(usb, endpoint_num);
                } else {
                        if (!devinfo->tx_pipe)
                                devinfo->tx_pipe =
                                        usb_sndbulkpipe(usb, endpoint_num);
                }
        }
        if (devinfo->rx_pipe == 0) {
                brcmf_err("No RX (in) Bulk EP found\n");
                ret = -ENODEV;
                goto fail;
        }
        if (devinfo->tx_pipe == 0) {
                brcmf_err("No TX (out) Bulk EP found\n");
                ret = -ENODEV;
                goto fail;
        }

        devinfo->ifnum = desc->bInterfaceNumber;

        if (usb->speed == USB_SPEED_SUPER_PLUS)
                brcmf_dbg(USB, "Broadcom super speed plus USB WLAN interface detected\n");
        else if (usb->speed == USB_SPEED_SUPER)
                brcmf_dbg(USB, "Broadcom super speed USB WLAN interface detected\n");
        else if (usb->speed == USB_SPEED_HIGH)
                brcmf_dbg(USB, "Broadcom high speed USB WLAN interface detected\n");
        else
                brcmf_dbg(USB, "Broadcom full speed USB WLAN interface detected\n");

        ret = brcmf_usb_probe_cb(devinfo);
        if (ret)
                goto fail;

        /* Success */
        return 0;

fail:
        complete(&devinfo->dev_init_done);
        kfree(devinfo);
        usb_set_intfdata(intf, NULL);
        return ret;
}

static void
brcmf_usb_disconnect(struct usb_interface *intf)
{
        struct brcmf_usbdev_info *devinfo;

        brcmf_dbg(USB, "Enter\n");
        devinfo = (struct brcmf_usbdev_info *)usb_get_intfdata(intf);

        if (devinfo) {
                wait_for_completion(&devinfo->dev_init_done);
                /* Make sure that devinfo still exists. Firmware probe routines
                 * may have released the device and cleared the intfdata.
                 */
                if (!usb_get_intfdata(intf))
                        goto done;

                brcmf_usb_disconnect_cb(devinfo);
                kfree(devinfo);
        }
done:
        brcmf_dbg(USB, "Exit\n");
}

/*
 * only need to signal the bus being down and update the state.
 */
static int brcmf_usb_suspend(struct usb_interface *intf, pm_message_t state)
{
        struct usb_device *usb = interface_to_usbdev(intf);
        struct brcmf_usbdev_info *devinfo = brcmf_usb_get_businfo(&usb->dev);

        brcmf_dbg(USB, "Enter\n");
        devinfo->bus_pub.state = BRCMFMAC_USB_STATE_SLEEP;
        if (devinfo->wowl_enabled) {
                brcmf_cancel_all_urbs(devinfo);
        } else {
                brcmf_detach(&usb->dev);
                brcmf_free(&usb->dev);
        }
        return 0;
}

/*
 * (re-) start the bus.
 */
static int brcmf_usb_resume(struct usb_interface *intf)
{
        struct usb_device *usb = interface_to_usbdev(intf);
        struct brcmf_usbdev_info *devinfo = brcmf_usb_get_businfo(&usb->dev);

        brcmf_dbg(USB, "Enter\n");
        if (!devinfo->wowl_enabled) {
                int err;

                err = brcmf_alloc(&usb->dev, devinfo->settings);
                if (err)
                        return err;

                err = brcmf_attach(devinfo->dev);
                if (err) {
                        brcmf_free(devinfo->dev);
                        return err;
                }
        }

        devinfo->bus_pub.state = BRCMFMAC_USB_STATE_UP;
        brcmf_usb_rx_fill_all(devinfo);
        return 0;
}

static int brcmf_usb_reset_resume(struct usb_interface *intf)
{
        struct usb_device *usb = interface_to_usbdev(intf);
        struct brcmf_usbdev_info *devinfo = brcmf_usb_get_businfo(&usb->dev);
        struct brcmf_fw_request *fwreq;
        int ret;

        brcmf_dbg(USB, "Enter\n");

        fwreq = brcmf_usb_prepare_fw_request(devinfo);
        if (!fwreq)
                return -ENOMEM;

        ret = brcmf_fw_get_firmwares(&usb->dev, fwreq, brcmf_usb_probe_phase2);
        if (ret < 0)
                kfree(fwreq);

        return ret;
}

#define BRCMF_USB_DEVICE(dev_id)        \
        { USB_DEVICE(BRCM_USB_VENDOR_ID_BROADCOM, dev_id) }

#define LINKSYS_USB_DEVICE(dev_id)      \
        { USB_DEVICE(BRCM_USB_VENDOR_ID_LINKSYS, dev_id) }

#define CYPRESS_USB_DEVICE(dev_id)      \
        { USB_DEVICE(CY_USB_VENDOR_ID_CYPRESS, dev_id) }

static const struct usb_device_id brcmf_usb_devid_table[] = {
        BRCMF_USB_DEVICE(BRCM_USB_43143_DEVICE_ID),
        BRCMF_USB_DEVICE(BRCM_USB_43236_DEVICE_ID),
        BRCMF_USB_DEVICE(BRCM_USB_43242_DEVICE_ID),
        BRCMF_USB_DEVICE(BRCM_USB_43569_DEVICE_ID),
        LINKSYS_USB_DEVICE(BRCM_USB_43235_LINKSYS_DEVICE_ID),
        CYPRESS_USB_DEVICE(CY_USB_4373_DEVICE_ID),
        { USB_DEVICE(BRCM_USB_VENDOR_ID_LG, BRCM_USB_43242_LG_DEVICE_ID) },
        /* special entry for device with firmware loaded and running */
        BRCMF_USB_DEVICE(BRCM_USB_BCMFW_DEVICE_ID),
        CYPRESS_USB_DEVICE(BRCM_USB_BCMFW_DEVICE_ID),
        { /* end: all zeroes */ }
};

MODULE_DEVICE_TABLE(usb, brcmf_usb_devid_table);

static struct usb_driver brcmf_usbdrvr = {
        .name = KBUILD_MODNAME,
        .probe = brcmf_usb_probe,
        .disconnect = brcmf_usb_disconnect,
        .id_table = brcmf_usb_devid_table,
        .suspend = brcmf_usb_suspend,
        .resume = brcmf_usb_resume,
        .reset_resume = brcmf_usb_reset_resume,
        .disable_hub_initiated_lpm = 1,
};

static int brcmf_usb_reset_device(struct device *dev, void *notused)
{
        /* device past is the usb interface so we
         * need to use parent here.
         */
        brcmf_dev_reset(dev->parent);
        return 0;
}

void brcmf_usb_exit(void)
{
        struct device_driver *drv = &brcmf_usbdrvr.drvwrap.driver;
        int ret;

        brcmf_dbg(USB, "Enter\n");
        ret = driver_for_each_device(drv, NULL, NULL,
                                     brcmf_usb_reset_device);
        usb_deregister(&brcmf_usbdrvr);
}

void brcmf_usb_register(void)
{
        int ret;

        brcmf_dbg(USB, "Enter\n");
        ret = usb_register(&brcmf_usbdrvr);
        if (ret)
                brcmf_err("usb_register failed %d\n", ret);
}