Quelle  dp_rx.c   Sprache: C

// SPDX-License-Identifier: BSD-3-Clause-Clear
/*
 * Copyright (c) 2018-2021 The Linux Foundation. All rights reserved.
 * Copyright (c) 2021-2025 Qualcomm Innovation Center, Inc. All rights reserved.
 */

#include <linux/ieee80211.h>
#include <linux/kernel.h>
#include <linux/skbuff.h>
#include <crypto/hash.h>
#include "core.h"
#include "debug.h"
#include "hal_desc.h"
#include "hw.h"
#include "dp_rx.h"
#include "hal_rx.h"
#include "dp_tx.h"
#include "peer.h"
#include "dp_mon.h"
#include "debugfs_htt_stats.h"

#define ATH12K_DP_RX_FRAGMENT_TIMEOUT_MS (2 * HZ)

static int ath12k_dp_rx_tid_delete_handler(struct ath12k_base *ab,
          struct ath12k_dp_rx_tid *rx_tid);

static enum hal_encrypt_type ath12k_dp_rx_h_enctype(struct ath12k_base *ab,
          struct hal_rx_desc *desc)
{
 if (!ab->hal_rx_ops->rx_desc_encrypt_valid(desc))
  return HAL_ENCRYPT_TYPE_OPEN;

 return ab->hal_rx_ops->rx_desc_get_encrypt_type(desc);
}

u8 ath12k_dp_rx_h_decap_type(struct ath12k_base *ab,
        struct hal_rx_desc *desc)
{
 return ab->hal_rx_ops->rx_desc_get_decap_type(desc);
}

static u8 ath12k_dp_rx_h_mesh_ctl_present(struct ath12k_base *ab,
       struct hal_rx_desc *desc)
{
 return ab->hal_rx_ops->rx_desc_get_mesh_ctl(desc);
}

static bool ath12k_dp_rx_h_seq_ctrl_valid(struct ath12k_base *ab,
       struct hal_rx_desc *desc)
{
 return ab->hal_rx_ops->rx_desc_get_mpdu_seq_ctl_vld(desc);
}

static bool ath12k_dp_rx_h_fc_valid(struct ath12k_base *ab,
        struct hal_rx_desc *desc)
{
 return ab->hal_rx_ops->rx_desc_get_mpdu_fc_valid(desc);
}

static bool ath12k_dp_rx_h_more_frags(struct ath12k_base *ab,
          struct sk_buff *skb)
{
 struct ieee80211_hdr *hdr;

 hdr = (struct ieee80211_hdr *)(skb->data + ab->hal.hal_desc_sz);
 return ieee80211_has_morefrags(hdr->frame_control);
}

static u16 ath12k_dp_rx_h_frag_no(struct ath12k_base *ab,
      struct sk_buff *skb)
{
 struct ieee80211_hdr *hdr;

 hdr = (struct ieee80211_hdr *)(skb->data + ab->hal.hal_desc_sz);
 return le16_to_cpu(hdr->seq_ctrl) & IEEE80211_SCTL_FRAG;
}

static u16 ath12k_dp_rx_h_seq_no(struct ath12k_base *ab,
     struct hal_rx_desc *desc)
{
 return ab->hal_rx_ops->rx_desc_get_mpdu_start_seq_no(desc);
}

static bool ath12k_dp_rx_h_msdu_done(struct ath12k_base *ab,
         struct hal_rx_desc *desc)
{
 return ab->hal_rx_ops->dp_rx_h_msdu_done(desc);
}

static bool ath12k_dp_rx_h_l4_cksum_fail(struct ath12k_base *ab,
      struct hal_rx_desc *desc)
{
 return ab->hal_rx_ops->dp_rx_h_l4_cksum_fail(desc);
}

static bool ath12k_dp_rx_h_ip_cksum_fail(struct ath12k_base *ab,
      struct hal_rx_desc *desc)
{
 return ab->hal_rx_ops->dp_rx_h_ip_cksum_fail(desc);
}

static bool ath12k_dp_rx_h_is_decrypted(struct ath12k_base *ab,
     struct hal_rx_desc *desc)
{
 return ab->hal_rx_ops->dp_rx_h_is_decrypted(desc);
}

u32 ath12k_dp_rx_h_mpdu_err(struct ath12k_base *ab,
       struct hal_rx_desc *desc)
{
 return ab->hal_rx_ops->dp_rx_h_mpdu_err(desc);
}

static u16 ath12k_dp_rx_h_msdu_len(struct ath12k_base *ab,
       struct hal_rx_desc *desc)
{
 return ab->hal_rx_ops->rx_desc_get_msdu_len(desc);
}

static u8 ath12k_dp_rx_h_sgi(struct ath12k_base *ab,
        struct hal_rx_desc *desc)
{
 return ab->hal_rx_ops->rx_desc_get_msdu_sgi(desc);
}

static u8 ath12k_dp_rx_h_rate_mcs(struct ath12k_base *ab,
      struct hal_rx_desc *desc)
{
 return ab->hal_rx_ops->rx_desc_get_msdu_rate_mcs(desc);
}

static u8 ath12k_dp_rx_h_rx_bw(struct ath12k_base *ab,
          struct hal_rx_desc *desc)
{
 return ab->hal_rx_ops->rx_desc_get_msdu_rx_bw(desc);
}

static u32 ath12k_dp_rx_h_freq(struct ath12k_base *ab,
          struct hal_rx_desc *desc)
{
 return ab->hal_rx_ops->rx_desc_get_msdu_freq(desc);
}

static u8 ath12k_dp_rx_h_pkt_type(struct ath12k_base *ab,
      struct hal_rx_desc *desc)
{
 return ab->hal_rx_ops->rx_desc_get_msdu_pkt_type(desc);
}

static u8 ath12k_dp_rx_h_nss(struct ath12k_base *ab,
        struct hal_rx_desc *desc)
{
 return hweight8(ab->hal_rx_ops->rx_desc_get_msdu_nss(desc));
}

static u8 ath12k_dp_rx_h_tid(struct ath12k_base *ab,
        struct hal_rx_desc *desc)
{
 return ab->hal_rx_ops->rx_desc_get_mpdu_tid(desc);
}

static u16 ath12k_dp_rx_h_peer_id(struct ath12k_base *ab,
      struct hal_rx_desc *desc)
{
 return ab->hal_rx_ops->rx_desc_get_mpdu_peer_id(desc);
}

u8 ath12k_dp_rx_h_l3pad(struct ath12k_base *ab,
   struct hal_rx_desc *desc)
{
 return ab->hal_rx_ops->rx_desc_get_l3_pad_bytes(desc);
}

static bool ath12k_dp_rx_h_first_msdu(struct ath12k_base *ab,
          struct hal_rx_desc *desc)
{
 return ab->hal_rx_ops->rx_desc_get_first_msdu(desc);
}

static bool ath12k_dp_rx_h_last_msdu(struct ath12k_base *ab,
         struct hal_rx_desc *desc)
{
 return ab->hal_rx_ops->rx_desc_get_last_msdu(desc);
}

static void ath12k_dp_rx_desc_end_tlv_copy(struct ath12k_base *ab,
        struct hal_rx_desc *fdesc,
        struct hal_rx_desc *ldesc)
{
 ab->hal_rx_ops->rx_desc_copy_end_tlv(fdesc, ldesc);
}

static void ath12k_dp_rxdesc_set_msdu_len(struct ath12k_base *ab,
       struct hal_rx_desc *desc,
       u16 len)
{
 ab->hal_rx_ops->rx_desc_set_msdu_len(desc, len);
}

u32 ath12k_dp_rxdesc_get_ppduid(struct ath12k_base *ab,
    struct hal_rx_desc *rx_desc)
{
 return ab->hal_rx_ops->rx_desc_get_mpdu_ppdu_id(rx_desc);
}

bool ath12k_dp_rxdesc_mpdu_valid(struct ath12k_base *ab,
     struct hal_rx_desc *rx_desc)
{
 u32 tlv_tag;

 tlv_tag = ab->hal_rx_ops->rx_desc_get_mpdu_start_tag(rx_desc);

 return tlv_tag == HAL_RX_MPDU_START;
}

static bool ath12k_dp_rx_h_is_da_mcbc(struct ath12k_base *ab,
          struct hal_rx_desc *desc)
{
 return (ath12k_dp_rx_h_first_msdu(ab, desc) &&
  ab->hal_rx_ops->rx_desc_is_da_mcbc(desc));
}

static bool ath12k_dp_rxdesc_mac_addr2_valid(struct ath12k_base *ab,
          struct hal_rx_desc *desc)
{
 return ab->hal_rx_ops->rx_desc_mac_addr2_valid(desc);
}

static u8 *ath12k_dp_rxdesc_get_mpdu_start_addr2(struct ath12k_base *ab,
       struct hal_rx_desc *desc)
{
 return ab->hal_rx_ops->rx_desc_mpdu_start_addr2(desc);
}

static void ath12k_dp_rx_desc_get_dot11_hdr(struct ath12k_base *ab,
         struct hal_rx_desc *desc,
         struct ieee80211_hdr *hdr)
{
 ab->hal_rx_ops->rx_desc_get_dot11_hdr(desc, hdr);
}

static void ath12k_dp_rx_desc_get_crypto_header(struct ath12k_base *ab,
      struct hal_rx_desc *desc,
      u8 *crypto_hdr,
      enum hal_encrypt_type enctype)
{
 ab->hal_rx_ops->rx_desc_get_crypto_header(desc, crypto_hdr, enctype);
}

static inline u8 ath12k_dp_rx_get_msdu_src_link(struct ath12k_base *ab,
      struct hal_rx_desc *desc)
{
 return ab->hal_rx_ops->rx_desc_get_msdu_src_link_id(desc);
}

static void ath12k_dp_clean_up_skb_list(struct sk_buff_head *skb_list)
{
 struct sk_buff *skb;

 while ((skb = __skb_dequeue(skb_list)))
  dev_kfree_skb_any(skb);
}

static size_t ath12k_dp_list_cut_nodes(struct list_head *list,
           struct list_head *head,
           size_t count)
{
 struct list_head *cur;
 struct ath12k_rx_desc_info *rx_desc;
 size_t nodes = 0;

 if (!count) {
  INIT_LIST_HEAD(list);
  goto out;
 }

 list_for_each(cur, head) {
  if (!count)
   break;

  rx_desc = list_entry(cur, struct ath12k_rx_desc_info, list);
  rx_desc->in_use = true;

  count--;
  nodes++;
 }

 list_cut_before(list, head, cur);
out:
 return nodes;
}

static void ath12k_dp_rx_enqueue_free(struct ath12k_dp *dp,
          struct list_head *used_list)
{
 struct ath12k_rx_desc_info *rx_desc, *safe;

 /* Reset the use flag */
 list_for_each_entry_safe(rx_desc, safe, used_list, list)
  rx_desc->in_use = false;

 spin_lock_bh(&dp->rx_desc_lock);
 list_splice_tail(used_list, &dp->rx_desc_free_list);
 spin_unlock_bh(&dp->rx_desc_lock);
}

/* Returns number of Rx buffers replenished */
int ath12k_dp_rx_bufs_replenish(struct ath12k_base *ab,
    struct dp_rxdma_ring *rx_ring,
    struct list_head *used_list,
    int req_entries)
{
 struct ath12k_buffer_addr *desc;
 struct hal_srng *srng;
 struct sk_buff *skb;
 int num_free;
 int num_remain;
 u32 cookie;
 dma_addr_t paddr;
 struct ath12k_dp *dp = &ab->dp;
 struct ath12k_rx_desc_info *rx_desc;
 enum hal_rx_buf_return_buf_manager mgr = ab->hw_params->hal_params->rx_buf_rbm;

 req_entries = min(req_entries, rx_ring->bufs_max);

 srng = &ab->hal.srng_list[rx_ring->refill_buf_ring.ring_id];

 spin_lock_bh(&srng->lock);

 ath12k_hal_srng_access_begin(ab, srng);

 num_free = ath12k_hal_srng_src_num_free(ab, srng, true);
 if (!req_entries && (num_free > (rx_ring->bufs_max * 3) / 4))
  req_entries = num_free;

 req_entries = min(num_free, req_entries);
 num_remain = req_entries;

 if (!num_remain)
  goto out;

 /* Get the descriptor from free list */
 if (list_empty(used_list)) {
  spin_lock_bh(&dp->rx_desc_lock);
  req_entries = ath12k_dp_list_cut_nodes(used_list,
             &dp->rx_desc_free_list,
             num_remain);
  spin_unlock_bh(&dp->rx_desc_lock);
  num_remain = req_entries;
 }

 while (num_remain > 0) {
  skb = dev_alloc_skb(DP_RX_BUFFER_SIZE +
        DP_RX_BUFFER_ALIGN_SIZE);
  if (!skb)
   break;

  if (!IS_ALIGNED((unsigned long)skb->data,
    DP_RX_BUFFER_ALIGN_SIZE)) {
   skb_pull(skb,
     PTR_ALIGN(skb->data, DP_RX_BUFFER_ALIGN_SIZE) -
     skb->data);
  }

  paddr = dma_map_single(ab->dev, skb->data,
           skb->len + skb_tailroom(skb),
           DMA_FROM_DEVICE);
  if (dma_mapping_error(ab->dev, paddr))
   goto fail_free_skb;

  rx_desc = list_first_entry_or_null(used_list,
         struct ath12k_rx_desc_info,
         list);
  if (!rx_desc)
   goto fail_dma_unmap;

  rx_desc->skb = skb;
  cookie = rx_desc->cookie;

  desc = ath12k_hal_srng_src_get_next_entry(ab, srng);
  if (!desc)
   goto fail_dma_unmap;

  list_del(&rx_desc->list);
  ATH12K_SKB_RXCB(skb)->paddr = paddr;

  num_remain--;

  ath12k_hal_rx_buf_addr_info_set(desc, paddr, cookie, mgr);
 }

 goto out;

fail_dma_unmap:
 dma_unmap_single(ab->dev, paddr, skb->len + skb_tailroom(skb),
    DMA_FROM_DEVICE);
fail_free_skb:
 dev_kfree_skb_any(skb);
out:
 ath12k_hal_srng_access_end(ab, srng);

 if (!list_empty(used_list))
  ath12k_dp_rx_enqueue_free(dp, used_list);

 spin_unlock_bh(&srng->lock);

 return req_entries - num_remain;
}

static int ath12k_dp_rxdma_mon_buf_ring_free(struct ath12k_base *ab,
          struct dp_rxdma_mon_ring *rx_ring)
{
 struct sk_buff *skb;
 int buf_id;

 spin_lock_bh(&rx_ring->idr_lock);
 idr_for_each_entry(&rx_ring->bufs_idr, skb, buf_id) {
  idr_remove(&rx_ring->bufs_idr, buf_id);
  /* TODO: Understand where internal driver does this dma_unmap
 * of rxdma_buffer.
 */
  dma_unmap_single(ab->dev, ATH12K_SKB_RXCB(skb)->paddr,
     skb->len + skb_tailroom(skb), DMA_FROM_DEVICE);
  dev_kfree_skb_any(skb);
 }

 idr_destroy(&rx_ring->bufs_idr);
 spin_unlock_bh(&rx_ring->idr_lock);

 return 0;
}

static int ath12k_dp_rxdma_buf_free(struct ath12k_base *ab)
{
 struct ath12k_dp *dp = &ab->dp;
 int i;

 ath12k_dp_rxdma_mon_buf_ring_free(ab, &dp->rxdma_mon_buf_ring);

 if (ab->hw_params->rxdma1_enable)
  return 0;

 for (i = 0; i < ab->hw_params->num_rxdma_per_pdev; i++)
  ath12k_dp_rxdma_mon_buf_ring_free(ab,
        &dp->rx_mon_status_refill_ring[i]);

 return 0;
}

static int ath12k_dp_rxdma_mon_ring_buf_setup(struct ath12k_base *ab,
           struct dp_rxdma_mon_ring *rx_ring,
           u32 ringtype)
{
 int num_entries;

 num_entries = rx_ring->refill_buf_ring.size /
  ath12k_hal_srng_get_entrysize(ab, ringtype);

 rx_ring->bufs_max = num_entries;

 if (ringtype == HAL_RXDMA_MONITOR_STATUS)
  ath12k_dp_mon_status_bufs_replenish(ab, rx_ring,
          num_entries);
 else
  ath12k_dp_mon_buf_replenish(ab, rx_ring, num_entries);

 return 0;
}

static int ath12k_dp_rxdma_ring_buf_setup(struct ath12k_base *ab,
       struct dp_rxdma_ring *rx_ring)
{
 LIST_HEAD(list);

 rx_ring->bufs_max = rx_ring->refill_buf_ring.size /
   ath12k_hal_srng_get_entrysize(ab, HAL_RXDMA_BUF);

 ath12k_dp_rx_bufs_replenish(ab, rx_ring, &list, 0);

 return 0;
}

static int ath12k_dp_rxdma_buf_setup(struct ath12k_base *ab)
{
 struct ath12k_dp *dp = &ab->dp;
 struct dp_rxdma_mon_ring *mon_ring;
 int ret, i;

 ret = ath12k_dp_rxdma_ring_buf_setup(ab, &dp->rx_refill_buf_ring);
 if (ret) {
  ath12k_warn(ab,
       "failed to setup HAL_RXDMA_BUF\n");
  return ret;
 }

 if (ab->hw_params->rxdma1_enable) {
  ret = ath12k_dp_rxdma_mon_ring_buf_setup(ab,
        &dp->rxdma_mon_buf_ring,
        HAL_RXDMA_MONITOR_BUF);
  if (ret)
   ath12k_warn(ab,
        "failed to setup HAL_RXDMA_MONITOR_BUF\n");
  return ret;
 }

 for (i = 0; i < ab->hw_params->num_rxdma_per_pdev; i++) {
  mon_ring = &dp->rx_mon_status_refill_ring[i];
  ret = ath12k_dp_rxdma_mon_ring_buf_setup(ab, mon_ring,
        HAL_RXDMA_MONITOR_STATUS);
  if (ret) {
   ath12k_warn(ab,
        "failed to setup HAL_RXDMA_MONITOR_STATUS\n");
   return ret;
  }
 }

 return 0;
}

static void ath12k_dp_rx_pdev_srng_free(struct ath12k *ar)
{
 struct ath12k_pdev_dp *dp = &ar->dp;
 struct ath12k_base *ab = ar->ab;
 int i;

 for (i = 0; i < ab->hw_params->num_rxdma_per_pdev; i++)
  ath12k_dp_srng_cleanup(ab, &dp->rxdma_mon_dst_ring[i]);
}

void ath12k_dp_rx_pdev_reo_cleanup(struct ath12k_base *ab)
{
 struct ath12k_dp *dp = &ab->dp;
 int i;

 for (i = 0; i < DP_REO_DST_RING_MAX; i++)
  ath12k_dp_srng_cleanup(ab, &dp->reo_dst_ring[i]);
}

int ath12k_dp_rx_pdev_reo_setup(struct ath12k_base *ab)
{
 struct ath12k_dp *dp = &ab->dp;
 int ret;
 int i;

 for (i = 0; i < DP_REO_DST_RING_MAX; i++) {
  ret = ath12k_dp_srng_setup(ab, &dp->reo_dst_ring[i],
        HAL_REO_DST, i, 0,
        DP_REO_DST_RING_SIZE);
  if (ret) {
   ath12k_warn(ab, "failed to setup reo_dst_ring\n");
   goto err_reo_cleanup;
  }
 }

 return 0;

err_reo_cleanup:
 ath12k_dp_rx_pdev_reo_cleanup(ab);

 return ret;
}

static int ath12k_dp_rx_pdev_srng_alloc(struct ath12k *ar)
{
 struct ath12k_pdev_dp *dp = &ar->dp;
 struct ath12k_base *ab = ar->ab;
 int i;
 int ret;
 u32 mac_id = dp->mac_id;

 for (i = 0; i < ab->hw_params->num_rxdma_per_pdev; i++) {
  ret = ath12k_dp_srng_setup(ar->ab,
        &dp->rxdma_mon_dst_ring[i],
        HAL_RXDMA_MONITOR_DST,
        0, mac_id + i,
        DP_RXDMA_MONITOR_DST_RING_SIZE(ab));
  if (ret) {
   ath12k_warn(ar->ab,
        "failed to setup HAL_RXDMA_MONITOR_DST\n");
   return ret;
  }
 }

 return 0;
}

void ath12k_dp_rx_reo_cmd_list_cleanup(struct ath12k_base *ab)
{
 struct ath12k_dp *dp = &ab->dp;
 struct ath12k_dp_rx_reo_cmd *cmd, *tmp;
 struct ath12k_dp_rx_reo_cache_flush_elem *cmd_cache, *tmp_cache;

 spin_lock_bh(&dp->reo_cmd_lock);
 list_for_each_entry_safe(cmd, tmp, &dp->reo_cmd_list, list) {
  list_del(&cmd->list);
  dma_unmap_single(ab->dev, cmd->data.qbuf.paddr_aligned,
     cmd->data.qbuf.size, DMA_BIDIRECTIONAL);
  kfree(cmd->data.qbuf.vaddr);
  kfree(cmd);
 }

 list_for_each_entry_safe(cmd_cache, tmp_cache,
     &dp->reo_cmd_cache_flush_list, list) {
  list_del(&cmd_cache->list);
  dp->reo_cmd_cache_flush_count--;
  dma_unmap_single(ab->dev, cmd_cache->data.qbuf.paddr_aligned,
     cmd_cache->data.qbuf.size, DMA_BIDIRECTIONAL);
  kfree(cmd_cache->data.qbuf.vaddr);
  kfree(cmd_cache);
 }
 spin_unlock_bh(&dp->reo_cmd_lock);
}

static void ath12k_dp_reo_cmd_free(struct ath12k_dp *dp, void *ctx,
       enum hal_reo_cmd_status status)
{
 struct ath12k_dp_rx_tid *rx_tid = ctx;

 if (status != HAL_REO_CMD_SUCCESS)
  ath12k_warn(dp->ab, "failed to flush rx tid hw desc, tid %d status %d\n",
       rx_tid->tid, status);

 dma_unmap_single(dp->ab->dev, rx_tid->qbuf.paddr_aligned, rx_tid->qbuf.size,
    DMA_BIDIRECTIONAL);
 kfree(rx_tid->qbuf.vaddr);
 rx_tid->qbuf.vaddr = NULL;
}

static int ath12k_dp_reo_cmd_send(struct ath12k_base *ab, struct ath12k_dp_rx_tid *rx_tid,
      enum hal_reo_cmd_type type,
      struct ath12k_hal_reo_cmd *cmd,
      void (*cb)(struct ath12k_dp *dp, void *ctx,
          enum hal_reo_cmd_status status))
{
 struct ath12k_dp *dp = &ab->dp;
 struct ath12k_dp_rx_reo_cmd *dp_cmd;
 struct hal_srng *cmd_ring;
 int cmd_num;

 cmd_ring = &ab->hal.srng_list[dp->reo_cmd_ring.ring_id];
 cmd_num = ath12k_hal_reo_cmd_send(ab, cmd_ring, type, cmd);

 /* cmd_num should start from 1, during failure return the error code */
 if (cmd_num < 0)
  return cmd_num;

 /* reo cmd ring descriptors has cmd_num starting from 1 */
 if (cmd_num == 0)
  return -EINVAL;

 if (!cb)
  return 0;

 /* Can this be optimized so that we keep the pending command list only
 * for tid delete command to free up the resource on the command status
 * indication?
 */
 dp_cmd = kzalloc(sizeof(*dp_cmd), GFP_ATOMIC);

 if (!dp_cmd)
  return -ENOMEM;

 memcpy(&dp_cmd->data, rx_tid, sizeof(*rx_tid));
 dp_cmd->cmd_num = cmd_num;
 dp_cmd->handler = cb;

 spin_lock_bh(&dp->reo_cmd_lock);
 list_add_tail(&dp_cmd->list, &dp->reo_cmd_list);
 spin_unlock_bh(&dp->reo_cmd_lock);

 return 0;
}

static void ath12k_dp_reo_cache_flush(struct ath12k_base *ab,
          struct ath12k_dp_rx_tid *rx_tid)
{
 struct ath12k_hal_reo_cmd cmd = {};
 unsigned long tot_desc_sz, desc_sz;
 int ret;

 tot_desc_sz = rx_tid->qbuf.size;
 desc_sz = ath12k_hal_reo_qdesc_size(0, HAL_DESC_REO_NON_QOS_TID);

 while (tot_desc_sz > desc_sz) {
  tot_desc_sz -= desc_sz;
  cmd.addr_lo = lower_32_bits(rx_tid->qbuf.paddr_aligned + tot_desc_sz);
  cmd.addr_hi = upper_32_bits(rx_tid->qbuf.paddr_aligned);
  ret = ath12k_dp_reo_cmd_send(ab, rx_tid,
          HAL_REO_CMD_FLUSH_CACHE, &cmd,
          NULL);
  if (ret)
   ath12k_warn(ab,
        "failed to send HAL_REO_CMD_FLUSH_CACHE, tid %d (%d)\n",
        rx_tid->tid, ret);
 }

 memset(&cmd, 0, sizeof(cmd));
 cmd.addr_lo = lower_32_bits(rx_tid->qbuf.paddr_aligned);
 cmd.addr_hi = upper_32_bits(rx_tid->qbuf.paddr_aligned);
 cmd.flag = HAL_REO_CMD_FLG_NEED_STATUS;
 ret = ath12k_dp_reo_cmd_send(ab, rx_tid,
         HAL_REO_CMD_FLUSH_CACHE,
         &cmd, ath12k_dp_reo_cmd_free);
 if (ret) {
  ath12k_err(ab, "failed to send HAL_REO_CMD_FLUSH_CACHE cmd, tid %d (%d)\n",
      rx_tid->tid, ret);
  dma_unmap_single(ab->dev, rx_tid->qbuf.paddr_aligned, rx_tid->qbuf.size,
     DMA_BIDIRECTIONAL);
  kfree(rx_tid->qbuf.vaddr);
  rx_tid->qbuf.vaddr = NULL;
 }
}

static void ath12k_dp_rx_tid_del_func(struct ath12k_dp *dp, void *ctx,
          enum hal_reo_cmd_status status)
{
 struct ath12k_base *ab = dp->ab;
 struct ath12k_dp_rx_tid *rx_tid = ctx;
 struct ath12k_dp_rx_reo_cache_flush_elem *elem, *tmp;

 if (status == HAL_REO_CMD_DRAIN) {
  goto free_desc;
 } else if (status != HAL_REO_CMD_SUCCESS) {
  /* Shouldn't happen! Cleanup in case of other failure? */
  ath12k_warn(ab, "failed to delete rx tid %d hw descriptor %d\n",
       rx_tid->tid, status);
  return;
 }

 elem = kzalloc(sizeof(*elem), GFP_ATOMIC);
 if (!elem)
  goto free_desc;

 elem->ts = jiffies;
 memcpy(&elem->data, rx_tid, sizeof(*rx_tid));

 spin_lock_bh(&dp->reo_cmd_lock);
 list_add_tail(&elem->list, &dp->reo_cmd_cache_flush_list);
 dp->reo_cmd_cache_flush_count++;

 /* Flush and invalidate aged REO desc from HW cache */
 list_for_each_entry_safe(elem, tmp, &dp->reo_cmd_cache_flush_list,
     list) {
  if (dp->reo_cmd_cache_flush_count > ATH12K_DP_RX_REO_DESC_FREE_THRES ||
      time_after(jiffies, elem->ts +
          msecs_to_jiffies(ATH12K_DP_RX_REO_DESC_FREE_TIMEOUT_MS))) {
   list_del(&elem->list);
   dp->reo_cmd_cache_flush_count--;

   /* Unlock the reo_cmd_lock before using ath12k_dp_reo_cmd_send()
 * within ath12k_dp_reo_cache_flush. The reo_cmd_cache_flush_list
 * is used in only two contexts, one is in this function called
 * from napi and the other in ath12k_dp_free during core destroy.
 * Before dp_free, the irqs would be disabled and would wait to
 * synchronize. Hence there wouldn’t be any race against add or
 * delete to this list. Hence unlock-lock is safe here.
 */
   spin_unlock_bh(&dp->reo_cmd_lock);

   ath12k_dp_reo_cache_flush(ab, &elem->data);
   kfree(elem);
   spin_lock_bh(&dp->reo_cmd_lock);
  }
 }
 spin_unlock_bh(&dp->reo_cmd_lock);

 return;
free_desc:
 dma_unmap_single(ab->dev, rx_tid->qbuf.paddr_aligned, rx_tid->qbuf.size,
    DMA_BIDIRECTIONAL);
 kfree(rx_tid->qbuf.vaddr);
 rx_tid->qbuf.vaddr = NULL;
}

static int ath12k_dp_rx_tid_delete_handler(struct ath12k_base *ab,
          struct ath12k_dp_rx_tid *rx_tid)
{
 struct ath12k_hal_reo_cmd cmd = {};

 cmd.flag = HAL_REO_CMD_FLG_NEED_STATUS;
 cmd.addr_lo = lower_32_bits(rx_tid->qbuf.paddr_aligned);
 cmd.addr_hi = upper_32_bits(rx_tid->qbuf.paddr_aligned);
 cmd.upd0 |= HAL_REO_CMD_UPD0_VLD;
 /* Observed flush cache failure, to avoid that set vld bit during delete */
 cmd.upd1 |= HAL_REO_CMD_UPD1_VLD;

 return ath12k_dp_reo_cmd_send(ab, rx_tid,
          HAL_REO_CMD_UPDATE_RX_QUEUE, &cmd,
          ath12k_dp_rx_tid_del_func);
}

static void ath12k_peer_rx_tid_qref_setup(struct ath12k_base *ab, u16 peer_id, u16 tid,
       dma_addr_t paddr)
{
 struct ath12k_reo_queue_ref *qref;
 struct ath12k_dp *dp = &ab->dp;
 bool ml_peer = false;

 if (!ab->hw_params->reoq_lut_support)
  return;

 if (peer_id & ATH12K_PEER_ML_ID_VALID) {
  peer_id &= ~ATH12K_PEER_ML_ID_VALID;
  ml_peer = true;
 }

 if (ml_peer)
  qref = (struct ath12k_reo_queue_ref *)dp->ml_reoq_lut.vaddr +
    (peer_id * (IEEE80211_NUM_TIDS + 1) + tid);
 else
  qref = (struct ath12k_reo_queue_ref *)dp->reoq_lut.vaddr +
    (peer_id * (IEEE80211_NUM_TIDS + 1) + tid);

 qref->info0 = u32_encode_bits(lower_32_bits(paddr),
          BUFFER_ADDR_INFO0_ADDR);
 qref->info1 = u32_encode_bits(upper_32_bits(paddr),
          BUFFER_ADDR_INFO1_ADDR) |
        u32_encode_bits(tid, DP_REO_QREF_NUM);
 ath12k_hal_reo_shared_qaddr_cache_clear(ab);
}

static void ath12k_peer_rx_tid_qref_reset(struct ath12k_base *ab, u16 peer_id, u16 tid)
{
 struct ath12k_reo_queue_ref *qref;
 struct ath12k_dp *dp = &ab->dp;
 bool ml_peer = false;

 if (!ab->hw_params->reoq_lut_support)
  return;

 if (peer_id & ATH12K_PEER_ML_ID_VALID) {
  peer_id &= ~ATH12K_PEER_ML_ID_VALID;
  ml_peer = true;
 }

 if (ml_peer)
  qref = (struct ath12k_reo_queue_ref *)dp->ml_reoq_lut.vaddr +
    (peer_id * (IEEE80211_NUM_TIDS + 1) + tid);
 else
  qref = (struct ath12k_reo_queue_ref *)dp->reoq_lut.vaddr +
    (peer_id * (IEEE80211_NUM_TIDS + 1) + tid);

 qref->info0 = u32_encode_bits(0, BUFFER_ADDR_INFO0_ADDR);
 qref->info1 = u32_encode_bits(0, BUFFER_ADDR_INFO1_ADDR) |
        u32_encode_bits(tid, DP_REO_QREF_NUM);
}

void ath12k_dp_rx_peer_tid_delete(struct ath12k *ar,
      struct ath12k_peer *peer, u8 tid)
{
 struct ath12k_dp_rx_tid *rx_tid = &peer->rx_tid[tid];
 int ret;

 if (!rx_tid->active)
  return;

 ret = ath12k_dp_rx_tid_delete_handler(ar->ab, rx_tid);
 if (ret) {
  ath12k_err(ar->ab, "failed to send HAL_REO_CMD_UPDATE_RX_QUEUE cmd, tid %d (%d)\n",
      tid, ret);
  dma_unmap_single(ar->ab->dev, rx_tid->qbuf.paddr_aligned,
     rx_tid->qbuf.size, DMA_BIDIRECTIONAL);
  kfree(rx_tid->qbuf.vaddr);
  rx_tid->qbuf.vaddr = NULL;
 }

 if (peer->mlo)
  ath12k_peer_rx_tid_qref_reset(ar->ab, peer->ml_id, tid);
 else
  ath12k_peer_rx_tid_qref_reset(ar->ab, peer->peer_id, tid);

 rx_tid->active = false;
}

int ath12k_dp_rx_link_desc_return(struct ath12k_base *ab,
      struct ath12k_buffer_addr *buf_addr_info,
      enum hal_wbm_rel_bm_act action)
{
 struct hal_wbm_release_ring *desc;
 struct ath12k_dp *dp = &ab->dp;
 struct hal_srng *srng;
 int ret = 0;

 srng = &ab->hal.srng_list[dp->wbm_desc_rel_ring.ring_id];

 spin_lock_bh(&srng->lock);

 ath12k_hal_srng_access_begin(ab, srng);

 desc = ath12k_hal_srng_src_get_next_entry(ab, srng);
 if (!desc) {
  ret = -ENOBUFS;
  goto exit;
 }

 ath12k_hal_rx_msdu_link_desc_set(ab, desc, buf_addr_info, action);

exit:
 ath12k_hal_srng_access_end(ab, srng);

 spin_unlock_bh(&srng->lock);

 return ret;
}

static void ath12k_dp_rx_frags_cleanup(struct ath12k_dp_rx_tid *rx_tid,
           bool rel_link_desc)
{
 struct ath12k_buffer_addr *buf_addr_info;
 struct ath12k_base *ab = rx_tid->ab;

 lockdep_assert_held(&ab->base_lock);

 if (rx_tid->dst_ring_desc) {
  if (rel_link_desc) {
   buf_addr_info = &rx_tid->dst_ring_desc->buf_addr_info;
   ath12k_dp_rx_link_desc_return(ab, buf_addr_info,
            HAL_WBM_REL_BM_ACT_PUT_IN_IDLE);
  }
  kfree(rx_tid->dst_ring_desc);
  rx_tid->dst_ring_desc = NULL;
 }

 rx_tid->cur_sn = 0;
 rx_tid->last_frag_no = 0;
 rx_tid->rx_frag_bitmap = 0;
 __skb_queue_purge(&rx_tid->rx_frags);
}

void ath12k_dp_rx_peer_tid_cleanup(struct ath12k *ar, struct ath12k_peer *peer)
{
 struct ath12k_dp_rx_tid *rx_tid;
 int i;

 lockdep_assert_held(&ar->ab->base_lock);

 for (i = 0; i <= IEEE80211_NUM_TIDS; i++) {
  rx_tid = &peer->rx_tid[i];

  ath12k_dp_rx_peer_tid_delete(ar, peer, i);
  ath12k_dp_rx_frags_cleanup(rx_tid, true);

  spin_unlock_bh(&ar->ab->base_lock);
  timer_delete_sync(&rx_tid->frag_timer);
  spin_lock_bh(&ar->ab->base_lock);
 }
}

static int ath12k_peer_rx_tid_reo_update(struct ath12k *ar,
      struct ath12k_peer *peer,
      struct ath12k_dp_rx_tid *rx_tid,
      u32 ba_win_sz, u16 ssn,
      bool update_ssn)
{
 struct ath12k_hal_reo_cmd cmd = {};
 int ret;

 cmd.addr_lo = lower_32_bits(rx_tid->qbuf.paddr_aligned);
 cmd.addr_hi = upper_32_bits(rx_tid->qbuf.paddr_aligned);
 cmd.flag = HAL_REO_CMD_FLG_NEED_STATUS;
 cmd.upd0 = HAL_REO_CMD_UPD0_BA_WINDOW_SIZE;
 cmd.ba_window_size = ba_win_sz;

 if (update_ssn) {
  cmd.upd0 |= HAL_REO_CMD_UPD0_SSN;
  cmd.upd2 = u32_encode_bits(ssn, HAL_REO_CMD_UPD2_SSN);
 }

 ret = ath12k_dp_reo_cmd_send(ar->ab, rx_tid,
         HAL_REO_CMD_UPDATE_RX_QUEUE, &cmd,
         NULL);
 if (ret) {
  ath12k_warn(ar->ab, "failed to update rx tid queue, tid %d (%d)\n",
       rx_tid->tid, ret);
  return ret;
 }

 rx_tid->ba_win_sz = ba_win_sz;

 return 0;
}

static int ath12k_dp_rx_assign_reoq(struct ath12k_base *ab,
        struct ath12k_sta *ahsta,
        struct ath12k_dp_rx_tid *rx_tid,
        u16 ssn, enum hal_pn_type pn_type)
{
 u32 ba_win_sz = rx_tid->ba_win_sz;
 struct ath12k_reoq_buf *buf;
 void *vaddr, *vaddr_aligned;
 dma_addr_t paddr_aligned;
 u8 tid = rx_tid->tid;
 u32 hw_desc_sz;
 int ret;

 buf = &ahsta->reoq_bufs[tid];
 if (!buf->vaddr) {
  /* TODO: Optimize the memory allocation for qos tid based on
 * the actual BA window size in REO tid update path.
 */
  if (tid == HAL_DESC_REO_NON_QOS_TID)
   hw_desc_sz = ath12k_hal_reo_qdesc_size(ba_win_sz, tid);
  else
   hw_desc_sz = ath12k_hal_reo_qdesc_size(DP_BA_WIN_SZ_MAX, tid);

  vaddr = kzalloc(hw_desc_sz + HAL_LINK_DESC_ALIGN - 1, GFP_ATOMIC);
  if (!vaddr)
   return -ENOMEM;

  vaddr_aligned = PTR_ALIGN(vaddr, HAL_LINK_DESC_ALIGN);

  ath12k_hal_reo_qdesc_setup(vaddr_aligned, tid, ba_win_sz,
        ssn, pn_type);

  paddr_aligned = dma_map_single(ab->dev, vaddr_aligned, hw_desc_sz,
            DMA_BIDIRECTIONAL);
  ret = dma_mapping_error(ab->dev, paddr_aligned);
  if (ret) {
   kfree(vaddr);
   return ret;
  }

  buf->vaddr = vaddr;
  buf->paddr_aligned = paddr_aligned;
  buf->size = hw_desc_sz;
 }

 rx_tid->qbuf = *buf;
 rx_tid->active = true;

 return 0;
}

int ath12k_dp_rx_peer_tid_setup(struct ath12k *ar, const u8 *peer_mac, int vdev_id,
    u8 tid, u32 ba_win_sz, u16 ssn,
    enum hal_pn_type pn_type)
{
 struct ath12k_base *ab = ar->ab;
 struct ath12k_dp *dp = &ab->dp;
 struct ath12k_peer *peer;
 struct ath12k_sta *ahsta;
 struct ath12k_dp_rx_tid *rx_tid;
 dma_addr_t paddr_aligned;
 int ret;

 spin_lock_bh(&ab->base_lock);

 peer = ath12k_peer_find(ab, vdev_id, peer_mac);
 if (!peer) {
  spin_unlock_bh(&ab->base_lock);
  ath12k_warn(ab, "failed to find the peer to set up rx tid\n");
  return -ENOENT;
 }

 if (ab->hw_params->dp_primary_link_only &&
     !peer->primary_link) {
  spin_unlock_bh(&ab->base_lock);
  return 0;
 }

 if (ab->hw_params->reoq_lut_support &&
     (!dp->reoq_lut.vaddr || !dp->ml_reoq_lut.vaddr)) {
  spin_unlock_bh(&ab->base_lock);
  ath12k_warn(ab, "reo qref table is not setup\n");
  return -EINVAL;
 }

 if (peer->peer_id > DP_MAX_PEER_ID || tid > IEEE80211_NUM_TIDS) {
  ath12k_warn(ab, "peer id of peer %d or tid %d doesn't allow reoq setup\n",
       peer->peer_id, tid);
  spin_unlock_bh(&ab->base_lock);
  return -EINVAL;
 }

 rx_tid = &peer->rx_tid[tid];
 /* Update the tid queue if it is already setup */
 if (rx_tid->active) {
  ret = ath12k_peer_rx_tid_reo_update(ar, peer, rx_tid,
          ba_win_sz, ssn, true);
  spin_unlock_bh(&ab->base_lock);
  if (ret) {
   ath12k_warn(ab, "failed to update reo for rx tid %d\n", tid);
   return ret;
  }

  if (!ab->hw_params->reoq_lut_support) {
   paddr_aligned = rx_tid->qbuf.paddr_aligned;
   ret = ath12k_wmi_peer_rx_reorder_queue_setup(ar, vdev_id,
             peer_mac,
             paddr_aligned, tid,
             1, ba_win_sz);
   if (ret) {
    ath12k_warn(ab, "failed to setup peer rx reorder queuefor tid %d: %d\n",
         tid, ret);
    return ret;
   }
  }

  return 0;
 }

 rx_tid->tid = tid;

 rx_tid->ba_win_sz = ba_win_sz;

 ahsta = ath12k_sta_to_ahsta(peer->sta);
 ret = ath12k_dp_rx_assign_reoq(ab, ahsta, rx_tid, ssn, pn_type);
 if (ret) {
  spin_unlock_bh(&ab->base_lock);
  ath12k_warn(ab, "failed to assign reoq buf for rx tid %u\n", tid);
  return ret;
 }

 paddr_aligned = rx_tid->qbuf.paddr_aligned;
 if (ab->hw_params->reoq_lut_support) {
  /* Update the REO queue LUT at the corresponding peer id
 * and tid with qaddr.
 */
  if (peer->mlo)
   ath12k_peer_rx_tid_qref_setup(ab, peer->ml_id, tid,
            paddr_aligned);
  else
   ath12k_peer_rx_tid_qref_setup(ab, peer->peer_id, tid,
            paddr_aligned);

  spin_unlock_bh(&ab->base_lock);
 } else {
  spin_unlock_bh(&ab->base_lock);
  ret = ath12k_wmi_peer_rx_reorder_queue_setup(ar, vdev_id, peer_mac,
            paddr_aligned, tid, 1,
            ba_win_sz);
 }

 return ret;
}

int ath12k_dp_rx_ampdu_start(struct ath12k *ar,
        struct ieee80211_ampdu_params *params,
        u8 link_id)
{
 struct ath12k_base *ab = ar->ab;
 struct ath12k_sta *ahsta = ath12k_sta_to_ahsta(params->sta);
 struct ath12k_link_sta *arsta;
 int vdev_id;
 int ret;

 lockdep_assert_wiphy(ath12k_ar_to_hw(ar)->wiphy);

 arsta = wiphy_dereference(ath12k_ar_to_hw(ar)->wiphy,
      ahsta->link[link_id]);
 if (!arsta)
  return -ENOLINK;

 vdev_id = arsta->arvif->vdev_id;

 ret = ath12k_dp_rx_peer_tid_setup(ar, arsta->addr, vdev_id,
       params->tid, params->buf_size,
       params->ssn, arsta->ahsta->pn_type);
 if (ret)
  ath12k_warn(ab, "failed to setup rx tid %d\n", ret);

 return ret;
}

int ath12k_dp_rx_ampdu_stop(struct ath12k *ar,
       struct ieee80211_ampdu_params *params,
       u8 link_id)
{
 struct ath12k_base *ab = ar->ab;
 struct ath12k_peer *peer;
 struct ath12k_sta *ahsta = ath12k_sta_to_ahsta(params->sta);
 struct ath12k_link_sta *arsta;
 int vdev_id;
 bool active;
 int ret;

 lockdep_assert_wiphy(ath12k_ar_to_hw(ar)->wiphy);

 arsta = wiphy_dereference(ath12k_ar_to_hw(ar)->wiphy,
      ahsta->link[link_id]);
 if (!arsta)
  return -ENOLINK;

 vdev_id = arsta->arvif->vdev_id;

 spin_lock_bh(&ab->base_lock);

 peer = ath12k_peer_find(ab, vdev_id, arsta->addr);
 if (!peer) {
  spin_unlock_bh(&ab->base_lock);
  ath12k_warn(ab, "failed to find the peer to stop rx aggregation\n");
  return -ENOENT;
 }

 active = peer->rx_tid[params->tid].active;

 if (!active) {
  spin_unlock_bh(&ab->base_lock);
  return 0;
 }

 ret = ath12k_peer_rx_tid_reo_update(ar, peer, peer->rx_tid, 1, 0, false);
 spin_unlock_bh(&ab->base_lock);
 if (ret) {
  ath12k_warn(ab, "failed to update reo for rx tid %d: %d\n",
       params->tid, ret);
  return ret;
 }

 return ret;
}

int ath12k_dp_rx_peer_pn_replay_config(struct ath12k_link_vif *arvif,
           const u8 *peer_addr,
           enum set_key_cmd key_cmd,
           struct ieee80211_key_conf *key)
{
 struct ath12k *ar = arvif->ar;
 struct ath12k_base *ab = ar->ab;
 struct ath12k_hal_reo_cmd cmd = {};
 struct ath12k_peer *peer;
 struct ath12k_dp_rx_tid *rx_tid;
 u8 tid;
 int ret = 0;

 /* NOTE: Enable PN/TSC replay check offload only for unicast frames.
 * We use mac80211 PN/TSC replay check functionality for bcast/mcast
 * for now.
 */
 if (!(key->flags & IEEE80211_KEY_FLAG_PAIRWISE))
  return 0;

 cmd.flag = HAL_REO_CMD_FLG_NEED_STATUS;
 cmd.upd0 = HAL_REO_CMD_UPD0_PN |
      HAL_REO_CMD_UPD0_PN_SIZE |
      HAL_REO_CMD_UPD0_PN_VALID |
      HAL_REO_CMD_UPD0_PN_CHECK |
      HAL_REO_CMD_UPD0_SVLD;

 switch (key->cipher) {
 case WLAN_CIPHER_SUITE_TKIP:
 case WLAN_CIPHER_SUITE_CCMP:
 case WLAN_CIPHER_SUITE_CCMP_256:
 case WLAN_CIPHER_SUITE_GCMP:
 case WLAN_CIPHER_SUITE_GCMP_256:
  if (key_cmd == SET_KEY) {
   cmd.upd1 |= HAL_REO_CMD_UPD1_PN_CHECK;
   cmd.pn_size = 48;
  }
  break;
 default:
  break;
 }

 spin_lock_bh(&ab->base_lock);

 peer = ath12k_peer_find(ab, arvif->vdev_id, peer_addr);
 if (!peer) {
  spin_unlock_bh(&ab->base_lock);
  ath12k_warn(ab, "failed to find the peer %pM to configure pn replay detection\n",
       peer_addr);
  return -ENOENT;
 }

 for (tid = 0; tid <= IEEE80211_NUM_TIDS; tid++) {
  rx_tid = &peer->rx_tid[tid];
  if (!rx_tid->active)
   continue;
  cmd.addr_lo = lower_32_bits(rx_tid->qbuf.paddr_aligned);
  cmd.addr_hi = upper_32_bits(rx_tid->qbuf.paddr_aligned);
  ret = ath12k_dp_reo_cmd_send(ab, rx_tid,
          HAL_REO_CMD_UPDATE_RX_QUEUE,
          &cmd, NULL);
  if (ret) {
   ath12k_warn(ab, "failed to configure rx tid %d queue of peer %pM for pn replay detection %d\n",
        tid, peer_addr, ret);
   break;
  }
 }

 spin_unlock_bh(&ab->base_lock);

 return ret;
}

static int ath12k_get_ppdu_user_index(struct htt_ppdu_stats *ppdu_stats,
          u16 peer_id)
{
 int i;

 for (i = 0; i < HTT_PPDU_STATS_MAX_USERS - 1; i++) {
  if (ppdu_stats->user_stats[i].is_valid_peer_id) {
   if (peer_id == ppdu_stats->user_stats[i].peer_id)
    return i;
  } else {
   return i;
  }
 }

 return -EINVAL;
}

static int ath12k_htt_tlv_ppdu_stats_parse(struct ath12k_base *ab,
        u16 tag, u16 len, const void *ptr,
        void *data)
{
 const struct htt_ppdu_stats_usr_cmpltn_ack_ba_status *ba_status;
 const struct htt_ppdu_stats_usr_cmpltn_cmn *cmplt_cmn;
 const struct htt_ppdu_stats_user_rate *user_rate;
 struct htt_ppdu_stats_info *ppdu_info;
 struct htt_ppdu_user_stats *user_stats;
 int cur_user;
 u16 peer_id;

 ppdu_info = data;

 switch (tag) {
 case HTT_PPDU_STATS_TAG_COMMON:
  if (len < sizeof(struct htt_ppdu_stats_common)) {
   ath12k_warn(ab, "Invalid len %d for the tag 0x%x\n",
        len, tag);
   return -EINVAL;
  }
  memcpy(&ppdu_info->ppdu_stats.common, ptr,
         sizeof(struct htt_ppdu_stats_common));
  break;
 case HTT_PPDU_STATS_TAG_USR_RATE:
  if (len < sizeof(struct htt_ppdu_stats_user_rate)) {
   ath12k_warn(ab, "Invalid len %d for the tag 0x%x\n",
        len, tag);
   return -EINVAL;
  }
  user_rate = ptr;
  peer_id = le16_to_cpu(user_rate->sw_peer_id);
  cur_user = ath12k_get_ppdu_user_index(&ppdu_info->ppdu_stats,
            peer_id);
  if (cur_user < 0)
   return -EINVAL;
  user_stats = &ppdu_info->ppdu_stats.user_stats[cur_user];
  user_stats->peer_id = peer_id;
  user_stats->is_valid_peer_id = true;
  memcpy(&user_stats->rate, ptr,
         sizeof(struct htt_ppdu_stats_user_rate));
  user_stats->tlv_flags |= BIT(tag);
  break;
 case HTT_PPDU_STATS_TAG_USR_COMPLTN_COMMON:
  if (len < sizeof(struct htt_ppdu_stats_usr_cmpltn_cmn)) {
   ath12k_warn(ab, "Invalid len %d for the tag 0x%x\n",
        len, tag);
   return -EINVAL;
  }

  cmplt_cmn = ptr;
  peer_id = le16_to_cpu(cmplt_cmn->sw_peer_id);
  cur_user = ath12k_get_ppdu_user_index(&ppdu_info->ppdu_stats,
            peer_id);
  if (cur_user < 0)
   return -EINVAL;
  user_stats = &ppdu_info->ppdu_stats.user_stats[cur_user];
  user_stats->peer_id = peer_id;
  user_stats->is_valid_peer_id = true;
  memcpy(&user_stats->cmpltn_cmn, ptr,
         sizeof(struct htt_ppdu_stats_usr_cmpltn_cmn));
  user_stats->tlv_flags |= BIT(tag);
  break;
 case HTT_PPDU_STATS_TAG_USR_COMPLTN_ACK_BA_STATUS:
  if (len <
      sizeof(struct htt_ppdu_stats_usr_cmpltn_ack_ba_status)) {
   ath12k_warn(ab, "Invalid len %d for the tag 0x%x\n",
        len, tag);
   return -EINVAL;
  }

  ba_status = ptr;
  peer_id = le16_to_cpu(ba_status->sw_peer_id);
  cur_user = ath12k_get_ppdu_user_index(&ppdu_info->ppdu_stats,
            peer_id);
  if (cur_user < 0)
   return -EINVAL;
  user_stats = &ppdu_info->ppdu_stats.user_stats[cur_user];
  user_stats->peer_id = peer_id;
  user_stats->is_valid_peer_id = true;
  memcpy(&user_stats->ack_ba, ptr,
         sizeof(struct htt_ppdu_stats_usr_cmpltn_ack_ba_status));
  user_stats->tlv_flags |= BIT(tag);
  break;
 }
 return 0;
}

int ath12k_dp_htt_tlv_iter(struct ath12k_base *ab, const void *ptr, size_t len,
      int (*iter)(struct ath12k_base *ar, u16 tag, u16 len,
           const void *ptr, void *data),
      void *data)
{
 const struct htt_tlv *tlv;
 const void *begin = ptr;
 u16 tlv_tag, tlv_len;
 int ret = -EINVAL;

 while (len > 0) {
  if (len < sizeof(*tlv)) {
   ath12k_err(ab, "htt tlv parse failure at byte %zd (%zu bytes left, %zu expected)\n",
       ptr - begin, len, sizeof(*tlv));
   return -EINVAL;
  }
  tlv = (struct htt_tlv *)ptr;
  tlv_tag = le32_get_bits(tlv->header, HTT_TLV_TAG);
  tlv_len = le32_get_bits(tlv->header, HTT_TLV_LEN);
  ptr += sizeof(*tlv);
  len -= sizeof(*tlv);

  if (tlv_len > len) {
   ath12k_err(ab, "htt tlv parse failure of tag %u at byte %zd (%zu bytes left, %u expected)\n",
       tlv_tag, ptr - begin, len, tlv_len);
   return -EINVAL;
  }
  ret = iter(ab, tlv_tag, tlv_len, ptr, data);
  if (ret == -ENOMEM)
   return ret;

  ptr += tlv_len;
  len -= tlv_len;
 }
 return 0;
}

static void
ath12k_update_per_peer_tx_stats(struct ath12k *ar,
    struct htt_ppdu_stats *ppdu_stats, u8 user)
{
 struct ath12k_base *ab = ar->ab;
 struct ath12k_peer *peer;
 struct ath12k_link_sta *arsta;
 struct htt_ppdu_stats_user_rate *user_rate;
 struct ath12k_per_peer_tx_stats *peer_stats = &ar->peer_tx_stats;
 struct htt_ppdu_user_stats *usr_stats = &ppdu_stats->user_stats[user];
 struct htt_ppdu_stats_common *common = &ppdu_stats->common;
 int ret;
 u8 flags, mcs, nss, bw, sgi, dcm, ppdu_type, rate_idx = 0;
 u32 v, succ_bytes = 0;
 u16 tones, rate = 0, succ_pkts = 0;
 u32 tx_duration = 0;
 u8 tid = HTT_PPDU_STATS_NON_QOS_TID;
 u16 tx_retry_failed = 0, tx_retry_count = 0;
 bool is_ampdu = false, is_ofdma;

 if (!(usr_stats->tlv_flags & BIT(HTT_PPDU_STATS_TAG_USR_RATE)))
  return;

 if (usr_stats->tlv_flags & BIT(HTT_PPDU_STATS_TAG_USR_COMPLTN_COMMON)) {
  is_ampdu =
   HTT_USR_CMPLTN_IS_AMPDU(usr_stats->cmpltn_cmn.flags);
  tx_retry_failed =
   __le16_to_cpu(usr_stats->cmpltn_cmn.mpdu_tried) -
   __le16_to_cpu(usr_stats->cmpltn_cmn.mpdu_success);
  tx_retry_count =
   HTT_USR_CMPLTN_LONG_RETRY(usr_stats->cmpltn_cmn.flags) +
   HTT_USR_CMPLTN_SHORT_RETRY(usr_stats->cmpltn_cmn.flags);
 }

 if (usr_stats->tlv_flags &
     BIT(HTT_PPDU_STATS_TAG_USR_COMPLTN_ACK_BA_STATUS)) {
  succ_bytes = le32_to_cpu(usr_stats->ack_ba.success_bytes);
  succ_pkts = le32_get_bits(usr_stats->ack_ba.info,
       HTT_PPDU_STATS_ACK_BA_INFO_NUM_MSDU_M);
  tid = le32_get_bits(usr_stats->ack_ba.info,
        HTT_PPDU_STATS_ACK_BA_INFO_TID_NUM);
 }

 if (common->fes_duration_us)
  tx_duration = le32_to_cpu(common->fes_duration_us);

 user_rate = &usr_stats->rate;
 flags = HTT_USR_RATE_PREAMBLE(user_rate->rate_flags);
 bw = HTT_USR_RATE_BW(user_rate->rate_flags) - 2;
 nss = HTT_USR_RATE_NSS(user_rate->rate_flags) + 1;
 mcs = HTT_USR_RATE_MCS(user_rate->rate_flags);
 sgi = HTT_USR_RATE_GI(user_rate->rate_flags);
 dcm = HTT_USR_RATE_DCM(user_rate->rate_flags);

 ppdu_type = HTT_USR_RATE_PPDU_TYPE(user_rate->info1);
 is_ofdma = (ppdu_type == HTT_PPDU_STATS_PPDU_TYPE_MU_OFDMA) ||
     (ppdu_type == HTT_PPDU_STATS_PPDU_TYPE_MU_MIMO_OFDMA);

 /* Note: If host configured fixed rates and in some other special
 * cases, the broadcast/management frames are sent in different rates.
 * Firmware rate's control to be skipped for this?
 */

 if (flags == WMI_RATE_PREAMBLE_HE && mcs > ATH12K_HE_MCS_MAX) {
  ath12k_warn(ab, "Invalid HE mcs %d peer stats",  mcs);
  return;
 }

 if (flags == WMI_RATE_PREAMBLE_VHT && mcs > ATH12K_VHT_MCS_MAX) {
  ath12k_warn(ab, "Invalid VHT mcs %d peer stats",  mcs);
  return;
 }

 if (flags == WMI_RATE_PREAMBLE_HT && (mcs > ATH12K_HT_MCS_MAX || nss < 1)) {
  ath12k_warn(ab, "Invalid HT mcs %d nss %d peer stats",
       mcs, nss);
  return;
 }

 if (flags == WMI_RATE_PREAMBLE_CCK || flags == WMI_RATE_PREAMBLE_OFDM) {
  ret = ath12k_mac_hw_ratecode_to_legacy_rate(mcs,
           flags,
           &rate_idx,
           &rate);
  if (ret < 0)
   return;
 }

 rcu_read_lock();
 spin_lock_bh(&ab->base_lock);
 peer = ath12k_peer_find_by_id(ab, usr_stats->peer_id);

 if (!peer || !peer->sta) {
  spin_unlock_bh(&ab->base_lock);
  rcu_read_unlock();
  return;
 }

 arsta = ath12k_peer_get_link_sta(ab, peer);
 if (!arsta) {
  spin_unlock_bh(&ab->base_lock);
  rcu_read_unlock();
  return;
 }

 memset(&arsta->txrate, 0, sizeof(arsta->txrate));

 arsta->txrate.bw = ath12k_mac_bw_to_mac80211_bw(bw);

 switch (flags) {
 case WMI_RATE_PREAMBLE_OFDM:
  arsta->txrate.legacy = rate;
  break;
 case WMI_RATE_PREAMBLE_CCK:
  arsta->txrate.legacy = rate;
  break;
 case WMI_RATE_PREAMBLE_HT:
  arsta->txrate.mcs = mcs + 8 * (nss - 1);
  arsta->txrate.flags = RATE_INFO_FLAGS_MCS;
  if (sgi)
   arsta->txrate.flags |= RATE_INFO_FLAGS_SHORT_GI;
  break;
 case WMI_RATE_PREAMBLE_VHT:
  arsta->txrate.mcs = mcs;
  arsta->txrate.flags = RATE_INFO_FLAGS_VHT_MCS;
  if (sgi)
   arsta->txrate.flags |= RATE_INFO_FLAGS_SHORT_GI;
  break;
 case WMI_RATE_PREAMBLE_HE:
  arsta->txrate.mcs = mcs;
  arsta->txrate.flags = RATE_INFO_FLAGS_HE_MCS;
  arsta->txrate.he_dcm = dcm;
  arsta->txrate.he_gi = ath12k_he_gi_to_nl80211_he_gi(sgi);
  tones = le16_to_cpu(user_rate->ru_end) -
   le16_to_cpu(user_rate->ru_start) + 1;
  v = ath12k_he_ru_tones_to_nl80211_he_ru_alloc(tones);
  arsta->txrate.he_ru_alloc = v;
  if (is_ofdma)
   arsta->txrate.bw = RATE_INFO_BW_HE_RU;
  break;
 case WMI_RATE_PREAMBLE_EHT:
  arsta->txrate.mcs = mcs;
  arsta->txrate.flags = RATE_INFO_FLAGS_EHT_MCS;
  arsta->txrate.he_dcm = dcm;
  arsta->txrate.eht_gi = ath12k_mac_eht_gi_to_nl80211_eht_gi(sgi);
  tones = le16_to_cpu(user_rate->ru_end) -
   le16_to_cpu(user_rate->ru_start) + 1;
  v = ath12k_mac_eht_ru_tones_to_nl80211_eht_ru_alloc(tones);
  arsta->txrate.eht_ru_alloc = v;
  if (is_ofdma)
   arsta->txrate.bw = RATE_INFO_BW_EHT_RU;
  break;
 }

 arsta->tx_retry_failed += tx_retry_failed;
 arsta->tx_retry_count += tx_retry_count;
 arsta->txrate.nss = nss;
 arsta->tx_duration += tx_duration;
 memcpy(&arsta->last_txrate, &arsta->txrate, sizeof(struct rate_info));

 /* PPDU stats reported for mgmt packet doesn't have valid tx bytes.
 * So skip peer stats update for mgmt packets.
 */
 if (tid < HTT_PPDU_STATS_NON_QOS_TID) {
  memset(peer_stats, 0, sizeof(*peer_stats));
  peer_stats->succ_pkts = succ_pkts;
  peer_stats->succ_bytes = succ_bytes;
  peer_stats->is_ampdu = is_ampdu;
  peer_stats->duration = tx_duration;
  peer_stats->ba_fails =
   HTT_USR_CMPLTN_LONG_RETRY(usr_stats->cmpltn_cmn.flags) +
   HTT_USR_CMPLTN_SHORT_RETRY(usr_stats->cmpltn_cmn.flags);
 }

 spin_unlock_bh(&ab->base_lock);
 rcu_read_unlock();
}

static void ath12k_htt_update_ppdu_stats(struct ath12k *ar,
      struct htt_ppdu_stats *ppdu_stats)
{
 u8 user;

 for (user = 0; user < HTT_PPDU_STATS_MAX_USERS - 1; user++)
  ath12k_update_per_peer_tx_stats(ar, ppdu_stats, user);
}

static
struct htt_ppdu_stats_info *ath12k_dp_htt_get_ppdu_desc(struct ath12k *ar,
       u32 ppdu_id)
{
 struct htt_ppdu_stats_info *ppdu_info;

 lockdep_assert_held(&ar->data_lock);
 if (!list_empty(&ar->ppdu_stats_info)) {
  list_for_each_entry(ppdu_info, &ar->ppdu_stats_info, list) {
   if (ppdu_info->ppdu_id == ppdu_id)
    return ppdu_info;
  }

  if (ar->ppdu_stat_list_depth > HTT_PPDU_DESC_MAX_DEPTH) {
   ppdu_info = list_first_entry(&ar->ppdu_stats_info,
           typeof(*ppdu_info), list);
   list_del(&ppdu_info->list);
   ar->ppdu_stat_list_depth--;
   ath12k_htt_update_ppdu_stats(ar, &ppdu_info->ppdu_stats);
   kfree(ppdu_info);
  }
 }

 ppdu_info = kzalloc(sizeof(*ppdu_info), GFP_ATOMIC);
 if (!ppdu_info)
  return NULL;

 list_add_tail(&ppdu_info->list, &ar->ppdu_stats_info);
 ar->ppdu_stat_list_depth++;

 return ppdu_info;
}

static void ath12k_copy_to_delay_stats(struct ath12k_peer *peer,
           struct htt_ppdu_user_stats *usr_stats)
{
 peer->ppdu_stats_delayba.sw_peer_id = le16_to_cpu(usr_stats->rate.sw_peer_id);
 peer->ppdu_stats_delayba.info0 = le32_to_cpu(usr_stats->rate.info0);
 peer->ppdu_stats_delayba.ru_end = le16_to_cpu(usr_stats->rate.ru_end);
 peer->ppdu_stats_delayba.ru_start = le16_to_cpu(usr_stats->rate.ru_start);
 peer->ppdu_stats_delayba.info1 = le32_to_cpu(usr_stats->rate.info1);
 peer->ppdu_stats_delayba.rate_flags = le32_to_cpu(usr_stats->rate.rate_flags);
 peer->ppdu_stats_delayba.resp_rate_flags =
  le32_to_cpu(usr_stats->rate.resp_rate_flags);

 peer->delayba_flag = true;
}

static void ath12k_copy_to_bar(struct ath12k_peer *peer,
          struct htt_ppdu_user_stats *usr_stats)
{
 usr_stats->rate.sw_peer_id = cpu_to_le16(peer->ppdu_stats_delayba.sw_peer_id);
 usr_stats->rate.info0 = cpu_to_le32(peer->ppdu_stats_delayba.info0);
 usr_stats->rate.ru_end = cpu_to_le16(peer->ppdu_stats_delayba.ru_end);
 usr_stats->rate.ru_start = cpu_to_le16(peer->ppdu_stats_delayba.ru_start);
 usr_stats->rate.info1 = cpu_to_le32(peer->ppdu_stats_delayba.info1);
 usr_stats->rate.rate_flags = cpu_to_le32(peer->ppdu_stats_delayba.rate_flags);
 usr_stats->rate.resp_rate_flags =
  cpu_to_le32(peer->ppdu_stats_delayba.resp_rate_flags);

 peer->delayba_flag = false;
}

static int ath12k_htt_pull_ppdu_stats(struct ath12k_base *ab,
          struct sk_buff *skb)
{
 struct ath12k_htt_ppdu_stats_msg *msg;
 struct htt_ppdu_stats_info *ppdu_info;
 struct ath12k_peer *peer = NULL;
 struct htt_ppdu_user_stats *usr_stats = NULL;
 u32 peer_id = 0;
 struct ath12k *ar;
 int ret, i;
 u8 pdev_id;
 u32 ppdu_id, len;

 msg = (struct ath12k_htt_ppdu_stats_msg *)skb->data;
 len = le32_get_bits(msg->info, HTT_T2H_PPDU_STATS_INFO_PAYLOAD_SIZE);
 if (len > (skb->len - struct_size(msg, data, 0))) {
  ath12k_warn(ab,
       "HTT PPDU STATS event has unexpected payload size %u, should be smaller than %u\n",
       len, skb->len);
  return -EINVAL;
 }

 pdev_id = le32_get_bits(msg->info, HTT_T2H_PPDU_STATS_INFO_PDEV_ID);
 ppdu_id = le32_to_cpu(msg->ppdu_id);

 rcu_read_lock();
 ar = ath12k_mac_get_ar_by_pdev_id(ab, pdev_id);
 if (!ar) {
  ret = -EINVAL;
  goto exit;
 }

 spin_lock_bh(&ar->data_lock);
 ppdu_info = ath12k_dp_htt_get_ppdu_desc(ar, ppdu_id);
 if (!ppdu_info) {
  spin_unlock_bh(&ar->data_lock);
  ret = -EINVAL;
  goto exit;
 }

 ppdu_info->ppdu_id = ppdu_id;
 ret = ath12k_dp_htt_tlv_iter(ab, msg->data, len,
         ath12k_htt_tlv_ppdu_stats_parse,
         (void *)ppdu_info);
 if (ret) {
  spin_unlock_bh(&ar->data_lock);
  ath12k_warn(ab, "Failed to parse tlv %d\n", ret);
  goto exit;
 }

 if (ppdu_info->ppdu_stats.common.num_users >= HTT_PPDU_STATS_MAX_USERS) {
  spin_unlock_bh(&ar->data_lock);
  ath12k_warn(ab,
       "HTT PPDU STATS event has unexpected num_users %u, should be smaller than %u\n",
       ppdu_info->ppdu_stats.common.num_users,
       HTT_PPDU_STATS_MAX_USERS);
  ret = -EINVAL;
  goto exit;
 }

 /* back up data rate tlv for all peers */
 if (ppdu_info->frame_type == HTT_STATS_PPDU_FTYPE_DATA &&
     (ppdu_info->tlv_bitmap & (1 << HTT_PPDU_STATS_TAG_USR_COMMON)) &&
     ppdu_info->delay_ba) {
  for (i = 0; i < ppdu_info->ppdu_stats.common.num_users; i++) {
   peer_id = ppdu_info->ppdu_stats.user_stats[i].peer_id;
   spin_lock_bh(&ab->base_lock);
   peer = ath12k_peer_find_by_id(ab, peer_id);
   if (!peer) {
    spin_unlock_bh(&ab->base_lock);
    continue;
   }

   usr_stats = &ppdu_info->ppdu_stats.user_stats[i];
   if (usr_stats->delay_ba)
    ath12k_copy_to_delay_stats(peer, usr_stats);
   spin_unlock_bh(&ab->base_lock);
  }
 }

 /* restore all peers' data rate tlv to mu-bar tlv */
 if (ppdu_info->frame_type == HTT_STATS_PPDU_FTYPE_BAR &&
     (ppdu_info->tlv_bitmap & (1 << HTT_PPDU_STATS_TAG_USR_COMMON))) {
  for (i = 0; i < ppdu_info->bar_num_users; i++) {
   peer_id = ppdu_info->ppdu_stats.user_stats[i].peer_id;
   spin_lock_bh(&ab->base_lock);
   peer = ath12k_peer_find_by_id(ab, peer_id);
   if (!peer) {
    spin_unlock_bh(&ab->base_lock);
    continue;
   }

   usr_stats = &ppdu_info->ppdu_stats.user_stats[i];
   if (peer->delayba_flag)
    ath12k_copy_to_bar(peer, usr_stats);
   spin_unlock_bh(&ab->base_lock);
  }
 }

 spin_unlock_bh(&ar->data_lock);

exit:
 rcu_read_unlock();

 return ret;
}

static void ath12k_htt_mlo_offset_event_handler(struct ath12k_base *ab,
      struct sk_buff *skb)
{
 struct ath12k_htt_mlo_offset_msg *msg;
 struct ath12k_pdev *pdev;
 struct ath12k *ar;
 u8 pdev_id;

 msg = (struct ath12k_htt_mlo_offset_msg *)skb->data;
 pdev_id = u32_get_bits(__le32_to_cpu(msg->info),
          HTT_T2H_MLO_OFFSET_INFO_PDEV_ID);

 rcu_read_lock();
 ar = ath12k_mac_get_ar_by_pdev_id(ab, pdev_id);
 if (!ar) {
  /* It is possible that the ar is not yet active (started).
 * The above function will only look for the active pdev
 * and hence %NULL return is possible. Just silently
 * discard this message
 */
  goto exit;
 }

 spin_lock_bh(&ar->data_lock);
 pdev = ar->pdev;

 pdev->timestamp.info = __le32_to_cpu(msg->info);
 pdev->timestamp.sync_timestamp_lo_us = __le32_to_cpu(msg->sync_timestamp_lo_us);
 pdev->timestamp.sync_timestamp_hi_us = __le32_to_cpu(msg->sync_timestamp_hi_us);
 pdev->timestamp.mlo_offset_lo = __le32_to_cpu(msg->mlo_offset_lo);
 pdev->timestamp.mlo_offset_hi = __le32_to_cpu(msg->mlo_offset_hi);
 pdev->timestamp.mlo_offset_clks = __le32_to_cpu(msg->mlo_offset_clks);
 pdev->timestamp.mlo_comp_clks = __le32_to_cpu(msg->mlo_comp_clks);
 pdev->timestamp.mlo_comp_timer = __le32_to_cpu(msg->mlo_comp_timer);

 spin_unlock_bh(&ar->data_lock);
exit:
 rcu_read_unlock();
}

void ath12k_dp_htt_htc_t2h_msg_handler(struct ath12k_base *ab,
           struct sk_buff *skb)
{
 struct ath12k_dp *dp = &ab->dp;
 struct htt_resp_msg *resp = (struct htt_resp_msg *)skb->data;
 enum htt_t2h_msg_type type;
 u16 peer_id;
 u8 vdev_id;
 u8 mac_addr[ETH_ALEN];
 u16 peer_mac_h16;
 u16 ast_hash = 0;
 u16 hw_peer_id;

 type = le32_get_bits(resp->version_msg.version, HTT_T2H_MSG_TYPE);

 ath12k_dbg(ab, ATH12K_DBG_DP_HTT, "dp_htt rx msg type :0x%0x\n", type);

 switch (type) {
 case HTT_T2H_MSG_TYPE_VERSION_CONF:
  dp->htt_tgt_ver_major = le32_get_bits(resp->version_msg.version,
            HTT_T2H_VERSION_CONF_MAJOR);
  dp->htt_tgt_ver_minor = le32_get_bits(resp->version_msg.version,
            HTT_T2H_VERSION_CONF_MINOR);
  complete(&dp->htt_tgt_version_received);
  break;
 /* TODO: remove unused peer map versions after testing */
 case HTT_T2H_MSG_TYPE_PEER_MAP:
  vdev_id = le32_get_bits(resp->peer_map_ev.info,
     HTT_T2H_PEER_MAP_INFO_VDEV_ID);
  peer_id = le32_get_bits(resp->peer_map_ev.info,
     HTT_T2H_PEER_MAP_INFO_PEER_ID);
  peer_mac_h16 = le32_get_bits(resp->peer_map_ev.info1,
          HTT_T2H_PEER_MAP_INFO1_MAC_ADDR_H16);
  ath12k_dp_get_mac_addr(le32_to_cpu(resp->peer_map_ev.mac_addr_l32),
           peer_mac_h16, mac_addr);
  ath12k_peer_map_event(ab, vdev_id, peer_id, mac_addr, 0, 0);
  break;
 case HTT_T2H_MSG_TYPE_PEER_MAP2:
  vdev_id = le32_get_bits(resp->peer_map_ev.info,
     HTT_T2H_PEER_MAP_INFO_VDEV_ID);
  peer_id = le32_get_bits(resp->peer_map_ev.info,
     HTT_T2H_PEER_MAP_INFO_PEER_ID);
  peer_mac_h16 = le32_get_bits(resp->peer_map_ev.info1,
          HTT_T2H_PEER_MAP_INFO1_MAC_ADDR_H16);
  ath12k_dp_get_mac_addr(le32_to_cpu(resp->peer_map_ev.mac_addr_l32),
           peer_mac_h16, mac_addr);
  ast_hash = le32_get_bits(resp->peer_map_ev.info2,
      HTT_T2H_PEER_MAP_INFO2_AST_HASH_VAL);
  hw_peer_id = le32_get_bits(resp->peer_map_ev.info1,
        HTT_T2H_PEER_MAP_INFO1_HW_PEER_ID);
  ath12k_peer_map_event(ab, vdev_id, peer_id, mac_addr, ast_hash,
          hw_peer_id);
  break;
 case HTT_T2H_MSG_TYPE_PEER_MAP3:
  vdev_id = le32_get_bits(resp->peer_map_ev.info,
     HTT_T2H_PEER_MAP_INFO_VDEV_ID);
  peer_id = le32_get_bits(resp->peer_map_ev.info,
     HTT_T2H_PEER_MAP_INFO_PEER_ID);
  peer_mac_h16 = le32_get_bits(resp->peer_map_ev.info1,
          HTT_T2H_PEER_MAP_INFO1_MAC_ADDR_H16);
  ath12k_dp_get_mac_addr(le32_to_cpu(resp->peer_map_ev.mac_addr_l32),
           peer_mac_h16, mac_addr);
  ast_hash = le32_get_bits(resp->peer_map_ev.info2,
      HTT_T2H_PEER_MAP3_INFO2_AST_HASH_VAL);
  hw_peer_id = le32_get_bits(resp->peer_map_ev.info2,
        HTT_T2H_PEER_MAP3_INFO2_HW_PEER_ID);
  ath12k_peer_map_event(ab, vdev_id, peer_id, mac_addr, ast_hash,
          hw_peer_id);
  break;
 case HTT_T2H_MSG_TYPE_PEER_UNMAP:
 case HTT_T2H_MSG_TYPE_PEER_UNMAP2:
  peer_id = le32_get_bits(resp->peer_unmap_ev.info,
     HTT_T2H_PEER_UNMAP_INFO_PEER_ID);
  ath12k_peer_unmap_event(ab, peer_id);
  break;
 case HTT_T2H_MSG_TYPE_PPDU_STATS_IND:
  ath12k_htt_pull_ppdu_stats(ab, skb);
  break;
 case HTT_T2H_MSG_TYPE_EXT_STATS_CONF:
  ath12k_debugfs_htt_ext_stats_handler(ab, skb);
  break;
 case HTT_T2H_MSG_TYPE_MLO_TIMESTAMP_OFFSET_IND:
  ath12k_htt_mlo_offset_event_handler(ab, skb);
  break;
 default:
  ath12k_dbg(ab, ATH12K_DBG_DP_HTT, "dp_htt event %d not handled\n",
      type);
  break;
 }

 dev_kfree_skb_any(skb);
}

static int ath12k_dp_rx_msdu_coalesce(struct ath12k *ar,
          struct sk_buff_head *msdu_list,
          struct sk_buff *first, struct sk_buff *last,
          u8 l3pad_bytes, int msdu_len)
{
 struct ath12k_base *ab = ar->ab;
 struct sk_buff *skb;
 struct ath12k_skb_rxcb *rxcb = ATH12K_SKB_RXCB(first);
 int buf_first_hdr_len, buf_first_len;
 struct hal_rx_desc *ldesc;
 int space_extra, rem_len, buf_len;
 u32 hal_rx_desc_sz = ar->ab->hal.hal_desc_sz;
 bool is_continuation;

 /* As the msdu is spread across multiple rx buffers,
 * find the offset to the start of msdu for computing
 * the length of the msdu in the first buffer.
 */
 buf_first_hdr_len = hal_rx_desc_sz + l3pad_bytes;
 buf_first_len = DP_RX_BUFFER_SIZE - buf_first_hdr_len;

 if (WARN_ON_ONCE(msdu_len <= buf_first_len)) {
  skb_put(first, buf_first_hdr_len + msdu_len);
  skb_pull(first, buf_first_hdr_len);
  return 0;
 }

 ldesc = (struct hal_rx_desc *)last->data;
 rxcb->is_first_msdu = ath12k_dp_rx_h_first_msdu(ab, ldesc);
 rxcb->is_last_msdu = ath12k_dp_rx_h_last_msdu(ab, ldesc);

 /* MSDU spans over multiple buffers because the length of the MSDU
 * exceeds DP_RX_BUFFER_SIZE - HAL_RX_DESC_SIZE. So assume the data
 * in the first buf is of length DP_RX_BUFFER_SIZE - HAL_RX_DESC_SIZE.
 */
 skb_put(first, DP_RX_BUFFER_SIZE);
 skb_pull(first, buf_first_hdr_len);

 /* When an MSDU spread over multiple buffers MSDU_END
 * tlvs are valid only in the last buffer. Copy those tlvs.
 */
 ath12k_dp_rx_desc_end_tlv_copy(ab, rxcb->rx_desc, ldesc);

 space_extra = msdu_len - (buf_first_len + skb_tailroom(first));
 if (space_extra > 0 &&
     (pskb_expand_head(first, 0, space_extra, GFP_ATOMIC) < 0)) {
  /* Free up all buffers of the MSDU */
  while ((skb = __skb_dequeue(msdu_list)) != NULL) {
   rxcb = ATH12K_SKB_RXCB(skb);
   if (!rxcb->is_continuation) {
    dev_kfree_skb_any(skb);
    break;
   }
   dev_kfree_skb_any(skb);
  }
  return -ENOMEM;
 }

 rem_len = msdu_len - buf_first_len;
 while ((skb = __skb_dequeue(msdu_list)) != NULL && rem_len > 0) {
  rxcb = ATH12K_SKB_RXCB(skb);
  is_continuation = rxcb->is_continuation;
  if (is_continuation)
   buf_len = DP_RX_BUFFER_SIZE - hal_rx_desc_sz;
  else
   buf_len = rem_len;

  if (buf_len > (DP_RX_BUFFER_SIZE - hal_rx_desc_sz)) {
   WARN_ON_ONCE(1);
   dev_kfree_skb_any(skb);
   return -EINVAL;
  }

  skb_put(skb, buf_len + hal_rx_desc_sz);
  skb_pull(skb, hal_rx_desc_sz);
  skb_copy_from_linear_data(skb, skb_put(first, buf_len),
       buf_len);
  dev_kfree_skb_any(skb);

  rem_len -= buf_len;
  if (!is_continuation)
   break;
 }

 return 0;
}

static struct sk_buff *ath12k_dp_rx_get_msdu_last_buf(struct sk_buff_head *msdu_list,
            struct sk_buff *first)
{
 struct sk_buff *skb;
 struct ath12k_skb_rxcb *rxcb = ATH12K_SKB_RXCB(first);

 if (!rxcb->is_continuation)
  return first;

 skb_queue_walk(msdu_list, skb) {
  rxcb = ATH12K_SKB_RXCB(skb);
  if (!rxcb->is_continuation)
   return skb;
 }

 return NULL;
}

static void ath12k_dp_rx_h_csum_offload(struct sk_buff *msdu,
     struct ath12k_dp_rx_info *rx_info)
{
 msdu->ip_summed = (rx_info->ip_csum_fail || rx_info->l4_csum_fail) ?
      CHECKSUM_NONE : CHECKSUM_UNNECESSARY;
}

int ath12k_dp_rx_crypto_mic_len(struct ath12k *ar, enum hal_encrypt_type enctype)
{
 switch (enctype) {
 case HAL_ENCRYPT_TYPE_OPEN:
 case HAL_ENCRYPT_TYPE_TKIP_NO_MIC:
 case HAL_ENCRYPT_TYPE_TKIP_MIC:
  return 0;
 case HAL_ENCRYPT_TYPE_CCMP_128:
  return IEEE80211_CCMP_MIC_LEN;
 case HAL_ENCRYPT_TYPE_CCMP_256:
  return IEEE80211_CCMP_256_MIC_LEN;
 case HAL_ENCRYPT_TYPE_GCMP_128:
 case HAL_ENCRYPT_TYPE_AES_GCMP_256:
  return IEEE80211_GCMP_MIC_LEN;
 case HAL_ENCRYPT_TYPE_WEP_40:
 case HAL_ENCRYPT_TYPE_WEP_104:
 case HAL_ENCRYPT_TYPE_WEP_128:
 case HAL_ENCRYPT_TYPE_WAPI_GCM_SM4:
 case HAL_ENCRYPT_TYPE_WAPI:
  break;
 }

 ath12k_warn(ar->ab, "unsupported encryption type %d for mic len\n", enctype);
 return 0;
}

static int ath12k_dp_rx_crypto_param_len(struct ath12k *ar,
      enum hal_encrypt_type enctype)
{
 switch (enctype) {
 case HAL_ENCRYPT_TYPE_OPEN:
  return 0;
 case HAL_ENCRYPT_TYPE_TKIP_NO_MIC:
 case HAL_ENCRYPT_TYPE_TKIP_MIC:
  return IEEE80211_TKIP_IV_LEN;
 case HAL_ENCRYPT_TYPE_CCMP_128:
  return IEEE80211_CCMP_HDR_LEN;
 case HAL_ENCRYPT_TYPE_CCMP_256:
  return IEEE80211_CCMP_256_HDR_LEN;
 case HAL_ENCRYPT_TYPE_GCMP_128:
 case HAL_ENCRYPT_TYPE_AES_GCMP_256:
  return IEEE80211_GCMP_HDR_LEN;
 case HAL_ENCRYPT_TYPE_WEP_40:
 case HAL_ENCRYPT_TYPE_WEP_104:
 case HAL_ENCRYPT_TYPE_WEP_128:
 case HAL_ENCRYPT_TYPE_WAPI_GCM_SM4:
 case HAL_ENCRYPT_TYPE_WAPI:
  break;
 }

 ath12k_warn(ar->ab, "unsupported encryption type %d\n", enctype);
 return 0;
}

static int ath12k_dp_rx_crypto_icv_len(struct ath12k *ar,
           enum hal_encrypt_type enctype)
{
 switch (enctype) {
 case HAL_ENCRYPT_TYPE_OPEN:
 case HAL_ENCRYPT_TYPE_CCMP_128:
 case HAL_ENCRYPT_TYPE_CCMP_256:
 case HAL_ENCRYPT_TYPE_GCMP_128:
 case HAL_ENCRYPT_TYPE_AES_GCMP_256:
  return 0;
 case HAL_ENCRYPT_TYPE_TKIP_NO_MIC:
 case HAL_ENCRYPT_TYPE_TKIP_MIC:
  return IEEE80211_TKIP_ICV_LEN;
 case HAL_ENCRYPT_TYPE_WEP_40:
 case HAL_ENCRYPT_TYPE_WEP_104:
 case HAL_ENCRYPT_TYPE_WEP_128:
 case HAL_ENCRYPT_TYPE_WAPI_GCM_SM4:
 case HAL_ENCRYPT_TYPE_WAPI:
  break;
 }

 ath12k_warn(ar->ab, "unsupported encryption type %d\n", enctype);
 return 0;
}

static void ath12k_dp_rx_h_undecap_nwifi(struct ath12k *ar,
      struct sk_buff *msdu,
      enum hal_encrypt_type enctype,
      struct ieee80211_rx_status *status)
{
 struct ath12k_base *ab = ar->ab;
 struct ath12k_skb_rxcb *rxcb = ATH12K_SKB_RXCB(msdu);
 u8 decap_hdr[DP_MAX_NWIFI_HDR_LEN];
 struct ieee80211_hdr *hdr;
 size_t hdr_len;
 u8 *crypto_hdr;
 u16 qos_ctl;

 /* pull decapped header */
 hdr = (struct ieee80211_hdr *)msdu->data;
 hdr_len = ieee80211_hdrlen(hdr->frame_control);
 skb_pull(msdu, hdr_len);

 /*  Rebuild qos header */
 hdr->frame_control |= __cpu_to_le16(IEEE80211_STYPE_QOS_DATA);

 /* Reset the order bit as the HT_Control header is stripped */
 hdr->frame_control &= ~(__cpu_to_le16(IEEE80211_FCTL_ORDER));

 qos_ctl = rxcb->tid;

 if (ath12k_dp_rx_h_mesh_ctl_present(ab, rxcb->rx_desc))
  qos_ctl |= IEEE80211_QOS_CTL_MESH_CONTROL_PRESENT;

 /* TODO: Add other QoS ctl fields when required */

 /* copy decap header before overwriting for reuse below */
 memcpy(decap_hdr, hdr, hdr_len);

 /* Rebuild crypto header for mac80211 use */
 if (!(status->flag & RX_FLAG_IV_STRIPPED)) {
  crypto_hdr = skb_push(msdu, ath12k_dp_rx_crypto_param_len(ar, enctype));
  ath12k_dp_rx_desc_get_crypto_header(ar->ab,
          rxcb->rx_desc, crypto_hdr,
          enctype);
 }

 memcpy(skb_push(msdu,
   IEEE80211_QOS_CTL_LEN), &qos_ctl,
   IEEE80211_QOS_CTL_LEN);
 memcpy(skb_push(msdu, hdr_len), decap_hdr, hdr_len);
}

static void ath12k_dp_rx_h_undecap_raw(struct ath12k *ar, struct sk_buff *msdu,
           enum hal_encrypt_type enctype,
           struct ieee80211_rx_status *status,
           bool decrypted)
{
 struct ath12k_skb_rxcb *rxcb = ATH12K_SKB_RXCB(msdu);
 struct ieee80211_hdr *hdr;
 size_t hdr_len;
 size_t crypto_len;

 if (!rxcb->is_first_msdu ||
     !(rxcb->is_first_msdu && rxcb->is_last_msdu)) {
  WARN_ON_ONCE(1);
  return;
 }

 skb_trim(msdu, msdu->len - FCS_LEN);

 if (!decrypted)
  return;

 hdr = (void *)msdu->data;

 /* Tail */
 if (status->flag & RX_FLAG_IV_STRIPPED) {
  skb_trim(msdu, msdu->len -
    ath12k_dp_rx_crypto_mic_len(ar, enctype));

  skb_trim(msdu, msdu->len -
    ath12k_dp_rx_crypto_icv_len(ar, enctype));
 } else {
  /* MIC */
  if (status->flag & RX_FLAG_MIC_STRIPPED)
   skb_trim(msdu, msdu->len -
     ath12k_dp_rx_crypto_mic_len(ar, enctype));

  /* ICV */
  if (status->flag & RX_FLAG_ICV_STRIPPED)
   skb_trim(msdu, msdu->len -
     ath12k_dp_rx_crypto_icv_len(ar, enctype));
 }

 /* MMIC */
 if ((status->flag & RX_FLAG_MMIC_STRIPPED) &&
     !ieee80211_has_morefrags(hdr->frame_control) &&
     enctype == HAL_ENCRYPT_TYPE_TKIP_MIC)
  skb_trim(msdu, msdu->len - IEEE80211_CCMP_MIC_LEN);

 /* Head */
 if (status->flag & RX_FLAG_IV_STRIPPED) {
  hdr_len = ieee80211_hdrlen(hdr->frame_control);
  crypto_len = ath12k_dp_rx_crypto_param_len(ar, enctype);

  memmove(msdu->data + crypto_len, msdu->data, hdr_len);
  skb_pull(msdu, crypto_len);
 }
}

static void ath12k_get_dot11_hdr_from_rx_desc(struct ath12k *ar,
           struct sk_buff *msdu,
           struct ath12k_skb_rxcb *rxcb,
           struct ieee80211_rx_status *status,
           enum hal_encrypt_type enctype)
{
 struct hal_rx_desc *rx_desc = rxcb->rx_desc;
 struct ath12k_base *ab = ar->ab;
 size_t hdr_len, crypto_len;
 struct ieee80211_hdr hdr;
 __le16 qos_ctl;
 u8 *crypto_hdr, mesh_ctrl;

 ath12k_dp_rx_desc_get_dot11_hdr(ab, rx_desc, &hdr);
 hdr_len = ieee80211_hdrlen(hdr.frame_control);
 mesh_ctrl = ath12k_dp_rx_h_mesh_ctl_present(ab, rx_desc);

 if (!(status->flag & RX_FLAG_IV_STRIPPED)) {
  crypto_len = ath12k_dp_rx_crypto_param_len(ar, enctype);
  crypto_hdr = skb_push(msdu, crypto_len);
  ath12k_dp_rx_desc_get_crypto_header(ab, rx_desc, crypto_hdr, enctype);
 }

 skb_push(msdu, hdr_len);
 memcpy(msdu->data, &hdr, min(hdr_len, sizeof(hdr)));

 if (rxcb->is_mcbc)
  status->flag &= ~RX_FLAG_PN_VALIDATED;

 /* Add QOS header */
 if (ieee80211_is_data_qos(hdr.frame_control)) {
  struct ieee80211_hdr *qos_ptr = (struct ieee80211_hdr *)msdu->data;

  qos_ctl = cpu_to_le16(rxcb->tid & IEEE80211_QOS_CTL_TID_MASK);
  if (mesh_ctrl)
   qos_ctl |= cpu_to_le16(IEEE80211_QOS_CTL_MESH_CONTROL_PRESENT);

  memcpy(ieee80211_get_qos_ctl(qos_ptr), &qos_ctl, IEEE80211_QOS_CTL_LEN);
 }
}

static void ath12k_dp_rx_h_undecap_eth(struct ath12k *ar,
           struct sk_buff *msdu,
           enum hal_encrypt_type enctype,
           struct ieee80211_rx_status *status)
{
 struct ieee80211_hdr *hdr;
 struct ethhdr *eth;
 u8 da[ETH_ALEN];
 u8 sa[ETH_ALEN];
 struct ath12k_skb_rxcb *rxcb = ATH12K_SKB_RXCB(msdu);
 struct ath12k_dp_rx_rfc1042_hdr rfc = {0xaa, 0xaa, 0x03, {0x00, 0x00, 0x00}};

 eth = (struct ethhdr *)msdu->data;
 ether_addr_copy(da, eth->h_dest);
 ether_addr_copy(sa, eth->h_source);
 rfc.snap_type = eth->h_proto;
 skb_pull(msdu, sizeof(*eth));
 memcpy(skb_push(msdu, sizeof(rfc)), &rfc,
        sizeof(rfc));
 ath12k_get_dot11_hdr_from_rx_desc(ar, msdu, rxcb, status, enctype);

 /* original 802.11 header has a different DA and in
 * case of 4addr it may also have different SA
 */
 hdr = (struct ieee80211_hdr *)msdu->data;
 ether_addr_copy(ieee80211_get_DA(hdr), da);
 ether_addr_copy(ieee80211_get_SA(hdr), sa);
}

static void ath12k_dp_rx_h_undecap(struct ath12k *ar, struct sk_buff *msdu,
       struct hal_rx_desc *rx_desc,
       enum hal_encrypt_type enctype,
       struct ieee80211_rx_status *status,
       bool decrypted)
{
 struct ath12k_base *ab = ar->ab;
 u8 decap;
 struct ethhdr *ehdr;

 decap = ath12k_dp_rx_h_decap_type(ab, rx_desc);

 switch (decap) {
 case DP_RX_DECAP_TYPE_NATIVE_WIFI:
  ath12k_dp_rx_h_undecap_nwifi(ar, msdu, enctype, status);
  break;
 case DP_RX_DECAP_TYPE_RAW:
  ath12k_dp_rx_h_undecap_raw(ar, msdu, enctype, status,
        decrypted);
  break;
 case DP_RX_DECAP_TYPE_ETHERNET2_DIX:
  ehdr = (struct ethhdr *)msdu->data;

  /* mac80211 allows fast path only for authorized STA */
  if (ehdr->h_proto == cpu_to_be16(ETH_P_PAE)) {
   ATH12K_SKB_RXCB(msdu)->is_eapol = true;
   ath12k_dp_rx_h_undecap_eth(ar, msdu, enctype, status);
   break;
  }

  /* PN for mcast packets will be validated in mac80211;
 * remove eth header and add 802.11 header.
 */
  if (ATH12K_SKB_RXCB(msdu)->is_mcbc && decrypted)
   ath12k_dp_rx_h_undecap_eth(ar, msdu, enctype, status);
  break;
 case DP_RX_DECAP_TYPE_8023:
  /* TODO: Handle undecap for these formats */
  break;
 }
}

struct ath12k_peer *
ath12k_dp_rx_h_find_peer(struct ath12k_base *ab, struct sk_buff *msdu,
    struct ath12k_dp_rx_info *rx_info)
{
 struct ath12k_skb_rxcb *rxcb = ATH12K_SKB_RXCB(msdu);
 struct ath12k_peer *peer = NULL;

 lockdep_assert_held(&ab->base_lock);

 if (rxcb->peer_id)
  peer = ath12k_peer_find_by_id(ab, rxcb->peer_id);

 if (peer)
  return peer;

 if (rx_info->addr2_present)
  peer = ath12k_peer_find_by_addr(ab, rx_info->addr2);

 return peer;
}

static void ath12k_dp_rx_h_mpdu(struct ath12k *ar,
    struct sk_buff *msdu,
    struct hal_rx_desc *rx_desc,
    struct ath12k_dp_rx_info *rx_info)
{
 struct ath12k_base *ab = ar->ab;
--> --------------------

--> maximum size reached

--> --------------------