zstd_common.c 2.0 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465
  1. // SPDX-License-Identifier: (GPL-2.0 or BSD-3-Clause-Clear)
  2. /**
  3. * Copyright (c) 2016-present, Yann Collet, Facebook, Inc.
  4. * All rights reserved.
  5. */
  6. /*-*************************************
  7. * Dependencies
  8. ***************************************/
  9. #include "error_private.h"
  10. #include "zstd_internal.h" /* declaration of ZSTD_isError, ZSTD_getErrorName, ZSTD_getErrorCode, ZSTD_getErrorString, ZSTD_versionNumber */
  11. #include <linux/kernel.h>
  12. /*=**************************************************************
  13. * Custom allocator
  14. ****************************************************************/
  15. #define stack_push(stack, size) \
  16. ({ \
  17. void *const ptr = ZSTD_PTR_ALIGN((stack)->ptr); \
  18. (stack)->ptr = (char *)ptr + (size); \
  19. (stack)->ptr <= (stack)->end ? ptr : NULL; \
  20. })
  21. ZSTD_customMem ZSTD_initStack(void *workspace, size_t workspaceSize)
  22. {
  23. ZSTD_customMem stackMem = {ZSTD_stackAlloc, ZSTD_stackFree, workspace};
  24. ZSTD_stack *stack = (ZSTD_stack *)workspace;
  25. /* Verify preconditions */
  26. if (!workspace || workspaceSize < sizeof(ZSTD_stack) || workspace != ZSTD_PTR_ALIGN(workspace)) {
  27. ZSTD_customMem error = {NULL, NULL, NULL};
  28. return error;
  29. }
  30. /* Initialize the stack */
  31. stack->ptr = workspace;
  32. stack->end = (char *)workspace + workspaceSize;
  33. stack_push(stack, sizeof(ZSTD_stack));
  34. return stackMem;
  35. }
  36. void *ZSTD_stackAllocAll(void *opaque, size_t *size)
  37. {
  38. ZSTD_stack *stack = (ZSTD_stack *)opaque;
  39. *size = (BYTE const *)stack->end - (BYTE *)ZSTD_PTR_ALIGN(stack->ptr);
  40. return stack_push(stack, *size);
  41. }
  42. void *ZSTD_stackAlloc(void *opaque, size_t size)
  43. {
  44. ZSTD_stack *stack = (ZSTD_stack *)opaque;
  45. return stack_push(stack, size);
  46. }
  47. void ZSTD_stackFree(void *opaque, void *address)
  48. {
  49. (void)opaque;
  50. (void)address;
  51. }
  52. void *ZSTD_malloc(size_t size, ZSTD_customMem customMem) { return customMem.customAlloc(customMem.opaque, size); }
  53. void ZSTD_free(void *ptr, ZSTD_customMem customMem)
  54. {
  55. if (ptr != NULL)
  56. customMem.customFree(customMem.opaque, ptr);
  57. }