| /* SPDX-License-Identifier: GPL-2.0-only */ |
| |
| #include <commonlib/helpers.h> |
| #include <commonlib/mem_pool.h> |
| |
| void *mem_pool_alloc(struct mem_pool *mp, size_t sz) |
| { |
| void *p; |
| |
| if (mp->alignment == 0) |
| return NULL; |
| |
| /* We assume that mp->buf started mp->alignment aligned */ |
| sz = ALIGN_UP(sz, mp->alignment); |
| |
| /* Determine if any space available. */ |
| if ((mp->size - mp->free_offset) < sz) |
| return NULL; |
| |
| p = &mp->buf[mp->free_offset]; |
| |
| mp->free_offset += sz; |
| mp->second_to_last_alloc = mp->last_alloc; |
| mp->last_alloc = p; |
| |
| return p; |
| } |
| |
| void mem_pool_free(struct mem_pool *mp, void *p) |
| { |
| /* Determine if p was the most recent allocation. */ |
| if (p == NULL || mp->last_alloc != p) |
| return; |
| |
| mp->free_offset = mp->last_alloc - mp->buf; |
| mp->last_alloc = mp->second_to_last_alloc; |
| /* No way to track allocation before this one. */ |
| mp->second_to_last_alloc = NULL; |
| } |