lib/librte_acl/tb_mem.c

   1 /*-
   2  *   BSD LICENSE
   3  *
   4  *   Copyright(c) 2010-2014 Intel Corporation. All rights reserved.
   5  *   All rights reserved.
   6  *
   7  *   Redistribution and use in source and binary forms, with or without
   8  *   modification, are permitted provided that the following conditions
   9  *   are met:
  10  *
  11  *     * Redistributions of source code must retain the above copyright
  12  *       notice, this list of conditions and the following disclaimer.
  13  *     * Redistributions in binary form must reproduce the above copyright
  14  *       notice, this list of conditions and the following disclaimer in
  15  *       the documentation and/or other materials provided with the
  16  *       distribution.
  17  *     * Neither the name of Intel Corporation nor the names of its
  18  *       contributors may be used to endorse or promote products derived
  19  *       from this software without specific prior written permission.
  20  *
  21  *   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  22  *   "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  23  *   LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  24  *   A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  25  *   OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  26  *   SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  27  *   LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  28  *   DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  29  *   THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  30  *   (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  31  *   OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  32  */
  33
  34 #include "tb_mem.h"
  35
  36 /*
  37  *  Memory managment routines for temporary memory.
  38  *  That memory is used only during build phase and is released after
  39  *  build is finished.
  40  */
  41
  42 static struct tb_mem_block *
  43 tb_pool(struct tb_mem_pool *pool, size_t sz)
  44 {
  45         struct tb_mem_block *block;
  46         uint8_t *ptr;
  47         size_t size;
  48
  49         size = sz + pool->alignment - 1;
  50         block = calloc(1, size + sizeof(*pool->block));
  51         if (block == NULL) {
  52                 RTE_LOG(ERR, MALLOC, "%s(%zu)\n failed, currently allocated "
  53                         "by pool: %zu bytes\n", __func__, sz, pool->alloc);
  54                 return NULL;
  55         }
  56
  57         block->pool = pool;
  58
  59         block->next = pool->block;
  60         pool->block = block;
  61
  62         pool->alloc += size;
  63
  64         ptr = (uint8_t *)(block + 1);
  65         block->mem = RTE_PTR_ALIGN_CEIL(ptr, pool->alignment);
  66         block->size = size - (block->mem - ptr);
  67
  68         return block;
  69 }
  70
  71 void *
  72 tb_alloc(struct tb_mem_pool *pool, size_t size)
  73 {
  74         struct tb_mem_block *block;
  75         void *ptr;
  76         size_t new_sz;
  77
  78         size = RTE_ALIGN_CEIL(size, pool->alignment);
  79
  80         block = pool->block;
  81         if (block == NULL || block->size < size) {
  82                 new_sz = (size > pool->min_alloc) ? size : pool->min_alloc;
  83                 block = tb_pool(pool, new_sz);
  84                 if (block == NULL)
  85                         return NULL;
  86         }
  87         ptr = block->mem;
  88         block->size -= size;
  89         block->mem += size;
  90         return ptr;
  91 }
  92
  93 void
  94 tb_free_pool(struct tb_mem_pool *pool)
  95 {
  96         struct tb_mem_block *next, *block;
  97
  98         for (block = pool->block; block != NULL; block = next) {
  99                 next = block->next;
 100                 free(block);
 101         }
 102         pool->block = NULL;
 103         pool->alloc = 0;
 104 }