ipc: fix spelling in log and comment
[dpdk.git] / lib / librte_eal / common / malloc_heap.c
index e243a8f..5a09247 100644 (file)
 #include <rte_string_fns.h>
 #include <rte_spinlock.h>
 #include <rte_memcpy.h>
+#include <rte_memzone.h>
 #include <rte_atomic.h>
 #include <rte_fbarray.h>
 
 #include "eal_internal_cfg.h"
 #include "eal_memalloc.h"
+#include "eal_memcfg.h"
+#include "eal_private.h"
 #include "malloc_elem.h"
 #include "malloc_heap.h"
 #include "malloc_mp.h"
@@ -238,6 +241,9 @@ heap_alloc(struct malloc_heap *heap, const char *type __rte_unused, size_t size,
        size = RTE_CACHE_LINE_ROUNDUP(size);
        align = RTE_CACHE_LINE_ROUNDUP(align);
 
+       /* roundup might cause an overflow */
+       if (size == 0)
+               return NULL;
        elem = find_suitable_element(heap, size, flags, align, bound, contig);
        if (elem != NULL) {
                elem = malloc_elem_alloc(elem, size, align, bound, contig);
@@ -485,10 +491,9 @@ try_expand_heap(struct malloc_heap *heap, uint64_t pg_sz, size_t elt_size,
                int socket, unsigned int flags, size_t align, size_t bound,
                bool contig)
 {
-       struct rte_mem_config *mcfg = rte_eal_get_configuration()->mem_config;
        int ret;
 
-       rte_rwlock_write_lock(&mcfg->memory_hotplug_lock);
+       rte_mcfg_mem_write_lock();
 
        if (rte_eal_process_type() == RTE_PROC_PRIMARY) {
                ret = try_expand_heap_primary(heap, pg_sz, elt_size, socket,
@@ -498,7 +503,7 @@ try_expand_heap(struct malloc_heap *heap, uint64_t pg_sz, size_t elt_size,
                                flags, align, bound, contig);
        }
 
-       rte_rwlock_write_unlock(&mcfg->memory_hotplug_lock);
+       rte_mcfg_mem_write_unlock();
        return ret;
 }
 
@@ -637,13 +642,15 @@ malloc_heap_alloc_on_heap_id(const char *type, size_t size,
        unsigned int size_flags = flags & ~RTE_MEMZONE_SIZE_HINT_ONLY;
        int socket_id;
        void *ret;
+       const struct internal_config *internal_conf =
+               eal_get_internal_configuration();
 
        rte_spinlock_lock(&(heap->lock));
 
        align = align == 0 ? 1 : align;
 
        /* for legacy mode, try once and with all flags */
-       if (internal_config.legacy_mem) {
+       if (internal_conf->legacy_mem) {
                ret = heap_alloc(heap, type, size, flags, align, bound, contig);
                goto alloc_unlock;
        }
@@ -821,13 +828,14 @@ malloc_heap_free_pages(void *aligned_start, size_t aligned_len)
 int
 malloc_heap_free(struct malloc_elem *elem)
 {
-       struct rte_mem_config *mcfg = rte_eal_get_configuration()->mem_config;
        struct malloc_heap *heap;
        void *start, *aligned_start, *end, *aligned_end;
        size_t len, aligned_len, page_sz;
        struct rte_memseg_list *msl;
        unsigned int i, n_segs, before_space, after_space;
        int ret;
+       const struct internal_config *internal_conf =
+               eal_get_internal_configuration();
 
        if (!malloc_elem_cookies_ok(elem) || elem->state != ELEM_BUSY)
                return -1;
@@ -850,7 +858,7 @@ malloc_heap_free(struct malloc_elem *elem)
        /* ...of which we can't avail if we are in legacy mode, or if this is an
         * externally allocated segment.
         */
-       if (internal_config.legacy_mem || (msl->external > 0))
+       if (internal_conf->legacy_mem || (msl->external > 0))
                goto free_unlock;
 
        /* check if we can free any memory back to the system */
@@ -861,7 +869,7 @@ malloc_heap_free(struct malloc_elem *elem)
         * we will defer freeing these hugepages until the entire original allocation
         * can be freed
         */
-       if (internal_config.match_allocations && elem->size != elem->orig_size)
+       if (internal_conf->match_allocations && elem->size != elem->orig_size)
                goto free_unlock;
 
        /* probably, but let's make sure, as we may not be using up full page */
@@ -935,7 +943,7 @@ malloc_heap_free(struct malloc_elem *elem)
 
        /* now we can finally free us some pages */
 
-       rte_rwlock_write_lock(&mcfg->memory_hotplug_lock);
+       rte_mcfg_mem_write_lock();
 
        /*
         * we allow secondary processes to clear the heap of this allocated
@@ -990,7 +998,7 @@ malloc_heap_free(struct malloc_elem *elem)
        RTE_LOG(DEBUG, EAL, "Heap on socket %d was shrunk by %zdMB\n",
                msl->socket_id, aligned_len >> 20ULL);
 
-       rte_rwlock_write_unlock(&mcfg->memory_hotplug_lock);
+       rte_mcfg_mem_write_unlock();
 free_unlock:
        rte_spinlock_unlock(&(heap->lock));
        return ret;
@@ -1074,12 +1082,9 @@ malloc_heap_dump(struct malloc_heap *heap, FILE *f)
 }
 
 static int
-destroy_seg(struct malloc_elem *elem, size_t len)
+destroy_elem(struct malloc_elem *elem, size_t len)
 {
        struct malloc_heap *heap = elem->heap;
-       struct rte_memseg_list *msl;
-
-       msl = elem->msl;
 
        /* notify all subscribers that a memory area is going to be removed */
        eal_memalloc_mem_event_notify(RTE_MEM_EVENT_FREE, elem, len);
@@ -1092,13 +1097,6 @@ destroy_seg(struct malloc_elem *elem, size_t len)
 
        memset(elem, 0, sizeof(*elem));
 
-       /* destroy the fbarray backing this memory */
-       if (rte_fbarray_destroy(&msl->memseg_arr) < 0)
-               return -1;
-
-       /* reset the memseg list */
-       memset(msl, 0, sizeof(*msl));
-
        return 0;
 }
 
@@ -1128,7 +1126,7 @@ malloc_heap_create_external_seg(void *va_addr, rte_iova_t iova_addrs[],
                return NULL;
        }
 
-       snprintf(fbarray_name, sizeof(fbarray_name) - 1, "%s_%p",
+       snprintf(fbarray_name, sizeof(fbarray_name), "%s_%p",
                        seg_name, va_addr);
 
        /* create the backing fbarray */
@@ -1165,6 +1163,62 @@ malloc_heap_create_external_seg(void *va_addr, rte_iova_t iova_addrs[],
        return msl;
 }
 
+struct extseg_walk_arg {
+       void *va_addr;
+       size_t len;
+       struct rte_memseg_list *msl;
+};
+
+static int
+extseg_walk(const struct rte_memseg_list *msl, void *arg)
+{
+       struct rte_mem_config *mcfg = rte_eal_get_configuration()->mem_config;
+       struct extseg_walk_arg *wa = arg;
+
+       if (msl->base_va == wa->va_addr && msl->len == wa->len) {
+               unsigned int found_idx;
+
+               /* msl is const */
+               found_idx = msl - mcfg->memsegs;
+               wa->msl = &mcfg->memsegs[found_idx];
+               return 1;
+       }
+       return 0;
+}
+
+struct rte_memseg_list *
+malloc_heap_find_external_seg(void *va_addr, size_t len)
+{
+       struct extseg_walk_arg wa;
+       int res;
+
+       wa.va_addr = va_addr;
+       wa.len = len;
+
+       res = rte_memseg_list_walk_thread_unsafe(extseg_walk, &wa);
+
+       if (res != 1) {
+               /* 0 means nothing was found, -1 shouldn't happen */
+               if (res == 0)
+                       rte_errno = ENOENT;
+               return NULL;
+       }
+       return wa.msl;
+}
+
+int
+malloc_heap_destroy_external_seg(struct rte_memseg_list *msl)
+{
+       /* destroy the fbarray backing this memory */
+       if (rte_fbarray_destroy(&msl->memseg_arr) < 0)
+               return -1;
+
+       /* reset the memseg list */
+       memset(msl, 0, sizeof(*msl));
+
+       return 0;
+}
+
 int
 malloc_heap_add_external_memory(struct malloc_heap *heap,
                struct rte_memseg_list *msl)
@@ -1213,7 +1267,7 @@ malloc_heap_remove_external_memory(struct malloc_heap *heap, void *va_addr,
                rte_errno = EBUSY;
                return -1;
        }
-       return destroy_seg(elem, len);
+       return destroy_elem(elem, len);
 }
 
 int
@@ -1273,10 +1327,11 @@ rte_eal_malloc_heap_init(void)
 {
        struct rte_mem_config *mcfg = rte_eal_get_configuration()->mem_config;
        unsigned int i;
+       const struct internal_config *internal_conf =
+               eal_get_internal_configuration();
 
-       if (internal_config.match_allocations) {
+       if (internal_conf->match_allocations)
                RTE_LOG(DEBUG, EAL, "Hugepages will be freed exactly as allocated.\n");
-       }
 
        if (rte_eal_process_type() == RTE_PROC_PRIMARY) {
                /* assign min socket ID to external heaps */
@@ -1288,7 +1343,7 @@ rte_eal_malloc_heap_init(void)
                        char heap_name[RTE_HEAP_NAME_MAX_LEN];
                        int socket_id = rte_socket_id_by_idx(i);
 
-                       snprintf(heap_name, sizeof(heap_name) - 1,
+                       snprintf(heap_name, sizeof(heap_name),
                                        "socket_%i", socket_id);
                        strlcpy(heap->name, heap_name, RTE_HEAP_NAME_MAX_LEN);
                        heap->socket_id = socket_id;
@@ -1298,7 +1353,7 @@ rte_eal_malloc_heap_init(void)
 
        if (register_mp_requests()) {
                RTE_LOG(ERR, EAL, "Couldn't register malloc multiprocess actions\n");
-               rte_rwlock_read_unlock(&mcfg->memory_hotplug_lock);
+               rte_mcfg_mem_read_unlock();
                return -1;
        }
 
@@ -1306,7 +1361,7 @@ rte_eal_malloc_heap_init(void)
         * even come before primary itself is fully initialized, and secondaries
         * do not need to initialize the heap.
         */
-       rte_rwlock_read_unlock(&mcfg->memory_hotplug_lock);
+       rte_mcfg_mem_read_unlock();
 
        /* secondary process does not need to initialize anything */
        if (rte_eal_process_type() != RTE_PROC_PRIMARY)