alloc.h 3.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596
  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. #ifndef HIREDIS_ALLOC_H
  31. #define HIREDIS_ALLOC_H
  32. #include <stddef.h> /* for size_t */
  33. #include <stdint.h>
  34. #ifdef __cplusplus
  35. extern "C" {
  36. #endif
  37. /* Structure pointing to our actually configured allocators */
  38. typedef struct hiredisAllocFuncs {
  39. void *(*mallocFn)(size_t);
  40. void *(*callocFn)(size_t,size_t);
  41. void *(*reallocFn)(void*,size_t);
  42. char *(*strdupFn)(const char*);
  43. void (*freeFn)(void*);
  44. } hiredisAllocFuncs;
  45. hiredisAllocFuncs hiredisSetAllocators(hiredisAllocFuncs *ha);
  46. void hiredisResetAllocators(void);
  47. #ifndef _WIN32
  48. /* Hiredis' configured allocator function pointer struct */
  49. extern hiredisAllocFuncs hiredisAllocFns;
  50. static inline void *hi_malloc(size_t size) {
  51. return hiredisAllocFns.mallocFn(size);
  52. }
  53. static inline void *hi_calloc(size_t nmemb, size_t size) {
  54. /* Overflow check as the user can specify any arbitrary allocator */
  55. if (SIZE_MAX / size < nmemb)
  56. return NULL;
  57. return hiredisAllocFns.callocFn(nmemb, size);
  58. }
  59. static inline void *hi_realloc(void *ptr, size_t size) {
  60. return hiredisAllocFns.reallocFn(ptr, size);
  61. }
  62. static inline char *hi_strdup(const char *str) {
  63. return hiredisAllocFns.strdupFn(str);
  64. }
  65. static inline void hi_free(void *ptr) {
  66. hiredisAllocFns.freeFn(ptr);
  67. }
  68. #else
  69. void *hi_malloc(size_t size);
  70. void *hi_calloc(size_t nmemb, size_t size);
  71. void *hi_realloc(void *ptr, size_t size);
  72. char *hi_strdup(const char *str);
  73. void hi_free(void *ptr);
  74. #endif
  75. #ifdef __cplusplus
  76. }
  77. #endif
  78. #endif /* HIREDIS_ALLOC_H */