/*
 * Copyright (c) 2014 Redpine Signals Inc.
 *
 * Permission to use, copy, modify, and/or distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 *
 */

#include <linux/module.h>
#include <linux/types.h>
#include <net/rsi_91x.h>
#include "rsi_usb.h"
#include "rsi_hal.h"
#include "rsi_coex.h"

/* Default operating mode is wlan STA + BT */
static u16 dev_oper_mode = DEV_OPMODE_STA_BT_DUAL;
module_param(dev_oper_mode, ushort, 0444);
MODULE_PARM_DESC(dev_oper_mode, DEV_OPMODE_PARAM_DESC);

static int rsi_rx_urb_submit(struct rsi_hw *adapter, u8 ep_num, gfp_t flags);

/**
 * rsi_usb_card_write() - This function writes to the USB Card.
 * @adapter: Pointer to the adapter structure.
 * @buf: Pointer to the buffer from where the data has to be taken.
 * @len: Length to be written.
 * @endpoint: Type of endpoint.
 *
 * Return: status: 0 on success, a negative error code on failure.
 */
static int rsi_usb_card_write(struct rsi_hw *adapter,
                              u8 *buf,
                              u16 len,
                              u8 endpoint)
{
        struct rsi_91x_usbdev *dev = adapter->rsi_dev;
        int status;
        u8 *seg = dev->tx_buffer;
        int transfer;
        int ep = dev->bulkout_endpoint_addr[endpoint - 1];

        memset(seg, 0, len + RSI_USB_TX_HEAD_ROOM);
        memcpy(seg + RSI_USB_TX_HEAD_ROOM, buf, len);
        len += RSI_USB_TX_HEAD_ROOM;
        transfer = len;
        status = usb_bulk_msg(dev->usbdev,
                              usb_sndbulkpipe(dev->usbdev, ep),
                              (void *)seg,
                              (int)len,
                              &transfer,
                              USB_CTRL_SET_TIMEOUT);

        if (status < 0) {
                rsi_dbg(ERR_ZONE,
                        "Card write failed with error code :%10d\n", status);
                dev->write_fail = 1;
        }
        return status;
}

/**
 * rsi_write_multiple() - This function writes multiple bytes of information
 *                        to the USB card.
 * @adapter: Pointer to the adapter structure.
 * @endpoint: Type of endpoint.
 * @data: Pointer to the data that has to be written.
 * @count: Number of multiple bytes to be written.
 *
 * Return: 0 on success, a negative error code on failure.
 */
static int rsi_write_multiple(struct rsi_hw *adapter,
                              u8 endpoint,
                              u8 *data,
                              u32 count)
{
        struct rsi_91x_usbdev *dev;

        if (!adapter)
                return -ENODEV;

        if (endpoint == 0)
                return -EINVAL;

        dev = adapter->rsi_dev;
        if (dev->write_fail)
                return -ENETDOWN;

        return rsi_usb_card_write(adapter, data, count, endpoint);
}

/**
 * rsi_find_bulk_in_and_out_endpoints() - This function initializes the bulk
 *                                        endpoints to the device.
 * @interface: Pointer to the USB interface structure.
 * @adapter: Pointer to the adapter structure.
 *
 * Return: ret_val: 0 on success, -ENOMEM on failure.
 */
static int rsi_find_bulk_in_and_out_endpoints(struct usb_interface *interface,
                                              struct rsi_hw *adapter)
{
        struct rsi_91x_usbdev *dev = adapter->rsi_dev;
        struct usb_host_interface *iface_desc;
        struct usb_endpoint_descriptor *endpoint;
        __le16 buffer_size;
        int ii, bin_found = 0, bout_found = 0;

        iface_desc = interface->cur_altsetting;

        for (ii = 0; ii < iface_desc->desc.bNumEndpoints; ++ii) {
                endpoint = &(iface_desc->endpoint[ii].desc);

                if (!dev->bulkin_endpoint_addr[bin_found] &&
                    (endpoint->bEndpointAddress & USB_DIR_IN) &&
                    ((endpoint->bmAttributes & USB_ENDPOINT_XFERTYPE_MASK) ==
                    USB_ENDPOINT_XFER_BULK)) {
                        buffer_size = endpoint->wMaxPacketSize;
                        dev->bulkin_size[bin_found] = buffer_size;
                        dev->bulkin_endpoint_addr[bin_found] =
                                endpoint->bEndpointAddress;
                        bin_found++;
                }

                if (!dev->bulkout_endpoint_addr[bout_found] &&
                    !(endpoint->bEndpointAddress & USB_DIR_IN) &&
                    ((endpoint->bmAttributes & USB_ENDPOINT_XFERTYPE_MASK) ==
                    USB_ENDPOINT_XFER_BULK)) {
                        buffer_size = endpoint->wMaxPacketSize;
                        dev->bulkout_endpoint_addr[bout_found] =
                                endpoint->bEndpointAddress;
                        dev->bulkout_size[bout_found] = buffer_size;
                        bout_found++;
                }

                if (bin_found >= MAX_BULK_EP || bout_found >= MAX_BULK_EP)
                        break;
        }

        if (!(dev->bulkin_endpoint_addr[0] && dev->bulkout_endpoint_addr[0])) {
                dev_err(&interface->dev, "missing wlan bulk endpoints\n");
                return -EINVAL;
        }

        if (adapter->priv->coex_mode > 1) {
                if (!dev->bulkin_endpoint_addr[1]) {
                        dev_err(&interface->dev, "missing bt bulk-in endpoint\n");
                        return -EINVAL;
                }
        }

        return 0;
}

#define RSI_USB_REQ_OUT (USB_TYPE_VENDOR | USB_DIR_OUT | USB_RECIP_DEVICE)
#define RSI_USB_REQ_IN  (USB_TYPE_VENDOR | USB_DIR_IN | USB_RECIP_DEVICE)

/* rsi_usb_reg_read() - This function reads data from given register address.
 * @usbdev: Pointer to the usb_device structure.
 * @reg: Address of the register to be read.
 * @value: Value to be read.
 * @len: length of data to be read.
 *
 * Return: status: 0 on success, a negative error code on failure.
 */
static int rsi_usb_reg_read(struct usb_device *usbdev,
                            u32 reg,
                            u16 *value,
                            u16 len)
{
        u8 *buf;
        int status = -ENOMEM;

        if (len > RSI_USB_CTRL_BUF_SIZE)
                return -EINVAL;

        buf  = kmalloc(RSI_USB_CTRL_BUF_SIZE, GFP_KERNEL);
        if (!buf)
                return status;

        status = usb_control_msg(usbdev,
                                 usb_rcvctrlpipe(usbdev, 0),
                                 USB_VENDOR_REGISTER_READ,
                                 RSI_USB_REQ_IN,
                                 ((reg & 0xffff0000) >> 16), (reg & 0xffff),
                                 (void *)buf,
                                 len,
                                 USB_CTRL_GET_TIMEOUT);

        *value = (buf[0] | (buf[1] << 8));
        if (status < 0) {
                rsi_dbg(ERR_ZONE,
                        "%s: Reg read failed with error code :%d\n",
                        __func__, status);
        }
        kfree(buf);

        return status;
}

/**
 * rsi_usb_reg_write() - This function writes the given data into the given
 *                       register address.
 * @usbdev: Pointer to the usb_device structure.
 * @reg: Address of the register.
 * @value: Value to write.
 * @len: Length of data to be written.
 *
 * Return: status: 0 on success, a negative error code on failure.
 */
static int rsi_usb_reg_write(struct usb_device *usbdev,
                             u32 reg,
                             u32 value,
                             u16 len)
{
        u8 *usb_reg_buf;
        int status = -ENOMEM;

        if (len > RSI_USB_CTRL_BUF_SIZE)
                return -EINVAL;

        usb_reg_buf  = kmalloc(RSI_USB_CTRL_BUF_SIZE, GFP_KERNEL);
        if (!usb_reg_buf)
                return status;

        usb_reg_buf[0] = value & 0x00ff;
        usb_reg_buf[1] = (value & 0xff00) >> 8;
        usb_reg_buf[2] = (value & 0x00ff0000) >> 16;
        usb_reg_buf[3] = (value & 0xff000000) >> 24;

        status = usb_control_msg(usbdev,
                                 usb_sndctrlpipe(usbdev, 0),
                                 USB_VENDOR_REGISTER_WRITE,
                                 RSI_USB_REQ_OUT,
                                 (reg & 0xffff0000) >> 16,
                                 reg & 0xffff,
                                 (void *)usb_reg_buf,
                                 len,
                                 USB_CTRL_SET_TIMEOUT);
        if (status < 0) {
                rsi_dbg(ERR_ZONE,
                        "%s: Reg write failed with error code :%d\n",
                        __func__, status);
        }
        kfree(usb_reg_buf);

        return status;
}

/**
 * rsi_rx_done_handler() - This function is called when a packet is received
 *                         from USB stack. This is callback to receive done.
 * @urb: Received URB.
 *
 * Return: None.
 */
static void rsi_rx_done_handler(struct urb *urb)
{
        struct rx_usb_ctrl_block *rx_cb = urb->context;
        struct rsi_91x_usbdev *dev = (struct rsi_91x_usbdev *)rx_cb->data;
        int status = -EINVAL;

        if (!rx_cb->rx_skb)
                return;

        if (urb->status) {
                dev_kfree_skb(rx_cb->rx_skb);
                rx_cb->rx_skb = NULL;
                return;
        }

        if (urb->actual_length <= 0 ||
            urb->actual_length > rx_cb->rx_skb->len) {
                rsi_dbg(INFO_ZONE, "%s: Invalid packet length = %d\n",
                        __func__, urb->actual_length);
                goto out;
        }
        if (skb_queue_len(&dev->rx_q) >= RSI_MAX_RX_PKTS) {
                rsi_dbg(INFO_ZONE, "Max RX packets reached\n");
                goto out;
        }
        skb_trim(rx_cb->rx_skb, urb->actual_length);
        skb_queue_tail(&dev->rx_q, rx_cb->rx_skb);

        rsi_set_event(&dev->rx_thread.event);
        status = 0;

out:
        if (rsi_rx_urb_submit(dev->priv, rx_cb->ep_num, GFP_ATOMIC))
                rsi_dbg(ERR_ZONE, "%s: Failed in urb submission", __func__);

        if (status) {
                dev_kfree_skb(rx_cb->rx_skb);
                rx_cb->rx_skb = NULL;
        }
}

static void rsi_rx_urb_kill(struct rsi_hw *adapter, u8 ep_num)
{
        struct rsi_91x_usbdev *dev = adapter->rsi_dev;
        struct rx_usb_ctrl_block *rx_cb = &dev->rx_cb[ep_num - 1];
        struct urb *urb = rx_cb->rx_urb;

        usb_kill_urb(urb);
}

/**
 * rsi_rx_urb_submit() - This function submits the given URB to the USB stack.
 * @adapter: Pointer to the adapter structure.
 * @ep_num: Endpoint number.
 * @mem_flags: The type of memory to allocate.
 *
 * Return: 0 on success, a negative error code on failure.
 */
static int rsi_rx_urb_submit(struct rsi_hw *adapter, u8 ep_num, gfp_t mem_flags)
{
        struct rsi_91x_usbdev *dev = adapter->rsi_dev;
        struct rx_usb_ctrl_block *rx_cb = &dev->rx_cb[ep_num - 1];
        struct urb *urb = rx_cb->rx_urb;
        int status;
        struct sk_buff *skb;
        u8 dword_align_bytes = 0;

        skb = dev_alloc_skb(RSI_MAX_RX_USB_PKT_SIZE);
        if (!skb)
                return -ENOMEM;
        skb_reserve(skb, MAX_DWORD_ALIGN_BYTES);
        skb_put(skb, RSI_MAX_RX_USB_PKT_SIZE - MAX_DWORD_ALIGN_BYTES);
        dword_align_bytes = (unsigned long)skb->data & 0x3f;
        if (dword_align_bytes > 0)
                skb_push(skb, dword_align_bytes);
        urb->transfer_buffer = skb->data;
        rx_cb->rx_skb = skb;

        usb_fill_bulk_urb(urb,
                          dev->usbdev,
                          usb_rcvbulkpipe(dev->usbdev,
                          dev->bulkin_endpoint_addr[ep_num - 1]),
                          urb->transfer_buffer,
                          skb->len,
                          rsi_rx_done_handler,
                          rx_cb);

        status = usb_submit_urb(urb, mem_flags);
        if (status) {
                rsi_dbg(ERR_ZONE, "%s: Failed in urb submission\n", __func__);
                dev_kfree_skb(skb);
        }

        return status;
}

static int rsi_usb_read_register_multiple(struct rsi_hw *adapter, u32 addr,
                                          u8 *data, u16 count)
{
        struct rsi_91x_usbdev *dev = adapter->rsi_dev;
        u8 *buf;
        u16 transfer;
        int status;

        if (!addr)
                return -EINVAL;

        buf = kzalloc(RSI_USB_BUF_SIZE, GFP_KERNEL);
        if (!buf)
                return -ENOMEM;

        while (count) {
                transfer = min_t(u16, count, RSI_USB_BUF_SIZE);
                status = usb_control_msg(dev->usbdev,
                                         usb_rcvctrlpipe(dev->usbdev, 0),
                                         USB_VENDOR_REGISTER_READ,
                                         RSI_USB_REQ_IN,
                                         ((addr & 0xffff0000) >> 16),
                                         (addr & 0xffff), (void *)buf,
                                         transfer, USB_CTRL_GET_TIMEOUT);
                if (status < 0) {
                        rsi_dbg(ERR_ZONE,
                                "Reg read failed with error code :%d\n",
                                 status);
                        kfree(buf);
                        return status;
                }
                memcpy(data, buf, transfer);
                count -= transfer;
                data += transfer;
                addr += transfer;
        }
        kfree(buf);
        return 0;
}

/**
 * rsi_usb_write_register_multiple() - This function writes multiple bytes of
 *                                     information to multiple registers.
 * @adapter: Pointer to the adapter structure.
 * @addr: Address of the register.
 * @data: Pointer to the data that has to be written.
 * @count: Number of multiple bytes to be written on to the registers.
 *
 * Return: status: 0 on success, a negative error code on failure.
 */
static int rsi_usb_write_register_multiple(struct rsi_hw *adapter, u32 addr,
                                           u8 *data, u16 count)
{
        struct rsi_91x_usbdev *dev = adapter->rsi_dev;
        u8 *buf;
        u16 transfer;
        int status = 0;

        buf = kzalloc(RSI_USB_BUF_SIZE, GFP_KERNEL);
        if (!buf)
                return -ENOMEM;

        while (count) {
                transfer = min_t(u16, count, RSI_USB_BUF_SIZE);
                memcpy(buf, data, transfer);
                status = usb_control_msg(dev->usbdev,
                                         usb_sndctrlpipe(dev->usbdev, 0),
                                         USB_VENDOR_REGISTER_WRITE,
                                         RSI_USB_REQ_OUT,
                                         ((addr & 0xffff0000) >> 16),
                                         (addr & 0xffff),
                                         (void *)buf,
                                         transfer,
                                         USB_CTRL_SET_TIMEOUT);
                if (status < 0) {
                        rsi_dbg(ERR_ZONE,
                                "Reg write failed with error code :%d\n",
                                status);
                        kfree(buf);
                        return status;
                }
                count -= transfer;
                data += transfer;
                addr += transfer;
        }

        kfree(buf);
        return 0;
}

/**
 *rsi_usb_host_intf_write_pkt() - This function writes the packet to the
 *                                 USB card.
 * @adapter: Pointer to the adapter structure.
 * @pkt: Pointer to the data to be written on to the card.
 * @len: Length of the data to be written on to the card.
 *
 * Return: 0 on success, a negative error code on failure.
 */
static int rsi_usb_host_intf_write_pkt(struct rsi_hw *adapter,
                                       u8 *pkt,
                                       u32 len)
{
        u32 queueno = ((pkt[1] >> 4) & 0x7);
        u8 endpoint;

        endpoint = ((queueno == RSI_WIFI_MGMT_Q || queueno == RSI_WIFI_DATA_Q ||
                     queueno == RSI_COEX_Q) ? WLAN_EP : BT_EP);

        return rsi_write_multiple(adapter,
                                  endpoint,
                                  (u8 *)pkt,
                                  len);
}

static int rsi_usb_master_reg_read(struct rsi_hw *adapter, u32 reg,
                                   u32 *value, u16 len)
{
        struct usb_device *usbdev =
                ((struct rsi_91x_usbdev *)adapter->rsi_dev)->usbdev;
        u16 temp;
        int ret;

        ret = rsi_usb_reg_read(usbdev, reg, &temp, len);
        if (ret < 0)
                return ret;
        *value = temp;

        return 0;
}

static int rsi_usb_master_reg_write(struct rsi_hw *adapter,
                                    unsigned long reg,
                                    unsigned long value, u16 len)
{
        struct usb_device *usbdev =
                ((struct rsi_91x_usbdev *)adapter->rsi_dev)->usbdev;

        return rsi_usb_reg_write(usbdev, reg, value, len);
}

static int rsi_usb_load_data_master_write(struct rsi_hw *adapter,
                                          u32 base_address,
                                          u32 instructions_sz, u16 block_size,
                                          u8 *ta_firmware)
{
        u16 num_blocks;
        u32 cur_indx, i;
        u8 temp_buf[256];
        int status;

        num_blocks = instructions_sz / block_size;
        rsi_dbg(INFO_ZONE, "num_blocks: %d\n", num_blocks);

        for (cur_indx = 0, i = 0; i < num_blocks; i++, cur_indx += block_size) {
                memcpy(temp_buf, ta_firmware + cur_indx, block_size);
                status = rsi_usb_write_register_multiple(adapter, base_address,
                                                         (u8 *)(temp_buf),
                                                         block_size);
                if (status < 0)
                        return status;

                rsi_dbg(INFO_ZONE, "%s: loading block: %d\n", __func__, i);
                base_address += block_size;
        }

        if (instructions_sz % block_size) {
                memset(temp_buf, 0, block_size);
                memcpy(temp_buf, ta_firmware + cur_indx,
                       instructions_sz % block_size);
                status = rsi_usb_write_register_multiple
                                                (adapter, base_address,
                                                 (u8 *)temp_buf,
                                                 instructions_sz % block_size);
                if (status < 0)
                        return status;
                rsi_dbg(INFO_ZONE,
                        "Written Last Block in Address 0x%x Successfully\n",
                        cur_indx);
        }
        return 0;
}

static struct rsi_host_intf_ops usb_host_intf_ops = {
        .write_pkt              = rsi_usb_host_intf_write_pkt,
        .read_reg_multiple      = rsi_usb_read_register_multiple,
        .write_reg_multiple     = rsi_usb_write_register_multiple,
        .master_reg_read        = rsi_usb_master_reg_read,
        .master_reg_write       = rsi_usb_master_reg_write,
        .load_data_master_write = rsi_usb_load_data_master_write,
};

/**
 * rsi_deinit_usb_interface() - This function deinitializes the usb interface.
 * @adapter: Pointer to the adapter structure.
 *
 * Return: None.
 */
static void rsi_deinit_usb_interface(struct rsi_hw *adapter)
{
        struct rsi_91x_usbdev *dev = adapter->rsi_dev;

        rsi_kill_thread(&dev->rx_thread);

        usb_free_urb(dev->rx_cb[0].rx_urb);
        if (adapter->priv->coex_mode > 1)
                usb_free_urb(dev->rx_cb[1].rx_urb);

        kfree(dev->tx_buffer);
}

static int rsi_usb_init_rx(struct rsi_hw *adapter)
{
        struct rsi_91x_usbdev *dev = adapter->rsi_dev;
        struct rx_usb_ctrl_block *rx_cb;
        u8 idx, num_rx_cb;

        num_rx_cb = (adapter->priv->coex_mode > 1 ? 2 : 1);

        for (idx = 0; idx < num_rx_cb; idx++) {
                rx_cb = &dev->rx_cb[idx];

                rx_cb->rx_urb = usb_alloc_urb(0, GFP_KERNEL);
                if (!rx_cb->rx_urb) {
                        rsi_dbg(ERR_ZONE, "Failed alloc rx urb[%d]\n", idx);
                        goto err;
                }
                rx_cb->ep_num = idx + 1;
                rx_cb->data = (void *)dev;
        }
        skb_queue_head_init(&dev->rx_q);
        rsi_init_event(&dev->rx_thread.event);
        if (rsi_create_kthread(adapter->priv, &dev->rx_thread,
                               rsi_usb_rx_thread, "RX-Thread")) {
                rsi_dbg(ERR_ZONE, "%s: Unable to init rx thrd\n", __func__);
                goto err;
        }

        return 0;

err:
        usb_free_urb(dev->rx_cb[0].rx_urb);
        if (adapter->priv->coex_mode > 1)
                usb_free_urb(dev->rx_cb[1].rx_urb);

        return -1;
}

/**
 * rsi_init_usb_interface() - This function initializes the usb interface.
 * @adapter: Pointer to the adapter structure.
 * @pfunction: Pointer to USB interface structure.
 *
 * Return: 0 on success, a negative error code on failure.
 */
static int rsi_init_usb_interface(struct rsi_hw *adapter,
                                  struct usb_interface *pfunction)
{
        struct rsi_91x_usbdev *rsi_dev;
        int status;

        rsi_dev = kzalloc_obj(*rsi_dev);
        if (!rsi_dev)
                return -ENOMEM;

        adapter->rsi_dev = rsi_dev;
        rsi_dev->usbdev = interface_to_usbdev(pfunction);
        rsi_dev->priv = (void *)adapter;

        if (rsi_find_bulk_in_and_out_endpoints(pfunction, adapter)) {
                status = -EINVAL;
                goto fail_eps;
        }

        adapter->device = &pfunction->dev;
        usb_set_intfdata(pfunction, adapter);

        rsi_dev->tx_buffer = kmalloc(2048, GFP_KERNEL);
        if (!rsi_dev->tx_buffer) {
                status = -ENOMEM;
                goto fail_eps;
        }

        if (rsi_usb_init_rx(adapter)) {
                rsi_dbg(ERR_ZONE, "Failed to init RX handle\n");
                status = -ENOMEM;
                goto fail_rx;
        }

        rsi_dev->tx_blk_size = 252;
        adapter->block_size = rsi_dev->tx_blk_size;

        /* Initializing function callbacks */
        adapter->check_hw_queue_status = rsi_usb_check_queue_status;
        adapter->determine_event_timeout = rsi_usb_event_timeout;
        adapter->rsi_host_intf = RSI_HOST_INTF_USB;
        adapter->host_intf_ops = &usb_host_intf_ops;

#ifdef CONFIG_RSI_DEBUGFS
        /* In USB, one less than the MAX_DEBUGFS_ENTRIES entries is required */
        adapter->num_debugfs_entries = (MAX_DEBUGFS_ENTRIES - 1);
#endif

        rsi_dbg(INIT_ZONE, "%s: Enabled the interface\n", __func__);
        return 0;

fail_rx:
        kfree(rsi_dev->tx_buffer);

fail_eps:

        return status;
}

static int usb_ulp_read_write(struct rsi_hw *adapter, u16 addr, u32 data,
                              u16 len_in_bits)
{
        int ret;

        ret = rsi_usb_master_reg_write
                        (adapter, RSI_GSPI_DATA_REG1,
                         ((addr << 6) | ((data >> 16) & 0xffff)), 2);
        if (ret < 0)
                return ret;

        ret = rsi_usb_master_reg_write(adapter, RSI_GSPI_DATA_REG0,
                                       (data & 0xffff), 2);
        if (ret < 0)
                return ret;

        /* Initializing GSPI for ULP read/writes */
        rsi_usb_master_reg_write(adapter, RSI_GSPI_CTRL_REG0,
                                 RSI_GSPI_CTRL_REG0_VALUE, 2);

        ret = rsi_usb_master_reg_write(adapter, RSI_GSPI_CTRL_REG1,
                                       ((len_in_bits - 1) | RSI_GSPI_TRIG), 2);
        if (ret < 0)
                return ret;

        msleep(20);

        return 0;
}

static int rsi_reset_card(struct rsi_hw *adapter)
{
        int ret;

        rsi_dbg(INFO_ZONE, "Resetting Card...\n");
        rsi_usb_master_reg_write(adapter, RSI_TA_HOLD_REG, 0xE, 4);

        /* This msleep will ensure Thread-Arch processor to go to hold
         * and any pending dma transfers to rf in device to finish.
         */
        msleep(100);

        ret = rsi_usb_master_reg_write(adapter, SWBL_REGOUT,
                                       RSI_FW_WDT_DISABLE_REQ,
                                       RSI_COMMON_REG_SIZE);
        if (ret < 0) {
                rsi_dbg(ERR_ZONE, "Disabling firmware watchdog timer failed\n");
                goto fail;
        }

        if (adapter->device_model != RSI_DEV_9116) {
                ret = usb_ulp_read_write(adapter, RSI_WATCH_DOG_TIMER_1,
                                         RSI_ULP_WRITE_2, 32);
                if (ret < 0)
                        goto fail;
                ret = usb_ulp_read_write(adapter, RSI_WATCH_DOG_TIMER_2,
                                         RSI_ULP_WRITE_0, 32);
                if (ret < 0)
                        goto fail;
                ret = usb_ulp_read_write(adapter, RSI_WATCH_DOG_DELAY_TIMER_1,
                                         RSI_ULP_WRITE_50, 32);
                if (ret < 0)
                        goto fail;
                ret = usb_ulp_read_write(adapter, RSI_WATCH_DOG_DELAY_TIMER_2,
                                         RSI_ULP_WRITE_0, 32);
                if (ret < 0)
                        goto fail;
                ret = usb_ulp_read_write(adapter, RSI_WATCH_DOG_TIMER_ENABLE,
                                         RSI_ULP_TIMER_ENABLE, 32);
                if (ret < 0)
                        goto fail;
        } else {
                ret = rsi_usb_master_reg_write(adapter,
                                               NWP_WWD_INTERRUPT_TIMER,
                                               NWP_WWD_INT_TIMER_CLKS,
                                               RSI_9116_REG_SIZE);
                if (ret < 0)
                        goto fail;
                ret = rsi_usb_master_reg_write(adapter,
                                               NWP_WWD_SYSTEM_RESET_TIMER,
                                               NWP_WWD_SYS_RESET_TIMER_CLKS,
                                               RSI_9116_REG_SIZE);
                if (ret < 0)
                        goto fail;
                ret = rsi_usb_master_reg_write(adapter,
                                               NWP_WWD_MODE_AND_RSTART,
                                               NWP_WWD_TIMER_DISABLE,
                                               RSI_9116_REG_SIZE);
                if (ret < 0)
                        goto fail;
        }

        rsi_dbg(INFO_ZONE, "Reset card done\n");
        return ret;

fail:
        rsi_dbg(ERR_ZONE, "Reset card failed\n");
        return ret;
}

/**
 * rsi_probe() - This function is called by kernel when the driver provided
 *               Vendor and device IDs are matched. All the initialization
 *               work is done here.
 * @pfunction: Pointer to the USB interface structure.
 * @id: Pointer to the usb_device_id structure.
 *
 * Return: 0 on success, a negative error code on failure.
 */
static int rsi_probe(struct usb_interface *pfunction,
                     const struct usb_device_id *id)
{
        struct rsi_hw *adapter;
        struct rsi_91x_usbdev *dev;
        u16 fw_status;
        int status;

        rsi_dbg(INIT_ZONE, "%s: Init function called\n", __func__);

        adapter = rsi_91x_init(dev_oper_mode);
        if (!adapter) {
                rsi_dbg(ERR_ZONE, "%s: Failed to init os intf ops\n",
                        __func__);
                return -ENOMEM;
        }
        adapter->rsi_host_intf = RSI_HOST_INTF_USB;

        status = rsi_init_usb_interface(adapter, pfunction);
        if (status) {
                rsi_dbg(ERR_ZONE, "%s: Failed to init usb interface\n",
                        __func__);
                goto err;
        }

        rsi_dbg(ERR_ZONE, "%s: Initialized os intf ops\n", __func__);

        if (id->idProduct == RSI_USB_PID_9113) {
                rsi_dbg(INIT_ZONE, "%s: 9113 module detected\n", __func__);
                adapter->device_model = RSI_DEV_9113;
        } else if (id->idProduct == RSI_USB_PID_9116) {
                rsi_dbg(INIT_ZONE, "%s: 9116 module detected\n", __func__);
                adapter->device_model = RSI_DEV_9116;
        } else {
                rsi_dbg(ERR_ZONE, "%s: Unsupported RSI device id 0x%x\n",
                        __func__, id->idProduct);
                status = -ENODEV;
                goto err1;
        }

        dev = adapter->rsi_dev;

        status = rsi_usb_reg_read(dev->usbdev, FW_STATUS_REG, &fw_status, 2);
        if (status < 0)
                goto err1;
        else
                fw_status &= 1;

        if (!fw_status) {
                rsi_dbg(INIT_ZONE, "Loading firmware...\n");
                status = rsi_hal_device_init(adapter);
                if (status) {
                        rsi_dbg(ERR_ZONE, "%s: Failed in device init\n",
                                __func__);
                        goto err1;
                }
                rsi_dbg(INIT_ZONE, "%s: Device Init Done\n", __func__);
        }

        status = rsi_rx_urb_submit(adapter, WLAN_EP, GFP_KERNEL);
        if (status)
                goto err1;

        if (adapter->priv->coex_mode > 1) {
                status = rsi_rx_urb_submit(adapter, BT_EP, GFP_KERNEL);
                if (status)
                        goto err_kill_wlan_urb;
        }

        return 0;

err_kill_wlan_urb:
        rsi_rx_urb_kill(adapter, WLAN_EP);
err1:
        rsi_deinit_usb_interface(adapter);
err:
        rsi_91x_deinit(adapter);
        rsi_dbg(ERR_ZONE, "%s: Failed in probe...Exiting\n", __func__);
        return status;
}

/**
 * rsi_disconnect() - This function performs the reverse of the probe function,
 *                    it deinitialize the driver structure.
 * @pfunction: Pointer to the USB interface structure.
 *
 * Return: None.
 */
static void rsi_disconnect(struct usb_interface *pfunction)
{
        struct rsi_hw *adapter = usb_get_intfdata(pfunction);

        if (!adapter)
                return;

        rsi_mac80211_detach(adapter);

        if (IS_ENABLED(CONFIG_RSI_COEX) && adapter->priv->coex_mode > 1 &&
            adapter->priv->bt_adapter) {
                rsi_bt_ops.detach(adapter->priv->bt_adapter);
                adapter->priv->bt_adapter = NULL;
        }

        if (adapter->priv->coex_mode > 1)
                rsi_rx_urb_kill(adapter, BT_EP);
        rsi_rx_urb_kill(adapter, WLAN_EP);

        rsi_reset_card(adapter);
        rsi_deinit_usb_interface(adapter);
        rsi_91x_deinit(adapter);

        rsi_dbg(INFO_ZONE, "%s: Deinitialization completed\n", __func__);
}

#ifdef CONFIG_PM
static int rsi_suspend(struct usb_interface *intf, pm_message_t message)
{
        /* Not yet implemented */
        return -ENOSYS;
}

static int rsi_resume(struct usb_interface *intf)
{
        /* Not yet implemented */
        return -ENOSYS;
}
#endif

static const struct usb_device_id rsi_dev_table[] = {
        { USB_DEVICE(RSI_USB_VENDOR_ID, RSI_USB_PID_9113) },
        { USB_DEVICE(RSI_USB_VENDOR_ID, RSI_USB_PID_9116) },
        { /* Blank */},
};

static struct usb_driver rsi_driver = {
        .name       = "RSI-USB WLAN",
        .probe      = rsi_probe,
        .disconnect = rsi_disconnect,
        .id_table   = rsi_dev_table,
#ifdef CONFIG_PM
        .suspend    = rsi_suspend,
        .resume     = rsi_resume,
#endif
};

module_usb_driver(rsi_driver);

MODULE_AUTHOR("Redpine Signals Inc");
MODULE_DESCRIPTION("Common USB layer for RSI drivers");
MODULE_DEVICE_TABLE(usb, rsi_dev_table);
MODULE_FIRMWARE(FIRMWARE_RSI9113);
MODULE_VERSION("0.1");
MODULE_LICENSE("Dual BSD/GPL");