40 lines
877 B
C
40 lines
877 B
C
/* 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;
|
|
}
|