numam-dpdk/app/test/test_cksum.c
Olivier Matz 30a65c6cb7 test/cksum: add L3/L4 checksum API test
Add a simple unit test for checksum API.

Signed-off-by: Olivier Matz <olivier.matz@6wind.com>
Reviewed-by: Andrew Rybchenko <andrew.rybchenko@oktetlabs.ru>
2021-07-02 19:03:03 +02:00

272 lines
7.7 KiB
C

/* SPDX-License-Identifier: BSD-3-Clause
* Copyright 2021 6WIND S.A.
*/
#include <string.h>
#include <stdio.h>
#include <stdlib.h>
#include <stdint.h>
#include <rte_net.h>
#include <rte_mbuf.h>
#include <rte_ip.h>
#include "test.h"
#define MEMPOOL_CACHE_SIZE 0
#define MBUF_DATA_SIZE 256
#define NB_MBUF 128
/*
* Test L3/L4 checksum API.
*/
#define GOTO_FAIL(str, ...) do { \
printf("cksum test FAILED (l.%d): <" str ">\n", \
__LINE__, ##__VA_ARGS__); \
goto fail; \
} while (0)
/* generated in scapy with Ether()/IP()/TCP())) */
static const char test_cksum_ipv4_tcp[] = {
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x08, 0x00, 0x45, 0x00,
0x00, 0x28, 0x00, 0x01, 0x00, 0x00, 0x40, 0x06,
0x7c, 0xcd, 0x7f, 0x00, 0x00, 0x01, 0x7f, 0x00,
0x00, 0x01, 0x00, 0x14, 0x00, 0x50, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x50, 0x02,
0x20, 0x00, 0x91, 0x7c, 0x00, 0x00,
};
/* generated in scapy with Ether()/IPv6()/TCP()) */
static const char test_cksum_ipv6_tcp[] = {
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x08, 0x00, 0x60, 0x00,
0x00, 0x00, 0x00, 0x14, 0x06, 0x40, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0x14,
0x00, 0x50, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x50, 0x02, 0x20, 0x00, 0x8f, 0x7d,
0x00, 0x00,
};
/* generated in scapy with Ether()/IP()/UDP()/Raw('x')) */
static const char test_cksum_ipv4_udp[] = {
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x08, 0x00, 0x45, 0x00,
0x00, 0x1d, 0x00, 0x01, 0x00, 0x00, 0x40, 0x11,
0x7c, 0xcd, 0x7f, 0x00, 0x00, 0x01, 0x7f, 0x00,
0x00, 0x01, 0x00, 0x35, 0x00, 0x35, 0x00, 0x09,
0x89, 0x6f, 0x78,
};
/* generated in scapy with Ether()/IPv6()/UDP()/Raw('x')) */
static const char test_cksum_ipv6_udp[] = {
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x86, 0xdd, 0x60, 0x00,
0x00, 0x00, 0x00, 0x09, 0x11, 0x40, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0x35,
0x00, 0x35, 0x00, 0x09, 0x87, 0x70, 0x78,
};
/* generated in scapy with Ether()/IP(options='\x00')/UDP()/Raw('x')) */
static const char test_cksum_ipv4_opts_udp[] = {
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x08, 0x00, 0x46, 0x00,
0x00, 0x21, 0x00, 0x01, 0x00, 0x00, 0x40, 0x11,
0x7b, 0xc9, 0x7f, 0x00, 0x00, 0x01, 0x7f, 0x00,
0x00, 0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x35,
0x00, 0x35, 0x00, 0x09, 0x89, 0x6f, 0x78,
};
/* test l3/l4 checksum api */
static int
test_l4_cksum(struct rte_mempool *pktmbuf_pool, const char *pktdata, size_t len)
{
struct rte_net_hdr_lens hdr_lens;
struct rte_mbuf *m = NULL;
uint32_t packet_type;
uint16_t prev_cksum;
void *l3_hdr;
void *l4_hdr;
uint32_t l3;
uint32_t l4;
char *data;
m = rte_pktmbuf_alloc(pktmbuf_pool);
if (m == NULL)
GOTO_FAIL("Cannot allocate mbuf");
data = rte_pktmbuf_append(m, len);
if (data == NULL)
GOTO_FAIL("Cannot append data");
memcpy(data, pktdata, len);
packet_type = rte_net_get_ptype(m, &hdr_lens, RTE_PTYPE_ALL_MASK);
l3 = packet_type & RTE_PTYPE_L3_MASK;
l4 = packet_type & RTE_PTYPE_L4_MASK;
l3_hdr = rte_pktmbuf_mtod_offset(m, void *, hdr_lens.l2_len);
l4_hdr = rte_pktmbuf_mtod_offset(m, void *,
hdr_lens.l2_len + hdr_lens.l3_len);
if (l3 == RTE_PTYPE_L3_IPV4 || l3 == RTE_PTYPE_L3_IPV4_EXT) {
struct rte_ipv4_hdr *ip = l3_hdr;
/* verify IPv4 checksum */
if (rte_ipv4_cksum(l3_hdr) != 0)
GOTO_FAIL("invalid IPv4 checksum verification");
/* verify bad IPv4 checksum */
ip->hdr_checksum++;
if (rte_ipv4_cksum(l3_hdr) == 0)
GOTO_FAIL("invalid IPv4 bad checksum verification");
ip->hdr_checksum--;
/* recalculate IPv4 checksum */
prev_cksum = ip->hdr_checksum;
ip->hdr_checksum = 0;
ip->hdr_checksum = rte_ipv4_cksum(ip);
if (ip->hdr_checksum != prev_cksum)
GOTO_FAIL("invalid IPv4 checksum calculation");
/* verify L4 checksum */
if (rte_ipv4_udptcp_cksum_verify(l3_hdr, l4_hdr) != 0)
GOTO_FAIL("invalid L4 checksum verification");
if (l4 == RTE_PTYPE_L4_TCP) {
struct rte_tcp_hdr *tcp = l4_hdr;
/* verify bad TCP checksum */
tcp->cksum++;
if (rte_ipv4_udptcp_cksum_verify(l3_hdr, l4_hdr) == 0)
GOTO_FAIL("invalid bad TCP checksum verification");
tcp->cksum--;
/* recalculate TCP checksum */
prev_cksum = tcp->cksum;
tcp->cksum = 0;
tcp->cksum = rte_ipv4_udptcp_cksum(l3_hdr, l4_hdr);
if (tcp->cksum != prev_cksum)
GOTO_FAIL("invalid TCP checksum calculation");
} else if (l4 == RTE_PTYPE_L4_UDP) {
struct rte_udp_hdr *udp = l4_hdr;
/* verify bad UDP checksum */
udp->dgram_cksum++;
if (rte_ipv4_udptcp_cksum_verify(l3_hdr, l4_hdr) == 0)
GOTO_FAIL("invalid bad UDP checksum verification");
udp->dgram_cksum--;
/* recalculate UDP checksum */
prev_cksum = udp->dgram_cksum;
udp->dgram_cksum = 0;
udp->dgram_cksum = rte_ipv4_udptcp_cksum(l3_hdr,
l4_hdr);
if (udp->dgram_cksum != prev_cksum)
GOTO_FAIL("invalid TCP checksum calculation");
}
} else if (l3 == RTE_PTYPE_L3_IPV6 || l3 == RTE_PTYPE_L3_IPV6_EXT) {
if (rte_ipv6_udptcp_cksum_verify(l3_hdr, l4_hdr) != 0)
GOTO_FAIL("invalid L4 checksum verification");
if (l4 == RTE_PTYPE_L4_TCP) {
struct rte_tcp_hdr *tcp = l4_hdr;
/* verify bad TCP checksum */
tcp->cksum++;
if (rte_ipv6_udptcp_cksum_verify(l3_hdr, l4_hdr) == 0)
GOTO_FAIL("invalid bad TCP checksum verification");
tcp->cksum--;
/* recalculate TCP checksum */
prev_cksum = tcp->cksum;
tcp->cksum = 0;
tcp->cksum = rte_ipv6_udptcp_cksum(l3_hdr, l4_hdr);
if (tcp->cksum != prev_cksum)
GOTO_FAIL("invalid TCP checksum calculation");
} else if (l4 == RTE_PTYPE_L4_UDP) {
struct rte_udp_hdr *udp = l4_hdr;
/* verify bad UDP checksum */
udp->dgram_cksum++;
if (rte_ipv6_udptcp_cksum_verify(l3_hdr, l4_hdr) == 0)
GOTO_FAIL("invalid bad UDP checksum verification");
udp->dgram_cksum--;
/* recalculate UDP checksum */
prev_cksum = udp->dgram_cksum;
udp->dgram_cksum = 0;
udp->dgram_cksum = rte_ipv6_udptcp_cksum(l3_hdr,
l4_hdr);
if (udp->dgram_cksum != prev_cksum)
GOTO_FAIL("invalid TCP checksum calculation");
}
}
rte_pktmbuf_free(m);
return 0;
fail:
if (m)
rte_pktmbuf_free(m);
return -1;
}
static int
test_cksum(void)
{
struct rte_mempool *pktmbuf_pool = NULL;
/* create pktmbuf pool if it does not exist */
pktmbuf_pool = rte_pktmbuf_pool_create("test_cksum_mbuf_pool",
NB_MBUF, MEMPOOL_CACHE_SIZE, 0, MBUF_DATA_SIZE,
SOCKET_ID_ANY);
if (pktmbuf_pool == NULL)
GOTO_FAIL("cannot allocate mbuf pool");
if (test_l4_cksum(pktmbuf_pool, test_cksum_ipv4_tcp,
sizeof(test_cksum_ipv4_tcp)) < 0)
GOTO_FAIL("checksum error on ipv4_tcp");
if (test_l4_cksum(pktmbuf_pool, test_cksum_ipv6_tcp,
sizeof(test_cksum_ipv6_tcp)) < 0)
GOTO_FAIL("checksum error on ipv6_tcp");
if (test_l4_cksum(pktmbuf_pool, test_cksum_ipv4_udp,
sizeof(test_cksum_ipv4_udp)) < 0)
GOTO_FAIL("checksum error on ipv4_udp");
if (test_l4_cksum(pktmbuf_pool, test_cksum_ipv6_udp,
sizeof(test_cksum_ipv6_udp)) < 0)
GOTO_FAIL("checksum error on ipv6_udp");
if (test_l4_cksum(pktmbuf_pool, test_cksum_ipv4_opts_udp,
sizeof(test_cksum_ipv4_opts_udp)) < 0)
GOTO_FAIL("checksum error on ipv4_opts_udp");
rte_mempool_free(pktmbuf_pool);
return 0;
fail:
rte_mempool_free(pktmbuf_pool);
return -1;
}
#undef GOTO_FAIL
REGISTER_TEST_COMMAND(cksum_autotest, test_cksum);