56 lines
1.6 KiB
C++
56 lines
1.6 KiB
C++
/*
|
|
* Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
* All rights reserved.
|
|
*
|
|
* This source code is licensed under both the BSD-style license (found in the
|
|
* LICENSE file in the root directory of this source tree) and the GPLv2 (found
|
|
* in the COPYING file in the root directory of this source tree).
|
|
* You may select, at your option, one of the above-listed licenses.
|
|
*/
|
|
|
|
/* This file provides custom allocation primitives
|
|
*/
|
|
|
|
#define ZSTD_DEPS_NEED_MALLOC
|
|
#include "zstd_deps.h" /* ZSTD_malloc, ZSTD_calloc, ZSTD_free, ZSTD_memset */
|
|
|
|
#include "mem.h" /* MEM_STATIC */
|
|
#define ZSTD_STATIC_LINKING_ONLY
|
|
#include "../zstd.h" /* ZSTD_customMem */
|
|
|
|
#ifndef ZSTD_ALLOCATIONS_H
|
|
#define ZSTD_ALLOCATIONS_H
|
|
|
|
/* custom memory allocation functions */
|
|
|
|
MEM_STATIC void* ZSTD_customMalloc(size_t size, ZSTD_customMem customMem)
|
|
{
|
|
if (customMem.customAlloc)
|
|
return customMem.customAlloc(customMem.opaque, size);
|
|
return ZSTD_malloc(size);
|
|
}
|
|
|
|
MEM_STATIC void* ZSTD_customCalloc(size_t size, ZSTD_customMem customMem)
|
|
{
|
|
if (customMem.customAlloc) {
|
|
/* calloc implemented as malloc+memset;
|
|
* not as efficient as calloc, but next best guess for custom malloc */
|
|
void* const ptr = customMem.customAlloc(customMem.opaque, size);
|
|
ZSTD_memset(ptr, 0, size);
|
|
return ptr;
|
|
}
|
|
return ZSTD_calloc(1, size);
|
|
}
|
|
|
|
MEM_STATIC void ZSTD_customFree(void* ptr, ZSTD_customMem customMem)
|
|
{
|
|
if (ptr!=NULL) {
|
|
if (customMem.customFree)
|
|
customMem.customFree(customMem.opaque, ptr);
|
|
else
|
|
ZSTD_free(ptr);
|
|
}
|
|
}
|
|
|
|
#endif /* ZSTD_ALLOCATIONS_H */
|