When mempool had been created with RTE_MEMPOOL_F_NO_IOVA_CONTIG flag but later populated with valid IOVA, RTE_MEMPOOL_F_NON_IO was set, while it should not be. The unit test did not catch this because rte_mempool_populate_default() it used was populating with RTE_BAD_IOVA.
When IOVA mode is IOVA-as-VA, RTE_MEMPOOL_NON_IO flag cannot be reliably inferred unless RTE_MEMPOOL_F_ON_IOVA_CONTIG is set. However, rte_mempool_populate_iova() was not checkig for IOVA mode. Keep setting RTE_MEMPOOL_NON_IO at an empty mempool creation and add an assert for it in the unit test (remove a separate test case). Reset the flag using the following logic: * Never reset if RTE_MEMPOOL_F_ON_IOVA_CONTIG is set. * Otherwise, always reset if IOVA mode is IOVA-as-VA. * Otherwise, reset if IOVA is RTE_BAD_IOVA. The unit test temporarily rewrites IOVA mode in EAL configuration before populationg mempools. While it theoretically can affect rte_malloc() inside rte_mempool_populate_iova() if it triggers a new hugepage mapping, it is unlikely to do so, because the allocation size is small. In exchange, the test run with any IOVA mode will check both modes. Fixes: 11541c5c81dd ("mempool: add non-IO flag") Reported-by: Tal Shnaiderman <tal...@nvidia.com> Signed-off-by: Dmitry Kozlyuk <dkozl...@nvidia.com> --- app/test/test_mempool.c | 88 ++++++++++++++++++++++++--------------- lib/mempool/rte_mempool.c | 7 +++- 2 files changed, 59 insertions(+), 36 deletions(-) diff --git a/app/test/test_mempool.c b/app/test/test_mempool.c index 4f399b461d..ae1692e6aa 100644 --- a/app/test/test_mempool.c +++ b/app/test/test_mempool.c @@ -28,6 +28,7 @@ #include <rte_malloc.h> #include <rte_mbuf_pool_ops.h> #include <rte_mbuf.h> +#include "eal_private.h" #include "test.h" @@ -738,36 +739,57 @@ test_mempool_events_safety(void) } while (0) static int -test_mempool_flag_non_io_set_when_no_iova_contig_set(void) +test_mempool_flag_non_io_with_no_iova_contig_set(enum rte_iova_mode mode) { + void *virt = NULL; + rte_iova_t iova; + size_t size = MEMPOOL_ELT_SIZE * 16; struct rte_mempool *mp = NULL; + enum rte_iova_mode saved_mode = rte_eal_iova_mode(); int ret; + virt = rte_malloc("test_mempool", size, rte_mem_page_size()); + RTE_TEST_ASSERT_NOT_NULL(virt, "Cannot allocate memory"); + iova = rte_mem_virt2iova(virt); + RTE_TEST_ASSERT_NOT_EQUAL(iova, RTE_BAD_IOVA, "Cannot get IOVA"); mp = rte_mempool_create_empty("empty", MEMPOOL_SIZE, MEMPOOL_ELT_SIZE, 0, 0, SOCKET_ID_ANY, RTE_MEMPOOL_F_NO_IOVA_CONTIG); RTE_TEST_ASSERT_NOT_NULL(mp, "Cannot create mempool: %s", rte_strerror(rte_errno)); rte_mempool_set_ops_byname(mp, rte_mbuf_best_mempool_ops(), NULL); - ret = rte_mempool_populate_default(mp); + + RTE_TEST_ASSERT(mp->flags & RTE_MEMPOOL_F_NON_IO, + "NON_IO flag is not set on an empty mempool"); + + /* + * Always use valid IOVA so that populate() has no other reason + * to infer that the mempool cannot be used for IO. + */ + rte_eal_get_configuration()->iova_mode = mode; + ret = rte_mempool_populate_iova(mp, virt, iova, size, NULL, NULL); RTE_TEST_ASSERT(ret > 0, "Failed to populate mempool: %s", rte_strerror(-ret)); RTE_TEST_ASSERT(mp->flags & RTE_MEMPOOL_F_NON_IO, "NON_IO flag is not set when NO_IOVA_CONTIG is set"); ret = TEST_SUCCESS; exit: + rte_eal_get_configuration()->iova_mode = saved_mode; rte_mempool_free(mp); + rte_free(virt); return ret; } static int -test_mempool_flag_non_io_unset_when_populated_with_valid_iova(void) +test_mempool_flag_non_io_when_populated(enum rte_iova_mode mode) { void *virt = NULL; rte_iova_t iova; size_t total_size = MEMPOOL_ELT_SIZE * MEMPOOL_SIZE; size_t block_size = total_size / 3; struct rte_mempool *mp = NULL; + bool using_pa = mode == RTE_IOVA_PA; + enum rte_iova_mode saved_mode = rte_eal_iova_mode(); int ret; /* @@ -785,55 +807,51 @@ test_mempool_flag_non_io_unset_when_populated_with_valid_iova(void) RTE_TEST_ASSERT_NOT_NULL(mp, "Cannot create mempool: %s", rte_strerror(rte_errno)); + RTE_TEST_ASSERT(mp->flags & RTE_MEMPOOL_F_NON_IO, + "NON_IO flag is not set on an empty mempool"); + + rte_eal_get_configuration()->iova_mode = mode; + ret = rte_mempool_populate_iova(mp, RTE_PTR_ADD(virt, 1 * block_size), RTE_BAD_IOVA, block_size, NULL, NULL); RTE_TEST_ASSERT(ret > 0, "Failed to populate mempool: %s", rte_strerror(-ret)); - RTE_TEST_ASSERT(mp->flags & RTE_MEMPOOL_F_NON_IO, - "NON_IO flag is not set when mempool is populated with only RTE_BAD_IOVA"); + if (using_pa) + RTE_TEST_ASSERT(mp->flags & RTE_MEMPOOL_F_NON_IO, + "NON_IO flag is not set when mempool is populated with only RTE_BAD_IOVA"); + else + RTE_TEST_ASSERT(!(mp->flags & RTE_MEMPOOL_F_NON_IO), + "NON_IO flag is set on a mempool with contiguous objects with IOVA-as-VA"); ret = rte_mempool_populate_iova(mp, virt, iova, block_size, NULL, NULL); RTE_TEST_ASSERT(ret > 0, "Failed to populate mempool: %s", rte_strerror(-ret)); - RTE_TEST_ASSERT(!(mp->flags & RTE_MEMPOOL_F_NON_IO), - "NON_IO flag is not unset when mempool is populated with valid IOVA"); + if (using_pa) + RTE_TEST_ASSERT(!(mp->flags & RTE_MEMPOOL_F_NON_IO), + "NON_IO flag is not unset when mempool is populated with valid IOVA"); + else + RTE_TEST_ASSERT(!(mp->flags & RTE_MEMPOOL_F_NON_IO), + "NON_IO flag is set when mempool is populated with valid IOVA with IOVA-as-VA"); ret = rte_mempool_populate_iova(mp, RTE_PTR_ADD(virt, 2 * block_size), RTE_BAD_IOVA, block_size, NULL, NULL); RTE_TEST_ASSERT(ret > 0, "Failed to populate mempool: %s", rte_strerror(-ret)); - RTE_TEST_ASSERT(!(mp->flags & RTE_MEMPOOL_F_NON_IO), - "NON_IO flag is set even when some objects have valid IOVA"); + if (using_pa) + RTE_TEST_ASSERT(!(mp->flags & RTE_MEMPOOL_F_NON_IO), + "NON_IO flag is set even when some objects have valid IOVA"); + else + RTE_TEST_ASSERT(!(mp->flags & RTE_MEMPOOL_F_NON_IO), + "NON_IO flag is set for a mempool with contiguous objects with IOVA-as-VA"); ret = TEST_SUCCESS; exit: + rte_eal_get_configuration()->iova_mode = saved_mode; rte_mempool_free(mp); rte_free(virt); return ret; } -static int -test_mempool_flag_non_io_unset_by_default(void) -{ - struct rte_mempool *mp; - int ret; - - mp = rte_mempool_create_empty("empty", MEMPOOL_SIZE, - MEMPOOL_ELT_SIZE, 0, 0, - SOCKET_ID_ANY, 0); - RTE_TEST_ASSERT_NOT_NULL(mp, "Cannot create mempool: %s", - rte_strerror(rte_errno)); - ret = rte_mempool_populate_default(mp); - RTE_TEST_ASSERT_EQUAL(ret, (int)mp->size, "Failed to populate mempool: %s", - rte_strerror(-ret)); - RTE_TEST_ASSERT(!(mp->flags & RTE_MEMPOOL_F_NON_IO), - "NON_IO flag is set by default"); - ret = TEST_SUCCESS; -exit: - rte_mempool_free(mp); - return ret; -} - #pragma pop_macro("RTE_TEST_TRACE_FAILURE") static int @@ -1022,11 +1040,13 @@ test_mempool(void) GOTO_ERR(ret, err); /* test NON_IO flag inference */ - if (test_mempool_flag_non_io_unset_by_default() < 0) + if (test_mempool_flag_non_io_with_no_iova_contig_set(RTE_IOVA_PA) < 0) + GOTO_ERR(ret, err); + if (test_mempool_flag_non_io_with_no_iova_contig_set(RTE_IOVA_VA) < 0) GOTO_ERR(ret, err); - if (test_mempool_flag_non_io_set_when_no_iova_contig_set() < 0) + if (test_mempool_flag_non_io_when_populated(RTE_IOVA_PA) < 0) GOTO_ERR(ret, err); - if (test_mempool_flag_non_io_unset_when_populated_with_valid_iova() < 0) + if (test_mempool_flag_non_io_when_populated(RTE_IOVA_VA) < 0) GOTO_ERR(ret, err); rte_mempool_list_dump(stdout); diff --git a/lib/mempool/rte_mempool.c b/lib/mempool/rte_mempool.c index b75d26c82a..ba81db6f67 100644 --- a/lib/mempool/rte_mempool.c +++ b/lib/mempool/rte_mempool.c @@ -327,6 +327,7 @@ rte_mempool_populate_iova(struct rte_mempool *mp, char *vaddr, unsigned i = 0; size_t off; struct rte_mempool_memhdr *memhdr; + bool always_io, never_io; int ret; ret = mempool_ops_alloc_once(mp); @@ -372,8 +373,10 @@ rte_mempool_populate_iova(struct rte_mempool *mp, char *vaddr, STAILQ_INSERT_TAIL(&mp->mem_list, memhdr, next); mp->nb_mem_chunks++; - /* At least some objects in the pool can now be used for IO. */ - if (iova != RTE_BAD_IOVA) + /* Check if at least some objects in the pool are now usable for IO. */ + never_io = mp->flags & RTE_MEMPOOL_F_NO_IOVA_CONTIG; + always_io = rte_eal_iova_mode() == RTE_IOVA_VA; + if (!never_io && (always_io || iova != RTE_BAD_IOVA)) mp->flags &= ~RTE_MEMPOOL_F_NON_IO; /* Report the mempool as ready only when fully populated. */ -- 2.25.1