| /* Copyright 2016 Google Inc. All Rights Reserved. |
| |
| Distributed under MIT license. |
| See file LICENSE for detail or copy at https://opensource.org/licenses/MIT |
| */ |
| |
| /* Macros for memory management. */ |
| |
| #ifndef BROTLI_ENC_MEMORY_H_ |
| #define BROTLI_ENC_MEMORY_H_ |
| |
| #include <string.h> /* memcpy */ |
| |
| #include <brotli/types.h> |
| |
| #include "../common/platform.h" |
| |
| #if defined(__cplusplus) || defined(c_plusplus) |
| extern "C" { |
| #endif |
| |
| #if !defined(BROTLI_ENCODER_CLEANUP_ON_OOM) && \ |
| !defined(BROTLI_ENCODER_EXIT_ON_OOM) |
| #define BROTLI_ENCODER_EXIT_ON_OOM |
| #endif |
| |
| #if !defined(BROTLI_ENCODER_EXIT_ON_OOM) |
| #if defined(BROTLI_EXPERIMENTAL) |
| #define BROTLI_ENCODER_MEMORY_MANAGER_SLOTS 32768 |
| #else /* BROTLI_EXPERIMENTAL */ |
| #define BROTLI_ENCODER_MEMORY_MANAGER_SLOTS 256 |
| #endif /* BROTLI_EXPERIMENTAL */ |
| #endif /* BROTLI_ENCODER_EXIT_ON_OOM */ |
| |
| typedef struct MemoryManager { |
| brotli_alloc_func alloc_func; |
| brotli_free_func free_func; |
| void* opaque; |
| #if !defined(BROTLI_ENCODER_EXIT_ON_OOM) |
| BROTLI_BOOL is_oom; |
| size_t perm_allocated; |
| size_t new_allocated; |
| size_t new_freed; |
| void* pointers[BROTLI_ENCODER_MEMORY_MANAGER_SLOTS]; |
| #endif /* BROTLI_ENCODER_EXIT_ON_OOM */ |
| } MemoryManager; |
| |
| BROTLI_INTERNAL void BrotliInitMemoryManager( |
| MemoryManager* m, brotli_alloc_func alloc_func, brotli_free_func free_func, |
| void* opaque); |
| |
| BROTLI_INTERNAL void* BrotliAllocate(MemoryManager* m, size_t n); |
| #define BROTLI_ALLOC(M, T, N) \ |
| ((N) > 0 ? ((T*)BrotliAllocate((M), (N) * sizeof(T))) : NULL) |
| |
| BROTLI_INTERNAL void BrotliFree(MemoryManager* m, void* p); |
| #define BROTLI_FREE(M, P) { \ |
| BrotliFree((M), (P)); \ |
| P = NULL; \ |
| } |
| |
| #if defined(BROTLI_ENCODER_EXIT_ON_OOM) |
| #define BROTLI_IS_OOM(M) (!!0) |
| #else /* BROTLI_ENCODER_EXIT_ON_OOM */ |
| #define BROTLI_IS_OOM(M) (!!(M)->is_oom) |
| #endif /* BROTLI_ENCODER_EXIT_ON_OOM */ |
| |
| /* |
| BROTLI_IS_NULL is a fake check, BROTLI_IS_OOM does the heavy lifting. |
| The only purpose of it is to explain static analyzers the state of things. |
| NB: use ONLY together with BROTLI_IS_OOM |
| AND ONLY for allocations in the current scope. |
| */ |
| #if defined(__clang_analyzer__) && !defined(BROTLI_ENCODER_EXIT_ON_OOM) |
| #define BROTLI_IS_NULL(A) ((A) == nullptr) |
| #else /* defined(__clang_analyzer__) */ |
| #define BROTLI_IS_NULL(A) (!!0) |
| #endif /* defined(__clang_analyzer__) */ |
| |
| BROTLI_INTERNAL void BrotliWipeOutMemoryManager(MemoryManager* m); |
| |
| /* |
| Dynamically grows array capacity to at least the requested size |
| M: MemoryManager |
| T: data type |
| A: array |
| C: capacity |
| R: requested size |
| */ |
| #define BROTLI_ENSURE_CAPACITY(M, T, A, C, R) { \ |
| if (C < (R)) { \ |
| size_t _new_size = (C == 0) ? (R) : C; \ |
| T* new_array; \ |
| while (_new_size < (R)) _new_size *= 2; \ |
| new_array = BROTLI_ALLOC((M), T, _new_size); \ |
| if (!BROTLI_IS_OOM(M) && !BROTLI_IS_NULL(new_array) && C != 0) \ |
| memcpy(new_array, A, C * sizeof(T)); \ |
| BROTLI_FREE((M), A); \ |
| A = new_array; \ |
| C = _new_size; \ |
| } \ |
| } |
| |
| /* |
| Appends value and dynamically grows array capacity when needed |
| M: MemoryManager |
| T: data type |
| A: array |
| C: array capacity |
| S: array size |
| V: value to append |
| */ |
| #define BROTLI_ENSURE_CAPACITY_APPEND(M, T, A, C, S, V) { \ |
| (S)++; \ |
| BROTLI_ENSURE_CAPACITY(M, T, A, C, S); \ |
| A[(S) - 1] = (V); \ |
| } |
| |
| /* "Bootstrap" allocations are not tracked by memory manager; should be used |
| only to allocate MemoryManager itself (or structure containing it). */ |
| BROTLI_INTERNAL void* BrotliBootstrapAlloc(size_t size, |
| brotli_alloc_func alloc_func, brotli_free_func free_func, void* opaque); |
| BROTLI_INTERNAL void BrotliBootstrapFree(void* address, MemoryManager* m); |
| |
| #if defined(__cplusplus) || defined(c_plusplus) |
| } /* extern "C" */ |
| #endif |
| |
| #endif /* BROTLI_ENC_MEMORY_H_ */ |