// SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause
/*
 * Copyright (C) 2022 - 2025 Intel Corporation
 */
#include "mvm.h"

static void iwl_mvm_mld_set_he_support(struct iwl_mvm *mvm,
                                       struct ieee80211_vif *vif,
                                       struct iwl_mac_config_cmd *cmd,
                                       int cmd_ver)
{
        if (vif->type == NL80211_IFTYPE_AP) {
                if (cmd_ver == 2)
                        cmd->wifi_gen_v2.he_ap_support = cpu_to_le16(1);
                else
                        cmd->wifi_gen.he_ap_support = 1;
        } else {
                if (cmd_ver == 2)
                        cmd->wifi_gen_v2.he_support = cpu_to_le16(1);
                else
                        cmd->wifi_gen.he_support = 1;
        }
}

static void iwl_mvm_mld_mac_ctxt_cmd_common(struct iwl_mvm *mvm,
                                            struct ieee80211_vif *vif,
                                            struct iwl_mac_config_cmd *cmd,
                                            u32 action)
{
        struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
        struct ieee80211_bss_conf *link_conf;
        unsigned int link_id;
        int cmd_ver = iwl_fw_lookup_cmd_ver(mvm->fw,
                                            WIDE_ID(MAC_CONF_GROUP,
                                                    MAC_CONFIG_CMD), 1);

        if (WARN_ON(cmd_ver > 3))
                return;

        cmd->id_and_color = cpu_to_le32(mvmvif->id);
        cmd->action = cpu_to_le32(action);

        cmd->mac_type = cpu_to_le32(iwl_mvm_get_mac_type(vif));

        memcpy(cmd->local_mld_addr, vif->addr, ETH_ALEN);

        cmd->wifi_gen_v2.he_support = 0;
        cmd->wifi_gen_v2.eht_support = 0;

        /* should be set by specific context type handler */
        cmd->filter_flags = 0;

        cmd->nic_not_ack_enabled =
                cpu_to_le32(!iwl_mvm_is_nic_ack_enabled(mvm, vif));

        if (iwlwifi_mod_params.disable_11ax)
                return;

        rcu_read_lock();
        for (link_id = 0; link_id < ARRAY_SIZE((vif)->link_conf); link_id++) {
                link_conf = rcu_dereference(vif->link_conf[link_id]);
                if (!link_conf)
                        continue;

                if (link_conf->he_support)
                        iwl_mvm_mld_set_he_support(mvm, vif, cmd, cmd_ver);

                /* It's not reasonable to have EHT without HE and FW API doesn't
                 * support it. Ignore EHT in this case.
                 */
                if (!link_conf->he_support && link_conf->eht_support)
                        continue;

                if (link_conf->eht_support) {
                        if (cmd_ver == 2)
                                cmd->wifi_gen_v2.eht_support = cpu_to_le32(1);
                        else
                                cmd->wifi_gen.eht_support = 1;
                        break;
                }
        }
        rcu_read_unlock();
}

static int iwl_mvm_mld_mac_ctxt_send_cmd(struct iwl_mvm *mvm,
                                         struct iwl_mac_config_cmd *cmd)
{
        int ret = iwl_mvm_send_cmd_pdu(mvm,
                                       WIDE_ID(MAC_CONF_GROUP, MAC_CONFIG_CMD),
                                       0, sizeof(*cmd), cmd);
        if (ret)
                IWL_ERR(mvm, "Failed to send MAC_CONFIG_CMD (action:%d): %d\n",
                        le32_to_cpu(cmd->action), ret);
        return ret;
}

static int iwl_mvm_mld_mac_ctxt_cmd_sta(struct iwl_mvm *mvm,
                                        struct ieee80211_vif *vif,
                                        u32 action, bool force_assoc_off)
{
        struct iwl_mac_config_cmd cmd = {};

        WARN_ON(vif->type != NL80211_IFTYPE_STATION);

        /* Fill the common data for all mac context types */
        iwl_mvm_mld_mac_ctxt_cmd_common(mvm, vif, &cmd, action);

        /*
         * We always want to hear MCAST frames, if we're not authorized yet,
         * we'll drop them.
         */
        cmd.filter_flags |= cpu_to_le32(MAC_CFG_FILTER_ACCEPT_GRP);

        if (vif->p2p)
                cmd.client.ctwin =
                        iwl_mvm_mac_ctxt_cmd_p2p_sta_get_oppps_ctwin(mvm, vif);

        if (vif->cfg.assoc && !force_assoc_off) {
                struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);

                cmd.client.is_assoc = 1;

                if (!mvmvif->authorized &&
                    fw_has_capa(&mvm->fw->ucode_capa,
                                IWL_UCODE_TLV_CAPA_COEX_HIGH_PRIO))
                        cmd.client.data_policy |=
                                cpu_to_le16(COEX_HIGH_PRIORITY_ENABLE);

        } else {
                cmd.client.is_assoc = 0;

                /* Allow beacons to pass through as long as we are not
                 * associated, or we do not have dtim period information.
                 */
                cmd.filter_flags |= cpu_to_le32(MAC_CFG_FILTER_ACCEPT_BEACON);
        }

        cmd.client.assoc_id = cpu_to_le16(vif->cfg.aid);

        if (vif->probe_req_reg && vif->cfg.assoc && vif->p2p)
                cmd.filter_flags |= cpu_to_le32(MAC_CFG_FILTER_ACCEPT_PROBE_REQ);

        if (vif->bss_conf.he_support && !iwlwifi_mod_params.disable_11ax)
                cmd.client.data_policy |=
                        cpu_to_le16(iwl_mvm_mac_ctxt_cmd_sta_get_twt_policy(mvm, vif));

        return iwl_mvm_mld_mac_ctxt_send_cmd(mvm, &cmd);
}

static int iwl_mvm_mld_mac_ctxt_cmd_listener(struct iwl_mvm *mvm,
                                             struct ieee80211_vif *vif,
                                             u32 action)
{
        struct iwl_mac_config_cmd cmd = {};

        WARN_ON(vif->type != NL80211_IFTYPE_MONITOR);

        iwl_mvm_mld_mac_ctxt_cmd_common(mvm, vif, &cmd, action);

        cmd.filter_flags = cpu_to_le32(MAC_CFG_FILTER_PROMISC |
                                       MAC_CFG_FILTER_ACCEPT_CONTROL_AND_MGMT |
                                       MAC_CFG_FILTER_ACCEPT_BEACON |
                                       MAC_CFG_FILTER_ACCEPT_PROBE_REQ |
                                       MAC_CFG_FILTER_ACCEPT_GRP);

        return iwl_mvm_mld_mac_ctxt_send_cmd(mvm, &cmd);
}

static int iwl_mvm_mld_mac_ctxt_cmd_ibss(struct iwl_mvm *mvm,
                                         struct ieee80211_vif *vif,
                                         u32 action)
{
        struct iwl_mac_config_cmd cmd = {};

        WARN_ON(vif->type != NL80211_IFTYPE_ADHOC);

        iwl_mvm_mld_mac_ctxt_cmd_common(mvm, vif, &cmd, action);

        cmd.filter_flags = cpu_to_le32(MAC_CFG_FILTER_ACCEPT_BEACON |
                                       MAC_CFG_FILTER_ACCEPT_PROBE_REQ |
                                       MAC_CFG_FILTER_ACCEPT_GRP);

        return iwl_mvm_mld_mac_ctxt_send_cmd(mvm, &cmd);
}

static int iwl_mvm_mld_mac_ctxt_cmd_p2p_device(struct iwl_mvm *mvm,
                                               struct ieee80211_vif *vif,
                                               u32 action)
{
        struct iwl_mac_config_cmd cmd = {};

        WARN_ON(vif->type != NL80211_IFTYPE_P2P_DEVICE);

        iwl_mvm_mld_mac_ctxt_cmd_common(mvm, vif, &cmd, action);

        cmd.p2p_dev.is_disc_extended =
                iwl_mac_ctxt_p2p_dev_has_extended_disc(mvm, vif);

        /* Override the filter flags to accept all management frames. This is
         * needed to support both P2P device discovery using probe requests and
         * P2P service discovery using action frames
         */
        cmd.filter_flags = cpu_to_le32(MAC_CFG_FILTER_ACCEPT_CONTROL_AND_MGMT);

        return iwl_mvm_mld_mac_ctxt_send_cmd(mvm, &cmd);
}

static int iwl_mvm_mld_mac_ctxt_cmd_ap_go(struct iwl_mvm *mvm,
                                          struct ieee80211_vif *vif,
                                          u32 action)
{
        struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
        struct iwl_mac_config_cmd cmd = {};

        WARN_ON(vif->type != NL80211_IFTYPE_AP);

        /* Fill the common data for all mac context types */
        iwl_mvm_mld_mac_ctxt_cmd_common(mvm, vif, &cmd, action);

        iwl_mvm_mac_ctxt_cmd_ap_set_filter_flags(mvm, mvmvif,
                                                 &cmd.filter_flags,
                                                 MAC_CFG_FILTER_ACCEPT_PROBE_REQ,
                                                 MAC_CFG_FILTER_ACCEPT_BEACON);

        return iwl_mvm_mld_mac_ctxt_send_cmd(mvm, &cmd);
}

static int iwl_mvm_mld_mac_ctx_send(struct iwl_mvm *mvm,
                                    struct ieee80211_vif *vif,
                                    u32 action, bool force_assoc_off)
{
        switch (vif->type) {
        case NL80211_IFTYPE_STATION:
                return iwl_mvm_mld_mac_ctxt_cmd_sta(mvm, vif, action,
                                                    force_assoc_off);
        case NL80211_IFTYPE_AP:
                return iwl_mvm_mld_mac_ctxt_cmd_ap_go(mvm, vif, action);
        case NL80211_IFTYPE_MONITOR:
                return iwl_mvm_mld_mac_ctxt_cmd_listener(mvm, vif, action);
        case NL80211_IFTYPE_P2P_DEVICE:
                return iwl_mvm_mld_mac_ctxt_cmd_p2p_device(mvm, vif, action);
        case NL80211_IFTYPE_ADHOC:
                return iwl_mvm_mld_mac_ctxt_cmd_ibss(mvm, vif, action);
        default:
                break;
        }

        return -EOPNOTSUPP;
}

int iwl_mvm_mld_mac_ctxt_add(struct iwl_mvm *mvm, struct ieee80211_vif *vif)
{
        struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
        int ret;

        if (WARN_ONCE(mvmvif->uploaded, "Adding active MAC %pM/%d\n",
                      vif->addr, ieee80211_vif_type_p2p(vif)))
                return -EIO;

        ret = iwl_mvm_mld_mac_ctx_send(mvm, vif, FW_CTXT_ACTION_ADD,
                                       true);
        if (ret)
                return ret;

        /* will only do anything at resume from D3 time */
        iwl_mvm_set_last_nonqos_seq(mvm, vif);

        mvmvif->uploaded = true;
        return 0;
}

int iwl_mvm_mld_mac_ctxt_changed(struct iwl_mvm *mvm,
                                 struct ieee80211_vif *vif,
                                 bool force_assoc_off)
{
        struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);

        if (WARN_ONCE(!mvmvif->uploaded, "Changing inactive MAC %pM/%d\n",
                      vif->addr, ieee80211_vif_type_p2p(vif)))
                return -EIO;

        return iwl_mvm_mld_mac_ctx_send(mvm, vif, FW_CTXT_ACTION_MODIFY,
                                        force_assoc_off);
}

int iwl_mvm_mld_mac_ctxt_remove(struct iwl_mvm *mvm, struct ieee80211_vif *vif)
{
        struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
        struct iwl_mac_config_cmd cmd = {
                .action = cpu_to_le32(FW_CTXT_ACTION_REMOVE),
                .id_and_color = cpu_to_le32(mvmvif->id),
        };
        int ret;

        if (WARN_ONCE(!mvmvif->uploaded, "Removing inactive MAC %pM/%d\n",
                      vif->addr, ieee80211_vif_type_p2p(vif)))
                return -EIO;

        ret = iwl_mvm_mld_mac_ctxt_send_cmd(mvm, &cmd);
        if (ret)
                return ret;

        mvmvif->uploaded = false;

        return 0;
}