// SPDX-License-Identifier: (GPL-2.0-only OR BSD-2-Clause)
/* Copyright (C) 2018 Netronome Systems, Inc */
/* Copyright (C) 2021 Corigine, Inc */

#include <net/xfrm.h>

#include "../nfp_net.h"
#include "nfd3.h"

void nfp_nfd3_ipsec_tx(struct nfp_nfd3_tx_desc *txd, struct sk_buff *skb)
{
        struct xfrm_state *x = xfrm_input_state(skb);
        struct xfrm_offload *xo = xfrm_offload(skb);
        struct iphdr *iph = ip_hdr(skb);
        int l4_proto;

        if (x->xso.dev && (x->xso.dev->features & NETIF_F_HW_ESP_TX_CSUM)) {
                txd->flags |= NFD3_DESC_TX_CSUM;

                if (iph->version == 4)
                        txd->flags |= NFD3_DESC_TX_IP4_CSUM;

                if (x->props.mode == XFRM_MODE_TRANSPORT)
                        l4_proto = xo->proto;
                else if (x->props.mode == XFRM_MODE_TUNNEL)
                        l4_proto = xo->inner_ipproto;
                else
                        return;

                switch (l4_proto) {
                case IPPROTO_UDP:
                        txd->flags |= NFD3_DESC_TX_UDP_CSUM;
                        return;
                case IPPROTO_TCP:
                        txd->flags |= NFD3_DESC_TX_TCP_CSUM;
                        return;
                }
        }
}