// SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause
/*
 * Copyright (C) 2012-2014, 2021-2022, 2024 Intel Corporation
 * Copyright (C) 2013-2014 Intel Mobile Communications GmbH
 * Copyright (C) 2015 Intel Deutschland GmbH
 */
#include <net/ipv6.h>
#include <net/addrconf.h>
#include <linux/bitops.h>
#include "mvm.h"

void iwl_mvm_set_wowlan_qos_seq(struct iwl_mvm_sta *mvm_ap_sta,
                                struct iwl_wowlan_config_cmd_v6 *cmd)
{
        int i;

        /*
         * For QoS counters, we store the one to use next, so subtract 0x10
         * since the uCode will add 0x10 *before* using the value while we
         * increment after using the value (i.e. store the next value to use).
         */
        for (i = 0; i < IWL_MAX_TID_COUNT; i++) {
                u16 seq = mvm_ap_sta->tid_data[i].seq_number;
                seq -= 0x10;
                cmd->qos_seq[i] = cpu_to_le16(seq);
        }
}

int iwl_mvm_send_proto_offload(struct iwl_mvm *mvm,
                               struct ieee80211_vif *vif,
                               bool disable_offloading,
                               bool offload_ns,
                               u32 cmd_flags,
                               u8 sta_id)
{
        union {
                struct iwl_proto_offload_cmd_v1 v1;
                struct iwl_proto_offload_cmd_v2 v2;
                struct iwl_proto_offload_cmd_v3_small v3s;
                struct iwl_proto_offload_cmd_v4 v4;
        } cmd = {};
        struct iwl_host_cmd hcmd = {
                .id = PROT_OFFLOAD_CONFIG_CMD,
                .flags = cmd_flags,
                .data[0] = &cmd,
                .dataflags[0] = IWL_HCMD_DFL_DUP,
        };
        struct iwl_proto_offload_cmd_common *common;
        u32 enabled = 0, size;
        u32 capa_flags = mvm->fw->ucode_capa.flags;
        int ver = iwl_fw_lookup_cmd_ver(mvm->fw, hcmd.id, 0);

#if IS_ENABLED(CONFIG_IPV6)
        struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
        int i;
        /*
         * Skip tentative address when ns offload is enabled to avoid
         * violating RFC4862.
         * Keep tentative address when ns offload is disabled so the NS packets
         * will not be filtered out and will wake up the host.
         */
        bool skip_tentative = offload_ns;

        if (capa_flags & IWL_UCODE_TLV_FLAGS_NEW_NSOFFL_SMALL ||
            capa_flags & IWL_UCODE_TLV_FLAGS_NEW_NSOFFL_LARGE) {
                struct iwl_ns_config *nsc;
                struct iwl_targ_addr *addrs;
                int n_nsc, n_addrs;
                int c;
                int num_skipped = 0;

                if (capa_flags & IWL_UCODE_TLV_FLAGS_NEW_NSOFFL_SMALL) {
                        nsc = cmd.v3s.ns_config;
                        n_nsc = IWL_PROTO_OFFLOAD_NUM_NS_CONFIG_V3S;
                        addrs = cmd.v3s.targ_addrs;
                        n_addrs = IWL_PROTO_OFFLOAD_NUM_IPV6_ADDRS_V3S;
                } else {
                        nsc = cmd.v4.ns_config;
                        n_nsc = IWL_PROTO_OFFLOAD_NUM_NS_CONFIG_V3L;
                        addrs = cmd.v4.targ_addrs;
                        n_addrs = IWL_PROTO_OFFLOAD_NUM_IPV6_ADDRS_V3L;
                }

                /*
                 * For each address we have (and that will fit) fill a target
                 * address struct and combine for NS offload structs with the
                 * solicited node addresses.
                 */
                for (i = 0, c = 0;
                     i < mvmvif->num_target_ipv6_addrs &&
                     i < n_addrs && c < n_nsc; i++) {
                        struct in6_addr solicited_addr;
                        int j;

                        if (skip_tentative &&
                            test_bit(i, mvmvif->tentative_addrs)) {
                                num_skipped++;
                                continue;
                        }

                        addrconf_addr_solict_mult(&mvmvif->target_ipv6_addrs[i],
                                                  &solicited_addr);
                        for (j = 0; j < c; j++)
                                if (ipv6_addr_cmp(&nsc[j].dest_ipv6_addr,
                                                  &solicited_addr) == 0)
                                        break;
                        if (j == c)
                                c++;
                        addrs[i].addr = mvmvif->target_ipv6_addrs[i];
                        addrs[i].config_num = cpu_to_le32(j);
                        nsc[j].dest_ipv6_addr = solicited_addr;
                        memcpy(nsc[j].target_mac_addr, vif->addr, ETH_ALEN);
                }

                if (mvmvif->num_target_ipv6_addrs - num_skipped)
                        enabled |= IWL_D3_PROTO_IPV6_VALID;

                if (capa_flags & IWL_UCODE_TLV_FLAGS_NEW_NSOFFL_SMALL)
                        cmd.v3s.num_valid_ipv6_addrs =
                                cpu_to_le32(i - num_skipped);
                else
                        cmd.v4.num_valid_ipv6_addrs =
                                cpu_to_le32(i - num_skipped);
        } else if (capa_flags & IWL_UCODE_TLV_FLAGS_D3_6_IPV6_ADDRS) {
                bool found = false;

                BUILD_BUG_ON(sizeof(cmd.v2.target_ipv6_addr[0]) !=
                             sizeof(mvmvif->target_ipv6_addrs[0]));

                for (i = 0; i < min(mvmvif->num_target_ipv6_addrs,
                                    IWL_PROTO_OFFLOAD_NUM_IPV6_ADDRS_V2); i++) {
                        if (skip_tentative &&
                            test_bit(i, mvmvif->tentative_addrs))
                                continue;

                        memcpy(cmd.v2.target_ipv6_addr[i],
                               &mvmvif->target_ipv6_addrs[i],
                               sizeof(cmd.v2.target_ipv6_addr[i]));

                        found = true;
                }
                if (found) {
                        enabled |= IWL_D3_PROTO_IPV6_VALID;
                        memcpy(cmd.v2.ndp_mac_addr, vif->addr, ETH_ALEN);
                }
        } else {
                bool found = false;
                BUILD_BUG_ON(sizeof(cmd.v1.target_ipv6_addr[0]) !=
                             sizeof(mvmvif->target_ipv6_addrs[0]));

                for (i = 0; i < min(mvmvif->num_target_ipv6_addrs,
                                    IWL_PROTO_OFFLOAD_NUM_IPV6_ADDRS_V1); i++) {
                        if (skip_tentative &&
                            test_bit(i, mvmvif->tentative_addrs))
                                continue;

                        memcpy(cmd.v1.target_ipv6_addr[i],
                               &mvmvif->target_ipv6_addrs[i],
                               sizeof(cmd.v1.target_ipv6_addr[i]));

                        found = true;
                }

                if (found) {
                        enabled |= IWL_D3_PROTO_IPV6_VALID;
                        memcpy(cmd.v1.ndp_mac_addr, vif->addr, ETH_ALEN);
                }
        }

        if (offload_ns && (enabled & IWL_D3_PROTO_IPV6_VALID))
                enabled |= IWL_D3_PROTO_OFFLOAD_NS;
#endif
        if (capa_flags & IWL_UCODE_TLV_FLAGS_NEW_NSOFFL_SMALL) {
                common = &cmd.v3s.common;
                size = sizeof(cmd.v3s);
        } else if (capa_flags & IWL_UCODE_TLV_FLAGS_NEW_NSOFFL_LARGE) {
                common = &cmd.v4.common;
                size = sizeof(cmd.v4);
                if (ver < 4) {
                        /*
                         * This basically uses iwl_proto_offload_cmd_v3_large
                         * which doesn't have the sta_id parameter before the
                         * common part.
                         */
                        size -= sizeof(cmd.v4.sta_id);
                        hcmd.data[0] = common;
                }
        } else if (capa_flags & IWL_UCODE_TLV_FLAGS_D3_6_IPV6_ADDRS) {
                common = &cmd.v2.common;
                size = sizeof(cmd.v2);
        } else {
                common = &cmd.v1.common;
                size = sizeof(cmd.v1);
        }

        if (vif->cfg.arp_addr_cnt) {
                enabled |= IWL_D3_PROTO_OFFLOAD_ARP | IWL_D3_PROTO_IPV4_VALID;
                common->host_ipv4_addr = vif->cfg.arp_addr_list[0];
                memcpy(common->arp_mac_addr, vif->addr, ETH_ALEN);
        }

        if (fw_has_capa(&mvm->fw->ucode_capa,
                        IWL_UCODE_TLV_CAPA_OFFLOAD_BTM_SUPPORT))
                enabled |= IWL_D3_PROTO_OFFLOAD_BTM;

        if (!disable_offloading)
                common->enabled = cpu_to_le32(enabled);

        if (ver >= 4)
                cmd.v4.sta_id = cpu_to_le32(sta_id);

        hcmd.len[0] = size;
        return iwl_mvm_send_cmd(mvm, &hcmd);
}