f22da1cde7
Change license to BSD-3-Clause Signed-off-by: Rafal Stefanowski <rafal.stefanowski@intel.com>
118 lines
2.1 KiB
C
118 lines
2.1 KiB
C
/*
|
|
* Copyright(c) 2012-2021 Intel Corporation
|
|
* SPDX-License-Identifier: BSD-3-Clause
|
|
*/
|
|
#include "ocf/ocf.h"
|
|
#include "utils_realloc.h"
|
|
#include "ocf_env.h"
|
|
|
|
#define OCF_REALLOC_K_MAX (128 * KiB)
|
|
|
|
static int _ocf_realloc_with_cp(void **mem, size_t size, size_t count,
|
|
size_t *limit, bool cp)
|
|
{
|
|
size_t alloc_size = size * count;
|
|
|
|
ENV_BUG_ON(!mem);
|
|
ENV_BUG_ON(!limit);
|
|
|
|
if (size && count) {
|
|
/* Memory reallocation request */
|
|
|
|
if (alloc_size > *limit) {
|
|
/* The space is not enough, we need allocate new one */
|
|
|
|
void *new_mem;
|
|
|
|
if (alloc_size > OCF_REALLOC_K_MAX)
|
|
new_mem = env_vzalloc(alloc_size);
|
|
else
|
|
new_mem = env_zalloc(alloc_size, ENV_MEM_NOIO);
|
|
|
|
if (!new_mem) {
|
|
/* Allocation error */
|
|
return -1;
|
|
}
|
|
|
|
/* Free previous memory */
|
|
if (*mem) {
|
|
if (cp) {
|
|
/* copy previous content into new allocated
|
|
* memory
|
|
*/
|
|
ENV_BUG_ON(env_memcpy(new_mem, alloc_size, *mem, *limit));
|
|
|
|
}
|
|
|
|
if (*limit > OCF_REALLOC_K_MAX)
|
|
env_vfree(*mem);
|
|
else
|
|
env_free(*mem);
|
|
}
|
|
|
|
/* Update limit */
|
|
*limit = alloc_size;
|
|
|
|
/* Update memory pointer */
|
|
*mem = new_mem;
|
|
|
|
return 0;
|
|
}
|
|
|
|
/*
|
|
* The memory space is enough, no action required.
|
|
* Space after allocation set to '0'
|
|
*/
|
|
if (cp)
|
|
ENV_BUG_ON(env_memset(*mem + alloc_size, *limit - alloc_size, 0));
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
if ((size == 0) && (count == 0)) {
|
|
|
|
if ((*mem) && (*limit)) {
|
|
/* Need to free memory */
|
|
if (*limit > OCF_REALLOC_K_MAX)
|
|
env_vfree(*mem);
|
|
else
|
|
env_free(*mem);
|
|
|
|
/* Update limit */
|
|
*((size_t *)limit) = 0;
|
|
*mem = NULL;
|
|
|
|
return 0;
|
|
}
|
|
|
|
if ((!*mem) && (*limit == 0)) {
|
|
/* No allocation before do nothing */
|
|
return 0;
|
|
|
|
}
|
|
}
|
|
|
|
ENV_BUG();
|
|
return -1;
|
|
}
|
|
|
|
int ocf_realloc(void **mem, size_t size, size_t count, size_t *limit)
|
|
{
|
|
return _ocf_realloc_with_cp(mem, size, count, limit, false);
|
|
}
|
|
|
|
int ocf_realloc_cp(void **mem, size_t size, size_t count, size_t *limit)
|
|
{
|
|
return _ocf_realloc_with_cp(mem, size, count, limit, true);
|
|
}
|
|
|
|
void ocf_realloc_init(void **mem, size_t *limit)
|
|
{
|
|
ENV_BUG_ON(!mem);
|
|
ENV_BUG_ON(!limit);
|
|
|
|
*mem = NULL;
|
|
*((size_t *)limit) = 0;
|
|
}
|