W dniu 17.04.2020 o 15:14, Aaron Conole pisze:
> Initial IP fragmentation unit test.
>
> Signed-off-by: Aaron Conole <acon...@redhat.com>
> ---
> MAINTAINERS | 1 +
> app/test/meson.build | 2 +
> app/test/test_ipfrag.c | 272 +++++++++++++++++++++++++++++++++++++++++
> 3 files changed, 275 insertions(+)
> create mode 100644 app/test/test_ipfrag.c
>
> diff --git a/MAINTAINERS b/MAINTAINERS
> index fe59f0224f..a77c7c17ce 100644
> --- a/MAINTAINERS
> +++ b/MAINTAINERS
> @@ -1228,6 +1228,7 @@ F: app/test/test_crc.c
> IP fragmentation & reassembly
> M: Konstantin Ananyev <konstantin.anan...@intel.com>
> F: lib/librte_ip_frag/
> +F: app/test/test_ipfrag.c
> F: doc/guides/prog_guide/ip_fragment_reassembly_lib.rst
> F: examples/ip_fragmentation/
> F: doc/guides/sample_app_ug/ip_frag.rst
> diff --git a/app/test/meson.build b/app/test/meson.build
> index 04b59cffa4..4b3c3852a2 100644
> --- a/app/test/meson.build
> +++ b/app/test/meson.build
> @@ -58,6 +58,7 @@ test_sources = files('commands.c',
> 'test_hash_perf.c',
> 'test_hash_readwrite_lf_perf.c',
> 'test_interrupts.c',
> + 'test_ipfrag.c',
> 'test_ipsec.c',
> 'test_ipsec_sad.c',
> 'test_kni.c',
> @@ -187,6 +188,7 @@ fast_tests = [
> ['flow_classify_autotest', false],
> ['hash_autotest', true],
> ['interrupt_autotest', true],
> + ['ipfrag_autotest', false],
> ['logs_autotest', true],
> ['lpm_autotest', true],
> ['lpm6_autotest', true],
> diff --git a/app/test/test_ipfrag.c b/app/test/test_ipfrag.c
> new file mode 100644
> index 0000000000..6a80581f0e
> --- /dev/null
> +++ b/app/test/test_ipfrag.c
> @@ -0,0 +1,272 @@
> +/* SPDX-License-Identifier: BSD-3-Clause
> + * Copyright(c) 2020 Red Hat, Inc.
> + */
> +
> +#include <time.h>
> +
> +#include <rte_common.h>
> +#include <rte_cycles.h>
> +#include <rte_hexdump.h>
> +#include <rte_ip.h>
> +#include <rte_ip_frag.h>
> +#include <rte_mbuf.h>
> +#include <rte_memcpy.h>
> +#include <rte_random.h>
> +
> +#include "test.h"
> +
> +#define NUM_MBUFS 128
> +#define BURST 32
> +
> +static struct rte_mempool *pkt_pool,
> + *direct_pool,
> + *indirect_pool;
> +
> +static int
> +setup_buf_pool(void)
> +{
> + if (!pkt_pool)
> + pkt_pool = rte_pktmbuf_pool_create("FRAG_MBUF_POOL",
> + NUM_MBUFS, BURST, 0,
> + RTE_MBUF_DEFAULT_BUF_SIZE,
> + SOCKET_ID_ANY);
> + if (pkt_pool == NULL) {
> + printf("%s: Error creating pkt mempool\n", __func__);
> + goto bad_setup;
> + }
> +
> + if (!direct_pool)
> + direct_pool = rte_pktmbuf_pool_create("FRAG_D_MBUF_POOL",
> + NUM_MBUFS, BURST, 0,
> + RTE_MBUF_DEFAULT_BUF_SIZE,
> + SOCKET_ID_ANY);
> + if (!direct_pool) {
> + printf("%s: Error creating direct mempool\n", __func__);
> + goto bad_setup;
> + }
> +
> + if (!indirect_pool)
> + indirect_pool = rte_pktmbuf_pool_create("FRAG_I_MBUF_POOL",
> + NUM_MBUFS, BURST, 0,
> + 0, SOCKET_ID_ANY);
> + if (!indirect_pool) {
> + printf("%s: Error creating indirect mempool\n", __func__);
> + goto bad_setup;
> + }
> +
> + return TEST_SUCCESS;
> +
> +bad_setup:
> + if (pkt_pool)
> + rte_mempool_free(pkt_pool);
> + pkt_pool = NULL;
> +
> + if (direct_pool)
> + rte_mempool_free(direct_pool);
> + direct_pool = NULL;
> +
> + return TEST_FAILED;
> +}
> +
> +static int testsuite_setup(void)
> +{
> + return setup_buf_pool();
> +}
> +
> +static void testsuite_teardown(void)
> +{
> + if (pkt_pool)
> + rte_mempool_free(pkt_pool);
> +
> + if (direct_pool)
> + rte_mempool_free(direct_pool);
> +
> + if (indirect_pool)
> + rte_mempool_free(indirect_pool);
> +
> + pkt_pool = NULL;
> + direct_pool = NULL;
> + indirect_pool = NULL;
> +}
> +
> +static int ut_setup(void)
> +{
> + return TEST_SUCCESS;
> +}
> +
> +static void ut_teardown(void)
> +{
> +}
> +
> +static void
> +v4_allocate_packet_of(struct rte_mbuf *b, int fill, size_t s, int df,
> + uint8_t ttl, uint8_t proto, uint16_t pktid)
> +{
> + /* Create a packet, 2k bytes long */
> + b->data_off = 0;
> + char *data = rte_pktmbuf_mtod(b, char *);
> +
> + memset(data, fill, sizeof(struct rte_ipv4_hdr) + s);
> +
> + struct rte_ipv4_hdr *hdr = (struct rte_ipv4_hdr *)data;
> +
> + hdr->version_ihl = 0x45; /* standard IP header... */
> + hdr->type_of_service = 0;
> + b->pkt_len = s + sizeof(struct rte_ipv4_hdr);
> + b->data_len = b->pkt_len;
> + hdr->total_length = rte_cpu_to_be_16(b->pkt_len);
> + hdr->packet_id = rte_cpu_to_be_16(pktid);
> + hdr->fragment_offset = 0;
> + if (df)
> + hdr->fragment_offset = rte_cpu_to_be_16(0x4000);
> +
> + if (!ttl)
> + ttl = 64; /* default to 64 */
> +
> + if (!proto)
> + proto = 1; /* icmp */
> +
> + hdr->time_to_live = ttl;
> + hdr->next_proto_id = proto;
> + hdr->hdr_checksum = 0;
> + hdr->src_addr = rte_cpu_to_be_32(0x8080808);
> + hdr->dst_addr = rte_cpu_to_be_32(0x8080404);
> +}
> +
> +static void
> +v6_allocate_packet_of(struct rte_mbuf *b, int fill, size_t s, uint8_t ttl,
> + uint8_t proto, uint16_t pktid)
> +{
> + /* Create a packet, 2k bytes long */
> + b->data_off = 0;
> + char *data = rte_pktmbuf_mtod(b, char *);
> +
> + memset(data, fill, sizeof(struct rte_ipv6_hdr) + s);
> +
> + struct rte_ipv6_hdr *hdr = (struct rte_ipv6_hdr *)data;
> + b->pkt_len = s + sizeof(struct rte_ipv6_hdr);
> + b->data_len = b->pkt_len;
> +
> + /* basic v6 header */
> + hdr->vtc_flow = rte_cpu_to_be_32(0x60 << 24 | pktid);
> + hdr->payload_len = rte_cpu_to_be_16(b->pkt_len);
> + hdr->proto = proto;
> + hdr->hop_limits = ttl;
> +
> + memset(hdr->src_addr, 0x08, sizeof(hdr->src_addr));
> + memset(hdr->dst_addr, 0x04, sizeof(hdr->src_addr));
> +}
> +
> +static inline void
> +test_free_fragments(struct rte_mbuf *mb[], uint32_t num)
> +{
> + uint32_t i;
> + for (i = 0; i < num; i++)
> + rte_pktmbuf_free(mb[i]);
> +}
> +
> +static int
> +test_ip_frag(void)
> +{
> + static const uint16_t RND_ID = UINT16_MAX;
> + int result = TEST_SUCCESS;
> + size_t i;
> +
> + struct test_ip_frags {
> + int ipv;
> + size_t mtu_size;
> + size_t pkt_size;
> + int set_df;
> + uint8_t ttl;
> + uint8_t proto;
> + uint16_t pkt_id;
> + int expected_frags;
> + } tests[] = {
> + {4, 1280, 1400, 0, 64, IPPROTO_ICMP, RND_ID, 2},
> + {4, 1280, 1400, 0, 64, IPPROTO_ICMP, 0, 2},
> + {4, 600, 1400, 0, 64, IPPROTO_ICMP, RND_ID, 3},
> + {4, 4, 1400, 0, 64, IPPROTO_ICMP, RND_ID, -EINVAL},
> + {4, 600, 1400, 1, 64, IPPROTO_ICMP, RND_ID, -ENOTSUP},
> + {4, 600, 1400, 0, 0, IPPROTO_ICMP, RND_ID, 3},
> +
> + {6, 1280, 1400, 0, 64, IPPROTO_ICMP, RND_ID, 2},
> + {6, 1300, 1400, 0, 64, IPPROTO_ICMP, RND_ID, 2},
> + {6, 4, 1400, 0, 64, IPPROTO_ICMP, RND_ID, -EINVAL},
> + {6, 1300, 1400, 0, 0, IPPROTO_ICMP, RND_ID, 2},
> + };
> +
> + for (i = 0; i < RTE_DIM(tests); i++) {
> + int32_t len = 0;
> + uint16_t pktid = tests[i].pkt_id;
> + struct rte_mbuf *pkts_out[BURST];
> + struct rte_mbuf *b = rte_pktmbuf_alloc(pkt_pool);
> +
> + RTE_TEST_ASSERT_NOT_EQUAL(b, NULL,
> + "Failed to allocate pkt.");
> +
> + if (tests[i].pkt_id == RND_ID)
> + pktid = rte_rand_max(UINT16_MAX);
> +
> + if (tests[i].ipv == 4) {
> + v4_allocate_packet_of(b, 0x41414141,
> + tests[i].pkt_size,
> + tests[i].set_df,
> + tests[i].ttl,
> + tests[i].proto,
> + pktid);
> + } else if (tests[i].ipv == 6) {
> + v6_allocate_packet_of(b, 0x41414141,
> + tests[i].pkt_size,
> + tests[i].ttl,
> + tests[i].proto,
> + pktid);
> + }
> +
> + if (tests[i].ipv == 4)
> + len = rte_ipv4_fragment_packet(b, pkts_out, BURST,
> + tests[i].mtu_size,
> + direct_pool,
> + indirect_pool);
> + else if (tests[i].ipv == 6)
> + len = rte_ipv6_fragment_packet(b, pkts_out, BURST,
> + tests[i].mtu_size,
> + direct_pool,
> + indirect_pool);
> +
> + rte_pktmbuf_free(b);
> +
> + if (len > 0)
> + test_free_fragments(pkts_out, len);
> +
> + printf("%zd: checking %d with %d\n", i, len,
> + tests[i].expected_frags);
> + RTE_TEST_ASSERT_EQUAL(len, tests[i].expected_frags,
> + "Failed case %zd.\n", i);
> +
> + }
> +
> + return result;
> +}
> +
> +static struct unit_test_suite ipfrag_testsuite = {
> + .suite_name = "IP Frag Unit Test Suite",
> + .setup = testsuite_setup,
> + .teardown = testsuite_teardown,
> + .unit_test_cases = {
> + TEST_CASE_ST(ut_setup, ut_teardown,
> + test_ip_frag),
> +
> + TEST_CASES_END() /**< NULL terminate unit test array */
> + }
> +};
> +
> +static int
> +test_ipfrag(void)
> +{
> + rte_log_set_global_level(RTE_LOG_DEBUG);
> + rte_log_set_level(RTE_LOGTYPE_EAL, RTE_LOG_DEBUG);
> +
> + return unit_test_suite_runner(&ipfrag_testsuite);
> +}
> +
> +REGISTER_TEST_COMMAND(ipfrag_autotest, test_ipfrag);
Perfect!
Reviewed-by: Lukasz Wojciechowski <l.wojciec...@partner.samsung.com>
Tested-by: Lukasz Wojciechowski <l.wojciec...@partner.samsung.com>
--
Lukasz Wojciechowski
Principal Software Engineer
Samsung R&D Institute Poland
Samsung Electronics
Office +48 22 377 88 25
l.wojciec...@partner.samsung.com
