test: move to app directory
[dpdk.git] / app / test / test_mbuf.c
index 8460db7..9e82a20 100644 (file)
@@ -1,34 +1,5 @@
-/*-
- *   BSD LICENSE
- *
- *   Copyright(c) 2010-2014 Intel Corporation. All rights reserved.
- *   All rights reserved.
- *
- *   Redistribution and use in source and binary forms, with or without
- *   modification, are permitted provided that the following conditions
- *   are met:
- *
- *     * Redistributions of source code must retain the above copyright
- *       notice, this list of conditions and the following disclaimer.
- *     * Redistributions in binary form must reproduce the above copyright
- *       notice, this list of conditions and the following disclaimer in
- *       the documentation and/or other materials provided with the
- *       distribution.
- *     * Neither the name of Intel Corporation nor the names of its
- *       contributors may be used to endorse or promote products derived
- *       from this software without specific prior written permission.
- *
- *   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
- *   "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
- *   LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
- *   A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
- *   OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
- *   SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
- *   LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
- *   DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
- *   THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
- *   (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
- *   OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+/* SPDX-License-Identifier: BSD-3-Clause
+ * Copyright(c) 2010-2014 Intel Corporation
  */
 
 #include <string.h>
 #include <rte_common.h>
 #include <rte_debug.h>
 #include <rte_log.h>
-#include <rte_common.h>
 #include <rte_memory.h>
 #include <rte_memcpy.h>
-#include <rte_memzone.h>
 #include <rte_launch.h>
 #include <rte_eal.h>
 #include <rte_per_lcore.h>
 
 #define MAKE_STRING(x)          # x
 
-static struct rte_mempool *pktmbuf_pool = NULL;
-static struct rte_mempool *pktmbuf_pool2 = NULL;
-
 #ifdef RTE_MBUF_REFCNT_ATOMIC
 
-static struct rte_mempool *refcnt_pool = NULL;
-static struct rte_ring *refcnt_mbuf_ring = NULL;
 static volatile uint32_t refcnt_stop_slaves;
 static unsigned refcnt_lcore[RTE_MAX_LCORE];
 
@@ -144,7 +108,7 @@ static unsigned refcnt_lcore[RTE_MAX_LCORE];
  * test data manipulation in mbuf with non-ascii data
  */
 static int
-test_pktmbuf_with_non_ascii_data(void)
+test_pktmbuf_with_non_ascii_data(struct rte_mempool *pktmbuf_pool)
 {
        struct rte_mbuf *m = NULL;
        char *data;
@@ -182,7 +146,7 @@ fail:
  * test data manipulation in mbuf
  */
 static int
-test_one_pktmbuf(void)
+test_one_pktmbuf(struct rte_mempool *pktmbuf_pool)
 {
        struct rte_mbuf *m = NULL;
        char *data, *data2, *hdr;
@@ -328,7 +292,7 @@ fail:
 }
 
 static int
-testclone_testupdate_testdetach(void)
+testclone_testupdate_testdetach(struct rte_mempool *pktmbuf_pool)
 {
        struct rte_mbuf *m = NULL;
        struct rte_mbuf *clone = NULL;
@@ -432,7 +396,8 @@ fail:
 }
 
 static int
-test_attach_from_different_pool(void)
+test_attach_from_different_pool(struct rte_mempool *pktmbuf_pool,
+                               struct rte_mempool *pktmbuf_pool2)
 {
        struct rte_mbuf *m = NULL;
        struct rte_mbuf *clone = NULL;
@@ -542,7 +507,7 @@ fail:
  * test allocation and free of mbufs
  */
 static int
-test_pktmbuf_pool(void)
+test_pktmbuf_pool(struct rte_mempool *pktmbuf_pool)
 {
        unsigned i;
        struct rte_mbuf *m[NB_MBUF];
@@ -583,7 +548,7 @@ test_pktmbuf_pool(void)
  * test that the pointer to the data on a packet mbuf is set properly
  */
 static int
-test_pktmbuf_pool_ptr(void)
+test_pktmbuf_pool_ptr(struct rte_mempool *pktmbuf_pool)
 {
        unsigned i;
        struct rte_mbuf *m[NB_MBUF];
@@ -636,7 +601,7 @@ test_pktmbuf_pool_ptr(void)
 }
 
 static int
-test_pktmbuf_free_segment(void)
+test_pktmbuf_free_segment(struct rte_mempool *pktmbuf_pool)
 {
        unsigned i;
        struct rte_mbuf *m[NB_MBUF];
@@ -674,16 +639,17 @@ test_pktmbuf_free_segment(void)
 /*
  * Stress test for rte_mbuf atomic refcnt.
  * Implies that RTE_MBUF_REFCNT_ATOMIC is defined.
- * For more efficency, recomended to run with RTE_LIBRTE_MBUF_DEBUG defined.
+ * For more efficiency, recommended to run with RTE_LIBRTE_MBUF_DEBUG defined.
  */
 
 #ifdef RTE_MBUF_REFCNT_ATOMIC
 
 static int
-test_refcnt_slave(__attribute__((unused)) void *arg)
+test_refcnt_slave(void *arg)
 {
        unsigned lcore, free;
        void *mp = 0;
+       struct rte_ring *refcnt_mbuf_ring = arg;
 
        lcore = rte_lcore_id();
        printf("%s started at lcore %u\n", __func__, lcore);
@@ -692,7 +658,7 @@ test_refcnt_slave(__attribute__((unused)) void *arg)
        while (refcnt_stop_slaves == 0) {
                if (rte_ring_dequeue(refcnt_mbuf_ring, &mp) == 0) {
                        free++;
-                       rte_pktmbuf_free((struct rte_mbuf *)mp);
+                       rte_pktmbuf_free(mp);
                }
        }
 
@@ -704,7 +670,9 @@ test_refcnt_slave(__attribute__((unused)) void *arg)
 }
 
 static void
-test_refcnt_iter(unsigned lcore, unsigned iter)
+test_refcnt_iter(unsigned int lcore, unsigned int iter,
+                struct rte_mempool *refcnt_pool,
+                struct rte_ring *refcnt_mbuf_ring)
 {
        uint16_t ref;
        unsigned i, n, tref, wn;
@@ -717,7 +685,7 @@ test_refcnt_iter(unsigned lcore, unsigned iter)
         * - increment it's reference up to N+1,
         * - enqueue it N times into the ring for slave cores to free.
         */
-       for (i = 0, n = rte_mempool_count(refcnt_pool);
+       for (i = 0, n = rte_mempool_avail_count(refcnt_pool);
            i != n && (m = rte_pktmbuf_alloc(refcnt_pool)) != NULL;
            i++) {
                ref = RTE_MAX(rte_rand() % REFCNT_MAX_REF, 1UL);
@@ -745,14 +713,14 @@ test_refcnt_iter(unsigned lcore, unsigned iter)
 
        /* check that all mbufs are back into mempool by now */
        for (wn = 0; wn != REFCNT_MAX_TIMEOUT; wn++) {
-               if ((i = rte_mempool_count(refcnt_pool)) == n) {
+               if ((i = rte_mempool_avail_count(refcnt_pool)) == n) {
                        refcnt_lcore[lcore] += tref;
                        printf("%s(lcore=%u, iter=%u) completed, "
                            "%u references processed\n",
                            __func__, lcore, iter, tref);
                        return;
                }
-               rte_delay_ms(1000);
+               rte_delay_ms(100);
        }
 
        rte_panic("(lcore=%u, iter=%u): after %us only "
@@ -760,7 +728,8 @@ test_refcnt_iter(unsigned lcore, unsigned iter)
 }
 
 static int
-test_refcnt_master(void)
+test_refcnt_master(struct rte_mempool *refcnt_pool,
+                  struct rte_ring *refcnt_mbuf_ring)
 {
        unsigned i, lcore;
 
@@ -768,7 +737,7 @@ test_refcnt_master(void)
        printf("%s started at lcore %u\n", __func__, lcore);
 
        for (i = 0; i != REFCNT_MAX_ITER; i++)
-               test_refcnt_iter(lcore, i);
+               test_refcnt_iter(lcore, i, refcnt_pool, refcnt_mbuf_ring);
 
        refcnt_stop_slaves = 1;
        rte_wmb();
@@ -783,9 +752,10 @@ static int
 test_refcnt_mbuf(void)
 {
 #ifdef RTE_MBUF_REFCNT_ATOMIC
-
        unsigned lnum, master, slave, tref;
-
+       int ret = -1;
+       struct rte_mempool *refcnt_pool = NULL;
+       struct rte_ring *refcnt_mbuf_ring = NULL;
 
        if ((lnum = rte_lcore_count()) == 1) {
                printf("skipping %s, number of lcores: %u is not enough\n",
@@ -797,31 +767,31 @@ test_refcnt_mbuf(void)
 
        /* create refcnt pool & ring if they don't exist */
 
-       if (refcnt_pool == NULL &&
-                       (refcnt_pool = rte_pktmbuf_pool_create(
-                               MAKE_STRING(refcnt_pool),
-                               REFCNT_MBUF_NUM, 0, 0, 0,
-                               SOCKET_ID_ANY)) == NULL) {
+       refcnt_pool = rte_pktmbuf_pool_create(MAKE_STRING(refcnt_pool),
+                                             REFCNT_MBUF_NUM, 0, 0, 0,
+                                             SOCKET_ID_ANY);
+       if (refcnt_pool == NULL) {
                printf("%s: cannot allocate " MAKE_STRING(refcnt_pool) "\n",
                    __func__);
                return -1;
        }
 
-       if (refcnt_mbuf_ring == NULL &&
-                       (refcnt_mbuf_ring = rte_ring_create("refcnt_mbuf_ring",
-                       REFCNT_RING_SIZE, SOCKET_ID_ANY,
-                       RING_F_SP_ENQ)) == NULL) {
+       refcnt_mbuf_ring = rte_ring_create("refcnt_mbuf_ring",
+                       rte_align32pow2(REFCNT_RING_SIZE), SOCKET_ID_ANY,
+                                       RING_F_SP_ENQ);
+       if (refcnt_mbuf_ring == NULL) {
                printf("%s: cannot allocate " MAKE_STRING(refcnt_mbuf_ring)
                    "\n", __func__);
-               return -1;
+               goto err;
        }
 
        refcnt_stop_slaves = 0;
        memset(refcnt_lcore, 0, sizeof (refcnt_lcore));
 
-       rte_eal_mp_remote_launch(test_refcnt_slave, NULL, SKIP_MASTER);
+       rte_eal_mp_remote_launch(test_refcnt_slave, refcnt_mbuf_ring,
+                                SKIP_MASTER);
 
-       test_refcnt_master();
+       test_refcnt_master(refcnt_pool, refcnt_mbuf_ring);
 
        rte_eal_mp_wait_lcore();
 
@@ -839,8 +809,15 @@ test_refcnt_mbuf(void)
        rte_mempool_dump(stdout, refcnt_pool);
        rte_ring_dump(stdout, refcnt_mbuf_ring);
 
-#endif
+       ret = 0;
+
+err:
+       rte_mempool_free(refcnt_pool);
+       rte_ring_free(refcnt_mbuf_ring);
+       return ret;
+#else
        return 0;
+#endif
 }
 
 #include <unistd.h>
@@ -870,7 +847,7 @@ verify_mbuf_check_panics(struct rte_mbuf *buf)
 }
 
 static int
-test_failing_mbuf_sanity_check(void)
+test_failing_mbuf_sanity_check(struct rte_mempool *pktmbuf_pool)
 {
        struct rte_mbuf *buf;
        struct rte_mbuf badbuf;
@@ -900,7 +877,7 @@ test_failing_mbuf_sanity_check(void)
        }
 
        badbuf = *buf;
-       badbuf.buf_physaddr = 0;
+       badbuf.buf_iova = 0;
        if (verify_mbuf_check_panics(&badbuf)) {
                printf("Error with bad-physaddr mbuf test\n");
                return -1;
@@ -930,57 +907,178 @@ test_failing_mbuf_sanity_check(void)
        return 0;
 }
 
+static int
+test_mbuf_linearize(struct rte_mempool *pktmbuf_pool, int pkt_len,
+                   int nb_segs)
+{
+
+       struct rte_mbuf *m = NULL, *mbuf = NULL;
+       uint8_t *data;
+       int data_len = 0;
+       int remain;
+       int seg, seg_len;
+       int i;
+
+       if (pkt_len < 1) {
+               printf("Packet size must be 1 or more (is %d)\n", pkt_len);
+               return -1;
+       }
+
+       if (nb_segs < 1) {
+               printf("Number of segments must be 1 or more (is %d)\n",
+                               nb_segs);
+               return -1;
+       }
+
+       seg_len = pkt_len / nb_segs;
+       if (seg_len == 0)
+               seg_len = 1;
+
+       remain = pkt_len;
+
+       /* Create chained mbuf_src and fill it generated data */
+       for (seg = 0; remain > 0; seg++) {
+
+               m = rte_pktmbuf_alloc(pktmbuf_pool);
+               if (m == NULL) {
+                       printf("Cannot create segment for source mbuf");
+                       goto fail;
+               }
+
+               /* Make sure if tailroom is zeroed */
+               memset(rte_pktmbuf_mtod(m, uint8_t *), 0,
+                               rte_pktmbuf_tailroom(m));
+
+               data_len = remain;
+               if (data_len > seg_len)
+                       data_len = seg_len;
+
+               data = (uint8_t *)rte_pktmbuf_append(m, data_len);
+               if (data == NULL) {
+                       printf("Cannot append %d bytes to the mbuf\n",
+                                       data_len);
+                       goto fail;
+               }
+
+               for (i = 0; i < data_len; i++)
+                       data[i] = (seg * seg_len + i) % 0x0ff;
+
+               if (seg == 0)
+                       mbuf = m;
+               else
+                       rte_pktmbuf_chain(mbuf, m);
+
+               remain -= data_len;
+       }
+
+       /* Create destination buffer to store coalesced data */
+       if (rte_pktmbuf_linearize(mbuf)) {
+               printf("Mbuf linearization failed\n");
+               goto fail;
+       }
+
+       if (!rte_pktmbuf_is_contiguous(mbuf)) {
+               printf("Source buffer should be contiguous after "
+                               "linearization\n");
+               goto fail;
+       }
+
+       data = rte_pktmbuf_mtod(mbuf, uint8_t *);
+
+       for (i = 0; i < pkt_len; i++)
+               if (data[i] != (i % 0x0ff)) {
+                       printf("Incorrect data in linearized mbuf\n");
+                       goto fail;
+               }
+
+       rte_pktmbuf_free(mbuf);
+       return 0;
+
+fail:
+       if (mbuf)
+               rte_pktmbuf_free(mbuf);
+       return -1;
+}
+
+static int
+test_mbuf_linearize_check(struct rte_mempool *pktmbuf_pool)
+{
+       struct test_mbuf_array {
+               int size;
+               int nb_segs;
+       } mbuf_array[] = {
+                       { 128, 1 },
+                       { 64, 64 },
+                       { 512, 10 },
+                       { 250, 11 },
+                       { 123, 8 },
+       };
+       unsigned int i;
+
+       printf("Test mbuf linearize API\n");
+
+       for (i = 0; i < RTE_DIM(mbuf_array); i++)
+               if (test_mbuf_linearize(pktmbuf_pool, mbuf_array[i].size,
+                               mbuf_array[i].nb_segs)) {
+                       printf("Test failed for %d, %d\n", mbuf_array[i].size,
+                                       mbuf_array[i].nb_segs);
+                       return -1;
+               }
+
+       return 0;
+}
 
 static int
 test_mbuf(void)
 {
+       int ret = -1;
+       struct rte_mempool *pktmbuf_pool = NULL;
+       struct rte_mempool *pktmbuf_pool2 = NULL;
+
+
        RTE_BUILD_BUG_ON(sizeof(struct rte_mbuf) != RTE_CACHE_LINE_MIN_SIZE * 2);
 
        /* create pktmbuf pool if it does not exist */
-       if (pktmbuf_pool == NULL) {
-               pktmbuf_pool = rte_pktmbuf_pool_create("test_pktmbuf_pool",
+       pktmbuf_pool = rte_pktmbuf_pool_create("test_pktmbuf_pool",
                        NB_MBUF, 32, 0, MBUF_DATA_SIZE, SOCKET_ID_ANY);
-       }
 
        if (pktmbuf_pool == NULL) {
                printf("cannot allocate mbuf pool\n");
-               return -1;
+               goto err;
        }
 
        /* create a specific pktmbuf pool with a priv_size != 0 and no data
         * room size */
-       if (pktmbuf_pool2 == NULL) {
-               pktmbuf_pool2 = rte_pktmbuf_pool_create("test_pktmbuf_pool2",
+       pktmbuf_pool2 = rte_pktmbuf_pool_create("test_pktmbuf_pool2",
                        NB_MBUF, 32, MBUF2_PRIV_SIZE, 0, SOCKET_ID_ANY);
-       }
 
        if (pktmbuf_pool2 == NULL) {
                printf("cannot allocate mbuf pool\n");
-               return -1;
+               goto err;
        }
 
        /* test multiple mbuf alloc */
-       if (test_pktmbuf_pool() < 0) {
+       if (test_pktmbuf_pool(pktmbuf_pool) < 0) {
                printf("test_mbuf_pool() failed\n");
-               return -1;
+               goto err;
        }
 
        /* do it another time to check that all mbufs were freed */
-       if (test_pktmbuf_pool() < 0) {
+       if (test_pktmbuf_pool(pktmbuf_pool) < 0) {
                printf("test_mbuf_pool() failed (2)\n");
-               return -1;
+               goto err;
        }
 
        /* test that the pointer to the data on a packet mbuf is set properly */
-       if (test_pktmbuf_pool_ptr() < 0) {
+       if (test_pktmbuf_pool_ptr(pktmbuf_pool) < 0) {
                printf("test_pktmbuf_pool_ptr() failed\n");
-               return -1;
+               goto err;
        }
 
        /* test data manipulation in mbuf */
-       if (test_one_pktmbuf() < 0) {
+       if (test_one_pktmbuf(pktmbuf_pool) < 0) {
                printf("test_one_mbuf() failed\n");
-               return -1;
+               goto err;
        }
 
 
@@ -988,46 +1086,52 @@ test_mbuf(void)
         * do it another time, to check that allocation reinitialize
         * the mbuf correctly
         */
-       if (test_one_pktmbuf() < 0) {
+       if (test_one_pktmbuf(pktmbuf_pool) < 0) {
                printf("test_one_mbuf() failed (2)\n");
-               return -1;
+               goto err;
        }
 
-       if (test_pktmbuf_with_non_ascii_data() < 0) {
+       if (test_pktmbuf_with_non_ascii_data(pktmbuf_pool) < 0) {
                printf("test_pktmbuf_with_non_ascii_data() failed\n");
-               return -1;
+               goto err;
        }
 
        /* test free pktmbuf segment one by one */
-       if (test_pktmbuf_free_segment() < 0) {
+       if (test_pktmbuf_free_segment(pktmbuf_pool) < 0) {
                printf("test_pktmbuf_free_segment() failed.\n");
-               return -1;
+               goto err;
        }
 
-       if (testclone_testupdate_testdetach()<0){
+       if (testclone_testupdate_testdetach(pktmbuf_pool) < 0) {
                printf("testclone_and_testupdate() failed \n");
-               return -1;
+               goto err;
        }
 
-       if (test_attach_from_different_pool() < 0) {
+       if (test_attach_from_different_pool(pktmbuf_pool, pktmbuf_pool2) < 0) {
                printf("test_attach_from_different_pool() failed\n");
-               return -1;
+               goto err;
        }
 
        if (test_refcnt_mbuf()<0){
                printf("test_refcnt_mbuf() failed \n");
-               return -1;
+               goto err;
        }
 
-       if (test_failing_mbuf_sanity_check() < 0) {
+       if (test_failing_mbuf_sanity_check(pktmbuf_pool) < 0) {
                printf("test_failing_mbuf_sanity_check() failed\n");
-               return -1;
+               goto err;
        }
-       return 0;
+
+       if (test_mbuf_linearize_check(pktmbuf_pool) < 0) {
+               printf("test_mbuf_linearize_check() failed\n");
+               goto err;
+       }
+       ret = 0;
+
+err:
+       rte_mempool_free(pktmbuf_pool);
+       rte_mempool_free(pktmbuf_pool2);
+       return ret;
 }
 
-static struct test_command mbuf_cmd = {
-       .command = "mbuf_autotest",
-       .callback = test_mbuf,
-};
-REGISTER_TEST_COMMAND(mbuf_cmd);
+REGISTER_TEST_COMMAND(mbuf_autotest, test_mbuf);