/******************************************************************************
 *
 * This file is provided under a dual BSD/GPLv2 license.  When using or
 * redistributing this file, you may do so under either license.
 *
 * GPL LICENSE SUMMARY
 *
 * Copyright(c) 2012 - 2014 Intel Corporation. All rights reserved.
 * Copyright(c) 2016 Intel Deutschland GmbH
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of version 2 of the GNU General Public License as
 * published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110,
 * USA
 *
 * The full GNU General Public License is included in this distribution
 * in the file called COPYING.
 *
 * Contact Information:
 *  Intel Linux Wireless <linuxwifi@intel.com>
 * Intel Corporation, 5200 N.E. Elam Young Parkway, Hillsboro, OR 97124-6497
 *
 * BSD LICENSE
 *
 * Copyright(c) 2012 - 2014 Intel Corporation. All rights reserved.
 * Copyright(c) 2016 Intel Deutschland GmbH
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 *  * Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 *  * Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in
 *    the documentation and/or other materials provided with the
 *    distribution.
 *  * Neither the name Intel Corporation nor the names of its
 *    contributors may be used to endorse or promote products derived
 *    from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 *
 *****************************************************************************/

#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);
                if (phyctxt->channel->band == NL80211_BAND_2GHZ ||
                    !iwl_mvm_is_cdb_supported(mvm))
                        cmd.lmac_id = cpu_to_le32(IWL_LMAC_24G_INDEX);
                else
                        cmd.lmac_id = cpu_to_le32(IWL_LMAC_5G_INDEX);
        } 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->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->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->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->phy_ctxt))
                return -EINVAL;

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

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

        return ret;
}