curl_memory.h 4.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157
  1. #ifndef HEADER_CURL_MEMORY_H
  2. #define HEADER_CURL_MEMORY_H
  3. /***************************************************************************
  4. * _ _ ____ _
  5. * Project ___| | | | _ \| |
  6. * / __| | | | |_) | |
  7. * | (__| |_| | _ <| |___
  8. * \___|\___/|_| \_\_____|
  9. *
  10. * Copyright (C) Daniel Stenberg, <[email protected]>, et al.
  11. *
  12. * This software is licensed as described in the file COPYING, which
  13. * you should have received as part of this distribution. The terms
  14. * are also available at https://curl.se/docs/copyright.html.
  15. *
  16. * You may opt to use, copy, modify, merge, publish, distribute and/or sell
  17. * copies of the Software, and permit persons to whom the Software is
  18. * furnished to do so, under the terms of the COPYING file.
  19. *
  20. * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
  21. * KIND, either express or implied.
  22. *
  23. * SPDX-License-Identifier: curl
  24. *
  25. ***************************************************************************/
  26. /*
  27. * Nasty internal details ahead...
  28. *
  29. * File curl_memory.h must be included by _all_ *.c source files
  30. * that use memory related functions strdup, malloc, calloc, realloc
  31. * or free, and given source file is used to build libcurl library.
  32. * It should be included immediately before memdebug.h as the last files
  33. * included to avoid undesired interaction with other memory function
  34. * headers in dependent libraries.
  35. *
  36. * There is nearly no exception to above rule. All libcurl source
  37. * files in 'lib' subdirectory as well as those living deep inside
  38. * 'packages' subdirectories and linked together in order to build
  39. * libcurl library shall follow it.
  40. *
  41. * File lib/strdup.c is an exception, given that it provides a strdup
  42. * clone implementation while using malloc. Extra care needed inside
  43. * this one.
  44. *
  45. * The need for curl_memory.h inclusion is due to libcurl's feature
  46. * of allowing library user to provide memory replacement functions,
  47. * memory callbacks, at runtime with curl_global_init_mem()
  48. *
  49. * Any *.c source file used to build libcurl library that does not
  50. * include curl_memory.h and uses any memory function of the five
  51. * mentioned above will compile without any indication, but it will
  52. * trigger weird memory related issues at runtime.
  53. *
  54. */
  55. #ifdef HEADER_CURL_MEMDEBUG_H
  56. /* cleanup after memdebug.h */
  57. #ifdef CURLDEBUG
  58. #undef strdup
  59. #undef malloc
  60. #undef calloc
  61. #undef realloc
  62. #undef free
  63. #undef send
  64. #undef recv
  65. #ifdef _WIN32
  66. #undef _tcsdup
  67. #endif
  68. #undef socket
  69. #undef accept
  70. #ifdef HAVE_ACCEPT4
  71. #undef accept4
  72. #endif
  73. #ifdef HAVE_SOCKETPAIR
  74. #undef socketpair
  75. #endif
  76. /* sclose is probably already defined, redefine it! */
  77. #undef sclose
  78. #define sclose(x) CURL_SCLOSE(x)
  79. #undef fopen
  80. #ifdef CURL_FOPEN
  81. #define fopen(fname, mode) CURL_FOPEN(fname, mode)
  82. #endif
  83. #undef fdopen
  84. #undef fclose
  85. #endif /* CURLDEBUG */
  86. #undef HEADER_CURL_MEMDEBUG_H
  87. #endif /* HEADER_CURL_MEMDEBUG_H */
  88. /*
  89. ** Following section applies even when CURLDEBUG is not defined.
  90. */
  91. #undef fake_sclose
  92. #ifndef CURL_DID_MEMORY_FUNC_TYPEDEFS /* only if not already done */
  93. /*
  94. * The following memory function replacement typedef's are COPIED from
  95. * curl/curl.h and MUST match the originals. We copy them to avoid having to
  96. * include curl/curl.h here. We avoid that include since it includes stdio.h
  97. * and other headers that may get messed up with defines done here.
  98. */
  99. typedef void *(*curl_malloc_callback)(size_t size);
  100. typedef void (*curl_free_callback)(void *ptr);
  101. typedef void *(*curl_realloc_callback)(void *ptr, size_t size);
  102. typedef char *(*curl_strdup_callback)(const char *str);
  103. typedef void *(*curl_calloc_callback)(size_t nmemb, size_t size);
  104. #define CURL_DID_MEMORY_FUNC_TYPEDEFS
  105. #endif
  106. extern curl_malloc_callback Curl_cmalloc;
  107. extern curl_free_callback Curl_cfree;
  108. extern curl_realloc_callback Curl_crealloc;
  109. extern curl_strdup_callback Curl_cstrdup;
  110. extern curl_calloc_callback Curl_ccalloc;
  111. #ifndef CURLDEBUG
  112. /*
  113. * libcurl's 'memory tracking' system defines strdup, malloc, calloc,
  114. * realloc and free, along with others, in memdebug.h in a different
  115. * way although still using memory callbacks forward declared above.
  116. * When using the 'memory tracking' system (CURLDEBUG defined) we do
  117. * not define here the five memory functions given that definitions
  118. * from memdebug.h are the ones that shall be used.
  119. */
  120. #undef strdup
  121. #define strdup(ptr) Curl_cstrdup(ptr)
  122. #undef malloc
  123. #define malloc(size) Curl_cmalloc(size)
  124. #undef calloc
  125. #define calloc(nbelem,size) Curl_ccalloc(nbelem, size)
  126. #undef realloc
  127. #define realloc(ptr,size) Curl_crealloc(ptr, size)
  128. #undef free
  129. #define free(ptr) Curl_cfree(ptr)
  130. #ifdef _WIN32
  131. #undef _tcsdup
  132. #ifdef UNICODE
  133. #define _tcsdup(ptr) Curl_wcsdup(ptr)
  134. #else
  135. #define _tcsdup(ptr) Curl_cstrdup(ptr)
  136. #endif
  137. #endif /* _WIN32 */
  138. #endif /* CURLDEBUG */
  139. #endif /* HEADER_CURL_MEMORY_H */