mysqlnd_alloc.h 4.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798
  1. /*
  2. +----------------------------------------------------------------------+
  3. | Copyright (c) The PHP Group |
  4. +----------------------------------------------------------------------+
  5. | This source file is subject to version 3.01 of the PHP license, |
  6. | that is bundled with this package in the file LICENSE, and is |
  7. | available through the world-wide-web at the following url: |
  8. | https://www.php.net/license/3_01.txt |
  9. | If you did not receive a copy of the PHP license and are unable to |
  10. | obtain it through the world-wide-web, please send a note to |
  11. | license@php.net so we can mail you a copy immediately. |
  12. +----------------------------------------------------------------------+
  13. | Authors: Andrey Hristov <andrey@php.net> |
  14. | Ulf Wendel <uw@php.net> |
  15. +----------------------------------------------------------------------+
  16. */
  17. #ifndef MYSQLND_ALLOC_H
  18. #define MYSQLND_ALLOC_H
  19. PHPAPI extern const char * mysqlnd_debug_std_no_trace_funcs[];
  20. #define MYSQLND_MEM_D ZEND_FILE_LINE_DC
  21. #define MYSQLND_MEM_C ZEND_FILE_LINE_CC
  22. struct st_mysqlnd_allocator_methods
  23. {
  24. void * (*m_emalloc)(size_t size MYSQLND_MEM_D);
  25. void * (*m_pemalloc)(size_t size, bool persistent MYSQLND_MEM_D);
  26. void * (*m_ecalloc)(unsigned int nmemb, size_t size MYSQLND_MEM_D);
  27. void * (*m_pecalloc)(unsigned int nmemb, size_t size, bool persistent MYSQLND_MEM_D);
  28. void * (*m_erealloc)(void *ptr, size_t new_size MYSQLND_MEM_D);
  29. void * (*m_perealloc)(void *ptr, size_t new_size, bool persistent MYSQLND_MEM_D);
  30. void (*m_efree)(void *ptr MYSQLND_MEM_D);
  31. void (*m_pefree)(void *ptr, bool persistent MYSQLND_MEM_D);
  32. char * (*m_pememdup)(const char * const ptr, size_t size, bool persistent MYSQLND_MEM_D);
  33. char * (*m_pestrndup)(const char * const ptr, size_t size, bool persistent MYSQLND_MEM_D);
  34. char * (*m_pestrdup)(const char * const ptr, bool persistent MYSQLND_MEM_D);
  35. };
  36. PHPAPI extern struct st_mysqlnd_allocator_methods mysqlnd_allocator;
  37. #define mnd_emalloc(size) mysqlnd_allocator.m_emalloc((size) MYSQLND_MEM_C)
  38. #define mnd_pemalloc(size, pers) mysqlnd_allocator.m_pemalloc((size), (pers) MYSQLND_MEM_C)
  39. #define mnd_ecalloc(nmemb, size) mysqlnd_allocator.m_ecalloc((nmemb), (size) MYSQLND_MEM_C)
  40. #define mnd_pecalloc(nmemb, size, p) mysqlnd_allocator.m_pecalloc((nmemb), (size), (p) MYSQLND_MEM_C)
  41. #define mnd_erealloc(ptr, new_size) mysqlnd_allocator.m_erealloc((ptr), (new_size) MYSQLND_MEM_C)
  42. #define mnd_perealloc(ptr, new_size, p) mysqlnd_allocator.m_perealloc((ptr), (new_size), (p) MYSQLND_MEM_C)
  43. #define mnd_efree(ptr) mysqlnd_allocator.m_efree((ptr) MYSQLND_MEM_C)
  44. #define mnd_pefree(ptr, pers) mysqlnd_allocator.m_pefree((ptr), (pers) MYSQLND_MEM_C)
  45. #define mnd_pememdup(ptr, size, pers) mysqlnd_allocator.m_pememdup((ptr), (size), (pers) MYSQLND_MEM_C)
  46. #define mnd_pestrndup(ptr, size, pers) mysqlnd_allocator.m_pestrndup((ptr), (size), (pers) MYSQLND_MEM_C)
  47. #define mnd_pestrdup(ptr, pers) mysqlnd_allocator.m_pestrdup((ptr), (pers) MYSQLND_MEM_C)
  48. #define mnd_sprintf(p, mx_len, fmt,...) spprintf((p), (mx_len), (fmt), __VA_ARGS__)
  49. #define mnd_vsprintf(p, mx_len, fmt,ap) vspprintf((p), (mx_len), (fmt), (ap))
  50. #define mnd_sprintf_free(p) efree((p))
  51. static inline MYSQLND_STRING mnd_dup_cstring(const MYSQLND_CSTRING str, const bool persistent)
  52. {
  53. const MYSQLND_STRING ret = {(char*) mnd_pemalloc(str.l + 1, persistent), str.l};
  54. if (ret.s) {
  55. memcpy(ret.s, str.s, str.l);
  56. ret.s[str.l] = '\0';
  57. }
  58. return ret;
  59. }
  60. static inline MYSQLND_CSTRING mnd_str2c(const MYSQLND_STRING str)
  61. {
  62. const MYSQLND_CSTRING ret = {str.s, str.l};
  63. return ret;
  64. }
  65. static inline void mysqlnd_set_string(MYSQLND_STRING *buf, const char *string, size_t len) {
  66. if (buf->s) {
  67. mnd_efree(buf->s);
  68. buf->s = NULL;
  69. buf->l = 0;
  70. }
  71. if (string) {
  72. buf->s = mnd_pestrndup(string, len, 0);
  73. buf->l = len;
  74. }
  75. }
  76. static inline void mysqlnd_set_persistent_string(MYSQLND_STRING *buf, const char *string, size_t len, bool persistent) {
  77. if (buf->s) {
  78. mnd_pefree(buf->s, persistent);
  79. buf->s = NULL;
  80. buf->l = 0;
  81. }
  82. if (string) {
  83. buf->s = mnd_pestrndup(string, len, persistent);
  84. buf->l = len;
  85. }
  86. }
  87. #endif /* MYSQLND_ALLOC_H */