Added mempool test cases with zero-copy get and put API's

Signed-off-by: Kamalakshitha Aligeri <kamalakshitha.alig...@arm.com>
---
Link: 
https://patchwork.dpdk.org/project/dpdk/patch/20221227151700.80887-1...@smartsharesystems.com/
1. Added mempool test cases with zero-copy get and put API's

 app/test/test_mempool.c | 124 +++++++++++++++++++++++++++++++++++++++-
 1 file changed, 122 insertions(+), 2 deletions(-)

diff --git a/app/test/test_mempool.c b/app/test/test_mempool.c
index 8e493eda47..a635a514a7 100644
--- a/app/test/test_mempool.c
+++ b/app/test/test_mempool.c
@@ -72,6 +72,122 @@ my_obj_init(struct rte_mempool *mp, __rte_unused void *arg,
        *objnum = i;
 }
 
+/* basic tests with zero-copy API's (done on one core) */
+static int
+test_mempool_basic_zc_api(struct rte_mempool *mp, int use_external_cache)
+{
+       uint32_t *objnum;
+       void **objtable;
+       void *obj, *obj2;
+       char *obj_data;
+       int ret = 0;
+       unsigned int i, j;
+       int offset;
+       struct rte_mempool_cache *cache;
+       void **cache_objs;
+
+       if (use_external_cache) {
+               /* Create a user-owned mempool cache. */
+               cache = rte_mempool_cache_create(RTE_MEMPOOL_CACHE_MAX_SIZE,
+                               SOCKET_ID_ANY);
+               if (cache == NULL)
+                       RET_ERR();
+       } else {
+               /* May be NULL if cache is disabled. */
+               cache = rte_mempool_default_cache(mp, rte_lcore_id());
+       }
+
+       /* dump the mempool status */
+       rte_mempool_dump(stdout, mp);
+
+       printf("get an object\n");
+       cache_objs = rte_mempool_cache_zc_get_bulk(cache, mp, 1);
+       obj = *cache_objs;
+       rte_mempool_dump(stdout, mp);
+
+       /* tests that improve coverage */
+       printf("get object count\n");
+       /* We have to count the extra caches, one in this case. */
+       offset = use_external_cache ? 1 * cache->len : 0;
+       if (rte_mempool_avail_count(mp) + offset != MEMPOOL_SIZE - 1)
+               GOTO_ERR(ret, out);
+
+       printf("get private data\n");
+       if (rte_mempool_get_priv(mp) != (char *)mp +
+                       RTE_MEMPOOL_HEADER_SIZE(mp, mp->cache_size))
+               GOTO_ERR(ret, out);
+
+#ifndef RTE_EXEC_ENV_FREEBSD /* rte_mem_virt2iova() not supported on bsd */
+       printf("get physical address of an object\n");
+       if (rte_mempool_virt2iova(obj) != rte_mem_virt2iova(obj))
+               GOTO_ERR(ret, out);
+#endif
+
+       printf("put the object back\n");
+       cache_objs = rte_mempool_cache_zc_put_bulk(cache, mp, 1);
+       rte_memcpy(cache_objs, &obj, sizeof(void *));
+       rte_mempool_dump(stdout, mp);
+
+       printf("get 2 objects\n");
+       cache_objs = rte_mempool_cache_zc_get_bulk(cache, mp, 1);
+       obj = *cache_objs;
+       cache_objs = rte_mempool_cache_zc_get_bulk(cache, mp, 1);
+       obj2 = *cache_objs;
+       rte_mempool_dump(stdout, mp);
+
+       printf("put the objects back\n");
+       cache_objs = rte_mempool_cache_zc_put_bulk(cache, mp, 1);
+       rte_memcpy(cache_objs, &obj, sizeof(void *));
+       cache_objs = rte_mempool_cache_zc_put_bulk(cache, mp, 1);
+       rte_memcpy(cache_objs, &obj2, sizeof(void *));
+       rte_mempool_dump(stdout, mp);
+
+       /*
+        * get many objects: we cannot get them all because the cache
+        * on other cores may not be empty.
+        */
+       objtable = malloc(MEMPOOL_SIZE * sizeof(void *));
+       if (objtable == NULL)
+               GOTO_ERR(ret, out);
+
+       for (i = 0; i < MEMPOOL_SIZE; i++) {
+               cache_objs = rte_mempool_cache_zc_get_bulk(cache, mp, 1);
+               objtable[i] = *cache_objs;
+       }
+
+       /*
+        * for each object, check that its content was not modified,
+        * and put objects back in pool
+        */
+       while (i--) {
+               obj = objtable[i];
+               obj_data = obj;
+               objnum = obj;
+               if (*objnum > MEMPOOL_SIZE) {
+                       printf("bad object number(%d)\n", *objnum);
+                       ret = -1;
+                       break;
+               }
+               for (j = sizeof(*objnum); j < mp->elt_size; j++) {
+                       if (obj_data[j] != 0)
+                               ret = -1;
+               }
+               cache_objs = rte_mempool_cache_zc_put_bulk(cache, mp, 1);
+               rte_memcpy(cache_objs, &objtable[i], sizeof(void *));
+       }
+
+       free(objtable);
+       if (ret == -1)
+               printf("objects were modified!\n");
+
+out:
+       if (use_external_cache) {
+               rte_mempool_cache_flush(cache, mp);
+               rte_mempool_cache_free(cache);
+       }
+
+       return ret;
+}
 /* basic tests (done on one core) */
 static int
 test_mempool_basic(struct rte_mempool *mp, int use_external_cache)
@@ -982,8 +1098,12 @@ test_mempool(void)
        if (test_mempool_basic(mp_nocache, 0) < 0)
                GOTO_ERR(ret, err);
 
-       /* basic tests with cache */
-       if (test_mempool_basic(mp_cache, 0) < 0)
+       /* basic tests with zero-copy API's */
+       if (test_mempool_basic_zc_api(mp_cache, 0) < 0)
+               GOTO_ERR(ret, err);
+
+       /* basic tests with user-owned cache and zero-copy API's */
+       if (test_mempool_basic_zc_api(mp_nocache, 1) < 0)
                GOTO_ERR(ret, err);
 
        /* basic tests with user-owned cache */
-- 
2.25.1

Reply via email to