// SPDX-License-Identifier: BSD-3-Clause-Clear
/* Copyright (C) 2023 MediaTek Inc. */

#include "mt7925.h"
#include "mcu.h"

static int
mt7925_reg_set(void *data, u64 val)
{
        struct mt792x_dev *dev = data;
        u32 regval = val;

        mt792x_mutex_acquire(dev);
        mt7925_mcu_regval(dev, dev->mt76.debugfs_reg, &regval, true);
        mt792x_mutex_release(dev);

        return 0;
}

static int
mt7925_reg_get(void *data, u64 *val)
{
        struct mt792x_dev *dev = data;
        u32 regval;
        int ret;

        mt792x_mutex_acquire(dev);
        ret = mt7925_mcu_regval(dev, dev->mt76.debugfs_reg, &regval, false);
        mt792x_mutex_release(dev);
        if (!ret)
                *val = regval;

        return 0;
}

DEFINE_DEBUGFS_ATTRIBUTE(fops_regval, mt7925_reg_get, mt7925_reg_set,
                         "0x%08llx\n");
static int
mt7925_fw_debug_set(void *data, u64 val)
{
        struct mt792x_dev *dev = data;

        mt792x_mutex_acquire(dev);

        dev->fw_debug = (u8)val;
        mt7925_mcu_fw_log_2_host(dev, dev->fw_debug);

        mt792x_mutex_release(dev);

        return 0;
}

static int
mt7925_fw_debug_get(void *data, u64 *val)
{
        struct mt792x_dev *dev = data;

        *val = dev->fw_debug;

        return 0;
}

DEFINE_DEBUGFS_ATTRIBUTE(fops_fw_debug, mt7925_fw_debug_get,
                         mt7925_fw_debug_set, "%lld\n");

DEFINE_SHOW_ATTRIBUTE(mt792x_tx_stats);

static void
mt7925_seq_puts_array(struct seq_file *file, const char *str,
                      s8 val[][2], int len, u8 band_idx)
{
        int i;

        seq_printf(file, "%-22s:", str);
        for (i = 0; i < len; i++)
                if (val[i][band_idx] == 127)
                        seq_printf(file, " %6s", "N.A");
                else
                        seq_printf(file, " %6d", val[i][band_idx]);
        seq_puts(file, "\n");
}

#define mt7925_print_txpwr_entry(prefix, rate, idx)     \
({                                                      \
        mt7925_seq_puts_array(s, #prefix " (tmac)",     \
                              txpwr->rate,              \
                              ARRAY_SIZE(txpwr->rate),  \
                              idx);                     \
})

static inline void
mt7925_eht_txpwr(struct seq_file *s, struct mt7925_txpwr *txpwr, u8 band_idx)
{
        seq_printf(s, "%-22s  %6s %6s %6s %6s %6s %6s %6s %6s %6s %6s %6s %6s %6s %6s %6s %6s\n",
                   " ", "mcs0", "mcs1", "mcs2", "mcs3", "mcs4", "mcs5",
                   "mcs6", "mcs7", "mcs8", "mcs9", "mcs10", "mcs11",
                   "mcs12", "mcs13", "mcs14", "mcs15");
        mt7925_print_txpwr_entry(EHT26, eht26, band_idx);
        mt7925_print_txpwr_entry(EHT52, eht52, band_idx);
        mt7925_print_txpwr_entry(EHT106, eht106, band_idx);
        mt7925_print_txpwr_entry(EHT242, eht242, band_idx);
        mt7925_print_txpwr_entry(EHT484, eht484, band_idx);

        mt7925_print_txpwr_entry(EHT996, eht996, band_idx);
        mt7925_print_txpwr_entry(EHT996x2, eht996x2, band_idx);
        mt7925_print_txpwr_entry(EHT996x4, eht996x4, band_idx);
        mt7925_print_txpwr_entry(EHT26_52, eht26_52, band_idx);
        mt7925_print_txpwr_entry(EHT26_106, eht26_106, band_idx);
        mt7925_print_txpwr_entry(EHT484_242, eht484_242, band_idx);
        mt7925_print_txpwr_entry(EHT996_484, eht996_484, band_idx);
        mt7925_print_txpwr_entry(EHT996_484_242, eht996_484_242, band_idx);
        mt7925_print_txpwr_entry(EHT996x2_484, eht996x2_484, band_idx);
        mt7925_print_txpwr_entry(EHT996x3, eht996x3, band_idx);
        mt7925_print_txpwr_entry(EHT996x3_484, eht996x3_484, band_idx);
}

static int
mt7925_txpwr(struct seq_file *s, void *data)
{
        struct mt792x_dev *dev = dev_get_drvdata(s->private);
        struct mt7925_txpwr *txpwr = NULL;
        u8 band_idx = dev->mphy.band_idx;
        int ret = 0;

        txpwr = devm_kmalloc(dev->mt76.dev, sizeof(*txpwr), GFP_KERNEL);

        if (!txpwr)
                return -ENOMEM;

        mt792x_mutex_acquire(dev);
        ret = mt7925_get_txpwr_info(dev, band_idx, txpwr);
        mt792x_mutex_release(dev);

        if (ret)
                goto out;

        seq_printf(s, "%-22s  %6s %6s %6s %6s\n",
                   " ", "1m", "2m", "5m", "11m");
        mt7925_print_txpwr_entry(CCK, cck, band_idx);

        seq_printf(s, "%-22s  %6s %6s %6s %6s %6s %6s %6s %6s\n",
                   " ", "6m", "9m", "12m", "18m", "24m", "36m",
                   "48m", "54m");
        mt7925_print_txpwr_entry(OFDM, ofdm, band_idx);

        seq_printf(s, "%-22s  %6s %6s %6s %6s %6s %6s %6s %6s\n",
                   " ", "mcs0", "mcs1", "mcs2", "mcs3", "mcs4", "mcs5",
                   "mcs6", "mcs7");
        mt7925_print_txpwr_entry(HT20, ht20, band_idx);

        seq_printf(s, "%-22s  %6s %6s %6s %6s %6s %6s %6s %6s %6s\n",
                   " ", "mcs0", "mcs1", "mcs2", "mcs3", "mcs4", "mcs5",
                   "mcs6", "mcs7", "mcs32");
        mt7925_print_txpwr_entry(HT40, ht40, band_idx);

        seq_printf(s, "%-22s  %6s %6s %6s %6s %6s %6s %6s %6s %6s %6s %6s %6s\n",
                   " ", "mcs0", "mcs1", "mcs2", "mcs3", "mcs4", "mcs5",
                   "mcs6", "mcs7", "mcs8", "mcs9", "mcs10", "mcs11");
        mt7925_print_txpwr_entry(VHT20, vht20, band_idx);
        mt7925_print_txpwr_entry(VHT40, vht40, band_idx);

        mt7925_print_txpwr_entry(VHT80, vht80, band_idx);
        mt7925_print_txpwr_entry(VHT160, vht160, band_idx);

        mt7925_print_txpwr_entry(HE26, he26, band_idx);
        mt7925_print_txpwr_entry(HE52, he52, band_idx);
        mt7925_print_txpwr_entry(HE106, he106, band_idx);
        mt7925_print_txpwr_entry(HE242, he242, band_idx);
        mt7925_print_txpwr_entry(HE484, he484, band_idx);

        mt7925_print_txpwr_entry(HE996, he996, band_idx);
        mt7925_print_txpwr_entry(HE996x2, he996x2, band_idx);

        mt7925_eht_txpwr(s, txpwr, band_idx);

out:
        devm_kfree(dev->mt76.dev, txpwr);
        return ret;
}

static int
mt7925_pm_set(void *data, u64 val)
{
        struct mt792x_dev *dev = data;
        struct mt76_connac_pm *pm = &dev->pm;

        if (mt76_is_usb(&dev->mt76))
                return -EOPNOTSUPP;

        mutex_lock(&dev->mt76.mutex);

        if (val == pm->enable_user)
                goto out;

        if (!pm->enable_user) {
                pm->stats.last_wake_event = jiffies;
                pm->stats.last_doze_event = jiffies;
        }
        /* make sure the chip is awake here and ps_work is scheduled
         * just at end of the this routine.
         */
        pm->enable = false;
        mt76_connac_pm_wake(&dev->mphy, pm);

        pm->enable_user = val;
        mt7925_set_runtime_pm(dev);
        mt76_connac_power_save_sched(&dev->mphy, pm);
out:
        mutex_unlock(&dev->mt76.mutex);

        return 0;
}

static int
mt7925_pm_get(void *data, u64 *val)
{
        struct mt792x_dev *dev = data;

        *val = dev->pm.enable_user;

        return 0;
}

DEFINE_DEBUGFS_ATTRIBUTE(fops_pm, mt7925_pm_get, mt7925_pm_set, "%lld\n");

static int
mt7925_deep_sleep_set(void *data, u64 val)
{
        struct mt792x_dev *dev = data;
        struct mt76_connac_pm *pm = &dev->pm;
        bool monitor = !!(dev->mphy.hw->conf.flags & IEEE80211_CONF_MONITOR);
        bool enable = !!val;

        if (mt76_is_usb(&dev->mt76))
                return -EOPNOTSUPP;

        mt792x_mutex_acquire(dev);
        if (pm->ds_enable_user == enable)
                goto out;

        pm->ds_enable_user = enable;
        pm->ds_enable = enable && !monitor;
        mt7925_mcu_set_deep_sleep(dev, pm->ds_enable);
out:
        mt792x_mutex_release(dev);

        return 0;
}

static int
mt7925_deep_sleep_get(void *data, u64 *val)
{
        struct mt792x_dev *dev = data;

        *val = dev->pm.ds_enable_user;

        return 0;
}

DEFINE_DEBUGFS_ATTRIBUTE(fops_ds, mt7925_deep_sleep_get,
                         mt7925_deep_sleep_set, "%lld\n");

DEFINE_DEBUGFS_ATTRIBUTE(fops_pm_idle_timeout, mt792x_pm_idle_timeout_get,
                         mt792x_pm_idle_timeout_set, "%lld\n");

static int mt7925_chip_reset(void *data, u64 val)
{
        struct mt792x_dev *dev = data;
        int ret = 0;

        switch (val) {
        case 1:
                /* Reset wifisys directly. */
                mt792x_reset(&dev->mt76);
                break;
        default:
                /* Collect the core dump before reset wifisys. */
                mt792x_mutex_acquire(dev);
                ret = mt7925_mcu_chip_config(dev, "assert");
                mt792x_mutex_release(dev);
                break;
        }

        return ret;
}

DEFINE_DEBUGFS_ATTRIBUTE(fops_reset, NULL, mt7925_chip_reset, "%lld\n");

int mt7925_init_debugfs(struct mt792x_dev *dev)
{
        struct dentry *dir;

        dir = mt76_register_debugfs_fops(&dev->mphy, &fops_regval);
        if (!dir)
                return -ENOMEM;

        if (mt76_is_mmio(&dev->mt76))
                debugfs_create_devm_seqfile(dev->mt76.dev, "xmit-queues",
                                            dir, mt792x_queues_read);
        else
                debugfs_create_devm_seqfile(dev->mt76.dev, "xmit-queues",
                                            dir, mt76_queues_read);

        debugfs_create_devm_seqfile(dev->mt76.dev, "acq", dir,
                                    mt792x_queues_acq);
        debugfs_create_devm_seqfile(dev->mt76.dev, "txpower_sku", dir,
                                    mt7925_txpwr);
        debugfs_create_file("tx_stats", 0400, dir, dev, &mt792x_tx_stats_fops);
        debugfs_create_file("fw_debug", 0600, dir, dev, &fops_fw_debug);
        debugfs_create_file("runtime-pm", 0600, dir, dev, &fops_pm);
        debugfs_create_file("idle-timeout", 0600, dir, dev,
                            &fops_pm_idle_timeout);
        debugfs_create_file("chip_reset", 0600, dir, dev, &fops_reset);
        debugfs_create_devm_seqfile(dev->mt76.dev, "runtime_pm_stats", dir,
                                    mt792x_pm_stats);
        debugfs_create_file("deep-sleep", 0600, dir, dev, &fops_ds);

        return 0;
}