// SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause
/*
 * Copyright (C) 2005-2014, 2018-2021, 2023, 2025 Intel Corporation
 * Copyright (C) 2015 Intel Mobile Communications GmbH
 */
#include <linux/types.h>
#include <linux/slab.h>
#include <linux/export.h>
#include "iwl-drv.h"
#include "iwl-modparams.h"
#include "iwl-nvm-utils.h"

int iwl_init_sband_channels(struct iwl_nvm_data *data,
       struct ieee80211_supported_band *sband,
       int n_channels, enum nl80211_band band)
{
 struct ieee80211_channel *chan = &data->channels[0];
 int n = 0, idx = 0;

 while (idx < n_channels && chan->band != band)
  chan = &data->channels[++idx];

 sband->channels = &data->channels[idx];

 while (idx < n_channels && chan->band == band) {
  chan = &data->channels[++idx];
  n++;
 }

 sband->n_channels = n;

 return n;
}
IWL_EXPORT_SYMBOL(iwl_init_sband_channels);

#define MAX_BIT_RATE_40_MHZ 150 /* Mbps */
#define MAX_BIT_RATE_20_MHZ 72 /* Mbps */

void iwl_init_ht_hw_capab(struct iwl_trans *trans,
     struct iwl_nvm_data *data,
     struct ieee80211_sta_ht_cap *ht_info,
     enum nl80211_band band,
     u8 tx_chains, u8 rx_chains)
{
 const struct iwl_rf_cfg *cfg = trans->cfg;
 int max_bit_rate = 0;

 tx_chains = hweight8(tx_chains);
 if (cfg->rx_with_siso_diversity)
  rx_chains = 1;
 else
  rx_chains = hweight8(rx_chains);

 if (!(data->sku_cap_11n_enable) ||
     (iwlwifi_mod_params.disable_11n & IWL_DISABLE_HT_ALL) ||
     /* there are no devices with HT but without HT40 entirely */
     !cfg->ht_params.ht40_bands) {
  ht_info->ht_supported = false;
  return;
 }

 if (data->sku_cap_mimo_disabled)
  rx_chains = 1;

 ht_info->ht_supported = true;
 ht_info->cap = IEEE80211_HT_CAP_DSSSCCK40;

 if (cfg->ht_params.stbc) {
  ht_info->cap |= (1 << IEEE80211_HT_CAP_RX_STBC_SHIFT);

  if (tx_chains > 1)
   ht_info->cap |= IEEE80211_HT_CAP_TX_STBC;
 }

 if (cfg->ht_params.ldpc)
  ht_info->cap |= IEEE80211_HT_CAP_LDPC_CODING;

 if (trans->mac_cfg->mq_rx_supported ||
     iwlwifi_mod_params.amsdu_size >= IWL_AMSDU_8K)
  ht_info->cap |= IEEE80211_HT_CAP_MAX_AMSDU;

 ht_info->ampdu_factor = IEEE80211_HT_MAX_AMPDU_64K;
 ht_info->ampdu_density = IEEE80211_HT_MPDU_DENSITY_4;

 ht_info->mcs.rx_mask[0] = 0xFF;
 ht_info->mcs.rx_mask[1] = 0x00;
 ht_info->mcs.rx_mask[2] = 0x00;

 if (rx_chains >= 2)
  ht_info->mcs.rx_mask[1] = 0xFF;
 if (rx_chains >= 3)
  ht_info->mcs.rx_mask[2] = 0xFF;

 if (cfg->ht_params.ht_greenfield_support)
  ht_info->cap |= IEEE80211_HT_CAP_GRN_FLD;
 ht_info->cap |= IEEE80211_HT_CAP_SGI_20;

 max_bit_rate = MAX_BIT_RATE_20_MHZ;

 if (cfg->ht_params.ht40_bands & BIT(band)) {
  ht_info->cap |= IEEE80211_HT_CAP_SUP_WIDTH_20_40;
  ht_info->cap |= IEEE80211_HT_CAP_SGI_40;
  max_bit_rate = MAX_BIT_RATE_40_MHZ;
 }

 /* Highest supported Rx data rate */
 max_bit_rate *= rx_chains;
 WARN_ON(max_bit_rate & ~IEEE80211_HT_MCS_RX_HIGHEST_MASK);
 ht_info->mcs.rx_highest = cpu_to_le16(max_bit_rate);

 /* Tx MCS capabilities */
 ht_info->mcs.tx_params = IEEE80211_HT_MCS_TX_DEFINED;
 if (tx_chains != rx_chains) {
  ht_info->mcs.tx_params |= IEEE80211_HT_MCS_TX_RX_DIFF;
  ht_info->mcs.tx_params |= ((tx_chains - 1) <<
    IEEE80211_HT_MCS_TX_MAX_STREAMS_SHIFT);
 }
}
IWL_EXPORT_SYMBOL(iwl_init_ht_hw_capab);