// SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause
/*
 * Copyright (C) 2005-2014, 2020-2021 Intel Corporation
 * Copyright (C) 2016 Intel Deutschland GmbH
 */
#include <linux/slab.h>
#include <linux/string.h>
#include <linux/export.h>

#include "iwl-drv.h"
#include "iwl-phy-db.h"
#include "iwl-debug.h"
#include "iwl-op-mode.h"
#include "iwl-trans.h"

struct iwl_phy_db_entry {
        u16     size;
        u8      *data;
};

/**
 * struct iwl_phy_db - stores phy configuration and calibration data.
 *
 * @cfg: phy configuration.
 * @calib_nch: non channel specific calibration data.
 * @n_group_papd: number of entries in papd channel group.
 * @calib_ch_group_papd: calibration data related to papd channel group.
 * @n_group_txp: number of entries in tx power channel group.
 * @calib_ch_group_txp: calibration data related to tx power chanel group.
 * @trans: transport layer
 */
struct iwl_phy_db {
        struct iwl_phy_db_entry cfg;
        struct iwl_phy_db_entry calib_nch;
        int n_group_papd;
        struct iwl_phy_db_entry *calib_ch_group_papd;
        int n_group_txp;
        struct iwl_phy_db_entry *calib_ch_group_txp;

        struct iwl_trans *trans;
};

enum iwl_phy_db_section_type {
        IWL_PHY_DB_CFG = 1,
        IWL_PHY_DB_CALIB_NCH,
        IWL_PHY_DB_UNUSED,
        IWL_PHY_DB_CALIB_CHG_PAPD,
        IWL_PHY_DB_CALIB_CHG_TXP,
        IWL_PHY_DB_MAX
};

#define PHY_DB_CMD 0x6c

/* for parsing of tx power channel group data that comes from the firmware*/
struct iwl_phy_db_chg_txp {
        __le32 space;
        __le16 max_channel_idx;
} __packed;

struct iwl_phy_db *iwl_phy_db_init(struct iwl_trans *trans)
{
        struct iwl_phy_db *phy_db = kzalloc_obj(struct iwl_phy_db);

        if (!phy_db)
                return phy_db;

        phy_db->trans = trans;

        phy_db->n_group_txp = -1;
        phy_db->n_group_papd = -1;

        /* TODO: add default values of the phy db. */
        return phy_db;
}
IWL_EXPORT_SYMBOL(iwl_phy_db_init);

/*
 * get phy db section: returns a pointer to a phy db section specified by
 * type and channel group id.
 */
static struct iwl_phy_db_entry *
iwl_phy_db_get_section(struct iwl_phy_db *phy_db,
                       enum iwl_phy_db_section_type type,
                       u16 chg_id)
{
        if (!phy_db || type >= IWL_PHY_DB_MAX)
                return NULL;

        switch (type) {
        case IWL_PHY_DB_CFG:
                return &phy_db->cfg;
        case IWL_PHY_DB_CALIB_NCH:
                return &phy_db->calib_nch;
        case IWL_PHY_DB_CALIB_CHG_PAPD:
                if (chg_id >= phy_db->n_group_papd)
                        return NULL;
                return &phy_db->calib_ch_group_papd[chg_id];
        case IWL_PHY_DB_CALIB_CHG_TXP:
                if (chg_id >= phy_db->n_group_txp)
                        return NULL;
                return &phy_db->calib_ch_group_txp[chg_id];
        default:
                return NULL;
        }
        return NULL;
}

static void iwl_phy_db_free_section(struct iwl_phy_db *phy_db,
                                    enum iwl_phy_db_section_type type,
                                    u16 chg_id)
{
        struct iwl_phy_db_entry *entry =
                                iwl_phy_db_get_section(phy_db, type, chg_id);
        if (!entry)
                return;

        kfree(entry->data);
        entry->data = NULL;
        entry->size = 0;
}

void iwl_phy_db_free(struct iwl_phy_db *phy_db)
{
        int i;

        if (!phy_db)
                return;

        iwl_phy_db_free_section(phy_db, IWL_PHY_DB_CFG, 0);
        iwl_phy_db_free_section(phy_db, IWL_PHY_DB_CALIB_NCH, 0);

        for (i = 0; i < phy_db->n_group_papd; i++)
                iwl_phy_db_free_section(phy_db, IWL_PHY_DB_CALIB_CHG_PAPD, i);
        kfree(phy_db->calib_ch_group_papd);

        for (i = 0; i < phy_db->n_group_txp; i++)
                iwl_phy_db_free_section(phy_db, IWL_PHY_DB_CALIB_CHG_TXP, i);
        kfree(phy_db->calib_ch_group_txp);

        kfree(phy_db);
}
IWL_EXPORT_SYMBOL(iwl_phy_db_free);

int iwl_phy_db_set_section(struct iwl_phy_db *phy_db,
                           struct iwl_rx_packet *pkt)
{
        unsigned int pkt_len = iwl_rx_packet_payload_len(pkt);
        struct iwl_calib_res_notif_phy_db *phy_db_notif =
                        (struct iwl_calib_res_notif_phy_db *)pkt->data;
        enum iwl_phy_db_section_type type;
        u16 size;
        struct iwl_phy_db_entry *entry;
        u16 chg_id = 0;

        if (pkt_len < sizeof(*phy_db_notif))
                return -EINVAL;

        type = le16_to_cpu(phy_db_notif->type);
        size = le16_to_cpu(phy_db_notif->length);

        if (pkt_len < sizeof(*phy_db_notif) + size)
                return -EINVAL;

        if (!phy_db)
                return -EINVAL;

        if (type == IWL_PHY_DB_CALIB_CHG_PAPD) {
                chg_id = le16_to_cpup((__le16 *)phy_db_notif->data);
                if (phy_db && !phy_db->calib_ch_group_papd) {
                        /*
                         * Firmware sends the largest index first, so we can use
                         * it to know how much we should allocate.
                         */
                        phy_db->calib_ch_group_papd = kzalloc_objs(struct iwl_phy_db_entry,
                                                                   chg_id + 1,
                                                                   GFP_ATOMIC);
                        if (!phy_db->calib_ch_group_papd)
                                return -ENOMEM;
                        phy_db->n_group_papd = chg_id + 1;
                }
        } else if (type == IWL_PHY_DB_CALIB_CHG_TXP) {
                chg_id = le16_to_cpup((__le16 *)phy_db_notif->data);
                if (phy_db && !phy_db->calib_ch_group_txp) {
                        /*
                         * Firmware sends the largest index first, so we can use
                         * it to know how much we should allocate.
                         */
                        phy_db->calib_ch_group_txp = kzalloc_objs(struct iwl_phy_db_entry,
                                                                  chg_id + 1,
                                                                  GFP_ATOMIC);
                        if (!phy_db->calib_ch_group_txp)
                                return -ENOMEM;
                        phy_db->n_group_txp = chg_id + 1;
                }
        }

        entry = iwl_phy_db_get_section(phy_db, type, chg_id);
        if (!entry)
                return -EINVAL;

        kfree(entry->data);
        entry->data = kmemdup(phy_db_notif->data, size, GFP_ATOMIC);
        if (!entry->data) {
                entry->size = 0;
                return -ENOMEM;
        }

        entry->size = size;

        IWL_DEBUG_INFO(phy_db->trans,
                       "%s(%d): [PHYDB]SET: Type %d , Size: %d\n",
                       __func__, __LINE__, type, size);

        return 0;
}
IWL_EXPORT_SYMBOL(iwl_phy_db_set_section);

static int is_valid_channel(u16 ch_id)
{
        if (ch_id <= 14 ||
            (36 <= ch_id && ch_id <= 64 && ch_id % 4 == 0) ||
            (100 <= ch_id && ch_id <= 140 && ch_id % 4 == 0) ||
            (145 <= ch_id && ch_id <= 165 && ch_id % 4 == 1))
                return 1;
        return 0;
}

static u8 ch_id_to_ch_index(u16 ch_id)
{
        if (WARN_ON(!is_valid_channel(ch_id)))
                return 0xff;

        if (ch_id <= 14)
                return ch_id - 1;
        if (ch_id <= 64)
                return (ch_id + 20) / 4;
        if (ch_id <= 140)
                return (ch_id - 12) / 4;
        return (ch_id - 13) / 4;
}


static u16 channel_id_to_papd(u16 ch_id)
{
        if (WARN_ON(!is_valid_channel(ch_id)))
                return 0xff;

        if (1 <= ch_id && ch_id <= 14)
                return 0;
        if (36 <= ch_id && ch_id <= 64)
                return 1;
        if (100 <= ch_id && ch_id <= 140)
                return 2;
        return 3;
}

static u16 channel_id_to_txp(struct iwl_phy_db *phy_db, u16 ch_id)
{
        struct iwl_phy_db_chg_txp *txp_chg;
        int i;
        u8 ch_index = ch_id_to_ch_index(ch_id);
        if (ch_index == 0xff)
                return 0xff;

        for (i = 0; i < phy_db->n_group_txp; i++) {
                txp_chg = (void *)phy_db->calib_ch_group_txp[i].data;
                if (!txp_chg)
                        return 0xff;
                /*
                 * Looking for the first channel group that its max channel is
                 * higher then wanted channel.
                 */
                if (le16_to_cpu(txp_chg->max_channel_idx) >= ch_index)
                        return i;
        }
        return 0xff;
}
static
int iwl_phy_db_get_section_data(struct iwl_phy_db *phy_db,
                                u32 type, u8 **data, u16 *size, u16 ch_id)
{
        struct iwl_phy_db_entry *entry;
        u16 ch_group_id = 0;

        if (!phy_db)
                return -EINVAL;

        /* find wanted channel group */
        if (type == IWL_PHY_DB_CALIB_CHG_PAPD)
                ch_group_id = channel_id_to_papd(ch_id);
        else if (type == IWL_PHY_DB_CALIB_CHG_TXP)
                ch_group_id = channel_id_to_txp(phy_db, ch_id);

        entry = iwl_phy_db_get_section(phy_db, type, ch_group_id);
        if (!entry)
                return -EINVAL;

        *data = entry->data;
        *size = entry->size;

        IWL_DEBUG_INFO(phy_db->trans,
                       "%s(%d): [PHYDB] GET: Type %d , Size: %d\n",
                       __func__, __LINE__, type, *size);

        return 0;
}

static int iwl_send_phy_db_cmd(struct iwl_phy_db *phy_db, u16 type,
                               u16 length, void *data)
{
        struct iwl_phy_db_cmd phy_db_cmd;
        struct iwl_host_cmd cmd = {
                .id = PHY_DB_CMD,
        };

        IWL_DEBUG_INFO(phy_db->trans,
                       "Sending PHY-DB hcmd of type %d, of length %d\n",
                       type, length);

        /* Set phy db cmd variables */
        phy_db_cmd.type = cpu_to_le16(type);
        phy_db_cmd.length = cpu_to_le16(length);

        /* Set hcmd variables */
        cmd.data[0] = &phy_db_cmd;
        cmd.len[0] = sizeof(struct iwl_phy_db_cmd);
        cmd.data[1] = data;
        cmd.len[1] = length;
        cmd.dataflags[1] = IWL_HCMD_DFL_NOCOPY;

        return iwl_trans_send_cmd(phy_db->trans, &cmd);
}

static int iwl_phy_db_send_all_channel_groups(
                                        struct iwl_phy_db *phy_db,
                                        enum iwl_phy_db_section_type type,
                                        u8 max_ch_groups)
{
        u16 i;
        int err;
        struct iwl_phy_db_entry *entry;

        /* Send all the  channel specific groups to operational fw */
        for (i = 0; i < max_ch_groups; i++) {
                entry = iwl_phy_db_get_section(phy_db,
                                               type,
                                               i);
                if (!entry)
                        return -EINVAL;

                if (!entry->size)
                        continue;

                /* Send the requested PHY DB section */
                err = iwl_send_phy_db_cmd(phy_db,
                                          type,
                                          entry->size,
                                          entry->data);
                if (err) {
                        IWL_ERR(phy_db->trans,
                                "Can't SEND phy_db section %d (%d), err %d\n",
                                type, i, err);
                        return err;
                }

                IWL_DEBUG_INFO(phy_db->trans,
                               "Sent PHY_DB HCMD, type = %d num = %d\n",
                               type, i);
        }

        return 0;
}

int iwl_send_phy_db_data(struct iwl_phy_db *phy_db)
{
        u8 *data = NULL;
        u16 size = 0;
        int err;

        IWL_DEBUG_INFO(phy_db->trans,
                       "Sending phy db data and configuration to runtime image\n");

        /* Send PHY DB CFG section */
        err = iwl_phy_db_get_section_data(phy_db, IWL_PHY_DB_CFG,
                                          &data, &size, 0);
        if (err) {
                IWL_ERR(phy_db->trans, "Cannot get Phy DB cfg section\n");
                return err;
        }

        err = iwl_send_phy_db_cmd(phy_db, IWL_PHY_DB_CFG, size, data);
        if (err) {
                IWL_ERR(phy_db->trans,
                        "Cannot send HCMD of  Phy DB cfg section\n");
                return err;
        }

        err = iwl_phy_db_get_section_data(phy_db, IWL_PHY_DB_CALIB_NCH,
                                          &data, &size, 0);
        if (err) {
                IWL_ERR(phy_db->trans,
                        "Cannot get Phy DB non specific channel section\n");
                return err;
        }

        err = iwl_send_phy_db_cmd(phy_db, IWL_PHY_DB_CALIB_NCH, size, data);
        if (err) {
                IWL_ERR(phy_db->trans,
                        "Cannot send HCMD of Phy DB non specific channel section\n");
                return err;
        }

        /* Send all the TXP channel specific data */
        err = iwl_phy_db_send_all_channel_groups(phy_db,
                                                 IWL_PHY_DB_CALIB_CHG_PAPD,
                                                 phy_db->n_group_papd);
        if (err) {
                IWL_ERR(phy_db->trans,
                        "Cannot send channel specific PAPD groups\n");
                return err;
        }

        /* Send all the TXP channel specific data */
        err = iwl_phy_db_send_all_channel_groups(phy_db,
                                                 IWL_PHY_DB_CALIB_CHG_TXP,
                                                 phy_db->n_group_txp);
        if (err) {
                IWL_ERR(phy_db->trans,
                        "Cannot send channel specific TX power groups\n");
                return err;
        }

        IWL_DEBUG_INFO(phy_db->trans,
                       "Finished sending phy db non channel data\n");
        return 0;
}
IWL_EXPORT_SYMBOL(iwl_send_phy_db_data);