// SPDX-License-Identifier: GPL-2.0-only
/*
 * NXP Wireless LAN device driver: Channel, Frequence and Power
 *
 * Copyright 2011-2020 NXP
 */

#include "decl.h"
#include "ioctl.h"
#include "util.h"
#include "fw.h"
#include "main.h"
#include "cfg80211.h"

/* 100mW */
#define MWIFIEX_TX_PWR_DEFAULT     20
/* 100mW */
#define MWIFIEX_TX_PWR_US_DEFAULT      20
/* 50mW */
#define MWIFIEX_TX_PWR_JP_DEFAULT      16
/* 100mW */
#define MWIFIEX_TX_PWR_FR_100MW        20
/* 10mW */
#define MWIFIEX_TX_PWR_FR_10MW         10
/* 100mW */
#define MWIFIEX_TX_PWR_EMEA_DEFAULT    20

static u8 adhoc_rates_b[B_SUPPORTED_RATES] = { 0x82, 0x84, 0x8b, 0x96, 0 };

static u8 adhoc_rates_g[G_SUPPORTED_RATES] = { 0x8c, 0x12, 0x98, 0x24,
            0xb0, 0x48, 0x60, 0x6c, 0 };

static u8 adhoc_rates_bg[BG_SUPPORTED_RATES] = { 0x82, 0x84, 0x8b, 0x96,
       0x0c, 0x12, 0x18, 0x24,
       0x30, 0x48, 0x60, 0x6c, 0 };

static u8 adhoc_rates_a[A_SUPPORTED_RATES] = { 0x8c, 0x12, 0x98, 0x24,
            0xb0, 0x48, 0x60, 0x6c, 0 };
static u8 supported_rates_a[A_SUPPORTED_RATES] = { 0x0c, 0x12, 0x18, 0x24,
     0xb0, 0x48, 0x60, 0x6c, 0 };
static u16 mwifiex_data_rates[MWIFIEX_SUPPORTED_RATES_EXT] = { 0x02, 0x04,
     0x0B, 0x16, 0x00, 0x0C, 0x12, 0x18,
     0x24, 0x30, 0x48, 0x60, 0x6C, 0x90,
     0x0D, 0x1A, 0x27, 0x34, 0x4E, 0x68,
     0x75, 0x82, 0x0C, 0x1B, 0x36, 0x51,
     0x6C, 0xA2, 0xD8, 0xF3, 0x10E, 0x00 };

static u8 supported_rates_b[B_SUPPORTED_RATES] = { 0x02, 0x04, 0x0b, 0x16, 0 };

static u8 supported_rates_g[G_SUPPORTED_RATES] = { 0x0c, 0x12, 0x18, 0x24,
     0x30, 0x48, 0x60, 0x6c, 0 };

static u8 supported_rates_bg[BG_SUPPORTED_RATES] = { 0x02, 0x04, 0x0b, 0x0c,
     0x12, 0x16, 0x18, 0x24, 0x30, 0x48,
     0x60, 0x6c, 0 };

u16 region_code_index[MWIFIEX_MAX_REGION_CODE] = { 0x00, 0x10, 0x20, 0x30,
      0x31, 0x32, 0x40, 0x41, 0x50 };

static u8 supported_rates_n[N_SUPPORTED_RATES] = { 0x02, 0x04, 0 };

/* For every mcs_rate line, the first 8 bytes are for stream 1x1,
 * and all 16 bytes are for stream 2x2.
 */
static const u16 mcs_rate[4][16] = {
 /* LGI 40M */
 { 0x1b, 0x36, 0x51, 0x6c, 0xa2, 0xd8, 0xf3, 0x10e,
   0x36, 0x6c, 0xa2, 0xd8, 0x144, 0x1b0, 0x1e6, 0x21c },

 /* SGI 40M */
 { 0x1e, 0x3c, 0x5a, 0x78, 0xb4, 0xf0, 0x10e, 0x12c,
   0x3c, 0x78, 0xb4, 0xf0, 0x168, 0x1e0, 0x21c, 0x258 },

 /* LGI 20M */
 { 0x0d, 0x1a, 0x27, 0x34, 0x4e, 0x68, 0x75, 0x82,
   0x1a, 0x34, 0x4e, 0x68, 0x9c, 0xd0, 0xea, 0x104 },

 /* SGI 20M */
 { 0x0e, 0x1c, 0x2b, 0x39, 0x56, 0x73, 0x82, 0x90,
   0x1c, 0x39, 0x56, 0x73, 0xad, 0xe7, 0x104, 0x120 }
};

/* AC rates */
static const u16 ac_mcs_rate_nss1[8][10] = {
 /* LG 160M */
 { 0x75, 0xEA, 0x15F, 0x1D4, 0x2BE, 0x3A8, 0x41D,
   0x492, 0x57C, 0x618 },

 /* SG 160M */
 { 0x82, 0x104, 0x186, 0x208, 0x30C, 0x410, 0x492,
   0x514, 0x618, 0x6C6 },

 /* LG 80M */
 { 0x3B, 0x75, 0xB0, 0xEA, 0x15F, 0x1D4, 0x20F,
   0x249, 0x2BE, 0x30C },

 /* SG 80M */
 { 0x41, 0x82, 0xC3, 0x104, 0x186, 0x208, 0x249,
   0x28A, 0x30C, 0x363 },

 /* LG 40M */
 { 0x1B, 0x36, 0x51, 0x6C, 0xA2, 0xD8, 0xF3,
   0x10E, 0x144, 0x168 },

 /* SG 40M */
 { 0x1E, 0x3C, 0x5A, 0x78, 0xB4, 0xF0, 0x10E,
   0x12C, 0x168, 0x190 },

 /* LG 20M */
 { 0xD, 0x1A, 0x27, 0x34, 0x4E, 0x68, 0x75, 0x82, 0x9C, 0x00 },

 /* SG 20M */
 { 0xF, 0x1D, 0x2C, 0x3A, 0x57, 0x74, 0x82, 0x91, 0xAE, 0x00 },
};

/* NSS2 note: the value in the table is 2 multiplier of the actual rate */
static const u16 ac_mcs_rate_nss2[8][10] = {
 /* LG 160M */
 { 0xEA, 0x1D4, 0x2BE, 0x3A8, 0x57C, 0x750, 0x83A,
   0x924, 0xAF8, 0xC30 },

 /* SG 160M */
 { 0x104, 0x208, 0x30C, 0x410, 0x618, 0x820, 0x924,
   0xA28, 0xC30, 0xD8B },

 /* LG 80M */
 { 0x75, 0xEA, 0x15F, 0x1D4, 0x2BE, 0x3A8, 0x41D,
   0x492, 0x57C, 0x618 },

 /* SG 80M */
 { 0x82, 0x104, 0x186, 0x208, 0x30C, 0x410, 0x492,
   0x514, 0x618, 0x6C6 },

 /* LG 40M */
 { 0x36, 0x6C, 0xA2, 0xD8, 0x144, 0x1B0, 0x1E6,
   0x21C, 0x288, 0x2D0 },

 /* SG 40M */
 { 0x3C, 0x78, 0xB4, 0xF0, 0x168, 0x1E0, 0x21C,
   0x258, 0x2D0, 0x320 },

 /* LG 20M */
 { 0x1A, 0x34, 0x4A, 0x68, 0x9C, 0xD0, 0xEA, 0x104,
   0x138, 0x00 },

 /* SG 20M */
 { 0x1D, 0x3A, 0x57, 0x74, 0xAE, 0xE6, 0x104, 0x121,
   0x15B, 0x00 },
};

struct region_code_mapping {
 u8 code;
 u8 region[IEEE80211_COUNTRY_STRING_LEN] __nonstring;
};

static const struct region_code_mapping region_code_mapping_t[] = {
 { 0x10, "US " }, /* US FCC */
 { 0x20, "CA " }, /* IC Canada */
 { 0x30, "FR " }, /* France */
 { 0x31, "ES " }, /* Spain */
 { 0x32, "FR " }, /* France */
 { 0x40, "JP " }, /* Japan */
 { 0x41, "JP " }, /* Japan */
 { 0x50, "CN " }, /* China */
};

/* This function converts integer code to region string */
const u8 *mwifiex_11d_code_2_region(u8 code)
{
 u8 i;

 /* Look for code in mapping table */
 for (i = 0; i < ARRAY_SIZE(region_code_mapping_t); i++)
  if (region_code_mapping_t[i].code == code)
   return region_code_mapping_t[i].region;

 return NULL;
}

/*
 * This function maps an index in supported rates table into
 * the corresponding data rate.
 */
u32 mwifiex_index_to_acs_data_rate(struct mwifiex_private *priv,
       u8 index, u8 ht_info)
{
 u32 rate = 0;
 u8 mcs_index = 0;
 u8 bw = 0;
 u8 gi = 0;

 if ((ht_info & 0x3) == MWIFIEX_RATE_FORMAT_VHT) {
  mcs_index = min(index & 0xF, 9);

  /* 20M: bw=0, 40M: bw=1, 80M: bw=2, 160M: bw=3 */
  bw = (ht_info & 0xC) >> 2;

  /* LGI: gi =0, SGI: gi = 1 */
  gi = (ht_info & 0x10) >> 4;

  if ((index >> 4) == 1) /* NSS = 2 */
   rate = ac_mcs_rate_nss2[2 * (3 - bw) + gi][mcs_index];
  else   /* NSS = 1 */
   rate = ac_mcs_rate_nss1[2 * (3 - bw) + gi][mcs_index];
 } else if ((ht_info & 0x3) == MWIFIEX_RATE_FORMAT_HT) {
  /* 20M: bw=0, 40M: bw=1 */
  bw = (ht_info & 0xC) >> 2;

  /* LGI: gi =0, SGI: gi = 1 */
  gi = (ht_info & 0x10) >> 4;

  if (index == MWIFIEX_RATE_BITMAP_MCS0) {
   if (gi == 1)
    rate = 0x0D;    /* MCS 32 SGI rate */
   else
    rate = 0x0C;    /* MCS 32 LGI rate */
  } else if (index < 16) {
   if ((bw == 1) || (bw == 0))
    rate = mcs_rate[2 * (1 - bw) + gi][index];
   else
    rate = mwifiex_data_rates[0];
  } else {
   rate = mwifiex_data_rates[0];
  }
 } else {
  /* 11n non-HT rates */
  if (index >= MWIFIEX_SUPPORTED_RATES_EXT)
   index = 0;
  rate = mwifiex_data_rates[index];
 }

 return rate;
}

/* This function maps an index in supported rates table into
 * the corresponding data rate.
 */
u32 mwifiex_index_to_data_rate(struct mwifiex_private *priv,
          u8 index, u8 ht_info)
{
 u32 mcs_num_supp =
  (priv->adapter->user_dev_mcs_support == HT_STREAM_2X2) ? 16 : 8;
 u32 rate;

 if (priv->adapter->is_hw_11ac_capable)
  return mwifiex_index_to_acs_data_rate(priv, index, ht_info);

 if (ht_info & BIT(0)) {
  if (index == MWIFIEX_RATE_BITMAP_MCS0) {
   if (ht_info & BIT(2))
    rate = 0x0D; /* MCS 32 SGI rate */
   else
    rate = 0x0C; /* MCS 32 LGI rate */
  } else if (index < mcs_num_supp) {
   if (ht_info & BIT(1)) {
    if (ht_info & BIT(2))
     /* SGI, 40M */
     rate = mcs_rate[1][index];
    else
     /* LGI, 40M */
     rate = mcs_rate[0][index];
   } else {
    if (ht_info & BIT(2))
     /* SGI, 20M */
     rate = mcs_rate[3][index];
    else
     /* LGI, 20M */
     rate = mcs_rate[2][index];
   }
  } else
   rate = mwifiex_data_rates[0];
 } else {
  if (index >= MWIFIEX_SUPPORTED_RATES_EXT)
   index = 0;
  rate = mwifiex_data_rates[index];
 }
 return rate;
}

/*
 * This function returns the current active data rates.
 *
 * The result may vary depending upon connection status.
 */
u32 mwifiex_get_active_data_rates(struct mwifiex_private *priv, u8 *rates)
{
 if (!priv->media_connected)
  return mwifiex_get_supported_rates(priv, rates);
 else
  return mwifiex_copy_rates(rates, 0,
       priv->curr_bss_params.data_rates,
       priv->curr_bss_params.num_of_rates);
}

/*
 * This function locates the Channel-Frequency-Power triplet based upon
 * band and channel/frequency parameters.
 */
struct mwifiex_chan_freq_power *
mwifiex_get_cfp(struct mwifiex_private *priv, u8 band, u16 channel, u32 freq)
{
 struct mwifiex_chan_freq_power *cfp = NULL;
 struct ieee80211_supported_band *sband;
 struct ieee80211_channel *ch = NULL;
 int i;

 if (!channel && !freq)
  return cfp;

 if (mwifiex_band_to_radio_type(band) == HostCmd_SCAN_RADIO_TYPE_BG)
  sband = priv->wdev.wiphy->bands[NL80211_BAND_2GHZ];
 else
  sband = priv->wdev.wiphy->bands[NL80211_BAND_5GHZ];

 if (!sband) {
  mwifiex_dbg(priv->adapter, ERROR,
       "%s: cannot find cfp by band %d\n",
       __func__, band);
  return cfp;
 }

 for (i = 0; i < sband->n_channels; i++) {
  ch = &sband->channels[i];

  if (ch->flags & IEEE80211_CHAN_DISABLED)
   continue;

  if (freq) {
   if (ch->center_freq == freq)
    break;
  } else {
   /* find by valid channel*/
   if (ch->hw_value == channel ||
       channel == FIRST_VALID_CHANNEL)
    break;
  }
 }
 if (i == sband->n_channels) {
  mwifiex_dbg(priv->adapter, WARN,
       "%s: cannot find cfp by band %d\t"
       "& channel=%d freq=%d\n",
       __func__, band, channel, freq);
 } else {
  if (!ch)
   return cfp;

  priv->cfp.channel = ch->hw_value;
  priv->cfp.freq = ch->center_freq;
  priv->cfp.max_tx_power = ch->max_power;
  cfp = &priv->cfp;
 }

 return cfp;
}

/*
 * This function checks if the data rate is set to auto.
 */
u8
mwifiex_is_rate_auto(struct mwifiex_private *priv)
{
 u32 i;
 int rate_num = 0;

 for (i = 0; i < ARRAY_SIZE(priv->bitmap_rates); i++)
  if (priv->bitmap_rates[i])
   rate_num++;

 if (rate_num > 1)
  return true;
 else
  return false;
}

/* This function gets the supported data rates from bitmask inside
 * cfg80211_scan_request.
 */
u32 mwifiex_get_rates_from_cfg80211(struct mwifiex_private *priv,
        u8 *rates, u8 radio_type)
{
 struct wiphy *wiphy = priv->adapter->wiphy;
 struct cfg80211_scan_request *request = priv->scan_request;
 u32 num_rates, rate_mask;
 struct ieee80211_supported_band *sband;
 int i;

 if (radio_type) {
  sband = wiphy->bands[NL80211_BAND_5GHZ];
  if (WARN_ON_ONCE(!sband))
   return 0;
  rate_mask = request->rates[NL80211_BAND_5GHZ];
 } else {
  sband = wiphy->bands[NL80211_BAND_2GHZ];
  if (WARN_ON_ONCE(!sband))
   return 0;
  rate_mask = request->rates[NL80211_BAND_2GHZ];
 }

 num_rates = 0;
 for (i = 0; i < sband->n_bitrates; i++) {
  if ((BIT(i) & rate_mask) == 0)
   continue; /* skip rate */
  rates[num_rates++] = (u8)(sband->bitrates[i].bitrate / 5);
 }

 return num_rates;
}

/* This function gets the supported data rates. The function works in
 * both Ad-Hoc and infra mode by printing the band and returning the
 * data rates.
 */
u32 mwifiex_get_supported_rates(struct mwifiex_private *priv, u8 *rates)
{
 u32 k = 0;
 struct mwifiex_adapter *adapter = priv->adapter;

 if (priv->bss_mode == NL80211_IFTYPE_STATION ||
     priv->bss_mode == NL80211_IFTYPE_P2P_CLIENT) {
  switch (adapter->config_bands) {
  case BAND_B:
   mwifiex_dbg(adapter, INFO, "info: infra band=%d\t"
        "supported_rates_b\n",
        adapter->config_bands);
   k = mwifiex_copy_rates(rates, k, supported_rates_b,
            sizeof(supported_rates_b));
   break;
  case BAND_G:
  case BAND_G | BAND_GN:
   mwifiex_dbg(adapter, INFO, "info: infra band=%d\t"
        "supported_rates_g\n",
        adapter->config_bands);
   k = mwifiex_copy_rates(rates, k, supported_rates_g,
            sizeof(supported_rates_g));
   break;
  case BAND_B | BAND_G:
  case BAND_A | BAND_B | BAND_G:
  case BAND_A | BAND_B:
  case BAND_A | BAND_B | BAND_G | BAND_GN | BAND_AN:
  case BAND_A | BAND_B | BAND_G | BAND_GN | BAND_AN | BAND_AAC:
  case BAND_B | BAND_G | BAND_GN:
   mwifiex_dbg(adapter, INFO, "info: infra band=%d\t"
        "supported_rates_bg\n",
        adapter->config_bands);
   k = mwifiex_copy_rates(rates, k, supported_rates_bg,
            sizeof(supported_rates_bg));
   break;
  case BAND_A:
  case BAND_A | BAND_G:
   mwifiex_dbg(adapter, INFO, "info: infra band=%d\t"
        "supported_rates_a\n",
        adapter->config_bands);
   k = mwifiex_copy_rates(rates, k, supported_rates_a,
            sizeof(supported_rates_a));
   break;
  case BAND_AN:
  case BAND_A | BAND_AN:
  case BAND_A | BAND_AN | BAND_AAC:
  case BAND_A | BAND_G | BAND_AN | BAND_GN:
  case BAND_A | BAND_G | BAND_AN | BAND_GN | BAND_AAC:
   mwifiex_dbg(adapter, INFO, "info: infra band=%d\t"
        "supported_rates_a\n",
        adapter->config_bands);
   k = mwifiex_copy_rates(rates, k, supported_rates_a,
            sizeof(supported_rates_a));
   break;
  case BAND_GN:
   mwifiex_dbg(adapter, INFO, "info: infra band=%d\t"
        "supported_rates_n\n",
        adapter->config_bands);
   k = mwifiex_copy_rates(rates, k, supported_rates_n,
            sizeof(supported_rates_n));
   break;
  }
 } else {
  /* Ad-hoc mode */
  switch (adapter->adhoc_start_band) {
  case BAND_B:
   mwifiex_dbg(adapter, INFO, "info: adhoc B\n");
   k = mwifiex_copy_rates(rates, k, adhoc_rates_b,
            sizeof(adhoc_rates_b));
   break;
  case BAND_G:
  case BAND_G | BAND_GN:
   mwifiex_dbg(adapter, INFO, "info: adhoc G only\n");
   k = mwifiex_copy_rates(rates, k, adhoc_rates_g,
            sizeof(adhoc_rates_g));
   break;
  case BAND_B | BAND_G:
  case BAND_B | BAND_G | BAND_GN:
   mwifiex_dbg(adapter, INFO, "info: adhoc BG\n");
   k = mwifiex_copy_rates(rates, k, adhoc_rates_bg,
            sizeof(adhoc_rates_bg));
   break;
  case BAND_A:
  case BAND_A | BAND_AN:
   mwifiex_dbg(adapter, INFO, "info: adhoc A\n");
   k = mwifiex_copy_rates(rates, k, adhoc_rates_a,
            sizeof(adhoc_rates_a));
   break;
  }
 }

 return k;
}

u8 mwifiex_adjust_data_rate(struct mwifiex_private *priv,
       u8 rx_rate, u8 rate_info)
{
 u8 rate_index = 0;

 /* HT40 */
 if ((rate_info & BIT(0)) && (rate_info & BIT(1)))
  rate_index = MWIFIEX_RATE_INDEX_MCS0 +
        MWIFIEX_BW20_MCS_NUM + rx_rate;
 else if (rate_info & BIT(0)) /* HT20 */
  rate_index = MWIFIEX_RATE_INDEX_MCS0 + rx_rate;
 else
  rate_index = (rx_rate > MWIFIEX_RATE_INDEX_OFDM0) ?
         rx_rate - 1 : rx_rate;

 if (rate_index >= MWIFIEX_MAX_AC_RX_RATES)
  rate_index = MWIFIEX_MAX_AC_RX_RATES - 1;

 return rate_index;
}