// SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause
/*
 * Copyright (C) 2012-2014, 2020 Intel Corporation
 * Copyright (C) 2016 Intel Deutschland GmbH
 * Copyright (C) 2022, 2024 Intel Corporation
 */
#include <net/mac80211.h>
#include "fw-api.h"
#include "mvm.h"

struct iwl_mvm_iface_iterator_data {
 struct ieee80211_vif *ignore_vif;
 int idx;

 struct iwl_mvm_phy_ctxt *phyctxt;

 u16 ids[MAX_MACS_IN_BINDING];
 u16 colors[MAX_MACS_IN_BINDING];
};

static int iwl_mvm_binding_cmd(struct iwl_mvm *mvm, u32 action,
          struct iwl_mvm_iface_iterator_data *data)
{
 struct iwl_binding_cmd cmd;
 struct iwl_mvm_phy_ctxt *phyctxt = data->phyctxt;
 int i, ret;
 u32 status;
 int size;

 memset(&cmd, 0, sizeof(cmd));

 if (fw_has_capa(&mvm->fw->ucode_capa,
   IWL_UCODE_TLV_CAPA_BINDING_CDB_SUPPORT)) {
  size = sizeof(cmd);
  cmd.lmac_id = cpu_to_le32(iwl_mvm_get_lmac_id(mvm,
             phyctxt->channel->band));
 } else {
  size = IWL_BINDING_CMD_SIZE_V1;
 }

 cmd.id_and_color = cpu_to_le32(FW_CMD_ID_AND_COLOR(phyctxt->id,
          phyctxt->color));
 cmd.action = cpu_to_le32(action);
 cmd.phy = cpu_to_le32(FW_CMD_ID_AND_COLOR(phyctxt->id,
        phyctxt->color));

 for (i = 0; i < MAX_MACS_IN_BINDING; i++)
  cmd.macs[i] = cpu_to_le32(FW_CTXT_INVALID);
 for (i = 0; i < data->idx; i++)
  cmd.macs[i] = cpu_to_le32(FW_CMD_ID_AND_COLOR(data->ids[i],
             data->colors[i]));

 status = 0;
 ret = iwl_mvm_send_cmd_pdu_status(mvm, BINDING_CONTEXT_CMD,
       size, &cmd, &status);
 if (ret) {
  IWL_ERR(mvm, "Failed to send binding (action:%d): %d\n",
   action, ret);
  return ret;
 }

 if (status) {
  IWL_ERR(mvm, "Binding command failed: %u\n", status);
  ret = -EIO;
 }

 return ret;
}

static void iwl_mvm_iface_iterator(void *_data, u8 *mac,
       struct ieee80211_vif *vif)
{
 struct iwl_mvm_iface_iterator_data *data = _data;
 struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);

 if (vif == data->ignore_vif)
  return;

 if (mvmvif->deflink.phy_ctxt != data->phyctxt)
  return;

 if (WARN_ON_ONCE(data->idx >= MAX_MACS_IN_BINDING))
  return;

 data->ids[data->idx] = mvmvif->id;
 data->colors[data->idx] = mvmvif->color;
 data->idx++;
}

static int iwl_mvm_binding_update(struct iwl_mvm *mvm,
      struct ieee80211_vif *vif,
      struct iwl_mvm_phy_ctxt *phyctxt,
      bool add)
{
 struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
 struct iwl_mvm_iface_iterator_data data = {
  .ignore_vif = vif,
  .phyctxt = phyctxt,
 };
 u32 action = FW_CTXT_ACTION_MODIFY;

 lockdep_assert_held(&mvm->mutex);

 ieee80211_iterate_active_interfaces_atomic(mvm->hw,
         IEEE80211_IFACE_ITER_NORMAL,
         iwl_mvm_iface_iterator,
         &data);

 /*
 * If there are no other interfaces yet we
 * need to create a new binding.
 */
 if (data.idx == 0) {
  if (add)
   action = FW_CTXT_ACTION_ADD;
  else
   action = FW_CTXT_ACTION_REMOVE;
 }

 if (add) {
  if (WARN_ON_ONCE(data.idx >= MAX_MACS_IN_BINDING))
   return -EINVAL;

  data.ids[data.idx] = mvmvif->id;
  data.colors[data.idx] = mvmvif->color;
  data.idx++;
 }

 return iwl_mvm_binding_cmd(mvm, action, &data);
}

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

 if (WARN_ON_ONCE(!mvmvif->deflink.phy_ctxt))
  return -EINVAL;

 /*
 * Update SF - Disable if needed. if this fails, SF might still be on
 * while many macs are bound, which is forbidden - so fail the binding.
 */
 if (iwl_mvm_sf_update(mvm, vif, false))
  return -EINVAL;

 return iwl_mvm_binding_update(mvm, vif, mvmvif->deflink.phy_ctxt,
          true);
}

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

 if (WARN_ON_ONCE(!mvmvif->deflink.phy_ctxt))
  return -EINVAL;

 ret = iwl_mvm_binding_update(mvm, vif, mvmvif->deflink.phy_ctxt,
         false);

 if (!ret && iwl_mvm_sf_update(mvm, vif, true))
  IWL_ERR(mvm, "Failed to update SF state\n");

 return ret;
}

u32 iwl_mvm_get_lmac_id(struct iwl_mvm *mvm, enum nl80211_band band)
{
 if (!fw_has_capa(&mvm->fw->ucode_capa, IWL_UCODE_TLV_CAPA_CDB_SUPPORT) ||
     band == NL80211_BAND_2GHZ)
  return IWL_LMAC_24G_INDEX;
 return IWL_LMAC_5G_INDEX;
}