8 #include "ext/standard/info.h"
16 # define PSI_ENGINE "jit"
21 # define PSI_ENGINE "ffi"
24 ZEND_DECLARE_MODULE_GLOBALS(psi
);
27 STD_PHP_INI_ENTRY("psi.engine", PSI_ENGINE
, PHP_INI_SYSTEM
, OnUpdateString
, engine
, zend_psi_globals
, psi_globals
)
28 STD_PHP_INI_ENTRY("psi.directory", "psi.d", PHP_INI_SYSTEM
, OnUpdateString
, directory
, zend_psi_globals
, psi_globals
)
31 void psi_error(int type
, const char *msg
, ...)
37 vslprintf(buf
, 0x1000, msg
, argv
);
43 size_t psi_t_alignment(token_t t
)
46 #define PSI_TAS_D(T) struct PSI_TAS_ ##T { \
50 #define PSI_TAS_P(T) struct PSI_TAS_ ## T ## _pointer { \
54 #define PSI_TAS_C(T) align = offsetof(struct PSI_TAS_ ##T, x)
55 #define PSI_TAS_CASE(T) { \
64 PSI_TAS_CASE(uint8_t);
67 PSI_TAS_CASE(int16_t);
70 PSI_TAS_CASE(uint16_t);
73 PSI_TAS_CASE(int32_t);
76 PSI_TAS_CASE(uint32_t);
79 PSI_TAS_CASE(int64_t);
82 PSI_TAS_CASE(uint64_t);
93 PSI_TAS_C(char_pointer
);
96 EMPTY_SWITCH_DEFAULT_CASE();
102 size_t psi_t_size(token_t t
)
127 size
= sizeof(float);
130 size
= sizeof(double);
133 size
= sizeof(char *);
135 EMPTY_SWITCH_DEFAULT_CASE();
140 size_t psi_t_align(token_t t
, size_t s
)
142 size_t a
= psi_t_alignment(t
);
143 return ((s
- 1) | (a
- 1)) + 1;
146 int psi_internal_type(impl_type
*type
)
148 switch (type
->type
) {
165 zend_internal_arg_info
*psi_internal_arginfo(impl
*impl
)
168 zend_internal_arg_info
*aip
;
169 zend_internal_function_info
*fi
;
171 aip
= calloc(impl
->func
->args
->count
+ 1, sizeof(*aip
));
173 fi
= (zend_internal_function_info
*) &aip
[0];
174 fi
->required_num_args
= psi_num_min_args(impl
);
175 fi
->return_reference
= impl
->func
->return_reference
;
176 fi
->type_hint
= psi_internal_type(impl
->func
->return_type
);
178 for (i
= 0; i
< impl
->func
->args
->count
; ++i
) {
179 impl_arg
*iarg
= impl
->func
->args
->args
[i
];
180 zend_internal_arg_info
*ai
= &aip
[i
+1];
182 ai
->name
= iarg
->var
->name
;
183 ai
->type_hint
= psi_internal_type(iarg
->type
);
184 if (iarg
->var
->reference
) {
185 ai
->pass_by_reference
= 1;
187 if (iarg
->var
->reference
|| (iarg
->def
&& iarg
->def
->type
== PSI_T_NULL
)) {
195 size_t psi_num_min_args(impl
*impl
)
197 size_t i
, n
= impl
->func
->args
->count
;
199 for (i
= 0; i
< impl
->func
->args
->count
; ++i
) {
200 if (impl
->func
->args
->args
[i
]->def
) {
207 void psi_to_bool(zval
*return_value
, set_value
*set
, impl_val
*ret_val
)
209 psi_to_int(return_value
, set
, ret_val
);
210 convert_to_boolean(return_value
);
213 void psi_to_int(zval
*return_value
, set_value
*set
, impl_val
*ret_val
)
215 decl_var
*var
= set
->vars
->vars
[0];
216 token_t t
= real_decl_type(var
->arg
->type
)->type
;
217 impl_val
*v
= deref_impl_val(ret_val
, var
);
221 RETVAL_DOUBLE((double) v
->fval
);
222 convert_to_long(return_value
);
225 RETVAL_DOUBLE(v
->dval
);
226 convert_to_long(return_value
);
244 #if UINT32_MAX >= ZEND_LONG_MAX
245 if (v
->u32
> ZEND_LONG_MAX
) {
248 RETVAL_STRING(zend_print_ulong_to_buf(&d
[10], v
->u32
));
252 #if UINT32_MAX >= ZEND_LONG_MAX
260 if (v
->u64
> ZEND_LONG_MAX
) {
263 RETVAL_STRING(zend_print_ulong_to_buf(&d
[22], v
->u64
));
268 EMPTY_SWITCH_DEFAULT_CASE();
272 void psi_to_double(zval
*return_value
, set_value
*set
, impl_val
*ret_val
)
274 decl_var
*var
= set
->vars
->vars
[0];
275 token_t t
= real_decl_type(var
->arg
->type
)->type
;
276 impl_val
*v
= deref_impl_val(ret_val
, var
);
280 RETVAL_DOUBLE((double) v
->fval
);
283 RETVAL_DOUBLE(v
->dval
);
286 RETVAL_DOUBLE((double) v
->lval
);
291 void psi_to_string(zval
*return_value
, set_value
*set
, impl_val
*ret_val
)
293 decl_var
*var
= set
->vars
->vars
[0];
294 token_t t
= real_decl_type(var
->arg
->type
)->type
;
299 if (!var
->arg
->var
->pointer_level
) {
300 RETVAL_STRINGL(&ret_val
->cval
, 1);
302 ret_val
= deref_impl_val(ret_val
, var
);
303 if (ret_val
&& ret_val
->ptr
) {
304 RETVAL_STRING(ret_val
->ptr
);
306 RETVAL_EMPTY_STRING();
311 RETVAL_DOUBLE((double) deref_impl_val(ret_val
, var
)->fval
);
314 RETVAL_DOUBLE(deref_impl_val(ret_val
, var
)->dval
);
317 RETVAL_LONG(deref_impl_val(ret_val
, var
)->lval
);
320 convert_to_string(return_value
);
324 static impl_val
*iterate(impl_val
*val
, size_t size
, unsigned i
, impl_val
*tmp
)
326 memset(tmp
, 0, sizeof(*tmp
));
327 memcpy(tmp
, ((void*) val
->ptr
) + size
* i
, size
);
331 void psi_from_zval(impl_val
*mem
, decl_arg
*spec
, zval
*zv
, void **tmp
)
333 decl_type
*type
= real_decl_type(spec
->type
);
335 switch (type
->type
) {
337 mem
->fval
= (float) zval_get_double(zv
);
340 mem
->dval
= zval_get_double(zv
);
344 if (spec
->var
->pointer_level
) {
345 zend_string
*zs
= zval_get_string(zv
);
346 *tmp
= mem
->ptr
= estrndup(zs
->val
, zs
->len
);
347 zend_string_release(zs
);
352 mem
->zend
.lval
= zval_get_long(zv
);
357 void *psi_array_to_struct(decl_struct
*s
, HashTable
*arr
)
360 char *mem
= ecalloc(1, s
->size
+ s
->args
->count
* sizeof(void *));
362 if (arr
) for (i
= 0; i
< s
->args
->count
; ++i
) {
363 decl_arg
*darg
= s
->args
->args
[i
];
364 zval
*entry
= zend_hash_str_find_ind(arr
, darg
->var
->name
, strlen(darg
->var
->name
));
370 memset(&tmp
, 0, sizeof(tmp
));
371 psi_from_zval(&val
, darg
, entry
, &tmp
);
372 memcpy(mem
+ darg
->layout
->pos
, &val
, darg
->layout
->len
);
374 ((void **)(mem
+ s
->size
))[j
++] = tmp
;
381 static inline impl_val
*struct_member_ref(decl_arg
*set_arg
, impl_val
*struct_ptr
, impl_val
**to_free
) {
382 void *ptr
= (char *) struct_ptr
->ptr
+ set_arg
->layout
->pos
;
383 impl_val
*val
= enref_impl_val(ptr
, set_arg
->var
);
391 void psi_to_array(zval
*return_value
, set_value
*set
, impl_val
*r_val
)
394 decl_var
*var
= set
->vars
->vars
[0];
395 token_t t
= real_decl_type(var
->arg
->type
)->type
;
396 impl_val tmp
, *ret_val
= deref_impl_val(r_val
, var
);
398 array_init(return_value
);
400 if (t
== PSI_T_STRUCT
) {
401 // decl_struct *s = real_decl_type(var->arg->type)->strct;
404 /* explicit member casts */
405 for (i
= 0; i
< set
->count
; ++i
) {
406 set_value
*sub_set
= set
->inner
[i
];
407 decl_var
*sub_var
= sub_set
->vars
->vars
[0];
409 sub_set
->outer
.val
= r_val
;
412 impl_val
*tmp
= NULL
, *val
;
415 val
= deref_impl_val(struct_member_ref(sub_var
->arg
, ret_val
, &tmp
), sub_var
);
416 sub_set
->func
->handler(&ztmp
, sub_set
, val
);
417 add_assoc_zval(return_value
, sub_var
->name
, &ztmp
);
428 if (var
->arg
->var
->array_size
) {
429 /* to_array(foo[NUMBER]) */
430 for (i
= 0; i
< var
->arg
->var
->array_size
; ++i
) {
431 size_t size
= psi_t_size(var
->arg
->var
->pointer_level
> 1 ? PSI_T_POINTER
: t
);
432 impl_val
*ptr
= iterate(ret_val
, size
, i
, &tmp
);
437 ZVAL_DOUBLE(&ele
, (double) ptr
->fval
);
440 ZVAL_DOUBLE(&ele
, ptr
->dval
);
443 ZVAL_LONG(&ele
, ptr
->lval
);
447 add_next_index_zval(return_value
, &ele
);
450 } else if (set
->vars
->count
> 1) {
451 /* to_array(arr_var, cnt_var[, cnt_var...], to_int(*arr_var))
452 * check for length in second var
457 if (set
->outer
.set
) {
459 for (i
= 1; i
< set
->vars
->count
; ++i
) {
460 impl_val
*tmp
= NULL
, *cnt_val
;
461 decl_var
*cnt_var
= set
->vars
->vars
[i
];
463 cnt_val
= struct_member_ref(cnt_var
->arg
, set
->outer
.val
, &tmp
);
464 count
+= deref_impl_val(cnt_val
, cnt_var
)->lval
;
474 for (i
= 0; i
< count
; ++i
) {
475 size_t size
= psi_t_size(var
->arg
->var
->pointer_level
? PSI_T_POINTER
: t
);
476 impl_val
*ptr
= iterate(ret_val
, size
, i
, &tmp
);
478 set
->inner
[0]->func
->handler(&ele
, set
->inner
[0], ptr
);
479 add_next_index_zval(return_value
, &ele
);
487 static inline ZEND_RESULT_CODE
psi_parse_args(zend_execute_data
*execute_data
, impl
*impl
)
491 if (!impl
->func
->args
->count
) {
492 return zend_parse_parameters_none();
495 ZEND_PARSE_PARAMETERS_START(psi_num_min_args(impl
), impl
->func
->args
->count
)
497 iarg
= impl
->func
->args
->args
[_i
];
501 if (PSI_T_BOOL
== iarg
->type
->type
) {
503 iarg
->val
.zend
.bval
= iarg
->def
->type
== PSI_T_TRUE
? 1 : 0;
505 Z_PARAM_BOOL(iarg
->val
.zend
.bval
);
506 } else if (PSI_T_INT
== iarg
->type
->type
) {
508 iarg
->val
.zend
.lval
= zend_atol(iarg
->def
->text
, strlen(iarg
->def
->text
));
510 Z_PARAM_LONG(iarg
->val
.zend
.lval
);
511 } else if (PSI_T_FLOAT
== iarg
->type
->type
|| PSI_T_DOUBLE
== iarg
->type
->type
) {
513 iarg
->val
.dval
= zend_strtod(iarg
->def
->text
, NULL
);
515 Z_PARAM_DOUBLE(iarg
->val
.dval
);
516 } else if (PSI_T_STRING
== iarg
->type
->type
) {
517 struct {char *val
; size_t len
;} str
;
520 str
.len
= strlen(iarg
->def
->text
) - 2;
521 str
.val
= &iarg
->def
->text
[1];
523 Z_PARAM_STR_EX(iarg
->val
.zend
.str
, 1, 0);
524 if (iarg
->val
.zend
.str
) {
525 zend_string_addref(iarg
->val
.zend
.str
);
526 } else if (iarg
->def
) {
527 iarg
->val
.zend
.str
= zend_string_init(str
.val
, str
.len
, 0);
529 } else if (PSI_T_ARRAY
== iarg
->type
->type
) {
530 /* handled as _zv in let or set */
533 error_code
= ZPP_ERROR_FAILURE
;
537 if (_i
< _max_num_args
) {
540 ZEND_PARSE_PARAMETERS_END_EX(return FAILURE
);
545 static inline void *psi_do_calloc(let_calloc
*alloc
)
547 decl_type
*type
= real_decl_type(alloc
->type
);
550 if (type
->type
== PSI_T_STRUCT
) {
551 /* psi_do_clean expects at least one NULL pointer after the struct */
552 size
= type
->strct
->size
+ sizeof(void *);
554 size
= psi_t_size(type
->type
);
557 return ecalloc(alloc
->n
, size
);
560 static inline void *psi_do_let(decl_arg
*darg
)
562 impl_arg
*iarg
= darg
->let
->arg
;
565 darg
->let
->ptr
= &darg
->let
->out
;
566 arg_val
= darg
->let
->ptr
;
569 /* let foo = calloc(1, long);
573 if (darg
->let
->val
->func
&& darg
->let
->val
->func
->type
== PSI_T_CALLOC
) {
574 arg_val
->ptr
= psi_do_calloc(darg
->let
->val
->func
->alloc
);
575 darg
->let
->mem
= arg_val
->ptr
;
576 } else if (darg
->var
->array_size
) {
577 arg_val
->ptr
= ecalloc(darg
->var
->array_size
, sizeof(*arg_val
));
578 darg
->let
->mem
= arg_val
->ptr
;
580 memset(arg_val
, 0, sizeof(*arg_val
));
584 switch (darg
->let
->val
->func
->type
) {
586 if (iarg
->type
->type
== PSI_T_BOOL
) {
587 arg_val
->cval
= iarg
->val
.zend
.bval
;
589 arg_val
->cval
= zend_is_true(iarg
->_zv
);
593 if (iarg
->type
->type
== PSI_T_INT
) {
594 arg_val
->lval
= iarg
->val
.zend
.lval
;
596 arg_val
->lval
= zval_get_long(iarg
->_zv
);
600 if (iarg
->type
->type
== PSI_T_STRING
) {
601 arg_val
->ptr
= estrdup(iarg
->val
.zend
.str
->val
);
602 darg
->let
->mem
= arg_val
->ptr
;
603 zend_string_release(iarg
->val
.zend
.str
);
605 zend_string
*zs
= zval_get_string(iarg
->_zv
);
606 arg_val
->ptr
= estrdup(zs
->val
);
607 darg
->let
->mem
= arg_val
->ptr
;
608 zend_string_release(zs
);
612 if (iarg
->type
->type
== PSI_T_STRING
) {
613 arg_val
->lval
= iarg
->val
.zend
.str
->len
;
614 zend_string_release(iarg
->val
.zend
.str
);
616 zend_string
*zs
= zval_get_string(iarg
->_zv
);
617 arg_val
->lval
= zs
->len
;
618 zend_string_release(zs
);
622 if (iarg
->type
->type
== PSI_T_ARRAY
) {
623 decl_type
*type
= real_decl_type(darg
->type
);
625 switch (type
->type
) {
627 arg_val
->ptr
= psi_array_to_struct(type
->strct
, HASH_OF(iarg
->_zv
));
628 darg
->let
->mem
= arg_val
->ptr
;
633 EMPTY_SWITCH_DEFAULT_CASE();
637 if (darg
->let
->val
&& darg
->let
->val
->is_reference
) {
638 return &darg
->let
->ptr
;
640 return darg
->let
->ptr
;
644 static inline void psi_do_set(zval
*return_value
, set_value
*set
)
646 ZVAL_DEREF(return_value
);
647 zval_dtor(return_value
);
649 set
->func
->handler(return_value
, set
, set
->vars
->vars
[0]->arg
->let
->ptr
);
652 static inline void psi_do_return(zval
*return_value
, return_stmt
*ret
, impl_val
*ret_val
)
654 ret
->set
->func
->handler(return_value
, ret
->set
, ret_val
);
657 static inline void psi_do_free(free_stmt
*fre
)
662 for (i
= 0; i
< fre
->calls
->count
; ++i
) {
663 free_call
*f
= fre
->calls
->list
[i
];
665 for (j
= 0; j
< f
->vars
->count
; ++j
) {
666 decl_var
*dvar
= f
->vars
->vars
[j
];
667 decl_arg
*darg
= dvar
->arg
;
669 f
->decl
->call
.args
[j
] = &darg
->let
->out
;
672 /* FIXME: check in validate_* that free functions return scalar */
673 PSI_ContextCall(&PSI_G(context
), &dummy
, f
->decl
);
677 static inline void psi_do_clean(impl
*impl
)
681 for (i
= 0; i
< impl
->func
->args
->count
; ++i
) {
682 impl_arg
*iarg
= impl
->func
->args
->args
[i
];
684 switch (iarg
->type
->type
) {
686 if (iarg
->val
.zend
.str
) {
687 zend_string_release(iarg
->val
.zend
.str
);
693 if (impl
->decl
->args
) for (i
= 0; i
< impl
->decl
->args
->count
; ++i
) {
694 decl_arg
*darg
= impl
->decl
->args
->args
[i
];
696 if (darg
->let
&& darg
->let
->mem
) {
697 decl_type
*type
= real_decl_type(darg
->type
);
699 if (type
->type
== PSI_T_STRUCT
) {
700 void **ptr
= (void **) ((char *) darg
->let
->mem
+ type
->strct
->size
);
706 efree(darg
->let
->mem
);
707 darg
->let
->mem
= NULL
;
712 void psi_call(zend_execute_data
*execute_data
, zval
*return_value
, impl
*impl
)
717 if (SUCCESS
!= psi_parse_args(execute_data
, impl
)) {
721 if (impl
->decl
->args
) {
722 for (i
= 0; i
< impl
->decl
->args
->count
; ++i
) {
723 decl_arg
*darg
= impl
->decl
->args
->args
[i
];
725 impl
->decl
->call
.args
[i
] = psi_do_let(darg
);
729 memset(&ret_val
, 0, sizeof(ret_val
));
730 PSI_ContextCall(&PSI_G(context
), &ret_val
, impl
->decl
);
732 psi_do_return(return_value
, impl
->stmts
->ret
.list
[0], &ret_val
);
734 for (i
= 0; i
< impl
->stmts
->set
.count
; ++i
) {
735 set_stmt
*set
= impl
->stmts
->set
.list
[i
];
738 psi_do_set(set
->arg
->_zv
, set
->val
);
742 for (i
= 0; i
< impl
->stmts
->fre
.count
; ++i
) {
743 free_stmt
*fre
= impl
->stmts
->fre
.list
[i
];
751 PHP_MINIT_FUNCTION(psi
)
753 PSI_ContextOps
*ops
= NULL
;
755 REGISTER_INI_ENTRIES();
758 if (!strcasecmp(PSI_G(engine
), "jit")) {
767 php_error(E_WARNING
, "No PSI engine found");
771 PSI_ContextInit(&PSI_G(context
), ops
, psi_error
);
772 PSI_ContextBuild(&PSI_G(context
), PSI_G(directory
));
774 if (getenv("PSI_DUMP")) {
775 PSI_ContextDump(&PSI_G(context
), STDOUT_FILENO
);
780 PHP_MSHUTDOWN_FUNCTION(psi
)
782 PSI_ContextDtor(&PSI_G(context
));
784 UNREGISTER_INI_ENTRIES();
789 /* Remove if there's nothing to do at request start */
790 /* {{{ PHP_RINIT_FUNCTION
792 PHP_RINIT_FUNCTION(psi
)
794 #if defined(COMPILE_DL_PSI) && defined(ZTS)
795 ZEND_TSRMLS_CACHE_UPDATE();
801 /* Remove if there's nothing to do at request end */
802 /* {{{ PHP_RSHUTDOWN_FUNCTION
804 PHP_RSHUTDOWN_FUNCTION(psi
)
810 PHP_MINFO_FUNCTION(psi
)
812 php_info_print_table_start();
813 php_info_print_table_header(2, "psi support", "enabled");
814 php_info_print_table_end();
816 DISPLAY_INI_ENTRIES();
818 const zend_function_entry psi_functions
[] = {
822 zend_module_entry psi_module_entry
= {
823 STANDARD_MODULE_HEADER
,
828 PHP_RINIT(psi
), /* Replace with NULL if there's nothing to do at request start */
829 PHP_RSHUTDOWN(psi
), /* Replace with NULL if there's nothing to do at request end */
832 STANDARD_MODULE_PROPERTIES
835 #ifdef COMPILE_DL_PSI
837 ZEND_TSRMLS_CACHE_DEFINE();
847 * vim600: noet sw=4 ts=4 fdm=marker
848 * vim<600: noet sw=4 ts=4