rte_thash.h

Go to the documentation of this file.

/* SPDX-License-Identifier: BSD-3-Clause
 * Copyright(c) 2015-2019 Vladimir Medvedkin <medvedkinv@gmail.com>
 */

#ifndef _RTE_THASH_H
#define _RTE_THASH_H

#ifdef __cplusplus
extern "C" {
#endif

#include <stdint.h>
#include <rte_byteorder.h>
#include <rte_config.h>
#include <rte_ip.h>
#include <rte_common.h>

#if defined(RTE_ARCH_X86) || defined(RTE_MACHINE_CPUFLAG_NEON)
#include <rte_vect.h>
#endif

#ifdef RTE_ARCH_X86
/* Byte swap mask used for converting IPv6 address
 * 4-byte chunks to CPU byte order
 */
static const __m128i rte_thash_ipv6_bswap_mask = {
                0x0405060700010203ULL, 0x0C0D0E0F08090A0BULL};
#endif

#define RTE_THASH_V4_L3_LEN     ((sizeof(struct rte_ipv4_tuple) -       \
                        sizeof(((struct rte_ipv4_tuple *)0)->sctp_tag)) / 4)

#define RTE_THASH_V4_L4_LEN      ((sizeof(struct rte_ipv4_tuple)) / 4)

#define RTE_THASH_V6_L3_LEN     ((sizeof(struct rte_ipv6_tuple) -       \
                        sizeof(((struct rte_ipv6_tuple *)0)->sctp_tag)) / 4)

#define RTE_THASH_V6_L4_LEN     ((sizeof(struct rte_ipv6_tuple)) / 4)

struct rte_ipv4_tuple {
        uint32_t        src_addr;
        uint32_t        dst_addr;
        RTE_STD_C11
        union {
                struct {
                        uint16_t dport;
                        uint16_t sport;
                };
                uint32_t        sctp_tag;
        };
};

struct rte_ipv6_tuple {
        uint8_t         src_addr[16];
        uint8_t         dst_addr[16];
        RTE_STD_C11
        union {
                struct {
                        uint16_t dport;
                        uint16_t sport;
                };
                uint32_t        sctp_tag;
        };
};

union rte_thash_tuple {
        struct rte_ipv4_tuple   v4;
        struct rte_ipv6_tuple   v6;
#ifdef RTE_ARCH_X86
} __attribute__((aligned(XMM_SIZE)));
#else
};
#endif

static inline void
rte_convert_rss_key(const uint32_t *orig, uint32_t *targ, int len)
{
        int i;

        for (i = 0; i < (len >> 2); i++)
                targ[i] = rte_be_to_cpu_32(orig[i]);
}

static inline void
rte_thash_load_v6_addrs(const struct ipv6_hdr *orig, union rte_thash_tuple *targ)
{
#ifdef RTE_ARCH_X86
        __m128i ipv6 = _mm_loadu_si128((const __m128i *)orig->src_addr);
        *(__m128i *)targ->v6.src_addr =
                        _mm_shuffle_epi8(ipv6, rte_thash_ipv6_bswap_mask);
        ipv6 = _mm_loadu_si128((const __m128i *)orig->dst_addr);
        *(__m128i *)targ->v6.dst_addr =
                        _mm_shuffle_epi8(ipv6, rte_thash_ipv6_bswap_mask);
#elif defined(RTE_MACHINE_CPUFLAG_NEON)
        uint8x16_t ipv6 = vld1q_u8((uint8_t const *)orig->src_addr);
        vst1q_u8((uint8_t *)targ->v6.src_addr, vrev32q_u8(ipv6));
        ipv6 = vld1q_u8((uint8_t const *)orig->dst_addr);
        vst1q_u8((uint8_t *)targ->v6.dst_addr, vrev32q_u8(ipv6));
#else
        int i;
        for (i = 0; i < 4; i++) {
                *((uint32_t *)targ->v6.src_addr + i) =
                        rte_be_to_cpu_32(*((const uint32_t *)orig->src_addr + i));
                *((uint32_t *)targ->v6.dst_addr + i) =
                        rte_be_to_cpu_32(*((const uint32_t *)orig->dst_addr + i));
        }
#endif
}

static inline uint32_t
rte_softrss(uint32_t *input_tuple, uint32_t input_len,
                const uint8_t *rss_key)
{
        uint32_t i, j, map, ret = 0;

        for (j = 0; j < input_len; j++) {
                for (map = input_tuple[j]; map; map &= (map - 1)) {
                        i = rte_bsf32(map);
                        ret ^= rte_cpu_to_be_32(((const uint32_t *)rss_key)[j]) << (31 - i) |
                                        (uint32_t)((uint64_t)(rte_cpu_to_be_32(((const uint32_t *)rss_key)[j + 1])) >>
                                        (i + 1));
                }
        }
        return ret;
}

static inline uint32_t
rte_softrss_be(uint32_t *input_tuple, uint32_t input_len,
                const uint8_t *rss_key)
{
        uint32_t i, j, map, ret = 0;

        for (j = 0; j < input_len; j++) {
                for (map = input_tuple[j]; map; map &= (map - 1)) {
                        i = rte_bsf32(map);
                        ret ^= ((const uint32_t *)rss_key)[j] << (31 - i) |
                                (uint32_t)((uint64_t)(((const uint32_t *)rss_key)[j + 1]) >> (i + 1));
                }
        }
        return ret;
}

#ifdef __cplusplus
}
#endif

#endif /* _RTE_THASH_H */