root/drivers/net/wireless/marvell/mwifiex/uap_cmd.c
// SPDX-License-Identifier: GPL-2.0-only
/*
 * NXP Wireless LAN device driver: AP specific command handling
 *
 * Copyright 2011-2020 NXP
 */

#include "main.h"
#include "11ac.h"
#include "11n.h"

/* This function parses security related parameters from cfg80211_ap_settings
 * and sets into FW understandable bss_config structure.
 */
int mwifiex_set_secure_params(struct mwifiex_private *priv,
                              struct mwifiex_uap_bss_param *bss_config,
                              struct cfg80211_ap_settings *params) {
        int i;
        struct mwifiex_wep_key wep_key;

        if (!params->privacy) {
                bss_config->protocol = PROTOCOL_NO_SECURITY;
                bss_config->key_mgmt = KEY_MGMT_NONE;
                bss_config->wpa_cfg.length = 0;
                priv->sec_info.wep_enabled = 0;
                priv->sec_info.wpa_enabled = 0;
                priv->sec_info.wpa2_enabled = 0;

                return 0;
        }

        switch (params->auth_type) {
        case NL80211_AUTHTYPE_OPEN_SYSTEM:
                bss_config->auth_mode = WLAN_AUTH_OPEN;
                break;
        case NL80211_AUTHTYPE_SHARED_KEY:
                bss_config->auth_mode = WLAN_AUTH_SHARED_KEY;
                break;
        case NL80211_AUTHTYPE_NETWORK_EAP:
                bss_config->auth_mode = WLAN_AUTH_LEAP;
                break;
        default:
                bss_config->auth_mode = MWIFIEX_AUTH_MODE_AUTO;
                break;
        }

        bss_config->key_mgmt_operation |= KEY_MGMT_ON_HOST;

        bss_config->protocol = 0;
        if (params->crypto.wpa_versions & NL80211_WPA_VERSION_1)
                bss_config->protocol |= PROTOCOL_WPA;
        if (params->crypto.wpa_versions & NL80211_WPA_VERSION_2)
                bss_config->protocol |= PROTOCOL_WPA2;

        bss_config->key_mgmt = 0;
        for (i = 0; i < params->crypto.n_akm_suites; i++) {
                switch (params->crypto.akm_suites[i]) {
                case WLAN_AKM_SUITE_8021X:
                        bss_config->key_mgmt |= KEY_MGMT_EAP;
                        break;
                case WLAN_AKM_SUITE_PSK:
                        bss_config->key_mgmt |= KEY_MGMT_PSK;
                        break;
                case WLAN_AKM_SUITE_PSK_SHA256:
                        bss_config->key_mgmt |= KEY_MGMT_PSK_SHA256;
                        break;
                case WLAN_AKM_SUITE_SAE:
                        bss_config->key_mgmt |= KEY_MGMT_SAE;
                        break;
                default:
                        break;
                }
        }
        for (i = 0; i < params->crypto.n_ciphers_pairwise; i++) {
                switch (params->crypto.ciphers_pairwise[i]) {
                case WLAN_CIPHER_SUITE_WEP40:
                case WLAN_CIPHER_SUITE_WEP104:
                        break;
                case WLAN_CIPHER_SUITE_TKIP:
                        if (params->crypto.wpa_versions & NL80211_WPA_VERSION_1)
                                bss_config->wpa_cfg.pairwise_cipher_wpa |=
                                                                CIPHER_TKIP;
                        if (params->crypto.wpa_versions & NL80211_WPA_VERSION_2)
                                bss_config->wpa_cfg.pairwise_cipher_wpa2 |=
                                                                CIPHER_TKIP;
                        break;
                case WLAN_CIPHER_SUITE_CCMP:
                        if (params->crypto.wpa_versions & NL80211_WPA_VERSION_1)
                                bss_config->wpa_cfg.pairwise_cipher_wpa |=
                                                                CIPHER_AES_CCMP;
                        if (params->crypto.wpa_versions & NL80211_WPA_VERSION_2)
                                bss_config->wpa_cfg.pairwise_cipher_wpa2 |=
                                                                CIPHER_AES_CCMP;
                        break;
                default:
                        break;
                }
        }

        switch (params->crypto.cipher_group) {
        case WLAN_CIPHER_SUITE_WEP40:
        case WLAN_CIPHER_SUITE_WEP104:
                if (priv->sec_info.wep_enabled) {
                        bss_config->protocol = PROTOCOL_STATIC_WEP;
                        bss_config->key_mgmt = KEY_MGMT_NONE;
                        bss_config->wpa_cfg.length = 0;

                        for (i = 0; i < NUM_WEP_KEYS; i++) {
                                wep_key = priv->wep_key[i];
                                bss_config->wep_cfg[i].key_index = i;

                                if (priv->wep_key_curr_index == i)
                                        bss_config->wep_cfg[i].is_default = 1;
                                else
                                        bss_config->wep_cfg[i].is_default = 0;

                                bss_config->wep_cfg[i].length =
                                                             wep_key.key_length;
                                memcpy(&bss_config->wep_cfg[i].key,
                                       &wep_key.key_material,
                                       wep_key.key_length);
                        }
                }
                break;
        case WLAN_CIPHER_SUITE_TKIP:
                bss_config->wpa_cfg.group_cipher = CIPHER_TKIP;
                break;
        case WLAN_CIPHER_SUITE_CCMP:
                bss_config->wpa_cfg.group_cipher = CIPHER_AES_CCMP;
                break;
        default:
                break;
        }

        return 0;
}

/* This function updates 11n related parameters from IE and sets them into
 * bss_config structure.
 */
void
mwifiex_set_ht_params(struct mwifiex_private *priv,
                      struct mwifiex_uap_bss_param *bss_cfg,
                      struct cfg80211_ap_settings *params)
{
        const u8 *ht_ie;

        if (!ISSUPP_11NENABLED(priv->adapter->fw_cap_info))
                return;

        ht_ie = cfg80211_find_ie(WLAN_EID_HT_CAPABILITY, params->beacon.tail,
                                 params->beacon.tail_len);
        if (ht_ie) {
                memcpy(&bss_cfg->ht_cap, ht_ie + 2,
                       sizeof(struct ieee80211_ht_cap));
                priv->ap_11n_enabled = 1;
        } else {
                memset(&bss_cfg->ht_cap, 0, sizeof(struct ieee80211_ht_cap));
                bss_cfg->ht_cap.cap_info = cpu_to_le16(MWIFIEX_DEF_HT_CAP);
                bss_cfg->ht_cap.ampdu_params_info = MWIFIEX_DEF_AMPDU;
        }

        return;
}

/* This function updates 11ac related parameters from IE
 * and sets them into bss_config structure.
 */
void mwifiex_set_vht_params(struct mwifiex_private *priv,
                            struct mwifiex_uap_bss_param *bss_cfg,
                            struct cfg80211_ap_settings *params)
{
        const u8 *vht_ie;

        vht_ie = cfg80211_find_ie(WLAN_EID_VHT_CAPABILITY, params->beacon.tail,
                                  params->beacon.tail_len);
        if (vht_ie) {
                memcpy(&bss_cfg->vht_cap, vht_ie + 2,
                       sizeof(struct ieee80211_vht_cap));
                priv->ap_11ac_enabled = 1;
        } else {
                priv->ap_11ac_enabled = 0;
        }

        return;
}

/* This function updates 11ac related parameters from IE
 * and sets them into bss_config structure.
 */
void mwifiex_set_tpc_params(struct mwifiex_private *priv,
                            struct mwifiex_uap_bss_param *bss_cfg,
                            struct cfg80211_ap_settings *params)
{
        const u8 *tpc_ie;

        tpc_ie = cfg80211_find_ie(WLAN_EID_TPC_REQUEST, params->beacon.tail,
                                  params->beacon.tail_len);
        if (tpc_ie)
                bss_cfg->power_constraint = *(tpc_ie + 2);
        else
                bss_cfg->power_constraint = 0;
}

/* Enable VHT only when cfg80211_ap_settings has VHT IE.
 * Otherwise disable VHT.
 */
void mwifiex_set_vht_width(struct mwifiex_private *priv,
                           enum nl80211_chan_width width,
                           bool ap_11ac_enable)
{
        struct mwifiex_adapter *adapter = priv->adapter;
        struct mwifiex_11ac_vht_cfg vht_cfg;

        vht_cfg.band_config = VHT_CFG_5GHZ;
        vht_cfg.cap_info = adapter->hw_dot_11ac_dev_cap;

        if (!ap_11ac_enable) {
                vht_cfg.mcs_tx_set = DISABLE_VHT_MCS_SET;
                vht_cfg.mcs_rx_set = DISABLE_VHT_MCS_SET;
        } else {
                vht_cfg.mcs_tx_set = DEFAULT_VHT_MCS_SET;
                vht_cfg.mcs_rx_set = DEFAULT_VHT_MCS_SET;
        }

        vht_cfg.misc_config  = VHT_CAP_UAP_ONLY;

        if (ap_11ac_enable && width >= NL80211_CHAN_WIDTH_80)
                vht_cfg.misc_config |= VHT_BW_80_160_80P80;

        mwifiex_send_cmd(priv, HostCmd_CMD_11AC_CFG,
                         HostCmd_ACT_GEN_SET, 0, &vht_cfg, true);

        return;
}

/* This function finds supported rates IE from beacon parameter and sets
 * these rates into bss_config structure.
 */
void
mwifiex_set_uap_rates(struct mwifiex_uap_bss_param *bss_cfg,
                      struct cfg80211_ap_settings *params)
{
        struct ieee_types_header *rate_ie;
        int var_offset = offsetof(struct ieee80211_mgmt, u.beacon.variable);
        const u8 *var_pos = params->beacon.head + var_offset;
        int len = params->beacon.head_len - var_offset;
        u8 rate_len = 0;

        rate_ie = (void *)cfg80211_find_ie(WLAN_EID_SUPP_RATES, var_pos, len);
        if (rate_ie) {
                if (rate_ie->len > MWIFIEX_SUPPORTED_RATES)
                        return;
                memcpy(bss_cfg->rates, rate_ie + 1, rate_ie->len);
                rate_len = rate_ie->len;
        }

        rate_ie = (void *)cfg80211_find_ie(WLAN_EID_EXT_SUPP_RATES,
                                           params->beacon.tail,
                                           params->beacon.tail_len);
        if (rate_ie) {
                if (rate_ie->len > MWIFIEX_SUPPORTED_RATES - rate_len)
                        return;
                memcpy(bss_cfg->rates + rate_len, rate_ie + 1, rate_ie->len);
        }

        return;
}

/* This function initializes some of mwifiex_uap_bss_param variables.
 * This helps FW in ignoring invalid values. These values may or may not
 * be get updated to valid ones at later stage.
 */
void mwifiex_set_sys_config_invalid_data(struct mwifiex_uap_bss_param *config)
{
        config->bcast_ssid_ctl = 0x7F;
        config->radio_ctl = 0x7F;
        config->dtim_period = 0x7F;
        config->beacon_period = 0x7FFF;
        config->auth_mode = 0x7F;
        config->rts_threshold = 0x7FFF;
        config->frag_threshold = 0x7FFF;
        config->retry_limit = 0x7F;
        config->qos_info = 0xFF;
}

/* This function parses BSS related parameters from structure
 * and prepares TLVs specific to WPA/WPA2 security.
 * These TLVs are appended to command buffer.
 */
static void
mwifiex_uap_bss_wpa(u8 **tlv_buf, void *cmd_buf, u16 *param_size)
{
        struct host_cmd_tlv_pwk_cipher *pwk_cipher;
        struct host_cmd_tlv_gwk_cipher *gwk_cipher;
        struct host_cmd_tlv_passphrase *passphrase;
        struct host_cmd_tlv_akmp *tlv_akmp;
        struct mwifiex_uap_bss_param *bss_cfg = cmd_buf;
        u16 cmd_size = *param_size;
        u8 *tlv = *tlv_buf;

        tlv_akmp = (struct host_cmd_tlv_akmp *)tlv;
        tlv_akmp->header.type = cpu_to_le16(TLV_TYPE_UAP_AKMP);
        tlv_akmp->header.len = cpu_to_le16(sizeof(struct host_cmd_tlv_akmp) -
                                        sizeof(struct mwifiex_ie_types_header));
        tlv_akmp->key_mgmt_operation = cpu_to_le16(bss_cfg->key_mgmt_operation);
        tlv_akmp->key_mgmt = cpu_to_le16(bss_cfg->key_mgmt);
        cmd_size += sizeof(struct host_cmd_tlv_akmp);
        tlv += sizeof(struct host_cmd_tlv_akmp);

        if (bss_cfg->wpa_cfg.pairwise_cipher_wpa & VALID_CIPHER_BITMAP) {
                pwk_cipher = (struct host_cmd_tlv_pwk_cipher *)tlv;
                pwk_cipher->header.type = cpu_to_le16(TLV_TYPE_PWK_CIPHER);
                pwk_cipher->header.len =
                        cpu_to_le16(sizeof(struct host_cmd_tlv_pwk_cipher) -
                                    sizeof(struct mwifiex_ie_types_header));
                pwk_cipher->proto = cpu_to_le16(PROTOCOL_WPA);
                pwk_cipher->cipher = bss_cfg->wpa_cfg.pairwise_cipher_wpa;
                cmd_size += sizeof(struct host_cmd_tlv_pwk_cipher);
                tlv += sizeof(struct host_cmd_tlv_pwk_cipher);
        }

        if (bss_cfg->wpa_cfg.pairwise_cipher_wpa2 & VALID_CIPHER_BITMAP) {
                pwk_cipher = (struct host_cmd_tlv_pwk_cipher *)tlv;
                pwk_cipher->header.type = cpu_to_le16(TLV_TYPE_PWK_CIPHER);
                pwk_cipher->header.len =
                        cpu_to_le16(sizeof(struct host_cmd_tlv_pwk_cipher) -
                                    sizeof(struct mwifiex_ie_types_header));
                pwk_cipher->proto = cpu_to_le16(PROTOCOL_WPA2);
                pwk_cipher->cipher = bss_cfg->wpa_cfg.pairwise_cipher_wpa2;
                cmd_size += sizeof(struct host_cmd_tlv_pwk_cipher);
                tlv += sizeof(struct host_cmd_tlv_pwk_cipher);
        }

        if (bss_cfg->wpa_cfg.group_cipher & VALID_CIPHER_BITMAP) {
                gwk_cipher = (struct host_cmd_tlv_gwk_cipher *)tlv;
                gwk_cipher->header.type = cpu_to_le16(TLV_TYPE_GWK_CIPHER);
                gwk_cipher->header.len =
                        cpu_to_le16(sizeof(struct host_cmd_tlv_gwk_cipher) -
                                    sizeof(struct mwifiex_ie_types_header));
                gwk_cipher->cipher = bss_cfg->wpa_cfg.group_cipher;
                cmd_size += sizeof(struct host_cmd_tlv_gwk_cipher);
                tlv += sizeof(struct host_cmd_tlv_gwk_cipher);
        }

        if (bss_cfg->wpa_cfg.length) {
                passphrase = (struct host_cmd_tlv_passphrase *)tlv;
                passphrase->header.type =
                                cpu_to_le16(TLV_TYPE_UAP_WPA_PASSPHRASE);
                passphrase->header.len = cpu_to_le16(bss_cfg->wpa_cfg.length);
                memcpy(passphrase->passphrase, bss_cfg->wpa_cfg.passphrase,
                       bss_cfg->wpa_cfg.length);
                cmd_size += sizeof(struct mwifiex_ie_types_header) +
                            bss_cfg->wpa_cfg.length;
                tlv += sizeof(struct mwifiex_ie_types_header) +
                                bss_cfg->wpa_cfg.length;
        }

        *param_size = cmd_size;
        *tlv_buf = tlv;

        return;
}

/* This function parses WMM related parameters from cfg80211_ap_settings
 * structure and updates bss_config structure.
 */
void
mwifiex_set_wmm_params(struct mwifiex_private *priv,
                       struct mwifiex_uap_bss_param *bss_cfg,
                       struct cfg80211_ap_settings *params)
{
        const u8 *vendor_ie;
        const u8 *wmm_ie;
        static const u8 wmm_oui[] = {0x00, 0x50, 0xf2, 0x02};

        vendor_ie = cfg80211_find_vendor_ie(WLAN_OUI_MICROSOFT,
                                            WLAN_OUI_TYPE_MICROSOFT_WMM,
                                            params->beacon.tail,
                                            params->beacon.tail_len);
        if (vendor_ie) {
                wmm_ie = vendor_ie;
                if (*(wmm_ie + 1) > sizeof(struct mwifiex_types_wmm_info))
                        return;
                memcpy(&bss_cfg->wmm_info, wmm_ie +
                       sizeof(struct ieee_types_header), *(wmm_ie + 1));
                priv->wmm_enabled = 1;
        } else {
                memset(&bss_cfg->wmm_info, 0, sizeof(bss_cfg->wmm_info));
                memcpy(&bss_cfg->wmm_info.oui, wmm_oui, sizeof(wmm_oui));
                bss_cfg->wmm_info.subtype = MWIFIEX_WMM_SUBTYPE;
                bss_cfg->wmm_info.version = MWIFIEX_WMM_VERSION;
                priv->wmm_enabled = 0;
        }

        bss_cfg->qos_info = 0x00;
        return;
}
/* This function parses BSS related parameters from structure
 * and prepares TLVs specific to WEP encryption.
 * These TLVs are appended to command buffer.
 */
static void
mwifiex_uap_bss_wep(u8 **tlv_buf, void *cmd_buf, u16 *param_size)
{
        struct host_cmd_tlv_wep_key *wep_key;
        u16 cmd_size = *param_size;
        int i;
        u8 *tlv = *tlv_buf;
        struct mwifiex_uap_bss_param *bss_cfg = cmd_buf;

        for (i = 0; i < NUM_WEP_KEYS; i++) {
                if (bss_cfg->wep_cfg[i].length &&
                    (bss_cfg->wep_cfg[i].length == WLAN_KEY_LEN_WEP40 ||
                     bss_cfg->wep_cfg[i].length == WLAN_KEY_LEN_WEP104)) {
                        wep_key = (struct host_cmd_tlv_wep_key *)tlv;
                        wep_key->header.type =
                                cpu_to_le16(TLV_TYPE_UAP_WEP_KEY);
                        wep_key->header.len =
                                cpu_to_le16(bss_cfg->wep_cfg[i].length + 2);
                        wep_key->key_index = bss_cfg->wep_cfg[i].key_index;
                        wep_key->is_default = bss_cfg->wep_cfg[i].is_default;
                        memcpy(wep_key->key, bss_cfg->wep_cfg[i].key,
                               bss_cfg->wep_cfg[i].length);
                        cmd_size += sizeof(struct mwifiex_ie_types_header) + 2 +
                                    bss_cfg->wep_cfg[i].length;
                        tlv += sizeof(struct mwifiex_ie_types_header) + 2 +
                                    bss_cfg->wep_cfg[i].length;
                }
        }

        *param_size = cmd_size;
        *tlv_buf = tlv;

        return;
}

/* This function enable 11D if userspace set the country IE.
 */
void mwifiex_config_uap_11d(struct mwifiex_private *priv,
                            struct cfg80211_beacon_data *beacon_data)
{
        enum state_11d_t state_11d;
        const u8 *country_ie;

        country_ie = cfg80211_find_ie(WLAN_EID_COUNTRY, beacon_data->tail,
                                      beacon_data->tail_len);
        if (country_ie) {
                /* Send cmd to FW to enable 11D function */
                state_11d = ENABLE_11D;
                if (mwifiex_send_cmd(priv, HostCmd_CMD_802_11_SNMP_MIB,
                                     HostCmd_ACT_GEN_SET, DOT11D_I,
                                     &state_11d, true)) {
                        mwifiex_dbg(priv->adapter, ERROR,
                                    "11D: failed to enable 11D\n");
                }
        }
}

/* This function parses BSS related parameters from structure
 * and prepares TLVs. These TLVs are appended to command buffer.
*/
static int
mwifiex_uap_bss_param_prepare(u8 *tlv, void *cmd_buf, u16 *param_size)
{
        struct host_cmd_tlv_mac_addr *mac_tlv;
        struct host_cmd_tlv_dtim_period *dtim_period;
        struct host_cmd_tlv_beacon_period *beacon_period;
        struct host_cmd_tlv_ssid *ssid;
        struct host_cmd_tlv_bcast_ssid *bcast_ssid;
        struct host_cmd_tlv_channel_band *chan_band;
        struct host_cmd_tlv_frag_threshold *frag_threshold;
        struct host_cmd_tlv_rts_threshold *rts_threshold;
        struct host_cmd_tlv_retry_limit *retry_limit;
        struct host_cmd_tlv_encrypt_protocol *encrypt_protocol;
        struct host_cmd_tlv_auth_type *auth_type;
        struct host_cmd_tlv_rates *tlv_rates;
        struct host_cmd_tlv_ageout_timer *ao_timer, *ps_ao_timer;
        struct host_cmd_tlv_power_constraint *pwr_ct;
        struct mwifiex_ie_types_htcap *htcap;
        struct mwifiex_ie_types_wmmcap *wmm_cap;
        struct mwifiex_uap_bss_param *bss_cfg = cmd_buf;
        int i;
        u16 cmd_size = *param_size;

        mac_tlv = (struct host_cmd_tlv_mac_addr *)tlv;
        mac_tlv->header.type = cpu_to_le16(TLV_TYPE_UAP_MAC_ADDRESS);
        mac_tlv->header.len = cpu_to_le16(ETH_ALEN);
        memcpy(mac_tlv->mac_addr, bss_cfg->mac_addr, ETH_ALEN);
        cmd_size += sizeof(struct host_cmd_tlv_mac_addr);
        tlv += sizeof(struct host_cmd_tlv_mac_addr);

        if (bss_cfg->ssid.ssid_len) {
                ssid = (struct host_cmd_tlv_ssid *)tlv;
                ssid->header.type = cpu_to_le16(TLV_TYPE_UAP_SSID);
                ssid->header.len = cpu_to_le16((u16)bss_cfg->ssid.ssid_len);
                memcpy(ssid->ssid, bss_cfg->ssid.ssid, bss_cfg->ssid.ssid_len);
                cmd_size += sizeof(struct mwifiex_ie_types_header) +
                            bss_cfg->ssid.ssid_len;
                tlv += sizeof(struct mwifiex_ie_types_header) +
                                bss_cfg->ssid.ssid_len;

                bcast_ssid = (struct host_cmd_tlv_bcast_ssid *)tlv;
                bcast_ssid->header.type = cpu_to_le16(TLV_TYPE_UAP_BCAST_SSID);
                bcast_ssid->header.len =
                                cpu_to_le16(sizeof(bcast_ssid->bcast_ctl));
                bcast_ssid->bcast_ctl = bss_cfg->bcast_ssid_ctl;
                cmd_size += sizeof(struct host_cmd_tlv_bcast_ssid);
                tlv += sizeof(struct host_cmd_tlv_bcast_ssid);
        }
        if (bss_cfg->rates[0]) {
                tlv_rates = (struct host_cmd_tlv_rates *)tlv;
                tlv_rates->header.type = cpu_to_le16(TLV_TYPE_UAP_RATES);

                for (i = 0; i < MWIFIEX_SUPPORTED_RATES && bss_cfg->rates[i];
                     i++)
                        tlv_rates->rates[i] = bss_cfg->rates[i];

                tlv_rates->header.len = cpu_to_le16(i);
                cmd_size += sizeof(struct host_cmd_tlv_rates) + i;
                tlv += sizeof(struct host_cmd_tlv_rates) + i;
        }
        if (bss_cfg->channel &&
            (((bss_cfg->band_cfg & BIT(0)) == BAND_CONFIG_BG &&
              bss_cfg->channel <= MAX_CHANNEL_BAND_BG) ||
            ((bss_cfg->band_cfg & BIT(0)) == BAND_CONFIG_A &&
             bss_cfg->channel <= MAX_CHANNEL_BAND_A))) {
                chan_band = (struct host_cmd_tlv_channel_band *)tlv;
                chan_band->header.type = cpu_to_le16(TLV_TYPE_CHANNELBANDLIST);
                chan_band->header.len =
                        cpu_to_le16(sizeof(struct host_cmd_tlv_channel_band) -
                                    sizeof(struct mwifiex_ie_types_header));
                chan_band->band_config = bss_cfg->band_cfg;
                chan_band->channel = bss_cfg->channel;
                cmd_size += sizeof(struct host_cmd_tlv_channel_band);
                tlv += sizeof(struct host_cmd_tlv_channel_band);
        }
        if (bss_cfg->beacon_period >= MIN_BEACON_PERIOD &&
            bss_cfg->beacon_period <= MAX_BEACON_PERIOD) {
                beacon_period = (struct host_cmd_tlv_beacon_period *)tlv;
                beacon_period->header.type =
                                        cpu_to_le16(TLV_TYPE_UAP_BEACON_PERIOD);
                beacon_period->header.len =
                        cpu_to_le16(sizeof(struct host_cmd_tlv_beacon_period) -
                                    sizeof(struct mwifiex_ie_types_header));
                beacon_period->period = cpu_to_le16(bss_cfg->beacon_period);
                cmd_size += sizeof(struct host_cmd_tlv_beacon_period);
                tlv += sizeof(struct host_cmd_tlv_beacon_period);
        }
        if (bss_cfg->dtim_period >= MIN_DTIM_PERIOD &&
            bss_cfg->dtim_period <= MAX_DTIM_PERIOD) {
                dtim_period = (struct host_cmd_tlv_dtim_period *)tlv;
                dtim_period->header.type =
                        cpu_to_le16(TLV_TYPE_UAP_DTIM_PERIOD);
                dtim_period->header.len =
                        cpu_to_le16(sizeof(struct host_cmd_tlv_dtim_period) -
                                    sizeof(struct mwifiex_ie_types_header));
                dtim_period->period = bss_cfg->dtim_period;
                cmd_size += sizeof(struct host_cmd_tlv_dtim_period);
                tlv += sizeof(struct host_cmd_tlv_dtim_period);
        }
        if (bss_cfg->rts_threshold <= MWIFIEX_RTS_MAX_VALUE) {
                rts_threshold = (struct host_cmd_tlv_rts_threshold *)tlv;
                rts_threshold->header.type =
                                        cpu_to_le16(TLV_TYPE_UAP_RTS_THRESHOLD);
                rts_threshold->header.len =
                        cpu_to_le16(sizeof(struct host_cmd_tlv_rts_threshold) -
                                    sizeof(struct mwifiex_ie_types_header));
                rts_threshold->rts_thr = cpu_to_le16(bss_cfg->rts_threshold);
                cmd_size += sizeof(struct host_cmd_tlv_frag_threshold);
                tlv += sizeof(struct host_cmd_tlv_frag_threshold);
        }
        if ((bss_cfg->frag_threshold >= MWIFIEX_FRAG_MIN_VALUE) &&
            (bss_cfg->frag_threshold <= MWIFIEX_FRAG_MAX_VALUE)) {
                frag_threshold = (struct host_cmd_tlv_frag_threshold *)tlv;
                frag_threshold->header.type =
                                cpu_to_le16(TLV_TYPE_UAP_FRAG_THRESHOLD);
                frag_threshold->header.len =
                        cpu_to_le16(sizeof(struct host_cmd_tlv_frag_threshold) -
                                    sizeof(struct mwifiex_ie_types_header));
                frag_threshold->frag_thr = cpu_to_le16(bss_cfg->frag_threshold);
                cmd_size += sizeof(struct host_cmd_tlv_frag_threshold);
                tlv += sizeof(struct host_cmd_tlv_frag_threshold);
        }
        if (bss_cfg->retry_limit <= MWIFIEX_RETRY_LIMIT) {
                retry_limit = (struct host_cmd_tlv_retry_limit *)tlv;
                retry_limit->header.type =
                        cpu_to_le16(TLV_TYPE_UAP_RETRY_LIMIT);
                retry_limit->header.len =
                        cpu_to_le16(sizeof(struct host_cmd_tlv_retry_limit) -
                                    sizeof(struct mwifiex_ie_types_header));
                retry_limit->limit = (u8)bss_cfg->retry_limit;
                cmd_size += sizeof(struct host_cmd_tlv_retry_limit);
                tlv += sizeof(struct host_cmd_tlv_retry_limit);
        }
        if ((bss_cfg->protocol & PROTOCOL_WPA) ||
            (bss_cfg->protocol & PROTOCOL_WPA2) ||
            (bss_cfg->protocol & PROTOCOL_EAP))
                mwifiex_uap_bss_wpa(&tlv, cmd_buf, &cmd_size);
        else
                mwifiex_uap_bss_wep(&tlv, cmd_buf, &cmd_size);

        if ((bss_cfg->auth_mode <= WLAN_AUTH_SHARED_KEY) ||
            (bss_cfg->auth_mode == MWIFIEX_AUTH_MODE_AUTO)) {
                auth_type = (struct host_cmd_tlv_auth_type *)tlv;
                auth_type->header.type = cpu_to_le16(TLV_TYPE_AUTH_TYPE);
                auth_type->header.len =
                        cpu_to_le16(sizeof(struct host_cmd_tlv_auth_type) -
                        sizeof(struct mwifiex_ie_types_header));
                auth_type->auth_type = (u8)bss_cfg->auth_mode;
                cmd_size += sizeof(struct host_cmd_tlv_auth_type);
                tlv += sizeof(struct host_cmd_tlv_auth_type);
        }
        if (bss_cfg->protocol) {
                encrypt_protocol = (struct host_cmd_tlv_encrypt_protocol *)tlv;
                encrypt_protocol->header.type =
                        cpu_to_le16(TLV_TYPE_UAP_ENCRY_PROTOCOL);
                encrypt_protocol->header.len =
                        cpu_to_le16(sizeof(struct host_cmd_tlv_encrypt_protocol)
                        - sizeof(struct mwifiex_ie_types_header));
                encrypt_protocol->proto = cpu_to_le16(bss_cfg->protocol);
                cmd_size += sizeof(struct host_cmd_tlv_encrypt_protocol);
                tlv += sizeof(struct host_cmd_tlv_encrypt_protocol);
        }

        if (bss_cfg->ht_cap.cap_info) {
                htcap = (struct mwifiex_ie_types_htcap *)tlv;
                htcap->header.type = cpu_to_le16(WLAN_EID_HT_CAPABILITY);
                htcap->header.len =
                                cpu_to_le16(sizeof(struct ieee80211_ht_cap));
                htcap->ht_cap.cap_info = bss_cfg->ht_cap.cap_info;
                htcap->ht_cap.ampdu_params_info =
                                             bss_cfg->ht_cap.ampdu_params_info;
                memcpy(&htcap->ht_cap.mcs, &bss_cfg->ht_cap.mcs,
                       sizeof(struct ieee80211_mcs_info));
                htcap->ht_cap.extended_ht_cap_info =
                                        bss_cfg->ht_cap.extended_ht_cap_info;
                htcap->ht_cap.tx_BF_cap_info = bss_cfg->ht_cap.tx_BF_cap_info;
                htcap->ht_cap.antenna_selection_info =
                                        bss_cfg->ht_cap.antenna_selection_info;
                cmd_size += sizeof(struct mwifiex_ie_types_htcap);
                tlv += sizeof(struct mwifiex_ie_types_htcap);
        }

        if (bss_cfg->wmm_info.qos_info != 0xFF) {
                wmm_cap = (struct mwifiex_ie_types_wmmcap *)tlv;
                wmm_cap->header.type = cpu_to_le16(WLAN_EID_VENDOR_SPECIFIC);
                wmm_cap->header.len = cpu_to_le16(sizeof(wmm_cap->wmm_info));
                memcpy(&wmm_cap->wmm_info, &bss_cfg->wmm_info,
                       sizeof(wmm_cap->wmm_info));
                cmd_size += sizeof(struct mwifiex_ie_types_wmmcap);
                tlv += sizeof(struct mwifiex_ie_types_wmmcap);
        }

        if (bss_cfg->sta_ao_timer) {
                ao_timer = (struct host_cmd_tlv_ageout_timer *)tlv;
                ao_timer->header.type = cpu_to_le16(TLV_TYPE_UAP_AO_TIMER);
                ao_timer->header.len = cpu_to_le16(sizeof(*ao_timer) -
                                        sizeof(struct mwifiex_ie_types_header));
                ao_timer->sta_ao_timer = cpu_to_le32(bss_cfg->sta_ao_timer);
                cmd_size += sizeof(*ao_timer);
                tlv += sizeof(*ao_timer);
        }

        if (bss_cfg->power_constraint) {
                pwr_ct = (void *)tlv;
                pwr_ct->header.type = cpu_to_le16(TLV_TYPE_PWR_CONSTRAINT);
                pwr_ct->header.len = cpu_to_le16(sizeof(u8));
                pwr_ct->constraint = bss_cfg->power_constraint;
                cmd_size += sizeof(*pwr_ct);
                tlv += sizeof(*pwr_ct);
        }

        if (bss_cfg->ps_sta_ao_timer) {
                ps_ao_timer = (struct host_cmd_tlv_ageout_timer *)tlv;
                ps_ao_timer->header.type =
                                cpu_to_le16(TLV_TYPE_UAP_PS_AO_TIMER);
                ps_ao_timer->header.len = cpu_to_le16(sizeof(*ps_ao_timer) -
                                sizeof(struct mwifiex_ie_types_header));
                ps_ao_timer->sta_ao_timer =
                                        cpu_to_le32(bss_cfg->ps_sta_ao_timer);
                cmd_size += sizeof(*ps_ao_timer);
                tlv += sizeof(*ps_ao_timer);
        }

        *param_size = cmd_size;

        return 0;
}

/* This function parses custom IEs from IE list and prepares command buffer */
static int mwifiex_uap_custom_ie_prepare(u8 *tlv, void *cmd_buf, u16 *ie_size)
{
        struct mwifiex_ie_list *ap_ie = cmd_buf;
        struct mwifiex_ie_types_header *tlv_ie = (void *)tlv;

        if (!ap_ie || !ap_ie->len)
                return -1;

        *ie_size += le16_to_cpu(ap_ie->len) +
                        sizeof(struct mwifiex_ie_types_header);

        tlv_ie->type = cpu_to_le16(TLV_TYPE_MGMT_IE);
        tlv_ie->len = ap_ie->len;
        tlv += sizeof(struct mwifiex_ie_types_header);

        memcpy(tlv, ap_ie->ie_list, le16_to_cpu(ap_ie->len));

        return 0;
}

/* Parse AP config structure and prepare TLV based command structure
 * to be sent to FW for uAP configuration
 */
static int
mwifiex_cmd_uap_sys_config(struct host_cmd_ds_command *cmd, u16 cmd_action,
                           u32 type, void *cmd_buf)
{
        u8 *tlv;
        u16 cmd_size, param_size, ie_size;
        struct host_cmd_ds_sys_config *sys_cfg;

        cmd->command = cpu_to_le16(HostCmd_CMD_UAP_SYS_CONFIG);
        cmd_size = (u16)(sizeof(struct host_cmd_ds_sys_config) + S_DS_GEN);
        sys_cfg = (struct host_cmd_ds_sys_config *)&cmd->params.uap_sys_config;
        sys_cfg->action = cpu_to_le16(cmd_action);
        tlv = sys_cfg->tlv;

        switch (type) {
        case UAP_BSS_PARAMS_I:
                param_size = cmd_size;
                if (mwifiex_uap_bss_param_prepare(tlv, cmd_buf, &param_size))
                        return -1;
                cmd->size = cpu_to_le16(param_size);
                break;
        case UAP_CUSTOM_IE_I:
                ie_size = cmd_size;
                if (mwifiex_uap_custom_ie_prepare(tlv, cmd_buf, &ie_size))
                        return -1;
                cmd->size = cpu_to_le16(ie_size);
                break;
        default:
                return -1;
        }

        return 0;
}

/* This function prepares AP start up command with or without host MLME
 */
static void mwifiex_cmd_uap_bss_start(struct mwifiex_private *priv,
                                     struct host_cmd_ds_command *cmd)
{
        struct mwifiex_ie_types_host_mlme *tlv;
        int size;

        cmd->command = cpu_to_le16(HostCmd_CMD_UAP_BSS_START);
        size = S_DS_GEN;

        if (priv->adapter->host_mlme_enabled) {
                tlv = (struct mwifiex_ie_types_host_mlme *)((u8 *)cmd + size);
                tlv->header.type = cpu_to_le16(TLV_TYPE_HOST_MLME);
                tlv->header.len = cpu_to_le16(sizeof(tlv->host_mlme));
                tlv->host_mlme = 1;
                size += sizeof(struct mwifiex_ie_types_host_mlme);
        }

        cmd->size = cpu_to_le16(size);
}

/* This function prepares AP specific deauth command with mac supplied in
 * function parameter.
 */
static int mwifiex_cmd_uap_sta_deauth(struct mwifiex_private *priv,
                                      struct host_cmd_ds_command *cmd, u8 *mac)
{
        struct host_cmd_ds_sta_deauth *sta_deauth = &cmd->params.sta_deauth;

        cmd->command = cpu_to_le16(HostCmd_CMD_UAP_STA_DEAUTH);
        memcpy(sta_deauth->mac, mac, ETH_ALEN);
        sta_deauth->reason = cpu_to_le16(WLAN_REASON_DEAUTH_LEAVING);

        cmd->size = cpu_to_le16(sizeof(struct host_cmd_ds_sta_deauth) +
                                S_DS_GEN);
        return 0;
}

/* This function prepares AP specific add station command.
 */
static int mwifiex_cmd_uap_add_station(struct mwifiex_private *priv,
                                       struct host_cmd_ds_command *cmd,
                                       u16 cmd_action, void *data_buf)
{
        struct host_cmd_ds_add_station *new_sta = &cmd->params.sta_info;
        struct mwifiex_sta_info *add_sta = (struct mwifiex_sta_info *)data_buf;
        struct station_parameters *params = add_sta->params;
        struct mwifiex_sta_node *sta_ptr;
        u8 *pos;
        u8 qos_capa;
        u16 header_len = sizeof(struct mwifiex_ie_types_header);
        u16 tlv_len;
        int size;
        struct mwifiex_ie_types_data *tlv;
        struct mwifiex_ie_types_sta_flag *sta_flag;
        int i;

        cmd->command = cpu_to_le16(HostCmd_CMD_ADD_NEW_STATION);
        new_sta->action = cpu_to_le16(cmd_action);
        size = sizeof(struct host_cmd_ds_add_station) + S_DS_GEN;

        if (cmd_action == HostCmd_ACT_ADD_STA)
                sta_ptr = mwifiex_add_sta_entry(priv, add_sta->peer_mac);
        else
                sta_ptr = mwifiex_get_sta_entry(priv, add_sta->peer_mac);

        if (!sta_ptr)
                return -1;

        memcpy(new_sta->peer_mac, add_sta->peer_mac, ETH_ALEN);

        if (cmd_action == HostCmd_ACT_REMOVE_STA) {
                cmd->size = cpu_to_le16(size);
                return 0;
        }

        new_sta->aid = cpu_to_le16(params->aid);
        new_sta->listen_interval = cpu_to_le32(params->listen_interval);
        new_sta->cap_info = cpu_to_le16(params->capability);

        pos = new_sta->tlv;

        if (params->sta_flags_set & NL80211_STA_FLAG_WME)
                sta_ptr->is_wmm_enabled = 1;
        sta_flag = (struct mwifiex_ie_types_sta_flag *)pos;
        sta_flag->header.type = cpu_to_le16(TLV_TYPE_UAP_STA_FLAGS);
        sta_flag->header.len = cpu_to_le16(sizeof(__le32));
        sta_flag->sta_flags = cpu_to_le32(params->sta_flags_set);
        pos += sizeof(struct mwifiex_ie_types_sta_flag);
        size += sizeof(struct mwifiex_ie_types_sta_flag);

        if (params->ext_capab_len) {
                tlv = (struct mwifiex_ie_types_data *)pos;
                tlv->header.type = cpu_to_le16(WLAN_EID_EXT_CAPABILITY);
                tlv_len = params->ext_capab_len;
                tlv->header.len = cpu_to_le16(tlv_len);
                memcpy(tlv->data, params->ext_capab, tlv_len);
                pos += (header_len + tlv_len);
                size += (header_len + tlv_len);
        }

        if (params->link_sta_params.supported_rates_len) {
                tlv = (struct mwifiex_ie_types_data *)pos;
                tlv->header.type = cpu_to_le16(WLAN_EID_SUPP_RATES);
                tlv_len = params->link_sta_params.supported_rates_len;
                tlv->header.len = cpu_to_le16(tlv_len);
                memcpy(tlv->data,
                       params->link_sta_params.supported_rates, tlv_len);
                pos += (header_len + tlv_len);
                size += (header_len + tlv_len);
        }

        if (params->uapsd_queues || params->max_sp) {
                tlv = (struct mwifiex_ie_types_data *)pos;
                tlv->header.type = cpu_to_le16(WLAN_EID_QOS_CAPA);
                tlv_len = sizeof(qos_capa);
                tlv->header.len = cpu_to_le16(tlv_len);
                qos_capa = params->uapsd_queues | (params->max_sp << 5);
                memcpy(tlv->data, &qos_capa, tlv_len);
                pos += (header_len + tlv_len);
                size += (header_len + tlv_len);
                sta_ptr->is_wmm_enabled = 1;
        }

        if (params->link_sta_params.ht_capa) {
                tlv = (struct mwifiex_ie_types_data *)pos;
                tlv->header.type = cpu_to_le16(WLAN_EID_HT_CAPABILITY);
                tlv_len = sizeof(struct ieee80211_ht_cap);
                tlv->header.len = cpu_to_le16(tlv_len);
                memcpy(tlv->data, params->link_sta_params.ht_capa, tlv_len);
                pos += (header_len + tlv_len);
                size += (header_len + tlv_len);
                sta_ptr->is_11n_enabled = 1;
                sta_ptr->max_amsdu =
                        le16_to_cpu(params->link_sta_params.ht_capa->cap_info) &
                        IEEE80211_HT_CAP_MAX_AMSDU ?
                        MWIFIEX_TX_DATA_BUF_SIZE_8K :
                        MWIFIEX_TX_DATA_BUF_SIZE_4K;
        }

        if (params->link_sta_params.vht_capa) {
                tlv = (struct mwifiex_ie_types_data *)pos;
                tlv->header.type = cpu_to_le16(WLAN_EID_VHT_CAPABILITY);
                tlv_len = sizeof(struct ieee80211_vht_cap);
                tlv->header.len = cpu_to_le16(tlv_len);
                memcpy(tlv->data, params->link_sta_params.vht_capa, tlv_len);
                pos += (header_len + tlv_len);
                size += (header_len + tlv_len);
                sta_ptr->is_11ac_enabled = 1;
        }

        if (params->link_sta_params.opmode_notif_used) {
                tlv = (struct mwifiex_ie_types_data *)pos;
                tlv->header.type = cpu_to_le16(WLAN_EID_OPMODE_NOTIF);
                tlv_len = sizeof(u8);
                tlv->header.len = cpu_to_le16(tlv_len);
                memcpy(tlv->data, &params->link_sta_params.opmode_notif,
                       tlv_len);
                pos += (header_len + tlv_len);
                size += (header_len + tlv_len);
        }

        for (i = 0; i < MAX_NUM_TID; i++) {
                if (sta_ptr->is_11n_enabled)
                        sta_ptr->ampdu_sta[i] =
                                      priv->aggr_prio_tbl[i].ampdu_user;
                else
                        sta_ptr->ampdu_sta[i] = BA_STREAM_NOT_ALLOWED;
        }

        memset(sta_ptr->rx_seq, 0xff, sizeof(sta_ptr->rx_seq));
        cmd->size = cpu_to_le16(size);

        return 0;
}

/* This function prepares the AP specific commands before sending them
 * to the firmware.
 * This is a generic function which calls specific command preparation
 * routines based upon the command number.
 */
int mwifiex_uap_prepare_cmd(struct mwifiex_private *priv, u16 cmd_no,
                            u16 cmd_action, u32 type,
                            void *data_buf, void *cmd_buf)
{
        struct host_cmd_ds_command *cmd = cmd_buf;

        switch (cmd_no) {
        case HostCmd_CMD_UAP_SYS_CONFIG:
                if (mwifiex_cmd_uap_sys_config(cmd, cmd_action, type, data_buf))
                        return -1;
                break;
        case HostCmd_CMD_UAP_BSS_START:
                mwifiex_cmd_uap_bss_start(priv, cmd);
                break;
        case HostCmd_CMD_UAP_BSS_STOP:
        case HOST_CMD_APCMD_SYS_RESET:
        case HOST_CMD_APCMD_STA_LIST:
                cmd->command = cpu_to_le16(cmd_no);
                cmd->size = cpu_to_le16(S_DS_GEN);
                break;
        case HostCmd_CMD_UAP_STA_DEAUTH:
                if (mwifiex_cmd_uap_sta_deauth(priv, cmd, data_buf))
                        return -1;
                break;
        case HostCmd_CMD_CHAN_REPORT_REQUEST:
                if (mwifiex_cmd_issue_chan_report_request(priv, cmd_buf,
                                                          data_buf))
                        return -1;
                break;
        case HostCmd_CMD_ADD_NEW_STATION:
                if (mwifiex_cmd_uap_add_station(priv, cmd, cmd_action,
                                                data_buf))
                        return -1;
                break;
        default:
                mwifiex_dbg(priv->adapter, ERROR,
                            "PREP_CMD: unknown cmd %#x\n", cmd_no);
                return -1;
        }

        return 0;
}

void mwifiex_uap_set_channel(struct mwifiex_private *priv,
                             struct mwifiex_uap_bss_param *bss_cfg,
                             struct cfg80211_chan_def chandef)
{
        u8 config_bands = 0, old_bands = priv->adapter->config_bands;

        priv->bss_chandef = chandef;

        bss_cfg->channel = ieee80211_frequency_to_channel(
                                                     chandef.chan->center_freq);

        /* Set appropriate bands */
        if (chandef.chan->band == NL80211_BAND_2GHZ) {
                bss_cfg->band_cfg = BAND_CONFIG_BG;
                config_bands = BAND_B | BAND_G;

                if (chandef.width > NL80211_CHAN_WIDTH_20_NOHT)
                        config_bands |= BAND_GN;
        } else {
                bss_cfg->band_cfg = BAND_CONFIG_A;
                config_bands = BAND_A;

                if (chandef.width > NL80211_CHAN_WIDTH_20_NOHT)
                        config_bands |= BAND_AN;

                if (chandef.width > NL80211_CHAN_WIDTH_40)
                        config_bands |= BAND_AAC;
        }

        switch (chandef.width) {
        case NL80211_CHAN_WIDTH_5:
        case NL80211_CHAN_WIDTH_10:
        case NL80211_CHAN_WIDTH_20_NOHT:
        case NL80211_CHAN_WIDTH_20:
                break;
        case NL80211_CHAN_WIDTH_40:
                if (chandef.center_freq1 < chandef.chan->center_freq)
                        bss_cfg->band_cfg |= MWIFIEX_SEC_CHAN_BELOW;
                else
                        bss_cfg->band_cfg |= MWIFIEX_SEC_CHAN_ABOVE;
                break;
        case NL80211_CHAN_WIDTH_80:
        case NL80211_CHAN_WIDTH_80P80:
        case NL80211_CHAN_WIDTH_160:
                bss_cfg->band_cfg |=
                    mwifiex_get_sec_chan_offset(bss_cfg->channel) << 4;
                break;
        default:
                mwifiex_dbg(priv->adapter,
                            WARN, "Unknown channel width: %d\n",
                            chandef.width);
                break;
        }

        priv->adapter->config_bands = config_bands;

        if (old_bands != config_bands) {
                mwifiex_send_domain_info_cmd_fw(priv->adapter->wiphy);
                mwifiex_dnld_txpwr_table(priv);
        }
}

int mwifiex_config_start_uap(struct mwifiex_private *priv,
                             struct mwifiex_uap_bss_param *bss_cfg)
{
        if (mwifiex_send_cmd(priv, HostCmd_CMD_UAP_SYS_CONFIG,
                             HostCmd_ACT_GEN_SET,
                             UAP_BSS_PARAMS_I, bss_cfg, true)) {
                mwifiex_dbg(priv->adapter, ERROR,
                            "Failed to set AP configuration\n");
                return -1;
        }

        if (mwifiex_send_cmd(priv, HostCmd_CMD_UAP_BSS_START,
                             HostCmd_ACT_GEN_SET, 0, NULL, true)) {
                mwifiex_dbg(priv->adapter, ERROR,
                            "Failed to start the BSS\n");
                return -1;
        }

        if (priv->sec_info.wep_enabled)
                priv->curr_pkt_filter |= HostCmd_ACT_MAC_WEP_ENABLE;
        else
                priv->curr_pkt_filter &= ~HostCmd_ACT_MAC_WEP_ENABLE;

        if (mwifiex_send_cmd(priv, HostCmd_CMD_MAC_CONTROL,
                             HostCmd_ACT_GEN_SET, 0,
                             &priv->curr_pkt_filter, true))
                return -1;

        return 0;
}