/*
 * Copyright (c) 2012-2019 The Linux Foundation. All rights reserved.
 *
 * 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/etherdevice.h>
#include <linux/moduleparam.h>
#include <linux/prefetch.h>
#include <linux/types.h>
#include <linux/list.h>
#include <linux/ip.h>
#include <linux/ipv6.h>
#include "wil6210.h"
#include "txrx_edma.h"
#include "txrx.h"
#include "trace.h"

/* Max number of entries (packets to complete) to update the hwtail of tx
 * status ring. Should be power of 2
 */
#define WIL_EDMA_TX_SRING_UPDATE_HW_TAIL 128
#define WIL_EDMA_MAX_DATA_OFFSET (2)
/* RX buffer size must be aligned to 4 bytes */
#define WIL_EDMA_RX_BUF_LEN_DEFAULT (2048)
#define MAX_INVALID_BUFF_ID_RETRY (3)

static void wil_tx_desc_unmap_edma(struct device *dev,
                                   union wil_tx_desc *desc,
                                   struct wil_ctx *ctx)
{
        struct wil_tx_enhanced_desc *d = (struct wil_tx_enhanced_desc *)desc;
        dma_addr_t pa = wil_tx_desc_get_addr_edma(&d->dma);
        u16 dmalen = le16_to_cpu(d->dma.length);

        switch (ctx->mapped_as) {
        case wil_mapped_as_single:
                dma_unmap_single(dev, pa, dmalen, DMA_TO_DEVICE);
                break;
        case wil_mapped_as_page:
                dma_unmap_page(dev, pa, dmalen, DMA_TO_DEVICE);
                break;
        default:
                break;
        }
}

static int wil_find_free_sring(struct wil6210_priv *wil)
{
        int i;

        for (i = 0; i < WIL6210_MAX_STATUS_RINGS; i++) {
                if (!wil->srings[i].va)
                        return i;
        }

        return -EINVAL;
}

static void wil_sring_free(struct wil6210_priv *wil,
                           struct wil_status_ring *sring)
{
        struct device *dev = wil_to_dev(wil);
        size_t sz;

        if (!sring || !sring->va)
                return;

        sz = sring->elem_size * sring->size;

        wil_dbg_misc(wil, "status_ring_free, size(bytes)=%zu, 0x%p:%pad\n",
                     sz, sring->va, &sring->pa);

        dma_free_coherent(dev, sz, (void *)sring->va, sring->pa);
        sring->pa = 0;
        sring->va = NULL;
}

static int wil_sring_alloc(struct wil6210_priv *wil,
                           struct wil_status_ring *sring)
{
        struct device *dev = wil_to_dev(wil);
        size_t sz = sring->elem_size * sring->size;

        wil_dbg_misc(wil, "status_ring_alloc: size=%zu\n", sz);

        if (sz == 0) {
                wil_err(wil, "Cannot allocate a zero size status ring\n");
                return -EINVAL;
        }

        sring->swhead = 0;

        /* Status messages are allocated and initialized to 0. This is necessary
         * since DR bit should be initialized to 0.
         */
        sring->va = dma_alloc_coherent(dev, sz, &sring->pa, GFP_KERNEL);
        if (!sring->va)
                return -ENOMEM;

        wil_dbg_misc(wil, "status_ring[%d] 0x%p:%pad\n", sring->size, sring->va,
                     &sring->pa);

        return 0;
}

static int wil_tx_init_edma(struct wil6210_priv *wil)
{
        int ring_id = wil_find_free_sring(wil);
        struct wil_status_ring *sring;
        int rc;
        u16 status_ring_size;

        if (wil->tx_status_ring_order < WIL_SRING_SIZE_ORDER_MIN ||
            wil->tx_status_ring_order > WIL_SRING_SIZE_ORDER_MAX)
                wil->tx_status_ring_order = WIL_TX_SRING_SIZE_ORDER_DEFAULT;

        status_ring_size = 1 << wil->tx_status_ring_order;

        wil_dbg_misc(wil, "init TX sring: size=%u, ring_id=%u\n",
                     status_ring_size, ring_id);

        if (ring_id < 0)
                return ring_id;

        /* Allocate Tx status ring. Tx descriptor rings will be
         * allocated on WMI connect event
         */
        sring = &wil->srings[ring_id];

        sring->is_rx = false;
        sring->size = status_ring_size;
        sring->elem_size = sizeof(struct wil_ring_tx_status);
        rc = wil_sring_alloc(wil, sring);
        if (rc)
                return rc;

        rc = wil_wmi_tx_sring_cfg(wil, ring_id);
        if (rc)
                goto out_free;

        sring->desc_rdy_pol = 1;
        wil->tx_sring_idx = ring_id;

        return 0;
out_free:
        wil_sring_free(wil, sring);
        return rc;
}

/**
 * Allocate one skb for Rx descriptor RING
 */
static int wil_ring_alloc_skb_edma(struct wil6210_priv *wil,
                                   struct wil_ring *ring, u32 i)
{
        struct device *dev = wil_to_dev(wil);
        unsigned int sz = wil->rx_buf_len;
        dma_addr_t pa;
        u16 buff_id;
        struct list_head *active = &wil->rx_buff_mgmt.active;
        struct list_head *free = &wil->rx_buff_mgmt.free;
        struct wil_rx_buff *rx_buff;
        struct wil_rx_buff *buff_arr = wil->rx_buff_mgmt.buff_arr;
        struct sk_buff *skb;
        struct wil_rx_enhanced_desc dd, *d = &dd;
        struct wil_rx_enhanced_desc *_d = (struct wil_rx_enhanced_desc *)
                &ring->va[i].rx.enhanced;

        if (unlikely(list_empty(free))) {
                wil->rx_buff_mgmt.free_list_empty_cnt++;
                return -EAGAIN;
        }

        skb = dev_alloc_skb(sz);
        if (unlikely(!skb))
                return -ENOMEM;

        skb_put(skb, sz);

        /**
         * Make sure that the network stack calculates checksum for packets
         * which failed the HW checksum calculation
         */
        skb->ip_summed = CHECKSUM_NONE;

        pa = dma_map_single(dev, skb->data, skb->len, DMA_FROM_DEVICE);
        if (unlikely(dma_mapping_error(dev, pa))) {
                kfree_skb(skb);
                return -ENOMEM;
        }

        /* Get the buffer ID - the index of the rx buffer in the buff_arr */
        rx_buff = list_first_entry(free, struct wil_rx_buff, list);
        buff_id = rx_buff->id;

        /* Move a buffer from the free list to the active list */
        list_move(&rx_buff->list, active);

        buff_arr[buff_id].skb = skb;

        wil_desc_set_addr_edma(&d->dma.addr, &d->dma.addr_high_high, pa);
        d->dma.length = cpu_to_le16(sz);
        d->mac.buff_id = cpu_to_le16(buff_id);
        *_d = *d;

        /* Save the physical address in skb->cb for later use in dma_unmap */
        memcpy(skb->cb, &pa, sizeof(pa));

        return 0;
}

static inline
void wil_get_next_rx_status_msg(struct wil_status_ring *sring, u8 *dr_bit,
                                void *msg)
{
        struct wil_rx_status_compressed *_msg;

        _msg = (struct wil_rx_status_compressed *)
                (sring->va + (sring->elem_size * sring->swhead));
        *dr_bit = WIL_GET_BITS(_msg->d0, 31, 31);
        /* make sure dr_bit is read before the rest of status msg */
        rmb();
        memcpy(msg, (void *)_msg, sring->elem_size);
}

static inline void wil_sring_advance_swhead(struct wil_status_ring *sring)
{
        sring->swhead = (sring->swhead + 1) % sring->size;
        if (sring->swhead == 0)
                sring->desc_rdy_pol = 1 - sring->desc_rdy_pol;
}

static int wil_rx_refill_edma(struct wil6210_priv *wil)
{
        struct wil_ring *ring = &wil->ring_rx;
        u32 next_head;
        int rc = 0;
        ring->swtail = *ring->edma_rx_swtail.va;

        for (; next_head = wil_ring_next_head(ring),
             (next_head != ring->swtail);
             ring->swhead = next_head) {
                rc = wil_ring_alloc_skb_edma(wil, ring, ring->swhead);
                if (unlikely(rc)) {
                        if (rc == -EAGAIN)
                                wil_dbg_txrx(wil, "No free buffer ID found\n");
                        else
                                wil_err_ratelimited(wil,
                                                    "Error %d in refill desc[%d]\n",
                                                    rc, ring->swhead);
                        break;
                }
        }

        /* make sure all writes to descriptors (shared memory) are done before
         * committing them to HW
         */
        wmb();

        wil_w(wil, ring->hwtail, ring->swhead);

        return rc;
}

static void wil_move_all_rx_buff_to_free_list(struct wil6210_priv *wil,
                                              struct wil_ring *ring)
{
        struct device *dev = wil_to_dev(wil);
        struct list_head *active = &wil->rx_buff_mgmt.active;
        dma_addr_t pa;

        if (!wil->rx_buff_mgmt.buff_arr)
                return;

        while (!list_empty(active)) {
                struct wil_rx_buff *rx_buff =
                        list_first_entry(active, struct wil_rx_buff, list);
                struct sk_buff *skb = rx_buff->skb;

                if (unlikely(!skb)) {
                        wil_err(wil, "No Rx skb at buff_id %d\n", rx_buff->id);
                } else {
                        rx_buff->skb = NULL;
                        memcpy(&pa, skb->cb, sizeof(pa));
                        dma_unmap_single(dev, pa, wil->rx_buf_len,
                                         DMA_FROM_DEVICE);
                        kfree_skb(skb);
                }

                /* Move the buffer from the active to the free list */
                list_move(&rx_buff->list, &wil->rx_buff_mgmt.free);
        }
}

static void wil_free_rx_buff_arr(struct wil6210_priv *wil)
{
        struct wil_ring *ring = &wil->ring_rx;

        if (!wil->rx_buff_mgmt.buff_arr)
                return;

        /* Move all the buffers to the free list in case active list is
         * not empty in order to release all SKBs before deleting the array
         */
        wil_move_all_rx_buff_to_free_list(wil, ring);

        kfree(wil->rx_buff_mgmt.buff_arr);
        wil->rx_buff_mgmt.buff_arr = NULL;
}

static int wil_init_rx_buff_arr(struct wil6210_priv *wil,
                                size_t size)
{
        struct wil_rx_buff *buff_arr;
        struct list_head *active = &wil->rx_buff_mgmt.active;
        struct list_head *free = &wil->rx_buff_mgmt.free;
        int i;

        wil->rx_buff_mgmt.buff_arr = kcalloc(size + 1,
                                             sizeof(struct wil_rx_buff),
                                             GFP_KERNEL);
        if (!wil->rx_buff_mgmt.buff_arr)
                return -ENOMEM;

        /* Set list heads */
        INIT_LIST_HEAD(active);
        INIT_LIST_HEAD(free);

        /* Linkify the list.
         * buffer id 0 should not be used (marks invalid id).
         */
        buff_arr = wil->rx_buff_mgmt.buff_arr;
        for (i = 1; i <= size; i++) {
                list_add(&buff_arr[i].list, free);
                buff_arr[i].id = i;
        }

        wil->rx_buff_mgmt.size = size + 1;

        return 0;
}

static int wil_init_rx_sring(struct wil6210_priv *wil,
                             u16 status_ring_size,
                             size_t elem_size,
                             u16 ring_id)
{
        struct wil_status_ring *sring = &wil->srings[ring_id];
        int rc;

        wil_dbg_misc(wil, "init RX sring: size=%u, ring_id=%u\n",
                     status_ring_size, ring_id);

        memset(&sring->rx_data, 0, sizeof(sring->rx_data));

        sring->is_rx = true;
        sring->size = status_ring_size;
        sring->elem_size = elem_size;
        rc = wil_sring_alloc(wil, sring);
        if (rc)
                return rc;

        rc = wil_wmi_rx_sring_add(wil, ring_id);
        if (rc)
                goto out_free;

        sring->desc_rdy_pol = 1;

        return 0;
out_free:
        wil_sring_free(wil, sring);
        return rc;
}

static int wil_ring_alloc_desc_ring(struct wil6210_priv *wil,
                                    struct wil_ring *ring)
{
        struct device *dev = wil_to_dev(wil);
        size_t sz = ring->size * sizeof(ring->va[0]);

        wil_dbg_misc(wil, "alloc_desc_ring:\n");

        BUILD_BUG_ON(sizeof(ring->va[0]) != 32);

        ring->swhead = 0;
        ring->swtail = 0;
        ring->ctx = kcalloc(ring->size, sizeof(ring->ctx[0]), GFP_KERNEL);
        if (!ring->ctx)
                goto err;

        ring->va = dma_alloc_coherent(dev, sz, &ring->pa, GFP_KERNEL);
        if (!ring->va)
                goto err_free_ctx;

        if (ring->is_rx) {
                sz = sizeof(*ring->edma_rx_swtail.va);
                ring->edma_rx_swtail.va =
                        dma_alloc_coherent(dev, sz, &ring->edma_rx_swtail.pa,
                                           GFP_KERNEL);
                if (!ring->edma_rx_swtail.va)
                        goto err_free_va;
        }

        wil_dbg_misc(wil, "%s ring[%d] 0x%p:%pad 0x%p\n",
                     ring->is_rx ? "RX" : "TX",
                     ring->size, ring->va, &ring->pa, ring->ctx);

        return 0;
err_free_va:
        dma_free_coherent(dev, ring->size * sizeof(ring->va[0]),
                          (void *)ring->va, ring->pa);
        ring->va = NULL;
err_free_ctx:
        kfree(ring->ctx);
        ring->ctx = NULL;
err:
        return -ENOMEM;
}

static void wil_ring_free_edma(struct wil6210_priv *wil, struct wil_ring *ring)
{
        struct device *dev = wil_to_dev(wil);
        size_t sz;
        int ring_index = 0;

        if (!ring->va)
                return;

        sz = ring->size * sizeof(ring->va[0]);

        lockdep_assert_held(&wil->mutex);
        if (ring->is_rx) {
                wil_dbg_misc(wil, "free Rx ring [%d] 0x%p:%pad 0x%p\n",
                             ring->size, ring->va,
                             &ring->pa, ring->ctx);

                wil_move_all_rx_buff_to_free_list(wil, ring);
                dma_free_coherent(dev, sizeof(*ring->edma_rx_swtail.va),
                                  ring->edma_rx_swtail.va,
                                  ring->edma_rx_swtail.pa);
                goto out;
        }

        /* TX ring */
        ring_index = ring - wil->ring_tx;

        wil_dbg_misc(wil, "free Tx ring %d [%d] 0x%p:%pad 0x%p\n",
                     ring_index, ring->size, ring->va,
                     &ring->pa, ring->ctx);

        while (!wil_ring_is_empty(ring)) {
                struct wil_ctx *ctx;

                struct wil_tx_enhanced_desc dd, *d = &dd;
                struct wil_tx_enhanced_desc *_d =
                        (struct wil_tx_enhanced_desc *)
                        &ring->va[ring->swtail].tx.enhanced;

                ctx = &ring->ctx[ring->swtail];
                if (!ctx) {
                        wil_dbg_txrx(wil,
                                     "ctx(%d) was already completed\n",
                                     ring->swtail);
                        ring->swtail = wil_ring_next_tail(ring);
                        continue;
                }
                *d = *_d;
                wil_tx_desc_unmap_edma(dev, (union wil_tx_desc *)d, ctx);
                if (ctx->skb)
                        dev_kfree_skb_any(ctx->skb);
                ring->swtail = wil_ring_next_tail(ring);
        }

out:
        dma_free_coherent(dev, sz, (void *)ring->va, ring->pa);
        kfree(ring->ctx);
        ring->pa = 0;
        ring->va = NULL;
        ring->ctx = NULL;
}

static int wil_init_rx_desc_ring(struct wil6210_priv *wil, u16 desc_ring_size,
                                 int status_ring_id)
{
        struct wil_ring *ring = &wil->ring_rx;
        int rc;

        wil_dbg_misc(wil, "init RX desc ring\n");

        ring->size = desc_ring_size;
        ring->is_rx = true;
        rc = wil_ring_alloc_desc_ring(wil, ring);
        if (rc)
                return rc;

        rc = wil_wmi_rx_desc_ring_add(wil, status_ring_id);
        if (rc)
                goto out_free;

        return 0;
out_free:
        wil_ring_free_edma(wil, ring);
        return rc;
}

static void wil_get_reorder_params_edma(struct wil6210_priv *wil,
                                        struct sk_buff *skb, int *tid,
                                        int *cid, int *mid, u16 *seq,
                                        int *mcast, int *retry)
{
        struct wil_rx_status_extended *s = wil_skb_rxstatus(skb);

        *tid = wil_rx_status_get_tid(s);
        *cid = wil_rx_status_get_cid(s);
        *mid = wil_rx_status_get_mid(s);
        *seq = le16_to_cpu(wil_rx_status_get_seq(wil, s));
        *mcast = wil_rx_status_get_mcast(s);
        *retry = wil_rx_status_get_retry(s);
}

static void wil_get_netif_rx_params_edma(struct sk_buff *skb, int *cid,
                                         int *security)
{
        struct wil_rx_status_extended *s = wil_skb_rxstatus(skb);

        *cid = wil_rx_status_get_cid(s);
        *security = wil_rx_status_get_security(s);
}

static int wil_rx_crypto_check_edma(struct wil6210_priv *wil,
                                    struct sk_buff *skb)
{
        struct wil_rx_status_extended *st;
        int cid, tid, key_id, mc;
        struct wil_sta_info *s;
        struct wil_tid_crypto_rx *c;
        struct wil_tid_crypto_rx_single *cc;
        const u8 *pn;

        /* In HW reorder, HW is responsible for crypto check */
        if (wil->use_rx_hw_reordering)
                return 0;

        st = wil_skb_rxstatus(skb);

        cid = wil_rx_status_get_cid(st);
        tid = wil_rx_status_get_tid(st);
        key_id = wil_rx_status_get_key_id(st);
        mc = wil_rx_status_get_mcast(st);
        s = &wil->sta[cid];
        c = mc ? &s->group_crypto_rx : &s->tid_crypto_rx[tid];
        cc = &c->key_id[key_id];
        pn = (u8 *)&st->ext.pn_15_0;

        if (!cc->key_set) {
                wil_err_ratelimited(wil,
                                    "Key missing. CID %d TID %d MCast %d KEY_ID %d\n",
                                    cid, tid, mc, key_id);
                return -EINVAL;
        }

        if (reverse_memcmp(pn, cc->pn, IEEE80211_GCMP_PN_LEN) <= 0) {
                wil_err_ratelimited(wil,
                                    "Replay attack. CID %d TID %d MCast %d KEY_ID %d PN %6phN last %6phN\n",
                                    cid, tid, mc, key_id, pn, cc->pn);
                return -EINVAL;
        }
        memcpy(cc->pn, pn, IEEE80211_GCMP_PN_LEN);

        return 0;
}

static bool wil_is_rx_idle_edma(struct wil6210_priv *wil)
{
        struct wil_status_ring *sring;
        struct wil_rx_status_extended msg1;
        void *msg = &msg1;
        u8 dr_bit;
        int i;

        for (i = 0; i < wil->num_rx_status_rings; i++) {
                sring = &wil->srings[i];
                if (!sring->va)
                        continue;

                wil_get_next_rx_status_msg(sring, &dr_bit, msg);

                /* Check if there are unhandled RX status messages */
                if (dr_bit == sring->desc_rdy_pol)
                        return false;
        }

        return true;
}

static void wil_rx_buf_len_init_edma(struct wil6210_priv *wil)
{
        /* RX buffer size must be aligned to 4 bytes */
        wil->rx_buf_len = rx_large_buf ?
                WIL_MAX_ETH_MTU : WIL_EDMA_RX_BUF_LEN_DEFAULT;
}

static int wil_rx_init_edma(struct wil6210_priv *wil, uint desc_ring_order)
{
        u16 status_ring_size, desc_ring_size = 1 << desc_ring_order;
        struct wil_ring *ring = &wil->ring_rx;
        int rc;
        size_t elem_size = wil->use_compressed_rx_status ?
                sizeof(struct wil_rx_status_compressed) :
                sizeof(struct wil_rx_status_extended);
        int i;

        /* In SW reorder one must use extended status messages */
        if (wil->use_compressed_rx_status && !wil->use_rx_hw_reordering) {
                wil_err(wil,
                        "compressed RX status cannot be used with SW reorder\n");
                return -EINVAL;
        }
        if (wil->rx_status_ring_order <= desc_ring_order)
                /* make sure sring is larger than desc ring */
                wil->rx_status_ring_order = desc_ring_order + 1;
        if (wil->rx_buff_id_count <= desc_ring_size)
                /* make sure we will not run out of buff_ids */
                wil->rx_buff_id_count = desc_ring_size + 512;
        if (wil->rx_status_ring_order < WIL_SRING_SIZE_ORDER_MIN ||
            wil->rx_status_ring_order > WIL_SRING_SIZE_ORDER_MAX)
                wil->rx_status_ring_order = WIL_RX_SRING_SIZE_ORDER_DEFAULT;

        status_ring_size = 1 << wil->rx_status_ring_order;

        wil_dbg_misc(wil,
                     "rx_init, desc_ring_size=%u, status_ring_size=%u, elem_size=%zu\n",
                     desc_ring_size, status_ring_size, elem_size);

        wil_rx_buf_len_init_edma(wil);

        /* Use debugfs dbg_num_rx_srings if set, reserve one sring for TX */
        if (wil->num_rx_status_rings > WIL6210_MAX_STATUS_RINGS - 1)
                wil->num_rx_status_rings = WIL6210_MAX_STATUS_RINGS - 1;

        wil_dbg_misc(wil, "rx_init: allocate %d status rings\n",
                     wil->num_rx_status_rings);

        rc = wil_wmi_cfg_def_rx_offload(wil, wil->rx_buf_len);
        if (rc)
                return rc;

        /* Allocate status ring */
        for (i = 0; i < wil->num_rx_status_rings; i++) {
                int sring_id = wil_find_free_sring(wil);

                if (sring_id < 0) {
                        rc = -EFAULT;
                        goto err_free_status;
                }
                rc = wil_init_rx_sring(wil, status_ring_size, elem_size,
                                       sring_id);
                if (rc)
                        goto err_free_status;
        }

        /* Allocate descriptor ring */
        rc = wil_init_rx_desc_ring(wil, desc_ring_size,
                                   WIL_DEFAULT_RX_STATUS_RING_ID);
        if (rc)
                goto err_free_status;

        if (wil->rx_buff_id_count >= status_ring_size) {
                wil_info(wil,
                         "rx_buff_id_count %d exceeds sring_size %d. set it to %d\n",
                         wil->rx_buff_id_count, status_ring_size,
                         status_ring_size - 1);
                wil->rx_buff_id_count = status_ring_size - 1;
        }

        /* Allocate Rx buffer array */
        rc = wil_init_rx_buff_arr(wil, wil->rx_buff_id_count);
        if (rc)
                goto err_free_desc;

        /* Fill descriptor ring with credits */
        rc = wil_rx_refill_edma(wil);
        if (rc)
                goto err_free_rx_buff_arr;

        return 0;
err_free_rx_buff_arr:
        wil_free_rx_buff_arr(wil);
err_free_desc:
        wil_ring_free_edma(wil, ring);
err_free_status:
        for (i = 0; i < wil->num_rx_status_rings; i++)
                wil_sring_free(wil, &wil->srings[i]);

        return rc;
}

static int wil_ring_init_tx_edma(struct wil6210_vif *vif, int ring_id,
                                 int size, int cid, int tid)
{
        struct wil6210_priv *wil = vif_to_wil(vif);
        int rc;
        struct wil_ring *ring = &wil->ring_tx[ring_id];
        struct wil_ring_tx_data *txdata = &wil->ring_tx_data[ring_id];

        lockdep_assert_held(&wil->mutex);

        wil_dbg_misc(wil,
                     "init TX ring: ring_id=%u, cid=%u, tid=%u, sring_id=%u\n",
                     ring_id, cid, tid, wil->tx_sring_idx);

        wil_tx_data_init(txdata);
        ring->size = size;
        rc = wil_ring_alloc_desc_ring(wil, ring);
        if (rc)
                goto out;

        wil->ring2cid_tid[ring_id][0] = cid;
        wil->ring2cid_tid[ring_id][1] = tid;
        if (!vif->privacy)
                txdata->dot1x_open = true;

        rc = wil_wmi_tx_desc_ring_add(vif, ring_id, cid, tid);
        if (rc) {
                wil_err(wil, "WMI_TX_DESC_RING_ADD_CMD failed\n");
                goto out_free;
        }

        if (txdata->dot1x_open && agg_wsize >= 0)
                wil_addba_tx_request(wil, ring_id, agg_wsize);

        return 0;
 out_free:
        spin_lock_bh(&txdata->lock);
        txdata->dot1x_open = false;
        txdata->enabled = 0;
        spin_unlock_bh(&txdata->lock);
        wil_ring_free_edma(wil, ring);
        wil->ring2cid_tid[ring_id][0] = wil->max_assoc_sta;
        wil->ring2cid_tid[ring_id][1] = 0;

 out:
        return rc;
}

static int wil_tx_ring_modify_edma(struct wil6210_vif *vif, int ring_id,
                                   int cid, int tid)
{
        struct wil6210_priv *wil = vif_to_wil(vif);

        wil_err(wil, "ring modify is not supported for EDMA\n");

        return -EOPNOTSUPP;
}

/* This function is used only for RX SW reorder */
static int wil_check_bar(struct wil6210_priv *wil, void *msg, int cid,
                         struct sk_buff *skb, struct wil_net_stats *stats)
{
        u8 ftype;
        u8 fc1;
        int mid;
        int tid;
        u16 seq;
        struct wil6210_vif *vif;

        ftype = wil_rx_status_get_frame_type(wil, msg);
        if (ftype == IEEE80211_FTYPE_DATA)
                return 0;

        fc1 = wil_rx_status_get_fc1(wil, msg);
        mid = wil_rx_status_get_mid(msg);
        tid = wil_rx_status_get_tid(msg);
        seq = le16_to_cpu(wil_rx_status_get_seq(wil, msg));
        vif = wil->vifs[mid];

        if (unlikely(!vif)) {
                wil_dbg_txrx(wil, "RX descriptor with invalid mid %d", mid);
                return -EAGAIN;
        }

        wil_dbg_txrx(wil,
                     "Non-data frame FC[7:0] 0x%02x MID %d CID %d TID %d Seq 0x%03x\n",
                     fc1, mid, cid, tid, seq);
        if (stats)
                stats->rx_non_data_frame++;
        if (wil_is_back_req(fc1)) {
                wil_dbg_txrx(wil,
                             "BAR: MID %d CID %d TID %d Seq 0x%03x\n",
                             mid, cid, tid, seq);
                wil_rx_bar(wil, vif, cid, tid, seq);
        } else {
                u32 sz = wil->use_compressed_rx_status ?
                        sizeof(struct wil_rx_status_compressed) :
                        sizeof(struct wil_rx_status_extended);

                /* print again all info. One can enable only this
                 * without overhead for printing every Rx frame
                 */
                wil_dbg_txrx(wil,
                             "Unhandled non-data frame FC[7:0] 0x%02x MID %d CID %d TID %d Seq 0x%03x\n",
                             fc1, mid, cid, tid, seq);
                wil_hex_dump_txrx("RxS ", DUMP_PREFIX_NONE, 32, 4,
                                  (const void *)msg, sz, false);
                wil_hex_dump_txrx("Rx ", DUMP_PREFIX_OFFSET, 16, 1,
                                  skb->data, skb_headlen(skb), false);
        }

        return -EAGAIN;
}

static int wil_rx_error_check_edma(struct wil6210_priv *wil,
                                   struct sk_buff *skb,
                                   struct wil_net_stats *stats)
{
        int l2_rx_status;
        void *msg = wil_skb_rxstatus(skb);

        l2_rx_status = wil_rx_status_get_l2_rx_status(msg);
        if (l2_rx_status != 0) {
                wil_dbg_txrx(wil, "L2 RX error, l2_rx_status=0x%x\n",
                             l2_rx_status);
                /* Due to HW issue, KEY error will trigger a MIC error */
                if (l2_rx_status == WIL_RX_EDMA_ERROR_MIC) {
                        wil_err_ratelimited(wil,
                                            "L2 MIC/KEY error, dropping packet\n");
                        stats->rx_mic_error++;
                }
                if (l2_rx_status == WIL_RX_EDMA_ERROR_KEY) {
                        wil_err_ratelimited(wil,
                                            "L2 KEY error, dropping packet\n");
                        stats->rx_key_error++;
                }
                if (l2_rx_status == WIL_RX_EDMA_ERROR_REPLAY) {
                        wil_err_ratelimited(wil,
                                            "L2 REPLAY error, dropping packet\n");
                        stats->rx_replay++;
                }
                if (l2_rx_status == WIL_RX_EDMA_ERROR_AMSDU) {
                        wil_err_ratelimited(wil,
                                            "L2 AMSDU error, dropping packet\n");
                        stats->rx_amsdu_error++;
                }
                return -EFAULT;
        }

        skb->ip_summed = wil_rx_status_get_checksum(msg, stats);

        return 0;
}

static struct sk_buff *wil_sring_reap_rx_edma(struct wil6210_priv *wil,
                                              struct wil_status_ring *sring)
{
        struct device *dev = wil_to_dev(wil);
        struct wil_rx_status_extended msg1;
        void *msg = &msg1;
        u16 buff_id;
        struct sk_buff *skb;
        dma_addr_t pa;
        struct wil_ring_rx_data *rxdata = &sring->rx_data;
        unsigned int sz = wil->rx_buf_len;
        struct wil_net_stats *stats = NULL;
        u16 dmalen;
        int cid;
        bool eop, headstolen;
        int delta;
        u8 dr_bit;
        u8 data_offset;
        struct wil_rx_status_extended *s;
        u16 sring_idx = sring - wil->srings;

        BUILD_BUG_ON(sizeof(struct wil_rx_status_extended) > sizeof(skb->cb));

again:
        wil_get_next_rx_status_msg(sring, &dr_bit, msg);

        /* Completed handling all the ready status messages */
        if (dr_bit != sring->desc_rdy_pol)
                return NULL;

        /* Extract the buffer ID from the status message */
        buff_id = le16_to_cpu(wil_rx_status_get_buff_id(msg));

        while (!buff_id) {
                struct wil_rx_status_extended *s;
                int invalid_buff_id_retry = 0;

                wil_dbg_txrx(wil,
                             "buff_id is not updated yet by HW, (swhead 0x%x)\n",
                             sring->swhead);
                if (++invalid_buff_id_retry > MAX_INVALID_BUFF_ID_RETRY)
                        break;

                /* Read the status message again */
                s = (struct wil_rx_status_extended *)
                        (sring->va + (sring->elem_size * sring->swhead));
                *(struct wil_rx_status_extended *)msg = *s;
                buff_id = le16_to_cpu(wil_rx_status_get_buff_id(msg));
        }

        if (unlikely(!wil_val_in_range(buff_id, 1, wil->rx_buff_mgmt.size))) {
                wil_err(wil, "Corrupt buff_id=%d, sring->swhead=%d\n",
                        buff_id, sring->swhead);
                wil_rx_status_reset_buff_id(sring);
                wil_sring_advance_swhead(sring);
                sring->invalid_buff_id_cnt++;
                goto again;
        }

        /* Extract the SKB from the rx_buff management array */
        skb = wil->rx_buff_mgmt.buff_arr[buff_id].skb;
        wil->rx_buff_mgmt.buff_arr[buff_id].skb = NULL;
        if (!skb) {
                wil_err(wil, "No Rx skb at buff_id %d\n", buff_id);
                wil_rx_status_reset_buff_id(sring);
                /* Move the buffer from the active list to the free list */
                list_move_tail(&wil->rx_buff_mgmt.buff_arr[buff_id].list,
                               &wil->rx_buff_mgmt.free);
                wil_sring_advance_swhead(sring);
                sring->invalid_buff_id_cnt++;
                goto again;
        }

        wil_rx_status_reset_buff_id(sring);
        wil_sring_advance_swhead(sring);

        memcpy(&pa, skb->cb, sizeof(pa));
        dma_unmap_single(dev, pa, sz, DMA_FROM_DEVICE);
        dmalen = le16_to_cpu(wil_rx_status_get_length(msg));

        trace_wil6210_rx_status(wil, wil->use_compressed_rx_status, buff_id,
                                msg);
        wil_dbg_txrx(wil, "Rx, buff_id=%u, sring_idx=%u, dmalen=%u bytes\n",
                     buff_id, sring_idx, dmalen);
        wil_hex_dump_txrx("RxS ", DUMP_PREFIX_NONE, 32, 4,
                          (const void *)msg, wil->use_compressed_rx_status ?
                          sizeof(struct wil_rx_status_compressed) :
                          sizeof(struct wil_rx_status_extended), false);

        /* Move the buffer from the active list to the free list */
        list_move_tail(&wil->rx_buff_mgmt.buff_arr[buff_id].list,
                       &wil->rx_buff_mgmt.free);

        eop = wil_rx_status_get_eop(msg);

        cid = wil_rx_status_get_cid(msg);
        if (unlikely(!wil_val_in_range(cid, 0, wil->max_assoc_sta))) {
                wil_err(wil, "Corrupt cid=%d, sring->swhead=%d\n",
                        cid, sring->swhead);
                rxdata->skipping = true;
                goto skipping;
        }
        stats = &wil->sta[cid].stats;

        if (unlikely(dmalen < ETH_HLEN)) {
                wil_dbg_txrx(wil, "Short frame, len = %d\n", dmalen);
                stats->rx_short_frame++;
                rxdata->skipping = true;
                goto skipping;
        }

        if (unlikely(dmalen > sz)) {
                wil_err(wil, "Rx size too large: %d bytes!\n", dmalen);
                stats->rx_large_frame++;
                rxdata->skipping = true;
        }

skipping:
        /* skipping indicates if a certain SKB should be dropped.
         * It is set in case there is an error on the current SKB or in case
         * of RX chaining: as long as we manage to merge the SKBs it will
         * be false. once we have a bad SKB or we don't manage to merge SKBs
         * it will be set to the !EOP value of the current SKB.
         * This guarantees that all the following SKBs until EOP will also
         * get dropped.
         */
        if (unlikely(rxdata->skipping)) {
                kfree_skb(skb);
                if (rxdata->skb) {
                        kfree_skb(rxdata->skb);
                        rxdata->skb = NULL;
                }
                rxdata->skipping = !eop;
                goto again;
        }

        skb_trim(skb, dmalen);

        prefetch(skb->data);

        if (!rxdata->skb) {
                rxdata->skb = skb;
        } else {
                if (likely(skb_try_coalesce(rxdata->skb, skb, &headstolen,
                                            &delta))) {
                        kfree_skb_partial(skb, headstolen);
                } else {
                        wil_err(wil, "failed to merge skbs!\n");
                        kfree_skb(skb);
                        kfree_skb(rxdata->skb);
                        rxdata->skb = NULL;
                        rxdata->skipping = !eop;
                        goto again;
                }
        }

        if (!eop)
                goto again;

        /* reaching here rxdata->skb always contains a full packet */
        skb = rxdata->skb;
        rxdata->skb = NULL;
        rxdata->skipping = false;

        if (stats) {
                stats->last_mcs_rx = wil_rx_status_get_mcs(msg);
                if (stats->last_mcs_rx < ARRAY_SIZE(stats->rx_per_mcs))
                        stats->rx_per_mcs[stats->last_mcs_rx]++;

                stats->last_cb_mode_rx  = wil_rx_status_get_cb_mode(msg);
        }

        if (!wil->use_rx_hw_reordering && !wil->use_compressed_rx_status &&
            wil_check_bar(wil, msg, cid, skb, stats) == -EAGAIN) {
                kfree_skb(skb);
                goto again;
        }

        /* Compensate for the HW data alignment according to the status
         * message
         */
        data_offset = wil_rx_status_get_data_offset(msg);
        if (data_offset == 0xFF ||
            data_offset > WIL_EDMA_MAX_DATA_OFFSET) {
                wil_err(wil, "Unexpected data offset %d\n", data_offset);
                kfree_skb(skb);
                goto again;
        }

        skb_pull(skb, data_offset);

        wil_hex_dump_txrx("Rx ", DUMP_PREFIX_OFFSET, 16, 1,
                          skb->data, skb_headlen(skb), false);

        /* Has to be done after dma_unmap_single as skb->cb is also
         * used for holding the pa
         */
        s = wil_skb_rxstatus(skb);
        memcpy(s, msg, sring->elem_size);

        return skb;
}

void wil_rx_handle_edma(struct wil6210_priv *wil, int *quota)
{
        struct net_device *ndev;
        struct wil_ring *ring = &wil->ring_rx;
        struct wil_status_ring *sring;
        struct sk_buff *skb;
        int i;

        if (unlikely(!ring->va)) {
                wil_err(wil, "Rx IRQ while Rx not yet initialized\n");
                return;
        }
        wil_dbg_txrx(wil, "rx_handle\n");

        for (i = 0; i < wil->num_rx_status_rings; i++) {
                sring = &wil->srings[i];
                if (unlikely(!sring->va)) {
                        wil_err(wil,
                                "Rx IRQ while Rx status ring %d not yet initialized\n",
                                i);
                        continue;
                }

                while ((*quota > 0) &&
                       (NULL != (skb =
                        wil_sring_reap_rx_edma(wil, sring)))) {
                        (*quota)--;
                        if (wil->use_rx_hw_reordering) {
                                void *msg = wil_skb_rxstatus(skb);
                                int mid = wil_rx_status_get_mid(msg);
                                struct wil6210_vif *vif = wil->vifs[mid];

                                if (unlikely(!vif)) {
                                        wil_dbg_txrx(wil,
                                                     "RX desc invalid mid %d",
                                                     mid);
                                        kfree_skb(skb);
                                        continue;
                                }
                                ndev = vif_to_ndev(vif);
                                wil_netif_rx_any(skb, ndev);
                        } else {
                                wil_rx_reorder(wil, skb);
                        }
                }

                wil_w(wil, sring->hwtail, (sring->swhead - 1) % sring->size);
        }

        wil_rx_refill_edma(wil);
}

static int wil_tx_desc_map_edma(union wil_tx_desc *desc,
                                dma_addr_t pa,
                                u32 len,
                                int ring_index)
{
        struct wil_tx_enhanced_desc *d =
                (struct wil_tx_enhanced_desc *)&desc->enhanced;

        memset(d, 0, sizeof(struct wil_tx_enhanced_desc));

        wil_desc_set_addr_edma(&d->dma.addr, &d->dma.addr_high_high, pa);

        /* 0..6: mac_length; 7:ip_version 0-IP6 1-IP4*/
        d->dma.length = cpu_to_le16((u16)len);
        d->mac.d[0] = (ring_index << WIL_EDMA_DESC_TX_MAC_CFG_0_QID_POS);
        /* translation type:  0 - bypass; 1 - 802.3; 2 - native wifi;
         * 3 - eth mode
         */
        d->mac.d[2] = BIT(MAC_CFG_DESC_TX_2_SNAP_HDR_INSERTION_EN_POS) |
                      (0x3 << MAC_CFG_DESC_TX_2_L2_TRANSLATION_TYPE_POS);

        return 0;
}

static inline void
wil_get_next_tx_status_msg(struct wil_status_ring *sring, u8 *dr_bit,
                           struct wil_ring_tx_status *msg)
{
        struct wil_ring_tx_status *_msg = (struct wil_ring_tx_status *)
                (sring->va + (sring->elem_size * sring->swhead));

        *dr_bit = _msg->desc_ready >> TX_STATUS_DESC_READY_POS;
        /* make sure dr_bit is read before the rest of status msg */
        rmb();
        *msg = *_msg;
}

/**
 * Clean up transmitted skb's from the Tx descriptor RING.
 * Return number of descriptors cleared.
 */
int wil_tx_sring_handler(struct wil6210_priv *wil,
                         struct wil_status_ring *sring)
{
        struct net_device *ndev;
        struct device *dev = wil_to_dev(wil);
        struct wil_ring *ring = NULL;
        struct wil_ring_tx_data *txdata;
        /* Total number of completed descriptors in all descriptor rings */
        int desc_cnt = 0;
        int cid;
        struct wil_net_stats *stats;
        struct wil_tx_enhanced_desc *_d;
        unsigned int ring_id;
        unsigned int num_descs, num_statuses = 0;
        int i;
        u8 dr_bit; /* Descriptor Ready bit */
        struct wil_ring_tx_status msg;
        struct wil6210_vif *vif;
        int used_before_complete;
        int used_new;

        wil_get_next_tx_status_msg(sring, &dr_bit, &msg);

        /* Process completion messages while DR bit has the expected polarity */
        while (dr_bit == sring->desc_rdy_pol) {
                num_descs = msg.num_descriptors;
                if (!num_descs) {
                        wil_err(wil, "invalid num_descs 0\n");
                        goto again;
                }

                /* Find the corresponding descriptor ring */
                ring_id = msg.ring_id;

                if (unlikely(ring_id >= WIL6210_MAX_TX_RINGS)) {
                        wil_err(wil, "invalid ring id %d\n", ring_id);
                        goto again;
                }
                ring = &wil->ring_tx[ring_id];
                if (unlikely(!ring->va)) {
                        wil_err(wil, "Tx irq[%d]: ring not initialized\n",
                                ring_id);
                        goto again;
                }
                txdata = &wil->ring_tx_data[ring_id];
                if (unlikely(!txdata->enabled)) {
                        wil_info(wil, "Tx irq[%d]: ring disabled\n", ring_id);
                        goto again;
                }
                vif = wil->vifs[txdata->mid];
                if (unlikely(!vif)) {
                        wil_dbg_txrx(wil, "invalid MID %d for ring %d\n",
                                     txdata->mid, ring_id);
                        goto again;
                }

                ndev = vif_to_ndev(vif);

                cid = wil->ring2cid_tid[ring_id][0];
                stats = (cid < wil->max_assoc_sta) ? &wil->sta[cid].stats :
                                                     NULL;

                wil_dbg_txrx(wil,
                             "tx_status: completed desc_ring (%d), num_descs (%d)\n",
                             ring_id, num_descs);

                used_before_complete = wil_ring_used_tx(ring);

                for (i = 0 ; i < num_descs; ++i) {
                        struct wil_ctx *ctx = &ring->ctx[ring->swtail];
                        struct wil_tx_enhanced_desc dd, *d = &dd;
                        u16 dmalen;
                        struct sk_buff *skb = ctx->skb;

                        _d = (struct wil_tx_enhanced_desc *)
                                &ring->va[ring->swtail].tx.enhanced;
                        *d = *_d;

                        dmalen = le16_to_cpu(d->dma.length);
                        trace_wil6210_tx_status(&msg, ring->swtail, dmalen);
                        wil_dbg_txrx(wil,
                                     "TxC[%2d][%3d] : %d bytes, status 0x%02x\n",
                                     ring_id, ring->swtail, dmalen,
                                     msg.status);
                        wil_hex_dump_txrx("TxS ", DUMP_PREFIX_NONE, 32, 4,
                                          (const void *)&msg, sizeof(msg),
                                          false);

                        wil_tx_desc_unmap_edma(dev,
                                               (union wil_tx_desc *)d,
                                               ctx);

                        if (skb) {
                                if (likely(msg.status == 0)) {
                                        ndev->stats.tx_packets++;
                                        ndev->stats.tx_bytes += skb->len;
                                        if (stats) {
                                                stats->tx_packets++;
                                                stats->tx_bytes += skb->len;

                                                wil_tx_latency_calc(wil, skb,
                                                        &wil->sta[cid]);
                                        }
                                } else {
                                        ndev->stats.tx_errors++;
                                        if (stats)
                                                stats->tx_errors++;
                                }

                                if (skb->protocol == cpu_to_be16(ETH_P_PAE))
                                        wil_tx_complete_handle_eapol(vif, skb);

                                wil_consume_skb(skb, msg.status == 0);
                        }
                        memset(ctx, 0, sizeof(*ctx));
                        /* Make sure the ctx is zeroed before updating the tail
                         * to prevent a case where wil_tx_ring will see
                         * this descriptor as used and handle it before ctx zero
                         * is completed.
                         */
                        wmb();

                        ring->swtail = wil_ring_next_tail(ring);

                        desc_cnt++;
                }

                /* performance monitoring */
                used_new = wil_ring_used_tx(ring);
                if (wil_val_in_range(wil->ring_idle_trsh,
                                     used_new, used_before_complete)) {
                        wil_dbg_txrx(wil, "Ring[%2d] idle %d -> %d\n",
                                     ring_id, used_before_complete, used_new);
                        txdata->last_idle = get_cycles();
                }

again:
                num_statuses++;
                if (num_statuses % WIL_EDMA_TX_SRING_UPDATE_HW_TAIL == 0)
                        /* update HW tail to allow HW to push new statuses */
                        wil_w(wil, sring->hwtail, sring->swhead);

                wil_sring_advance_swhead(sring);

                wil_get_next_tx_status_msg(sring, &dr_bit, &msg);
        }

        /* shall we wake net queues? */
        if (desc_cnt)
                wil_update_net_queues(wil, vif, NULL, false);

        if (num_statuses % WIL_EDMA_TX_SRING_UPDATE_HW_TAIL != 0)
                /* Update the HW tail ptr (RD ptr) */
                wil_w(wil, sring->hwtail, (sring->swhead - 1) % sring->size);

        return desc_cnt;
}

/**
 * Sets the descriptor @d up for csum and/or TSO offloading. The corresponding
 * @skb is used to obtain the protocol and headers length.
 * @tso_desc_type is a descriptor type for TSO: 0 - a header, 1 - first data,
 * 2 - middle, 3 - last descriptor.
 */
static void wil_tx_desc_offload_setup_tso_edma(struct wil_tx_enhanced_desc *d,
                                               int tso_desc_type, bool is_ipv4,
                                               int tcp_hdr_len,
                                               int skb_net_hdr_len,
                                               int mss)
{
        /* Number of descriptors */
        d->mac.d[2] |= 1;
        /* Maximum Segment Size */
        d->mac.tso_mss |= cpu_to_le16(mss >> 2);
        /* L4 header len: TCP header length */
        d->dma.l4_hdr_len |= tcp_hdr_len & DMA_CFG_DESC_TX_0_L4_LENGTH_MSK;
        /* EOP, TSO desc type, Segmentation enable,
         * Insert IPv4 and TCP / UDP Checksum
         */
        d->dma.cmd |= BIT(WIL_EDMA_DESC_TX_CFG_EOP_POS) |
                      tso_desc_type << WIL_EDMA_DESC_TX_CFG_TSO_DESC_TYPE_POS |
                      BIT(WIL_EDMA_DESC_TX_CFG_SEG_EN_POS) |
                      BIT(WIL_EDMA_DESC_TX_CFG_INSERT_IP_CHKSUM_POS) |
                      BIT(WIL_EDMA_DESC_TX_CFG_INSERT_TCP_CHKSUM_POS);
        /* Calculate pseudo-header */
        d->dma.w1 |= BIT(WIL_EDMA_DESC_TX_CFG_PSEUDO_HEADER_CALC_EN_POS) |
                     BIT(WIL_EDMA_DESC_TX_CFG_L4_TYPE_POS);
        /* IP Header Length */
        d->dma.ip_length |= skb_net_hdr_len;
        /* MAC header length and IP address family*/
        d->dma.b11 |= ETH_HLEN |
                      is_ipv4 << DMA_CFG_DESC_TX_OFFLOAD_CFG_L3T_IPV4_POS;
}

static int wil_tx_tso_gen_desc(struct wil6210_priv *wil, void *buff_addr,
                               int len, uint i, int tso_desc_type,
                               skb_frag_t *frag, struct wil_ring *ring,
                               struct sk_buff *skb, bool is_ipv4,
                               int tcp_hdr_len, int skb_net_hdr_len,
                               int mss, int *descs_used)
{
        struct device *dev = wil_to_dev(wil);
        struct wil_tx_enhanced_desc *_desc = (struct wil_tx_enhanced_desc *)
                &ring->va[i].tx.enhanced;
        struct wil_tx_enhanced_desc desc_mem, *d = &desc_mem;
        int ring_index = ring - wil->ring_tx;
        dma_addr_t pa;

        if (len == 0)
                return 0;

        if (!frag) {
                pa = dma_map_single(dev, buff_addr, len, DMA_TO_DEVICE);
                ring->ctx[i].mapped_as = wil_mapped_as_single;
        } else {
                pa = skb_frag_dma_map(dev, frag, 0, len, DMA_TO_DEVICE);
                ring->ctx[i].mapped_as = wil_mapped_as_page;
        }
        if (unlikely(dma_mapping_error(dev, pa))) {
                wil_err(wil, "TSO: Skb DMA map error\n");
                return -EINVAL;
        }

        wil->txrx_ops.tx_desc_map((union wil_tx_desc *)d, pa,
                                   len, ring_index);
        wil_tx_desc_offload_setup_tso_edma(d, tso_desc_type, is_ipv4,
                                           tcp_hdr_len,
                                           skb_net_hdr_len, mss);

        /* hold reference to skb
         * to prevent skb release before accounting
         * in case of immediate "tx done"
         */
        if (tso_desc_type == wil_tso_type_lst)
                ring->ctx[i].skb = skb_get(skb);

        wil_hex_dump_txrx("TxD ", DUMP_PREFIX_NONE, 32, 4,
                          (const void *)d, sizeof(*d), false);

        *_desc = *d;
        (*descs_used)++;

        return 0;
}

static int __wil_tx_ring_tso_edma(struct wil6210_priv *wil,
                                  struct wil6210_vif *vif,
                                  struct wil_ring *ring,
                                  struct sk_buff *skb)
{
        int ring_index = ring - wil->ring_tx;
        struct wil_ring_tx_data *txdata = &wil->ring_tx_data[ring_index];
        int nr_frags = skb_shinfo(skb)->nr_frags;
        int min_desc_required = nr_frags + 2; /* Headers, Head, Fragments */
        int used, avail = wil_ring_avail_tx(ring);
        int f, hdrlen, headlen;
        int gso_type;
        bool is_ipv4;
        u32 swhead = ring->swhead;
        int descs_used = 0; /* total number of used descriptors */
        int rc = -EINVAL;
        int tcp_hdr_len;
        int skb_net_hdr_len;
        int mss = skb_shinfo(skb)->gso_size;

        wil_dbg_txrx(wil, "tx_ring_tso: %d bytes to ring %d\n", skb->len,
                     ring_index);

        if (unlikely(!txdata->enabled))
                return -EINVAL;

        if (unlikely(avail < min_desc_required)) {
                wil_err_ratelimited(wil,
                                    "TSO: Tx ring[%2d] full. No space for %d fragments\n",
                                    ring_index, min_desc_required);
                return -ENOMEM;
        }

        gso_type = skb_shinfo(skb)->gso_type & (SKB_GSO_TCPV6 | SKB_GSO_TCPV4);
        switch (gso_type) {
        case SKB_GSO_TCPV4:
                is_ipv4 = true;
                break;
        case SKB_GSO_TCPV6:
                is_ipv4 = false;
                break;
        default:
                return -EINVAL;
        }

        if (skb->ip_summed != CHECKSUM_PARTIAL)
                return -EINVAL;

        /* tcp header length and skb network header length are fixed for all
         * packet's descriptors - read them once here
         */
        tcp_hdr_len = tcp_hdrlen(skb);
        skb_net_hdr_len = skb_network_header_len(skb);

        /* First descriptor must contain the header only
         * Header Length = MAC header len + IP header len + TCP header len
         */
        hdrlen = ETH_HLEN + tcp_hdr_len + skb_net_hdr_len;
        wil_dbg_txrx(wil, "TSO: process header descriptor, hdrlen %u\n",
                     hdrlen);
        rc = wil_tx_tso_gen_desc(wil, skb->data, hdrlen, swhead,
                                 wil_tso_type_hdr, NULL, ring, skb,
                                 is_ipv4, tcp_hdr_len, skb_net_hdr_len,
                                 mss, &descs_used);
        if (rc)
                return -EINVAL;

        /* Second descriptor contains the head */
        headlen = skb_headlen(skb) - hdrlen;
        wil_dbg_txrx(wil, "TSO: process skb head, headlen %u\n", headlen);
        rc = wil_tx_tso_gen_desc(wil, skb->data + hdrlen, headlen,
                                 (swhead + descs_used) % ring->size,
                                 (nr_frags != 0) ? wil_tso_type_first :
                                 wil_tso_type_lst, NULL, ring, skb,
                                 is_ipv4, tcp_hdr_len, skb_net_hdr_len,
                                 mss, &descs_used);
        if (rc)
                goto mem_error;

        /* Rest of the descriptors are from the SKB fragments */
        for (f = 0; f < nr_frags; f++) {
                skb_frag_t *frag = &skb_shinfo(skb)->frags[f];
                int len = skb_frag_size(frag);

                wil_dbg_txrx(wil, "TSO: frag[%d]: len %u, descs_used %d\n", f,
                             len, descs_used);

                rc = wil_tx_tso_gen_desc(wil, NULL, len,
                                         (swhead + descs_used) % ring->size,
                                         (f != nr_frags - 1) ?
                                         wil_tso_type_mid : wil_tso_type_lst,
                                         frag, ring, skb, is_ipv4,
                                         tcp_hdr_len, skb_net_hdr_len,
                                         mss, &descs_used);
                if (rc)
                        goto mem_error;
        }

        /* performance monitoring */
        used = wil_ring_used_tx(ring);
        if (wil_val_in_range(wil->ring_idle_trsh,
                             used, used + descs_used)) {
                txdata->idle += get_cycles() - txdata->last_idle;
                wil_dbg_txrx(wil,  "Ring[%2d] not idle %d -> %d\n",
                             ring_index, used, used + descs_used);
        }

        /* advance swhead */
        wil_ring_advance_head(ring, descs_used);
        wil_dbg_txrx(wil, "TSO: Tx swhead %d -> %d\n", swhead, ring->swhead);

        /* make sure all writes to descriptors (shared memory) are done before
         * committing them to HW
         */
        wmb();

        if (wil->tx_latency)
                *(ktime_t *)&skb->cb = ktime_get();
        else
                memset(skb->cb, 0, sizeof(ktime_t));

        wil_w(wil, ring->hwtail, ring->swhead);

        return 0;

mem_error:
        while (descs_used > 0) {
                struct device *dev = wil_to_dev(wil);
                struct wil_ctx *ctx;
                int i = (swhead + descs_used - 1) % ring->size;
                struct wil_tx_enhanced_desc dd, *d = &dd;
                struct wil_tx_enhanced_desc *_desc =
                        (struct wil_tx_enhanced_desc *)
                        &ring->va[i].tx.enhanced;

                *d = *_desc;
                ctx = &ring->ctx[i];
                wil_tx_desc_unmap_edma(dev, (union wil_tx_desc *)d, ctx);
                memset(ctx, 0, sizeof(*ctx));
                descs_used--;
        }
        return rc;
}

static int wil_ring_init_bcast_edma(struct wil6210_vif *vif, int ring_id,
                                    int size)
{
        struct wil6210_priv *wil = vif_to_wil(vif);
        struct wil_ring *ring = &wil->ring_tx[ring_id];
        int rc;
        struct wil_ring_tx_data *txdata = &wil->ring_tx_data[ring_id];

        wil_dbg_misc(wil, "init bcast: ring_id=%d, sring_id=%d\n",
                     ring_id, wil->tx_sring_idx);

        lockdep_assert_held(&wil->mutex);

        wil_tx_data_init(txdata);
        ring->size = size;
        ring->is_rx = false;
        rc = wil_ring_alloc_desc_ring(wil, ring);
        if (rc)
                goto out;

        wil->ring2cid_tid[ring_id][0] = WIL6210_MAX_CID; /* CID */
        wil->ring2cid_tid[ring_id][1] = 0; /* TID */
        if (!vif->privacy)
                txdata->dot1x_open = true;

        rc = wil_wmi_bcast_desc_ring_add(vif, ring_id);
        if (rc)
                goto out_free;

        return 0;

 out_free:
        spin_lock_bh(&txdata->lock);
        txdata->enabled = 0;
        txdata->dot1x_open = false;
        spin_unlock_bh(&txdata->lock);
        wil_ring_free_edma(wil, ring);

out:
        return rc;
}

static void wil_tx_fini_edma(struct wil6210_priv *wil)
{
        struct wil_status_ring *sring = &wil->srings[wil->tx_sring_idx];

        wil_dbg_misc(wil, "free TX sring\n");

        wil_sring_free(wil, sring);
}

static void wil_rx_data_free(struct wil_status_ring *sring)
{
        if (!sring)
                return;

        kfree_skb(sring->rx_data.skb);
        sring->rx_data.skb = NULL;
}

static void wil_rx_fini_edma(struct wil6210_priv *wil)
{
        struct wil_ring *ring = &wil->ring_rx;
        int i;

        wil_dbg_misc(wil, "rx_fini_edma\n");

        wil_ring_free_edma(wil, ring);

        for (i = 0; i < wil->num_rx_status_rings; i++) {
                wil_rx_data_free(&wil->srings[i]);
                wil_sring_free(wil, &wil->srings[i]);
        }

        wil_free_rx_buff_arr(wil);
}

void wil_init_txrx_ops_edma(struct wil6210_priv *wil)
{
        wil->txrx_ops.configure_interrupt_moderation =
                wil_configure_interrupt_moderation_edma;
        /* TX ops */
        wil->txrx_ops.ring_init_tx = wil_ring_init_tx_edma;
        wil->txrx_ops.ring_fini_tx = wil_ring_free_edma;
        wil->txrx_ops.ring_init_bcast = wil_ring_init_bcast_edma;
        wil->txrx_ops.tx_init = wil_tx_init_edma;
        wil->txrx_ops.tx_fini = wil_tx_fini_edma;
        wil->txrx_ops.tx_desc_map = wil_tx_desc_map_edma;
        wil->txrx_ops.tx_desc_unmap = wil_tx_desc_unmap_edma;
        wil->txrx_ops.tx_ring_tso = __wil_tx_ring_tso_edma;
        wil->txrx_ops.tx_ring_modify = wil_tx_ring_modify_edma;
        /* RX ops */
        wil->txrx_ops.rx_init = wil_rx_init_edma;
        wil->txrx_ops.wmi_addba_rx_resp = wmi_addba_rx_resp_edma;
        wil->txrx_ops.get_reorder_params = wil_get_reorder_params_edma;
        wil->txrx_ops.get_netif_rx_params = wil_get_netif_rx_params_edma;
        wil->txrx_ops.rx_crypto_check = wil_rx_crypto_check_edma;
        wil->txrx_ops.rx_error_check = wil_rx_error_check_edma;
        wil->txrx_ops.is_rx_idle = wil_is_rx_idle_edma;
        wil->txrx_ops.rx_fini = wil_rx_fini_edma;
}