projects / m6w6 / ext-psi / blob
? search:
summary | shortlog | log | commit | commitdiff | tree | github
history | raw | HEAD
flush
[m6w6/ext-psi] / src / module.c
1
2 #ifdef HAVE_CONFIG_H
3 #include "config.h"
4 #endif
5
6 #include "php.h"
7 #include "php_ini.h"
8 #include "ext/standard/info.h"
9 #include "zend_exceptions.h"
10 #include "zend_constants.h"
11 #include "zend_operators.h"
12
13 #include "php_psi.h"
14
15 #if HAVE_LIBJIT
16 # include "libjit.h"
17 # ifndef HAVE_LIBFFI
18 # define PSI_ENGINE "jit"
19 # endif
20 #endif
21 #if HAVE_LIBFFI
22 # include "libffi.h"
23 # define PSI_ENGINE "ffi"
24 #endif
25
26 ZEND_DECLARE_MODULE_GLOBALS(psi);
27
28 PHP_INI_BEGIN()
29 STD_PHP_INI_ENTRY("psi.engine", PSI_ENGINE, PHP_INI_SYSTEM, OnUpdateString, engine, zend_psi_globals, psi_globals)
30 STD_PHP_INI_ENTRY("psi.directory", "psi.d", PHP_INI_SYSTEM, OnUpdateString, directory, zend_psi_globals, psi_globals)
31 PHP_INI_END();
32
33 static zend_object_handlers psi_object_handlers;
34 static zend_class_entry *psi_class_entry;
35
36 void psi_error_wrapper(PSI_Token *t, int type, const char *msg, ...)
37 {
38 va_list argv;
39
40 va_start(argv, msg);
41 psi_verror(type, t?t->file:"Unknown", t?*t->line:0, msg, argv);
42 va_end(argv);
43 }
44 void psi_error(int type, const char *fn, unsigned ln, const char *msg, ...)
45 {
46 va_list argv;
47
48 va_start(argv, msg);
49 psi_verror(type, fn, ln, msg, argv);
50 va_end(argv);
51 }
52 void psi_verror(int type, const char *fn, unsigned ln, const char *msg, va_list argv)
53 {
54 zend_error_cb(type, fn, ln, msg, argv);
55 }
56
57 size_t psi_t_alignment(token_t t)
58 {
59 #define PSI_ALIGNOF(T) case PSI_T_## T: return ALIGNOF_## T ##_T;
60 switch (t) {
61 PSI_ALIGNOF(INT8);
62 PSI_ALIGNOF(UINT8);
63 PSI_ALIGNOF(INT16);
64 PSI_ALIGNOF(UINT16);
65 PSI_ALIGNOF(INT32);
66 PSI_ALIGNOF(UINT32);
67 PSI_ALIGNOF(INT64);
68 PSI_ALIGNOF(UINT64);
69 case PSI_T_FLOAT:
70 return ALIGNOF_FLOAT;
71 case PSI_T_DOUBLE:
72 return ALIGNOF_DOUBLE;
73 case PSI_T_POINTER:
74 return ALIGNOF_VOID_P;
75 EMPTY_SWITCH_DEFAULT_CASE();
76 }
77 return 0;
78 }
79
80 size_t psi_t_size(token_t t)
81 {
82 #define PSI_SIZEOF(T) case PSI_T_## T : return SIZEOF_## T ##_T;
83 switch (t) {
84 PSI_SIZEOF(INT8);
85 PSI_SIZEOF(UINT8);
86 PSI_SIZEOF(INT16);
87 PSI_SIZEOF(UINT16);
88 PSI_SIZEOF(INT32);
89 PSI_SIZEOF(UINT32);
90 PSI_SIZEOF(INT64);
91 PSI_SIZEOF(UINT64);
92 case PSI_T_FLOAT:
93 return SIZEOF_FLOAT;
94 case PSI_T_DOUBLE:
95 return SIZEOF_DOUBLE;
96 case PSI_T_POINTER:
97 return SIZEOF_VOID_P;
98 EMPTY_SWITCH_DEFAULT_CASE();
99 }
100 return 0;
101 }
102
103 size_t psi_t_align(token_t t, size_t s)
104 {
105 size_t a = psi_t_alignment(t);
106 return ((s - 1) | (a - 1)) + 1;
107 }
108
109 int psi_internal_type(impl_type *type)
110 {
111 switch (type->type) {
112 case PSI_T_BOOL:
113 return _IS_BOOL;
114 case PSI_T_INT:
115 return IS_LONG;
116 case PSI_T_FLOAT:
117 case PSI_T_DOUBLE:
118 return IS_DOUBLE;
119 case PSI_T_STRING:
120 return IS_STRING;
121 case PSI_T_ARRAY:
122 return IS_ARRAY;
123 default:
124 return 0;
125 }
126 }
127
128 zend_internal_arg_info *psi_internal_arginfo(impl *impl)
129 {
130 size_t i;
131 zend_internal_arg_info *aip;
132 zend_internal_function_info *fi;
133
134 aip = calloc(impl->func->args->count + 1 + !!impl->func->args->vararg.name, sizeof(*aip));
135
136 fi = (zend_internal_function_info *) &aip[0];
137 fi->allow_null = 1;
138 fi->required_num_args = psi_num_min_args(impl);
139 fi->return_reference = impl->func->return_reference;
140 fi->type_hint = psi_internal_type(impl->func->return_type);
141
142 if (impl->func->args->vararg.name) {
143 impl_arg *vararg = impl->func->args->vararg.name;
144 zend_internal_arg_info *ai = &aip[impl->func->args->count];
145
146 ai->name = vararg->var->name;
147 ai->type_hint = psi_internal_type(vararg->type);
148 if (vararg->var->reference) {
149 ai->pass_by_reference = 1;
150 }
151 ai->is_variadic = 1;
152 }
153
154 for (i = 0; i < impl->func->args->count; ++i) {
155 impl_arg *iarg = impl->func->args->args[i];
156 zend_internal_arg_info *ai = &aip[i+1];
157
158 ai->name = iarg->var->name;
159 ai->type_hint = psi_internal_type(iarg->type);
160 if (iarg->var->reference) {
161 ai->pass_by_reference = 1;
162 }
163 if (iarg->var->reference || (iarg->def && iarg->def->type == PSI_T_NULL)) {
164 ai->allow_null = 1;
165 }
166 }
167
168 return aip;
169 }
170
171 size_t psi_num_min_args(impl *impl)
172 {
173 size_t i, n = impl->func->args->count;
174
175 for (i = 0; i < impl->func->args->count; ++i) {
176 if (impl->func->args->args[i]->def) {
177 --n;
178 }
179 }
180 return n;
181 }
182
183 void psi_to_void(zval *return_value, set_value *set, impl_val *ret_val)
184 {
185 RETVAL_NULL();
186 }
187
188 void psi_to_bool(zval *return_value, set_value *set, impl_val *ret_val)
189 {
190 psi_to_int(return_value, set, ret_val);
191 convert_to_boolean(return_value);
192 }
193
194 void psi_to_int(zval *return_value, set_value *set, impl_val *ret_val)
195 {
196 decl_var *var = set->vars->vars[0];
197 token_t t = real_decl_type(var->arg->type)->type;
198 impl_val *v = deref_impl_val(ret_val, var);
199
200 switch (t) {
201 case PSI_T_FLOAT:
202 RETVAL_DOUBLE((double) v->fval);
203 convert_to_long(return_value);
204 break;
205 case PSI_T_DOUBLE:
206 RETVAL_DOUBLE(v->dval);
207 convert_to_long(return_value);
208 break;
209 case PSI_T_INT8:
210 RETVAL_LONG(v->i8);
211 break;
212 case PSI_T_UINT8:
213 RETVAL_LONG(v->u8);
214 break;
215 case PSI_T_INT16:
216 RETVAL_LONG(v->i16);
217 break;
218 case PSI_T_UINT16:
219 RETVAL_LONG(v->u16);
220 break;
221 case PSI_T_INT32:
222 RETVAL_LONG(v->i32);
223 break;
224 case PSI_T_UINT32:
225 #if UINT32_MAX >= ZEND_LONG_MAX
226 if (v->u32 > ZEND_LONG_MAX) {
227 char d[12] = {0};
228
229 RETVAL_STRING(zend_print_ulong_to_buf(&d[10], v->u32));
230 } else {
231 #endif
232 RETVAL_LONG(v->u32);
233 #if UINT32_MAX >= ZEND_LONG_MAX
234 }
235 #endif
236 break;
237 case PSI_T_INT64:
238 RETVAL_LONG(v->i64);
239 break;
240 case PSI_T_UINT64:
241 if (v->u64 > ZEND_LONG_MAX) {
242 char d[24] = {0};
243
244 RETVAL_STRING(zend_print_ulong_to_buf(&d[22], v->u64));
245 } else {
246 RETVAL_LONG(v->u64);
247 }
248 break;
249 EMPTY_SWITCH_DEFAULT_CASE();
250 }
251 }
252
253 void psi_to_double(zval *return_value, set_value *set, impl_val *ret_val)
254 {
255 decl_var *var = set->vars->vars[0];
256 token_t t = real_decl_type(var->arg->type)->type;
257 impl_val *v = deref_impl_val(ret_val, var);
258
259 switch (t) {
260 case PSI_T_FLOAT:
261 RETVAL_DOUBLE((double) v->fval);
262 break;
263 case PSI_T_DOUBLE:
264 RETVAL_DOUBLE(v->dval);
265 break;
266 case PSI_T_INT8:
267 RETVAL_DOUBLE((double) v->i8);
268 break;
269 case PSI_T_UINT8:
270 RETVAL_DOUBLE((double) v->u8);
271 break;
272 case PSI_T_INT16:
273 RETVAL_DOUBLE((double) v->i16);
274 break;
275 case PSI_T_UINT16:
276 RETVAL_DOUBLE((double) v->u16);
277 break;
278 case PSI_T_INT32:
279 RETVAL_DOUBLE((double) v->i32);
280 break;
281 case PSI_T_UINT32:
282 RETVAL_DOUBLE((double) v->u32);
283 break;
284 case PSI_T_INT64:
285 RETVAL_DOUBLE((double) v->i64);
286 break;
287 case PSI_T_UINT64:
288 RETVAL_DOUBLE((double) v->u64);
289 break;
290 EMPTY_SWITCH_DEFAULT_CASE();
291 }
292 }
293
294 void psi_to_string(zval *return_value, set_value *set, impl_val *ret_val)
295 {
296 decl_var *var = set->vars->vars[0];
297 token_t t = real_decl_type(var->arg->type)->type;
298
299 switch (t) {
300 case PSI_T_FLOAT:
301 RETVAL_DOUBLE((double) deref_impl_val(ret_val, var)->fval);
302 break;
303 case PSI_T_DOUBLE:
304 RETVAL_DOUBLE(deref_impl_val(ret_val, var)->dval);
305 break;
306 default:
307 case PSI_T_VOID:
308 case PSI_T_INT8:
309 case PSI_T_UINT8:
310 if (!var->arg->var->pointer_level) {
311 RETVAL_STRINGL(&ret_val->cval, 1);
312 } else {
313 ret_val = deref_impl_val(ret_val, var);
314 if (ret_val && ret_val->ptr) {
315 if (set->num) {
316 RETVAL_STRINGL(ret_val->ptr, psi_long_num_exp(set->num, set->outer.val));
317 } else {
318 RETVAL_STRING(ret_val->ptr);
319 }
320 } else {
321 RETVAL_EMPTY_STRING();
322 }
323 }
324 return;
325 psi_to_int(return_value, set, ret_val);
326 break;
327 }
328 convert_to_string(return_value);
329 }
330
331
332 static impl_val *iterate(impl_val *val, size_t size, unsigned i, impl_val *tmp)
333 {
334 memset(tmp, 0, sizeof(*tmp));
335 memcpy(tmp, ((void*) val->ptr) + size * i, size);
336 return tmp;
337 }
338
339 void psi_from_zval(impl_val *mem, decl_arg *spec, zval *zv, void **tmp)
340 {
341 decl_type *type = real_decl_type(spec->type);
342
343 switch (type->type) {
344 case PSI_T_FLOAT:
345 mem->fval = (float) zval_get_double(zv);
346 break;
347 case PSI_T_DOUBLE:
348 mem->dval = zval_get_double(zv);
349 break;
350 case PSI_T_INT8:
351 case PSI_T_UINT8:
352 if (spec->var->pointer_level) {
353 zend_string *zs = zval_get_string(zv);
354 *tmp = mem->ptr = estrndup(zs->val, zs->len);
355 zend_string_release(zs);
356 break;
357 }
358 /* no break */
359 default:
360 mem->zend.lval = zval_get_long(zv);
361 break;
362 }
363 }
364
365 void *psi_array_to_struct(decl_struct *s, HashTable *arr)
366 {
367 size_t i, j = 0;
368 char *mem = ecalloc(1, s->size + s->args->count * sizeof(void *));
369
370 if (arr) for (i = 0; i < s->args->count; ++i) {
371 decl_arg *darg = s->args->args[i];
372 zval *entry = zend_hash_str_find_ind(arr, darg->var->name, strlen(darg->var->name));
373
374 if (entry) {
375 impl_val val;
376 void *tmp = NULL;
377
378 memset(&tmp, 0, sizeof(tmp));
379 psi_from_zval(&val, darg, entry, &tmp);
380 memcpy(mem + darg->layout->pos, &val, darg->layout->len);
381 if (tmp) {
382 ((void **)(mem + s->size))[j++] = tmp;
383 }
384 }
385 }
386 return mem;
387 }
388
389 void psi_to_recursive(zval *return_value, set_value *set, impl_val *r_val)
390 {
391 set->outer.set->func->handler(return_value, set->outer.set, r_val->ptr);
392 }
393
394 void psi_to_array(zval *return_value, set_value *set, impl_val *r_val)
395 {
396 size_t i;
397 decl_var *var = set->vars->vars[0];
398 token_t t = real_decl_type(var->arg->type)->type;
399 impl_val tmp, *ret_val = deref_impl_val(r_val, var);
400
401 if ((intptr_t) ret_val->ptr <= (intptr_t) 0) {
402 RETURN_NULL();
403 }
404
405 array_init(return_value);
406
407 if (t == PSI_T_STRUCT) {
408 // decl_struct *s = real_decl_type(var->arg->type)->strct;
409
410 if (set->count) {
411 /* explicit member casts */
412 for (i = 0; i < set->count; ++i) {
413 set_value *sub_set = set->inner[i];
414 decl_var *sub_var = sub_set->vars->vars[0];
415
416 sub_set->outer.val = r_val;
417
418 if (sub_var->arg) {
419 impl_val *tmp = NULL, *val;
420 zval ztmp;
421
422 val = deref_impl_val(struct_member_ref(sub_var->arg, ret_val, &tmp), sub_var);
423 sub_set->func->handler(&ztmp, sub_set, val);
424 add_assoc_zval(return_value, sub_var->name, &ztmp);
425
426 if (tmp) {
427 free(tmp);
428 }
429 }
430 }
431 }
432 return;
433 }
434
435 if (var->arg->var->array_size) {
436 /* to_array(foo[NUMBER]) */
437 for (i = 0; i < var->arg->var->array_size; ++i) {
438 size_t size = psi_t_size(var->arg->var->pointer_level > 1 ? PSI_T_POINTER : t);
439 impl_val *ptr = iterate(ret_val, size, i, &tmp);
440 zval ele;
441
442 switch (t) {
443 case PSI_T_FLOAT:
444 ZVAL_DOUBLE(&ele, (double) ptr->fval);
445 break;
446 case PSI_T_DOUBLE:
447 ZVAL_DOUBLE(&ele, ptr->dval);
448 break;
449 default:
450 ZVAL_LONG(&ele, ptr->lval);
451 break;
452 }
453
454 add_next_index_zval(return_value, &ele);
455 }
456 return;
457 } else if (set->num) {
458 /* to_array(arr_var, num_expr, to_int(*arr_var))
459 */
460 zval ele;
461 zend_long i, n = psi_long_num_exp(set->num, set->outer.val);
462
463 for (i = 0; i < n; ++i) {
464 size_t size = psi_t_size(var->arg->var->pointer_level ? PSI_T_POINTER : t);
465 impl_val *ptr = iterate(ret_val, size, i, &tmp);
466
467 set->inner[0]->func->handler(&ele, set->inner[0], ptr);
468 add_next_index_zval(return_value, &ele);
469 }
470 } else {
471 ZEND_ASSERT(0);
472 }
473 }
474
475 void psi_to_object(zval *return_value, set_value *set, impl_val *r_val)
476 {
477 decl_var *var = set->vars->vars[0];
478 impl_val *ret_val = deref_impl_val(r_val, var);
479 psi_object *obj;
480
481 if ((intptr_t) ret_val->ptr > (intptr_t) 0) {
482 object_init_ex(return_value, psi_class_entry);
483 obj = PSI_OBJ(return_value, NULL);
484 obj->data = ret_val->ptr;
485 } else {
486 RETVAL_NULL();
487 }
488 }
489
490 static inline ZEND_RESULT_CODE psi_parse_args(zend_execute_data *execute_data, impl *impl)
491 {
492 size_t i;
493 impl_arg *iarg;
494 zend_error_handling zeh;
495
496 zend_replace_error_handling(EH_THROW, zend_exception_get_default(), &zeh);
497
498 if (!impl->func->args->count) {
499 ZEND_RESULT_CODE rv;
500
501 rv = zend_parse_parameters_none();
502 zend_restore_error_handling(&zeh);
503 return rv;
504 }
505
506 ZEND_PARSE_PARAMETERS_START(psi_num_min_args(impl), impl->func->args->vararg.name ? -1 : impl->func->args->count)
507 nextarg:
508 if (impl->func->args->vararg.name && _i >= impl->func->args->count) {
509 impl_arg *varg = impl->func->args->vararg.name;
510 iarg = init_impl_arg(
511 init_impl_type(varg->type->type, varg->type->name),
512 init_impl_var(varg->var->name, varg->var->reference),
513 NULL);
514
515 Z_PARAM_OPTIONAL;
516 if (_i == impl->func->args->count) {
517 impl->func->args->vararg.args = init_impl_args(iarg);
518 } else {
519 add_impl_arg(impl->func->args->vararg.args, iarg);
520 }
521 } else {
522 iarg = impl->func->args->args[_i];
523 if (iarg->def) {
524 Z_PARAM_OPTIONAL;
525 }
526 }
527 if (PSI_T_BOOL == iarg->type->type) {
528 Z_PARAM_BOOL(iarg->val.zend.bval);
529 } else if (PSI_T_INT == iarg->type->type) {
530 Z_PARAM_LONG(iarg->val.zend.lval);
531 } else if (PSI_T_FLOAT == iarg->type->type || PSI_T_DOUBLE == iarg->type->type) {
532 Z_PARAM_DOUBLE(iarg->val.dval);
533 } else if (PSI_T_STRING == iarg->type->type) {
534 Z_PARAM_STR_EX(iarg->val.zend.str, 1, iarg->var->reference);
535 if (iarg->val.zend.str) {
536 zend_string_addref(iarg->val.zend.str);
537 }
538 } else if (PSI_T_ARRAY == iarg->type->type) {
539 Z_PARAM_PROLOGUE(0);
540 } else if (PSI_T_OBJECT == iarg->type->type) {
541 Z_PARAM_PROLOGUE(0);
542 } else if (PSI_T_MIXED == iarg->type->type) {
543 Z_PARAM_PROLOGUE(0);
544 } else {
545 error_code = ZPP_ERROR_FAILURE;
546 break;
547 }
548 iarg->_zv = _arg;
549 ZVAL_DEREF(iarg->_zv);
550 if (_i < _num_args) {
551 goto nextarg;
552 }
553 ZEND_PARSE_PARAMETERS_END_EX(
554 zend_restore_error_handling(&zeh);
555 return FAILURE
556 );
557
558 /* set up defaults */
559 for (i = 0; i < impl->func->args->count; ++i) {
560 if (i >= EX_NUM_ARGS() && iarg->def) {
561 iarg = impl->func->args->args[i];
562
563 switch (iarg->type->type) {
564 case PSI_T_BOOL:
565 iarg->val.zend.bval = iarg->def->type == PSI_T_TRUE ? 1 : 0;
566 break;
567 case PSI_T_INT:
568 iarg->val.zend.lval = zend_atol(iarg->def->text, strlen(iarg->def->text));
569 break;
570 case PSI_T_FLOAT:
571 case PSI_T_DOUBLE:
572 iarg->val.dval = zend_strtod(iarg->def->text, NULL);
573 break;
574 case PSI_T_STRING:
575 /* FIXME */
576 iarg->val.zend.str = zend_string_init(&iarg->def->text[1], strlen(iarg->def->text) - 2, 0);
577 break;
578 }
579 }
580 }
581
582 zend_restore_error_handling(&zeh);
583 return SUCCESS;
584 }
585
586 static inline void *psi_do_calloc(let_calloc *alloc)
587 {
588 zend_long n = psi_long_num_exp(alloc->nmemb, NULL), s = psi_long_num_exp(alloc->size, NULL);
589 void *mem = safe_emalloc(n, s, sizeof(void *));
590 memset(mem, 0, n * s + sizeof(void *));
591 return mem;
592 }
593
594 static inline ZEND_RESULT_CODE psi_let_val(token_t let_func, impl_arg *iarg, impl_val *arg_val, decl_struct *strct, void **to_free)
595 {
596 switch (let_func) {
597 case PSI_T_BOOLVAL:
598 if (iarg->type->type == PSI_T_BOOL) {
599 arg_val->cval = iarg->val.zend.bval;
600 } else {
601 arg_val->cval = zend_is_true(iarg->_zv);
602 }
603 break;
604 case PSI_T_INTVAL:
605 if (iarg->type->type == PSI_T_INT) {
606 arg_val->lval = iarg->val.zend.lval;
607 } else {
608 arg_val->lval = zval_get_long(iarg->_zv);
609 }
610 break;
611 case PSI_T_FLOATVAL:
612 if (iarg->type->type == PSI_T_FLOAT || iarg->type->type == PSI_T_DOUBLE) {
613 arg_val->dval = iarg->val.dval;
614 } else {
615 arg_val->dval = zval_get_double(iarg->_zv);
616 }
617 break;
618 case PSI_T_PATHVAL:
619 case PSI_T_STRVAL:
620 if (iarg->type->type == PSI_T_STRING) {
621 if (iarg->val.zend.str) {
622 arg_val->ptr = estrdup(iarg->val.zend.str->val);
623 *to_free = arg_val->ptr;
624 } else {
625 arg_val->ptr = "";
626 }
627 } else {
628 zend_string *zs = zval_get_string(iarg->_zv);
629 arg_val->ptr = estrdup(zs->val);
630 *to_free = arg_val->ptr;
631 zend_string_release(zs);
632 }
633 if (PSI_T_PATHVAL == let_func) {
634 if (SUCCESS != php_check_open_basedir(arg_val->ptr)) {
635 efree(arg_val->ptr);
636 return FAILURE;
637 }
638 }
639 break;
640 case PSI_T_STRLEN:
641 if (iarg->type->type == PSI_T_STRING) {
642 if (iarg->val.zend.str) {
643 arg_val->lval = iarg->val.zend.str->len;
644 } else {
645 arg_val->lval = 0;
646 }
647 } else {
648 zend_string *zs = zval_get_string(iarg->_zv);
649 arg_val->lval = zs->len;
650 zend_string_release(zs);
651 }
652 break;
653 case PSI_T_ARRVAL:
654 if (iarg->type->type == PSI_T_ARRAY) {
655 arg_val->ptr = psi_array_to_struct(strct, HASH_OF(iarg->_zv));
656 *to_free = arg_val->ptr;
657 }
658 break;
659 case PSI_T_OBJVAL:
660 if (iarg->type->type == PSI_T_OBJECT) {
661 psi_object *obj;
662
663 if (!instanceof_function(Z_OBJCE_P(iarg->_zv), psi_class_entry)) {
664 return FAILURE;
665 }
666
667 obj = PSI_OBJ(iarg->_zv, NULL);
668 arg_val->ptr = obj->data;
669 }
670 break;
671 EMPTY_SWITCH_DEFAULT_CASE();
672 }
673 return SUCCESS;
674 }
675
676 static inline void *psi_do_let(let_stmt *let)
677 {
678 decl_arg *darg = let->var->arg;
679 impl_val *arg_val = darg->ptr;
680 impl_arg *iarg;
681
682 switch (let->val ? let->val->kind : PSI_LET_NULL) {
683 case PSI_LET_TMP:
684 arg_val->ptr = deref_impl_val(let->val->data.var->arg->val.ptr, let->var);
685 break;
686 case PSI_LET_NULL:
687 if (darg->var->array_size) {
688 arg_val->ptr = ecalloc(darg->var->array_size, sizeof(*arg_val));
689 darg->mem = arg_val->ptr;
690 } else {
691 memset(arg_val, 0, sizeof(*arg_val));
692 }
693 break;
694 case PSI_LET_CALLOC:
695 arg_val->ptr = psi_do_calloc(let->val->data.alloc);
696 darg->mem = arg_val->ptr;
697 break;
698 case PSI_LET_NUMEXP:
699 arg_val->zend.lval = psi_long_num_exp(let->val->data.num, NULL);
700 break;
701 case PSI_LET_FUNC:
702 iarg = let->val->data.func->arg;
703
704 if (SUCCESS != psi_let_val(let->val->data.func->type, iarg, darg->ptr, real_decl_type(darg->type)->strct, &darg->mem)) {
705 return NULL;
706 }
707 }
708
709 if (let->val && let->val->flags.one.is_reference) {
710 return let->ptr = &darg->ptr;
711 } else {
712 return let->ptr = darg->ptr;
713 }
714 }
715
716 static inline void psi_do_set(zval *return_value, set_value *set)
717 {
718 zval_dtor(return_value);
719 set->func->handler(return_value, set, set->vars->vars[0]->arg->ptr);
720 }
721
722 static inline void psi_do_return(zval *return_value, return_stmt *ret)
723 {
724 ret->set->func->handler(return_value, ret->set, ret->set->vars->vars[0]->arg->ptr);
725 }
726
727 static inline void psi_do_free(free_stmt *fre)
728 {
729 size_t i, j;
730
731 for (i = 0; i < fre->calls->count; ++i) {
732 free_call *f = fre->calls->list[i];
733
734 for (j = 0; j < f->vars->count; ++j) {
735 decl_var *dvar = f->vars->vars[j];
736 decl_arg *darg = dvar->arg;
737
738 f->decl->call.args[j] = &darg->val;
739 }
740
741 /* FIXME: check in validate_* that free functions return scalar */
742 PSI_ContextCall(&PSI_G(context), &f->decl->call, NULL);
743 }
744 }
745
746 static inline void psi_do_clean(impl *impl)
747 {
748 size_t i;
749
750 for (i = 0; i < impl->func->args->count; ++i ) {
751 impl_arg *iarg = impl->func->args->args[i];
752
753 switch (iarg->type->type) {
754 case PSI_T_STRING:
755 if (iarg->val.zend.str) {
756 zend_string_release(iarg->val.zend.str);
757 }
758 break;
759 }
760 }
761
762 if (impl->decl->args) for (i = 0; i < impl->decl->args->count; ++i) {
763 decl_arg *darg = impl->decl->args->args[i];
764
765 if (darg->mem) {
766 decl_type *type = real_decl_type(darg->type);
767
768 if (type->type == PSI_T_STRUCT) {
769 void **ptr = (void **) ((char *) darg->mem + type->strct->size);
770
771 while (*ptr) {
772 efree(*ptr++);
773 }
774 }
775 efree(darg->mem);
776 darg->mem = NULL;
777 }
778 }
779
780 if (impl->func->args->vararg.args) {
781 free_impl_args(impl->func->args->vararg.args);
782 impl->func->args->vararg.args = NULL;
783 }
784 if (impl->func->args->vararg.types) {
785 efree(impl->func->args->vararg.types);
786 impl->func->args->vararg.types = NULL;
787 }
788 if (impl->func->args->vararg.values) {
789 efree(impl->func->args->vararg.values);
790 impl->func->args->vararg.values = NULL;
791 }
792 if (impl->func->args->vararg.free_list) {
793 void **list = impl->func->args->vararg.free_list;
794
795 while (*list) {
796 efree(*list++);
797 }
798
799 efree(impl->func->args->vararg.free_list);
800 impl->func->args->vararg.free_list = NULL;
801 }
802 }
803
804 static inline int psi_calc_num_exp_value(num_exp *exp, impl_val *strct, impl_val *res) {
805 impl_val *ref, *tmp = NULL;
806
807 switch (exp->t) {
808 case PSI_T_NUMBER:
809 switch (is_numeric_string(exp->u.numb, strlen(exp->u.numb), (zend_long *) res, (double *) res, 0)) {
810 case IS_LONG:
811 return PSI_T_INT64;
812 case IS_DOUBLE:
813 return PSI_T_DOUBLE;
814 }
815 break;
816
817 case PSI_T_NSNAME:
818 switch (exp->u.cnst->type->type) {
819 case PSI_T_INT:
820 res->i64 = zend_get_constant_str(exp->u.cnst->name, strlen(exp->u.cnst->name))->value.lval;
821 return PSI_T_INT64;
822 case PSI_T_FLOAT:
823 res->dval = zend_get_constant_str(exp->u.cnst->name, strlen(exp->u.cnst->name))->value.dval;
824 return PSI_T_DOUBLE;
825 default:
826 return 0;
827 }
828 break;
829
830 case PSI_T_NAME:
831 if (strct) {
832 ref = struct_member_ref(exp->u.dvar->arg, strct, &tmp);
833 } else {
834 ref = exp->u.dvar->arg->ptr;
835 }
836 switch (real_decl_type(exp->u.dvar->arg->type)->type) {
837 case PSI_T_INT8:
838 case PSI_T_UINT8:
839 case PSI_T_INT16:
840 case PSI_T_UINT16:
841 case PSI_T_INT32:
842 case PSI_T_UINT32:
843 case PSI_T_INT64:
844 case PSI_T_UINT64:
845 memcpy(res, deref_impl_val(ref, exp->u.dvar), sizeof(*res));
846 if (tmp) {
847 free(tmp);
848 }
849 return real_decl_type(exp->u.dvar->arg->type)->type;
850
851 case PSI_T_FLOAT:
852 case PSI_T_DOUBLE:
853 memcpy(res, deref_impl_val(ref, exp->u.dvar), sizeof(*res));
854 if (tmp) {
855 free(tmp);
856 }
857 return real_decl_type(exp->u.dvar->arg->type)->type;
858
859 EMPTY_SWITCH_DEFAULT_CASE();
860 }
861 break;
862
863 EMPTY_SWITCH_DEFAULT_CASE();
864 }
865 return 0;
866 }
867
868 int psi_calc_num_exp(num_exp *exp, impl_val *strct, impl_val *res) {
869 impl_val num = {0};
870 int num_type = psi_calc_num_exp_value(exp, strct, &num);
871
872 if (exp->operand) {
873 impl_val tmp = {0};
874 int tmp_type = psi_calc_num_exp(exp->operand, strct, &tmp);
875
876 return exp->calculator(num_type, &num, tmp_type, &tmp, res);
877 }
878
879 memcpy(res, &num, sizeof(*res));
880 return num_type;
881 }
882
883 #define PRIfval "f"
884 #define PRIdval "lf"
885
886 #define PSI_CALC_OP(var) do { \
887 const char *fmt = "calc %" PRI##var ", %" PRI##var ": %" PRI##var "\n"; \
888 res->var = PSI_CALC(v1->var, v2->var); \
889 if (!res->var) fprintf(stderr, fmt, v1->var, v2->var, res->var); \
890 } while (0)
891 #define PSI_CALC_OP2(vres, var1, var2) do { \
892 const char *fmt = "calc %" PRI##var1 ", %" PRI##var2 ": %" PRI##vres "\n"; \
893 res->vres = PSI_CALC(v1->var1, v2->var2); \
894 if (!res->vres) fprintf(stderr, fmt, v1->var1, v2->var2, res->vres); \
895 } while(0)
896 #define PSI_CALC_FN(op) int psi_calc_##op(int t1, impl_val *v1, int t2, impl_val *v2, impl_val *res) \
897 { \
898 if (t1 == t2) { \
899 switch (t1) { \
900 case PSI_T_FLOAT: PSI_CALC_OP(fval); break; \
901 case PSI_T_DOUBLE: PSI_CALC_OP(dval); break; \
902 case PSI_T_INT8: PSI_CALC_OP(i8); break; \
903 case PSI_T_UINT8: PSI_CALC_OP(u8); break; \
904 case PSI_T_INT16: PSI_CALC_OP(i16); break; \
905 case PSI_T_UINT16: PSI_CALC_OP(u16); break; \
906 case PSI_T_INT32: PSI_CALC_OP(i32); break; \
907 case PSI_T_UINT32: PSI_CALC_OP(u32); break; \
908 case PSI_T_INT64: PSI_CALC_OP(i64); break; \
909 case PSI_T_UINT64: PSI_CALC_OP(u64); break; \
910 EMPTY_SWITCH_DEFAULT_CASE(); \
911 } \
912 return t1; \
913 } else if (t1 == PSI_T_DOUBLE) { \
914 switch (t2) { \
915 case PSI_T_FLOAT: PSI_CALC_OP2(dval, dval, fval); break; \
916 case PSI_T_INT8: PSI_CALC_OP2(dval, dval, i8); break; \
917 case PSI_T_UINT8: PSI_CALC_OP2(dval, dval, u8); break; \
918 case PSI_T_INT16: PSI_CALC_OP2(dval, dval, i16); break; \
919 case PSI_T_UINT16: PSI_CALC_OP2(dval, dval, u16); break; \
920 case PSI_T_INT32: PSI_CALC_OP2(dval, dval, i32); break; \
921 case PSI_T_UINT32: PSI_CALC_OP2(dval, dval, u32); break; \
922 case PSI_T_INT64: PSI_CALC_OP2(dval, dval, i64); break; \
923 case PSI_T_UINT64: PSI_CALC_OP2(dval, dval, u64); break; \
924 EMPTY_SWITCH_DEFAULT_CASE(); \
925 } \
926 return t1; \
927 } else if (t2 == PSI_T_DOUBLE) { \
928 switch (t1) { \
929 case PSI_T_FLOAT: PSI_CALC_OP2(dval, fval, dval); break; \
930 case PSI_T_INT8: PSI_CALC_OP2(dval, i8, dval); break; \
931 case PSI_T_UINT8: PSI_CALC_OP2(dval, u8, dval); break; \
932 case PSI_T_INT16: PSI_CALC_OP2(dval, i16, dval); break; \
933 case PSI_T_UINT16: PSI_CALC_OP2(dval, u16, dval); break; \
934 case PSI_T_INT32: PSI_CALC_OP2(dval, i32, dval); break; \
935 case PSI_T_UINT32: PSI_CALC_OP2(dval, u32, dval); break; \
936 case PSI_T_INT64: PSI_CALC_OP2(dval, i64, dval); break; \
937 case PSI_T_UINT64: PSI_CALC_OP2(dval, u64, dval); break; \
938 EMPTY_SWITCH_DEFAULT_CASE(); \
939 } \
940 return t2; \
941 } else if (t1 == PSI_T_FLOAT) { \
942 switch (t2) { \
943 case PSI_T_DOUBLE: PSI_CALC_OP2(dval, fval, dval); return t2; \
944 case PSI_T_INT8: PSI_CALC_OP2(fval, fval, i8); break; \
945 case PSI_T_UINT8: PSI_CALC_OP2(fval, fval, u8); break; \
946 case PSI_T_INT16: PSI_CALC_OP2(fval, fval, i16); break; \
947 case PSI_T_UINT16: PSI_CALC_OP2(fval, fval, u16); break; \
948 case PSI_T_INT32: PSI_CALC_OP2(fval, fval, i32); break; \
949 case PSI_T_UINT32: PSI_CALC_OP2(fval, fval, u32); break; \
950 case PSI_T_INT64: PSI_CALC_OP2(fval, fval, i64); break; \
951 case PSI_T_UINT64: PSI_CALC_OP2(fval, fval, u64); break; \
952 EMPTY_SWITCH_DEFAULT_CASE(); \
953 } \
954 return t1; \
955 } else if (t2 == PSI_T_FLOAT) { \
956 switch (t1) { \
957 case PSI_T_DOUBLE: PSI_CALC_OP2(dval, dval, fval); return t1; \
958 case PSI_T_INT8: PSI_CALC_OP2(fval, i8, fval); break; \
959 case PSI_T_UINT8: PSI_CALC_OP2(fval, u8, fval); break; \
960 case PSI_T_INT16: PSI_CALC_OP2(fval, i16, fval); break; \
961 case PSI_T_UINT16: PSI_CALC_OP2(fval, u16, fval); break; \
962 case PSI_T_INT32: PSI_CALC_OP2(fval, i32, fval); break; \
963 case PSI_T_UINT32: PSI_CALC_OP2(fval, u32, fval); break; \
964 case PSI_T_INT64: PSI_CALC_OP2(fval, i64, fval); break; \
965 case PSI_T_UINT64: PSI_CALC_OP2(fval, u64, fval); break; \
966 EMPTY_SWITCH_DEFAULT_CASE(); \
967 } \
968 return t2; \
969 } else { \
970 int64_t sval1 = v1->i64, sval2 = v2->i64; \
971 uint64_t uval1 = v1->u64, uval2 = v2->u64; \
972 switch (t1) { \
973 case PSI_T_INT8: sval1 >>= 8; \
974 case PSI_T_INT16: sval1 >>= 8; \
975 case PSI_T_INT32: sval1 >>= 8; \
976 case PSI_T_INT64: \
977 switch (t2) { \
978 case PSI_T_INT8: sval2 >>= 8; \
979 case PSI_T_INT16: sval2 >>= 8; \
980 case PSI_T_INT32: sval2 >>= 8; \
981 case PSI_T_INT64: \
982 res->i64 = PSI_CALC(sval1 , sval2); \
983 return PSI_T_INT64; \
984 case PSI_T_UINT8: uval2 >>= 8; \
985 case PSI_T_UINT16: uval2 >>= 8; \
986 case PSI_T_UINT32: uval2 >>= 8; \
987 case PSI_T_UINT64: \
988 res->i64 = PSI_CALC(sval1, uval2); \
989 return PSI_T_INT64; \
990 } \
991 break; \
992 case PSI_T_UINT8: uval1 >>= 8; \
993 case PSI_T_UINT16: uval1 >>= 8; \
994 case PSI_T_UINT32: uval1 >>= 8; \
995 case PSI_T_UINT64: \
996 switch (t2) { \
997 case PSI_T_INT8: sval2 >>= 8; \
998 case PSI_T_INT16: sval2 >>= 8; \
999 case PSI_T_INT32: sval2 >>= 8; \
1000 case PSI_T_INT64: \
1001 res->i64 = PSI_CALC(uval1, sval2); \
1002 return PSI_T_INT64; \
1003 case PSI_T_UINT8: uval2 >>= 8; \
1004 case PSI_T_UINT16: uval2 >>= 8; \
1005 case PSI_T_UINT32: uval2 >>= 8; \
1006 case PSI_T_UINT64: \
1007 res->u64 = PSI_CALC(uval1, uval2); \
1008 return PSI_T_UINT64; \
1009 } \
1010 break; \
1011 } \
1012 } \
1013 ZEND_ASSERT(0); \
1014 return 0; \
1015 }
1016
1017 #undef PSI_CALC
1018 #define PSI_CALC(var1, var2) (var1) + (var2)
1019 PSI_CALC_FN(add)
1020 #undef PSI_CALC
1021 #define PSI_CALC(var1, var2) (var1) * (var2)
1022 PSI_CALC_FN(mul)
1023 #undef PSI_CALC
1024 #define PSI_CALC(var1, var2) (var1) - (var2)
1025 PSI_CALC_FN(sub)
1026 #undef PSI_CALC
1027 #define PSI_CALC(var1, var2) (var1) / (var2)
1028 PSI_CALC_FN(div)
1029
1030 static inline void psi_do_args(impl *impl) {
1031 size_t i;
1032
1033 for (i = 0; i < impl->decl->args->count; ++i) {
1034 impl->decl->call.args[i] = impl->decl->args->args[i]->let->ptr;
1035 }
1036 }
1037
1038 static inline impl_vararg *psi_do_varargs(impl *impl) {
1039 size_t i, j;
1040 impl_vararg *va = &impl->func->args->vararg;
1041 size_t vacount = va->args->count;
1042
1043
1044 if (!vacount) {
1045 return NULL;
1046 }
1047
1048 va->types = ecalloc(vacount, sizeof(*va->types));
1049 va->values = ecalloc(vacount, sizeof(*va->values));
1050
1051 for (i = 0, j = 0; i < vacount; ++i) {
1052 impl_arg *vaarg = va->args->args[i];
1053 void *to_free = NULL;
1054 token_t let_fn, vatype = va->name->type->type;
1055
1056 if (vatype == PSI_T_MIXED) {
1057 switch (Z_TYPE_P(vaarg->_zv)) {
1058 case IS_TRUE:
1059 case IS_FALSE: vatype = PSI_T_BOOL; break;
1060 case IS_LONG: vatype = PSI_T_INT; break;
1061 case IS_DOUBLE: vatype = PSI_T_FLOAT; break;
1062 default: vatype = PSI_T_STRING; break;
1063 }
1064 }
1065
1066
1067 switch (vatype) {
1068 case PSI_T_BOOL: let_fn = PSI_T_BOOLVAL; break;
1069 case PSI_T_INT: let_fn = PSI_T_INTVAL; break;
1070 case PSI_T_FLOAT:
1071 case PSI_T_DOUBLE: let_fn = PSI_T_FLOATVAL;break;
1072 case PSI_T_STRING: let_fn = PSI_T_STRVAL; break;
1073 EMPTY_SWITCH_DEFAULT_CASE();
1074 }
1075
1076 va->types[i] = vatype;
1077 psi_let_val(let_fn, vaarg, &va->values[i], NULL, &to_free);
1078
1079 if (to_free) {
1080 if (!va->free_list) {
1081 va->free_list = ecalloc(vacount - i + 1, sizeof(*va->free_list));
1082 }
1083 va->free_list[j++] = to_free;
1084 }
1085 }
1086
1087 return va;
1088 }
1089
1090 void psi_call(zend_execute_data *execute_data, zval *return_value, impl *impl)
1091 {
1092 size_t i;
1093 impl_vararg *va = NULL;
1094
1095 memset(impl->decl->func->ptr, 0, sizeof(impl_val));
1096
1097 if (SUCCESS != psi_parse_args(execute_data, impl)) {
1098 return;
1099 }
1100
1101 for (i = 0; i < impl->stmts->let.count; ++i) {
1102 let_stmt *let = impl->stmts->let.list[i];
1103
1104 if (!psi_do_let(let)) {
1105 psi_do_return(return_value, impl->stmts->ret.list[0]);
1106 psi_do_clean(impl);
1107 return;
1108 }
1109 }
1110
1111 if (impl->decl->args) {
1112 psi_do_args(impl);
1113
1114 if (impl->func->args->vararg.args) {
1115 va = psi_do_varargs(impl);
1116 }
1117 }
1118
1119 PSI_ContextCall(&PSI_G(context), &impl->decl->call, va);
1120 psi_do_return(return_value, impl->stmts->ret.list[0]);
1121
1122 for (i = 0; i < impl->stmts->set.count; ++i) {
1123 set_stmt *set = impl->stmts->set.list[i];
1124
1125 if (set->arg->_zv) {
1126 psi_do_set(set->arg->_zv, set->val);
1127 }
1128 }
1129
1130 for (i = 0; i < impl->stmts->fre.count; ++i) {
1131 free_stmt *fre = impl->stmts->fre.list[i];
1132
1133 psi_do_free(fre);
1134 }
1135 psi_do_clean(impl);
1136 }
1137
1138 static void psi_object_free(zend_object *o)
1139 {
1140 psi_object *obj = PSI_OBJ(NULL, o);
1141
1142 if (obj->data) {
1143 // free(obj->data);
1144 obj->data = NULL;
1145 }
1146 zend_object_std_dtor(o);
1147 }
1148
1149 static zend_object *psi_object_init(zend_class_entry *ce)
1150 {
1151 psi_object *o = ecalloc(1, sizeof(*o) + zend_object_properties_size(ce));
1152
1153 zend_object_std_init(&o->std, ce);
1154 object_properties_init(&o->std, ce);
1155 o->std.handlers = &psi_object_handlers;
1156 return &o->std;
1157 }
1158
1159 PHP_MINIT_FUNCTION(psi)
1160 {
1161 PSI_ContextOps *ops = NULL;
1162 zend_class_entry ce = {0};
1163
1164 REGISTER_INI_ENTRIES();
1165
1166 INIT_NS_CLASS_ENTRY(ce, "psi", "object", NULL);
1167 psi_class_entry = zend_register_internal_class_ex(&ce, NULL);
1168 psi_class_entry->create_object = psi_object_init;
1169
1170 memcpy(&psi_object_handlers, zend_get_std_object_handlers(), sizeof(zend_object_handlers));
1171 psi_object_handlers.offset = XtOffsetOf(psi_object, std);
1172 psi_object_handlers.free_obj = psi_object_free;
1173 psi_object_handlers.clone_obj = NULL;
1174
1175 #ifdef HAVE_LIBJIT
1176 if (!strcasecmp(PSI_G(engine), "jit")) {
1177 ops = PSI_Libjit();
1178 } else
1179 #endif
1180 #ifdef HAVE_LIBFFI
1181 ops = PSI_Libffi();
1182 #endif
1183
1184 if (!ops) {
1185 php_error(E_WARNING, "No PSI engine found");
1186 return FAILURE;
1187 }
1188
1189 PSI_ContextInit(&PSI_G(context), ops, psi_error_wrapper);
1190 PSI_ContextBuild(&PSI_G(context), PSI_G(directory));
1191
1192 if (psi_check_env("PSI_DUMP")) {
1193 PSI_ContextDump(&PSI_G(context), STDOUT_FILENO);
1194 }
1195
1196 return SUCCESS;
1197 }
1198
1199 PHP_MSHUTDOWN_FUNCTION(psi)
1200 {
1201 PSI_ContextDtor(&PSI_G(context));
1202
1203 UNREGISTER_INI_ENTRIES();
1204
1205 return SUCCESS;
1206 }
1207
1208 #if defined(COMPILE_DL_PSI) && defined(ZTS)
1209 PHP_RINIT_FUNCTION(psi)
1210 {
1211 ZEND_TSRMLS_CACHE_UPDATE();
1212 return SUCCESS;
1213 }
1214 #endif
1215
1216 PHP_MINFO_FUNCTION(psi)
1217 {
1218 php_info_print_table_start();
1219 php_info_print_table_header(2, "psi support", "enabled");
1220 php_info_print_table_end();
1221
1222 DISPLAY_INI_ENTRIES();
1223 }
1224 const zend_function_entry psi_functions[] = {
1225 PHP_FE_END
1226 };
1227
1228 zend_module_entry psi_module_entry = {
1229 STANDARD_MODULE_HEADER,
1230 "psi",
1231 psi_functions,
1232 PHP_MINIT(psi),
1233 PHP_MSHUTDOWN(psi),
1234 #if defined(COMPILE_DL_PSI) && defined(ZTS)
1235 PHP_RINIT(psi), /* Replace with NULL if there's nothing to do at request start */
1236 #else
1237 NULL,
1238 #endif
1239 NULL,
1240 PHP_MINFO(psi),
1241 PHP_PSI_VERSION,
1242 STANDARD_MODULE_PROPERTIES
1243 };
1244
1245 #ifdef COMPILE_DL_PSI
1246 #ifdef ZTS
1247 ZEND_TSRMLS_CACHE_DEFINE();
1248 #endif
1249 ZEND_GET_MODULE(psi)
1250 #endif
1251
1252 /*
1253 * Local variables:
1254 * tab-width: 4
1255 * c-basic-offset: 4
1256 * End:
1257 * vim600: noet sw=4 ts=4 fdm=marker
1258 * vim<600: noet sw=4 ts=4
1259 */
PHP System Interface / POSIX Standard Interface