root/drivers/net/wireless/ath/ath11k/debugfs_sta.c
// SPDX-License-Identifier: BSD-3-Clause-Clear
/*
 * Copyright (c) 2018-2019 The Linux Foundation. All rights reserved.
 * Copyright (c) Qualcomm Technologies, Inc. and/or its subsidiaries.
 */

#include <linux/vmalloc.h>

#include "debugfs_sta.h"
#include "core.h"
#include "peer.h"
#include "debug.h"
#include "dp_tx.h"
#include "debugfs_htt_stats.h"

void ath11k_debugfs_sta_add_tx_stats(struct ath11k_sta *arsta,
                                     struct ath11k_per_peer_tx_stats *peer_stats,
                                     u8 legacy_rate_idx)
{
        struct rate_info *txrate = &arsta->txrate;
        struct ath11k_htt_tx_stats *tx_stats;
        int gi, mcs, bw, nss;

        if (!arsta->tx_stats)
                return;

        tx_stats = arsta->tx_stats;
        gi = FIELD_GET(RATE_INFO_FLAGS_SHORT_GI, arsta->txrate.flags);
        mcs = txrate->mcs;
        bw = ath11k_mac_mac80211_bw_to_ath11k_bw(txrate->bw);
        nss = txrate->nss - 1;

#define STATS_OP_FMT(name) tx_stats->stats[ATH11K_STATS_TYPE_##name]

        if (txrate->flags & RATE_INFO_FLAGS_HE_MCS) {
                STATS_OP_FMT(SUCC).he[0][mcs] += peer_stats->succ_bytes;
                STATS_OP_FMT(SUCC).he[1][mcs] += peer_stats->succ_pkts;
                STATS_OP_FMT(FAIL).he[0][mcs] += peer_stats->failed_bytes;
                STATS_OP_FMT(FAIL).he[1][mcs] += peer_stats->failed_pkts;
                STATS_OP_FMT(RETRY).he[0][mcs] += peer_stats->retry_bytes;
                STATS_OP_FMT(RETRY).he[1][mcs] += peer_stats->retry_pkts;
        } else if (txrate->flags & RATE_INFO_FLAGS_VHT_MCS) {
                STATS_OP_FMT(SUCC).vht[0][mcs] += peer_stats->succ_bytes;
                STATS_OP_FMT(SUCC).vht[1][mcs] += peer_stats->succ_pkts;
                STATS_OP_FMT(FAIL).vht[0][mcs] += peer_stats->failed_bytes;
                STATS_OP_FMT(FAIL).vht[1][mcs] += peer_stats->failed_pkts;
                STATS_OP_FMT(RETRY).vht[0][mcs] += peer_stats->retry_bytes;
                STATS_OP_FMT(RETRY).vht[1][mcs] += peer_stats->retry_pkts;
        } else if (txrate->flags & RATE_INFO_FLAGS_MCS) {
                STATS_OP_FMT(SUCC).ht[0][mcs] += peer_stats->succ_bytes;
                STATS_OP_FMT(SUCC).ht[1][mcs] += peer_stats->succ_pkts;
                STATS_OP_FMT(FAIL).ht[0][mcs] += peer_stats->failed_bytes;
                STATS_OP_FMT(FAIL).ht[1][mcs] += peer_stats->failed_pkts;
                STATS_OP_FMT(RETRY).ht[0][mcs] += peer_stats->retry_bytes;
                STATS_OP_FMT(RETRY).ht[1][mcs] += peer_stats->retry_pkts;
        } else {
                mcs = legacy_rate_idx;

                STATS_OP_FMT(SUCC).legacy[0][mcs] += peer_stats->succ_bytes;
                STATS_OP_FMT(SUCC).legacy[1][mcs] += peer_stats->succ_pkts;
                STATS_OP_FMT(FAIL).legacy[0][mcs] += peer_stats->failed_bytes;
                STATS_OP_FMT(FAIL).legacy[1][mcs] += peer_stats->failed_pkts;
                STATS_OP_FMT(RETRY).legacy[0][mcs] += peer_stats->retry_bytes;
                STATS_OP_FMT(RETRY).legacy[1][mcs] += peer_stats->retry_pkts;
        }

        if (peer_stats->is_ampdu) {
                tx_stats->ba_fails += peer_stats->ba_fails;

                if (txrate->flags & RATE_INFO_FLAGS_HE_MCS) {
                        STATS_OP_FMT(AMPDU).he[0][mcs] +=
                        peer_stats->succ_bytes + peer_stats->retry_bytes;
                        STATS_OP_FMT(AMPDU).he[1][mcs] +=
                        peer_stats->succ_pkts + peer_stats->retry_pkts;
                } else if (txrate->flags & RATE_INFO_FLAGS_MCS) {
                        STATS_OP_FMT(AMPDU).ht[0][mcs] +=
                        peer_stats->succ_bytes + peer_stats->retry_bytes;
                        STATS_OP_FMT(AMPDU).ht[1][mcs] +=
                        peer_stats->succ_pkts + peer_stats->retry_pkts;
                } else {
                        STATS_OP_FMT(AMPDU).vht[0][mcs] +=
                        peer_stats->succ_bytes + peer_stats->retry_bytes;
                        STATS_OP_FMT(AMPDU).vht[1][mcs] +=
                        peer_stats->succ_pkts + peer_stats->retry_pkts;
                }
                STATS_OP_FMT(AMPDU).bw[0][bw] +=
                        peer_stats->succ_bytes + peer_stats->retry_bytes;
                STATS_OP_FMT(AMPDU).nss[0][nss] +=
                        peer_stats->succ_bytes + peer_stats->retry_bytes;
                STATS_OP_FMT(AMPDU).gi[0][gi] +=
                        peer_stats->succ_bytes + peer_stats->retry_bytes;
                STATS_OP_FMT(AMPDU).bw[1][bw] +=
                        peer_stats->succ_pkts + peer_stats->retry_pkts;
                STATS_OP_FMT(AMPDU).nss[1][nss] +=
                        peer_stats->succ_pkts + peer_stats->retry_pkts;
                STATS_OP_FMT(AMPDU).gi[1][gi] +=
                        peer_stats->succ_pkts + peer_stats->retry_pkts;
        } else {
                tx_stats->ack_fails += peer_stats->ba_fails;
        }

        STATS_OP_FMT(SUCC).bw[0][bw] += peer_stats->succ_bytes;
        STATS_OP_FMT(SUCC).nss[0][nss] += peer_stats->succ_bytes;
        STATS_OP_FMT(SUCC).gi[0][gi] += peer_stats->succ_bytes;

        STATS_OP_FMT(SUCC).bw[1][bw] += peer_stats->succ_pkts;
        STATS_OP_FMT(SUCC).nss[1][nss] += peer_stats->succ_pkts;
        STATS_OP_FMT(SUCC).gi[1][gi] += peer_stats->succ_pkts;

        STATS_OP_FMT(FAIL).bw[0][bw] += peer_stats->failed_bytes;
        STATS_OP_FMT(FAIL).nss[0][nss] += peer_stats->failed_bytes;
        STATS_OP_FMT(FAIL).gi[0][gi] += peer_stats->failed_bytes;

        STATS_OP_FMT(FAIL).bw[1][bw] += peer_stats->failed_pkts;
        STATS_OP_FMT(FAIL).nss[1][nss] += peer_stats->failed_pkts;
        STATS_OP_FMT(FAIL).gi[1][gi] += peer_stats->failed_pkts;

        STATS_OP_FMT(RETRY).bw[0][bw] += peer_stats->retry_bytes;
        STATS_OP_FMT(RETRY).nss[0][nss] += peer_stats->retry_bytes;
        STATS_OP_FMT(RETRY).gi[0][gi] += peer_stats->retry_bytes;

        STATS_OP_FMT(RETRY).bw[1][bw] += peer_stats->retry_pkts;
        STATS_OP_FMT(RETRY).nss[1][nss] += peer_stats->retry_pkts;
        STATS_OP_FMT(RETRY).gi[1][gi] += peer_stats->retry_pkts;

        tx_stats->tx_duration += peer_stats->duration;
}

void ath11k_debugfs_sta_update_txcompl(struct ath11k *ar,
                                       struct hal_tx_status *ts)
{
        ath11k_dp_tx_update_txcompl(ar, ts);
}

static ssize_t ath11k_dbg_sta_dump_tx_stats(struct file *file,
                                            char __user *user_buf,
                                            size_t count, loff_t *ppos)
{
        struct ieee80211_sta *sta = file->private_data;
        struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
        struct ath11k *ar = arsta->arvif->ar;
        struct ath11k_htt_data_stats *stats;
        static const char *str_name[ATH11K_STATS_TYPE_MAX] = {"succ", "fail",
                                                              "retry", "ampdu"};
        static const char *str[ATH11K_COUNTER_TYPE_MAX] = {"bytes", "packets"};
        int len = 0, i, j, k, retval = 0;
        const int size = 2 * 4096;
        char *buf;

        if (!arsta->tx_stats)
                return -ENOENT;

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

        mutex_lock(&ar->conf_mutex);

        spin_lock_bh(&ar->data_lock);
        for (k = 0; k < ATH11K_STATS_TYPE_MAX; k++) {
                for (j = 0; j < ATH11K_COUNTER_TYPE_MAX; j++) {
                        stats = &arsta->tx_stats->stats[k];
                        len += scnprintf(buf + len, size - len, "%s_%s\n",
                                         str_name[k],
                                         str[j]);
                        len += scnprintf(buf + len, size - len,
                                         " HE MCS %s\n",
                                         str[j]);
                        for (i = 0; i < ATH11K_HE_MCS_NUM; i++)
                                len += scnprintf(buf + len, size - len,
                                                 "  %llu ",
                                                 stats->he[j][i]);
                        len += scnprintf(buf + len, size - len, "\n");
                        len += scnprintf(buf + len, size - len,
                                         " VHT MCS %s\n",
                                         str[j]);
                        for (i = 0; i < ATH11K_VHT_MCS_NUM; i++)
                                len += scnprintf(buf + len, size - len,
                                                 "  %llu ",
                                                 stats->vht[j][i]);
                        len += scnprintf(buf + len, size - len, "\n");
                        len += scnprintf(buf + len, size - len, " HT MCS %s\n",
                                         str[j]);
                        for (i = 0; i < ATH11K_HT_MCS_NUM; i++)
                                len += scnprintf(buf + len, size - len,
                                                 "  %llu ", stats->ht[j][i]);
                        len += scnprintf(buf + len, size - len, "\n");
                        len += scnprintf(buf + len, size - len,
                                        " BW %s (20,40,80,160 MHz)\n", str[j]);
                        len += scnprintf(buf + len, size - len,
                                         "  %llu %llu %llu %llu\n",
                                         stats->bw[j][0], stats->bw[j][1],
                                         stats->bw[j][2], stats->bw[j][3]);
                        len += scnprintf(buf + len, size - len,
                                         " NSS %s (1x1,2x2,3x3,4x4)\n", str[j]);
                        len += scnprintf(buf + len, size - len,
                                         "  %llu %llu %llu %llu\n",
                                         stats->nss[j][0], stats->nss[j][1],
                                         stats->nss[j][2], stats->nss[j][3]);
                        len += scnprintf(buf + len, size - len,
                                         " GI %s (0.4us,0.8us,1.6us,3.2us)\n",
                                         str[j]);
                        len += scnprintf(buf + len, size - len,
                                         "  %llu %llu %llu %llu\n",
                                         stats->gi[j][0], stats->gi[j][1],
                                         stats->gi[j][2], stats->gi[j][3]);
                        len += scnprintf(buf + len, size - len,
                                         " legacy rate %s (1,2 ... Mbps)\n  ",
                                         str[j]);
                        for (i = 0; i < ATH11K_LEGACY_NUM; i++)
                                len += scnprintf(buf + len, size - len, "%llu ",
                                                 stats->legacy[j][i]);
                        len += scnprintf(buf + len, size - len, "\n");
                }
        }

        len += scnprintf(buf + len, size - len,
                         "\nTX duration\n %llu usecs\n",
                         arsta->tx_stats->tx_duration);
        len += scnprintf(buf + len, size - len,
                        "BA fails\n %llu\n", arsta->tx_stats->ba_fails);
        len += scnprintf(buf + len, size - len,
                        "ack fails\n %llu\n", arsta->tx_stats->ack_fails);
        spin_unlock_bh(&ar->data_lock);

        if (len > size)
                len = size;
        retval = simple_read_from_buffer(user_buf, count, ppos, buf, len);
        kfree(buf);

        mutex_unlock(&ar->conf_mutex);
        return retval;
}

static const struct file_operations fops_tx_stats = {
        .read = ath11k_dbg_sta_dump_tx_stats,
        .open = simple_open,
        .owner = THIS_MODULE,
        .llseek = default_llseek,
};

#ifdef CONFIG_ATH11K_CFR
static ssize_t ath11k_dbg_sta_write_cfr_capture(struct file *file,
                                                const char __user *user_buf,
                                                size_t count, loff_t *ppos)
{
        struct ieee80211_sta *sta = file->private_data;
        struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
        struct ath11k *ar = arsta->arvif->ar;
        struct ath11k_cfr *cfr = &ar->cfr;
        struct wmi_peer_cfr_capture_conf_arg arg;
        u32 cfr_capture_enable = 0, cfr_capture_bw  = 0;
        u32 cfr_capture_method = 0, cfr_capture_period = 0;
        char buf[64] = {};
        int ret;

        simple_write_to_buffer(buf, sizeof(buf) - 1, ppos, user_buf, count);

        guard(mutex)(&ar->conf_mutex);

        if (ar->state != ATH11K_STATE_ON)
                return -ENETDOWN;

        if (!ar->cfr_enabled)
                return -EINVAL;

        ret = sscanf(buf, "%u %u %u %u", &cfr_capture_enable, &cfr_capture_bw,
                     &cfr_capture_period, &cfr_capture_method);

        if (ret < 1 || (cfr_capture_enable && ret != 4))
                return -EINVAL;

        if (cfr_capture_enable == arsta->cfr_capture.cfr_enable &&
            (cfr_capture_period &&
             cfr_capture_period == arsta->cfr_capture.cfr_period) &&
            cfr_capture_bw == arsta->cfr_capture.cfr_bw &&
            cfr_capture_method == arsta->cfr_capture.cfr_method)
                return count;

        if (!cfr_capture_enable &&
            cfr_capture_enable == arsta->cfr_capture.cfr_enable)
                return count;

        if (cfr_capture_enable > WMI_PEER_CFR_CAPTURE_ENABLE ||
            cfr_capture_bw > WMI_PEER_CFR_CAPTURE_BW_80 ||
            cfr_capture_method > ATH11K_CFR_CAPTURE_METHOD_NULL_FRAME_WITH_PHASE ||
            cfr_capture_period > WMI_PEER_CFR_PERIODICITY_MAX)
                return -EINVAL;

        /* Target expects cfr period in multiple of 10 */
        if (cfr_capture_period % 10) {
                ath11k_err(ar->ab, "periodicity should be 10x\n");
                return -EINVAL;
        }

        if (ar->cfr.cfr_enabled_peer_cnt >= ATH11K_MAX_CFR_ENABLED_CLIENTS &&
            !arsta->cfr_capture.cfr_enable) {
                ath11k_err(ar->ab, "CFR enable peer threshold reached %u\n",
                           ar->cfr.cfr_enabled_peer_cnt);
                return -EINVAL;
        }

        if (!cfr_capture_enable) {
                cfr_capture_bw = arsta->cfr_capture.cfr_bw;
                cfr_capture_period = arsta->cfr_capture.cfr_period;
                cfr_capture_method = arsta->cfr_capture.cfr_method;
        }

        arg.request = cfr_capture_enable;
        arg.periodicity = cfr_capture_period;
        arg.bw = cfr_capture_bw;
        arg.method = cfr_capture_method;

        ret = ath11k_wmi_peer_set_cfr_capture_conf(ar, arsta->arvif->vdev_id,
                                                   sta->addr, &arg);
        if (ret) {
                ath11k_warn(ar->ab,
                            "failed to send cfr capture info: vdev_id %u peer %pM: %d\n",
                            arsta->arvif->vdev_id, sta->addr, ret);
                return ret;
        }

        spin_lock_bh(&ar->cfr.lock);

        if (cfr_capture_enable &&
            cfr_capture_enable != arsta->cfr_capture.cfr_enable)
                cfr->cfr_enabled_peer_cnt++;
        else if (!cfr_capture_enable)
                cfr->cfr_enabled_peer_cnt--;

        spin_unlock_bh(&ar->cfr.lock);

        arsta->cfr_capture.cfr_enable = cfr_capture_enable;
        arsta->cfr_capture.cfr_period = cfr_capture_period;
        arsta->cfr_capture.cfr_bw = cfr_capture_bw;
        arsta->cfr_capture.cfr_method = cfr_capture_method;

        return count;
}

static ssize_t ath11k_dbg_sta_read_cfr_capture(struct file *file,
                                               char __user *user_buf,
                                               size_t count, loff_t *ppos)
{
        struct ieee80211_sta *sta = file->private_data;
        struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
        struct ath11k *ar = arsta->arvif->ar;
        char buf[512] = {};
        int len = 0;

        mutex_lock(&ar->conf_mutex);

        len += scnprintf(buf + len, sizeof(buf) - len, "cfr_enabled = %d\n",
                         arsta->cfr_capture.cfr_enable);
        len += scnprintf(buf + len, sizeof(buf) - len, "bandwidth = %d\n",
                         arsta->cfr_capture.cfr_bw);
        len += scnprintf(buf + len, sizeof(buf) - len, "period = %d\n",
                         arsta->cfr_capture.cfr_period);
        len += scnprintf(buf + len, sizeof(buf) - len, "cfr_method = %d\n",
                         arsta->cfr_capture.cfr_method);

        mutex_unlock(&ar->conf_mutex);

        return simple_read_from_buffer(user_buf, count, ppos, buf, len);
}

static const struct file_operations fops_peer_cfr_capture = {
        .write = ath11k_dbg_sta_write_cfr_capture,
        .read = ath11k_dbg_sta_read_cfr_capture,
        .open = simple_open,
        .owner = THIS_MODULE,
        .llseek = default_llseek,
};
#endif /* CONFIG_ATH11K_CFR */

static ssize_t ath11k_dbg_sta_dump_rx_stats(struct file *file,
                                            char __user *user_buf,
                                            size_t count, loff_t *ppos)
{
        struct ieee80211_sta *sta = file->private_data;
        struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
        struct ath11k *ar = arsta->arvif->ar;
        struct ath11k_rx_peer_stats *rx_stats = arsta->rx_stats;
        int len = 0, i, retval = 0;
        const int size = 4096;
        char *buf;

        if (!rx_stats)
                return -ENOENT;

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

        mutex_lock(&ar->conf_mutex);
        spin_lock_bh(&ar->ab->base_lock);

        len += scnprintf(buf + len, size - len, "RX peer stats:\n");
        len += scnprintf(buf + len, size - len, "Num of MSDUs: %llu\n",
                         rx_stats->num_msdu);
        len += scnprintf(buf + len, size - len, "Num of MSDUs with TCP L4: %llu\n",
                         rx_stats->tcp_msdu_count);
        len += scnprintf(buf + len, size - len, "Num of MSDUs with UDP L4: %llu\n",
                         rx_stats->udp_msdu_count);
        len += scnprintf(buf + len, size - len, "Num of MSDUs part of AMPDU: %llu\n",
                         rx_stats->ampdu_msdu_count);
        len += scnprintf(buf + len, size - len, "Num of MSDUs not part of AMPDU: %llu\n",
                         rx_stats->non_ampdu_msdu_count);
        len += scnprintf(buf + len, size - len, "Num of MSDUs using STBC: %llu\n",
                         rx_stats->stbc_count);
        len += scnprintf(buf + len, size - len, "Num of MSDUs beamformed: %llu\n",
                         rx_stats->beamformed_count);
        len += scnprintf(buf + len, size - len, "Num of MPDUs with FCS ok: %llu\n",
                         rx_stats->num_mpdu_fcs_ok);
        len += scnprintf(buf + len, size - len, "Num of MPDUs with FCS error: %llu\n",
                         rx_stats->num_mpdu_fcs_err);
        len += scnprintf(buf + len, size - len,
                         "GI: 0.8us %llu 0.4us %llu 1.6us %llu 3.2us %llu\n",
                         rx_stats->gi_count[0], rx_stats->gi_count[1],
                         rx_stats->gi_count[2], rx_stats->gi_count[3]);
        len += scnprintf(buf + len, size - len,
                         "BW: 20Mhz %llu 40Mhz %llu 80Mhz %llu 160Mhz %llu\n",
                         rx_stats->bw_count[0], rx_stats->bw_count[1],
                         rx_stats->bw_count[2], rx_stats->bw_count[3]);
        len += scnprintf(buf + len, size - len, "BCC %llu LDPC %llu\n",
                         rx_stats->coding_count[0], rx_stats->coding_count[1]);
        len += scnprintf(buf + len, size - len,
                         "preamble: 11A %llu 11B %llu 11N %llu 11AC %llu 11AX %llu\n",
                         rx_stats->pream_cnt[0], rx_stats->pream_cnt[1],
                         rx_stats->pream_cnt[2], rx_stats->pream_cnt[3],
                         rx_stats->pream_cnt[4]);
        len += scnprintf(buf + len, size - len,
                         "reception type: SU %llu MU_MIMO %llu MU_OFDMA %llu MU_OFDMA_MIMO %llu\n",
                         rx_stats->reception_type[0], rx_stats->reception_type[1],
                         rx_stats->reception_type[2], rx_stats->reception_type[3]);
        len += scnprintf(buf + len, size - len, "TID(0-15) Legacy TID(16):");
        for (i = 0; i <= IEEE80211_NUM_TIDS; i++)
                len += scnprintf(buf + len, size - len, "%llu ", rx_stats->tid_count[i]);
        len += scnprintf(buf + len, size - len, "\nMCS(0-11) Legacy MCS(12):");
        for (i = 0; i < HAL_RX_MAX_MCS + 1; i++)
                len += scnprintf(buf + len, size - len, "%llu ", rx_stats->mcs_count[i]);
        len += scnprintf(buf + len, size - len, "\nNSS(1-8):");
        for (i = 0; i < HAL_RX_MAX_NSS; i++)
                len += scnprintf(buf + len, size - len, "%llu ", rx_stats->nss_count[i]);
        len += scnprintf(buf + len, size - len, "\nRX Duration:%llu ",
                         rx_stats->rx_duration);
        len += scnprintf(buf + len, size - len,
                         "\nDCM: %llu\nRU: 26 %llu 52: %llu 106: %llu 242: %llu 484: %llu 996: %llu\n",
                         rx_stats->dcm_count, rx_stats->ru_alloc_cnt[0],
                         rx_stats->ru_alloc_cnt[1], rx_stats->ru_alloc_cnt[2],
                         rx_stats->ru_alloc_cnt[3], rx_stats->ru_alloc_cnt[4],
                         rx_stats->ru_alloc_cnt[5]);

        len += scnprintf(buf + len, size - len, "\n");

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

        if (len > size)
                len = size;
        retval = simple_read_from_buffer(user_buf, count, ppos, buf, len);
        kfree(buf);

        mutex_unlock(&ar->conf_mutex);
        return retval;
}

static const struct file_operations fops_rx_stats = {
        .read = ath11k_dbg_sta_dump_rx_stats,
        .open = simple_open,
        .owner = THIS_MODULE,
        .llseek = default_llseek,
};

static int
ath11k_dbg_sta_open_htt_peer_stats(struct inode *inode, struct file *file)
{
        struct ieee80211_sta *sta = inode->i_private;
        struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
        struct ath11k *ar = arsta->arvif->ar;
        struct debug_htt_stats_req *stats_req;
        int type = ar->debug.htt_stats.type;
        int ret;

        if ((type != ATH11K_DBG_HTT_EXT_STATS_PEER_INFO &&
             type != ATH11K_DBG_HTT_EXT_STATS_PEER_CTRL_PATH_TXRX_STATS) ||
            type == ATH11K_DBG_HTT_EXT_STATS_RESET)
                return -EPERM;

        stats_req = vzalloc(sizeof(*stats_req) + ATH11K_HTT_STATS_BUF_SIZE);
        if (!stats_req)
                return -ENOMEM;

        mutex_lock(&ar->conf_mutex);
        ar->debug.htt_stats.stats_req = stats_req;
        stats_req->type = type;
        memcpy(stats_req->peer_addr, sta->addr, ETH_ALEN);
        ret = ath11k_debugfs_htt_stats_req(ar);
        mutex_unlock(&ar->conf_mutex);
        if (ret < 0)
                goto out;

        file->private_data = stats_req;
        return 0;
out:
        vfree(stats_req);
        ar->debug.htt_stats.stats_req = NULL;
        return ret;
}

static int
ath11k_dbg_sta_release_htt_peer_stats(struct inode *inode, struct file *file)
{
        struct ieee80211_sta *sta = inode->i_private;
        struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
        struct ath11k *ar = arsta->arvif->ar;

        mutex_lock(&ar->conf_mutex);
        vfree(file->private_data);
        ar->debug.htt_stats.stats_req = NULL;
        mutex_unlock(&ar->conf_mutex);

        return 0;
}

static ssize_t ath11k_dbg_sta_read_htt_peer_stats(struct file *file,
                                                  char __user *user_buf,
                                                  size_t count, loff_t *ppos)
{
        struct debug_htt_stats_req *stats_req = file->private_data;
        char *buf;
        u32 length = 0;

        buf = stats_req->buf;
        length = min_t(u32, stats_req->buf_len, ATH11K_HTT_STATS_BUF_SIZE);
        return simple_read_from_buffer(user_buf, count, ppos, buf, length);
}

static const struct file_operations fops_htt_peer_stats = {
        .open = ath11k_dbg_sta_open_htt_peer_stats,
        .release = ath11k_dbg_sta_release_htt_peer_stats,
        .read = ath11k_dbg_sta_read_htt_peer_stats,
        .owner = THIS_MODULE,
        .llseek = default_llseek,
};

static ssize_t ath11k_dbg_sta_write_peer_pktlog(struct file *file,
                                                const char __user *buf,
                                                size_t count, loff_t *ppos)
{
        struct ieee80211_sta *sta = file->private_data;
        struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
        struct ath11k *ar = arsta->arvif->ar;
        int ret, enable;

        mutex_lock(&ar->conf_mutex);

        if (ar->state != ATH11K_STATE_ON) {
                ret = -ENETDOWN;
                goto out;
        }

        ret = kstrtoint_from_user(buf, count, 0, &enable);
        if (ret)
                goto out;

        ar->debug.pktlog_peer_valid = enable;
        memcpy(ar->debug.pktlog_peer_addr, sta->addr, ETH_ALEN);

        /* Send peer based pktlog enable/disable */
        ret = ath11k_wmi_pdev_peer_pktlog_filter(ar, sta->addr, enable);
        if (ret) {
                ath11k_warn(ar->ab, "failed to set peer pktlog filter %pM: %d\n",
                            sta->addr, ret);
                goto out;
        }

        ath11k_dbg(ar->ab, ATH11K_DBG_WMI, "peer pktlog filter set to %d\n",
                   enable);
        ret = count;

out:
        mutex_unlock(&ar->conf_mutex);
        return ret;
}

static ssize_t ath11k_dbg_sta_read_peer_pktlog(struct file *file,
                                               char __user *ubuf,
                                               size_t count, loff_t *ppos)
{
        struct ieee80211_sta *sta = file->private_data;
        struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
        struct ath11k *ar = arsta->arvif->ar;
        char buf[32] = {};
        int len;

        mutex_lock(&ar->conf_mutex);
        len = scnprintf(buf, sizeof(buf), "%08x %pM\n",
                        ar->debug.pktlog_peer_valid,
                        ar->debug.pktlog_peer_addr);
        mutex_unlock(&ar->conf_mutex);

        return simple_read_from_buffer(ubuf, count, ppos, buf, len);
}

static const struct file_operations fops_peer_pktlog = {
        .write = ath11k_dbg_sta_write_peer_pktlog,
        .read = ath11k_dbg_sta_read_peer_pktlog,
        .open = simple_open,
        .owner = THIS_MODULE,
        .llseek = default_llseek,
};

static ssize_t ath11k_dbg_sta_write_delba(struct file *file,
                                          const char __user *user_buf,
                                          size_t count, loff_t *ppos)
{
        struct ieee80211_sta *sta = file->private_data;
        struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
        struct ath11k *ar = arsta->arvif->ar;
        u32 tid, initiator, reason;
        int ret;
        char buf[64] = {};

        ret = simple_write_to_buffer(buf, sizeof(buf) - 1, ppos,
                                     user_buf, count);
        if (ret <= 0)
                return ret;

        ret = sscanf(buf, "%u %u %u", &tid, &initiator, &reason);
        if (ret != 3)
                return -EINVAL;

        /* Valid TID values are 0 through 15 */
        if (tid > HAL_DESC_REO_NON_QOS_TID - 1)
                return -EINVAL;

        mutex_lock(&ar->conf_mutex);
        if (ar->state != ATH11K_STATE_ON ||
            arsta->aggr_mode != ATH11K_DBG_AGGR_MODE_MANUAL) {
                ret = count;
                goto out;
        }

        ret = ath11k_wmi_delba_send(ar, arsta->arvif->vdev_id, sta->addr,
                                    tid, initiator, reason);
        if (ret) {
                ath11k_warn(ar->ab, "failed to send delba: vdev_id %u peer %pM tid %u initiator %u reason %u\n",
                            arsta->arvif->vdev_id, sta->addr, tid, initiator,
                            reason);
        }
        ret = count;
out:
        mutex_unlock(&ar->conf_mutex);
        return ret;
}

static const struct file_operations fops_delba = {
        .write = ath11k_dbg_sta_write_delba,
        .open = simple_open,
        .owner = THIS_MODULE,
        .llseek = default_llseek,
};

static ssize_t ath11k_dbg_sta_write_addba_resp(struct file *file,
                                               const char __user *user_buf,
                                               size_t count, loff_t *ppos)
{
        struct ieee80211_sta *sta = file->private_data;
        struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
        struct ath11k *ar = arsta->arvif->ar;
        u32 tid, status;
        int ret;
        char buf[64] = {};

        ret = simple_write_to_buffer(buf, sizeof(buf) - 1, ppos,
                                     user_buf, count);
        if (ret <= 0)
                return ret;

        ret = sscanf(buf, "%u %u", &tid, &status);
        if (ret != 2)
                return -EINVAL;

        /* Valid TID values are 0 through 15 */
        if (tid > HAL_DESC_REO_NON_QOS_TID - 1)
                return -EINVAL;

        mutex_lock(&ar->conf_mutex);
        if (ar->state != ATH11K_STATE_ON ||
            arsta->aggr_mode != ATH11K_DBG_AGGR_MODE_MANUAL) {
                ret = count;
                goto out;
        }

        ret = ath11k_wmi_addba_set_resp(ar, arsta->arvif->vdev_id, sta->addr,
                                        tid, status);
        if (ret) {
                ath11k_warn(ar->ab, "failed to send addba response: vdev_id %u peer %pM tid %u status%u\n",
                            arsta->arvif->vdev_id, sta->addr, tid, status);
        }
        ret = count;
out:
        mutex_unlock(&ar->conf_mutex);
        return ret;
}

static const struct file_operations fops_addba_resp = {
        .write = ath11k_dbg_sta_write_addba_resp,
        .open = simple_open,
        .owner = THIS_MODULE,
        .llseek = default_llseek,
};

static ssize_t ath11k_dbg_sta_write_addba(struct file *file,
                                          const char __user *user_buf,
                                          size_t count, loff_t *ppos)
{
        struct ieee80211_sta *sta = file->private_data;
        struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
        struct ath11k *ar = arsta->arvif->ar;
        u32 tid, buf_size;
        int ret;
        char buf[64] = {};

        ret = simple_write_to_buffer(buf, sizeof(buf) - 1, ppos,
                                     user_buf, count);
        if (ret <= 0)
                return ret;

        ret = sscanf(buf, "%u %u", &tid, &buf_size);
        if (ret != 2)
                return -EINVAL;

        /* Valid TID values are 0 through 15 */
        if (tid > HAL_DESC_REO_NON_QOS_TID - 1)
                return -EINVAL;

        mutex_lock(&ar->conf_mutex);
        if (ar->state != ATH11K_STATE_ON ||
            arsta->aggr_mode != ATH11K_DBG_AGGR_MODE_MANUAL) {
                ret = count;
                goto out;
        }

        ret = ath11k_wmi_addba_send(ar, arsta->arvif->vdev_id, sta->addr,
                                    tid, buf_size);
        if (ret) {
                ath11k_warn(ar->ab, "failed to send addba request: vdev_id %u peer %pM tid %u buf_size %u\n",
                            arsta->arvif->vdev_id, sta->addr, tid, buf_size);
        }

        ret = count;
out:
        mutex_unlock(&ar->conf_mutex);
        return ret;
}

static const struct file_operations fops_addba = {
        .write = ath11k_dbg_sta_write_addba,
        .open = simple_open,
        .owner = THIS_MODULE,
        .llseek = default_llseek,
};

static ssize_t ath11k_dbg_sta_read_aggr_mode(struct file *file,
                                             char __user *user_buf,
                                             size_t count, loff_t *ppos)
{
        struct ieee80211_sta *sta = file->private_data;
        struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
        struct ath11k *ar = arsta->arvif->ar;
        char buf[64];
        int len = 0;

        mutex_lock(&ar->conf_mutex);
        len = scnprintf(buf, sizeof(buf) - len,
                        "aggregation mode: %s\n\n%s\n%s\n",
                        (arsta->aggr_mode == ATH11K_DBG_AGGR_MODE_AUTO) ?
                        "auto" : "manual", "auto = 0", "manual = 1");
        mutex_unlock(&ar->conf_mutex);

        return simple_read_from_buffer(user_buf, count, ppos, buf, len);
}

static ssize_t ath11k_dbg_sta_write_aggr_mode(struct file *file,
                                              const char __user *user_buf,
                                              size_t count, loff_t *ppos)
{
        struct ieee80211_sta *sta = file->private_data;
        struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
        struct ath11k *ar = arsta->arvif->ar;
        u32 aggr_mode;
        int ret;

        if (kstrtouint_from_user(user_buf, count, 0, &aggr_mode))
                return -EINVAL;

        if (aggr_mode >= ATH11K_DBG_AGGR_MODE_MAX)
                return -EINVAL;

        mutex_lock(&ar->conf_mutex);
        if (ar->state != ATH11K_STATE_ON ||
            aggr_mode == arsta->aggr_mode) {
                ret = count;
                goto out;
        }

        ret = ath11k_wmi_addba_clear_resp(ar, arsta->arvif->vdev_id, sta->addr);
        if (ret) {
                ath11k_warn(ar->ab, "failed to clear addba session ret: %d\n",
                            ret);
                goto out;
        }

        arsta->aggr_mode = aggr_mode;
out:
        mutex_unlock(&ar->conf_mutex);
        return ret;
}

static const struct file_operations fops_aggr_mode = {
        .read = ath11k_dbg_sta_read_aggr_mode,
        .write = ath11k_dbg_sta_write_aggr_mode,
        .open = simple_open,
        .owner = THIS_MODULE,
        .llseek = default_llseek,
};

static ssize_t
ath11k_write_htt_peer_stats_reset(struct file *file,
                                  const char __user *user_buf,
                                  size_t count, loff_t *ppos)
{
        struct ieee80211_sta *sta = file->private_data;
        struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
        struct ath11k *ar = arsta->arvif->ar;
        struct htt_ext_stats_cfg_params cfg_params = {};
        int ret;
        u8 type;

        ret = kstrtou8_from_user(user_buf, count, 0, &type);
        if (ret)
                return ret;

        if (!type)
                return ret;

        mutex_lock(&ar->conf_mutex);
        cfg_params.cfg0 = HTT_STAT_PEER_INFO_MAC_ADDR;
        cfg_params.cfg0 |= FIELD_PREP(GENMASK(15, 1),
                                HTT_PEER_STATS_REQ_MODE_FLUSH_TQM);

        cfg_params.cfg1 = HTT_STAT_DEFAULT_PEER_REQ_TYPE;

        cfg_params.cfg2 |= FIELD_PREP(GENMASK(7, 0), sta->addr[0]);
        cfg_params.cfg2 |= FIELD_PREP(GENMASK(15, 8), sta->addr[1]);
        cfg_params.cfg2 |= FIELD_PREP(GENMASK(23, 16), sta->addr[2]);
        cfg_params.cfg2 |= FIELD_PREP(GENMASK(31, 24), sta->addr[3]);

        cfg_params.cfg3 |= FIELD_PREP(GENMASK(7, 0), sta->addr[4]);
        cfg_params.cfg3 |= FIELD_PREP(GENMASK(15, 8), sta->addr[5]);

        cfg_params.cfg3 |= ATH11K_HTT_PEER_STATS_RESET;

        ret = ath11k_dp_tx_htt_h2t_ext_stats_req(ar,
                                                 ATH11K_DBG_HTT_EXT_STATS_PEER_INFO,
                                                 &cfg_params,
                                                 0ULL);
        if (ret) {
                ath11k_warn(ar->ab, "failed to send htt peer stats request: %d\n", ret);
                mutex_unlock(&ar->conf_mutex);
                return ret;
        }

        mutex_unlock(&ar->conf_mutex);

        ret = count;

        return ret;
}

static const struct file_operations fops_htt_peer_stats_reset = {
        .write = ath11k_write_htt_peer_stats_reset,
        .open = simple_open,
        .owner = THIS_MODULE,
        .llseek = default_llseek,
};

static ssize_t ath11k_dbg_sta_read_peer_ps_state(struct file *file,
                                                 char __user *user_buf,
                                                 size_t count, loff_t *ppos)
{
        struct ieee80211_sta *sta = file->private_data;
        struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
        struct ath11k *ar = arsta->arvif->ar;
        char buf[20];
        int len;

        spin_lock_bh(&ar->data_lock);

        len = scnprintf(buf, sizeof(buf), "%d\n", arsta->peer_ps_state);

        spin_unlock_bh(&ar->data_lock);

        return simple_read_from_buffer(user_buf, count, ppos, buf, len);
}

static const struct file_operations fops_peer_ps_state = {
        .open = simple_open,
        .read = ath11k_dbg_sta_read_peer_ps_state,
        .owner = THIS_MODULE,
        .llseek = default_llseek,
};

static ssize_t ath11k_dbg_sta_read_current_ps_duration(struct file *file,
                                                       char __user *user_buf,
                                                       size_t count,
                                                       loff_t *ppos)
{
        struct ieee80211_sta *sta = file->private_data;
        struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
        struct ath11k *ar = arsta->arvif->ar;
        u64 time_since_station_in_power_save;
        char buf[20];
        int len;

        spin_lock_bh(&ar->data_lock);

        if (arsta->peer_ps_state == WMI_PEER_PS_STATE_ON &&
            arsta->peer_current_ps_valid)
                time_since_station_in_power_save = jiffies_to_msecs(jiffies
                                                - arsta->ps_start_jiffies);
        else
                time_since_station_in_power_save = 0;

        len = scnprintf(buf, sizeof(buf), "%llu\n",
                        time_since_station_in_power_save);
        spin_unlock_bh(&ar->data_lock);

        return simple_read_from_buffer(user_buf, count, ppos, buf, len);
}

static const struct file_operations fops_current_ps_duration = {
        .open = simple_open,
        .read = ath11k_dbg_sta_read_current_ps_duration,
        .owner = THIS_MODULE,
        .llseek = default_llseek,
};

static ssize_t ath11k_dbg_sta_read_total_ps_duration(struct file *file,
                                                     char __user *user_buf,
                                                     size_t count, loff_t *ppos)
{
        struct ieee80211_sta *sta = file->private_data;
        struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
        struct ath11k *ar = arsta->arvif->ar;
        char buf[20];
        u64 power_save_duration;
        int len;

        spin_lock_bh(&ar->data_lock);

        if (arsta->peer_ps_state == WMI_PEER_PS_STATE_ON &&
            arsta->peer_current_ps_valid)
                power_save_duration = jiffies_to_msecs(jiffies
                                                - arsta->ps_start_jiffies)
                                                + arsta->ps_total_duration;
        else
                power_save_duration = arsta->ps_total_duration;

        len = scnprintf(buf, sizeof(buf), "%llu\n", power_save_duration);

        spin_unlock_bh(&ar->data_lock);

        return simple_read_from_buffer(user_buf, count, ppos, buf, len);
}

static const struct file_operations fops_total_ps_duration = {
        .open = simple_open,
        .read = ath11k_dbg_sta_read_total_ps_duration,
        .owner = THIS_MODULE,
        .llseek = default_llseek,
};

void ath11k_debugfs_sta_op_add(struct ieee80211_hw *hw, struct ieee80211_vif *vif,
                               struct ieee80211_sta *sta, struct dentry *dir)
{
        struct ath11k *ar = hw->priv;

        if (ath11k_debugfs_is_extd_tx_stats_enabled(ar))
                debugfs_create_file("tx_stats", 0400, dir, sta,
                                    &fops_tx_stats);
        if (ath11k_debugfs_is_extd_rx_stats_enabled(ar))
                debugfs_create_file("rx_stats", 0400, dir, sta,
                                    &fops_rx_stats);

        debugfs_create_file("htt_peer_stats", 0400, dir, sta,
                            &fops_htt_peer_stats);

        debugfs_create_file("peer_pktlog", 0644, dir, sta,
                            &fops_peer_pktlog);

        debugfs_create_file("aggr_mode", 0644, dir, sta, &fops_aggr_mode);
        debugfs_create_file("addba", 0200, dir, sta, &fops_addba);
        debugfs_create_file("addba_resp", 0200, dir, sta, &fops_addba_resp);
        debugfs_create_file("delba", 0200, dir, sta, &fops_delba);

        if (test_bit(WMI_TLV_SERVICE_PER_PEER_HTT_STATS_RESET,
                     ar->ab->wmi_ab.svc_map))
                debugfs_create_file("htt_peer_stats_reset", 0600, dir, sta,
                                    &fops_htt_peer_stats_reset);

#ifdef CONFIG_ATH11K_CFR
        if (test_bit(WMI_TLV_SERVICE_CFR_CAPTURE_SUPPORT,
                     ar->ab->wmi_ab.svc_map))
                debugfs_create_file("cfr_capture", 0600, dir, sta,
                                    &fops_peer_cfr_capture);
#endif/* CONFIG_ATH11K_CFR */

        debugfs_create_file("peer_ps_state", 0400, dir, sta,
                            &fops_peer_ps_state);

        if (test_bit(WMI_TLV_SERVICE_PEER_POWER_SAVE_DURATION_SUPPORT,
                     ar->ab->wmi_ab.svc_map)) {
                debugfs_create_file("current_ps_duration", 0440, dir, sta,
                                    &fops_current_ps_duration);
                debugfs_create_file("total_ps_duration", 0440, dir, sta,
                                    &fops_total_ps_duration);
        }
}