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>

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

int
rte_net_skip_ip6_ext(uint16_t proto, const struct rte_mbuf *m, uint32_t *off,
    int *frag);

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)
{
    const uint64_t inner_requests = RTE_MBUF_F_TX_IP_CKSUM | RTE_MBUF_F_TX_L4_MASK |
        RTE_MBUF_F_TX_TCP_SEG | RTE_MBUF_F_TX_UDP_SEG;
    const uint64_t outer_requests = RTE_MBUF_F_TX_OUTER_IP_CKSUM |
        RTE_MBUF_F_TX_OUTER_UDP_CKSUM;
    /* Initialise ipv4_hdr to avoid false positive compiler warnings. */
    struct rte_ipv4_hdr *ipv4_hdr = NULL;
    struct rte_ipv6_hdr *ipv6_hdr;
    struct rte_tcp_hdr *tcp_hdr;
    struct rte_udp_hdr *udp_hdr;
    uint64_t inner_l3_offset = m->l2_len;

    /*
     * Does packet set any of available offloads?
     * Mainly it is required to avoid fragmented headers check if
     * no offloads are requested.
     */
    if (!(ol_flags & (inner_requests | outer_requests)))
        return 0;

    if (ol_flags & (RTE_MBUF_F_TX_OUTER_IPV4 | RTE_MBUF_F_TX_OUTER_IPV6)) {
        inner_l3_offset += m->outer_l2_len + m->outer_l3_len;
        /*
         * prepare outer IPv4 header checksum by setting it to 0,
         * in order to be computed by hardware NICs.
         */
        if (ol_flags & RTE_MBUF_F_TX_OUTER_IP_CKSUM) {
            ipv4_hdr = rte_pktmbuf_mtod_offset(m,
                    struct rte_ipv4_hdr *, m->outer_l2_len);
            ipv4_hdr->hdr_checksum = 0;
        }
        if (ol_flags & RTE_MBUF_F_TX_OUTER_UDP_CKSUM || ol_flags & inner_requests) {
            if (ol_flags & RTE_MBUF_F_TX_OUTER_IPV4) {
                ipv4_hdr = rte_pktmbuf_mtod_offset(m, struct rte_ipv4_hdr *,
                    m->outer_l2_len);
                udp_hdr = (struct rte_udp_hdr *)((char *)ipv4_hdr +
                    m->outer_l3_len);
                if (ol_flags & RTE_MBUF_F_TX_OUTER_UDP_CKSUM)
                    udp_hdr->dgram_cksum = rte_ipv4_phdr_cksum(ipv4_hdr,
                        m->ol_flags);
                else if (ipv4_hdr->next_proto_id == IPPROTO_UDP)
                    udp_hdr->dgram_cksum = 0;
            } else {
                ipv6_hdr = rte_pktmbuf_mtod_offset(m, struct rte_ipv6_hdr *,
                    m->outer_l2_len);
                udp_hdr = rte_pktmbuf_mtod_offset(m, struct rte_udp_hdr *,
                     m->outer_l2_len + m->outer_l3_len);
                if (ol_flags & RTE_MBUF_F_TX_OUTER_UDP_CKSUM)
                    udp_hdr->dgram_cksum = rte_ipv6_phdr_cksum(ipv6_hdr,
                        m->ol_flags);
                else if (ipv6_hdr->proto == IPPROTO_UDP)
                    udp_hdr->dgram_cksum = 0;
            }
        }
    }

    /*
     * Check if headers are fragmented.
     * The check could be less strict depending on which offloads are
     * requested and headers to be used, but let's keep it simple.
     */
    if (unlikely(rte_pktmbuf_data_len(m) <
             inner_l3_offset + m->l3_len + m->l4_len))
        return -ENOTSUP;

    if (ol_flags & RTE_MBUF_F_TX_IPV4) {
        ipv4_hdr = rte_pktmbuf_mtod_offset(m, struct rte_ipv4_hdr *,
                inner_l3_offset);

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

    if ((ol_flags & RTE_MBUF_F_TX_L4_MASK) == RTE_MBUF_F_TX_UDP_CKSUM ||
            (ol_flags & RTE_MBUF_F_TX_UDP_SEG)) {
        if (ol_flags & RTE_MBUF_F_TX_IPV4) {
            udp_hdr = (struct rte_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 rte_ipv6_hdr *, inner_l3_offset);
            /* non-TSO udp */
            udp_hdr = rte_pktmbuf_mtod_offset(m,
                    struct rte_udp_hdr *,
                    inner_l3_offset + m->l3_len);
            udp_hdr->dgram_cksum = rte_ipv6_phdr_cksum(ipv6_hdr,
                    ol_flags);
        }
    } else if ((ol_flags & RTE_MBUF_F_TX_L4_MASK) == RTE_MBUF_F_TX_TCP_CKSUM ||
            (ol_flags & RTE_MBUF_F_TX_TCP_SEG)) {
        if (ol_flags & RTE_MBUF_F_TX_IPV4) {
            /* non-TSO tcp or TSO */
            tcp_hdr = (struct rte_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 rte_ipv6_hdr *, inner_l3_offset);
            /* non-TSO tcp or TSO */
            tcp_hdr = rte_pktmbuf_mtod_offset(m,
                    struct rte_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_ */