eal: add uuid API
[dpdk.git] / lib / librte_eal / common / malloc_elem.c
index eb41200..efcb826 100644 (file)
@@ -1,10 +1,12 @@
 /* SPDX-License-Identifier: BSD-3-Clause
  * Copyright(c) 2010-2014 Intel Corporation
  */
+#include <inttypes.h>
 #include <stdint.h>
 #include <stddef.h>
 #include <stdio.h>
 #include <string.h>
+#include <unistd.h>
 #include <sys/queue.h>
 
 #include <rte_memory.h>
 #include <rte_common.h>
 #include <rte_spinlock.h>
 
+#include "eal_internal_cfg.h"
+#include "eal_memalloc.h"
 #include "malloc_elem.h"
 #include "malloc_heap.h"
 
-#define MIN_DATA_SIZE (RTE_CACHE_LINE_SIZE)
+size_t
+malloc_elem_find_max_iova_contig(struct malloc_elem *elem, size_t align)
+{
+       void *cur_page, *contig_seg_start, *page_end, *cur_seg_end;
+       void *data_start, *data_end;
+       rte_iova_t expected_iova;
+       struct rte_memseg *ms;
+       size_t page_sz, cur, max;
+
+       page_sz = (size_t)elem->msl->page_sz;
+       data_start = RTE_PTR_ADD(elem, MALLOC_ELEM_HEADER_LEN);
+       data_end = RTE_PTR_ADD(elem, elem->size - MALLOC_ELEM_TRAILER_LEN);
+       /* segment must start after header and with specified alignment */
+       contig_seg_start = RTE_PTR_ALIGN_CEIL(data_start, align);
+
+       /* if we're in IOVA as VA mode, or if we're in legacy mode with
+        * hugepages, all elements are IOVA-contiguous.
+        */
+       if (rte_eal_iova_mode() == RTE_IOVA_VA ||
+                       (internal_config.legacy_mem && rte_eal_has_hugepages()))
+               return RTE_PTR_DIFF(data_end, contig_seg_start);
+
+       cur_page = RTE_PTR_ALIGN_FLOOR(contig_seg_start, page_sz);
+       ms = rte_mem_virt2memseg(cur_page, elem->msl);
+
+       /* do first iteration outside the loop */
+       page_end = RTE_PTR_ADD(cur_page, page_sz);
+       cur_seg_end = RTE_MIN(page_end, data_end);
+       cur = RTE_PTR_DIFF(cur_seg_end, contig_seg_start) -
+                       MALLOC_ELEM_TRAILER_LEN;
+       max = cur;
+       expected_iova = ms->iova + page_sz;
+       /* memsegs are contiguous in memory */
+       ms++;
+
+       cur_page = RTE_PTR_ADD(cur_page, page_sz);
+
+       while (cur_page < data_end) {
+               page_end = RTE_PTR_ADD(cur_page, page_sz);
+               cur_seg_end = RTE_MIN(page_end, data_end);
+
+               /* reset start of contiguous segment if unexpected iova */
+               if (ms->iova != expected_iova) {
+                       /* next contiguous segment must start at specified
+                        * alignment.
+                        */
+                       contig_seg_start = RTE_PTR_ALIGN(cur_page, align);
+                       /* new segment start may be on a different page, so find
+                        * the page and skip to next iteration to make sure
+                        * we're not blowing past data end.
+                        */
+                       ms = rte_mem_virt2memseg(contig_seg_start, elem->msl);
+                       cur_page = ms->addr;
+                       /* don't trigger another recalculation */
+                       expected_iova = ms->iova;
+                       continue;
+               }
+               /* cur_seg_end ends on a page boundary or on data end. if we're
+                * looking at data end, then malloc trailer is already included
+                * in the calculations. if we're looking at page end, then we
+                * know there's more data past this page and thus there's space
+                * for malloc element trailer, so don't count it here.
+                */
+               cur = RTE_PTR_DIFF(cur_seg_end, contig_seg_start);
+               /* update max if cur value is bigger */
+               if (cur > max)
+                       max = cur;
+
+               /* move to next page */
+               cur_page = page_end;
+               expected_iova = ms->iova + page_sz;
+               /* memsegs are contiguous in memory */
+               ms++;
+       }
+
+       return max;
+}
 
 /*
  * Initialize a general malloc_elem header structure
  */
 void
-malloc_elem_init(struct malloc_elem *elem,
-               struct malloc_heap *heap, const struct rte_memseg *ms, size_t size)
+malloc_elem_init(struct malloc_elem *elem, struct malloc_heap *heap,
+               struct rte_memseg_list *msl, size_t size)
 {
        elem->heap = heap;
-       elem->ms = ms;
+       elem->msl = msl;
        elem->prev = NULL;
        elem->next = NULL;
        memset(&elem->free_list, 0, sizeof(elem->free_list));
@@ -46,6 +126,12 @@ malloc_elem_insert(struct malloc_elem *elem)
        struct malloc_elem *prev_elem, *next_elem;
        struct malloc_heap *heap = elem->heap;
 
+       /* first and last elements must be both NULL or both non-NULL */
+       if ((heap->first == NULL) != (heap->last == NULL)) {
+               RTE_LOG(ERR, EAL, "Heap is probably corrupt\n");
+               return;
+       }
+
        if (heap->first == NULL && heap->last == NULL) {
                /* if empty heap */
                heap->first = elem;
@@ -92,6 +178,18 @@ malloc_elem_insert(struct malloc_elem *elem)
                next_elem->prev = elem;
 }
 
+/*
+ * Attempt to find enough physically contiguous memory in this block to store
+ * our data. Assume that element has at least enough space to fit in the data,
+ * so we just check the page addresses.
+ */
+static bool
+elem_check_phys_contig(const struct rte_memseg_list *msl,
+               void *start, size_t size)
+{
+       return eal_memalloc_is_contig(msl, start, size);
+}
+
 /*
  * calculate the starting point of where data of the requested size
  * and alignment would fit in the current element. If the data doesn't
@@ -99,27 +197,59 @@ malloc_elem_insert(struct malloc_elem *elem)
  */
 static void *
 elem_start_pt(struct malloc_elem *elem, size_t size, unsigned align,
-               size_t bound)
+               size_t bound, bool contig)
 {
-       const size_t bmask = ~(bound - 1);
-       uintptr_t end_pt = (uintptr_t)elem +
-                       elem->size - MALLOC_ELEM_TRAILER_LEN;
-       uintptr_t new_data_start = RTE_ALIGN_FLOOR((end_pt - size), align);
-       uintptr_t new_elem_start;
-
-       /* check boundary */
-       if ((new_data_start & bmask) != ((end_pt - 1) & bmask)) {
-               end_pt = RTE_ALIGN_FLOOR(end_pt, bound);
-               new_data_start = RTE_ALIGN_FLOOR((end_pt - size), align);
-               end_pt = new_data_start + size;
-               if (((end_pt - 1) & bmask) != (new_data_start & bmask))
-                       return NULL;
-       }
+       size_t elem_size = elem->size;
 
-       new_elem_start = new_data_start - MALLOC_ELEM_HEADER_LEN;
+       /*
+        * we're allocating from the end, so adjust the size of element by
+        * alignment size.
+        */
+       while (elem_size >= size) {
+               const size_t bmask = ~(bound - 1);
+               uintptr_t end_pt = (uintptr_t)elem +
+                               elem_size - MALLOC_ELEM_TRAILER_LEN;
+               uintptr_t new_data_start = RTE_ALIGN_FLOOR((end_pt - size),
+                               align);
+               uintptr_t new_elem_start;
+
+               /* check boundary */
+               if ((new_data_start & bmask) != ((end_pt - 1) & bmask)) {
+                       end_pt = RTE_ALIGN_FLOOR(end_pt, bound);
+                       new_data_start = RTE_ALIGN_FLOOR((end_pt - size),
+                                       align);
+                       end_pt = new_data_start + size;
+
+                       if (((end_pt - 1) & bmask) != (new_data_start & bmask))
+                               return NULL;
+               }
+
+               new_elem_start = new_data_start - MALLOC_ELEM_HEADER_LEN;
+
+               /* if the new start point is before the exist start,
+                * it won't fit
+                */
+               if (new_elem_start < (uintptr_t)elem)
+                       return NULL;
 
-       /* if the new start point is before the exist start, it won't fit */
-       return (new_elem_start < (uintptr_t)elem) ? NULL : (void *)new_elem_start;
+               if (contig) {
+                       size_t new_data_size = end_pt - new_data_start;
+
+                       /*
+                        * if physical contiguousness was requested and we
+                        * couldn't fit all data into one physically contiguous
+                        * block, try again with lower addresses.
+                        */
+                       if (!elem_check_phys_contig(elem->msl,
+                                       (void *)new_data_start,
+                                       new_data_size)) {
+                               elem_size -= align;
+                               continue;
+                       }
+               }
+               return (void *)new_elem_start;
+       }
+       return NULL;
 }
 
 /*
@@ -128,9 +258,9 @@ elem_start_pt(struct malloc_elem *elem, size_t size, unsigned align,
  */
 int
 malloc_elem_can_hold(struct malloc_elem *elem, size_t size,    unsigned align,
-               size_t bound)
+               size_t bound, bool contig)
 {
-       return elem_start_pt(elem, size, align, bound) != NULL;
+       return elem_start_pt(elem, size, align, bound, contig) != NULL;
 }
 
 /*
@@ -144,7 +274,7 @@ split_elem(struct malloc_elem *elem, struct malloc_elem *split_pt)
        const size_t old_elem_size = (uintptr_t)split_pt - (uintptr_t)elem;
        const size_t new_elem_size = elem->size - old_elem_size;
 
-       malloc_elem_init(split_pt, elem->heap, elem->ms, new_elem_size);
+       malloc_elem_init(split_pt, elem->heap, elem->msl, new_elem_size);
        split_pt->prev = elem;
        split_pt->next = next_elem;
        if (next_elem)
@@ -244,8 +374,8 @@ malloc_elem_free_list_insert(struct malloc_elem *elem)
 /*
  * Remove the specified element from its heap's free list.
  */
-static void
-elem_free_list_remove(struct malloc_elem *elem)
+void
+malloc_elem_free_list_remove(struct malloc_elem *elem)
 {
        LIST_REMOVE(elem, free_list);
 }
@@ -258,14 +388,15 @@ elem_free_list_remove(struct malloc_elem *elem)
  */
 struct malloc_elem *
 malloc_elem_alloc(struct malloc_elem *elem, size_t size, unsigned align,
-               size_t bound)
+               size_t bound, bool contig)
 {
-       struct malloc_elem *new_elem = elem_start_pt(elem, size, align, bound);
+       struct malloc_elem *new_elem = elem_start_pt(elem, size, align, bound,
+                       contig);
        const size_t old_elem_size = (uintptr_t)new_elem - (uintptr_t)elem;
        const size_t trailer_size = elem->size - old_elem_size - size -
                MALLOC_ELEM_OVERHEAD;
 
-       elem_free_list_remove(elem);
+       malloc_elem_free_list_remove(elem);
 
        if (trailer_size > MALLOC_ELEM_OVERHEAD + MIN_DATA_SIZE) {
                /* split it, too much free space after elem */
@@ -324,8 +455,8 @@ join_elem(struct malloc_elem *elem1, struct malloc_elem *elem2)
        elem1->next = next;
 }
 
-static struct malloc_elem *
-elem_join_adjacent_free(struct malloc_elem *elem)
+struct malloc_elem *
+malloc_elem_join_adjacent_free(struct malloc_elem *elem)
 {
        /*
         * check if next element exists, is adjacent and is free, if so join
@@ -334,16 +465,18 @@ elem_join_adjacent_free(struct malloc_elem *elem)
        if (elem->next != NULL && elem->next->state == ELEM_FREE &&
                        next_elem_is_adjacent(elem)) {
                void *erase;
+               size_t erase_len;
 
                /* we will want to erase the trailer and header */
                erase = RTE_PTR_SUB(elem->next, MALLOC_ELEM_TRAILER_LEN);
+               erase_len = MALLOC_ELEM_OVERHEAD + elem->next->pad;
 
                /* remove from free list, join to this one */
-               elem_free_list_remove(elem->next);
+               malloc_elem_free_list_remove(elem->next);
                join_elem(elem, elem->next);
 
-               /* erase header and trailer */
-               memset(erase, 0, MALLOC_ELEM_OVERHEAD);
+               /* erase header, trailer and pad */
+               memset(erase, 0, erase_len);
        }
 
        /*
@@ -354,18 +487,20 @@ elem_join_adjacent_free(struct malloc_elem *elem)
                        prev_elem_is_adjacent(elem)) {
                struct malloc_elem *new_elem;
                void *erase;
+               size_t erase_len;
 
                /* we will want to erase trailer and header */
                erase = RTE_PTR_SUB(elem, MALLOC_ELEM_TRAILER_LEN);
+               erase_len = MALLOC_ELEM_OVERHEAD + elem->pad;
 
                /* remove from free list, join to this one */
-               elem_free_list_remove(elem->prev);
+               malloc_elem_free_list_remove(elem->prev);
 
                new_elem = elem->prev;
                join_elem(new_elem, elem);
 
-               /* erase header and trailer */
-               memset(erase, 0, MALLOC_ELEM_OVERHEAD);
+               /* erase header, trailer and pad */
+               memset(erase, 0, erase_len);
 
                elem = new_elem;
        }
@@ -378,25 +513,74 @@ elem_join_adjacent_free(struct malloc_elem *elem)
  * blocks either immediately before or immediately after newly freed block
  * are also free, the blocks are merged together.
  */
-int
+struct malloc_elem *
 malloc_elem_free(struct malloc_elem *elem)
 {
        void *ptr;
        size_t data_len;
 
-       ptr = RTE_PTR_ADD(elem, sizeof(*elem));
-       data_len = elem->size - MALLOC_ELEM_OVERHEAD;
+       ptr = RTE_PTR_ADD(elem, MALLOC_ELEM_HEADER_LEN + elem->pad);
+       data_len = elem->size - elem->pad - MALLOC_ELEM_OVERHEAD;
 
-       elem = elem_join_adjacent_free(elem);
+       elem = malloc_elem_join_adjacent_free(elem);
 
        malloc_elem_free_list_insert(elem);
 
+       elem->pad = 0;
+
        /* decrease heap's count of allocated elements */
        elem->heap->alloc_count--;
 
        memset(ptr, 0, data_len);
 
-       return 0;
+       return elem;
+}
+
+/* assume all checks were already done */
+void
+malloc_elem_hide_region(struct malloc_elem *elem, void *start, size_t len)
+{
+       struct malloc_elem *hide_start, *hide_end, *prev, *next;
+       size_t len_before, len_after;
+
+       hide_start = start;
+       hide_end = RTE_PTR_ADD(start, len);
+
+       prev = elem->prev;
+       next = elem->next;
+
+       /* we cannot do anything with non-adjacent elements */
+       if (next && next_elem_is_adjacent(elem)) {
+               len_after = RTE_PTR_DIFF(next, hide_end);
+               if (len_after >= MALLOC_ELEM_OVERHEAD + MIN_DATA_SIZE) {
+                       /* split after */
+                       split_elem(elem, hide_end);
+
+                       malloc_elem_free_list_insert(hide_end);
+               } else if (len_after > 0) {
+                       RTE_LOG(ERR, EAL, "Unaligned element, heap is probably corrupt\n");
+                       return;
+               }
+       }
+
+       /* we cannot do anything with non-adjacent elements */
+       if (prev && prev_elem_is_adjacent(elem)) {
+               len_before = RTE_PTR_DIFF(hide_start, elem);
+               if (len_before >= MALLOC_ELEM_OVERHEAD + MIN_DATA_SIZE) {
+                       /* split before */
+                       split_elem(elem, hide_start);
+
+                       prev = elem;
+                       elem = hide_start;
+
+                       malloc_elem_free_list_insert(prev);
+               } else if (len_before > 0) {
+                       RTE_LOG(ERR, EAL, "Unaligned element, heap is probably corrupt\n");
+                       return;
+               }
+       }
+
+       remove_elem(elem);
 }
 
 /*
@@ -422,7 +606,7 @@ malloc_elem_resize(struct malloc_elem *elem, size_t size)
        /* we now know the element fits, so remove from free list,
         * join the two
         */
-       elem_free_list_remove(elem->next);
+       malloc_elem_free_list_remove(elem->next);
        join_elem(elem, elem->next);
 
        if (elem->size - new_size >= MIN_DATA_SIZE + MALLOC_ELEM_OVERHEAD) {
@@ -434,3 +618,26 @@ malloc_elem_resize(struct malloc_elem *elem, size_t size)
        }
        return 0;
 }
+
+static inline const char *
+elem_state_to_str(enum elem_state state)
+{
+       switch (state) {
+       case ELEM_PAD:
+               return "PAD";
+       case ELEM_BUSY:
+               return "BUSY";
+       case ELEM_FREE:
+               return "FREE";
+       }
+       return "ERROR";
+}
+
+void
+malloc_elem_dump(const struct malloc_elem *elem, FILE *f)
+{
+       fprintf(f, "Malloc element at %p (%s)\n", elem,
+                       elem_state_to_str(elem->state));
+       fprintf(f, "  len: 0x%zx pad: 0x%" PRIx32 "\n", elem->size, elem->pad);
+       fprintf(f, "  prev: %p next: %p\n", elem->prev, elem->next);
+}