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

#include "ath9k.h"

/*************/
/* node_aggr */
/*************/

static ssize_t read_file_node_aggr(struct file *file, char __user *user_buf,
                                   size_t count, loff_t *ppos)
{
        struct ath_node *an = file->private_data;
        struct ath_softc *sc = an->sc;
        struct ath_atx_tid *tid;
        struct ath_txq *txq;
        u32 len = 0, size = 4096;
        char *buf;
        size_t retval;
        int tidno;

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

        if (!an->sta->deflink.ht_cap.ht_supported) {
                len = scnprintf(buf, size, "%s\n",
                                "HT not supported");
                goto exit;
        }

        len = scnprintf(buf, size, "Max-AMPDU: %d\n",
                        an->maxampdu);
        len += scnprintf(buf + len, size - len, "MPDU Density: %d\n\n",
                         an->mpdudensity);

        len += scnprintf(buf + len, size - len,
                         "\n%3s%11s%10s%10s%10s%10s%9s%6s%8s\n",
                         "TID", "SEQ_START", "SEQ_NEXT", "BAW_SIZE",
                         "BAW_HEAD", "BAW_TAIL", "BAR_IDX", "SCHED", "PAUSED");

        for (tidno = 0; tidno < IEEE80211_NUM_TIDS; tidno++) {
                tid = ath_node_to_tid(an, tidno);
                txq = tid->txq;
                ath_txq_lock(sc, txq);
                if (tid->active) {
                        len += scnprintf(buf + len, size - len,
                                         "%3d%11d%10d%10d%10d%10d%9d%6d\n",
                                         tid->tidno,
                                         tid->seq_start,
                                         tid->seq_next,
                                         tid->baw_size,
                                         tid->baw_head,
                                         tid->baw_tail,
                                         tid->bar_index,
                                         !list_empty(&tid->list));
                }
                ath_txq_unlock(sc, txq);
        }
exit:
        retval = simple_read_from_buffer(user_buf, count, ppos, buf, len);
        kfree(buf);

        return retval;
}

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

/*************/
/* node_recv */
/*************/

void ath_debug_rate_stats(struct ath_softc *sc,
                          struct ath_rx_status *rs,
                          struct sk_buff *skb)
{
        struct ieee80211_hdr *hdr = (struct ieee80211_hdr *) skb->data;
        struct ath_hw *ah = sc->sc_ah;
        struct ieee80211_rx_status *rxs;
        struct ath_rx_rate_stats *rstats;
        struct ieee80211_sta *sta;
        struct ath_node *an;

        if (!ieee80211_is_data(hdr->frame_control))
                return;

        rcu_read_lock();

        sta = ieee80211_find_sta_by_ifaddr(sc->hw, hdr->addr2, NULL);
        if (!sta)
                goto exit;

        an = (struct ath_node *) sta->drv_priv;
        rstats = &an->rx_rate_stats;
        rxs = IEEE80211_SKB_RXCB(skb);

        if (IS_HT_RATE(rs->rs_rate)) {
                if (rxs->rate_idx >= ARRAY_SIZE(rstats->ht_stats))
                        goto exit;

                if (rxs->bw == RATE_INFO_BW_40)
                        rstats->ht_stats[rxs->rate_idx].ht40_cnt++;
                else
                        rstats->ht_stats[rxs->rate_idx].ht20_cnt++;

                if (rxs->enc_flags & RX_ENC_FLAG_SHORT_GI)
                        rstats->ht_stats[rxs->rate_idx].sgi_cnt++;
                else
                        rstats->ht_stats[rxs->rate_idx].lgi_cnt++;

                goto exit;
        }

        if (IS_CCK_RATE(rs->rs_rate)) {
                if (rxs->enc_flags & RX_ENC_FLAG_SHORTPRE)
                        rstats->cck_stats[rxs->rate_idx].cck_sp_cnt++;
                else
                        rstats->cck_stats[rxs->rate_idx].cck_lp_cnt++;

                goto exit;
        }

        if (IS_OFDM_RATE(rs->rs_rate)) {
                if (ah->curchan->chan->band == NL80211_BAND_2GHZ)
                        rstats->ofdm_stats[rxs->rate_idx - 4].ofdm_cnt++;
                else
                        rstats->ofdm_stats[rxs->rate_idx].ofdm_cnt++;
        }
exit:
        rcu_read_unlock();
}

#define PRINT_CCK_RATE(str, i, sp)                                      \
        do {                                                            \
                len += scnprintf(buf + len, size - len,                 \
                         "%11s : %10u\n",                               \
                         str,                                           \
                         (sp) ? rstats->cck_stats[i].cck_sp_cnt :       \
                         rstats->cck_stats[i].cck_lp_cnt);              \
        } while (0)

#define PRINT_OFDM_RATE(str, i)                                 \
        do {                                                    \
                len += scnprintf(buf + len, size - len,         \
                         "%11s : %10u\n",                       \
                         str,                                   \
                         rstats->ofdm_stats[i].ofdm_cnt);       \
        } while (0)

static ssize_t read_file_node_recv(struct file *file, char __user *user_buf,
                                   size_t count, loff_t *ppos)
{
        struct ath_node *an = file->private_data;
        struct ath_softc *sc = an->sc;
        struct ath_hw *ah = sc->sc_ah;
        struct ath_rx_rate_stats *rstats;
        struct ieee80211_sta *sta = an->sta;
        enum nl80211_band band;
        u32 len = 0, size = 4096;
        char *buf;
        size_t retval;
        int i;

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

        band = ah->curchan->chan->band;
        rstats = &an->rx_rate_stats;

        if (!sta->deflink.ht_cap.ht_supported)
                goto legacy;

        len += scnprintf(buf + len, size - len,
                         "%24s%10s%10s%10s\n",
                         "HT20", "HT40", "SGI", "LGI");

        for (i = 0; i < 24; i++) {
                len += scnprintf(buf + len, size - len,
                                 "%8s%3u : %10u%10u%10u%10u\n",
                                 "MCS", i,
                                 rstats->ht_stats[i].ht20_cnt,
                                 rstats->ht_stats[i].ht40_cnt,
                                 rstats->ht_stats[i].sgi_cnt,
                                 rstats->ht_stats[i].lgi_cnt);
        }

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

legacy:
        if (band == NL80211_BAND_2GHZ) {
                PRINT_CCK_RATE("CCK-1M/LP", 0, false);
                PRINT_CCK_RATE("CCK-2M/LP", 1, false);
                PRINT_CCK_RATE("CCK-5.5M/LP", 2, false);
                PRINT_CCK_RATE("CCK-11M/LP", 3, false);

                PRINT_CCK_RATE("CCK-2M/SP", 1, true);
                PRINT_CCK_RATE("CCK-5.5M/SP", 2, true);
                PRINT_CCK_RATE("CCK-11M/SP", 3, true);
        }

        PRINT_OFDM_RATE("OFDM-6M", 0);
        PRINT_OFDM_RATE("OFDM-9M", 1);
        PRINT_OFDM_RATE("OFDM-12M", 2);
        PRINT_OFDM_RATE("OFDM-18M", 3);
        PRINT_OFDM_RATE("OFDM-24M", 4);
        PRINT_OFDM_RATE("OFDM-36M", 5);
        PRINT_OFDM_RATE("OFDM-48M", 6);
        PRINT_OFDM_RATE("OFDM-54M", 7);

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

        return retval;
}

#undef PRINT_OFDM_RATE
#undef PRINT_CCK_RATE

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

void ath9k_sta_add_debugfs(struct ieee80211_hw *hw,
                           struct ieee80211_vif *vif,
                           struct ieee80211_sta *sta,
                           struct dentry *dir)
{
        struct ath_node *an = (struct ath_node *)sta->drv_priv;

        debugfs_create_file("node_aggr", 0444, dir, an, &fops_node_aggr);
        debugfs_create_file("node_recv", 0444, dir, an, &fops_node_recv);
}