rte_net.h

/* SPDX-License-Identifier: BSD-3-Clause
 * Copyright 2016 6WIND S.A.
 */

#ifndef _RTE_NET_PTYPE_H_
#define _RTE_NET_PTYPE_H_

#ifdef __cplusplus
extern "C" {
#endif

#include <rte_ip.h>
#include <rte_udp.h>
#include <rte_tcp.h>
#include <rte_sctp.h>

struct rte_net_hdr_lens {
        uint8_t l2_len;
        uint8_t l3_len;
        uint8_t l4_len;
        uint8_t tunnel_len;
        uint8_t inner_l2_len;
        uint8_t inner_l3_len;
        uint8_t inner_l4_len;
};

uint32_t rte_net_get_ptype(const struct rte_mbuf *m,
        struct rte_net_hdr_lens *hdr_lens, uint32_t layers);

static inline int
rte_net_intel_cksum_flags_prepare(struct rte_mbuf *m, uint64_t ol_flags)
{
        struct ipv4_hdr *ipv4_hdr;
        struct ipv6_hdr *ipv6_hdr;
        struct tcp_hdr *tcp_hdr;
        struct udp_hdr *udp_hdr;
        uint64_t inner_l3_offset = m->l2_len;

        if ((ol_flags & PKT_TX_OUTER_IP_CKSUM) ||
                (ol_flags & PKT_TX_OUTER_IPV6))
                inner_l3_offset += m->outer_l2_len + m->outer_l3_len;

        if ((ol_flags & PKT_TX_UDP_CKSUM) == PKT_TX_UDP_CKSUM) {
                if (ol_flags & PKT_TX_IPV4) {
                        ipv4_hdr = rte_pktmbuf_mtod_offset(m, struct ipv4_hdr *,
                                        inner_l3_offset);

                        if (ol_flags & PKT_TX_IP_CKSUM)
                                ipv4_hdr->hdr_checksum = 0;

                        udp_hdr = (struct udp_hdr *)((char *)ipv4_hdr +
                                        m->l3_len);
                        udp_hdr->dgram_cksum = rte_ipv4_phdr_cksum(ipv4_hdr,
                                        ol_flags);
                } else {
                        ipv6_hdr = rte_pktmbuf_mtod_offset(m, struct ipv6_hdr *,
                                        inner_l3_offset);
                        /* non-TSO udp */
                        udp_hdr = rte_pktmbuf_mtod_offset(m, struct udp_hdr *,
                                        inner_l3_offset + m->l3_len);
                        udp_hdr->dgram_cksum = rte_ipv6_phdr_cksum(ipv6_hdr,
                                        ol_flags);
                }
        } else if ((ol_flags & PKT_TX_TCP_CKSUM) ||
                        (ol_flags & PKT_TX_TCP_SEG)) {
                if (ol_flags & PKT_TX_IPV4) {
                        ipv4_hdr = rte_pktmbuf_mtod_offset(m, struct ipv4_hdr *,
                                        inner_l3_offset);

                        if (ol_flags & PKT_TX_IP_CKSUM)
                                ipv4_hdr->hdr_checksum = 0;

                        /* non-TSO tcp or TSO */
                        tcp_hdr = (struct tcp_hdr *)((char *)ipv4_hdr +
                                        m->l3_len);
                        tcp_hdr->cksum = rte_ipv4_phdr_cksum(ipv4_hdr,
                                        ol_flags);
                } else {
                        ipv6_hdr = rte_pktmbuf_mtod_offset(m, struct ipv6_hdr *,
                                        inner_l3_offset);
                        /* non-TSO tcp or TSO */
                        tcp_hdr = rte_pktmbuf_mtod_offset(m, struct tcp_hdr *,
                                        inner_l3_offset + m->l3_len);
                        tcp_hdr->cksum = rte_ipv6_phdr_cksum(ipv6_hdr,
                                        ol_flags);
                }
        }

        return 0;
}

static inline int
rte_net_intel_cksum_prepare(struct rte_mbuf *m)
{
        return rte_net_intel_cksum_flags_prepare(m, m->ol_flags);
}

#ifdef __cplusplus
}
#endif


#endif /* _RTE_NET_PTYPE_H_ */