autoconf: split into subdirectories; redirect calls in debug builds
[m6w6/ext-psi] / src / engine.c
1 #ifdef HAVE_CONFIG_H
2 # include "config.h"
3 #else
4 # include "php_config.h"
5 #endif
6
7 #include "php.h"
8 #include "php_psi.h"
9
10 #include "zend_exceptions.h"
11
12 #include "parser.h"
13 #include "engine.h"
14 #include "calc.h"
15 #include "marshal.h"
16
17 static inline void psi_do_set(zval *return_value, set_value *set)
18 {
19 decl_arg *set_arg = set->vars->vars[0]->arg;
20
21 zval_dtor(return_value);
22 set->func->handler(return_value, set, set_arg->let);
23 }
24
25 int psi_internal_type(impl_type *type)
26 {
27 switch (type->type) {
28 case PSI_T_BOOL:
29 return _IS_BOOL;
30 case PSI_T_INT:
31 return IS_LONG;
32 case PSI_T_FLOAT:
33 case PSI_T_DOUBLE:
34 return IS_DOUBLE;
35 case PSI_T_STRING:
36 return IS_STRING;
37 case PSI_T_ARRAY:
38 return IS_ARRAY;
39 default:
40 return 0;
41 }
42 }
43
44 zend_internal_arg_info *psi_internal_arginfo(impl *impl)
45 {
46 size_t i;
47 zend_internal_arg_info *aip;
48 zend_internal_function_info *fi;
49
50 aip = calloc(impl->func->args->count + 1 + !!impl->func->args->vararg.name, sizeof(*aip));
51
52 fi = (zend_internal_function_info *) &aip[0];
53 fi->allow_null = 1;
54 fi->required_num_args = psi_num_min_args(impl);
55 fi->return_reference = impl->func->return_reference;
56 fi->type_hint = psi_internal_type(impl->func->return_type);
57
58 if (impl->func->args->vararg.name) {
59 impl_arg *vararg = impl->func->args->vararg.name;
60 zend_internal_arg_info *ai = &aip[impl->func->args->count];
61
62 ai->name = vararg->var->name;
63 ai->allow_null = 1;
64 ai->type_hint = psi_internal_type(vararg->type);
65 if (vararg->var->reference) {
66 ai->pass_by_reference = 1;
67 }
68 ai->is_variadic = 1;
69 }
70
71 for (i = 0; i < impl->func->args->count; ++i) {
72 impl_arg *iarg = impl->func->args->args[i];
73 zend_internal_arg_info *ai = &aip[i+1];
74
75 ai->name = iarg->var->name;
76 ai->type_hint = psi_internal_type(iarg->type);
77 if (iarg->var->reference) {
78 ai->pass_by_reference = 1;
79 }
80 //if (iarg->var->reference || (iarg->def && iarg->def->type == PSI_T_NULL)) {
81 ai->allow_null = 1;
82 //}
83 }
84
85 return aip;
86 }
87
88 size_t psi_num_min_args(impl *impl)
89 {
90 size_t i, n = impl->func->args->count;
91
92 for (i = 0; i < impl->func->args->count; ++i) {
93 if (impl->func->args->args[i]->def) {
94 --n;
95 }
96 }
97 return n;
98 }
99
100
101 static inline ZEND_RESULT_CODE psi_parse_args(zend_execute_data *execute_data, impl *impl)
102 {
103 size_t i;
104 impl_arg *iarg;
105 zend_error_handling zeh;
106
107 zend_replace_error_handling(EH_THROW, zend_exception_get_default(), &zeh);
108
109 if (!impl->func->args->count) {
110 ZEND_RESULT_CODE rv;
111
112 rv = zend_parse_parameters_none();
113 zend_restore_error_handling(&zeh);
114 return rv;
115 }
116
117 ZEND_PARSE_PARAMETERS_START(psi_num_min_args(impl), impl->func->args->vararg.name ? -1 : impl->func->args->count)
118 nextarg:
119 if (impl->func->args->vararg.name && _i >= impl->func->args->count) {
120 impl_arg *varg = impl->func->args->vararg.name;
121 iarg = init_impl_arg(
122 init_impl_type(varg->type->type, varg->type->name),
123 init_impl_var(varg->var->name, varg->var->reference),
124 NULL);
125
126 Z_PARAM_OPTIONAL;
127 if (_i == impl->func->args->count) {
128 impl->func->args->vararg.args = init_impl_args(iarg);
129 } else {
130 add_impl_arg(impl->func->args->vararg.args, iarg);
131 }
132 } else {
133 iarg = impl->func->args->args[_i];
134 if (iarg->def) {
135 Z_PARAM_OPTIONAL;
136 }
137 }
138 if (PSI_T_BOOL == iarg->type->type) {
139 Z_PARAM_BOOL(iarg->val.zend.bval);
140 } else if (PSI_T_INT == iarg->type->type) {
141 Z_PARAM_LONG(iarg->val.zend.lval);
142 } else if (PSI_T_FLOAT == iarg->type->type || PSI_T_DOUBLE == iarg->type->type) {
143 Z_PARAM_DOUBLE(iarg->val.dval);
144 } else if (PSI_T_STRING == iarg->type->type) {
145 Z_PARAM_STR_EX(iarg->val.zend.str, 1, iarg->var->reference);
146 if (iarg->val.zend.str) {
147 zend_string_addref(iarg->val.zend.str);
148 }
149 } else if (PSI_T_ARRAY == iarg->type->type) {
150 Z_PARAM_PROLOGUE(0);
151 } else if (PSI_T_OBJECT == iarg->type->type) {
152 Z_PARAM_PROLOGUE(0);
153 } else if (PSI_T_MIXED == iarg->type->type) {
154 Z_PARAM_PROLOGUE(0);
155 } else if (PSI_T_CALLABLE == iarg->type->type) {
156 zend_fcall_info fci;
157 zend_fcall_info_cache fcc;
158
159 Z_PARAM_FUNC_EX(fci, fcc, 1, 0);
160
161 if (fci.size) {
162 iarg->val.zend.cb = ecalloc(1, sizeof(zend_fcall));
163 iarg->val.zend.cb->fci = fci;
164 iarg->val.zend.cb->fcc = fcc;
165 }
166 } else {
167 error_code = ZPP_ERROR_FAILURE;
168 break;
169 }
170 iarg->_zv = _arg;
171 ZVAL_DEREF(iarg->_zv);
172 if (_i < _num_args) {
173 goto nextarg;
174 }
175 ZEND_PARSE_PARAMETERS_END_EX(
176 zend_restore_error_handling(&zeh);
177 return FAILURE
178 );
179
180 /* set up defaults */
181 for (i = 0; i < impl->func->args->count; ++i) {
182 if (i >= EX_NUM_ARGS() && iarg->def) {
183 iarg = impl->func->args->args[i];
184
185 switch (iarg->type->type) {
186 case PSI_T_BOOL:
187 iarg->val.zend.bval = iarg->def->type == PSI_T_TRUE ? 1 : 0;
188 break;
189 case PSI_T_INT:
190 iarg->val.zend.lval = zend_atol(iarg->def->text, strlen(iarg->def->text));
191 break;
192 case PSI_T_FLOAT:
193 case PSI_T_DOUBLE:
194 iarg->val.dval = zend_strtod(iarg->def->text, NULL);
195 break;
196 case PSI_T_STRING:
197 /* FIXME */
198 iarg->val.zend.str = zend_string_init(&iarg->def->text[1], strlen(iarg->def->text) - 2, 0);
199 break;
200 }
201 }
202 }
203
204 zend_restore_error_handling(&zeh);
205 return SUCCESS;
206 }
207
208 static inline void *psi_do_calloc(let_calloc *alloc)
209 {
210 zend_long n = psi_long_num_exp(alloc->nmemb, NULL), s = psi_long_num_exp(alloc->size, NULL);
211 void *mem = safe_emalloc(n, s, sizeof(void *));
212 memset(mem, 0, n * s + sizeof(void *));
213 #if 0
214 fprintf(stderr, "calloc: %p\n", mem);
215 #endif
216 return mem;
217 }
218
219 static inline impl_val *psi_let_func(let_func *func, decl_arg *darg);
220
221 static inline void *psi_let_val(let_val *val, decl_arg *darg)
222 {
223 ZEND_ASSERT(darg);
224
225 switch (val ? val->kind : PSI_LET_NULL) {
226 case PSI_LET_TMP:
227 memcpy(darg->ptr, deref_impl_val(val->data.var->arg->let, val->data.var), sizeof(impl_val));
228 break;
229 case PSI_LET_NULL:
230 if (darg->var->array_size) {
231 darg->val.ptr = ecalloc(darg->var->array_size, sizeof(impl_val));
232 darg->mem = darg->val.ptr;
233 } else {
234 memset(&darg->val, 0, sizeof(impl_val));
235 }
236 break;
237 case PSI_LET_CALLOC:
238 darg->val.ptr = psi_do_calloc(val->data.alloc);
239 darg->mem = darg->val.ptr;
240 break;
241 case PSI_LET_NUMEXP:
242 darg->val.zend.lval = psi_long_num_exp(val->data.num, NULL);
243 break;
244 case PSI_LET_CALLBACK:
245 darg->val.ptr = val->data.callback->decl->call.sym;
246 break;
247 case PSI_LET_FUNC:
248 if (!psi_let_func(val->data.func, darg)) {
249 return NULL;
250 }
251 break;
252 }
253
254 if (val && val->flags.one.is_reference) {
255 return darg->let = &darg->ptr;
256 } else {
257 return darg->let = darg->ptr;
258 }
259 }
260
261 static void marshal_func(void *cb_ctx, impl_val **ptr, decl_arg *spec, token_t cast, zval *zv, void **tmp) {
262 let_vals *vals = cb_ctx;
263 size_t i;
264
265 for (i = 0; i < vals->count; ++i) {
266 impl_var *var = locate_let_val_impl_var(vals->vals[i]);
267
268 if (!strcmp(&var->name[1], spec->var->name)) {
269 *ptr = psi_let_val(vals->vals[i], spec);
270 *tmp = spec->mem;
271 break;
272 }
273 }
274 }
275
276 static inline impl_val *psi_let_func_ex(let_func *func, void *dptr, decl_type *dtype, token_t itype, impl_val *ival, zval *zvalue, void **to_free) {
277 switch (func->type) {
278 case PSI_T_BOOLVAL:
279 return psi_let_boolval(dptr, dtype, itype, ival, zvalue, to_free);
280 case PSI_T_INTVAL:
281 return psi_let_intval(dptr, dtype, itype, ival, zvalue, to_free);
282 case PSI_T_FLOATVAL:
283 return psi_let_floatval(dptr, dtype, itype, ival, zvalue, to_free);
284 case PSI_T_STRVAL:
285 return psi_let_strval(dptr, dtype, itype, ival, zvalue, to_free);
286 case PSI_T_STRLEN:
287 return psi_let_strlen(dptr, dtype, itype, ival, zvalue, to_free);
288 case PSI_T_PATHVAL:
289 return psi_let_pathval(dptr, dtype, itype, ival, zvalue, to_free);
290 case PSI_T_OBJVAL:
291 return psi_let_objval(dptr, dtype, itype, ival, zvalue, to_free);
292 case PSI_T_ZVAL:
293 return psi_let_zval(dptr, dtype, itype, ival, zvalue, to_free);
294 case PSI_T_VOID:
295 return psi_let_void(dptr, dtype, itype, ival, zvalue, to_free);
296 break;
297 case PSI_T_ARRVAL:
298 if (func->inner) {
299 decl_type *real = real_decl_type(dtype);
300
301 if (itype != PSI_T_ARRAY) {
302 SEPARATE_ARG_IF_REF(zvalue);
303 convert_to_array(zvalue);
304 }
305
306 return *to_free = psi_array_to_struct_ex(real->real.strct, Z_ARRVAL_P(zvalue), marshal_func, func->inner);
307 } else {
308 return psi_let_arrval(dptr, dtype, itype, ival, zvalue, to_free);
309 }
310 break;
311 default:
312 assert(0);
313 }
314 return NULL;
315 }
316
317 static inline impl_val *psi_let_func(let_func *func, decl_arg *darg) {
318 impl_arg *iarg = func->var->arg;
319
320 if (func->outer && !iarg) {
321 impl_arg *outer_arg = locate_let_val_impl_var(func->outer)->arg;
322 iarg = init_impl_arg(
323 init_impl_type(PSI_T_MIXED, "mixed"),
324 copy_impl_var(func->var), NULL);
325
326
327 if (!(iarg->_zv = zend_hash_str_find(Z_ARRVAL_P(outer_arg->_zv), &iarg->var->name[1], strlen(iarg->var->name)-1))) {
328 iarg->_zv = zend_hash_str_add_empty_element(Z_ARRVAL_P(outer_arg->_zv), &iarg->var->name[1], strlen(iarg->var->name)-1);
329 }
330 }
331 return darg->ptr = psi_let_func_ex(func, darg->ptr, darg->type, iarg->type->type, &iarg->val, iarg->_zv, &darg->mem);
332 }
333
334 static inline void *psi_do_let(let_stmt *let)
335 {
336 return psi_let_val(let->val, let->var->arg);
337 }
338
339 static inline void psi_do_return(zval *return_value, return_stmt *ret)
340 {
341 ret->set->func->handler(return_value, ret->set, ret->set->vars->vars[0]->arg->ptr);
342 }
343
344 static inline void psi_do_free(free_stmt *fre)
345 {
346 size_t i, j;
347
348 for (i = 0; i < fre->calls->count; ++i) {
349 free_call *f = fre->calls->list[i];
350
351 for (j = 0; j < f->vars->count; ++j) {
352 decl_var *dvar = f->vars->vars[j];
353 decl_arg *darg = dvar->arg;
354 impl_val *fval = darg->let;
355
356 f->decl->call.args[j] = deref_impl_val(fval, dvar);
357 }
358
359 /* FIXME: check in validate_* that free functions return scalar */
360 psi_context_call(&PSI_G(context), &f->decl->call, NULL);
361 }
362 }
363
364 static inline void psi_clean_array_struct(let_val *val, decl_arg *darg) {
365 if (val->kind == PSI_LET_FUNC
366 && val->data.func->type == PSI_T_ARRVAL) {
367 decl_type *type = real_decl_type(darg->type);
368 decl_args *args = NULL;
369
370 if (type->type == PSI_T_STRUCT) {
371 void **ptr = (void **) ((char *) darg->mem + type->real.strct->size);
372
373 while (*ptr) {
374 efree(*ptr++);
375 }
376 args = type->real.strct->args;
377 } else if (type->type == PSI_T_UNION) {
378 void **ptr = (void **) ((char *) darg->mem + type->real.unn->size);
379
380 if (*ptr) {
381 efree(*ptr);
382 }
383 args = type->real.unn->args;
384 }
385 if (args && val->data.func->inner) {
386 size_t i;
387
388 for (i = 0; i < val->data.func->inner->count; ++i) {
389 let_val *inner = val->data.func->inner->vals[i];
390 decl_var *refvar = locate_let_val_inner_ref(inner)->var;
391 decl_arg *subarg = locate_decl_arg(args, refvar->name);
392
393 if (subarg) {
394 psi_clean_array_struct(val->data.func->inner->vals[i], subarg);
395 if (subarg && subarg->mem) {
396 efree(subarg->mem);
397 subarg->mem = NULL;
398 }
399 }
400 }
401 }
402 }
403 }
404
405 static inline void psi_do_clean(impl *impl)
406 {
407 size_t i;
408
409 if (impl->decl->func->ptr != &impl->decl->func->val) {
410 efree(impl->decl->func->ptr);
411 impl->decl->func->ptr = &impl->decl->func->val;
412 }
413
414 for (i = 0; i < impl->func->args->count; ++i ) {
415 impl_arg *iarg = impl->func->args->args[i];
416
417 switch (iarg->type->type) {
418 case PSI_T_STRING:
419 if (iarg->val.zend.str) {
420 zend_string_release(iarg->val.zend.str);
421 }
422 break;
423 case PSI_T_CALLABLE:
424 if (iarg->val.zend.cb) {
425 if (iarg->val.zend.cb->fci.size) {
426 zend_fcall_info_args_clear(&iarg->val.zend.cb->fci, 1);
427 }
428 efree(iarg->val.zend.cb);
429 }
430 break;
431 }
432 }
433
434 for (i = 0; i < impl->stmts->let.count; ++i) {
435 let_stmt *let = impl->stmts->let.list[i];
436 decl_arg *darg = let->var->arg;
437
438 if (darg->mem) {
439 psi_clean_array_struct(let->val, darg);
440 efree(darg->mem);
441 darg->mem = NULL;
442 }
443 darg->ptr = &darg->val;
444 darg->let = darg->ptr;
445 }
446
447 if (impl->func->args->vararg.args) {
448 free_impl_args(impl->func->args->vararg.args);
449 impl->func->args->vararg.args = NULL;
450 }
451 if (impl->func->args->vararg.types) {
452 efree(impl->func->args->vararg.types);
453 impl->func->args->vararg.types = NULL;
454 }
455 if (impl->func->args->vararg.values) {
456 efree(impl->func->args->vararg.values);
457 impl->func->args->vararg.values = NULL;
458 }
459 if (impl->func->args->vararg.free_list) {
460 void **list = impl->func->args->vararg.free_list;
461
462 while (*list) {
463 efree(*list++);
464 }
465
466 efree(impl->func->args->vararg.free_list);
467 impl->func->args->vararg.free_list = NULL;
468 }
469 }
470
471
472 static inline void psi_do_args(impl *impl) {
473 size_t i;
474
475 for (i = 0; i < impl->decl->args->count; ++i) {
476 impl->decl->call.args[i] = impl->decl->args->args[i]->let;
477 }
478
479 if (!impl->decl->func->var->pointer_level) {
480 decl_type *real = real_decl_type(impl->decl->func->type);
481
482 switch (real->type) {
483 case PSI_T_STRUCT:
484 impl->decl->func->ptr = psi_array_to_struct(real->real.strct, NULL);
485 break;
486 case PSI_T_UNION:
487 impl->decl->func->ptr = psi_array_to_union(real->real.unn, NULL);
488 break;
489 }
490 }
491 }
492
493 static inline impl_vararg *psi_do_varargs(impl *impl) {
494 size_t i, j;
495 impl_vararg *va = &impl->func->args->vararg;
496 size_t vacount = va->args->count;
497
498
499 if (!vacount) {
500 return NULL;
501 }
502
503 va->types = ecalloc(vacount, sizeof(*va->types));
504 va->values = ecalloc(vacount, sizeof(*va->values));
505
506 for (i = 0, j = 0; i < vacount; ++i) {
507 impl_arg *vaarg = va->args->args[i];
508 void *to_free = NULL;
509 token_t vatype = va->name->type->type;
510 psi_marshal_let let_fn;
511
512 if (vatype == PSI_T_MIXED) {
513 switch (Z_TYPE_P(vaarg->_zv)) {
514 case IS_TRUE:
515 case IS_FALSE: vatype = PSI_T_BOOL; break;
516 case IS_LONG: vatype = PSI_T_INT; break;
517 case IS_DOUBLE: vatype = PSI_T_FLOAT; break;
518 default: vatype = PSI_T_STRING; break;
519 }
520 }
521
522
523 switch (vatype) {
524 case PSI_T_BOOL: let_fn = psi_let_boolval; break;
525 case PSI_T_INT: let_fn = psi_let_intval; break;
526 case PSI_T_FLOAT:
527 case PSI_T_DOUBLE: let_fn = psi_let_floatval; break;
528 case PSI_T_STRING: let_fn = psi_let_strval; break;
529 EMPTY_SWITCH_DEFAULT_CASE();
530 }
531
532 va->types[i] = vatype;
533
534 /* FIXME: varargs with struct-by-value :) */
535 //if (!psi_let_val(let_fn, vaarg, &va->values[i], NULL, &to_free)) {
536 if (!let_fn(&va->values[i], NULL, vaarg->type->type, &vaarg->val, vaarg->_zv, &to_free)) {
537 return NULL;
538 }
539
540 if (to_free) {
541 if (!va->free_list) {
542 va->free_list = ecalloc(vacount - i + 1, sizeof(*va->free_list));
543 }
544 va->free_list[j++] = to_free;
545 }
546 }
547
548 return va;
549 }
550
551 ZEND_RESULT_CODE psi_call(zend_execute_data *execute_data, zval *return_value, impl *impl)
552 {
553 size_t i;
554 impl_vararg *va = NULL;
555
556 memset(impl->decl->func->ptr, 0, sizeof(impl_val));
557
558 if (SUCCESS != psi_parse_args(execute_data, impl)) {
559 return FAILURE;
560 }
561
562 for (i = 0; i < impl->stmts->let.count; ++i) {
563 let_stmt *let = impl->stmts->let.list[i];
564
565 if (!psi_do_let(let)) {
566 psi_do_return(return_value, impl->stmts->ret.list[0]);
567 psi_do_clean(impl);
568 return FAILURE;
569 }
570 }
571
572 if (impl->decl->args) {
573 psi_do_args(impl);
574
575 if (impl->func->args->vararg.args) {
576 va = psi_do_varargs(impl);
577 }
578 }
579
580 psi_context_call(&PSI_G(context), &impl->decl->call, va);
581 psi_do_return(return_value, impl->stmts->ret.list[0]);
582
583 for (i = 0; i < impl->stmts->set.count; ++i) {
584 set_stmt *set = impl->stmts->set.list[i];
585
586 if (set->arg->_zv) {
587 psi_do_set(set->arg->_zv, set->val);
588 }
589 }
590
591 for (i = 0; i < impl->stmts->fre.count; ++i) {
592 free_stmt *fre = impl->stmts->fre.list[i];
593
594 psi_do_free(fre);
595 }
596 psi_do_clean(impl);
597
598 return SUCCESS;
599 }
600
601 ZEND_RESULT_CODE psi_callback(let_callback *cb, void *retval, unsigned argc, void **argv)
602 {
603 size_t i;
604 decl *decl_cb = cb->decl;
605 impl_arg *iarg = cb->func->var->arg;
606 zval return_value, *zargv = calloc(argc, sizeof(*zargv));
607 void *result, *to_free = NULL;
608
609 ZEND_ASSERT(argc == cb->decl->args->count);
610
611 /* prepare args for the userland call */
612 for (i = 0; i < argc; ++i) {
613 cb->decl->args->args[i]->let = argv[i];
614 }
615 for (i = 0; i < cb->args->count; ++i) {
616 psi_do_set(&zargv[i], cb->args->vals[i]);
617 }
618 zend_fcall_info_argp(&iarg->val.zend.cb->fci, cb->args->count, zargv);
619
620 /* callback into userland */
621 ZVAL_UNDEF(&return_value);
622 iarg->_zv = &return_value;
623 zend_fcall_info_call(&iarg->val.zend.cb->fci, &iarg->val.zend.cb->fcc, iarg->_zv, NULL);
624
625 /* marshal return value of the userland call
626 switch (iarg->type->type) {
627 case PSI_T_BOOL: zend_parse_arg_bool(iarg->_zv, &iarg->val.zend.bval, NULL, 0); break;
628 case PSI_T_LONG: zend_parse_arg_long(iarg->_zv, &iarg->val.zend.lval, NULL, 0, 1); break;
629 case PSI_T_FLOAT:
630 case PSI_T_DOUBLE: zend_parse_arg_double(iarg->_zv, &iarg->val.dval, NULL, 0); break;
631 case PSI_T_STRING: zend_parse_arg_str(iarg->_zv, &iarg->val.zend.str, 0); break;
632 }
633 */
634 result = psi_let_func_ex(cb->func, retval, decl_cb->func->type, 0, &iarg->val, iarg->_zv, &to_free);
635 // result = cb->func->handler(retval, decl_cb->func->type, iarg, &to_free);
636
637 if (result != retval) {
638 *(void **)retval = result;
639 }
640
641 zend_fcall_info_args_clear(&iarg->val.zend.cb->fci, 0);
642 for (i = 0; i < cb->args->count; ++i) {
643 zval_ptr_dtor(&zargv[i]);
644 }
645 free(zargv);
646
647 return SUCCESS;
648
649 }