alloc.c 2.7 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586
  1. /*
  2. * Copyright (c) 2020, Michael Grunder <michael dot grunder at gmail dot com>
  3. *
  4. * All rights reserved.
  5. *
  6. * Redistribution and use in source and binary forms, with or without
  7. * modification, are permitted provided that the following conditions are met:
  8. *
  9. * * Redistributions of source code must retain the above copyright notice,
  10. * this list of conditions and the following disclaimer.
  11. * * Redistributions in binary form must reproduce the above copyright
  12. * notice, this list of conditions and the following disclaimer in the
  13. * documentation and/or other materials provided with the distribution.
  14. * * Neither the name of Redis nor the names of its contributors may be used
  15. * to endorse or promote products derived from this software without
  16. * specific prior written permission.
  17. *
  18. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  19. * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  20. * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  21. * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
  22. * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  23. * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  24. * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  25. * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  26. * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  27. * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  28. * POSSIBILITY OF SUCH DAMAGE.
  29. */
  30. #include "fmacros.h"
  31. #include "alloc.h"
  32. #include <string.h>
  33. #include <stdlib.h>
  34. hiredisAllocFuncs hiredisAllocFns = {
  35. .mallocFn = malloc,
  36. .callocFn = calloc,
  37. .reallocFn = realloc,
  38. .strdupFn = strdup,
  39. .freeFn = free,
  40. };
  41. /* Override hiredis' allocators with ones supplied by the user */
  42. hiredisAllocFuncs hiredisSetAllocators(hiredisAllocFuncs *override) {
  43. hiredisAllocFuncs orig = hiredisAllocFns;
  44. hiredisAllocFns = *override;
  45. return orig;
  46. }
  47. /* Reset allocators to use libc defaults */
  48. void hiredisResetAllocators(void) {
  49. hiredisAllocFns = (hiredisAllocFuncs) {
  50. .mallocFn = malloc,
  51. .callocFn = calloc,
  52. .reallocFn = realloc,
  53. .strdupFn = strdup,
  54. .freeFn = free,
  55. };
  56. }
  57. #ifdef _WIN32
  58. void *hi_malloc(size_t size) {
  59. return hiredisAllocFns.mallocFn(size);
  60. }
  61. void *hi_calloc(size_t nmemb, size_t size) {
  62. return hiredisAllocFns.callocFn(nmemb, size);
  63. }
  64. void *hi_realloc(void *ptr, size_t size) {
  65. return hiredisAllocFns.reallocFn(ptr, size);
  66. }
  67. char *hi_strdup(const char *str) {
  68. return hiredisAllocFns.strdupFn(str);
  69. }
  70. void hi_free(void *ptr) {
  71. hiredisAllocFns.freeFn(ptr);
  72. }
  73. #endif