Currently, only scenario with valid IOVA table is tested. Fix this
by also testing without IOVA table - in these cases, EAL should
always return RTE_BAD_IOVA for all memsegs.

Signed-off-by: Anatoly Burakov <anatoly.bura...@intel.com>
---
 test/test/test_external_mem.c | 68 +++++++++++++++++++++++------------
 1 file changed, 45 insertions(+), 23 deletions(-)

diff --git a/test/test/test_external_mem.c b/test/test/test_external_mem.c
index 6df42e3ae..845c625d6 100644
--- a/test/test/test_external_mem.c
+++ b/test/test/test_external_mem.c
@@ -42,7 +42,11 @@ check_mem(void *addr, rte_iova_t *iova, size_t pgsz, int 
n_pages)
                        printf("%s():%i: VA mismatch\n", __func__, __LINE__);
                        return -1;
                }
-               if (ms->iova != iova[i]) {
+               if (iova == NULL && ms->iova != RTE_BAD_IOVA) {
+                       printf("%s():%i: IOVA is not invalid\n", __func__,
+                                       __LINE__);
+                       return -1;
+               } else if (iova != NULL && ms->iova != iova[i]) {
                        printf("%s():%i: IOVA mismatch\n", __func__, __LINE__);
                        return -1;
                }
@@ -218,24 +222,29 @@ test_malloc_invalid_param(void *addr, size_t len, size_t 
pgsz, rte_iova_t *iova,
                goto fail;
        }
 
-       /* wrong page count */
-       if (rte_malloc_heap_memory_add(valid_name, addr, len,
-                       iova, 0, pgsz) >= 0 || rte_errno != EINVAL) {
-               printf("%s():%i: Added memory with invalid parameters\n",
-                       __func__, __LINE__);
-               goto fail;
-       }
-       if (rte_malloc_heap_memory_add(valid_name, addr, len,
-                       iova, n_pages - 1, pgsz) >= 0 || rte_errno != EINVAL) {
-               printf("%s():%i: Added memory with invalid parameters\n",
-                       __func__, __LINE__);
-               goto fail;
-       }
-       if (rte_malloc_heap_memory_add(valid_name, addr, len,
-                       iova, n_pages + 1, pgsz) >= 0 || rte_errno != EINVAL) {
-               printf("%s():%i: Added memory with invalid parameters\n",
-                       __func__, __LINE__);
-               goto fail;
+       /* the following tests are only valid if IOVA table is not NULL */
+       if (iova != NULL) {
+               /* wrong page count */
+               if (rte_malloc_heap_memory_add(valid_name, addr, len,
+                               iova, 0, pgsz) >= 0 || rte_errno != EINVAL) {
+                       printf("%s():%i: Added memory with invalid 
parameters\n",
+                               __func__, __LINE__);
+                       goto fail;
+               }
+               if (rte_malloc_heap_memory_add(valid_name, addr, len,
+                               iova, n_pages - 1, pgsz) >= 0 ||
+                               rte_errno != EINVAL) {
+                       printf("%s():%i: Added memory with invalid 
parameters\n",
+                               __func__, __LINE__);
+                       goto fail;
+               }
+               if (rte_malloc_heap_memory_add(valid_name, addr, len,
+                               iova, n_pages + 1, pgsz) >= 0 ||
+                               rte_errno != EINVAL) {
+                       printf("%s():%i: Added memory with invalid 
parameters\n",
+                               __func__, __LINE__);
+                       goto fail;
+               }
        }
 
        /* tests passed, destroy heap */
@@ -257,7 +266,7 @@ test_malloc_basic(void *addr, size_t len, size_t pgsz, 
rte_iova_t *iova,
        const char *heap_name = "heap";
        void *ptr = NULL;
        int socket_id;
-       const struct rte_memzone *mz = NULL;
+       const struct rte_memzone *mz = NULL, *contig_mz = NULL;
 
        /* create heap */
        if (rte_malloc_heap_create(heap_name) != 0) {
@@ -322,12 +331,19 @@ test_malloc_basic(void *addr, size_t len, size_t pgsz, 
rte_iova_t *iova,
                goto fail;
        }
 
+       /* try allocating a memzone */
+       mz = rte_memzone_reserve("heap_test", pgsz * 2, socket_id, 0);
+       if (mz == NULL) {
+               printf("%s():%i: Failed to reserve memzone\n",
+                       __func__, __LINE__);
+               goto fail;
+       }
        /* try allocating an IOVA-contiguous memzone - this should succeed
-        * because we've set up a contiguous IOVA table.
+        * if we've set up a contiguous IOVA table, and fail if we haven't.
         */
-       mz = rte_memzone_reserve("heap_test", pgsz * 2, socket_id,
+       contig_mz = rte_memzone_reserve("heap_test_contig", pgsz * 2, socket_id,
                        RTE_MEMZONE_IOVA_CONTIG);
-       if (mz == NULL) {
+       if (iova != NULL && contig_mz == NULL) {
                printf("%s():%i: Failed to reserve memzone\n",
                        __func__, __LINE__);
                goto fail;
@@ -342,6 +358,8 @@ test_malloc_basic(void *addr, size_t len, size_t pgsz, 
rte_iova_t *iova,
 
        rte_memzone_free(mz);
        mz = NULL;
+       rte_memzone_free(contig_mz);
+       contig_mz = NULL;
 
        if (rte_malloc_heap_memory_remove(heap_name, addr, len) != 0) {
                printf("%s():%i: Removing memory from heap failed\n",
@@ -356,6 +374,7 @@ test_malloc_basic(void *addr, size_t len, size_t pgsz, 
rte_iova_t *iova,
 
        return 0;
 fail:
+       rte_memzone_free(contig_mz);
        rte_memzone_free(mz);
        rte_free(ptr);
        /* even if something failed, attempt to clean up */
@@ -393,6 +412,9 @@ test_external_mem(void)
 
        ret = test_malloc_invalid_param(addr, len, pgsz, iova, n_pages);
        ret |= test_malloc_basic(addr, len, pgsz, iova, n_pages);
+       /* when iova table is NULL, everything should still work */
+       ret |= test_malloc_invalid_param(addr, len, pgsz, NULL, n_pages);
+       ret |= test_malloc_basic(addr, len, pgsz, NULL, n_pages);
 
        munmap(addr, len);
 
-- 
2.17.1

Reply via email to