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