Mon, 21 Jul 2014 13:18:32 +0200
added some string tests
universe@103 | 1 | /* |
universe@103 | 2 | * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER. |
olaf@13 | 3 | * |
universe@177 | 4 | * Copyright 2014 Olaf Wintermann. All rights reserved. |
universe@103 | 5 | * |
universe@103 | 6 | * Redistribution and use in source and binary forms, with or without |
universe@103 | 7 | * modification, are permitted provided that the following conditions are met: |
universe@103 | 8 | * |
universe@103 | 9 | * 1. Redistributions of source code must retain the above copyright |
universe@103 | 10 | * notice, this list of conditions and the following disclaimer. |
universe@103 | 11 | * |
universe@103 | 12 | * 2. Redistributions in binary form must reproduce the above copyright |
universe@103 | 13 | * notice, this list of conditions and the following disclaimer in the |
universe@103 | 14 | * documentation and/or other materials provided with the distribution. |
universe@103 | 15 | * |
universe@103 | 16 | * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" |
universe@103 | 17 | * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
universe@103 | 18 | * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
universe@103 | 19 | * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE |
universe@103 | 20 | * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR |
universe@103 | 21 | * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF |
universe@103 | 22 | * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS |
universe@103 | 23 | * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN |
universe@103 | 24 | * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) |
universe@103 | 25 | * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE |
universe@103 | 26 | * POSSIBILITY OF SUCH DAMAGE. |
olaf@13 | 27 | */ |
olaf@13 | 28 | |
universe@135 | 29 | /** |
universe@135 | 30 | * @file mempool.h |
universe@135 | 31 | * |
universe@135 | 32 | * Memory pool implementation. |
universe@135 | 33 | * |
universe@135 | 34 | * @author Mike Becker |
universe@135 | 35 | * @author Olaf Wintermann |
universe@135 | 36 | */ |
universe@135 | 37 | |
olaf@120 | 38 | #ifndef UCX_MEMPOOL_H |
olaf@120 | 39 | #define UCX_MEMPOOL_H |
olaf@13 | 40 | |
universe@69 | 41 | #include "ucx.h" |
universe@38 | 42 | #include <stddef.h> |
universe@50 | 43 | #include "allocator.h" |
universe@38 | 44 | |
olaf@13 | 45 | #ifdef __cplusplus |
olaf@13 | 46 | extern "C" { |
olaf@13 | 47 | #endif |
olaf@13 | 48 | |
universe@135 | 49 | /** |
universe@135 | 50 | * A function pointer to a destructor function. |
universe@135 | 51 | * @see ucx_mempool_setdestr() |
universe@135 | 52 | * @see ucx_mempool_regdestr() |
universe@135 | 53 | */ |
olaf@13 | 54 | typedef void(*ucx_destructor)(void*); |
olaf@13 | 55 | |
universe@135 | 56 | /** |
universe@135 | 57 | * UCX mempool structure. |
universe@135 | 58 | */ |
olaf@13 | 59 | typedef struct { |
olaf@158 | 60 | /** UcxAllocator based on this pool */ |
olaf@158 | 61 | UcxAllocator *allocator; |
olaf@158 | 62 | |
universe@135 | 63 | /** List of pointers to pooled memory. */ |
olaf@158 | 64 | void **data; |
universe@146 | 65 | |
universe@135 | 66 | /** Count of pooled memory items. */ |
olaf@158 | 67 | size_t ndata; |
universe@146 | 68 | |
universe@135 | 69 | /** Memory pool size. */ |
olaf@158 | 70 | size_t size; |
olaf@13 | 71 | } UcxMempool; |
olaf@13 | 72 | |
universe@135 | 73 | /** Shorthand for a new default memory pool with a capacity of 16 elements. */ |
universe@135 | 74 | #define ucx_mempool_new_default() ucx_mempool_new(16) |
universe@48 | 75 | |
universe@135 | 76 | |
universe@135 | 77 | /** |
universe@135 | 78 | * Creates a memory pool with the specified initial size. |
universe@135 | 79 | * |
universe@135 | 80 | * As the created memory pool automatically grows in size by 16 elements, when |
universe@135 | 81 | * trying to allocate memory on a full pool, it is recommended that you use |
universe@135 | 82 | * a multiple of 16 for the initial size. |
universe@135 | 83 | * |
universe@135 | 84 | * @param n initial pool size (should be a multiple of 16) |
universe@135 | 85 | * @return a pointer to the new memory pool |
universe@135 | 86 | */ |
olaf@13 | 87 | UcxMempool *ucx_mempool_new(size_t n); |
universe@135 | 88 | |
universe@135 | 89 | /** |
universe@135 | 90 | * Resizes a memory pool. |
universe@135 | 91 | * |
universe@135 | 92 | * @param pool the pool to resize |
universe@135 | 93 | * @param newcap the new capacity |
universe@135 | 94 | * @return <code>EXIT_SUCCESS</code> on success or |
universe@135 | 95 | * <code>EXIT_FAILURE</code> on failure |
universe@135 | 96 | */ |
universe@15 | 97 | int ucx_mempool_chcap(UcxMempool *pool, size_t newcap); |
olaf@13 | 98 | |
universe@135 | 99 | /** |
universe@141 | 100 | * Changes the pool size to the next smallest multiple of 16. |
universe@141 | 101 | * |
universe@141 | 102 | * You may use this macro, to reduce the pool size after freeing |
universe@141 | 103 | * many pooled memory items. |
universe@141 | 104 | * |
universe@141 | 105 | * @param pool the pool to clamp |
universe@141 | 106 | * @return <code>EXIT_SUCCESS</code> on success or |
universe@141 | 107 | * <code>EXIT_FAILURE</code> on failure |
universe@141 | 108 | */ |
universe@141 | 109 | #define ucx_mempool_clamp(pool) ucx_mempool_chcap(pool, \ |
universe@141 | 110 | (pool->ndata & ~0xF)+0x10) |
universe@141 | 111 | |
universe@141 | 112 | /** |
universe@135 | 113 | * Allocates pooled memory. |
universe@135 | 114 | * |
universe@135 | 115 | * @param pool the memory pool |
universe@135 | 116 | * @param n amount of memory to allocate |
universe@135 | 117 | * @return a pointer to the allocated memory |
universe@135 | 118 | * @see ucx_allocator_malloc() |
universe@135 | 119 | */ |
olaf@13 | 120 | void *ucx_mempool_malloc(UcxMempool *pool, size_t n); |
universe@135 | 121 | /** |
universe@135 | 122 | * Allocates a pooled memory array. |
universe@135 | 123 | * |
universe@181 | 124 | * The content of the allocated memory is set to zero. |
universe@135 | 125 | * |
universe@135 | 126 | * @param pool the memory pool |
universe@135 | 127 | * @param nelem amount of elements to allocate |
universe@135 | 128 | * @param elsize amount of memory per element |
universe@135 | 129 | * @return a pointer to the allocated memory |
universe@135 | 130 | * @see ucx_allocator_calloc() |
universe@135 | 131 | */ |
olaf@13 | 132 | void *ucx_mempool_calloc(UcxMempool *pool, size_t nelem, size_t elsize); |
universe@146 | 133 | |
universe@135 | 134 | /** |
universe@135 | 135 | * Reallocates pooled memory. |
universe@135 | 136 | * |
universe@141 | 137 | * If the memory to be reallocated is not contained by the specified pool, this |
universe@141 | 138 | * function will possibly fail. In case the memory had to be moved to another |
universe@141 | 139 | * location, this function will print out a message to <code>stderr</code> |
universe@141 | 140 | * and exit the program with error code <code>EXIT_FAILURE</code>. |
universe@141 | 141 | * |
universe@135 | 142 | * @param pool the memory pool |
universe@135 | 143 | * @param ptr a pointer to the memory that shall be reallocated |
universe@135 | 144 | * @param n the new size of the memory |
universe@181 | 145 | * @return a pointer to the new location of the memory |
universe@135 | 146 | * @see ucx_allocator_realloc() |
universe@135 | 147 | */ |
olaf@13 | 148 | void *ucx_mempool_realloc(UcxMempool *pool, void *ptr, size_t n); |
universe@146 | 149 | |
universe@135 | 150 | /** |
universe@135 | 151 | * Frees pooled memory. |
universe@135 | 152 | * |
universe@135 | 153 | * Before freeing the memory, the specified destructor function (if any) |
universe@135 | 154 | * is called. |
universe@135 | 155 | * |
universe@135 | 156 | * If you specify memory, that is not pooled by the specified memory pool, this |
universe@135 | 157 | * is considered as a severe error and an error message is written to |
universe@135 | 158 | * <code>stderr</code> and the application is shut down with code |
universe@135 | 159 | * <code>EXIT_FAILURE</code>. |
universe@135 | 160 | * |
universe@135 | 161 | * @param pool the memory pool |
universe@135 | 162 | * @param ptr a pointer to the memory that shall be freed |
universe@135 | 163 | * @see ucx_mempool_set_destr() |
universe@135 | 164 | */ |
olaf@113 | 165 | void ucx_mempool_free(UcxMempool *pool, void *ptr); |
olaf@13 | 166 | |
universe@135 | 167 | /** |
universe@135 | 168 | * Destroys a memory pool. |
universe@135 | 169 | * |
universe@135 | 170 | * For each element the destructor function (if any) is called and the element |
universe@135 | 171 | * is freed. |
universe@135 | 172 | * |
universe@135 | 173 | * Each of the registered destructor function that has no corresponding element |
universe@135 | 174 | * within the pool (namely those registered by ucx_mempool_reg_destr) is |
universe@135 | 175 | * called interleaving with the element destruction, but with guarantee to the |
universe@135 | 176 | * order in which they were registered (FIFO order). |
universe@135 | 177 | * |
universe@135 | 178 | * |
universe@135 | 179 | * @param pool the mempool to destroy |
universe@135 | 180 | */ |
olaf@113 | 181 | void ucx_mempool_destroy(UcxMempool *pool); |
olaf@13 | 182 | |
universe@135 | 183 | /** |
universe@135 | 184 | * Sets a destructor function for the specified memory. |
universe@135 | 185 | * |
universe@135 | 186 | * The destructor is automatically called when the memory is freed or the |
universe@135 | 187 | * pool is destroyed. |
universe@135 | 188 | * |
universe@135 | 189 | * The only requirement for the specified memory is, that it <b>MUST</b> be |
universe@135 | 190 | * pooled memory by an UcxMempool or an element-compatible mempool. The pointer |
universe@135 | 191 | * to the destructor function is saved in a reserved area before the actual |
universe@135 | 192 | * memory. |
universe@135 | 193 | * |
universe@135 | 194 | * @param ptr pooled memory |
universe@135 | 195 | * @param func a pointer to the destructor function |
universe@135 | 196 | * @see ucx_mempool_free() |
universe@135 | 197 | * @see ucx_mempool_destroy() |
universe@135 | 198 | */ |
olaf@13 | 199 | void ucx_mempool_set_destr(void *ptr, ucx_destructor func); |
universe@135 | 200 | |
universe@135 | 201 | /** |
universe@135 | 202 | * Registers a destructor function for the specified (non-pooled) memory. |
universe@135 | 203 | * |
universe@135 | 204 | * This is useful, if you have memory that has not been allocated by a mempool, |
universe@135 | 205 | * but shall be managed by a mempool. |
universe@135 | 206 | * |
universe@135 | 207 | * This function creates an entry in the specified mempool and the memory will |
universe@135 | 208 | * therefore (logically) convert to pooled memory. |
universe@135 | 209 | * |
universe@135 | 210 | * @param pool the memory pool |
universe@135 | 211 | * @param ptr data the destructor is registered for |
universe@135 | 212 | * @param destr a pointer to the destructor function |
universe@135 | 213 | */ |
olaf@13 | 214 | void ucx_mempool_reg_destr(UcxMempool *pool, void *ptr, ucx_destructor destr); |
olaf@13 | 215 | |
olaf@13 | 216 | #ifdef __cplusplus |
olaf@13 | 217 | } |
olaf@13 | 218 | #endif |
olaf@13 | 219 | |
olaf@120 | 220 | #endif /* UCX_MEMPOOL_H */ |
olaf@13 | 221 |