// SPDX-License-Identifier: GPL-2.0+
/* Microchip VCAP TC
 *
 * Copyright (c) 2023 Microchip Technology Inc. and its subsidiaries.
 */

#include <net/flow_offload.h>
#include <net/ipv6.h>
#include <net/tcp.h>

#include "vcap_api_client.h"
#include "vcap_tc.h"

enum vcap_is2_arp_opcode {
        VCAP_IS2_ARP_REQUEST,
        VCAP_IS2_ARP_REPLY,
        VCAP_IS2_RARP_REQUEST,
        VCAP_IS2_RARP_REPLY,
};

enum vcap_arp_opcode {
        VCAP_ARP_OP_RESERVED,
        VCAP_ARP_OP_REQUEST,
        VCAP_ARP_OP_REPLY,
};

int vcap_tc_flower_handler_ethaddr_usage(struct vcap_tc_flower_parse_usage *st)
{
        enum vcap_key_field smac_key = VCAP_KF_L2_SMAC;
        enum vcap_key_field dmac_key = VCAP_KF_L2_DMAC;
        struct flow_match_eth_addrs match;
        struct vcap_u48_key smac, dmac;
        int err = 0;

        flow_rule_match_eth_addrs(st->frule, &match);

        if (!is_zero_ether_addr(match.mask->src)) {
                vcap_netbytes_copy(smac.value, match.key->src, ETH_ALEN);
                vcap_netbytes_copy(smac.mask, match.mask->src, ETH_ALEN);
                err = vcap_rule_add_key_u48(st->vrule, smac_key, &smac);
                if (err)
                        goto out;
        }

        if (!is_zero_ether_addr(match.mask->dst)) {
                vcap_netbytes_copy(dmac.value, match.key->dst, ETH_ALEN);
                vcap_netbytes_copy(dmac.mask, match.mask->dst, ETH_ALEN);
                err = vcap_rule_add_key_u48(st->vrule, dmac_key, &dmac);
                if (err)
                        goto out;
        }

        st->used_keys |= BIT_ULL(FLOW_DISSECTOR_KEY_ETH_ADDRS);

        return err;

out:
        NL_SET_ERR_MSG_MOD(st->fco->common.extack, "eth_addr parse error");
        return err;
}
EXPORT_SYMBOL_GPL(vcap_tc_flower_handler_ethaddr_usage);

int vcap_tc_flower_handler_ipv4_usage(struct vcap_tc_flower_parse_usage *st)
{
        int err = 0;

        if (st->l3_proto == ETH_P_IP) {
                struct flow_match_ipv4_addrs mt;

                flow_rule_match_ipv4_addrs(st->frule, &mt);
                if (mt.mask->src) {
                        err = vcap_rule_add_key_u32(st->vrule,
                                                    VCAP_KF_L3_IP4_SIP,
                                                    be32_to_cpu(mt.key->src),
                                                    be32_to_cpu(mt.mask->src));
                        if (err)
                                goto out;
                }
                if (mt.mask->dst) {
                        err = vcap_rule_add_key_u32(st->vrule,
                                                    VCAP_KF_L3_IP4_DIP,
                                                    be32_to_cpu(mt.key->dst),
                                                    be32_to_cpu(mt.mask->dst));
                        if (err)
                                goto out;
                }
        }

        st->used_keys |= BIT_ULL(FLOW_DISSECTOR_KEY_IPV4_ADDRS);

        return err;

out:
        NL_SET_ERR_MSG_MOD(st->fco->common.extack, "ipv4_addr parse error");
        return err;
}
EXPORT_SYMBOL_GPL(vcap_tc_flower_handler_ipv4_usage);

int vcap_tc_flower_handler_ipv6_usage(struct vcap_tc_flower_parse_usage *st)
{
        int err = 0;

        if (st->l3_proto == ETH_P_IPV6) {
                struct flow_match_ipv6_addrs mt;
                struct vcap_u128_key sip;
                struct vcap_u128_key dip;

                flow_rule_match_ipv6_addrs(st->frule, &mt);
                /* Check if address masks are non-zero */
                if (!ipv6_addr_any(&mt.mask->src)) {
                        vcap_netbytes_copy(sip.value, mt.key->src.s6_addr, 16);
                        vcap_netbytes_copy(sip.mask, mt.mask->src.s6_addr, 16);
                        err = vcap_rule_add_key_u128(st->vrule,
                                                     VCAP_KF_L3_IP6_SIP, &sip);
                        if (err)
                                goto out;
                }
                if (!ipv6_addr_any(&mt.mask->dst)) {
                        vcap_netbytes_copy(dip.value, mt.key->dst.s6_addr, 16);
                        vcap_netbytes_copy(dip.mask, mt.mask->dst.s6_addr, 16);
                        err = vcap_rule_add_key_u128(st->vrule,
                                                     VCAP_KF_L3_IP6_DIP, &dip);
                        if (err)
                                goto out;
                }
        }
        st->used_keys |= BIT_ULL(FLOW_DISSECTOR_KEY_IPV6_ADDRS);
        return err;
out:
        NL_SET_ERR_MSG_MOD(st->fco->common.extack, "ipv6_addr parse error");
        return err;
}
EXPORT_SYMBOL_GPL(vcap_tc_flower_handler_ipv6_usage);

int vcap_tc_flower_handler_portnum_usage(struct vcap_tc_flower_parse_usage *st)
{
        struct flow_match_ports mt;
        u16 value, mask;
        int err = 0;

        flow_rule_match_ports(st->frule, &mt);

        if (mt.mask->src) {
                value = be16_to_cpu(mt.key->src);
                mask = be16_to_cpu(mt.mask->src);
                err = vcap_rule_add_key_u32(st->vrule, VCAP_KF_L4_SPORT, value,
                                            mask);
                if (err)
                        goto out;
        }

        if (mt.mask->dst) {
                value = be16_to_cpu(mt.key->dst);
                mask = be16_to_cpu(mt.mask->dst);
                err = vcap_rule_add_key_u32(st->vrule, VCAP_KF_L4_DPORT, value,
                                            mask);
                if (err)
                        goto out;
        }

        st->used_keys |= BIT_ULL(FLOW_DISSECTOR_KEY_PORTS);

        return err;

out:
        NL_SET_ERR_MSG_MOD(st->fco->common.extack, "port parse error");
        return err;
}
EXPORT_SYMBOL_GPL(vcap_tc_flower_handler_portnum_usage);

int vcap_tc_flower_handler_cvlan_usage(struct vcap_tc_flower_parse_usage *st)
{
        enum vcap_key_field vid_key = VCAP_KF_8021Q_VID0;
        enum vcap_key_field pcp_key = VCAP_KF_8021Q_PCP0;
        struct flow_match_vlan mt;
        u16 tpid;
        int err;

        flow_rule_match_cvlan(st->frule, &mt);

        tpid = be16_to_cpu(mt.key->vlan_tpid);

        if (tpid == ETH_P_8021Q) {
                vid_key = VCAP_KF_8021Q_VID1;
                pcp_key = VCAP_KF_8021Q_PCP1;
        }

        if (mt.mask->vlan_id) {
                err = vcap_rule_add_key_u32(st->vrule, vid_key,
                                            mt.key->vlan_id,
                                            mt.mask->vlan_id);
                if (err)
                        goto out;
        }

        if (mt.mask->vlan_priority) {
                err = vcap_rule_add_key_u32(st->vrule, pcp_key,
                                            mt.key->vlan_priority,
                                            mt.mask->vlan_priority);
                if (err)
                        goto out;
        }

        st->used_keys |= BIT_ULL(FLOW_DISSECTOR_KEY_CVLAN);

        return 0;
out:
        NL_SET_ERR_MSG_MOD(st->fco->common.extack, "cvlan parse error");
        return err;
}
EXPORT_SYMBOL_GPL(vcap_tc_flower_handler_cvlan_usage);

int vcap_tc_flower_handler_vlan_usage(struct vcap_tc_flower_parse_usage *st,
                                      enum vcap_key_field vid_key,
                                      enum vcap_key_field pcp_key)
{
        struct flow_match_vlan mt;
        int err;

        flow_rule_match_vlan(st->frule, &mt);

        if (mt.mask->vlan_id) {
                err = vcap_rule_add_key_u32(st->vrule, vid_key,
                                            mt.key->vlan_id,
                                            mt.mask->vlan_id);
                if (err)
                        goto out;
        }

        if (mt.mask->vlan_priority) {
                err = vcap_rule_add_key_u32(st->vrule, pcp_key,
                                            mt.key->vlan_priority,
                                            mt.mask->vlan_priority);
                if (err)
                        goto out;
        }

        if (mt.mask->vlan_tpid)
                st->tpid = be16_to_cpu(mt.key->vlan_tpid);

        st->used_keys |= BIT_ULL(FLOW_DISSECTOR_KEY_VLAN);

        return 0;
out:
        NL_SET_ERR_MSG_MOD(st->fco->common.extack, "vlan parse error");
        return err;
}
EXPORT_SYMBOL_GPL(vcap_tc_flower_handler_vlan_usage);

int vcap_tc_flower_handler_tcp_usage(struct vcap_tc_flower_parse_usage *st)
{
        struct flow_match_tcp mt;
        u16 tcp_flags_mask;
        u16 tcp_flags_key;
        enum vcap_bit val;
        int err = 0;

        flow_rule_match_tcp(st->frule, &mt);
        tcp_flags_key = be16_to_cpu(mt.key->flags);
        tcp_flags_mask = be16_to_cpu(mt.mask->flags);

        if (tcp_flags_mask & TCPHDR_FIN) {
                val = VCAP_BIT_0;
                if (tcp_flags_key & TCPHDR_FIN)
                        val = VCAP_BIT_1;
                err = vcap_rule_add_key_bit(st->vrule, VCAP_KF_L4_FIN, val);
                if (err)
                        goto out;
        }

        if (tcp_flags_mask & TCPHDR_SYN) {
                val = VCAP_BIT_0;
                if (tcp_flags_key & TCPHDR_SYN)
                        val = VCAP_BIT_1;
                err = vcap_rule_add_key_bit(st->vrule, VCAP_KF_L4_SYN, val);
                if (err)
                        goto out;
        }

        if (tcp_flags_mask & TCPHDR_RST) {
                val = VCAP_BIT_0;
                if (tcp_flags_key & TCPHDR_RST)
                        val = VCAP_BIT_1;
                err = vcap_rule_add_key_bit(st->vrule, VCAP_KF_L4_RST, val);
                if (err)
                        goto out;
        }

        if (tcp_flags_mask & TCPHDR_PSH) {
                val = VCAP_BIT_0;
                if (tcp_flags_key & TCPHDR_PSH)
                        val = VCAP_BIT_1;
                err = vcap_rule_add_key_bit(st->vrule, VCAP_KF_L4_PSH, val);
                if (err)
                        goto out;
        }

        if (tcp_flags_mask & TCPHDR_ACK) {
                val = VCAP_BIT_0;
                if (tcp_flags_key & TCPHDR_ACK)
                        val = VCAP_BIT_1;
                err = vcap_rule_add_key_bit(st->vrule, VCAP_KF_L4_ACK, val);
                if (err)
                        goto out;
        }

        if (tcp_flags_mask & TCPHDR_URG) {
                val = VCAP_BIT_0;
                if (tcp_flags_key & TCPHDR_URG)
                        val = VCAP_BIT_1;
                err = vcap_rule_add_key_bit(st->vrule, VCAP_KF_L4_URG, val);
                if (err)
                        goto out;
        }

        st->used_keys |= BIT_ULL(FLOW_DISSECTOR_KEY_TCP);

        return err;

out:
        NL_SET_ERR_MSG_MOD(st->fco->common.extack, "tcp_flags parse error");
        return err;
}
EXPORT_SYMBOL_GPL(vcap_tc_flower_handler_tcp_usage);

int vcap_tc_flower_handler_arp_usage(struct vcap_tc_flower_parse_usage *st)
{
        struct flow_match_arp mt;
        u16 value, mask;
        u32 ipval, ipmsk;
        int err;

        flow_rule_match_arp(st->frule, &mt);

        if (mt.mask->op) {
                mask = 0x3;
                if (st->l3_proto == ETH_P_ARP) {
                        value = mt.key->op == VCAP_ARP_OP_REQUEST ?
                                        VCAP_IS2_ARP_REQUEST :
                                        VCAP_IS2_ARP_REPLY;
                } else { /* RARP */
                        value = mt.key->op == VCAP_ARP_OP_REQUEST ?
                                        VCAP_IS2_RARP_REQUEST :
                                        VCAP_IS2_RARP_REPLY;
                }
                err = vcap_rule_add_key_u32(st->vrule, VCAP_KF_ARP_OPCODE,
                                            value, mask);
                if (err)
                        goto out;
        }

        /* The IS2 ARP keyset does not support ARP hardware addresses */
        if (!is_zero_ether_addr(mt.mask->sha) ||
            !is_zero_ether_addr(mt.mask->tha)) {
                err = -EINVAL;
                goto out;
        }

        if (mt.mask->sip) {
                ipval = be32_to_cpu((__force __be32)mt.key->sip);
                ipmsk = be32_to_cpu((__force __be32)mt.mask->sip);

                err = vcap_rule_add_key_u32(st->vrule, VCAP_KF_L3_IP4_SIP,
                                            ipval, ipmsk);
                if (err)
                        goto out;
        }

        if (mt.mask->tip) {
                ipval = be32_to_cpu((__force __be32)mt.key->tip);
                ipmsk = be32_to_cpu((__force __be32)mt.mask->tip);

                err = vcap_rule_add_key_u32(st->vrule, VCAP_KF_L3_IP4_DIP,
                                            ipval, ipmsk);
                if (err)
                        goto out;
        }

        st->used_keys |= BIT_ULL(FLOW_DISSECTOR_KEY_ARP);

        return 0;

out:
        NL_SET_ERR_MSG_MOD(st->fco->common.extack, "arp parse error");
        return err;
}
EXPORT_SYMBOL_GPL(vcap_tc_flower_handler_arp_usage);

int vcap_tc_flower_handler_ip_usage(struct vcap_tc_flower_parse_usage *st)
{
        struct flow_match_ip mt;
        int err = 0;

        flow_rule_match_ip(st->frule, &mt);

        if (mt.mask->tos) {
                err = vcap_rule_add_key_u32(st->vrule, VCAP_KF_L3_TOS,
                                            mt.key->tos,
                                            mt.mask->tos);
                if (err)
                        goto out;
        }

        st->used_keys |= BIT_ULL(FLOW_DISSECTOR_KEY_IP);

        return err;

out:
        NL_SET_ERR_MSG_MOD(st->fco->common.extack, "ip_tos parse error");
        return err;
}
EXPORT_SYMBOL_GPL(vcap_tc_flower_handler_ip_usage);