hc
2024-08-16 a24a44ff9ca902811b99aa9663d697cf452e08ef
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
/**
 * Copyright (c) 2016-present, Yann Collet, Facebook, Inc.
 * All rights reserved.
 *
 * This source code is licensed under the BSD-style license found in the
 * LICENSE file in the root directory of https://github.com/facebook/zstd.
 * An additional grant of patent rights can be found in the PATENTS file in the
 * same directory.
 *
 * This program is free software; you can redistribute it and/or modify it under
 * the terms of the GNU General Public License version 2 as published by the
 * Free Software Foundation. This program is dual-licensed; you may select
 * either version 2 of the GNU General Public License ("GPL") or BSD license
 * ("BSD").
 */
 
/*-*************************************
*  Dependencies
***************************************/
#include "error_private.h"
#include "zstd_internal.h" /* declaration of ZSTD_isError, ZSTD_getErrorName, ZSTD_getErrorCode, ZSTD_getErrorString, ZSTD_versionNumber */
#include <linux/kernel.h>
 
/*=**************************************************************
*  Custom allocator
****************************************************************/
 
#define stack_push(stack, size)                                 \
   ({                                                      \
       void *const ptr = ZSTD_PTR_ALIGN((stack)->ptr); \
       (stack)->ptr = (char *)ptr + (size);            \
       (stack)->ptr <= (stack)->end ? ptr : NULL;      \
   })
 
ZSTD_customMem ZSTD_initStack(void *workspace, size_t workspaceSize)
{
   ZSTD_customMem stackMem = {ZSTD_stackAlloc, ZSTD_stackFree, workspace};
   ZSTD_stack *stack = (ZSTD_stack *)workspace;
   /* Verify preconditions */
   if (!workspace || workspaceSize < sizeof(ZSTD_stack) || workspace != ZSTD_PTR_ALIGN(workspace)) {
       ZSTD_customMem error = {NULL, NULL, NULL};
       return error;
   }
   /* Initialize the stack */
   stack->ptr = workspace;
   stack->end = (char *)workspace + workspaceSize;
   stack_push(stack, sizeof(ZSTD_stack));
   return stackMem;
}
 
void *ZSTD_stackAllocAll(void *opaque, size_t *size)
{
   ZSTD_stack *stack = (ZSTD_stack *)opaque;
   *size = (BYTE const *)stack->end - (BYTE *)ZSTD_PTR_ALIGN(stack->ptr);
   return stack_push(stack, *size);
}
 
void *ZSTD_stackAlloc(void *opaque, size_t size)
{
   ZSTD_stack *stack = (ZSTD_stack *)opaque;
   return stack_push(stack, size);
}
void ZSTD_stackFree(void *opaque, void *address)
{
   (void)opaque;
   (void)address;
}
 
void *ZSTD_malloc(size_t size, ZSTD_customMem customMem) { return customMem.customAlloc(customMem.opaque, size); }
 
void ZSTD_free(void *ptr, ZSTD_customMem customMem)
{
   if (ptr != NULL)
       customMem.customFree(customMem.opaque, ptr);
}