123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183 |
- /*
- +----------------------------------------------------------------------+
- | PHP Version 7 |
- +----------------------------------------------------------------------+
- | Copyright (c) 1997-2018 The PHP Group |
- +----------------------------------------------------------------------+
- | This source file is subject to version 3.01 of the PHP license, |
- | that is bundled with this package in the file LICENSE, and is |
- | available through the world-wide-web at the following url: |
- | http://www.php.net/license/3_01.txt |
- | If you did not receive a copy of the PHP license and are unable to |
- | obtain it through the world-wide-web, please send a note to |
- | license@php.net so we can mail you a copy immediately. |
- +----------------------------------------------------------------------+
- | Author: Dmitry Stogov <dmitry@php.net> |
- +----------------------------------------------------------------------+
- */
- #include <zend.h>
- #include "zend_smart_str.h"
- #include "zend_smart_string.h"
- #define SMART_STR_OVERHEAD (ZEND_MM_OVERHEAD + _ZSTR_HEADER_SIZE + 1)
- #define SMART_STR_START_SIZE 256
- #define SMART_STR_START_LEN (SMART_STR_START_SIZE - SMART_STR_OVERHEAD)
- #define SMART_STR_PAGE 4096
- #define SMART_STR_NEW_LEN(len) \
- (ZEND_MM_ALIGNED_SIZE_EX(len + SMART_STR_OVERHEAD, SMART_STR_PAGE) - SMART_STR_OVERHEAD)
- ZEND_API void ZEND_FASTCALL smart_str_erealloc(smart_str *str, size_t len)
- {
- if (UNEXPECTED(!str->s)) {
- str->a = len <= SMART_STR_START_LEN
- ? SMART_STR_START_LEN
- : SMART_STR_NEW_LEN(len);
- str->s = zend_string_alloc(str->a, 0);
- ZSTR_LEN(str->s) = 0;
- } else {
- str->a = SMART_STR_NEW_LEN(len);
- str->s = (zend_string *) erealloc2(str->s, str->a + _ZSTR_HEADER_SIZE + 1, _ZSTR_HEADER_SIZE + ZSTR_LEN(str->s));
- }
- }
- ZEND_API void ZEND_FASTCALL smart_str_realloc(smart_str *str, size_t len)
- {
- if (UNEXPECTED(!str->s)) {
- str->a = len <= SMART_STR_START_SIZE
- ? SMART_STR_START_LEN
- : SMART_STR_NEW_LEN(len);
- str->s = zend_string_alloc(str->a, 1);
- ZSTR_LEN(str->s) = 0;
- } else {
- str->a = SMART_STR_NEW_LEN(len);
- str->s = (zend_string *) perealloc(str->s, str->a + _ZSTR_HEADER_SIZE + 1, 1);
- }
- }
- /* Windows uses VK_ESCAPE instead of \e */
- #ifndef VK_ESCAPE
- #define VK_ESCAPE '\e'
- #endif
- static size_t zend_compute_escaped_string_len(const char *s, size_t l) {
- size_t i, len = l;
- for (i = 0; i < l; ++i) {
- char c = s[i];
- if (c == '\n' || c == '\r' || c == '\t' ||
- c == '\f' || c == '\v' || c == '\\' || c == VK_ESCAPE) {
- len += 1;
- } else if (c < 32 || c > 126) {
- len += 3;
- }
- }
- return len;
- }
- ZEND_API void ZEND_FASTCALL smart_str_append_escaped(smart_str *str, const char *s, size_t l) {
- char *res;
- size_t i, len = zend_compute_escaped_string_len(s, l);
- smart_str_alloc(str, len, 0);
- res = &ZSTR_VAL(str->s)[ZSTR_LEN(str->s)];
- ZSTR_LEN(str->s) += len;
- for (i = 0; i < l; ++i) {
- unsigned char c = s[i];
- if (c < 32 || c == '\\' || c > 126) {
- *res++ = '\\';
- switch (c) {
- case '\n': *res++ = 'n'; break;
- case '\r': *res++ = 'r'; break;
- case '\t': *res++ = 't'; break;
- case '\f': *res++ = 'f'; break;
- case '\v': *res++ = 'v'; break;
- case '\\': *res++ = '\\'; break;
- case VK_ESCAPE: *res++ = 'e'; break;
- default:
- *res++ = 'x';
- if ((c >> 4) < 10) {
- *res++ = (c >> 4) + '0';
- } else {
- *res++ = (c >> 4) + 'A' - 10;
- }
- if ((c & 0xf) < 10) {
- *res++ = (c & 0xf) + '0';
- } else {
- *res++ = (c & 0xf) + 'A' - 10;
- }
- }
- } else {
- *res++ = c;
- }
- }
- }
- ZEND_API void smart_str_append_printf(smart_str *dest, const char *format, ...) {
- va_list arg;
- va_start(arg, format);
- zend_printf_to_smart_str(dest, format, arg);
- va_end(arg);
- }
- #define SMART_STRING_OVERHEAD (ZEND_MM_OVERHEAD + 1)
- #define SMART_STRING_START_SIZE 256
- #define SMART_STRING_START_LEN (SMART_STRING_START_SIZE - SMART_STRING_OVERHEAD)
- #define SMART_STRING_PAGE 4096
- ZEND_API void ZEND_FASTCALL _smart_string_alloc_persistent(smart_string *str, size_t len)
- {
- if (!str->c) {
- str->len = 0;
- if (len <= SMART_STRING_START_LEN) {
- str->a = SMART_STRING_START_LEN;
- } else {
- str->a = ZEND_MM_ALIGNED_SIZE_EX(len + SMART_STRING_OVERHEAD, SMART_STRING_PAGE) - SMART_STRING_OVERHEAD;
- }
- str->c = pemalloc(str->a + 1, 1);
- } else {
- if (UNEXPECTED((size_t) len > SIZE_MAX - str->len)) {
- zend_error(E_ERROR, "String size overflow");
- }
- len += str->len;
- str->a = ZEND_MM_ALIGNED_SIZE_EX(len + SMART_STRING_OVERHEAD, SMART_STRING_PAGE) - SMART_STRING_OVERHEAD;
- str->c = perealloc(str->c, str->a + 1, 1);
- }
- }
- ZEND_API void ZEND_FASTCALL _smart_string_alloc(smart_string *str, size_t len)
- {
- if (!str->c) {
- str->len = 0;
- if (len <= SMART_STRING_START_LEN) {
- str->a = SMART_STRING_START_LEN;
- str->c = emalloc(SMART_STRING_START_LEN + 1);
- } else {
- str->a = ZEND_MM_ALIGNED_SIZE_EX(len + SMART_STRING_OVERHEAD, SMART_STRING_PAGE) - SMART_STRING_OVERHEAD;
- if (EXPECTED(str->a < (ZEND_MM_CHUNK_SIZE - SMART_STRING_OVERHEAD))) {
- str->c = emalloc_large(str->a + 1);
- } else {
- /* allocate a huge chunk */
- str->c = emalloc(str->a + 1);
- }
- }
- } else {
- if (UNEXPECTED((size_t) len > SIZE_MAX - str->len)) {
- zend_error(E_ERROR, "String size overflow");
- }
- len += str->len;
- str->a = ZEND_MM_ALIGNED_SIZE_EX(len + SMART_STRING_OVERHEAD, SMART_STRING_PAGE) - SMART_STRING_OVERHEAD;
- str->c = erealloc2(str->c, str->a + 1, str->len);
- }
- }
- /*
- * Local variables:
- * tab-width: 4
- * c-basic-offset: 4
- * indent-tabs-mode: t
- * End:
- * vim600: sw=4 ts=4 fdm=marker
- * vim<600: sw=4 ts=4
- */
|