This source file includes following definitions.
- ZSTD_initStack
- ZSTD_stackAllocAll
- ZSTD_stackAlloc
- ZSTD_stackFree
- ZSTD_malloc
- ZSTD_free
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20 #include "error_private.h"
21 #include "zstd_internal.h"
22 #include <linux/kernel.h>
23
24
25
26
27
28 #define stack_push(stack, size) \
29 ({ \
30 void *const ptr = ZSTD_PTR_ALIGN((stack)->ptr); \
31 (stack)->ptr = (char *)ptr + (size); \
32 (stack)->ptr <= (stack)->end ? ptr : NULL; \
33 })
34
35 ZSTD_customMem ZSTD_initStack(void *workspace, size_t workspaceSize)
36 {
37 ZSTD_customMem stackMem = {ZSTD_stackAlloc, ZSTD_stackFree, workspace};
38 ZSTD_stack *stack = (ZSTD_stack *)workspace;
39
40 if (!workspace || workspaceSize < sizeof(ZSTD_stack) || workspace != ZSTD_PTR_ALIGN(workspace)) {
41 ZSTD_customMem error = {NULL, NULL, NULL};
42 return error;
43 }
44
45 stack->ptr = workspace;
46 stack->end = (char *)workspace + workspaceSize;
47 stack_push(stack, sizeof(ZSTD_stack));
48 return stackMem;
49 }
50
51 void *ZSTD_stackAllocAll(void *opaque, size_t *size)
52 {
53 ZSTD_stack *stack = (ZSTD_stack *)opaque;
54 *size = (BYTE const *)stack->end - (BYTE *)ZSTD_PTR_ALIGN(stack->ptr);
55 return stack_push(stack, *size);
56 }
57
58 void *ZSTD_stackAlloc(void *opaque, size_t size)
59 {
60 ZSTD_stack *stack = (ZSTD_stack *)opaque;
61 return stack_push(stack, size);
62 }
63 void ZSTD_stackFree(void *opaque, void *address)
64 {
65 (void)opaque;
66 (void)address;
67 }
68
69 void *ZSTD_malloc(size_t size, ZSTD_customMem customMem) { return customMem.customAlloc(customMem.opaque, size); }
70
71 void ZSTD_free(void *ptr, ZSTD_customMem customMem)
72 {
73 if (ptr != NULL)
74 customMem.customFree(customMem.opaque, ptr);
75 }