eal: add single file segments option

[dpdk.git] / lib / librte_eal / common / rte_malloc.c

diff --git a/lib/librte_eal/common/rte_malloc.c b/lib/librte_eal/common/rte_malloc.c
index 2cda48e..c6d3e57 100644 (file)
--- a/lib/librte_eal/common/rte_malloc.c
+++ b/lib/librte_eal/common/rte_malloc.c
@@ -37,7 +37,8 @@ void rte_free(void *addr)
  * Allocate memory on specified heap.
  */
 void *
-rte_malloc_socket(const char *type, size_t size, unsigned align, int socket_arg)
+rte_malloc_socket(const char *type, size_t size, unsigned int align,
+               int socket_arg)
 {
        struct rte_mem_config *mcfg = rte_eal_get_configuration()->mem_config;
        int socket, i;
@@ -60,7 +61,7 @@ rte_malloc_socket(const char *type, size_t size, unsigned align, int socket_arg)
                return NULL;
 
        ret = malloc_heap_alloc(&mcfg->malloc_heaps[socket], type,
-                               size, 0, align == 0 ? 1 : align, 0);
+                               size, 0, align == 0 ? 1 : align, 0, false);
        if (ret != NULL || socket_arg != SOCKET_ID_ANY)
                return ret;
 
@@ -71,7 +72,7 @@ rte_malloc_socket(const char *type, size_t size, unsigned align, int socket_arg)
                        continue;
 
                ret = malloc_heap_alloc(&mcfg->malloc_heaps[i], type,
-                                       size, 0, align == 0 ? 1 : align, 0);
+                               size, 0, align == 0 ? 1 : align, 0, false);
                if (ret != NULL)
                        return ret;
        }
@@ -241,17 +242,21 @@ rte_malloc_set_limit(__rte_unused const char *type,
 rte_iova_t
 rte_malloc_virt2iova(const void *addr)
 {
-       rte_iova_t iova;
-       const struct malloc_elem *elem = malloc_elem_from_data(addr);
+       const struct rte_memseg *ms;
+       struct malloc_elem *elem = malloc_elem_from_data(addr);
+
        if (elem == NULL)
                return RTE_BAD_IOVA;
-       if (elem->ms->iova == RTE_BAD_IOVA)
-               return RTE_BAD_IOVA;
 
        if (rte_eal_iova_mode() == RTE_IOVA_VA)
-               iova = (uintptr_t)addr;
-       else
-               iova = elem->ms->iova +
-                       RTE_PTR_DIFF(addr, elem->ms->addr);
-       return iova;
+               return (uintptr_t) addr;
+
+       ms = rte_mem_virt2memseg(addr, elem->msl);
+       if (ms == NULL)
+               return RTE_BAD_IOVA;
+
+       if (ms->iova == RTE_BAD_IOVA)
+               return RTE_BAD_IOVA;
+
+       return ms->iova + RTE_PTR_DIFF(addr, ms->addr);
 }