56 lines · 1.7 KB · C
RawDownload
/* SPDX-License-Identifier: GPL-2.0+ OR BSD-3-Clause *//* * 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 "compiler.h" /* MEM_STATIC */#define ZSTD_STATIC_LINKING_ONLY#include <linux/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 */