1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556 |
- /*
- * 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 */
|