mbuf: add a copy routine
[dpdk.git] / app / test / test_mbuf.c
index 9e82a20..b9c2b25 100644 (file)
@@ -27,6 +27,7 @@
 #include <rte_mbuf.h>
 #include <rte_random.h>
 #include <rte_cycles.h>
+#include <rte_malloc.h>
 
 #include "test.h"
 
@@ -331,8 +332,11 @@ testclone_testupdate_testdetach(struct rte_mempool *pktmbuf_pool)
        m->next = rte_pktmbuf_alloc(pktmbuf_pool);
        if (m->next == NULL)
                GOTO_FAIL("Next Pkt Null\n");
+       m->nb_segs = 2;
 
        rte_pktmbuf_append(m->next, sizeof(uint32_t));
+       m->pkt_len = 2 * sizeof(uint32_t);
+
        data = rte_pktmbuf_mtod(m->next, unaligned_uint32_t *);
        *data = MAGIC_DATA;
 
@@ -395,6 +399,158 @@ fail:
        return -1;
 }
 
+static int
+test_pktmbuf_copy(struct rte_mempool *pktmbuf_pool)
+{
+       struct rte_mbuf *m = NULL;
+       struct rte_mbuf *copy = NULL;
+       struct rte_mbuf *copy2 = NULL;
+       struct rte_mbuf *clone = NULL;
+       unaligned_uint32_t *data;
+
+       /* alloc a mbuf */
+       m = rte_pktmbuf_alloc(pktmbuf_pool);
+       if (m == NULL)
+               GOTO_FAIL("ooops not allocating mbuf");
+
+       if (rte_pktmbuf_pkt_len(m) != 0)
+               GOTO_FAIL("Bad length");
+
+       rte_pktmbuf_append(m, sizeof(uint32_t));
+       data = rte_pktmbuf_mtod(m, unaligned_uint32_t *);
+       *data = MAGIC_DATA;
+
+       /* copy the allocated mbuf */
+       copy = rte_pktmbuf_copy(m, pktmbuf_pool, 0, UINT32_MAX);
+       if (copy == NULL)
+               GOTO_FAIL("cannot copy data\n");
+
+       if (rte_pktmbuf_pkt_len(copy) != sizeof(uint32_t))
+               GOTO_FAIL("copy length incorrect\n");
+
+       if (rte_pktmbuf_data_len(copy) != sizeof(uint32_t))
+               GOTO_FAIL("copy data length incorrect\n");
+
+       data = rte_pktmbuf_mtod(copy, unaligned_uint32_t *);
+       if (*data != MAGIC_DATA)
+               GOTO_FAIL("invalid data in copy\n");
+
+       /* free the copy */
+       rte_pktmbuf_free(copy);
+       copy = NULL;
+
+       /* same test with a cloned mbuf */
+       clone = rte_pktmbuf_clone(m, pktmbuf_pool);
+       if (clone == NULL)
+               GOTO_FAIL("cannot clone data\n");
+
+       if (!RTE_MBUF_CLONED(clone))
+               GOTO_FAIL("clone did not give a cloned mbuf\n");
+
+       copy = rte_pktmbuf_copy(clone, pktmbuf_pool, 0, UINT32_MAX);
+       if (copy == NULL)
+               GOTO_FAIL("cannot copy cloned mbuf\n");
+
+       if (RTE_MBUF_CLONED(copy))
+               GOTO_FAIL("copy of clone is cloned?\n");
+
+       if (rte_pktmbuf_pkt_len(copy) != sizeof(uint32_t))
+               GOTO_FAIL("copy clone length incorrect\n");
+
+       if (rte_pktmbuf_data_len(copy) != sizeof(uint32_t))
+               GOTO_FAIL("copy clone data length incorrect\n");
+
+       data = rte_pktmbuf_mtod(copy, unaligned_uint32_t *);
+       if (*data != MAGIC_DATA)
+               GOTO_FAIL("invalid data in clone copy\n");
+       rte_pktmbuf_free(clone);
+       rte_pktmbuf_free(copy);
+       copy = NULL;
+       clone = NULL;
+
+
+       /* same test with a chained mbuf */
+       m->next = rte_pktmbuf_alloc(pktmbuf_pool);
+       if (m->next == NULL)
+               GOTO_FAIL("Next Pkt Null\n");
+       m->nb_segs = 2;
+
+       rte_pktmbuf_append(m->next, sizeof(uint32_t));
+       m->pkt_len = 2 * sizeof(uint32_t);
+       data = rte_pktmbuf_mtod(m->next, unaligned_uint32_t *);
+       *data = MAGIC_DATA + 1;
+
+       copy = rte_pktmbuf_copy(m, pktmbuf_pool, 0, UINT32_MAX);
+       if (copy == NULL)
+               GOTO_FAIL("cannot copy data\n");
+
+       if (rte_pktmbuf_pkt_len(copy) != 2 * sizeof(uint32_t))
+               GOTO_FAIL("chain copy length incorrect\n");
+
+       if (rte_pktmbuf_data_len(copy) != 2 * sizeof(uint32_t))
+               GOTO_FAIL("chain copy data length incorrect\n");
+
+       data = rte_pktmbuf_mtod(copy, unaligned_uint32_t *);
+       if (data[0] != MAGIC_DATA || data[1] != MAGIC_DATA + 1)
+               GOTO_FAIL("invalid data in copy\n");
+
+       rte_pktmbuf_free(copy2);
+
+       /* test offset copy */
+       copy2 = rte_pktmbuf_copy(copy, pktmbuf_pool,
+                                sizeof(uint32_t), UINT32_MAX);
+       if (copy2 == NULL)
+               GOTO_FAIL("cannot copy the copy\n");
+
+       if (rte_pktmbuf_pkt_len(copy2) != sizeof(uint32_t))
+               GOTO_FAIL("copy with offset, length incorrect\n");
+
+       if (rte_pktmbuf_data_len(copy2) != sizeof(uint32_t))
+               GOTO_FAIL("copy with offset, data length incorrect\n");
+
+       data = rte_pktmbuf_mtod(copy2, unaligned_uint32_t *);
+       if (data[0] != MAGIC_DATA + 1)
+               GOTO_FAIL("copy with offset, invalid data\n");
+
+       rte_pktmbuf_free(copy2);
+
+       /* test truncation copy */
+       copy2 = rte_pktmbuf_copy(copy, pktmbuf_pool,
+                                0, sizeof(uint32_t));
+       if (copy2 == NULL)
+               GOTO_FAIL("cannot copy the copy\n");
+
+       if (rte_pktmbuf_pkt_len(copy2) != sizeof(uint32_t))
+               GOTO_FAIL("copy with truncate, length incorrect\n");
+
+       if (rte_pktmbuf_data_len(copy2) != sizeof(uint32_t))
+               GOTO_FAIL("copy with truncate, data length incorrect\n");
+
+       data = rte_pktmbuf_mtod(copy2, unaligned_uint32_t *);
+       if (data[0] != MAGIC_DATA)
+               GOTO_FAIL("copy with truncate, invalid data\n");
+
+       /* free mbuf */
+       rte_pktmbuf_free(m);
+       rte_pktmbuf_free(copy);
+       rte_pktmbuf_free(copy2);
+
+       m = NULL;
+       copy = NULL;
+       copy2 = NULL;
+       printf("%s ok\n", __func__);
+       return 0;
+
+fail:
+       if (m)
+               rte_pktmbuf_free(m);
+       if (copy)
+               rte_pktmbuf_free(copy);
+       if (copy2)
+               rte_pktmbuf_free(copy2);
+       return -1;
+}
+
 static int
 test_attach_from_different_pool(struct rte_mempool *pktmbuf_pool,
                                struct rte_mempool *pktmbuf_pool2)
@@ -752,18 +908,17 @@ static int
 test_refcnt_mbuf(void)
 {
 #ifdef RTE_MBUF_REFCNT_ATOMIC
-       unsigned lnum, master, slave, tref;
+       unsigned int 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",
-                   __func__, lnum);
-               return 0;
+       if (rte_lcore_count() < 2) {
+               printf("Not enough cores for test_refcnt_mbuf, expecting at least 2\n");
+               return TEST_SKIPPED;
        }
 
-       printf("starting %s, at %u lcores\n", __func__, lnum);
+       printf("starting %s, at %u lcores\n", __func__, rte_lcore_count());
 
        /* create refcnt pool & ring if they don't exist */
 
@@ -1028,6 +1183,99 @@ test_mbuf_linearize_check(struct rte_mempool *pktmbuf_pool)
        return 0;
 }
 
+/*
+ * Helper function for test_tx_ofload
+ */
+static inline void
+set_tx_offload(struct rte_mbuf *mb, uint64_t il2, uint64_t il3, uint64_t il4,
+       uint64_t tso, uint64_t ol3, uint64_t ol2)
+{
+       mb->l2_len = il2;
+       mb->l3_len = il3;
+       mb->l4_len = il4;
+       mb->tso_segsz = tso;
+       mb->outer_l3_len = ol3;
+       mb->outer_l2_len = ol2;
+}
+
+static int
+test_tx_offload(void)
+{
+       struct rte_mbuf *mb;
+       uint64_t tm, v1, v2;
+       size_t sz;
+       uint32_t i;
+
+       static volatile struct {
+               uint16_t l2;
+               uint16_t l3;
+               uint16_t l4;
+               uint16_t tso;
+       } txof;
+
+       const uint32_t num = 0x10000;
+
+       txof.l2 = rte_rand() % (1 <<  RTE_MBUF_L2_LEN_BITS);
+       txof.l3 = rte_rand() % (1 <<  RTE_MBUF_L3_LEN_BITS);
+       txof.l4 = rte_rand() % (1 <<  RTE_MBUF_L4_LEN_BITS);
+       txof.tso = rte_rand() % (1 <<   RTE_MBUF_TSO_SEGSZ_BITS);
+
+       printf("%s started, tx_offload = {\n"
+               "\tl2_len=%#hx,\n"
+               "\tl3_len=%#hx,\n"
+               "\tl4_len=%#hx,\n"
+               "\ttso_segsz=%#hx,\n"
+               "\touter_l3_len=%#x,\n"
+               "\touter_l2_len=%#x,\n"
+               "};\n",
+               __func__,
+               txof.l2, txof.l3, txof.l4, txof.tso, txof.l3, txof.l2);
+
+       sz = sizeof(*mb) * num;
+       mb = rte_zmalloc(NULL, sz, RTE_CACHE_LINE_SIZE);
+       if (mb == NULL) {
+               printf("%s failed, out of memory\n", __func__);
+               return -ENOMEM;
+       }
+
+       memset(mb, 0, sz);
+       tm = rte_rdtsc_precise();
+
+       for (i = 0; i != num; i++)
+               set_tx_offload(mb + i, txof.l2, txof.l3, txof.l4,
+                       txof.tso, txof.l3, txof.l2);
+
+       tm = rte_rdtsc_precise() - tm;
+       printf("%s set tx_offload by bit-fields: %u iterations, %"
+               PRIu64 " cycles, %#Lf cycles/iter\n",
+               __func__, num, tm, (long double)tm / num);
+
+       v1 = mb[rte_rand() % num].tx_offload;
+
+       memset(mb, 0, sz);
+       tm = rte_rdtsc_precise();
+
+       for (i = 0; i != num; i++)
+               mb[i].tx_offload = rte_mbuf_tx_offload(txof.l2, txof.l3,
+                       txof.l4, txof.tso, txof.l3, txof.l2, 0);
+
+       tm = rte_rdtsc_precise() - tm;
+       printf("%s set raw tx_offload: %u iterations, %"
+               PRIu64 " cycles, %#Lf cycles/iter\n",
+               __func__, num, tm, (long double)tm / num);
+
+       v2 = mb[rte_rand() % num].tx_offload;
+
+       rte_free(mb);
+
+       printf("%s finished\n"
+               "expected tx_offload value: 0x%" PRIx64 ";\n"
+               "rte_mbuf_tx_offload value: 0x%" PRIx64 ";\n",
+               __func__, v1, v2);
+
+       return (v1 == v2) ? 0 : -EINVAL;
+}
+
 static int
 test_mbuf(void)
 {
@@ -1107,12 +1355,17 @@ test_mbuf(void)
                goto err;
        }
 
+       if (test_pktmbuf_copy(pktmbuf_pool) < 0) {
+               printf("test_pktmbuf_copy() failed\n");
+               goto err;
+       }
+
        if (test_attach_from_different_pool(pktmbuf_pool, pktmbuf_pool2) < 0) {
                printf("test_attach_from_different_pool() failed\n");
                goto err;
        }
 
-       if (test_refcnt_mbuf()<0){
+       if (test_refcnt_mbuf() < 0) {
                printf("test_refcnt_mbuf() failed \n");
                goto err;
        }
@@ -1126,8 +1379,13 @@ test_mbuf(void)
                printf("test_mbuf_linearize_check() failed\n");
                goto err;
        }
-       ret = 0;
 
+       if (test_tx_offload() < 0) {
+               printf("test_tx_offload() failed\n");
+               goto err;
+       }
+
+       ret = 0;
 err:
        rte_mempool_free(pktmbuf_pool);
        rte_mempool_free(pktmbuf_pool2);