zend_smart_str.h 5.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162
  1. /*
  2. +----------------------------------------------------------------------+
  3. | PHP Version 7 |
  4. +----------------------------------------------------------------------+
  5. | Copyright (c) 1997-2018 The PHP Group |
  6. +----------------------------------------------------------------------+
  7. | This source file is subject to version 3.01 of the PHP license, |
  8. | that is bundled with this package in the file LICENSE, and is |
  9. | available through the world-wide-web at the following url: |
  10. | http://www.php.net/license/3_01.txt |
  11. | If you did not receive a copy of the PHP license and are unable to |
  12. | obtain it through the world-wide-web, please send a note to |
  13. | license@php.net so we can mail you a copy immediately. |
  14. +----------------------------------------------------------------------+
  15. | Author: Sascha Schumann <sascha@schumann.cx> |
  16. +----------------------------------------------------------------------+
  17. */
  18. #ifndef ZEND_SMART_STR_H
  19. #define ZEND_SMART_STR_H
  20. #include <zend.h>
  21. #include "zend_globals.h"
  22. #include "zend_smart_str_public.h"
  23. #define smart_str_appends_ex(dest, src, what) \
  24. smart_str_appendl_ex((dest), (src), strlen(src), (what))
  25. #define smart_str_appends(dest, src) \
  26. smart_str_appendl((dest), (src), strlen(src))
  27. #define smart_str_extend(dest, len) \
  28. smart_str_extend_ex((dest), (len), 0)
  29. #define smart_str_appendc(dest, c) \
  30. smart_str_appendc_ex((dest), (c), 0)
  31. #define smart_str_appendl(dest, src, len) \
  32. smart_str_appendl_ex((dest), (src), (len), 0)
  33. #define smart_str_append(dest, src) \
  34. smart_str_append_ex((dest), (src), 0)
  35. #define smart_str_append_smart_str(dest, src) \
  36. smart_str_append_smart_str_ex((dest), (src), 0)
  37. #define smart_str_sets(dest, src) \
  38. smart_str_setl((dest), (src), strlen(src));
  39. #define smart_str_append_long(dest, val) \
  40. smart_str_append_long_ex((dest), (val), 0)
  41. #define smart_str_append_unsigned(dest, val) \
  42. smart_str_append_unsigned_ex((dest), (val), 0)
  43. #define smart_str_free(dest) \
  44. smart_str_free_ex((dest), 0)
  45. BEGIN_EXTERN_C()
  46. ZEND_API void ZEND_FASTCALL smart_str_erealloc(smart_str *str, size_t len);
  47. ZEND_API void ZEND_FASTCALL smart_str_realloc(smart_str *str, size_t len);
  48. ZEND_API void ZEND_FASTCALL smart_str_append_escaped(smart_str *str, const char *s, size_t l);
  49. ZEND_API void smart_str_append_printf(smart_str *dest, const char *format, ...)
  50. ZEND_ATTRIBUTE_FORMAT(printf, 2, 3);
  51. END_EXTERN_C()
  52. static zend_always_inline size_t smart_str_alloc(smart_str *str, size_t len, zend_bool persistent) {
  53. if (UNEXPECTED(!str->s)) {
  54. goto do_smart_str_realloc;
  55. } else {
  56. len += ZSTR_LEN(str->s);
  57. if (UNEXPECTED(len >= str->a)) {
  58. do_smart_str_realloc:
  59. if (persistent) {
  60. smart_str_realloc(str, len);
  61. } else {
  62. smart_str_erealloc(str, len);
  63. }
  64. }
  65. }
  66. return len;
  67. }
  68. static zend_always_inline char* smart_str_extend_ex(smart_str *dest, size_t len, zend_bool persistent) {
  69. size_t new_len = smart_str_alloc(dest, len, persistent);
  70. char *ret = ZSTR_VAL(dest->s) + ZSTR_LEN(dest->s);
  71. ZSTR_LEN(dest->s) = new_len;
  72. return ret;
  73. }
  74. static zend_always_inline void smart_str_free_ex(smart_str *str, zend_bool persistent) {
  75. if (str->s) {
  76. zend_string_release_ex(str->s, persistent);
  77. str->s = NULL;
  78. }
  79. str->a = 0;
  80. }
  81. static zend_always_inline void smart_str_0(smart_str *str) {
  82. if (str->s) {
  83. ZSTR_VAL(str->s)[ZSTR_LEN(str->s)] = '\0';
  84. }
  85. }
  86. static zend_always_inline size_t smart_str_get_len(smart_str *str) {
  87. return str->s ? ZSTR_LEN(str->s) : 0;
  88. }
  89. static zend_always_inline zend_string *smart_str_extract(smart_str *str) {
  90. if (str->s) {
  91. zend_string *res;
  92. smart_str_0(str);
  93. res = str->s;
  94. str->s = NULL;
  95. return res;
  96. } else {
  97. return ZSTR_EMPTY_ALLOC();
  98. }
  99. }
  100. static zend_always_inline void smart_str_appendc_ex(smart_str *dest, char ch, zend_bool persistent) {
  101. size_t new_len = smart_str_alloc(dest, 1, persistent);
  102. ZSTR_VAL(dest->s)[new_len - 1] = ch;
  103. ZSTR_LEN(dest->s) = new_len;
  104. }
  105. static zend_always_inline void smart_str_appendl_ex(smart_str *dest, const char *str, size_t len, zend_bool persistent) {
  106. size_t new_len = smart_str_alloc(dest, len, persistent);
  107. memcpy(ZSTR_VAL(dest->s) + ZSTR_LEN(dest->s), str, len);
  108. ZSTR_LEN(dest->s) = new_len;
  109. }
  110. static zend_always_inline void smart_str_append_ex(smart_str *dest, const zend_string *src, zend_bool persistent) {
  111. smart_str_appendl_ex(dest, ZSTR_VAL(src), ZSTR_LEN(src), persistent);
  112. }
  113. static zend_always_inline void smart_str_append_smart_str_ex(smart_str *dest, const smart_str *src, zend_bool persistent) {
  114. if (src->s && ZSTR_LEN(src->s)) {
  115. smart_str_append_ex(dest, src->s, persistent);
  116. }
  117. }
  118. static zend_always_inline void smart_str_append_long_ex(smart_str *dest, zend_long num, zend_bool persistent) {
  119. char buf[32];
  120. char *result = zend_print_long_to_buf(buf + sizeof(buf) - 1, num);
  121. smart_str_appendl_ex(dest, result, buf + sizeof(buf) - 1 - result, persistent);
  122. }
  123. static zend_always_inline void smart_str_append_unsigned_ex(smart_str *dest, zend_ulong num, zend_bool persistent) {
  124. char buf[32];
  125. char *result = zend_print_ulong_to_buf(buf + sizeof(buf) - 1, num);
  126. smart_str_appendl_ex(dest, result, buf + sizeof(buf) - 1 - result, persistent);
  127. }
  128. static zend_always_inline void smart_str_setl(smart_str *dest, const char *src, size_t len) {
  129. smart_str_free(dest);
  130. smart_str_appendl(dest, src, len);
  131. }
  132. #endif
  133. /*
  134. * Local variables:
  135. * tab-width: 4
  136. * c-basic-offset: 4
  137. * indent-tabs-mode: t
  138. * End:
  139. * vim600: sw=4 ts=4 fdm=marker
  140. * vim<600: sw=4 ts=4
  141. */