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"
35 struct psi_let_func
*psi_let_func_init(token_t type
, const char *name
,
36 struct psi_impl_var
*var
)
38 struct psi_let_func
*func
= calloc(1, sizeof(*func
));
40 func
->name
= strdup(name
);
45 void psi_let_func_free(struct psi_let_func
**func_ptr
)
48 struct psi_let_func
*func
= *func_ptr
;
54 psi_impl_var_free(&func
->var
);
57 psi_plist_free(func
->inner
);
63 void psi_let_func_dump(int fd
, struct psi_let_func
*func
, unsigned level
)
65 dprintf(fd
, "%s(%s\t/* fqn=%s */", func
->name
, func
->var
->name
, func
->var
->fqn
);
68 size_t i
= 0, count
= psi_plist_count(func
->inner
);
69 struct psi_let_exp
*inner
;
73 while (psi_plist_get(func
->inner
, i
++, &inner
)) {
75 psi_let_exp_dump(fd
, inner
, level
, i
== count
);
79 dprintf(fd
, "%s", psi_t_indent(level
));
84 static inline int validate_let_func_type(struct psi_data
*data
,
85 struct psi_let_func
*func
, struct psi_validate_scope
*scope
)
102 data
->error(data
, func
->token
, PSI_WARNING
,
103 "Expected sub-expressions in `arrval` expression");
109 data
->error(data
, func
->var
->token
, PSI_WARNING
,
110 "Unknown `let` cast function '%s' of implementation '%s'",
111 func
->name
, scope
->impl
->func
->name
);
116 static inline bool validate_let_func_inner(struct psi_data
*data
,
117 struct psi_let_func
*func
, struct psi_validate_scope
*scope
)
119 struct psi_let_exp
*exp
= scope
->current_let
;
122 struct psi_decl_var
*let_var
= psi_let_exp_get_decl_var(exp
);
123 struct psi_decl_type
*var_typ
;
124 struct psi_plist
*sub_args
;
126 sub_args
= psi_decl_type_get_args(let_var
->arg
->type
, &var_typ
);
128 if (func
->type
== PSI_T_ARRVAL
&& sub_args
) {
129 /* struct = arrval($array,
130 * member = strval($member) ...)
133 struct psi_let_exp
*inner
;
135 while (psi_plist_get(func
->inner
, i
++, &inner
)) {
136 const char *name
= psi_let_exp_get_decl_var_name(inner
);
137 struct psi_decl_arg
*sub_arg
;
142 sub_arg
= psi_decl_arg_get_by_name(sub_args
, name
);
144 if (!name
|| !sub_arg
) {
145 /* remove expr for portability with different struct members */
146 psi_plist_del(func
->inner
, --i
, NULL
);
147 psi_let_exp_free(&inner
);
149 scope
->current_let
= inner
;
150 if (!psi_let_exp_validate(data
, inner
, scope
)) {
151 scope
->current_let
= exp
;
154 scope
->current_let
= exp
;
157 } else if (func
->type
== PSI_T_ARRVAL
158 && psi_plist_count(func
->inner
) == 1
159 && let_var
->arg
->var
->pointer_level
) {
160 /* array = arrval($array,
161 * strval($array)) // cast foreach entry
163 struct psi_let_exp
*inner
;
164 struct psi_impl_var
*sub_var
;
165 struct psi_impl_var
*sub_ref
;
167 psi_plist_get(func
->inner
, 0, &inner
);
170 sub_var
= psi_let_exp_get_impl_var(exp
);
171 sub_ref
= psi_let_exp_get_impl_var(inner
);
172 if (strcmp(sub_var
->name
, sub_ref
->name
)) {
173 data
->error(data
, sub_var
->token
, E_WARNING
,
174 "Inner `set` statement casts on pointers must"
175 " reference the same variable");
178 scope
->current_let
= inner
;
179 if (!psi_let_exp_validate(data
, inner
, scope
)) {
180 scope
->current_let
= exp
;
183 scope
->current_let
= exp
;
185 data
->error(data
, let_var
->token
, PSI_WARNING
,
186 "Inner let statement's values must refer to a structure or"
187 " array type, got '%s%s' for '%s'", var_typ
->name
,
188 psi_t_indirection(let_var
->arg
->var
->pointer_level
),
193 if (!psi_plist_count(func
->inner
)) {
194 data
->error(data
, func
->token
, PSI_WARNING
,
195 "No valid sub-expressions left");
203 bool psi_let_func_validate(struct psi_data
*data
, struct psi_let_func
*func
,
204 struct psi_validate_scope
*scope
)
206 if (scope
->impl
->func
->args
) {
207 /* FIXME, func->var does not need to be referring to a func arg */
208 psi_impl_get_arg(scope
->impl
, func
->var
);
211 if (!psi_impl_var_validate(data
, func
->var
, scope
)) {
215 if (!validate_let_func_type(data
, func
, scope
)) {
218 if (!validate_let_func_inner(data
, func
, scope
)) {
224 void exec_let_func_arrval_inner(struct psi_let_func
*func
,
225 struct psi_decl_arg
*darg
, struct psi_decl_arg
*inner_decl_arg
,
226 struct psi_call_frame_argument
*frame_arg
,
227 struct psi_let_exp
*inner_let_exp
, void *container
,
228 struct psi_call_frame
*frame
)
230 struct psi_let_func
*inner_func
= psi_let_exp_get_func(inner_let_exp
);
231 struct psi_impl_var
*inner_var
= inner_func
->var
;
233 if (psi_decl_type_get_real(darg
->type
)->type
== PSI_T_UNION
) {
234 /* skip if there's no entry in the outer_zval;
235 * we only want to set supplied data on unions
237 if (!zend_symtable_str_exists(Z_ARRVAL_P(frame_arg
->zval_ptr
),
238 &inner_var
->name
[1], strlen(&inner_var
->name
[1]))) {
242 psi_call_frame_sub_argument(frame
, inner_var
, frame_arg
->zval_ptr
,
243 inner_func
->var
->fqn
);
245 /* example from dm_store/dbm_fetch with arrval($string) conversion:
246 let key = arrval($key,
254 full_func_var_name = $key.0
255 inner_decl_arg = key.dptr
257 psi_let_exp_exec(inner_let_exp
, inner_decl_arg
,
258 ((char *) container
) + inner_decl_arg
->layout
->pos
,
259 inner_decl_arg
->layout
->len
, frame
);
262 static void *exec_let_func_arrval(struct psi_let_exp
*val
,
263 struct psi_let_func
*func
, struct psi_decl_arg
*darg
,
264 struct psi_call_frame
*frame
);
266 void exec_let_func_arrval_seq(struct psi_let_func
*func
,
267 struct psi_decl_arg
*darg
,
268 struct psi_call_frame_argument
*frame_arg
,
269 struct psi_let_exp
*inner_let_exp
, void *container
,
270 struct psi_call_frame
*frame
)
272 zval
*zval_ptr
= NULL
;
273 psi_marshal_let let_fn
;
275 struct psi_decl_var
*dvar
;
277 if (inner_let_exp
->var
) {
278 /* arrval($foo, *foo = intval($foo)); */
279 dvar
= inner_let_exp
->var
;
281 /* arrval($foo, intval($foo)); */
282 dvar
= psi_decl_var_copy(darg
->var
);
283 assert(dvar
->pointer_level
);
284 --dvar
->pointer_level
;
287 dvar
->pointer_level
+= inner_let_exp
->is_reference
;
288 size
= psi_decl_var_get_size(dvar
);
289 dvar
->pointer_level
-= inner_let_exp
->is_reference
;
291 let_fn
= locate_let_func_fn(inner_let_exp
->data
.func
->type
);
293 ZEND_HASH_FOREACH_VAL_IND(Z_ARRVAL_P(frame_arg
->zval_ptr
), zval_ptr
)
296 impl_val val
= {0}, *ptr
, *sub
;
299 ptr
= let_fn(&val
, darg
, 0, NULL
, zval_ptr
, &temp
);
301 psi_call_frame_push_auto(frame
, temp
);
303 } else if (func
->type
== PSI_T_ARRVAL
) {
304 ptr
= exec_let_func_arrval(inner_let_exp
,
305 inner_let_exp
->data
.func
, darg
, frame
);
310 sub
= deref_impl_val(ptr
, dvar
);
312 memcpy(&((char *) container
)[size
* i
++], &sub
, size
);
314 ZEND_HASH_FOREACH_END();
316 if (dvar
!= inner_let_exp
->var
) {
317 psi_decl_var_free(&dvar
);
321 static void *exec_let_func_arrval(struct psi_let_exp
*val
,
322 struct psi_let_func
*func
, struct psi_decl_arg
*darg
,
323 struct psi_call_frame
*frame
)
325 void *container
= NULL
;
326 struct psi_call_frame_argument
*frame_arg
;
327 struct psi_plist
*darg_members
;
329 darg_members
= psi_decl_type_get_args(darg
->type
, NULL
);
330 frame_arg
= psi_call_frame_get_argument(frame
, func
->var
->fqn
);
332 if (frame_arg
->zval_ptr
&& Z_TYPE_P(frame_arg
->zval_ptr
) != IS_ARRAY
) {
333 convert_to_array(frame_arg
->zval_ptr
);
336 if (darg_members
&& func
->inner
) {
339 * str = strval($str),
340 * num = intval($num));
343 struct psi_let_exp
*inner
;
344 struct psi_decl_arg
*darg_member
;
346 val
->var
->pointer_level
+= val
->is_reference
;
347 size
= psi_decl_var_get_size(val
->var
);
348 container
= ecalloc(1, size
);
349 val
->var
->pointer_level
-= val
->is_reference
;
351 if (frame_arg
->zval_ptr
) {
352 while (psi_plist_get(func
->inner
, i
++, &inner
)) {
353 darg_member
= psi_decl_arg_get_by_name(darg_members
,
354 psi_let_exp_get_decl_var_name(inner
));
356 exec_let_func_arrval_inner(func
, darg
, darg_member
, frame_arg
,
357 inner
, container
, frame
);
360 } else if (func
->inner
) {
362 * arrval($foo, foo = intval($foo))
365 struct psi_let_exp
*inner
;
367 if (!frame_arg
->zval_ptr
) {
370 assert(psi_plist_count(func
->inner
) == 1);
372 arcount
= zend_array_count(Z_ARRVAL_P(frame_arg
->zval_ptr
));
373 psi_plist_get(func
->inner
, 0, &inner
);
375 inner
->var
->pointer_level
+= inner
->is_reference
;
376 container
= ecalloc(arcount
+ 1, psi_decl_var_get_size(inner
->var
));
377 inner
->var
->pointer_level
-= inner
->is_reference
;
379 exec_let_func_arrval_seq(func
, darg
, frame_arg
, inner
, container
, frame
);
384 return *psi_call_frame_push_auto(frame
, container
);
387 void *psi_let_func_exec(struct psi_let_exp
*val
, struct psi_let_func
*func
,
388 struct psi_decl_arg
*darg
, struct psi_call_frame
*frame
)
390 struct psi_call_frame_symbol
*frame_sym
;
391 psi_marshal_let let_fn
= locate_let_func_fn(func
->type
);
393 frame_sym
= psi_call_frame_fetch_symbol(frame
, val
->var
);
397 struct psi_call_frame_argument
*iarg
;
399 iarg
= psi_call_frame_get_argument(frame
, func
->var
->fqn
);
403 frame_sym
->ival_ptr
= let_fn(&frame_sym
->temp_val
, darg
,
404 iarg
->spec
? iarg
->spec
->type
->type
: 0, iarg
->ival_ptr
,
405 iarg
->zval_ptr
, &temp
);
407 psi_call_frame_push_auto(frame
, temp
);
409 } else if (func
->type
== PSI_T_ARRVAL
) {
410 frame_sym
->ival_ptr
= exec_let_func_arrval(val
, func
, darg
, frame
);
413 return frame_sym
->ival_ptr
;