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