net/mac80211/debugfs_sta.c

/*
 * Copyright 2003-2005	Devicescape Software, Inc.
 * Copyright (c) 2006	Jiri Benc <jbenc@suse.cz>
 * Copyright 2007	Johannes Berg <johannes@sipsolutions.net>
 * Copyright 2013-2014  Intel Mobile Communications GmbH
 * Copyright(c) 2016 Intel Deutschland GmbH
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 */

#include <linux/debugfs.h>
#include <linux/ieee80211.h>
#include "ieee80211_i.h"
#include "debugfs.h"
#include "debugfs_sta.h"
#include "sta_info.h"
#include "driver-ops.h"

/* sta attributtes */

#define STA_READ(name, field, format_string)				\
static ssize_t sta_ ##name## _read(struct file *file,			\
				   char __user *userbuf,		\
				   size_t count, loff_t *ppos)		\
{									\
	struct sta_info *sta = file->private_data;			\
	return mac80211_format_buffer(userbuf, count, ppos, 		\
				      format_string, sta->field);	\
}
#define STA_READ_D(name, field) STA_READ(name, field, "%d\n")

#define STA_OPS(name)							\
static const struct file_operations sta_ ##name## _ops = {		\
	.read = sta_##name##_read,					\
	.open = simple_open,						\
	.llseek = generic_file_llseek,					\
}

#define STA_OPS_RW(name)						\
static const struct file_operations sta_ ##name## _ops = {		\
	.read = sta_##name##_read,					\
	.write = sta_##name##_write,					\
	.open = simple_open,						\
	.llseek = generic_file_llseek,					\
}

#define STA_FILE(name, field, format)					\
		STA_READ_##format(name, field)				\
		STA_OPS(name)

STA_FILE(aid, sta.aid, D);

static const char * const sta_flag_names[] = {
#define FLAG(F) [WLAN_STA_##F] = #F
	FLAG(AUTH),
	FLAG(ASSOC),
	FLAG(PS_STA),
	FLAG(AUTHORIZED),
	FLAG(SHORT_PREAMBLE),
	FLAG(WDS),
	FLAG(CLEAR_PS_FILT),
	FLAG(MFP),
	FLAG(BLOCK_BA),
	FLAG(PS_DRIVER),
	FLAG(PSPOLL),
	FLAG(TDLS_PEER),
	FLAG(TDLS_PEER_AUTH),
	FLAG(TDLS_INITIATOR),
	FLAG(TDLS_CHAN_SWITCH),
	FLAG(TDLS_OFF_CHANNEL),
	FLAG(TDLS_WIDER_BW),
	FLAG(UAPSD),
	FLAG(SP),
	FLAG(4ADDR_EVENT),
	FLAG(INSERTED),
	FLAG(RATE_CONTROL),
	FLAG(TOFFSET_KNOWN),
	FLAG(MPSP_OWNER),
	FLAG(MPSP_RECIPIENT),
	FLAG(PS_DELIVER),
	FLAG(USES_ENCRYPTION),
#undef FLAG
};

static ssize_t sta_flags_read(struct file *file, char __user *userbuf,
			      size_t count, loff_t *ppos)
{
	char buf[16 * NUM_WLAN_STA_FLAGS], *pos = buf;
	char *end = buf + sizeof(buf) - 1;
	struct sta_info *sta = file->private_data;
	unsigned int flg;

	BUILD_BUG_ON(ARRAY_SIZE(sta_flag_names) != NUM_WLAN_STA_FLAGS);

	for (flg = 0; flg < NUM_WLAN_STA_FLAGS; flg++) {
		if (test_sta_flag(sta, flg))
			pos += scnprintf(pos, end - pos, "%s\n",
					 sta_flag_names[flg]);
	}

	return simple_read_from_buffer(userbuf, count, ppos, buf, strlen(buf));
}
STA_OPS(flags);

static ssize_t sta_num_ps_buf_frames_read(struct file *file,
					  char __user *userbuf,
					  size_t count, loff_t *ppos)
{
	struct sta_info *sta = file->private_data;
	char buf[17*IEEE80211_NUM_ACS], *p = buf;
	int ac;

	for (ac = 0; ac < IEEE80211_NUM_ACS; ac++)
		p += scnprintf(p, sizeof(buf)+buf-p, "AC%d: %d\n", ac,
			       skb_queue_len(&sta->ps_tx_buf[ac]) +
			       skb_queue_len(&sta->tx_filtered[ac]));
	return simple_read_from_buffer(userbuf, count, ppos, buf, p - buf);
}
STA_OPS(num_ps_buf_frames);

static ssize_t sta_last_seq_ctrl_read(struct file *file, char __user *userbuf,
				      size_t count, loff_t *ppos)
{
	char buf[15*IEEE80211_NUM_TIDS], *p = buf;
	int i;
	struct sta_info *sta = file->private_data;
	for (i = 0; i < IEEE80211_NUM_TIDS; i++)
		p += scnprintf(p, sizeof(buf)+buf-p, "%x ",
			       le16_to_cpu(sta->last_seq_ctrl[i]));
	p += scnprintf(p, sizeof(buf)+buf-p, "\n");
	return simple_read_from_buffer(userbuf, count, ppos, buf, p - buf);
}
STA_OPS(last_seq_ctrl);

#define AQM_TXQ_ENTRY_LEN 130

static ssize_t sta_aqm_read(struct file *file, char __user *userbuf,
			size_t count, loff_t *ppos)
{
	struct sta_info *sta = file->private_data;
	struct ieee80211_local *local = sta->local;
	size_t bufsz = AQM_TXQ_ENTRY_LEN*(IEEE80211_NUM_TIDS+1);
	char *buf = kzalloc(bufsz, GFP_KERNEL), *p = buf;
	struct txq_info *txqi;
	ssize_t rv;
	int i;

	if (!buf)
		return -ENOMEM;

	spin_lock_bh(&local->fq.lock);
	rcu_read_lock();
	p += scnprintf(p, bufsz + buf - p, "tid target interval ecn\n");
	for (i = 0; i < IEEE80211_NUM_TIDS; i++) {
		p += scnprintf(p,
		       bufsz+buf-p,
		       "%u %uus %uus %s\n", i,
		       codel_time_to_us(sta->cparams[i].target),
		       codel_time_to_us(sta->cparams[i].interval),
		       sta->cparams[i].ecn ? "yes" : "no");
	}
	p += scnprintf(p,
		       bufsz+buf-p,
		       "tid ac backlog-bytes backlog-packets new-flows drops marks overlimit collisions tx-bytes tx-packets flags\n");

	for (i = 0; i < IEEE80211_NUM_TIDS; i++) {
		txqi = to_txq_info(sta->sta.txq[i]);
		p += scnprintf(p, bufsz+buf-p,
			       "%d %d %u %u %u %u %u %u %u %u %u 0x%lx(%s%s%s)\n",
			       txqi->txq.tid,
			       txqi->txq.ac,
			       txqi->tin.backlog_bytes,
			       txqi->tin.backlog_packets,
			       txqi->tin.flows,
			       txqi->cstats.drop_count,
			       txqi->cstats.ecn_mark,
			       txqi->tin.overlimit,
			       txqi->tin.collisions,
			       txqi->tin.tx_bytes,
			       txqi->tin.tx_packets,
			       txqi->flags,
			       txqi->flags & (1<<IEEE80211_TXQ_STOP) ? "STOP" : "RUN",
			       txqi->flags & (1<<IEEE80211_TXQ_AMPDU) ? " AMPDU" : "",
			       txqi->flags & (1<<IEEE80211_TXQ_NO_AMSDU) ? " NO-AMSDU" : "");
	}

	rcu_read_unlock();
	spin_unlock_bh(&local->fq.lock);

	rv = simple_read_from_buffer(userbuf, count, ppos, buf, p - buf);
	kfree(buf);
	return rv;
}
STA_OPS(aqm);

static ssize_t sta_airtime_read(struct file *file, char __user *userbuf,
				size_t count, loff_t *ppos)
{
	struct sta_info *sta = file->private_data;
	struct ieee80211_local *local = sta->sdata->local;
	size_t bufsz = 600;
	char *buf = kzalloc(bufsz, GFP_KERNEL), *p = buf;
	u64 rx_airtime = 0, tx_airtime = 0;
	s64 deficit[IEEE80211_NUM_TIDS];
	ssize_t rv;
	int tid;

	if (!buf)
		return -ENOMEM;

	for (tid = 0; tid < IEEE80211_NUM_TIDS; tid++) {
		int ac = ieee80211_ac_from_tid(tid);
		spin_lock_bh(&local->active_txq_lock[ac]);
		rx_airtime += sta->airtime[tid].rx_airtime;
		tx_airtime += sta->airtime[tid].tx_airtime;
		deficit[tid] = sta->airtime[tid].deficit;
		spin_unlock_bh(&local->active_txq_lock[ac]);
	}

	p += scnprintf(p, bufsz + buf - p,
		       "RX: %llu us\nTX: %llu us\nWeight: %u"
		       " Total pending tx airtime: %u us\n",
		       rx_airtime, tx_airtime, sta->airtime_weight,
		       atomic_read(&sta->local->aql_total_pending_airtime));

	p += scnprintf(p, bufsz + buf - p,
		       "TID:\t Deficit: Queue Depth:\n");
	for (tid = 0; tid < IEEE80211_NUM_TIDS; tid++) {
		p += scnprintf(p, bufsz + buf - p,
			       "%d\t %lld\t %u\n", tid, deficit[tid],
			       atomic_read(&sta->airtime[tid].aql_tx_pending));
	}

	rv = simple_read_from_buffer(userbuf, count, ppos, buf, p - buf);
	kfree(buf);
	return rv;
}

static ssize_t sta_airtime_write(struct file *file, const char __user *userbuf,
				 size_t count, loff_t *ppos)
{
	struct sta_info *sta = file->private_data;
	struct ieee80211_local *local = sta->sdata->local;
	int tid;

	for (tid = 0; tid < IEEE80211_NUM_TIDS; tid++) {
		int ac = ieee80211_ac_from_tid(tid);
		spin_lock_bh(&local->active_txq_lock[ac]);
		sta->airtime[tid].rx_airtime = 0;
		sta->airtime[tid].tx_airtime = 0;
		sta->airtime[tid].deficit = sta->airtime_weight;
		spin_unlock_bh(&local->active_txq_lock[ac]);
	}

	return count;
}
STA_OPS_RW(airtime);

static ssize_t sta_aql_read(struct file *file, char __user *userbuf,
				size_t count, loff_t *ppos)
{
	struct sta_info *sta = file->private_data;
	struct ieee80211_local *local = sta->sdata->local;
	size_t bufsz = 600;
	char *buf = kzalloc(bufsz, GFP_KERNEL), *p = buf;
	s64 q_depth[IEEE80211_NUM_TIDS];
	s64 q_limit_l[IEEE80211_NUM_TIDS];
	s64 q_limit_h[IEEE80211_NUM_TIDS];
	ssize_t rv;
	int tid;

	if (!buf)
		return -ENOMEM;

	for (tid = 0; tid < IEEE80211_NUM_TIDS; tid++) {
		int ac = ieee80211_ac_from_tid(tid);
		spin_lock_bh(&local->active_txq_lock[ac]);
		q_limit_l[tid] = sta->airtime[tid].aql_limit_low;
		q_limit_h[tid] = sta->airtime[tid].aql_limit_high;
		spin_unlock_bh(&local->active_txq_lock[ac]);
		q_depth[tid] = atomic_read(&sta->airtime[tid].aql_tx_pending);
	}

	p += scnprintf(p, bufsz + buf - p,
		"TID:\tQueue Limit Low:\tQueue Limit High:\tQueue Depth:\n");
	for (tid = 0; tid < IEEE80211_NUM_TIDS; tid++) {
		p += scnprintf(p, bufsz + buf - p,
			       "%d\t %llu\t\t %llu\t\t %llu\n",
			       tid,
			       q_limit_l[tid], q_limit_h[tid],
			       q_depth[tid]);
	}

	rv = simple_read_from_buffer(userbuf, count, ppos, buf, p - buf);
	kfree(buf);
	return rv;
}

static ssize_t sta_aql_write(struct file *file, const char __user *userbuf,
				 size_t count, loff_t *ppos)
{
	struct sta_info *sta = file->private_data;
	u32 tid, q_limit_l, q_limit_h;
	char _buf[100] = {}, *buf = _buf;

	if (count > sizeof(_buf))
		return -EINVAL;

	if (copy_from_user(buf, userbuf, count))
		return -EFAULT;

	buf[sizeof(_buf) - 1] = '\0';
	if (sscanf(buf, "queue limit %u %u %u", &tid, &q_limit_l, &q_limit_h)
	    != 3)
		return -EINVAL;

	if (tid >= IEEE80211_NUM_TIDS)
		return -EINVAL;

	sta->airtime[tid].aql_limit_low = q_limit_l;
	sta->airtime[tid].aql_limit_high = q_limit_h;

	return count;
}
STA_OPS_RW(aql);


static ssize_t sta_agg_status_read(struct file *file, char __user *userbuf,
					size_t count, loff_t *ppos)
{
	char buf[71 + IEEE80211_NUM_TIDS * 40], *p = buf;
	int i;
	struct sta_info *sta = file->private_data;
	struct tid_ampdu_rx *tid_rx;
	struct tid_ampdu_tx *tid_tx;

	rcu_read_lock();

	p += scnprintf(p, sizeof(buf) + buf - p, "next dialog_token: %#02x\n",
			sta->ampdu_mlme.dialog_token_allocator + 1);
	p += scnprintf(p, sizeof(buf) + buf - p,
		       "TID\t\tRX\tDTKN\tSSN\t\tTX\tDTKN\tpending\n");

	for (i = 0; i < IEEE80211_NUM_TIDS; i++) {
		bool tid_rx_valid;

		tid_rx = rcu_dereference(sta->ampdu_mlme.tid_rx[i]);
		tid_tx = rcu_dereference(sta->ampdu_mlme.tid_tx[i]);
		tid_rx_valid = test_bit(i, sta->ampdu_mlme.agg_session_valid);

		p += scnprintf(p, sizeof(buf) + buf - p, "%02d", i);
		p += scnprintf(p, sizeof(buf) + buf - p, "\t\t%x",
			       tid_rx_valid);
		p += scnprintf(p, sizeof(buf) + buf - p, "\t%#.2x",
			       tid_rx_valid ?
					sta->ampdu_mlme.tid_rx_token[i] : 0);
		p += scnprintf(p, sizeof(buf) + buf - p, "\t%#.3x",
				tid_rx ? tid_rx->ssn : 0);

		p += scnprintf(p, sizeof(buf) + buf - p, "\t\t%x", !!tid_tx);
		p += scnprintf(p, sizeof(buf) + buf - p, "\t%#.2x",
				tid_tx ? tid_tx->dialog_token : 0);
		p += scnprintf(p, sizeof(buf) + buf - p, "\t%03d",
				tid_tx ? skb_queue_len(&tid_tx->pending) : 0);
		p += scnprintf(p, sizeof(buf) + buf - p, "\n");
	}
	rcu_read_unlock();

	return simple_read_from_buffer(userbuf, count, ppos, buf, p - buf);
}

static ssize_t sta_agg_status_write(struct file *file, const char __user *userbuf,
				    size_t count, loff_t *ppos)
{
	char _buf[25] = {}, *buf = _buf;
	struct sta_info *sta = file->private_data;
	bool start, tx;
	unsigned long tid;
	char *pos;
	int ret, timeout = 5000;

	if (count > sizeof(_buf))
		return -EINVAL;

	if (copy_from_user(buf, userbuf, count))
		return -EFAULT;

	buf[sizeof(_buf) - 1] = '\0';
	pos = buf;
	buf = strsep(&pos, " ");
	if (!buf)
		return -EINVAL;

	if (!strcmp(buf, "tx"))
		tx = true;
	else if (!strcmp(buf, "rx"))
		tx = false;
	else
		return -EINVAL;

	buf = strsep(&pos, " ");
	if (!buf)
		return -EINVAL;
	if (!strcmp(buf, "start")) {
		start = true;
		if (!tx)
			return -EINVAL;
	} else if (!strcmp(buf, "stop")) {
		start = false;
	} else {
		return -EINVAL;
	}

	buf = strsep(&pos, " ");
	if (!buf)
		return -EINVAL;
	if (sscanf(buf, "timeout=%d", &timeout) == 1) {
		buf = strsep(&pos, " ");
		if (!buf || !tx || !start)
			return -EINVAL;
	}

	ret = kstrtoul(buf, 0, &tid);
	if (ret || tid >= IEEE80211_NUM_TIDS)
		return -EINVAL;

	if (tx) {
		if (start)
			ret = ieee80211_start_tx_ba_session(&sta->sta, tid,
							    timeout);
		else
			ret = ieee80211_stop_tx_ba_session(&sta->sta, tid);
	} else {
		__ieee80211_stop_rx_ba_session(sta, tid, WLAN_BACK_RECIPIENT,
					       3, true);
		ret = 0;
	}

	return ret ?: count;
}
STA_OPS_RW(agg_status);

static ssize_t sta_ht_capa_read(struct file *file, char __user *userbuf,
				size_t count, loff_t *ppos)
{
#define PRINT_HT_CAP(_cond, _str) \
	do { \
	if (_cond) \
			p += scnprintf(p, sizeof(buf)+buf-p, "\t" _str "\n"); \
	} while (0)
	char buf[512], *p = buf;
	int i;
	struct sta_info *sta = file->private_data;
	struct ieee80211_sta_ht_cap *htc = &sta->sta.ht_cap;

	p += scnprintf(p, sizeof(buf) + buf - p, "ht %ssupported\n",
			htc->ht_supported ? "" : "not ");
	if (htc->ht_supported) {
		p += scnprintf(p, sizeof(buf)+buf-p, "cap: %#.4x\n", htc->cap);

		PRINT_HT_CAP((htc->cap & BIT(0)), "RX LDPC");
		PRINT_HT_CAP((htc->cap & BIT(1)), "HT20/HT40");
		PRINT_HT_CAP(!(htc->cap & BIT(1)), "HT20");

		PRINT_HT_CAP(((htc->cap >> 2) & 0x3) == 0, "Static SM Power Save");
		PRINT_HT_CAP(((htc->cap >> 2) & 0x3) == 1, "Dynamic SM Power Save");
		PRINT_HT_CAP(((htc->cap >> 2) & 0x3) == 3, "SM Power Save disabled");

		PRINT_HT_CAP((htc->cap & BIT(4)), "RX Greenfield");
		PRINT_HT_CAP((htc->cap & BIT(5)), "RX HT20 SGI");
		PRINT_HT_CAP((htc->cap & BIT(6)), "RX HT40 SGI");
		PRINT_HT_CAP((htc->cap & BIT(7)), "TX STBC");

		PRINT_HT_CAP(((htc->cap >> 8) & 0x3) == 0, "No RX STBC");
		PRINT_HT_CAP(((htc->cap >> 8) & 0x3) == 1, "RX STBC 1-stream");
		PRINT_HT_CAP(((htc->cap >> 8) & 0x3) == 2, "RX STBC 2-streams");
		PRINT_HT_CAP(((htc->cap >> 8) & 0x3) == 3, "RX STBC 3-streams");

		PRINT_HT_CAP((htc->cap & BIT(10)), "HT Delayed Block Ack");

		PRINT_HT_CAP(!(htc->cap & BIT(11)), "Max AMSDU length: "
			     "3839 bytes");
		PRINT_HT_CAP((htc->cap & BIT(11)), "Max AMSDU length: "
			     "7935 bytes");

		/*
		 * For beacons and probe response this would mean the BSS
		 * does or does not allow the usage of DSSS/CCK HT40.
		 * Otherwise it means the STA does or does not use
		 * DSSS/CCK HT40.
		 */
		PRINT_HT_CAP((htc->cap & BIT(12)), "DSSS/CCK HT40");
		PRINT_HT_CAP(!(htc->cap & BIT(12)), "No DSSS/CCK HT40");

		/* BIT(13) is reserved */

		PRINT_HT_CAP((htc->cap & BIT(14)), "40 MHz Intolerant");

		PRINT_HT_CAP((htc->cap & BIT(15)), "L-SIG TXOP protection");

		p += scnprintf(p, sizeof(buf)+buf-p, "ampdu factor/density: %d/%d\n",
				htc->ampdu_factor, htc->ampdu_density);
		p += scnprintf(p, sizeof(buf)+buf-p, "MCS mask:");

		for (i = 0; i < IEEE80211_HT_MCS_MASK_LEN; i++)
			p += scnprintf(p, sizeof(buf)+buf-p, " %.2x",
					htc->mcs.rx_mask[i]);
		p += scnprintf(p, sizeof(buf)+buf-p, "\n");

		/* If not set this is meaningless */
		if (le16_to_cpu(htc->mcs.rx_highest)) {
			p += scnprintf(p, sizeof(buf)+buf-p,
				       "MCS rx highest: %d Mbps\n",
				       le16_to_cpu(htc->mcs.rx_highest));
		}

		p += scnprintf(p, sizeof(buf)+buf-p, "MCS tx params: %x\n",
				htc->mcs.tx_params);
	}

	return simple_read_from_buffer(userbuf, count, ppos, buf, p - buf);
}
STA_OPS(ht_capa);

static ssize_t sta_rx_stats_read(struct file *file, char __user *userbuf,
				 size_t count, loff_t *ppos)
{
#define PRINT_LINE_TAB(n,max) \
	do { \
		if (((i + 1) % n == 0) && (i != (max-1))) \
			len += scnprintf(buf + len, size - len, "\n\t\t"); \
	} while (0)
	int retval = 0, len = 0;
	char *buf;
	const int size = 4096;
	struct sta_info *sta = file->private_data;
	struct ieee80211_local *local = sta->local;
	int i;
	char *bw_str[IEEE80211_BW_NUM] = {"20", "40", "80", "160"};
	char *nss_str[IEEE80211_NSS_NUM] = {"1x1", "2x2", "3x3", "4x4"};
	char *gi_str[IEEE80211_GI_NUM] = {"LGI", "SGI"};
	char *legacy_str[IEEE80211_LEGACY_RATE_NUM] = {"1", "2", "5.5",
						       "11", "6", "9",
						       "12", "18", "24",
						       "36", "48", "54"};
	if (!local->rx_stats_enabled)
		return -ENOTSUPP;

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

	len += scnprintf(buf + len, size - len, "rx_succ_pkts:\n");
	len += scnprintf(buf + len, size - len, "VHT MCS packets: ");
	for (i = 0; i < IEEE80211_VHT_MCS_NUM; i++) {
		len += scnprintf(buf + len, size - len, "MCS %d: %llu, ",
				 i, sta->rx_vht_pkt[i]);
		PRINT_LINE_TAB(5, IEEE80211_VHT_MCS_NUM);
	}

	len += scnprintf(buf + len, size - len, "\nHT MCS packets: ");
	for (i = 0; i < IEEE80211_HT_MCS_NUM; i++) {
		len += scnprintf(buf + len, size - len, "MCS %d: %llu, ",
				 i, sta->rx_ht_pkt[i]);
		PRINT_LINE_TAB(5, IEEE80211_HT_MCS_NUM);
	}

	len += scnprintf(buf + len, size - len, "\nBW packets: ");
	for (i = 0; i < IEEE80211_BW_NUM; i++)
		len += scnprintf(buf + len, size - len, "\t%sMhz: %llu",
				 bw_str[i], sta->rx_bw_pkt[i]);

	len += scnprintf(buf + len, size - len, "\nNSS packets: ");
	for (i = 0; i < IEEE80211_NSS_NUM; i++)
		len += scnprintf(buf + len, size - len, "\t%s: %llu",
				 nss_str[i], sta->rx_nss_pkt[i]);

	len += scnprintf(buf + len, size - len, "\nGI packets: ");
	for (i = 0; i < IEEE80211_GI_NUM; i++)
		len += scnprintf(buf + len, size - len, "\t%s: %llu",
				 gi_str[i], sta->rx_gi_pkt[i]);

	len += scnprintf(buf + len, size - len, "\nlegacy rate packets: ");
	for (i = 0; i < IEEE80211_LEGACY_RATE_NUM; i++) {
		len += scnprintf(buf + len, size - len, "\t%sMbps: %llu",
				 legacy_str[i], sta->rx_legacy_pkt[i]);
		PRINT_LINE_TAB(4, IEEE80211_LEGACY_RATE_NUM);
	}

	len += scnprintf(buf + len, size - len, "\nRate table packets:  ");
	for (i = 0; i < IEEE80211_RATE_TABLE_NUM; i++) {
		len += scnprintf(buf + len, size - len, "\t%llu",
				 sta->rx_rate_pkt[i]);
		PRINT_LINE_TAB(8, IEEE80211_RATE_TABLE_NUM);
	}

	len += scnprintf(buf + len, size - len, "\n\nrx_succ_bytes: ");
	len += scnprintf(buf + len, size - len, "\nVHT MCS bytes: ");
	for (i = 0; i < IEEE80211_VHT_MCS_NUM; i++) {
		len += scnprintf(buf + len, size - len, "MCS %d: %llu, ",
				 i, sta->rx_vht_byte[i]);
		PRINT_LINE_TAB(5, IEEE80211_VHT_MCS_NUM);
	}

	len += scnprintf(buf + len, size - len, "\nHT MCS bytes: ");
	for (i = 0; i < IEEE80211_HT_MCS_NUM; i++) {
		len += scnprintf(buf + len, size - len, "MCS %d: %llu, ",
				 i, sta->rx_ht_byte[i]);
		PRINT_LINE_TAB(5, IEEE80211_HT_MCS_NUM);
	}

	len += scnprintf(buf + len, size - len, "\nNSS bytes: ");
	for (i = 0; i < IEEE80211_NSS_NUM; i++)
		len += scnprintf(buf + len, size - len, "\t%s: %llu",
				 nss_str[i], sta->rx_nss_byte[i]);

	len += scnprintf(buf + len, size - len, "\nBW bytes: ");
	for (i = 0; i < IEEE80211_BW_NUM; i++)
		len += scnprintf(buf + len, size - len, "\t%sMhz: %llu",
				 bw_str[i], sta->rx_bw_byte[i]);

	len += scnprintf(buf + len, size - len, "\nGI bytes: ");
	for (i = 0; i < IEEE80211_GI_NUM; i++)
		len += scnprintf(buf + len, size - len, "\t%s: %llu",
				 gi_str[i], sta->rx_gi_byte[i]);

	len += scnprintf(buf + len, size - len, "\nlegacy rate bytes: ");
	for (i = 0; i < IEEE80211_LEGACY_RATE_NUM; i++) {
		len += scnprintf(buf + len, size - len, "\t%sMbps: %llu",
				 legacy_str[i], sta->rx_legacy_byte[i]);
		PRINT_LINE_TAB(4, IEEE80211_LEGACY_RATE_NUM);
	}

	len += scnprintf(buf + len, size - len, "\nRate table bytes:  ");
	for (i = 0; i < IEEE80211_RATE_TABLE_NUM; i++) {
		len += scnprintf(buf + len, size - len, "\t%llu",
				 sta->rx_rate_byte[i]);
		PRINT_LINE_TAB(8, IEEE80211_RATE_TABLE_NUM);
	}
	len += scnprintf(buf + len, size - len, "\n");

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

#undef PRINT_LINE_TAB
	return retval;
}
STA_OPS(rx_stats);

static ssize_t sta_reset_rx_stats_read(struct file *file, char __user *userbuf,
				       size_t count, loff_t *ppos)
{
	return 0;
}

static ssize_t sta_reset_rx_stats_write(struct file *file,
					const char __user *userbuf,
					size_t count, loff_t *ppos)
{
	char _buf[2] = {0}, *buf = _buf;
	struct sta_info *sta = file->private_data;
	struct ieee80211_local *local = sta->local;
	struct station_info sinfo = {};
	int i;
	bool val;

	if (count > sizeof(_buf))
		return -EINVAL;

	if (copy_from_user(buf, userbuf, count))
		return -EFAULT;

	buf[sizeof(_buf) - 1] = '\0';

	if (strtobool(buf, &val) != 0)
		return -EINVAL;

	if (!val || !local->rx_stats_enabled)
		return -ENOTSUPP;

	mutex_lock(&local->sta_mtx);
	sinfo.filled = 0;
	sta_set_sinfo(sta, &sinfo);
	mutex_unlock(&local->sta_mtx);

	for (i = 0; i < IEEE80211_VHT_MCS_NUM; i++)
		sta->rx_vht_pkt[i] = 0;

	for (i = 0; i < IEEE80211_HT_MCS_NUM; i++)
		sta->rx_ht_pkt[i] = 0;

	for (i = 0; i < IEEE80211_BW_NUM; i++)
		sta->rx_bw_pkt[i] = 0;

	for (i = 0; i < IEEE80211_NSS_NUM; i++)
		sta->rx_nss_pkt[i] = 0;

	for (i = 0; i < IEEE80211_GI_NUM; i++)
		sta->rx_gi_pkt[i] = 0;

	for (i = 0; i < IEEE80211_LEGACY_RATE_NUM; i++)
		sta->rx_legacy_pkt[i] = 0;

	for (i = 0; i < IEEE80211_RATE_TABLE_NUM; i++)
		sta->rx_rate_pkt[i] = 0;

	for (i = 0; i < IEEE80211_NUM_SIGNAL; i++)
		sta->sdata->vif.rx_signal_pkt[i] = 0;

	for (i = 0; i < IEEE80211_VHT_MCS_NUM; i++)
		sta->rx_vht_byte[i] = 0;

	for (i = 0; i < IEEE80211_HT_MCS_NUM; i++)
		sta->rx_ht_byte[i] = 0;

	for (i = 0; i < IEEE80211_NSS_NUM; i++)
		sta->rx_nss_byte[i] = 0;

	for (i = 0; i < IEEE80211_BW_NUM; i++)
		sta->rx_bw_byte[i] = 0;

	for (i = 0; i < IEEE80211_GI_NUM; i++)
		sta->rx_gi_byte[i] = 0;

	for (i = 0; i < IEEE80211_LEGACY_RATE_NUM; i++)
		sta->rx_legacy_byte[i] = 0;

	for (i = 0; i < IEEE80211_RATE_TABLE_NUM; i++)
		sta->rx_rate_byte[i] = 0;

	return count;
}

STA_OPS_RW(reset_rx_stats);

static ssize_t sta_rx_duration_read(struct file *file, char __user *userbuf,
				    size_t count, loff_t *ppos)
{
	struct sta_info *sta = file->private_data;
	struct station_info sinfo = {};
	int ret = 0, len = 0;
	const int size = 32;
	char *buf;

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

	sinfo.filled = 0;
	sta_set_sinfo(sta, &sinfo);

	len = scnprintf(buf, size, "rx duration: %llu\n", sinfo.rx_duration);

	ret = simple_read_from_buffer(userbuf, count, ppos, buf, len);

	return ret;
}

STA_OPS(rx_duration);

static ssize_t sta_tx_fail_cnt_read(struct file *file, char __user *userbuf,
				    size_t count, loff_t *ppos)
{
	char buf[10 * MAX_TX_FAIL_CNT], *p = buf;
	int i;
	struct sta_info *sta = file->private_data;

	for (i = 0; i < MAX_TX_FAIL_CNT; i++) {
		if (!sta->mesh->tx_fail_cnt[i])
			continue;

		p += scnprintf(p, sizeof(buf) + buf - p, "%d : %u ",
			       i, sta->mesh->tx_fail_cnt[i]);
	}
	p += scnprintf(p, (sizeof(buf) + buf - p), "\n");
	return simple_read_from_buffer(userbuf, count, ppos, buf, p - buf);
}

static ssize_t sta_tx_fail_cnt_write(struct file *file,
				     const char __user *userbuf,
				     size_t count, loff_t *ppos)
{
	struct sta_info *sta = file->private_data;
	char _buf[2], *buf = _buf;
	int ret;
	u8 val;

	if (count > sizeof(_buf))
		return -EINVAL;

	if (copy_from_user(buf, userbuf, count))
		return -EFAULT;

	buf[sizeof(_buf) - 1] = '\0';

	ret = kstrtou8_from_user(userbuf, count, 0, &val);

	if (ret || val > 2)
		return -EINVAL;

	if (val != MESH_RESET_TX_FAIL_COUNT) {
		sta->mesh->tx_fail_log = val;
		return count;
	}

	memset(sta->mesh->tx_fail_cnt, 0, sizeof(u32) * MAX_TX_FAIL_CNT);

	return count;
}

STA_OPS_RW(tx_fail_cnt);

static ssize_t sta_vht_capa_read(struct file *file, char __user *userbuf,
				 size_t count, loff_t *ppos)
{
	char buf[512], *p = buf;
	struct sta_info *sta = file->private_data;
	struct ieee80211_sta_vht_cap *vhtc = &sta->sta.vht_cap;

	p += scnprintf(p, sizeof(buf) + buf - p, "VHT %ssupported\n",
			vhtc->vht_supported ? "" : "not ");
	if (vhtc->vht_supported) {
		p += scnprintf(p, sizeof(buf) + buf - p, "cap: %#.8x\n",
			       vhtc->cap);
#define PFLAG(a, b)							\
		do {							\
			if (vhtc->cap & IEEE80211_VHT_CAP_ ## a)	\
				p += scnprintf(p, sizeof(buf) + buf - p, \
					       "\t\t%s\n", b);		\
		} while (0)

		switch (vhtc->cap & 0x3) {
		case IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_3895:
			p += scnprintf(p, sizeof(buf) + buf - p,
				       "\t\tMAX-MPDU-3895\n");
			break;
		case IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_7991:
			p += scnprintf(p, sizeof(buf) + buf - p,
				       "\t\tMAX-MPDU-7991\n");
			break;
		case IEEE80211_VHT_CAP_MAX_MPDU_LENGTH_11454:
			p += scnprintf(p, sizeof(buf) + buf - p,
				       "\t\tMAX-MPDU-11454\n");
			break;
		default:
			p += scnprintf(p, sizeof(buf) + buf - p,
				       "\t\tMAX-MPDU-UNKNOWN\n");
		};
		switch (vhtc->cap & IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_MASK) {
		case 0:
			p += scnprintf(p, sizeof(buf) + buf - p,
				       "\t\t80Mhz\n");
			break;
		case IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160MHZ:
			p += scnprintf(p, sizeof(buf) + buf - p,
				       "\t\t160Mhz\n");
			break;
		case IEEE80211_VHT_CAP_SUPP_CHAN_WIDTH_160_80PLUS80MHZ:
			p += scnprintf(p, sizeof(buf) + buf - p,
				       "\t\t80+80Mhz\n");
			break;
		default:
			p += scnprintf(p, sizeof(buf) + buf - p,
				       "\t\tUNKNOWN-MHZ: 0x%x\n",
				       (vhtc->cap >> 2) & 0x3);
		};
		PFLAG(RXLDPC, "RXLDPC");
		PFLAG(SHORT_GI_80, "SHORT-GI-80");
		PFLAG(SHORT_GI_160, "SHORT-GI-160");
		PFLAG(TXSTBC, "TXSTBC");
		p += scnprintf(p, sizeof(buf) + buf - p,
			       "\t\tRXSTBC_%d\n", (vhtc->cap >> 8) & 0x7);
		PFLAG(SU_BEAMFORMER_CAPABLE, "SU-BEAMFORMER-CAPABLE");
		PFLAG(SU_BEAMFORMEE_CAPABLE, "SU-BEAMFORMEE-CAPABLE");
		p += scnprintf(p, sizeof(buf) + buf - p,
			"\t\tBEAMFORMEE-STS: 0x%x\n",
			(vhtc->cap & IEEE80211_VHT_CAP_BEAMFORMEE_STS_MASK) >>
			IEEE80211_VHT_CAP_BEAMFORMEE_STS_SHIFT);
		p += scnprintf(p, sizeof(buf) + buf - p,
			"\t\tSOUNDING-DIMENSIONS: 0x%x\n",
			(vhtc->cap & IEEE80211_VHT_CAP_SOUNDING_DIMENSIONS_MASK)
			>> IEEE80211_VHT_CAP_SOUNDING_DIMENSIONS_SHIFT);
		PFLAG(MU_BEAMFORMER_CAPABLE, "MU-BEAMFORMER-CAPABLE");
		PFLAG(MU_BEAMFORMEE_CAPABLE, "MU-BEAMFORMEE-CAPABLE");
		PFLAG(VHT_TXOP_PS, "TXOP-PS");
		PFLAG(HTC_VHT, "HTC-VHT");
		p += scnprintf(p, sizeof(buf) + buf - p,
			"\t\tMPDU-LENGTH-EXPONENT: 0x%x\n",
			(vhtc->cap & IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_MASK) >>
			IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_SHIFT);
		PFLAG(VHT_LINK_ADAPTATION_VHT_UNSOL_MFB,
		      "LINK-ADAPTATION-VHT-UNSOL-MFB");
		p += scnprintf(p, sizeof(buf) + buf - p,
			"\t\tLINK-ADAPTATION-VHT-MRQ-MFB: 0x%x\n",
			(vhtc->cap & IEEE80211_VHT_CAP_VHT_LINK_ADAPTATION_VHT_MRQ_MFB) >> 26);
		PFLAG(RX_ANTENNA_PATTERN, "RX-ANTENNA-PATTERN");
		PFLAG(TX_ANTENNA_PATTERN, "TX-ANTENNA-PATTERN");

		p += scnprintf(p, sizeof(buf)+buf-p, "RX MCS: %.4x\n",
			       le16_to_cpu(vhtc->vht_mcs.rx_mcs_map));
		if (vhtc->vht_mcs.rx_highest)
			p += scnprintf(p, sizeof(buf)+buf-p,
				       "MCS RX highest: %d Mbps\n",
				       le16_to_cpu(vhtc->vht_mcs.rx_highest));
		p += scnprintf(p, sizeof(buf)+buf-p, "TX MCS: %.4x\n",
			       le16_to_cpu(vhtc->vht_mcs.tx_mcs_map));
		if (vhtc->vht_mcs.tx_highest)
			p += scnprintf(p, sizeof(buf)+buf-p,
				       "MCS TX highest: %d Mbps\n",
				       le16_to_cpu(vhtc->vht_mcs.tx_highest));
	}

	return simple_read_from_buffer(userbuf, count, ppos, buf, p - buf);
}
STA_OPS(vht_capa);


static ssize_t sta_mc_bc_stats_read(struct file *file, char __user *userbuf,
				    size_t count, loff_t *ppos)
{
	char buf[100], *p = buf;
	struct sta_info *sta = file->private_data;

	rcu_read_lock();
	p += scnprintf(p, sizeof(buf) + buf - p,
		       "mc_pkts: %llu\nmc_bytes: %llu\n",
		       sta->mc_bc_stat.mc_pkts, sta->mc_bc_stat.mc_bytes);
	p += scnprintf(p, sizeof(buf) + buf - p,
		       "bc_pkts: %llu\nbc_bytes: %llu\n",
		       sta->mc_bc_stat.bc_pkts, sta->mc_bc_stat.bc_bytes);

	rcu_read_unlock();

	return simple_read_from_buffer(userbuf, count, ppos, buf, p - buf);
}

static ssize_t sta_mc_bc_stats_write(struct file *file,
				     const char __user *userbuf,
				     size_t count, loff_t *ppos)
{
	char buf[8] = {}, *pbuf = buf;
	struct sta_info *sta = file->private_data;

	if (!sta)
		return -ENOENT;

	if (count > sizeof(buf))
		return -EINVAL;

	if (copy_from_user(pbuf, userbuf, count))
		return -EFAULT;

	pbuf[sizeof(buf) - 1] = '\0';

	if (strncmp(pbuf, "reset", 5) == 0) {
		rcu_read_lock();
		spin_lock_bh(&sta->lock);
		sta->mc_bc_stat.mc_pkts = 0;
		sta->mc_bc_stat.mc_bytes = 0;
		sta->mc_bc_stat.bc_pkts = 0;
		sta->mc_bc_stat.bc_bytes = 0;
		spin_unlock_bh(&sta->lock);
		rcu_read_unlock();
	} else {
		return -EINVAL;
	}

	return count;
}

STA_OPS_RW(mc_bc_stats);

static ssize_t sta_mc_bc_rx_limit_read(struct file *file, char __user *userbuf,
				       size_t count, loff_t *ppos)
{
	char buf[100], *p = buf;
	struct sta_info *sta = file->private_data;

	rcu_read_lock();
	p += scnprintf(p, sizeof(buf) + buf - p, "mc: %u\nbc: %u\n",
		       sta->mc_rx_limit.rate, sta->bc_rx_limit.rate);

	rcu_read_unlock();

	return simple_read_from_buffer(userbuf, count, ppos, buf, p - buf);
}

static ssize_t sta_mc_bc_rx_limit_write(struct file *file,
					const char __user *userbuf,
					size_t count, loff_t *ppos)
{
	char buf[32] = {}, *pbuf = buf;
	struct sta_info *sta = file->private_data;
	int ret = 0;
	bool mc = false, bc = false;
	u32 rate = 0;

	if (!sta)
		return -ENOENT;

	if (count > sizeof(buf))
		return -EINVAL;

	if (copy_from_user(pbuf, userbuf, count))
		return -EFAULT;

	pbuf[sizeof(buf) - 1] = '\0';

	if (strncmp(pbuf, "mc:", 3) == 0) {
		pbuf += 3;
		mc = true;
	} else if (strncmp(pbuf, "bc:", 3) == 0) {
		pbuf += 3;
		bc = true;
	} else if (strncmp(pbuf, "reset", 5) == 0) {
		rcu_read_lock();
		spin_lock_bh(&sta->lock);
		/* reset both multicast and broadcast to
		 * interface default rate
		 */
		sta->mc_rx_limit.rate = sta->sdata->mc_rx_limit_rate;
		sta->bc_rx_limit.rate = sta->sdata->bc_rx_limit_rate;
		/* Calculate the bc/mc receive frame burst size */
		mc_bc_burst_size(sta);
		spin_unlock_bh(&sta->lock);
		rcu_read_unlock();
	} else {
		return -EINVAL;
	}

	if (mc || bc) {
		ret = kstrtouint(pbuf, 0, &rate);
		/* Allow the rates less than 100000Kbps(100Mbps) */
		if (rate <= 100000) {
			rcu_read_lock();
			spin_lock_bh(&sta->lock);
			mc > bc ? (sta->mc_rx_limit.rate = rate) :
				  (sta->bc_rx_limit.rate = rate);
			/* Calculate the bc/mc receive frame burst size */
			mc_bc_burst_size(sta);
			spin_unlock_bh(&sta->lock);
			rcu_read_unlock();
		} else {
			return -EINVAL;
		}
	}

	return ret ?: count;
}

STA_OPS_RW(mc_bc_rx_limit);

static ssize_t sta_per_sta_per_txq_stats_read(struct file *file,
					      char __user *userbuf,
					      size_t count, loff_t *ppos)
{
	struct sta_info *sta = file->private_data;
	struct ieee80211_local *local = sta->local;
	size_t bufsz = AQM_TXQ_ENTRY_LEN * (IEEE80211_NUM_TIDS + 2);
	char *buf = kzalloc(bufsz, GFP_KERNEL), *p = buf;
	struct txq_info *txqi;
	ssize_t rv;
	int i;

	if (!buf)
		return -ENOMEM;

	spin_lock_bh(&local->fq.lock);
	rcu_read_lock();

	p += scnprintf(p,
		       bufsz + buf - p,
		       "netdev_packets: tid tx-packets backlog-pkts drops\n");
	for (i = 0; i < ARRAY_SIZE(sta->sta.txq); i++) {
		if (!sta->sta.txq[i])
			continue;
		txqi = to_txq_info(sta->sta.txq[i]);
		p += scnprintf(p, bufsz + buf - p,
			       "%d %u %u %u\n",
			       txqi->txq.tid,
			       txqi->tin.tx_packets,
			       txqi->tin.backlog_packets,
			       txqi->cstats.drop_count);
	}

	p += scnprintf(p, bufsz + buf - p,
		       "Tx- broadcast packets: %llu multicast packets: %llu\n",
		       sta->tx_bc_packets, sta->tx_mc_packets);

	rcu_read_unlock();
	spin_unlock_bh(&local->fq.lock);

	rv = simple_read_from_buffer(userbuf, count, ppos, buf, p - buf);
	kfree(buf);
	return rv;
}

STA_OPS(per_sta_per_txq_stats);

static ssize_t sta_per_sta_rx_stats_read(struct file *file,
					 char __user *userbuf,
					 size_t count, loff_t *ppos)
{
	int retval = 0, len = 0;
	char *buf;
	const int size = 4096;
	struct sta_info *sta = file->private_data;
	struct ieee80211_local *local = sta->local;
	struct ieee80211_sub_if_data *sdata = sta->sdata;
	int i;

	if (!local->rx_stats_enabled)
		return -ENOTSUPP;

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

	rcu_read_lock();

	len += scnprintf(buf + len, size - len, "per sta rx stats:\n");
	for (i = 0; i < IEEE80211_BW_NUM; i++)
		sta->rx_packets += sta->rx_bw_pkt[i];
	len += scnprintf(buf + len, size - len, "Rx packets : %llu\n",
			 sta->rx_packets);

	len += scnprintf(buf + len, size - len,
			 "Rx multi-cast/broad-cast packets:\n");
	len += scnprintf(buf + len, size - len, "mc_pkts: %llu\n",
			 sta->mc_bc_stat.mc_pkts);
	len += scnprintf(buf + len, size - len, "bc_pkts: %llu\n",
			 sta->mc_bc_stat.bc_pkts);
	len += scnprintf(buf + len, size - len, "dropped rx_pkts: %llu\n",
			 sta->rx_stats.dropped);
	len += scnprintf(buf + len, size - len, "rx forwarded_pkts: %llu\n",
			 sta->rx_forwarded_packets);

	if (sdata->vif.type == NL80211_IFTYPE_MESH_POINT)
		len +=
		scnprintf(buf + len, size - len,
			  "\nMesh packets\nFwded_frames: %d\nFwded_mcast: %d\nFwded_unicast: %d\nDropped_frames_ttl: %d\nDropped_frames_no_route: %d\nDropped_frames_congestion: %d\n",
			  sdata->u.mesh.mshstats.fwded_frames,
			  sdata->u.mesh.mshstats.fwded_mcast,
			  sdata->u.mesh.mshstats.fwded_unicast,
			  sdata->u.mesh.mshstats.dropped_frames_ttl,
			  sdata->u.mesh.mshstats.dropped_frames_no_route,
			  sdata->u.mesh.mshstats.dropped_frames_congestion);

	rcu_read_unlock();

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

	return retval;
}

STA_OPS(per_sta_rx_stats);

#define DEBUGFS_ADD(name) \
	debugfs_create_file(#name, 0400, \
		sta->debugfs_dir, sta, &sta_ ##name## _ops);

#define DEBUGFS_ADD_MODE(name, mode) \
	debugfs_create_file(#name, mode, \
		sta->debugfs_dir, sta, &sta_ ##name## _ops)

#define DEBUGFS_ADD_COUNTER(name, field)				\
	if (sizeof(sta->field) == sizeof(u32))				\
		debugfs_create_u32(#name, 0400, sta->debugfs_dir,	\
			(u32 *) &sta->field);				\
	else								\
		debugfs_create_u64(#name, 0400, sta->debugfs_dir,	\
			(u64 *) &sta->field);

static int legacy_rate_to_index(u16 rate)
{
	int legacy_rate[] = {10, 20, 55, 110, 60, 90, 120,
			     180, 240, 360, 480, 540};
	int i;

	for (i = 0; i < IEEE80211_LEGACY_RATE_NUM; i++)
		if (rate == legacy_rate[i])
			return i;
	return -1;
}

void ieee80211_rx_h_sta_stats(struct sta_info *sta, struct sk_buff *skb)
{
	struct ieee80211_rx_status *status = IEEE80211_SKB_RXCB(skb);
	struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)skb->data;
	struct ieee80211_local *local = sta->local;
	unsigned int pkt_len = skb->len;
	unsigned int bw_idx, gi_idx, mcs_idx = 0, nss_idx = 0, i;

	if (!local->rx_stats_enabled)
		return;

	if (!ieee80211_is_data(hdr->frame_control))
		goto out;

	/* Not support 5Mhz and 10Mhz currently  */
	if (status->bw == RATE_INFO_BW_5 || status->bw == RATE_INFO_BW_10)
		goto out;

	if (status->bw == RATE_INFO_BW_160)
		bw_idx = 3;
	else if (status->bw == RATE_INFO_BW_80)
		bw_idx = 2;
	else if (status->bw == RATE_INFO_BW_40)
		bw_idx = 1;
	else
		bw_idx = 0;

	i = abs(status->signal);
	sta->sdata->vif.rx_signal_pkt[i]++;

	if (status->encoding & RX_ENC_HT) {
		mcs_idx = status->rate_idx;
		nss_idx = mcs_idx >> 3;

		if (status->rate_idx > IEEE80211_HT_MCS_NUM - 1 ||
		    nss_idx > IEEE80211_NSS_NUM - 1)
			goto out;

		sta->rx_ht_pkt[mcs_idx]++;
		sta->rx_ht_byte[mcs_idx] += pkt_len;
		sta->rx_nss_pkt[nss_idx]++;
		sta->rx_nss_byte[nss_idx] += pkt_len;
		/* To fit into rate table for HT packets */
		mcs_idx = mcs_idx % 8;
	} else if (status->encoding == RX_ENC_VHT) {
		mcs_idx = status->rate_idx;
		nss_idx = status->nss - 1;

		if (nss_idx > IEEE80211_NSS_NUM - 1 ||
		    mcs_idx > (IEEE80211_VHT_MCS_NUM - 1))
			goto out;

		sta->rx_vht_pkt[mcs_idx]++;
		sta->rx_vht_byte[mcs_idx] += pkt_len;
		sta->rx_nss_pkt[nss_idx]++;
		sta->rx_nss_byte[nss_idx] += pkt_len;
	}

	gi_idx = (status->enc_flags & RX_ENC_FLAG_SHORT_GI) ? 1 : 0;
	sta->rx_gi_pkt[gi_idx]++;
	sta->rx_gi_byte[gi_idx] += pkt_len;
	sta->rx_bw_pkt[bw_idx]++;
	sta->rx_bw_byte[bw_idx] += pkt_len;

	if (status->encoding == RX_ENC_HT || status->encoding == RX_ENC_VHT) {
		/* Update Rate table for HT and VHT packets */
		i = mcs_idx * 8 + 8 * 10 * nss_idx;
		i += bw_idx * 2 + gi_idx;
		sta->rx_rate_pkt[i]++;
		sta->rx_rate_byte[i] += pkt_len;
	} else {
		struct ieee80211_supported_band *sband;
		int shift = ieee80211_vif_get_shift(&sta->sdata->vif);
		u16 brate, legacy_rate;
		int legacy_idx;

		if (status->rate_idx > IEEE80211_LEGACY_RATE_NUM - 1)
			goto out;

		sband = ieee80211_get_sband(sta->sdata);
		if (sband) {
			brate = sband->bitrates[status->rate_idx].bitrate;
			legacy_rate = DIV_ROUND_UP(brate, 1 << shift);
		}

		legacy_idx = legacy_rate_to_index(legacy_rate);
		if (legacy_idx < 0)
			goto out;

		sta->rx_legacy_pkt[legacy_idx]++;
		sta->rx_legacy_byte[legacy_idx] += pkt_len;
		sta->rx_nss_pkt[nss_idx]++;
		sta->rx_nss_byte[nss_idx] += pkt_len;
	}
out:
	return;
}

void ieee80211_sta_debugfs_add(struct sta_info *sta)
{
	struct ieee80211_local *local = sta->local;
	struct ieee80211_sub_if_data *sdata = sta->sdata;
	struct dentry *stations_dir = sta->sdata->debugfs.subdir_stations;
	u8 mac[3*ETH_ALEN];

	if (!stations_dir)
		return;

	snprintf(mac, sizeof(mac), "%pM", sta->sta.addr);

	/*
	 * This might fail due to a race condition:
	 * When mac80211 unlinks a station, the debugfs entries
	 * remain, but it is already possible to link a new
	 * station with the same address which triggers adding
	 * it to debugfs; therefore, if the old station isn't
	 * destroyed quickly enough the old station's debugfs
	 * dir might still be around.
	 */
	sta->debugfs_dir = debugfs_create_dir(mac, stations_dir);
	if (!sta->debugfs_dir)
		return;

	DEBUGFS_ADD(flags);
	DEBUGFS_ADD(aid);
	DEBUGFS_ADD(num_ps_buf_frames);
	DEBUGFS_ADD(last_seq_ctrl);
	DEBUGFS_ADD(agg_status);
	DEBUGFS_ADD(ht_capa);
	DEBUGFS_ADD(vht_capa);
	DEBUGFS_ADD_MODE(rx_stats, 0444);
	DEBUGFS_ADD_MODE(reset_rx_stats, 0666);
	DEBUGFS_ADD(rx_duration);
	DEBUGFS_ADD(mc_bc_stats);
	DEBUGFS_ADD(mc_bc_rx_limit);
	DEBUGFS_ADD(per_sta_rx_stats);
#ifdef CONFIG_MAC80211_MESH
	DEBUGFS_ADD(tx_fail_cnt);
#endif

	DEBUGFS_ADD_COUNTER(rx_duplicates, rx_stats.num_duplicates);
	DEBUGFS_ADD_COUNTER(rx_fragments, rx_stats.fragments);
	DEBUGFS_ADD_COUNTER(tx_filtered, status_stats.filtered);

	if (local->ops->wake_tx_queue) {
		DEBUGFS_ADD(aqm);
		DEBUGFS_ADD(airtime);
		DEBUGFS_ADD(per_sta_per_txq_stats);
	}

	if (sizeof(sta->driver_buffered_tids) == sizeof(u32))
		debugfs_create_x32("driver_buffered_tids", 0400,
				   sta->debugfs_dir,
				   (u32 *)&sta->driver_buffered_tids);
	else
		debugfs_create_x64("driver_buffered_tids", 0400,
				   sta->debugfs_dir,
				   (u64 *)&sta->driver_buffered_tids);

	if (wiphy_ext_feature_isset(local->hw.wiphy,
				    NL80211_EXT_FEATURE_AQL))
		DEBUGFS_ADD(aql);

	drv_sta_add_debugfs(local, sdata, &sta->sta, sta->debugfs_dir);
}

void ieee80211_sta_debugfs_remove(struct sta_info *sta)
{
	debugfs_remove_recursive(sta->debugfs_dir);
	sta->debugfs_dir = NULL;
}