1 /*******************************************************************************
2 Copyright (c) 2016, Michael Wallner <mike@php.net>.
5 Redistribution and use in source and binary forms, with or without
6 modification, are permitted provided that the following conditions are met:
8 * Redistributions of source code must retain the above copyright notice,
9 this list of conditions and the following disclaimer.
10 * Redistributions in binary form must reproduce the above copyright
11 notice, this list of conditions and the following disclaimer in the
12 documentation and/or other materials provided with the distribution.
14 THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
15 AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16 IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
17 DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE
18 FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19 DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
20 SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
21 CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
22 OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
23 OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
24 *******************************************************************************/
26 #include "php_psi_stdinc.h"
30 #include <Zend/zend_smart_str.h>
35 #ifndef PSI_DEBUG_TOKEN_ALLOC
36 # define PSI_DEBUG_TOKEN_ALLOC 0
39 struct psi_token
*psi_token_init(token_t token_typ
, const char *token_txt
,
40 size_t token_len
, unsigned col
, unsigned line
, zend_string
*file
)
44 T
= pecalloc(1, sizeof(*T
), 1);
48 T
->file
= zend_string_copy(file
);
49 T
->text
= zend_string_init_interned(token_txt
, token_len
, 1);
50 #if PSI_DEBUG_TOKEN_ALLOC
51 fprintf(stderr
, "PSI: token_init %p\n", T
);
56 void psi_token_free(struct psi_token
**token_ptr
) {
58 struct psi_token
*token
= *token_ptr
;
59 #if PSI_DEBUG_TOKEN_ALLOC
60 fprintf(stderr
, "PSI: token_free %p\n", token
);
63 zend_string_release(token
->text
);
64 zend_string_release(token
->file
);
69 struct psi_token
*psi_token_copy(struct psi_token
*src
) {
70 struct psi_token
*ptr
= pemalloc(sizeof(*ptr
), 1);
73 #if PSI_DEBUG_TOKEN_ALLOC
74 fprintf(stderr
, "PSI: token_copy %p <= %p\n", ptr
, src
);
76 ptr
->text
= zend_string_copy(ptr
->text
);
77 ptr
->file
= zend_string_copy(ptr
->file
);
82 void psi_token_copy_ctor(struct psi_token
**tok
) {
83 *tok
= psi_token_copy(*tok
);
86 /* concatenate `argc` number of tokens separated by `sep` into a newly allocated token */
87 struct psi_token
*psi_token_cat(const char *sep
, unsigned argc
, ...) {
90 size_t sep_len
= sep
? strlen(sep
) : 0;
91 struct psi_token
*T
= pemalloc(sizeof(*T
), 1);
96 *T
= *(struct psi_token
*) va_arg(argv
, struct psi_token
*);
98 T
->file
= zend_string_copy(T
->file
);
100 smart_str_append_ex(&text
, T
->text
, 1);
102 for (i
= 1; i
< argc
; ++i
) {
103 struct psi_token
*arg
= va_arg(argv
, struct psi_token
*);
105 smart_str_appendl_ex(&text
, sep
, sep_len
, 1);
106 smart_str_append_ex(&text
, arg
->text
, 1);
110 T
->text
= smart_str_extract(&text
);
112 #if PSI_DEBUG_TOKEN_ALLOC
113 fprintf(stderr
, "PSI: token_cat %p\n", T
);
118 /* append `argc` number of C strings separated by `sep` to token `T` */
119 struct psi_token
*psi_token_append(const char *sep
, struct psi_token
*T
, unsigned argc
, ...) {
122 size_t sep_len
= sep
? strlen(sep
) : 0;
123 smart_str text
= {0};
125 smart_str_append_ex(&text
, T
->text
, 1);
127 va_start(argv
, argc
);
128 for (i
= 0; i
< argc
; ++i
) {
129 char *str
= va_arg(argv
, char *);
130 size_t str_len
= strlen(str
);
132 if (sep_len
&& text
.a
) {
133 smart_str_appendl_ex(&text
, sep
, sep_len
, 1);
136 smart_str_appendl_ex(&text
, str
, str_len
, 1);
140 zend_string_release(T
->text
);
141 T
->text
= smart_str_extract(&text
);
146 char *php_strtr(char *str
, size_t len
, char *str_from
, char *str_to
, size_t trlen
);
147 struct psi_token
*psi_token_translit(struct psi_token
*T
, char *from
, char *to
) {
148 zend_string
*tmp
= zend_string_init(T
->text
->val
, T
->text
->len
, 1);
150 zend_string_release(T
->text
);
153 php_strtr(T
->text
->val
, T
->text
->len
, from
, to
, MIN(strlen(from
), strlen(to
)));
154 zend_string_forget_hash_val(T
->text
);
159 static inline uint64_t psi_hash(char *digest_buf
, ...)
161 uint64_t hash
= 5381;
166 va_start(argv
, digest_buf
);
167 while ((ptr
= va_arg(argv
, const uint8_t *))) {
168 while ((c
= *ptr
++)) {
169 hash
= ((hash
<< 5) + hash
) + c
;
175 sprintf(digest_buf
, "%" PRIx64
, hash
);
181 uint64_t psi_token_hash(struct psi_token
*t
, char *digest_buf
) {
184 sprintf(loc_buf
, "%u%u", t
->line
, t
->col
);
185 return psi_hash(digest_buf
, t
->file
->val
, loc_buf
, (char *) NULL
);
188 void psi_token_dump(int fd
, struct psi_token
*t
)
192 dprintf(fd
, "TOKEN %p (%u) ", t
, t
->type
);
193 if (t
->type
== PSI_T_EOF
) {
197 for (i
= 0; i
< t
->text
->len
; ++i
) {
198 switch (t
->text
->val
[i
]) {
227 if (isprint(t
->text
->val
[i
])) {
228 dprintf(fd
, "%c", t
->text
->val
[i
]);
230 dprintf(fd
, "\\x%02hhX", t
->text
->val
[i
]);
237 dprintf(fd
, " at col %u in %s on line %u\n", t
->col
, t
->file
->val
, t
->line
);